jaskell.sql.With Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jaskell-java8 Show documentation
This is a utils library for java 8 project. It include parsec combinators and sql generators library.
There is a newer version: 2.9.2
package jaskell.sql;

import jaskell.script.Directive;
import jaskell.script.Parameter;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

public class With extends Query implements ThenSelect {
    Name _name;
    List _fields = new ArrayList<>();

    With(){

    }

    With(String name){
        _name = new Name(name);
    }

    With(Name name){
        _name = name;
    }

    With(String name, String fs){
        _name = new Name(name);
        _fields.addAll(Arrays.stream(fs.split(",")).map(Name::new).collect(Collectors.toList()));
    }

    With(Name name, String... fs){
        _name = name;
        _fields.addAll(Arrays.stream(fs).map(Name::new).collect(Collectors.toList()));
    }

    With(Name name, Name... fs){
        _name = name;
        _fields.addAll(Arrays.asList(fs));
    }

    public Name name(){
        return _name;
    }

    public With name(String name){
        _name = new Name(name);
        return this;
    }

    public With name(Name name){
        _name = name;
        return this;
    }

    public As as(Query query){
        As re =  new As(query);
        re._prefix = this;
        return re;
    }

    Recursive recursive(){
        Recursive re = new Recursive();
        re._prefix = this;
        return re;
    }

    @Override
    public String script() {
        if(_fields.isEmpty()) {
            return String.format("WITH %s", _name.script());
        }else{
            return String.format("WITH %s(%s)", _name.script(),
                    _fields.stream().map(Directive::script).collect(Collectors.joining(", ")));
        }
    }

    @Override
    public List> parameters() {
        return new ArrayList<>();
    }

    public static class Recursive extends With {
        With _prefix;

        Recursive(){

        }

        Recursive(String name, String fs){
            super(name, fs);
        }

        Recursive(Name name, String... fs){
            super(name, fs);
        }

        Recursive(Name name, Name... fs){
            super(name, fs);
        }

        @Override
        public String script() {
            if(_fields.isEmpty()) {
                return String.format("WITH RECURSIVE %s", _name.script());
            }else{
                return String.format("WITH RECURSIVE %s(%s)", _name.script(),
                        _fields.stream().map(Directive::script).collect(Collectors.joining(", ")));
            }
        }
    }

    public static class As implements Directive {
        Directive _prefix;
        Query _query;
        As(Query query){
            _query = query;
        }

        @Override
        public String script() {
            return String.format("%s AS (%s)", _prefix.script(), _query.script());
        }

        @Override
        public List> parameters() {
            List> re = _prefix.parameters();
            re.addAll(_query.parameters());
            return re;
        }

        public CommonTableExpression cte(String name){
            CommonTableExpression re = new CommonTableExpression(name);
            re._prefix = this;
            return re;
        }

        public CommonTableExpression cte(Name name){
            CommonTableExpression re = new CommonTableExpression(name);
            re._prefix = this;
            return re;
        }


        public CommonTableExpression cte(String name, String fs){
            CommonTableExpression re = new CommonTableExpression(name, fs);
            re._prefix = this;
            return re;
        }

        public CommonTableExpression cte(Name name, String fs){
            CommonTableExpression re = new CommonTableExpression(name, fs);
            re._prefix = this;
            return re;
        }

        public CommonTableExpression cte(String name, String ... fs){
            CommonTableExpression re = new CommonTableExpression(name, fs);
            re._prefix = this;
            return re;
        }

        public CommonTableExpression cte(Name name, String ... fs){
            CommonTableExpression re = new CommonTableExpression(name, fs);
            re._prefix = this;
            return re;
        }

        public CommonTableExpression cte(String name, Name ... fs){
            CommonTableExpression re = new CommonTableExpression(name, fs);
            re._prefix = this;
            return re;
        }

        public CommonTableExpression cte(Name name, Name ... fs){
            CommonTableExpression re = new CommonTableExpression(name, fs);
            re._prefix = this;
            return re;
        }

