/**
* 解析数据关联关系
*
* @param list
* @param fieldName
* @param keyMethod
* @param table
* @param refKey
* @param ids
* @throws NoSuchFieldException
* @throws NoSuchMethodException
* @throws IllegalAccessException
* @throws InvocationTargetException
*/
private void parseReference(
List<T> list,
String fieldName,
Method keyMethod,
String table,
String refKey,
StringBuffer ids)
throws NoSuchFieldException, NoSuchMethodException, IllegalAccessException,
InvocationTargetException {
Coupler<Entity> childCoupler = childCouplers.get(fieldName);
Class<Entity> childClass = childCoupler.getClazz();
TreeDao<Entity> childDao = DaoFactory.getTreeDao(childClass);
String childSql = buildChildQuerySql(table, refKey, ids, ordersMap, fieldName);
if (entityClass == childClass) {
int count = dao.count(childSql);
if (count == 0) {
return;
}
}
List<Entity> childEntities = childDao.queryForTree(childSql);
Class<?> fileType = entityClass.getDeclaredField(fieldName).getType();
if (fileType == List.class) {
setListValue(list, keyMethod, fieldName, childEntities, childClass, refKey);
} else {
setSingleValue(list, keyMethod, fieldName, childEntities, childClass, fileType, refKey);
}
}
public SimpleTreeDao(Class<T> clazz) {
this.entityClass = clazz;
dao = DaoFactory.getDao(entityClass);
treeCoupler = CouplerFactory.getTreeCoupler(entityClass);
coupler = treeCoupler.getCoupler();
childCouplers = treeCoupler.getChildCouplers();
oneToOneMap = coupler.getOneToOneMap();
oneToManyMap = coupler.getOneToManyMap();
manyToOneMap = coupler.getManyToOneMap();
ordersMap = coupler.getOrdersMap();
}
/**
* 树形结构的数据查询
*
* @param sql
* @param values
* @return
*/
@Override
public List<T> queryForTree(String sql, Object... values) {
List<T> list = dao.find(sql, values);
if (list == null || list.isEmpty()) {
return list;
}
if (childCouplers.isEmpty()) {
return list;
}
try {
Method majorMethod = entityClass.getDeclaredMethod(coupler.getMajorFieldGetterName());
StringBuffer ids = new StringBuffer();
int i = 0;
for (Entity entity : list) {
ids.append(majorMethod.invoke(entity));
if (i++ < list.size() - 1) {
ids.append(",");
}
}
for (String fieldName : oneToOneMap.keySet()) {
OneToOne oneToOne = oneToOneMap.get(fieldName);
String table = oneToOne.table();
String refKey = oneToOne.foreignKey();
parseReference(list, fieldName, majorMethod, table, refKey, ids);
}
for (String fieldName : oneToManyMap.keySet()) {
OneToMany oneToMany = oneToManyMap.get(fieldName);
String table = oneToMany.table();
String refKey = oneToMany.foreignKey();
parseReference(list, fieldName, majorMethod, table, refKey, ids);
}
for (String fieldName : manyToOneMap.keySet()) {
ManyToOne manyToOne = manyToOneMap.get(fieldName);
String table = manyToOne.table();
String refKey = manyToOne.referenceKey();
Method keyMethod =
entityClass.getDeclaredMethod("get".concat(StringUtils.firstToUpperCase(refKey)));
StringBuffer refIds = new StringBuffer();
int j = 0;
for (Entity entity : list) {
refIds.append(keyMethod.invoke(entity));
if (j++ < list.size() - 1) {
refIds.append(",");
}
}
parseReference(list, fieldName, keyMethod, table, refKey, refIds);
}
} catch (NoSuchMethodException e) {
log.error("SimpleTreeDao.queryForTree throws NoSuchMethodException", e);
} catch (InvocationTargetException e) {
log.error("SimpleTreeDao.queryForTree throws InvocationTargetException", e);
} catch (IllegalAccessException e) {
log.error("SimpleTreeDao.queryForTree throws IllegalAccessException", e);
} catch (NoSuchFieldException e) {
log.error("SimpleTreeDao.queryForTree throws NoSuchFieldException", e);
}
return list;
}