// Copyright (c) 2003-2014, Jodd Team (jodd.org). All Rights Reserved.
package jodd.db.oom.sqlgen.chunks;
import jodd.db.oom.DbEntityColumnDescriptor;
import jodd.db.oom.DbEntityDescriptor;
import jodd.db.oom.sqlgen.DbSqlBuilderException;
import jodd.util.StringPool;
/**
* Resolves column and table references. Reference is given in format: <code>tableRef.propertyName</code>.
* The <code>propertyName</code> may be '+' (e.g. <code>tableRef.+</code>), indicating the identity columns.
* If property name is omitted (e.g. <code>tableRef</code>), only table name will be rendered.
* If table reference is omitted (e.g. <code>.propertyName</code>), only property i.e. column name
* will be rendered.
*/
public class ReferenceChunk extends SqlChunk {
protected final String tableRef;
protected final String columnRef;
protected final boolean onlyId;
public ReferenceChunk(String tableRef, String columnRef) {
this(tableRef, columnRef, false);
}
public ReferenceChunk(String tableRef, String columnRef, boolean onlyId) {
super(CHUNK_REFERENCE);
this.tableRef = tableRef;
this.columnRef = columnRef;
this.onlyId = onlyId;
}
public ReferenceChunk(String reference) {
super(CHUNK_REFERENCE);
int dotNdx = reference.indexOf('.');
if (dotNdx == -1) {
this.tableRef = reference;
this.columnRef = null;
this.onlyId = false;
} else {
String ref = reference.substring(0, dotNdx);
if (ref.length() == 0) {
ref = null;
}
this.tableRef = ref;
ref = reference.substring(dotNdx + 1);
if (ref.length() == 0) {
ref = null;
}
this.columnRef = ref;
onlyId = columnRef != null && columnRef.equals(StringPool.PLUS);
}
}
// ---------------------------------------------------------------- process
@Override
public void process(StringBuilder out) {
DbEntityDescriptor ded;
if (tableRef != null) {
ded = lookupTableRef(tableRef);
String tableName = resolveTable(tableRef, ded);
out.append(tableName);
} else {
ded = findColumnRef(columnRef);
}
if (onlyId == true) {
if (tableRef != null) {
out.append('.');
}
out.append(ded.getIdColumnName());
} else if (columnRef != null) {
DbEntityColumnDescriptor dec = ded.findByPropertyName(columnRef);
templateData.lastColumnDec = dec;
String column = dec == null ? null : dec.getColumnName();
//String column = ded.getColumnName(columnRef);
if (column == null) {
throw new DbSqlBuilderException("Invalid column reference: " + tableRef + '.' + columnRef);
}
if (tableRef != null) {
out.append('.');
}
out.append(column);
}
}
}