com.oceanbase.tools.sqlparser.statement.droptable.DropTable Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of ob-sql-parser Show documentation
ob-sql-parser parses an SQL statement into a Java classes.
There is a newer version: 1.3.0
/*
 * Copyright (c) 2023 OceanBase.
 *
 * Licensed 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 com.oceanbase.tools.sqlparser.statement.droptable;

import java.util.Collections;
import java.util.List;

import org.antlr.v4.runtime.ParserRuleContext;
import org.apache.commons.lang3.StringUtils;

import com.oceanbase.tools.sqlparser.statement.BaseStatement;
import com.oceanbase.tools.sqlparser.statement.common.RelationFactor;

import lombok.EqualsAndHashCode;
import lombok.Getter;
import lombok.NonNull;
import lombok.Setter;

/**
 * @Author: Lebie
 * @Date: 2024/8/21 17:17
 * @Description: []
 */
@Getter
@Setter
@EqualsAndHashCode(callSuper = false)
public class DropTable extends BaseStatement {
    private boolean temporary;
    private boolean materialized;
    private boolean ifExists;
    private boolean cascade;
    private boolean cascadeConstraints;
    private boolean restrict;
    private boolean purge;
    private TableList tableList;
    private final List relations;

    public DropTable(@NonNull ParserRuleContext context, @NonNull TableList tableList, boolean temporary,
            boolean materialized,
            boolean ifExists, boolean cascade, boolean restrict) {
        super(context);
        this.tableList = tableList;
        this.relations = tableList.getRelations();
        this.temporary = temporary;
        this.materialized = materialized;
        this.ifExists = ifExists;
        this.cascade = cascade;
        this.restrict = restrict;
    }

    public DropTable(@NonNull ParserRuleContext context, @NonNull RelationFactor relation, boolean cascadeConstraints,
            boolean purge) {
        super(context);
        this.relations = Collections.singletonList(relation);
        this.cascadeConstraints = cascadeConstraints;
        this.purge = purge;
    }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder("DROP ");
        if (temporary) {
            sb.append("TEMPORARY ");
        }
        if (materialized) {
            sb.append("MATERIALIZED ");
        }
        sb.append("TABLE ");
        if (ifExists) {
            sb.append("IF EXISTS ");
        }
        if (relations != null) {
            sb.append(StringUtils.join(relations, ','));
        }
        if (cascade) {
            sb.append(" CASCADE ");
        } else if (restrict) {
            sb.append(" RESTRICT ");
        }
        if (cascadeConstraints) {
            sb.append(" CASCADE CONSTRAINTS ");
        }
        if (purge) {
            sb.append(" PURGE");
        }
        return sb.toString();
    }
}