com.pugwoo.dbhelper.impl.DBHelperContext Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of nimble-orm Show documentation
Show all versions of nimble-orm Show documentation
nimble-orm an ORM based on Spring jdbcTemplate.
package com.pugwoo.dbhelper.impl;
import com.pugwoo.dbhelper.utils.InnerCommonUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.time.LocalDate;
import java.time.LocalDateTime;
import java.time.LocalTime;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
/**
* DBHelper线程上下文，支持:
* 1. 自定义表名
* 2. 线程级别的SQL注释
*/
public class DBHelperContext {
private static final Logger LOGGER = LoggerFactory.getLogger(DBHelperContext.class);
/**自定义表名，适合于分表场景*/
private static final ThreadLocal, String>> tableNames = new ThreadLocal<>();
/**全局的SQL注释*/
private static String globalComment;
/**线程上下文注释*/
private static final ThreadLocal comment = new ThreadLocal<>();
/**是否开启缓存，默认开启*/
private static boolean isCacheEnabled = true;
/**存放类对应的插入数据默认值的集合，以集合为单位提供给Table注解使用
* 内置的集合有：
* 1) clickhouse 适用于clickhouse数据库
*/
private static final Map, Object>> insertDefaultValueMap = new ConcurrentHashMap<>();
/*初始化类对应的默认值*/
static {
{
Map, Object> clickhouse = new ConcurrentHashMap<>();
clickhouse.put(Integer.class, 0);
clickhouse.put(Long.class, 0L);
clickhouse.put(Float.class, 0.0f);
clickhouse.put(Double.class, 0.0d);
clickhouse.put(Boolean.class, false);
clickhouse.put(String.class, "");
clickhouse.put(java.util.Date.class, new java.util.Date(0));
clickhouse.put(java.sql.Date.class, new java.sql.Date(0));
clickhouse.put(java.sql.Timestamp.class, new java.sql.Timestamp(0));
clickhouse.put(java.sql.Time.class, new java.sql.Time(0));
clickhouse.put(BigDecimal.class, new BigDecimal(0));
clickhouse.put(BigInteger.class, new BigInteger("0"));
clickhouse.put(LocalDate.class, LocalDate.of(1970, 1, 1));
clickhouse.put(LocalDateTime.class, LocalDateTime.of(1970, 1, 1, 0, 0, 0));
clickhouse.put(LocalTime.class, LocalTime.of(0, 0, 0));
insertDefaultValueMap.put("clickhouse", clickhouse);
}
{
Map, Object> mysql = new ConcurrentHashMap<>();
mysql.put(Integer.class, 0);
mysql.put(Long.class, 0L);
mysql.put(Float.class, 0.0f);
mysql.put(Double.class, 0.0d);
mysql.put(Boolean.class, false);
mysql.put(String.class, "");
mysql.put(java.util.Date.class, new java.util.Date(0));
mysql.put(java.sql.Date.class, new java.sql.Date(0));
mysql.put(java.sql.Timestamp.class, new java.sql.Timestamp(0));
mysql.put(java.sql.Time.class, new java.sql.Time(0));
mysql.put(BigDecimal.class, new BigDecimal(0));
mysql.put(BigInteger.class, new BigInteger("0"));
mysql.put(LocalDate.class, LocalDate.of(1970, 1, 1));
mysql.put(LocalDateTime.class, LocalDateTime.of(1970, 1, 1, 0, 0, 0));
mysql.put(LocalTime.class, LocalTime.of(0, 0, 0));
insertDefaultValueMap.put("mysql", mysql);
}
}
/**
* 动态获得类对应的自定义表名，不存在返回null
*/
public static String getTableName(Class clazz) {
Map, String> tableNames = DBHelperContext.tableNames.get();
if(tableNames == null) {
return null;
}
return tableNames.get(clazz);
}
/**
* 设置类对应的自定义表名
* @param tableName 不需要加反引号`，如果为null表示清除自定义表名
*/
public static void setTableName(Class clazz, String tableName) {
if(clazz == null) {
return;
}
if (tableName == null) { // 清除表名
Map, String> tableNames = DBHelperContext.tableNames.get();
if (tableNames != null) {
tableNames.remove(clazz);
}
}
if(InnerCommonUtils.isBlank(tableName)) { // 空字符串不允许，不处理
return;
}
Map, String> tableNames = DBHelperContext.tableNames.get();
if(tableNames == null) {
tableNames = new HashMap<>();
DBHelperContext.tableNames.set(tableNames);
}
tableNames.put(clazz, tableName);
}
/**
* 清空自定义表名
*/
public static void resetTableName() {
tableNames.set(null);
}
public static void setGlobalComment(String comment) {
DBHelperContext.globalComment = comment;
}
public static String getGlobalComment() {
return DBHelperContext.globalComment;
}
/**
* 设置线程上下文的SQL注释
*/
public static void setThreadLocalComment(String comment) {
DBHelperContext.comment.set(comment);
}
/**
* 获取线程上下文的SQL注释
*/
public static String getThreadLocalComment() {
return DBHelperContext.comment.get();
}
/**全局开启缓存*/
public static synchronized void enableCache() {
isCacheEnabled = true;
}
public static synchronized void disableCache() {
isCacheEnabled = false;
}
public static boolean isCacheEnabled() {
return isCacheEnabled;
}
public static Map, Object> getInsertDefaultValueMap(String insertDefaultValueMapName) {
if (InnerCommonUtils.isBlank(insertDefaultValueMapName)) {
return null;
}
Map, Object> map = insertDefaultValueMap.get(insertDefaultValueMapName);
if (map == null) {
LOGGER.error("getInsertDefaultValueMap is not exist, insertDefaultValueMapName:{}", insertDefaultValueMapName);
}
return map;
}
/**
* 设置插入时的默认值
* @param insertDefaultValueMapName 插入默认值的集合名称
* @param clazz 类的类型
* @param value 默认值
*/
public static void setInsertDefaultValue(String insertDefaultValueMapName, Class clazz, Object value) {
if (InnerCommonUtils.isBlank(insertDefaultValueMapName)) {
LOGGER.error("setInsertDefaultValue insertDefaultValueMapName is blank, ignore set");
return;
}
if (clazz == null) {
LOGGER.error("setInsertDefaultValue clazz is null, ignore set");
return;
}
Map, Object> map = insertDefaultValueMap.get(insertDefaultValueMapName);
if (map == null) {
map = new ConcurrentHashMap<>();
insertDefaultValueMap.put(insertDefaultValueMapName, map);
}
map.put(clazz, value);
}
}