JSqlParser/src/main/java/net/sf/jsqlparser/util/deparser/AlterDeParser.java at b2614e408d133efeecf50e044b8779d8609f13b5 · JSQLParser/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
101
102
103
104
105
106
107
108
109
110
/*-
 * #%L
 * JSQLParser library
 * %%
 * Copyright (C) 2004 - 2019 JSQLParser
 * %%
 * Dual licensed under GNU LGPL 2.1 or Apache License 2.0
 * #L%
 */
package net.sf.jsqlparser.util.deparser;

import net.sf.jsqlparser.statement.alter.Alter;
import net.sf.jsqlparser.statement.alter.AlterForeignDataOptions;
import net.sf.jsqlparser.expression.ExpressionVisitor;
import net.sf.jsqlparser.statement.alter.AlterExpression;
import net.sf.jsqlparser.statement.alter.AlterExpressionPartition;
import net.sf.jsqlparser.statement.alter.AlterExpressionPrimaryKey;
import net.sf.jsqlparser.statement.alter.RelationAlterAction;
import net.sf.jsqlparser.statement.alter.AlterOperation;
import net.sf.jsqlparser.statement.select.PlainSelect;
import java.util.Iterator;

public class AlterDeParser extends AbstractDeParser<Alter> {
    private final ExpressionVisitor<StringBuilder> expressionVisitor;

    public AlterDeParser(StringBuilder buffer) {
        this(buffer, new StatementDeParser(buffer).getExpressionDeParser());
    }

    public AlterDeParser(StringBuilder buffer, ExpressionVisitor<StringBuilder> expressionVisitor) {
        super(buffer);
        this.expressionVisitor = expressionVisitor;
    }

    @Override
    public void deParse(Alter alter) {
        alter.appendTargetTo(builder);
        for (Iterator<AlterExpression> iterator = alter.getAlterExpressions().iterator(); iterator
                .hasNext();) {
            deParseAction(iterator.next());
            if (iterator.hasNext()) {
                builder.append(", ");
            }
        }
    }

    private void deParseAction(AlterExpression action) {
        if (action instanceof AlterForeignDataOptions) {
            ((AlterForeignDataOptions) action).appendTo(builder,
                    expression -> expression.accept(expressionVisitor, null));
            return;
        }
        if (action instanceof net.sf.jsqlparser.statement.alter.AlterExpressionOrderBy) {
            ((net.sf.jsqlparser.statement.alter.AlterExpressionOrderBy) action).appendTo(builder,
                    expression -> expression.accept(expressionVisitor, null));
            return;
        }
        if (action instanceof RelationAlterAction) {
            ((RelationAlterAction) action).appendTo(builder,
                    expression -> expression.accept(expressionVisitor, null));
            return;
        }
        if (action instanceof AlterExpressionPartition) {
            ((AlterExpressionPartition) action).appendTo(builder,
                    expression -> expression.accept(expressionVisitor, null));
            return;
        }
        if (action instanceof AlterExpressionPrimaryKey) {
            ((AlterExpressionPrimaryKey) action).appendTo(builder,
                    expression -> expression.accept(expressionVisitor, null));
            return;
        }
        if (action.getOperation() == AlterOperation.ADD && action.getIndex() != null
                && action.getConstraintType() == null) {
            builder.append(action.getOperation()).append(' ');
            new TableElementDeParser(builder, expressionVisitor).deParse(action.getIndex());
            if (action.getConstraints() != null && !action.getConstraints().isEmpty()) {
                builder.append(' ')
                        .append(PlainSelect.getStringList(action.getConstraints(), false, false));
            }
            if (action.getUseEqual()) {
                builder.append('=');
            }
            deParseTail(action);
            return;
        }
        if (!action.getColumnDefaults().isEmpty()) {
            action.appendColumnActionTo(builder,
                    expression -> expression.accept(expressionVisitor, null));
            return;
        }
        if (action.getColDataTypeList() != null) {
            action.appendColumnDefinitionsTo(builder,
                    expression -> expression.accept(expressionVisitor, null));
        } else {
            builder.append(action);
        }
    }

    private void deParseTail(AlterExpression action) {
        if (action.getParameters() != null && !action.getParameters().isEmpty()) {
            builder.append(' ')
                    .append(PlainSelect.getStringList(action.getParameters(), false, false));
        }
        if (action.getIndex() != null && action.getIndex().getCommentText() != null) {
            builder.append(" COMMENT ").append(action.getIndex().getCommentText());
        }
    }

}