@@ -0,0 +1,3234 @@
+/*%0A * Licensed to the Apache Software Foundation (ASF) under one or more%0A * contributor license agreements.  See the NOTICE file distributed with%0A * this work for additional information regarding copyright ownership.%0A * The ASF licenses this file to You under the Apache License, Version 2.0%0A * (the %22License%22); you may not use this file except in compliance with%0A * the License.  You may obtain a copy of the License at%0A *%0A *     http://www.apache.org/licenses/LICENSE-2.0%0A *%0A * Unless required by applicable law or agreed to in writing, software%0A * distributed under the License is distributed on an %22AS IS%22 BASIS,%0A * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.%0A * See the License for the specific language governing permissions and%0A * limitations under the License.%0A */%0A%0Apackage org.apache.shardingsphere.core.rewrite.token.generator;%0A%0Aimport com.google.common.base.Optional;%0Aimport org.apache.shardingsphere.core.parse.sql.segment.dml.column.InsertColumnsSegment;%0Aimport org.apache.shardingsphere.core.parse.sql.statement.SQLStatement;%0Aimport org.apache.shardingsphere.core.parse.sql.statement.dml.InsertStatement;%0Aimport org.apache.shardingsphere.core.parse.sql.token.impl.InsertColumnsToken;%0Aimport org.apache.shardingsphere.core.rule.EncryptRule;%0A%0A/**%0A * Insert columns token generator.%0A *%0A * @author panjuan%0A */%0Apublic final class ShardingInsertColumnsTokenGenerator implements OptionalSQLTokenGenerator%3CEncryptRule%3E %7B%0A    %0A    @Override%0A    public Optional%3CInsertColumnsToken%3E generateSQLToken(final SQLStatement sqlStatement, final EncryptRule encryptRule) %7B%0A        Optional%3CInsertColumnsSegment%3E insertColumnsSegment = sqlStatement.findSQLSegment(InsertColumnsSegment.class);%0A        if (!(sqlStatement instanceof InsertStatement && insertColumnsSegment.isPresent())) %7B%0A            return Optional.absent();%0A        %7D%0A        return Optional.of(createInsertColumnsToken((InsertStatement) sqlStatement, insertColumnsSegment.get(), encryptRule));%0A    %7D%0A    %0A    private InsertColumnsToken createInsertColumnsToken(final InsertStatement insertStatement, final InsertColumnsSegment segment, final EncryptRule encryptRule) %7B%0A        InsertColumnsToken result;%0A        if (segment.getColumns().isEmpty()) %7B%0A            result = new InsertColumnsToken(segment.getStopIndex(), false);%0A            result.getColumns().addAll(insertStatement.getColumnNames());%0A        %7D else %7B%0A            result = new InsertColumnsToken(segment.getStopIndex(), true);%0A        %7D%0A        fillWithQueryAssistedColumn(insertStatement, encryptRule, result);%0A        return result;%0A    %7D%0A    %0A    private void fillWithQueryAssistedColumn(final InsertStatement insertStatement, final EncryptRule encryptRule, final InsertColumnsToken insertColumnsToken) %7B%0A        for (String each : insertStatement.getColumnNames()) %7B%0A            Optional%3CString%3E assistedColumnName = encryptRule.getEncryptorEngine().getAssistedQueryColumn(insertStatement.getTables().getSingleTableName(), each);%0A            if (assistedColumnName.isPresent()) %7B%0A                insertColumnsToken.getColumns().remove(assistedColumnName.get());%0A                insertColumnsToken.getColumns().add(assistedColumnName.get());%0A            %7D%0A        %7D%0A    %7D%0A%7D%0A

add ShardingInsertColumnsTokenGenerator