You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
57 lines
1.3 KiB
Go
57 lines
1.3 KiB
Go
package main
|
|
|
|
import (
|
|
"log"
|
|
"net/http"
|
|
"os"
|
|
"os/signal"
|
|
"syscall"
|
|
|
|
"github.com/noahlann/nnet/pkg/nnet"
|
|
)
|
|
|
|
func main() {
|
|
// HTTP mux 用于暴露 metrics 与健康检查
|
|
metricsMux := http.NewServeMux()
|
|
|
|
server, err := nnet.NewPreset().
|
|
WithAddress("tcp://:6995").
|
|
WithMetricsHandler(metricsMux, "/metrics").
|
|
WithHealthHandler(metricsMux, "/healthz").
|
|
WithRouteString("/hello", func(ctx nnet.Context) error {
|
|
return ctx.Response().WriteString("hello from preset\n")
|
|
}).
|
|
WithRouteString("/ping", func(ctx nnet.Context) error {
|
|
return ctx.Response().WriteString("pong\n")
|
|
}).
|
|
Build()
|
|
if err != nil {
|
|
log.Fatal("preset build failed:", err)
|
|
}
|
|
|
|
// 启动监控 HTTP 服务
|
|
go func() {
|
|
log.Println("metrics & health available on :9090 (/metrics, /healthz)")
|
|
if err := http.ListenAndServe(":9090", metricsMux); err != nil {
|
|
log.Fatal("metrics server failed:", err)
|
|
}
|
|
}()
|
|
|
|
// 监听退出信号
|
|
done := make(chan os.Signal, 1)
|
|
signal.Notify(done, syscall.SIGINT, syscall.SIGTERM)
|
|
|
|
go func() {
|
|
log.Println("nnet preset server listening on :6995")
|
|
if err := server.Start(); err != nil {
|
|
log.Fatal("server start failed:", err)
|
|
}
|
|
}()
|
|
|
|
<-done
|
|
log.Println("shutting down...")
|
|
if err := server.Stop(); err != nil {
|
|
log.Fatal("server stop failed:", err)
|
|
}
|
|
}
|