winpost: Run on single node context

deps/apiinfo.go

@@ -81,9 +81,13 @@ func GetFullNodeAPIV1Curio(ctx *cli.Context, ainfoCfg []string) (api.Chain, json
 
 type contextKey string
 
+var retryNodeKey = contextKey("retry-node")
+
+// OnSingleNode returns a new context that will try to perform all calls on the same node.
+// If the backing node fails, the calls will be retried on a different node, and further calls will be made on that node.
 // Not thread safe
 func OnSingleNode(ctx context.Context) context.Context {
-	return context.WithValue(ctx, contextKey("retry-node"), new(*int))
+	return context.WithValue(ctx, retryNodeKey, new(*int))
 }
 
 type httpHead struct {
@@ -229,12 +233,11 @@ func FullNodeProxy[T api.Chain](ins []T, outstr *api.ChainStruct) {
 
 				// for calls that need to be performed on the same node
 				// primarily for miner when calling create block and submit block subsequently
-				key := contextKey("retry-node")
-				if ctx.Value(key) != nil {
-					if (*ctx.Value(key).(**int)) == nil {
-						*ctx.Value(key).(**int) = preferredProvider
+				if ctx.Value(retryNodeKey) != nil {
+					if (*ctx.Value(retryNodeKey).(**int)) == nil {
+						*ctx.Value(retryNodeKey).(**int) = preferredProvider
 					} else {
-						preferredProvider = *ctx.Value(key).(**int)
+						preferredProvider = *ctx.Value(retryNodeKey).(**int)
 					}
 				}
 

tasks/winning/winning_task.go

@@ -22,6 +22,7 @@ import (
 	prooftypes "github.com/filecoin-project/go-state-types/proof"
 
 	"github.com/filecoin-project/curio/build"
+	"github.com/filecoin-project/curio/deps"
 	"github.com/filecoin-project/curio/harmony/harmonydb"
 	"github.com/filecoin-project/curio/harmony/harmonytask"
 	"github.com/filecoin-project/curio/harmony/resources"
@@ -95,7 +96,7 @@ func NewWinPostTask(max int, db *harmonydb.DB, remote *paths.Remote, verifier st
 func (t *WinPostTask) Do(taskID harmonytask.TaskID, stillOwned func() bool) (done bool, err error) {
 	log.Debugw("WinPostTask.Do()", "taskID", taskID)
 
-	ctx := context.TODO()
+	ctx := deps.OnSingleNode(context.Background())
 
 	type BlockCID struct {
 		CID string
@@ -397,7 +398,7 @@ func (t *WinPostTask) Do(taskID harmonytask.TaskID, stillOwned func() bool) (don
 	{
 		uts := base.TipSet.MinTimestamp() + build.BlockDelaySecs*(uint64(base.AddRounds)+1)
 
-		blockMsg, err = t.api.MinerCreateBlock(context.TODO(), &api.BlockTemplate{
+		blockMsg, err = t.api.MinerCreateBlock(ctx, &api.BlockTemplate{
 			Miner:            maddr,
 			Parents:          base.TipSet.Key(),
 			Ticket:           ticket,