101 lines
2.1 KiB
Go
101 lines
2.1 KiB
Go
|
package cmd
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"errors"
|
|||
|
|
"github.com/spf13/cobra"
|
|||
|
|
"net/http"
|
|||
|
|
"os"
|
|||
|
|
"os/signal"
|
|||
|
|
"strconv"
|
|||
|
|
"syscall"
|
|||
|
|
"time"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
func init() {
|
|||
|
|
RootCmd.AddCommand(httpCmd)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var httpCmd = &cobra.Command{
|
|||
|
|
Use: "http",
|
|||
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|||
|
|
initHttpServer()
|
|||
|
|
},
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
func initHttpServer() {
|
|||
|
|
app, err := CreateApp()
|
|||
|
|
if err != nil {
|
|||
|
|
panic(err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if app.Config.Http.Host == "" {
|
|||
|
|
app.Config.Http.Host = "0.0.0.0"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if app.Config.Http.Port == 0 {
|
|||
|
|
app.Config.Http.Port = 8000
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
var bizServer *http.Server
|
|||
|
|
var metricServer *http.Server
|
|||
|
|
|
|||
|
|
bizServer = &http.Server{
|
|||
|
|
Addr: ":8080",
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
bizServer.Addr = app.Config.Http.Host + ":" + strconv.Itoa(app.Config.Http.Port)
|
|||
|
|
bizServer.Handler = app.HttpServer.BizRouter()
|
|||
|
|
|
|||
|
|
// 启动 http
|
|||
|
|
go func() {
|
|||
|
|
// refresh
|
|||
|
|
app.Service.Jwks.SetupAuthRefresh()
|
|||
|
|
|
|||
|
|
app.Logger.Sugar.Info("Listening and serving HTTP on ", bizServer.Addr)
|
|||
|
|
err = bizServer.ListenAndServe()
|
|||
|
|
if err != nil && !errors.Is(http.ErrServerClosed, err) {
|
|||
|
|
panic(err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
// 启动 metrics
|
|||
|
|
if app.Config.Metrics.Enabled {
|
|||
|
|
metricServer = &http.Server{
|
|||
|
|
Addr: ":8080",
|
|||
|
|
}
|
|||
|
|
metricServer.Addr = app.Config.Metrics.Host + ":" + strconv.Itoa(app.Config.Metrics.Port)
|
|||
|
|
go func() {
|
|||
|
|
app.Logger.Sugar.Info("Metrics and serving HTTP on ", metricServer.Addr)
|
|||
|
|
|
|||
|
|
metricServer.Handler = app.HttpServer.MetricRouter()
|
|||
|
|
|
|||
|
|
err = metricServer.ListenAndServe()
|
|||
|
|
if err != nil && !errors.Is(http.ErrServerClosed, err) {
|
|||
|
|
panic(err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
}()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 等待一个 INT 或 TERM 信号
|
|||
|
|
quit := make(chan os.Signal)
|
|||
|
|
signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
|
|||
|
|
<-quit
|
|||
|
|
app.Logger.Sugar.Info("Shutdown Server ...")
|
|||
|
|
// 创建超时上下文，Shutdown 可以让未处理的连接在这个时间内关闭
|
|||
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|||
|
|
defer cancel()
|
|||
|
|
|
|||
|
|
// 停止HTTP服务器
|
|||
|
|
if err := bizServer.Shutdown(ctx); err != nil {
|
|||
|
|
app.Logger.Sugar.Fatal("Biz Server Shutdown Error:", err)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if err := metricServer.Shutdown(ctx); err != nil {
|
|||
|
|
app.Logger.Sugar.Fatal("Metric Server Shutdown Error:", err)
|
|||
|
|
}
|
|||
|
|
}
|