com.artfii.fluentsql.core.Attr Maven / Gradle / Ivy
package com.artfii.fluentsql.core;
import com.artfii.fluentsql.core.mock.User;
import com.trigersoft.jaque.expression.*;
import javax.persistence.Entity;
import javax.persistence.Id;
import javax.persistence.Table;
import java.io.Serializable;
import java.lang.reflect.Field;
import java.lang.reflect.Member;
import java.util.List;
import java.util.Set;
import java.util.function.Function;
/**
* @function : 分析 lambda 方法
* @autor: leeton
* @date : 2018-03-27 6:36 PM
*/
public class Attr {
private Class clz; // 方法所属的类
private String name; // 属性名
private String column; // 对应的字段名称
private Member member;
private String tableName;
private Object val;
// this interface is required to make the lambda Serializable, which removes a need for
// jdk.internal.lambda.dumpProxyClasses system property. See below.
public interface Property extends Function, Serializable { }
public Attr() {
}
public Attr(Property propertyRef) {
parse(propertyRef);
}
/**
*
* 传入 lambda 静态方法，返回方法属性名及对应的字段名
* @param propertyRef
* @return Attr
*/
public Attr parse(Property propertyRef){
LambdaExpression expression = getExpression(propertyRef);
List params = expression.getParameters();
this.member = getMember(expression);
this.clz = params.get(0).getResultType();
this.name = attrName(member);
this.column = getUnderLineName(this.name);
this.tableName = getRealTableName(this.clz);
return this;
}
public LambdaExpression getExpression(Property propertyRef) {
return LambdaExpression.parse(propertyRef);
}
public static Member getMember(LambdaExpression lambdaExpression) {
Expression methodCall = lambdaExpression.getBody();
// remove casts
while (methodCall instanceof UnaryExpression) {
methodCall = ((UnaryExpression) methodCall).getFirst();
}
// checks are omitted for brevity
Member member = ((MemberExpression) ((InvocationExpression) methodCall).getTarget()).getMember();
return member;
}
public String attrName(Member member) {
return getCamelName(member.getName());
}
private String getCamelName(String method) {
method = method.startsWith("get") ? method.substring(3) : method.substring(2);
return StringKit.toLowerCaseFirstOne(method);
}
public String[] attrName(Member... member) {
String[] cols = new String[member.length];
int i = 0;
for (Member fun : member) {
cols[i++] = attrName(fun);
}
return cols;
}
public String attrName(Property fun) {
Member member = getMember(getExpression(fun));
return getCamelName(member.getName());
}
public String[] attrName(Property... funs) {
String[] cols = new String[funs.length];
int i = 0;
for (Property fun : funs) {
cols[i++] = attrName(fun);
}
return cols;
}
public static String getRealTableName(Class c) {
Table table = (Table) c.getAnnotation(Table.class);
Entity entity = (Entity) c.getAnnotation(Entity.class);
if (table != null && !table.name().isEmpty()) {
return table.name();
} else if (entity != null && !entity.name().isEmpty()) {
return entity.name();
}
return StringKit.toUnderline(c.getSimpleName());
}
/**
* 取得实体类的真实ID与值
* @param entity
*/
public static Attr getRealIdAttr(T entity) {
Set fields = BeanMapUtils.getFields(entity.getClass());
Attr attr = new Attr();
try {
for (Field f : fields) {
if(null != f.getAnnotation(Id.class) || f.getName().equals("id")){
f.setAccessible(true);
attr.clz = entity.getClass();
attr.name = f.getName();
attr.column = f.getName().equals("id")?"id":StringKit.toUnderline(f.getName());
attr.val = f.get(entity);
return attr;
}
}
} catch (IllegalAccessException e) {
e.printStackTrace();
}
return attr;
}
public String getUnderLineName(String arrtName) {
return StringKit.toUnderline(arrtName);
}
public static String[] getColumns(Attr... methods) {
String[] cols = new String[methods.length];
int i = 0;
for (Attr method : methods) {
cols[i++] = method.getColumn();
}
return cols;
}
public String getTableName() {
return tableName;
}
public void setTableName(String tableName) {
this.tableName = tableName;
}
public Class getClz() {
return clz;
}
public void setClz(Class clz) {
this.clz = clz;
}
public String getName() {
return name;
}
public void setName(String name) {
this.name = name;
}
public String getColumn() {
return column;
}
public void setColumn(String column) {
this.column = column;
}
public Object getVal() {
return val;
}
public void setVal(Object val) {
this.val = val;
}
@Override
public String toString() {
final StringBuilder sb = new StringBuilder("Attr{");
sb.append("clz=").append(clz);
sb.append(", name='").append(name).append('\'');
sb.append(", column='").append(column).append('\'');
sb.append(", member=").append(member);
sb.append(", tableName='").append(tableName).append('\'');
sb.append(", val=").append(val);
sb.append('}');
return sb.toString();
}
public static void main(String[] args) {
User user = new User();
user.setId(100);
Attr objs = Attr.getRealIdAttr(user);
System.err.println(objs);
}
}