org.onetwo.dbm.id.SequenceIdGenerator 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.sql.SQLException;
import java.util.List;

import org.onetwo.common.convert.Types;
import org.onetwo.common.db.sql.SequenceNameManager;
import org.onetwo.common.profiling.TimeCounter;
import org.onetwo.common.reflect.ReflectUtils;
import org.onetwo.common.utils.Assert;
import org.onetwo.dbm.core.spi.DbmSessionImplementor;
import org.springframework.jdbc.BadSqlGrammarException;

/**
 * @author wayshall
 * 

 */
public class SequenceIdGenerator extends AbstractIdentifierGenerator {
	
	final private SequenceGeneratorAttrs attrs;
//	private Queue seqQueue;
	
	public SequenceIdGenerator(SequenceGeneratorAttrs attrs) {
		super(attrs.getName());
		this.attrs = attrs;
//		this.seqQueue = new LinkedList();
	}
	
	
	@Override
	protected int getAllocationSize() {
		return attrs.getAllocationSize();
	}


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


//	private Long generateOneSeq(DbmSessionImplementor session) {
//		SequenceNameManager sequenceNameManager = session.getSequenceNameManager();
//		String seqSql = sequenceNameManager.getSequenceSql(attrs.getSequenceName(), null);
//		Long id = null;
//		try {
//			id = session.getDbmJdbcOperations().queryForObject(seqSql, Long.class);
//		} catch (BadSqlGrammarException e) {
//			if(createSeqIfNecessary(e, session)){
//				id = session.getDbmJdbcOperations().queryForObject(seqSql, Long.class);
//				if(id==null)
//					throw e;
//			}
//		}
//		return id;
//	}
	
	private boolean createSeqIfNecessary(BadSqlGrammarException e, DbmSessionImplementor session){
		SequenceNameManager sequenceNameManager = session.getSequenceNameManager();
		//ORA-02289: 序列不存在
		SQLException sqe = e.getSQLException();
		int vendorCode = Types.convertValue(ReflectUtils.getFieldValue(sqe, "vendorCode"), int.class);
		if(vendorCode==2289){
			session.getDbmJdbcOperations().execute(sequenceNameManager.getCreateSequence(attrs.getSequenceName(), attrs.getInitialValue()));
			return true;
		}
		return false;
	}

	@Override
	public List batchGenerate(DbmSessionImplementor session, int batchSize) {
		Assert.notNull(session);
		Assert.isTrue(batchSize>0);
		
		SequenceNameManager sequenceNameManager = session.getSequenceNameManager();
		TimeCounter counter = new TimeCounter("select batch seq...");
		counter.start();
		String seqSql = sequenceNameManager.getSequenceSql(attrs.getSequenceName(), batchSize);
		List seqs = null;
		try {
			seqs = session.getDbmJdbcOperations().queryForList(seqSql, Long.class, batchSize);
			Assert.isTrue(batchSize==seqs.size(), "the size of seq is not equals to data, seq size:"+seqs.size()+", data size:"+batchSize);
		} catch (BadSqlGrammarException e) {
			if(createSeqIfNecessary(e, session)){
				seqs = session.getDbmJdbcOperations().queryForList(seqSql, Long.class, batchSize);
				if(seqs==null)
					throw e;
				Assert.isTrue(batchSize==seqs.size(), "the size of seq is not equals to data, seq size:"+seqs.size()+", data size:"+batchSize);
			}
		}
		counter.stop();
		return seqs;
	}

	

}