package org.yeshi.utils.generater.entity;
|
|
import org.springframework.data.annotation.Id;
|
import org.springframework.data.annotation.Transient;
|
import org.yeshi.utils.generater.annotation.admin.DaoQueryCondition;
|
|
import java.lang.annotation.Annotation;
|
import java.lang.reflect.Field;
|
import java.util.ArrayList;
|
import java.util.HashSet;
|
import java.util.List;
|
import java.util.Set;
|
|
public class MongoDBDaoData extends BaseData {
|
|
private String daoName;
|
private String baseDaoClassName;
|
private String entityClassName;
|
private Set<String> importClasses;
|
|
private ColumnData identityColumn;
|
private List<ColumnData> columnDataList;
|
|
private List<DaoQueryColumnData> daoQueryColumnDataList;
|
|
public static class Builder {
|
|
private Class baseDaoClass;
|
|
private Class entityClass;
|
|
private String daoPackageName;
|
|
private String daoName;
|
|
|
public Builder setBaseDaoClass(Class clazz) {
|
this.baseDaoClass = clazz;
|
return this;
|
}
|
|
public Builder setEntityClass(Class clazz) {
|
this.entityClass = clazz;
|
return this;
|
}
|
|
public Builder setDaoPackageName(String daoPackageName) {
|
this.daoPackageName = daoPackageName;
|
return this;
|
}
|
|
public Builder setDaoName(String daoName) {
|
this.daoName = daoName;
|
return this;
|
}
|
|
|
public MongoDBDaoData create() {
|
|
if (daoName == null) {
|
daoName = entityClass.getSimpleName() + "Dao";
|
}
|
|
Field[] fields = entityClass.getDeclaredFields();
|
String identityColumnName = null;
|
String identityColumnType = null;
|
List<ColumnData> columlist = new ArrayList<>();
|
List<DaoQueryColumnData> daoQueryColumnList = new ArrayList<>();
|
for (Field fd : fields) {
|
//private/public 非final，static属性才会参与解析
|
if (fd.getModifiers() != 1 && fd.getModifiers() != 2) {
|
continue;
|
}
|
|
String columName = fd.getName();
|
String columnDataType = fd.getGenericType().getTypeName();
|
Annotation[] as = fd.getAnnotations();
|
|
|
if (columName != null) {
|
for (Annotation a : as) {
|
//主键
|
if (a instanceof Id) {
|
identityColumnName = new String(columName);
|
identityColumnType = columnDataType;
|
columName = null;
|
break;
|
}
|
}
|
}
|
|
for (Annotation a : as) {
|
//不参与解析
|
if (a instanceof Transient || a instanceof java.beans.Transient) {
|
columName = null;
|
break;
|
}
|
}
|
|
//获取查询条件
|
for (Annotation a : as) {
|
if (a instanceof DaoQueryCondition) {
|
daoQueryColumnList.add(new DaoQueryColumnData(columName, columnDataType, ((DaoQueryCondition) a).queryType()));
|
break;
|
}
|
}
|
|
if (columName != null) {
|
columlist.add(new ColumnData(columName, columnDataType));
|
}
|
}
|
|
List<ColumnData> columnDataList = new ArrayList<>();
|
for (ColumnData column : columlist) {
|
//更新时间不参与更新
|
if (!"updateTime".equalsIgnoreCase(column.getColumnName())) {
|
columnDataList.add(column);
|
}
|
}
|
|
Set<String> importClasses = new HashSet<>();
|
for (ColumnData columnData : columnDataList) {
|
if (columnData.getType().indexOf(".") > -1) {
|
importClasses.add(columnData.getType());
|
columnData.setType(columnData.getType().split("\\.")[columnData.getType().split("\\.").length - 1]);
|
}
|
}
|
|
importClasses.add("java.util.List");
|
importClasses.add("java.util.ArrayList");
|
|
|
MongoDBDaoData data = new MongoDBDaoData();
|
data.setBaseDaoClassName(baseDaoClass.getSimpleName());
|
data.setColumnDataList(columnDataList);
|
data.setDaoQueryColumnDataList(daoQueryColumnList);
|
data.setDaoName(daoName);
|
data.setEntityClass(entityClass.getName());
|
data.setEntityClassName(entityClass.getSimpleName());
|
data.setIdentityColumn(new ColumnData(identityColumnName, identityColumnType));
|
data.setPackageName(daoPackageName);
|
|
importClasses.add(baseDaoClass.getName());
|
importClasses.add(entityClass.getName());
|
data.setImportClasses(importClasses);
|
|
return data;
|
}
|
}
|
|
|
public String getDaoName() {
|
return daoName;
|
}
|
|
public void setDaoName(String daoName) {
|
this.daoName = daoName;
|
}
|
|
public String getBaseDaoClassName() {
|
return baseDaoClassName;
|
}
|
|
public void setBaseDaoClassName(String baseDaoClassName) {
|
this.baseDaoClassName = baseDaoClassName;
|
}
|
|
public String getEntityClassName() {
|
return entityClassName;
|
}
|
|
public void setEntityClassName(String entityClassName) {
|
this.entityClassName = entityClassName;
|
}
|
|
|
public ColumnData getIdentityColumn() {
|
return identityColumn;
|
}
|
|
public void setIdentityColumn(ColumnData identityColumn) {
|
this.identityColumn = identityColumn;
|
}
|
|
|
public List<ColumnData> getColumnDataList() {
|
return columnDataList;
|
}
|
|
public void setColumnDataList(List<ColumnData> columnDataList) {
|
this.columnDataList = columnDataList;
|
}
|
|
public Set<String> getImportClasses() {
|
return importClasses;
|
}
|
|
public void setImportClasses(Set<String> importClasses) {
|
this.importClasses = importClasses;
|
}
|
|
|
public List<DaoQueryColumnData> getDaoQueryColumnDataList() {
|
return daoQueryColumnDataList;
|
}
|
|
public void setDaoQueryColumnDataList(List<DaoQueryColumnData> daoQueryColumnDataList) {
|
this.daoQueryColumnDataList = daoQueryColumnDataList;
|
}
|
}
|
