com.easycodebox.jdbc.support.DefaultJdbcPreHandler Maven / Gradle / Ivy

Go to download
package com.easycodebox.jdbc.support;

import java.beans.PropertyDescriptor;
import java.lang.reflect.Method;
import java.lang.reflect.Modifier;
import java.util.Date;

import org.apache.commons.beanutils.PropertyUtils;

import com.easycodebox.common.log.slf4j.Logger;
import com.easycodebox.common.log.slf4j.LoggerFactory;
import com.easycodebox.common.security.SecurityUtils;
import com.easycodebox.jdbc.Property;
import com.easycodebox.jdbc.entity.CreateEntity;
import com.easycodebox.jdbc.entity.Entity;
import com.easycodebox.jdbc.entity.ModifyEntity;
import com.easycodebox.jdbc.grammar.SqlGrammar;

/**
 * @author WangXiaoJin
 *
 */
public class DefaultJdbcPreHandler implements JdbcPreHandler {
	
	private static final Logger LOG = LoggerFactory.getLogger(DefaultJdbcPreHandler.class);
	
	public static final Object SYS_USER_ID = "0";
	public static final String SYS_USERNAME = "系统";
	
	public static final String CREATOR_FIELD_NAME = "creator";
	public static final String CREATE_TIME_FIELD_NAME = "createTime";
	public static final String MODIFIER_FIELD_NAME = "modifier";
	public static final String MODIFY_TIME_FIELD_NAME = "modifyTime";
	
	private Object sysUserId = SYS_USER_ID;
	private String creatorFieldName = CREATOR_FIELD_NAME;
	private String createTimeFieldName = CREATE_TIME_FIELD_NAME;
	private String modifierFieldName = MODIFIER_FIELD_NAME;
	private String modifyTimeFieldName = MODIFY_TIME_FIELD_NAME;
	
	private SecurityUser securityUser = new SecurityUser() {

		@Override
		public Object getUserId() {
			return SecurityUtils.getUserId();
		}

	};
	
	private DateFactory dateFactory = new DateFactory() {

		@Override
		public Object instance() {
			return new Date();
		}
		
	};
	
	@Override
	public void beforeSave(Entity entity) {
		Object userId = securityUser.getUserId() == null ? sysUserId : securityUser.getUserId();
		if(entity instanceof CreateEntity) {
			CreateEntity e = (CreateEntity) entity;
			if(e.getCreator() == null)
				e.setCreator((String)userId);
			if(e.getCreateTime() == null)
				e.setCreateTime((Date)dateFactory.instance());
		}else {
			setBeanProperty(entity, creatorFieldName, userId);
			setBeanProperty(entity, createTimeFieldName, dateFactory.instance());
		}
		beforeUpdate(entity);
	}

	@Override
	public void beforeUpdate(Entity entity) {
		Object userId = securityUser.getUserId() == null ? sysUserId : securityUser.getUserId();
		if(entity instanceof ModifyEntity) {
			ModifyEntity e = (ModifyEntity) entity;
			if(e.getModifier() == null)
				e.setModifier((String)userId);
			if(e.getModifyTime() == null)
				e.setModifyTime((Date)dateFactory.instance());
		}else {
			setBeanProperty(entity, modifierFieldName, userId);
			setBeanProperty(entity, modifyTimeFieldName, dateFactory.instance());
		}
	}
	
	@Override
	public void beforeUpdate(SqlGrammar sqlGrammar) {
		Object userId = securityUser.getUserId() == null ? sysUserId : securityUser.getUserId();
		boolean isModifyEntity = ModifyEntity.class.isAssignableFrom(sqlGrammar.getEntity());
		if(!updateSqlHad(sqlGrammar, modifierFieldName)
				&& (
						isModifyEntity
						|| existProperty(sqlGrammar.getEntity(), modifierFieldName)
						)) {
			sqlGrammar.update(Property.instance(modifierFieldName, sqlGrammar.getEntity(), false), userId);
		}
		if(!updateSqlHad(sqlGrammar, modifyTimeFieldName)
				&& (
						isModifyEntity
						|| existProperty(sqlGrammar.getEntity(), modifyTimeFieldName)
						)) {
			sqlGrammar.update(Property.instance(modifyTimeFieldName, sqlGrammar.getEntity(), false), dateFactory.instance());
		}
	}
	
