1

common/base_dao.go

@@ -0,0 +1,53 @@
+package common
+
+import (
+	"context"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/gtime"
+	"github.com/gogf/gf/v2/util/gconv"
+)
+
+func prepareInsertData(data any) map[string]any {
+	m := gconv.Map(data, gconv.MapOption{Tags: []string{"orm"}})
+	delete(m, "id")
+	m["created_at"] = gtime.Now().Format("Y-m-d H:i:s")
+	m["updated_at"] = gtime.Now().Format("Y-m-d H:i:s")
+	delete(m, "deleted_at")
+	return m
+}
+
+func InsertAndReturnId(ctx context.Context, table string, data any) (id int64, err error) {
+	m := prepareInsertData(data)
+	r, err := g.DB().Model(table).Ctx(ctx).Data(m).Insert()
+	if err != nil {
+		return 0, err
+	}
+	if r == nil {
+		return 0, nil
+	}
+	return r.LastInsertId()
+}
+
+func GetOneByPk[T any](ctx context.Context, table string, pk int64) (res *T, err error) {
+	r, err := g.DB().Model(table).Ctx(ctx).
+		Where("id", pk).One()
+	if err != nil {
+		return nil, err
+	}
+	if r == nil {
+		return nil, nil
+	}
+	err = r.Struct(&res)
+	return
+}
+
+func UpdateByPk(ctx context.Context, table string, pk int64, data any) error {
+	_, err := g.DB().Model(table).Ctx(ctx).Data(data).Where("id", pk).Update()
+	return err
+}
+
+func DeleteByPk(ctx context.Context, table string, pk int64) error {
+	_, err := g.DB().Model(table).Ctx(ctx).Unscoped().Where("id", pk).Delete()
+	return err
+}

common/http.go

@@ -0,0 +1,72 @@
+package common
+
+import (
+	"fmt"
+	"os"
+	"path/filepath"
+	"reflect"
+	"regexp"
+	"strings"
+
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/net/ghttp"
+	"github.com/gogf/gf/v2/os/gtime"
+)
+
+var Httpserver = g.Server()
+
+func init() {
+	err := gtime.SetTimeZone("Asia/Shanghai")
+	if err != nil {
+		panic("设置时区失败")
+	}
+	Httpserver.SetOpenApiPath("/api.json")
+	Httpserver.BindMiddlewareDefault(ghttp.MiddlewareHandlerResponse)
+	Httpserver.BindMiddlewareDefault(func(r *ghttp.Request) {
+		r.Response.CORS(r.Response.DefaultCORSOptions())
+		r.Middleware.Next()
+	})
+}
+
+// RouteRegister 根据控制器结构体名称自动注册路由
+func RouteRegister(controllers []interface{}) {
+	re := regexp.MustCompile("[A-Z]")
+	for _, t := range controllers {
+		sName := reflect.ValueOf(t).Elem().Type().Name()
+		convertedStr := re.ReplaceAllStringFunc(sName, func(s string) string {
+			return fmt.Sprintf("-%s", strings.ToLower(s))
+		})
+		if len(convertedStr) > 0 && convertedStr[0] == '-' {
+			convertedStr = convertedStr[1:]
+		}
+		Httpserver.Group("/"+convertedStr, func(group *ghttp.RouterGroup) {
+			group.Bind(t)
+		})
+	}
+}
+
+// ServeFrontend 配置前端静态文件服务
+func ServeFrontend(distDir string) {
+	absDir, err := filepath.Abs(distDir)
+	if err != nil {
+		g.Log().Fatalf(nil, "前端目录路径无效: %v", err)
+	}
+
+	// 静态资源（JS/CSS/图片）
+	Httpserver.AddStaticPath("/assets", absDir+"/assets")
+
+	// SPA 兜底：非API路由返回 index.html
+	Httpserver.BindHandler("/*", func(r *ghttp.Request) {
+		indexFile := filepath.Join(absDir, "index.html")
+		if fileExists(indexFile) {
+			r.Response.ServeFile(indexFile)
+		} else {
+			r.Response.WriteStatus(404)
+		}
+	})
+}
+
+func fileExists(path string) bool {
+	_, err := os.Stat(path)
+	return err == nil
+}