Fix[2306] - Adds support for Trino UDF

Trino SQL allows you to specify User Defined functions in the WITH clause ahead of a statement. This adds support for this type of statement.

Author: David Hayes

src/main/java/net/sf/jsqlparser/statement/select/WithFunctionDeclaration.java

@@ -20,10 +20,10 @@ public class WithFunctionDeclaration implements Serializable {
     private String returnType;
     private Expression returnExpression;
 
-    public WithFunctionDeclaration() {
-    }
+    public WithFunctionDeclaration() {}
 
-    public WithFunctionDeclaration(String functionName, List<WithFunctionParameter> parameters, String returnType, Expression returnExpression) {
+    public WithFunctionDeclaration(String functionName, List<WithFunctionParameter> parameters,
+            String returnType, Expression returnExpression) {
         this.functionName = functionName;
         this.parameters = parameters;
         this.returnType = returnType;

src/main/java/net/sf/jsqlparser/statement/select/WithFunctionParameter.java

@@ -15,8 +15,7 @@ public class WithFunctionParameter implements Serializable {
     private String name;
     private String type; // e.g., INT
 
-    public WithFunctionParameter() {
-    }
+    public WithFunctionParameter() {}
 
     public WithFunctionParameter(String name, String type) {
         this.name = name;

src/main/java/net/sf/jsqlparser/statement/select/WithItem.java

@@ -130,7 +130,8 @@ public void setWithFunctionDeclaration(WithFunctionDeclaration withFunctionDecla
         this.withFunctionDeclaration = withFunctionDeclaration;
     }
 
-    public WithItem<K> withWithFunctionDeclaration(WithFunctionDeclaration withFunctionDeclaration) {
+    public WithItem<K> withWithFunctionDeclaration(
+            WithFunctionDeclaration withFunctionDeclaration) {
         this.setWithFunctionDeclaration(withFunctionDeclaration);
         return this;
     }

src/test/java/net/sf/jsqlparser/statement/select/WithFunctionDeclarationTest.java

@@ -41,7 +41,8 @@ void fullConstructorAndGetters() {
     void defaultConstructorAndSetters() {
         withFunctionDeclaration = new WithFunctionDeclaration();
         withFunctionDeclaration.setFunctionName(FUNCTION_NAME);
-        withFunctionDeclaration.setParameters(List.of(withFunctionParameter1, withFunctionParameter2));
+        withFunctionDeclaration
+                .setParameters(List.of(withFunctionParameter1, withFunctionParameter2));
         withFunctionDeclaration.setReturnType(RETURN_TYPE);
         withFunctionDeclaration.setReturnExpression(expression);
         assertThat(withFunctionDeclaration.getFunctionName()).isEqualTo(FUNCTION_NAME);
@@ -79,14 +80,17 @@ void toStringTestWithParameters() {
         withFunctionDeclaration = new WithFunctionDeclaration(FUNCTION_NAME,
                 List.of(withFunctionParameter1, withFunctionParameter2), RETURN_TYPE, expression);
 
-        assertThat(withFunctionDeclaration.toString()).isEqualTo("FUNCTION func1(param1 bigint, param2 double) RETURNS integer RETURN 1 + 1");
+        assertThat(withFunctionDeclaration.toString()).isEqualTo(
+                "FUNCTION func1(param1 bigint, param2 double) RETURNS integer RETURN 1 + 1");
     }
 
     @Test
     void toStringTestWithNoParameters() {
         when(expression.toString()).thenReturn("1 + 1");
-        withFunctionDeclaration = new WithFunctionDeclaration(FUNCTION_NAME, List.of(), RETURN_TYPE, expression);
+        withFunctionDeclaration =
+                new WithFunctionDeclaration(FUNCTION_NAME, List.of(), RETURN_TYPE, expression);
 
-        assertThat(withFunctionDeclaration.toString()).isEqualTo("FUNCTION func1() RETURNS integer RETURN 1 + 1");
+        assertThat(withFunctionDeclaration.toString())
+                .isEqualTo("FUNCTION func1() RETURNS integer RETURN 1 + 1");
     }
 }