Пример #1
0
 public static String[][] generateColumnNames(Type[] types, SessionFactoryImplementor f)
     throws MappingException {
   String[][] columnNames = new String[types.length][];
   for (int i = 0; i < types.length; i++) {
     int span = types[i].getColumnSpan(f);
     columnNames[i] = new String[span];
     for (int j = 0; j < span; j++) {
       columnNames[i][j] = NameGenerator.scalarName(i, j);
     }
   }
   return columnNames;
 }
 /** Generates the scalar column AST nodes for a given array of SQL columns */
 public static void generateScalarColumns(HqlSqlWalkerNode node, String sqlColumns[], int i) {
   if (sqlColumns.length == 1) {
     generateSingleScalarColumn(node, i);
   } else {
     ASTFactory factory = node.getASTFactory();
     AST n = node;
     n.setText(sqlColumns[0]); // Use the DOT node to emit the first column name.
     // Create the column names, folled by the column aliases.
     for (int j = 0; j < sqlColumns.length; j++) {
       if (j > 0) {
         n = ASTUtil.createSibling(factory, SqlTokenTypes.SQL_TOKEN, sqlColumns[j], n);
       }
       n =
           ASTUtil.createSibling(
               factory, SqlTokenTypes.SELECT_COLUMNS, " as " + NameGenerator.scalarName(i, j), n);
     }
   }
 }
Пример #3
0
 public String generateAlias(String sqlExpression) {
   return NameGenerator.scalarName(base, counter++);
 }
  /** WARNING: side-effecty */
  private String renderScalarSelect() {

    boolean isSubselect = superQuery != null;

    StringBuffer buf = new StringBuffer(20);

    if (scalarTypes.size() == 0) {
      // ie. no select clause
      int size = returnedTypes.size();
      for (int k = 0; k < size; k++) {

        scalarTypes.add(TypeFactory.manyToOne(persisters[k].getEntityName(), shallowQuery));

        String[] idColumnNames = persisters[k].getIdentifierColumnNames();
        for (int i = 0; i < idColumnNames.length; i++) {
          buf.append(returnedTypes.get(k)).append('.').append(idColumnNames[i]);
          if (!isSubselect) buf.append(" as ").append(NameGenerator.scalarName(k, i));
          if (i != idColumnNames.length - 1 || k != size - 1) buf.append(", ");
        }
      }

    } else {
      // there _was_ a select clause
      Iterator iter = scalarSelectTokens.iterator();
      int c = 0;
      boolean nolast = false; // real hacky...
      int parenCount = 0; // used to count the nesting of parentheses
      while (iter.hasNext()) {
        Object next = iter.next();
        if (next instanceof String) {
          String token = (String) next;

          if ("(".equals(token)) {
            parenCount++;
          } else if (")".equals(token)) {
            parenCount--;
          }

          String lc = token.toLowerCase();
          if (lc.equals(", ")) {
            if (nolast) {
              nolast = false;
            } else {
              if (!isSubselect && parenCount == 0) {
                int x = c++;
                buf.append(" as ").append(NameGenerator.scalarName(x, 0));
              }
            }
          }
          buf.append(token);
          if (lc.equals("distinct") || lc.equals("all")) {
            buf.append(' ');
          }
        } else {
          nolast = true;
          String[] tokens = (String[]) next;
          for (int i = 0; i < tokens.length; i++) {
            buf.append(tokens[i]);
            if (!isSubselect) {
              buf.append(" as ").append(NameGenerator.scalarName(c, i));
            }
            if (i != tokens.length - 1) buf.append(", ");
          }
          c++;
        }
      }
      if (!isSubselect && !nolast) {
        int x = c++;
        buf.append(" as ").append(NameGenerator.scalarName(x, 0));
      }
    }

    return buf.toString();
  }
 static void generateSingleScalarColumn(HqlSqlWalkerNode node, int i) {
   ASTFactory factory = node.getASTFactory();
   ASTUtil.createSibling(
       factory, SqlTokenTypes.SELECT_COLUMNS, " as " + NameGenerator.scalarName(i, 0), node);
 }