	/**
	 * 验证是否存在指定的property
	 * @param clazz
	 * @param name
	 * @return
	 */
	private boolean existProperty(Class clazz, String name) {
		PropertyDescriptor[] pds = PropertyUtils.getPropertyDescriptors(clazz);
		for (PropertyDescriptor pd : pds) {
			if (name.equals(pd.getName())) {
				return true;
			}
		}
		return false;
	}
	
	/**
	 * 设置指定属性的值，该属性值为null设值，不为null则不设值
	 */
	private void setBeanProperty(Object obj, String name, Object val) {
		PropertyDescriptor prop = null;
		PropertyDescriptor[] pds = PropertyUtils.getPropertyDescriptors(obj);
		for (PropertyDescriptor pd : pds) {
			if (name.equals(pd.getName())) {
				prop = pd;
				break;
			}
		}
		if(prop != null) {
			try {
				Method read = prop.getReadMethod();
				if (!Modifier.isPublic(read.getModifiers())) {
					read.setAccessible(true);
				}
				if (read.invoke(obj) == null) {
					Method write = prop.getWriteMethod();
					if (!Modifier.isPublic(write.getModifiers())) {
						write.setAccessible(true);
					}
					//设置值
					write.invoke(obj, val);
				}
			} catch (Exception e) {
				LOG.warn("read or write object({0}) property({1}) error.", e, obj, name);
			}
		}
	}
	
	private boolean updateSqlHad(SqlGrammar sqlGrammar, String propertyName) {
		String updSql = sqlGrammar.getUpdateSql().toString();
		if(updSql.matches("(\\s+|,|\\.)" + propertyName + "(\\s+|,)")) {
			return true;
		}
		return false;
	}
	
	/**
	 * 获取用户信息接口
	 * @author WangXiaoJin
	 *
	 */
	public static interface SecurityUser {
		
		/**
		 * 返回当前操作人id
		 * @return
		 */
		Object getUserId();
		
	}
	
	/**
	 * 创建日期类型的工厂
	 * @author WangXiaoJin
	 *
	 */
	public static interface DateFactory {
		
		Object instance();
		
	}
	
	public Object getSysUserId() {
		return sysUserId;
	}

	public void setSysUserId(Object sysUserId) {
		this.sysUserId = sysUserId;
	}

	public String getCreatorFieldName() {
		return creatorFieldName;
	}

	public void setCreatorFieldName(String creatorFieldName) {
		this.creatorFieldName = creatorFieldName;
	}

	public String getCreateTimeFieldName() {
		return createTimeFieldName;
	}

	public void setCreateTimeFieldName(String createTimeFieldName) {
		this.createTimeFieldName = createTimeFieldName;
	}

	public String getModifierFieldName() {
		return modifierFieldName;
	}

	public void setModifierFieldName(String modifierFieldName) {
		this.modifierFieldName = modifierFieldName;
	}

	public String getModifyTimeFieldName() {
		return modifyTimeFieldName;
	}

	public void setModifyTimeFieldName(String modifyTimeFieldName) {
		this.modifyTimeFieldName = modifyTimeFieldName;
	}

	public SecurityUser getSecurityUser() {
		return securityUser;
	}

	public void setSecurityUser(SecurityUser securityUser) {
		this.securityUser = securityUser;
	}

	public DateFactory getDateFactory() {
		return dateFactory;
	}

	public void setDateFactory(DateFactory dateFactory) {
		this.dateFactory = dateFactory;
	}
	
}