♻️ Refactoring code. 发号器优化适配Oracle 、PG数据库

2022-01-19 15:35:11 +08:00
1 parent b533e6441b
commit a06c9bc69a
@@ -1,6 +1,7 @@
 package com.pig4cloud.pigx.common.sequence.range.impl.db;

 import cn.hutool.core.util.RandomUtil;
+import cn.hutool.core.util.StrUtil;
 import cn.hutool.extra.spring.SpringUtil;
 import com.pig4cloud.pigx.common.sequence.exception.SeqException;
 import com.pig4cloud.pigx.common.sequence.range.impl.db.provider.SqlProviderFactory;
@@ -58,7 +59,7 @@ abstract class BaseDbHelper {
 		try {
 			conn = dataSource.getConnection();
 			stmt = conn.prepareStatement(SQL_PROVIDER_FACTORY.getInsertRangeSql().replace("#tableName", tableName));
-			stmt.setLong(1, RandomUtil.randomLong());
+			stmt.setLong(1, RandomUtil.randomLong(12));
 			stmt.setString(2, name);
 			stmt.setLong(3, stepStart);
 			stmt.setTimestamp(4, new Timestamp(System.currentTimeMillis()));
@@ -169,4 +170,40 @@ abstract class BaseDbHelper {
 		}
 	}

+	public static Boolean existTable(DataSource dataSource, String tableName) {
+
+		String existTableSql = SQL_PROVIDER_FACTORY.getExistTableSql();
+		if (StrUtil.isBlank(existTableSql)) {
+			return true;
+		}
+		Connection conn = null;
+		Statement stmt = null;
+
+		try {
+			conn = dataSource.getConnection();
+			stmt = conn.createStatement();
+			ResultSet resultSet = stmt
+					.executeQuery(SQL_PROVIDER_FACTORY.getExistTableSql().replace("#tableName", tableName));
+
+			if (!resultSet.next()) {
+				return false;
+			}
+
+			int count = resultSet.getInt(1);
+
+			if (0 == count) {
+				return true;
+			}
+		}
+		catch (SQLException e) {
+			throw new SeqException(e);
+		}
+		finally {
+			closeQuietly(stmt);
+			closeQuietly(conn);
+		}
+
+		return false;
+	}
+
 }
