lotus/node/modules/services.go

118 lines
3.2 KiB
Go
Raw Normal View History

2019-07-08 14:07:09 +00:00
package modules
import (
2019-08-01 17:12:41 +00:00
"context"
2019-07-08 14:07:09 +00:00
"github.com/libp2p/go-libp2p-core/host"
inet "github.com/libp2p/go-libp2p-core/network"
2020-02-17 05:51:18 +00:00
peer "github.com/libp2p/go-libp2p-peer"
2019-07-08 14:07:09 +00:00
pubsub "github.com/libp2p/go-libp2p-pubsub"
"go.uber.org/fx"
"github.com/filecoin-project/go-fil-markets/retrievalmarket"
"github.com/filecoin-project/go-fil-markets/retrievalmarket/discovery"
2020-01-10 18:01:48 +00:00
"github.com/filecoin-project/go-fil-markets/storagemarket"
"github.com/filecoin-project/lotus/chain"
2019-11-09 23:00:22 +00:00
"github.com/filecoin-project/lotus/chain/blocksync"
2019-12-01 23:11:43 +00:00
"github.com/filecoin-project/lotus/chain/messagepool"
"github.com/filecoin-project/lotus/chain/sub"
2020-02-17 05:51:18 +00:00
"github.com/filecoin-project/lotus/chain/types"
"github.com/filecoin-project/lotus/node/hello"
"github.com/filecoin-project/lotus/node/modules/dtypes"
"github.com/filecoin-project/lotus/node/modules/helpers"
2019-10-17 08:57:56 +00:00
"github.com/filecoin-project/lotus/peermgr"
2019-07-08 14:07:09 +00:00
)
2020-02-17 05:51:18 +00:00
const BlocksTopic = "/fil/blocks"
const MessagesTopic = "/fil/messages"
2019-07-08 14:07:09 +00:00
func RunHello(mctx helpers.MetricsCtx, lc fx.Lifecycle, h host.Host, svc *hello.Service) {
h.SetStreamHandler(hello.ProtocolID, svc.HandleStream)
bundle := inet.NotifyBundle{
ConnectedF: func(_ inet.Network, c inet.Conn) {
go func() {
if err := svc.SayHello(helpers.LifecycleCtx(mctx, lc), c.RemotePeer()); err != nil {
log.Warnw("failed to say hello", "error", err)
return
}
}()
},
}
h.Network().Notify(&bundle)
}
2019-10-17 08:57:56 +00:00
func RunPeerMgr(mctx helpers.MetricsCtx, lc fx.Lifecycle, pmgr *peermgr.PeerMgr) {
go pmgr.Run(helpers.LifecycleCtx(mctx, lc))
}
2019-11-09 23:00:22 +00:00
func RunBlockSync(h host.Host, svc *blocksync.BlockSyncService) {
h.SetStreamHandler(blocksync.BlockSyncProtocolID, svc.HandleStream)
2019-07-08 14:07:09 +00:00
}
2020-02-17 05:51:18 +00:00
func HandleIncomingBlocks(mctx helpers.MetricsCtx, lc fx.Lifecycle, ps *pubsub.PubSub, s *chain.Syncer, h host.Host) {
2019-07-08 14:07:09 +00:00
ctx := helpers.LifecycleCtx(mctx, lc)
2020-02-17 05:51:18 +00:00
blocksub, err := ps.Subscribe(BlocksTopic)
2019-07-08 14:07:09 +00:00
if err != nil {
panic(err)
}
2020-02-17 05:51:18 +00:00
v := sub.NewBlockValidator(ps.BlacklistPeer)
if err := ps.RegisterTopicValidator(BlocksTopic, v.Validate); err != nil {
panic(err)
}
go sub.HandleIncomingBlocks(ctx, blocksub, s, h.ConnManager())
2019-07-08 14:07:09 +00:00
}
2020-02-17 05:51:18 +00:00
func HandleIncomingMessages(mctx helpers.MetricsCtx, lc fx.Lifecycle, ps *pubsub.PubSub, mpool *messagepool.MessagePool) {
2019-07-08 14:07:09 +00:00
ctx := helpers.LifecycleCtx(mctx, lc)
2020-02-17 05:51:18 +00:00
msgsub, err := ps.Subscribe(MessagesTopic)
2019-07-08 14:07:09 +00:00
if err != nil {
panic(err)
}
2020-02-17 05:51:18 +00:00
v := func(ctx context.Context, pid peer.ID, msg *pubsub.Message) bool {
m, err := types.DecodeSignedMessage(msg.GetData())
if err != nil {
log.Errorf("got incorrectly formatted Message: %s", err)
return false
}
msg.ValidatorData = m
return true
}
if err := ps.RegisterTopicValidator(MessagesTopic, v); err != nil {
panic(err)
}
2019-07-08 14:07:09 +00:00
go sub.HandleIncomingMessages(ctx, mpool, msgsub)
}
2019-08-01 17:12:41 +00:00
func RunDealClient(mctx helpers.MetricsCtx, lc fx.Lifecycle, c storagemarket.StorageClient) {
2019-09-10 12:35:43 +00:00
ctx := helpers.LifecycleCtx(mctx, lc)
2019-08-01 17:12:41 +00:00
lc.Append(fx.Hook{
OnStart: func(context.Context) error {
2019-09-10 12:35:43 +00:00
c.Run(ctx)
2019-08-01 17:12:41 +00:00
return nil
},
OnStop: func(context.Context) error {
c.Stop()
return nil
},
})
}
2019-08-14 20:27:10 +00:00
func NewLocalDiscovery(ds dtypes.MetadataDS) *discovery.Local {
return discovery.NewLocal(ds)
}
func RetrievalResolver(l *discovery.Local) retrievalmarket.PeerResolver {
2019-08-26 13:45:36 +00:00
return discovery.Multi(l)
}