@@ -0,0 +1,3756 @@
+/*%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.assignment.AssignmentSegment;%0Aimport org.apache.shardingsphere.core.parse.sql.segment.dml.assignment.SetAssignmentsSegment;%0Aimport org.apache.shardingsphere.core.parse.sql.segment.dml.expr.simple.ParameterMarkerExpressionSegment;%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.InsertSetEncryptValueToken;%0Aimport org.apache.shardingsphere.core.rule.EncryptRule;%0Aimport org.apache.shardingsphere.spi.encrypt.ShardingEncryptor;%0A%0Aimport java.util.Collection;%0Aimport java.util.Collections;%0Aimport java.util.LinkedList;%0A%0A/**%0A * Insert set encrypt value token generator.%0A *%0A * @author panjuan%0A */%0Apublic final class InsertSetEncryptValueTokenGenerator implements CollectionSQLTokenGenerator%3CEncryptRule%3E %7B%0A    %0A    @Override%0A    public Collection%3CInsertSetEncryptValueToken%3E generateSQLTokens(final SQLStatement sqlStatement, final EncryptRule encryptRule) %7B%0A        Optional%3CSetAssignmentsSegment%3E setAssignmentsSegment = sqlStatement.findSQLSegment(SetAssignmentsSegment.class);%0A        if (!(sqlStatement instanceof InsertStatement && setAssignmentsSegment.isPresent())) %7B%0A            return Collections.emptyList();%0A        %7D%0A        return createInsertSetEncryptValueTokens((InsertStatement) sqlStatement, encryptRule, setAssignmentsSegment.get());%0A    %7D%0A    %0A    private Collection%3CInsertSetEncryptValueToken%3E createInsertSetEncryptValueTokens(final InsertStatement insertStatement, final EncryptRule encryptRule, final SetAssignmentsSegment segment) %7B%0A        Collection%3CInsertSetEncryptValueToken%3E result = new LinkedList%3C%3E();%0A        for (AssignmentSegment each : segment.getAssignments()) %7B%0A            Optional%3CInsertSetEncryptValueToken%3E insertSetEncryptValueToken = getInsertSetEncryptValueToken(insertStatement, encryptRule, each);%0A            if (insertSetEncryptValueToken.isPresent()) %7B%0A                result.add(insertSetEncryptValueToken.get());%0A            %7D%0A        %7D%0A        return result;%0A    %7D%0A    %0A    private Optional%3CInsertSetEncryptValueToken%3E getInsertSetEncryptValueToken(final InsertStatement insertStatement, final EncryptRule encryptRule, final AssignmentSegment segment) %7B%0A        Optional%3CShardingEncryptor%3E shardingEncryptor = encryptRule.getEncryptorEngine().getShardingEncryptor(insertStatement.getTables().getSingleTableName(), segment.getColumn().getName());%0A        if (shardingEncryptor.isPresent() && !(segment.getValue() instanceof ParameterMarkerExpressionSegment)) %7B%0A            return Optional.of(new InsertSetEncryptValueToken(segment.getValue().getStartIndex(), segment.getValue().getStopIndex(), segment.getColumn().getName()));%0A        %7D%0A        return Optional.absent();%0A    %7D%0A%7D%0A

add InsertSetEncryptValueTokenGenerator