119 lines
2.2 KiB
Go
119 lines
2.2 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"crypto/tls"
|
|
"fmt"
|
|
"loveuer/utodo/pkg/handler"
|
|
"loveuer/utodo/pkg/logger"
|
|
"net"
|
|
|
|
"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 {
|
|
logger.WarnCtx(ctx, "app is nil, use default app!!!")
|
|
logger.WarnCtx(ctx, "app is nil, use default app!!!")
|
|
logger.WarnCtx(ctx, "app is nil, use default app!!!")
|
|
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
|
|
}
|