Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.jn.sqlhelper.jfinal.dialect.JFinalCommonDialect Maven / Gradle / Ivy
package com.jn.sqlhelper.jfinal.dialect;
import com.jfinal.plugin.activerecord.Record;
import com.jfinal.plugin.activerecord.Table;
import com.jfinal.plugin.activerecord.dialect.*;
import com.jn.langx.util.collection.Collects;
import com.jn.langx.util.function.Consumer2;
import com.jn.langx.util.function.Function2;
import com.jn.sqlhelper.common.ddl.SQLSyntaxCompatTable;
import com.jn.sqlhelper.dialect.instrument.SQLInstrumentorConfig;
import com.jn.sqlhelper.dialect.instrument.SQLStatementInstrumentor;
import com.jn.sqlhelper.dialect.internal.OracleDialect;
import com.jn.sqlhelper.dialect.pagination.RowSelection;
import com.jn.sqlhelper.dialect.parameter.ArrayBasedParameterSetter;
import com.jn.sqlhelper.dialect.parameter.ArrayBasedQueryParameters;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.List;
import java.util.Map;
import java.util.Set;
public class JFinalCommonDialect extends Dialect {
protected com.jn.sqlhelper.dialect.Dialect delegate;
private static final Map builtInDialectMap = Collects.emptyHashMap();
protected String databaseId;
private SQLStatementInstrumentor instrumentor;
private ThreadLocal pagingRequestHolder = new ThreadLocal();
public JFinalCommonDialect(String databaseId) {
this.databaseId = databaseId.toLowerCase();
this.instrumentor = new SQLStatementInstrumentor();
if (this.instrumentor.getConfig() == null) {
SQLInstrumentorConfig config = new SQLInstrumentorConfig();
config.setDialect(databaseId);
this.instrumentor.setConfig(config);
}
this.instrumentor.init();
if (instrumentor.beginIfSupportsLimit(databaseId)) {
delegate = instrumentor.getCurrentDialect();
}
}
static {
builtInDialectMap.put("mysql", new MysqlDialect());
builtInDialectMap.put("oracle", new com.jfinal.plugin.activerecord.dialect.OracleDialect());
builtInDialectMap.put("postgresql", new PostgreSqlDialect());
builtInDialectMap.put("sqlserver", new SqlServerDialect());
builtInDialectMap.put("sqlite", new Sqlite3Dialect());
builtInDialectMap.put("sqlite3", new Sqlite3Dialect());
}
@Override
public String forTableBuilderDoBuild(String tableName) {
return "select * from " + tableName + " where 1 = 2";
}
@Override
public String forPaginate(int pageNumber, int pageSize, StringBuilder findSql) {
RowSelection rowSelection = new RowSelection();
rowSelection.setOffset(pageNumber <= 0 ? 0L : (pageNumber - 1) * pageSize);
rowSelection.setLimit(pageSize);
if (instrumentor.beginIfSupportsLimit(databaseId)) {
pagingRequestHolder.set(rowSelection);
return instrumentor.instrumentLimitSql(findSql.toString(), rowSelection);
} else {
return findSql.toString();
}
}
@Override
public void fillStatement(PreparedStatement pst, List paras) throws SQLException {
fillStatement(pst, paras.toArray());
}
@Override
public void fillStatement(PreparedStatement pst, Object... paras) throws SQLException {
RowSelection rowSelection = pagingRequestHolder.get();
if (rowSelection != null) {
pagingRequestHolder.remove();
ArrayBasedQueryParameters queryParameters = new ArrayBasedQueryParameters();
queryParameters.setRowSelection(rowSelection);
instrumentor.bindParameters(pst, new ArrayBasedParameterSetter(), queryParameters, true);
} else {
super.fillStatement(pst, paras);
}
}
@Override
public boolean isOracle() {
return delegate == null ? "oracle".equals(databaseId) : delegate instanceof OracleDialect;
}
private Dialect findCompatibleDialect() {
Set compatDatabases = SQLSyntaxCompatTable.getInstance().getCompatDatabases(databaseId);
return Collects.firstMap(compatDatabases, new Function2() {
@Override
public Dialect apply(Integer index, String compatDatabase) {
return JFinalCommonDialect.builtInDialectMap.get(compatDatabase);
}
});
}
@Override
public String forModelFindById(Table table, String columns) {
Dialect builtInDelegate = findCompatibleDialect();
if (builtInDelegate != null) {
return builtInDelegate.forModelFindById(table, columns);
}
StringBuilder sql = new StringBuilder("select ").append(columns).append(" from ");
sql.append(table.getName());
sql.append(" where ");
String[] pKeys = table.getPrimaryKey();
appendWhereParamters(sql, pKeys);
return sql.toString();
}
@Override
public String forModelDeleteById(Table table) {
Dialect builtInDelegate = findCompatibleDialect();
if (builtInDelegate != null) {
return builtInDelegate.forModelDeleteById(table);
}
String[] pKeys = table.getPrimaryKey();
StringBuilder sql = new StringBuilder(45);
sql.append("delete from ");
sql.append(table.getName());
sql.append(" where ");
appendWhereParamters(sql, pKeys);
return sql.toString();
}
@Override
public void forModelSave(Table table, Map attrs, StringBuilder sql, List paras) {
Dialect builtInDelegate = findCompatibleDialect();
if (builtInDelegate != null) {
builtInDelegate.forModelSave(table, attrs, sql, paras);
return;
}
sql.append("insert into ").append(table.getName()).append('(');
StringBuilder temp = new StringBuilder(") values(");
for (Map.Entry e : attrs.entrySet()) {
String colName = e.getKey();
if (table.hasColumnLabel(colName)) {
if (!paras.isEmpty()) {
sql.append(", ");
temp.append(", ");
}
sql.append(colName);
temp.append('?');
paras.add(e.getValue());
}
}
sql.append(temp.toString()).append(')');
}
@Override
public void forModelUpdate(Table table, Map attrs, Set modifyFlag, StringBuilder sql, List paras) {
Dialect builtInDelegate = findCompatibleDialect();
if (builtInDelegate != null) {
builtInDelegate.forModelUpdate(table, attrs, modifyFlag, sql, paras);
return;
}
sql.append("update ").append(table.getName()).append(" set ");
String[] pKeys = table.getPrimaryKey();
for (Map.Entry e : attrs.entrySet()) {
String colName = e.getKey();
if (modifyFlag.contains(colName) && !isPrimaryKey(colName, pKeys) && table.hasColumnLabel(colName)) {
if (!paras.isEmpty()) {
sql.append(", ");
}
sql.append(colName).append(" = ? ");
paras.add(e.getValue());
}
}
sql.append(" where ");
for (int i = 0; i < pKeys.length; i++) {
if (i > 0) {
sql.append(" and ");
}
sql.append(pKeys[i]).append(" = ?");
paras.add(attrs.get(pKeys[i]));
}
}
@Override
public String forDbFindById(String tableName, String[] pKeys) {
Dialect builtInDelegate = findCompatibleDialect();
if (builtInDelegate != null) {
return builtInDelegate.forDbFindById(tableName, pKeys);
}
tableName = tableName.trim();
trimPrimaryKeys(pKeys);
StringBuilder sql = new StringBuilder("select * from ").append(tableName).append(" where ");
appendWhereParamters(sql, pKeys);
return sql.toString();
}
@Override
public String forDbDeleteById(String tableName, String[] pKeys) {
Dialect builtInDelegate = findCompatibleDialect();
if (builtInDelegate != null) {
return builtInDelegate.forDbDeleteById(tableName, pKeys);
}
tableName = tableName.trim();
trimPrimaryKeys(pKeys);
final StringBuilder sql = new StringBuilder("delete from ").append(tableName).append(" where ");
appendWhereParamters(sql, pKeys);
return sql.toString();
}
@Override
public void forDbSave(String tableName, String[] pKeys, Record record, StringBuilder sql, List paras) {
Dialect builtInDelegate = findCompatibleDialect();
if (builtInDelegate != null) {
builtInDelegate.forDbSave(tableName, pKeys, record, sql, paras);
return;
}
tableName = tableName.trim();
trimPrimaryKeys(pKeys);
sql.append("insert into ");
sql.append(tableName).append('(');
StringBuilder temp = new StringBuilder();
temp.append(") values(");
int count = 0;
for (Map.Entry e : record.getColumns().entrySet()) {
String colName = e.getKey();
if (count++ > 0) {
sql.append(", ");
temp.append(", ");
}
sql.append(colName);
Object value = e.getValue();
if (value instanceof String && isPrimaryKey(colName, pKeys) && ((String) value).endsWith(".nextval")) {
temp.append(value);
} else {
temp.append('?');
paras.add(value);
}
}
sql.append(temp.toString()).append(')');
}
@Override
public void forDbUpdate(String tableName, String[] pKeys, Object[] ids, Record record, final StringBuilder sql, List paras) {
Dialect builtInDelegate = findCompatibleDialect();
if (builtInDelegate != null) {
builtInDelegate.forDbUpdate(tableName, pKeys, ids, record, sql, paras);
return;
}
tableName = tableName.trim();
trimPrimaryKeys(pKeys);
sql.append("update ").append(tableName).append(" set ");
for (Map.Entry e : record.getColumns().entrySet()) {
String colName = e.getKey();
if (!isPrimaryKey(colName, pKeys)) {
if (!paras.isEmpty()) {
sql.append(", ");
}
sql.append(colName).append(" = ? ");
paras.add(e.getValue());
}
}
sql.append(" where ");
appendWhereParamters(sql, pKeys);
}
private void appendWhereParamters(final StringBuilder sql, String[] pKeys) {
Collects.forEach(pKeys, new Consumer2() {
@Override
public void accept(Integer i, String s) {
if (i > 0) {
sql.append(" and ");
}
sql.append(s).append(" = ?");
}
});
}
}
