package com.todoroo.andlib.sql; import static com.todoroo.andlib.sql.SqlConstants.ALL; import static com.todoroo.andlib.sql.SqlConstants.COMMA; import static com.todoroo.andlib.sql.SqlConstants.FROM; import static com.todoroo.andlib.sql.SqlConstants.GROUP_BY; import static com.todoroo.andlib.sql.SqlConstants.LEFT_PARENTHESIS; import static com.todoroo.andlib.sql.SqlConstants.LIMIT; import static com.todoroo.andlib.sql.SqlConstants.ORDER_BY; import static com.todoroo.andlib.sql.SqlConstants.RIGHT_PARENTHESIS; import static com.todoroo.andlib.sql.SqlConstants.SELECT; import static com.todoroo.andlib.sql.SqlConstants.SPACE; import static com.todoroo.andlib.sql.SqlConstants.WHERE; import static com.todoroo.andlib.sql.SqlTable.table; import static java.util.Arrays.asList; import java.util.ArrayList; import com.todoroo.andlib.data.Property; public final class Query { private SqlTable table; private String queryTemplate = null; private final ArrayList criterions = new ArrayList(); private final ArrayList fields = new ArrayList(); private final ArrayList joins = new ArrayList(); private final ArrayList groupBies = new ArrayList(); private final ArrayList orders = new ArrayList(); private final ArrayList havings = new ArrayList(); private int limits = -1; private Query(Field... fields) { this.fields.addAll(asList(fields)); } public static Query select(Field... fields) { return new Query(fields); } public Query from(SqlTable fromTable) { this.table = fromTable; return this; } public Query join(Join... join) { joins.addAll(asList(join)); return this; } public Query where(Criterion criterion) { criterions.add(criterion); return this; } public Query groupBy(Field... groupBy) { groupBies.addAll(asList(groupBy)); return this; } public Query orderBy(Order... order) { orders.addAll(asList(order)); return this; } public Query limit(int limit) { limits = limit; return this; } public Query appendSelectFields(Property... selectFields) { this.fields.addAll(asList(selectFields)); return this; } @Override public boolean equals(Object o) { return this == o || !(o == null || getClass() != o.getClass()) && this.toString().equals(o.toString()); } @Override public int hashCode() { return toString().hashCode(); } @Override public String toString() { StringBuilder sql = new StringBuilder(); visitSelectClause(sql); visitFromClause(sql); visitJoinClause(sql); if(queryTemplate == null) { visitWhereClause(sql); visitGroupByClause(sql); visitOrderByClause(sql); visitLimitClause(sql); } else { if(groupBies.size() > 0 || orders.size() > 0 || havings.size() > 0) throw new IllegalStateException("Can't have extras AND query template"); //$NON-NLS-1$ sql.append(queryTemplate); } return sql.toString(); } private void visitOrderByClause(StringBuilder sql) { if (orders.isEmpty()) { return; } sql.append(ORDER_BY); for (Order order : orders) { sql.append(SPACE).append(order).append(COMMA); } sql.deleteCharAt(sql.length() - 1).append(SPACE); } @SuppressWarnings("nls") private void visitGroupByClause(StringBuilder sql) { if (groupBies.isEmpty()) { return; } sql.append(GROUP_BY); for (Field groupBy : groupBies) { sql.append(SPACE).append(groupBy).append(COMMA); } sql.deleteCharAt(sql.length() - 1).append(SPACE); if (havings.isEmpty()) { return; } sql.append("HAVING"); for (Criterion havingCriterion : havings) { sql.append(SPACE).append(havingCriterion).append(COMMA); } sql.deleteCharAt(sql.length() - 1).append(SPACE); } private void visitWhereClause(StringBuilder sql) { if (criterions.isEmpty()) { return; } sql.append(WHERE); for (Criterion criterion : criterions) { sql.append(SPACE).append(criterion).append(SPACE); } } private void visitJoinClause(StringBuilder sql) { for (Join join : joins) { sql.append(join).append(SPACE); } } private void visitFromClause(StringBuilder sql) { if (table == null) { return; } sql.append(FROM).append(SPACE).append(table).append(SPACE); } private void visitSelectClause(StringBuilder sql) { sql.append(SELECT).append(SPACE); if (fields.isEmpty()) { sql.append(ALL).append(SPACE); return; } for (Field field : fields) { sql.append(field.toStringInSelect()).append(COMMA); } sql.deleteCharAt(sql.length() - 1).append(SPACE); } private void visitLimitClause(StringBuilder sql) { if(limits > -1) sql.append(LIMIT).append(SPACE).append(limits).append(SPACE); } public SqlTable as(String alias) { return table(LEFT_PARENTHESIS + this.toString() + RIGHT_PARENTHESIS).as(alias); } public Query having(Criterion criterion) { this.havings.add(criterion); return this; } /** * Gets a list of fields returned by this query * @return */ public Property[] getFields() { return fields.toArray(new Property[fields.size()]); } /** * Add the SQL query template (comes after the "from") * @param sqlQuery * @return */ public Query withQueryTemplate(String template) { queryTemplate = template; return this; } }