86 lines
1.7 KiB
Go
86 lines
1.7 KiB
Go
|
package cmd
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"github.com/spf13/cobra"
|
|||
|
|
"go.uber.org/zap"
|
|||
|
|
"net/http"
|
|||
|
|
"os"
|
|||
|
|
"os/signal"
|
|||
|
|
"packet-ui/internal/model"
|
|||
|
|
"packet-ui/internal/server"
|
|||
|
|
"syscall"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// svrCmd represents the svr command
|
|||
|
|
// go build -ldflags "-w -s" -o packet-admin.out . && ./packet-admin.out svr
|
|||
|
|
var svrCmd = &cobra.Command{
|
|||
|
|
Use: "svr",
|
|||
|
|
Short: "server run",
|
|||
|
|
Long: `server run，default on ` + model.DefaultServerAddress,
|
|||
|
|
Run: RunSvr,
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
rootCmd.AddCommand(svrCmd)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// RunSvr 启动 server
|
|||
|
|
func RunSvr(_ *cobra.Command, _ []string) {
|
|||
|
|
if err := InitServer(); err != nil {
|
|||
|
|
zap.S().Panicf("init svr err, cause: %v", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
runHttpServer()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func runHttpServer() {
|
|||
|
|
var httpServer *http.Server
|
|||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
|||
|
|
|
|||
|
|
defer func() {
|
|||
|
|
if httpServer == nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
err := httpServer.Shutdown(ctx)
|
|||
|
|
|
|||
|
|
if err != nil {
|
|||
|
|
zap.S().Errorf("server stop error(%v)", err)
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
defer func() {
|
|||
|
|
select {
|
|||
|
|
case <-ctx.Done():
|
|||
|
|
default:
|
|||
|
|
cancel()
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
config := model.GetConfig()
|
|||
|
|
if config == nil || config.Svr == nil {
|
|||
|
|
zap.S().Panic("server config nil!", ErrorInitFundamental)
|
|||
|
|
}
|
|||
|
|
httpServer = server.RunServer(config, cancel)
|
|||
|
|
|
|||
|
|
SignalWatchWithContext(ctx, cancel)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func SignalWatchWithContext(ctx context.Context, cancel context.CancelFunc) {
|
|||
|
|
sigterm := make(chan os.Signal, 1)
|
|||
|
|
signal.Notify(sigterm, syscall.SIGINT, syscall.SIGTERM)
|
|||
|
|
select {
|
|||
|
|
case <-ctx.Done():
|
|||
|
|
zap.S().Infof("terminating: context cancelled")
|
|||
|
|
case <-sigterm:
|
|||
|
|
cancel()
|
|||
|
|
zap.S().Infof("terminating: via signal")
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func SignalWatch() {
|
|||
|
|
sigterm := make(chan os.Signal, 1)
|
|||
|
|
sig := <-sigterm
|
|||
|
|
zap.S().Infof("terminating: via signal, %v", sig)
|
|||
|
|
}
|