discovery: cache orchestrator stake for current round when DBOrchestr…

…atorPoolcache is created

discovery/db_discovery.go

@@ -58,6 +58,10 @@ func NewDBOrchestratorPoolCache(ctx context.Context, node *core.LivepeerNode, rm
 		return nil, err
 	}
 
+	if err := dbo.cacheOrchestratorStake(); err != nil {
+		return nil, err
+	}
+
 	if err := dbo.pollOrchestratorInfo(ctx); err != nil {
 		return nil, err
 	}
@@ -145,6 +149,53 @@ func (dbo *DBOrchestratorPoolCache) cacheTranscoderPool() error {
 	return nil
 }
 
+func (dbo *DBOrchestratorPoolCache) cacheOrchestratorStake() error {
+	orchs, err := dbo.store.SelectOrchs(
+		&common.DBOrchFilter{
+			CurrentRound: dbo.rm.LastInitializedRound(),
+		},
+	)
+	if err != nil {
+		return fmt.Errorf("could not retrieve orchestrators from DB: %v", err)
+	}
+
+	resc, errc := make(chan *common.DBOrch), make(chan error)
+	ctx, cancel := context.WithTimeout(context.Background(), getOrchestratorsTimeoutLoop)
+	defer cancel()
+
+	currentRound := dbo.rm.LastInitializedRound()
+
+	getStake := func(o *common.DBOrch) {
+		ep, err := dbo.lpEth.GetTranscoderEarningsPoolForRound(ethcommon.HexToAddress(o.EthereumAddr), currentRound)
+		if err != nil {
+			errc <- err
+			return
+		}
+		o.Stake = ep.TotalStake.String()
+		resc <- o
+	}
+
+	for _, o := range orchs {
+		go getStake(o)
+	}
+
+	for i := 0; i < len(orchs); i++ {
+		select {
+		case res := <-resc:
+			if err := dbo.store.UpdateOrch(res); err != nil {
+				glog.Error("Error updating Orchestrator in DB: ", err)
+			}
+		case err := <-errc:
+			glog.Errorln(err)
+		case <-ctx.Done():
+			glog.Info("Done fetching stake for orchestrators, context timeout")
+			break
+		}
+	}
+
+	return nil
+}
+
 func (dbo *DBOrchestratorPoolCache) pollOrchestratorInfo(ctx context.Context) error {
 	if err := dbo.cacheDBOrchs(); err != nil {
 		return err

discovery/discovery_test.go

@@ -211,6 +211,7 @@ func TestNewDBOrchestratorPoolCache_GivenListOfOrchs_CreatesPoolCacheCorrectly(t
 		Database: dbh,
 		Eth: &eth.StubClient{
 			Orchestrators: orchestrators,
+			TotalStake:    big.NewInt(5000),
 		},
 		Sender: sender,
 	}
@@ -235,6 +236,7 @@ func TestNewDBOrchestratorPoolCache_GivenListOfOrchs_CreatesPoolCacheCorrectly(t
 	for _, o := range dbOrchs {
 		test := toOrchTest(o.EthereumAddr, o.ServiceURI, o.PricePerPixel)
 		assert.Contains(testOrchs, test)
+		assert.Equal(o.Stake, big.NewInt(5000).String())
 	}
 
 	urls := pool.GetURLs()