package com.qianwen.core.sequence.range.db;
|
|
import cn.hutool.core.util.StrUtil;
|
import com.baomidou.mybatisplus.core.incrementer.DefaultIdentifierGenerator;
|
import java.sql.Connection;
|
import java.sql.PreparedStatement;
|
import java.sql.ResultSet;
|
import java.sql.SQLException;
|
import java.sql.Statement;
|
import java.sql.Timestamp;
|
import javax.sql.DataSource;
|
import com.qianwen.core.sequence.exception.SeqException;
|
import com.qianwen.core.sequence.range.db.provider.SqlProviderFactory;
|
import com.qianwen.core.tool.utils.SpringUtil;
|
|
|
public class BaseDbHelper {
|
private static final long DELTA = 100000000;
|
private static final SqlProviderFactory SQL_PROVIDER_FACTORY = (SqlProviderFactory) SpringUtil.getBean(SqlProviderFactory.class);
|
private static final DefaultIdentifierGenerator identifierGenerator = new DefaultIdentifierGenerator();
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
public static void createTable(DataSource dataSource, String tableName) {
|
Connection conn = null;
|
Statement stmt = null;
|
try {
|
try {
|
conn = dataSource.getConnection();
|
stmt = conn.createStatement();
|
stmt.executeUpdate(SQL_PROVIDER_FACTORY.getCreateTableSql().replace("#tableName", tableName));
|
closeQuietly(stmt);
|
closeQuietly(conn);
|
} catch (SQLException e) {
|
throw new SeqException(e);
|
}
|
} catch (Throwable th) {
|
closeQuietly(stmt);
|
closeQuietly(conn);
|
throw th;
|
}
|
}
|
|
private static void insertRange(DataSource dataSource, String tableName, String name, long stepStart) {
|
Connection conn = null;
|
PreparedStatement stmt = null;
|
try {
|
try {
|
conn = dataSource.getConnection();
|
stmt = conn.prepareStatement(SQL_PROVIDER_FACTORY.getInsertRangeSql().replace("#tableName", tableName));
|
stmt.setLong(1, identifierGenerator.nextId((Object) null).longValue());
|
stmt.setString(2, name);
|
stmt.setLong(3, stepStart);
|
stmt.setTimestamp(4, new Timestamp(System.currentTimeMillis()));
|
stmt.setTimestamp(5, new Timestamp(System.currentTimeMillis()));
|
stmt.executeUpdate();
|
closeQuietly(stmt);
|
closeQuietly(conn);
|
} catch (SQLException e) {
|
throw new SeqException(e);
|
}
|
} catch (Throwable th) {
|
closeQuietly(stmt);
|
closeQuietly(conn);
|
throw th;
|
}
|
}
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
public static boolean updateRange(DataSource dataSource, String tableName, Long newValue, Long oldValue, String name) {
|
Connection conn = null;
|
PreparedStatement stmt = null;
|
try {
|
try {
|
conn = dataSource.getConnection();
|
stmt = conn.prepareStatement(SQL_PROVIDER_FACTORY.getUpdateRangeSql().replace("#tableName", tableName));
|
stmt.setLong(1, newValue.longValue());
|
stmt.setTimestamp(2, new Timestamp(System.currentTimeMillis()));
|
stmt.setString(3, name);
|
stmt.setLong(4, oldValue.longValue());
|
int affectedRows = stmt.executeUpdate();
|
boolean z = affectedRows > 0;
|
closeQuietly(stmt);
|
closeQuietly(conn);
|
return z;
|
} catch (SQLException e) {
|
throw new SeqException(e);
|
}
|
} catch (Throwable th) {
|
closeQuietly(stmt);
|
closeQuietly(conn);
|
throw th;
|
}
|
}
|
|
/* JADX INFO: Access modifiers changed from: package-private */
|
public static Long selectRange(DataSource dataSource, String tableName, String name, long stepStart) {
|
try {
|
try {
|
Connection conn = dataSource.getConnection();
|
PreparedStatement stmt = conn.prepareStatement(SQL_PROVIDER_FACTORY.getSelectRangeSql().replace("#tableName", tableName));
|
stmt.setString(1, name);
|
ResultSet rs = stmt.executeQuery();
|
if (!rs.next()) {
|
insertRange(dataSource, tableName, name, stepStart);
|
closeQuietly(rs);
|
closeQuietly(stmt);
|
closeQuietly(conn);
|
return null;
|
}
|
long oldValue = rs.getLong(1);
|
if (oldValue < 0) {
|
String msg = "Sequence value cannot be less than zero, value = " + oldValue + ", please check table sequence" + tableName;
|
throw new SeqException(msg);
|
} else if (oldValue > 9223372036754775807L) {
|
String msg2 = "Sequence value overflow, value = " + oldValue + ", please check table sequence" + tableName;
|
throw new SeqException(msg2);
|
} else {
|
Long valueOf = Long.valueOf(oldValue);
|
closeQuietly(rs);
|
closeQuietly(stmt);
|
closeQuietly(conn);
|
return valueOf;
|
}
|
} catch (SQLException e) {
|
throw new SeqException(e);
|
}
|
} catch (Throwable th) {
|
closeQuietly(null);
|
closeQuietly(null);
|
closeQuietly(null);
|
throw th;
|
}
|
}
|
|
public static Boolean existTable(DataSource dataSource, String tableName) {
|
String existTableSql = SQL_PROVIDER_FACTORY.getExistTableSql();
|
if (StrUtil.isBlank(existTableSql)) {
|
return true;
|
}
|
try {
|
try {
|
Connection conn = dataSource.getConnection();
|
Statement stmt = conn.createStatement();
|
ResultSet resultSet = stmt.executeQuery(SQL_PROVIDER_FACTORY.getExistTableSql().replace("#tableName", tableName));
|
if (!resultSet.next()) {
|
closeQuietly(stmt);
|
closeQuietly(conn);
|
return false;
|
}
|
int count = resultSet.getInt(1);
|
if (0 == count) {
|
closeQuietly(stmt);
|
closeQuietly(conn);
|
return true;
|
}
|
closeQuietly(stmt);
|
closeQuietly(conn);
|
return false;
|
} catch (SQLException e) {
|
throw new SeqException(e);
|
}
|
} catch (Throwable th) {
|
closeQuietly(null);
|
closeQuietly(null);
|
throw th;
|
}
|
}
|
|
private static void closeQuietly(AutoCloseable closeable) {
|
if (null != closeable) {
|
try {
|
closeable.close();
|
} catch (Throwable th) {
|
}
|
}
|
}
|
}
|
