dzhh-backend/.svn/pristine/cb/cbc426d99a9c0917d51a5a601649665c44f56abb.svn-base

//package org.jeecg.config.mybatis;
//
//import com.baomidou.mybatisplus.extension.plugins.tenant.TenantSqlParser;
//import net.sf.jsqlparser.expression.BinaryExpression;
//import net.sf.jsqlparser.expression.Expression;
//import net.sf.jsqlparser.expression.Parenthesis;
//import net.sf.jsqlparser.expression.operators.conditional.AndExpression;
//import net.sf.jsqlparser.expression.operators.conditional.OrExpression;
//import net.sf.jsqlparser.expression.operators.relational.*;
//import net.sf.jsqlparser.schema.Column;
//import net.sf.jsqlparser.schema.Table;
//import net.sf.jsqlparser.statement.select.*;
//
//import java.util.List;
//
///**
// * 复写租户条件
// */
//public class JeecgTenantParser extends TenantSqlParser {
//
//    /**
//     * @param expression
//     * @param table
//     * @return
//     */
//    protected Expression processTableAlias(Expression expression, Table table) {
//        String tableAliasName;
//        if (table.getAlias() == null) {
//            tableAliasName = table.getName();
//        } else {
//            tableAliasName = table.getAlias().getName();
//        }
//
//        // in
//        if (expression instanceof InExpression) {
//            InExpression in = (InExpression) expression;
//            if (in.getLeftExpression() instanceof Column) {
//                setTableAliasNameForColumn((Column) in.getLeftExpression(), tableAliasName);
//            }
//
//            // 比较操作
//        } else if (expression instanceof BinaryExpression) {
//            BinaryExpression compare = (BinaryExpression) expression;
//            if (compare.getLeftExpression() instanceof Column) {
//                setTableAliasNameForColumn((Column) compare.getLeftExpression(), tableAliasName);
//            } else if (compare.getRightExpression() instanceof Column) {
//                setTableAliasNameForColumn((Column) compare.getRightExpression(), tableAliasName);
//            }
//
//            // between
//        } else if (expression instanceof Between) {
//            Between between = (Between) expression;
//            if (between.getLeftExpression() instanceof Column) {
//                setTableAliasNameForColumn((Column) between.getLeftExpression(), tableAliasName);
//            }
//        }
//        return expression;
//    }
//
//    private void setTableAliasNameForColumn(Column column, String tableAliasName) {
//        column.setColumnName(tableAliasName + "." + column.getColumnName());
//    }
//
//    /**
//     * 默认是按 tenant_id=1 按等于条件追加
//     *
//     * @param currentExpression 现有的条件：比如你原来的sql查询条件
//     * @param table
//     * @return
//     */
//    @Override
//    protected Expression builderExpression(Expression currentExpression, Table table) {
//        final Expression tenantExpression = this.getTenantHandler().getTenantId(true);
//        Expression appendExpression;
//        if (!(tenantExpression instanceof SupportsOldOracleJoinSyntax)) {
//            appendExpression = new EqualsTo();
//            ((EqualsTo) appendExpression).setLeftExpression(this.getAliasColumn(table));
//            ((EqualsTo) appendExpression).setRightExpression(tenantExpression);
//        } else {
//            appendExpression = processTableAlias(tenantExpression, table);
//        }
//        if (currentExpression == null) {
//            return appendExpression;
//        }
//        if (currentExpression instanceof BinaryExpression) {
//            BinaryExpression binaryExpression = (BinaryExpression) currentExpression;
//            if (binaryExpression.getLeftExpression() instanceof FromItem) {
//                processFromItem((FromItem) binaryExpression.getLeftExpression());
//            }
//            if (binaryExpression.getRightExpression() instanceof FromItem) {
//                processFromItem((FromItem) binaryExpression.getRightExpression());
//            }
//        } else if (currentExpression instanceof InExpression) {
//            InExpression inExp = (InExpression) currentExpression;
//            ItemsList rightItems = inExp.getRightItemsList();
//            if (rightItems instanceof SubSelect) {
//                processSelectBody(((SubSelect) rightItems).getSelectBody());
//            }
//        }
//        if (currentExpression instanceof OrExpression) {
//            return new AndExpression(new Parenthesis(currentExpression), appendExpression);
//        } else {
//            return new AndExpression(currentExpression, appendExpression);
//        }
//    }
//
//    @Override
//    protected void processPlainSelect(PlainSelect plainSelect, boolean addColumn) {
//        FromItem fromItem = plainSelect.getFromItem();
//        if (fromItem instanceof Table) {
//            Table fromTable = (Table) fromItem;
//            if (!this.getTenantHandler().doTableFilter(fromTable.getName())) {
//                plainSelect.setWhere(builderExpression(plainSelect.getWhere(), fromTable));
//                if (addColumn) {
//                    plainSelect.getSelectItems().add(new SelectExpressionItem(new Column(this.getTenantHandler().getTenantIdColumn())));
//                }
//            }
//        } else {
//            processFromItem(fromItem);
//        }
//        List<Join> joins = plainSelect.getJoins();
//        if (joins != null && joins.size() > 0) {
//            joins.forEach(j -> {
//                processJoin(j);
//                processFromItem(j.getRightItem());
//            });
//        }
//    }
//
//}