package com.avaje.ebeaninternal.server.deploy;
import javax.persistence.PersistenceException;
import com.avaje.ebeaninternal.server.lib.util.StringHelper;
public class BeanSqlSelect {
public static final String HAVING_PREDICATES = "${HAVING_PREDICATES}";
public static final String WHERE_PREDICATES = "${WHERE_PREDICATES}";
public static final String AND_PREDICATES = "${AND_PREDICATES}";
public static final String ORDER_BY = "${ORDER_BY}";
public enum PredicatesType {
HAVING, WHERE, AND, NONE
}
final String sql;
final PredicatesType predicatesType;
final boolean hasOrderBy;
public BeanSqlSelect(String sql, PredicatesType predicatesType, boolean hasOrderBy) {
this.sql = sql;
this.predicatesType = predicatesType;
this.hasOrderBy = hasOrderBy;
}
public String getSql() {
return sql;
}
public PredicatesType getPredicatesType() {
return predicatesType;
}
public boolean hasOrderBy() {
return hasOrderBy;
}
public String addPredicates(String query, String predicates) {
if (predicates == null) {
switch (predicatesType) {
case HAVING:
return StringHelper.replaceString(query, HAVING_PREDICATES, "");
case WHERE:
return StringHelper.replaceString(query, WHERE_PREDICATES, "");
case AND:
return StringHelper.replaceString(query, AND_PREDICATES, "");
case NONE:
return query;
default:
throw new PersistenceException("predicatesType " + predicatesType + " not handled");
}
} else {
switch (predicatesType) {
case HAVING:
return StringHelper
.replaceString(query, HAVING_PREDICATES, " HAVING " + predicates);
case WHERE:
return StringHelper.replaceString(query, WHERE_PREDICATES, " WHERE " + predicates);
case AND:
return StringHelper.replaceString(query, AND_PREDICATES, " AND " + predicates);
case NONE:
return query;
default:
throw new PersistenceException("predicatesType " + predicatesType + " not handled");
}
}
}
public String addOrderBy(String query, String orderBy) {
if (!hasOrderBy) {
return query;
}
if (orderBy == null) {
return StringHelper.replaceString(query, ORDER_BY, "");
} else {
return StringHelper.replaceString(query, ORDER_BY, " ORDER BY " + orderBy);
}
}
}