273ac513fc
Co-authored-by: Raúl Kripalani <raul@protocol.ai>
121 lines
3.5 KiB
Go
121 lines
3.5 KiB
Go
package modules
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
"github.com/multiformats/go-varint"
|
|
"go.uber.org/fx"
|
|
|
|
"github.com/filecoin-project/go-address"
|
|
"github.com/filecoin-project/go-state-types/abi"
|
|
builtintypes "github.com/filecoin-project/go-state-types/builtin"
|
|
|
|
"github.com/filecoin-project/lotus/chain/events"
|
|
"github.com/filecoin-project/lotus/chain/events/filter"
|
|
"github.com/filecoin-project/lotus/chain/messagepool"
|
|
"github.com/filecoin-project/lotus/chain/stmgr"
|
|
"github.com/filecoin-project/lotus/chain/store"
|
|
"github.com/filecoin-project/lotus/chain/types"
|
|
"github.com/filecoin-project/lotus/node/config"
|
|
"github.com/filecoin-project/lotus/node/impl/full"
|
|
"github.com/filecoin-project/lotus/node/modules/helpers"
|
|
)
|
|
|
|
type EventAPI struct {
|
|
fx.In
|
|
|
|
full.ChainAPI
|
|
full.StateAPI
|
|
}
|
|
|
|
var _ events.EventAPI = &EventAPI{}
|
|
|
|
func EthEventAPI(cfg config.ActorEventConfig) func(helpers.MetricsCtx, fx.Lifecycle, *store.ChainStore, *stmgr.StateManager, EventAPI, *messagepool.MessagePool) (*full.EthEvent, error) {
|
|
return func(mctx helpers.MetricsCtx, lc fx.Lifecycle, cs *store.ChainStore, sm *stmgr.StateManager, evapi EventAPI, mp *messagepool.MessagePool) (*full.EthEvent, error) {
|
|
ee := &full.EthEvent{
|
|
Chain: cs,
|
|
MaxFilterHeightRange: abi.ChainEpoch(cfg.MaxFilterHeightRange),
|
|
}
|
|
|
|
if !cfg.EnableRealTimeFilterAPI && !cfg.EnableHistoricFilterAPI {
|
|
// all event functionality is disabled
|
|
return ee, nil
|
|
}
|
|
|
|
ee.FilterStore = filter.NewMemFilterStore(cfg.MaxFilters)
|
|
|
|
// Start garbage collection for filters
|
|
lc.Append(fx.Hook{
|
|
OnStart: func(ctx context.Context) error {
|
|
go ee.GC(ctx, time.Duration(cfg.FilterTTL))
|
|
return nil
|
|
},
|
|
})
|
|
|
|
if cfg.EnableRealTimeFilterAPI {
|
|
ee.EventFilterManager = &filter.EventFilterManager{
|
|
ChainStore: cs,
|
|
AddressResolver: func(ctx context.Context, emitter abi.ActorID, ts *types.TipSet) (address.Address, bool) {
|
|
// we only want to match using f4 addresses
|
|
idAddr, err := address.NewIDAddress(uint64(emitter))
|
|
if err != nil {
|
|
return address.Undef, false
|
|
}
|
|
addr, err := sm.LookupRobustAddress(ctx, idAddr, ts)
|
|
if err != nil {
|
|
return address.Undef, false
|
|
}
|
|
// if robust address is not f4 then we won't match against it so bail early
|
|
if addr.Protocol() != address.Delegated {
|
|
return address.Undef, false
|
|
}
|
|
// we have an f4 address, make sure it's assigned by the EAM
|
|
if namespace, _, err := varint.FromUvarint(addr.Payload()); err != nil || namespace != builtintypes.EthereumAddressManagerActorID {
|
|
return address.Undef, false
|
|
}
|
|
return addr, true
|
|
},
|
|
|
|
MaxFilterResults: cfg.MaxFilterResults,
|
|
}
|
|
ee.TipSetFilterManager = &filter.TipSetFilterManager{
|
|
MaxFilterResults: cfg.MaxFilterResults,
|
|
}
|
|
ee.MemPoolFilterManager = &filter.MemPoolFilterManager{
|
|
MaxFilterResults: cfg.MaxFilterResults,
|
|
}
|
|
|
|
const ChainHeadConfidence = 1
|
|
|
|
ctx := helpers.LifecycleCtx(mctx, lc)
|
|
lc.Append(fx.Hook{
|
|
OnStart: func(context.Context) error {
|
|
ev, err := events.NewEventsWithConfidence(ctx, &evapi, ChainHeadConfidence)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
// ignore returned tipsets
|
|
_ = ev.Observe(ee.EventFilterManager)
|
|
_ = ev.Observe(ee.TipSetFilterManager)
|
|
|
|
ch, err := mp.Updates(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
go ee.MemPoolFilterManager.WaitForMpoolUpdates(ctx, ch)
|
|
|
|
return nil
|
|
},
|
|
})
|
|
|
|
}
|
|
|
|
if cfg.EnableHistoricFilterAPI {
|
|
// TODO: enable indexer
|
|
}
|
|
|
|
return ee, nil
|
|
}
|
|
}
|