Android Open Source - DKO Sql Context

From Project

Back to project page DKO.
License

The source code is released under:
GNU Lesser General Public License
If you think the Android project DKO listed in this page is inappropriate, such as containing malicious code/tools or violating the copyright, please email info at java2s dot com, thanks.
Java Source Code

package org.kered.dko;
/* w w w . j av  a2  s . com*/
import java.util.ArrayList;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;

import org.kered.dko.Constants.DB_TYPE;
import org.kered.dko.DBQuery.JoinInfo;
import org.kered.dko.TemporaryTableFactory.DummyTableWithName;

class SqlContext {

  private final DBQuery<?> q;

  SqlContext(final DBQuery<?> q) {
    this(q, null);
  }

  public SqlContext(final DBQuery<?> q, final SqlContext parentContext) {
    tableNameMap = q.tableNameMap;
    tableInfos = new ArrayList<TableInfo>(q.tableInfos);
    for (final JoinInfo join : q.joins) tableInfos.add(join.reffedTableInfo);
    for (final JoinInfo join : q.joinsToOne) tableInfos.add(join.reffedTableInfo);
    for (final JoinInfo join : q.joinsToMany) tableInfos.add(join.reffingTableInfo);
    this.parentContext = parentContext;
    dbType = parentContext == null ? q.getDBType() : parentContext.dbType;
    this.q = q;
  }
  
  public SqlContext(DB_TYPE dbType) {
    this.dbType = dbType;
    q = null;
    tableInfos = new ArrayList<TableInfo>();
  }

  Map<String, Set<String>> tableNameMap = null;
  List<TableInfo> tableInfos = null;
  DB_TYPE dbType = null;
  SqlContext parentContext = null;
  Map<Field,String> fieldNameOverrides = null;
  public int maxFields = Integer.MAX_VALUE;

  boolean inInnerQuery() {
    return parentContext != null;
  }

  DBQuery<?> getRootQuery() {
    if (parentContext != null) return parentContext.getRootQuery();
    return q;
  }

  String getFullTableName(final TableInfo ti) {
    if (ti.dummyTable != null) {
      return dbType==Constants.DB_TYPE.SQLSERVER ? "#"+ti.dummyTable.name : ti.dummyTable.name;
    }
    return getFullTableName(ti.tableClass);
  }

  String getFullTableName(final Class<? extends Table> tableClass) {
    final StringBuilder sb = new StringBuilder();
    String schema = Util.getSchemaName(tableClass);
    if (schema != null && !"".equals(schema)) {
      schema = Context.getSchemaToUse(getRootQuery().getDataSource(), schema);
      if (schema != null && !"".equals(schema)) {
        sb.append(schema);
        sb.append(dbType==Constants.DB_TYPE.SQLSERVER ? ".dbo." : ".");
      }
    }
    sb.append(Util.getTableName(tableClass));
    return sb.toString();
  }

  Set<String> getPossibleTableMatches(Class<? extends Table> table) {
    Set<String> names = new HashSet<String>();
    if (q!=null && q.tableInfos!=null) {
      for (TableInfo ti : q.tableInfos) {
        if (table.equals(ti.tableClass)) names.add(ti.tableName);
      }
    }
    if (q!=null && q.joins!=null) {
      for (JoinInfo ji : q.joins) {
        if (table.equals(ji.reffedTableInfo.tableClass)) names.add(ji.reffedTableInfo.tableName);
        if (table.equals(ji.reffingTableInfo.tableClass)) names.add(ji.reffingTableInfo.tableName);
      }
    }
    if (q!=null && q.joinsToOne!=null) {
      for (JoinInfo ji : q.joinsToOne) {
        if (table.equals(ji.reffedTableInfo.tableClass)) names.add(ji.reffedTableInfo.tableName);
        if (table.equals(ji.reffingTableInfo.tableClass)) names.add(ji.reffingTableInfo.tableName);
      }
    }
    if (q!=null && q.joinsToMany!=null) {
      for (JoinInfo ji : q.joinsToMany) {
        if (table.equals(ji.reffedTableInfo.tableClass)) names.add(ji.reffedTableInfo.tableName);
        if (table.equals(ji.reffingTableInfo.tableClass)) names.add(ji.reffingTableInfo.tableName);
      }
    }
    return names;
  }

  public void setFieldNameOverride(Field<?> key, String value) {
    if (fieldNameOverrides==null) fieldNameOverrides = new HashMap<Field,String>();
    fieldNameOverrides.put(key, value);
  }
  
}
Java Source Code List
