lotus/node/impl/full/mpool.go

117 lines
2.7 KiB
Go
Raw Normal View History

2019-08-20 16:48:33 +00:00
package full
import (
"context"
2019-10-02 18:00:08 +00:00
2019-12-03 19:33:29 +00:00
"github.com/ipfs/go-cid"
2019-08-20 16:48:33 +00:00
"go.uber.org/fx"
2019-09-16 14:17:08 +00:00
"golang.org/x/xerrors"
2019-08-20 16:48:33 +00:00
"github.com/filecoin-project/go-address"
2019-11-17 07:44:06 +00:00
"github.com/filecoin-project/lotus/api"
2019-12-01 23:11:43 +00:00
"github.com/filecoin-project/lotus/chain/messagepool"
2019-12-03 19:33:29 +00:00
"github.com/filecoin-project/lotus/chain/store"
"github.com/filecoin-project/lotus/chain/types"
2019-08-20 16:48:33 +00:00
)
type MpoolAPI struct {
fx.In
2019-09-16 14:17:08 +00:00
WalletAPI
2019-12-03 19:33:29 +00:00
Chain *store.ChainStore
2019-12-01 23:11:43 +00:00
Mpool *messagepool.MessagePool
2019-08-20 16:48:33 +00:00
}
func (a *MpoolAPI) MpoolPending(ctx context.Context, tsk types.TipSetKey) ([]*types.SignedMessage, error) {
ts, err := a.Chain.GetTipSetFromKey(tsk)
if err != nil {
return nil, xerrors.Errorf("loading tipset %s: %w", tsk, err)
}
2019-12-03 19:33:29 +00:00
pending, mpts := a.Mpool.Pending()
haveCids := map[cid.Cid]struct{}{}
for _, m := range pending {
haveCids[m.Cid()] = struct{}{}
}
2019-12-03 19:34:31 +00:00
if ts == nil || mpts.Height() > ts.Height() {
2019-12-03 19:33:29 +00:00
return pending, nil
}
for {
if mpts.Height() == ts.Height() {
if mpts.Equals(ts) {
return pending, nil
}
// different blocks in tipsets
have, err := a.Mpool.MessagesForBlocks(ts.Blocks())
if err != nil {
return nil, xerrors.Errorf("getting messages for base ts: %w", err)
}
for _, m := range have {
haveCids[m.Cid()] = struct{}{}
}
}
msgs, err := a.Mpool.MessagesForBlocks(ts.Blocks())
if err != nil {
return nil, xerrors.Errorf(": %w", err)
}
for _, m := range msgs {
if _, ok := haveCids[m.Cid()]; ok {
continue
}
haveCids[m.Cid()] = struct{}{}
pending = append(pending, m)
}
if mpts.Height() >= ts.Height() {
return pending, nil
}
ts, err = a.Chain.LoadTipSet(ts.Parents())
if err != nil {
return nil, xerrors.Errorf("loading parent tipset: %w", err)
}
}
2019-08-20 16:48:33 +00:00
}
2020-01-07 16:44:55 +00:00
func (a *MpoolAPI) MpoolPush(ctx context.Context, smsg *types.SignedMessage) (cid.Cid, error) {
2019-09-16 14:17:08 +00:00
return a.Mpool.Push(smsg)
}
2019-09-17 08:15:26 +00:00
func (a *MpoolAPI) MpoolPushMessage(ctx context.Context, msg *types.Message) (*types.SignedMessage, error) {
2019-09-16 14:17:08 +00:00
if msg.Nonce != 0 {
2019-09-17 08:15:26 +00:00
return nil, xerrors.Errorf("MpoolPushMessage expects message nonce to be 0, was %d", msg.Nonce)
2019-08-20 16:48:33 +00:00
}
2019-09-16 14:17:08 +00:00
return a.Mpool.PushWithNonce(msg.From, func(nonce uint64) (*types.SignedMessage, error) {
msg.Nonce = nonce
b, err := a.WalletBalance(ctx, msg.From)
if err != nil {
return nil, xerrors.Errorf("mpool push: getting origin balance: %w", err)
}
if b.LessThan(msg.Value) {
return nil, xerrors.Errorf("mpool push: not enough funds: %s < %s", b, msg.Value)
}
2019-09-16 14:17:08 +00:00
return a.WalletSignMessage(ctx, msg.From, msg)
})
2019-08-20 16:48:33 +00:00
}
func (a *MpoolAPI) MpoolGetNonce(ctx context.Context, addr address.Address) (uint64, error) {
return a.Mpool.GetNonce(addr)
}
2019-11-17 07:44:06 +00:00
func (a *MpoolAPI) MpoolSub(ctx context.Context) (<-chan api.MpoolUpdate, error) {
return a.Mpool.Updates(ctx)
}