Change name TaskRegistry Registry

Signed-off-by: joshvanl <[email protected]>

Author: JoshVanL

workflow/client.go

@@ -26,7 +26,7 @@ func NewClientWithLogger(cc grpc.ClientConnInterface, logger backend.Logger) *Cl
 }
 
 // StartWorker starts the workflow runtime to process workflows.
-func (c *Client) StartWorker(ctx context.Context, r *TaskRegistry) error {
+func (c *Client) StartWorker(ctx context.Context, r *Registry) error {
 	return c.thgc.StartWorkItemListener(ctx, r.registry)
 }
 

workflow/registry.go

@@ -4,46 +4,46 @@ import (
 	"github.com/dapr/durabletask-go/task"
 )
 
-// TaskRegistry contains maps of names to corresponding orchestrator and
-// activity functions.
-type TaskRegistry struct {
+// Registry contains maps of names to corresponding orchestrator and activity
+// functions.
+type Registry struct {
 	registry *task.TaskRegistry
 }
 
-// NewTaskRegistry returns a new [TaskRegistry] struct.
-func NewTaskRegistry() *TaskRegistry {
-	return &TaskRegistry{
+// NewRegistry returns a new Registry struct.
+func NewRegistry() *Registry {
+	return &Registry{
 		registry: task.NewTaskRegistry(),
 	}
 }
 
 // AddWorkflow adds an orchestrator function to the registry. The name of the orchestrator
 // function is determined using reflection.
-func (r *TaskRegistry) AddWorkflow(w Workflow) error {
+func (r *Registry) AddWorkflow(w Workflow) error {
 	return r.registry.AddOrchestrator(func(ctx *task.OrchestrationContext) (any, error) {
 		return w(&WorkflowContext{ctx})
 	})
 }
 
 // AddWorkflowN adds an orchestrator function to the registry with a
 // specified name.
-func (r *TaskRegistry) AddWorkflowN(name string, w Workflow) error {
+func (r *Registry) AddWorkflowN(name string, w Workflow) error {
 	return r.registry.AddOrchestratorN(name, func(ctx *task.OrchestrationContext) (any, error) {
 		return w(&WorkflowContext{ctx})
 	})
 }
 
 // AddActivity adds an activity function to the registry. The name of the
 // activity function is determined using reflection.
-func (r *TaskRegistry) AddActivity(a Activity) error {
+func (r *Registry) AddActivity(a Activity) error {
 	return r.registry.AddActivity(func(ctx task.ActivityContext) (any, error) {
 		return a(ctx.(ActivityContext))
 	})
 }
 
 // AddActivityN adds an activity function to the registry with a specified
 // name.
-func (r *TaskRegistry) AddActivityN(name string, a Activity) error {
+func (r *Registry) AddActivityN(name string, a Activity) error {
 	return r.registry.AddActivityN(name, func(ctx task.ActivityContext) (any, error) {
 		return a(ctx.(ActivityContext))
 	})