81 lines
1.3 KiB
Go
81 lines
1.3 KiB
Go
package floodsub
|
|
|
|
import (
|
|
"context"
|
|
"time"
|
|
|
|
pb "github.com/libp2p/go-floodsub/pb"
|
|
|
|
host "github.com/libp2p/go-libp2p-host"
|
|
peer "github.com/libp2p/go-libp2p-peer"
|
|
protocol "github.com/libp2p/go-libp2p-protocol"
|
|
)
|
|
|
|
const (
|
|
GossipSubID = protocol.ID("/meshsub/1.0.0")
|
|
)
|
|
|
|
func NewGossipSub(ctx context.Context, h host.Host, opts ...Option) (*PubSub, error) {
|
|
rt := &GossipSubRouter{}
|
|
return NewPubSub(ctx, h, rt, opts...)
|
|
}
|
|
|
|
type GossipSubRouter struct {
|
|
p *PubSub
|
|
}
|
|
|
|
func (fs *GossipSubRouter) Protocols() []protocol.ID {
|
|
return []protocol.ID{GossipSubID, FloodSubID}
|
|
}
|
|
|
|
func (fs *GossipSubRouter) Attach(p *PubSub) {
|
|
fs.p = p
|
|
go fs.heartbeatTimer()
|
|
}
|
|
|
|
func (fs *GossipSubRouter) AddPeer(peer.ID, protocol.ID) {
|
|
|
|
}
|
|
|
|
func (fs *GossipSubRouter) RemovePeer(peer.ID) {
|
|
|
|
}
|
|
|
|
func (fs *GossipSubRouter) HandleRPC(rpc *RPC) {
|
|
|
|
}
|
|
|
|
func (fs *GossipSubRouter) Publish(from peer.ID, msg *pb.Message) {
|
|
|
|
}
|
|
|
|
func (fs *GossipSubRouter) Join(topic string) {
|
|
|
|
}
|
|
|
|
func (fs *GossipSubRouter) Leave(topic string) {
|
|
|
|
}
|
|
|
|
func (fs *GossipSubRouter) heartbeatTimer() {
|
|
ticker := time.NewTicker(1 * time.Second)
|
|
defer ticker.Stop()
|
|
|
|
for {
|
|
select {
|
|
case <-ticker.C:
|
|
select {
|
|
case fs.p.eval <- fs.heartbeat:
|
|
case <-fs.p.ctx.Done():
|
|
return
|
|
}
|
|
case <-fs.p.ctx.Done():
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
func (fs *GossipSubRouter) heartbeat() {
|
|
|
|
}
|