|
|
//go:build client
|
|
|
|
|
|
package main
|
|
|
|
|
|
import (
|
|
|
"flag"
|
|
|
"fmt"
|
|
|
"log"
|
|
|
"os"
|
|
|
"time"
|
|
|
|
|
|
"github.com/noahlann/nnet/pkg/nnet"
|
|
|
)
|
|
|
|
|
|
// This example demonstrates client testing interceptor validation.
|
|
|
// Usage:
|
|
|
//
|
|
|
// go run ./examples/interceptor/client.go -addr tcp://127.0.0.1:8084 -cmd validate -data "hello"
|
|
|
// go run ./examples/interceptor/client.go -addr tcp://127.0.0.1:8084 -cmd validate -data "hi" # too short
|
|
|
// go run ./examples/interceptor/client.go -addr tcp://127.0.0.1:8084 -cmd validate -data "forbidden" # rejected
|
|
|
// go run ./examples/interceptor/client.go -addr tcp://127.0.0.1:8084 -cmd echo -data "test"
|
|
|
// go run ./examples/interceptor/client.go -addr tcp://127.0.0.1:8084 -cmd test -data "valid data"
|
|
|
func main() {
|
|
|
addr := flag.String("addr", "tcp://127.0.0.1:8084", "server address")
|
|
|
cmd := flag.String("cmd", "validate", "command to send (validate, echo, test)")
|
|
|
data := flag.String("data", "hello world", "data to send")
|
|
|
timeout := flag.Duration("timeout", 3*time.Second, "request timeout")
|
|
|
flag.Parse()
|
|
|
|
|
|
cfg := &nnet.ClientConfig{
|
|
|
Addr: *addr,
|
|
|
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
|
|
|
switch *cmd {
|
|
|
case "validate":
|
|
|
// validate命令：发送"validate " + 数据
|
|
|
if *data != "" {
|
|
|
reqData = "validate " + *data
|
|
|
} else {
|
|
|
reqData = "validate"
|
|
|
}
|
|
|
case "echo", "test":
|
|
|
// echo和test命令：只发送命令
|
|
|
reqData = *cmd
|
|
|
default:
|
|
|
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))
|
|
|
}
|
|
|
|