        public CommonTableQuery query(Query query){
            CommonTableQuery re = new CommonTableQuery();
            re._prefix = this;
            re._query = query;
            return re;
        }

        public Select select(String names){
            Select re = new Select(names);
            re._prefix = this;
            return re;
        }

        public Select select(String... names){
            Select re = new Select(names);
            re._prefix = this;
            return re;
        }

        public Select select(Directive... names){
            Select re = new Select(names);
            re._prefix = this;
            return re;
        }

        public Insert insert() {
            Insert re = new Insert();
            re._prefix = this;
            return re;
        }

        public Update update(String name) {
            Update re = new Update(name);
            re._prefix = this;
            return re;
        }

        public Update update(Name name) {
            Update re = new Update(name);
            re._prefix = this;
            return re;
        }

        public Delete delete() {
            Delete re = new Delete();
            re._prefix = this;
            return re;
        }

    }

    public static class Insert extends jaskell.sql.Insert {
        Directive _prefix;

        @Override
        public String script() {
            return String.format("%s %s", _prefix.script(), super.script());
        }

        @Override
        public List> parameters() {
            List> re = _prefix.parameters();
            re.addAll(super.parameters());
            return re;
        }
    }

    public static class Delete extends jaskell.sql.Delete {
        Directive _prefix;

        @Override
        public String script() {
            return String.format("%s %s", _prefix.script(), super.script());
        }

        @Override
        public List> parameters() {
            List> re = _prefix.parameters();
            re.addAll(super.parameters());
            return re;
        }

    }

    public static class Update extends jaskell.sql.Update {
        Directive _prefix;

        Update(String name) {
            super(name);
        }

        Update(Name name) {
            super(name);
        }

        @Override
        public String script() {
            return String.format("%s %s", _prefix.script(), super.script());
        }

        @Override
        public List> parameters() {
            List> re = _prefix.parameters();
            re.addAll(super.parameters());
            return re;
        }
    }

    public static class CommonTableExpression implements Directive {
        Directive _prefix;
        Name _name;
        List _fields = new ArrayList<>();

        public As as(Query query){
            As re =  new As(query);
            re._prefix = this;
            return re;
        }

        CommonTableExpression(String name){
            this._name = new Name(name);
        }

        CommonTableExpression(Name name){
            this._name = name;
        }

        CommonTableExpression(String name, String fs){
            this._name = new Name(name);
            this._fields.addAll(Arrays.stream(fs.split(",")).map(Name::new).collect(Collectors.toList()));
        }

        CommonTableExpression(String name, String... fs){
            this._name = new Name(name);
            this._fields.addAll(Arrays.stream(fs).map(Name::new).collect(Collectors.toList()));
        }

        CommonTableExpression(String name, Name... fs){
            this._name = new Name(name);
            this._fields.addAll(Arrays.asList(fs));
        }

        CommonTableExpression(Name name, String fs){
            this._name = name;
            this._fields.addAll(Arrays.stream(fs.split(",")).map(Name::new).collect(Collectors.toList()));
        }

        CommonTableExpression(Name name, String... fs){
            this._name = name;
            this._fields.addAll(Arrays.stream(fs).map(Name::new).collect(Collectors.toList()));
        }

        CommonTableExpression(Name name, Name... fs){
            this._name = name;
            this._fields.addAll(Arrays.asList(fs));
        }

        @Override
        public String script() {
            return String.format("%s, %s", _prefix.script(), _name.script());
        }

        @Override
        public List> parameters() {
            return _prefix.parameters();
        }
    }

    public static class CommonTableQuery extends Query {
        Directive _prefix;
        Query _query;

        @Override
        public String script() {
            return String.format("%s %s", _prefix.script(), _query.script());
        }

        @Override
        public List> parameters() {
            List> re = _prefix.parameters();
            re.addAll(_query.parameters());
            return re;
        }
    }

}