This commit is contained in:
oneao committed 2026-04-06 21:15:11 +08:00
1 parent f08db1959b
commit c6671c8a8c
68 files changed
+13904 -1131

No files matched your search

+37
View File
@@ -0,0 +1,37 @@
*.o
*.a
*.so
_obj
_test
*.[568vq]
[568vq].out
*.cgo1.go
*.cgo2.c
_cgo_defun.c
_cgo_gotypes.go
_cgo_export.*
_testmain.go
*.exe
*.exe~
*.test
*.prof
*.rar
*.zip
*.gz
*.psd
*.bmd
*.cfg
*.pptx
*.log
*nohup.out
*settings.pyc
*.sublime-project
*.sublime-workspace
!.gitkeep
.DS_Store
/.idea
/.vscode
/output
*.local.yml
dumped_hertz_remote_config.json
+13 -7
View File
@@ -3,15 +3,21 @@ module allapp-go
go 1.25.0
require (
github.com/go-playground/validator/v10 v10.30.2
github.com/gofiber/fiber/v3 v3.1.0
github.com/golang-jwt/jwt/v5 v5.3.1
github.com/jackc/pgx/v5 v5.9.1
github.com/spf13/viper v1.21.0
go.uber.org/zap v1.27.1
gopkg.in/natefinch/lumberjack.v2 v2.2.1
)
require (
github.com/andybalholm/brotli v1.2.0 // indirect
github.com/fsnotify/fsnotify v1.9.0 // indirect
github.com/gabriel-vasile/mimetype v1.4.13 // indirect
github.com/go-playground/locales v0.14.1 // indirect
github.com/go-playground/universal-translator v0.18.1 // indirect
github.com/go-viper/mapstructure/v2 v2.4.0 // indirect
github.com/gofiber/schema v1.7.0 // indirect
github.com/gofiber/utils/v2 v2.0.2 // indirect
@@ -20,10 +26,12 @@ require (
github.com/jackc/pgservicefile v0.0.0-20240606120523-5a60cdf6a761 // indirect
github.com/jackc/puddle/v2 v2.2.2 // indirect
github.com/klauspost/compress v1.18.4 // indirect
github.com/leodido/go-urn v1.4.0 // indirect
github.com/mattn/go-colorable v0.1.14 // indirect
github.com/mattn/go-isatty v0.0.20 // indirect
github.com/pelletier/go-toml/v2 v2.2.4 // indirect
github.com/philhofer/fwd v1.2.0 // indirect
github.com/pkg/errors v0.9.1 // indirect
github.com/sagikazarmark/locafero v0.11.0 // indirect
github.com/sourcegraph/conc v0.3.1-0.20240121214520-5f936abd7ae8 // indirect
github.com/spf13/afero v1.15.0 // indirect
@@ -34,12 +42,10 @@ require (
github.com/valyala/bytebufferpool v1.0.0 // indirect
github.com/valyala/fasthttp v1.69.0 // indirect
go.uber.org/multierr v1.10.0 // indirect
go.uber.org/zap v1.27.1 // indirect
go.yaml.in/yaml/v3 v3.0.4 // indirect
golang.org/x/crypto v0.48.0 // indirect
golang.org/x/net v0.50.0 // indirect
golang.org/x/sync v0.19.0 // indirect
golang.org/x/sys v0.41.0 // indirect
golang.org/x/text v0.34.0 // indirect
gopkg.in/natefinch/lumberjack.v2 v2.2.1 // indirect
golang.org/x/crypto v0.49.0 // indirect
golang.org/x/net v0.51.0 // indirect
golang.org/x/sync v0.20.0 // indirect
golang.org/x/sys v0.42.0 // indirect
golang.org/x/text v0.35.0 // indirect
)
+26 -10
View File
@@ -9,6 +9,16 @@ github.com/fsnotify/fsnotify v1.9.0 h1:2Ml+OJNzbYCTzsxtv8vKSFD9PbJjmhYF14k/jKC7S
github.com/fsnotify/fsnotify v1.9.0/go.mod h1:8jBTzvmWwFyi3Pb8djgCCO5IBqzKJ/Jwo8TRcHyHii0=
github.com/fxamacker/cbor/v2 v2.9.0 h1:NpKPmjDBgUfBms6tr6JZkTHtfFGcMKsw3eGcmD/sapM=
github.com/fxamacker/cbor/v2 v2.9.0/go.mod h1:vM4b+DJCtHn+zz7h3FFp/hDAI9WNWCsZj23V5ytsSxQ=
github.com/gabriel-vasile/mimetype v1.4.13 h1:46nXokslUBsAJE/wMsp5gtO500a4F3Nkz9Ufpk2AcUM=
github.com/gabriel-vasile/mimetype v1.4.13/go.mod h1:d+9Oxyo1wTzWdyVUPMmXFvp4F9tea18J8ufA774AB3s=
github.com/go-playground/assert/v2 v2.2.0 h1:JvknZsQTYeFEAhQwI4qEt9cyV5ONwRHC+lYKSsYSR8s=
github.com/go-playground/assert/v2 v2.2.0/go.mod h1:VDjEfimB/XKnb+ZQfWdccd7VUvScMdVu0Titje2rxJ4=
github.com/go-playground/locales v0.14.1 h1:EWaQ/wswjilfKLTECiXz7Rh+3BjFhfDFKv/oXslEjJA=
github.com/go-playground/locales v0.14.1/go.mod h1:hxrqLVvrK65+Rwrd5Fc6F2O76J/NuW9t0sjnWqG1slY=
github.com/go-playground/universal-translator v0.18.1 h1:Bcnm0ZwsGyWbCzImXv+pAJnYK9S473LQFuzCbDbfSFY=
github.com/go-playground/universal-translator v0.18.1/go.mod h1:xekY+UJKNuX9WP91TpwSH2VMlDf28Uj24BCp08ZFTUY=
github.com/go-playground/validator/v10 v10.30.2 h1:JiFIMtSSHb2/XBUbWM4i/MpeQm9ZK2xqPNk8vgvu5JQ=
github.com/go-playground/validator/v10 v10.30.2/go.mod h1:mAf2pIOVXjTEBrwUMGKkCWKKPs9NheYGabeB04txQSc=
github.com/go-viper/mapstructure/v2 v2.4.0 h1:EBsztssimR/CONLSZZ04E8qAkxNYq4Qp9LvH92wZUgs=
github.com/go-viper/mapstructure/v2 v2.4.0/go.mod h1:oJDH3BJKyqBA2TXFhDsKDGDTlndYOZ6rGS0BRZIxGhM=
github.com/gofiber/fiber/v3 v3.1.0 h1:1p4I820pIa+FGxfwWuQZ5rAyX0WlGZbGT6Hnuxt6hKY=
@@ -37,6 +47,8 @@ github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE=
github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk=
github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY=
github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE=
github.com/leodido/go-urn v1.4.0 h1:WT9HwE9SGECu3lg4d/dIA+jxlljEa1/ffXKmRjqdmIQ=
github.com/leodido/go-urn v1.4.0/go.mod h1:bvxc+MVxLKB4z00jd1z+Dvzr47oO32F/QSNjSBOlFxI=
github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
@@ -45,6 +57,8 @@ github.com/pelletier/go-toml/v2 v2.2.4 h1:mye9XuhQ6gvn5h28+VilKrrPoQVanw5PMw/TB0
github.com/pelletier/go-toml/v2 v2.2.4/go.mod h1:2gIqNv+qfxSVS7cM2xJQKtLSTLUE9V8t9Stt+h56mCY=
github.com/philhofer/fwd v1.2.0 h1:e6DnBTl7vGY+Gz322/ASL4Gyp1FspeMvx1RNDoToZuM=
github.com/philhofer/fwd v1.2.0/go.mod h1:RqIHx9QI14HlwKwm98g9Re5prTQ6LdeRQn+gXJFxsJM=
github.com/pkg/errors v0.9.1 h1:FEBLx1zS214owpjy7qsBeixbURkuhQAwrK5UwLGTwt4=
github.com/pkg/errors v0.9.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0=
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/rogpeppe/go-internal v1.9.0 h1:73kH8U+JUqXU8lRuOHeVHaa/SZPifC7BkcraZVejAe8=
@@ -80,23 +94,25 @@ github.com/x448/float16 v0.8.4 h1:qLwI1I70+NjRFUR3zs1JPUCgaCXSh3SW62uAKT1mSBM=
github.com/x448/float16 v0.8.4/go.mod h1:14CWIYCyZA/cWjXOioeEpHeN/83MdbZDRQHoFcYsOfg=
github.com/xyproto/randomstring v1.0.5 h1:YtlWPoRdgMu3NZtP45drfy1GKoojuR7hmRcnhZqKjWU=
github.com/xyproto/randomstring v1.0.5/go.mod h1:rgmS5DeNXLivK7YprL0pY+lTuhNQW3iGxZ18UQApw/E=
go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto=
go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE=
go.uber.org/multierr v1.10.0 h1:S0h4aNzvfcFsC3dRF1jLoaov7oRaKqRGC/pUEJ2yvPQ=
go.uber.org/multierr v1.10.0/go.mod h1:20+QtiLqy0Nd6FdQB9TLXag12DsQkrbs3htMFfDN80Y=
go.uber.org/zap v1.27.1 h1:08RqriUEv8+ArZRYSTXy1LeBScaMpVSTBhCeaZYfMYc=
go.uber.org/zap v1.27.1/go.mod h1:GB2qFLM7cTU87MWRP2mPIjqfIDnGu+VIO4V/SdhGo2E=
go.yaml.in/yaml/v3 v3.0.4 h1:tfq32ie2Jv2UxXFdLJdh3jXuOzWiL1fo0bu/FbuKpbc=
go.yaml.in/yaml/v3 v3.0.4/go.mod h1:DhzuOOF2ATzADvBadXxruRBLzYTpT36CKvDb3+aBEFg=
golang.org/x/crypto v0.48.0 h1:/VRzVqiRSggnhY7gNRxPauEQ5Drw9haKdM0jqfcCFts=
golang.org/x/crypto v0.48.0/go.mod h1:r0kV5h3qnFPlQnBSrULhlsRfryS2pmewsg+XfMgkVos=
golang.org/x/net v0.50.0 h1:ucWh9eiCGyDR3vtzso0WMQinm2Dnt8cFMuQa9K33J60=
golang.org/x/net v0.50.0/go.mod h1:UgoSli3F/pBgdJBHCTc+tp3gmrU4XswgGRgtnwWTfyM=
golang.org/x/sync v0.19.0 h1:vV+1eWNmZ5geRlYjzm2adRgW2/mcpevXNg50YZtPCE4=
golang.org/x/sync v0.19.0/go.mod h1:9KTHXmSnoGruLpwFjVSX0lNNA75CykiMECbovNTZqGI=
golang.org/x/crypto v0.49.0 h1:+Ng2ULVvLHnJ/ZFEq4KdcDd/cfjrrjjNSXNzxg0Y4U4=
golang.org/x/crypto v0.49.0/go.mod h1:ErX4dUh2UM+CFYiXZRTcMpEcN8b/1gxEuv3nODoYtCA=
golang.org/x/net v0.51.0 h1:94R/GTO7mt3/4wIKpcR5gkGmRLOuE/2hNGeWq/GBIFo=
golang.org/x/net v0.51.0/go.mod h1:aamm+2QF5ogm02fjy5Bb7CQ0WMt1/WVM7FtyaTLlA9Y=
golang.org/x/sync v0.20.0 h1:e0PTpb7pjO8GAtTs2dQ6jYa5BWYlMuX047Dco/pItO4=
golang.org/x/sync v0.20.0/go.mod h1:9xrNwdLfx4jkKbNva9FpL6vEN7evnE43NNNJQ2LF3+0=
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
golang.org/x/sys v0.41.0 h1:Ivj+2Cp/ylzLiEU89QhWblYnOE9zerudt9Ftecq2C6k=
golang.org/x/sys v0.41.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
golang.org/x/text v0.34.0 h1:oL/Qq0Kdaqxa1KbNeMKwQq0reLCCaFtqu2eNuSeNHbk=
golang.org/x/text v0.34.0/go.mod h1:homfLqTYRFyVYemLBFl5GgL/DWEiH5wcsQ5gSh1yziA=
golang.org/x/sys v0.42.0 h1:omrd2nAlyT5ESRdCLYdm3+fMfNFE/+Rf4bDIQImRJeo=
golang.org/x/sys v0.42.0/go.mod h1:4GL1E5IUh+htKOUEOaiffhrAeqysfVGipDYzABqnCmw=
golang.org/x/text v0.35.0 h1:JOVx6vVDFokkpaq1AEptVzLTpDe9KGpj5tR4/X+ybL8=
golang.org/x/text v0.35.0/go.mod h1:khi/HExzZJ2pGnjenulevKNX1W67CUy0AsXcNubPGCA=
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c h1:Hei/4ADfdWqJk1ZMxUNpqntNwaWcugrBjAiHlqqRiVk=
gopkg.in/check.v1 v1.0.0-20201130134442-10cb98267c6c/go.mod h1:JHkPIbrfpd72SG/EVd6muEfDQjcINNoR0C8j2r3qZ4Q=
@@ -3,6 +3,7 @@ package bootstrap
import (
"allapp-go/internal/bootstrap/boots"
"allapp-go/internal/config"
"allapp-go/pkg/db"
"allapp-go/pkg/logger"
"github.com/jackc/pgx/v5/pgxpool"
@@ -32,6 +33,8 @@ func NewApp() (*App, error) {
return nil, err
}
db.Init(pgsql)
// 其他组件
boots.InitUniqueId(cfg)
boots.InitJwt(cfg)
@@ -2,29 +2,33 @@ package bootstrap
import (
"allapp-go/internal/config"
"allapp-go/internal/errors"
"allapp-go/internal/middleware"
"allapp-go/internal/router"
"allapp-go/pkg/logger"
"fmt"
"github.com/gofiber/fiber/v3"
"github.com/gofiber/fiber/v3/middleware/recover"
"github.com/gofiber/fiber/v3/middleware/requestid"
)
func RunServer(cfg *config.Config) error {
app := fiber.New()
app := fiber.New(fiber.Config{
ErrorHandler: errors.ErrorHandler(),
})
// ===== middleware =====
app.Use(recover.New())
app.Use(requestid.New())
app.Use(middleware.Logger())
// ===== router =====
router.Setup(app)
router.SetupRouter(app)
addr := fmt.Sprintf(":%d", cfg.App.Port)
logger.Log.Info("服务启动")
// ❗ 关键:直接阻塞
return app.Listen(addr)
}
@@ -0,0 +1,10 @@
package errors
import "github.com/pkg/errors"
func WithStack(err error) error {
if err == nil {
return nil
}
return errors.WithStack(err)
}
@@ -0,0 +1,31 @@
package errors
import (
"allapp-go/internal/httpx"
"allapp-go/pkg/logger"
"errors"
"fmt"
"github.com/gofiber/fiber/v3"
"go.uber.org/zap"
)
func ErrorHandler() fiber.ErrorHandler {
return func(c fiber.Ctx, err error) error {
code := fiber.StatusInternalServerError
msg := "系统异常,请稍后再试!"
var e *fiber.Error
if errors.As(err, &e) {
code = e.Code
msg = e.Message
}
logger.FromCtx(c.Context()).Error("HTTP 请求异常",
zap.Int("status", code),
zap.String("error", fmt.Sprintf("%+v", err)),
)
return httpx.SystemError(c, msg)
}
}
@@ -0,0 +1,50 @@
package handle
import (
"allapp-go/internal/errors"
"allapp-go/internal/httpx"
"allapp-go/internal/types"
"allapp-go/pkg/db"
"github.com/gofiber/fiber/v3"
)
func LoadData(c fiber.Ctx) error {
var req types.LoadDataReq
err := httpx.BindAndValidate(c, &req)
if err != nil {
return errors.WithStack(err)
}
dbClient := db.New()
// 查询
data, err := dbClient.LoadData(c.Context(), req.ViewName, req.SearchCondition, req.OrderBy, req.SearchColumns, req.Args)
if err != nil {
return errors.WithStack(err)
}
return httpx.OK(c, data)
}
func LoadDataBySql(c fiber.Ctx) error {
var req types.LoadDataBySqlReq
err := httpx.BindAndValidate(c, &req)
if err != nil {
return errors.WithStack(err)
}
dbClient := db.New()
data, err := dbClient.LoadDataBySQL(c.Context(), req.Sql, req.Args)
if err != nil {
return errors.WithStack(err)
}
return httpx.OK(c, data)
}
@@ -0,0 +1,56 @@
package handle
import (
"allapp-go/internal/httpx"
"allapp-go/internal/types"
"allapp-go/pkg/db"
"github.com/gofiber/fiber/v3"
)
func SaveData(c fiber.Ctx) error {
var req []types.SaveDataReq
if err := httpx.BindAndValidate(c, &req); err != nil {
return err
}
if len(req) == 0 {
return httpx.OK(c, nil)
}
ctx := c.Context()
dbClient := db.New()
err := dbClient.WithTx(ctx, func(tx *db.Client) error {
for _, dataReq := range req {
tableName := dataReq.TableName
keyField := dataReq.KeyField
for _, insert := range dataReq.Inserts {
if err := tx.Insert(ctx, tableName, keyField, insert); err != nil {
return err
}
}
for _, update := range dataReq.Updates {
if err := tx.Update(ctx, tableName, keyField, update); err != nil {
return err
}
}
for _, del := range dataReq.Deletes {
if err := tx.Delete(ctx, tableName, keyField, del); err != nil {
return err
}
}
}
return nil
})
if err != nil {
return err
}
return httpx.OK(c, nil)
}
@@ -0,0 +1,20 @@
package httpx
import (
"github.com/go-playground/validator/v10"
"github.com/gofiber/fiber/v3"
)
var validate = validator.New()
func BindAndValidate(c fiber.Ctx, v any) error {
if err := c.Bind().Body(v); err != nil {
return err
}
if err := validate.Struct(v); err != nil {
return err
}
return nil
}
@@ -0,0 +1,63 @@
package httpx
import "github.com/gofiber/fiber/v3"
type Result struct {
Code int `json:"code"`
Message string `json:"message"`
Data interface{} `json:"data"`
TrackId string `json:"track_id,omitempty"`
}
var BizCode = struct {
Success int // 成功
Unauthorized int // 未授权/Token失效
BusinessError int // 业务异常,例如参数错误、操作不允许
SystemError int // 系统异常,例如数据库崩溃、未知错误
}{
Success: 1000,
Unauthorized: 1001,
BusinessError: 2000,
SystemError: 3000,
}
// ------------------- 统一响应方法 -------------------
// Respond 返回通用响应,code和msg必填,data可选
func Respond(c fiber.Ctx, code int, msg string, data interface{}) error {
trackId := ""
if v := c.Context().Value("request_id"); v != nil {
if s, ok := v.(string); ok {
trackId = s
}
}
return c.JSON(Result{
Code: code,
Message: msg,
Data: data,
TrackId: trackId,
})
}
// ------------------- 便捷封装 -------------------
// OK 成功响应
func OK(c fiber.Ctx, data interface{}) error {
return Respond(c, BizCode.Success, "success", data)
}
// Fail 业务异常
func Fail(c fiber.Ctx, msg string) error {
return Respond(c, BizCode.BusinessError, msg, nil)
}
// SystemError 系统异常
func SystemError(c fiber.Ctx, msg string) error {
return Respond(c, BizCode.SystemError, msg, nil)
}
// Unauthorized 授权失败
func Unauthorized(c fiber.Ctx, msg string) error {
return Respond(c, BizCode.Unauthorized, msg, nil)
}
@@ -0,0 +1,47 @@
package middleware
import (
"allapp-go/internal/httpx"
"allapp-go/pkg/jwtx"
"strings"
"github.com/gofiber/fiber/v3"
)
const CtxUserIDKey = "user_id"
// Auth 登录校验中间件
func Auth() fiber.Handler {
return func(c fiber.Ctx) error {
token := strings.TrimSpace(c.Get("Token"))
if token == "" {
return httpx.Unauthorized(c, "未登录")
}
verifyToken, ok := jwtx.VerifyToken(c.Context(), token)
if !ok {
return httpx.Unauthorized(c, "登录已过期")
}
c.Locals(CtxUserIDKey, verifyToken.Data["user_id"])
return c.Next()
}
}
func GetUserID(c fiber.Ctx) (int64, error) {
v := c.Locals(CtxUserIDKey)
if v == nil {
return 0, httpx.Unauthorized(c, "登录已过期")
}
userID, ok := v.(int64)
if !ok {
return 0, httpx.Unauthorized(c, "登录已过期")
}
return userID, nil
}
@@ -1,22 +1,14 @@
package router
import (
"allapp-go/pkg/logger"
"allapp-go/internal/handle"
"github.com/gofiber/fiber/v3"
)
func Setup(app *fiber.App) {
// 健康检查
app.Get("/ping", func(c fiber.Ctx) error {
log := logger.FromCtx(c)
log.Info("测试")
return c.SendString("pong")
})
// 你后面可以这样分组:
// setupUserRoutes(app)
// setupOrderRoutes(app)
func SetupRouter(app *fiber.App) {
// ==================== data ====================
app.Post("/data/loadData", handle.LoadData)
app.Post("/data/loadDataBySql", handle.LoadDataBySql)
app.Post("/data/saveData", handle.SaveData)
}
@@ -0,0 +1,14 @@
package types
type LoadDataReq struct {
ViewName string `json:"view_name" validate:"required"`
SearchCondition string `json:"search_condition"`
OrderBy string `json:"order_by"`
SearchColumns []string `json:"search_columns"`
Args []any `json:"args"`
}
type LoadDataBySqlReq struct {
Sql string `json:"sql" validate:"required"`
Args []any `json:"args"`
}
@@ -0,0 +1,9 @@
package types
type SaveDataReq struct {
TableName string `json:"table_name"`
KeyField string `json:"key_field"`
Inserts []map[string]any `json:"inserts"`
Updates []map[string]any `json:"updates"`
Deletes []map[string]any `json:"deletes"`
}
@@ -62,3 +62,22 @@
{"level":"INFO","time":"2026-04-03T17:25:10.656+0800","msg":"服务启动"}
{"level":"INFO","time":"2026-04-03T17:25:12.876+0800","msg":"测试","trace_id":"HP6MsYAn3gEeurwlsBPImOgJHuEcwOcaPeTIUiFJnbA"}
{"level":"INFO","time":"2026-04-03T17:25:12.876+0800","msg":"HTTP 请求","trace_id":"HP6MsYAn3gEeurwlsBPImOgJHuEcwOcaPeTIUiFJnbA","method":"GET","path":"/ping","status":200,"ip":"127.0.0.1","duration":"31µs"}
{"level":"INFO","time":"2026-04-03T18:47:56.384+0800","msg":"服务启动"}
{"level":"INFO","time":"2026-04-03T23:10:20.276+0800","msg":"服务启动"}
{"level":"INFO","time":"2026-04-03T23:23:07.557+0800","msg":"服务启动"}
{"level":"INFO","time":"2026-04-03T23:45:48.868+0800","msg":"服务启动"}
{"level":"INFO","time":"2026-04-03T23:45:57.051+0800","msg":"测试"}
{"level":"INFO","time":"2026-04-03T23:45:57.130+0800","msg":"HTTP 请求","trace_id":"3eBTZ1s6zqCSVoAmfMYeClE68qEY_DMSdCWj4J7nd1U","method":"GET","path":"/ping","status":200,"ip":"127.0.0.1","duration":"79.3204ms"}
{"level":"INFO","time":"2026-04-03T23:45:57.278+0800","msg":"HTTP 请求","trace_id":"1dMP6oPPdH0gq-BUf62cYxRY7ftD-6320hMKTDIqDdw","method":"GET","path":"/favicon.ico","status":200,"ip":"127.0.0.1","duration":"0s"}
{"level":"INFO","time":"2026-04-03T23:46:02.757+0800","msg":"测试"}
{"level":"INFO","time":"2026-04-03T23:46:02.863+0800","msg":"HTTP 请求","trace_id":"LSzqK436_1KC_65lEuHBJk1IF4zYu4K5bFiUOkGdt0o","method":"GET","path":"/ping","status":200,"ip":"127.0.0.1","duration":"106.201ms"}
{"level":"INFO","time":"2026-04-03T23:50:09.300+0800","msg":"服务启动"}
{"level":"INFO","time":"2026-04-03T23:50:13.016+0800","msg":"测试"}
{"level":"INFO","time":"2026-04-03T23:50:13.087+0800","msg":"HTTP 请求","trace_id":"HkojD0HcU_UTgoM4fxTfzLjd9XfOfVLq8lnHo0uj-TM","method":"GET","path":"/ping","status":200,"ip":"127.0.0.1","duration":"70.5617ms"}
{"level":"INFO","time":"2026-04-03T23:50:33.781+0800","msg":"服务启动"}
{"level":"INFO","time":"2026-04-03T23:50:49.853+0800","msg":"服务启动"}
{"level":"INFO","time":"2026-04-03T23:50:51.495+0800","msg":"测试"}
{"level":"INFO","time":"2026-04-03T23:50:51.564+0800","msg":"HTTP 请求","trace_id":"k2PZJ9uhIck2jbS32ObSU2O_enz-VpPHjL0ibvv8GWs","method":"GET","path":"/ping","status":200,"ip":"127.0.0.1","duration":"69.1125ms"}
{"level":"INFO","time":"2026-04-03T23:51:10.476+0800","msg":"服务启动"}
{"level":"INFO","time":"2026-04-03T23:51:12.453+0800","msg":"测试"}
{"level":"INFO","time":"2026-04-03T23:51:12.537+0800","msg":"HTTP 请求","trace_id":"ejaNSeN3XTsuYU3jpKO1Q3Ayq5RjdNaJTc0iRWbTfn8","method":"GET","path":"/ping","status":200,"ip":"127.0.0.1","duration":"84.1231ms"}
+52
View File
@@ -0,0 +1,52 @@
package db
import (
"fmt"
"strconv"
"strings"
)
func validateData(table string, keyField string, data map[string]any) (any, error) {
if table == "" {
return nil, fmt.Errorf("表名不能为空")
}
if len(data) == 0 {
return nil, fmt.Errorf("数据不能为空")
}
if keyField == "" {
return nil, fmt.Errorf("主键字段不能为空")
}
val, ok := data[keyField]
if !ok {
return nil, fmt.Errorf("缺少主键字段: %s", keyField)
}
if val == nil {
return nil, fmt.Errorf("主键字段值不能为空")
}
return val, nil
}
// 高效 placeholder 转换
func convertPlaceholder(sql string) string {
var sb strings.Builder
sb.Grow(len(sql))
argIndex := 1
for i := 0; i < len(sql); i++ {
if sql[i] == '?' {
sb.WriteByte('$')
sb.WriteString(strconv.Itoa(argIndex))
argIndex++
} else {
sb.WriteByte(sql[i])
}
}
return sb.String()
}
+42
View File
@@ -0,0 +1,42 @@
package db
import (
"context"
"github.com/jackc/pgx/v5"
"github.com/jackc/pgx/v5/pgconn"
"github.com/jackc/pgx/v5/pgxpool"
)
var defaultClient *Client
type Executor interface {
Exec(ctx context.Context, sql string, args ...any) (pgconn.CommandTag, error)
Query(ctx context.Context, sql string, args ...any) (pgx.Rows, error)
QueryRow(ctx context.Context, sql string, args ...any) pgx.Row
}
type Client struct {
pool *pgxpool.Pool
tx pgx.Tx
}
// Init 初始化
func Init(pool *pgxpool.Pool) {
defaultClient = &Client{pool: pool}
}
// New 创建会话
func New() *Client {
return &Client{
pool: defaultClient.pool,
}
}
// 内部获取执行器(关键)
func (c *Client) exec() Executor {
if c.tx != nil {
return c.tx
}
return c.pool
}
+83
View File
@@ -0,0 +1,83 @@
package db
import (
"context"
"fmt"
"strings"
)
func (c *Client) Delete(
ctx context.Context,
table string,
keyField string,
data map[string]any,
) error {
if keyField == "" {
return fmt.Errorf("keyField 不能为空")
}
keyVal, err := validateData(table, keyField, data)
if err != nil {
return err
}
sql := fmt.Sprintf(
"DELETE FROM %s WHERE %s = $1",
table,
keyField,
)
_, err = c.exec().Exec(ctx, sql, keyVal)
return err
}
func (c *Client) BatchDelete(
ctx context.Context,
table string,
keyField string,
dataList []map[string]any,
) error {
if table == "" {
return fmt.Errorf("表名不能为空")
}
if keyField == "" {
return fmt.Errorf("主键字段不能为空")
}
if len(dataList) == 0 {
return fmt.Errorf("数据不能为空")
}
var (
placeholders []string
args []any
argIndex = 1
)
for _, data := range dataList {
keyVal, err := validateData(table, keyField, data)
if err != nil {
return err
}
if keyVal == nil {
return fmt.Errorf("主键字段值不能为空")
}
placeholders = append(placeholders, fmt.Sprintf("$%d", argIndex))
args = append(args, keyVal)
argIndex++
}
sql := fmt.Sprintf(
"DELETE FROM %s WHERE %s IN (%s)",
table,
keyField,
strings.Join(placeholders, ", "),
)
_, err := c.exec().Exec(ctx, sql, args...)
return err
}
+120
View File
@@ -0,0 +1,120 @@
package db
import (
"context"
"fmt"
"sort"
"strings"
)
// 构建 INSERT SQL(单条)
func buildInsertSQL(table string, data map[string]any) (string, []any) {
var (
columns []string
placeholders []string
args []any
)
i := 1
for col, val := range data {
columns = append(columns, col)
placeholders = append(placeholders, fmt.Sprintf("$%d", i))
args = append(args, val)
i++
}
sql := fmt.Sprintf(
"INSERT INTO %s (%s) VALUES (%s)",
table,
strings.Join(columns, ", "),
strings.Join(placeholders, ", "),
)
return sql, args
}
func (c *Client) Insert(
ctx context.Context,
table string,
keyField string,
data map[string]any,
) error {
if _, err := validateData(table, keyField, data); err != nil {
return err
}
sql, args := buildInsertSQL(table, data)
_, err := c.exec().Exec(ctx, sql, args...)
return err
}
func (c *Client) BatchInsert(
ctx context.Context,
table string,
keyField string,
dataList []map[string]any,
) error {
if len(dataList) == 0 {
return fmt.Errorf("数据不能为空")
}
// 用第一条确定列顺序
first := dataList[0]
if _, err := validateData(table, keyField, first); err != nil {
return err
}
var columns []string
for col := range first {
columns = append(columns, col)
}
sort.Strings(columns)
var (
valueStrings []string
args []any
argIndex = 1
)
for _, data := range dataList {
// 统一校验
if _, err := validateData(table, keyField, data); err != nil {
return err
}
// 字段数量检查
if len(data) != len(columns) {
return fmt.Errorf("批量插入失败:数据字段不一致")
}
var placeholders []string
for _, col := range columns {
val, ok := data[col]
if !ok {
return fmt.Errorf("缺少字段: %s", col)
}
placeholders = append(placeholders, fmt.Sprintf("$%d", argIndex))
args = append(args, val)
argIndex++
}
valueStrings = append(
valueStrings,
fmt.Sprintf("(%s)", strings.Join(placeholders, ", ")),
)
}
sql := fmt.Sprintf(
"INSERT INTO %s (%s) VALUES %s",
table,
strings.Join(columns, ", "),
strings.Join(valueStrings, ", "),
)
_, err := c.exec().Exec(ctx, sql, args...)
return err
}
+82
View File
@@ -0,0 +1,82 @@
package db
import (
"context"
"fmt"
"strings"
)
func (c *Client) LoadData(
ctx context.Context,
viewName string,
searchCondition string,
orderBy string,
searchColumns []string,
args []any,
) ([][]any, error) {
var where string
// WHERE 构造
if searchCondition != "" && len(searchColumns) > 0 {
var conditions []string
for _, col := range searchColumns {
conditions = append(conditions, fmt.Sprintf("%s LIKE ?", col))
}
where = " WHERE (" + strings.Join(conditions, " OR ") + ")"
// 自动加 %
for i := range args {
if s, ok := args[i].(string); ok {
args[i] = "%" + s + "%"
}
}
}
// ORDER BY
var order string
if orderBy != "" {
order = " ORDER BY " + orderBy
}
sql := fmt.Sprintf("SELECT * FROM %s%s%s", viewName, where, order)
return c.query(ctx, sql, args...)
}
func (c *Client) LoadDataBySQL(
ctx context.Context,
sql string,
args []any,
) ([][]any, error) {
return c.query(ctx, sql, args...)
}
// 统一查询方法(核心优化)
func (c *Client) query(ctx context.Context, sql string, args ...any) ([][]any, error) {
sql = convertPlaceholder(sql)
rows, err := c.exec().Query(ctx, sql, args...)
if err != nil {
return nil, err
}
defer rows.Close()
result := make([][]any, 0, 16)
for rows.Next() {
values, err := rows.Values()
if err != nil {
return nil, err
}
row := make([]any, len(values))
copy(row, values)
result = append(result, row)
}
return result, rows.Err()
}
+34
View File
@@ -0,0 +1,34 @@
package db
import (
"context"
"github.com/jackc/pgx/v5"
)
// WithTx 开启事务(在当前 client 上)
func (c *Client) WithTx(ctx context.Context, fn func(tx *Client) error) error {
tx, err := c.pool.Begin(ctx)
if err != nil {
return err
}
// 创建事务 client
txClient := &Client{
pool: c.pool,
tx: tx,
}
defer func(tx pgx.Tx, ctx context.Context) {
err := tx.Rollback(ctx)
if err != nil {
}
}(tx, ctx)
if err := fn(txClient); err != nil {
return err
}
return tx.Commit(ctx)
}
+140
View File
@@ -0,0 +1,140 @@
package db
import (
"context"
"fmt"
"sort"
"strings"
)
func (c *Client) Update(
ctx context.Context,
table string,
keyField string,
data map[string]any,
) error {
keyVal, err := validateData(table, keyField, data)
if err != nil {
return err
}
var (
setClauses []string
args []any
i = 1
)
for col, val := range data {
if col == keyField {
continue
}
setClauses = append(setClauses, fmt.Sprintf("%s=$%d", col, i))
args = append(args, val)
i++
}
// WHERE 条件
where := fmt.Sprintf("%s=$%d", keyField, i)
args = append(args, keyVal)
sql := fmt.Sprintf(
"UPDATE %s SET %s WHERE %s",
table,
strings.Join(setClauses, ", "),
where,
)
_, err = c.exec().Exec(ctx, sql, args...)
return err
}
func (c *Client) BatchUpdate(
ctx context.Context,
table string,
keyField string,
dataList []map[string]any,
) error {
if len(dataList) == 0 {
return fmt.Errorf("数据不能为空")
}
// 用第一条数据确定字段
first := dataList[0]
if _, err := validateData(table, keyField, first); err != nil {
return err
}
// 提取字段(排除主键)+ 排序(关键)
var columns []string
for col := range first {
if col != keyField {
columns = append(columns, col)
}
}
sort.Strings(columns)
var (
args []any
argIndex = 1
)
// CASE 语句
var setClauses []string
for _, col := range columns {
var caseBuilder strings.Builder
caseBuilder.WriteString(fmt.Sprintf("%s = CASE %s ", col, keyField))
for _, data := range dataList {
// 校验 key
keyVal, err := validateData(table, keyField, data)
if err != nil {
return err
}
val, ok := data[col]
if !ok {
return fmt.Errorf("缺少字段: %s", col)
}
caseBuilder.WriteString(fmt.Sprintf(
"WHEN $%d THEN $%d ",
argIndex,
argIndex+1,
))
args = append(args, keyVal, val)
argIndex += 2
}
caseBuilder.WriteString("END")
setClauses = append(setClauses, caseBuilder.String())
}
// WHERE IN
var wherePlaceholders []string
for _, data := range dataList {
keyVal, ok := data[keyField]
if !ok || keyVal == nil {
return fmt.Errorf("缺少主键字段值: %s", keyField)
}
wherePlaceholders = append(wherePlaceholders, fmt.Sprintf("$%d", argIndex))
args = append(args, keyVal)
argIndex++
}
sql := fmt.Sprintf(
"UPDATE %s SET %s WHERE %s IN (%s)",
table,
strings.Join(setClauses, ", "),
keyField,
strings.Join(wherePlaceholders, ", "),
)
_, err := c.exec().Exec(ctx, sql, args...)
return err
}
+48 -16
View File
@@ -45,52 +45,86 @@ func Init(cfg Config) {
level := parseLevel(cfg.Level)
// ======================
// encoder 配置
// ======================
encoderConfig := zapcore.EncoderConfig{
TimeKey: "time",
LevelKey: "level",
MessageKey: "msg",
CallerKey: "caller",
EncodeLevel: zapcore.CapitalLevelEncoder,
EncodeTime: zapcore.ISO8601TimeEncoder,
EncodeCaller: zapcore.ShortCallerEncoder,
}
encoder := zapcore.NewJSONEncoder(encoderConfig)
// 👉 文件 JSON encoder
jsonEncoder := zapcore.NewJSONEncoder(func() zapcore.EncoderConfig {
cfg := encoderConfig
cfg.EncodeLevel = zapcore.CapitalLevelEncoder
cfg.EncodeTime = zapcore.TimeEncoderOfLayout("2006-01-02 15:04:05")
return cfg
}())
// 👉 控制台 encoder(可读)
consoleEncoder := zapcore.NewConsoleEncoder(func() zapcore.EncoderConfig {
cfg := encoderConfig
cfg.EncodeLevel = zapcore.CapitalColorLevelEncoder
cfg.EncodeTime = zapcore.TimeEncoderOfLayout("2006-01-02 15:04:05")
return cfg
}())
// ======================
// writer
// ======================
// 控制台
consoleWriter := zapcore.AddSync(os.Stdout)
// 文件 writer(按天)
infoWriter := getWriter(cfg, "info")
errorWriter := getWriter(cfg, "error")
// info core
// ======================
// core
// ======================
// 控制台(所有级别)
consoleCore := zapcore.NewCore(
consoleEncoder,
consoleWriter,
level,
)
// info 文件
infoCore := zapcore.NewCore(
encoder,
zapcore.NewMultiWriteSyncer(consoleWriter, infoWriter),
jsonEncoder,
infoWriter,
zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
return lvl < zapcore.ErrorLevel && lvl >= level
}),
)
// error core
// error 文件
errorCore := zapcore.NewCore(
encoder,
zapcore.NewMultiWriteSyncer(consoleWriter, errorWriter),
jsonEncoder,
errorWriter,
zap.LevelEnablerFunc(func(lvl zapcore.Level) bool {
return lvl >= zapcore.ErrorLevel
}),
)
core := zapcore.NewTee(infoCore, errorCore)
core := zapcore.NewTee(
consoleCore,
infoCore,
errorCore,
)
Log = zap.New(
core,
zap.AddCaller(), // 打印调用位置
zap.AddCallerSkip(1), // 跳过封装层
)
}
// ======================
// Daily Writer(核心)
// Daily Writer(按天切割)
// ======================
type DailyWriter struct {
@@ -110,7 +144,6 @@ func newDailyWriter(cfg Config, level string) *DailyWriter {
func (w *DailyWriter) Write(p []byte) (n int, err error) {
today := time.Now().Format("2006-01-02")
// 跨天自动切换
if w.log == nil || w.date != today {
dir := filepath.Join(w.cfg.Dir, w.level)
_ = os.MkdirAll(dir, os.ModePerm)
@@ -172,7 +205,6 @@ func SetTraceID(c fiber.Ctx, traceID string) {
return
}
// context
ctx := context.WithValue(c.Context(), TraceIDKey, traceID)
c.SetContext(ctx)
}