feat: 完成基本的 model

pkg/api/api.go

@@ -0,0 +1,115 @@
+package api
+
+import (
+	"context"
+	"crypto/tls"
+	"fmt"
+	"loveuer/utodo/pkg/handler"
+	"net"
+
+	"gitea.loveuer.com/yizhisec/packages/logger"
+	"github.com/gofiber/fiber/v3"
+	l3 "github.com/gofiber/fiber/v3/middleware/logger"
+	r3 "github.com/gofiber/fiber/v3/middleware/recover"
+)
+
+type Option func(*option)
+
+type option struct {
+	name      string
+	version   string
+	address   string
+	app       *fiber.App
+	tlsConfig *tls.Config
+}
+
+func WithName(name string) Option {
+	return func(o *option) {
+		if name != "" {
+			o.name = name
+		}
+	}
+}
+
+func WithAddress(address string) Option {
+	return func(o *option) {
+		if address != "" {
+			o.address = address
+		}
+	}
+}
+
+func WithApp(app *fiber.App) Option {
+	return func(o *option) {
+		if app != nil {
+			o.app = app
+		}
+	}
+}
+
+func WithVersion(version string) Option {
+	return func(o *option) {
+		if version != "" {
+			o.version = version
+		}
+	}
+}
+
+func WithTLSConfig(tlsConfig *tls.Config) Option {
+	return func(o *option) {
+		if tlsConfig != nil {
+			o.tlsConfig = tlsConfig
+		}
+	}
+}
+
+func Start(ctx context.Context, opts ...Option) (func(context.Context) error, error) {
+	var (
+		err error
+		fn  func(context.Context) error
+		ln  net.Listener
+		opt = &option{
+			name:    "utodo",
+			version: "0.0.1",
+			address: ":80",
+		}
+	)
+
+	for _, o := range opts {
+		o(opt)
+	}
+
+	if opt.app == nil {
+		opt.app = fiber.New(fiber.Config{BodyLimit: 1024 * 1024 * 5})
+		opt.app.Use(l3.New())
+		opt.app.Use(r3.New())
+		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", opt.address)
+	}
+
+	if err != nil {
+		return nil, err
+	}
+
+	go func() {
+		if err := opt.app.Listener(ln); err != nil {
+			logger.ErrorCtx(ctx, "service[%s:%s] running failed, err = %v", opt.name, opt.version, err)
+		}
+	}()
+
+	fn = func(_ctx context.Context) error {
+		logger.WarnCtx(_ctx, "service[%s] shutdown...", opt.name)
+		if err := opt.app.ShutdownWithContext(_ctx); err != nil {
+			return fmt.Errorf("service[%s:%s] shutdown failed, err = %w", opt.name, opt.version, err)
+		}
+
+		return nil
+	}
+
+	return fn, nil
+}

pkg/handler/healthz.go

@@ -0,0 +1,19 @@
+package handler
+
+import (
+	"time"
+
+	"github.com/gofiber/fiber/v3"
+)
+
+func Healthz(name, version string) fiber.Handler {
+	start := time.Now()
+	return func(c fiber.Ctx) error {
+		return c.JSON(fiber.Map{
+			"name":     name,
+			"version":  version,
+			"start_at": start,
+			"uptime":   time.Since(start).String(),
+		})
+	}
+}