Support PostgreSQL DO wrappers with preserved routine body literals

Author: minleejae

src/main/java/net/sf/jsqlparser/parser/feature/Feature.java

@@ -457,7 +457,7 @@ public enum Feature {
      */
     oracleBlock,
 
-    execute, executeExec, executeCall, executeExecute,
+    execute, executeExec, executeCall, executeExecute, doStatement,
 
     /**
      * SQL "EXECUTE" statement is allowed

src/main/java/net/sf/jsqlparser/statement/DoStatement.java

@@ -0,0 +1,78 @@
+/*-
+ * #%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;
+
+import java.util.function.Consumer;
+import net.sf.jsqlparser.expression.StringValue;
+
+/** A PostgreSQL anonymous routine. The language-specific body remains a string literal. */
+public class DoStatement implements Statement {
+    private StringValue code;
+    private String language;
+    private boolean languageBeforeCode;
+
+    public StringValue getCode() {
+        return code;
+    }
+
+    public void setCode(StringValue code) {
+        this.code = code;
+    }
+
+    public DoStatement withCode(StringValue code) {
+        setCode(code);
+        return this;
+    }
+
+    /** Returns the explicit language, or null when omitted. */
+    public String getLanguage() {
+        return language;
+    }
+
+    public void setLanguage(String language) {
+        this.language = language;
+    }
+
+    public DoStatement withLanguage(String language) {
+        setLanguage(language);
+        return this;
+    }
+
+    public boolean isLanguageBeforeCode() {
+        return languageBeforeCode;
+    }
+
+    public void setLanguageBeforeCode(boolean languageBeforeCode) {
+        this.languageBeforeCode = languageBeforeCode;
+    }
+
+    public StringBuilder appendTo(StringBuilder builder, Consumer<StringValue> codePrinter) {
+        builder.append("DO ");
+        if (languageBeforeCode && language != null) {
+            builder.append("LANGUAGE ").append(language).append(' ');
+        }
+        codePrinter.accept(code);
+        if (!languageBeforeCode && language != null) {
+            builder.append(" LANGUAGE ").append(language);
+        }
+        return builder;
+    }
+
+    @Override
+    public <T, S> T accept(StatementVisitor<T> visitor, S context) {
+        return visitor.visit(this, context);
+    }
+
+    @Override
+    public String toString() {
+        StringBuilder builder = new StringBuilder();
+        return appendTo(builder, value -> builder.append(value)).toString();
+    }
+}

src/main/java/net/sf/jsqlparser/statement/StatementFeatureVisitor.java

@@ -598,6 +598,13 @@ public <S> Void visit(Execute execute, S context) {
         return null;
     }
 
+    @Override
+    public <S> Void visit(DoStatement statement, S context) {
+        analysis.claimTopLevel();
+        analysis.opaque("DO");
+        return null;
+    }
+
     @Override
     public <S> Void visit(CreateFunctionalStatement createFunctionalStatement, S context) {
         analysis.claimTopLevel();

src/main/java/net/sf/jsqlparser/statement/StatementVisitor.java

@@ -210,6 +210,14 @@ default void visit(Statements statements) {
 
     <S> T visit(Execute execute, S context);
 
+    default <S> T visit(DoStatement statement, S context) {
+        return null;
+    }
+
+    default void visit(DoStatement statement) {
+        this.visit(statement, null);
+    }
+
     default void visit(Execute execute) {
         this.visit(execute, null);
     }

src/main/java/net/sf/jsqlparser/statement/StatementVisitorAdapter.java

@@ -423,6 +423,12 @@ public <S> T visit(Execute execute, S context) {
         return null;
     }
 
+    @Override
+    public <S> T visit(DoStatement statement, S context) {
+        expressionVisitor.visitExpression(statement.getCode(), context);
+        return null;
+    }
+
     @Override
     public <S> T visit(LockStatement lock, S context) {
 

src/main/java/net/sf/jsqlparser/util/TablesNamesFinder.java

@@ -91,6 +91,7 @@
 import net.sf.jsqlparser.statement.CreateFunctionalStatement;
 import net.sf.jsqlparser.statement.DeclareStatement;
 import net.sf.jsqlparser.statement.DescribeStatement;
+import net.sf.jsqlparser.statement.DoStatement;
 import net.sf.jsqlparser.statement.ExplainStatement;
 import net.sf.jsqlparser.statement.IfElseStatement;
 import net.sf.jsqlparser.statement.OutputClause;
@@ -1564,6 +1565,12 @@ public <S> Void visit(CreateIndex createIndex, S context) {
         return null;
     }
 
+    @Override
+    public <S> Void visit(DoStatement statement, S context) {
+        throwUnsupported(statement);
+        return null;
+    }
+
     @Override
     public void visit(CreateIndex createIndex) {
         StatementVisitor.super.visit(createIndex);

src/main/java/net/sf/jsqlparser/util/deparser/StatementDeParser.java

@@ -38,6 +38,7 @@
 import net.sf.jsqlparser.statement.CreateFunctionalStatement;
 import net.sf.jsqlparser.statement.DeclareStatement;
 import net.sf.jsqlparser.statement.DescribeStatement;
+import net.sf.jsqlparser.statement.DoStatement;
 import net.sf.jsqlparser.statement.ExplainStatement;
 import net.sf.jsqlparser.statement.IfElseStatement;
 import net.sf.jsqlparser.statement.PurgeStatement;
@@ -151,6 +152,11 @@ public <S> StringBuilder visit(CreateIndex createIndex, S context) {
         return builder;
     }
 
+    @Override
+    public <S> StringBuilder visit(DoStatement statement, S context) {
+        return statement.appendTo(builder, code -> code.accept(expressionDeParser, context));
+    }
+
     @Override
     public <S> StringBuilder visit(CreateTable createTable, S context) {
         CreateTableDeParser createTableDeParser = new CreateTableDeParser(this, builder);

src/main/java/net/sf/jsqlparser/util/validation/feature/PostgresqlVersion.java

@@ -27,6 +27,7 @@ public enum PostgresqlVersion implements Version {
             EnumSet.of(// supported if used with jdbc
                     Feature.jdbcParameter,
                     Feature.jdbcNamedParameter, // expressions
+                    Feature.doStatement,
                     Feature.exprLike,
                     Feature.exprSimilarTo, // https://www.postgresql.org/docs/current/sql-select.html
                     Feature.select,

src/main/java/net/sf/jsqlparser/util/validation/validator/StatementValidator.java

@@ -35,6 +35,7 @@
 import net.sf.jsqlparser.statement.Block;
 import net.sf.jsqlparser.statement.Commit;
 import net.sf.jsqlparser.statement.CreateFunctionalStatement;
+import net.sf.jsqlparser.statement.DoStatement;
 import net.sf.jsqlparser.statement.DeclareStatement;
 import net.sf.jsqlparser.statement.DescribeStatement;
 import net.sf.jsqlparser.statement.ExplainStatement;
@@ -232,6 +233,12 @@ public <S> Void visit(Execute execute, S context) {
         return null;
     }
 
+    @Override
+    public <S> Void visit(DoStatement statement, S context) {
+        validateFeature(Feature.doStatement);
+        return null;
+    }
+
     @Override
     public <S> Void visit(SetStatement set, S context) {
         getValidator(SetStatementValidator.class).validate(set);

src/main/jjtree/net/sf/jsqlparser/parser/JSqlParserCC.jjt

@@ -2764,6 +2764,9 @@ Statement SingleStatement() :
             |
             stm = Execute()
             |
+            LOOKAHEAD({ getToken(1).kind == K_DO && Dialect.POSTGRESQL.name().equals(getAsString(Feature.dialect)) })
+            stm = DoStatement()
+            |
             stm = Set()
             |
             stm = Reset()
@@ -2883,6 +2886,26 @@ OracleBlock.ExceptionHandler OracleExceptionHandler():
     <K_THEN> body=OracleBlockStatements() { handler.setStatements(body); return handler; }
 }
 
+DoStatement DoStatement():
+{
+    DoStatement statement = new DoStatement();
+    String language;
+    Token code;
+}
+{
+    <K_DO>
+    (
+        LOOKAHEAD({ isKeywordAhead("LANGUAGE") })
+        AccessKeyword("LANGUAGE") language=RelObjectName() code=<S_CHAR_LITERAL>
+        { statement.setLanguage(language); statement.setLanguageBeforeCode(true); }
+        |
+        code=<S_CHAR_LITERAL>
+        [ LOOKAHEAD({ isKeywordAhead("LANGUAGE") }) AccessKeyword("LANGUAGE")
+            language=RelObjectName() { statement.setLanguage(language); } ]
+    )
+    { statement.setCode(new StringValue(code.image)); return statement; }
+}
+
 Block Block() #Block : {
     Statements stmts = new Statements();
     List<Statement> list = new ArrayList<Statement>();