Java Code Examples for net.sf.jsqlparser.schema.Column#getTable()

@Override
public int isNullable(int column) throws SQLException {
  Column col = getColumn(column);
  if (col != null && col.getTable() != null) {
    String schema = Strings.isNullOrEmpty(col.getTable().getSchemaName()) ? ""
        : CloudSpannerDriver.unquoteIdentifier(col.getTable().getSchemaName());
    String tableName = CloudSpannerDriver.unquoteIdentifier(col.getTable().getName());
    String colName = CloudSpannerDriver.unquoteIdentifier(col.getColumnName());
    try (java.sql.ResultSet rs =
        statement.getConnection().getMetaData().getColumns("", schema, tableName, colName)) {
      if (rs.next()) {
        return rs.getInt("NULLABLE");
      }
    }
  }
  return columnNullableUnknown;
}
 

private void routeTable(TableWrapper tab, Column column, Object sdValue) {
    if (tab == null) {//
        return;
    }
    if (tab == AMBIGUOUS_TABLE) {
        throw new RuntimeException("Shard value '" + column.toString() + "' in where clause is ambiguous. Sql is ["
                                   + sql + "]");
    }
    ShardRouteInfo routeInfo = getRouteInfo(tab, sdValue);
    // 当没有设置别名,但分表字段使用了表前缀时,别前缀需要根据路由结果进行重写;
    // 这里之所有没有采用将table取名的方式是因为update/delete/insert不全支持别名;
    // 由于select都是支持别名的,所有都会被添加别名,因此不会执行下面的操作;
    Table columnTable = column.getTable();
    if (tab.getAlias() == null && columnTable != null && columnTable.getName() != null
        && columnTable.getName().length() > 0) {
        if (columnTable.getSchemaName() != null) {
            columnTable.setSchemaName(routeInfo.getScName());
        }
        columnTable.setName(routeInfo.getTbName());
    }
    route0(tab, routeInfo);
}
 

@Override
  public void visit(Column column) {
if(SqlUtils.isColumn(column)) {
	String columnName;
	if(column.getTable() != null) {
		columnName = SqlUtils.getColumnNameFromColumn(column);
	}
	else {
		columnName = column.getColumnName();
	}
	if(!SqlUtils.isTableAliasOfColumn(column, joinAliasTable) ) {
		if(column.getTable() == null || SqlUtils.isTableAliasOfColumn(column, baseAliasTable)) {//we know let var don't have table inside
			column.setColumnName("$$" + columnName.replace(".", "_").toLowerCase());
		}
		else {
			column.setColumnName("$$" + column.getName(false).replace(".", "_").toLowerCase());
		}
		column.setTable(null);
	}
	else {
		column.setTable(null);
		column.setColumnName("$" + columnName);
	}
}
  }
 

@Override
public void visit(Column tableColumn) {
	clear();
	boolean prevIsTopLevel = isTopLevel;
	isTopLevel = false;
	if (tableColumn.getTable()!=null && tableColumn.getTable().getName()!=null) {
		if (tableColumn.getTable().getName().equalsIgnoreCase(leftTable.getAlias()!=null? leftTable.getAlias().getName(): leftTable.getName())) {
			leftTableFound = true;
		}
		if (tableColumn.getTable().getName().equalsIgnoreCase(rightTable.getAlias()!=null? rightTable.getAlias().getName(): rightTable.getName())) {
			rightTableFound = true;
		}
	}
	isTopLevel = prevIsTopLevel;
	defaultTopLevelProcessing(tableColumn);
}
 

@Override
public int getPrecision(int column) throws SQLException {
  int colType = getColumnType(column);
  switch (colType) {
    case Types.BOOLEAN:
      return 1;
    case Types.DATE:
      return 10;
    case Types.DOUBLE:
      return 14;
    case Types.BIGINT:
      return 10;
    case Types.TIMESTAMP:
      return 24;
    default:
      // Not fixed size, try to get it from INFORMATION_SCHEMA
  }
  Column col = getColumn(column);
  if (col != null && col.getTable() != null) {
    String schema = Strings.isNullOrEmpty(col.getTable().getSchemaName()) ? ""
        : CloudSpannerDriver.unquoteIdentifier(col.getTable().getSchemaName());
    String tableName = CloudSpannerDriver.unquoteIdentifier(col.getTable().getName());
    String colName = CloudSpannerDriver.unquoteIdentifier(col.getColumnName());
    try (java.sql.ResultSet rs =
        statement.getConnection().getMetaData().getColumns("", schema, tableName, colName)) {
      if (rs.next()) {
        return rs.getInt("COLUMN_SIZE");
      }
    }
  }
  return 0;
}
 

@Override
public String getTableName(int column) throws SQLException {
  Column col = getColumn(column);
  if (col != null && col.getTable() != null)
    return col.getTable().getName();

  return "";
}
 

@Override
public boolean isReadOnly(int column) throws SQLException {
  Column col = getColumn(column);
  if (col == null || col.getTable() == null)
    return true;
  // Primary key columns are only insertable and never updatable, all
  // other columns are insertable and updatable. This however still means
  // that a primary key column is writable.
  return false;
}
 

@Override
public void visit(Column arg0) {
	String tableName = null;
	if (arg0.getTable() != null) {
		if (arg0.getTable().getAlias() != null)
			tableName = arg0.getTable().getAlias().getName();
		else 
			tableName = arg0.getTable().getName();
	}
	
	// If it is outer select, don't add, otherwise do add
	if (!(currentState.isOuterState && currentState.phase == SelectPhase.SELECT)) {
		addUsedColumn(arg0.getColumnName(), tableName);
	}
}
 

@Override
public void visit(Column arg0) {
	Table table = arg0.getTable();
	if (table != null) {
		table.accept(this);
	}
	arg0.setColumnName(secureName(arg0.getColumnName()));
}
 

private Expression getExpression(SelectItem item, Table table) {
	SelectExpressionItem rightSelectItem = (SelectExpressionItem) item;
	Expression rightExp = rightSelectItem.getExpression();
	if (rightExp instanceof Column) {
		Column c = (Column) rightExp;
		if (c.getTable() == null) {
			c.setTable(table);
		}
	}
	return rightExp;
}
 

@Override
public boolean isSearchable(int column) throws SQLException {
  Column col = getColumn(column);
  return col != null && col.getTable() != null;
}
 

public void visit(Column tableColumn) {
    if (tableColumn.getTable() != null) {
        tableColumn.setTable(null);
    }
}
 

public void visit(Column tableColumn) throws Exception {
    if (tableColumn.getTable() != null) {
        tableColumn.getTable().accept(this);
    }
}