122 lines
2.5 KiB
Go
122 lines
2.5 KiB
Go
package storage
|
|
|
|
import (
|
|
"context"
|
|
"github.com/filecoin-project/lotus/api"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
cid "github.com/ipfs/go-cid"
|
|
)
|
|
|
|
type SealTicket struct {
|
|
BlockHeight uint64
|
|
TicketBytes []byte
|
|
}
|
|
|
|
type SectorInfo struct {
|
|
State api.SectorState
|
|
SectorID uint64
|
|
|
|
// PreCommit
|
|
CommD []byte
|
|
CommR []byte
|
|
Ticket SealTicket
|
|
|
|
PreCommitMessage *cid.Cid
|
|
|
|
// PreCommitted
|
|
RandHeight uint64
|
|
RandTs *types.TipSet
|
|
|
|
// Committing
|
|
CommitMessage *cid.Cid
|
|
}
|
|
|
|
type sectorUpdate struct {
|
|
newState api.SectorState
|
|
id uint64
|
|
err error
|
|
mut func(*SectorInfo)
|
|
}
|
|
|
|
func (m *Miner) sectorStateLoop(ctx context.Context) {
|
|
// TODO: restore state
|
|
|
|
go func() {
|
|
defer log.Warn("quitting deal provider loop")
|
|
defer close(m.stopped)
|
|
|
|
for {
|
|
select {
|
|
case sector := <-m.sectorIncoming:
|
|
m.onSectorIncoming(sector)
|
|
case update := <-m.sectorUpdated:
|
|
m.onSectorUpdated(ctx, update)
|
|
case <-m.stop:
|
|
return
|
|
}
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (m *Miner) onSectorIncoming(sector *SectorInfo) {
|
|
if err := m.sectors.Begin(sector.SectorID, sector); err != nil {
|
|
// We may have re-sent the proposal
|
|
log.Errorf("deal tracking failed: %s", err)
|
|
m.failSector(sector.SectorID, err)
|
|
return
|
|
}
|
|
|
|
go func() {
|
|
m.sectorUpdated <- sectorUpdate{
|
|
newState: api.Unsealed,
|
|
id: sector.SectorID,
|
|
}
|
|
}()
|
|
}
|
|
|
|
func (m *Miner) onSectorUpdated(ctx context.Context, update sectorUpdate) {
|
|
log.Infof("Sector %s updated state to %s", update.id, api.SectorStateStr(update.newState))
|
|
var sector SectorInfo
|
|
err := m.sectors.Mutate(update.id, func(s *SectorInfo) error {
|
|
s.State = update.newState
|
|
sector = *s
|
|
return nil
|
|
})
|
|
if update.err != nil {
|
|
log.Errorf("deal %s failed: %s", update.id, update.err)
|
|
m.failSector(update.id, update.err)
|
|
return
|
|
}
|
|
if err != nil {
|
|
m.failSector(update.id, err)
|
|
return
|
|
}
|
|
|
|
switch update.newState {
|
|
case api.Unsealed:
|
|
m.handle(ctx, sector, m.sealPreCommit, api.PreCommitting)
|
|
case api.PreCommitting:
|
|
m.handle(ctx, sector, m.preCommit, api.PreCommitted)
|
|
case api.PreCommitted:
|
|
m.handle(ctx, sector, m.preCommitted, api.SectorNoUpdate)
|
|
case api.Committing:
|
|
m.handle(ctx, sector, m.committing, api.Proving)
|
|
}
|
|
}
|
|
|
|
func (m *Miner) failSector(id uint64, err error) {
|
|
panic(err) // todo: better error handling strategy
|
|
}
|
|
|
|
func (m *Miner) SealSector(ctx context.Context, sid uint64) error {
|
|
select {
|
|
case m.sectorIncoming <- &SectorInfo{
|
|
State: api.UndefinedSectorState,
|
|
SectorID: sid,
|
|
}:
|
|
return nil
|
|
case <-ctx.Done():
|
|
return ctx.Err()
|
|
}
|
|
}
|