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

Go to download
/*
 * 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.mapping.SqlNormalizer;
import io.xream.sqli.util.EnumUtil;
import io.xream.sqli.util.SqliStringUtil;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Objects;

/**
 * @Author Sim
 */
public class ConditionBuilder implements SqlNormalizer {

    private ConditionBuilder instance;

    private transient List bbList;
    private transient boolean isOr;

    private transient List tempList;
    private transient List> subsList;

    protected ConditionBuilder(){
        this.instance = this;
    }

    protected ConditionBuilder(List bbList){
        this.instance = this;
        this.instance.bbList = bbList;
    }

    protected void init(List bbList){
        this.bbList = bbList;
    }

    private List> getSubsList(){
        if (this.subsList == null) {
            this.subsList = new LinkedList<>();
        }
        return this.subsList;
    }

    public static ConditionBuilder builder(List bbList){
        return new ConditionBuilder(bbList);
    }

    public static ConditionBuilder builder() {
        return new ConditionBuilder(new ArrayList<>());
    }

    public CriteriaCondition build() {
        return () -> bbList;
    }

    public ConditionBuilder eq(String property, Object value){
        return doGle(Op.EQ, property, value);
    }

    public ConditionBuilder lt(String property, Object value){
        return doGle(Op.LT, property, value);
    }

    public ConditionBuilder lte(String property, Object value){
        return doGle(Op.LTE, property, value);
    }

    public ConditionBuilder gt(String property, Object value){
        return doGle(Op.GT, property, value);
    }

    public ConditionBuilder gte(String property, Object value){
        return doGle(Op.GTE, property, value);
    }

    public ConditionBuilder ne(String property, Object value){
        return doGle(Op.NE, property, value);
    }

    public ConditionBuilder like(String property, String value){
        if (SqliStringUtil.isNullOrEmpty(value)) {
            isOr();
            return instance;
        }
        String likeValue = SqlScript.LIKE_HOLDER + value + SqlScript.LIKE_HOLDER;
        return doLike(Op.LIKE,property,likeValue);
    }

    public ConditionBuilder likeRight(String property, String value){
        if (SqliStringUtil.isNullOrEmpty(value)) {
            isOr();
            return instance;
        }
        String likeValue = value + SqlScript.LIKE_HOLDER;
        return doLike(Op.LIKE,property,likeValue);
    }

    public ConditionBuilder notLike(String property, String value){
        if (SqliStringUtil.isNullOrEmpty(value)) {
            isOr();
            return instance;
        }
        String likeValue = SqlScript.LIKE_HOLDER + value + SqlScript.LIKE_HOLDER;
        return doLike(Op.NOT_LIKE,property,likeValue);
    }

    public ConditionBuilder in(String property, List list){
        return doIn(Op.IN,property,list);
    }

    public ConditionBuilder nin(String property, List list){
        return doIn(Op.NOT_IN,property,list);
    }

    public ConditionBuilder nonNull(String property){
        return doNull(Op.IS_NOT_NULL,property);
    }

    public ConditionBuilder isNull(String property){
        return doNull(Op.IS_NULL,property);
    }

    public ConditionBuilder x(String sqlSegment, Object... values){

        if (SqliStringUtil.isNullOrEmpty(sqlSegment)){
            isOr();
            return instance;
        }

        String sql = normalizeSql(sqlSegment);

        Object[] arr = null;
        if (values != null){
            int length = values.length;
            arr = new Object[length];
            for (int i=0; i subList = new ArrayList<>();
        bb.setSubList(subList);
        this.add(bb);

        this.tempList = subList;
        this.getSubsList().add(subList);

        return instance;
    }
    
    public ConditionBuilder endSub(){
        isOr();
        int size = getSubsList().size();
        if (--size >= 0)
            getSubsList().remove(size);
        if (--size >= 0){
            this.tempList = getSubsList().get(size);
        }else {
            this.tempList = null;
        }

        return this.instance;
    }


    private ConditionBuilder doGle(Op p, String property, Object value) {
        if (value == null){
            isOr();
            return instance;
        }
        if (SqliStringUtil.isNullOrEmpty(value)){
            isOr();
            return instance;
        }
        Bb bb = new Bb(isOr());
        bb.setP(p);
        bb.setKey(property);
        bb.setValue(value);
        this.add(bb);

        return instance;
    }

    private ConditionBuilder doLike(Op p, String property, String likeWalue){

        Bb bb = new Bb(isOr());
        bb.setP(p);
        bb.setKey(property);
        bb.setValue(likeWalue);
        this.add(bb);
        return instance;
    }


    private ConditionBuilder doIn(Op p, String property, List list){

        if (list == null || list.isEmpty()){
            isOr();
            return instance;
        }

        List