cn.vonce.sql.spring.service.BaseSqlBeanServiceImpl Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of vonce-sqlbean-spring Show documentation
Show all versions of vonce-sqlbean-spring Show documentation
This is a tool that uses java object-oriented idea to write and generate SQL statements. On this basis,
it also implements lightweight plug-in support similar to JPA for mybatis and spring JDBC. A large number of
common SQL execution methods are built in plug-ins to improve development efficiency, reduce a large number of
SQL statement writing, and make developers focus on business code writing.
The newest version!
package cn.vonce.sql.spring.service;
import cn.vonce.sql.annotation.SqlId;
import cn.vonce.sql.bean.ColumnInfo;
import cn.vonce.sql.config.SqlBeanConfig;
import cn.vonce.sql.config.SqlBeanDB;
import cn.vonce.sql.enumerate.DbType;
import cn.vonce.sql.enumerate.IdType;
import cn.vonce.sql.enumerate.JdbcType;
import cn.vonce.sql.uitls.ReflectUtil;
import cn.vonce.sql.uitls.SqlBeanUtil;
import cn.vonce.sql.uitls.StringUtil;
import java.lang.reflect.Field;
import java.sql.DatabaseMetaData;
import java.sql.SQLException;
import java.util.Collection;
import java.util.List;
import java.util.Objects;
/**
* SqlBeanServiceImpl 抽象类 获取配置通用实现
*
* @param
* @author Jovi
* @version 1.0
* @email [email protected]
*/
public abstract class BaseSqlBeanServiceImpl {
private SqlBeanDB sqlBeanDB;
abstract SqlBeanConfig getSqlBeanConfig();
abstract SqlBeanDB initDBInfo();
public SqlBeanDB getSqlBeanDB() {
if (sqlBeanDB == null) {
sqlBeanDB = initDBInfo();
sqlBeanDB.setSqlBeanConfig(getSqlBeanConfig());
//如果用户未进行配置
boolean isUserConfig = true;
if (sqlBeanDB.getSqlBeanConfig() == null) {
isUserConfig = false;
sqlBeanDB.setSqlBeanConfig(new SqlBeanConfig());
}
//如果用户未进行配置则对某些数据库进行设置
if (!isUserConfig) {
switch (Objects.requireNonNull(sqlBeanDB.getDbType())) {
case Oracle:
case DB2:
case Derby:
case Hsql:
case H2:
sqlBeanDB.getSqlBeanConfig().setToUpperCase(true);
break;
}
}
}
return sqlBeanDB;
}
/**
* 处理字段信息
*
* @param columnInfoList
*/
public void handleColumnInfo(List columnInfoList) {
DbType dbType = getSqlBeanDB().getDbType();
if (dbType == DbType.Derby) {
for (ColumnInfo info : columnInfoList) {
String[] values = info.getType().split(" ");
//设置字段类型
info.setType(StringUtil.getWord(values[0]));
//如果存在空格分割说明字段名后面存在NOT NULL，即不能为空
if (values.length > 1) {
info.setNotnull(true);
} else {
info.setNotnull(false);
}
//设置字段长度范围
String range[] = StringUtil.getBracketContent(values[0]).split(",");
info.setLength("".equals(range[0]) ? 0 : Long.parseLong(range[0]));
info.setScale(range.length == 1 ? 0 : Integer.parseInt(range[1]));
info.setDfltValue(processDefaultValue(info.getDfltValue()));
}
} else if (dbType == DbType.H2 || dbType == DbType.Postgresql) {
Boolean toUpperCase = getSqlBeanDB().getSqlBeanConfig().getToUpperCase();
for (ColumnInfo info : columnInfoList) {
if ("CHARACTER VARYING".equalsIgnoreCase(info.getType())) {
info.setType(JdbcType.VARCHAR.getName(toUpperCase));
} else if ("CHARACTER LARGE OBJECT".equalsIgnoreCase(info.getType())) {
info.setType(JdbcType.LONGTEXT.getName(toUpperCase));
} else if ("BINARY VARYING".equalsIgnoreCase(info.getType())) {
info.setType(JdbcType.BLOB.getName(toUpperCase));
} else if ("BINARY LARGE OBJECT".equalsIgnoreCase(info.getType())) {
info.setType(JdbcType.LONGBLOB.getName(toUpperCase));
} else if ("TIMESTAMP WITHOUT TIME ZONE".equalsIgnoreCase(info.getType())) {
info.setType(JdbcType.TIMESTAMP.getName(toUpperCase));
} else if ("TIME WITHOUT TIME ZONE".equalsIgnoreCase(info.getType())) {
info.setType(JdbcType.TIME.getName(toUpperCase));
}
String deftValue = info.getDfltValue();
if (deftValue != null && StringUtil.isNotBlank(deftValue) && deftValue.indexOf("'::") > -1) {
info.setDfltValue(deftValue.substring(1, deftValue.indexOf("'::")));
}
}
} else if (dbType == DbType.Oracle || dbType == DbType.DB2) {
for (ColumnInfo info : columnInfoList) {
info.setDfltValue(processDefaultValue(info.getDfltValue()));
}
} else if (dbType == DbType.SQLite) {
for (ColumnInfo info : columnInfoList) {
if (info.getType().indexOf("(") > -1 && info.getType().indexOf(")") > -1) {
String range = info.getType().substring(info.getType().indexOf("(") + 1, info.getType().indexOf(")"));
if (range.indexOf(",") > -1) {
String ranges[] = range.split(",");
info.setLength(Long.parseLong(ranges[0]));
info.setScale(Integer.parseInt(ranges[1]));
} else {
info.setLength(Long.parseLong(range));
}
info.setType(info.getType().substring(0, info.getType().indexOf("(")));
}
info.setDfltValue(processDefaultValue(info.getDfltValue()));
}
}
}
private String processDefaultValue(String dfltValue) {
if (dfltValue != null) {
dfltValue = dfltValue.trim();
if (dfltValue.charAt(0) == '\'' && dfltValue.charAt(dfltValue.length() - 1) == '\'') {
return dfltValue.substring(1, dfltValue.length() - 1);
} else {
return dfltValue;
}
}
return null;
}
/**
* 填充数据
*
* @param sqlBeanDB
* @param metaData
* @throws SQLException
*/
public void sqlBeanDBFill(SqlBeanDB sqlBeanDB, DatabaseMetaData metaData) throws SQLException {
sqlBeanDB.setProductName(metaData.getDatabaseProductName());
sqlBeanDB.setDatabaseMajorVersion(metaData.getDatabaseMajorVersion());
sqlBeanDB.setDatabaseMinorVersion(metaData.getDatabaseMinorVersion());
sqlBeanDB.setDatabaseProductVersion(metaData.getDatabaseProductVersion());
sqlBeanDB.setJdbcMajorVersion(metaData.getJDBCMajorVersion());
sqlBeanDB.setJdbcMinorVersion(metaData.getJDBCMinorVersion());
sqlBeanDB.setDriverMajorVersion(metaData.getDatabaseMajorVersion());
sqlBeanDB.setDriverMinorVersion(metaData.getDriverMinorVersion());
sqlBeanDB.setDriverVersion(metaData.getDriverVersion());
sqlBeanDB.setDriverName(metaData.getDriverName());
sqlBeanDB.setDbType(DbType.getDbType(sqlBeanDB.getProductName()));
}
/**
* 获取自增id
*
* @return
*/
abstract Long getAutoIncrId();
/**
* 设置自增id
*
* @param clazz
* @param beanList
* @param
*/
public void setAutoIncrId(Class clazz, Collection beanList) {
if (beanList == null || beanList.size() == 0) {
return;
}
Field idField = SqlBeanUtil.getIdField(clazz);
if (idField != null && idField.getAnnotation(SqlId.class).type() == IdType.AUTO) {
Long idBeginValue = getAutoIncrId();
if (idBeginValue == null) {
return;
}
int i = 0;
for (T t : beanList) {
Field field = SqlBeanUtil.getIdField(t.getClass());
if (Number.class.isAssignableFrom(field.getType())) {
Long value = idBeginValue + i;
Object newValue = ReflectUtil.instance().invoke(field.getType(), null, "valueOf", new Class[]{String.class}, new Object[]{value.toString()});
ReflectUtil.instance().set(t.getClass(), t, field.getName(), newValue);
}
i++;
}
}
}
}