fixes from testing

littlehorse-enterprises · Oct 30, 2024 · 24c507d · 24c507d
1 parent 55004ec
commit 24c507d
Show file tree

Hide file tree

Showing 4 changed files with 30 additions and 18 deletions.
diff --git a/sdk-java/src/main/java/io/littlehorse/sdk/wfsdk/WfRunVariable.java b/sdk-java/src/main/java/io/littlehorse/sdk/wfsdk/WfRunVariable.java
@@ -123,21 +123,21 @@ public interface WfRunVariable extends Serializable {
 
     WorkflowCondition isNotIn(Serializable rhs);
 
-    void assignTo(Serializable rhs);
+    void assignTo(Object rhs);
 
-    void add(Serializable rhs);
+    void add(Object rhs);
 
-    void subtract(Serializable rhs);
+    void subtract(Object rhs);
 
-    void divide(Serializable rhs);
+    void divide(Object rhs);
 
-    void multiply(Serializable rhs);
+    void multiply(Object rhs);
 
-    void extend(Serializable rhs);
+    void extend(Object rhs);
 
-    void removeIndex(Serializable rhs);
+    void removeIndex(Object rhs);
 
-    void removeIfPresent(Serializable rhs);
+    void removeIfPresent(Object rhs);
 
-    void removeKey(Serializable rhs);
+    void removeKey(Object rhs);
 }
diff --git a/sdk-java/src/main/java/io/littlehorse/sdk/wfsdk/WorkflowThread.java b/sdk-java/src/main/java/io/littlehorse/sdk/wfsdk/WorkflowThread.java
@@ -134,6 +134,10 @@ public interface WorkflowThread {
 
     WfRunVariable declareJsonObj(String name, Map<String, Object> defaultVal);
 
+    WfRunVariable declareBool(String name);
+
+    WfRunVariable declareBool(String name, boolean defaultVal);
+
     /**
      * Defines a Variable in the `ThreadSpec` and returns a handle to it.
      *

diff --git a/sdk-java/src/main/java/io/littlehorse/sdk/wfsdk/internal/WfRunVariableImpl.java b/sdk-java/src/main/java/io/littlehorse/sdk/wfsdk/internal/WfRunVariableImpl.java
@@ -161,47 +161,47 @@ public WorkflowConditionImpl isNotIn(Serializable rhs) {
     }
 
     @Override
-    public void assignTo(Serializable rhs) {
+    public void assignTo(Object rhs) {
         parent.mutate(this, VariableMutationType.ASSIGN, rhs);
     }
 
     @Override
-    public void divide(Serializable rhs) {
+    public void divide(Object rhs) {
         parent.mutate(this, VariableMutationType.DIVIDE, rhs);
     }
 
     @Override
-    public void add(Serializable rhs) {
+    public void add(Object rhs) {
         parent.mutate(this, VariableMutationType.ADD, rhs);
     }
 
     @Override
-    public void subtract(Serializable rhs) {
+    public void subtract(Object rhs) {
         parent.mutate(this, VariableMutationType.SUBTRACT, rhs);
     }
 
     @Override
-    public void multiply(Serializable rhs) {
+    public void multiply(Object rhs) {
         parent.mutate(this, VariableMutationType.MULTIPLY, rhs);
     }
 
     @Override
-    public void extend(Serializable rhs) {
+    public void extend(Object rhs) {
         parent.mutate(this, VariableMutationType.EXTEND, rhs);
     }
 
     @Override
-    public void removeIndex(Serializable rhs) {
+    public void removeIndex(Object rhs) {
         parent.mutate(this, VariableMutationType.REMOVE_INDEX, rhs);
     }
 
     @Override
-    public void removeKey(Serializable rhs) {
+    public void removeKey(Object rhs) {
         parent.mutate(this, VariableMutationType.REMOVE_KEY, rhs);
     }
 
     @Override
-    public void removeIfPresent(Serializable rhs) {
+    public void removeIfPresent(Object rhs) {
         parent.mutate(this, VariableMutationType.REMOVE_IF_PRESENT, rhs);
     }
 

diff --git a/sdk-java/src/main/java/io/littlehorse/sdk/wfsdk/internal/WorkflowThreadImpl.java b/sdk-java/src/main/java/io/littlehorse/sdk/wfsdk/internal/WorkflowThreadImpl.java
@@ -382,6 +382,14 @@ public WfRunVariableImpl addVariable(String name, Object typeOrDefaultVal) {
         return wfRunVariable;
     }
 
+    public WfRunVariable declareBool(String name) {
+        return addVariable(name, VariableType.BOOL);
+    }
+
+    public WfRunVariable declareBool(String name, boolean defaultVal) {
+        return addVariable(name, defaultVal);
+    }
+
     public WfRunVariable declareInt(String name) {
         return addVariable(name, VariableType.INT);
     }