//go:build client

package main

import (
	"flag"
	"fmt"
	"log"
	"os"
	"time"

	"github.com/noahlann/nnet/pkg/nnet"
)

func main() {
	addr := flag.String("addr", "udp://127.0.0.1:8084", "server address")
	data := flag.String("data", "hello-udp", "message")
	flag.Parse()

	cfg := &nnet.ClientConfig{
		Addr:           *addr,
		ConnectTimeout: 1 * time.Second,
		ReadTimeout:    1 * time.Second,
		WriteTimeout:   1 * time.Second,
	}
	c := nnet.NewClient(cfg)
	defer c.Close()

	if err := c.Connect(); err != nil {
		log.Fatalf("connect failed: %v", err)
	}
	defer c.Disconnect()

	if err := c.Send([]byte(*data)); err != nil {
		fmt.Fprintf(os.Stderr, "send error: %v\n", err)
		os.Exit(1)
	}
	resp, err := c.Receive()
	if err != nil {
		fmt.Fprintf(os.Stderr, "receive error: %v\n", err)
		os.Exit(1)
	}
	fmt.Printf("echo: %s\n", string(resp))
}