public DAO setClass(Class<?> clazz) {
this.clazz = clazz;
if (Map.class.isAssignableFrom(clazz)) {
if (this.map != null) {
selectAllColumn = ORMConfigBeanUtil.getSelectAllColumn(map);
table = (String) map.get("table");
}
} else {
this.table = ORMConfigBeanUtil.getTable(clazz);
selectAllColumn = ORMConfigBeanUtil.getSelectAllColumn(clazz);
}
if (selectAllColumn == null || selectAllColumn.trim().length() == 0) selectAllColumn = "*";
return this;
}
private void init(Class<?> clazz, String dsName) {
if (dsName == null) dsName = DAOConfigConstant.MYDBINFO;
this.clazz = clazz;
this.dsName = dsName;
this.ds = DataSourceWrapCache.get(dsName);
dbType = DBInfoConfigBeanCache.get(dsName).getDataBaseType();
if (Map.class.isAssignableFrom(clazz)) {
if (this.map != null) {
selectAllColumn = ORMConfigBeanUtil.getSelectAllColumn(map);
table = (String) map.get("table");
}
} else {
this.table = ORMConfigBeanUtil.getTable(clazz);
selectAllColumn = ORMConfigBeanUtil.getSelectAllColumn(clazz);
}
if (selectAllColumn == null || selectAllColumn.trim().length() == 0) selectAllColumn = "*";
this.buffer.put("orderField", ORMConfigBeanUtil.getIdField(clazz));
this.buffer.put("orderType", OrderType.DESC_ORDER);
}
/**
* 多对多级联查询 1.当主对象没有包含任何一个关联对象时，默认查询所有与之关联的对象 2.当主对象中包含了关联对象时（含有其id值），则只查询这些关联的对象
*
* @param <T>
* @throws InvocationTargetException
* @throws IllegalAccessException
* @throws IllegalArgumentException
*/
public void select() throws DAOException {
if (this.fields == null || this.fields.size() == 0) return;
// select %s from {tarTable} where {referencedColumn} in (select {to} from {relTable} where
// {from} = {idVal})
String format = "SELECT %s FROM %s WHERE %s IN (SELECT %s FROM %s WHERE %s = ?) ";
if (idVal == null || "0".equals(idVal) || "".equals(idVal)) return;
for (Field f : fields) {
Method tarGetter = ru.getGetter(f.getName());
if (tarGetter == null) continue;
ManyToMany ann = tarGetter.getAnnotation(ManyToMany.class);
if (ann == null) {
ann = f.getAnnotation(ManyToMany.class);
if (ann == null) continue;
}
JoinTable join = tarGetter.getAnnotation(JoinTable.class);
if (join == null) {
join = f.getAnnotation(JoinTable.class);
if (join == null) continue;
}
JoinColumn[] froms = join.joinColumns();
if (froms == null || froms.length == 0) continue;
JoinColumn[] tos = join.inverseJoinColumns();
if (tos == null || tos.length == 0) continue;
// 多对多关系目标Class
Class<?> tarClass = ann.targetEntity();
if (void.class.isAssignableFrom(tarClass)) {
tarClass = ClassUtil.getGenericType(f);
}
String tarTable = ORMConfigBeanUtil.getTable(tarClass, true);
// 目标类对应的数据库表Id字段
String referencedColumn = tos[0].referencedColumnName();
if (referencedColumn == null || referencedColumn.trim().length() == 0)
referencedColumn = ORMConfigBeanUtil.getIdColumn(tarClass);
// 目标类在第三方关系表中的字段名
String to = tos[0].name();
// 第三方关系表
String relTable = join.name();
// 主类在第三方关系表中的字段名
String from = froms[0].name();
try {
List<?> tarList = null;
tarList = (List<?>) tarGetter.invoke(t);
if (tarList != null && tarList.size() > 0) {
for (int i = 0; i < tarList.size(); i++) {
Object tarObj = tarList.get(i);
ReflectUtil tarRu = new ReflectUtil(tarObj);
String tarIdField = ORMConfigBeanUtil.getIdField(tarClass);
Method tarIdGetter = tarRu.getGetter(tarIdField);
Object tarIdValObj = tarIdGetter.invoke(tarObj);
if (tarIdValObj == null) continue;
String tarIdVal = String.valueOf(tarIdValObj);
// 查询 select %s from {tarTable} where {tarIdColumn} = {tarIdVal}
tarObj =
DAOFactory.getSelectDAO(dsName)
.selectOne(tarClass, new String[] {tarIdField}, new String[] {tarIdVal});
}
} else {
String sql =
String.format(
format,
ORMConfigBeanUtil.getSelectAllColumn(tarClass),
tarTable,
referencedColumn,
to,
relTable,
from);
// 从数据库中取出与当前主对象id关联的所有目标对象，
tarList = DAOFactory.getSelectDAO(dsName).selectBySQL(tarClass, sql, idVal);
}
// 并注入到当前主对象的属性中
Method tarSetter = ru.getSetter(f.getName());
tarSetter.invoke(t, tarList);
} catch (Exception e) {
e.printStackTrace();
throw new DAOException("", e);
}
}
}
/** 多对一级联查询 1.获取当前idVal，然后作为条件查询出其外键值，接着通过其外键值查出主对象数据，注入到当前 */
public void select() {
if (this.fields == null || this.fields.size() == 0) return;
if (idVal == null || "0".equals(idVal) || "".equals(idVal)) {
return;
}
// else if (DAOFactory.getSelectDAO(dsName).selectOne(t, this.idField) == null) {
// // 检查一下当前对象的ID是否存在于数据库
// return;
// }
for (Field f : fields) {
Method tarGetter = ru.getGetter(f.getName());
if (tarGetter == null) continue;
String fk = null;
OneToOne ann = tarGetter.getAnnotation(OneToOne.class);
if (ann == null) ann = f.getAnnotation(OneToOne.class);
if (ann == null) {
ManyToOne moAn = tarGetter.getAnnotation(ManyToOne.class);
if (moAn == null) moAn = f.getAnnotation(ManyToOne.class);
if (moAn == null) continue;
}
String refCol = null;
JoinColumn joinCol = f.getAnnotation(JoinColumn.class);
if (joinCol == null) joinCol = tarGetter.getAnnotation(JoinColumn.class);
if (joinCol == null) fk = f.getName() + "_id";
else {
fk = joinCol.name();
refCol = joinCol.referencedColumnName();
}
Class<?> tarClass = f.getType();
if (refCol == null || refCol.trim().length() == 0)
refCol = ORMConfigBeanUtil.getIdColumn(tarClass);
String refField = ORMConfigBeanUtil.getField(tarClass, refCol);
try {
Object _tarObj = tarGetter.invoke(t);
Object tarObj = null;
boolean flag = false;
if (_tarObj != null) {
Method refFieldGetter = new ReflectUtil(_tarObj).getGetter(refField);
if (refFieldGetter != null && refFieldGetter.invoke(_tarObj) != null)
tarObj = DAOFactory.getSelectDAO(dsName).selectOne(_tarObj, refField);
else flag = true;
} else flag = true;
if (flag) {
// select * from {tarTable} where {referencedColumn} = (select {fk} from {table} where
// {idColumn} = {idVal})
String format = "select %s from %s where %s = (select %s from %s where %s = %s )";
String tarTable = ORMConfigBeanUtil.getTable(tarClass, true);
String sql =
String.format(
format,
ORMConfigBeanUtil.getSelectAllColumn(tarClass),
tarTable,
refCol,
fk,
table,
idColumn,
idVal);
List<?> tarList = DAOFactory.getSelectDAO(dsName).selectBySQL(tarClass, sql);
if (tarList == null || tarList.size() == 0) continue;
tarObj = tarList.get(0);
}
if (tarObj == null) continue;
Method tarSetter = ru.getSetter(f.getName());
tarSetter.invoke(t, tarObj);
} catch (Exception e) {
throw new DAOException("", e);
}
}
}
