org.onetwo.dbm.mapping.IdGeneratorFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of onetwo-dbm Show documentation
onetwo-dbm
The newest version!
package org.onetwo.dbm.mapping;

import java.io.Serializable;
import java.util.HashMap;
import java.util.Optional;

import javax.persistence.SequenceGenerator;
import javax.persistence.TableGenerator;

import org.apache.commons.lang3.StringUtils;
import org.onetwo.common.annotation.AnnotationInfo;
import org.onetwo.common.jackson.JsonMapper;
import org.onetwo.common.reflect.ReflectUtils;
import org.onetwo.common.spring.SpringUtils;
import org.onetwo.common.spring.Springs;
import org.onetwo.dbm.annotation.DbmIdGenerator;
import org.onetwo.dbm.annotation.DbmIdGenerator.GeneratorCreator;
import org.onetwo.dbm.id.CustomIdGenerator;
import org.onetwo.dbm.id.CustomerIdGeneratorAdapter;
import org.onetwo.dbm.id.IdentifierGenerator;
import org.onetwo.dbm.id.SequenceGeneratorAttrs;
import org.onetwo.dbm.id.SequenceIdGenerator;
import org.onetwo.dbm.id.SnowflakeGenerator;
import org.onetwo.dbm.id.SnowflakeIdGenerator;
import org.onetwo.dbm.id.TableGeneratorAttrs;
import org.onetwo.dbm.id.TableIdGenerator;

/**
 * @author wayshall
 * 

 */
public class IdGeneratorFactory {
//	private static final Logger logger = JFishLoggerFactory.getLogger(IdGeneratorFactory.class);
//	private static final SnowflakeGenerator DefaultSnowflakeGenerator = new SnowflakeGenerator();
	
	public static Optional> createSequenceGenerator(AnnotationInfo annotationInfo){
		SequenceGenerator sg = annotationInfo.getAnnotation(SequenceGenerator.class);
		if(sg==null){
			return Optional.empty();
		}
		SequenceGeneratorAttrs sgAttrs = new SequenceGeneratorAttrs(sg.name(), sg.sequenceName(), sg.initialValue(), sg.allocationSize());
		SequenceIdGenerator generator = new SequenceIdGenerator(sgAttrs);
		return Optional.of(generator);
	}
	
	public static Optional> createTableGenerator(AnnotationInfo annotationInfo){
		TableGenerator tg = annotationInfo.getAnnotation(TableGenerator.class);
		if(tg==null){
			return Optional.empty();
		}
		TableGeneratorAttrs tgAttrs = new TableGeneratorAttrs(tg.name(), 
																tg.allocationSize(), 
																tg.table(), 
																tg.pkColumnName(), 
																tg.valueColumnName(), 
																tg.pkColumnValue(),
																tg.initialValue());
		IdentifierGenerator generator = createTableGenerator(tgAttrs);
		return Optional.of(generator);
	}
	
	/***
	 * 此方法使用下面的默认表:
	 * CREATE TABLE `gen_ids` (
		`gen_name`  varchar(255) NOT NULL ,
		`gen_value`  bigint NOT NULL ,
		PRIMARY KEY (`gen_name`)
		);
	 * @author weishao zeng
	 * @param genenratorName
	 * @return
	 */
	public static IdentifierGenerator createTableGenerator(String genenratorName){
		String tableName = "gen_ids";
		int allocationSize = 1;
		int initialValue = 1;
		TableGeneratorAttrs tgAttrs = new TableGeneratorAttrs(genenratorName, 
				allocationSize, 
				tableName, 
				"gen_name", 
				"gen_value", 
				genenratorName,
				initialValue);
		return createTableGenerator(tgAttrs);
	}
	
	public static IdentifierGenerator createTableGenerator(TableGeneratorAttrs tgAttrs){
		if (tgAttrs==null) {
			throw new IllegalArgumentException("TableGeneratorAttrs can not be null!");
		}
		TableIdGenerator generator = new TableIdGenerator(tgAttrs);
		return generator;
	}
	
	@SuppressWarnings("unchecked")
	public static Optional> createDbmIdGenerator(AnnotationInfo annotationInfo){
		DbmIdGenerator dg = annotationInfo.getAnnotation(DbmIdGenerator.class);
		if (dg==null) {
			return Optional.empty();
		}
		GeneratorCreator creator = dg.creator();
		if (creator==GeneratorCreator.SPRING) {
			CustomIdGenerator customIdGenerator = Springs.getInstance().getBean(dg.generatorClass());
			IdentifierGenerator idGenerator = new CustomerIdGeneratorAdapter<>(dg.name(), customIdGenerator, dg.valueType());
			return Optional.of(idGenerator);
		}
		
		String attribute = dg.attributes();
		CustomIdGenerator customIdGenerator = null;
		if (dg.generatorClass()==SnowflakeGenerator.class) {
			SnowflakeIdGenerator idg = Springs.getInstance().getBean(SnowflakeIdGenerator.class);
			customIdGenerator = new SnowflakeGenerator(idg);
		} else {
			customIdGenerator = ReflectUtils.newInstance(dg.generatorClass());
		}
		
		if(StringUtils.isNotBlank(attribute)){
			HashMap jsonMap = JsonMapper.IGNORE_EMPTY.fromJson(attribute, HashMap.class);
			SpringUtils.getMapToBean().injectBeanProperties(jsonMap, customIdGenerator);
		}
		customIdGenerator.initGenerator();
		IdentifierGenerator idGenerator = new CustomerIdGeneratorAdapter<>(dg.name(), customIdGenerator, dg.valueType());
		return Optional.of(idGenerator);
	}

}