STSSRE-1430: Make getWorkflow public (#89)

Co-authored-by: Serg Tyatin <[email protected]>

Author: 2rba

iwf/client_impl.go

@@ -17,7 +17,7 @@ type clientImpl struct {
 
 func (c *clientImpl) StartWorkflow(ctx context.Context, workflow ObjectWorkflow, workflowId string, timeoutSecs int32, input interface{}, options *WorkflowOptions) (string, error) {
 	wfType := GetFinalWorkflowType(workflow)
-	wf := c.registry.getWorkflow(wfType)
+	wf := c.registry.GetWorkflow(wfType)
 	if wf == nil {
 		return "", NewInvalidArgumentError("workflow is not registered")
 	}

iwf/registry.go

@@ -10,8 +10,10 @@ type Registry interface {
 	// GetAllRegisteredWorkflowTypes returns all the workflow types that have been registered
 	GetAllRegisteredWorkflowTypes() []string
 
+	// GetWorkflow returns the workflow with the given type
+	GetWorkflow(wfType string) ObjectWorkflow
+
 	// below are all for internal implementation
-	getWorkflow(wfType string) ObjectWorkflow
 	getWorkflowStartingState(wfType string) WorkflowState
 	getWorkflowStateDef(wfType string, id string) StateDef
 	getWorkflowRPC(wfType string, rpcMethod string) CommunicationMethodDef

iwf/registry_impl.go

@@ -46,6 +46,10 @@ func (r *registryImpl) GetAllRegisteredWorkflowTypes() []string {
 	return res
 }
 
+func (r *registryImpl) GetWorkflow(wfType string) ObjectWorkflow {
+	return r.workflowStore[wfType]
+}
+
 func (r *registryImpl) getWorkflowStartingState(wfType string) WorkflowState {
 	return r.workflowStartingState[wfType]
 }
@@ -74,10 +78,6 @@ func (r *registryImpl) getWorkflowRPC(wfType string, rpcMethod string) Communica
 	return r.workflowRPCStore[wfType][rpcMethod]
 }
 
-func (r *registryImpl) getWorkflow(wfType string) ObjectWorkflow {
-	return r.workflowStore[wfType]
-}
-
 func (r *registryImpl) registerWorkflow(wf ObjectWorkflow) error {
 	wfType := GetFinalWorkflowType(wf)
 	_, ok := r.workflowStore[wfType]