@@ -69,7 +69,11 @@ public class DbSeqRangeMgr implements SeqRangeMgr {
 	@Override
 	public void init() {
 		checkParam();
-		BaseDbHelper.createTable(getDataSource(), getRealTableName());
+
+		// 判断是否需要创建表
+		if (BaseDbHelper.existTable(getDataSource(), getRealTableName())) {
+			BaseDbHelper.createTable(getDataSource(), getRealTableName());
+		}
 	}

 	private boolean isEmpty(String str) {
@@ -13,12 +13,14 @@ import org.springframework.stereotype.Component;
 public class MysqlSqlProvider implements SqlProvider {

 	/**
-	 * 获取插入范围语句
+	 * 获取建表语句
 	 * @return
 	 */
 	@Override
-	public String getInsertRangeSql() {
-		return "INSERT IGNORE INTO #tableName(id,name,value,gmt_create,gmt_modified)" + " VALUE(?,?,?,?,?)";
+	public String getCreateTableSql() {
+		return "CREATE TABLE IF NOT EXISTS #tableName(" + "id bigint(20) NOT NULL AUTO_INCREMENT,"
+				+ "value bigint(20) NOT NULL," + "name varchar(32) NOT NULL," + "gmt_create DATETIME NOT NULL,"
+				+ "gmt_modified DATETIME NOT NULL," + "PRIMARY KEY (`id`),UNIQUE uk_name (`name`)" + ")";
 	}

 	@Override
@@ -12,9 +12,40 @@ import org.springframework.stereotype.Component;
@Component
 public class OracleSqlProvider implements SqlProvider {

+	/**
+	 * 获取表是否存在
+	 * @return
+	 */
 	@Override
-	public String getInsertRangeSql() {
-		return "INSERT INTO #tableName(id,name,value,gmt_create,gmt_modified)" + " VALUE(?,?,?,?,?)";
+	public String getExistTableSql() {
+		return "select count(*) from user_tables where table_name =upper('#tableName')";
+	}
+
+	/**
+	 * 获取建表语句
+	 * @return SQL
+	 */
+	@Override
+	public String getCreateTableSql() {
+		return "CREATE TABLE #tableName " + "(id NUMBER (20,0) VISIBLE NOT NULL,value NUMBER (20,0) VISIBLE NOT NULL"
+				+ ",name VARCHAR2 (96 BYTE) VISIBLE,gmt_create DATE VISIBLE NOT NULL"
+				+ ",gmt_modified DATE VISIBLE NOT NULL)";
+	}
+
+	/**
+	 * 获取更新范围语句
+	 * @return
+	 */
+	public String getUpdateRangeSql() {
+		return "UPDATE #tableName SET value=?,gmt_modified=? WHERE name=? AND value=?";
+	}
+
+	/**
+	 * 获取查询范围语句
+	 * @return
+	 */
+	public String getSelectRangeSql() {
+		return "SELECT value FROM #tableName WHERE name=?";
 	}

 	@Override
@@ -12,9 +12,15 @@ import org.springframework.stereotype.Component;
@Component
 public class PostgreSqlProvider implements SqlProvider {

+	/**
+	 * 获取建表语句
+	 * @return
+	 */
 	@Override
-	public String getInsertRangeSql() {
-		return "INSERT INTO #tableName(id,name,value,gmt_create,gmt_modified)" + " VALUE(?,?,?,?,?)";
+	public String getCreateTableSql() {
+		return "CREATE TABLE IF NOT EXISTS #tableName (ID int8 PRIMARY KEY NOT NULL"
+				+ ",VALUE int8 NOT NULL,NAME VARCHAR (266) NOT NULL,gmt_create TIMESTAMP (6) NOT NULL"
+				+ ",gmt_modified TIMESTAMP (6) NOT NULL)";
 	}

 	@Override
@@ -10,28 +10,34 @@ import com.baomidou.mybatisplus.annotation.DbType;
 */
 public interface SqlProvider {

+	/**
+	 * 获取表是否存在
+	 * @return
+	 */
+	default String getExistTableSql() {
+		return null;
+	}
+
 	/**
 	 * 获取建表语句
 	 * @return
 	 */
-	default String getCreateTableSql() {
-		return "CREATE TABLE IF NOT EXISTS #tableName(" + "id bigint(20) NOT NULL AUTO_INCREMENT,"
-				+ "value bigint(20) NOT NULL," + "name varchar(32) NOT NULL," + "gmt_create DATETIME NOT NULL,"
-				+ "gmt_modified DATETIME NOT NULL," + "PRIMARY KEY (`id`),UNIQUE uk_name (`name`)" + ")";
-	}
+	String getCreateTableSql();

 	/**
 	 * 获取插入范围语句
 	 * @return
 	 */
-	String getInsertRangeSql();
+	default String getInsertRangeSql() {
+		return "INSERT INTO #tableName (id,name,value,gmt_create,gmt_modified)" + " VALUES(?,?,?,?,?)";
+	}

 	/**
 	 * 获取更新范围语句
 	 * @return
 	 */
 	default String getUpdateRangeSql() {
-		return "UPDATE #tableName SET value=?,gmt_modified=? WHERE name=? AND " + "value=?";
+		return "UPDATE #tableName SET value=?,gmt_modified=? WHERE name=? AND value=?";
 	}

 	/**
@@ -71,4 +71,13 @@ public class SqlProviderFactory {
 		return sqlProvider.getUpdateRangeSql();
 	}

+	/**
+	 * 获取是否创建表语句
+	 * @return SQL语句
+	 */
+	public String getExistTableSql() {
+		SqlProvider sqlProvider = getSqlProvider();
+		return sqlProvider.getExistTableSql();
+	}
+
 }
@@ -1,25 +0,0 @@
-package com.pig4cloud.pigx.common.sequence.range.impl.db.provider;
-
-import com.baomidou.mybatisplus.annotation.DbType;
-import org.springframework.stereotype.Component;
-
-/**
- * SQL server 数据源提供者
- *
- * @author lengleng
- * @date 2022-01-13
- */
-@Component
-public class SqlServerSqlProvider implements SqlProvider {
-
-	@Override
-	public String getInsertRangeSql() {
-		return "INSERT INTO #tableName(id,name,value,gmt_create,gmt_modified)" + " VALUE(?,?,?,?,?)";
-	}
-
-	@Override
-	public Boolean support(DbType dbType) {
-		return DbType.SQL_SERVER.equals(dbType);
-	}
-
-}