//go:build client

package main

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

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

// This example demonstrates client connecting to a protocol version server.
// Usage:
//
//	go run ./examples/protocol_version/client.go -addr tcp://127.0.0.1:8082 -cmd version
//	go run ./examples/protocol_version/client.go -addr tcp://127.0.0.1:8082 -cmd echo -data "hello"
func main() {
	addr := flag.String("addr", "tcp://127.0.0.1:8082", "server address")
	cmd := flag.String("cmd", "version", "command to send (version, echo)")
	data := flag.String("data", "test", "data to send for echo command")
	timeout := flag.Duration("timeout", 3*time.Second, "request timeout")
	flag.Parse()

	cfg := &nnet.ClientConfig{
		Addr:                *addr,
		ApplicationProtocol: "nnet", // 指定应用层协议
		ConnectTimeout:      3 * time.Second,
		ReadTimeout:         3 * time.Second,
		WriteTimeout:        3 * time.Second,
	}

	c := nnet.NewClient(cfg)
	defer c.Close()

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

	// 发送命令
	var reqData string
	if *cmd == "echo" {
		reqData = *data
	} else {
		reqData = *cmd
	}

	resp, err := c.Request([]byte(reqData), *timeout)
	if err != nil {
		fmt.Fprintf(os.Stderr, "request error: %v\n", err)
		os.Exit(1)
	}

	fmt.Printf("response: %s\n", string(resp))
}