Beispiel #1
0
  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;
  }
Beispiel #2
0
  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);
      }
    }
  }