|
|
package main
|
|
|
|
|
|
import (
|
|
|
"log"
|
|
|
"os"
|
|
|
"os/signal"
|
|
|
"syscall"
|
|
|
|
|
|
"github.com/noahlann/nnet/pkg/nnet"
|
|
|
)
|
|
|
|
|
|
// User 用户结构
|
|
|
type User struct {
|
|
|
ID int `json:"id"`
|
|
|
Name string `json:"name"`
|
|
|
Age int `json:"age"`
|
|
|
}
|
|
|
|
|
|
// Response 响应结构
|
|
|
type Response struct {
|
|
|
Code int `json:"code"`
|
|
|
Message string `json:"message"`
|
|
|
Data interface{} `json:"data"`
|
|
|
}
|
|
|
|
|
|
func main() {
|
|
|
// 创建服务器配置
|
|
|
cfg := &nnet.Config{
|
|
|
Addr: "tcp://:6995",
|
|
|
Codec: &nnet.CodecConfig{
|
|
|
DefaultCodec: "json", // 默认使用JSON编解码器
|
|
|
EnableProtocolEncode: false, // 不启用协议层编码
|
|
|
},
|
|
|
}
|
|
|
|
|
|
// 创建服务器
|
|
|
server, err := nnet.NewServer(cfg)
|
|
|
if err != nil {
|
|
|
log.Fatal("Failed to create server:", err)
|
|
|
}
|
|
|
|
|
|
// 注册路由处理器
|
|
|
router := server.Router()
|
|
|
|
|
|
// 示例1：写入字符串
|
|
|
router.RegisterString("hello", helloHandler)
|
|
|
|
|
|
// 示例2：写入结构体（自动JSON编码）
|
|
|
router.RegisterString("get_user", getUserHandler)
|
|
|
|
|
|
// 示例3：写入数组
|
|
|
router.RegisterString("list_users", listUsersHandler)
|
|
|
|
|
|
// 示例4：使用指定的编解码器
|
|
|
router.RegisterString("get_user_msgpack", getUserWithMsgpackHandler)
|
|
|
|
|
|
// 启动服务器
|
|
|
log.Println("Starting server on :6995")
|
|
|
log.Println("Try:")
|
|
|
log.Println(" echo 'hello' | nc localhost 6995")
|
|
|
log.Println(" echo 'get_user' | nc localhost 6995")
|
|
|
log.Println(" echo 'list_users' | nc localhost 6995")
|
|
|
log.Println(" echo 'get_user_msgpack' | nc localhost 6995")
|
|
|
|
|
|
if err := server.Start(); err != nil {
|
|
|
log.Fatal("Failed to start server:", err)
|
|
|
}
|
|
|
|
|
|
// 等待中断信号
|
|
|
quit := make(chan os.Signal, 1)
|
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|
|
<-quit
|
|
|
|
|
|
log.Println("Shutting down server...")
|
|
|
if err := server.Stop(); err != nil {
|
|
|
log.Fatal("Failed to stop server:", err)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
// helloHandler 写入字符串
|
|
|
func helloHandler(ctx nnet.Context) error {
|
|
|
return ctx.Response().WriteString("Hello, World!\n")
|
|
|
}
|
|
|
|
|
|
// getUserHandler 返回单个用户（使用默认JSON编解码器）
|
|
|
func getUserHandler(ctx nnet.Context) error {
|
|
|
user := User{
|
|
|
ID: 1,
|
|
|
Name: "Alice",
|
|
|
Age: 25,
|
|
|
}
|
|
|
|
|
|
resp := Response{
|
|
|
Code: 200,
|
|
|
Message: "success",
|
|
|
Data: user,
|
|
|
}
|
|
|
|
|
|
// 自动使用JSON编码
|
|
|
return ctx.Response().Write(resp)
|
|
|
}
|
|
|
|
|
|
// listUsersHandler 返回用户列表
|
|
|
func listUsersHandler(ctx nnet.Context) error {
|
|
|
users := []User{
|
|
|
{ID: 1, Name: "Alice", Age: 25},
|
|
|
{ID: 2, Name: "Bob", Age: 30},
|
|
|
{ID: 3, Name: "Charlie", Age: 35},
|
|
|
}
|
|
|
|
|
|
resp := Response{
|
|
|
Code: 200,
|
|
|
Message: "success",
|
|
|
Data: users,
|
|
|
}
|
|
|
|
|
|
return ctx.Response().Write(resp)
|
|
|
}
|
|
|
|
|
|
// getUserWithMsgpackHandler 使用MessagePack编解码器
|
|
|
func getUserWithMsgpackHandler(ctx nnet.Context) error {
|
|
|
user := User{
|
|
|
ID: 1,
|
|
|
Name: "Alice",
|
|
|
Age: 25,
|
|
|
}
|
|
|
|
|
|
resp := Response{
|
|
|
Code: 200,
|
|
|
Message: "success",
|
|
|
Data: user,
|
|
|
}
|
|
|
|
|
|
// 使用MessagePack编码
|
|
|
return ctx.Response().WriteWithCodec(resp, "msgpack")
|
|
|
}
|