root/huangguo_server
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Initial commit

Browse Source 
Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
This commit is contained in:
root
2026-09-15 13:57:10 +08:00
co-authored by Claude Opus 5
commit 8679200f41
1897 changed files with 257900 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
generate/common/generate.go
+364
View File
@@ -0,0 +1,364 @@
package common
import (
"errors"
"fmt"
"os"
"path"
"path/filepath"
"reflect"
"strings"
"91porn-server/generate/tpl"
)
func NewGen(moduleName, tblComment string) *G {
rootPath := RootPath()
return &G{
AppGenerate: true,
WebGenerate: true,
RootPath: rootPath,
ModuleName: moduleName,
TableComment: tblComment,
}
}
type Field struct {
Name string
TypeName string
Tag string
BsonField string
FormTag string
UpdateTag string
Comment string
SubField []Field
SubStructName string
}
type G struct {
AppGenerate bool
WebGenerate bool
RootPath string
ModuleName string
StructName string
FileName string
TableName string
TableComment string
Fields []Field
}
// Generate 生成...
func (o *G) Generate(s interface{}) {
var err error
// 获取model结构
t := reflect.TypeOf(s)
o.Fields, err = o.explodeStruct(t)
if err != nil {
log(err.Error(), "error")
return
}
o.StructName = FirstUpper(t.Name())
if o.TableName == "" {
o.TableName = o.StructName
}
o.FileName = strings.ToLower(o.TableName)
// 生成变量表
v := make(map[string]string)
v["FileName"] = strings.ToLower(o.TableName)
v["ModuleName"] = o.ModuleName
v["TableComment"] = o.TableComment
v["TableName"] = o.TableName
v["StructName"] = o.StructName
v["PackageTableName"] = strings.ToLower(o.TableName)
v["RouterName"] = FirstLower(o.TableName)
v["StructDefine"] = o.genStructDefine(o.StructName, o.Fields)
v["AppRespStructFields"] = o.genAppRespStructFields(o.Fields)
v["AppRespTransferStatement"] = o.genAppRespTransferStatement(o.Fields)
v["CreateReqFields"] = o.genCreateStructFields(o.Fields)
v["CreateStatement"] = o.genCreateStatement(o.Fields)
v["UpdateReqFields"] = o.genUpdateStructFields(o.Fields)
v["UpdateStatement"] = o.genUpdateStatement(o.Fields)
// 循环配置进行生成
for tpDir, c := range tpl.Config {
// 判断是否生成 app web
if !o.AppGenerate && strings.HasPrefix(tpDir, "/app") {
continue
}
if !o.WebGenerate && strings.HasPrefix(tpDir, "/web") {
continue
}
// 遍历目录
tplPath := path.Join(o.RootPath, "/generate/tpl", tpDir)
var tpls []string
err := filepath.Walk(tplPath, func(path string, info os.FileInfo, err error) error {
if !info.IsDir() && strings.HasSuffix(info.Name(), ".tpl") {
tpls = append(tpls, path)
}
return nil
})
if err != nil {
log(err.Error(), "error")
return
}
// 创建文件
for _, tf := range tpls {
err = o.CreateFile(tpDir, tf, c, v)
if err != nil {
log(err.Error(), "error")
}
}
}
// 执行完成
log("生成完毕！", "info")
}
// CreateFile 根据模板创建文件
func (o *G) CreateFile(dir, tf string, conf tpl.Tpl, val map[string]string) (err error) {
val["VariablePrefix"] = FirstUpper(conf.VariablePrefix)
_, f := path.Split(tf)
f = strings.Replace(TplReplace(f, val), ".tpl", ".go", -1)
fPath := path.Join(o.RootPath, dir)
if conf.SplitDir {
fPath = path.Join(fPath, o.FileName+conf.DirSuffix)
}
// 创建目录
err = EnsurePath(fPath)
if err != nil {
return err
}
fPath = path.Join(fPath, f)
// 检查文件是否存在，存在则跳过
//_, err = os.Stat(fPath)
//if err == nil {
// log("目标文件已存在，不可覆盖！File:"+fPath, "warn")
// return nil
//}
log("准备写入目标文件："+fPath, "info")
// 读取文件并替换内容
b, err := os.ReadFile(tf)
if err != nil {
return err
}
tplContent := string(b)
tplContent = TplReplace(tplContent, val)
// 写入目标文件
if err := os.WriteFile(fPath, []byte(tplContent), 0755); err != nil {
log("写入目标文件："+fPath+"失败！", "error")
return err
}
log("写入目标文件："+fPath+"成功！", "error")
return nil
}
// genStructDefine 生成结构体定义语句
func (o *G) genStructDefine(structName string, fields []Field) string {
var s []string
sub := make(map[string][]Field)
for _, f := range fields {
tmp := fmt.Sprintf(" %s\t%s\t`%s`\t// %s", f.Name, f.TypeName, f.Tag, f.Comment)
s = append(s, tmp)
if len(f.SubField) > 0 {
sub[f.SubStructName] = f.SubField
}
}
ms := fmt.Sprintf("type %s struct {\n%s\n} \n\n", structName, strings.Join(s, "\n"))
if len(sub) == 0 {
return ms
}
var subs []string
for n, v := range sub {
subs = append(subs, o.genStructDefine(n, v))
}
ss := strings.Join(subs, "\n\n")
return ms + ss
}
// genAppRespStructFields 生成app返回结构
func (o *G) genAppRespStructFields(fields []Field) string {
var s []string
for _, f := range fields {
tmp := fmt.Sprintf(" %s\t%s\t`%s`\t// %s", f.Name, f.TypeName, f.UpdateTag, f.Comment)
s = append(s, tmp)
}
return strings.Join(s, "\n")
}
// genAppRespTransferStatement 生成app返回转换过程代码
func (o *G) genAppRespTransferStatement(fields []Field) string {
var s []string
for _, f := range fields {
tmp := fmt.Sprintf(" %s: item.%s,", f.Name, f.Name)
s = append(s, tmp)
}
str := strings.Join(s, "\n")
return str
}
// genCreateStructFields 生成创建数据字段定义
func (o *G) genCreateStructFields(fields []Field) string {
var s []string
for _, f := range fields {
fn := strings.ToLower(f.Name)
if fn == "id" || fn == "createdat" || fn == "updatedat" {
continue
}
tmp := fmt.Sprintf(" %s\t%s\t`%s`\t// %s", f.Name, f.TypeName, f.FormTag, f.Comment)
s = append(s, tmp)
}
return strings.Join(s, "\n")
}
// genCreateStatement 生成创建过程代码
func (o *G) genCreateStatement(fields []Field) string {
var s []string
hasCreateTime := false
hasUpdateTime := false
for _, f := range fields {
fn := strings.ToLower(f.Name)
if fn == "id" {
continue
}
if fn == "createdat" {
hasCreateTime = true
continue
}
if fn == "updatedat" {
hasUpdateTime = true
continue
}
tmp := fmt.Sprintf(" %s: p.%s,", f.Name, f.Name)
s = append(s, tmp)
}
str := strings.Join(s, "\n")
if hasCreateTime {
str += "\n CreatedAt: time.Now(),"
}
if hasUpdateTime {
str += "\n UpdatedAt: time.Now(),"
}
return str
}
// genUpdateStructFields 生成更新数据字段定义
func (o *G) genUpdateStructFields(fields []Field) string {
var s []string
for _, f := range fields {
fn := strings.ToLower(f.Name)
if fn == "id" || fn == "createdat" || fn == "updatedat" {
continue
}
tmp := fmt.Sprintf(" %s\t*%s\t`%s`\t// %s", f.Name, f.TypeName, f.UpdateTag, f.Comment)
s = append(s, tmp)
}
return strings.Join(s, "\n")
}
// genUpdateStatement 生成更新过程代码
func (o *G) genUpdateStatement(fields []Field) string {
var s []string
hasUpdateTime := false
for _, f := range fields {
fn := strings.ToLower(f.Name)
if fn == "id" || fn == "createdat" {
continue
}
if fn == "updatedat" {
hasUpdateTime = true
continue
}
tmp := fmt.Sprintf(" if p.%s != nil {\n\t\tdata[\"%s\"] = *p.%s\n\t}", f.Name, f.BsonField, f.Name)
s = append(s, tmp)
}
str := strings.Join(s, "\n")
if hasUpdateTime {
str += "\n\tdata[\"updatedAt\"] = time.Now()\n"
}
return str
}
// SetTableName 自定义表名
func (o *G) SetTableName(n string) {
o.TableName = FirstUpper(n)
}
// NoApp 不生成app
func (o *G) NoApp() {
o.AppGenerate = false
}
// NoWeb 不生成web
func (o *G) NoWeb() {
o.WebGenerate = false
}
// explodeStruct 解剖struct
func (o *G) explodeStruct(t reflect.Type) (f []Field, err error) {
if t.Kind() != reflect.Struct {
return nil, errors.New("不支持结构体以外的类型！")
}
// 遍历结构体的字段
for i := 0; i < t.NumField(); i++ {
field := t.Field(i)
jsonTag := field.Tag.Get("json")
if jsonTag == "" {
jsonTag = "-"
}
bsonTag := field.Tag.Get("bson")
if bsonTag == "" {
bsonTag = "-"
}
tag := fmt.Sprintf("json:\"%s\" bson:\"%s\"", jsonTag, bsonTag)
fTag := fmt.Sprintf("json:\"%s\" form:\"%s\" binding:\"required\"", jsonTag, jsonTag)
uTag := fmt.Sprintf("json:\"%s\"", jsonTag)
isDefinedType := strings.Count(field.Type.String(), "main.") > 0
typeName := strings.Replace(field.Type.String(), "main.", "", 1)
tmp := Field{
Name: field.Name,
TypeName: typeName,
Tag: tag,
BsonField: bsonTag,
FormTag: fTag,
UpdateTag: uTag,
Comment: field.Tag.Get("comment"),
}
if isDefinedType {
if field.Type.Kind() == reflect.Slice || field.Type.Kind() == reflect.Pointer {
tmp.SubField, _ = o.explodeStruct(field.Type.Elem())
if field.Type.Kind() == reflect.Slice {
tmp.SubStructName = strings.TrimLeft(typeName, "[]")
} else {
tmp.SubStructName = strings.TrimLeft(typeName, "*")
}
} else if field.Type.Kind() == reflect.Struct {
tmp.SubField, _ = o.explodeStruct(field.Type)
tmp.SubStructName = typeName
}
}
f = append(f, tmp)
}
return
}
generate/common/helper.go
+52
View File
@@ -0,0 +1,52 @@
package common
import (
"fmt"
"os"
"path/filepath"
"strings"
)
// EnsurePath 创建目标目录
func EnsurePath(p string) error {
s, err := os.Stat(p)
if err != nil || !s.IsDir() {
return os.MkdirAll(p, 0755)
}
return nil
}
// TplReplace 模板变量替换
func TplReplace(content string, val map[string]string) string {
for k, v := range val {
content = strings.ReplaceAll(content, "{{"+k+"}}", v)
}
return content
}
func log(msg, level string) {
fmt.Printf("[%s] %s \n", level, msg)
}
// FirstUpper 字符串首字母大写
func FirstUpper(s string) string {
if s == "" {
return ""
}
return strings.ToUpper(s[:1]) + s[1:]
}
// FirstLower 字符串首字母小写
func FirstLower(s string) string {
if s == "" {
return ""
}
return strings.ToLower(s[:1]) + s[1:]
}
func RootPath() string {
p, _ := filepath.Abs("./")
return p
}
generate/model/aiplaza/main.go
+32
View File
@@ -0,0 +1,32 @@
package main
import (
"91porn-server/generate/common"
"go.mongodb.org/mongo-driver/bson/primitive"
"time"
)
type AiPlaza struct {
ID primitive.ObjectID `bson:"_id,omitempty" json:"id" comment:"文档id"`
Type int `json:"type" bson:"type"` // 1-ai图片换脸 2-ai视频换脸 3-ai脱衣 4-ai图生视频 5-ai绘画
ModuleId string `json:"moduleId" bson:"moduleId" comment:"关联模版id"` // 关联模版id
Uid uint64 `json:"uid" bson:"uid"`
Title string `json:"title" bson:"title" comment:"标题"`
OriginalImage string `json:"originalImage" bson:"originalImage" comment:"原图"` // 原图
OriginalVideo string `json:"originalVideo" bson:"originalVideo" comment:"原视频"` // 原视频
OriginContent string `json:"originContent" bson:"originContent" comment:"输入的原文"` // 输入的原文
GenerateImage string `json:"generateImage" bson:"generateImage" comment:"生成的图"` // 生成的图
GenerateVideo string `json:"generateVideo" bson:"generateVideo" comment:"生成的视频"` // 生成的视频
Status int `json:"status" bson:"status" comment:"0-待审核 1-审核通过 2-拒绝 3-下架"` // 0-待审核 1-审核通过 2-拒绝 3-下架
Reason string `json:"reason" bson:"reason" comment:"原因"`
SortCode int `json:"sortCode" bson:"sortCode" comment:"置顶排序号，大于0就是置顶"`
ReviewAt time.Time `json:"reviewAt" bson:"reviewAt" comment:"审核时间"` // 审核时间
CreatedAt time.Time `json:"createdAt" bson:"createdAt" comment:"文档创建时间"` // 文档创建时间
UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt" comment:"文档更新时间"` // 文档更新时间
}
func main() {
g := common.NewGen("91porn-server", "ai广场帖子")
g.AppGenerate = true
g.Generate(AiPlaza{})
}
generate/model/hotsearch/main.go
+33
View File
@@ -0,0 +1,33 @@
package main
import (
"time"
"91porn-server/generate/common"
"go.mongodb.org/mongo-driver/bson/primitive"
)
type MediaBookshelf struct {
ID primitive.ObjectID `json:"id" bson:"_id,omitempty" comment:"文档id"`
Type int `json:"type" bson:"type" comment:"类型"`
Name string `json:"name" bson:"name" comment:"名称"`
MID primitive.ObjectID `json:"mid" bson:"mid" comment:"媒体ID"`
ReadHistory ReadHistory `json:"readHistory" bson:"readHistory" comment:"阅读记录"`
UID uint64 `json:"uid" bson:"uid" comment:"用户ID"`
IsDelete bool `json:"isDelete" bson:"isDelete" comment:"是否删除"`
ReadAt time.Time `json:"readAt" bson:"readAt" comment:"阅读时间"`
CreatedAt time.Time `json:"createdAt" bson:"createdAt" comment:"创建时间"`
UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt" comment:"更新时间"`
}
type ReadHistory struct {
HasRead bool `json:"hasRead" bson:"hasRead"` // 是否已读
CID primitive.ObjectID `json:"cid" bson:"cid"` // 子集ID
Name string `json:"name" bson:"name"` // 续看子集名称
}
func main() {
g := common.NewGen("91porn-server", "媒体书架列表")
g.Generate(MediaBookshelf{})
}
generate/model/imgroup/main.go
+26
View File
@@ -0,0 +1,26 @@
package main
import (
"91porn-server/generate/common"
"go.mongodb.org/mongo-driver/bson/primitive"
"time"
)
type ImGroup struct {
ID primitive.ObjectID `bson:"_id,omitempty" json:"id"`
GroupId int64 `json:"groupId" bson:"groupId" comment:"群id"` // 群id
Name string `json:"name" bson:"name" comment:"群组名"` // 群组名
Cover string `json:"cover" bson:"cover" comment:"封面"`
Summary string `json:"summary" bson:"summary" comment:"简介"` // 简介
MemberNum int64 `json:"memberNum" bson:"memberNum" comment:"成员数量（真实）"` // 成员数量（真实）
FakeMemberNum int64 `json:"fakeMemberNum" bson:"fakeMemberNum" comment:"成员数量（假的）"` // 成员数量（假的）
Price int64 `bson:"price" json:"price" comment:"加入群聊价格 0-免费"` // 加入群聊价格 0-免费
CreatedAt time.Time `json:"createdAt" bson:"createdAt" comment:"文档创建时间"` // 文档创建时间
UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt" comment:"文档更新时间"` // 文档更新时间
}
func main() {
g := common.NewGen("91porn-server", "im群组")
g.AppGenerate = true
g.Generate(ImGroup{})
}
generate/model/imgroupmember/main.go
+20
View File
@@ -0,0 +1,20 @@
package main
import (
"91porn-server/generate/common"
"go.mongodb.org/mongo-driver/bson/primitive"
"time"
)
type ImGroupMember struct {
ID primitive.ObjectID `bson:"_id,omitempty" json:"id" comment:"文档id"`
Uid int64 `json:"uid" bson:"uid" comment:"uid"`
Price int64 `json:"price" bson:"price" comment:"进群花费价格"`
CreatedAt time.Time `json:"createdAt" bson:"createdAt" comment:"创建时间"`
}
func main() {
g := common.NewGen("91porn-server", "im群组成员")
g.AppGenerate = true
g.Generate(ImGroupMember{})
}
generate/model/immessage/main.go
+22
View File
@@ -0,0 +1,22 @@
package main
import (
"91porn-server/generate/common"
"go.mongodb.org/mongo-driver/bson/primitive"
"time"
)
type ImMessage struct {
ID primitive.ObjectID `bson:"_id,omitempty" json:"id" comment:"文档id"`
GroupId int64 `json:"groupId" bson:"groupId" comment:"群组id"`
Uid int64 `json:"uid" bson:"uid" comment:"用户id"`
Content string `json:"content" bson:"content" comment:"消息内容"`
Image string `json:"image" bson:"image" comment:"图片"`
CreatedAt time.Time `json:"createdAt" bson:"createdAt" comment:"创建时间"`
}
func main() {
g := common.NewGen("91porn-server", "im消息")
g.AppGenerate = true
g.Generate(ImMessage{})
}
generate/model/nakechatorder/main.go
+27
View File
@@ -0,0 +1,27 @@
package main
import (
"91porn-server/generate/common"
"go.mongodb.org/mongo-driver/bson/primitive"
"time"
)
type NakedChatOrder struct {
ID primitive.ObjectID `bson:"_id,omitempty" json:"id" comment:"文档id"`
Nid primitive.ObjectID `json:"nid" bson:"nid" comment:"裸聊id"`
Uid int64 `json:"uid" bson:"uid" comment:"uid"`
UserContact string `json:"userContact " bson:"userContact" comment:"用户联系方式"`
Num int64 `json:"num" bson:"num" comment:"购买数量"`
Price int64 `json:"price" bson:"price" comment:"购买的单价"`
Amount int64 `json:"amount" bson:"amount" comment:"总额"`
Remark string `json:"remark" bson:"remark" comment:"订单备注信息"`
Status int `json:"status" bson:"status" comment:"0-已下单 1-已完成"`
CreatedAt time.Time `json:"createdAt" bson:"createdAt" comment:"文档创建时间"` // 文档创建时间
UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt" comment:"文档更新时间"` // 文档更新时间
}
func main() {
g := common.NewGen("91porn-server", "裸聊订单")
g.AppGenerate = true
g.Generate(NakedChatOrder{})
}
generate/model/nakedchat/main.go
+33
View File
@@ -0,0 +1,33 @@
package main
import (
"91porn-server/generate/common"
"go.mongodb.org/mongo-driver/bson/primitive"
"time"
)
type NakedChat struct {
ID primitive.ObjectID `bson:"_id,omitempty" json:"id" comment:"文档id"`
Title string `json:"title" bson:"title" comment:"标题"`
Cover string `json:"cover" bson:"cover" comment:"封面"`
Price int64 `json:"price" bson:"price" comment:"价格 n金币/分钟"`
Options []int64 `json:"options" bson:"options" comment:"可供购买选项 ，多少分钟"`
Images []string `json:"images" bson:"images" comment:"图片列表"`
Video string `json:"video" bson:"video" comment:"展示的视频"`
Contact string `json:"contact " bson:"contact" comment:"联系方式"`
Age int `json:"age" bson:"age" comment:"年龄 单位 岁"`
Weight int `json:"weight" bson:"weight" comment:"体重 单位 kg"`
Cup string `json:"cup" bson:"cup" comment:"罩杯"`
SaleNum int64 `json:"saleNum" bson:"saleNum" comment:"销售数量"` // 销售数量
FakeSaleNum int64 `json:"fakeSaleNum" bson:"fakeSaleNum" comment:"销售数量(假)"` // 销售数量(假)
BusinessHours string `json:"businessHours" bson:"businessHours" comment:"连线时间"` // 连线时间
Summary string `json:"summary" bson:"summary" comment:"简介"` // 简介
CreatedAt time.Time `json:"createdAt" bson:"createdAt" comment:"文档创建时间"` // 文档创建时间
UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt" comment:"文档更新时间"` // 文档更新时间
}
func main() {
g := common.NewGen("91porn-server", "裸聊")
g.AppGenerate = true
g.Generate(NakedChat{})
}
generate/model/sysconf/main.go
+31
View File
@@ -0,0 +1,31 @@
package main
import (
"time"
"91porn-server/generate/common"
"go.mongodb.org/mongo-driver/bson/primitive"
)
type SysConf struct {
ID primitive.ObjectID `json:"id" bson:"_id,omitempty" comment:"文档id"`
GroupName string `json:"groupName" bson:"groupName" comment:"分组名"`
GpCode string `json:"gpCode" bson:"gpCode" comment:"分组编码"`
VCode string `json:"vCode" bson:"vCode" comment:"变量名"`
Title string `json:"title" bson:"title" comment:"变量标题"`
Tip string `json:"tip" bson:"tip" comment:"变量描述"`
Type string `json:"type" bson:"type" comment:"类型:text,string,img,int,bool,object,text-array,string-array"`
Value string `json:"value" bson:"value" comment:"变量值"`
IsRequired bool `json:"is_required" bson:"is_required" comment:"是否必填"`
SortOrder int `json:"sort_order" bson:"sort_order" comment:"排序值"`
CreatedAt time.Time `json:"createdAt" bson:"createdAt" comment:"创建时间"`
UpdatedAt time.Time `json:"updatedAt" bson:"updatedAt" comment:"更新时间"`
}
func main() {
g := common.NewGen("91porn-server", "系统通用配置")
g.AppGenerate = false
g.Generate(SysConf{})
}
generate/tpl/app/api/{{PackageTableName}}.tpl
+66
View File
@@ -0,0 +1,66 @@
package {{PackageTableName}}ctrl
import (
"{{ModuleName}}/app/service/{{PackageTableName}}ser"
"{{ModuleName}}/common"
"{{ModuleName}}/common/stderr"
"github.com/gin-gonic/gin"
)
// List doc
//
// @Summary 获取{{TableComment}}列表接口
// @Description 获取{{TableComment}}列表
// @Tags 移动端-{{TableComment}}
// @Accept mpfd,json
// @Produce json
// @Param q query {{PackageTableName}}ser.{{VariablePrefix}}QueryListReq false "请求参数"
// @Success 200 object {{PackageTableName}}ser.{{VariablePrefix}}ListRes "成功"
// @Failure 400 {string} json "{"msg": "操作失败"}"
// @Router /api/app/{{PackageTableName}}/list [get]
func List(ctx *gin.Context) {
p := &{{PackageTableName}}ser.{{VariablePrefix}}QueryListReq{}
if err := ctx.ShouldBind(&p); err != nil {
common.ServeJSON(ctx, stderr.ErrParamError, nil)
return
}
// 获取用户当前配置
// var err error
// p.Uid, err = common.GetUID(ctx)
// if err != nil {
// common.ServeJSON(ctx, stderr.ErrNoToken, err)
// return
// }
list := p.GetList()
common.ServeJSON(ctx, stderr.Success, list)
}
// Info doc
//
// @Summary 获取{{TableComment}}详情接口
// @Description 获取{{TableComment}}详情
// @Tags 移动端-{{TableComment}}
// @Accept mpfd,json
// @Produce json
// @Param q query {{PackageTableName}}ser.{{VariablePrefix}}QueryInfoReq false "请求参数"
// @Success 200 object {{PackageTableName}}ser.{{VariablePrefix}}QueryInfoRes "成功"
// @Failure 400 {string} json "{"msg": "操作失败"}"
// @Router /api/app/{{PackageTableName}}/info [get]
func Info(ctx *gin.Context) {
p := &{{PackageTableName}}ser.{{VariablePrefix}}QueryInfoReq{}
if err := ctx.ShouldBind(&p); err != nil {
common.ServeJSON(ctx, stderr.ErrParamError, nil)
return
}
data, err := p.GetInfo()
if err != nil {
common.ServeJSON(ctx, stderr.Failure, err.Error())
return
}
common.ServeJSON(ctx, stderr.Success, data)
}
generate/tpl/app/router/{{PackageTableName}}.tpl
+15
View File
@@ -0,0 +1,15 @@
package router
import (
"github.com/gin-gonic/gin"
"{{ModuleName}}/app/api/{{PackageTableName}}ctrl"
)
// {{RouterName}} 移动端接口 todo: 在 app/router/base.go 中新增路由
func {{RouterName}}Router(router *gin.RouterGroup) {
group := router.Group("/{{PackageTableName}}")
{
group.GET("/list", {{PackageTableName}}ctrl.List)
group.GET("/info", {{PackageTableName}}ctrl.Info)
}
}
generate/tpl/app/service/{{PackageTableName}}.tpl
+63
View File
@@ -0,0 +1,63 @@
package {{PackageTableName}}ser
import (
"{{ModuleName}}/common/log"
"{{ModuleName}}/models/cache/{{PackageTableName}}data"
"{{ModuleName}}/models/commod"
"{{ModuleName}}/models/v/{{PackageTableName}}mod"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
)
type {{VariablePrefix}}QueryListReq struct {
commod.Page
}
type {{VariablePrefix}}ListRes struct {
Total int64 `json:"total"`
HasNext bool `json:"hasNext"`
List []*{{PackageTableName}}mod.{{StructName}}Info `json:"list"`
}
// GetList 获取列表
func (p *{{VariablePrefix}}QueryListReq) GetList() {{VariablePrefix}}ListRes {
var res {{VariablePrefix}}ListRes
var err error
filter := bson.M{}
sort := bson.D{{"_id", -1}}
// 获取列表
var data []{{PackageTableName}}mod.{{StructName}}
data, res.Total, res.HasNext, err = {{PackageTableName}}data.GetListFromCache(filter, int64(p.Skip()), int64(p.Limit()), sort)
if err != nil {
log.Error("获取{{TableComment}}列表数据错误", log.Any("Params", *p), log.E(err))
return res
}
res.List = {{PackageTableName}}data.FormatAppDataList(data)
return res
}
type {{VariablePrefix}}QueryInfoReq struct {
ID string `json:"id" form:"id"` // id
}
type {{VariablePrefix}}QueryInfoRes = *{{PackageTableName}}mod.{{StructName}}Info
// GetInfo 获取详情
func (p *{{VariablePrefix}}QueryInfoReq) GetInfo() (res {{VariablePrefix}}QueryInfoRes, err error) {
oid, err := primitive.ObjectIDFromHex(p.ID)
if err != nil {
return res, err
}
var item {{PackageTableName}}mod.{{StructName}}
item, err = {{PackageTableName}}data.GetInfoFromCache(oid)
if err != nil {
log.Error("获取{{TableComment}}详情数据错误", log.Any("ID", p.ID), log.E(err))
return
}
res = {{PackageTableName}}data.FormatAppData(item)
return
}
generate/tpl/config.go
+44
View File
@@ -0,0 +1,44 @@
package tpl
type Tpl struct {
SplitDir bool
DirSuffix string
VariablePrefix string
}
var Config = map[string]Tpl{
"/app/api": {
SplitDir: true,
DirSuffix: "ctrl",
VariablePrefix: "App",
},
"/app/router": {
SplitDir: false,
},
"/app/service": {
SplitDir: true,
DirSuffix: "ser",
VariablePrefix: "App",
},
"/models/cache": {
SplitDir: true,
DirSuffix: "data",
},
"/models/v": {
SplitDir: true,
DirSuffix: "mod",
},
"/web/api": {
SplitDir: true,
DirSuffix: "ctrl",
VariablePrefix: "Web",
},
"/web/router": {
SplitDir: false,
},
"/web/service": {
SplitDir: true,
DirSuffix: "ser",
VariablePrefix: "Web",
},
}
generate/tpl/models/cache/data.tpl
Vendored
+34
View File
@@ -0,0 +1,34 @@
package {{PackageTableName}}data
import (
"{{ModuleName}}/models/v/{{PackageTableName}}mod"
)
// FormatAppDataList 格式化app列表数据
func FormatAppDataList(origin []{{PackageTableName}}mod.{{StructName}}) (res []*{{PackageTableName}}mod.{{StructName}}Info) {
res = make([]*{{PackageTableName}}mod.{{StructName}}Info, len(origin))
if len(origin) == 0 {
return
}
// 组装返回数据
for i, item := range origin {
res[i] = FormatAppData(item)
}
return
}
// FormatAppData 格式化app数据
func FormatAppData(item {{PackageTableName}}mod.{{StructName}}) (res *{{PackageTableName}}mod.{{StructName}}Info) {
if item.ID.IsZero() {
return
}
// 组装返回数据
res = &{{PackageTableName}}mod.{{StructName}}Info{
{{AppRespTransferStatement}}
}
return
}
generate/tpl/models/cache/{{PackageTableName}}.tpl
Vendored
+89
View File
@@ -0,0 +1,89 @@
package {{PackageTableName}}data
import (
"time"
"{{ModuleName}}/common/cachev2"
"{{ModuleName}}/common/db"
"{{ModuleName}}/models"
"{{ModuleName}}/models/v/{{PackageTableName}}mod"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
)
type ListRes struct {
List []{{PackageTableName}}mod.{{StructName}}
Count int64
HasNext bool
}
// GetListFromCache 从缓存获取列表数据
func GetListFromCache(cond bson.M, skip, limit int64, sort bson.D) ([]{{PackageTableName}}mod.{{StructName}}, int64, bool, error) {
var data ListRes
_, err := cachev2.Classes().CacheTime(10*time.Minute).AutoListKey(models.{{TableName}}).ResBind(&data).Cache(GetListData, cond, skip, limit, sort)
return data.List, data.Count, data.HasNext, err
}
// GetListData 获取列表数据并组装适合缓存格式返回
func GetListData(cond bson.M, skip, limit int64, sort bson.D) (res ListRes, err error) {
res.List, res.Count, res.HasNext, err = {{PackageTableName}}mod.GetList(cond, skip, limit, sort)
return
}
// GetInfoFromCache 获取数据详情
func GetInfoFromCache(id primitive.ObjectID) (res {{PackageTableName}}mod.{{StructName}}, err error) {
_, err = cachev2.Classes().CacheTime(30*time.Minute).AutoInfoKey(models.{{TableName}}, id.Hex()).ResBind(&res).Cache({{PackageTableName}}mod.GetInfo, id)
return
}
// GetAllFromCache 获取所有数据
func GetAllFromCache(cond bson.M, sort bson.D) (res []{{PackageTableName}}mod.{{StructName}}, err error) {
_, err = cachev2.Classes().CacheTime(30*time.Minute).AutoListKey(models.{{TableName}}+"-all").ResBind(&res).Cache({{PackageTableName}}mod.GetAll, cond, sort)
return
}
// InsertData 插入数据
func InsertData(t *db.MongoTool, data {{PackageTableName}}mod.{{StructName}}) (primitive.ObjectID, error) {
newID, err := {{PackageTableName}}mod.Insert(t, data)
if err != nil {
return newID, err
}
// 清除缓存数据
_, err = cachev2.Classes().Table(models.{{TableName}}).AutoClear(true, nil)
if err != nil {
return newID, err
}
return newID, nil
}
// UpdateData 更新数据
func UpdateData(t *db.MongoTool, id string, data map[string]interface{}) (int64, error) {
objID, err := primitive.ObjectIDFromHex(id)
if err != nil {
return 0, err
}
affect, err := {{PackageTableName}}mod.UpdateByID(t, objID, data)
if err != nil {
return 0, err
}
// 清理缓存
_, _ = cachev2.Classes().Table(models.{{TableName}}).AutoClear(true, &id)
return affect, err
}
// DeleteData 删除数据
func DeleteData(t *db.MongoTool, id primitive.ObjectID) (err error) {
err = {{PackageTableName}}mod.DeleteByID(t, id)
if err != nil {
return
}
// 清理缓存
all := "*"
_, _ = cachev2.Classes().Table(models.{{TableName}}).AutoClear(true, &all)
return
}
generate/tpl/models/v/app.tpl
+12
View File
@@ -0,0 +1,12 @@
package {{PackageTableName}}mod
import (
"time"
"go.mongodb.org/mongo-driver/bson/primitive"
)
// {{StructName}}Info 移动端返回内容
type {{StructName}}Info struct {
{{AppRespStructFields}}
}
generate/tpl/models/v/type.tpl
+44
View File
@@ -0,0 +1,44 @@
package {{PackageTableName}}mod
import (
"fmt"
"time"
"{{ModuleName}}/common/db"
"{{ModuleName}}/models"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo"
)
var mdb *db.MongoDB
// todo: 在models/table.go中创建表名
const table = models.{{TableName}}
{{StructDefine}}
func coll(t *db.MongoTool) *db.MongoTool {
if t == nil {
return mdb.Coll(table)
}
return t.Coll(table)
}
// Init 初始化索引 todo: 在models/Init/init.go中调用初始化
func Init() {
mdb = db.Init(table)
initIndex()
}
func initIndex() {
many := []mongo.IndexModel{
{
Keys: bson.D{{Key: "_id", Value: 1}},
},
}
if _, err := coll(nil).CreateIndex(many); err != nil {
panic(fmt.Sprintf("%s model set index err ==>[%+v]", table, err))
}
}
generate/tpl/models/v/{{PackageTableName}}.tpl
+144
View File
@@ -0,0 +1,144 @@
package {{PackageTableName}}mod
import (
"encoding/json"
"errors"
"fmt"
"{{ModuleName}}/common/db"
"{{ModuleName}}/common/log"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/mongo/options"
)
// GetList 获取列表
func GetList(cond bson.M, skip, limit int64, sort bson.D) (res []{{StructName}}, count int64, hasNext bool, err error) {
count, err = coll(nil).Count(cond)
if err != nil {
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetList", table, "Count", err),
log.Any("skip", skip),
log.Any("limit", limit),
log.Any("cond", cond),
log.Any("sort", sort),
)
return
}
if len(sort) == 0 {
sort = bson.D{{Key: "_id", Value: -1}}
}
opts := options.Find().SetSort(sort).SetSkip(skip).SetLimit(limit + 1)
if err = coll(nil).Find(&res, cond, opts); err != nil {
return
}
// 判断下一页
if len(res) > int(limit) {
hasNext = true
res = res[:limit]
}
return
}
// GetAll 查询全部文档
func GetAll(filter primitive.M, sort bson.D) (out []*{{StructName}}, err error) {
if len(sort) == 0 {
sort = bson.D{{Key: "_id", Value: -1}}
}
opts := options.Find().SetSort(sort)
if err = coll(nil).Find(&out, filter, opts); err != nil {
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetAll", table, "Find", err),
log.Any("filter", filter),
log.Any("opts", opts),
)
return nil, err
}
return
}
// QueryAllCount 查询文档条目数
func QueryAllCount(filter primitive.M) (int64, error) {
if count, err := coll(nil).Count(filter); err != nil {
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "QueryAllCount", table, "Count", err),
log.Any("filter", filter),
)
return 0, err
} else {
return count, nil
}
}
// GetInfo 通过id获取详细信息
func GetInfo(id primitive.ObjectID) ({{StructName}}, error) {
v := {{StructName}}{}
if err := coll(nil).FindOne(&v, bson.M{"_id": id}); err != nil {
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "GetInfo", table, "FindOne", err),
log.Any("id", id),
)
return v, err
}
if v.ID.IsZero() {
return v, errors.New("record not found")
}
return v, nil
}
// Insert 插入记录
func Insert(t *db.MongoTool, d {{StructName}}) (data primitive.ObjectID, err error) {
result, err := coll(t).InsertOne(&d)
if err != nil {
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "Insert", table, "InsertOne", err))
return
}
byteID, err := json.Marshal(result.InsertedID)
if err != nil {
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "Insert", table, "Marshal", err))
return
}
if err = data.UnmarshalJSON(byteID); err != nil {
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "Insert", table, "UnmarshalJSON", err))
return
}
return
}
// UpdateByID 根据id更新数据
func UpdateByID(t *db.MongoTool, id primitive.ObjectID, data map[string]interface{}) (int64, error) {
cond := bson.M{"_id": id}
return update(t, cond, data)
}
// update 更新数据
func update(t *db.MongoTool, cond primitive.M, data map[string]interface{}) (int64, error) {
result, err := coll(t).UpdateMany(cond, bson.M{"$set": bson.M(data)})
if err != nil {
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "Update", table, "UpdateMany", err),
log.Any("cond", cond),
log.Any("update", data),
)
return 0, err
}
return result.ModifiedCount, nil
}
// DeleteByID 删除数据
func DeleteByID(t *db.MongoTool, id primitive.ObjectID) error {
_, err := coll(t).DeleteMany(bson.M{"_id": id})
if err != nil {
if err == mongo.ErrNoDocuments {
return nil
} else {
log.Error(fmt.Sprintf("[METHOD-%s]==> Model %s %s fail error:%+v:", "DeleteByID", table, "DeleteMany", err), log.Any("id", id))
}
}
return err
}
generate/tpl/web/api/{{PackageTableName}}.tpl
+142
View File
@@ -0,0 +1,142 @@
package {{PackageTableName}}ctrl
import (
"encoding/json"
"github.com/gin-gonic/gin"
"{{ModuleName}}/common"
"{{ModuleName}}/common/stderr"
"{{ModuleName}}/models/l/operatorlgmod"
"{{ModuleName}}/web/service/{{PackageTableName}}ser"
)
// List doc
// @Summary 获取{{TableComment}}列表
// @Description 获取{{TableComment}}列表
// @Tags 后台-{{TableComment}}
// @Accept mpfd,json
// @Produce json
// @Param q query {{PackageTableName}}ser.{{VariablePrefix}}ListReq false "请求参数"
// @Success 200 object {{PackageTableName}}ser.{{VariablePrefix}}ListRes "成功"
// @Failure 400 {string} json "{"msg": "操作失败"}"
// @Router /api/web/admin/{{PackageTableName}}/list [get]
func List(ctx *gin.Context) {
var req = &{{PackageTableName}}ser.{{VariablePrefix}}ListReq{}
if err := ctx.ShouldBind(&req); err != nil {
common.ServeJSON(ctx, stderr.ErrParamError, err.Error())
return
}
res, err := req.GetList()
if err != nil {
common.ServeJSON(ctx, stderr.ErrDbQueryError, "")
return
}
common.ServeJSON(ctx, stderr.Success, res)
}
// Create doc
// @Summary 新增{{TableComment}}
// @Description 新增{{TableComment}}
// @Tags 后台-{{TableComment}}
// @Accept mpfd,json
// @Produce json
// @Param q body {{PackageTableName}}ser.{{VariablePrefix}}CreateReq false "请求参数"
// @Success 200 {string} json "{"msg": "操作成功"}"
// @Failure 400 {string} json "{"msg": "操作失败"}"
// @Router /api/web/admin/{{PackageTableName}}/create [post]
func Create(ctx *gin.Context) {
manager, err := common.GetAdminAct(ctx)
if err != nil {
common.ServeJSON(ctx, stderr.AdminIDErr, err.Error())
return
}
p := &{{PackageTableName}}ser.{{VariablePrefix}}CreateReq{}
err = ctx.ShouldBindJSON(&p)
if err != nil {
common.ServeJSON(ctx, stderr.ErrParamError, err.Error())
return
}
// 创建
err = p.Create()
if err != nil {
common.ServeJSON(ctx, stderr.Failure, err.Error())
return
}
log, _ := json.Marshal(p)
_ = operatorlgmod.RecordOperation(manager, "{{TableComment}}管理", "创建", string(log), ctx.Request.URL.RequestURI())
common.ServeJSON(ctx, stderr.Success, "")
}
// Update doc
// @Summary 更新{{TableComment}}
// @Description 更新{{TableComment}}
// @Tags 后台-{{TableComment}}
// @Accept mpfd,json
// @Produce json
// @Param q body {{PackageTableName}}ser.{{VariablePrefix}}UpdateReq false "请求参数"
// @Success 200 {string} json "{"msg": "操作成功"}"
// @Failure 400 {string} json "{"msg": "操作失败"}"
// @Router /api/web/admin/{{PackageTableName}}/update [post]
func Update(ctx *gin.Context) {
manager, err := common.GetAdminAct(ctx)
if err != nil {
common.ServeJSON(ctx, stderr.AdminIDErr, err.Error())
return
}
p := &{{PackageTableName}}ser.{{VariablePrefix}}UpdateReq{}
err = ctx.ShouldBindJSON(&p)
if err != nil {
common.ServeJSON(ctx, stderr.ErrParamError, err.Error())
return
}
// 更新
err = p.Update()
if err != nil {
common.ServeJSON(ctx, stderr.Failure, err.Error())
return
}
log, _ := json.Marshal(p)
_ = operatorlgmod.RecordOperation(manager, "{{TableComment}}管理", "更新", string(log), ctx.Request.URL.RequestURI())
common.ServeJSON(ctx, stderr.Success, "")
}
// Delete doc
// @Summary 删除{{TableComment}}
// @Description 删除{{TableComment}}
// @Tags 后台-{{TableComment}}
// @Accept mpfd,json
// @Produce json
// @Param q body {{PackageTableName}}ser.{{VariablePrefix}}DeleteReq false "请求参数"
// @Success 200 {string} json "{"msg": "操作成功"}"
// @Failure 400 {string} json "{"msg": "操作失败"}"
// @Router /api/web/admin/{{PackageTableName}}/delete [post]
func Delete(ctx *gin.Context) {
manager, err := common.GetAdminAct(ctx)
if err != nil {
common.ServeJSON(ctx, stderr.AdminIDErr, err.Error())
return
}
p := &{{PackageTableName}}ser.{{VariablePrefix}}DeleteReq{}
err = ctx.ShouldBindJSON(&p)
if err != nil {
common.ServeJSON(ctx, stderr.ErrParamError, err.Error())
return
}
// 删除
err = p.Delete()
if err != nil {
common.ServeJSON(ctx, stderr.Failure, err.Error())
return
}
log, _ := json.Marshal(p)
_ = operatorlgmod.RecordOperation(manager, "{{TableComment}}管理", "删除", string(log), ctx.Request.URL.RequestURI())
common.ServeJSON(ctx, stderr.Success, "")
}
generate/tpl/web/router/{{PackageTableName}}.tpl
+17
View File
@@ -0,0 +1,17 @@
package router
import (
"github.com/gin-gonic/gin"
"{{ModuleName}}/web/api/{{PackageTableName}}ctrl"
)
// {{RouterName}}Router {{TableComment}}管理 todo: 在 web/router/base.go 中新增路由
func {{RouterName}}Router(router *gin.RouterGroup) {
group := router.Group("/{{PackageTableName}}")
{
group.GET("/list", {{PackageTableName}}ctrl.List)
group.POST("/create", {{PackageTableName}}ctrl.Create)
group.POST("/update", {{PackageTableName}}ctrl.Update)
group.POST("/delete", {{PackageTableName}}ctrl.Delete)
}
}
generate/tpl/web/service/{{PackageTableName}}.tpl
+81
View File
@@ -0,0 +1,81 @@
package {{PackageTableName}}ser
import (
"time"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/bson/primitive"
"{{ModuleName}}/models/commod"
"{{ModuleName}}/models/cache/{{PackageTableName}}data"
"{{ModuleName}}/models/v/{{PackageTableName}}mod"
)
type {{VariablePrefix}}ListReq struct {
commod.Page
}
type {{VariablePrefix}}ListRes struct {
Total int64 `json:"total"`
HasNext bool `json:"hasNext"`
List []{{PackageTableName}}mod.{{StructName}} `json:"list"`
}
// GetList 获取列表
func (q *{{VariablePrefix}}ListReq) GetList() (res {{VariablePrefix}}ListRes, err error) {
filter := bson.M{}
res.List, res.Total, res.HasNext, err = {{PackageTableName}}mod.GetList(filter, int64(q.Skip()), int64(q.Limit()), q.GetSort())
return
}
type {{VariablePrefix}}CreateReq struct {
{{CreateReqFields}}
}
// Create 发布数据
func (p *{{VariablePrefix}}CreateReq) Create() error {
data := {{PackageTableName}}mod.{{StructName}}{
{{CreateStatement}}
}
// 创建数据
if _, err := {{PackageTableName}}data.InsertData(nil, data); err != nil {
return err
}
return nil
}
type {{VariablePrefix}}UpdateReq struct {
ID primitive.ObjectID `json:"id" binding:"required"`
{{UpdateReqFields}}
}
// Update 更新数据
func (p *{{VariablePrefix}}UpdateReq) Update() error {
_, err := {{PackageTableName}}mod.GetInfo(p.ID)
if err != nil {
return err
}
data := make(map[string]interface{})
{{UpdateStatement}}
if _, err = {{PackageTableName}}data.UpdateData(nil, p.ID.Hex(), data); err != nil {
return err
}
return nil
}
type {{VariablePrefix}}DeleteReq struct {
ID primitive.ObjectID `json:"id" binding:"required"`
}
// Delete 删除数据
func (p *{{VariablePrefix}}DeleteReq) Delete() error {
if err := {{PackageTableName}}data.DeleteData(nil, p.ID); err != nil {
return err
}
return nil
}