package chain import ( "sync" "time" "github.com/filecoin-project/go-state-types/abi" "github.com/filecoin-project/lotus/api" "github.com/filecoin-project/lotus/build" "github.com/filecoin-project/lotus/chain/types" ) type SyncerState struct { lk sync.Mutex Target *types.TipSet Base *types.TipSet Stage api.SyncStateStage Height abi.ChainEpoch Message string Start time.Time End time.Time } func (ss *SyncerState) SetStage(v api.SyncStateStage) { if ss == nil { return } ss.lk.Lock() defer ss.lk.Unlock() ss.Stage = v if v == api.StageSyncComplete { ss.End = build.Clock.Now() } } func (ss *SyncerState) Init(base, target *types.TipSet) { if ss == nil { return } ss.lk.Lock() defer ss.lk.Unlock() ss.Target = target ss.Base = base ss.Stage = api.StageHeaders ss.Height = 0 ss.Message = "" ss.Start = build.Clock.Now() ss.End = time.Time{} } func (ss *SyncerState) SetHeight(h abi.ChainEpoch) { if ss == nil { return } ss.lk.Lock() defer ss.lk.Unlock() ss.Height = h } func (ss *SyncerState) Error(err error) { if ss == nil { return } ss.lk.Lock() defer ss.lk.Unlock() ss.Message = err.Error() ss.Stage = api.StageSyncErrored ss.End = build.Clock.Now() } func (ss *SyncerState) Snapshot() SyncerState { ss.lk.Lock() defer ss.lk.Unlock() return SyncerState{ Base: ss.Base, Target: ss.Target, Stage: ss.Stage, Height: ss.Height, Message: ss.Message, Start: ss.Start, End: ss.End, } }