DataSqlMapper

This commit is contained in:
redkale
2024-01-09 14:39:27 +08:00
parent 853f501ffd
commit 6c331f733b
6 changed files with 86 additions and 9 deletions

View File

@@ -74,6 +74,10 @@ public abstract class AsmMethodBoost<T> {
return rs; return rs;
} }
public static String getMethodBeanKey(Method method) {
return method.getName() + ":" + Type.getMethodDescriptor(method);
}
/** /**
* 获取需屏蔽的方法上的注解 * 获取需屏蔽的方法上的注解
* *

View File

@@ -0,0 +1,47 @@
/*
*
*/
package org.redkale.source;
import java.util.ArrayList;
import java.util.List;
import org.redkale.convert.ConvertDisabled;
/**
*
* 原生的sql解析基本信息对象 <br>
*
* <p>
* 详情见: https://redkale.org
*
* @author zhangjx
* @since 2.8.0
*/
public class DataNativeSqlInfo {
//原始sql语句
protected String rawSql;
//jdbc版的sql语句, 只有numberSignNames为空时才有值
protected String jdbcSql;
protected final List<String> rootParamNames = new ArrayList<>();
@ConvertDisabled
public boolean isDynamic() {
return jdbcSql == null;
}
public String getRawSql() {
return rawSql;
}
public String getJdbcSql() {
return jdbcSql;
}
public List<String> getRootParamNames() {
return rootParamNames;
}
}

View File

@@ -24,7 +24,9 @@ import org.redkale.util.RedkaleClassLoader;
*/ */
public interface DataNativeSqlParser { public interface DataNativeSqlParser {
DataNativeSqlStatement parse(IntFunction<String> signFunc, String dbtype, String rawSql, Map<String, Object> params); public DataNativeSqlInfo parse(IntFunction<String> signFunc, String dbtype, String rawSql);
public DataNativeSqlStatement parse(IntFunction<String> signFunc, String dbtype, String rawSql, Map<String, Object> params);
public static DataNativeSqlParser loadFirst() { public static DataNativeSqlParser loadFirst() {
if (DataNativeSqlStatement._first_parser != DataNativeSqlStatement.PARSER_NIL) { if (DataNativeSqlStatement._first_parser != DataNativeSqlStatement.PARSER_NIL) {

View File

@@ -22,10 +22,16 @@ import org.redkale.convert.json.JsonConvert;
public class DataNativeSqlStatement { public class DataNativeSqlStatement {
static final DataNativeSqlParser PARSER_NIL = new DataNativeSqlParser() { static final DataNativeSqlParser PARSER_NIL = new DataNativeSqlParser() {
@Override
public DataNativeSqlInfo parse(IntFunction<String> signFunc, String dbtype, String rawSql) {
throw new UnsupportedOperationException("Not supported yet.");
}
@Override @Override
public DataNativeSqlStatement parse(IntFunction<String> signFunc, String dbtype, String rawSql, Map<String, Object> params) { public DataNativeSqlStatement parse(IntFunction<String> signFunc, String dbtype, String rawSql, Map<String, Object> params) {
throw new UnsupportedOperationException("Not supported yet."); throw new UnsupportedOperationException("Not supported yet.");
} }
}; };
static DataNativeSqlParser _first_parser = PARSER_NIL; static DataNativeSqlParser _first_parser = PARSER_NIL;

View File

@@ -3,19 +3,17 @@
*/ */
package org.redkale.source.spi; package org.redkale.source.spi;
import java.lang.reflect.Field; import org.redkale.source.DataNativeSqlParser;
import org.redkale.inject.ResourceFactory; import org.redkale.source.DataSqlMapper;
import org.redkale.inject.ResourceTypeLoader; import org.redkale.source.DataSqlSource;
/** /**
* *
* @author zhangjx * @author zhangjx
*/ */
public class DataSqlMapperBuilder implements ResourceTypeLoader { public final class DataSqlMapperBuilder {
@Override public static <T, M extends DataSqlMapper<T>> M createMapper(DataNativeSqlParser nativeSqlParser, DataSqlSource source, Class<M> mapperType) {
public Object load(ResourceFactory factory, String srcResourceName, Object srcObj, String resourceName, Field field, Object attachment) {
return null; return null;
} }
} }

View File

@@ -137,7 +137,7 @@ public class SourceModuleEngine extends ModuleEngine implements SourceManager {
//--------------------------------- 注册 DataSource、CacheSource --------------------------------- //--------------------------------- 注册 DataSource、CacheSource ---------------------------------
resourceFactory.register(new DataSourceLoader(), DataSource.class); resourceFactory.register(new DataSourceLoader(), DataSource.class);
resourceFactory.register(new CacheSourceLoader(), CacheSource.class); resourceFactory.register(new CacheSourceLoader(), CacheSource.class);
resourceFactory.register(new DataSqlMapperBuilder(), DataSqlMapper.class); resourceFactory.register(new DataSqlMapperLoader(), DataSqlMapper.class);
} }
/** /**
@@ -508,6 +508,26 @@ public class SourceModuleEngine extends ModuleEngine implements SourceManager {
return conf; return conf;
} }
private class DataSqlMapperLoader implements ResourceTypeLoader {
@Override
public Object load(ResourceFactory rf, String srcResourceName, final Object srcObj, final String resourceName, Field field, final Object attachment) {
try {
if ((srcObj instanceof Service) && Sncp.isRemote((Service) srcObj)) {
return null; //远程模式不得注入
}
DataSource source = loadDataSource(resourceName, false);
Class mapperType = field.getType();
DataSqlMapper mapper = DataSqlMapperBuilder.createMapper(nativeSqlParser, (DataSqlSource) source, mapperType);
field.set(srcObj, mapper);
return mapper;
} catch (Exception e) {
logger.log(Level.SEVERE, DataSqlMapper.class.getSimpleName() + " inject to " + srcObj + " error", e);
return null;
}
}
}
private class DataSourceLoader implements ResourceTypeLoader { private class DataSourceLoader implements ResourceTypeLoader {
@Override @Override