feat: configurable retry after duration

cmd/opinitd/start.go

@@ -18,6 +18,7 @@ import (
 
 const (
 	flagPollingInterval = "polling-interval"
+	flagRetryAfter      = "retry-after"
 )
 
 func startCmd(ctx *cmdContext) *cobra.Command {
@@ -56,6 +57,11 @@ Currently supported bots:
 				return err
 			}
 			ctx = types.WithPollingInterval(ctx, interval)
+			retryAfter, err := cmd.Flags().GetDuration(flagRetryAfter)
+			if err != nil {
+				return err
+			}
+			ctx = types.WithRetryAfter(ctx, retryAfter)
 			err = bot.Initialize(ctx)
 			if err != nil {
 				return err
@@ -65,7 +71,8 @@ Currently supported bots:
 	}
 
 	cmd = configFlag(ctx.v, cmd)
-	cmd.Flags().Duration(flagPollingInterval, time.Second, "Polling interval in milliseconds")
+	cmd.Flags().Duration(flagPollingInterval, 100*time.Millisecond, "Polling interval")
+	cmd.Flags().Duration(flagRetryAfter, time.Minute, "Waiting time for a retry")
 	return cmd
 }
 

node/process.go

@@ -60,7 +60,7 @@ func (n *Node) blockProcessLooper(ctx context.Context, processType nodetypes.Blo
 				if err != nil {
 					n.logger.Error("failed to handle new block", zap.String("error", err.Error()))
 					if errors.Is(err, nodetypes.ErrIgnoreAndTryLater) {
-						sleep := time.NewTimer(time.Minute)
+						sleep := time.NewTimer(types.RetryAfter(ctx))
 						select {
 						case <-ctx.Done():
 							return nil

types/context.go

@@ -12,6 +12,7 @@ type contextKey string
 var (
 	ContextKeyErrGrp          = contextKey("ErrGrp")
 	ContextKeyPollingInterval = contextKey("PollingInterval")
+	ContextKeyRetryAfter      = contextKey("RetryAfter")
 	ContextKeyTxTimeout       = contextKey("TxTimeout")
 )
 
@@ -32,5 +33,17 @@ func PollingInterval(ctx context.Context) time.Duration {
 	if interval == nil {
 		return 100 * time.Millisecond
 	}
-	return ctx.Value(ContextKeyPollingInterval).(time.Duration)
+	return interval.(time.Duration)
+}
+
+func WithRetryAfter(ctx context.Context, duration time.Duration) context.Context {
+	return context.WithValue(ctx, ContextKeyRetryAfter, duration)
+}
+
+func RetryAfter(ctx context.Context) time.Duration {
+	duration := ctx.Value(ContextKeyRetryAfter)
+	if duration == nil {
+		return time.Minute
+	}
+	return duration.(time.Duration)
 }