Added new DbColumn class that encapsulates column name, SQL type and the
relation associated with the column. This helps us fix a bug where no column info is available because a column has no relation associated with it. It also helps us streamline the Relation lookup when creating a Node from a ResultSet. Use StringBuffers instead of + for query string composition where possible.
This commit is contained in:
parent
3d7af041bd
commit
f2bfe5b9cf
4 changed files with 176 additions and 63 deletions
46
src/helma/objectmodel/db/DbColumn.java
Normal file
46
src/helma/objectmodel/db/DbColumn.java
Normal file
|
@ -0,0 +1,46 @@
|
|||
// DbColumn.java
|
||||
// Copyright 2002 Hannes Wallnoefer, Helma.org
|
||||
|
||||
package helma.objectmodel.db;
|
||||
|
||||
/**
|
||||
* A class that encapsulates the Column name and data type of a
|
||||
* column in a relational table.
|
||||
*/
|
||||
public final class DbColumn {
|
||||
|
||||
private final String name;
|
||||
private final int type;
|
||||
private final Relation relation;
|
||||
|
||||
/**
|
||||
* Constructor
|
||||
*/
|
||||
public DbColumn (String name, int type, Relation rel) {
|
||||
this.name = name;
|
||||
this.type = type;
|
||||
this.relation = rel;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the column name.
|
||||
*/
|
||||
public String getName() {
|
||||
return name;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get this columns SQL data type.
|
||||
*/
|
||||
public int getType() {
|
||||
return type;
|
||||
}
|
||||
|
||||
/**
|
||||
* Return the relation associated with this column. May be null.
|
||||
*/
|
||||
public Relation getRelation() {
|
||||
return relation;
|
||||
}
|
||||
|
||||
}
|
|
@ -43,15 +43,21 @@ public final class DbMapping implements Updatable {
|
|||
Relation subRelation;
|
||||
Relation propRelation;
|
||||
|
||||
// if this defines a subnode mapping with groupby layer, we need a DbMapping for those groupby nodes
|
||||
// if this defines a subnode mapping with groupby layer,
|
||||
// we need a DbMapping for those groupby nodes
|
||||
DbMapping groupbyMapping;
|
||||
|
||||
// Map of property names to Relations objects
|
||||
HashMap prop2db;
|
||||
// Map of db columns to Relations objects
|
||||
// Map of db columns to Relations objects.
|
||||
// Case insensitive, keys are stored in upper case so
|
||||
// lookups must do a toUpperCase().
|
||||
HashMap db2prop;
|
||||
|
||||
// list of columns to fetch from db
|
||||
String[] columns = null;
|
||||
DbColumn[] columns = null;
|
||||
// Map of db columns by name
|
||||
HashMap columnMap;
|
||||
// pre-rendered select statement
|
||||
String select = null;
|
||||
|
||||
|
@ -116,6 +122,8 @@ public final class DbMapping implements Updatable {
|
|||
prop2db = new HashMap ();
|
||||
db2prop = new HashMap ();
|
||||
|
||||
columnMap = new HashMap ();
|
||||
|
||||
parent = null;
|
||||
|
||||
idField = null;
|
||||
|
@ -185,6 +193,7 @@ public final class DbMapping implements Updatable {
|
|||
keydef = null;
|
||||
// same with columns and select string
|
||||
columns = null;
|
||||
columnMap.clear();
|
||||
select = null;
|
||||
|
||||
|
||||
|
@ -567,11 +576,11 @@ public final class DbMapping implements Updatable {
|
|||
|
||||
|
||||
/**
|
||||
* Return a Village Schema object for this DbMapping.
|
||||
* Return an array of DbColumns for the relational table mapped by this DbMapping.
|
||||
*/
|
||||
public synchronized String[] getColumns() throws ClassNotFoundException, SQLException {
|
||||
public synchronized DbColumn[] getColumns() throws ClassNotFoundException, SQLException {
|
||||
if (!isRelational ())
|
||||
throw new SQLException ("Can't get Schema for non-relational data mapping");
|
||||
throw new SQLException ("Can't get columns for non-relational data mapping "+this);
|
||||
if (source == null && parentMapping != null)
|
||||
return parentMapping.getColumns ();
|
||||
// Use local variable cols to avoid synchronization (schema may be nulled elsewhere)
|
||||
|
@ -580,25 +589,46 @@ public final class DbMapping implements Updatable {
|
|||
// and build a string of column names.
|
||||
Connection con = getConnection ();
|
||||
Statement stmt = con.createStatement ();
|
||||
ResultSet rs = stmt.executeQuery ("select * from "+getTableName()+" where 1 = 0");
|
||||
String t = getTableName();
|
||||
if (t == null)
|
||||
throw new SQLException ("Table name is null in getColumns() for "+this);
|
||||
ResultSet rs = stmt.executeQuery (
|
||||
new StringBuffer("SELECT * FROM ")
|
||||
.append(t).append(" WHERE 1 = 0").toString());
|
||||
if (rs == null)
|
||||
throw new SQLException ("Error retrieving DB scheme for "+this);
|
||||
throw new SQLException ("Error retrieving columns for "+this);
|
||||
ResultSetMetaData meta = rs.getMetaData ();
|
||||
// ok, we have the meta data, now loop through mapping...
|
||||
int ncols = meta.getColumnCount ();
|
||||
columns = new String[ncols];
|
||||
columns = new DbColumn[ncols];
|
||||
for (int i=0; i<ncols; i++) {
|
||||
columns[i] = meta.getColumnName (i+1);
|
||||
Relation rel = columnNameToRelation (columns[i]);
|
||||
if (rel == null || (rel.reftype != Relation.PRIMITIVE &&
|
||||
rel.reftype != Relation.REFERENCE))
|
||||
continue;
|
||||
rel.setColumnType (meta.getColumnType (i+1));
|
||||
String colName = meta.getColumnName (i+1);
|
||||
Relation rel = columnNameToRelation (colName);
|
||||
columns[i] = new DbColumn (colName, meta.getColumnType (i+1), rel);
|
||||
}
|
||||
}
|
||||
return columns;
|
||||
}
|
||||
|
||||
public DbColumn getColumn (String columnName) throws ClassNotFoundException, SQLException {
|
||||
DbColumn col = (DbColumn) columnMap.get(columnName);
|
||||
if (col == null) {
|
||||
DbColumn[] cols = columns;
|
||||
if (cols == null)
|
||||
cols = getColumns();
|
||||
for (int i=0; i<cols.length; i++) {
|
||||
if (columnName.equalsIgnoreCase (cols[i].getName())) {
|
||||
col = cols[i];
|
||||
break;
|
||||
}
|
||||
}
|
||||
if (col == null)
|
||||
throw new SQLException ("Column "+columnName+" not found in "+this);
|
||||
columnMap.put (columnName, col);
|
||||
}
|
||||
return col;
|
||||
}
|
||||
|
||||
public StringBuffer getSelect () throws SQLException, ClassNotFoundException {
|
||||
String sel = select;
|
||||
if (sel != null)
|
||||
|
@ -619,13 +649,8 @@ public final class DbMapping implements Updatable {
|
|||
if (table == null && parentMapping != null)
|
||||
return parentMapping.needsQuotes (columnName);
|
||||
try {
|
||||
Relation rel = (Relation) db2prop.get (columnName.toUpperCase());
|
||||
if (rel == null)
|
||||
throw new SQLException ("Error retrieving relational schema for "+this);
|
||||
// make sure columns are initialized and up to date
|
||||
if (columns == null)
|
||||
getColumns();
|
||||
switch (rel.getColumnType()) {
|
||||
DbColumn col = getColumn (columnName);
|
||||
switch (col.getType()) {
|
||||
case Types.CHAR:
|
||||
case Types.VARCHAR:
|
||||
case Types.LONGVARCHAR:
|
||||
|
|
|
@ -234,7 +234,7 @@ public final class Node implements INode, Serializable {
|
|||
/**
|
||||
* Constructor used for nodes being stored in a relational database table.
|
||||
*/
|
||||
public Node (DbMapping dbm, ResultSet rs, String[] columns, WrappedNodeManager nmgr)
|
||||
public Node (DbMapping dbm, ResultSet rs, DbColumn[] columns, WrappedNodeManager nmgr)
|
||||
throws SQLException {
|
||||
|
||||
this.nmgr = nmgr;
|
||||
|
@ -265,35 +265,35 @@ public final class Node implements INode, Serializable {
|
|||
|
||||
for (int i=0; i<columns.length; i++) {
|
||||
|
||||
Relation rel = dbm.columnNameToRelation (columns[i]);
|
||||
Relation rel = columns[i].getRelation();
|
||||
if (rel == null || (rel.reftype != Relation.PRIMITIVE &&
|
||||
rel.reftype != Relation.REFERENCE))
|
||||
continue;
|
||||
|
||||
Property newprop = new Property (rel.propName, this);
|
||||
|
||||
switch (rel.getColumnType()) {
|
||||
switch (columns[i].getType()) {
|
||||
|
||||
case Types.BIT:
|
||||
newprop.setBooleanValue (rs.getBoolean(columns[i]));
|
||||
newprop.setBooleanValue (rs.getBoolean(columns[i].getName()));
|
||||
break;
|
||||
|
||||
case Types.TINYINT:
|
||||
case Types.BIGINT:
|
||||
case Types.SMALLINT:
|
||||
case Types.INTEGER:
|
||||
newprop.setIntegerValue (rs.getLong(columns[i]));
|
||||
newprop.setIntegerValue (rs.getLong(columns[i].getName()));
|
||||
break;
|
||||
|
||||
case Types.REAL:
|
||||
case Types.FLOAT:
|
||||
case Types.DOUBLE:
|
||||
newprop.setFloatValue (rs.getDouble(columns[i]));
|
||||
newprop.setFloatValue (rs.getDouble(columns[i].getName()));
|
||||
break;
|
||||
|
||||
case Types.DECIMAL:
|
||||
case Types.NUMERIC:
|
||||
BigDecimal num = rs.getBigDecimal (columns[i]);
|
||||
BigDecimal num = rs.getBigDecimal (columns[i].getName());
|
||||
if (num == null)
|
||||
break;
|
||||
if (num.scale() > 0)
|
||||
|
@ -305,20 +305,20 @@ public final class Node implements INode, Serializable {
|
|||
case Types.LONGVARBINARY:
|
||||
case Types.VARBINARY:
|
||||
case Types.BINARY:
|
||||
newprop.setStringValue (rs.getString(columns[i]));
|
||||
newprop.setStringValue (rs.getString(columns[i].getName()));
|
||||
break;
|
||||
|
||||
case Types.LONGVARCHAR:
|
||||
case Types.CHAR:
|
||||
case Types.VARCHAR:
|
||||
case Types.OTHER:
|
||||
newprop.setStringValue (rs.getString(columns[i]));
|
||||
newprop.setStringValue (rs.getString(columns[i].getName()));
|
||||
break;
|
||||
|
||||
case Types.DATE:
|
||||
case Types.TIME:
|
||||
case Types.TIMESTAMP:
|
||||
newprop.setDateValue (rs.getTimestamp(columns[i]));
|
||||
newprop.setDateValue (rs.getTimestamp(columns[i].getName()));
|
||||
break;
|
||||
|
||||
case Types.NULL:
|
||||
|
@ -327,7 +327,7 @@ public final class Node implements INode, Serializable {
|
|||
// continue;
|
||||
|
||||
default:
|
||||
newprop.setStringValue (rs.getString(columns[i]));
|
||||
newprop.setStringValue (rs.getString(columns[i].getName()));
|
||||
break;
|
||||
}
|
||||
|
||||
|
|
|
@ -580,7 +580,13 @@ public final class NodeManager {
|
|||
try {
|
||||
Connection con = dbm.getConnection ();
|
||||
st = con.createStatement ();
|
||||
st.executeUpdate ("DELETE FROM "+dbm.getTableName ()+" WHERE "+dbm.getIDField ()+" = "+node.getID ());
|
||||
st.executeUpdate (new StringBuffer ("DELETE FROM ")
|
||||
.append(dbm.getTableName ())
|
||||
.append(" WHERE ")
|
||||
.append(dbm.getIDField())
|
||||
.append(" = ")
|
||||
.append(node.getID())
|
||||
.toString());
|
||||
} finally {
|
||||
if (st != null) try {
|
||||
st.close ();
|
||||
|
@ -605,7 +611,11 @@ public final class NodeManager {
|
|||
Statement stmt = null;
|
||||
try {
|
||||
Connection con = map.getConnection ();
|
||||
String q = "SELECT MAX("+map.getIDField()+") FROM "+map.getTableName();
|
||||
String q = new StringBuffer("SELECT MAX(")
|
||||
.append(map.getIDField())
|
||||
.append(") FROM ")
|
||||
.append(map.getTableName())
|
||||
.toString();
|
||||
stmt = con.createStatement ();
|
||||
ResultSet rs = stmt.executeQuery (q);
|
||||
// check for empty table
|
||||
|
@ -639,7 +649,10 @@ public final class NodeManager {
|
|||
String retval = null;
|
||||
try {
|
||||
Connection con = map.getConnection ();
|
||||
String q = "SELECT "+map.getIDgen()+".nextval FROM dual";
|
||||
String q = new StringBuffer("SELECT ")
|
||||
.append(map.getIDgen())
|
||||
.append(".nextval FROM dual")
|
||||
.toString();
|
||||
stmt = con.createStatement();
|
||||
ResultSet rs = stmt.executeQuery (q);
|
||||
if (!rs.next ())
|
||||
|
@ -682,10 +695,23 @@ public final class NodeManager {
|
|||
|
||||
if (home.getSubnodeRelation() != null) {
|
||||
// subnode relation was explicitly set
|
||||
q = "SELECT "+idfield+" FROM "+table+" "+home.getSubnodeRelation();
|
||||
q = new StringBuffer("SELECT ")
|
||||
.append(idfield)
|
||||
.append(" FROM ")
|
||||
.append(table)
|
||||
.append(" ")
|
||||
.append(home.getSubnodeRelation())
|
||||
.toString();
|
||||
} else {
|
||||
// let relation object build the query
|
||||
q = "SELECT "+idfield+" FROM "+table + rel.buildQuery (home, home.getNonVirtualParent (), null, " WHERE ", true);
|
||||
q = new StringBuffer("SELECT ")
|
||||
.append(idfield)
|
||||
.append(" FROM ")
|
||||
.append(table)
|
||||
.append(rel.buildQuery (home,
|
||||
home.getNonVirtualParent (), null,
|
||||
" WHERE ", true))
|
||||
.toString();
|
||||
}
|
||||
|
||||
if (logSql)
|
||||
|
@ -750,7 +776,7 @@ public final class NodeManager {
|
|||
|
||||
Connection con = dbm.getConnection ();
|
||||
Statement stmt = con.createStatement ();
|
||||
String[] columns = dbm.getColumns ();
|
||||
DbColumn[] columns = dbm.getColumns ();
|
||||
StringBuffer q = dbm.getSelect ();
|
||||
try {
|
||||
if (home.getSubnodeRelation() != null) {
|
||||
|
@ -808,7 +834,7 @@ public final class NodeManager {
|
|||
if (missing > 0) {
|
||||
Connection con = dbm.getConnection ();
|
||||
Statement stmt = con.createStatement ();
|
||||
String[] columns = dbm.getColumns ();
|
||||
DbColumn[] columns = dbm.getColumns ();
|
||||
StringBuffer q = dbm.getSelect ();
|
||||
try {
|
||||
String idfield = rel.groupby != null ? rel.groupby : dbm.getIDField ();
|
||||
|
@ -835,8 +861,10 @@ public final class NodeManager {
|
|||
q.append (") ");
|
||||
if (rel.groupby != null) {
|
||||
q.append (rel.renderConstraints (home, home.getNonVirtualParent ()));
|
||||
if (rel.order != null)
|
||||
q.append (" ORDER BY "+rel.order);
|
||||
if (rel.order != null) {
|
||||
q.append (" ORDER BY ");
|
||||
q.append (rel.order);
|
||||
}
|
||||
}
|
||||
|
||||
if (logSql)
|
||||
|
@ -941,10 +969,18 @@ public final class NodeManager {
|
|||
String q = null;
|
||||
if (home.getSubnodeRelation() != null) {
|
||||
// use the manually set subnoderelation of the home node
|
||||
q = "SELECT count(*) FROM "+table+" "+home.getSubnodeRelation();
|
||||
q = new StringBuffer("SELECT count(*) FROM ")
|
||||
.append(table)
|
||||
.append(" ")
|
||||
.append(home.getSubnodeRelation())
|
||||
.toString();
|
||||
} else {
|
||||
// let relation object build the query
|
||||
q = "SELECT count(*) FROM "+table + rel.buildQuery (home, home.getNonVirtualParent (), null, " WHERE ", false);
|
||||
q = new StringBuffer("SELECT count(*) FROM ")
|
||||
.append(table)
|
||||
.append(rel.buildQuery (home, home.getNonVirtualParent(),
|
||||
null, " WHERE ", false))
|
||||
.toString();
|
||||
}
|
||||
|
||||
if (logSql)
|
||||
|
@ -990,7 +1026,13 @@ public final class NodeManager {
|
|||
Statement stmt = null;
|
||||
|
||||
try {
|
||||
String q = "SELECT "+namefield+" FROM "+table+" ORDER BY "+namefield;
|
||||
String q = new StringBuffer("SELECT ")
|
||||
.append(namefield)
|
||||
.append(" FROM ")
|
||||
.append(table)
|
||||
.append(" ORDER BY ")
|
||||
.append(namefield)
|
||||
.toString();
|
||||
stmt = con.createStatement ();
|
||||
|
||||
if (logSql)
|
||||
|
@ -1037,7 +1079,7 @@ public final class NodeManager {
|
|||
Connection con = dbm.getConnection ();
|
||||
stmt = con.createStatement ();
|
||||
|
||||
String[] columns = dbm.getColumns ();
|
||||
DbColumn[] columns = dbm.getColumns ();
|
||||
StringBuffer q = dbm.getSelect ();
|
||||
q.append ("WHERE ");
|
||||
q.append (idfield);
|
||||
|
@ -1099,7 +1141,7 @@ public final class NodeManager {
|
|||
DbMapping dbm = rel.otherType;
|
||||
|
||||
Connection con = dbm.getConnection ();
|
||||
String[] columns = dbm.getColumns ();
|
||||
DbColumn[] columns = dbm.getColumns ();
|
||||
StringBuffer q = dbm.getSelect ();
|
||||
if (home.getSubnodeRelation () != null) {
|
||||
// combine our key with the constraints in the manually set subnode relation
|
||||
|
|
Loading…
Add table
Reference in a new issue