103 lines
2.5 KiB
Go
103 lines
2.5 KiB
Go
package sealing
|
|
|
|
import (
|
|
"context"
|
|
"sync"
|
|
|
|
"go.opencensus.io/stats"
|
|
"go.opencensus.io/tag"
|
|
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
|
|
"github.com/filecoin-project/lotus/metrics"
|
|
"github.com/filecoin-project/lotus/storage/pipeline/sealiface"
|
|
)
|
|
|
|
type statSectorState int
|
|
|
|
const (
|
|
sstStaging statSectorState = iota
|
|
sstSealing
|
|
sstFailed
|
|
sstProving
|
|
nsst
|
|
)
|
|
|
|
type SectorStats struct {
|
|
lk sync.Mutex
|
|
|
|
bySector map[abi.SectorID]SectorState
|
|
byState map[SectorState]int64
|
|
totals [nsst]uint64
|
|
}
|
|
|
|
func (ss *SectorStats) updateSector(ctx context.Context, cfg sealiface.Config, id abi.SectorID, st SectorState) (updateInput bool) {
|
|
ss.lk.Lock()
|
|
defer ss.lk.Unlock()
|
|
|
|
preSealing := ss.curSealingLocked()
|
|
preStaging := ss.curStagingLocked()
|
|
|
|
// update totals
|
|
oldst, found := ss.bySector[id]
|
|
if found {
|
|
ss.totals[toStatState(oldst, cfg.FinalizeEarly)]--
|
|
ss.byState[oldst]--
|
|
|
|
mctx, _ := tag.New(ctx, tag.Upsert(metrics.SectorState, string(oldst)))
|
|
stats.Record(mctx, metrics.SectorStates.M(ss.byState[oldst]))
|
|
}
|
|
|
|
sst := toStatState(st, cfg.FinalizeEarly)
|
|
ss.bySector[id] = st
|
|
ss.totals[sst]++
|
|
ss.byState[st]++
|
|
|
|
mctx, _ := tag.New(ctx, tag.Upsert(metrics.SectorState, string(st)))
|
|
stats.Record(mctx, metrics.SectorStates.M(ss.byState[st]))
|
|
|
|
// check if we may need be able to process more deals
|
|
sealing := ss.curSealingLocked()
|
|
staging := ss.curStagingLocked()
|
|
|
|
log.Debugw("sector stats", "sealing", sealing, "staging", staging)
|
|
|
|
if cfg.MaxSealingSectorsForDeals > 0 && // max sealing deal sector limit set
|
|
preSealing >= cfg.MaxSealingSectorsForDeals && // we were over limit
|
|
sealing < cfg.MaxSealingSectorsForDeals { // and we're below the limit now
|
|
updateInput = true
|
|
}
|
|
|
|
if cfg.MaxWaitDealsSectors > 0 && // max waiting deal sector limit set
|
|
preStaging >= cfg.MaxWaitDealsSectors && // we were over limit
|
|
staging < cfg.MaxWaitDealsSectors { // and we're below the limit now
|
|
updateInput = true
|
|
}
|
|
|
|
return updateInput
|
|
}
|
|
|
|
func (ss *SectorStats) curSealingLocked() uint64 {
|
|
return ss.totals[sstStaging] + ss.totals[sstSealing] + ss.totals[sstFailed]
|
|
}
|
|
|
|
func (ss *SectorStats) curStagingLocked() uint64 {
|
|
return ss.totals[sstStaging]
|
|
}
|
|
|
|
// return the number of sectors currently in the sealing pipeline
|
|
func (ss *SectorStats) curSealing() uint64 {
|
|
ss.lk.Lock()
|
|
defer ss.lk.Unlock()
|
|
|
|
return ss.curSealingLocked()
|
|
}
|
|
|
|
// return the number of sectors waiting to enter the sealing pipeline
|
|
func (ss *SectorStats) curStaging() uint64 {
|
|
ss.lk.Lock()
|
|
defer ss.lk.Unlock()
|
|
|
|
return ss.curStagingLocked()
|
|
}
|