com.tsc9526.monalisa.orm.dialect.Dialect Maven / Gradle / Ivy
/*******************************************************************************************
* Copyright (c) 2016, zzg.zhou([email protected])
*
* Monalisa is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see hTables=new ConcurrentHashMap();
public abstract String getUrlPrefix();
public abstract String getDriver();
public abstract String getSchema(String jdbcUrl);
public abstract String getColumnName(String name);
public abstract String getTableName(String name);
public abstract Query getLimitQuery(Query origin,int limit ,int offset);
public abstract CreateTable getCreateTable(DBConfig db,String tableName);
public boolean tableExist(DBConfig db,String name,boolean includeView){
name=getRealname(name);
Set names=db.getTables(includeView);
return names.contains(name);
}
/**
* 实现该接口用于保持连接，在连接空闲时将执行该SQL语句。
* 默认空闲5分钟执行一次检查， 参考：{@link com.tsc9526.monalisa.orm.datasource.DbProp#CFG_CONNECT_IDLE_INTERVALS}
*
* @return 在定时检查时需执行的SQL语句
*/
public abstract String getIdleValidationQuery();
public synchronized void createTable(DBConfig db,CreateTable table){
String key=db.getKey()+":"+table.getTableName();
if(!hTables.containsKey(key)){
String sql=table.getCreateSQL();
logger.info(sql);
db.execute(table.getCreateSQL());
hTables.put(key, table);
}
}
protected String getTableName(Table table) {
String tableName=table.value();
if(tableName==null || tableName.trim().length()==0){
tableName=table.name();
}
return getTableName(tableName);
}
public Query insert(Model model,boolean updateOnDuplicateKey){
Query query=new Query();
if(updateOnDuplicateKey){
query.add("REPLACE ");
}else{
query.add("INSERT ");
}
query.add("INTO "+getTableName(model.table())+"(");
for(Object o:model.changedFields()){
FGS fgs=(FGS)o;
Column c=fgs.getAnnotation(Column.class);
Object v=getValue(fgs,model);
if(c.auto()==false || v!=null){
if(query.parameterCount()>0){
query.add(", ");
}
query.add(getColumnName(c.name()),v);
}
}
query.add(")VALUES(");
for(int i=0;i0?", ?":"?");
}
query.add(")");
return query;
}
public Query deleteAll(Model model){
Query query=new Query();
query.add("DELETE FROM "+getTableName(model.table()));
return query;
}
public Query truncate(Model model){
Query query=new Query();
query.add("TRUNCATE TABLE "+getTableName(model.table()));
return query;
}
public Query delete(Model model){
Query q=findWhereKey(model);
return delete(model,q.getSql(),q.getParameters());
}
public Query delete(Model model,String whereStatement,Object ... args){
if(whereStatement==null || whereStatement.trim().length()==0){
throw new RuntimeException("Model: "+model.getClass()+" delete fail, no where cause.");
}
Query query=new Query();
query.add("DELETE FROM "+getTableName(model.table())+" ");
if(whereStatement.toUpperCase().trim().startsWith("WHERE")){
query.add(whereStatement, args);
}else{
query.add("WHERE ").add(whereStatement, args);
}
return query;
}
public Query update(Model model){
Query q=findWhereKey(model);
return update(model,q.getSql(),q.getParameters());
}
public Query update(Model model,String whereStatement,Object ... args){
if(whereStatement==null || whereStatement.trim().length()==0){
throw new RuntimeException("Model: "+model.getClass()+" update fail, no where cause.");
}
Query query=new Query();
query.add("UPDATE "+getTableName(model.table())+" SET ");
for(Object o:model.changedFields()){
FGS fgs=(FGS)o;
Column c=fgs.getAnnotation(Column.class);
Object v=getValue(fgs,model);
if(c.key()==false || (model.updateKey() && v!=null)){
if(query.parameterCount()>0){
query.add(", ");
}
query.add(getColumnName(c.name())+"=?",v);
}
}
query.add(" ");
List kws=MelpSQL.splitKeyWords(whereStatement);
String w=kws.get(0);
if(w.equalsIgnoreCase("WHERE")){
query.add(whereStatement, args);
}else{
query.add("WHERE ").add(whereStatement, args);
}
return query;
}
public Query load(Model model){
Query query=new Query();
query.add("SELECT "+model.filterFields()+" FROM ").add(getTableName(model.table())).add(" WHERE ");
Query w=findWhereKey(model);
if(w.parameterCount()<1){
throw new RuntimeException("Model: "+model.getClass()+" load fail, no primary key.");
}
query.addQuery(w);
return query;
}
public Query selectOne(Model model,String whereStatement,Object ... args){
Query query=select(model, whereStatement, args);
return getLimitQuery(query,1,0);
}
public Query select(final Model model,String whereStatement,Object ... args){
Query query=new Query();
if(isJoinStatement(whereStatement)){
String x=model.filterFields();
if(x.equals("*")){
x="a.*";
}else{
StringBuffer sb=new StringBuffer();
for(String s:x.split(",")){
s="a."+s.trim();
if(sb.length()>0){
sb.append(", ");
}
sb.append(s);
}
x=sb.toString();
}
query.add("SELECT "+x+" FROM ").add(getTableName(model.table()));
query.add(" a ");
query.add(whereStatement, args);
}else{
query.add("SELECT "+model.filterFields()+" FROM ").add(getTableName(model.table()));
if(whereStatement!=null){
whereStatement=whereStatement.trim();
if(whereStatement.length()>0){
query.add(" ");
List kws=MelpSQL.splitKeyWords(whereStatement);
String w=kws.get(0);
if(w.equalsIgnoreCase("WHERE") || w.equalsIgnoreCase("ORDER")){
query.add(whereStatement, args);
}else{
query.add("WHERE ").add(whereStatement, args);
}
}
}
}
return query;
}
public Query count(Model model,String whereStatement,Object ... args){
Query query=new Query();
if(isJoinStatement(whereStatement)){
query.add("SELECT COUNT(*) FROM ").add(getTableName(model.table()));
query.add(" a ");
query.add(whereStatement, args);
}else{
query.add("SELECT COUNT(*) FROM ").add(getTableName(model.table()));
if(whereStatement!=null){
whereStatement=whereStatement.trim();
if(whereStatement.length()>0){
query.add(" ");
List kws=MelpSQL.splitKeyWords(whereStatement);
String w=kws.get(0);
if(w.equalsIgnoreCase("WHERE") || w.equalsIgnoreCase("ORDER")){
query.add(whereStatement, args);
}else{
query.add("WHERE ").add(whereStatement, args);
}
}
}
}
return query;
}
public Query notin(Query query,Object[] values){
return inOrNotIn("NOT IN",query,values);
}
public Query in(Query query,Object[] values){
return inOrNotIn("IN",query,values);
}
protected Query inOrNotIn(String keyInOrNotin,Query query,Object[] values){
String sql=query.getSql();
if(sql.length()>0){
char c=sql.charAt(sql.length()-1);
if(c!=' ' && c!='\n' && c!='\t'){
query.add(" ");
}
}
query.add(keyInOrNotin).add("(");
addQueryVars(query,values);
query.add(")");
return query;
}
protected void addQueryVars(Query query,Object value){
if(value instanceof Collection){
for(Object v:(Collection)value){
addQueryVars(query,v);
}
}else if(value!=null && value.getClass().isArray()){
for(Object v:(Object[])value){
addQueryVars(query,v);
}
}else{
if(query.getSql().endsWith("?")){
query.add(", ?",value);
}else{
query.add("?",value);
}
}
}
protected Query findWhereKey(Model model) {
Query query=getWhereByPrimaryKey(model);
if(query==null){
query=getWhereByUniqueKey(model);
}
if(query!=null){
return query;
}else{
throw new RuntimeException("Model: "+model.getClass()+", Primary key is null, or unique key is null");
}
}
protected Query getWhereByPrimaryKey(Model model){
Query query=new Query();
int keyType=-1; //-1: 初始化, 0-无匹配的键, 1-primary key, 2-unique key
for(Object o:model.fields()){
FGS fgs=(FGS)o;
Column c=fgs.getAnnotation(Column.class);
if(c.key()){
keyType=1;
Object v=getValue(fgs,model);
if(v!=null){
if(!query.isEmpty()){
query.add(" AND ");
}
query.add(getColumnName(c.name())+" = ?",v);
}else{
keyType=0;
break;
}
}
}
if(keyType==1){
return query;
}else{
return null;
}
}
protected Query getWhereByUniqueKey(Model model){
for(Object x:model.uniqueIndexes()){
ModelIndex index=(ModelIndex)x;
Query query=new Query();
List fs=index.getFields();
boolean keyExists=fs.size()>0;
for(FGS fgs:fs){
Column c=fgs.getAnnotation(Column.class);
Object v=getValue(fgs,model);
if(v!=null){
if(!query.isEmpty()){
query.add(" AND ");
}
query.add(getColumnName(c.name())+" = ?",v);
}else{
keyExists=false;
break;
}
}
if(keyExists){
return query;
}
}
return null;
}
protected Object getValue(FGS fgs,Model model) {
Object v=fgs.getObject(model);
if(v!=null){
Column c=fgs.getAnnotation(Column.class);
String type=MelpTypes.getJavaType(c.jdbcType());
if(v.getClass().isEnum()){
if(type.equals("String")){
return MelpEnum.getStringValue((Enum)v);
}else{
return MelpEnum.getIntValue((Enum)v);
}
}else if(v.getClass() == Boolean.class || v.getClass()==boolean.class){
if( (Boolean)v ){
if(type.equals("String")){
return "TRUE";
}else{
return 1;
}
}else{
if(type.equals("String")){
return "FALSE";
}else{
return 0;
}
}
}else if(v.getClass().isArray()){
if(v.getClass()==byte[].class){
return v;
}else{
JsonArray array=new JsonArray();
Object[] os=(Object[])new ArrayTypeConversion().convert(v, Object[].class);
for(Object o:os){
if(o!=null){
array.add(toJsonPrimitive(o));
}
}
return array.toString();
}
}else if(List.class.isAssignableFrom(v.getClass())){
JsonArray array=new JsonArray();
for(Object o:(List)v){
if(o!=null){
array.add(toJsonPrimitive(o));
}
}
return array.toString();
}else if(v.getClass()==JsonObject.class){
return v.toString();
}else if(v.getClass().isPrimitive() || v.getClass().getName().startsWith("java.")){
return v;
}else if(type.equals("String")){
return MelpJson.getGson().toJson(v);
}
}
return v;
}
private JsonPrimitive toJsonPrimitive(Object o){
Class clazz=o.getClass();
if(clazz==int.class || clazz==Integer.class){
return new JsonPrimitive((Integer)o);
}if(clazz==float.class || clazz==Float.class){
return new JsonPrimitive((Float)o);
}if(clazz==double.class || clazz==Double.class){
return new JsonPrimitive((Double)o);
}if(clazz==long.class || clazz==Long.class){
return new JsonPrimitive((Long)o);
}else{
return new JsonPrimitive(o.toString());
}
}
protected boolean isJoinStatement(String whereStatement){
if(whereStatement==null){
return false;
}
whereStatement=whereStatement.trim();
if(whereStatement.length()<1){
return false;
}
if(whereStatement.startsWith(",")){
return true;
}
List kws=MelpSQL.splitKeyWords(whereStatement);
if(kws.contains("JOIN")){
return true;
}
return false;
}
public Query getCountQuery(Query origin){
Query query=new Query();
query.use(origin.getDb());
String sql=origin.getSql().toLowerCase();
query.add(getCountSql(sql), origin.getParameters());
return query;
}
protected String getCountSql(String sql){
String cql=sql;
String loweredString = sql.toLowerCase();
int orderByIndex = loweredString.indexOf("order by");
if(orderByIndex>0){
cql=sql.substring(0,orderByIndex);
}
int groupByIndex=loweredString.indexOf("group by");
if(groupByIndex>0){
return "SELECT COUNT(*) AS cnt FROM( "+cql+") AS tmp";
}else{
int p=loweredString.indexOf("from");
while(p>0){
char left =loweredString.charAt(p-1);
char right=loweredString.charAt(p+4);
if(isSplitChar(left) && isSplitChar(right)){
break;
}else{
p=loweredString.indexOf("from",p+4);
}
}
if(p>0){
cql="SELECT COUNT(*) AS cnt "+cql.substring(p);
return cql;
}else{
return sql;
}
}
}
protected boolean isSplitChar(char c){
return c==' '|| c=='\t' || c=='\r' || c=='\n';
}
public static String getRealname(String name){
if(name==null){
return null;
}
if(name.length()<2){
return name;
}
char c=name.charAt(0);
if(c=='`' || c=='[' || c=='"' || c=='\''){
return name.substring(1,name.length()-1);
}
return name;
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy