Modify methods to support having in Siddhi app string

MaheshMadushan · Jun 7, 2023 · ac7ede2 · ac7ede2
1 parent fb74c5d
commit ac7ede2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/main/java/SiddhiAppComposites/SiddhiApp.java b/src/main/java/SiddhiAppComposites/SiddhiApp.java
@@ -13,6 +13,8 @@
 import SiddhiAppComposites.Statement.From.IFromStatement;
 import SiddhiAppComposites.Statement.GroupBy.GroupByStatement;
 import SiddhiAppComposites.Statement.GroupBy.IGroupByStatement;
+import SiddhiAppComposites.Statement.Having.HavingExpression;
+import SiddhiAppComposites.Statement.Having.IHavingExpression;
 import SiddhiAppComposites.Statement.Insert.IInsertStatement;
 import SiddhiAppComposites.Statement.Insert.InsertStatement;
 import SiddhiAppComposites.Statement.JoinStatement.IJoinStatement;
@@ -41,6 +43,7 @@ public class SiddhiApp {
     private final IFromStatement joinStatement = new FromStatement(); // from statement
 
     private final IFilterExpression filterExpression = new FilterExpression(); // filter statement
+    private final IHavingExpression havingExpression = new HavingExpression(); // filter statement
     private final IInsertStatement insertStatement = new InsertStatement(); // insert into statement
     private final App annotationApp; // Annotation @app
     private final ISource annotationSource; // Annotation @source
@@ -102,6 +105,10 @@ public void addGroupByExpression(String column){
         groupByStatement.addAttribute(column);
     }
 
+    public void addHavingExpression(String symbol){
+        ((HavingExpression) this.havingExpression).addSymbol(symbol);
+    }
+
     public void setStreamNamePrefix(String inputOutputStreamNamePrefix) {
         this.inputOutputStreamNamePrefix = inputOutputStreamNamePrefix;
     }
@@ -225,6 +232,9 @@ public String getSiddhiAppStringRepresentation(){
             groupByStatement.setColumnWithStreamMap(selectStatement.getAttributesListWithStreamName());
             stringSiddhiApp.append(groupByStatement.getSiddhiAppCompositeAsString());
         }
+        if (havingExpression.containsExpressions()) {
+            stringSiddhiApp.append(havingExpression.getSiddhiAppCompositeAsString());
+        }
         stringSiddhiApp.append(insertStatement.getSiddhiAppCompositeAsString());
         return stringSiddhiApp.toString();
     }