org.onetwo.dbm.id.TableIdGenerator 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.id;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.onetwo.common.expr.Expression;
import org.onetwo.common.expr.ExpressionFacotry;
import org.onetwo.common.reflect.ReflectUtils;
import org.onetwo.dbm.core.spi.DbmSessionImplementor;
import org.onetwo.dbm.core.spi.DbmTransaction;
import org.onetwo.dbm.exception.DbmException;
import org.springframework.dao.EmptyResultDataAccessException;
import org.springframework.jdbc.BadSqlGrammarException;
import org.springframework.transaction.TransactionDefinition;
import org.springframework.transaction.support.DefaultTransactionDefinition;

/**
 * @author wayshall
 * 

 */
public class TableIdGenerator extends AbstractIdentifierGenerator {
	
	/****
	 * CREATE TABLE `gen_ids` (
		`gen_name`  varchar(255) NOT NULL ,
		`gen_value`  bigint NOT NULL ,
		PRIMARY KEY (`gen_name`)
		)
		;
	 */
	private static final String TEMPLATE_CREATE_TABLE_SQL = " CREATE TABLE `${table}` ("
			+ "`${pkColumnName}`  varchar(255) NOT NULL ,"
			+ "`${valueColumnName}`  bigint NOT NULL ,"
			+ "PRIMARY KEY (`${pkColumnName}`)"
			+ ");";
	
	private static final String TEMPLATE_SELECT_SQL = "select ${valueColumnName} from ${table} where ${pkColumnName} = ? for update";
	private static final String TEMPLATE_INSERT_SQL = "insert into ${table} ( ${pkColumnName}, ${valueColumnName} ) " + " values ( ?, ? )";
	private static final String TEMPLATE_UPDATE_SQL = "update ${table} set ${valueColumnName}=?  where ${pkColumnName}=? and ${valueColumnName}=?";
	

	final private Expression parser = ExpressionFacotry.DOLOR;

	final private String createTableSql;
	final private String selectSql;
	final private String insertSql;
	final private String updateSql;
	final private TableGeneratorAttrs attrs;
	/***
	 * 默认使用 PROPAGATION_REQUIRES_NEW 
	 */
	private int transactionPropagation = TransactionDefinition.PROPAGATION_REQUIRES_NEW;
	
	public TableIdGenerator(TableGeneratorAttrs attrs) {
		super(attrs.getName());
		this.attrs = attrs;
		Map provider = ReflectUtils.toMap(attrs);
		this.createTableSql = parser.parseByProvider(TEMPLATE_CREATE_TABLE_SQL, provider);
		this.selectSql = parser.parseByProvider(TEMPLATE_SELECT_SQL, provider);
		this.insertSql = parser.parseByProvider(TEMPLATE_INSERT_SQL, provider);
		this.updateSql = parser.parseByProvider(TEMPLATE_UPDATE_SQL, provider);
		this.transactionPropagation = attrs.getTransactionPropagation();
	}


	@Override
	protected int getAllocationSize() {
		return attrs.getAllocationSize();
	}

	@Override
	public StrategyType getStrategyType() {
		return StrategyType.TABLE;
	}


	/****
	 */
	@Override
	public List batchGenerate(DbmSessionImplementor contextSession, int batchSize) {
//		System.out.println("==========> batchGenerate");
		DbmSessionImplementor session = (DbmSessionImplementor)contextSession.getSessionFactory().openSession();
		DefaultTransactionDefinition definition = new DefaultTransactionDefinition();
		definition.setPropagationBehavior(transactionPropagation);
		DbmTransaction transaction = session.beginTransaction(definition);
		try {
			Long currentId = null;
			try {
				currentId = session.getDbmJdbcOperations().queryForObject(selectSql, Long.class, attrs.getPkColumnValue());
			} catch (BadSqlGrammarException e) {
				//MySQLSyntaxErrorException SQLState=42S02 vendorCode=1146
//				throw e;
				this.createSeqTable(session);
			}catch (EmptyResultDataAccessException e) {
				//no result
			}
			if(currentId==null){
				int res = session.getDbmJdbcOperations().update(insertSql, attrs.getPkColumnValue(), attrs.getInitialValue());
				if(res!=1){
					throw new DbmException("error insert table sequeue : "+attrs.getPkColumnValue());
				}
				currentId = Long.valueOf(attrs.getInitialValue());
			}
			Long maxId = currentId + attrs.getAllocationSize();
			int res = session.getDbmJdbcOperations().update(updateSql, maxId, attrs.getPkColumnValue(), currentId);
			if(res!=1){
				throw new DbmException("error update table sequeue : "+attrs.getPkColumnValue()+", updateSql:"+updateSql);
			}
			transaction.commit();
			List ids = new ArrayList(batchSize);
			long nextId = currentId;
			for (; nextId < maxId; nextId++) {
				ids.add(nextId);
			}
			if(ids.size()!=batchSize){
				throw new DbmException("generate batch id count error. actual:"+ids.size()+", expect:"+batchSize);
			}
			return ids;
		} catch (DbmException e) {
			transaction.rollback();
			throw e;
		} catch (Exception e) {
			transaction.rollback();
			throw new DbmException("generate id error", e);
		} 
	}
	
	private void createSeqTable(DbmSessionImplementor session){
		try {
			session.executeUpdate(createTableSql);
		} catch (Exception e) {
			throw new DbmException("create table error , create table sql:"+createTableSql);
		}
	}
	
}