Add new loading policy fields

indeedeng · Oct 17, 2024 · 37fe022 · 37fe022
1 parent 67c528e
commit 37fe022
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 15 deletions.
diff --git a/iwf/internal_mapper.go b/iwf/internal_mapper.go
@@ -124,13 +124,17 @@ func toIdlStateOptions(skipWaitUntil bool, stateOptions *StateOptions) *iwfidl.W
 	}
 
 	idlStOptions := &iwfidl.WorkflowStateOptions{
-		SearchAttributesLoadingPolicy: stateOptions.SearchAttributesLoadingPolicy,
-		DataAttributesLoadingPolicy:   stateOptions.DataAttributesLoadingPolicy,
-		WaitUntilApiTimeoutSeconds:    stateOptions.WaitUntilApiTimeoutSeconds,
-		ExecuteApiTimeoutSeconds:      stateOptions.ExecuteApiTimeoutSeconds,
-		WaitUntilApiRetryPolicy:       stateOptions.WaitUntilApiRetryPolicy,
-		ExecuteApiRetryPolicy:         stateOptions.ExecuteApiRetryPolicy,
-		WaitUntilApiFailurePolicy:     stateOptions.WaitUntilApiFailurePolicy,
+		WaitUntilApiSearchAttributesLoadingPolicy: stateOptions.WaitUntilApiSearchAttributesLoadingPolicy,
+		ExecuteApiSearchAttributesLoadingPolicy:   stateOptions.ExecuteApiSearchAttributesLoadingPolicy,
+		WaitUntilApiDataAttributesLoadingPolicy:   stateOptions.WaitUntilApiDataAttributesLoadingPolicy,
+		ExecuteApiDataAttributesLoadingPolicy:     stateOptions.ExecuteApiDataAttributesLoadingPolicy,
+		SearchAttributesLoadingPolicy:             stateOptions.SearchAttributesLoadingPolicy,
+		DataAttributesLoadingPolicy:               stateOptions.DataAttributesLoadingPolicy,
+		WaitUntilApiTimeoutSeconds:                stateOptions.WaitUntilApiTimeoutSeconds,
+		ExecuteApiTimeoutSeconds:                  stateOptions.ExecuteApiTimeoutSeconds,
+		WaitUntilApiRetryPolicy:                   stateOptions.WaitUntilApiRetryPolicy,
+		ExecuteApiRetryPolicy:                     stateOptions.ExecuteApiRetryPolicy,
+		WaitUntilApiFailurePolicy:                 stateOptions.WaitUntilApiFailurePolicy,
 	}
 
 	if skipWaitUntil {

diff --git a/iwf/state_options.go b/iwf/state_options.go
@@ -5,12 +5,16 @@ import (
 )
 
 type StateOptions struct {
-	SearchAttributesLoadingPolicy *iwfidl.PersistenceLoadingPolicy
-	DataAttributesLoadingPolicy   *iwfidl.PersistenceLoadingPolicy
-	WaitUntilApiTimeoutSeconds    *int32
-	ExecuteApiTimeoutSeconds      *int32
-	WaitUntilApiRetryPolicy       *iwfidl.RetryPolicy
-	ExecuteApiRetryPolicy         *iwfidl.RetryPolicy
-	WaitUntilApiFailurePolicy     *iwfidl.WaitUntilApiFailurePolicy
-	ExecuteApiFailureProceedState WorkflowState
+	WaitUntilApiSearchAttributesLoadingPolicy *iwfidl.PersistenceLoadingPolicy
+	ExecuteApiSearchAttributesLoadingPolicy   *iwfidl.PersistenceLoadingPolicy
+	SearchAttributesLoadingPolicy             *iwfidl.PersistenceLoadingPolicy
+	WaitUntilApiDataAttributesLoadingPolicy   *iwfidl.PersistenceLoadingPolicy
+	ExecuteApiDataAttributesLoadingPolicy     *iwfidl.PersistenceLoadingPolicy
+	DataAttributesLoadingPolicy               *iwfidl.PersistenceLoadingPolicy
+	WaitUntilApiTimeoutSeconds                *int32
+	ExecuteApiTimeoutSeconds                  *int32
+	WaitUntilApiRetryPolicy                   *iwfidl.RetryPolicy
+	ExecuteApiRetryPolicy                     *iwfidl.RetryPolicy
+	WaitUntilApiFailurePolicy                 *iwfidl.WaitUntilApiFailurePolicy
+	ExecuteApiFailureProceedState             WorkflowState
 }