@@ -2,8 +2,6 @@ package local
22
33import (
44 "context"
5- "fmt"
6- "strconv"
75 "sync"
86
97 "github.com/dapr/durabletask-go/api"
@@ -37,14 +35,15 @@ func (be *TasksBackend) CompleteActivityTask(ctx context.Context, response *prot
3735 if be .deletePendingActivityTask (response .GetInstanceId (), response .GetTaskId (), response ) {
3836 return nil
3937 }
40- return fmt .Errorf ("unknown instance ID/task ID combo: %s" , response .GetInstanceId ()+ "/" + strconv .FormatInt (int64 (response .GetTaskId ()), 10 ))
38+
39+ return api .NewUnknownTaskIDError (response .GetInstanceId (), response .GetTaskId ())
4140}
4241
4342func (be * TasksBackend ) CancelActivityTask (ctx context.Context , instanceID api.InstanceID , taskID int32 ) error {
4443 if be .deletePendingActivityTask (string (instanceID ), taskID , nil ) {
4544 return nil
4645 }
47- return fmt . Errorf ( "unknown instance ID/task ID combo: %s" , string ( instanceID ) + "/" + strconv . FormatInt ( int64 ( taskID ), 10 ) )
46+ return api . NewUnknownTaskIDError ( instanceID . String ( ), taskID )
4847}
4948
5049func (be * TasksBackend ) WaitForActivityCompletion (ctx context.Context , request * protos.ActivityRequest ) (* protos.ActivityResponse , error ) {
@@ -70,14 +69,14 @@ func (be *TasksBackend) CompleteOrchestratorTask(ctx context.Context, response *
7069 if be .deletePendingOrchestrator (response .GetInstanceId (), response ) {
7170 return nil
7271 }
73- return fmt . Errorf ( "unknown instance ID: %s" , response .GetInstanceId ())
72+ return api . NewUnknownInstanceIDError ( response .GetInstanceId ())
7473}
7574
7675func (be * TasksBackend ) CancelOrchestratorTask (ctx context.Context , instanceID api.InstanceID ) error {
7776 if be .deletePendingOrchestrator (string (instanceID ), nil ) {
7877 return nil
7978 }
80- return fmt . Errorf ( "unknown instance ID: %s" , instanceID )
79+ return api . NewUnknownInstanceIDError ( instanceID . String () )
8180}
8281
8382func (be * TasksBackend ) WaitForOrchestratorCompletion (ctx context.Context , request * protos.OrchestratorRequest ) (* protos.OrchestratorResponse , error ) {
0 commit comments