89 lines
1.5 KiB
Go
89 lines
1.5 KiB
Go
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,
|
|
}
|
|
}
|