cn.sylinx.hbatis.plugin.optlock.OptimisticLockWrapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hbatis-core Show documentation
hbatis is a simple orm framework
The newest version!
package cn.sylinx.hbatis.plugin.optlock;

import java.lang.reflect.Field;
import java.math.BigDecimal;
import java.math.BigInteger;

import cn.sylinx.hbatis.db.mapper.ModelBuilder;
import cn.sylinx.hbatis.db.mapper.anno.Version;
import cn.sylinx.hbatis.log.GLog;

public enum OptimisticLockWrapper {

	ME;

	private boolean enable = false;

	public boolean isEnable() {
		return enable;
	}

	void setEnable(boolean enable) {
		this.enable = enable;
	}

	public static  OptimisticLockInsert parseOptimisticLockInsert(T t) {

		if (!ME.enable) {
			// 乐观锁不可用
			return null;
		}

		if (t == null) {
			return null;
		}

		Field f = ModelBuilder.getVersionField(t.getClass());
		if (f == null) {
			// 没有乐观锁字段
			return null;
		}

		Number initVersionValue = getInitVersionValue(f.getType());
		if (initVersionValue == null) {
			return null;
		}

		OptimisticLockInsert oli = new OptimisticLockInsert();
		oli.setVersionField(f);
		oli.setVersionFieldAttr(f.getName());
		oli.setInitVersionValue(initVersionValue);

		return oli;
	}

	public static  OptimisticLockUpdate parseOptimisticLockUpdate(T t) {

		if (!ME.enable) {
			// 乐观锁不可用
			return null;
		}

		if (t == null) {
			return null;
		}

		Field f = ModelBuilder.getVersionField(t.getClass());
		if (f == null) {
			// 没有乐观锁字段
			return null;
		}

		Number versionValue = null;

		try {
			f.setAccessible(true);
			versionValue = (Number) f.get(t);
		} catch (Exception e) {
			GLog.error("getVersionField error", e);
		}

		if (versionValue == null) {
			return null;
		}

		Number newVersionValue = getNewVersionValue(versionValue);
		if (newVersionValue == null) {
			return null;
		}

		OptimisticLockUpdate olu = new OptimisticLockUpdate();
		olu.setNewVersionValue(newVersionValue);
		olu.setOldVersionValue(versionValue);
		olu.setVersionField(f);
		olu.setVersionFieldAttr(f.getName());

		return olu;
	}

	private static Number getNewVersionValue(Number oldVersionValue) {

		// 判断字段类型
		Class clz = oldVersionValue.getClass();
		if (clz == Version.VERSION_TYPE_INTEGER || clz == Version.VERSION_TYPE_INTEGER_PRIMARY) {
			return new Integer(((Integer) oldVersionValue).intValue() + 1);
		}

		if (clz == Version.VERSION_TYPE_LONG || clz == Version.VERSION_TYPE_LONG_PRIMARY) {
			return new Long(((Long) oldVersionValue).longValue() + 1);
		}

		if (clz == Version.VERSION_TYPE_BIGINTEGER) {
			return ((BigInteger) oldVersionValue).add(new BigInteger("1"));
		}

		if (clz == Version.VERSION_TYPE_BIGDECIMAL) {
			return ((BigDecimal) oldVersionValue).add(new BigDecimal("1"));
		}

		return null;
	}

	private static Number getInitVersionValue(Class clz) {

		// 判断字段类型
		if (clz == Version.VERSION_TYPE_INTEGER || clz == Version.VERSION_TYPE_INTEGER_PRIMARY) {
			return new Integer(1);
		}

		if (clz == Version.VERSION_TYPE_LONG || clz == Version.VERSION_TYPE_LONG_PRIMARY) {
			return new Long(1);
		}

		if (clz == Version.VERSION_TYPE_BIGINTEGER) {
			return new BigInteger("1");
		}

		if (clz == Version.VERSION_TYPE_BIGDECIMAL) {
			return new BigDecimal("1");
		}

		return null;
	}

}