初始化第一版influxORM
This commit is contained in:
68
src/main/java/com/njcn/influx/base/ProxyMapper.java
Normal file
68
src/main/java/com/njcn/influx/base/ProxyMapper.java
Normal file
@@ -0,0 +1,68 @@
|
||||
package com.njcn.influx.base;
|
||||
|
||||
|
||||
import com.njcn.influx.ano.Delete;
|
||||
import com.njcn.influx.ano.Select;
|
||||
import com.njcn.influx.ano.Insert;
|
||||
import com.njcn.influx.core.InfluxExecutor;
|
||||
import com.njcn.influx.core.ParameterHandler;
|
||||
import com.njcn.influx.core.ResultSetHandler;
|
||||
|
||||
import java.lang.annotation.Annotation;
|
||||
import java.lang.reflect.InvocationHandler;
|
||||
import java.lang.reflect.Method;
|
||||
import java.lang.reflect.Parameter;
|
||||
import java.util.List;
|
||||
|
||||
public class ProxyMapper implements InvocationHandler {
|
||||
|
||||
private ParameterHandler parameterHandler;
|
||||
private InfluxExecutor executor;
|
||||
private ResultSetHandler resultSetHandler;
|
||||
|
||||
public ProxyMapper(ParameterHandler parameterHandler, InfluxExecutor executor, ResultSetHandler resultSetHandler) {
|
||||
this.parameterHandler = parameterHandler;
|
||||
this.executor = executor;
|
||||
this.resultSetHandler = resultSetHandler;
|
||||
}
|
||||
|
||||
@Override
|
||||
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
|
||||
Annotation[] annotations = method.getAnnotations();
|
||||
if (annotations.length == 1) {
|
||||
Annotation annotation = annotations[0];
|
||||
Class<? extends Annotation> annotationType = annotation.annotationType();
|
||||
//是查询的
|
||||
if (annotationType == Select.class) {
|
||||
Select selectAnnotation = method.getAnnotation(Select.class);
|
||||
//拼接sql
|
||||
String sql = selectAnnotation.value();
|
||||
Parameter[] parameters = method.getParameters();
|
||||
sql = parameterHandler.handleParameter(parameters, args, sql);
|
||||
//查询结果
|
||||
Class resultType = selectAnnotation.resultType();
|
||||
List<Object> list = executor.select(sql, resultType);
|
||||
//根据返回类型返回结果
|
||||
Class<?> returnType = method.getReturnType();
|
||||
return resultSetHandler.handleResultSet(list, returnType);
|
||||
}
|
||||
//是插入的
|
||||
if (annotationType == Insert.class) {
|
||||
executor.insert(args);
|
||||
}
|
||||
//是删除的
|
||||
if (annotationType == Delete.class) {
|
||||
Delete deleteAnnotation = method.getAnnotation(Delete.class);
|
||||
//拼接sql
|
||||
String sql = deleteAnnotation.value();
|
||||
Parameter[] parameters = method.getParameters();
|
||||
|
||||
String database = deleteAnnotation.database();
|
||||
sql = parameterHandler.handleParameter(parameters, args, sql);
|
||||
//执行sql
|
||||
executor.delete(sql, database);
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user