libp2p - pubsub com Golang

package main

import (
    "bufio"
    "context"
    "fmt"
    "os"
    "time"

    "github.com/libp2p/go-libp2p"
    "github.com/libp2p/go-libp2p-core/host"
    "github.com/libp2p/go-libp2p-core/peer"
    pubsub "github.com/libp2p/go-libp2p-pubsub"
    "github.com/libp2p/go-libp2p/p2p/discovery/mdns"
)

// DiscoveryInterval é a frequência com que republicamos nossos registros mDNS.
const DiscoveryInterval = time.Hour

// DiscoveryServiceTag é usado em nossos anúncios mDNS para descobrir outros pares para chat.
const DiscoveryServiceTag = "librum-pubsub"

func main() {
    ctx := context.Background()

    // criar um novo host libp2p que acessa em uma porta TCP aleatória
    // podemos especificar uma porta como /ip4/0.0.0.0/tcp/3326
    host, err := libp2p.New(libp2p.ListenAddrStrings("/ip4/0.0.0.0/tcp/0"))
    if err != nil {
        panic(err)
    }

    // ver detalhes e endereços do host
    fmt.Printf("host ID %s\n", host.ID().Pretty())
    fmt.Printf("following are the assigned addresses\n")
    for _, addr := range host.Addrs() {
        fmt.Printf("%s\n", addr.String())
    }
    fmt.Printf("\n")

    // criar um novo serviço PubSub usando o roteador GossipSub
    gossipSub, err := pubsub.NewGossipSub(ctx, host)
    if err != nil {
        panic(err)
    }

    // configurar descoberta de mDNS local
    if err := setupDiscovery(host); err != nil {
        panic(err)
    }

    // juntar-se ao tópico pubsub chamado librum
    room := "librum"
    topic, err := gossipSub.Join(room)
    if err != nil {
        panic(err)
    }

    // criar publicador
    publish(ctx, topic)
}

// iniciar publicador para tópico
func publish(ctx context.Context, topic *pubsub.Topic) {
    for {
        scanner := bufio.NewScanner(os.Stdin)
        for scanner.Scan() {
            fmt.Printf("digite a mensagem para publicar: \n")

            msg := scanner.Text()
            if len(msg) != 0 {
                // publicar mensagem no tópico
                bytes := []byte(msg)
                topic.Publish(ctx, bytes)
            }
        }
    }
}

// DiscoveryNotifee é notificado quando encontramos um novo par por meio da descoberta mDNS
type discoveryNotifee struct {
    h host.Host
}

// HandlePeerFound se conecta a pares descobertos via mDNS. Assim que estiverem conectados, o sistema PubSub começará a interagir automaticamente com eles se também suportarem o PubSub
func (n *discoveryNotifee) HandlePeerFound(pi peer.AddrInfo) {
    fmt.Printf("discovered new peer %s\n", pi.ID.Pretty())
    err := n.h.Connect(context.Background(), pi)
    if err != nil {
        fmt.Printf("error connecting to peer %s: %s\n", pi.ID.Pretty(), err)
    }
}

// setupDiscovery cria um serviço de descoberta mDNS e o anexa ao Host libp2p.
// Isso nos permite descobrir automaticamente pares na mesma LAN e nos conectar a eles.
func setupDiscovery(h host.Host) error {
    // configurar descoberta de mDNS para encontrar pares locais
    s := mdns.NewMdnsService(h, DiscoveryServiceTag, &discoveryNotifee{h: h})
    return s.Start()
}

package main

import (
    "context"
    "fmt"
    "time"

    "github.com/libp2p/go-libp2p"
    "github.com/libp2p/go-libp2p-core/host"
    "github.com/libp2p/go-libp2p-core/peer"
    pubsub "github.com/libp2p/go-libp2p-pubsub"
    "github.com/libp2p/go-libp2p/p2p/discovery/mdns"
)

// DiscoveryInterval é a frequência com que republicamos nossos registros mDNS.
const DiscoveryInterval = time.Hour

// DiscoveryServiceTag é usado em nossos anúncios mDNS para descobrir outros pares para chat.
const DiscoveryServiceTag = "librum-pubsub"

func main() {
    ctx := context.Background()

    // criar um novo host libp2p que acessa em uma porta TCP aleatória
    host, err := libp2p.New(libp2p.ListenAddrStrings("/ip4/0.0.0.0/tcp/0"))
    if err != nil {
        panic(err)
    }

    // ver detalhes e endereços do host
    fmt.Printf("host ID %s\n", host.ID().Pretty())
    fmt.Printf("following are the assigned addresses\n")
    for _, addr := range host.Addrs() {
        fmt.Printf("%s\n", addr.String())
    }
    fmt.Printf("\n")

    // criar um novo serviço PubSub usando o roteador GossipSub
    gossipSub, err := pubsub.NewGossipSub(ctx, host)
    if err != nil {
        panic(err)
    }

    // configurar descoberta de mDNS local
    if err := setupDiscovery(host); err != nil {
        panic(err)
    }

    // juntar-se ao tópico pubsub chamado librum
    room := "librum"
    topic, err := gossipSub.Join(room)
    if err != nil {
        panic(err)
    }

    // inscrever no tópico
    subscriber, err := topic.Subscribe()
    if err != nil {
        panic(err)
    }
    subscribe(subscriber, ctx, host.ID())
}

// iniciar o subscritor do tópico
func subscribe(subscriber *pubsub.Subscription, ctx context.Context, hostID peer.ID) {
    for {
        msg, err := subscriber.Next(ctx)
        if err != nil {
            panic(err)
        }

        // considerar apenas mensagens entregues por outros pares
        if msg.ReceivedFrom == hostID {
            continue
        }

        fmt.Printf("got message: %s, from: %s\n", string(msg.Data), msg.ReceivedFrom.Pretty())
    }
}

// DiscoveryNotifee é notificado quando encontramos um novo par por meio da descoberta mDNS
type discoveryNotifee struct {
    h host.Host
}

// HandlePeerFound se conecta a pares descobertos via mDNS. Assim que estiverem conectados, o sistema PubSub começará a interagir automaticamente com eles se também suportarem o PubSub.
func (n *discoveryNotifee) HandlePeerFound(pi peer.AddrInfo) {
    fmt.Printf("discovered new peer %s\n", pi.ID.Pretty())
    err := n.h.Connect(context.Background(), pi)
    if err != nil {
        fmt.Printf("error connecting to peer %s: %s\n", pi.ID.Pretty(), err)
    }
}

// setupDiscovery cria um serviço de descoberta mDNS e o anexa ao Host libp2p.
// Isso nos permite descobrir automaticamente pares na mesma LAN e nos conectar a eles.
func setupDiscovery(h host.Host) error {
    // configurar descoberta de mDNS para encontrar pares locais
    s := mdns.NewMdnsService(h, DiscoveryServiceTag, &discoveryNotifee{h: h})
    return s.Start()
}

# instalar dependências do golang  
go mod init
go mod tidy

---

# criar publicador e subscritores no diretório ./build
go build -o build/publisher src/publisher.go
go build -o build/subscriber src/subsciber.go

---

# executar subscritores1 (no terminal 1)
./build/subscriber

# executar subscritores2 (no terminal 2)
./build/subscriber

# executar publicador (no terminal 3)
./build/publisher