103 lines
2.3 KiB
Go
103 lines
2.3 KiB
Go
package sealing
|
|
|
|
import (
|
|
"context"
|
|
|
|
"golang.org/x/xerrors"
|
|
|
|
"github.com/filecoin-project/specs-actors/actors/abi"
|
|
"github.com/filecoin-project/specs-actors/actors/abi/big"
|
|
"github.com/filecoin-project/specs-actors/actors/builtin/miner"
|
|
)
|
|
|
|
func (m *Sealing) IsMarkedForUpgrade(id abi.SectorNumber) bool {
|
|
m.upgradeLk.Lock()
|
|
_, found := m.toUpgrade[id]
|
|
m.upgradeLk.Unlock()
|
|
return found
|
|
}
|
|
|
|
func (m *Sealing) MarkForUpgrade(id abi.SectorNumber) error {
|
|
m.upgradeLk.Lock()
|
|
defer m.upgradeLk.Unlock()
|
|
|
|
_, found := m.toUpgrade[id]
|
|
if found {
|
|
return xerrors.Errorf("sector %d already marked for upgrade", id)
|
|
}
|
|
|
|
si, err := m.GetSectorInfo(id)
|
|
if err != nil {
|
|
return xerrors.Errorf("getting sector info: %w", err)
|
|
}
|
|
|
|
if si.State != Proving {
|
|
return xerrors.Errorf("can't mark sectors not in the 'Proving' state for upgrade")
|
|
}
|
|
|
|
if len(si.Pieces) != 1 {
|
|
return xerrors.Errorf("not a committed-capacity sector, expected 1 piece")
|
|
}
|
|
|
|
if si.Pieces[0].DealInfo != nil {
|
|
return xerrors.Errorf("not a committed-capacity sector, has deals")
|
|
}
|
|
|
|
// TODO: more checks to match actor constraints
|
|
|
|
m.toUpgrade[id] = struct{}{}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (m *Sealing) tryUpgradeSector(ctx context.Context, params *miner.SectorPreCommitInfo) big.Int {
|
|
if len(params.DealIDs) == 0 {
|
|
return big.Zero()
|
|
}
|
|
replace := m.maybeUpgradableSector()
|
|
if replace != nil {
|
|
loc, err := m.api.StateSectorPartition(ctx, m.maddr, *replace, nil)
|
|
if err != nil {
|
|
log.Errorf("error calling StateSectorPartition for replaced sector: %+v", err)
|
|
return big.Zero()
|
|
}
|
|
|
|
params.ReplaceCapacity = true
|
|
params.ReplaceSectorNumber = *replace
|
|
params.ReplaceSectorDeadline = loc.Deadline
|
|
params.ReplaceSectorPartition = loc.Partition
|
|
|
|
ri, err := m.api.StateSectorGetInfo(ctx, m.maddr, *replace, nil)
|
|
if err != nil {
|
|
log.Errorf("error calling StateSectorGetInfo for replaced sector: %+v", err)
|
|
return big.Zero()
|
|
}
|
|
|
|
if params.Expiration < ri.Expiration {
|
|
// TODO: Some limit on this
|
|
params.Expiration = ri.Expiration
|
|
}
|
|
|
|
return ri.InitialPledge
|
|
}
|
|
|
|
return big.Zero()
|
|
}
|
|
|
|
func (m *Sealing) maybeUpgradableSector() *abi.SectorNumber {
|
|
m.upgradeLk.Lock()
|
|
defer m.upgradeLk.Unlock()
|
|
for number := range m.toUpgrade {
|
|
// TODO: checks to match actor constraints
|
|
|
|
// this one looks good
|
|
/*if checks */
|
|
{
|
|
delete(m.toUpgrade, number)
|
|
return &number
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|