com.bixuebihui.r2dbc.sequence.KeyInfo Maven / Gradle / Ivy

Go to download
package com.bixuebihui.r2dbc.sequence;

import com.bixuebihui.r2dbc.sql.IDbHelper;
import io.r2dbc.spi.Connection;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import reactor.core.publisher.Mono;

import java.util.concurrent.atomic.AtomicBoolean;
import java.util.concurrent.atomic.AtomicLong;


/**
 * Created by IntelliJ IDEA.

 * User: leizhimin

 * Date: 2008-4-2 15:24:52

 * Note: Sequence载体 DROP TABLE IF EXISTS t_sequence; CREATE TABLE
 * t_sequence( KEYNAME varchar(24) NOT NULL ,--COMMENT 'Sequence名称', KEYVALUE
 * numeric(20) DEFAULT 10000,-- COMMENT 'Sequence最大值', PRIMARY KEY (KEYNAME) ) ;
 *
 * @author xingwx
 * @version $Id: $Id
 */
public class KeyInfo {
    /**
     * Constant SEQUENCE_TABLE="t_sequence"
     */
    public static final String SEQUENCE_TABLE = "t_sequence";
    /** Constant SEQUENCE_DBHELPER_NAME="sequenceDbHelper" */
    public static final String SEQUENCE_DBHELPER_NAME = "sequenceDbHelper";
    /** 当前Sequence载体的最大值 */
	private long maxKey;
    /** 当前Sequence载体的最小值 */
	private long minKey;
    /** 下一个Sequence值 */
	private long nextKey;
    /** Sequence值缓存大小 */
	private int poolSize;
    /** Sequence的名称 */
	private String keyName;
	private static final String SQL_UPDATE = "UPDATE " + SEQUENCE_TABLE
			+ " SET KEYVALUE = KEYVALUE + ? WHERE KEYNAME = ? and KEYVALUE=?";
	private static final String MOVE_TO_MAX = "UPDATE " + SEQUENCE_TABLE
			+ " SET KEYVALUE =  ? WHERE KEYNAME = ? ";
	private static final String SQL_QUERY = "SELECT KEYVALUE FROM "
			+ SEQUENCE_TABLE + " WHERE KEYNAME = ?";
	static final String INSTALL =
			"CREATE TABLE  IF NOT EXISTS  t_sequence ( KEYNAME varchar(24) NOT NULL PRIMARY KEY ,"+
							" KEYVALUE INT default 10000"+
								"  );";

	private IDbHelper dbHelper;

	protected static Logger LOG = LoggerFactory.getLogger(KeyInfo.class);

    private void init(String keyName, int poolSize, IDbHelper dbHelper) {
        this.dbHelper = dbHelper;

        this.poolSize = poolSize;
        this.keyName = keyName;

        retrieveFromDb();
    }

    /**
     * Constructor for KeyInfo.
     *
     * @param keyName a {@link String} object.
     * @param poolSize a int.
     * @param dbHelper a {@link IDbHelper} object.
     */
    public KeyInfo(String keyName, int poolSize, IDbHelper dbHelper)  {
        init(keyName, poolSize, dbHelper);
    }


    /**
     * Getter for the field keyName.
     *
     * @return a {@link String} object.
     */
    public String getKeyName() {
        return keyName;
    }

    /**
     * Getter for the field maxKey.
     *
     * @return a long.
     */
    public long getMaxKey() {
        return maxKey;
    }

    /**
     * Getter for the field minKey.
     *
     * @return a long.
     */
    public long getMinKey() {
        return minKey;
    }

    /**
     * Getter for the field poolSize.
     *
     * @return a int.
     */
    public int getPoolSize() {
        return poolSize;
    }

	/**
	 * 获取下一个Sequence值
     *
	 * @return 下一个Sequence值
     */
	public Mono getNextKey()  {
		if (nextKey > maxKey) {
			 return  retrieveFromDb();
		}
		return Mono.just(nextKey++);
	}

    Mono