JSqlParser/src/main/java/net/sf/jsqlparser/statement/select/WithSearchClause.java at 66dac44ebd01f96be31441f2fdea28b93f037da3 · dengliming/JSqlParser · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
/*-
 * #%L
 * JSQLParser library
 * %%
 * Copyright (C) 2004 - 2026 JSQLParser
 * %%
 * Dual licensed under GNU LGPL 2.1 or Apache License 2.0
 * #L%
 */
package net.sf.jsqlparser.statement.select;

import java.io.Serializable;
import java.util.Collection;

import net.sf.jsqlparser.expression.operators.relational.ExpressionList;
import net.sf.jsqlparser.schema.Column;

public class WithSearchClause implements Serializable {
    public enum SearchOrder {
        BREADTH, DEPTH
    }

    private SearchOrder searchOrder;
    private ExpressionList<Column> searchColumns;
    private String sequenceColumnName;

    public WithSearchClause() {}

    public WithSearchClause(SearchOrder searchOrder, ExpressionList<Column> searchColumns,
            String sequenceColumnName) {
        this.searchOrder = searchOrder;
        this.searchColumns = searchColumns;
        this.sequenceColumnName = sequenceColumnName;
    }

    public SearchOrder getSearchOrder() {
        return searchOrder;
    }

    public void setSearchOrder(SearchOrder searchOrder) {
        this.searchOrder = searchOrder;
    }

    public WithSearchClause withSearchOrder(SearchOrder searchOrder) {
        this.setSearchOrder(searchOrder);
        return this;
    }

    public ExpressionList<Column> getSearchColumns() {
        return searchColumns;
    }

    public void setSearchColumns(ExpressionList<Column> searchColumns) {
        this.searchColumns = searchColumns;
    }

    public WithSearchClause withSearchColumns(ExpressionList<Column> searchColumns) {
        this.setSearchColumns(searchColumns);
        return this;
    }

    public WithSearchClause addSearchColumns(Column... searchColumns) {
        ExpressionList<Column> collection =
                getSearchColumns() != null ? getSearchColumns() : new ExpressionList<>();
        collection.addExpressions(searchColumns);
        return this.withSearchColumns(collection);
    }

    public WithSearchClause addSearchColumns(Collection<? extends Column> searchColumns) {
        ExpressionList<Column> collection =
                getSearchColumns() != null ? getSearchColumns() : new ExpressionList<>();
        collection.addAll(searchColumns);
        return this.withSearchColumns(collection);
    }

    public String getSequenceColumnName() {
        return sequenceColumnName;
    }

    public void setSequenceColumnName(String sequenceColumnName) {
        this.sequenceColumnName = sequenceColumnName;
    }

    public WithSearchClause withSequenceColumnName(String sequenceColumnName) {
        this.setSequenceColumnName(sequenceColumnName);
        return this;
    }

    @Override
    public String toString() {
        return new StringBuilder()
                .append("SEARCH ")
                .append(searchOrder)
                .append(" FIRST BY ")
                .append(Select.getStringList(searchColumns))
                .append(" SET ")
                .append(sequenceColumnName)
                .toString();
    }
}