Java Column类代码示例

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
private Mutation createUpdateMutation(Update update, boolean generateParameterMetaData) throws SQLException
{
	if (update.getTables().isEmpty())
		throw new CloudSpannerSQLException("No table found in update statement", Code.INVALID_ARGUMENT);
	if (update.getTables().size() > 1)
		throw new CloudSpannerSQLException("Update statements for multiple tables at once are not supported",
				Code.INVALID_ARGUMENT);
	String table = unquoteIdentifier(update.getTables().get(0).getFullyQualifiedName());
	getParameterStore().setTable(table);
	List<Expression> expressions = update.getExpressions();
	WriteBuilder builder = Mutation.newUpdateBuilder(table);
	int index = 0;
	for (Column col : update.getColumns())
	{
		String columnName = unquoteIdentifier(col.getFullyQualifiedName());
		expressions.get(index).accept(new ValueBinderExpressionVisitorAdapter<>(getParameterStore(),
				builder.set(columnName), columnName));
		index++;
	}
	visitUpdateWhereClause(update.getWhere(), builder, generateParameterMetaData);

	return builder.build();
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
private void visitDeleteWhereClause(Expression where, DeleteKeyBuilder keyBuilder,
		boolean generateParameterMetaData) throws SQLException
{
	if (where != null)
	{
		DMLWhereClauseVisitor whereClauseVisitor = new DMLWhereClauseVisitor(getParameterStore())
		{

			@Override
			protected void visitExpression(Column col, Expression expression)
			{
				String columnName = unquoteIdentifier(col.getFullyQualifiedName());
				keyBuilder.set(columnName);
				expression.accept(
						new KeyBuilderExpressionVisitorAdapter(getParameterStore(), columnName, keyBuilder));
			}

		};
		where.accept(whereClauseVisitor);
		if (!generateParameterMetaData && !whereClauseVisitor.isValid())
		{
			throw new CloudSpannerSQLException(INVALID_WHERE_CLAUSE_DELETE_MESSAGE, Code.INVALID_ARGUMENT);
		}
	}
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
private boolean isSingleRowWhereClause(TableKeyMetaData table, Expression where)
{
	if (where != null)
	{
		SingleRowWhereClauseValidator validator = new SingleRowWhereClauseValidator(table);
		DMLWhereClauseVisitor whereClauseVisitor = new DMLWhereClauseVisitor(getParameterStore())
		{

			@Override
			protected void visitExpression(Column col, Expression expression)
			{
				String columnName = unquoteIdentifier(col.getFullyQualifiedName());
				validator.set(columnName);
				expression.accept(
						new SingleRowWhereClauseValidatorExpressionVisitorAdapter(getParameterStore(), validator));
			}

		};
		where.accept(whereClauseVisitor);
		return whereClauseVisitor.isValid() && validator.isValid();
	}
	return false;
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
/**
 * 将sql转换为count查询
 *
 * @param select
 */
public void sqlToCount(Select select, String name) {
    SelectBody selectBody = select.getSelectBody();
    // 是否能简化count查询
    List<SelectItem> COUNT_ITEM = new ArrayList<SelectItem>();
    COUNT_ITEM.add(new SelectExpressionItem(new Column("count(" + name +")")));
    if (selectBody instanceof PlainSelect && isSimpleCount((PlainSelect) selectBody)) {
        ((PlainSelect) selectBody).setSelectItems(COUNT_ITEM);
    } else {
        PlainSelect plainSelect = new PlainSelect();
        SubSelect subSelect = new SubSelect();
        subSelect.setSelectBody(selectBody);
        subSelect.setAlias(TABLE_ALIAS);
        plainSelect.setFromItem(subSelect);
        plainSelect.setSelectItems(COUNT_ITEM);
        select.setSelectBody(plainSelect);
    }
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
@Override
public void visit(PivotXml pivot) {
    List<Column> forColumns = pivot.getForColumns();
    buffer.append(" PIVOT XML (")
            .append(PlainSelect.getStringList(pivot.getFunctionItems()))
            .append(" FOR ")
            .append(PlainSelect.getStringList(forColumns, true, forColumns != null && forColumns.size() > 1))
            .append(" IN (");
    if (pivot.isInAny()) {
        buffer.append("ANY");
    } else if (pivot.getInSelect() != null) {
        buffer.append(pivot.getInSelect());
    } else {
        buffer.append(PlainSelect.getStringList(pivot.getInItems()));
    }
    buffer.append("))");
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
@Override
public void visit(Column tableColumn) {
    final Table table = tableColumn.getTable();
    String tableName = null;
    if (table != null) {
        if (table.getAlias() != null) {
            tableName = table.getAlias().getName();
        } else {
            tableName = table.getFullyQualifiedName();
        }
    }
    if (tableName != null && !tableName.isEmpty()) {
        buffer.append(tableName).append(".");
    }

    buffer.append(tableColumn.getColumnName());
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
/**
 * 最外层的SQL查询需要增加ROW_NUMBER()
 *
 * @param plainSelect
 */
private void addRowNumber(PlainSelect plainSelect, String orderBy) {
    //增加ROW_NUMBER()
    StringBuilder orderByBuilder = new StringBuilder();
    orderByBuilder.append("ROW_NUMBER() OVER (");
    if (isNotEmptyString(orderBy)) {
        //使用设置的值
        orderByBuilder.append(orderBy);
    } else if (isNotEmptyList(plainSelect.getOrderByElements())) {
        //注意：order by别名的时候有错,由于没法判断一个列是否为别名，所以不能解决
        orderByBuilder.append(PlainSelect.orderByToString(false, plainSelect.getOrderByElements()));
    } else {
        throw new RuntimeException("请您指定order by参数或者在sql中包含order by语句!");
    }
    //需要把改orderby清空
    if (isNotEmptyList(plainSelect.getOrderByElements())) {
        plainSelect.setOrderByElements(null);
    }
    orderByBuilder.append(") ");
    orderByBuilder.append(PAGE_ROW_NUMBER);
    Column orderByColumn = new Column(orderByBuilder.toString());
    plainSelect.getSelectItems().add(0, new SelectExpressionItem(orderByColumn));
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
/**
 * 举例 left join tab2 on tab1.a = tab2.A
 *
 * @param joinTableType   tab2_left
 * @param righExpEqualsTo tab1.a = tab2.A
 */
public void saveJoinCondition(String joinTableType, EqualsTo righExpEqualsTo) {
    //region 保存至JoinConditon
    Column leftCol = (Column) righExpEqualsTo.getLeftExpression();
    Column rightCol = (Column) righExpEqualsTo.getRightExpression();
    TCItem leftTCItem = new TCItem();
    leftTCItem.setColName(leftCol.getColumnName());
    leftTCItem.setTableName(leftCol.getTable().getName());
    TCItem rightTCItem = new TCItem();
    rightTCItem.setColName(rightCol.getColumnName());
    rightTCItem.setTableName(rightCol.getTable().getName());

    JoinTwoTable joinTwoTable = new JoinTwoTable();
    joinTwoTable.setTcItemLeft(leftTCItem);
    joinTwoTable.setTcItemRight(rightTCItem);
    JoinCondition.linkTablemap.put(joinTableType, joinTwoTable);
    //endregion
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
/**
     * 解析group by的语义，并保存到GoupBy 中
     *
     * @param defalutTable 默认的表名
     * @param plainSelect  SQL
     */
    private void getGroups(String defalutTable, PlainSelect plainSelect) {
        List<Expression> groupExps = plainSelect.getGroupByColumnReferences();
        if (groupExps != null) {
            for (Expression item : groupExps) {
                Column column = (Column) item;
                TCItem tcItem = new TCItem();
                tcItem.setColName(column.getColumnName());
                if (column.getTable().getName() == null || column.getTable().getName().isEmpty()) {
                    tcItem.setTableName(defalutTable);
                } else {
                    tcItem.setTableName(column.getTable().getName());
                }
                GroupBy.groupList.add(tcItem);
            }
        }
//        assertEquals(2,GroupBy.groupList.size());
    }
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
/**
 * Get type
 * @param c The connection
 * @param column The column
 * @param query The query
 * @return The value
 */
private static Integer getType(Connection c, Column column, String query) throws Exception
{
   String tableName = column.getTable().getName();

   if (tableName != null && aliases.containsKey(tableName))
      tableName = aliases.get(tableName);

   if (tableName == null)
      tableName = currentTableName;

   tableName = tableName.toLowerCase();

   Map<String, Integer> tableData = tables.get(tableName);

   if (tableData == null)
   {
      getUsedTables(c, query);
      tableData = tables.get(tableName);
   }

   return tableData.get(column.getColumnName().toLowerCase());
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
@Override
public void visit(Insert insert) {
    this.getStack().push(new FrameContext());
    ShardRouteConfig routeConfig = shardRouter.getRouteConfig(insert.getTable().getSchemaName(),
                                                              insert.getTable().getName());
    if (routeConfig != null) {
        TableWrapper table = new TableWrapper(insert.getTable(), routeConfig);
        addRoutedTableIntoContext(table, routeConfig, false);
        List<Column> columns = insert.getColumns();
        if (columns != null) {
            ExpressionList expressionList = (ExpressionList) insert.getItemsList();
            List<Expression> valueList = expressionList.getExpressions();
            for (int i = 0; i < columns.size(); i++) {
                Column column = columns.get(i);
                TableWrapper tab = getTableFromContext(column);
                if (tab != null) {
                    Expression expression = valueList.get(i);
                    routeTable(tab, column, expression);
                }
            }
        }
    }
    super.visit(insert);
    afterVisitBaseStatement();
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
private Object getRouteValue(Column column, Expression obj) {
    if (obj == null) {
        return null;
    }
    if (obj instanceof LongValue) {
        return ((LongValue) obj).getValue();
    } else if (obj instanceof StringValue) {
        return ((StringValue) obj).getValue();
    } else if (obj instanceof HexValue) {
        return Long.parseLong(((HexValue) obj).getValue(), 16);
    } else if (obj instanceof DateValue) {
        return ((DateValue) obj).getValue();
    } else if (obj instanceof DoubleValue) {
        return ((DoubleValue) obj).getValue();
    } else if (obj instanceof TimeValue) {
        return ((TimeValue) obj).getValue();
    } else if (obj instanceof TimestampValue) {
        return ((TimestampValue) obj).getValue();
    } else {// NullValue
        throw new UnsupportedSQLParameterTypeException("Type '" + obj.getClass()
                                                       + "' is not supported for shard value '" + column.toString()
                                                       + "'. Sql is [" + sql + "]");
    }
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
@Override
public byte[] computeNewValue(Record previous, StatementEvaluationContext context, TableContext tableContext) throws StatementExecutionException {
    Map<String, Object> res = previous != null ? new HashMap<>(previous.toBean(table)) : new HashMap<>();
    DataAccessor bean = previous != null ? previous.getDataAccessor(table) : DataAccessor.NULL;
    for (int i = 0; i < columns.size(); i++) {
        CompiledSQLExpression e = expressions.get(i);
        String columnName = columns.get(i);
        herddb.model.Column column = table.getColumn(columnName);
        if (column == null) {
            throw new StatementExecutionException("unknown column " + columnName + " in table " + table.name);
        }
        columnName = column.name;
        Object value = RecordSerializer.convert(column.type, e.evaluate(bean, context));
        res.put(columnName, value);
    }
    return RecordSerializer.toRecord(res, table).value.data;
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
/**
 * <p>
 * insert 语句处理
 * </p>
 */
@Override
public void processInsert(Insert insert) {
    if (this.tenantHandler.doTableFilter(insert.getTable().getName())) {
        // 过滤退出执行
        return;
    }
    insert.getColumns().add(new Column(this.tenantHandler.getTenantIdColumn()));
    if (insert.getSelect() != null) {
        processPlainSelect((PlainSelect) insert.getSelect().getSelectBody(), true);
    } else if (insert.getItemsList() != null) {
        ((ExpressionList) insert.getItemsList()).getExpressions().add(tenantHandler.getTenantId());
    } else {
        throw new MybatisPlusException("Failed to process multiple-table update, please exclude the tableName or statementId");
    }
}
 

import net.sf.jsqlparser.schema.Column; //导入依赖的package包/类
/**
 * <p>
 * 处理 PlainSelect
 * </p>
 *
 * @param plainSelect
 * @param addColumn   是否添加租户列,insert into select语句中需要
 */
protected void processPlainSelect(PlainSelect plainSelect, boolean addColumn) {
    FromItem fromItem = plainSelect.getFromItem();
    if (fromItem instanceof Table) {
        Table fromTable = (Table) fromItem;
        if (this.tenantHandler.doTableFilter(fromTable.getName())) {
            // 过滤退出执行
            return;
        }
        plainSelect.setWhere(builderExpression(plainSelect.getWhere(), fromTable));
        if (addColumn) {
            plainSelect.getSelectItems().add(new SelectExpressionItem(new Column(this.tenantHandler.getTenantIdColumn())));
        }
    } else {
        processFromItem(fromItem);
    }
    List<Join> joins = plainSelect.getJoins();
    if (joins != null && joins.size() > 0) {
        for (Join join : joins) {
            processJoin(join);
            processFromItem(join.getRightItem());
        }
    }
}