Class DropDbStmt

fe/src/main/java/org/apache/impala/analysis/DropDbStmt.java:28–101

Represents a DROP [IF EXISTS] DATABASE [CASCADE | RESTRICT] statement

 * Represents a DROP [IF EXISTS] DATABASE [CASCADE | RESTRICT] statement
 */
public class DropDbStmt extends StatementBase {
  private final String dbName_;
  private final boolean ifExists_;
  private final boolean cascade_;

  // Server name needed for privileges. Set during analysis.
  private String serverName_;

  /**
   * Constructor for building the drop statement. If ifExists is true, an error will not
   * be thrown if the database does not exist. If cascade is true, all the tables in the
   * database will be dropped.
   */
  public DropDbStmt(String dbName, boolean ifExists, boolean cascade) {
    this.dbName_ = dbName;
    this.ifExists_ = ifExists;
    this.cascade_ = cascade;
  }

  @Override
  public String getParsedDb() { return dbName_; }
  public String getDb() { return dbName_; }
  public boolean getIfExists() { return ifExists_; }
  public boolean getCascade() { return cascade_; }

  @Override
  public String toSql(ToSqlOptions options) {
    StringBuilder sb = new StringBuilder("DROP DATABASE");
    if (ifExists_) sb.append(" IF EXISTS ");
    sb.append(getDb());
    if (cascade_) sb.append(" CASCADE");
    return sb.toString();
  }

  public TDropDbParams toThrift() {
    TDropDbParams params = new TDropDbParams();
    params.setDb(getDb());
    params.setIf_exists(getIfExists());
    params.setCascade(getCascade());
    params.setServer_name(serverName_);
    return params;
  }

  @Override
  public void analyze(Analyzer analyzer) throws AnalysisException {
    // Set the servername here if authorization is enabled because analyzer_ is not
    // available in the toThrift() method.
    serverName_ = analyzer.getServerName();
    // Fetch the owner information if the db exists.
    FeDb db = analyzer.getDb(dbName_, /*ThrowIfNotExists*/ false);
    String ownerUser = db == null ? null : db.getOwnerUser();
    if (ifExists_) {
      // Start with ANY privilege in case of IF EXISTS, and register DROP privilege
      // later only if the database exists. See IMPALA-8851 for more explanation.
      analyzer.registerPrivReq(builder ->
          builder.allOf(Privilege.ANY)
              .onDb(dbName_, ownerUser)
              .build());