feat: 添加提示语

api/api.go

@@ -4,34 +4,89 @@ import (
 	"context"
 	"crypto/tls"
 	"errors"
+	"gitea.loveuer.com/yizhisec/packages/handler"
 	"gitea.loveuer.com/yizhisec/packages/logger"
 	"github.com/gin-gonic/gin"
 	"net"
 	"net/http"
 )
 
-type Api struct {
-	Address   string
-	Name      string
-	App       *gin.Engine
-	TlsConfig *tls.Config
+type Option func(*option)
+
+type option struct {
+	name      string
+	version   string
+	address   string
+	app       *gin.Engine
+	tlsConfig *tls.Config
 }
 
-func New(ctx context.Context, api *Api) (func(context.Context) error, error) {
+func WithName(name string) Option {
+	return func(o *option) {
+		if name == "" {
+			o.name = name
+		}
+	}
+}
+
+func WithVersion(version string) Option {
+	return func(o *option) {
+		if version == "" {
+			o.version = version
+		}
+	}
+}
+
+func WithAddress(address string) Option {
+	return func(o *option) {
+		if address == "" {
+			o.address = address
+		}
+	}
+}
+
+func WithApp(app *gin.Engine) Option {
+	return func(o *option) {
+		if app != nil {
+			o.app = app
+		}
+	}
+}
+
+func WithTLSConfig(tlsConfig *tls.Config) Option {
+	return func(o *option) {
+		if tlsConfig != nil {
+			o.tlsConfig = tlsConfig
+		}
+	}
+}
+
+func New(ctx context.Context, optFns ...Option) (func(context.Context) error, error) {
 	var (
 		err error
 		fn  func(context.Context) error
 		ln  net.Listener
+		opt = &option{
+			name:      "unknown",
+			version:   "v0.0.1",
+			address:   "127.0.0.1:9119",
+			tlsConfig: nil,
+		}
 	)
 
-	if api == nil {
-		return fn, errors.New("api is nil")
+	for _, ofn := range optFns {
+		ofn(opt)
 	}
 
-	if api.TlsConfig != nil {
-		ln, err = tls.Listen("tcp", api.Address, api.TlsConfig)
+	if opt.app == nil {
+		opt.app = gin.Default()
+		opt.app.GET("/healthz", handler.Healthz(opt.name, opt.version))
+	}
+
+	if opt.tlsConfig != nil {
+		ln, err = tls.Listen("tcp", opt.address, opt.tlsConfig)
 	} else {
-		ln, err = net.Listen("tcp", api.Address)
+		ln, err = net.Listen("tcp", opt.address)
 	}
 
 	if err != nil {
@@ -39,11 +94,11 @@ func New(ctx context.Context, api *Api) (func(context.Context) error, error) {
 	}
 
 	svc := &http.Server{
-		Handler: api.App,
+		Handler: opt.app,
 	}
 
 	go func() {
-		logger.InfoCtx(ctx, "[%s] api svc running at: %s", api.Name, api.Address)
+		logger.InfoCtx(ctx, "[%s] api svc running at: %s", opt.name, opt.address)
 		if err = svc.Serve(ln); err != nil {
 			if !errors.Is(err, http.ErrServerClosed) {
 				logger.ErrorCtx(ctx, "api svc run failed, err = %s", err.Error())
@@ -52,7 +107,7 @@ func New(ctx context.Context, api *Api) (func(context.Context) error, error) {
 	}()
 
 	fn = func(timeout context.Context) error {
-		logger.WarnCtx(ctx, "[%s] api svc shutdown...", api.Name)
+		logger.WarnCtx(ctx, "[%s] api svc shutdown...", opt.name)
 		return svc.Shutdown(timeout)
 	}
 

handler/healthz.go

@@ -0,0 +1,19 @@
+package handler
+
+import (
+	"gitea.loveuer.com/yizhisec/packages/resp"
+	"github.com/gin-gonic/gin"
+	"time"
+)
+
+func Healthz(name, version string) gin.HandlerFunc {
+	start := time.Now()
+	return func(c *gin.Context) {
+		resp.R200(c, gin.H{
+			"name":     name,
+			"version":  version,
+			"start_at": start,
+			"uptime":   time.Since(start).String(),
+		})
+	}
+}

resp/msg.go

@@ -1,15 +1,16 @@
 package resp
 
 const (
-	Msg200        = "操作成功"
-	Msg400        = "参数错误"
-	Msg401        = "该账号登录已失效, 请重新登录"
-	Msg401NoMulti = "用户已在其他地方登录"
-	Msg403        = "权限不足"
-	Msg404        = "资源不存在"
-	Msg500        = "服务器开小差了"
-	Msg501        = "服务不可用"
-	Msg503        = "服务不可用或正在升级, 请联系管理员"
+	Msg200         = "操作成功"
+	Msg400         = "参数错误"
+	Msg401         = "该账号登录已失效, 请重新登录"
+	Msg401NoMulti  = "用户已在其他地方登录"
+	Msg401Inactive = "当前用户尚未生效, 请稍后再试"
+	Msg403         = "权限不足"
+	Msg404         = "资源不存在"
+	Msg500         = "服务器开小差了"
+	Msg501         = "服务不可用"
+	Msg503         = "服务不可用或正在升级, 请联系管理员"
 )
 
 func Msg(status int) string {

resp/resp.go

@@ -43,23 +43,45 @@ func RE(c *gin.Context, err error) {
 
 func _r(c *gin.Context, r *res, args ...any) {
 	length := len(args)
-	switch length {
-	case 0:
-		break
-	case 1:
-		if msg, ok := args[0].(string); ok {
-			r.Msg = msg
-		}
-	case 2:
-		r.Data = args[1]
-	case 3:
-		r.Err = args[2]
-	case 4:
-		if code, err := cast.ToIntE(args[3]); err == nil {
-			r.Code = code
-		}
+
+	if length == 0 {
+		goto END
 	}
 
+	if length >= 4 {
+		goto H4
+	}
+
+	if length >= 3 {
+		goto H3
+	}
+
+	if length >= 2 {
+		goto H2
+	}
+	if length >= 1 {
+		goto H1
+	}
+
+H4:
+	if code, err := cast.ToIntE(args[3]); err == nil {
+		r.Code = code
+	}
+H3:
+	if es, ok := args[2].(error); ok {
+		r.Err = es.Error()
+	} else {
+		r.Err = args[2]
+	}
+H2:
+	r.Data = args[1]
+H1:
+	if msg, ok := args[0].(string); ok {
+		r.Msg = msg
+	}
+
+END:
+
 	if r.Msg == "" {
 		r.Msg = Msg(r.Status)
 	}

resp/resp_test.go

@@ -0,0 +1,9 @@
+package resp
+
+import (
+	"testing"
+)
+
+func TestResp(t *testing.T) {
+
+}