feat: load SectorsSummmary from SectorStats instead of calling API (faster)

2023-11-03 15:38:19 +01:00
parent 12b30c0069
commit 02929dd166
3 changed files with 11 additions and 8 deletions
@@ -336,16 +336,14 @@ func (sm *StorageMinerAPI) SectorsListInStates(ctx context.Context, states []api
 	return sns, nil
 }

+// Use SectorsSummary from stats (prometheus) for faster result
 func (sm *StorageMinerAPI) SectorsSummary(ctx context.Context) (map[api.SectorState]int, error) {
-	sectors, err := sm.Miner.ListSectors()
-	if err != nil {
-		return nil, err
-	}
-
+	sectorStats := sm.Miner.SectorsSummary(ctx)
 	out := make(map[api.SectorState]int)
-	for i := range sectors {
-		state := api.SectorState(sectors[i].State)
-		out[state]++
+
+	for st, count := range sectorStats {
+		state := api.SectorState(st)
+		out[state] = int(count)
 	}

 	return out, nil