io.xream.sqli.builder.CriteriaBuilder Maven / Gradle / Ivy
/*
* 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.page.Paged;
import io.xream.sqli.parser.Parsed;
import io.xream.sqli.parser.Parser;
import io.xream.sqli.util.SqliStringUtil;
import java.util.ArrayList;
import java.util.List;
import java.util.Objects;
/**
* @author Sim
*/
public class CriteriaBuilder extends ConditionBuilder {
private Criteria criteria;
private PageBuilder pageBuilder;
protected SourceScript sourceScriptTemp;
public CriteriaBuilder routeKey(Object routeKey) {
this.criteria.setRouteKey(routeKey);
return this;
}
public PageBuilder paged() {
if ( this.pageBuilder != null)
return this.pageBuilder;
this.pageBuilder = new PageBuilder() {
@Override
public PageBuilder ignoreTotalRows() {
criteria.setTotalRowsIgnored(true);
return this;
}
@Override
public PageBuilder rows(int rows) {
criteria.setRows(rows);
return this;
}
@Override
public PageBuilder page(int page) {
criteria.setPage(page);
return this;
}
@Override
public PageBuilder last(long last) {
criteria.setLast(last);
return this;
}
};
return this.pageBuilder;
}
public void paged(Paged paged) {
criteria.paged(paged);
}
public CriteriaBuilder sortIn(String porperty, List inList) {
if (Objects.nonNull(inList) && inList.size() > 0) {
KV kv = new KV(porperty, inList);
List fixedSortList = criteria.getFixedSortList();
if (fixedSortList == null){
fixedSortList = new ArrayList<>();
criteria.setFixedSortList(fixedSortList);
}
fixedSortList.add(kv);
}
return this;
}
public CriteriaBuilder sort(String orderBy, Direction direction) {
if (SqliStringUtil.isNullOrEmpty(orderBy))
return this;
List sortList = criteria.getSortList();
if (sortList == null) {
sortList = new ArrayList<>();
criteria.setSortList(sortList);
}
Sort sort = new Sort(orderBy, direction);
sortList.add(sort);
return this;
}
private CriteriaBuilder(Criteria criteria) {
super(criteria.getBbList());
this.criteria = criteria;
}
public static CriteriaBuilder builder(Class clz) {
Criteria criteria = new Criteria();
criteria.setClzz(clz);
CriteriaBuilder builder = new CriteriaBuilder(criteria);
if (criteria.getParsed() == null) {
Parsed parsed = Parser.get(clz);
criteria.setParsed(parsed);
}
return builder;
}
public static ResultMapBuilder resultMapBuilder() {
Criteria.ResultMapCriteria resultMapCriteria = new Criteria.ResultMapCriteria();
return new ResultMapBuilder(resultMapCriteria);
}
public Class getClz() {
return this.criteria.getClzz();
}
protected Criteria get() {
return this.criteria;
}
public Criteria build(){
return this.criteria;
}
public void clear(){
this.criteria = null;
}
public static final class ResultMapBuilder extends CriteriaBuilder {
private SourceScriptBuilder sourceScriptBuilder = new SourceScriptBuilder() {
@Override
public SourceScriptBuilder source(String source) {
sourceScriptTemp.setSource(source);
return this;
}
@Override
public SourceScriptBuilder sub(Sub sub) {
ResultMapBuilder subBuilder = CriteriaBuilder.resultMapBuilder();
sub.buildBy(subBuilder);
Criteria.ResultMapCriteria resultMapCriteria = subBuilder.build();
sourceScriptTemp.setSubCriteria(resultMapCriteria);
subBuilder.clear();
return this;
}
@Override
public SourceScriptBuilder alia(String alia) {
sourceScriptTemp.setAlia(alia);
return this;
}
@Override
public SourceScriptBuilder join(JoinType joinType) {
sourceScriptTemp.setJoinType(joinType);
return this;
}
@Override
public SourceScriptBuilder join(String joinStr) {
sourceScriptTemp.setJoinStr(joinStr);
return this;
}
@Override
public SourceScriptBuilder on(String key, JoinFrom joinFrom) {
if (key.contains("."))
throw new IllegalArgumentException("On key can not contains '.'");
On on = new On();
on.setKey(key);
on.setOp(Op.EQ.sql());
on.setJoinFrom(joinFrom);
sourceScriptTemp.setOn(on);
return this;
}
@Override
public SourceScriptBuilder on(String key, Op op, JoinFrom joinFrom) {
if (key.contains("."))
throw new IllegalArgumentException("On key can not contains '.'");
On on = new On();
on.setKey(key);
on.setOp(op.sql());
on.setJoinFrom(joinFrom);
sourceScriptTemp.setOn(on);
return this;
}
@Override
public ConditionBuilder more() {
List bbList = new ArrayList<>();
sourceScriptTemp.setBbList(bbList);
return builder(bbList);
}
};
public SourceScriptBuilder sourceBuilder() {
sourceScriptTemp = new SourceScript();
get().getSourceScripts().add(sourceScriptTemp);
return this.sourceScriptBuilder;
}
public ResultMapBuilder withoutOptimization() {
get().setWithoutOptimization(true);
return this;
}
@Override
protected Criteria.ResultMapCriteria get() {
return (Criteria.ResultMapCriteria) super.get();
}
@Override
public Criteria.ResultMapCriteria build() {
return (Criteria.ResultMapCriteria) super.get();
}
public ResultMapBuilder(Criteria criteria) {
super(criteria);
}
public ResultMapBuilder resultKey(String resultKey) {
if (SqliStringUtil.isNullOrEmpty(resultKey))
return this;
get().getResultKeyList().add(resultKey);
return this;
}
/**
*
* @param resultKey
* @param alia
* @return resultKey set by framework, not alia, (temporaryRepository.findToCreate)
*
*/
public ResultMapBuilder resultKey(String resultKey, String alia) {
if (SqliStringUtil.isNullOrEmpty(resultKey))
return this;
Objects.requireNonNull(alia,"resultKeyAssignedAlia(), alia can not null");
get().getResultKeyAssignedAliaList().add(new KV(resultKey,alia));
return this;
}
public ResultMapBuilder resultWithDottedKey() {
get().setResultWithDottedKey(true);
return this;
}
/**
* @param functionScript FUNCTION(?,?)
* @param values "test", 1000
*/
public ResultMapBuilder resultKeyFunction(ResultKeyAlia resultKeyAlia, String functionScript, String... values) {
if (SqliStringUtil.isNullOrEmpty(functionScript) || values == null)
return this;
Objects.requireNonNull(resultKeyAlia, "function no alia");
Objects.requireNonNull(resultKeyAlia.getAlia());
FunctionResultKey functionResultKey = new FunctionResultKey();
functionResultKey.setScript(functionScript);
functionResultKey.setAlia(resultKeyAlia.getAlia());
functionResultKey.setValues(values);
get().getResultFunctionList().add(functionResultKey);
return this;
}
public ResultMapBuilder sourceScript(String sourceScript) {
if (SqliStringUtil.isNullOrEmpty(sourceScript))
return this;
get().setSourceScript(sourceScript);
return this;
}
public ResultMapBuilder distinct(String... objs) {
if (objs == null)
throw new IllegalArgumentException("distinct non resultKey");
Criteria.ResultMapCriteria resultMapped = get();
Distinct distinct = resultMapped.getDistinct();
if (Objects.isNull(distinct)) {
distinct = new Distinct();
resultMapped.setDistinct(distinct);
}
for (String obj : objs) {
distinct.add(obj);
}
return this;
}
public ResultMapBuilder groupBy(String property) {
get().setGroupBy(property);
return this;
}
public ResultMapBuilder xAggr(String function, Object...values) {
List list = get().getAggrList();
if (list == null) {
list = new ArrayList<>();
get().setAggrList(list);
}
Bb bb = new Bb();
bb.setKey(function);
bb.setValue(values);
bb.setP(Op.X_AGGR);
list.add(bb);
return this;
}
public ResultMapBuilder having(ResultKeyAlia resultKeyAlia, Op op, Object value) {
Having having = Having.of(op,value);
having.setAliaOrFunction(resultKeyAlia.getKey());
get().getHavingList().add(having);
return this;
}
public ResultMapBuilder having(String functionScript, Op op, Object value) {
Having having = Having.of(op,value);
having.setAliaOrFunction(functionScript);
get().getHavingList().add(having);
return this;
}
public ResultMapBuilder reduce(ReduceType type, String property) {
Reduce reduce = new Reduce();
reduce.setType(type);
reduce.setProperty(property);
get().getReduceList().add(reduce);
return this;
}
@Override
public ResultMapBuilder sort(String orderBy, Direction direction){
return (ResultMapBuilder) super.sort(orderBy,direction);
}
/**
* @param type
* @param property
* @param having paged().totalRowsIgnored(true), if isTotalRowsIgnored == false，will throw Exception
*/
public ResultMapBuilder reduce(ReduceType type, String property, Having having) {
Reduce reduce = new Reduce();
reduce.setType(type);
reduce.setProperty(property);
reduce.setHaving(having);
get().getReduceList().add(reduce);
return this;
}
public ResultMapBuilder eq(String key, Object value) {
return (ResultMapBuilder) super.eq(key,value);
}
public ResultMapBuilder gt(String key, Object value) {
return (ResultMapBuilder) super.gt(key,value);
}
public ResultMapBuilder gte(String key, Object value) {
return (ResultMapBuilder) super.gte(key,value);
}
public ResultMapBuilder lt(String key, Object value) {
return (ResultMapBuilder) super.lt(key,value);
}
public ResultMapBuilder lte(String key, Object value) {
return (ResultMapBuilder) super.lte(key,value);
}
public ResultMapBuilder ne(String property, Object value) {
return (ResultMapBuilder) super.ne(property, value);
}
public ResultMapBuilder like(String property, String value) {
return (ResultMapBuilder) super.like(property, value);
}
public ResultMapBuilder likeRight(String property, String value) {
return (ResultMapBuilder) super.likeRight(property, value);
}
public ResultMapBuilder notLike(String property, String value) {
return (ResultMapBuilder) super.notLike(property, value);
}
public ResultMapBuilder in(String property, List list) {
return (ResultMapBuilder) super.in(property,list);
}
public ResultMapBuilder nin(String property, List list) {
return (ResultMapBuilder) super.nin(property,list);
}
public ResultMapBuilder nonNull(String property){
return (ResultMapBuilder) super.nonNull(property);
}
public ResultMapBuilder isNull(String property){
return (ResultMapBuilder) super.isNull(property);
}
public ResultMapBuilder x(String sqlSegment){
return (ResultMapBuilder) super.x(sqlSegment);
}
public ResultMapBuilder x(String sqlSegment, Object...values){
return (ResultMapBuilder) super.x(sqlSegment, values);
}
public ResultMapBuilder beginSub(){
return (ResultMapBuilder) super.beginSub();
}
public ResultMapBuilder endSub(){
return (ResultMapBuilder) super.endSub();
}
public void clear(){
super.clear();
}
}
} © 2015 - 2025 Weber Informatics LLC | Privacy Policy