io.xream.sqli.builder.SourceScriptOptimizable Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sqli-builder Show documentation
There is a newer version: 2.1.10
/*
 * Copyright 2020 io.xream.sqli
 *
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.xream.sqli.builder;

import io.xream.sqli.util.SqliStringUtil;

import java.util.Iterator;
import java.util.List;
import java.util.Set;

/**
 * @Author Sim
 */
public interface SourceScriptOptimizable {

    default void addConditonBeforeOptimization(String key, Set conditionSet){
        conditionSet.add(key);
    }

    default void addConditionBeforeOptimization(List bbList, Set conditionSet) {
        if (bbList == null)
            return;
        for (Bb bb : bbList) {
            conditionSet.add(bb.getKey());
            List subList = bb.getSubList();
            if (subList != null && !subList.isEmpty()) {
                addConditionBeforeOptimization(subList, conditionSet);
            }
        }
    }

    default void optimizeSourceScript(Set conditionSet, List sourceScripts) {
        if (sourceScripts.size() <= 1)
            return;
        if (conditionSet.size() > 0) {
            for (String test : conditionSet) {
                if (test != null) {
                    if (test.contains("."))
                        break;
                    return;
                }
            }
        }
        for (SourceScript sourceScript : sourceScripts) {
            if (sourceScript.getSubCriteria() != null) {
                sourceScript.used();
                continue;
            }
            for (String key : conditionSet) {
                if (key == null)
                    continue;
                if (SqliStringUtil.isNullOrEmpty(sourceScript.getAlia())) {
                    if (key.contains(sourceScript.getSource() + ".")) {
                        sourceScript.used();
                        break;
                    }
                } else {
                    if (key.contains(sourceScript.getAlia() + ".")) {
                        sourceScript.used();
                        break;
                    }
                }
            }
        }

        int size = sourceScripts.size();
        for (int i = size - 1; i >= 0; i--) {
            SourceScript sourceScript = sourceScripts.get(i);
            if (sourceScript.getSubCriteria() != null) {
                sourceScript.targeted();
                continue;
            }
            if (!sourceScript.isUsed() && !sourceScript.isTargeted())
                continue;
            for (int j = i - 1; j >= 0; j--) {
                SourceScript sc = sourceScripts.get(j);
                if (sourceScript.getSource().equals(sc.getSource()))
                    continue;
                //FIXME
                On on = sourceScript.getOn();
                if (on == null || on.getJoinFrom() == null)
                    continue;
                if (sc.alia().equals(on.getJoinFrom().getAlia())) {
                    sc.targeted();
                    break;
                }
            }
        }

        Iterator ite = sourceScripts.iterator();
        while (ite.hasNext()) {
            SourceScript sourceScript = ite.next();
            if (!sourceScript.isUsed() && !sourceScript.isTargeted())
                ite.remove();
        }
    }

}