worker: avoid fetching contract set contracts separately

api/contract.go

@@ -220,3 +220,13 @@ func (c Contract) RemainingCollateral() types.Currency {
 	}
 	return c.Revision.MissedHostPayout().Sub(c.ContractPrice)
 }
+
+// InSet returns whether the contract is in the given set.
+func (cm ContractMetadata) InSet(set string) bool {
+	for _, s := range cm.ContractSets {
+		if s == set {
+			return true
+		}
+	}
+	return false
+}

autopilot/accounts.go

@@ -135,17 +135,14 @@ func (a *accounts) refillWorkerAccounts(ctx context.Context, w Worker) {
 		return
 	}
 
-	// fetch all contract set contracts
-	contractSetContracts, err := a.c.Contracts(ctx, api.ContractsOpts{ContractSet: cfg.Config.Contracts.Set})
-	if err != nil {
-		a.l.Errorw(fmt.Sprintf("failed to fetch contract set contracts: %v", err))
-		return
-	}
-
-	// build a map of contract set contracts
+	// filter all contract set contracts
+	var contractSetContracts []api.ContractMetadata
 	inContractSet := make(map[types.FileContractID]struct{})
-	for _, contract := range contractSetContracts {
-		inContractSet[contract.ID] = struct{}{}
+	for _, c := range contracts {
+		if c.InSet(cfg.Config.Contracts.Set) {
+			contractSetContracts = append(contractSetContracts, c)
+			inContractSet[c.ID] = struct{}{}
+		}
 	}
 
 	// refill accounts in separate goroutines

worker/worker.go

@@ -786,10 +786,12 @@ func (w *worker) slabMigrateHandler(jc jape.Context) {
 		return
 	}
 
-	// fetch upload contracts
-	ulContracts, err := w.bus.Contracts(ctx, api.ContractsOpts{ContractSet: up.ContractSet})
-	if jc.Check("couldn't fetch contracts from bus", err) != nil {
-		return
+	// filter upload contracts
+	var ulContracts []api.ContractMetadata
+	for _, c := range dlContracts {
+		if c.InSet(up.ContractSet) {
+			ulContracts = append(ulContracts, c)
+		}
 	}
 
 	// migrate the slab