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)
|
||
}
|
||
}
|