dont recover upload if gin context is cancelled

utils/recovery/recovery.go

@@ -11,6 +11,20 @@ import (
 	"github.com/gotd/td/tgerr"
 )
 
+func hasError(err error, target string) bool {
+	for err != nil {
+		if err.Error() == target {
+			return true
+		}
+		if unwrapper, ok := err.(interface{ Unwrap() error }); ok {
+			err = unwrapper.Unwrap()
+		} else {
+			break
+		}
+	}
+	return false
+}
+
 type recovery struct {
 	ctx     context.Context
 	backoff backoff.BackOff
@@ -47,7 +61,10 @@ func (r *recovery) shouldRecover(err error) bool {
 	default:
 	}
 
+	//recover only if context is not cancelled and error is not a rpc error
+	isContextErr := hasError(err, "context canceled")
+
 	_, ok := tgerr.As(err)
 
-	return !ok
+	return !isContextErr && !ok
 }

utils/retry/retry.go

@@ -11,6 +11,11 @@ import (
 	"github.com/gotd/td/tgerr"
 )
 
+var internalErrors = []string{
+	"Timedout",
+	"No workers running",
+}
+
 type retry struct {
 	max    int
 	errors []string
@@ -38,8 +43,7 @@ func (r retry) Handle(next tg.Invoker) telegram.InvokeFunc {
 
 func New(max int, errors ...string) telegram.Middleware {
 	return retry{
-		max: max,
-		errors: append(errors,
-			"Timedout"),
+		max:    max,
+		errors: append(errors, internalErrors...),
 	}
 }

utils/tgc/tgc.go

@@ -30,10 +30,9 @@ func deviceConfig(appConfig *utils.Config) telegram.DeviceConfig {
 	return config
 }
 func NewDefaultMiddlewares(ctx context.Context) ([]telegram.Middleware, error) {
-	_clock := tdclock.System
 
 	return []telegram.Middleware{
-		recovery.New(ctx, Backoff(_clock)),
+		recovery.New(ctx, Backoff(tdclock.System)),
 		retry.New(5),
 		floodwait.NewSimpleWaiter(),
 	}, nil