go-peer/pkg/network/examples/echo/main.go

package main

import (
	"context"
	"fmt"
	"time"

	"github.com/number571/go-peer/pkg/network"
	"github.com/number571/go-peer/pkg/network/conn"
	"github.com/number571/go-peer/pkg/network/message"
	"github.com/number571/go-peer/pkg/payload"
)

// client <-> service

const (
	serviceHeader  = 0xDEADBEAF
	serviceAddress = "127.0.0.1:8080"
)

var handler = func(ctx context.Context, node network.INode, c conn.IConn, msg message.IMessage) error {
	resp := fmt.Sprintf("echo: [%s]", string(msg.GetPayload().GetBody()))
	_ = c.WriteMessage(
		ctx,
		message.NewMessage(
			message.NewConstructSettings(&message.SConstructSettings{
				FSettings: node.GetSettings().GetConnSettings().GetMessageSettings(),
			}),
			payload.NewPayload32(serviceHeader, []byte(resp)),
		),
	)
	return nil
}

func main() {
	var (
		_       = runServiceNode()
		ctx     = context.Background()
		conn, _ = conn.Connect(ctx, connSettings(), serviceAddress)
	)

	_ = conn.WriteMessage(
		ctx,
		message.NewMessage(
			message.NewConstructSettings(&message.SConstructSettings{
				FSettings: conn.GetSettings().GetMessageSettings(),
			}),
			payload.NewPayload32(serviceHeader, []byte("hello, world!")),
		),
	)

	readCh := make(chan struct{})
	go func() { <-readCh }()

	recvMsg, _ := conn.ReadMessage(ctx, readCh)
	fmt.Println(string(recvMsg.GetPayload().GetBody()))
}

func runServiceNode() network.INode {
	ctx := context.Background()
	node := newNode(serviceAddress).HandleFunc(serviceHeader, handler)

	go func() { _ = node.Listen(ctx) }()

	time.Sleep(time.Second) // wait listener
	return node
}