commit c7a30fb094f0081edf0a43b908e4e79ed8719de6 Author: liangzai <2440983361@qq.com> Date: Sat Dec 13 17:44:18 2025 +0800 f diff --git a/.cursor/rules/api.mdc b/.cursor/rules/api.mdc new file mode 100644 index 0000000..c55f0bf --- /dev/null +++ b/.cursor/rules/api.mdc @@ -0,0 +1,186 @@ +# Cursor 工作流程和规范 + +## 目录结构说明 + +### API 定义目录 (`app/main/api/desc/`) +``` +desc/ +├── admin/ # 后台管理接口 +│ ├── admin_user.api # 管理员用户相关接口 +│ ├── platform_user.api # 平台用户相关接口 +│ ├── order.api # 订单管理接口 +│ ├── promotion.api # 促销活动接口 +│ ├── menu.api # 菜单管理接口 +│ ├── role.api # 角色权限接口 +│ └── auth.api # 后台认证接口 +│ +├── front/ # 前台业务接口 +│ ├── user.api # 用户相关接口 +│ ├── product.api # 产品相关接口 +│ ├── pay.api # 支付相关接口 +│ ├── query.api # 查询相关接口 +│ ├── auth/ # 前台认证相关子模块 +│ ├── product/ # 产品相关子模块 +│ ├── query/ # 查询相关子模块 +│ ├── user/ # 用户相关子模块 +│ └── pay/ # 支付相关子模块 +│ +└── main.api # API 主入口文件，用于引入所有子模块 +``` + +### 目录规范 +1. 后台管理接口统一放在 `admin/` 目录下 + - 所有后台管理相关的 API 定义文件都放在此目录 + - 文件名应清晰表明模块功能，如 `order.api`、`user.api` 等 + - 后台接口统一使用 `/api/v1/admin/` 前缀 + +2. 前台业务接口统一放在 `front/` 目录下 + - 所有面向用户的 API 定义文件都放在此目录 + - 可以根据业务模块创建子目录，如 `user/`、`product/` 等 + - 前台接口统一使用 `/api/v1/` 前缀 + +3. 主入口文件 `main.api` + - 用于引入所有子模块的 API 定义 + - 保持清晰的模块分类和注释 + +## API 开发流程 + +### 1. API 定义 +- 根据业务类型选择正确的目录： + - 后台管理接口：`app/main/api/desc/admin/` 目录 + - 前台业务接口：`app/main/api/desc/front/` 目录 +- 创建新的 `.api` 文件，遵循以下命名规范： + - 后台接口：`[模块名].api`，如 `order.api`、`user.api` + - 前台接口：`[模块名].api`，如 `product.api`、`pay.api` +- API 定义规范： + - 使用 RESTful 风格 + - 请求/响应结构体命名规范：`[模块名][操作名][Req/Resp]` + - 字段类型使用 string 而不是 int64 来存储枚举值 + - 添加必要的注释说明 + - 请求/响应参数定义规范： + ```go + type ( + // 创建类请求（必填字段） + AdminCreateXXXReq { + Field1 string `json:"field1"` // 字段1说明 + Field2 int64 `json:"field2"` // 字段2说明 + Field3 string `json:"field3"` // 字段3说明 + } + + // 更新类请求（可选字段使用指针类型） + AdminUpdateXXXReq { + Id int64 `path:"id"` // ID（路径参数） + Field1 *string `json:"field1,optional"` // 字段1说明（可选） + Field2 *int64 `json:"field2,optional"` // 字段2说明（可选） + Field3 *string `json:"field3,optional"` // 字段3说明（可选） + } + + // 查询列表请求（分页参数 + 可选查询条件） + AdminGetXXXListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + Field1 *string `form:"field1,optional"` // 查询条件1（可选） + Field2 *int64 `form:"field2,optional"` // 查询条件2（可选） + Field3 *string `form:"field3,optional"` // 查询条件3（可选） + } + + // 列表项结构体（用于列表响应） + XXXListItem { + Id int64 `json:"id"` // ID + Field1 string `json:"field1"` // 字段1 + Field2 string `json:"field2"` // 字段2 + Field3 string `json:"field3"` // 字段3 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } + + // 列表响应 + AdminGetXXXListResp { + Total int64 `json:"total"` // 总数 + Items []XXXListItem `json:"items"` // 列表数据 + } + ) + ``` + + - 参数定义注意事项： + 1. 创建类请求（Create）： + - 所有字段都是必填的，使用普通类型（非指针） + - 使用 `json` 标签，不需要 `optional` 标记 + - 必须添加字段说明注释 + + 2. 更新类请求（Update）： + - 除 ID 外的所有字段都是可选的，使用指针类型（如 `*string`、`*int64`） + - 使用 `json` 标签，并添加 `optional` 标记 + - ID 字段使用 `path` 标签，因为是路径参数 + - 必须添加字段说明注释 + + 3. 查询列表请求（GetList）： + - 分页参数（page、pageSize）使用普通类型 + - 查询条件字段都是可选的，使用指针类型 + - 使用 `form` 标签，并添加 `optional` 标记 + - 必须添加字段说明注释 + + 4. 响应结构体： + - 列表响应使用 `Total` 和 `Items` 字段 + - 列表项使用单独的结构体定义 + - 时间字段统一使用 string 类型 + - 必须添加字段说明注释 + + 5. 标签使用规范： + - 路径参数：`path:"id"` + - JSON 参数：`json:"field_name"` + - 表单参数：`form:"field_name"` + - 可选字段：添加 `optional` 标记 + - 所有字段必须添加说明注释 + + - 示例： + ```go + // 通知管理接口 + @server( + prefix: /api/v1/admin/notification + group: admin_notification + ) + service main { + // 创建通知 + @handler AdminCreateNotification + post /create (AdminCreateNotificationReq) returns (AdminCreateNotificationResp) + + // 更新通知 + @handler AdminUpdateNotification + put /update/:id (AdminUpdateNotificationReq) returns (AdminUpdateNotificationResp) + + // 获取通知列表 + @handler AdminGetNotificationList + get /list (AdminGetNotificationListReq) returns (AdminGetNotificationListResp) + } + ``` + +### 2. 引入 API +- 在 `app/main/api/main.api` 中引入新定义的 API 文件： + ```go + import "desc/order.api" + ``` + +### 3. 生成代码 +- 在项目根目录运行 `gen_api.ps1` 脚本生成相关代码： + ```powershell + ./gen_api.ps1 + ``` +- 生成的文件包括： + - `app/main/api/internal/handler/` - 处理器 + - `app/main/api/internal/logic/` - 业务逻辑 + - `app/main/api/internal/svc/` - 服务上下文 + - `app/main/api/internal/types/` - 类型定义 + +### 4. 实现业务逻辑 +- 在 `app/main/api/internal/logic/` 目录下实现各个接口的业务逻辑 +- 具体实现规范请参考 `logic.mdc` 文件 + +## 注意事项 +1. 所有枚举类型使用 string 而不是 int64 +2. 错误处理必须使用 errors.Wrapf 和 xerr 包 +3. 涉及多表操作时使用事务 +4. 并发操作时注意使用互斥锁保护共享资源 +5. 时间类型统一使用 "2006-01-02 15:04:05" 格式 +6. 代码生成后需要检查并完善业务逻辑实现 +7. 保持代码风格统一，添加必要的注释 \ No newline at end of file diff --git a/.cursor/rules/logic.mdc b/.cursor/rules/logic.mdc new file mode 100644 index 0000000..ec720c1 --- /dev/null +++ b/.cursor/rules/logic.mdc @@ -0,0 +1,270 @@ +# Your rule content + +- You can @ files here +- You can use markdown but dont have to + +# Logic 实现规范 + +## 目录结构 +``` +app/main/api/internal/logic/ +├── admin_notification/ # 通知管理模块 +│ ├── admincreatenotificationlogic.go # 创建通知 +│ ├── admindeletnotificationlogic.go # 删除通知 +│ ├── admingetnotificationdetaillogic.go # 获取通知详情 +│ ├── admingetnotificationlistlogic.go # 获取通知列表 +│ └── adminupdatenotificationlogic.go # 更新通知 +└── [其他模块]/ +``` + +## Logic 实现规范 + +### 1. 基础结构 +每个 Logic 文件都应包含以下基础结构： +```go +package [模块名] + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type [操作名]Logic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func New[操作名]Logic(ctx context.Context, svcCtx *svc.ServiceContext) *[操作名]Logic { + return &[操作名]Logic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} +``` + +### 2. 增删改查实现规范 + +#### 2.1 创建操作（Create） +```go +func (l *[操作名]Logic) [操作名](req *types.[操作名]Req) (resp *types.[操作名]Resp, err error) { + // 1. 数据转换和验证 + data := &model.[表名]{ + Field1: req.Field1, + Field2: req.Field2, + // ... 其他字段映射 + } + + // 2. 数据库操作 + result, err := l.svcCtx.[表名]Model.Insert(l.ctx, nil, data) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "创建[操作对象]失败, err: %v, req: %+v", err, req) + } + + // 3. 返回结果 + id, _ := result.LastInsertId() + return &types.[操作名]Resp{Id: id}, nil +} +``` + +#### 2.2 删除操作（Delete） +```go +func (l *[操作名]Logic) [操作名](req *types.[操作名]Req) (resp *types.[操作名]Resp, err error) { + // 1. 查询记录是否存在 + record, err := l.svcCtx.[表名]Model.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查找[操作对象]失败, err: %v, id: %d", err, req.Id) + } + + // 2. 执行删除操作（软删除） + err = l.svcCtx.[表名]Model.DeleteSoft(l.ctx, nil, record) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "删除[操作对象]失败, err: %v, id: %d", err, req.Id) + } + + // 3. 返回结果 + return &types.[操作名]Resp{Success: true}, nil +} +``` + +#### 2.3 更新操作（Update） +```go +func (l *[操作名]Logic) [操作名](req *types.[操作名]Req) (resp *types.[操作名]Resp, err error) { + // 1. 查询记录是否存在 + record, err := l.svcCtx.[表名]Model.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查找[操作对象]失败, err: %v, id: %d", err, req.Id) + } + + // 2. 更新字段（使用指针判断是否更新） + if req.Field1 != nil { + record.Field1 = *req.Field1 + } + if req.Field2 != nil { + record.Field2 = *req.Field2 + } + // ... 其他字段更新 + + // 3. 执行更新操作 + _, err = l.svcCtx.[表名]Model.Update(l.ctx, nil, record) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "更新[操作对象]失败, err: %v, req: %+v", err, req) + } + + // 4. 返回结果 + return &types.[操作名]Resp{Success: true}, nil +} +``` + +#### 2.4 查询详情（GetDetail） +```go +func (l *[操作名]Logic) [操作名](req *types.[操作名]Req) (resp *types.[操作名]Resp, err error) { + // 1. 查询记录 + record, err := l.svcCtx.[表名]Model.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查找[操作对象]失败, err: %v, id: %d", err, req.Id) + } + + // 2. 构建响应 + resp = &types.[操作名]Resp{ + Id: record.Id, + Field1: record.Field1, + Field2: record.Field2, + CreateTime: record.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: record.UpdateTime.Format("2006-01-02 15:04:05"), + } + + // 3. 处理可选字段（如时间字段） + if record.OptionalField.Valid { + resp.OptionalField = record.OptionalField.Time.Format("2006-01-02") + } + + return resp, nil +} +``` + +#### 2.5 查询列表（GetList） +```go +func (l *[操作名]Logic) [操作名](req *types.[操作名]Req) (resp *types.[操作名]Resp, err error) { + // 1. 构建查询条件 + builder := l.svcCtx.[表名]Model.SelectBuilder() + + // 2. 添加查询条件（使用指针判断是否添加条件） + if req.Field1 != nil { + builder = builder.Where("field1 LIKE ?", "%"+*req.Field1+"%") + } + if req.Field2 != nil { + builder = builder.Where("field2 = ?", *req.Field2) + } + // ... 其他查询条件 + + // 3. 执行分页查询 + list, total, err := l.svcCtx.[表名]Model.FindPageListByPageWithTotal( + l.ctx, builder, req.Page, req.PageSize, "id DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询[操作对象]列表失败, err: %v, req: %+v", err, req) + } + + // 4. 构建响应列表 + items := make([]types.[列表项类型], 0, len(list)) + for _, item := range list { + listItem := types.[列表项类型]{ + Id: item.Id, + Field1: item.Field1, + Field2: item.Field2, + CreateTime: item.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: item.UpdateTime.Format("2006-01-02 15:04:05"), + } + // 处理可选字段 + if item.OptionalField.Valid { + listItem.OptionalField = item.OptionalField.Time.Format("2006-01-02") + } + items = append(items, listItem) + } + + // 5. 返回结果 + return &types.[操作名]Resp{ + Total: total, + Items: items, + }, nil +} +``` + +### 3. 错误处理规范 +1. 使用 `errors.Wrapf` 包装错误 +2. 使用 `xerr.NewErrCode` 创建业务错误 +3. 错误信息应包含： + - 操作类型（创建/更新/删除/查询） + - 具体错误描述 + - 相关参数信息（ID、请求参数等） + +### 4. 时间处理规范 +1. 时间格式化： + - 日期时间：`"2006-01-02 15:04:05"` + - 仅日期：`"2006-01-02"` +2. 可选时间字段处理： + ```go + if field.Valid { + resp.Field = field.Time.Format("2006-01-02") + } + ``` + +### 5. 数据库操作规范 +1. 查询条件构建： + ```go + builder := l.svcCtx.[表名]Model.SelectBuilder() + builder = builder.Where("field = ?", value) + ``` +2. 分页查询： + ```go + list, total, err := l.svcCtx.[表名]Model.FindPageListByPageWithTotal( + ctx, builder, page, pageSize, "id DESC") + ``` +3. 事务处理： + ```go + err = l.svcCtx.[表名]Model.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 事务操作 + return nil + }) + ``` + +### 6. 并发处理规范 +```go +var mu sync.Mutex +err = mr.MapReduceVoid(func(source chan<- interface{}) { + // 并发处理 +}, func(item interface{}, writer mr.Writer[struct{}], cancel func(error)) { + // 处理单个项目 +}, func(pipe <-chan struct{}, cancel func(error)) { + // 完成处理 +}) +``` + +### 7. 注意事项 +1. 所有数据库操作必须进行错误处理 +2. 更新操作必须使用指针类型判断字段是否更新 +3. 查询列表必须支持分页 +4. 时间字段必须统一格式化 +5. 可选字段必须进行空值判断 +6. 保持代码风格统一，添加必要的注释 +7. 涉及多表操作时使用事务 +8. 并发操作时注意使用互斥锁保护共享资源 +9. 错误处理必须使用 errors.Wrapf 和 xerr 包 +10. 时间类型统一使用 "2006-01-02 15:04:05" 格式 + +# Your rule content + +- You can @ files here +- You can use markdown but dont have to diff --git a/.cursorrules b/.cursorrules new file mode 100644 index 0000000..87e9bb4 --- /dev/null +++ b/.cursorrules @@ -0,0 +1,202 @@ +你是一位精通Go-Zero框架的AI编程助手，专门帮助开发基于Go-Zero的微服务API项目。 + +熟悉Go-Zero的项目结构和架构模式，包括： +- api服务开发 +- rpc服务开发 +- model层数据库操作 +- 中间件实现 +- 配置文件管理 +- JWT认证体系 +- 分布式事务处理 +- 使用goctl工具生成代码 + +项目目录结构说明： +``` +qnc-server/ # 项目根目录 +├── app/ # 应用服务目录 +│ └── user/ # 用户服务 +│ ├── cmd/ # 服务启动入口 +│ │ ├── api/ # API服务 +│ │ │ ├── desc/ # API接口定义目录 +│ │ │ │ ├── user/ # 用户模块API定义 +│ │ │ │ │ └── user.api # 用户API类型定义 +│ │ │ │ └── main.api # 主API文件 +│ │ │ ├── etc/ # 配置文件目录 +│ │ │ │ └── user.yaml # 服务配置文件 +│ │ │ └── internal/ # 内部代码 +│ │ │ ├── config/ # 配置结构定义 +│ │ │ ├── handler/ # HTTP处理器 +│ │ │ ├── logic/ # 业务逻辑 +│ │ │ ├── middleware/ # 中间件 +│ │ │ ├── svc/ # 服务上下文 +│ │ │ └── types/ # 类型定义 +│ │ └── rpc/ # RPC服务（如果有） +│ └── model/ # 数据库模型 +├── common/ # 公共代码 +│ ├── ctxdata/ # 上下文数据处理 +│ ├── globalkey/ # 全局键值定义 +│ ├── interceptor/ # 拦截器 +│ ├── jwt/ # JWT认证 +│ ├── kqueue/ # 消息队列 +│ ├── middleware/ # 中间件 +│ ├── result/ # 统一返回结果 +│ ├── tool/ # 工具函数 +│ ├── uniqueid/ # 唯一ID生成 +│ ├── wxminisub/ # 微信小程序订阅 +│ └── xerr/ # 错误处理 +├── data/ # 数据文件目录 +├── deploy/ # 部署相关文件 +│ └── template/ # goctl模板 +├── pkg/ # 可复用的包 +│ └── lzkit/ # 工具包 +└── tmp/ # 临时文件目录 +``` + +目录作用说明： +1. app/user/cmd/api/：API服务目录 + - desc/：API接口定义，包含各模块的API文件 + - main.api：主API文件，导入所有模块API并定义路由 + - user/user.api：用户模块的请求响应参数定义 + - order/order.api：订单模块的请求响应参数定义 + - 其他模块API定义文件 + - etc/：配置文件目录，存放yaml配置 + - user.yaml：包含数据库、缓存、JWT等配置信息 + - internal/：服务内部代码 + - config/：配置结构定义，对应etc下的yaml文件 + - handler/：HTTP请求处理器，负责解析请求和返回响应 + - logic/：业务逻辑实现，处理具体业务 + - middleware/：HTTP中间件，如认证、日志等 + - svc/：服务上下文，管理服务依赖（如DB、Cache等） + - types/：请求响应的结构体定义，由goctl根据API文件生成(不允许自己修改) + +2. app/user/model/：数据库模型层 + - userModel.go：用户表模型定义及CRUD方法 + - userModel_gen.go：goctl工具生成的基础数据库操作代码(不允许自己修改) + - vars.go：定义数据库相关变量和常量 + - 其他数据表模型文件 + +3. common/：存放公共代码 + - ctxdata/：上下文数据处理 + - globalkey/：全局键值定义 + - interceptor/：拦截器 + - jwt/：JWT认证相关 + - kqueue/：消息队列处理 + - middleware/：全局中间件 + - result/：统一返回结果处理 + - tool/：通用工具函数 + - uniqueid/：唯一ID生成器 + - wxminisub/：微信小程序订阅功能 + - xerr/：统一错误处理 + +4. pkg/：可在多个服务间共享的包 + - lzkit/：通用工具包 + +5. deploy/：部署相关文件 + - template/：goctl代码生成模板 + +6. data/：数据文件目录 + - 用于存放项目相关的数据文件 + +7. tmp/：临时文件目录 + - 用于存放临时生成的文件 + +使用goctl生成API服务的步骤： +1. 首先确保API定义目录存在： + ```bash + mkdir -p app/user/cmd/api/desc/user + ``` + +2. API文件组织结构（单体服务模式）： + ``` + app/user/cmd/api/desc/ + ├── user/ + │ └── user.api # 用户模块的请求响应参数定义 + ├── order/ + │ └── order.api # 订单模块的请求响应参数定义 + └── main.api # 主API文件，集中管理所有模块的API定义 + ``` + +3. 在app/user/cmd/api/desc/main.api中集中管理所有API： + ``` + syntax = "v1" + + info( + title: "单体服务API" + desc: "集中管理所有模块的API" + author: "team" + version: "v1" + ) + + // 导入各模块的类型定义 + import "user/user.api" + import "order/order.api" + + // 各模块下定义路由，例如user模块 desc/user.api + @server ( + prefix: api/v1 + group: user + ) + service main { + // 用户模块接口 + @handler Login + post /login (LoginReq) returns (LoginResp) + } + ``` + +4. 各模块在下一层定义类型，例如在app/user/cmd/api/desc/user/user.api中只定义用户模块的接口的类型： + ``` + type ( + LoginReq { + Username string `json:"username"` + Password string `json:"password"` + } + + LoginResp { + Token string `json:"token"` + ExpireAt int64 `json:"expireAt"` + } + ) + ``` + +5. 使用goctl生成API代码（始终使用main.api）： + ```bash + goctl api go -api app/user/cmd/api/desc/main.api -dir app/user/cmd/api --home ./deploy/template + ``` + +注意：无论修改哪个模块的API文件，都需要执行main.api来生成代码，因为这是单体服务模式。 +6. goctl生成的文件和目录结构： + ``` + app/user/cmd/api/ + ├── desc/ # API接口定义目录（已存在） + ├── etc/ # 配置文件目录 + │ └── main.yaml # 服务配置文件 + ├── internal/ # 内部代码 + │ ├── config/ # 配置结构定义 + │ │ └── config.go # 配置结构体 + │ ├── handler/ # HTTP处理器 + │ │ ├── routes.go # 路由注册 + │ │ └── user/ # 用户模块处理器 + │ │ └── login_handler.go # 登录处理器 + │ ├── logic/ # 业务逻辑 + │ │ └── user/ # 用户模块逻辑 + │ │ └── login_logic.go # 登录逻辑 + │ ├── middleware/ # 中间件 + │ ├── svc/ # 服务上下文 + │ │ └── service_context.go # 服务上下文定义 + │ └── types/ # 类型定义 + │ └── types.go # 请求响应类型定义 + └── main.go # 服务入口文件 + ``` + +7. 生成代码后，才能够实现具体的业务逻辑，例如： + - user.api中的`mobileLogin`接口生成的逻辑文件在`app/user/cmd/api/internal/logic/user/mobile_login_logic.go` + - user.api中的`wxMiniAuth`接口生成的逻辑文件在`app/user/cmd/api/internal/logic/user/wx_mini_auth_logic.go` + - query.api中的`queryService`接口生成的逻辑文件在`app/user/cmd/api/internal/logic/query/query_service_logic.go` + + 生成的逻辑文件中需要实现`Logic`结构体的`XXX`方法（方法名与接口名对应），这是业务逻辑的核心部分。 + +代码说明尽量简洁，但关键逻辑和go-zero特有模式需要添加注释。 + +始终关注性能、安全性和可维护性。 + +在回答问题时，优先考虑Go-Zero的特性和设计理念，而不是通用的Go编程模式。 \ No newline at end of file diff --git a/.gitignore b/.gitignore new file mode 100644 index 0000000..832cc89 --- /dev/null +++ b/.gitignore @@ -0,0 +1,31 @@ +# idea +.idea +.idea/ +*.iml +.DS_Store +**/.DS_Store + +#deploy data +data/* +!data/.gitkeep + +# gitlab ci +.cache + +#vscode +.vscode +.vscode/ + +/tmp/ + +# 打包出来的可执行文件 +/app/api +/app/main/api/main +/app/main/api/_* + + +# 文档目录 +documents/* +!documents/.gitkeep + +deploy/script/js diff --git a/app/main/api/Dockerfile b/app/main/api/Dockerfile new file mode 100644 index 0000000..12f447e --- /dev/null +++ b/app/main/api/Dockerfile @@ -0,0 +1,33 @@ +FROM golang:1.23.4-alpine AS builder + +LABEL stage=gobuilder + +ENV CGO_ENABLED 0 +ENV GOPROXY https://goproxy.cn,direct +RUN sed -i 's/dl-cdn.alpinelinux.org/mirrors.aliyun.com/g' /etc/apk/repositories + +RUN apk update --no-cache && apk add --no-cache tzdata + +WORKDIR /build + +ADD go.mod . +ADD go.sum . +RUN go mod download +COPY . . +COPY app/main/api/etc /app/etc +COPY app/main/api/static /app/static +RUN go build -ldflags="-s -w" -o /app/main app/main/api/main.go + + +FROM scratch + +COPY --from=builder /etc/ssl/certs/ca-certificates.crt /etc/ssl/certs/ca-certificates.crt +COPY --from=builder /usr/share/zoneinfo/Asia/Shanghai /usr/share/zoneinfo/Asia/Shanghai +ENV TZ Asia/Shanghai + +WORKDIR /app +COPY --from=builder /app/main /app/main +COPY --from=builder /app/etc /app/etc +COPY --from=builder /app/static /app/static + +CMD ["./main", "-f", "etc/main.yaml"] diff --git a/app/main/api/desc/admin/admin_agent.api b/app/main/api/desc/admin/admin_agent.api new file mode 100644 index 0000000..90bb543 --- /dev/null +++ b/app/main/api/desc/admin/admin_agent.api @@ -0,0 +1,426 @@ +syntax = "v1" + +info ( + title: "后台代理管理服务" + desc: "新代理系统后台管理接口" + author: "team" + version: "v1" +) + +// ============================================ +// 代理管理接口 +// ============================================ +@server ( + prefix: /api/v1/admin/agent + group: admin_agent + middleware: AdminAuthInterceptor +) +service main { + // 代理分页查询 + @handler AdminGetAgentList + get /list (AdminGetAgentListReq) returns (AdminGetAgentListResp) + + // 代理审核 + @handler AdminAuditAgent + post /audit (AdminAuditAgentReq) returns (AdminAuditAgentResp) + + // 代理推广链接分页查询 + @handler AdminGetAgentLinkList + get /link/list (AdminGetAgentLinkListReq) returns (AdminGetAgentLinkListResp) + + // 代理订单分页查询 + @handler AdminGetAgentOrderList + get /order/list (AdminGetAgentOrderListReq) returns (AdminGetAgentOrderListResp) + + // 代理佣金分页查询 + @handler AdminGetAgentCommissionList + get /commission/list (AdminGetAgentCommissionListReq) returns (AdminGetAgentCommissionListResp) + + // 代理返佣分页查询 + @handler AdminGetAgentRebateList + get /rebate/list (AdminGetAgentRebateListReq) returns (AdminGetAgentRebateListResp) + + // 代理升级记录分页查询 + @handler AdminGetAgentUpgradeList + get /upgrade/list (AdminGetAgentUpgradeListReq) returns (AdminGetAgentUpgradeListResp) + + // 代理提现分页查询 + @handler AdminGetAgentWithdrawalList + get /withdrawal/list (AdminGetAgentWithdrawalListReq) returns (AdminGetAgentWithdrawalListResp) + + // 代理提现审核 + @handler AdminAuditWithdrawal + post /withdrawal/audit (AdminAuditWithdrawalReq) returns (AdminAuditWithdrawalResp) + + // 代理实名认证分页查询 + @handler AdminGetAgentRealNameList + get /real_name/list (AdminGetAgentRealNameListReq) returns (AdminGetAgentRealNameListResp) + + // 代理实名认证审核（已废弃：实名认证改为三要素核验，无需审核） + // @handler AdminAuditRealName + // post /real_name/audit (AdminAuditRealNameReq) returns (AdminAuditRealNameResp) + // 系统配置查询 + @handler AdminGetAgentConfig + get /config returns (AdminGetAgentConfigResp) + + // 系统配置更新 + @handler AdminUpdateAgentConfig + post /config/update (AdminUpdateAgentConfigReq) returns (AdminUpdateAgentConfigResp) + + // 产品配置分页查询 + @handler AdminGetAgentProductConfigList + get /product_config/list (AdminGetAgentProductConfigListReq) returns (AdminGetAgentProductConfigListResp) + + // 产品配置更新 + @handler AdminUpdateAgentProductConfig + post /product_config/update (AdminUpdateAgentProductConfigReq) returns (AdminUpdateAgentProductConfigResp) + + // 生成钻石邀请码 + @handler AdminGenerateDiamondInviteCode + post /invite_code/diamond/generate (AdminGenerateDiamondInviteCodeReq) returns (AdminGenerateDiamondInviteCodeResp) + + // 邀请码列表查询 + @handler AdminGetInviteCodeList + get /invite_code/list (AdminGetInviteCodeListReq) returns (AdminGetInviteCodeListResp) +} + +type ( + // 代理分页查询 + AdminGetAgentListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + Mobile *string `form:"mobile,optional"` // 手机号（可选） + Region *string `form:"region,optional"` // 区域（可选） + Level *int64 `form:"level,optional"` // 等级（可选） + TeamLeaderId *string `form:"team_leader_id,optional"` // 团队首领ID（可选） + } + AgentListItem { + Id string `json:"id"` // 主键 + UserId string `json:"user_id"` // 用户ID + Level int64 `json:"level"` // 等级：1=普通，2=黄金，3=钻石 + LevelName string `json:"level_name"` // 等级名称 + Region string `json:"region"` // 区域 + Mobile string `json:"mobile"` // 手机号 + WechatId string `json:"wechat_id"` // 微信号 + TeamLeaderId string `json:"team_leader_id"` // 团队首领ID + AgentCode int64 `json:"agent_code"` + Balance float64 `json:"balance"` // 钱包余额 + TotalEarnings float64 `json:"total_earnings"` // 累计收益 + FrozenBalance float64 `json:"frozen_balance"` // 冻结余额 + WithdrawnAmount float64 `json:"withdrawn_amount"` // 提现总额 + IsRealName bool `json:"is_real_name"` // 是否已实名 + CreateTime string `json:"create_time"` // 创建时间 + } + AdminGetAgentListResp { + Total int64 `json:"total"` // 总数 + Items []AgentListItem `json:"items"` // 列表数据 + } + // 代理审核 + AdminAuditAgentReq { + AuditId int64 `json:"audit_id"` // 审核记录ID + Status int64 `json:"status"` // 审核状态：1=通过，2=拒绝 + AuditReason string `json:"audit_reason"` // 审核原因（拒绝时必填） + } + AdminAuditAgentResp { + Success bool `json:"success"` + } + // 推广链接分页查询 + AdminGetAgentLinkListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + ProductId *string `form:"product_id,optional"` // 产品ID（可选） + LinkIdentifier *string `form:"link_identifier,optional"` // 推广码（可选） + } + AgentLinkListItem { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + ProductId string `json:"product_id"` // 产品ID + ProductName string `json:"product_name"` // 产品名称 + SetPrice float64 `json:"set_price"` // 设定价格 + ActualBasePrice float64 `json:"actual_base_price"` // 实际底价 + LinkIdentifier string `json:"link_identifier"` // 推广码 + CreateTime string `json:"create_time"` // 创建时间 + } + AdminGetAgentLinkListResp { + Total int64 `json:"total"` // 总数 + Items []AgentLinkListItem `json:"items"` // 列表数据 + } + // 代理订单分页查询 + AdminGetAgentOrderListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + OrderId *string `form:"order_id,optional"` // 订单ID（可选） + ProcessStatus *int64 `form:"process_status,optional"` // 处理状态（可选） + } + AgentOrderListItem { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + OrderId string `json:"order_id"` // 订单ID + ProductId string `json:"product_id"` // 产品ID + ProductName string `json:"product_name"` // 产品名称 + OrderAmount float64 `json:"order_amount"` // 订单金额 + SetPrice float64 `json:"set_price"` // 设定价格 + ActualBasePrice float64 `json:"actual_base_price"` // 实际底价 + PriceCost float64 `json:"price_cost"` // 提价成本 + AgentProfit float64 `json:"agent_profit"` // 代理收益 + ProcessStatus int64 `json:"process_status"` // 处理状态 + CreateTime string `json:"create_time"` // 创建时间 + } + AdminGetAgentOrderListResp { + Total int64 `json:"total"` // 总数 + Items []AgentOrderListItem `json:"items"` // 列表数据 + } + // 代理佣金分页查询 + AdminGetAgentCommissionListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + OrderId *string `form:"order_id,optional"` // 订单ID（可选） + Status *int64 `form:"status,optional"` // 状态（可选） + } + AgentCommissionListItem { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + OrderId string `json:"order_id"` // 订单ID + ProductName string `json:"product_name"` // 产品名称 + Amount float64 `json:"amount"` // 金额 + Status int64 `json:"status"` // 状态 + CreateTime string `json:"create_time"` // 创建时间 + } + AdminGetAgentCommissionListResp { + Total int64 `json:"total"` // 总数 + Items []AgentCommissionListItem `json:"items"` // 列表数据 + } + // 代理返佣分页查询 + AdminGetAgentRebateListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + SourceAgentId *string `form:"source_agent_id,optional"` // 来源代理ID（可选） + RebateType *int64 `form:"rebate_type,optional"` // 返佣类型（可选） + } + AgentRebateListItem { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 获得返佣的代理ID + SourceAgentId string `json:"source_agent_id"` // 来源代理ID + OrderId string `json:"order_id"` // 订单ID + RebateType int64 `json:"rebate_type"` // 返佣类型 + Amount float64 `json:"amount"` // 金额 + CreateTime string `json:"create_time"` // 创建时间 + } + AdminGetAgentRebateListResp { + Total int64 `json:"total"` // 总数 + Items []AgentRebateListItem `json:"items"` // 列表数据 + } + // 代理升级记录分页查询 + AdminGetAgentUpgradeListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + UpgradeType *int64 `form:"upgrade_type,optional"` // 升级类型（可选） + Status *int64 `form:"status,optional"` // 状态（可选） + } + AgentUpgradeListItem { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + FromLevel int64 `json:"from_level"` // 原等级 + ToLevel int64 `json:"to_level"` // 目标等级 + UpgradeType int64 `json:"upgrade_type"` // 升级类型 + UpgradeFee float64 `json:"upgrade_fee"` // 升级费用 + RebateAmount float64 `json:"rebate_amount"` // 返佣金额 + Status int64 `json:"status"` // 状态 + CreateTime string `json:"create_time"` // 创建时间 + } + AdminGetAgentUpgradeListResp { + Total int64 `json:"total"` // 总数 + Items []AgentUpgradeListItem `json:"items"` // 列表数据 + } + // 代理提现分页查询 + AdminGetAgentWithdrawalListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + Status *int64 `form:"status,optional"` // 状态（可选） + WithdrawNo *string `form:"withdraw_no,optional"` // 提现单号（可选） + } + AgentWithdrawalListItem { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + WithdrawNo string `json:"withdraw_no"` // 提现单号 + Amount float64 `json:"amount"` // 金额 + TaxAmount float64 `json:"tax_amount"` // 税费金额 + ActualAmount float64 `json:"actual_amount"` // 实际到账金额 + Status int64 `json:"status"` // 状态 + PayeeAccount string `json:"payee_account"` // 收款账户 + PayeeName string `json:"payee_name"` // 收款人姓名 + Remark string `json:"remark"` // 备注 + CreateTime string `json:"create_time"` // 创建时间 + } + AdminGetAgentWithdrawalListResp { + Total int64 `json:"total"` // 总数 + Items []AgentWithdrawalListItem `json:"items"` // 列表数据 + } + // 代理提现审核 + AdminAuditWithdrawalReq { + WithdrawalId string `json:"withdrawal_id"` // 提现记录ID + Status int64 `json:"status"` // 审核状态：2=通过，3=拒绝 + Remark string `json:"remark"` // 备注 + } + AdminAuditWithdrawalResp { + Success bool `json:"success"` + } + // 代理实名认证分页查询 + AdminGetAgentRealNameListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + Status *int64 `form:"status,optional"` // 状态（可选）：1=未验证，2=已通过 + } + AgentRealNameListItem { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + Name string `json:"name"` // 姓名 + IdCard string `json:"id_card"` // 身份证号 + Mobile string `json:"mobile"` // 手机号 + Status int64 `json:"status"` // 状态：1=未验证，2=已通过（verify_time不为空表示已通过） + VerifyTime string `json:"verify_time"` // 验证时间（三要素核验通过时间） + CreateTime string `json:"create_time"` // 创建时间 + } + AdminGetAgentRealNameListResp { + Total int64 `json:"total"` // 总数 + Items []AgentRealNameListItem `json:"items"` // 列表数据 + } + // 代理实名认证审核 + AdminAuditRealNameReq { + RealNameId int64 `json:"real_name_id"` // 实名认证记录ID + Status int64 `json:"status"` // 审核状态：2=通过，3=拒绝 + AuditReason string `json:"audit_reason"` // 审核原因（拒绝时必填） + } + AdminAuditRealNameResp { + Success bool `json:"success"` + } + // 系统配置查询（价格配置已移除，改为产品配置表管理） + AdminGetAgentConfigResp { + LevelBonus LevelBonusConfig `json:"level_bonus"` // 等级加成配置 + UpgradeFee UpgradeFeeConfig `json:"upgrade_fee"` // 升级费用配置 + UpgradeRebate UpgradeRebateConfig `json:"upgrade_rebate"` // 升级返佣配置 + DirectParentRebate DirectParentRebateConfig `json:"direct_parent_rebate"` // 直接上级返佣配置 + MaxGoldRebateAmount float64 `json:"max_gold_rebate_amount"` // 黄金代理最大返佣金额 + CommissionFreeze CommissionFreezeConfig `json:"commission_freeze"` // 佣金冻结配置 + TaxRate float64 `json:"tax_rate"` // 税率 + TaxExemptionAmount float64 `json:"tax_exemption_amount"` // 免税额度 + GoldMaxUpliftAmount float64 `json:"gold_max_uplift_amount"` + DiamondMaxUpliftAmount float64 `json:"diamond_max_uplift_amount"` + } + LevelBonusConfig { + Diamond int64 `json:"diamond"` // 钻石加成：0 + Gold int64 `json:"gold"` // 黄金加成：3 + Normal int64 `json:"normal"` // 普通加成：6 + } + UpgradeFeeConfig { + NormalToGold float64 `json:"normal_to_gold"` // 普通→黄金：199 + NormalToDiamond float64 `json:"normal_to_diamond"` // 普通→钻石：980 + } + UpgradeRebateConfig { + NormalToGoldRebate float64 `json:"normal_to_gold_rebate"` // 普通→黄金返佣：139 + ToDiamondRebate float64 `json:"to_diamond_rebate"` // 升级为钻石返佣：680 + } + DirectParentRebateConfig { + Diamond float64 `json:"diamond"` // 直接上级是钻石的返佣金额（6元） + Gold float64 `json:"gold"` // 直接上级是黄金的返佣金额（3元） + Normal float64 `json:"normal"` // 直接上级是普通的返佣金额（2元） + } + CommissionFreezeConfig { + Ratio float64 `json:"ratio"` // 佣金冻结比例（例如：0.1表示10%） + Threshold float64 `json:"threshold"` // 佣金冻结阈值（订单单价达到此金额才触发冻结，单位：元） + Days int64 `json:"days"` // 佣金冻结解冻天数（单位：天，例如：30表示30天后解冻） + } + // 系统配置更新（价格配置已移除，改为产品配置表管理） + AdminUpdateAgentConfigReq { + LevelBonus *LevelBonusConfig `json:"level_bonus,optional"` // 等级加成配置 + UpgradeFee *UpgradeFeeConfig `json:"upgrade_fee,optional"` // 升级费用配置 + UpgradeRebate *UpgradeRebateConfig `json:"upgrade_rebate,optional"` // 升级返佣配置 + DirectParentRebate *DirectParentRebateConfig `json:"direct_parent_rebate,optional"` // 直接上级返佣配置 + MaxGoldRebateAmount *float64 `json:"max_gold_rebate_amount,optional"` // 黄金代理最大返佣金额 + CommissionFreeze *CommissionFreezeConfig `json:"commission_freeze,optional"` // 佣金冻结配置 + TaxRate *float64 `json:"tax_rate,optional"` // 税率 + TaxExemptionAmount *float64 `json:"tax_exemption_amount,optional"` // 免税额度 + GoldMaxUpliftAmount *float64 `json:"gold_max_uplift_amount,optional"` + DiamondMaxUpliftAmount *float64 `json:"diamond_max_uplift_amount,optional"` + } + AdminUpdateAgentConfigResp { + Success bool `json:"success"` + } + // 产品配置分页查询 + AdminGetAgentProductConfigListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + ProductId *string `form:"product_id,optional"` // 产品ID（可选） + ProductName *string `form:"product_name,optional"` // 产品名称（可选，用于搜索） + } + AgentProductConfigItem { + Id string `json:"id"` // 主键 + ProductId string `json:"product_id"` // 产品ID + ProductName string `json:"product_name"` // 产品名称 + BasePrice float64 `json:"base_price"` // 基础底价 + PriceRangeMin float64 `json:"price_range_min"` // 最低定价 + PriceRangeMax float64 `json:"price_range_max"` // 最高定价 + PriceThreshold float64 `json:"price_threshold"` // 提价标准阈值 + PriceFeeRate float64 `json:"price_fee_rate"` // 提价手续费比例 + CreateTime string `json:"create_time"` // 创建时间 + } + AdminGetAgentProductConfigListResp { + Total int64 `json:"total"` // 总数 + Items []AgentProductConfigItem `json:"items"` // 列表数据 + } + // 产品配置更新 + AdminUpdateAgentProductConfigReq { + Id string `json:"id"` // 主键 + BasePrice float64 `json:"base_price"` // 基础底价 + PriceRangeMax float64 `json:"price_range_max"` // 最高定价（对应数据库 system_max_price） + PriceThreshold *float64 `json:"price_threshold,optional"` // 提价标准阈值（可选） + PriceFeeRate *float64 `json:"price_fee_rate,optional"` // 提价手续费比例（可选） + } + AdminUpdateAgentProductConfigResp { + Success bool `json:"success"` + } + // 生成钻石邀请码 + AdminGenerateDiamondInviteCodeReq { + Count int64 `json:"count"` // 生成数量 + ExpireDays int64 `json:"expire_days,optional"` // 过期天数（可选，0表示不过期） + Remark string `json:"remark,optional"` // 备注（可选） + } + AdminGenerateDiamondInviteCodeResp { + Codes []string `json:"codes"` // 生成的邀请码列表 + } + // 邀请码列表查询 + AdminGetInviteCodeListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + Code *string `form:"code,optional"` // 邀请码（可选） + AgentId *string `form:"agent_id,optional"` // 发放代理ID（可选，NULL表示平台发放） + TargetLevel *int64 `form:"target_level,optional"` // 目标等级（可选） + Status *int64 `form:"status,optional"` // 状态（可选） + } + InviteCodeListItem { + Id string `json:"id"` // 主键 + Code string `json:"code"` // 邀请码 + AgentId string `json:"agent_id"` // 发放代理ID（0表示平台发放） + AgentMobile string `json:"agent_mobile"` // 发放代理手机号 + TargetLevel int64 `json:"target_level"` // 目标等级 + Status int64 `json:"status"` // 状态：0=未使用，1=已使用，2=已失效 + UsedUserId string `json:"used_user_id"` // 使用用户ID + UsedAgentId string `json:"used_agent_id"` // 使用代理ID + UsedTime string `json:"used_time"` // 使用时间 + ExpireTime string `json:"expire_time"` // 过期时间 + Remark string `json:"remark"` // 备注 + CreateTime string `json:"create_time"` // 创建时间 + } + AdminGetInviteCodeListResp { + Total int64 `json:"total"` // 总数 + Items []InviteCodeListItem `json:"items"` // 列表数据 + } +) + diff --git a/app/main/api/desc/admin/admin_api.api b/app/main/api/desc/admin/admin_api.api new file mode 100644 index 0000000..87db11b --- /dev/null +++ b/app/main/api/desc/admin/admin_api.api @@ -0,0 +1,131 @@ +syntax = "v1" + +info( + title: "Admin API管理" + desc: "管理员API管理接口" + author: "team" + version: "v1" +) + +type ( + // API列表请求 + AdminGetApiListReq { + Page int64 `form:"page,default=1"` + PageSize int64 `form:"page_size,default=20"` + ApiName string `form:"api_name,optional"` + Method string `form:"method,optional"` + Status int64 `form:"status,optional"` + } + + // API列表响应 + AdminGetApiListResp { + Items []AdminApiInfo `json:"items"` + Total int64 `json:"total"` + } + + // API信息 + AdminApiInfo { + Id string `json:"id"` + ApiName string `json:"api_name"` + ApiCode string `json:"api_code"` + Method string `json:"method"` + Url string `json:"url"` + Status int64 `json:"status"` + Description string `json:"description"` + CreateTime string `json:"create_time"` + UpdateTime string `json:"update_time"` + } + + // API详情请求 + AdminGetApiDetailReq { + Id string `path:"id"` + } + + // API详情响应 + AdminGetApiDetailResp { + AdminApiInfo + } + + // 创建API请求 + AdminCreateApiReq { + ApiName string `json:"api_name"` + ApiCode string `json:"api_code"` + Method string `json:"method"` + Url string `json:"url"` + Status int64 `json:"status,default=1"` + Description string `json:"description,optional"` + } + + // 创建API响应 + AdminCreateApiResp { + Id string `json:"id"` + } + + // 更新API请求 + AdminUpdateApiReq { + Id string `path:"id"` + ApiName string `json:"api_name"` + ApiCode string `json:"api_code"` + Method string `json:"method"` + Url string `json:"url"` + Status int64 `json:"status"` + Description string `json:"description,optional"` + } + + // 更新API响应 + AdminUpdateApiResp { + Success bool `json:"success"` + } + + // 删除API请求 + AdminDeleteApiReq { + Id string `path:"id"` + } + + // 删除API响应 + AdminDeleteApiResp { + Success bool `json:"success"` + } + + // 批量更新API状态请求 + AdminBatchUpdateApiStatusReq { + Ids []string `json:"ids"` + Status int64 `json:"status"` + } + + // 批量更新API状态响应 + AdminBatchUpdateApiStatusResp { + Success bool `json:"success"` + } +) + +@server ( + prefix: api/v1 + group: admin_api + middleware: AdminAuthInterceptor +) +service main { + // 获取API列表 + @handler AdminGetApiList + get /admin/api/list (AdminGetApiListReq) returns (AdminGetApiListResp) + + // 获取API详情 + @handler AdminGetApiDetail + get /admin/api/detail/:id (AdminGetApiDetailReq) returns (AdminGetApiDetailResp) + + // 创建API + @handler AdminCreateApi + post /admin/api/create (AdminCreateApiReq) returns (AdminCreateApiResp) + + // 更新API + @handler AdminUpdateApi + put /admin/api/update/:id (AdminUpdateApiReq) returns (AdminUpdateApiResp) + + // 删除API + @handler AdminDeleteApi + delete /admin/api/delete/:id (AdminDeleteApiReq) returns (AdminDeleteApiResp) + + // 批量更新API状态 + @handler AdminBatchUpdateApiStatus + put /admin/api/batch-update-status (AdminBatchUpdateApiStatusReq) returns (AdminBatchUpdateApiStatusResp) +} diff --git a/app/main/api/desc/admin/admin_feature.api b/app/main/api/desc/admin/admin_feature.api new file mode 100644 index 0000000..03a7212 --- /dev/null +++ b/app/main/api/desc/admin/admin_feature.api @@ -0,0 +1,128 @@ +syntax = "v1" + +info ( + title: "后台功能管理服务" + desc: "后台功能管理相关接口" + version: "v1" +) + +// 功能管理接口 +@server ( + prefix: /api/v1/admin/feature + group: admin_feature + middleware: AdminAuthInterceptor +) +service main { + // 创建功能 + @handler AdminCreateFeature + post /create (AdminCreateFeatureReq) returns (AdminCreateFeatureResp) + + // 更新功能 + @handler AdminUpdateFeature + put /update/:id (AdminUpdateFeatureReq) returns (AdminUpdateFeatureResp) + + // 删除功能 + @handler AdminDeleteFeature + delete /delete/:id (AdminDeleteFeatureReq) returns (AdminDeleteFeatureResp) + + // 获取功能列表 + @handler AdminGetFeatureList + get /list (AdminGetFeatureListReq) returns (AdminGetFeatureListResp) + + // 获取功能详情 + @handler AdminGetFeatureDetail + get /detail/:id (AdminGetFeatureDetailReq) returns (AdminGetFeatureDetailResp) + + // 配置功能示例数据 + @handler AdminConfigFeatureExample + post /config-example (AdminConfigFeatureExampleReq) returns (AdminConfigFeatureExampleResp) + + // 查看功能示例数据 + @handler AdminGetFeatureExample + get /example/:feature_id (AdminGetFeatureExampleReq) returns (AdminGetFeatureExampleResp) +} + +type ( + // 创建功能请求 + AdminCreateFeatureReq { + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 + } + // 创建功能响应 + AdminCreateFeatureResp { + Id string `json:"id"` // 功能ID + } + // 更新功能请求 + AdminUpdateFeatureReq { + Id string `path:"id"` // 功能ID + ApiId *string `json:"api_id,optional"` // API标识 + Name *string `json:"name,optional"` // 描述 + } + // 更新功能响应 + AdminUpdateFeatureResp { + Success bool `json:"success"` // 是否成功 + } + // 删除功能请求 + AdminDeleteFeatureReq { + Id string `path:"id"` // 功能ID + } + // 删除功能响应 + AdminDeleteFeatureResp { + Success bool `json:"success"` // 是否成功 + } + // 获取功能列表请求 + AdminGetFeatureListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + ApiId *string `form:"api_id,optional"` // API标识 + Name *string `form:"name,optional"` // 描述 + } + // 功能列表项 + FeatureListItem { + Id string `json:"id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } + // 获取功能列表响应 + AdminGetFeatureListResp { + Total int64 `json:"total"` // 总数 + Items []FeatureListItem `json:"items"` // 列表数据 + } + // 获取功能详情请求 + AdminGetFeatureDetailReq { + Id string `path:"id"` // 功能ID + } + // 获取功能详情响应 + AdminGetFeatureDetailResp { + Id string `json:"id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } + // 配置功能示例数据请求 + AdminConfigFeatureExampleReq { + FeatureId string `json:"feature_id"` // 功能ID + Data string `json:"data"` // 示例数据JSON + } + // 配置功能示例数据响应 + AdminConfigFeatureExampleResp { + Success bool `json:"success"` // 是否成功 + } + // 查看功能示例数据请求 + AdminGetFeatureExampleReq { + FeatureId string `path:"feature_id"` // 功能ID + } + // 查看功能示例数据响应 + AdminGetFeatureExampleResp { + Id string `json:"id"` // 示例数据ID + FeatureId string `json:"feature_id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Data string `json:"data"` // 示例数据JSON + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } +) + diff --git a/app/main/api/desc/admin/admin_product.api b/app/main/api/desc/admin/admin_product.api new file mode 100644 index 0000000..c3ba2c7 --- /dev/null +++ b/app/main/api/desc/admin/admin_product.api @@ -0,0 +1,174 @@ +syntax = "v1" + +info ( + title: "后台产品管理服务" + desc: "后台产品管理相关接口" + version: "v1" +) + +// 产品管理接口 +@server( + prefix: /api/v1/admin/product + group: admin_product + middleware: AdminAuthInterceptor +) +service main { + // 创建产品 + @handler AdminCreateProduct + post /create (AdminCreateProductReq) returns (AdminCreateProductResp) + + // 更新产品 + @handler AdminUpdateProduct + put /update/:id (AdminUpdateProductReq) returns (AdminUpdateProductResp) + + // 删除产品 + @handler AdminDeleteProduct + delete /delete/:id (AdminDeleteProductReq) returns (AdminDeleteProductResp) + + // 获取产品列表 + @handler AdminGetProductList + get /list (AdminGetProductListReq) returns (AdminGetProductListResp) + + // 获取产品详情 + @handler AdminGetProductDetail + get /detail/:id (AdminGetProductDetailReq) returns (AdminGetProductDetailResp) + + // 获取产品功能列表 + @handler AdminGetProductFeatureList + get /feature/list/:product_id (AdminGetProductFeatureListReq) returns ([]AdminGetProductFeatureListResp) + + // 更新产品功能关联（批量） + @handler AdminUpdateProductFeatures + put /feature/update/:product_id (AdminUpdateProductFeaturesReq) returns (AdminUpdateProductFeaturesResp) +} + +type ( + // 创建产品请求 + AdminCreateProductReq { + ProductName string `json:"product_name"` // 服务名 + ProductEn string `json:"product_en"` // 英文名 + Description string `json:"description"` // 描述 + Notes string `json:"notes,optional"` // 备注 + CostPrice float64 `json:"cost_price"` // 成本 + SellPrice float64 `json:"sell_price"` // 售价 + } + + // 创建产品响应 + AdminCreateProductResp { + Id string `json:"id"` // 产品ID + } + + // 更新产品请求 + AdminUpdateProductReq { + Id string `path:"id"` // 产品ID + ProductName *string `json:"product_name,optional"` // 服务名 + ProductEn *string `json:"product_en,optional"` // 英文名 + Description *string `json:"description,optional"` // 描述 + Notes *string `json:"notes,optional"` // 备注 + CostPrice *float64 `json:"cost_price,optional"` // 成本 + SellPrice *float64 `json:"sell_price,optional"` // 售价 + } + + // 更新产品响应 + AdminUpdateProductResp { + Success bool `json:"success"` // 是否成功 + } + + // 删除产品请求 + AdminDeleteProductReq { + Id string `path:"id"` // 产品ID + } + + // 删除产品响应 + AdminDeleteProductResp { + Success bool `json:"success"` // 是否成功 + } + + // 获取产品列表请求 + AdminGetProductListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + ProductName *string `form:"product_name,optional"` // 服务名 + ProductEn *string `form:"product_en,optional"` // 英文名 + } + + // 产品列表项 + ProductListItem { + Id string `json:"id"` // 产品ID + ProductName string `json:"product_name"` // 服务名 + ProductEn string `json:"product_en"` // 英文名 + Description string `json:"description"` // 描述 + Notes string `json:"notes"` // 备注 + CostPrice float64 `json:"cost_price"` // 成本 + SellPrice float64 `json:"sell_price"` // 售价 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } + + // 获取产品列表响应 + AdminGetProductListResp { + Total int64 `json:"total"` // 总数 + Items []ProductListItem `json:"items"` // 列表数据 + } + + // 获取产品详情请求 + AdminGetProductDetailReq { + Id string `path:"id"` // 产品ID + } + + // 获取产品详情响应 + AdminGetProductDetailResp { + Id string `json:"id"` // 产品ID + ProductName string `json:"product_name"` // 服务名 + ProductEn string `json:"product_en"` // 英文名 + Description string `json:"description"` // 描述 + Notes string `json:"notes"` // 备注 + CostPrice float64 `json:"cost_price"` // 成本 + SellPrice float64 `json:"sell_price"` // 售价 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } + + // 获取产品功能列表请求 + AdminGetProductFeatureListReq { + ProductId string `path:"product_id"` // 产品ID + } + + // 获取产品功能列表响应Item + AdminGetProductFeatureListResp { + Id string `json:"id"` // 关联ID + ProductId string `json:"product_id"` // 产品ID + FeatureId string `json:"feature_id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 功能描述 + Sort int64 `json:"sort"` // 排序 + Enable int64 `json:"enable"` // 是否启用 + IsImportant int64 `json:"is_important"` // 是否重要 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } + + // // 获取产品功能列表响应 + // AdminGetProductFeatureListResp { + // Items []ProductFeatureListItem `json:"items"` // 列表数据 + // } + + // 产品功能关联项 + ProductFeatureItem { + FeatureId string `json:"feature_id"` // 功能ID + Sort int64 `json:"sort"` // 排序 + Enable int64 `json:"enable"` // 是否启用 + IsImportant int64 `json:"is_important"` // 是否重要 + } + + // 更新产品功能关联请求（批量） + AdminUpdateProductFeaturesReq { + ProductId string `path:"product_id"` // 产品ID + Features []ProductFeatureItem `json:"features"` // 功能列表 + } + + // 更新产品功能关联响应 + AdminUpdateProductFeaturesResp { + Success bool `json:"success"` // 是否成功 + } +) diff --git a/app/main/api/desc/admin/admin_query.api b/app/main/api/desc/admin/admin_query.api new file mode 100644 index 0000000..1d3bb41 --- /dev/null +++ b/app/main/api/desc/admin/admin_query.api @@ -0,0 +1,133 @@ +syntax = "v1" + +info ( + title: "查询服务" + desc: "查询服务" + version: "v1" +) + +@server ( + prefix: api/v1/admin/query + group: admin_query + middleware: AdminAuthInterceptor +) +service main { + @doc "获取查询详情" + @handler AdminGetQueryDetailByOrderId + get /detail/:order_id (AdminGetQueryDetailByOrderIdReq) returns (AdminGetQueryDetailByOrderIdResp) + + @doc "获取清理日志列表" + @handler AdminGetQueryCleanupLogList + get /cleanup/logs (AdminGetQueryCleanupLogListReq) returns (AdminGetQueryCleanupLogListResp) + + @doc "获取清理详情列表" + @handler AdminGetQueryCleanupDetailList + get /cleanup/details/:log_id (AdminGetQueryCleanupDetailListReq) returns (AdminGetQueryCleanupDetailListResp) + + @doc "获取清理配置列表" + @handler AdminGetQueryCleanupConfigList + get /cleanup/configs (AdminGetQueryCleanupConfigListReq) returns (AdminGetQueryCleanupConfigListResp) + + @doc "更新清理配置" + @handler AdminUpdateQueryCleanupConfig + put /cleanup/config (AdminUpdateQueryCleanupConfigReq) returns (AdminUpdateQueryCleanupConfigResp) +} + + type AdminGetQueryDetailByOrderIdReq { + OrderId string `path:"order_id"` +} + + type AdminGetQueryDetailByOrderIdResp { + Id string `json:"id"` // 主键ID + OrderId string `json:"order_id"` // 订单ID + UserId string `json:"user_id"` // 用户ID + ProductName string `json:"product_name"` // 产品ID + QueryParams map[string]interface{} `json:"query_params"` + QueryData []AdminQueryItem `json:"query_data"` + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + QueryState string `json:"query_state"` // 查询状态 +} + +type AdminQueryItem { + Feature interface{} `json:"feature"` + Data interface{} `json:"data"` // 这里可以是 map 或 具体的 struct +} + +// 清理日志相关请求响应定义 +type AdminGetQueryCleanupLogListReq { + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"page_size,default=20"` // 每页数量 + Status int64 `form:"status,optional"` // 状态：1-成功，2-失败 + StartTime string `form:"start_time,optional"` // 开始时间 + EndTime string `form:"end_time,optional"` // 结束时间 +} + +type AdminGetQueryCleanupLogListResp { + Total int64 `json:"total"` // 总数 + Items []QueryCleanupLogItem `json:"items"` // 列表 +} + + type QueryCleanupLogItem { + Id string `json:"id"` // 主键ID + CleanupTime string `json:"cleanup_time"` // 清理时间 + CleanupBefore string `json:"cleanup_before"` // 清理截止时间 + Status int64 `json:"status"` // 状态：1-成功，2-失败 + AffectedRows int64 `json:"affected_rows"` // 影响行数 + ErrorMsg string `json:"error_msg"` // 错误信息 + Remark string `json:"remark"` // 备注 + CreateTime string `json:"create_time"` // 创建时间 +} + +// 清理详情相关请求响应定义 + type AdminGetQueryCleanupDetailListReq { + LogId string `path:"log_id"` // 清理日志ID + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"page_size,default=20"` // 每页数量 +} + +type AdminGetQueryCleanupDetailListResp { + Total int64 `json:"total"` // 总数 + Items []QueryCleanupDetailItem `json:"items"` // 列表 +} + + type QueryCleanupDetailItem { + Id string `json:"id"` // 主键ID + CleanupLogId string `json:"cleanup_log_id"` // 清理日志ID + QueryId string `json:"query_id"` // 查询ID + OrderId string `json:"order_id"` // 订单ID + UserId string `json:"user_id"` // 用户ID + ProductName string `json:"product_name"` // 产品名称 + QueryState string `json:"query_state"` // 查询状态 + CreateTimeOld string `json:"create_time_old"` // 原创建时间 + CreateTime string `json:"create_time"` // 创建时间 +} + +// 清理配置相关请求响应定义 +type AdminGetQueryCleanupConfigListReq { + Status int64 `form:"status,optional"` // 状态：1-启用，0-禁用 +} + +type AdminGetQueryCleanupConfigListResp { + Items []QueryCleanupConfigItem `json:"items"` // 配置列表 +} + + type QueryCleanupConfigItem { + Id string `json:"id"` // 主键ID + ConfigKey string `json:"config_key"` // 配置键 + ConfigValue string `json:"config_value"` // 配置值 + ConfigDesc string `json:"config_desc"` // 配置描述 + Status int64 `json:"status"` // 状态：1-启用，0-禁用 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + + type AdminUpdateQueryCleanupConfigReq { + Id string `json:"id"` // 主键ID + ConfigValue string `json:"config_value"` // 配置值 + Status int64 `json:"status"` // 状态：1-启用，0-禁用 +} + +type AdminUpdateQueryCleanupConfigResp { + Success bool `json:"success"` // 是否成功 +} diff --git a/app/main/api/desc/admin/admin_role_api.api b/app/main/api/desc/admin/admin_role_api.api new file mode 100644 index 0000000..42322e2 --- /dev/null +++ b/app/main/api/desc/admin/admin_role_api.api @@ -0,0 +1,103 @@ +syntax = "v1" + +info( + title: "Admin 角色API权限管理" + desc: "管理员角色API权限管理接口" + author: "team" + version: "v1" +) + +type ( + // 获取角色API权限列表请求 + AdminGetRoleApiListReq { + RoleId string `path:"role_id"` + } + + // 获取角色API权限列表响应 + AdminGetRoleApiListResp { + Items []AdminRoleApiInfo `json:"items"` + } + + // 角色API权限信息 + AdminRoleApiInfo { + Id string `json:"id"` + RoleId string `json:"role_id"` + ApiId string `json:"api_id"` + ApiName string `json:"api_name"` + ApiCode string `json:"api_code"` + Method string `json:"method"` + Url string `json:"url"` + Status int64 `json:"status"` + Description string `json:"description"` + } + + // 分配角色API权限请求 + AdminAssignRoleApiReq { + RoleId string `json:"role_id"` + ApiIds []string `json:"api_ids"` + } + + // 分配角色API权限响应 + AdminAssignRoleApiResp { + Success bool `json:"success"` + } + + // 移除角色API权限请求 + AdminRemoveRoleApiReq { + RoleId string `json:"role_id"` + ApiIds []string `json:"api_ids"` + } + + // 移除角色API权限响应 + AdminRemoveRoleApiResp { + Success bool `json:"success"` + } + + // 更新角色API权限请求 + AdminUpdateRoleApiReq { + RoleId string `json:"role_id"` + ApiIds []string `json:"api_ids"` + } + + // 更新角色API权限响应 + AdminUpdateRoleApiResp { + Success bool `json:"success"` + } + + // 获取所有API列表（用于权限分配） + AdminGetAllApiListReq { + Status int64 `form:"status,optional,default=1"` + } + + // 获取所有API列表响应 + AdminGetAllApiListResp { + Items []AdminRoleApiInfo `json:"items"` + } +) + +@server ( + prefix: api/v1 + group: admin_role_api + middleware: AdminAuthInterceptor +) +service main { + // 获取角色API权限列表 + @handler AdminGetRoleApiList + get /admin/role/:role_id/api/list (AdminGetRoleApiListReq) returns (AdminGetRoleApiListResp) + + // 分配角色API权限 + @handler AdminAssignRoleApi + post /admin/role/api/assign (AdminAssignRoleApiReq) returns (AdminAssignRoleApiResp) + + // 移除角色API权限 + @handler AdminRemoveRoleApi + post /admin/role/api/remove (AdminRemoveRoleApiReq) returns (AdminRemoveRoleApiResp) + + // 更新角色API权限 + @handler AdminUpdateRoleApi + put /admin/role/api/update (AdminUpdateRoleApiReq) returns (AdminUpdateRoleApiResp) + + // 获取所有API列表（用于权限分配） + @handler AdminGetAllApiList + get /admin/api/all (AdminGetAllApiListReq) returns (AdminGetAllApiListResp) +} diff --git a/app/main/api/desc/admin/admin_user.api b/app/main/api/desc/admin/admin_user.api new file mode 100644 index 0000000..9dd8000 --- /dev/null +++ b/app/main/api/desc/admin/admin_user.api @@ -0,0 +1,144 @@ +syntax = "v1" + +info ( + title: "后台用户中心服务" + desc: "后台用户中心服务" + version: "v1" +) + +@server ( + prefix: api/v1/admin/user + group: admin_user + middleware: AdminAuthInterceptor +) +service main { + @doc "获取用户列表" + @handler AdminGetUserList + get /list (AdminGetUserListReq) returns (AdminGetUserListResp) + + @doc "获取用户详情" + @handler AdminGetUserDetail + get /detail/:id (AdminGetUserDetailReq) returns (AdminGetUserDetailResp) + + @doc "创建用户" + @handler AdminCreateUser + post /create (AdminCreateUserReq) returns (AdminCreateUserResp) + + @doc "更新用户" + @handler AdminUpdateUser + put /update/:id (AdminUpdateUserReq) returns (AdminUpdateUserResp) + + @doc "删除用户" + @handler AdminDeleteUser + delete /delete/:id (AdminDeleteUserReq) returns (AdminDeleteUserResp) + + @doc "用户信息" + @handler AdminUserInfo + get /info (AdminUserInfoReq) returns (AdminUserInfoResp) + + @doc "重置管理员密码" + @handler AdminResetPassword + put /reset-password/:id (AdminResetPasswordReq) returns (AdminResetPasswordResp) +} + +type ( + // 列表请求 + AdminGetUserListReq { + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"pageSize,default=20"` // 每页数量 + Username string `form:"username,optional"` // 用户名 + RealName string `form:"real_name,optional"` // 真实姓名 + Status int64 `form:"status,optional,default=-1"` // 状态：0-禁用，1-启用 + } + + // 列表响应 + AdminGetUserListResp { + Total int64 `json:"total"` // 总数 + Items []AdminUserListItem `json:"items"` // 列表 + } + + // 列表项 + AdminUserListItem { + Id string `json:"id"` // 用户ID + Username string `json:"username"` // 用户名 + RealName string `json:"real_name"` // 真实姓名 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + CreateTime string `json:"create_time"` // 创建时间 + RoleIds []string `json:"role_ids"` // 关联的角色ID列表 + } + + // 详情请求 + AdminGetUserDetailReq { + Id string `path:"id"` // 用户ID + } + + // 详情响应 + AdminGetUserDetailResp { + Id string `json:"id"` // 用户ID + Username string `json:"username"` // 用户名 + RealName string `json:"real_name"` // 真实姓名 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + RoleIds []string `json:"role_ids"` // 关联的角色ID列表 + } + + // 创建请求 + AdminCreateUserReq { + Username string `json:"username"` // 用户名 + RealName string `json:"real_name"` // 真实姓名 + Status int64 `json:"status,default=1"` // 状态：0-禁用，1-启用 + RoleIds []string `json:"role_ids"` // 关联的角色ID列表 + } + + // 创建响应 + AdminCreateUserResp { + Id string `json:"id"` // 用户ID + } + + // 更新请求 + AdminUpdateUserReq { + Id string `path:"id"` // 用户ID + Username *string `json:"username,optional"` // 用户名 + RealName *string `json:"real_name,optional"` // 真实姓名 + Status *int64 `json:"status,optional"` // 状态：0-禁用，1-启用 + RoleIds []string `json:"role_ids,optional"` // 关联的角色ID列表 + } + + // 更新响应 + AdminUpdateUserResp { + Success bool `json:"success"` // 是否成功 + } + + // 删除请求 + AdminDeleteUserReq { + Id string `path:"id"` // 用户ID + } + + // 删除响应 + AdminDeleteUserResp { + Success bool `json:"success"` // 是否成功 + } + + // 用户信息请求 + AdminUserInfoReq { + } + + // 用户信息响应 + AdminUserInfoResp { + Username string `json:"username"` // 用户名 + RealName string `json:"real_name"` // 真实姓名 + Roles []string `json:"roles"` // 角色编码列表 + } + + // 重置密码请求 + AdminResetPasswordReq { + Id string `path:"id"` // 用户ID + Password string `json:"password"` // 新密码 + } + + // 重置密码响应 + AdminResetPasswordResp { + Success bool `json:"success"` // 是否成功 + } +) diff --git a/app/main/api/desc/admin/auth.api b/app/main/api/desc/admin/auth.api new file mode 100644 index 0000000..71c79a9 --- /dev/null +++ b/app/main/api/desc/admin/auth.api @@ -0,0 +1,32 @@ +syntax = "v1" + +info ( + title: "认证中心服务" + desc: "认证中心服务" + version: "v1" +) + +@server ( + prefix: api/v1/admin/auth + group: admin_auth +) +service main { + @doc "登录" + @handler AdminLogin + post /login (AdminLoginReq) returns (AdminLoginResp) + +} + +type ( + AdminLoginReq { + Username string `json:"username" validate:"required"` + Password string `json:"password" validate:"required"` + Captcha bool `json:"captcha" validate:"required"` + } + AdminLoginResp { + AccessToken string `json:"access_token"` + AccessExpire int64 `json:"access_expire"` + RefreshAfter int64 `json:"refresh_after"` + Roles []string `json:"roles"` + } +) \ No newline at end of file diff --git a/app/main/api/desc/admin/menu.api b/app/main/api/desc/admin/menu.api new file mode 100644 index 0000000..b420c44 --- /dev/null +++ b/app/main/api/desc/admin/menu.api @@ -0,0 +1,136 @@ +syntax = "v1" + +info ( + title: "菜单中心服务" + desc: "菜单中心服务" + version: "v1" +) + +@server ( + prefix: api/v1/admin/menu + group: admin_menu + middleware: AdminAuthInterceptor +) +service main { + @doc "获取菜单列表" + @handler GetMenuList + get /list (GetMenuListReq) returns ([]MenuListItem) + + @doc "获取菜单详情" + @handler GetMenuDetail + get /detail/:id (GetMenuDetailReq) returns (GetMenuDetailResp) + + @doc "创建菜单" + @handler CreateMenu + post /create (CreateMenuReq) returns (CreateMenuResp) + + @doc "更新菜单" + @handler UpdateMenu + put /update/:id (UpdateMenuReq) returns (UpdateMenuResp) + + @doc "删除菜单" + @handler DeleteMenu + delete /delete/:id (DeleteMenuReq) returns (DeleteMenuResp) + + @doc "获取所有菜单(树形结构)" + @handler GetMenuAll + get /all (GetMenuAllReq) returns ([]GetMenuAllResp) +} + +type ( + // 列表请求 + GetMenuListReq { + Name string `form:"name,optional"` // 菜单名称 + Path string `form:"path,optional"` // 路由路径 + Status int64 `form:"status,optional,default=-1"` // 状态：0-禁用，1-启用 + Type string `form:"type,optional"` // 类型 + } + // 列表项 + MenuListItem { + Id string `json:"id"` // 菜单ID + Pid string `json:"pid"` // 父菜单ID + Name string `json:"name"` // 路由名称 + Path string `json:"path"` // 路由路径 + Component string `json:"component"` // 组件路径 + Redirect string `json:"redirect"` // 重定向路径 + Meta map[string]interface{} `json:"meta"` // 路由元数据 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + Type string `json:"type"` // 类型 + Sort int64 `json:"sort"` // 排序 + CreateTime string `json:"createTime"` // 创建时间 + Children []MenuListItem `json:"children"` // 子菜单 + } + // 详情请求 + GetMenuDetailReq { + Id string `path:"id"` // 菜单ID + } + // 详情响应 + GetMenuDetailResp { + Id string `json:"id"` // 菜单ID + Pid string `json:"pid"` // 父菜单ID + Name string `json:"name"` // 路由名称 + Path string `json:"path"` // 路由路径 + Component string `json:"component"` // 组件路径 + Redirect string `json:"redirect"` // 重定向路径 + Meta map[string]interface{} `json:"meta"` // 路由元数据 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + Type string `json:"type"` // 类型 + Sort int64 `json:"sort"` // 排序 + CreateTime string `json:"createTime"` // 创建时间 + UpdateTime string `json:"updateTime"` // 更新时间 + } + // 创建请求 + CreateMenuReq { + Pid string `json:"pid,optional"` // 父菜单ID + Name string `json:"name"` // 路由名称 + Path string `json:"path,optional"` // 路由路径 + Component string `json:"component,optional"` // 组件路径 + Redirect string `json:"redirect,optional"` // 重定向路径 + Meta map[string]interface{} `json:"meta"` // 路由元数据 + Status int64 `json:"status,optional,default=1"` // 状态：0-禁用，1-启用 + Type string `json:"type"` // 类型 + Sort int64 `json:"sort,optional"` // 排序 + } + // 创建响应 + CreateMenuResp { + Id string `json:"id"` // 菜单ID + } + // 更新请求 + UpdateMenuReq { + Id string `path:"id"` // 菜单ID + Pid *string `json:"pid,optional"` // 父菜单ID + Name string `json:"name"` // 路由名称 + Path string `json:"path,optional"` // 路由路径 + Component string `json:"component,optional"` // 组件路径 + Redirect string `json:"redirect,optional"` // 重定向路径 + Meta map[string]interface{} `json:"meta"` // 路由元数据 + Status int64 `json:"status,optional"` // 状态：0-禁用，1-启用 + Type string `json:"type"` // 类型 + Sort int64 `json:"sort,optional"` // 排序 + } + // 更新响应 + UpdateMenuResp { + Success bool `json:"success"` // 是否成功 + } + // 删除请求 + DeleteMenuReq { + Id string `path:"id"` // 菜单ID + } + // 删除响应 + DeleteMenuResp { + Success bool `json:"success"` // 是否成功 + } + // 获取所有菜单请求 + GetMenuAllReq {} + // 获取所有菜单响应 + GetMenuAllResp { + Name string `json:"name"` + Path string `json:"path"` + Redirect string `json:"redirect,omitempty"` + Component string `json:"component,omitempty"` + Sort int64 `json:"sort"` + Meta map[string]interface{} `json:"meta"` + Children []GetMenuAllResp `json:"children"` + } +) + diff --git a/app/main/api/desc/admin/notification.api b/app/main/api/desc/admin/notification.api new file mode 100644 index 0000000..18f0f7c --- /dev/null +++ b/app/main/api/desc/admin/notification.api @@ -0,0 +1,128 @@ +syntax = "v1" + +type ( + // 创建通知请求 + AdminCreateNotificationReq { + Title string `json:"title"` // 通知标题 + NotificationPage string `json:"notification_page"` // 通知页面 + Content string `json:"content"` // 通知内容 + StartDate string `json:"start_date"` // 生效开始日期（yyyy-MM-dd） + StartTime string `json:"start_time"` // 生效开始时间（HH:mm:ss） + EndDate string `json:"end_date"` // 生效结束日期（yyyy-MM-dd） + EndTime string `json:"end_time"` // 生效结束时间（HH:mm:ss） + Status int64 `json:"status"` // 状态：1-启用，0-禁用 + } + + // 创建通知响应 + AdminCreateNotificationResp { + Id string `json:"id"` // 通知ID + } + + // 更新通知请求 + AdminUpdateNotificationReq { + Id string `path:"id"` // 通知ID + Title *string `json:"title,optional"` // 通知标题 + Content *string `json:"content,optional"` // 通知内容 + NotificationPage *string `json:"notification_page,optional"` // 通知页面 + StartDate *string `json:"start_date,optional"` // 生效开始日期 + StartTime *string `json:"start_time,optional"` // 生效开始时间 + EndDate *string `json:"end_date,optional"` // 生效结束日期 + EndTime *string `json:"end_time,optional"` // 生效结束时间 + Status *int64 `json:"status,optional"` // 状态 + } + + // 更新通知响应 + AdminUpdateNotificationResp { + Success bool `json:"success"` // 是否成功 + } + + // 删除通知请求 + AdminDeleteNotificationReq { + Id string `path:"id"` // 通知ID + } + + // 删除通知响应 + AdminDeleteNotificationResp { + Success bool `json:"success"` // 是否成功 + } + + // 获取通知详情请求 + AdminGetNotificationDetailReq { + Id string `path:"id"` // 通知ID + } + + // 获取通知详情响应 + AdminGetNotificationDetailResp { + Id string `json:"id"` // 通知ID + Title string `json:"title"` // 通知标题 + Content string `json:"content"` // 通知内容 + NotificationPage string `json:"notification_page"` // 通知页面 + StartDate string `json:"start_date"` // 生效开始日期 + StartTime string `json:"start_time"` // 生效开始时间 + EndDate string `json:"end_date"` // 生效结束日期 + EndTime string `json:"end_time"` // 生效结束时间 + Status int64 `json:"status"` // 状态 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } + + // 获取通知列表请求 + AdminGetNotificationListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + Title *string `form:"title,optional"` // 通知标题（可选） + NotificationPage *string `form:"notification_page,optional"` // 通知页面（可选） + Status *int64 `form:"status,optional"` // 状态（可选） + StartDate *string `form:"start_date,optional"` // 开始日期范围（可选） + EndDate *string `form:"end_date,optional"` // 结束日期范围（可选） + } + + // 通知列表项 + NotificationListItem { + Id string `json:"id"` // 通知ID + Title string `json:"title"` // 通知标题 + NotificationPage string `json:"notification_page"` // 通知页面 + Content string `json:"content"` // 通知内容 + StartDate string `json:"start_date"` // 生效开始日期 + StartTime string `json:"start_time"` // 生效开始时间 + EndDate string `json:"end_date"` // 生效结束日期 + EndTime string `json:"end_time"` // 生效结束时间 + Status int64 `json:"status"` // 状态 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } + + // 获取通知列表响应 + AdminGetNotificationListResp { + Total int64 `json:"total"` // 总数 + Items []NotificationListItem `json:"items"` // 列表数据 + } +) + +// 通知管理接口 +@server( + prefix: /api/v1/admin/notification + group: admin_notification + middleware: AdminAuthInterceptor +) +service main { + // 创建通知 + @handler AdminCreateNotification + post /create (AdminCreateNotificationReq) returns (AdminCreateNotificationResp) + + // 更新通知 + @handler AdminUpdateNotification + put /update/:id (AdminUpdateNotificationReq) returns (AdminUpdateNotificationResp) + + // 删除通知 + @handler AdminDeleteNotification + delete /delete/:id (AdminDeleteNotificationReq) returns (AdminDeleteNotificationResp) + + // 获取通知详情 + @handler AdminGetNotificationDetail + get /detail/:id (AdminGetNotificationDetailReq) returns (AdminGetNotificationDetailResp) + + // 获取通知列表 + @handler AdminGetNotificationList + get /list (AdminGetNotificationListReq) returns (AdminGetNotificationListResp) +} diff --git a/app/main/api/desc/admin/order.api b/app/main/api/desc/admin/order.api new file mode 100644 index 0000000..47c7d1f --- /dev/null +++ b/app/main/api/desc/admin/order.api @@ -0,0 +1,168 @@ +syntax = "v1" + +info ( + title: "订单服务" + desc: "订单服务" + version: "v1" +) + +@server ( + prefix: api/v1/admin/order + group: admin_order + middleware: AdminAuthInterceptor +) +service main { + @doc "获取订单列表" + @handler AdminGetOrderList + get /list (AdminGetOrderListReq) returns (AdminGetOrderListResp) + + @doc "获取订单详情" + @handler AdminGetOrderDetail + get /detail/:id (AdminGetOrderDetailReq) returns (AdminGetOrderDetailResp) + + @doc "创建订单" + @handler AdminCreateOrder + post /create (AdminCreateOrderReq) returns (AdminCreateOrderResp) + + @doc "更新订单" + @handler AdminUpdateOrder + put /update/:id (AdminUpdateOrderReq) returns (AdminUpdateOrderResp) + + @doc "删除订单" + @handler AdminDeleteOrder + delete /delete/:id (AdminDeleteOrderReq) returns (AdminDeleteOrderResp) + + @doc "订单退款" + @handler AdminRefundOrder + post /refund/:id (AdminRefundOrderReq) returns (AdminRefundOrderResp) + + @doc "重新执行代理处理" + @handler AdminRetryAgentProcess + post /retry-agent-process/:id (AdminRetryAgentProcessReq) returns (AdminRetryAgentProcessResp) +} + +type ( + // 列表请求 + AdminGetOrderListReq { + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"pageSize,default=20"` // 每页数量 + OrderNo string `form:"order_no,optional"` // 商户订单号 + PlatformOrderId string `form:"platform_order_id,optional"` // 支付订单号 + ProductName string `form:"product_name,optional"` // 产品名称 + PaymentPlatform string `form:"payment_platform,optional"` // 支付方式 + PaymentScene string `form:"payment_scene,optional"` // 支付平台 + Amount float64 `form:"amount,optional"` // 金额 + Status string `form:"status,optional"` // 支付状态：pending-待支付，paid-已支付，refunded-已退款，closed-已关闭，failed-支付失败 + CreateTimeStart string `form:"create_time_start,optional"` // 创建时间开始 + CreateTimeEnd string `form:"create_time_end,optional"` // 创建时间结束 + PayTimeStart string `form:"pay_time_start,optional"` // 支付时间开始 + PayTimeEnd string `form:"pay_time_end,optional"` // 支付时间结束 + RefundTimeStart string `form:"refund_time_start,optional"` // 退款时间开始 + RefundTimeEnd string `form:"refund_time_end,optional"` // 退款时间结束 + } + // 列表响应 + AdminGetOrderListResp { + Total int64 `json:"total"` // 总数 + Items []OrderListItem `json:"items"` // 列表 + } + // 列表项 + OrderListItem { + Id string `json:"id"` // 订单ID + OrderNo string `json:"order_no"` // 商户订单号 + PlatformOrderId string `json:"platform_order_id"` // 支付订单号 + ProductName string `json:"product_name"` // 产品名称 + PaymentPlatform string `json:"payment_platform"` // 支付方式 + PaymentScene string `json:"payment_scene"` // 支付平台 + Amount float64 `json:"amount"` // 金额 + Status string `json:"status"` // 支付状态：pending-待支付，paid-已支付，refunded-已退款，closed-已关闭，failed-支付失败 + QueryState string `json:"query_state"` // 查询状态：pending-待查询，success-查询成功，failed-查询失败 processing-查询中 + CreateTime string `json:"create_time"` // 创建时间 + PayTime string `json:"pay_time"` // 支付时间 + RefundTime string `json:"refund_time"` // 退款时间 + IsAgentOrder bool `json:"is_agent_order"` // 是否是代理订单 + AgentProcessStatus string `json:"agent_process_status"` // 代理事务处理状态：not_agent-非代理订单，success-处理成功，failed-处理失败，pending-待处理 + } + // 详情请求 + AdminGetOrderDetailReq { + Id string `path:"id"` // 订单ID + } + // 详情响应 + AdminGetOrderDetailResp { + Id string `json:"id"` // 订单ID + OrderNo string `json:"order_no"` // 商户订单号 + PlatformOrderId string `json:"platform_order_id"` // 支付订单号 + ProductName string `json:"product_name"` // 产品名称 + PaymentPlatform string `json:"payment_platform"` // 支付方式 + PaymentScene string `json:"payment_scene"` // 支付平台 + Amount float64 `json:"amount"` // 金额 + Status string `json:"status"` // 支付状态：pending-待支付，paid-已支付，refunded-已退款，closed-已关闭，failed-支付失败 + QueryState string `json:"query_state"` // 查询状态：pending-待查询，success-查询成功，failed-查询失败 processing-查询中 + CreateTime string `json:"create_time"` // 创建时间 + PayTime string `json:"pay_time"` // 支付时间 + RefundTime string `json:"refund_time"` // 退款时间 + UpdateTime string `json:"update_time"` // 更新时间 + IsAgentOrder bool `json:"is_agent_order"` // 是否是代理订单 + AgentProcessStatus string `json:"agent_process_status"` // 代理事务处理状态：not_agent-非代理订单，success-处理成功，failed-处理失败，pending-待处理 + } + // 创建请求 + AdminCreateOrderReq { + OrderNo string `json:"order_no"` // 商户订单号 + PlatformOrderId string `json:"platform_order_id"` // 支付订单号 + ProductName string `json:"product_name"` // 产品名称 + PaymentPlatform string `json:"payment_platform"` // 支付方式 + PaymentScene string `json:"payment_scene"` // 支付平台 + Amount float64 `json:"amount"` // 金额 + Status string `json:"status,default=pending"` // 支付状态：pending-待支付，paid-已支付，refunded-已退款，closed-已关闭，failed-支付失败 + } + // 创建响应 + AdminCreateOrderResp { + Id string `json:"id"` // 订单ID + } + // 更新请求 + AdminUpdateOrderReq { + Id string `path:"id"` // 订单ID + OrderNo *string `json:"order_no,optional"` // 商户订单号 + PlatformOrderId *string `json:"platform_order_id,optional"` // 支付订单号 + ProductName *string `json:"product_name,optional"` // 产品名称 + PaymentPlatform *string `json:"payment_platform,optional"` // 支付方式 + PaymentScene *string `json:"payment_scene,optional"` // 支付平台 + Amount *float64 `json:"amount,optional"` // 金额 + Status *string `json:"status,optional"` // 支付状态：pending-待支付，paid-已支付，refunded-已退款，closed-已关闭，failed-支付失败 + PayTime *string `json:"pay_time,optional"` // 支付时间 + RefundTime *string `json:"refund_time,optional"` // 退款时间 + } + // 更新响应 + AdminUpdateOrderResp { + Success bool `json:"success"` // 是否成功 + } + // 删除请求 + AdminDeleteOrderReq { + Id string `path:"id"` // 订单ID + } + // 删除响应 + AdminDeleteOrderResp { + Success bool `json:"success"` // 是否成功 + } + // 退款请求 + AdminRefundOrderReq { + Id string `path:"id"` // 订单ID + RefundAmount float64 `json:"refund_amount"` // 退款金额 + RefundReason string `json:"refund_reason"` // 退款原因 + } + // 退款响应 + AdminRefundOrderResp { + Status string `json:"status"` // 退款状态 + RefundNo string `json:"refund_no"` // 退款单号 + Amount float64 `json:"amount"` // 退款金额 + } + // 重新执行代理处理请求 + AdminRetryAgentProcessReq { + Id string `path:"id"` // 订单ID + } + // 重新执行代理处理响应 + AdminRetryAgentProcessResp { + Status string `json:"status"` // 执行状态：success-成功，already_processed-已处理，failed-失败 + Message string `json:"message"` // 执行结果消息 + ProcessedAt string `json:"processed_at"` // 处理时间 + } +) diff --git a/app/main/api/desc/admin/platform_user.api b/app/main/api/desc/admin/platform_user.api new file mode 100644 index 0000000..1b3b56f --- /dev/null +++ b/app/main/api/desc/admin/platform_user.api @@ -0,0 +1,113 @@ +syntax = "v1" + +info ( + title: "平台用户管理" + desc: "平台用户管理" + version: "v1" +) + +// 平台用户管理接口 +@server ( + prefix: /api/v1/admin/platform_user + group: admin_platform_user + middleware: AdminAuthInterceptor +) +service main { + // 创建平台用户 + @handler AdminCreatePlatformUser + post /create (AdminCreatePlatformUserReq) returns (AdminCreatePlatformUserResp) + + // 更新平台用户 + @handler AdminUpdatePlatformUser + put /update/:id (AdminUpdatePlatformUserReq) returns (AdminUpdatePlatformUserResp) + + // 删除平台用户 + @handler AdminDeletePlatformUser + delete /delete/:id (AdminDeletePlatformUserReq) returns (AdminDeletePlatformUserResp) + + // 获取平台用户分页列表 + @handler AdminGetPlatformUserList + get /list (AdminGetPlatformUserListReq) returns (AdminGetPlatformUserListResp) + + // 获取平台用户详情 + @handler AdminGetPlatformUserDetail + get /detail/:id (AdminGetPlatformUserDetailReq) returns (AdminGetPlatformUserDetailResp) +} + +type ( + // 分页列表请求 + AdminGetPlatformUserListReq { + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"pageSize,default=20"` // 每页数量 + Mobile string `form:"mobile,optional"` // 手机号 + Nickname string `form:"nickname,optional"` // 昵称 + Inside int64 `form:"inside,optional"` // 是否内部用户 1-是 0-否 + CreateTimeStart string `form:"create_time_start,optional"` // 创建时间开始 + CreateTimeEnd string `form:"create_time_end,optional"` // 创建时间结束 + OrderBy string `form:"order_by,optional"` // 排序字段 + OrderType string `form:"order_type,optional"` // 排序类型 + } + // 分页列表响应 + AdminGetPlatformUserListResp { + Total int64 `json:"total"` // 总数 + Items []PlatformUserListItem `json:"items"` // 列表 + } + // 列表项 + PlatformUserListItem { + Id string `json:"id"` // 用户ID + Mobile string `json:"mobile"` // 手机号 + Nickname string `json:"nickname"` // 昵称 + Info string `json:"info"` // 备注信息 + Inside int64 `json:"inside"` // 是否内部用户 1-是 0-否 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } + // 详情请求 + AdminGetPlatformUserDetailReq { + Id string `path:"id"` // 用户ID + } + // 详情响应 + AdminGetPlatformUserDetailResp { + Id string `json:"id"` // 用户ID + Mobile string `json:"mobile"` // 手机号 + Nickname string `json:"nickname"` // 昵称 + Info string `json:"info"` // 备注信息 + Inside int64 `json:"inside"` // 是否内部用户 1-是 0-否 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + } + // 创建请求 + AdminCreatePlatformUserReq { + Mobile string `json:"mobile"` // 手机号 + Password string `json:"password"` // 密码 + Nickname string `json:"nickname"` // 昵称 + Info string `json:"info"` // 备注信息 + Inside int64 `json:"inside"` // 是否内部用户 1-是 0-否 + } + // 创建响应 + AdminCreatePlatformUserResp { + Id string `json:"id"` // 用户ID + } + // 更新请求 + AdminUpdatePlatformUserReq { + Id string `path:"id"` // 用户ID + Mobile *string `json:"mobile,optional"` // 手机号 + Password *string `json:"password,optional"` // 密码 + Nickname *string `json:"nickname,optional"` // 昵称 + Info *string `json:"info,optional"` // 备注信息 + Inside *int64 `json:"inside,optional"` // 是否内部用户 1-是 0-否 + } + // 更新响应 + AdminUpdatePlatformUserResp { + Success bool `json:"success"` // 是否成功 + } + // 删除请求 + AdminDeletePlatformUserReq { + Id string `path:"id"` // 用户ID + } + // 删除响应 + AdminDeletePlatformUserResp { + Success bool `json:"success"` // 是否成功 + } +) + diff --git a/app/main/api/desc/admin/role.api b/app/main/api/desc/admin/role.api new file mode 100644 index 0000000..b6c18e5 --- /dev/null +++ b/app/main/api/desc/admin/role.api @@ -0,0 +1,122 @@ +syntax = "v1" + +info ( + title: "角色服务" + desc: "角色服务" + version: "v1" +) + +@server ( + prefix: api/v1/admin/role + group: admin_role + middleware: AdminAuthInterceptor +) +service main { + @doc "获取角色列表" + @handler GetRoleList + get /list (GetRoleListReq) returns (GetRoleListResp) + + @doc "获取角色详情" + @handler GetRoleDetail + get /detail/:id (GetRoleDetailReq) returns (GetRoleDetailResp) + + @doc "创建角色" + @handler CreateRole + post /create (CreateRoleReq) returns (CreateRoleResp) + + @doc "更新角色" + @handler UpdateRole + put /update/:id (UpdateRoleReq) returns (UpdateRoleResp) + + @doc "删除角色" + @handler DeleteRole + delete /delete/:id (DeleteRoleReq) returns (DeleteRoleResp) +} + +type ( + // 列表请求 + GetRoleListReq { + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"pageSize,default=20"` // 每页数量 + Name string `form:"name,optional"` // 角色名称 + Code string `form:"code,optional"` // 角色编码 + Status int64 `form:"status,optional,default=-1"` // 状态：0-禁用，1-启用 + } + + // 列表响应 + GetRoleListResp { + Total int64 `json:"total"` // 总数 + Items []RoleListItem `json:"items"` // 列表 + } + + // 列表项 + RoleListItem { + Id string `json:"id"` // 角色ID + RoleName string `json:"role_name"` // 角色名称 + RoleCode string `json:"role_code"` // 角色编码 + Description string `json:"description"` // 角色描述 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + Sort int64 `json:"sort"` // 排序 + CreateTime string `json:"create_time"` // 创建时间 + MenuIds []string `json:"menu_ids"` // 关联的菜单ID列表 + } + + // 详情请求 + GetRoleDetailReq { + Id string `path:"id"` // 角色ID + } + + // 详情响应 + GetRoleDetailResp { + Id string `json:"id"` // 角色ID + RoleName string `json:"role_name"` // 角色名称 + RoleCode string `json:"role_code"` // 角色编码 + Description string `json:"description"` // 角色描述 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + Sort int64 `json:"sort"` // 排序 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + MenuIds []string `json:"menu_ids"` // 关联的菜单ID列表 + } + + // 创建请求 + CreateRoleReq { + RoleName string `json:"role_name"` // 角色名称 + RoleCode string `json:"role_code"` // 角色编码 + Description string `json:"description"` // 角色描述 + Status int64 `json:"status,default=1"` // 状态：0-禁用，1-启用 + Sort int64 `json:"sort,default=0"` // 排序 + MenuIds []string `json:"menu_ids"` // 关联的菜单ID列表 + } + + // 创建响应 + CreateRoleResp { + Id string `json:"id"` // 角色ID + } + + // 更新请求 + UpdateRoleReq { + Id string `path:"id"` // 角色ID + RoleName *string `json:"role_name,optional"` // 角色名称 + RoleCode *string `json:"role_code,optional"` // 角色编码 + Description *string `json:"description,optional"` // 角色描述 + Status *int64 `json:"status,optional"` // 状态：0-禁用，1-启用 + Sort *int64 `json:"sort,optional"` // 排序 + MenuIds []string `json:"menu_ids,optional"` // 关联的菜单ID列表 + } + + // 更新响应 + UpdateRoleResp { + Success bool `json:"success"` // 是否成功 + } + + // 删除请求 + DeleteRoleReq { + Id string `path:"id"` // 角色ID + } + + // 删除响应 + DeleteRoleResp { + Success bool `json:"success"` // 是否成功 + } +) diff --git a/app/main/api/desc/front/agent.api b/app/main/api/desc/front/agent.api new file mode 100644 index 0000000..a9d726b --- /dev/null +++ b/app/main/api/desc/front/agent.api @@ -0,0 +1,601 @@ +syntax = "v1" + +info ( + title: "代理服务" + desc: "新代理系统接口" + version: "v1" +) + +@server ( + prefix: api/v1/agent + group: agent +) +service main { + // 获取推广链接数据 + @handler GetLinkData + get /link (GetLinkDataReq) returns (GetLinkDataResp) +} + +@server ( + prefix: api/v1/agent + group: agent + middleware: AuthInterceptor +) +service main { + // 通过邀请码申请成为代理（必须提供邀请码） + @handler ApplyForAgent + post /apply (AgentApplyReq) returns (AgentApplyResp) + + // 通过邀请码注册（同时注册用户和代理） + @handler RegisterByInviteCode + post /register/invite (RegisterByInviteCodeReq) returns (RegisterByInviteCodeResp) +} + +type ( + GetLinkDataReq { + LinkIdentifier string `form:"link_identifier"` // 推广链接标识 + } + GetLinkDataResp { + AgentId string `json:"agent_id"` // 代理ID + ProductId string `json:"product_id"` // 产品ID + SetPrice float64 `json:"set_price"` // 代理设定价格 + ActualBasePrice float64 `json:"actual_base_price"` // 实际底价 + ProductName string `json:"product_name"` // 产品名称 + ProductEn string `json:"product_en"` // 产品英文标识 + SellPrice float64 `json:"sell_price"` // 销售价格（使用代理设定价格） + Description string `json:"description"` // 产品描述 + Features []Feature `json:"features"` // 产品功能列表 + } + AgentApplyReq { + Region string `json:"region,optional"` + Mobile string `json:"mobile"` + Code string `json:"code"` + Referrer string `json:"referrer"` + InviteCode string `json:"invite_code,optional"` + AgentCode int64 `json:"agent_code,optional"` + } + AgentApplyResp { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` + AgentCode int64 `json:"agent_code"` + } + // 通过邀请码注册 + RegisterByInviteCodeReq { + Referrer string `json:"referrer"` + InviteCode string `json:"invite_code,optional"` + AgentCode int64 `json:"agent_code,optional"` + Mobile string `json:"mobile"` + Code string `json:"code"` + Region string `json:"region,optional"` + WechatId string `json:"wechat_id,optional"` + } + RegisterByInviteCodeResp { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` + AgentId string `json:"agent_id"` // 代理ID + Level int64 `json:"level"` // 代理等级 + LevelName string `json:"level_name"` // 等级名称 + AgentCode int64 `json:"agent_code"` + } + // 生成邀请码 + GenerateInviteCodeReq { + Count int64 `json:"count"` // 生成数量 + ExpireDays int64 `json:"expire_days,optional"` // 过期天数（可选，0表示不过期） + Remark string `json:"remark,optional"` // 备注（可选） + } + GenerateInviteCodeResp { + Codes []string `json:"codes"` // 生成的邀请码列表 + } + // 获取邀请码列表 + GetInviteCodeListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + Status int64 `form:"status,optional"` // 状态（可选） + } + GetInviteCodeListResp { + Total int64 `json:"total"` // 总数 + List []InviteCodeItem `json:"list"` // 列表 + } + InviteCodeItem { + Id string `json:"id"` // 记录ID + Code string `json:"code"` // 邀请码 + TargetLevel int64 `json:"target_level"` // 目标等级 + Status int64 `json:"status"` // 状态：0=未使用，1=已使用，2=已失效 + UsedTime string `json:"used_time"` // 使用时间 + ExpireTime string `json:"expire_time"` // 过期时间 + Remark string `json:"remark"` // 备注 + CreateTime string `json:"create_time"` // 创建时间 + } + // 删除邀请码 + DeleteInviteCodeReq { + Id string `json:"id"` // 邀请码ID + } + DeleteInviteCodeResp {} + // 获取邀请链接 + GetInviteLinkReq { + InviteCode string `form:"invite_code"` // 邀请码 + TargetPath string `form:"target_path,optional"` // 目标地址（可选，默认为注册页面） + } + GetInviteLinkResp { + InviteLink string `json:"invite_link"` // 邀请链接 + } + // 获取代理等级特权信息 + GetLevelPrivilegeResp { + Levels []LevelPrivilegeItem `json:"levels"` + UpgradeToGoldFee float64 `json:"upgrade_to_gold_fee"` + UpgradeToDiamondFee float64 `json:"upgrade_to_diamond_fee"` + UpgradeToGoldRebate float64 `json:"upgrade_to_gold_rebate"` + UpgradeToDiamondRebate float64 `json:"upgrade_to_diamond_rebate"` + } + LevelPrivilegeItem { + Level int64 `json:"level"` // 等级：1=普通，2=黄金，3=钻石 + LevelName string `json:"level_name"` // 等级名称 + LevelBonus float64 `json:"level_bonus"` // 等级加成（元） + PriceReduction float64 `json:"price_reduction"` // 底价降低（元，相对于当前等级） + PromoteRebate string `json:"promote_rebate"` // 推广返佣说明 + MaxPromoteRebate float64 `json:"max_promote_rebate"` // 最高推广返佣金额（元） + UpgradeRebate string `json:"upgrade_rebate"` // 下级升级返佣说明 + Privileges []string `json:"privileges"` // 特权列表 + CanUpgradeSubordinate bool `json:"can_upgrade_subordinate"` // 是否可以升级下级 + } +) + +// ============================================ +// 需要登录的接口 +// ============================================ +@server ( + prefix: api/v1/agent + group: agent + jwt: JwtAuth + middleware: UserAuthInterceptor +) +service main { + // 查看代理信息 + @handler GetAgentInfo + get /info returns (AgentInfoResp) + + // 生成推广链接 + @handler GeneratingLink + post /generating_link (AgentGeneratingLinkReq) returns (AgentGeneratingLinkResp) + + // 获取产品配置 + @handler GetAgentProductConfig + get /product_config returns (AgentProductConfigResp) + + // 获取团队统计 + @handler GetTeamStatistics + get /team/statistics returns (TeamStatisticsResp) + + // 获取转化率统计 + @handler GetConversionRate + get /conversion/rate returns (ConversionRateResp) + + // 获取下级列表 + @handler GetSubordinateList + get /subordinate/list (GetSubordinateListReq) returns (GetSubordinateListResp) + + // 获取下级贡献详情 + @handler GetSubordinateContributionDetail + get /subordinate/contribution/detail (GetSubordinateContributionDetailReq) returns (GetSubordinateContributionDetailResp) + + // 获取团队列表 + @handler GetTeamList + get /team/list (GetTeamListReq) returns (GetTeamListResp) + + // 获取收益信息 + @handler GetRevenueInfo + get /revenue returns (GetRevenueInfoResp) + + // 获取佣金记录 + @handler GetCommissionList + get /commission/list (GetCommissionListReq) returns (GetCommissionListResp) + + // 获取返佣记录（推广返佣） + @handler GetRebateList + get /rebate/list (GetRebateListReq) returns (GetRebateListResp) + + // 获取升级返佣记录 + @handler GetUpgradeRebateList + get /rebate/upgrade/list (GetUpgradeRebateListReq) returns (GetUpgradeRebateListResp) + + // 获取升级记录 + @handler GetUpgradeList + get /upgrade/list (GetUpgradeListReq) returns (GetUpgradeListResp) + + // 申请升级 + @handler ApplyUpgrade + post /upgrade/apply (ApplyUpgradeReq) returns (ApplyUpgradeResp) + + // 钻石代理升级下级 + @handler UpgradeSubordinate + post /upgrade/subordinate (UpgradeSubordinateReq) returns (UpgradeSubordinateResp) + + // 获取提现列表 + @handler GetWithdrawalList + get /withdrawal/list (GetWithdrawalListReq) returns (GetWithdrawalListResp) + + // 申请提现 + @handler ApplyWithdrawal + post /withdrawal/apply (ApplyWithdrawalReq) returns (ApplyWithdrawalResp) + + // 实名认证 + @handler RealNameAuth + post /real_name (RealNameAuthReq) returns (RealNameAuthResp) + + // 生成邀请码 + @handler GenerateInviteCode + post /invite_code/generate (GenerateInviteCodeReq) returns (GenerateInviteCodeResp) + + // 获取邀请码列表 + @handler GetInviteCodeList + get /invite_code/list (GetInviteCodeListReq) returns (GetInviteCodeListResp) + + // 删除邀请码 + @handler DeleteInviteCode + post /invite_code/delete (DeleteInviteCodeReq) returns (DeleteInviteCodeResp) + + // 获取邀请链接（根据邀请码生成） + @handler GetInviteLink + get /invite_link (GetInviteLinkReq) returns (GetInviteLinkResp) + + // 获取代理等级特权信息 + @handler GetLevelPrivilege + get /level/privilege returns (GetLevelPrivilegeResp) + + // 获取推广查询报告列表 + @handler GetPromotionQueryList + get /promotion/query/list (GetPromotionQueryListReq) returns (GetPromotionQueryListResp) +} + +type ( + // 代理信息 + AgentInfoResp { + AgentId string `json:"agent_id"` + Level int64 `json:"level"` + LevelName string `json:"level_name"` + Region string `json:"region"` + Mobile string `json:"mobile"` + WechatId string `json:"wechat_id"` + TeamLeaderId string `json:"team_leader_id"` + IsRealName bool `json:"is_real_name"` + AgentCode int64 `json:"agent_code"` + } + // 生成推广链接 + AgentGeneratingLinkReq { + ProductId string `json:"product_id"` // 产品ID + SetPrice float64 `json:"set_price"` // 设定价格 + TargetPath string `json:"target_path,optional"` // 目标地址（可选，默认为推广报告页面） + } + AgentGeneratingLinkResp { + LinkIdentifier string `json:"link_identifier"` // 推广链接标识 + FullLink string `json:"full_link"` // 完整短链URL + } + // 产品配置 + AgentProductConfigResp { + List []ProductConfigItem `json:"list"` + } + ProductConfigItem { + ProductId string `json:"product_id"` // 产品ID + ProductName string `json:"product_name"` // 产品名称 + ProductEn string `json:"product_en"` // 产品英文标识 + ActualBasePrice float64 `json:"actual_base_price"` // 实际底价 + PriceRangeMin float64 `json:"price_range_min"` // 最低价格 + PriceRangeMax float64 `json:"price_range_max"` // 最高价格 + PriceThreshold float64 `json:"price_threshold"` // 提价标准阈值 + PriceFeeRate float64 `json:"price_fee_rate"` // 提价手续费比例 + } + // 团队统计 + TeamStatisticsResp { + TotalCount int64 `json:"total_count"` // 团队总人数（不包括自己） + DirectCount int64 `json:"direct_count"` // 直接下级数量 + IndirectCount int64 `json:"indirect_count"` // 间接下级数量 + GoldCount int64 `json:"gold_count"` // 黄金代理数量 + NormalCount int64 `json:"normal_count"` // 普通代理数量 + TodayNewMembers int64 `json:"today_new_members"` // 今日新增成员 + MonthNewMembers int64 `json:"month_new_members"` // 本月新增成员 + } + // 转化率统计 + ConversionRateResp { + MyConversionRate ConversionRateData `json:"my_conversion_rate"` // 我的转化率 + SubordinateConversionRate ConversionRateData `json:"subordinate_conversion_rate"` // 我的下级转化率 + } + ConversionRateData { + Daily []PeriodConversionData `json:"daily"` // 日统计（今日、昨日、前日） + Weekly []PeriodConversionData `json:"weekly"` // 周统计（这周、前一周、前两周） + Monthly []PeriodConversionData `json:"monthly"` // 月统计（本月、上月、前两月） + } + PeriodConversionData { + PeriodLabel string `json:"period_label"` // 时间段标签（如：今日、昨日、前日） + QueryUsers int64 `json:"query_users"` // 查询订单数（订单数量，不是用户数） + PaidUsers int64 `json:"paid_users"` // 付费订单数（订单数量，不是用户数） + TotalAmount float64 `json:"total_amount"` // 总金额（已支付订单的总金额） + QueryUserCount int64 `json:"query_user_count"` // 查询用户数（去重user_id后的用户数量） + PaidUserCount int64 `json:"paid_user_count"` // 付费用户数（去重user_id后的用户数量） + } + // 下级列表 + GetSubordinateListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + } + GetSubordinateListResp { + Total int64 `json:"total"` // 总数 + List []SubordinateItem `json:"list"` // 列表 + } + SubordinateItem { + AgentId string `json:"agent_id"` // 代理ID + Level int64 `json:"level"` // 等级 + LevelName string `json:"level_name"` // 等级名称 + Mobile string `json:"mobile"` // 手机号 + CreateTime string `json:"create_time"` // 创建时间 + TotalOrders int64 `json:"total_orders"` // 总订单数 + TotalAmount float64 `json:"total_amount"` // 总金额 + } + // 下级贡献详情 + GetSubordinateContributionDetailReq { + SubordinateId string `form:"subordinate_id"` // 下级代理ID + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + TabType string `form:"tab_type,optional"` // 标签页类型：order=订单列表，invite=邀请列表 + } + GetSubordinateContributionDetailResp { + Mobile string `json:"mobile"` // 手机号 + LevelName string `json:"level_name"` // 等级名称 + CreateTime string `json:"create_time"` // 创建时间 + OrderStats OrderStatistics `json:"order_stats"` // 订单统计（仅统计有返佣的订单） + RebateStats RebateStatistics `json:"rebate_stats"` // 返佣统计 + InviteStats InviteStatistics `json:"invite_stats"` // 邀请统计 + OrderList []OrderItem `json:"order_list"` // 订单列表（仅有贡献的订单） + InviteList []InviteItem `json:"invite_list"` // 邀请列表 + OrderListTotal int64 `json:"order_list_total"` // 订单列表总数 + InviteListTotal int64 `json:"invite_list_total"` // 邀请列表总数 + } + OrderStatistics { + TotalOrders int64 `json:"total_orders"` // 总订单量（仅统计有返佣的订单） + MonthOrders int64 `json:"month_orders"` // 月订单 + TodayOrders int64 `json:"today_orders"` // 今日订单 + } + RebateStatistics { + TotalRebateAmount float64 `json:"total_rebate_amount"` // 总贡献返佣金额 + TodayRebateAmount float64 `json:"today_rebate_amount"` // 今日贡献返佣金额 + MonthRebateAmount float64 `json:"month_rebate_amount"` // 月贡献返佣金额 + } + InviteStatistics { + TotalInvites int64 `json:"total_invites"` // 总的邀请 + TodayInvites int64 `json:"today_invites"` // 今日邀请 + MonthInvites int64 `json:"month_invites"` // 月邀请 + } + OrderItem { + OrderNo string `json:"order_no"` // 订单号 + ProductId string `json:"product_id"` // 产品ID + ProductName string `json:"product_name"` // 产品名称 + OrderAmount float64 `json:"order_amount"` // 订单金额 + RebateAmount float64 `json:"rebate_amount"` // 返佣金额 + CreateTime string `json:"create_time"` // 创建时间 + } + InviteItem { + AgentId string `json:"agent_id"` // 代理ID + Level int64 `json:"level"` // 等级 + LevelName string `json:"level_name"` // 等级名称 + Mobile string `json:"mobile"` // 手机号 + CreateTime string `json:"create_time"` // 邀请时间 + } + // 团队列表 + GetTeamListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + Mobile string `form:"mobile,optional"` // 手机号（可选，用于搜索） + } + GetTeamListResp { + Statistics TeamStatistics `json:"statistics"` // 统计数据 + Total int64 `json:"total"` // 总数 + List []TeamMemberItem `json:"list"` // 列表 + } + TeamStatistics { + TotalMembers int64 `json:"total_members"` // 团队成员总数 + TodayNewMembers int64 `json:"today_new_members"` // 团队人员今日新增 + MonthNewMembers int64 `json:"month_new_members"` // 团队人员月新增 + TotalQueries int64 `json:"total_queries"` // 团队总查询总量 + TodayPromotions int64 `json:"today_promotions"` // 团队今日推广量 + MonthPromotions int64 `json:"month_promotions"` // 团队月推广量 + TotalEarnings float64 `json:"total_earnings"` // 依靠团队得到的总收益 + TodayEarnings float64 `json:"today_earnings"` // 今日收益 + MonthEarnings float64 `json:"month_earnings"` // 月收益 + } + TeamMemberItem { + AgentId string `json:"agent_id"` // 代理ID + Level int64 `json:"level"` // 等级 + LevelName string `json:"level_name"` // 等级名称 + Mobile string `json:"mobile"` // 手机号 + CreateTime string `json:"create_time"` // 加入团队时间 + TodayQueries int64 `json:"today_queries"` // 当日查询量 + TotalQueries int64 `json:"total_queries"` // 总查询量 + TotalRebateAmount float64 `json:"total_rebate_amount"` // 返佣给我的总金额 + TodayRebateAmount float64 `json:"today_rebate_amount"` // 返佣给我的今日金额 + TotalInvites int64 `json:"total_invites"` // 邀请加入团队的总人数 + TodayInvites int64 `json:"today_invites"` // 邀请加入团队的今日人数 + IsDirect bool `json:"is_direct"` // 是否直接下级 + } + // 收益信息 + GetRevenueInfoResp { + Balance float64 `json:"balance"` // 可用余额 + FrozenBalance float64 `json:"frozen_balance"` // 冻结余额 + TotalEarnings float64 `json:"total_earnings"` // 累计收益（钱包总收益） + WithdrawnAmount float64 `json:"withdrawn_amount"` // 累计提现 + CommissionTotal float64 `json:"commission_total"` // 佣金累计总收益（推广订单获得的佣金） + CommissionToday float64 `json:"commission_today"` // 佣金今日收益 + CommissionMonth float64 `json:"commission_month"` // 佣金本月收益 + RebateTotal float64 `json:"rebate_total"` // 返佣累计总收益（包括推广返佣和升级返佣） + RebateToday float64 `json:"rebate_today"` // 返佣今日收益 + RebateMonth float64 `json:"rebate_month"` // 返佣本月收益 + } + // 佣金记录 + GetCommissionListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + } + GetCommissionListResp { + Total int64 `json:"total"` // 总数 + List []CommissionItem `json:"list"` // 列表 + } + CommissionItem { + Id string `json:"id"` // 记录ID + OrderId string `json:"order_id"` // 订单ID + OrderNo string `json:"order_no"` // 订单号 + ProductName string `json:"product_name"` // 产品名称 + Amount float64 `json:"amount"` // 佣金金额 + Status int64 `json:"status"` // 状态：1=已发放，2=已冻结，3=已取消 + CreateTime string `json:"create_time"` // 创建时间 + } + // 返佣记录（推广返佣） + GetRebateListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + RebateType *int64 `form:"rebate_type,optional"` // 返佣类型（可选）：1=直接上级返佣，2=钻石上级返佣，3=黄金上级返佣 + } + GetRebateListResp { + Total int64 `json:"total"` // 总数 + List []RebateItem `json:"list"` // 列表 + } + RebateItem { + Id string `json:"id"` // 记录ID + SourceAgentId string `json:"source_agent_id"` // 来源代理ID + SourceAgentMobile string `json:"source_agent_mobile"` // 来源代理手机号 + SourceAgentLevel int64 `json:"source_agent_level"` // 来源代理等级：1=普通，2=黄金，3=钻石 + OrderId string `json:"order_id"` // 订单ID + OrderNo string `json:"order_no"` // 订单号 + RebateType int64 `json:"rebate_type"` // 返佣类型：1=直接上级，2=钻石上级，3=黄金上级 + Amount float64 `json:"amount"` // 返佣金额 + CreateTime string `json:"create_time"` // 创建时间 + } + // 升级返佣记录 + GetUpgradeRebateListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + } + GetUpgradeRebateListResp { + Total int64 `json:"total"` // 总数 + List []UpgradeRebateItem `json:"list"` // 列表 + } + UpgradeRebateItem { + Id string `json:"id"` // 记录ID + SourceAgentId string `json:"source_agent_id"` // 来源代理ID（升级的代理） + SourceAgentMobile string `json:"source_agent_mobile"` // 来源代理手机号 + OrderNo string `json:"order_no"` // 订单号 + FromLevel int64 `json:"from_level"` // 原等级 + ToLevel int64 `json:"to_level"` // 目标等级 + Amount float64 `json:"amount"` // 返佣金额 + CreateTime string `json:"create_time"` // 创建时间 + } + // 升级记录 + GetUpgradeListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + } + GetUpgradeListResp { + Total int64 `json:"total"` // 总数 + List []UpgradeItem `json:"list"` // 列表 + } + UpgradeItem { + Id string `json:"id"` // 记录ID + AgentId string `json:"agent_id"` // 代理ID + FromLevel int64 `json:"from_level"` // 原等级 + ToLevel int64 `json:"to_level"` // 目标等级 + UpgradeType int64 `json:"upgrade_type"` // 升级类型：1=自主付费，2=钻石升级 + UpgradeFee float64 `json:"upgrade_fee"` // 升级费用 + RebateAmount float64 `json:"rebate_amount"` // 返佣金额 + Status int64 `json:"status"` // 状态：1=待支付，2=已支付，3=已完成，4=已取消 + CreateTime string `json:"create_time"` // 创建时间 + } + // 申请升级 + ApplyUpgradeReq { + ToLevel int64 `json:"to_level"` // 目标等级：2=黄金，3=钻石 + } + ApplyUpgradeResp { + UpgradeId string `json:"upgrade_id"` // 升级记录ID + OrderNo string `json:"order_no"` // 支付订单号 + } + // 钻石升级下级 + UpgradeSubordinateReq { + SubordinateId string `json:"subordinate_id"` // 下级代理ID + ToLevel int64 `json:"to_level"` // 目标等级（只能是2=黄金） + } + UpgradeSubordinateResp { + Success bool `json:"success"` + } + // 提现列表 + GetWithdrawalListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + } + GetWithdrawalListResp { + Total int64 `json:"total"` // 总数 + List []WithdrawalItem `json:"list"` // 列表 + } + WithdrawalItem { + Id string `json:"id"` // 记录ID + WithdrawalNo string `json:"withdrawal_no"` // 提现单号 + Amount float64 `json:"amount"` // 提现金额 + TaxAmount float64 `json:"tax_amount"` // 税费金额 + ActualAmount float64 `json:"actual_amount"` // 实际到账金额 + Status int64 `json:"status"` // 状态：1=待审核，2=审核通过，3=审核拒绝，4=提现中，5=提现成功，6=提现失败 + PayeeAccount string `json:"payee_account"` // 收款账户 + PayeeName string `json:"payee_name"` // 收款人姓名 + Remark string `json:"remark"` // 备注 + CreateTime string `json:"create_time"` // 创建时间 + } + // 申请提现 + ApplyWithdrawalReq { + Amount float64 `json:"amount"` // 提现金额 + PayeeAccount string `json:"payee_account"` // 收款账户 + PayeeName string `json:"payee_name"` // 收款人姓名 + } + ApplyWithdrawalResp { + WithdrawalId string `json:"withdrawal_id"` // 提现记录ID + WithdrawalNo string `json:"withdrawal_no"` // 提现单号 + } + // 实名认证 + RealNameAuthReq { + Name string `json:"name"` // 姓名 + IdCard string `json:"id_card"` // 身份证号 + Mobile string `json:"mobile"` // 手机号 + Code string `json:"code"` // 验证码 + } + RealNameAuthResp { + Status string `json:"status"` // 状态：pending=待审核，approved=已通过，rejected=已拒绝 + } + // 推广查询列表 + GetPromotionQueryListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + } + GetPromotionQueryListResp { + Total int64 `json:"total"` // 总数 + List []PromotionQueryItem `json:"list"` // 列表 + } + PromotionQueryItem { + Id string `json:"id"` // 查询ID + OrderId string `json:"order_id"` // 订单ID + ProductName string `json:"product_name"` // 产品名称 + CreateTime string `json:"create_time"` // 创建时间 + QueryState string `json:"query_state"` // 查询状态 + } +) + +// ============================================ +// 短链重定向接口（公开访问，不需要认证和api前缀） +// ============================================ +@server ( + group: agent +) +service main { + // 短链重定向 + @handler ShortLinkRedirect + get /s/:shortCode returns (ShortLinkRedirectResp) +} + +type ( + ShortLinkRedirectResp {} +) + diff --git a/app/main/api/desc/front/app.api b/app/main/api/desc/front/app.api new file mode 100644 index 0000000..73a830c --- /dev/null +++ b/app/main/api/desc/front/app.api @@ -0,0 +1,47 @@ +syntax = "v1" + +info ( + title: "APP服务" + desc: "APP服务" + version: "v1" +) + +@server ( + prefix: api/v1 + group: app +) +service main { + @doc ( + summary: "心跳检测接口" + ) + @handler healthCheck + get /health/check returns (HealthCheckResp) + + @handler getAppVersion + get /app/version returns (getAppVersionResp) + + @handler getAppConfig + get /app/config returns (getAppConfigResp) +} + +type ( + // 心跳检测响应 + HealthCheckResp { + Status string `json:"status"` // 服务状态 + Message string `json:"message"` // 状态信息 + } +) + +type ( + getAppVersionResp { + Version string `json:"version"` + WgtUrl string `json:"wgtUrl"` + } +) + +type ( + getAppConfigResp { + QueryRetentionDays int64 `json:"query_retention_days"` + } +) + diff --git a/app/main/api/desc/front/authorization.api b/app/main/api/desc/front/authorization.api new file mode 100644 index 0000000..3886183 --- /dev/null +++ b/app/main/api/desc/front/authorization.api @@ -0,0 +1,69 @@ +type ( + // GetAuthorizationDocumentReq 获取授权书请求 + GetAuthorizationDocumentReq { + DocumentId string `json:"documentId" validate:"required"` // 授权书ID + } + // GetAuthorizationDocumentResp 获取授权书响应 + GetAuthorizationDocumentResp { + DocumentId string `json:"documentId"` // 授权书ID + UserId string `json:"userId"` // 用户ID + OrderId string `json:"orderId"` // 订单ID + QueryId string `json:"queryId"` // 查询ID + FileName string `json:"fileName"` // 文件名 + FileUrl string `json:"fileUrl"` // 文件访问URL + FileSize int64 `json:"fileSize"` // 文件大小 + FileType string `json:"fileType"` // 文件类型 + Status string `json:"status"` // 状态 + CreateTime string `json:"createTime"` // 创建时间 + } + // GetAuthorizationDocumentByOrderReq 根据订单ID获取授权书请求 + GetAuthorizationDocumentByOrderReq { + OrderId string `json:"orderId" validate:"required"` // 订单ID + } + // GetAuthorizationDocumentByOrderResp 根据订单ID获取授权书响应 + GetAuthorizationDocumentByOrderResp { + Documents []AuthorizationDocumentInfo `json:"documents"` // 授权书列表 + } + // AuthorizationDocumentInfo 授权书信息 + AuthorizationDocumentInfo { + DocumentId string `json:"documentId"` // 授权书ID + UserId string `json:"userId"` // 用户ID + OrderId string `json:"orderId"` // 订单ID + QueryId string `json:"queryId"` // 查询ID + FileName string `json:"fileName"` // 文件名 + FileUrl string `json:"fileUrl"` // 文件访问URL + FileSize int64 `json:"fileSize"` // 文件大小 + FileType string `json:"fileType"` // 文件类型 + Status string `json:"status"` // 状态 + CreateTime string `json:"createTime"` // 创建时间 + } + // DownloadAuthorizationDocumentReq 下载授权书请求 + DownloadAuthorizationDocumentReq { + DocumentId string `json:"documentId" validate:"required"` // 授权书ID + } + // DownloadAuthorizationDocumentResp 下载授权书响应 + DownloadAuthorizationDocumentResp { + FileName string `json:"fileName"` // 文件名 + FileUrl string `json:"fileUrl"` // 文件访问URL + } +) + +// 授权书相关接口 +@server ( + prefix: api/v1 + group: authorization +) +service main { + // 获取授权书信息 + @handler GetAuthorizationDocument + get /authorization/document/:documentId (GetAuthorizationDocumentReq) returns (GetAuthorizationDocumentResp) + + // 根据订单ID获取授权书列表 + @handler GetAuthorizationDocumentByOrder + get /authorization/document/order/:orderId (GetAuthorizationDocumentByOrderReq) returns (GetAuthorizationDocumentByOrderResp) + + // 下载授权书文件 + @handler DownloadAuthorizationDocument + get /authorization/download/:documentId (DownloadAuthorizationDocumentReq) returns (DownloadAuthorizationDocumentResp) +} + diff --git a/app/main/api/desc/front/pay.api b/app/main/api/desc/front/pay.api new file mode 100644 index 0000000..afc4f93 --- /dev/null +++ b/app/main/api/desc/front/pay.api @@ -0,0 +1,71 @@ +syntax = "v1" + +info ( + title: "支付服务" + desc: "支付服务" + version: "v1" +) + +@server ( + prefix: api/v1 + group: pay +) +service main { + // 微信支付回调 + @handler WechatPayCallback + post /pay/wechat/callback + + // 支付宝支付回调 + @handler AlipayCallback + post /pay/alipay/callback + + // 微信退款回调 + @handler WechatPayRefundCallback + post /pay/wechat/refund_callback +} + +@server ( + prefix: api/v1 + group: pay + jwt: JwtAuth + middleware: AuthInterceptor +) +service main { + // 支付 + @handler Payment + post /pay/payment (PaymentReq) returns (PaymentResp) + + @handler IapCallback + post /pay/iap_callback (IapCallbackReq) + + @handler PaymentCheck + post /pay/check (PaymentCheckReq) returns (PaymentCheckResp) +} + +type ( + PaymentReq { + Id string `json:"id"` + PayMethod string `json:"pay_method"` // 支付方式: wechat, alipay, appleiap, test(仅开发环境), test_empty(仅开发环境-空报告模式) + PayType string `json:"pay_type" validate:"required,oneof=query agent_vip agent_upgrade"` + } + PaymentResp { + PrepayData interface{} `json:"prepay_data"` + PrepayId string `json:"prepay_id"` + OrderNo string `json:"order_no"` + } + PaymentCheckReq { + OrderNo string `json:"order_no" validate:"required"` + } + PaymentCheckResp { + Type string `json:"type"` + Status string `json:"status"` + } +) + +type ( + IapCallbackReq { + OrderID string `json:"order_id" validate:"required"` + TransactionReceipt string `json:"transaction_receipt" validate:"required"` + } +) + diff --git a/app/main/api/desc/front/product.api b/app/main/api/desc/front/product.api new file mode 100644 index 0000000..8b6549f --- /dev/null +++ b/app/main/api/desc/front/product.api @@ -0,0 +1,55 @@ +syntax = "v1" + +info ( + title: "产品服务" + desc: "产品服务" + version: "v1" +) +type Feature { + ID string `json:"id"` // 功能ID + ApiID string `json:"api_id"` // API标识 + Name string `json:"name"` // 功能描述 +} +// 产品基本类型定义 +type Product { + ProductName string `json:"product_name"` + ProductEn string `json:"product_en"` + Description string `json:"description"` + Notes string `json:"notes,optional"` + SellPrice float64 `json:"sell_price"` + Features []Feature `json:"features"` // 关联功能列表 +} + +@server ( + prefix: api/v1/product + group: product + +) +service main { + @handler GetProductByID + get /:id (GetProductByIDRequest) returns (ProductResponse) + + @handler GetProductByEn + get /en/:product_en (GetProductByEnRequest) returns (ProductResponse) +} + +type GetProductByIDRequest { + Id string `path:"id"` +} + +type GetProductByEnRequest { + ProductEn string `path:"product_en"` +} + +type ProductResponse { + Product +} + +@server ( + prefix: api/v1/product + group: product +) +service main { + @handler GetProductAppByEn + get /app_en/:product_en (GetProductByEnRequest) returns (ProductResponse) +} diff --git a/app/main/api/desc/front/query.api b/app/main/api/desc/front/query.api new file mode 100644 index 0000000..7a71035 --- /dev/null +++ b/app/main/api/desc/front/query.api @@ -0,0 +1,219 @@ +syntax = "v1" + +info ( + title: "产品查询服务" + desc: "产品查询服务" + version: "v1" +) + +//============================> query v1 <============================ +// 查询基本类型定义 +type Query { + Id string `json:"id"` // 主键ID + OrderId string `json:"order_id"` // 订单ID + UserId string `json:"user_id"` // 用户ID + Product string `json:"product"` // 产品ID + ProductName string `json:"product_name"` // 产品ID + QueryParams map[string]interface{} `json:"query_params"` + QueryData []QueryItem `json:"query_data"` + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + QueryState string `json:"query_state"` // 查询状态 +} + +type QueryItem { + Feature interface{} `json:"feature"` + Data interface{} `json:"data"` // 这里可以是 map 或 具体的 struct +} + +@server ( + prefix: api/v1 + group: query + middleware: AuthInterceptor +) +service main { + @doc "query service agent" + @handler queryServiceAgent + post /query/service_agent/:product (QueryServiceReq) returns (QueryServiceResp) + + @handler queryServiceApp + post /query/service_app/:product (QueryServiceReq) returns (QueryServiceResp) +} + +type ( + QueryReq { + Data string `json:"data" validate:"required"` + } + QueryResp { + Id string `json:"id"` + } +) + +type ( + QueryServiceReq { + Product string `path:"product"` + Data string `json:"data" validate:"required"` + AgentIdentifier string `json:"agent_identifier,optional"` + App bool `json:"app,optional"` + } + QueryServiceResp { + Id string `json:"id"` + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` + } +) + +@server ( + prefix: api/v1 + group: query + jwt: JwtAuth + middleware: AuthInterceptor +) +service main { + @doc "query service" + @handler queryService + post /query/service/:product (QueryServiceReq) returns (QueryServiceResp) +} + +@server ( + prefix: api/v1 + group: query + jwt: JwtAuth + middleware: AuthInterceptor +) +service main { + @doc "获取查询临时订单" + @handler queryProvisionalOrder + get /query/provisional_order/:id (QueryProvisionalOrderReq) returns (QueryProvisionalOrderResp) + + @doc "查询列表" + @handler queryList + get /query/list (QueryListReq) returns (QueryListResp) + + @doc "查询详情 按订单号 付款查询时" + @handler queryDetailByOrderId + get /query/orderId/:order_id (QueryDetailByOrderIdReq) returns (string) + + @doc "查询详情 按订单号" + @handler queryDetailByOrderNo + get /query/orderNo/:order_no (QueryDetailByOrderNoReq) returns (string) + + @doc "重试查询" + @handler queryRetry + post /query/retry/:id (QueryRetryReq) returns (QueryRetryResp) + + @doc "更新查询数据" + @handler updateQueryData + post /query/update_data (UpdateQueryDataReq) returns (UpdateQueryDataResp) + + @doc "生成分享链接" + @handler QueryGenerateShareLink + post /query/generate_share_link (QueryGenerateShareLinkReq) returns (QueryGenerateShareLinkResp) +} + +type ( + QueryGenerateShareLinkReq { + OrderId *string `json:"order_id,optional"` + OrderNo *string `json:"order_no,optional"` + } + QueryGenerateShareLinkResp { + ShareLink string `json:"share_link"` + } +) + +// 获取查询临时订单 +type ( + QueryProvisionalOrderReq { + Id string `path:"id"` + } + QueryProvisionalOrderResp { + Name string `json:"name"` + IdCard string `json:"id_card"` + Mobile string `json:"mobile"` + Product Product `json:"product"` + } +) + +type ( + QueryListReq { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数据量 + } + QueryListResp { + Total int64 `json:"total"` // 总记录数 + List []Query `json:"list"` // 查询列表 + } +) + +type ( + QueryExampleReq { + Feature string `form:"feature"` + } +) + +type ( + QueryDetailByOrderIdReq { + OrderId string `path:"order_id"` + } +) + +type ( + QueryDetailByOrderNoReq { + OrderNo string `path:"order_no"` + } +) + +type ( + QueryRetryReq { + Id string `path:"id"` + } + QueryRetryResp { + Query + } +) + +type ( + UpdateQueryDataReq { + Id string `json:"id"` // 查询ID + QueryData string `json:"query_data"` // 查询数据(未加密的JSON) + } + UpdateQueryDataResp { + Id string `json:"id"` + UpdatedAt string `json:"updated_at"` // 更新时间 + } +) + +@server ( + prefix: api/v1 + group: query +) +service main { + @handler querySingleTest + post /query/single/test (QuerySingleTestReq) returns (QuerySingleTestResp) + + @doc "查询详情" + @handler queryShareDetail + get /query/share/:id (QueryShareDetailReq) returns (string) + + @doc "查询示例" + @handler queryExample + get /query/example (QueryExampleReq) returns (string) +} + +type ( + QueryShareDetailReq { + Id string `path:"id"` + } +) + +type QuerySingleTestReq { + Params map[string]interface{} `json:"params"` + Api string `json:"api"` +} + +type QuerySingleTestResp { + Data interface{} `json:"data"` + Api string `json:"api"` +} + diff --git a/app/main/api/desc/front/user.api b/app/main/api/desc/front/user.api new file mode 100644 index 0000000..683c887 --- /dev/null +++ b/app/main/api/desc/front/user.api @@ -0,0 +1,192 @@ +syntax = "v1" + +info ( + title: "用户中心服务" + desc: "用户中心服务" + version: "v1" +) + +//============================> user v1 <============================ +// 用户基本类型定义 +type User { + Id string `json:"id"` + Mobile string `json:"mobile"` + NickName string `json:"nickName"` + UserType int64 `json:"userType"` +} + +//no need login +@server ( + prefix: api/v1 + group: user +) +service main { + @doc "unified auth" + @handler auth + post /user/auth (AuthReq) returns (AuthResp) + @doc "mobile code login" + @handler mobileCodeLogin + post /user/mobileCodeLogin (MobileCodeLoginReq) returns (MobileCodeLoginResp) + + @doc "wechat mini auth" + @handler wxMiniAuth + post /user/wxMiniAuth (WXMiniAuthReq) returns (WXMiniAuthResp) + + @doc "wechat h5 auth" + @handler wxH5Auth + post /user/wxh5Auth (WXH5AuthReq) returns (WXH5AuthResp) + + @handler getSignature + post /wechat/getSignature (GetSignatureReq) returns (GetSignatureResp) +} + +type ( + AuthReq { + Platform string `json:"platform"` // browser|wxh5|wxmini + Code string `json:"code,optional"` + } + AuthResp { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` + UserType int64 `json:"userType"` + HasMobile bool `json:"hasMobile"` + IsAgent bool `json:"isAgent"` + } + MobileCodeLoginReq { + Mobile string `json:"mobile"` + Code string `json:"code" validate:"required"` + } + MobileCodeLoginResp { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` + } +) + +type ( + WXMiniAuthReq { + Code string `json:"code"` + } + WXMiniAuthResp { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` + } +) + +type ( + GetSignatureReq { + Url string `json:"url"` + } + GetSignatureResp { + AppId string `json:"appId"` + Timestamp int64 `json:"timestamp"` + NonceStr string `json:"nonceStr"` + Signature string `json:"signature"` + } +) + +type ( + WXH5AuthReq { + Code string `json:"code"` + } + WXH5AuthResp { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` + } +) + +@server ( + prefix: api/v1 + group: user + middleware: AuthInterceptor +) +service main { + @doc "绑定手机号" + @handler bindMobile + post /user/bindMobile (BindMobileReq) returns (BindMobileResp) +} + +//need login +@server ( + prefix: api/v1 + group: user + jwt: JwtAuth +) +service main { + @doc "get user info" + @handler detail + get /user/detail returns (UserInfoResp) + + @doc "get new token" + @handler getToken + post /user/getToken returns (MobileCodeLoginResp) + + @handler cancelOut + post /user/cancelOut +} + +type ( + UserInfoResp { + UserInfo User `json:"userInfo"` + } + BindMobileReq { + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` + } + BindMobileResp { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` + } +) + +//============================> auth v1 <============================ +@server ( + prefix: api/v1 + group: auth +) +service main { + @doc "get mobile verify code" + @handler sendSms + post /auth/sendSms (sendSmsReq) +} + +type ( + sendSmsReq { + Mobile string `json:"mobile" validate:"required,mobile"` + ActionType string `json:"actionType" validate:"required,oneof=login register query agentApply realName bindMobile"` + } +) + +//============================> notification v1 <============================ +@server ( + prefix: api/v1 + group: notification +) +service main { + @doc "get notifications" + @handler getNotifications + get /notification/list returns (GetNotificationsResp) +} + +type Notification { + Title string `json:"title"` // 通知标题 + Content string `json:"content"` // 通知内容 (富文本) + NotificationPage string `json:"notificationPage"` // 通知页面 + StartDate string `json:"startDate"` // 通知开始日期，格式 "YYYY-MM-DD" + EndDate string `json:"endDate"` // 通知结束日期，格式 "YYYY-MM-DD" + StartTime string `json:"startTime"` // 每天通知开始时间，格式 "HH:MM:SS" + EndTime string `json:"endTime"` // 每天通知结束时间，格式 "HH:MM:SS" +} + +type ( + // 获取通知响应体（分页） + GetNotificationsResp { + Notifications []Notification `json:"notifications"` // 通知列表 + Total int64 `json:"total"` // 总记录数 + } +) + diff --git a/app/main/api/desc/main.api b/app/main/api/desc/main.api new file mode 100644 index 0000000..855660b --- /dev/null +++ b/app/main/api/desc/main.api @@ -0,0 +1,30 @@ +syntax = "v1" + +info ( + title: "单体服务中心" + desc: "单体服务中心" + version: "v1" +) + +// 前台 +import "./front/user.api" +import "./front/query.api" +import "./front/pay.api" +import "./front/product.api" +import "./front/agent.api" +import "./front/app.api" +import "./front/authorization.api" +// 后台 +import "./admin/auth.api" +import "./admin/menu.api" +import "./admin/role.api" +import "./admin/order.api" +import "./admin/admin_user.api" +import "./admin/platform_user.api" +import "./admin/notification.api" +import "./admin/admin_product.api" +import "./admin/admin_feature.api" +import "./admin/admin_query.api" +import "./admin/admin_agent.api" +import "./admin/admin_api.api" +import "./admin/admin_role_api.api" diff --git a/app/main/api/etc/main.dev.yaml b/app/main/api/etc/main.dev.yaml new file mode 100644 index 0000000..7a323ca --- /dev/null +++ b/app/main/api/etc/main.dev.yaml @@ -0,0 +1,92 @@ +Name: main +Host: 0.0.0.0 +Port: 8888 +Timeout: 0 +DataSource: "qnc:5vg67b3UNHu8@tcp(127.0.0.1:21201)/qnc?charset=utf8mb4&parseTime=True&loc=Local" +CacheRedis: + - Host: "127.0.0.1:21202" + Pass: "3m3WsgyCKWqz" # Redis 密码，如果未设置则留空 + Type: "node" # 单节点模式 +JwtAuth: + AccessSecret: "WUvoIwL-FK0qnlxhvxR9tV6SjfOpeJMpKmY2QvT99lA" + AccessExpire: 2592000 + RefreshAfter: 1296000 +VerifyCode: + AccessKeyID: "LTAI5tKGB3TVJbMHSoZN3yr9" + AccessKeySecret: "OCQ30GWp4yENMjmfOAaagksE18bp65" + EndpointURL: "dysmsapi.aliyuncs.com" + SignName: "天远查" + TemplateCode: "SMS_302641455" + ValidTime: 300 +Encrypt: + SecretKey: "ff83609b2b24fc73196aac3d3dfb874f" +WestConfig: + Url: "http://proxy.tianyuanapi.com/api/invoke" + Key: "121a1e41fc1690dd6b90afbcacd80cf4" + SecretId: "449159" + SecretSecondId: "296804" +YushanConfig: + ApiKey: "4c566c4a4b543164535455685655316c" + AcctID: "YSSJ843926726" + Url: "https://api.yushanshuju.com/credit-gw/service" +Alipay: + AppID: "2021004165608254" + PrivateKey: "MIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQCPsXuwFJeHAL8CwI0QdD9GP7xQ8eejIoQKg6J3/peu26su68JCtGSRhlDm/7vbLHJcFR6h7at+INoz2juc7SqlmNO7i9wKc3+Ua0487y1G2fCsneRNxTTqbceBZwqjj9/AAN0u5/4nSl0bcqTeMddofdpTGOvwGvIJh6CZgCglnhMZnH4D6H6yiIyZf7Q6k2d/qBpVGK8kluYEtSnf/vEQCHhxRx+/DgTL7V1LjbA3BYoPTELZ15JAj0uIzuxextAtxOm4+Huli0RJFAN3q/to2L1Zs8yYY1gKJyTaPWKsJWBx8zI+gZcC/e45k6CZnGgh1Fn3+Xqkf7eGxJGGHs1fAgMBAAECggEAM2rkApbrvdBDiV2TXK7sMVv/K8vUAmkIbKa7zUpZxqUuNSUBp1LbpcM1UeNyujPGXDLmejUMp55j1igiKr4nA4iTQ0oBm+/GWDqpjV5cijzURUBegIGvtK9Bs4lGok6KVy839l/nbvHKLVcxrZySIv7dz9xcGNfbghN5IVRdiU/kOokNbtwQNC837piG5q4PHL6bzwIUGbrLED/RDmw/IwVMMmZovcQQ2JAuWJBo9CS9LB0Nc3I4MOPNx/0Rl+5URSSfmJAriL5ihlWckocQCUHwhRpSGQ6Q4xAXFYvb8OsApAQG3WU9SciBfs2wg/QfGNFzwQgGFofPcTQg3DTeuQKBgQDUHBTsgoe3WXnGo6qZKw1zA4OtF67IJJoltHo5JtkBRKCNVU3BJ+q+6i/fn0MBwScKQ1mhPjWe3h+qTRT207RRxGaxb6ljATOiU+BxmpHvu6jP+DVYtP5F3M7MCAGqpDAEoXgoAWttxmijqk+5YuLOLe0j/btCmpzuH7zwxSnqlQKBgQCtbTvaS/g+Jeu4Ml6iv7xi5//JCjeTn2wUJpXnNmN0jn+riRwEO81z4GWuOI8WukZHHAnufI6qWk2sLH0gcdLQ/STsMnl2L3NbeUyO8o5w2JSAlnZDYfaFfasGqFkGJrBLqG6bh/Bk1DP3/Bl6iMEwDbmu7Ptoy8ihokng9dEPIwKBgBHdi6WgGO5IiwlAH85m4eseEKkzpXUWICWs3d6SdxS0QxGkbbgnNI6ACyg6sdoj+rXSlmoOY1XOP7yIYYuoqTd542xui0XbhA3YIr9u1XvrwnxB27xtAj3AK2rkAb/ttF2ve/9inznPzGB8p9plidTz6VVuuacSfsVPxwpAkRdBAoGAR7c9Ifd6b1DFGkWSBuEc6RWhG6Si+OPbELYoFRXTqNZoiynGsSV9v2ZTBemTmkVrXGqG3N0bLezr47/9+lW3ZP7ZrubsfWf/3xrZAt/g8V9OgaI2w4SWKfuepsElFzsWeiLroltjmH58Axd3/cjhgpqaZ3DOQjbK/7QZsvJUAlsCgYEAqTQVhKLizM7BvXu1N6Z2K8trfJbiN+f90XhZIRPkIIcom0PnOfXhRtT76MCxz9n+lwf+alOKOfbQFy0pZtWG/eaFSYroQlXL+EfmqlFPXZR6D0NQLeygWAKH8161IQUh2VF3Qkhle6g6ZkyJA3Ev4RmqH2BYGv8hcZTTHsZ3Ic4=" + AlipayPublicKey: "MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA2CqoCp95w/JV3RT/gzF4/8QmVT1HQNaeW7yUp+mA7x9AbjvlTW/+eRn6oGAL/XhZLjvHD0XjKLVKX0MJVS1aUQHEHEbOJN4Eu8II45OavD4iZISa7Kp9V6AM+i4qTyaeV2wNDnGxHQBaLVUGCfMR+56EK2YpORdE1H9uy72SSQseVb3bmpsV9EW/IJNmcVL/ut3uA1JWAoRmzlQ7ekxg7p8AYXzYPEHQr1tl7W+M4zv9wO9GKZCxIqMA8U3RP5npPfRaCfIRGzXzCqFEEUvWuidOB7frsvN4jiPD07qpL2Bi9LM1X/ee2kC/oM8Uhd7ERZhG8MbZfijZKxgrsDKBcwIDAQAB" + AppCertPath: "etc/merchant/appCertPublicKey_2021004165608254.crt" + AlipayCertPath: "etc/merchant/alipayCertPublicKey_RSA2.crt" + AlipayRootCertPath: "etc/merchant/alipayRootCert.crt" + IsProduction: true + NotifyUrl: "https://6m4685017o.goho.co/api/v1/pay/alipay/callback" + ReturnURL: "http://localhost:5678/inquire" + +Wxpay: + AppID: "wx442ee1ac1ee75917" + MchID: "1682635136" + MchCertificateSerialNumber: "5369B8AEEBDCF7AF274510252E6A8C0659C30F61" + MchApiv3Key: "e3ea4cf0765f1e71b01bb387dfcdbc9f" + MchPrivateKeyPath: "etc/merchant/apiclient_key.pem" + MchPublicKeyID: "PUB_KEY_ID_0116826351362025060900382267001601" + MchPublicKeyPath: "etc/merchant/pub_key.pem" + MchPlatformRAS: "1FFEC3F62E31885FAB4C91ADCB8D7557E9488781" + NotifyUrl: "https://6m4685017o.goho.co/api/v1/pay/wechat/callback" + RefundNotifyUrl: "https://6m4685017o.goho.co/api/v1/paywechat/refund_callback" +Applepay: + ProductionVerifyURL: "https://api.storekit.itunes.apple.com/inApps/v1/transactions/receipt" + SandboxVerifyURL: "https://api.storekit-sandbox.itunes.apple.com/inApps/v1/transactions/receipt" + Sandbox: false + BundleID: "com.allinone.check" + IssuerID: "bf828d85-5269-4914-9660-c066e09cd6ef" + KeyID: "LAY65829DQ" + LoadPrivateKeyPath: "etc/merchant/AuthKey_LAY65829DQ.p8" +Ali: + Code: "d55b58829efb41c8aa8e86769cba4844" +SystemConfig: + ThreeVerify: false +WechatH5: + AppID: "wx442ee1ac1ee75917" + AppSecret: "c80474909db42f63913b7a307b3bee17" +WechatMini: + AppID: "wx781abb66b3368963" # 小程序的AppID + AppSecret: "c7d02cdb0fc23c35c93187af9243b00d" # 小程序的AppSecret + TycAppID: "wxe74617f3dd56c196" + TycAppSecret: "c8207e54aef5689b2a7c1f91ed7ae8a0" +Query: + ShareLinkExpire: 604800 # 7天 = 7 * 24 * 60 * 60 = 604800秒 +AdminConfig: + AccessSecret: "jK8nP3qR7tV2xZ5aB9cD1eF6gH4iJ0kL8mN5oP6qR7sT" + AccessExpire: 604800 + RefreshAfter: 302400 +TaxConfig: + TaxRate: 0.06 + TaxExemptionAmount: 0.00 +Tianyuanapi: + AccessID: "7f8a9b2c4d5e6f1a" + Key: "9e4f8a1b3c6d7e2f5a8b9c0d1e4f7a2b" + BaseURL: "https://api.tianyuanapi.com" + Timeout: 60 +Authorization: + FileBaseURL: "https://www.quannengcha.com/api/v1/auth-docs" # 授权书文件访问基础URL +Promotion: + PromotionDomain: "http://localhost:8888" # 推广域名（用于生成短链） + OfficialDomain: "http://localhost:5678" # 正式站点域名（短链重定向的目标域名） diff --git a/app/main/api/etc/main.yaml b/app/main/api/etc/main.yaml new file mode 100644 index 0000000..30394a1 --- /dev/null +++ b/app/main/api/etc/main.yaml @@ -0,0 +1,79 @@ +Name: main +Host: 0.0.0.0 +Port: 8888 +DataSource: "qnc:5vg67b3UNHu8@tcp(qnc_mysql:3306)/qnc?charset=utf8mb4&parseTime=True&loc=Local" +CacheRedis: + - Host: "qnc_redis:6379" + Pass: "3m3WsgyCKWqz" # Redis 密码，如果未设置则留空 + Type: "node" # 单节点模式 + +JwtAuth: + AccessSecret: "WUvoIwL-FK0qnlxhvxR9tV6SjfOpeJMpKmY2QvT99lA" + AccessExpire: 2592000 + RefreshAfter: 1296000 + +VerifyCode: + AccessKeyID: "LTAI5tKGB3TVJbMHSoZN3yr9" + AccessKeySecret: "OCQ30GWp4yENMjmfOAaagksE18bp65" + EndpointURL: "dysmsapi.aliyuncs.com" + SignName: "天远查" + TemplateCode: "SMS_302641455" + ValidTime: 300 +Encrypt: + SecretKey: "ff83609b2b24fc73196aac3d3dfb874f" +Alipay: + AppID: "2021004165608254" + PrivateKey: "MIIEvQIBADANBgkqhkiG9w0BAQEFAASCBKcwggSjAgEAAoIBAQCPsXuwFJeHAL8CwI0QdD9GP7xQ8eejIoQKg6J3/peu26su68JCtGSRhlDm/7vbLHJcFR6h7at+INoz2juc7SqlmNO7i9wKc3+Ua0487y1G2fCsneRNxTTqbceBZwqjj9/AAN0u5/4nSl0bcqTeMddofdpTGOvwGvIJh6CZgCglnhMZnH4D6H6yiIyZf7Q6k2d/qBpVGK8kluYEtSnf/vEQCHhxRx+/DgTL7V1LjbA3BYoPTELZ15JAj0uIzuxextAtxOm4+Huli0RJFAN3q/to2L1Zs8yYY1gKJyTaPWKsJWBx8zI+gZcC/e45k6CZnGgh1Fn3+Xqkf7eGxJGGHs1fAgMBAAECggEAM2rkApbrvdBDiV2TXK7sMVv/K8vUAmkIbKa7zUpZxqUuNSUBp1LbpcM1UeNyujPGXDLmejUMp55j1igiKr4nA4iTQ0oBm+/GWDqpjV5cijzURUBegIGvtK9Bs4lGok6KVy839l/nbvHKLVcxrZySIv7dz9xcGNfbghN5IVRdiU/kOokNbtwQNC837piG5q4PHL6bzwIUGbrLED/RDmw/IwVMMmZovcQQ2JAuWJBo9CS9LB0Nc3I4MOPNx/0Rl+5URSSfmJAriL5ihlWckocQCUHwhRpSGQ6Q4xAXFYvb8OsApAQG3WU9SciBfs2wg/QfGNFzwQgGFofPcTQg3DTeuQKBgQDUHBTsgoe3WXnGo6qZKw1zA4OtF67IJJoltHo5JtkBRKCNVU3BJ+q+6i/fn0MBwScKQ1mhPjWe3h+qTRT207RRxGaxb6ljATOiU+BxmpHvu6jP+DVYtP5F3M7MCAGqpDAEoXgoAWttxmijqk+5YuLOLe0j/btCmpzuH7zwxSnqlQKBgQCtbTvaS/g+Jeu4Ml6iv7xi5//JCjeTn2wUJpXnNmN0jn+riRwEO81z4GWuOI8WukZHHAnufI6qWk2sLH0gcdLQ/STsMnl2L3NbeUyO8o5w2JSAlnZDYfaFfasGqFkGJrBLqG6bh/Bk1DP3/Bl6iMEwDbmu7Ptoy8ihokng9dEPIwKBgBHdi6WgGO5IiwlAH85m4eseEKkzpXUWICWs3d6SdxS0QxGkbbgnNI6ACyg6sdoj+rXSlmoOY1XOP7yIYYuoqTd542xui0XbhA3YIr9u1XvrwnxB27xtAj3AK2rkAb/ttF2ve/9inznPzGB8p9plidTz6VVuuacSfsVPxwpAkRdBAoGAR7c9Ifd6b1DFGkWSBuEc6RWhG6Si+OPbELYoFRXTqNZoiynGsSV9v2ZTBemTmkVrXGqG3N0bLezr47/9+lW3ZP7ZrubsfWf/3xrZAt/g8V9OgaI2w4SWKfuepsElFzsWeiLroltjmH58Axd3/cjhgpqaZ3DOQjbK/7QZsvJUAlsCgYEAqTQVhKLizM7BvXu1N6Z2K8trfJbiN+f90XhZIRPkIIcom0PnOfXhRtT76MCxz9n+lwf+alOKOfbQFy0pZtWG/eaFSYroQlXL+EfmqlFPXZR6D0NQLeygWAKH8161IQUh2VF3Qkhle6g6ZkyJA3Ev4RmqH2BYGv8hcZTTHsZ3Ic4=" + AlipayPublicKey: "MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA2CqoCp95w/JV3RT/gzF4/8QmVT1HQNaeW7yUp+mA7x9AbjvlTW/+eRn6oGAL/XhZLjvHD0XjKLVKX0MJVS1aUQHEHEbOJN4Eu8II45OavD4iZISa7Kp9V6AM+i4qTyaeV2wNDnGxHQBaLVUGCfMR+56EK2YpORdE1H9uy72SSQseVb3bmpsV9EW/IJNmcVL/ut3uA1JWAoRmzlQ7ekxg7p8AYXzYPEHQr1tl7W+M4zv9wO9GKZCxIqMA8U3RP5npPfRaCfIRGzXzCqFEEUvWuidOB7frsvN4jiPD07qpL2Bi9LM1X/ee2kC/oM8Uhd7ERZhG8MbZfijZKxgrsDKBcwIDAQAB" + AppCertPath: "etc/merchant/appCertPublicKey_2021004165608254.crt" + AlipayCertPath: "etc/merchant/alipayCertPublicKey_RSA2.crt" + AlipayRootCertPath: "etc/merchant/alipayRootCert.crt" + IsProduction: true + NotifyUrl: "https://www.quannengcha.com/api/v1/pay/alipay/callback" + ReturnURL: "https://www.quannengcha.com/payment/result" +Wxpay: + AppID: "wx442ee1ac1ee75917" + MchID: "1682635136" + MchCertificateSerialNumber: "5369B8AEEBDCF7AF274510252E6A8C0659C30F61" + MchApiv3Key: "e3ea4cf0765f1e71b01bb387dfcdbc9f" + MchPrivateKeyPath: "etc/merchant/apiclient_key.pem" + MchPublicKeyID: "PUB_KEY_ID_0116826351362025060900382267001601" + MchPublicKeyPath: "etc/merchant/pub_key.pem" + MchPlatformRAS: "1FFEC3F62E31885FAB4C91ADCB8D7557E9488781" + NotifyUrl: "https://www.quannengcha.com/api/v1/pay/wechat/callback" + RefundNotifyUrl: "https://www.quannengcha.com/api/v1/pay/wechat/refund_callback" +Applepay: + ProductionVerifyURL: "https://api.storekit.itunes.apple.com/inApps/v1/transactions/receipt" + SandboxVerifyURL: "https://api.storekit-sandbox.itunes.apple.com/inApps/v1/transactions/receipt" + Sandbox: true + BundleID: "com.allinone.check" + IssuerID: "bf828d85-5269-4914-9660-c066e09cd6ef" + KeyID: "LAY65829DQ" + LoadPrivateKeyPath: "etc/merchant/AuthKey_LAY65829DQ.p8" +SystemConfig: + ThreeVerify: true +WechatH5: + AppID: "wx442ee1ac1ee75917" + AppSecret: "c80474909db42f63913b7a307b3bee17" +WechatMini: + AppID: "wx5bacc94add2da981" # 小程序的AppID + AppSecret: "48a2c1e8ff1b7d4c0ff82fbefa64d2d0" # 小程序的AppSecret +Query: + ShareLinkExpire: 604800 # 7天 = 7 * 24 * 60 * 60 = 604800秒 +AdminConfig: + AccessSecret: "jK8nP3qR7tV2xZ5aB9cD1eF6gH4iJ0kL8mN5oP6qR7sT" + AccessExpire: 604800 + RefreshAfter: 302400 +TaxConfig: + TaxRate: 0.06 + TaxExemptionAmount: 0.00 +Tianyuanapi: + AccessID: "7f8a9b2c4d5e6f1a" + Key: "9e4f8a1b3c6d7e2f5a8b9c0d1e4f7a2b" + BaseURL: "https://api.tianyuanapi.com" + Timeout: 60 +Authorization: + FileBaseURL: "https://www.quannengcha.com/api/v1/auth-docs" # 授权书文件访问基础URL +Promotion: + PromotionDomain: "https://p.quannengcha.com" # 推广域名（用于生成短链） + OfficialDomain: "https://www.quannengcha.com" # 正式站点域名（短链重定向的目标域名） diff --git a/app/main/api/etc/merchant/AuthKey_LAY65829DQ.p8 b/app/main/api/etc/merchant/AuthKey_LAY65829DQ.p8 new file mode 100644 index 0000000..b448586 --- /dev/null +++ b/app/main/api/etc/merchant/AuthKey_LAY65829DQ.p8 @@ -0,0 +1,6 @@ +-----BEGIN PRIVATE KEY----- +MIGTAgEAMBMGByqGSM49AgEGCCqGSM49AwEHBHkwdwIBAQQgkidSHV1OeJN84sDD +xWLGIVjTyhn6sAQDyHfqKW6lxnGgCgYIKoZIzj0DAQehRANCAAQSAlAcuuuRNFqk +aMPVpXxsiR/pwhyM62tFhdFsbULq1C7MItQxKVMKCiwz3r5rZZy7HcbkqL47LPZ1 +q6V8Wyop +-----END PRIVATE KEY----- \ No newline at end of file diff --git a/app/main/api/etc/merchant/alipayCertPublicKey_RSA2.crt b/app/main/api/etc/merchant/alipayCertPublicKey_RSA2.crt new file mode 100644 index 0000000..a1de812 --- /dev/null +++ b/app/main/api/etc/merchant/alipayCertPublicKey_RSA2.crt @@ -0,0 +1,43 @@ +-----BEGIN CERTIFICATE----- +MIIDuDCCAqCgAwIBAgIQICQIJEqhy5G63s3j7VrljTANBgkqhkiG9w0BAQsFADCBgjELMAkGA1UE +BhMCQ04xFjAUBgNVBAoMDUFudCBGaW5hbmNpYWwxIDAeBgNVBAsMF0NlcnRpZmljYXRpb24gQXV0 +aG9yaXR5MTkwNwYDVQQDDDBBbnQgRmluYW5jaWFsIENlcnRpZmljYXRpb24gQXV0aG9yaXR5IENs +YXNzIDIgUjEwHhcNMjQwODI0MTMzMDA3WhcNMjkwODIzMTMzMDA3WjCBmDELMAkGA1UEBhMCQ04x +MzAxBgNVBAoMKua1t+WNl+ecgeWtpuWuh+aAnee9kee7nOenkeaKgOaciemZkOWFrOWPuDEPMA0G +A1UECwwGQWxpcGF5MUMwQQYDVQQDDDrmlK/ku5jlrp0o5Lit5Zu9Kee9kee7nOaKgOacr+aciemZ +kOWFrOWPuC0yMDg4ODQxODczNDU0MDUzMIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEA +2CqoCp95w/JV3RT/gzF4/8QmVT1HQNaeW7yUp+mA7x9AbjvlTW/+eRn6oGAL/XhZLjvHD0XjKLVK +X0MJVS1aUQHEHEbOJN4Eu8II45OavD4iZISa7Kp9V6AM+i4qTyaeV2wNDnGxHQBaLVUGCfMR+56E +K2YpORdE1H9uy72SSQseVb3bmpsV9EW/IJNmcVL/ut3uA1JWAoRmzlQ7ekxg7p8AYXzYPEHQr1tl +7W+M4zv9wO9GKZCxIqMA8U3RP5npPfRaCfIRGzXzCqFEEUvWuidOB7frsvN4jiPD07qpL2Bi9LM1 +X/ee2kC/oM8Uhd7ERZhG8MbZfijZKxgrsDKBcwIDAQABoxIwEDAOBgNVHQ8BAf8EBAMCA/gwDQYJ +KoZIhvcNAQELBQADggEBADsqTEJB7QQc+zFteTgZ608BnTvVlZlPHK+5pLFSG9N1XK3LCv2wj1Tc +FfDo6VXq5YhLmIIp8LlyUWTn5Gwd+/3Rbv50p52CvzsdeqZ9kSJAqFyO4iWmAPUB4f63yLH81q+1 +eyUjc4GkUmZ3EZmB2+vyG2iRHvyG27TYbWrGPWz4AaPECU5jn3rIPDVvR6C9JXjODiiuU5PCVJ9j +pCmGgmllTYAHSvhkOF8zM+qt1Fz0iEtB2ZPMzKexBD4uz8ULkwp+x3QvuLD/ebBbMr2R3BJMCD+3 +2dx0wQJ0OhXuXGOXbuwyAo17LmRKheucoadlT+7Ilr+i5JrAwApBLQSQI6w= +-----END CERTIFICATE----- +-----BEGIN CERTIFICATE----- +MIIE4jCCAsqgAwIBAgIIYsSr5bKAMl8wDQYJKoZIhvcNAQELBQAwejELMAkGA1UEBhMCQ04xFjAU +BgNVBAoMDUFudCBGaW5hbmNpYWwxIDAeBgNVBAsMF0NlcnRpZmljYXRpb24gQXV0aG9yaXR5MTEw +LwYDVQQDDChBbnQgRmluYW5jaWFsIENlcnRpZmljYXRpb24gQXV0aG9yaXR5IFIxMB4XDTE4MDMy +MjE0MzQxNVoXDTM3MTEyNjE0MzQxNVowgYIxCzAJBgNVBAYTAkNOMRYwFAYDVQQKDA1BbnQgRmlu +YW5jaWFsMSAwHgYDVQQLDBdDZXJ0aWZpY2F0aW9uIEF1dGhvcml0eTE5MDcGA1UEAwwwQW50IEZp +bmFuY2lhbCBDZXJ0aWZpY2F0aW9uIEF1dGhvcml0eSBDbGFzcyAyIFIxMIIBIjANBgkqhkiG9w0B +AQEFAAOCAQ8AMIIBCgKCAQEAsLMfYaoRoPRbmDcAfXPCmKf43pWRN5yTXa/KJWO0l+mrgQvs89bA +NEvbDUxlkGwycwtwi5DgBuBgVhLliXu+R9CYgr2dXs8D8Hx/gsggDcyGPLmVrDOnL+dyeauheARZ +fA3du60fwEwwbGcVIpIxPa/4n3IS/ElxQa6DNgqxh8J9Xwh7qMGl0JK9+bALuxf7B541Gr4p0WEN +G8fhgjBV4w4ut9eQLOoa1eddOUSZcy46Z7allwowwgt7b5VFfx/P1iKJ3LzBMgkCK7GZ2kiLrL7R +iqV+h482J7hkJD+ardoc6LnrHO/hIZymDxok+VH9fVeUdQa29IZKrIDVj65THQIDAQABo2MwYTAf +BgNVHSMEGDAWgBRfdLQEwE8HWurlsdsio4dBspzhATAdBgNVHQ4EFgQUSqHkYINtUSAtDPnS8Xoy +oP9p7qEwDwYDVR0TAQH/BAUwAwEB/zAOBgNVHQ8BAf8EBAMCAQYwDQYJKoZIhvcNAQELBQADggIB +AIQ8TzFy4bVIVb8+WhHKCkKNPcJe2EZuIcqvRoi727lZTJOfYy/JzLtckyZYfEI8J0lasZ29wkTt +a1IjSo+a6XdhudU4ONVBrL70U8Kzntplw/6TBNbLFpp7taRALjUgbCOk4EoBMbeCL0GiYYsTS0mw +7xdySzmGQku4GTyqutIGPQwKxSj9iSFw1FCZqr4VP4tyXzMUgc52SzagA6i7AyLedd3tbS6lnR5B +L+W9Kx9hwT8L7WANAxQzv/jGldeuSLN8bsTxlOYlsdjmIGu/C9OWblPYGpjQQIRyvs4Cc/mNhrh+ +14EQgwuemIIFDLOgcD+iISoN8CqegelNcJndFw1PDN6LkVoiHz9p7jzsge8RKay/QW6C03KNDpWZ +EUCgCUdfHfo8xKeR+LL1cfn24HKJmZt8L/aeRZwZ1jwePXFRVtiXELvgJuM/tJDIFj2KD337iV64 +fWcKQ/ydDVGqfDZAdcU4hQdsrPWENwPTQPfVPq2NNLMyIH9+WKx9Ed6/WzeZmIy5ZWpX1TtTolo6 +OJXQFeItMAjHxW/ZSZTok5IS3FuRhExturaInnzjYpx50a6kS34c5+c8hYq7sAtZ/CNLZmBnBCFD +aMQqT8xFZJ5uolUaSeXxg7JFY1QsYp5RKvj4SjFwCGKJ2+hPPe9UyyltxOidNtxjaknOCeBHytOr +-----END CERTIFICATE----- diff --git a/app/main/api/etc/merchant/alipayRootCert.crt b/app/main/api/etc/merchant/alipayRootCert.crt new file mode 100644 index 0000000..76417c5 --- /dev/null +++ b/app/main/api/etc/merchant/alipayRootCert.crt @@ -0,0 +1,88 @@ +-----BEGIN CERTIFICATE----- +MIIBszCCAVegAwIBAgIIaeL+wBcKxnswDAYIKoEcz1UBg3UFADAuMQswCQYDVQQG +EwJDTjEOMAwGA1UECgwFTlJDQUMxDzANBgNVBAMMBlJPT1RDQTAeFw0xMjA3MTQw +MzExNTlaFw00MjA3MDcwMzExNTlaMC4xCzAJBgNVBAYTAkNOMQ4wDAYDVQQKDAVO +UkNBQzEPMA0GA1UEAwwGUk9PVENBMFkwEwYHKoZIzj0CAQYIKoEcz1UBgi0DQgAE +MPCca6pmgcchsTf2UnBeL9rtp4nw+itk1Kzrmbnqo05lUwkwlWK+4OIrtFdAqnRT +V7Q9v1htkv42TsIutzd126NdMFswHwYDVR0jBBgwFoAUTDKxl9kzG8SmBcHG5Yti +W/CXdlgwDAYDVR0TBAUwAwEB/zALBgNVHQ8EBAMCAQYwHQYDVR0OBBYEFEwysZfZ +MxvEpgXBxuWLYlvwl3ZYMAwGCCqBHM9VAYN1BQADSAAwRQIgG1bSLeOXp3oB8H7b +53W+CKOPl2PknmWEq/lMhtn25HkCIQDaHDgWxWFtnCrBjH16/W3Ezn7/U/Vjo5xI +pDoiVhsLwg== +-----END CERTIFICATE----- + +-----BEGIN CERTIFICATE----- +MIIF0zCCA7ugAwIBAgIIH8+hjWpIDREwDQYJKoZIhvcNAQELBQAwejELMAkGA1UE +BhMCQ04xFjAUBgNVBAoMDUFudCBGaW5hbmNpYWwxIDAeBgNVBAsMF0NlcnRpZmlj +YXRpb24gQXV0aG9yaXR5MTEwLwYDVQQDDChBbnQgRmluYW5jaWFsIENlcnRpZmlj +YXRpb24gQXV0aG9yaXR5IFIxMB4XDTE4MDMyMTEzNDg0MFoXDTM4MDIyODEzNDg0 +MFowejELMAkGA1UEBhMCQ04xFjAUBgNVBAoMDUFudCBGaW5hbmNpYWwxIDAeBgNV +BAsMF0NlcnRpZmljYXRpb24gQXV0aG9yaXR5MTEwLwYDVQQDDChBbnQgRmluYW5j +aWFsIENlcnRpZmljYXRpb24gQXV0aG9yaXR5IFIxMIICIjANBgkqhkiG9w0BAQEF +AAOCAg8AMIICCgKCAgEAtytTRcBNuur5h8xuxnlKJetT65cHGemGi8oD+beHFPTk +rUTlFt9Xn7fAVGo6QSsPb9uGLpUFGEdGmbsQ2q9cV4P89qkH04VzIPwT7AywJdt2 +xAvMs+MgHFJzOYfL1QkdOOVO7NwKxH8IvlQgFabWomWk2Ei9WfUyxFjVO1LVh0Bp +dRBeWLMkdudx0tl3+21t1apnReFNQ5nfX29xeSxIhesaMHDZFViO/DXDNW2BcTs6 +vSWKyJ4YIIIzStumD8K1xMsoaZBMDxg4itjWFaKRgNuPiIn4kjDY3kC66Sl/6yTl +YUz8AybbEsICZzssdZh7jcNb1VRfk79lgAprm/Ktl+mgrU1gaMGP1OE25JCbqli1 +Pbw/BpPynyP9+XulE+2mxFwTYhKAwpDIDKuYsFUXuo8t261pCovI1CXFzAQM2w7H +DtA2nOXSW6q0jGDJ5+WauH+K8ZSvA6x4sFo4u0KNCx0ROTBpLif6GTngqo3sj+98 +SZiMNLFMQoQkjkdN5Q5g9N6CFZPVZ6QpO0JcIc7S1le/g9z5iBKnifrKxy0TQjtG +PsDwc8ubPnRm/F82RReCoyNyx63indpgFfhN7+KxUIQ9cOwwTvemmor0A+ZQamRe +9LMuiEfEaWUDK+6O0Gl8lO571uI5onYdN1VIgOmwFbe+D8TcuzVjIZ/zvHrAGUcC +AwEAAaNdMFswCwYDVR0PBAQDAgEGMAwGA1UdEwQFMAMBAf8wHQYDVR0OBBYEFF90 +tATATwda6uWx2yKjh0GynOEBMB8GA1UdIwQYMBaAFF90tATATwda6uWx2yKjh0Gy +nOEBMA0GCSqGSIb3DQEBCwUAA4ICAQCVYaOtqOLIpsrEikE5lb+UARNSFJg6tpkf +tJ2U8QF/DejemEHx5IClQu6ajxjtu0Aie4/3UnIXop8nH/Q57l+Wyt9T7N2WPiNq +JSlYKYbJpPF8LXbuKYG3BTFTdOVFIeRe2NUyYh/xs6bXGr4WKTXb3qBmzR02FSy3 +IODQw5Q6zpXj8prYqFHYsOvGCEc1CwJaSaYwRhTkFedJUxiyhyB5GQwoFfExCVHW +05ZFCAVYFldCJvUzfzrWubN6wX0DD2dwultgmldOn/W/n8at52mpPNvIdbZb2F41 +T0YZeoWnCJrYXjq/32oc1cmifIHqySnyMnavi75DxPCdZsCOpSAT4j4lAQRGsfgI +kkLPGQieMfNNkMCKh7qjwdXAVtdqhf0RVtFILH3OyEodlk1HYXqX5iE5wlaKzDop +PKwf2Q3BErq1xChYGGVS+dEvyXc/2nIBlt7uLWKp4XFjqekKbaGaLJdjYP5b2s7N +1dM0MXQ/f8XoXKBkJNzEiM3hfsU6DOREgMc1DIsFKxfuMwX3EkVQM1If8ghb6x5Y +jXayv+NLbidOSzk4vl5QwngO/JYFMkoc6i9LNwEaEtR9PhnrdubxmrtM+RjfBm02 +77q3dSWFESFQ4QxYWew4pHE0DpWbWy/iMIKQ6UZ5RLvB8GEcgt8ON7BBJeMc+Dyi +kT9qhqn+lw== +-----END CERTIFICATE----- + +-----BEGIN CERTIFICATE----- +MIICiDCCAgygAwIBAgIIQX76UsB/30owDAYIKoZIzj0EAwMFADB6MQswCQYDVQQG +EwJDTjEWMBQGA1UECgwNQW50IEZpbmFuY2lhbDEgMB4GA1UECwwXQ2VydGlmaWNh +dGlvbiBBdXRob3JpdHkxMTAvBgNVBAMMKEFudCBGaW5hbmNpYWwgQ2VydGlmaWNh +dGlvbiBBdXRob3JpdHkgRTEwHhcNMTkwNDI4MTYyMDQ0WhcNNDkwNDIwMTYyMDQ0 +WjB6MQswCQYDVQQGEwJDTjEWMBQGA1UECgwNQW50IEZpbmFuY2lhbDEgMB4GA1UE +CwwXQ2VydGlmaWNhdGlvbiBBdXRob3JpdHkxMTAvBgNVBAMMKEFudCBGaW5hbmNp +YWwgQ2VydGlmaWNhdGlvbiBBdXRob3JpdHkgRTEwdjAQBgcqhkjOPQIBBgUrgQQA +IgNiAASCCRa94QI0vR5Up9Yr9HEupz6hSoyjySYqo7v837KnmjveUIUNiuC9pWAU +WP3jwLX3HkzeiNdeg22a0IZPoSUCpasufiLAnfXh6NInLiWBrjLJXDSGaY7vaokt +rpZvAdmjXTBbMAsGA1UdDwQEAwIBBjAMBgNVHRMEBTADAQH/MB0GA1UdDgQWBBRZ +4ZTgDpksHL2qcpkFkxD2zVd16TAfBgNVHSMEGDAWgBRZ4ZTgDpksHL2qcpkFkxD2 +zVd16TAMBggqhkjOPQQDAwUAA2gAMGUCMQD4IoqT2hTUn0jt7oXLdMJ8q4vLp6sg +wHfPiOr9gxreb+e6Oidwd2LDnC4OUqCWiF8CMAzwKs4SnDJYcMLf2vpkbuVE4dTH +Rglz+HGcTLWsFs4KxLsq7MuU+vJTBUeDJeDjdA== +-----END CERTIFICATE----- + +-----BEGIN CERTIFICATE----- +MIIDxTCCAq2gAwIBAgIUEMdk6dVgOEIS2cCP0Q43P90Ps5YwDQYJKoZIhvcNAQEF +BQAwajELMAkGA1UEBhMCQ04xEzARBgNVBAoMCmlUcnVzQ2hpbmExHDAaBgNVBAsM +E0NoaW5hIFRydXN0IE5ldHdvcmsxKDAmBgNVBAMMH2lUcnVzQ2hpbmEgQ2xhc3Mg +MiBSb290IENBIC0gRzMwHhcNMTMwNDE4MDkzNjU2WhcNMzMwNDE4MDkzNjU2WjBq +MQswCQYDVQQGEwJDTjETMBEGA1UECgwKaVRydXNDaGluYTEcMBoGA1UECwwTQ2hp +bmEgVHJ1c3QgTmV0d29yazEoMCYGA1UEAwwfaVRydXNDaGluYSBDbGFzcyAyIFJv +b3QgQ0EgLSBHMzCCASIwDQYJKoZIhvcNAQEBBQADggEPADCCAQoCggEBAOPPShpV +nJbMqqCw6Bz1kehnoPst9pkr0V9idOwU2oyS47/HjJXk9Rd5a9xfwkPO88trUpz5 +4GmmwspDXjVFu9L0eFaRuH3KMha1Ak01citbF7cQLJlS7XI+tpkTGHEY5pt3EsQg +wykfZl/A1jrnSkspMS997r2Gim54cwz+mTMgDRhZsKK/lbOeBPpWtcFizjXYCqhw +WktvQfZBYi6o4sHCshnOswi4yV1p+LuFcQ2ciYdWvULh1eZhLxHbGXyznYHi0dGN +z+I9H8aXxqAQfHVhbdHNzi77hCxFjOy+hHrGsyzjrd2swVQ2iUWP8BfEQqGLqM1g +KgWKYfcTGdbPB1MCAwEAAaNjMGEwHQYDVR0OBBYEFG/oAMxTVe7y0+408CTAK8hA +uTyRMB8GA1UdIwQYMBaAFG/oAMxTVe7y0+408CTAK8hAuTyRMA8GA1UdEwEB/wQF +MAMBAf8wDgYDVR0PAQH/BAQDAgEGMA0GCSqGSIb3DQEBBQUAA4IBAQBLnUTfW7hp +emMbuUGCk7RBswzOT83bDM6824EkUnf+X0iKS95SUNGeeSWK2o/3ALJo5hi7GZr3 +U8eLaWAcYizfO99UXMRBPw5PRR+gXGEronGUugLpxsjuynoLQu8GQAeysSXKbN1I +UugDo9u8igJORYA+5ms0s5sCUySqbQ2R5z/GoceyI9LdxIVa1RjVX8pYOj8JFwtn +DJN3ftSFvNMYwRuILKuqUYSHc2GPYiHVflDh5nDymCMOQFcFG3WsEuB+EYQPFgIU +1DHmdZcz7Llx8UOZXX2JupWCYzK1XhJb+r4hK5ncf/w8qGtYlmyJpxk3hr1TfUJX +Yf4Zr0fJsGuv +-----END CERTIFICATE----- \ No newline at end of file diff --git a/app/main/api/etc/merchant/apiclient_key.pem b/app/main/api/etc/merchant/apiclient_key.pem new file mode 100644 index 0000000..246c1df --- /dev/null +++ b/app/main/api/etc/merchant/apiclient_key.pem @@ -0,0 +1,28 @@ +-----BEGIN PRIVATE KEY----- +MIIEvgIBADANBgkqhkiG9w0BAQEFAASCBKgwggSkAgEAAoIBAQDCP6fWm1vXXybH +m3Ne6PjacGrN2+iMrzWZlzdHCZ31udDPqSUYaZ+78b441KZK/CJFQWeSJ/1h//A+ +BGsQDKvE/fj2QzN1KkOuQ8WJXNGpixE5uu5bv/QTN/ukurGdA1aO2aFCANumlOmB +HkB/B2so57ii8iQQjwK2xM4r3oOU/IfcFGKL+9/QjLGFFp9PJXCDBCgrxxlZGaj1 +3wowlfVOzlaX94gemQsCYVkuAFIYMAnFHs9cKNZQIU80somW/yy2Gy38N6n7NnbD +nvFSaq4GoDROqRgKbRZ5e706d/p7A3aS/2oRqq1jomUIugK8g++LmoHFTgfhfQkI +v1aG/nPzAgMBAAECggEAD2RN31J2J42xm/V0YdviBCUOQXugZK1peN8jkSxw6Myt +gBbuCo4sCw9vvD8VYjGyYXx6QXmLuV03YyKkfSQT5EsflBvlEu6jaEaUe3rwXhfX +6JQoWPrP00oHVZk5g7CFBlK2VW2N+hgonIOSJr6mvhoGZlr7gphiZasYjx9Vm9N3 +Pbnfru5ttzplYNniwH3DF6ph8VmdbD1nnbWSKLXvHCsXQT2wBcnsIagIH3vyq6K1 +pc5abWsQJrixOPebpI8jD5w0HxHAqVLx58H/OC2zW/roAw1WS2AkueJ1j7dQ7Z0C +mc9Xexz5gcAP0nMAQv+LP7iYqsa/niFhfcTFWfdxkQKBgQD5JkKNmInU2/IVYCwO +c483MCSv1+MnbRXlb7vut8T0IupHTU6hCge6C3q3HsjbKSBn8bRChtPUzvw9JFxK +QWKiQqQDPLDJ08AIKhfQD2JiLtoikkZN0bF6OTL+Soney1yGx51mlfHM194+PcCJ +jF7iWdMVbcBwHbgydNxxIS5cKQKBgQDHlvQ4lw6gvLILpGK494/vNYSJP/Jmd66V +3oSGYi84YRKTSwH4NlbBVVieb3Dv+pPugbsXEuFHBif7WsivbYgNTE9++8Yvt0gh +duB1G4yh7m/ylQeSuipgQU9tozrU/15cWwmcCRV50wWXBGoVEM0kf7mzEKSxmjYk +Qzko/zxSuwKBgQCY6Bc+SViFz3qSDdTcBaXma+CIHsmlH7ipd9px1kzEvEzl95cD +FGHLl1H34qfIgUQHJvrHPXHyEBoT+CW/2MMM7DM2XV/ubctT92ln4pkxwqlTQExv +Y/s1FLesAtj8Z/hgK0/5bprYab9WmZV5lTGCXzhB1XqeFE9AgCHuODv4iQKBgQC8 +g2uwd5ytXQydymokYk9klJvWNrvw5GHV1BJAC0Smb6lnzZTSqCBRAxdsrb1yLK7E +u2vGY2K7/qiM1DZw23eBd+4t9gg+0VIjqXBfq+GsoNTDvtckUwnrWER5PY831ut9 +N89fvYS3SAUjmlvIAdKBAtKWusWTqiAxJ/05J7oGOQKBgB5PSr5i0LlupIbKui9t +XtXnRqGPxxrZZUpTkyrGOAnlCz/zq2QiwFpBWo/NMHOp0KmxzJpQ8yEY2LWlRZ61 +Oc9m0J/HtPw3Ohi1treBosEVG/0NOI9Tq1Obny23N51MVibdW6zEIyGUp/DbFS8h +5DljdOYX9IYIHHn3Ig4GeTGe +-----END PRIVATE KEY----- diff --git a/app/main/api/etc/merchant/appCertPublicKey_2021004165608254.crt b/app/main/api/etc/merchant/appCertPublicKey_2021004165608254.crt new file mode 100644 index 0000000..55f1aaa --- /dev/null +++ b/app/main/api/etc/merchant/appCertPublicKey_2021004165608254.crt @@ -0,0 +1,23 @@ +-----BEGIN CERTIFICATE----- +MIIEpzCCA4+gAwIBAgIQICUEEJni7ld+YaII5tXCjTANBgkqhkiG9w0BAQsFADCBgjELMAkGA1UE +BhMCQ04xFjAUBgNVBAoMDUFudCBGaW5hbmNpYWwxIDAeBgNVBAsMF0NlcnRpZmljYXRpb24gQXV0 +aG9yaXR5MTkwNwYDVQQDDDBBbnQgRmluYW5jaWFsIENlcnRpZmljYXRpb24gQXV0aG9yaXR5IENs +YXNzIDEgUjEwHhcNMjUwNDEwMDQwNTQyWhcNMzAwNDA5MDQwNTQyWjBuMQswCQYDVQQGEwJDTjEz +MDEGA1UECgwq5rW35Y2X55yB5a2m5a6H5oCd572R57uc56eR5oqA5pyJ6ZmQ5YWs5Y+4MQ8wDQYD +VQQLDAZBbGlwYXkxGTAXBgNVBAMMEDIwODg4NDE4NzM0NTQwNTMwggEiMA0GCSqGSIb3DQEBAQUA +A4IBDwAwggEKAoIBAQCPsXuwFJeHAL8CwI0QdD9GP7xQ8eejIoQKg6J3/peu26su68JCtGSRhlDm +/7vbLHJcFR6h7at+INoz2juc7SqlmNO7i9wKc3+Ua0487y1G2fCsneRNxTTqbceBZwqjj9/AAN0u +5/4nSl0bcqTeMddofdpTGOvwGvIJh6CZgCglnhMZnH4D6H6yiIyZf7Q6k2d/qBpVGK8kluYEtSnf +/vEQCHhxRx+/DgTL7V1LjbA3BYoPTELZ15JAj0uIzuxextAtxOm4+Huli0RJFAN3q/to2L1Zs8yY +Y1gKJyTaPWKsJWBx8zI+gZcC/e45k6CZnGgh1Fn3+Xqkf7eGxJGGHs1fAgMBAAGjggEqMIIBJjAf +BgNVHSMEGDAWgBRxB+IEYRbk5fJl6zEPyeD0PJrVkTAdBgNVHQ4EFgQUb4Kgf8qXsgDnWyG5deLm +Rw1DhPowQAYDVR0gBDkwNzA1BgdggRwBbgEBMCowKAYIKwYBBQUHAgEWHGh0dHA6Ly9jYS5hbGlw +YXkuY29tL2Nwcy5wZGYwDgYDVR0PAQH/BAQDAgbAMDAGA1UdHwQpMCcwJaAjoCGGH2h0dHA6Ly9j +YS5hbGlwYXkuY29tL2NybDEwMC5jcmwwYAYIKwYBBQUHAQEEVDBSMCgGCCsGAQUFBzAChhxodHRw +Oi8vY2EuYWxpcGF5LmNvbS9jYTYuY2VyMCYGCCsGAQUFBzABhhpodHRwOi8vY2EuYWxpcGF5LmNv +bTo4MzQwLzANBgkqhkiG9w0BAQsFAAOCAQEAYFgnBMeYL/+DjWpimqTtlGnlDsjPUDOjpZopzSa+ +HdL4nxWZBW48K+EzN8x45Ua7b1VKNmHWXe1O3X6W2Cz2H53NslVglzipPnxg9REx3Acr+8spkxWe +oxX9+1xwbSzBnpwG4UayambP0I6L2V9YQxSiLlLr6t2VuDxwCHVLu83rvnrOdlKjyjZop6oufhC9 +29BqfVmR6xuZv/VsCRx+BuRqNtHDoOZuP26kRVaHdnZMPPyTVw++cBt5Xh0Pq6vrYr1Nr9Kkp7wy +8RW951YTarYY5rDQ8RdaEFxPc+QWKQujVFQCvvt3ktTzrVa1mCgW/vX5qI0hXdPGGbq0DH3dxg== +-----END CERTIFICATE----- \ No newline at end of file diff --git a/app/main/api/etc/merchant/pub_key.pem b/app/main/api/etc/merchant/pub_key.pem new file mode 100644 index 0000000..9356d8d --- /dev/null +++ b/app/main/api/etc/merchant/pub_key.pem @@ -0,0 +1,9 @@ +-----BEGIN PUBLIC KEY----- +MIIBIjANBgkqhkiG9w0BAQEFAAOCAQ8AMIIBCgKCAQEAvwSy7dS/ICZV38tI0HxM +SAIE7+Ug92qryuNlkNyaNDRjfsykHsrPCSsUUQEZblBNmZOLfLQxmAaWC+cQqWCv +zfy4rXGAHE1widWFkHGzQzaw6cB0VdDXatK9yAt1PgXdp5jzBRzOn9Z3u4t0s771 +2zjuxCnLxMq84DovNgh2y0LBiuorWbtuTFTd8SXUGk2Jyuojq/02U3KTuyh+7SmW +ffJXKrzhrKwSpGh59e/fFxqX2xGlVoJ1kdohMZPo/7k+e5jP7qjrf93l7JVeUKYa +V27hNVowJ4oho21WVCJ1AYo41IbPJWI+6WxlaVeoR4zKix0Mb2timaWayyLoN53y +aQIDAQAB +-----END PUBLIC KEY----- diff --git a/app/main/api/internal/config/config.go b/app/main/api/internal/config/config.go new file mode 100644 index 0000000..4117ad1 --- /dev/null +++ b/app/main/api/internal/config/config.go @@ -0,0 +1,118 @@ +package config + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/rest" +) + +type Config struct { + rest.RestConf + DataSource string + CacheRedis cache.CacheConf + JwtAuth JwtAuth // JWT 鉴权相关配置 + VerifyCode VerifyCode + Encrypt Encrypt + Alipay AlipayConfig + Wxpay WxpayConfig + Applepay ApplepayConfig + Tianyuanapi TianyuanapiConfig + SystemConfig SystemConfig + WechatH5 WechatH5Config + Authorization AuthorizationConfig // 授权书配置 + WechatMini WechatMiniConfig + Query QueryConfig + AdminConfig AdminConfig + TaxConfig TaxConfig + Promotion PromotionConfig // 推广链接配置 +} + +// JwtAuth 用于 JWT 鉴权配置 +type JwtAuth struct { + AccessSecret string // JWT 密钥，用于签发 Token + AccessExpire int64 // Token 过期时间，单位为秒 + RefreshAfter int64 +} +type VerifyCode struct { + AccessKeyID string + AccessKeySecret string + EndpointURL string + SignName string + TemplateCode string + ValidTime int +} +type Encrypt struct { + SecretKey string +} + +type AlipayConfig struct { + AppID string + PrivateKey string + AlipayPublicKey string + AppCertPath string // 应用公钥证书路径 + AlipayCertPath string // 支付宝公钥证书路径 + AlipayRootCertPath string // 根证书路径 + IsProduction bool + NotifyUrl string + ReturnURL string +} +type WxpayConfig struct { + AppID string + MchID string + MchCertificateSerialNumber string + MchApiv3Key string + MchPrivateKeyPath string + MchPublicKeyID string + MchPublicKeyPath string + MchPlatformRAS string + NotifyUrl string + RefundNotifyUrl string +} +type ApplepayConfig struct { + ProductionVerifyURL string + SandboxVerifyURL string // 沙盒环境的验证 URL + Sandbox bool + BundleID string + IssuerID string + KeyID string + LoadPrivateKeyPath string +} +type SystemConfig struct { + ThreeVerify bool +} +type WechatH5Config struct { + AppID string + AppSecret string +} +type WechatMiniConfig struct { + AppID string + AppSecret string +} +type QueryConfig struct { + ShareLinkExpire int64 +} +type AdminConfig struct { + AccessSecret string + AccessExpire int64 + RefreshAfter int64 +} + +type TaxConfig struct { + TaxRate float64 + TaxExemptionAmount float64 +} +type TianyuanapiConfig struct { + AccessID string + Key string + BaseURL string + Timeout int64 +} + +type AuthorizationConfig struct { + FileBaseURL string // 授权书文件访问基础URL +} + +// PromotionConfig 推广链接配置 +type PromotionConfig struct { + PromotionDomain string // 推广域名（用于生成短链） + OfficialDomain string // 正式站点域名（短链重定向的目标域名） +} diff --git a/app/main/api/internal/handler/admin_agent/adminauditagenthandler.go b/app/main/api/internal/handler/admin_agent/adminauditagenthandler.go new file mode 100644 index 0000000..42b7c99 --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/adminauditagenthandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminAuditAgentHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminAuditAgentReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminAuditAgentLogic(r.Context(), svcCtx) + resp, err := l.AdminAuditAgent(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/adminauditrealnamehandler.go b/app/main/api/internal/handler/admin_agent/adminauditrealnamehandler.go new file mode 100644 index 0000000..6167bdf --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/adminauditrealnamehandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminAuditRealNameHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminAuditRealNameReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminAuditRealNameLogic(r.Context(), svcCtx) + resp, err := l.AdminAuditRealName(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/adminauditwithdrawalhandler.go b/app/main/api/internal/handler/admin_agent/adminauditwithdrawalhandler.go new file mode 100644 index 0000000..9b658bb --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/adminauditwithdrawalhandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminAuditWithdrawalHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminAuditWithdrawalReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminAuditWithdrawalLogic(r.Context(), svcCtx) + resp, err := l.AdminAuditWithdrawal(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingeneratediamondinvitecodehandler.go b/app/main/api/internal/handler/admin_agent/admingeneratediamondinvitecodehandler.go new file mode 100644 index 0000000..f04e58c --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingeneratediamondinvitecodehandler.go @@ -0,0 +1,29 @@ +package admin_agent + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func AdminGenerateDiamondInviteCodeHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGenerateDiamondInviteCodeReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminGenerateDiamondInviteCodeLogic(r.Context(), svcCtx) + resp, err := l.AdminGenerateDiamondInviteCode(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingetagentcommissionlisthandler.go b/app/main/api/internal/handler/admin_agent/admingetagentcommissionlisthandler.go new file mode 100644 index 0000000..1508e0e --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingetagentcommissionlisthandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetAgentCommissionListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetAgentCommissionListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminGetAgentCommissionListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetAgentCommissionList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingetagentconfighandler.go b/app/main/api/internal/handler/admin_agent/admingetagentconfighandler.go new file mode 100644 index 0000000..0c7a9dd --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingetagentconfighandler.go @@ -0,0 +1,17 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func AdminGetAgentConfigHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := admin_agent.NewAdminGetAgentConfigLogic(r.Context(), svcCtx) + resp, err := l.AdminGetAgentConfig() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingetagentlinklisthandler.go b/app/main/api/internal/handler/admin_agent/admingetagentlinklisthandler.go new file mode 100644 index 0000000..8393926 --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingetagentlinklisthandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetAgentLinkListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetAgentLinkListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminGetAgentLinkListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetAgentLinkList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingetagentlisthandler.go b/app/main/api/internal/handler/admin_agent/admingetagentlisthandler.go new file mode 100644 index 0000000..ebd7986 --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingetagentlisthandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetAgentListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetAgentListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminGetAgentListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetAgentList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingetagentorderlisthandler.go b/app/main/api/internal/handler/admin_agent/admingetagentorderlisthandler.go new file mode 100644 index 0000000..383cd8a --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingetagentorderlisthandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetAgentOrderListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetAgentOrderListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminGetAgentOrderListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetAgentOrderList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingetagentproductconfiglisthandler.go b/app/main/api/internal/handler/admin_agent/admingetagentproductconfiglisthandler.go new file mode 100644 index 0000000..2daa54e --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingetagentproductconfiglisthandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetAgentProductConfigListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetAgentProductConfigListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminGetAgentProductConfigListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetAgentProductConfigList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingetagentrealnamelisthandler.go b/app/main/api/internal/handler/admin_agent/admingetagentrealnamelisthandler.go new file mode 100644 index 0000000..c6f0811 --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingetagentrealnamelisthandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetAgentRealNameListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetAgentRealNameListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminGetAgentRealNameListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetAgentRealNameList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingetagentrebatelisthandler.go b/app/main/api/internal/handler/admin_agent/admingetagentrebatelisthandler.go new file mode 100644 index 0000000..0ac7eed --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingetagentrebatelisthandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetAgentRebateListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetAgentRebateListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminGetAgentRebateListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetAgentRebateList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingetagentupgradelisthandler.go b/app/main/api/internal/handler/admin_agent/admingetagentupgradelisthandler.go new file mode 100644 index 0000000..a377f35 --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingetagentupgradelisthandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetAgentUpgradeListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetAgentUpgradeListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminGetAgentUpgradeListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetAgentUpgradeList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingetagentwithdrawallisthandler.go b/app/main/api/internal/handler/admin_agent/admingetagentwithdrawallisthandler.go new file mode 100644 index 0000000..3fc18a0 --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingetagentwithdrawallisthandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetAgentWithdrawalListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetAgentWithdrawalListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminGetAgentWithdrawalListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetAgentWithdrawalList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/admingetinvitecodelisthandler.go b/app/main/api/internal/handler/admin_agent/admingetinvitecodelisthandler.go new file mode 100644 index 0000000..f4ac72f --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/admingetinvitecodelisthandler.go @@ -0,0 +1,29 @@ +package admin_agent + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func AdminGetInviteCodeListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetInviteCodeListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminGetInviteCodeListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetInviteCodeList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/adminupdateagentconfighandler.go b/app/main/api/internal/handler/admin_agent/adminupdateagentconfighandler.go new file mode 100644 index 0000000..b10f6be --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/adminupdateagentconfighandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdateAgentConfigHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdateAgentConfigReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminUpdateAgentConfigLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdateAgentConfig(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_agent/adminupdateagentproductconfighandler.go b/app/main/api/internal/handler/admin_agent/adminupdateagentproductconfighandler.go new file mode 100644 index 0000000..17d1bcf --- /dev/null +++ b/app/main/api/internal/handler/admin_agent/adminupdateagentproductconfighandler.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdateAgentProductConfigHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdateAgentProductConfigReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_agent.NewAdminUpdateAgentProductConfigLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdateAgentProductConfig(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_api/adminbatchupdateapistatushandler.go b/app/main/api/internal/handler/admin_api/adminbatchupdateapistatushandler.go new file mode 100644 index 0000000..86bda53 --- /dev/null +++ b/app/main/api/internal/handler/admin_api/adminbatchupdateapistatushandler.go @@ -0,0 +1,30 @@ +package admin_api + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_api" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminBatchUpdateApiStatusHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminBatchUpdateApiStatusReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_api.NewAdminBatchUpdateApiStatusLogic(r.Context(), svcCtx) + resp, err := l.AdminBatchUpdateApiStatus(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_api/admincreateapihandler.go b/app/main/api/internal/handler/admin_api/admincreateapihandler.go new file mode 100644 index 0000000..55fac4d --- /dev/null +++ b/app/main/api/internal/handler/admin_api/admincreateapihandler.go @@ -0,0 +1,30 @@ +package admin_api + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_api" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminCreateApiHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminCreateApiReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_api.NewAdminCreateApiLogic(r.Context(), svcCtx) + resp, err := l.AdminCreateApi(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_api/admindeleteapihandler.go b/app/main/api/internal/handler/admin_api/admindeleteapihandler.go new file mode 100644 index 0000000..74240cf --- /dev/null +++ b/app/main/api/internal/handler/admin_api/admindeleteapihandler.go @@ -0,0 +1,30 @@ +package admin_api + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_api" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminDeleteApiHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminDeleteApiReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_api.NewAdminDeleteApiLogic(r.Context(), svcCtx) + resp, err := l.AdminDeleteApi(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_api/admingetapidetailhandler.go b/app/main/api/internal/handler/admin_api/admingetapidetailhandler.go new file mode 100644 index 0000000..87e77c3 --- /dev/null +++ b/app/main/api/internal/handler/admin_api/admingetapidetailhandler.go @@ -0,0 +1,30 @@ +package admin_api + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_api" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetApiDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetApiDetailReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_api.NewAdminGetApiDetailLogic(r.Context(), svcCtx) + resp, err := l.AdminGetApiDetail(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_api/admingetapilisthandler.go b/app/main/api/internal/handler/admin_api/admingetapilisthandler.go new file mode 100644 index 0000000..3af28ac --- /dev/null +++ b/app/main/api/internal/handler/admin_api/admingetapilisthandler.go @@ -0,0 +1,30 @@ +package admin_api + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_api" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetApiListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetApiListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_api.NewAdminGetApiListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetApiList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_api/adminupdateapihandler.go b/app/main/api/internal/handler/admin_api/adminupdateapihandler.go new file mode 100644 index 0000000..774b467 --- /dev/null +++ b/app/main/api/internal/handler/admin_api/adminupdateapihandler.go @@ -0,0 +1,30 @@ +package admin_api + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_api" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdateApiHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdateApiReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_api.NewAdminUpdateApiLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdateApi(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_auth/adminloginhandler.go b/app/main/api/internal/handler/admin_auth/adminloginhandler.go new file mode 100644 index 0000000..5e6f77f --- /dev/null +++ b/app/main/api/internal/handler/admin_auth/adminloginhandler.go @@ -0,0 +1,30 @@ +package admin_auth + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_auth" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminLoginHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminLoginReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_auth.NewAdminLoginLogic(r.Context(), svcCtx) + resp, err := l.AdminLogin(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_feature/adminconfigfeatureexamplehandler.go b/app/main/api/internal/handler/admin_feature/adminconfigfeatureexamplehandler.go new file mode 100644 index 0000000..93083e2 --- /dev/null +++ b/app/main/api/internal/handler/admin_feature/adminconfigfeatureexamplehandler.go @@ -0,0 +1,30 @@ +package admin_feature + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_feature" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminConfigFeatureExampleHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminConfigFeatureExampleReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_feature.NewAdminConfigFeatureExampleLogic(r.Context(), svcCtx) + resp, err := l.AdminConfigFeatureExample(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_feature/admincreatefeaturehandler.go b/app/main/api/internal/handler/admin_feature/admincreatefeaturehandler.go new file mode 100644 index 0000000..8522883 --- /dev/null +++ b/app/main/api/internal/handler/admin_feature/admincreatefeaturehandler.go @@ -0,0 +1,30 @@ +package admin_feature + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_feature" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminCreateFeatureHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminCreateFeatureReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_feature.NewAdminCreateFeatureLogic(r.Context(), svcCtx) + resp, err := l.AdminCreateFeature(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_feature/admindeletefeaturehandler.go b/app/main/api/internal/handler/admin_feature/admindeletefeaturehandler.go new file mode 100644 index 0000000..973df2d --- /dev/null +++ b/app/main/api/internal/handler/admin_feature/admindeletefeaturehandler.go @@ -0,0 +1,30 @@ +package admin_feature + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_feature" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminDeleteFeatureHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminDeleteFeatureReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_feature.NewAdminDeleteFeatureLogic(r.Context(), svcCtx) + resp, err := l.AdminDeleteFeature(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_feature/admingetfeaturedetailhandler.go b/app/main/api/internal/handler/admin_feature/admingetfeaturedetailhandler.go new file mode 100644 index 0000000..d4c78ce --- /dev/null +++ b/app/main/api/internal/handler/admin_feature/admingetfeaturedetailhandler.go @@ -0,0 +1,30 @@ +package admin_feature + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_feature" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetFeatureDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetFeatureDetailReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_feature.NewAdminGetFeatureDetailLogic(r.Context(), svcCtx) + resp, err := l.AdminGetFeatureDetail(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_feature/admingetfeatureexamplehandler.go b/app/main/api/internal/handler/admin_feature/admingetfeatureexamplehandler.go new file mode 100644 index 0000000..d5aea8c --- /dev/null +++ b/app/main/api/internal/handler/admin_feature/admingetfeatureexamplehandler.go @@ -0,0 +1,30 @@ +package admin_feature + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_feature" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetFeatureExampleHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetFeatureExampleReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_feature.NewAdminGetFeatureExampleLogic(r.Context(), svcCtx) + resp, err := l.AdminGetFeatureExample(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_feature/admingetfeaturelisthandler.go b/app/main/api/internal/handler/admin_feature/admingetfeaturelisthandler.go new file mode 100644 index 0000000..b7fba37 --- /dev/null +++ b/app/main/api/internal/handler/admin_feature/admingetfeaturelisthandler.go @@ -0,0 +1,30 @@ +package admin_feature + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_feature" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetFeatureListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetFeatureListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_feature.NewAdminGetFeatureListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetFeatureList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_feature/adminupdatefeaturehandler.go b/app/main/api/internal/handler/admin_feature/adminupdatefeaturehandler.go new file mode 100644 index 0000000..540af7b --- /dev/null +++ b/app/main/api/internal/handler/admin_feature/adminupdatefeaturehandler.go @@ -0,0 +1,30 @@ +package admin_feature + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_feature" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdateFeatureHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdateFeatureReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_feature.NewAdminUpdateFeatureLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdateFeature(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_menu/createmenuhandler.go b/app/main/api/internal/handler/admin_menu/createmenuhandler.go new file mode 100644 index 0000000..cd62bc4 --- /dev/null +++ b/app/main/api/internal/handler/admin_menu/createmenuhandler.go @@ -0,0 +1,30 @@ +package admin_menu + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_menu" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func CreateMenuHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.CreateMenuReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_menu.NewCreateMenuLogic(r.Context(), svcCtx) + resp, err := l.CreateMenu(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_menu/deletemenuhandler.go b/app/main/api/internal/handler/admin_menu/deletemenuhandler.go new file mode 100644 index 0000000..b6ee2d0 --- /dev/null +++ b/app/main/api/internal/handler/admin_menu/deletemenuhandler.go @@ -0,0 +1,30 @@ +package admin_menu + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_menu" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func DeleteMenuHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.DeleteMenuReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_menu.NewDeleteMenuLogic(r.Context(), svcCtx) + resp, err := l.DeleteMenu(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_menu/getmenuallhandler.go b/app/main/api/internal/handler/admin_menu/getmenuallhandler.go new file mode 100644 index 0000000..47985d1 --- /dev/null +++ b/app/main/api/internal/handler/admin_menu/getmenuallhandler.go @@ -0,0 +1,30 @@ +package admin_menu + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_menu" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetMenuAllHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetMenuAllReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_menu.NewGetMenuAllLogic(r.Context(), svcCtx) + resp, err := l.GetMenuAll(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_menu/getmenudetailhandler.go b/app/main/api/internal/handler/admin_menu/getmenudetailhandler.go new file mode 100644 index 0000000..b167051 --- /dev/null +++ b/app/main/api/internal/handler/admin_menu/getmenudetailhandler.go @@ -0,0 +1,30 @@ +package admin_menu + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_menu" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetMenuDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetMenuDetailReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_menu.NewGetMenuDetailLogic(r.Context(), svcCtx) + resp, err := l.GetMenuDetail(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_menu/getmenulisthandler.go b/app/main/api/internal/handler/admin_menu/getmenulisthandler.go new file mode 100644 index 0000000..86aac68 --- /dev/null +++ b/app/main/api/internal/handler/admin_menu/getmenulisthandler.go @@ -0,0 +1,30 @@ +package admin_menu + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_menu" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetMenuListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetMenuListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_menu.NewGetMenuListLogic(r.Context(), svcCtx) + resp, err := l.GetMenuList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_menu/updatemenuhandler.go b/app/main/api/internal/handler/admin_menu/updatemenuhandler.go new file mode 100644 index 0000000..4f5d7b5 --- /dev/null +++ b/app/main/api/internal/handler/admin_menu/updatemenuhandler.go @@ -0,0 +1,30 @@ +package admin_menu + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_menu" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func UpdateMenuHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.UpdateMenuReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_menu.NewUpdateMenuLogic(r.Context(), svcCtx) + resp, err := l.UpdateMenu(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_notification/admincreatenotificationhandler.go b/app/main/api/internal/handler/admin_notification/admincreatenotificationhandler.go new file mode 100644 index 0000000..2f6a523 --- /dev/null +++ b/app/main/api/internal/handler/admin_notification/admincreatenotificationhandler.go @@ -0,0 +1,30 @@ +package admin_notification + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_notification" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminCreateNotificationHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminCreateNotificationReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_notification.NewAdminCreateNotificationLogic(r.Context(), svcCtx) + resp, err := l.AdminCreateNotification(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_notification/admindeletenotificationhandler.go b/app/main/api/internal/handler/admin_notification/admindeletenotificationhandler.go new file mode 100644 index 0000000..33585a8 --- /dev/null +++ b/app/main/api/internal/handler/admin_notification/admindeletenotificationhandler.go @@ -0,0 +1,30 @@ +package admin_notification + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_notification" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminDeleteNotificationHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminDeleteNotificationReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_notification.NewAdminDeleteNotificationLogic(r.Context(), svcCtx) + resp, err := l.AdminDeleteNotification(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_notification/admingetnotificationdetailhandler.go b/app/main/api/internal/handler/admin_notification/admingetnotificationdetailhandler.go new file mode 100644 index 0000000..e3fba3b --- /dev/null +++ b/app/main/api/internal/handler/admin_notification/admingetnotificationdetailhandler.go @@ -0,0 +1,30 @@ +package admin_notification + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_notification" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetNotificationDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetNotificationDetailReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_notification.NewAdminGetNotificationDetailLogic(r.Context(), svcCtx) + resp, err := l.AdminGetNotificationDetail(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_notification/admingetnotificationlisthandler.go b/app/main/api/internal/handler/admin_notification/admingetnotificationlisthandler.go new file mode 100644 index 0000000..8db5cae --- /dev/null +++ b/app/main/api/internal/handler/admin_notification/admingetnotificationlisthandler.go @@ -0,0 +1,30 @@ +package admin_notification + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_notification" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetNotificationListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetNotificationListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_notification.NewAdminGetNotificationListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetNotificationList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_notification/adminupdatenotificationhandler.go b/app/main/api/internal/handler/admin_notification/adminupdatenotificationhandler.go new file mode 100644 index 0000000..14f0e41 --- /dev/null +++ b/app/main/api/internal/handler/admin_notification/adminupdatenotificationhandler.go @@ -0,0 +1,30 @@ +package admin_notification + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_notification" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdateNotificationHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdateNotificationReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_notification.NewAdminUpdateNotificationLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdateNotification(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_order/admincreateorderhandler.go b/app/main/api/internal/handler/admin_order/admincreateorderhandler.go new file mode 100644 index 0000000..2c0a260 --- /dev/null +++ b/app/main/api/internal/handler/admin_order/admincreateorderhandler.go @@ -0,0 +1,30 @@ +package admin_order + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_order" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminCreateOrderHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminCreateOrderReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_order.NewAdminCreateOrderLogic(r.Context(), svcCtx) + resp, err := l.AdminCreateOrder(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_order/admindeleteorderhandler.go b/app/main/api/internal/handler/admin_order/admindeleteorderhandler.go new file mode 100644 index 0000000..1e73b7f --- /dev/null +++ b/app/main/api/internal/handler/admin_order/admindeleteorderhandler.go @@ -0,0 +1,30 @@ +package admin_order + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_order" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminDeleteOrderHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminDeleteOrderReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_order.NewAdminDeleteOrderLogic(r.Context(), svcCtx) + resp, err := l.AdminDeleteOrder(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_order/admingetorderdetailhandler.go b/app/main/api/internal/handler/admin_order/admingetorderdetailhandler.go new file mode 100644 index 0000000..f6e3206 --- /dev/null +++ b/app/main/api/internal/handler/admin_order/admingetorderdetailhandler.go @@ -0,0 +1,30 @@ +package admin_order + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_order" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetOrderDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetOrderDetailReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_order.NewAdminGetOrderDetailLogic(r.Context(), svcCtx) + resp, err := l.AdminGetOrderDetail(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_order/admingetorderlisthandler.go b/app/main/api/internal/handler/admin_order/admingetorderlisthandler.go new file mode 100644 index 0000000..b87dde2 --- /dev/null +++ b/app/main/api/internal/handler/admin_order/admingetorderlisthandler.go @@ -0,0 +1,30 @@ +package admin_order + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_order" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetOrderListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetOrderListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_order.NewAdminGetOrderListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetOrderList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_order/adminrefundorderhandler.go b/app/main/api/internal/handler/admin_order/adminrefundorderhandler.go new file mode 100644 index 0000000..3ed3a51 --- /dev/null +++ b/app/main/api/internal/handler/admin_order/adminrefundorderhandler.go @@ -0,0 +1,30 @@ +package admin_order + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_order" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminRefundOrderHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminRefundOrderReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_order.NewAdminRefundOrderLogic(r.Context(), svcCtx) + resp, err := l.AdminRefundOrder(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_order/adminretryagentprocesshandler.go b/app/main/api/internal/handler/admin_order/adminretryagentprocesshandler.go new file mode 100644 index 0000000..734f760 --- /dev/null +++ b/app/main/api/internal/handler/admin_order/adminretryagentprocesshandler.go @@ -0,0 +1,29 @@ +package admin_order + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/admin_order" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func AdminRetryAgentProcessHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminRetryAgentProcessReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_order.NewAdminRetryAgentProcessLogic(r.Context(), svcCtx) + resp, err := l.AdminRetryAgentProcess(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_order/adminupdateorderhandler.go b/app/main/api/internal/handler/admin_order/adminupdateorderhandler.go new file mode 100644 index 0000000..443f13c --- /dev/null +++ b/app/main/api/internal/handler/admin_order/adminupdateorderhandler.go @@ -0,0 +1,30 @@ +package admin_order + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_order" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdateOrderHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdateOrderReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_order.NewAdminUpdateOrderLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdateOrder(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_platform_user/admincreateplatformuserhandler.go b/app/main/api/internal/handler/admin_platform_user/admincreateplatformuserhandler.go new file mode 100644 index 0000000..5cdf26b --- /dev/null +++ b/app/main/api/internal/handler/admin_platform_user/admincreateplatformuserhandler.go @@ -0,0 +1,30 @@ +package admin_platform_user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_platform_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminCreatePlatformUserHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminCreatePlatformUserReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_platform_user.NewAdminCreatePlatformUserLogic(r.Context(), svcCtx) + resp, err := l.AdminCreatePlatformUser(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_platform_user/admindeleteplatformuserhandler.go b/app/main/api/internal/handler/admin_platform_user/admindeleteplatformuserhandler.go new file mode 100644 index 0000000..d68faca --- /dev/null +++ b/app/main/api/internal/handler/admin_platform_user/admindeleteplatformuserhandler.go @@ -0,0 +1,30 @@ +package admin_platform_user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_platform_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminDeletePlatformUserHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminDeletePlatformUserReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_platform_user.NewAdminDeletePlatformUserLogic(r.Context(), svcCtx) + resp, err := l.AdminDeletePlatformUser(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_platform_user/admingetplatformuserdetailhandler.go b/app/main/api/internal/handler/admin_platform_user/admingetplatformuserdetailhandler.go new file mode 100644 index 0000000..0a75b02 --- /dev/null +++ b/app/main/api/internal/handler/admin_platform_user/admingetplatformuserdetailhandler.go @@ -0,0 +1,30 @@ +package admin_platform_user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_platform_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetPlatformUserDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetPlatformUserDetailReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_platform_user.NewAdminGetPlatformUserDetailLogic(r.Context(), svcCtx) + resp, err := l.AdminGetPlatformUserDetail(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_platform_user/admingetplatformuserlisthandler.go b/app/main/api/internal/handler/admin_platform_user/admingetplatformuserlisthandler.go new file mode 100644 index 0000000..1fdf121 --- /dev/null +++ b/app/main/api/internal/handler/admin_platform_user/admingetplatformuserlisthandler.go @@ -0,0 +1,30 @@ +package admin_platform_user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_platform_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetPlatformUserListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetPlatformUserListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_platform_user.NewAdminGetPlatformUserListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetPlatformUserList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_platform_user/adminupdateplatformuserhandler.go b/app/main/api/internal/handler/admin_platform_user/adminupdateplatformuserhandler.go new file mode 100644 index 0000000..cb2a40d --- /dev/null +++ b/app/main/api/internal/handler/admin_platform_user/adminupdateplatformuserhandler.go @@ -0,0 +1,30 @@ +package admin_platform_user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_platform_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdatePlatformUserHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdatePlatformUserReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_platform_user.NewAdminUpdatePlatformUserLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdatePlatformUser(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_product/admincreateproducthandler.go b/app/main/api/internal/handler/admin_product/admincreateproducthandler.go new file mode 100644 index 0000000..c20563f --- /dev/null +++ b/app/main/api/internal/handler/admin_product/admincreateproducthandler.go @@ -0,0 +1,30 @@ +package admin_product + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_product" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminCreateProductHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminCreateProductReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_product.NewAdminCreateProductLogic(r.Context(), svcCtx) + resp, err := l.AdminCreateProduct(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_product/admindeleteproducthandler.go b/app/main/api/internal/handler/admin_product/admindeleteproducthandler.go new file mode 100644 index 0000000..cc5a55d --- /dev/null +++ b/app/main/api/internal/handler/admin_product/admindeleteproducthandler.go @@ -0,0 +1,30 @@ +package admin_product + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_product" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminDeleteProductHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminDeleteProductReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_product.NewAdminDeleteProductLogic(r.Context(), svcCtx) + resp, err := l.AdminDeleteProduct(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_product/admingetproductdetailhandler.go b/app/main/api/internal/handler/admin_product/admingetproductdetailhandler.go new file mode 100644 index 0000000..7113218 --- /dev/null +++ b/app/main/api/internal/handler/admin_product/admingetproductdetailhandler.go @@ -0,0 +1,30 @@ +package admin_product + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_product" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetProductDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetProductDetailReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_product.NewAdminGetProductDetailLogic(r.Context(), svcCtx) + resp, err := l.AdminGetProductDetail(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_product/admingetproductfeaturelisthandler.go b/app/main/api/internal/handler/admin_product/admingetproductfeaturelisthandler.go new file mode 100644 index 0000000..b1d1d31 --- /dev/null +++ b/app/main/api/internal/handler/admin_product/admingetproductfeaturelisthandler.go @@ -0,0 +1,30 @@ +package admin_product + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_product" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetProductFeatureListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetProductFeatureListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_product.NewAdminGetProductFeatureListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetProductFeatureList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_product/admingetproductlisthandler.go b/app/main/api/internal/handler/admin_product/admingetproductlisthandler.go new file mode 100644 index 0000000..9b82722 --- /dev/null +++ b/app/main/api/internal/handler/admin_product/admingetproductlisthandler.go @@ -0,0 +1,30 @@ +package admin_product + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_product" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetProductListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetProductListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_product.NewAdminGetProductListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetProductList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_product/adminupdateproductfeatureshandler.go b/app/main/api/internal/handler/admin_product/adminupdateproductfeatureshandler.go new file mode 100644 index 0000000..e8ad28c --- /dev/null +++ b/app/main/api/internal/handler/admin_product/adminupdateproductfeatureshandler.go @@ -0,0 +1,30 @@ +package admin_product + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_product" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdateProductFeaturesHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdateProductFeaturesReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_product.NewAdminUpdateProductFeaturesLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdateProductFeatures(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_product/adminupdateproducthandler.go b/app/main/api/internal/handler/admin_product/adminupdateproducthandler.go new file mode 100644 index 0000000..aefe484 --- /dev/null +++ b/app/main/api/internal/handler/admin_product/adminupdateproducthandler.go @@ -0,0 +1,30 @@ +package admin_product + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_product" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdateProductHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdateProductReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_product.NewAdminUpdateProductLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdateProduct(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_query/admingetquerycleanupconfiglisthandler.go b/app/main/api/internal/handler/admin_query/admingetquerycleanupconfiglisthandler.go new file mode 100644 index 0000000..f3ba709 --- /dev/null +++ b/app/main/api/internal/handler/admin_query/admingetquerycleanupconfiglisthandler.go @@ -0,0 +1,30 @@ +package admin_query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetQueryCleanupConfigListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetQueryCleanupConfigListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_query.NewAdminGetQueryCleanupConfigListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetQueryCleanupConfigList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_query/admingetquerycleanupdetaillisthandler.go b/app/main/api/internal/handler/admin_query/admingetquerycleanupdetaillisthandler.go new file mode 100644 index 0000000..c5e0aa9 --- /dev/null +++ b/app/main/api/internal/handler/admin_query/admingetquerycleanupdetaillisthandler.go @@ -0,0 +1,30 @@ +package admin_query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetQueryCleanupDetailListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetQueryCleanupDetailListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_query.NewAdminGetQueryCleanupDetailListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetQueryCleanupDetailList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_query/admingetquerycleanuploglisthandler.go b/app/main/api/internal/handler/admin_query/admingetquerycleanuploglisthandler.go new file mode 100644 index 0000000..e3ccc3e --- /dev/null +++ b/app/main/api/internal/handler/admin_query/admingetquerycleanuploglisthandler.go @@ -0,0 +1,30 @@ +package admin_query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetQueryCleanupLogListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetQueryCleanupLogListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_query.NewAdminGetQueryCleanupLogListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetQueryCleanupLogList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_query/admingetquerydetailbyorderidhandler.go b/app/main/api/internal/handler/admin_query/admingetquerydetailbyorderidhandler.go new file mode 100644 index 0000000..f36c0cd --- /dev/null +++ b/app/main/api/internal/handler/admin_query/admingetquerydetailbyorderidhandler.go @@ -0,0 +1,30 @@ +package admin_query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetQueryDetailByOrderIdHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetQueryDetailByOrderIdReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_query.NewAdminGetQueryDetailByOrderIdLogic(r.Context(), svcCtx) + resp, err := l.AdminGetQueryDetailByOrderId(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_query/adminupdatequerycleanupconfighandler.go b/app/main/api/internal/handler/admin_query/adminupdatequerycleanupconfighandler.go new file mode 100644 index 0000000..8153bed --- /dev/null +++ b/app/main/api/internal/handler/admin_query/adminupdatequerycleanupconfighandler.go @@ -0,0 +1,30 @@ +package admin_query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdateQueryCleanupConfigHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdateQueryCleanupConfigReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_query.NewAdminUpdateQueryCleanupConfigLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdateQueryCleanupConfig(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_role/createrolehandler.go b/app/main/api/internal/handler/admin_role/createrolehandler.go new file mode 100644 index 0000000..01350a1 --- /dev/null +++ b/app/main/api/internal/handler/admin_role/createrolehandler.go @@ -0,0 +1,30 @@ +package admin_role + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_role" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func CreateRoleHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.CreateRoleReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_role.NewCreateRoleLogic(r.Context(), svcCtx) + resp, err := l.CreateRole(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_role/deleterolehandler.go b/app/main/api/internal/handler/admin_role/deleterolehandler.go new file mode 100644 index 0000000..3d0cd25 --- /dev/null +++ b/app/main/api/internal/handler/admin_role/deleterolehandler.go @@ -0,0 +1,30 @@ +package admin_role + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_role" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func DeleteRoleHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.DeleteRoleReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_role.NewDeleteRoleLogic(r.Context(), svcCtx) + resp, err := l.DeleteRole(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_role/getroledetailhandler.go b/app/main/api/internal/handler/admin_role/getroledetailhandler.go new file mode 100644 index 0000000..e2f0c67 --- /dev/null +++ b/app/main/api/internal/handler/admin_role/getroledetailhandler.go @@ -0,0 +1,30 @@ +package admin_role + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_role" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetRoleDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetRoleDetailReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_role.NewGetRoleDetailLogic(r.Context(), svcCtx) + resp, err := l.GetRoleDetail(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_role/getrolelisthandler.go b/app/main/api/internal/handler/admin_role/getrolelisthandler.go new file mode 100644 index 0000000..ba34aea --- /dev/null +++ b/app/main/api/internal/handler/admin_role/getrolelisthandler.go @@ -0,0 +1,30 @@ +package admin_role + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_role" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetRoleListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetRoleListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_role.NewGetRoleListLogic(r.Context(), svcCtx) + resp, err := l.GetRoleList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_role/updaterolehandler.go b/app/main/api/internal/handler/admin_role/updaterolehandler.go new file mode 100644 index 0000000..5b61959 --- /dev/null +++ b/app/main/api/internal/handler/admin_role/updaterolehandler.go @@ -0,0 +1,30 @@ +package admin_role + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_role" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func UpdateRoleHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.UpdateRoleReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_role.NewUpdateRoleLogic(r.Context(), svcCtx) + resp, err := l.UpdateRole(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_role_api/adminassignroleapihandler.go b/app/main/api/internal/handler/admin_role_api/adminassignroleapihandler.go new file mode 100644 index 0000000..18d3b0e --- /dev/null +++ b/app/main/api/internal/handler/admin_role_api/adminassignroleapihandler.go @@ -0,0 +1,30 @@ +package admin_role_api + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_role_api" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminAssignRoleApiHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminAssignRoleApiReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_role_api.NewAdminAssignRoleApiLogic(r.Context(), svcCtx) + resp, err := l.AdminAssignRoleApi(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_role_api/admingetallapilisthandler.go b/app/main/api/internal/handler/admin_role_api/admingetallapilisthandler.go new file mode 100644 index 0000000..24ba487 --- /dev/null +++ b/app/main/api/internal/handler/admin_role_api/admingetallapilisthandler.go @@ -0,0 +1,30 @@ +package admin_role_api + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_role_api" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetAllApiListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetAllApiListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_role_api.NewAdminGetAllApiListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetAllApiList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_role_api/admingetroleapilisthandler.go b/app/main/api/internal/handler/admin_role_api/admingetroleapilisthandler.go new file mode 100644 index 0000000..63e226d --- /dev/null +++ b/app/main/api/internal/handler/admin_role_api/admingetroleapilisthandler.go @@ -0,0 +1,30 @@ +package admin_role_api + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_role_api" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetRoleApiListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetRoleApiListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_role_api.NewAdminGetRoleApiListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetRoleApiList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_role_api/adminremoveroleapihandler.go b/app/main/api/internal/handler/admin_role_api/adminremoveroleapihandler.go new file mode 100644 index 0000000..0e3300c --- /dev/null +++ b/app/main/api/internal/handler/admin_role_api/adminremoveroleapihandler.go @@ -0,0 +1,30 @@ +package admin_role_api + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_role_api" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminRemoveRoleApiHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminRemoveRoleApiReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_role_api.NewAdminRemoveRoleApiLogic(r.Context(), svcCtx) + resp, err := l.AdminRemoveRoleApi(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_role_api/adminupdateroleapihandler.go b/app/main/api/internal/handler/admin_role_api/adminupdateroleapihandler.go new file mode 100644 index 0000000..e4b55d4 --- /dev/null +++ b/app/main/api/internal/handler/admin_role_api/adminupdateroleapihandler.go @@ -0,0 +1,30 @@ +package admin_role_api + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_role_api" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdateRoleApiHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdateRoleApiReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_role_api.NewAdminUpdateRoleApiLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdateRoleApi(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_user/admincreateuserhandler.go b/app/main/api/internal/handler/admin_user/admincreateuserhandler.go new file mode 100644 index 0000000..3a3f93e --- /dev/null +++ b/app/main/api/internal/handler/admin_user/admincreateuserhandler.go @@ -0,0 +1,30 @@ +package admin_user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminCreateUserHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminCreateUserReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_user.NewAdminCreateUserLogic(r.Context(), svcCtx) + resp, err := l.AdminCreateUser(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_user/admindeleteuserhandler.go b/app/main/api/internal/handler/admin_user/admindeleteuserhandler.go new file mode 100644 index 0000000..b7f47a5 --- /dev/null +++ b/app/main/api/internal/handler/admin_user/admindeleteuserhandler.go @@ -0,0 +1,30 @@ +package admin_user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminDeleteUserHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminDeleteUserReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_user.NewAdminDeleteUserLogic(r.Context(), svcCtx) + resp, err := l.AdminDeleteUser(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_user/admingetuserdetailhandler.go b/app/main/api/internal/handler/admin_user/admingetuserdetailhandler.go new file mode 100644 index 0000000..47b5784 --- /dev/null +++ b/app/main/api/internal/handler/admin_user/admingetuserdetailhandler.go @@ -0,0 +1,30 @@ +package admin_user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetUserDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetUserDetailReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_user.NewAdminGetUserDetailLogic(r.Context(), svcCtx) + resp, err := l.AdminGetUserDetail(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_user/admingetuserlisthandler.go b/app/main/api/internal/handler/admin_user/admingetuserlisthandler.go new file mode 100644 index 0000000..cce7bcd --- /dev/null +++ b/app/main/api/internal/handler/admin_user/admingetuserlisthandler.go @@ -0,0 +1,30 @@ +package admin_user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminGetUserListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminGetUserListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_user.NewAdminGetUserListLogic(r.Context(), svcCtx) + resp, err := l.AdminGetUserList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_user/adminresetpasswordhandler.go b/app/main/api/internal/handler/admin_user/adminresetpasswordhandler.go new file mode 100644 index 0000000..5ac07f5 --- /dev/null +++ b/app/main/api/internal/handler/admin_user/adminresetpasswordhandler.go @@ -0,0 +1,29 @@ +package admin_user + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/admin_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func AdminResetPasswordHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminResetPasswordReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_user.NewAdminResetPasswordLogic(r.Context(), svcCtx) + resp, err := l.AdminResetPassword(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_user/adminupdateuserhandler.go b/app/main/api/internal/handler/admin_user/adminupdateuserhandler.go new file mode 100644 index 0000000..d8b8d09 --- /dev/null +++ b/app/main/api/internal/handler/admin_user/adminupdateuserhandler.go @@ -0,0 +1,30 @@ +package admin_user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUpdateUserHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUpdateUserReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_user.NewAdminUpdateUserLogic(r.Context(), svcCtx) + resp, err := l.AdminUpdateUser(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/admin_user/adminuserinfohandler.go b/app/main/api/internal/handler/admin_user/adminuserinfohandler.go new file mode 100644 index 0000000..6b79247 --- /dev/null +++ b/app/main/api/internal/handler/admin_user/adminuserinfohandler.go @@ -0,0 +1,30 @@ +package admin_user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/admin_user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AdminUserInfoHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AdminUserInfoReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := admin_user.NewAdminUserInfoLogic(r.Context(), svcCtx) + resp, err := l.AdminUserInfo(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/applyforagenthandler.go b/app/main/api/internal/handler/agent/applyforagenthandler.go new file mode 100644 index 0000000..bf48b85 --- /dev/null +++ b/app/main/api/internal/handler/agent/applyforagenthandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func ApplyForAgentHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AgentApplyReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewApplyForAgentLogic(r.Context(), svcCtx) + resp, err := l.ApplyForAgent(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/applyupgradehandler.go b/app/main/api/internal/handler/agent/applyupgradehandler.go new file mode 100644 index 0000000..abb4bf4 --- /dev/null +++ b/app/main/api/internal/handler/agent/applyupgradehandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func ApplyUpgradeHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.ApplyUpgradeReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewApplyUpgradeLogic(r.Context(), svcCtx) + resp, err := l.ApplyUpgrade(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/applywithdrawalhandler.go b/app/main/api/internal/handler/agent/applywithdrawalhandler.go new file mode 100644 index 0000000..d90b7be --- /dev/null +++ b/app/main/api/internal/handler/agent/applywithdrawalhandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func ApplyWithdrawalHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.ApplyWithdrawalReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewApplyWithdrawalLogic(r.Context(), svcCtx) + resp, err := l.ApplyWithdrawal(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/deleteinvitecodehandler.go b/app/main/api/internal/handler/agent/deleteinvitecodehandler.go new file mode 100644 index 0000000..166d5e7 --- /dev/null +++ b/app/main/api/internal/handler/agent/deleteinvitecodehandler.go @@ -0,0 +1,29 @@ +package agent + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func DeleteInviteCodeHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.DeleteInviteCodeReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewDeleteInviteCodeLogic(r.Context(), svcCtx) + resp, err := l.DeleteInviteCode(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/generateinvitecodehandler.go b/app/main/api/internal/handler/agent/generateinvitecodehandler.go new file mode 100644 index 0000000..8e84ef4 --- /dev/null +++ b/app/main/api/internal/handler/agent/generateinvitecodehandler.go @@ -0,0 +1,29 @@ +package agent + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func GenerateInviteCodeHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GenerateInviteCodeReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGenerateInviteCodeLogic(r.Context(), svcCtx) + resp, err := l.GenerateInviteCode(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/generatinglinkhandler.go b/app/main/api/internal/handler/agent/generatinglinkhandler.go new file mode 100644 index 0000000..3030876 --- /dev/null +++ b/app/main/api/internal/handler/agent/generatinglinkhandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GeneratingLinkHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AgentGeneratingLinkReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGeneratingLinkLogic(r.Context(), svcCtx) + resp, err := l.GeneratingLink(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getagentinfohandler.go b/app/main/api/internal/handler/agent/getagentinfohandler.go new file mode 100644 index 0000000..b258afe --- /dev/null +++ b/app/main/api/internal/handler/agent/getagentinfohandler.go @@ -0,0 +1,17 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func GetAgentInfoHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := agent.NewGetAgentInfoLogic(r.Context(), svcCtx) + resp, err := l.GetAgentInfo() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getagentproductconfighandler.go b/app/main/api/internal/handler/agent/getagentproductconfighandler.go new file mode 100644 index 0000000..3f1066a --- /dev/null +++ b/app/main/api/internal/handler/agent/getagentproductconfighandler.go @@ -0,0 +1,17 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func GetAgentProductConfigHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := agent.NewGetAgentProductConfigLogic(r.Context(), svcCtx) + resp, err := l.GetAgentProductConfig() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getcommissionlisthandler.go b/app/main/api/internal/handler/agent/getcommissionlisthandler.go new file mode 100644 index 0000000..1a7cfba --- /dev/null +++ b/app/main/api/internal/handler/agent/getcommissionlisthandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetCommissionListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetCommissionListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetCommissionListLogic(r.Context(), svcCtx) + resp, err := l.GetCommissionList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getconversionratehandler.go b/app/main/api/internal/handler/agent/getconversionratehandler.go new file mode 100644 index 0000000..1c91a91 --- /dev/null +++ b/app/main/api/internal/handler/agent/getconversionratehandler.go @@ -0,0 +1,17 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func GetConversionRateHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := agent.NewGetConversionRateLogic(r.Context(), svcCtx) + resp, err := l.GetConversionRate() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getinvitecodelisthandler.go b/app/main/api/internal/handler/agent/getinvitecodelisthandler.go new file mode 100644 index 0000000..7ee4eb8 --- /dev/null +++ b/app/main/api/internal/handler/agent/getinvitecodelisthandler.go @@ -0,0 +1,29 @@ +package agent + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func GetInviteCodeListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetInviteCodeListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetInviteCodeListLogic(r.Context(), svcCtx) + resp, err := l.GetInviteCodeList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getinvitelinkhandler.go b/app/main/api/internal/handler/agent/getinvitelinkhandler.go new file mode 100644 index 0000000..e475636 --- /dev/null +++ b/app/main/api/internal/handler/agent/getinvitelinkhandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetInviteLinkHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetInviteLinkReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetInviteLinkLogic(r.Context(), svcCtx) + resp, err := l.GetInviteLink(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getlevelprivilegehandler.go b/app/main/api/internal/handler/agent/getlevelprivilegehandler.go new file mode 100644 index 0000000..87c28c3 --- /dev/null +++ b/app/main/api/internal/handler/agent/getlevelprivilegehandler.go @@ -0,0 +1,18 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func GetLevelPrivilegeHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := agent.NewGetLevelPrivilegeLogic(r.Context(), svcCtx) + resp, err := l.GetLevelPrivilege() + result.HttpResult(r, w, resp, err) + } +} + diff --git a/app/main/api/internal/handler/agent/getlinkdatahandler.go b/app/main/api/internal/handler/agent/getlinkdatahandler.go new file mode 100644 index 0000000..232fe91 --- /dev/null +++ b/app/main/api/internal/handler/agent/getlinkdatahandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetLinkDataHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetLinkDataReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetLinkDataLogic(r.Context(), svcCtx) + resp, err := l.GetLinkData(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getpromotionquerylisthandler.go b/app/main/api/internal/handler/agent/getpromotionquerylisthandler.go new file mode 100644 index 0000000..eda0f67 --- /dev/null +++ b/app/main/api/internal/handler/agent/getpromotionquerylisthandler.go @@ -0,0 +1,31 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetPromotionQueryListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetPromotionQueryListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetPromotionQueryListLogic(r.Context(), svcCtx) + resp, err := l.GetPromotionQueryList(&req) + result.HttpResult(r, w, resp, err) + } +} + diff --git a/app/main/api/internal/handler/agent/getrebatelisthandler.go b/app/main/api/internal/handler/agent/getrebatelisthandler.go new file mode 100644 index 0000000..c107b06 --- /dev/null +++ b/app/main/api/internal/handler/agent/getrebatelisthandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetRebateListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetRebateListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetRebateListLogic(r.Context(), svcCtx) + resp, err := l.GetRebateList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getrevenueinfohandler.go b/app/main/api/internal/handler/agent/getrevenueinfohandler.go new file mode 100644 index 0000000..561dd78 --- /dev/null +++ b/app/main/api/internal/handler/agent/getrevenueinfohandler.go @@ -0,0 +1,17 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func GetRevenueInfoHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := agent.NewGetRevenueInfoLogic(r.Context(), svcCtx) + resp, err := l.GetRevenueInfo() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getsubordinatecontributiondetailhandler.go b/app/main/api/internal/handler/agent/getsubordinatecontributiondetailhandler.go new file mode 100644 index 0000000..17b8dfc --- /dev/null +++ b/app/main/api/internal/handler/agent/getsubordinatecontributiondetailhandler.go @@ -0,0 +1,29 @@ +package agent + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func GetSubordinateContributionDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetSubordinateContributionDetailReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetSubordinateContributionDetailLogic(r.Context(), svcCtx) + resp, err := l.GetSubordinateContributionDetail(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getsubordinatelisthandler.go b/app/main/api/internal/handler/agent/getsubordinatelisthandler.go new file mode 100644 index 0000000..509c13c --- /dev/null +++ b/app/main/api/internal/handler/agent/getsubordinatelisthandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetSubordinateListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetSubordinateListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetSubordinateListLogic(r.Context(), svcCtx) + resp, err := l.GetSubordinateList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getteamlisthandler.go b/app/main/api/internal/handler/agent/getteamlisthandler.go new file mode 100644 index 0000000..3472287 --- /dev/null +++ b/app/main/api/internal/handler/agent/getteamlisthandler.go @@ -0,0 +1,29 @@ +package agent + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func GetTeamListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetTeamListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetTeamListLogic(r.Context(), svcCtx) + resp, err := l.GetTeamList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getteamstatisticshandler.go b/app/main/api/internal/handler/agent/getteamstatisticshandler.go new file mode 100644 index 0000000..f5ba9c4 --- /dev/null +++ b/app/main/api/internal/handler/agent/getteamstatisticshandler.go @@ -0,0 +1,17 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func GetTeamStatisticsHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := agent.NewGetTeamStatisticsLogic(r.Context(), svcCtx) + resp, err := l.GetTeamStatistics() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getupgradelisthandler.go b/app/main/api/internal/handler/agent/getupgradelisthandler.go new file mode 100644 index 0000000..8d2c321 --- /dev/null +++ b/app/main/api/internal/handler/agent/getupgradelisthandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetUpgradeListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetUpgradeListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetUpgradeListLogic(r.Context(), svcCtx) + resp, err := l.GetUpgradeList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getupgraderebatelisthandler.go b/app/main/api/internal/handler/agent/getupgraderebatelisthandler.go new file mode 100644 index 0000000..fe86135 --- /dev/null +++ b/app/main/api/internal/handler/agent/getupgraderebatelisthandler.go @@ -0,0 +1,29 @@ +package agent + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func GetUpgradeRebateListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetUpgradeRebateListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetUpgradeRebateListLogic(r.Context(), svcCtx) + resp, err := l.GetUpgradeRebateList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/getwithdrawallisthandler.go b/app/main/api/internal/handler/agent/getwithdrawallisthandler.go new file mode 100644 index 0000000..9fbd5a4 --- /dev/null +++ b/app/main/api/internal/handler/agent/getwithdrawallisthandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetWithdrawalListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetWithdrawalListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewGetWithdrawalListLogic(r.Context(), svcCtx) + resp, err := l.GetWithdrawalList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/promotionredirecthandler.go b/app/main/api/internal/handler/agent/promotionredirecthandler.go new file mode 100644 index 0000000..0f650b4 --- /dev/null +++ b/app/main/api/internal/handler/agent/promotionredirecthandler.go @@ -0,0 +1,21 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func PromotionRedirectHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := agent.NewPromotionRedirectLogic(r.Context(), svcCtx) + err := l.PromotionRedirect(r, w) + if err != nil { + httpx.ErrorCtx(r.Context(), w, err) + } + } +} + diff --git a/app/main/api/internal/handler/agent/realnameauthhandler.go b/app/main/api/internal/handler/agent/realnameauthhandler.go new file mode 100644 index 0000000..9e5da90 --- /dev/null +++ b/app/main/api/internal/handler/agent/realnameauthhandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func RealNameAuthHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.RealNameAuthReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewRealNameAuthLogic(r.Context(), svcCtx) + resp, err := l.RealNameAuth(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/registerbyinvitecodehandler.go b/app/main/api/internal/handler/agent/registerbyinvitecodehandler.go new file mode 100644 index 0000000..9c73ff9 --- /dev/null +++ b/app/main/api/internal/handler/agent/registerbyinvitecodehandler.go @@ -0,0 +1,29 @@ +package agent + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func RegisterByInviteCodeHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.RegisterByInviteCodeReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewRegisterByInviteCodeLogic(r.Context(), svcCtx) + resp, err := l.RegisterByInviteCode(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/agent/shortlinkredirecthandler.go b/app/main/api/internal/handler/agent/shortlinkredirecthandler.go new file mode 100644 index 0000000..542791c --- /dev/null +++ b/app/main/api/internal/handler/agent/shortlinkredirecthandler.go @@ -0,0 +1,29 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func ShortLinkRedirectHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + // 从URL.Path解析shortCode + // 路径格式：/s/{shortCode} + path := r.URL.Path + var shortCode string + if len(path) >= 3 && path[:3] == "/s/" { + shortCode = path[3:] + } + + l := agent.NewShortLinkRedirectLogic(r.Context(), svcCtx) + err := l.ShortLinkRedirect(shortCode, r, w) + if err != nil { + httpx.ErrorCtx(r.Context(), w, err) + } + } +} + diff --git a/app/main/api/internal/handler/agent/upgradesubordinatehandler.go b/app/main/api/internal/handler/agent/upgradesubordinatehandler.go new file mode 100644 index 0000000..17236d8 --- /dev/null +++ b/app/main/api/internal/handler/agent/upgradesubordinatehandler.go @@ -0,0 +1,30 @@ +package agent + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/agent" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func UpgradeSubordinateHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.UpgradeSubordinateReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := agent.NewUpgradeSubordinateLogic(r.Context(), svcCtx) + resp, err := l.UpgradeSubordinate(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/app/getappconfighandler.go b/app/main/api/internal/handler/app/getappconfighandler.go new file mode 100644 index 0000000..5271c94 --- /dev/null +++ b/app/main/api/internal/handler/app/getappconfighandler.go @@ -0,0 +1,17 @@ +package app + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/app" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func GetAppConfigHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := app.NewGetAppConfigLogic(r.Context(), svcCtx) + resp, err := l.GetAppConfig() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/app/getappversionhandler.go b/app/main/api/internal/handler/app/getappversionhandler.go new file mode 100644 index 0000000..e339907 --- /dev/null +++ b/app/main/api/internal/handler/app/getappversionhandler.go @@ -0,0 +1,17 @@ +package app + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/app" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func GetAppVersionHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := app.NewGetAppVersionLogic(r.Context(), svcCtx) + resp, err := l.GetAppVersion() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/app/healthcheckhandler.go b/app/main/api/internal/handler/app/healthcheckhandler.go new file mode 100644 index 0000000..d9442a8 --- /dev/null +++ b/app/main/api/internal/handler/app/healthcheckhandler.go @@ -0,0 +1,17 @@ +package app + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/app" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func HealthCheckHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := app.NewHealthCheckLogic(r.Context(), svcCtx) + resp, err := l.HealthCheck() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/auth/sendsmshandler.go b/app/main/api/internal/handler/auth/sendsmshandler.go new file mode 100644 index 0000000..ccd5aa3 --- /dev/null +++ b/app/main/api/internal/handler/auth/sendsmshandler.go @@ -0,0 +1,30 @@ +package auth + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/auth" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func SendSmsHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.SendSmsReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := auth.NewSendSmsLogic(r.Context(), svcCtx) + err := l.SendSms(&req) + result.HttpResult(r, w, nil, err) + } +} diff --git a/app/main/api/internal/handler/authorization/downloadauthorizationdocumenthandler.go b/app/main/api/internal/handler/authorization/downloadauthorizationdocumenthandler.go new file mode 100644 index 0000000..89990b9 --- /dev/null +++ b/app/main/api/internal/handler/authorization/downloadauthorizationdocumenthandler.go @@ -0,0 +1,30 @@ +package authorization + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/authorization" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func DownloadAuthorizationDocumentHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.DownloadAuthorizationDocumentReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := authorization.NewDownloadAuthorizationDocumentLogic(r.Context(), svcCtx) + resp, err := l.DownloadAuthorizationDocument(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/authorization/getauthorizationdocumentbyorderhandler.go b/app/main/api/internal/handler/authorization/getauthorizationdocumentbyorderhandler.go new file mode 100644 index 0000000..b981861 --- /dev/null +++ b/app/main/api/internal/handler/authorization/getauthorizationdocumentbyorderhandler.go @@ -0,0 +1,30 @@ +package authorization + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/authorization" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetAuthorizationDocumentByOrderHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetAuthorizationDocumentByOrderReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := authorization.NewGetAuthorizationDocumentByOrderLogic(r.Context(), svcCtx) + resp, err := l.GetAuthorizationDocumentByOrder(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/authorization/getauthorizationdocumenthandler.go b/app/main/api/internal/handler/authorization/getauthorizationdocumenthandler.go new file mode 100644 index 0000000..6d23bce --- /dev/null +++ b/app/main/api/internal/handler/authorization/getauthorizationdocumenthandler.go @@ -0,0 +1,30 @@ +package authorization + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/authorization" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetAuthorizationDocumentHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetAuthorizationDocumentReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := authorization.NewGetAuthorizationDocumentLogic(r.Context(), svcCtx) + resp, err := l.GetAuthorizationDocument(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/notification/getnotificationshandler.go b/app/main/api/internal/handler/notification/getnotificationshandler.go new file mode 100644 index 0000000..1c72950 --- /dev/null +++ b/app/main/api/internal/handler/notification/getnotificationshandler.go @@ -0,0 +1,17 @@ +package notification + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/notification" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func GetNotificationsHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := notification.NewGetNotificationsLogic(r.Context(), svcCtx) + resp, err := l.GetNotifications() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/pay/alipaycallbackhandler.go b/app/main/api/internal/handler/pay/alipaycallbackhandler.go new file mode 100644 index 0000000..e1b69e2 --- /dev/null +++ b/app/main/api/internal/handler/pay/alipaycallbackhandler.go @@ -0,0 +1,17 @@ +package pay + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/pay" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func AlipayCallbackHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := pay.NewAlipayCallbackLogic(r.Context(), svcCtx) + err := l.AlipayCallback(w, r) + result.HttpResult(r, w, nil, err) + } +} diff --git a/app/main/api/internal/handler/pay/iapcallbackhandler.go b/app/main/api/internal/handler/pay/iapcallbackhandler.go new file mode 100644 index 0000000..9c10b33 --- /dev/null +++ b/app/main/api/internal/handler/pay/iapcallbackhandler.go @@ -0,0 +1,30 @@ +package pay + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/pay" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func IapCallbackHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.IapCallbackReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := pay.NewIapCallbackLogic(r.Context(), svcCtx) + err := l.IapCallback(&req) + result.HttpResult(r, w, nil, err) + } +} diff --git a/app/main/api/internal/handler/pay/paymentcheckhandler.go b/app/main/api/internal/handler/pay/paymentcheckhandler.go new file mode 100644 index 0000000..9287e8e --- /dev/null +++ b/app/main/api/internal/handler/pay/paymentcheckhandler.go @@ -0,0 +1,30 @@ +package pay + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/pay" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func PaymentCheckHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.PaymentCheckReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := pay.NewPaymentCheckLogic(r.Context(), svcCtx) + resp, err := l.PaymentCheck(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/pay/paymenthandler.go b/app/main/api/internal/handler/pay/paymenthandler.go new file mode 100644 index 0000000..c3e4432 --- /dev/null +++ b/app/main/api/internal/handler/pay/paymenthandler.go @@ -0,0 +1,30 @@ +package pay + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/pay" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func PaymentHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.PaymentReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := pay.NewPaymentLogic(r.Context(), svcCtx) + resp, err := l.Payment(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/pay/wechatpaycallbackhandler.go b/app/main/api/internal/handler/pay/wechatpaycallbackhandler.go new file mode 100644 index 0000000..2772107 --- /dev/null +++ b/app/main/api/internal/handler/pay/wechatpaycallbackhandler.go @@ -0,0 +1,17 @@ +package pay + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/pay" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func WechatPayCallbackHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := pay.NewWechatPayCallbackLogic(r.Context(), svcCtx) + err := l.WechatPayCallback(w, r) + result.HttpResult(r, w, nil, err) + } +} diff --git a/app/main/api/internal/handler/pay/wechatpayrefundcallbackhandler.go b/app/main/api/internal/handler/pay/wechatpayrefundcallbackhandler.go new file mode 100644 index 0000000..de16101 --- /dev/null +++ b/app/main/api/internal/handler/pay/wechatpayrefundcallbackhandler.go @@ -0,0 +1,17 @@ +package pay + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/pay" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func WechatPayRefundCallbackHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := pay.NewWechatPayRefundCallbackLogic(r.Context(), svcCtx) + err := l.WechatPayRefundCallback(w, r) + result.HttpResult(r, w, nil, err) + } +} diff --git a/app/main/api/internal/handler/product/getproductappbyenhandler.go b/app/main/api/internal/handler/product/getproductappbyenhandler.go new file mode 100644 index 0000000..437264b --- /dev/null +++ b/app/main/api/internal/handler/product/getproductappbyenhandler.go @@ -0,0 +1,30 @@ +package product + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/product" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetProductAppByEnHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetProductByEnRequest + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := product.NewGetProductAppByEnLogic(r.Context(), svcCtx) + resp, err := l.GetProductAppByEn(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/product/getproductbyenhandler.go b/app/main/api/internal/handler/product/getproductbyenhandler.go new file mode 100644 index 0000000..ade590b --- /dev/null +++ b/app/main/api/internal/handler/product/getproductbyenhandler.go @@ -0,0 +1,30 @@ +package product + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/product" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetProductByEnHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetProductByEnRequest + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := product.NewGetProductByEnLogic(r.Context(), svcCtx) + resp, err := l.GetProductByEn(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/product/getproductbyidhandler.go b/app/main/api/internal/handler/product/getproductbyidhandler.go new file mode 100644 index 0000000..a66b6f4 --- /dev/null +++ b/app/main/api/internal/handler/product/getproductbyidhandler.go @@ -0,0 +1,30 @@ +package product + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/product" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func GetProductByIDHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetProductByIDRequest + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := product.NewGetProductByIDLogic(r.Context(), svcCtx) + resp, err := l.GetProductByID(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/querydetailbyorderidhandler.go b/app/main/api/internal/handler/query/querydetailbyorderidhandler.go new file mode 100644 index 0000000..c2929d5 --- /dev/null +++ b/app/main/api/internal/handler/query/querydetailbyorderidhandler.go @@ -0,0 +1,30 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QueryDetailByOrderIdHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QueryDetailByOrderIdReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewQueryDetailByOrderIdLogic(r.Context(), svcCtx) + resp, err := l.QueryDetailByOrderId(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/querydetailbyordernohandler.go b/app/main/api/internal/handler/query/querydetailbyordernohandler.go new file mode 100644 index 0000000..e34752c --- /dev/null +++ b/app/main/api/internal/handler/query/querydetailbyordernohandler.go @@ -0,0 +1,30 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QueryDetailByOrderNoHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QueryDetailByOrderNoReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewQueryDetailByOrderNoLogic(r.Context(), svcCtx) + resp, err := l.QueryDetailByOrderNo(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/queryexamplehandler.go b/app/main/api/internal/handler/query/queryexamplehandler.go new file mode 100644 index 0000000..36b7bd8 --- /dev/null +++ b/app/main/api/internal/handler/query/queryexamplehandler.go @@ -0,0 +1,30 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QueryExampleHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QueryExampleReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewQueryExampleLogic(r.Context(), svcCtx) + resp, err := l.QueryExample(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/querygeneratesharelinkhandler.go b/app/main/api/internal/handler/query/querygeneratesharelinkhandler.go new file mode 100644 index 0000000..e1458fa --- /dev/null +++ b/app/main/api/internal/handler/query/querygeneratesharelinkhandler.go @@ -0,0 +1,30 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QueryGenerateShareLinkHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QueryGenerateShareLinkReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewQueryGenerateShareLinkLogic(r.Context(), svcCtx) + resp, err := l.QueryGenerateShareLink(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/querylisthandler.go b/app/main/api/internal/handler/query/querylisthandler.go new file mode 100644 index 0000000..19cb5f6 --- /dev/null +++ b/app/main/api/internal/handler/query/querylisthandler.go @@ -0,0 +1,30 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QueryListHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QueryListReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewQueryListLogic(r.Context(), svcCtx) + resp, err := l.QueryList(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/queryprovisionalorderhandler.go b/app/main/api/internal/handler/query/queryprovisionalorderhandler.go new file mode 100644 index 0000000..17943dd --- /dev/null +++ b/app/main/api/internal/handler/query/queryprovisionalorderhandler.go @@ -0,0 +1,30 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QueryProvisionalOrderHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QueryProvisionalOrderReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewQueryProvisionalOrderLogic(r.Context(), svcCtx) + resp, err := l.QueryProvisionalOrder(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/queryretryhandler.go b/app/main/api/internal/handler/query/queryretryhandler.go new file mode 100644 index 0000000..e6dca30 --- /dev/null +++ b/app/main/api/internal/handler/query/queryretryhandler.go @@ -0,0 +1,30 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QueryRetryHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QueryRetryReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewQueryRetryLogic(r.Context(), svcCtx) + resp, err := l.QueryRetry(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/queryserviceagenthandler.go b/app/main/api/internal/handler/query/queryserviceagenthandler.go new file mode 100644 index 0000000..d10ad03 --- /dev/null +++ b/app/main/api/internal/handler/query/queryserviceagenthandler.go @@ -0,0 +1,30 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QueryServiceAgentHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QueryServiceReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewQueryServiceLogic(r.Context(), svcCtx) + resp, err := l.QueryService(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/queryserviceapphandler.go b/app/main/api/internal/handler/query/queryserviceapphandler.go new file mode 100644 index 0000000..2d3fb22 --- /dev/null +++ b/app/main/api/internal/handler/query/queryserviceapphandler.go @@ -0,0 +1,30 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QueryServiceAppHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QueryServiceReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewQueryServiceLogic(r.Context(), svcCtx) + resp, err := l.QueryService(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/queryservicehandler.go b/app/main/api/internal/handler/query/queryservicehandler.go new file mode 100644 index 0000000..0123a7a --- /dev/null +++ b/app/main/api/internal/handler/query/queryservicehandler.go @@ -0,0 +1,25 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QueryServiceHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QueryServiceReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + l := query.NewQueryServiceLogic(r.Context(), svcCtx) + resp, err := l.QueryService(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/querysharedetailhandler.go b/app/main/api/internal/handler/query/querysharedetailhandler.go new file mode 100644 index 0000000..a7ae365 --- /dev/null +++ b/app/main/api/internal/handler/query/querysharedetailhandler.go @@ -0,0 +1,30 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QueryShareDetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QueryShareDetailReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewQueryShareDetailLogic(r.Context(), svcCtx) + resp, err := l.QueryShareDetail(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/querysingletesthandler.go b/app/main/api/internal/handler/query/querysingletesthandler.go new file mode 100644 index 0000000..261f3b7 --- /dev/null +++ b/app/main/api/internal/handler/query/querysingletesthandler.go @@ -0,0 +1,30 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func QuerySingleTestHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.QuerySingleTestReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewQuerySingleTestLogic(r.Context(), svcCtx) + resp, err := l.QuerySingleTest(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/query/updatequerydatahandler.go b/app/main/api/internal/handler/query/updatequerydatahandler.go new file mode 100644 index 0000000..450ef33 --- /dev/null +++ b/app/main/api/internal/handler/query/updatequerydatahandler.go @@ -0,0 +1,31 @@ +package query + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/query" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +// 更新查询数据 +func UpdateQueryDataHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.UpdateQueryDataReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := query.NewUpdateQueryDataLogic(r.Context(), svcCtx) + resp, err := l.UpdateQueryData(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/routes.go b/app/main/api/internal/handler/routes.go new file mode 100644 index 0000000..37bea1c --- /dev/null +++ b/app/main/api/internal/handler/routes.go @@ -0,0 +1,1084 @@ +// Code generated by goctl. DO NOT EDIT. +package handler + +import ( + "net/http" + + admin_agent "qnc-server/app/main/api/internal/handler/admin_agent" + admin_api "qnc-server/app/main/api/internal/handler/admin_api" + admin_auth "qnc-server/app/main/api/internal/handler/admin_auth" + admin_feature "qnc-server/app/main/api/internal/handler/admin_feature" + admin_menu "qnc-server/app/main/api/internal/handler/admin_menu" + admin_notification "qnc-server/app/main/api/internal/handler/admin_notification" + admin_order "qnc-server/app/main/api/internal/handler/admin_order" + admin_platform_user "qnc-server/app/main/api/internal/handler/admin_platform_user" + admin_product "qnc-server/app/main/api/internal/handler/admin_product" + admin_query "qnc-server/app/main/api/internal/handler/admin_query" + admin_role "qnc-server/app/main/api/internal/handler/admin_role" + admin_role_api "qnc-server/app/main/api/internal/handler/admin_role_api" + admin_user "qnc-server/app/main/api/internal/handler/admin_user" + agent "qnc-server/app/main/api/internal/handler/agent" + app "qnc-server/app/main/api/internal/handler/app" + auth "qnc-server/app/main/api/internal/handler/auth" + authorization "qnc-server/app/main/api/internal/handler/authorization" + notification "qnc-server/app/main/api/internal/handler/notification" + pay "qnc-server/app/main/api/internal/handler/pay" + product "qnc-server/app/main/api/internal/handler/product" + query "qnc-server/app/main/api/internal/handler/query" + user "qnc-server/app/main/api/internal/handler/user" + "qnc-server/app/main/api/internal/svc" + + "github.com/zeromicro/go-zero/rest" +) + +func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) { + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + Method: http.MethodPost, + Path: "/audit", + Handler: admin_agent.AdminAuditAgentHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/commission/list", + Handler: admin_agent.AdminGetAgentCommissionListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/config", + Handler: admin_agent.AdminGetAgentConfigHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/config/update", + Handler: admin_agent.AdminUpdateAgentConfigHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/invite_code/diamond/generate", + Handler: admin_agent.AdminGenerateDiamondInviteCodeHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/invite_code/list", + Handler: admin_agent.AdminGetInviteCodeListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/link/list", + Handler: admin_agent.AdminGetAgentLinkListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/list", + Handler: admin_agent.AdminGetAgentListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/order/list", + Handler: admin_agent.AdminGetAgentOrderListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/product_config/list", + Handler: admin_agent.AdminGetAgentProductConfigListHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/product_config/update", + Handler: admin_agent.AdminUpdateAgentProductConfigHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/real_name/list", + Handler: admin_agent.AdminGetAgentRealNameListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/rebate/list", + Handler: admin_agent.AdminGetAgentRebateListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/upgrade/list", + Handler: admin_agent.AdminGetAgentUpgradeListHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/withdrawal/audit", + Handler: admin_agent.AdminAuditWithdrawalHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/withdrawal/list", + Handler: admin_agent.AdminGetAgentWithdrawalListHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/admin/agent"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + Method: http.MethodPut, + Path: "/admin/api/batch-update-status", + Handler: admin_api.AdminBatchUpdateApiStatusHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/admin/api/create", + Handler: admin_api.AdminCreateApiHandler(serverCtx), + }, + { + Method: http.MethodDelete, + Path: "/admin/api/delete/:id", + Handler: admin_api.AdminDeleteApiHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/admin/api/detail/:id", + Handler: admin_api.AdminGetApiDetailHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/admin/api/list", + Handler: admin_api.AdminGetApiListHandler(serverCtx), + }, + { + Method: http.MethodPut, + Path: "/admin/api/update/:id", + Handler: admin_api.AdminUpdateApiHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + []rest.Route{ + { + // 登录 + Method: http.MethodPost, + Path: "/login", + Handler: admin_auth.AdminLoginHandler(serverCtx), + }, + }, + rest.WithPrefix("/api/v1/admin/auth"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + Method: http.MethodPost, + Path: "/config-example", + Handler: admin_feature.AdminConfigFeatureExampleHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/create", + Handler: admin_feature.AdminCreateFeatureHandler(serverCtx), + }, + { + Method: http.MethodDelete, + Path: "/delete/:id", + Handler: admin_feature.AdminDeleteFeatureHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/detail/:id", + Handler: admin_feature.AdminGetFeatureDetailHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/example/:feature_id", + Handler: admin_feature.AdminGetFeatureExampleHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/list", + Handler: admin_feature.AdminGetFeatureListHandler(serverCtx), + }, + { + Method: http.MethodPut, + Path: "/update/:id", + Handler: admin_feature.AdminUpdateFeatureHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/admin/feature"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + // 获取所有菜单(树形结构) + Method: http.MethodGet, + Path: "/all", + Handler: admin_menu.GetMenuAllHandler(serverCtx), + }, + { + // 创建菜单 + Method: http.MethodPost, + Path: "/create", + Handler: admin_menu.CreateMenuHandler(serverCtx), + }, + { + // 删除菜单 + Method: http.MethodDelete, + Path: "/delete/:id", + Handler: admin_menu.DeleteMenuHandler(serverCtx), + }, + { + // 获取菜单详情 + Method: http.MethodGet, + Path: "/detail/:id", + Handler: admin_menu.GetMenuDetailHandler(serverCtx), + }, + { + // 获取菜单列表 + Method: http.MethodGet, + Path: "/list", + Handler: admin_menu.GetMenuListHandler(serverCtx), + }, + { + // 更新菜单 + Method: http.MethodPut, + Path: "/update/:id", + Handler: admin_menu.UpdateMenuHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/admin/menu"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + Method: http.MethodPost, + Path: "/create", + Handler: admin_notification.AdminCreateNotificationHandler(serverCtx), + }, + { + Method: http.MethodDelete, + Path: "/delete/:id", + Handler: admin_notification.AdminDeleteNotificationHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/detail/:id", + Handler: admin_notification.AdminGetNotificationDetailHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/list", + Handler: admin_notification.AdminGetNotificationListHandler(serverCtx), + }, + { + Method: http.MethodPut, + Path: "/update/:id", + Handler: admin_notification.AdminUpdateNotificationHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/admin/notification"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + // 创建订单 + Method: http.MethodPost, + Path: "/create", + Handler: admin_order.AdminCreateOrderHandler(serverCtx), + }, + { + // 删除订单 + Method: http.MethodDelete, + Path: "/delete/:id", + Handler: admin_order.AdminDeleteOrderHandler(serverCtx), + }, + { + // 获取订单详情 + Method: http.MethodGet, + Path: "/detail/:id", + Handler: admin_order.AdminGetOrderDetailHandler(serverCtx), + }, + { + // 获取订单列表 + Method: http.MethodGet, + Path: "/list", + Handler: admin_order.AdminGetOrderListHandler(serverCtx), + }, + { + // 订单退款 + Method: http.MethodPost, + Path: "/refund/:id", + Handler: admin_order.AdminRefundOrderHandler(serverCtx), + }, + { + // 重新执行代理处理 + Method: http.MethodPost, + Path: "/retry-agent-process/:id", + Handler: admin_order.AdminRetryAgentProcessHandler(serverCtx), + }, + { + // 更新订单 + Method: http.MethodPut, + Path: "/update/:id", + Handler: admin_order.AdminUpdateOrderHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/admin/order"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + Method: http.MethodPost, + Path: "/create", + Handler: admin_platform_user.AdminCreatePlatformUserHandler(serverCtx), + }, + { + Method: http.MethodDelete, + Path: "/delete/:id", + Handler: admin_platform_user.AdminDeletePlatformUserHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/detail/:id", + Handler: admin_platform_user.AdminGetPlatformUserDetailHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/list", + Handler: admin_platform_user.AdminGetPlatformUserListHandler(serverCtx), + }, + { + Method: http.MethodPut, + Path: "/update/:id", + Handler: admin_platform_user.AdminUpdatePlatformUserHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/admin/platform_user"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + Method: http.MethodPost, + Path: "/create", + Handler: admin_product.AdminCreateProductHandler(serverCtx), + }, + { + Method: http.MethodDelete, + Path: "/delete/:id", + Handler: admin_product.AdminDeleteProductHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/detail/:id", + Handler: admin_product.AdminGetProductDetailHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/feature/list/:product_id", + Handler: admin_product.AdminGetProductFeatureListHandler(serverCtx), + }, + { + Method: http.MethodPut, + Path: "/feature/update/:product_id", + Handler: admin_product.AdminUpdateProductFeaturesHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/list", + Handler: admin_product.AdminGetProductListHandler(serverCtx), + }, + { + Method: http.MethodPut, + Path: "/update/:id", + Handler: admin_product.AdminUpdateProductHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/admin/product"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + // 更新清理配置 + Method: http.MethodPut, + Path: "/cleanup/config", + Handler: admin_query.AdminUpdateQueryCleanupConfigHandler(serverCtx), + }, + { + // 获取清理配置列表 + Method: http.MethodGet, + Path: "/cleanup/configs", + Handler: admin_query.AdminGetQueryCleanupConfigListHandler(serverCtx), + }, + { + // 获取清理详情列表 + Method: http.MethodGet, + Path: "/cleanup/details/:log_id", + Handler: admin_query.AdminGetQueryCleanupDetailListHandler(serverCtx), + }, + { + // 获取清理日志列表 + Method: http.MethodGet, + Path: "/cleanup/logs", + Handler: admin_query.AdminGetQueryCleanupLogListHandler(serverCtx), + }, + { + // 获取查询详情 + Method: http.MethodGet, + Path: "/detail/:order_id", + Handler: admin_query.AdminGetQueryDetailByOrderIdHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/admin/query"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + // 创建角色 + Method: http.MethodPost, + Path: "/create", + Handler: admin_role.CreateRoleHandler(serverCtx), + }, + { + // 删除角色 + Method: http.MethodDelete, + Path: "/delete/:id", + Handler: admin_role.DeleteRoleHandler(serverCtx), + }, + { + // 获取角色详情 + Method: http.MethodGet, + Path: "/detail/:id", + Handler: admin_role.GetRoleDetailHandler(serverCtx), + }, + { + // 获取角色列表 + Method: http.MethodGet, + Path: "/list", + Handler: admin_role.GetRoleListHandler(serverCtx), + }, + { + // 更新角色 + Method: http.MethodPut, + Path: "/update/:id", + Handler: admin_role.UpdateRoleHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/admin/role"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + Method: http.MethodGet, + Path: "/admin/api/all", + Handler: admin_role_api.AdminGetAllApiListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/admin/role/:role_id/api/list", + Handler: admin_role_api.AdminGetRoleApiListHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/admin/role/api/assign", + Handler: admin_role_api.AdminAssignRoleApiHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/admin/role/api/remove", + Handler: admin_role_api.AdminRemoveRoleApiHandler(serverCtx), + }, + { + Method: http.MethodPut, + Path: "/admin/role/api/update", + Handler: admin_role_api.AdminUpdateRoleApiHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AdminAuthInterceptor}, + []rest.Route{ + { + // 创建用户 + Method: http.MethodPost, + Path: "/create", + Handler: admin_user.AdminCreateUserHandler(serverCtx), + }, + { + // 删除用户 + Method: http.MethodDelete, + Path: "/delete/:id", + Handler: admin_user.AdminDeleteUserHandler(serverCtx), + }, + { + // 获取用户详情 + Method: http.MethodGet, + Path: "/detail/:id", + Handler: admin_user.AdminGetUserDetailHandler(serverCtx), + }, + { + // 用户信息 + Method: http.MethodGet, + Path: "/info", + Handler: admin_user.AdminUserInfoHandler(serverCtx), + }, + { + // 获取用户列表 + Method: http.MethodGet, + Path: "/list", + Handler: admin_user.AdminGetUserListHandler(serverCtx), + }, + { + // 重置管理员密码 + Method: http.MethodPut, + Path: "/reset-password/:id", + Handler: admin_user.AdminResetPasswordHandler(serverCtx), + }, + { + // 更新用户 + Method: http.MethodPut, + Path: "/update/:id", + Handler: admin_user.AdminUpdateUserHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/admin/user"), + ) + + server.AddRoutes( + []rest.Route{ + { + Method: http.MethodGet, + Path: "/link", + Handler: agent.GetLinkDataHandler(serverCtx), + }, + }, + rest.WithPrefix("/api/v1/agent"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AuthInterceptor}, + []rest.Route{ + { + Method: http.MethodPost, + Path: "/apply", + Handler: agent.ApplyForAgentHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/register/invite", + Handler: agent.RegisterByInviteCodeHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1/agent"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.UserAuthInterceptor}, + []rest.Route{ + { + Method: http.MethodGet, + Path: "/commission/list", + Handler: agent.GetCommissionListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/conversion/rate", + Handler: agent.GetConversionRateHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/generating_link", + Handler: agent.GeneratingLinkHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/info", + Handler: agent.GetAgentInfoHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/invite_code/delete", + Handler: agent.DeleteInviteCodeHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/invite_code/generate", + Handler: agent.GenerateInviteCodeHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/invite_code/list", + Handler: agent.GetInviteCodeListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/invite_link", + Handler: agent.GetInviteLinkHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/level/privilege", + Handler: agent.GetLevelPrivilegeHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/product_config", + Handler: agent.GetAgentProductConfigHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/promotion/query/list", + Handler: agent.GetPromotionQueryListHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/real_name", + Handler: agent.RealNameAuthHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/rebate/list", + Handler: agent.GetRebateListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/rebate/upgrade/list", + Handler: agent.GetUpgradeRebateListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/revenue", + Handler: agent.GetRevenueInfoHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/subordinate/contribution/detail", + Handler: agent.GetSubordinateContributionDetailHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/subordinate/list", + Handler: agent.GetSubordinateListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/team/list", + Handler: agent.GetTeamListHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/team/statistics", + Handler: agent.GetTeamStatisticsHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/upgrade/apply", + Handler: agent.ApplyUpgradeHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/upgrade/list", + Handler: agent.GetUpgradeListHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/upgrade/subordinate", + Handler: agent.UpgradeSubordinateHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/withdrawal/apply", + Handler: agent.ApplyWithdrawalHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/withdrawal/list", + Handler: agent.GetWithdrawalListHandler(serverCtx), + }, + }..., + ), + rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret), + rest.WithPrefix("/api/v1/agent"), + ) + + server.AddRoutes( + []rest.Route{ + { + Method: http.MethodGet, + Path: "/s/:shortCode", + Handler: agent.ShortLinkRedirectHandler(serverCtx), + }, + }, + ) + + server.AddRoutes( + []rest.Route{ + { + Method: http.MethodGet, + Path: "/app/config", + Handler: app.GetAppConfigHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/app/version", + Handler: app.GetAppVersionHandler(serverCtx), + }, + { + // 心跳检测接口 + Method: http.MethodGet, + Path: "/health/check", + Handler: app.HealthCheckHandler(serverCtx), + }, + }, + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + []rest.Route{ + { + // get mobile verify code + Method: http.MethodPost, + Path: "/auth/sendSms", + Handler: auth.SendSmsHandler(serverCtx), + }, + }, + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + []rest.Route{ + { + Method: http.MethodGet, + Path: "/authorization/document/:documentId", + Handler: authorization.GetAuthorizationDocumentHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/authorization/document/order/:orderId", + Handler: authorization.GetAuthorizationDocumentByOrderHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/authorization/download/:documentId", + Handler: authorization.DownloadAuthorizationDocumentHandler(serverCtx), + }, + }, + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + []rest.Route{ + { + // get notifications + Method: http.MethodGet, + Path: "/notification/list", + Handler: notification.GetNotificationsHandler(serverCtx), + }, + }, + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + []rest.Route{ + { + Method: http.MethodPost, + Path: "/pay/alipay/callback", + Handler: pay.AlipayCallbackHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/pay/wechat/callback", + Handler: pay.WechatPayCallbackHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/pay/wechat/refund_callback", + Handler: pay.WechatPayRefundCallbackHandler(serverCtx), + }, + }, + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AuthInterceptor}, + []rest.Route{ + { + Method: http.MethodPost, + Path: "/pay/check", + Handler: pay.PaymentCheckHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/pay/iap_callback", + Handler: pay.IapCallbackHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/pay/payment", + Handler: pay.PaymentHandler(serverCtx), + }, + }..., + ), + rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret), + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + []rest.Route{ + { + Method: http.MethodGet, + Path: "/:id", + Handler: product.GetProductByIDHandler(serverCtx), + }, + { + Method: http.MethodGet, + Path: "/en/:product_en", + Handler: product.GetProductByEnHandler(serverCtx), + }, + }, + rest.WithPrefix("/api/v1/product"), + ) + + server.AddRoutes( + []rest.Route{ + { + Method: http.MethodGet, + Path: "/app_en/:product_en", + Handler: product.GetProductAppByEnHandler(serverCtx), + }, + }, + rest.WithPrefix("/api/v1/product"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AuthInterceptor}, + []rest.Route{ + { + // query service agent + Method: http.MethodPost, + Path: "/query/service_agent/:product", + Handler: query.QueryServiceAgentHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/query/service_app/:product", + Handler: query.QueryServiceAppHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AuthInterceptor}, + []rest.Route{ + { + // query service + Method: http.MethodPost, + Path: "/query/service/:product", + Handler: query.QueryServiceHandler(serverCtx), + }, + }..., + ), + rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret), + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AuthInterceptor}, + []rest.Route{ + { + // 生成分享链接 + Method: http.MethodPost, + Path: "/query/generate_share_link", + Handler: query.QueryGenerateShareLinkHandler(serverCtx), + }, + { + // 查询列表 + Method: http.MethodGet, + Path: "/query/list", + Handler: query.QueryListHandler(serverCtx), + }, + { + // 查询详情 按订单号 付款查询时 + Method: http.MethodGet, + Path: "/query/orderId/:order_id", + Handler: query.QueryDetailByOrderIdHandler(serverCtx), + }, + { + // 查询详情 按订单号 + Method: http.MethodGet, + Path: "/query/orderNo/:order_no", + Handler: query.QueryDetailByOrderNoHandler(serverCtx), + }, + { + // 获取查询临时订单 + Method: http.MethodGet, + Path: "/query/provisional_order/:id", + Handler: query.QueryProvisionalOrderHandler(serverCtx), + }, + { + // 重试查询 + Method: http.MethodPost, + Path: "/query/retry/:id", + Handler: query.QueryRetryHandler(serverCtx), + }, + { + // 更新查询数据 + Method: http.MethodPost, + Path: "/query/update_data", + Handler: query.UpdateQueryDataHandler(serverCtx), + }, + }..., + ), + rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret), + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + []rest.Route{ + { + // 查询示例 + Method: http.MethodGet, + Path: "/query/example", + Handler: query.QueryExampleHandler(serverCtx), + }, + { + // 查询详情 + Method: http.MethodGet, + Path: "/query/share/:id", + Handler: query.QueryShareDetailHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/query/single/test", + Handler: query.QuerySingleTestHandler(serverCtx), + }, + }, + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + []rest.Route{ + { + // unified auth + Method: http.MethodPost, + Path: "/user/auth", + Handler: user.AuthHandler(serverCtx), + }, + { + // mobile code login + Method: http.MethodPost, + Path: "/user/mobileCodeLogin", + Handler: user.MobileCodeLoginHandler(serverCtx), + }, + { + // wechat mini auth + Method: http.MethodPost, + Path: "/user/wxMiniAuth", + Handler: user.WxMiniAuthHandler(serverCtx), + }, + { + // wechat h5 auth + Method: http.MethodPost, + Path: "/user/wxh5Auth", + Handler: user.WxH5AuthHandler(serverCtx), + }, + { + Method: http.MethodPost, + Path: "/wechat/getSignature", + Handler: user.GetSignatureHandler(serverCtx), + }, + }, + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + rest.WithMiddlewares( + []rest.Middleware{serverCtx.AuthInterceptor}, + []rest.Route{ + { + // 绑定手机号 + Method: http.MethodPost, + Path: "/user/bindMobile", + Handler: user.BindMobileHandler(serverCtx), + }, + }..., + ), + rest.WithPrefix("/api/v1"), + ) + + server.AddRoutes( + []rest.Route{ + { + Method: http.MethodPost, + Path: "/user/cancelOut", + Handler: user.CancelOutHandler(serverCtx), + }, + { + // get user info + Method: http.MethodGet, + Path: "/user/detail", + Handler: user.DetailHandler(serverCtx), + }, + { + // get new token + Method: http.MethodPost, + Path: "/user/getToken", + Handler: user.GetTokenHandler(serverCtx), + }, + }, + rest.WithJwt(serverCtx.Config.JwtAuth.AccessSecret), + rest.WithPrefix("/api/v1"), + ) +} diff --git a/app/main/api/internal/handler/user/authhandler.go b/app/main/api/internal/handler/user/authhandler.go new file mode 100644 index 0000000..ff5ff30 --- /dev/null +++ b/app/main/api/internal/handler/user/authhandler.go @@ -0,0 +1,25 @@ +package user + +import ( + "net/http" + + logic "qnc-server/app/main/api/internal/logic/user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func AuthHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.AuthReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + l := logic.NewAuthLogic(r.Context(), svcCtx) + resp, err := l.Auth(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/user/bindmobilehandler.go b/app/main/api/internal/handler/user/bindmobilehandler.go new file mode 100644 index 0000000..16b3654 --- /dev/null +++ b/app/main/api/internal/handler/user/bindmobilehandler.go @@ -0,0 +1,30 @@ +package user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func BindMobileHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.BindMobileReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := user.NewBindMobileLogic(r.Context(), svcCtx) + resp, err := l.BindMobile(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/user/cancelouthandler.go b/app/main/api/internal/handler/user/cancelouthandler.go new file mode 100644 index 0000000..4a4ebf9 --- /dev/null +++ b/app/main/api/internal/handler/user/cancelouthandler.go @@ -0,0 +1,17 @@ +package user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func CancelOutHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := user.NewCancelOutLogic(r.Context(), svcCtx) + err := l.CancelOut() + result.HttpResult(r, w, nil, err) + } +} diff --git a/app/main/api/internal/handler/user/detailhandler.go b/app/main/api/internal/handler/user/detailhandler.go new file mode 100644 index 0000000..0689335 --- /dev/null +++ b/app/main/api/internal/handler/user/detailhandler.go @@ -0,0 +1,17 @@ +package user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func DetailHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := user.NewDetailLogic(r.Context(), svcCtx) + resp, err := l.Detail() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/user/getsignaturehandler.go b/app/main/api/internal/handler/user/getsignaturehandler.go new file mode 100644 index 0000000..23e940d --- /dev/null +++ b/app/main/api/internal/handler/user/getsignaturehandler.go @@ -0,0 +1,29 @@ +package user + +import ( + "net/http" + + "github.com/zeromicro/go-zero/rest/httpx" + "qnc-server/app/main/api/internal/logic/user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" +) + +func GetSignatureHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.GetSignatureReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := user.NewGetSignatureLogic(r.Context(), svcCtx) + resp, err := l.GetSignature(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/user/gettokenhandler.go b/app/main/api/internal/handler/user/gettokenhandler.go new file mode 100644 index 0000000..b25a120 --- /dev/null +++ b/app/main/api/internal/handler/user/gettokenhandler.go @@ -0,0 +1,17 @@ +package user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/common/result" +) + +func GetTokenHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + l := user.NewGetTokenLogic(r.Context(), svcCtx) + resp, err := l.GetToken() + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/user/mobilecodeloginhandler.go b/app/main/api/internal/handler/user/mobilecodeloginhandler.go new file mode 100644 index 0000000..28a1ac8 --- /dev/null +++ b/app/main/api/internal/handler/user/mobilecodeloginhandler.go @@ -0,0 +1,30 @@ +package user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func MobileCodeLoginHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.MobileCodeLoginReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := user.NewMobileCodeLoginLogic(r.Context(), svcCtx) + resp, err := l.MobileCodeLogin(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/user/wxh5authhandler.go b/app/main/api/internal/handler/user/wxh5authhandler.go new file mode 100644 index 0000000..eb33694 --- /dev/null +++ b/app/main/api/internal/handler/user/wxh5authhandler.go @@ -0,0 +1,30 @@ +package user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func WxH5AuthHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.WXH5AuthReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := user.NewWxH5AuthLogic(r.Context(), svcCtx) + resp, err := l.WxH5Auth(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/handler/user/wxminiauthhandler.go b/app/main/api/internal/handler/user/wxminiauthhandler.go new file mode 100644 index 0000000..5c8f276 --- /dev/null +++ b/app/main/api/internal/handler/user/wxminiauthhandler.go @@ -0,0 +1,30 @@ +package user + +import ( + "net/http" + + "qnc-server/app/main/api/internal/logic/user" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + + "github.com/zeromicro/go-zero/rest/httpx" +) + +func WxMiniAuthHandler(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + var req types.WXMiniAuthReq + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r, w, err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + l := user.NewWxMiniAuthLogic(r.Context(), svcCtx) + resp, err := l.WxMiniAuth(&req) + result.HttpResult(r, w, resp, err) + } +} diff --git a/app/main/api/internal/logic/admin_agent/adminauditagentlogic.go b/app/main/api/internal/logic/admin_agent/adminauditagentlogic.go new file mode 100644 index 0000000..9f2d31f --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/adminauditagentlogic.go @@ -0,0 +1,30 @@ +package admin_agent + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminAuditAgentLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminAuditAgentLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminAuditAgentLogic { + return &AdminAuditAgentLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminAuditAgentLogic) AdminAuditAgent(req *types.AdminAuditAgentReq) (resp *types.AdminAuditAgentResp, err error) { + // todo: add your logic here and delete this line + + return +} diff --git a/app/main/api/internal/logic/admin_agent/adminauditrealnamelogic.go b/app/main/api/internal/logic/admin_agent/adminauditrealnamelogic.go new file mode 100644 index 0000000..1b06fb8 --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/adminauditrealnamelogic.go @@ -0,0 +1,33 @@ +package admin_agent + +import ( + "context" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminAuditRealNameLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminAuditRealNameLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminAuditRealNameLogic { + return &AdminAuditRealNameLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +// AdminAuditRealName 实名认证审核（已废弃：实名认证改为三要素核验，无需审核） +func (l *AdminAuditRealNameLogic) AdminAuditRealName(req *types.AdminAuditRealNameReq) (resp *types.AdminAuditRealNameResp, err error) { + // 该接口已废弃，实名认证现在通过三要素核验自动完成，无需人工审核 + return nil, errors.Wrapf(xerr.NewErrMsg("该接口已废弃，实名认证改为三要素核验，无需审核"), "") +} diff --git a/app/main/api/internal/logic/admin_agent/adminauditwithdrawallogic.go b/app/main/api/internal/logic/admin_agent/adminauditwithdrawallogic.go new file mode 100644 index 0000000..5628edb --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/adminauditwithdrawallogic.go @@ -0,0 +1,175 @@ +package admin_agent + +import ( + "context" + "database/sql" + "fmt" + "time" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/sqlx" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminAuditWithdrawalLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminAuditWithdrawalLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminAuditWithdrawalLogic { + return &AdminAuditWithdrawalLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminAuditWithdrawalLogic) AdminAuditWithdrawal(req *types.AdminAuditWithdrawalReq) (resp *types.AdminAuditWithdrawalResp, err error) { + // 1. 查询提现记录 + withdrawal, err := l.svcCtx.AgentWithdrawalModel.FindOne(l.ctx, req.WithdrawalId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询提现记录失败, %v", err) + } + + // 3. 检查状态（必须是待审核状态） + if withdrawal.Status != 1 { + return nil, errors.Wrapf(xerr.NewErrMsg("该提现记录已处理"), "") + } + + // 4. 使用事务处理审核 + err = l.svcCtx.AgentWithdrawalModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + if req.Status == 2 { // 审核通过 + // 4.1 更新提现记录状态为提现中 + withdrawal.Status = 4 // 提现中 + withdrawal.Remark = sql.NullString{String: req.Remark, Valid: true} + if err := l.svcCtx.AgentWithdrawalModel.UpdateWithVersion(transCtx, session, withdrawal); err != nil { + return errors.Wrapf(err, "更新提现记录失败") + } + + // 4.2 调用支付宝转账接口 + outBizNo := withdrawal.WithdrawNo + transferResp, err := l.svcCtx.AlipayService.AliTransfer(transCtx, withdrawal.PayeeAccount, withdrawal.PayeeName, withdrawal.ActualAmount, "代理提现", outBizNo) + if err != nil { + // 转账失败，更新状态为失败 + withdrawal.Status = 6 // 提现失败 + withdrawal.Remark = sql.NullString{String: fmt.Sprintf("转账失败: %v", err), Valid: true} + l.svcCtx.AgentWithdrawalModel.UpdateWithVersion(transCtx, session, withdrawal) + + // 解冻余额 + wallet, err := l.svcCtx.AgentWalletModel.FindOneByAgentId(transCtx, withdrawal.AgentId) + if err == nil { + wallet.FrozenBalance -= withdrawal.Amount + wallet.Balance += withdrawal.Amount + l.svcCtx.AgentWalletModel.UpdateWithVersion(transCtx, session, wallet) + } + + return errors.Wrapf(err, "支付宝转账失败") + } + + // 4.3 根据转账结果更新状态 + switch transferResp.Status { + case "SUCCESS": + // 转账成功 + withdrawal.Status = 5 // 提现成功 + if err := l.svcCtx.AgentWithdrawalModel.UpdateWithVersion(transCtx, session, withdrawal); err != nil { + return errors.Wrapf(err, "更新提现记录失败") + } + + // 更新钱包（解冻并扣除） + wallet, err := l.svcCtx.AgentWalletModel.FindOneByAgentId(transCtx, withdrawal.AgentId) + if err != nil { + return errors.Wrapf(err, "查询钱包失败") + } + wallet.FrozenBalance -= withdrawal.Amount + wallet.WithdrawnAmount += withdrawal.Amount + if err := l.svcCtx.AgentWalletModel.UpdateWithVersion(transCtx, session, wallet); err != nil { + return errors.Wrapf(err, "更新钱包失败") + } + + // 更新扣税记录状态 + taxBuilder := l.svcCtx.AgentWithdrawalTaxModel.SelectBuilder(). + Where("withdrawal_id = ? AND del_state = ?", withdrawal.Id, globalkey.DelStateNo) + taxRecords, err := l.svcCtx.AgentWithdrawalTaxModel.FindAll(transCtx, taxBuilder, "") + if err == nil && len(taxRecords) > 0 { + taxRecord := taxRecords[0] + taxRecord.TaxStatus = 2 // 已扣税 + taxRecord.TaxTime = lzUtils.TimeToNullTime(time.Now()) + l.svcCtx.AgentWithdrawalTaxModel.UpdateWithVersion(transCtx, session, taxRecord) + } + + case "FAIL": + // 转账失败 + withdrawal.Status = 6 // 提现失败 + errorMsg := l.mapAlipayError(transferResp.SubCode) + withdrawal.Remark = sql.NullString{String: errorMsg, Valid: true} + if err := l.svcCtx.AgentWithdrawalModel.UpdateWithVersion(transCtx, session, withdrawal); err != nil { + return errors.Wrapf(err, "更新提现记录失败") + } + + // 解冻余额 + wallet, err := l.svcCtx.AgentWalletModel.FindOneByAgentId(transCtx, withdrawal.AgentId) + if err == nil { + wallet.FrozenBalance -= withdrawal.Amount + wallet.Balance += withdrawal.Amount + l.svcCtx.AgentWalletModel.UpdateWithVersion(transCtx, session, wallet) + } + + case "DEALING": + // 处理中，保持提现中状态，后续通过轮询更新 + // 状态已经是4（提现中），无需更新 + } + + } else if req.Status == 3 { // 审核拒绝 + // 4.1 更新提现记录状态为拒绝 + withdrawal.Status = 3 // 审核拒绝 + withdrawal.Remark = sql.NullString{String: req.Remark, Valid: true} + if err := l.svcCtx.AgentWithdrawalModel.UpdateWithVersion(transCtx, session, withdrawal); err != nil { + return errors.Wrapf(err, "更新提现记录失败") + } + + // 4.2 解冻余额 + wallet, err := l.svcCtx.AgentWalletModel.FindOneByAgentId(transCtx, withdrawal.AgentId) + if err != nil { + return errors.Wrapf(err, "查询钱包失败") + } + wallet.FrozenBalance -= withdrawal.Amount + wallet.Balance += withdrawal.Amount + if err := l.svcCtx.AgentWalletModel.UpdateWithVersion(transCtx, session, wallet); err != nil { + return errors.Wrapf(err, "更新钱包失败") + } + } + + return nil + }) + + if err != nil { + return nil, err + } + + return &types.AdminAuditWithdrawalResp{ + Success: true, + }, nil +} + +// mapAlipayError 映射支付宝错误码 +func (l *AdminAuditWithdrawalLogic) mapAlipayError(code string) string { + errorMapping := map[string]string{ + "PAYEE_USERINFO_ERROR": "收款方姓名或信息不匹配", + "PAYEE_CARD_INFO_ERROR": "收款支付宝账号及户名不一致", + "PAYEE_IDENTITY_NOT_MATCH": "收款方身份信息不匹配", + "PAYEE_USER_IS_INST": "收款方为金融机构，不能使用提现功能", + "PAYEE_USER_TYPE_ERROR": "该支付宝账号类型不支持提现", + } + if msg, ok := errorMapping[code]; ok { + return msg + } + return "系统错误，请联系客服" +} diff --git a/app/main/api/internal/logic/admin_agent/admingeneratediamondinvitecodelogic.go b/app/main/api/internal/logic/admin_agent/admingeneratediamondinvitecodelogic.go new file mode 100644 index 0000000..c472407 --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingeneratediamondinvitecodelogic.go @@ -0,0 +1,102 @@ +package admin_agent + +import ( + "context" + "database/sql" + "time" + "qnc-server/app/main/model" + "qnc-server/common/tool" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/sqlx" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" + "github.com/google/uuid" +) + +type AdminGenerateDiamondInviteCodeLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGenerateDiamondInviteCodeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGenerateDiamondInviteCodeLogic { + return &AdminGenerateDiamondInviteCodeLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGenerateDiamondInviteCodeLogic) AdminGenerateDiamondInviteCode(req *types.AdminGenerateDiamondInviteCodeReq) (resp *types.AdminGenerateDiamondInviteCodeResp, err error) { + // 1. 验证生成数量 + if req.Count <= 0 || req.Count > 100 { + return nil, errors.Wrapf(xerr.NewErrMsg("生成数量必须在1-100之间"), "") + } + + // 2. 生成钻石邀请码（平台发放，agent_id为NULL，target_level为3） + codes := make([]string, 0, req.Count) + var expireTime sql.NullTime + if req.ExpireDays > 0 { + expireTime = sql.NullTime{ + Time: time.Now().AddDate(0, 0, int(req.ExpireDays)), + Valid: true, + } + } + + err = l.svcCtx.AgentInviteCodeModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + for i := int64(0); i < req.Count; i++ { + // 生成8位随机邀请码（大小写字母+数字） + var code string + maxRetries := 10 // 最大重试次数 + for retry := 0; retry < maxRetries; retry++ { + code = tool.Krand(8, tool.KC_RAND_KIND_ALL) + // 检查邀请码是否已存在 + _, err := l.svcCtx.AgentInviteCodeModel.FindOneByCode(transCtx, code) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + // 邀请码不存在，可以使用 + break + } + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "检查邀请码失败, %v", err) + } + // 邀请码已存在，继续生成 + if retry == maxRetries-1 { + return errors.Wrapf(xerr.NewErrMsg("生成邀请码失败，请重试"), "") + } + } + + // 创建邀请码记录（平台发放的钻石邀请码） + inviteCode := &model.AgentInviteCode{ + Id: uuid.NewString(), + Code: code, + AgentId: sql.NullString{Valid: false}, // NULL表示平台发放 + TargetLevel: 3, // 钻石代理 + Status: 0, // 未使用 + ExpireTime: expireTime, + Remark: sql.NullString{String: req.Remark, Valid: req.Remark != ""}, + } + + _, err := l.svcCtx.AgentInviteCodeModel.Insert(transCtx, session, inviteCode) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建邀请码失败, %v", err) + } + + codes = append(codes, code) + } + + return nil + }) + + if err != nil { + return nil, err + } + + return &types.AdminGenerateDiamondInviteCodeResp{ + Codes: codes, + }, nil +} diff --git a/app/main/api/internal/logic/admin_agent/admingetagentcommissionlistlogic.go b/app/main/api/internal/logic/admin_agent/admingetagentcommissionlistlogic.go new file mode 100644 index 0000000..c9f31af --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingetagentcommissionlistlogic.go @@ -0,0 +1,74 @@ +package admin_agent + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/Masterminds/squirrel" + "github.com/jinzhu/copier" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetAgentCommissionListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetAgentCommissionListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetAgentCommissionListLogic { + return &AdminGetAgentCommissionListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetAgentCommissionListLogic) AdminGetAgentCommissionList(req *types.AdminGetAgentCommissionListReq) (resp *types.AdminGetAgentCommissionListResp, err error) { + builder := l.svcCtx.AgentCommissionModel.SelectBuilder() + if req.AgentId != nil { + builder = builder.Where(squirrel.Eq{"agent_id": *req.AgentId}) + } + if req.Status != nil { + builder = builder.Where(squirrel.Eq{"status": *req.Status}) + } + // 产品名称筛选功能已移除，如需可按product_id筛选 + + list, total, err := l.svcCtx.AgentCommissionModel.FindPageListByPageWithTotal(l.ctx, builder, req.Page, req.PageSize, "create_time DESC") + if err != nil { + return nil, err + } + + // 批量查product_name + productIds := make(map[string]struct{}) + for _, v := range list { + productIds[v.ProductId] = struct{}{} + } + productNameMap := make(map[string]string) + if len(productIds) > 0 { + ids := make([]string, 0, len(productIds)) + for id := range productIds { + ids = append(ids, id) + } + builder := l.svcCtx.ProductModel.SelectBuilder().Where(squirrel.Eq{"id": ids}) + products, _ := l.svcCtx.ProductModel.FindAll(l.ctx, builder, "") + for _, p := range products { + productNameMap[p.Id] = p.ProductName + } + } + + items := make([]types.AgentCommissionListItem, 0, len(list)) + for _, v := range list { + item := types.AgentCommissionListItem{} + _ = copier.Copy(&item, v) + item.ProductName = productNameMap[v.ProductId] + item.CreateTime = v.CreateTime.Format("2006-01-02 15:04:05") + items = append(items, item) + } + resp = &types.AdminGetAgentCommissionListResp{ + Total: total, + Items: items, + } + return +} diff --git a/app/main/api/internal/logic/admin_agent/admingetagentconfiglogic.go b/app/main/api/internal/logic/admin_agent/admingetagentconfiglogic.go new file mode 100644 index 0000000..1d99af0 --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingetagentconfiglogic.go @@ -0,0 +1,104 @@ +package admin_agent + +import ( + "context" + "strconv" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetAgentConfigLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetAgentConfigLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetAgentConfigLogic { + return &AdminGetAgentConfigLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetAgentConfigLogic) AdminGetAgentConfig() (resp *types.AdminGetAgentConfigResp, err error) { + // 获取配置值的辅助函数 + getConfigFloat := func(key string) float64 { + config, err := l.svcCtx.AgentConfigModel.FindOneByConfigKey(l.ctx, key) + if err != nil { + return 0 + } + value, _ := strconv.ParseFloat(config.ConfigValue, 64) + return value + } + getConfigInt := func(key string) int64 { + config, err := l.svcCtx.AgentConfigModel.FindOneByConfigKey(l.ctx, key) + if err != nil { + return 0 + } + value, _ := strconv.ParseInt(config.ConfigValue, 10, 64) + return value + } + + // 获取等级加成配置 + level1Bonus := getConfigFloat("level_1_bonus") + level2Bonus := getConfigFloat("level_2_bonus") + level3Bonus := getConfigFloat("level_3_bonus") + + // 获取升级费用配置 + upgradeToGoldFee := getConfigFloat("upgrade_to_gold_fee") + upgradeToDiamondFee := getConfigFloat("upgrade_to_diamond_fee") + + // 获取升级返佣配置 + upgradeToGoldRebate := getConfigFloat("upgrade_to_gold_rebate") + upgradeToDiamondRebate := getConfigFloat("upgrade_to_diamond_rebate") + + // 获取直接上级返佣配置 + directParentAmountDiamond := getConfigFloat("direct_parent_amount_diamond") + directParentAmountGold := getConfigFloat("direct_parent_amount_gold") + directParentAmountNormal := getConfigFloat("direct_parent_amount_normal") + + // 获取黄金代理最大返佣金额 + maxGoldRebateAmount := getConfigFloat("max_gold_rebate_amount") + + // 获取佣金冻结配置 + commissionFreezeRatio := getConfigFloat("commission_freeze_ratio") + commissionFreezeThreshold := getConfigFloat("commission_freeze_threshold") + commissionFreezeDays := getConfigInt("commission_freeze_days") + + goldUplift := getConfigFloat("gold_max_uplift_amount") + diamondUplift := getConfigFloat("diamond_max_uplift_amount") + return &types.AdminGetAgentConfigResp{ + LevelBonus: types.LevelBonusConfig{ + Normal: int64(level1Bonus), + Gold: int64(level2Bonus), + Diamond: int64(level3Bonus), + }, + UpgradeFee: types.UpgradeFeeConfig{ + NormalToGold: upgradeToGoldFee, + NormalToDiamond: upgradeToDiamondFee, + }, + UpgradeRebate: types.UpgradeRebateConfig{ + NormalToGoldRebate: upgradeToGoldRebate, + ToDiamondRebate: upgradeToDiamondRebate, + }, + DirectParentRebate: types.DirectParentRebateConfig{ + Diamond: directParentAmountDiamond, + Gold: directParentAmountGold, + Normal: directParentAmountNormal, + }, + MaxGoldRebateAmount: maxGoldRebateAmount, + CommissionFreeze: types.CommissionFreezeConfig{ + Ratio: commissionFreezeRatio, + Threshold: commissionFreezeThreshold, + Days: commissionFreezeDays, + }, + TaxRate: getConfigFloat("tax_rate"), + TaxExemptionAmount: getConfigFloat("tax_exemption_amount"), + GoldMaxUpliftAmount: goldUplift, + DiamondMaxUpliftAmount: diamondUplift, + }, nil +} diff --git a/app/main/api/internal/logic/admin_agent/admingetagentlinklistlogic.go b/app/main/api/internal/logic/admin_agent/admingetagentlinklistlogic.go new file mode 100644 index 0000000..312cf22 --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingetagentlinklistlogic.go @@ -0,0 +1,82 @@ +package admin_agent + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/Masterminds/squirrel" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetAgentLinkListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetAgentLinkListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetAgentLinkListLogic { + return &AdminGetAgentLinkListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetAgentLinkListLogic) AdminGetAgentLinkList(req *types.AdminGetAgentLinkListReq) (resp *types.AdminGetAgentLinkListResp, err error) { + builder := l.svcCtx.AgentLinkModel.SelectBuilder() + if req.AgentId != nil { + builder = builder.Where("agent_id = ?", *req.AgentId) + } + if req.LinkIdentifier != nil && *req.LinkIdentifier != "" { + builder = builder.Where("link_identifier = ?", *req.LinkIdentifier) + } + + // 如果传入ProductId，添加筛选条件 + if req.ProductId != nil { + builder = builder.Where("product_id = ?", *req.ProductId) + } + + links, total, err := l.svcCtx.AgentLinkModel.FindPageListByPageWithTotal(l.ctx, builder, req.Page, req.PageSize, "id DESC") + if err != nil { + return nil, err + } + + // 批量查product_id->name，避免N+1 + productIdSet := make(map[string]struct{}) + for _, link := range links { + productIdSet[link.ProductId] = struct{}{} + } + productIdList := make([]string, 0, len(productIdSet)) + for id := range productIdSet { + productIdList = append(productIdList, id) + } + productNameMap := make(map[string]string) + if len(productIdList) > 0 { + products, _ := l.svcCtx.ProductModel.FindAll(l.ctx, l.svcCtx.ProductModel.SelectBuilder().Where(squirrel.Eq{"id": productIdList}), "") + for _, p := range products { + productNameMap[p.Id] = p.ProductName + } + } + + items := make([]types.AgentLinkListItem, 0, len(links)) + for _, link := range links { + items = append(items, types.AgentLinkListItem{ + Id: link.Id, + AgentId: link.AgentId, + ProductId: link.ProductId, + ProductName: productNameMap[link.ProductId], + SetPrice: link.SetPrice, + ActualBasePrice: link.ActualBasePrice, + LinkIdentifier: link.LinkIdentifier, + CreateTime: link.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + resp = &types.AdminGetAgentLinkListResp{ + Total: total, + Items: items, + } + return +} diff --git a/app/main/api/internal/logic/admin_agent/admingetagentlistlogic.go b/app/main/api/internal/logic/admin_agent/admingetagentlistlogic.go new file mode 100644 index 0000000..15c749a --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingetagentlistlogic.go @@ -0,0 +1,137 @@ +package admin_agent + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetAgentListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetAgentListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetAgentListLogic { + return &AdminGetAgentListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetAgentListLogic) AdminGetAgentList(req *types.AdminGetAgentListReq) (resp *types.AdminGetAgentListResp, err error) { + builder := l.svcCtx.AgentModel.SelectBuilder() + + // 如果传入TeamLeaderId，则查找该团队首领下的所有代理 + if req.TeamLeaderId != nil { + builder = builder.Where(squirrel.Eq{"team_leader_id": *req.TeamLeaderId}) + } + if req.Level != nil { + builder = builder.Where(squirrel.Eq{"level": *req.Level}) + } + if req.Mobile != nil && *req.Mobile != "" { + // 加密手机号进行查询 + encryptedMobile, err := crypto.EncryptMobile(*req.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "加密手机号失败: %v", err) + } + builder = builder.Where(squirrel.Eq{"mobile": encryptedMobile}) + } + if req.Region != nil && *req.Region != "" { + // 注意：region字段现在是可选的，查询时需要处理NULL值 + // 如果region字段是NULL，使用IS NULL查询；否则使用等值查询 + // 这里简化处理，直接使用等值查询（如果数据库中有NULL值，需要特殊处理） + builder = builder.Where(squirrel.Eq{"region": *req.Region}) + } + + agents, total, err := l.svcCtx.AgentModel.FindPageListByPageWithTotal(l.ctx, builder, req.Page, req.PageSize, "id DESC") + if err != nil { + return nil, err + } + + items := make([]types.AgentListItem, 0, len(agents)) + + for _, agent := range agents { + // 获取等级名称 + levelName := "" + switch agent.Level { + case 1: + levelName = "普通" + case 2: + levelName = "黄金" + case 3: + levelName = "钻石" + } + + agent.Mobile, err = crypto.DecryptMobile(agent.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取代理信息, 解密手机号失败: %v", err) + } + + // 查询钱包信息 + wallet, _ := l.svcCtx.AgentWalletModel.FindOneByAgentId(l.ctx, agent.Id) + + // 查询实名认证信息 + realNameInfo, _ := l.svcCtx.AgentRealNameModel.FindOneByAgentId(l.ctx, agent.Id) + isRealName := false + if realNameInfo != nil && realNameInfo.VerifyTime.Valid { + isRealName = true // verify_time不为空表示已通过三要素核验 + } + + wechatId := "" + if agent.WechatId.Valid { + wechatId = agent.WechatId.String + } + teamLeaderId := "" + if agent.TeamLeaderId.Valid { + teamLeaderId = agent.TeamLeaderId.String + } + + // 获取区域 + region := "" + if agent.Region.Valid { + region = agent.Region.String + } + + item := types.AgentListItem{ + Id: agent.Id, + UserId: agent.UserId, + Level: agent.Level, + LevelName: levelName, + Region: region, + Mobile: agent.Mobile, + WechatId: wechatId, + TeamLeaderId: teamLeaderId, + AgentCode: agent.AgentCode, + Balance: 0, + TotalEarnings: 0, + FrozenBalance: 0, + WithdrawnAmount: 0, + IsRealName: isRealName, + CreateTime: agent.CreateTime.Format("2006-01-02 15:04:05"), + } + + if wallet != nil { + item.Balance = wallet.Balance + item.TotalEarnings = wallet.TotalEarnings + item.FrozenBalance = wallet.FrozenBalance + item.WithdrawnAmount = wallet.WithdrawnAmount + } + + items = append(items, item) + } + + resp = &types.AdminGetAgentListResp{ + Total: total, + Items: items, + } + return +} diff --git a/app/main/api/internal/logic/admin_agent/admingetagentorderlistlogic.go b/app/main/api/internal/logic/admin_agent/admingetagentorderlistlogic.go new file mode 100644 index 0000000..8e12bb5 --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingetagentorderlistlogic.go @@ -0,0 +1,100 @@ +package admin_agent + +import ( + "context" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetAgentOrderListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetAgentOrderListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetAgentOrderListLogic { + return &AdminGetAgentOrderListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetAgentOrderListLogic) AdminGetAgentOrderList(req *types.AdminGetAgentOrderListReq) (resp *types.AdminGetAgentOrderListResp, err error) { + builder := l.svcCtx.AgentOrderModel.SelectBuilder(). + Where("del_state = ?", globalkey.DelStateNo) + + if req.AgentId != nil { + builder = builder.Where("agent_id = ?", *req.AgentId) + } + if req.OrderId != nil { + builder = builder.Where("order_id = ?", *req.OrderId) + } + if req.ProcessStatus != nil { + builder = builder.Where("process_status = ?", *req.ProcessStatus) + } + + // 分页查询 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + + orders, total, err := l.svcCtx.AgentOrderModel.FindPageListByPageWithTotal(l.ctx, builder, page, pageSize, "create_time DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理订单列表失败, %v", err) + } + + // 批量查询产品名称 + productIdSet := make(map[string]struct{}) + for _, order := range orders { + productIdSet[order.ProductId] = struct{}{} + } + productIdList := make([]string, 0, len(productIdSet)) + for id := range productIdSet { + productIdList = append(productIdList, id) + } + productNameMap := make(map[string]string) + if len(productIdList) > 0 { + products, _ := l.svcCtx.ProductModel.FindAll(l.ctx, l.svcCtx.ProductModel.SelectBuilder().Where(squirrel.Eq{"id": productIdList}), "") + for _, p := range products { + productNameMap[p.Id] = p.ProductName + } + } + + // 组装响应 + items := make([]types.AgentOrderListItem, 0, len(orders)) + for _, order := range orders { + items = append(items, types.AgentOrderListItem{ + Id: order.Id, + AgentId: order.AgentId, + OrderId: order.OrderId, + ProductId: order.ProductId, + ProductName: productNameMap[order.ProductId], + OrderAmount: order.OrderAmount, + SetPrice: order.SetPrice, + ActualBasePrice: order.ActualBasePrice, + PriceCost: order.PriceCost, + AgentProfit: order.AgentProfit, + ProcessStatus: order.ProcessStatus, + CreateTime: order.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return &types.AdminGetAgentOrderListResp{ + Total: total, + Items: items, + }, nil +} diff --git a/app/main/api/internal/logic/admin_agent/admingetagentproductconfiglistlogic.go b/app/main/api/internal/logic/admin_agent/admingetagentproductconfiglistlogic.go new file mode 100644 index 0000000..f27293d --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingetagentproductconfiglistlogic.go @@ -0,0 +1,99 @@ +package admin_agent + +import ( + "context" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetAgentProductConfigListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetAgentProductConfigListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetAgentProductConfigListLogic { + return &AdminGetAgentProductConfigListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetAgentProductConfigListLogic) AdminGetAgentProductConfigList(req *types.AdminGetAgentProductConfigListReq) (resp *types.AdminGetAgentProductConfigListResp, err error) { + builder := l.svcCtx.AgentProductConfigModel.SelectBuilder(). + Where("del_state = ?", globalkey.DelStateNo) + + // 如果提供了产品ID，直接过滤 + if req.ProductId != nil { + builder = builder.Where("product_id = ?", *req.ProductId) + } + + // 如果提供了产品名称，通过关联查询 product 表过滤 + if req.ProductName != nil && *req.ProductName != "" { + builder = builder.Where("product_id IN (SELECT id FROM product WHERE product_name LIKE ? AND del_state = ?)", "%"+*req.ProductName+"%", globalkey.DelStateNo) + } + + // 分页查询 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + + configs, total, err := l.svcCtx.AgentProductConfigModel.FindPageListByPageWithTotal(l.ctx, builder, page, pageSize, "id DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询产品配置列表失败, %v", err) + } + + // 组装响应（通过 product_id 查询产品名称） + items := make([]types.AgentProductConfigItem, 0, len(configs)) + for _, config := range configs { + // 通过 product_id 查询产品信息获取产品名称 + product, err := l.svcCtx.ProductModel.FindOne(l.ctx, config.ProductId) + productName := "" + if err == nil { + productName = product.ProductName + } else { + // 如果产品不存在，记录日志但不影响主流程 + l.Infof("查询产品信息失败, productId: %d, err: %v", config.ProductId, err) + } + + priceThreshold := 0.0 + if config.PriceThreshold.Valid { + priceThreshold = config.PriceThreshold.Float64 + } + + priceFeeRate := 0.0 + if config.PriceFeeRate.Valid { + priceFeeRate = config.PriceFeeRate.Float64 + } + + items = append(items, types.AgentProductConfigItem{ + Id: config.Id, + ProductId: config.ProductId, + ProductName: productName, + BasePrice: config.BasePrice, + PriceRangeMin: config.BasePrice, // 最低定价等于基础底价 + PriceRangeMax: config.SystemMaxPrice, + PriceThreshold: priceThreshold, + PriceFeeRate: priceFeeRate, + CreateTime: config.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return &types.AdminGetAgentProductConfigListResp{ + Total: total, + Items: items, + }, nil +} diff --git a/app/main/api/internal/logic/admin_agent/admingetagentrealnamelistlogic.go b/app/main/api/internal/logic/admin_agent/admingetagentrealnamelistlogic.go new file mode 100644 index 0000000..bdd7f03 --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingetagentrealnamelistlogic.go @@ -0,0 +1,115 @@ +package admin_agent + +import ( + "context" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetAgentRealNameListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetAgentRealNameListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetAgentRealNameListLogic { + return &AdminGetAgentRealNameListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetAgentRealNameListLogic) AdminGetAgentRealNameList(req *types.AdminGetAgentRealNameListReq) (resp *types.AdminGetAgentRealNameListResp, err error) { + builder := l.svcCtx.AgentRealNameModel.SelectBuilder(). + Where("del_state = ?", globalkey.DelStateNo) + + if req.AgentId != nil { + builder = builder.Where("agent_id = ?", *req.AgentId) + } + if req.Status != nil { + // 根据状态过滤：1=未验证（verify_time为NULL），2=已通过（verify_time不为NULL） + switch *req.Status { + case 1: // 未验证 + builder = builder.Where("verify_time IS NULL") + case 2: // 已通过 + builder = builder.Where("verify_time IS NOT NULL") + } + } + + // 分页查询 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + + realNames, total, err := l.svcCtx.AgentRealNameModel.FindPageListByPageWithTotal(l.ctx, builder, page, pageSize, "create_time DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询实名认证列表失败, %v", err) + } + + // 组装响应 + items := make([]types.AgentRealNameListItem, 0, len(realNames)) + for _, realName := range realNames { + // 解密手机号（仅显示部分） + mobile := "" + if realName.Mobile != "" { + decrypted, err := crypto.DecryptMobile(realName.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err == nil { + mobile = decrypted + } + } + + // 解密身份证号（仅显示部分） + idCard := "" + if realName.IdCard != "" { + decrypted, err := crypto.DecryptIDCard(realName.IdCard, []byte(l.svcCtx.Config.Encrypt.SecretKey)) + if err == nil { + // 脱敏显示 + if len(decrypted) > 10 { + idCard = decrypted[:3] + "***********" + decrypted[len(decrypted)-4:] + } else { + idCard = decrypted + } + } + } + + // 根据verify_time判断状态：NULL=未验证(1)，不为NULL=已通过(2) + statusInt := int64(1) // 默认未验证 + verifyTime := "" + if realName.VerifyTime.Valid { + statusInt = 2 // 已通过 + verifyTime = realName.VerifyTime.Time.Format("2006-01-02 15:04:05") + } + + item := types.AgentRealNameListItem{ + Id: realName.Id, + AgentId: realName.AgentId, + Name: realName.Name, + IdCard: idCard, + Mobile: mobile, + Status: statusInt, + CreateTime: realName.CreateTime.Format("2006-01-02 15:04:05"), + } + // TODO: 重新生成接口后，取消注释下面的代码 + item.VerifyTime = verifyTime + items = append(items, item) + } + + return &types.AdminGetAgentRealNameListResp{ + Total: total, + Items: items, + }, nil +} diff --git a/app/main/api/internal/logic/admin_agent/admingetagentrebatelistlogic.go b/app/main/api/internal/logic/admin_agent/admingetagentrebatelistlogic.go new file mode 100644 index 0000000..a13aeeb --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingetagentrebatelistlogic.go @@ -0,0 +1,95 @@ +package admin_agent + +import ( + "context" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetAgentRebateListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetAgentRebateListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetAgentRebateListLogic { + return &AdminGetAgentRebateListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetAgentRebateListLogic) AdminGetAgentRebateList(req *types.AdminGetAgentRebateListReq) (resp *types.AdminGetAgentRebateListResp, err error) { + builder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("del_state = ?", globalkey.DelStateNo) + + if req.AgentId != nil { + builder = builder.Where("agent_id = ?", *req.AgentId) + } + if req.SourceAgentId != nil { + builder = builder.Where("source_agent_id = ?", *req.SourceAgentId) + } + if req.RebateType != nil { + builder = builder.Where("rebate_type = ?", *req.RebateType) + } + + // 分页查询 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + + rebates, total, err := l.svcCtx.AgentRebateModel.FindPageListByPageWithTotal(l.ctx, builder, page, pageSize, "create_time DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询返佣列表失败, %v", err) + } + + // 批量查询产品名称 + productIdSet := make(map[string]struct{}) + for _, rebate := range rebates { + productIdSet[rebate.ProductId] = struct{}{} + } + productIdList := make([]string, 0, len(productIdSet)) + for id := range productIdSet { + productIdList = append(productIdList, id) + } + productNameMap := make(map[string]string) + if len(productIdList) > 0 { + products, _ := l.svcCtx.ProductModel.FindAll(l.ctx, l.svcCtx.ProductModel.SelectBuilder().Where(squirrel.Eq{"id": productIdList}), "") + for _, p := range products { + productNameMap[p.Id] = p.ProductName + } + } + + // 组装响应 + items := make([]types.AgentRebateListItem, 0, len(rebates)) + for _, rebate := range rebates { + items = append(items, types.AgentRebateListItem{ + Id: rebate.Id, + AgentId: rebate.AgentId, + SourceAgentId: rebate.SourceAgentId, + OrderId: rebate.OrderId, + RebateType: rebate.RebateType, + Amount: rebate.RebateAmount, + CreateTime: rebate.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return &types.AdminGetAgentRebateListResp{ + Total: total, + Items: items, + }, nil +} diff --git a/app/main/api/internal/logic/admin_agent/admingetagentupgradelistlogic.go b/app/main/api/internal/logic/admin_agent/admingetagentupgradelistlogic.go new file mode 100644 index 0000000..0595c1d --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingetagentupgradelistlogic.go @@ -0,0 +1,79 @@ +package admin_agent + +import ( + "context" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetAgentUpgradeListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetAgentUpgradeListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetAgentUpgradeListLogic { + return &AdminGetAgentUpgradeListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetAgentUpgradeListLogic) AdminGetAgentUpgradeList(req *types.AdminGetAgentUpgradeListReq) (resp *types.AdminGetAgentUpgradeListResp, err error) { + builder := l.svcCtx.AgentUpgradeModel.SelectBuilder(). + Where("del_state = ?", globalkey.DelStateNo) + + if req.AgentId != nil { + builder = builder.Where("agent_id = ?", *req.AgentId) + } + if req.UpgradeType != nil { + builder = builder.Where("upgrade_type = ?", *req.UpgradeType) + } + if req.Status != nil { + builder = builder.Where("status = ?", *req.Status) + } + + // 分页查询 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + + upgrades, total, err := l.svcCtx.AgentUpgradeModel.FindPageListByPageWithTotal(l.ctx, builder, page, pageSize, "create_time DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询升级记录列表失败, %v", err) + } + + // 组装响应 + items := make([]types.AgentUpgradeListItem, 0, len(upgrades)) + for _, upgrade := range upgrades { + items = append(items, types.AgentUpgradeListItem{ + Id: upgrade.Id, + AgentId: upgrade.AgentId, + FromLevel: upgrade.FromLevel, + ToLevel: upgrade.ToLevel, + UpgradeType: upgrade.UpgradeType, + UpgradeFee: upgrade.UpgradeFee, + RebateAmount: upgrade.RebateAmount, + Status: upgrade.Status, + CreateTime: upgrade.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return &types.AdminGetAgentUpgradeListResp{ + Total: total, + Items: items, + }, nil +} diff --git a/app/main/api/internal/logic/admin_agent/admingetagentwithdrawallistlogic.go b/app/main/api/internal/logic/admin_agent/admingetagentwithdrawallistlogic.go new file mode 100644 index 0000000..c74dafe --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingetagentwithdrawallistlogic.go @@ -0,0 +1,59 @@ +package admin_agent + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/Masterminds/squirrel" + "github.com/jinzhu/copier" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetAgentWithdrawalListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetAgentWithdrawalListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetAgentWithdrawalListLogic { + return &AdminGetAgentWithdrawalListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetAgentWithdrawalListLogic) AdminGetAgentWithdrawalList(req *types.AdminGetAgentWithdrawalListReq) (resp *types.AdminGetAgentWithdrawalListResp, err error) { + builder := l.svcCtx.AgentWithdrawalModel.SelectBuilder() + if req.AgentId != nil { + builder = builder.Where(squirrel.Eq{"agent_id": *req.AgentId}) + } + if req.Status != nil { + builder = builder.Where(squirrel.Eq{"status": *req.Status}) + } + if req.WithdrawNo != nil && *req.WithdrawNo != "" { + builder = builder.Where(squirrel.Eq{"withdraw_no": *req.WithdrawNo}) + } + list, total, err := l.svcCtx.AgentWithdrawalModel.FindPageListByPageWithTotal(l.ctx, builder, req.Page, req.PageSize, "create_time DESC") + if err != nil { + return nil, err + } + items := make([]types.AgentWithdrawalListItem, 0, len(list)) + for _, v := range list { + item := types.AgentWithdrawalListItem{} + _ = copier.Copy(&item, v) + item.Remark = "" + if v.Remark.Valid { + item.Remark = v.Remark.String + } + item.CreateTime = v.CreateTime.Format("2006-01-02 15:04:05") + items = append(items, item) + } + resp = &types.AdminGetAgentWithdrawalListResp{ + Total: total, + Items: items, + } + return +} diff --git a/app/main/api/internal/logic/admin_agent/admingetinvitecodelistlogic.go b/app/main/api/internal/logic/admin_agent/admingetinvitecodelistlogic.go new file mode 100644 index 0000000..1fdf7b3 --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/admingetinvitecodelistlogic.go @@ -0,0 +1,128 @@ +package admin_agent + +import ( + "context" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetInviteCodeListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetInviteCodeListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetInviteCodeListLogic { + return &AdminGetInviteCodeListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetInviteCodeListLogic) AdminGetInviteCodeList(req *types.AdminGetInviteCodeListReq) (resp *types.AdminGetInviteCodeListResp, err error) { + // 1. 构建查询条件 + builder := l.svcCtx.AgentInviteCodeModel.SelectBuilder(). + Where("del_state = ?", globalkey.DelStateNo) + + if req.Code != nil && *req.Code != "" { + builder = builder.Where("code = ?", *req.Code) + } + if req.AgentId != nil && *req.AgentId != "" { + if *req.AgentId == "0" { + // agent_id = 0 表示查询平台发放的邀请码（agent_id为NULL） + builder = builder.Where("agent_id IS NULL") + } else { + builder = builder.Where("agent_id = ?", *req.AgentId) + } + } + if req.TargetLevel != nil { + builder = builder.Where("target_level = ?", *req.TargetLevel) + } + if req.Status != nil { + builder = builder.Where("status = ?", *req.Status) + } + + // 2. 分页查询 + list, total, err := l.svcCtx.AgentInviteCodeModel.FindPageListByPageWithTotal(l.ctx, builder, req.Page, req.PageSize, "create_time DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询邀请码列表失败, %v", err) + } + + // 3. 批量查询代理信息（用于显示代理手机号） + agentIds := make(map[string]struct{}) + for _, v := range list { + if v.AgentId.Valid && v.AgentId.String != "" { + agentIds[v.AgentId.String] = struct{}{} + } + } + + agentMobileMap := make(map[string]string) + if len(agentIds) > 0 { + agentIdList := make([]string, 0, len(agentIds)) + for id := range agentIds { + agentIdList = append(agentIdList, id) + } + agents, _ := l.svcCtx.AgentModel.FindAll(l.ctx, l.svcCtx.AgentModel.SelectBuilder().Where(squirrel.Eq{"id": agentIdList}), "") + secretKey := l.svcCtx.Config.Encrypt.SecretKey + for _, agent := range agents { + mobile, decErr := crypto.DecryptMobile(agent.Mobile, secretKey) + if decErr == nil { + agentMobileMap[agent.Id] = mobile + } else { + l.Logger.Errorf("解密代理手机号失败: %v", decErr) + } + } + } + + // 4. 格式化返回数据 + items := make([]types.InviteCodeListItem, 0, len(list)) + for _, v := range list { + item := types.InviteCodeListItem{ + Id: v.Id, + Code: v.Code, + AgentId: "", + AgentMobile: "", + TargetLevel: v.TargetLevel, + Status: v.Status, + CreateTime: v.CreateTime.Format("2006-01-02 15:04:05"), + } + + if v.AgentId.Valid { + item.AgentId = v.AgentId.String + item.AgentMobile = agentMobileMap[v.AgentId.String] + } + + if v.UsedUserId.Valid { + item.UsedUserId = v.UsedUserId.String + } + if v.UsedAgentId.Valid { + item.UsedAgentId = v.UsedAgentId.String + } + if v.UsedTime.Valid { + item.UsedTime = v.UsedTime.Time.Format("2006-01-02 15:04:05") + } + if v.ExpireTime.Valid { + item.ExpireTime = v.ExpireTime.Time.Format("2006-01-02 15:04:05") + } + if v.Remark.Valid { + item.Remark = v.Remark.String + } + + items = append(items, item) + } + + return &types.AdminGetInviteCodeListResp{ + Total: total, + Items: items, + }, nil +} diff --git a/app/main/api/internal/logic/admin_agent/adminupdateagentconfiglogic.go b/app/main/api/internal/logic/admin_agent/adminupdateagentconfiglogic.go new file mode 100644 index 0000000..2722f86 --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/adminupdateagentconfiglogic.go @@ -0,0 +1,193 @@ +package admin_agent + +import ( + "context" + "database/sql" + "strconv" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminUpdateAgentConfigLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdateAgentConfigLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateAgentConfigLogic { + return &AdminUpdateAgentConfigLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdateAgentConfigLogic) AdminUpdateAgentConfig(req *types.AdminUpdateAgentConfigReq) (resp *types.AdminUpdateAgentConfigResp, err error) { + configTypeForKey := func(key string) string { + switch key { + case "level_1_bonus", "level_2_bonus", "level_3_bonus": + return "bonus" + case "upgrade_to_gold_fee", "upgrade_to_diamond_fee", "upgrade_to_gold_rebate", "upgrade_to_diamond_rebate": + return "upgrade" + case "direct_parent_amount_diamond", "direct_parent_amount_gold", "direct_parent_amount_normal", "max_gold_rebate_amount": + return "rebate" + case "commission_freeze_ratio", "commission_freeze_threshold", "commission_freeze_days": + return "rebate" + case "tax_rate", "tax_exemption_amount": + return "tax" + case "gold_max_uplift_amount", "diamond_max_uplift_amount": + return "price" + default: + return "rebate" + } + } + + updateConfig := func(key string, value *float64) error { + if value == nil { + return nil + } + valStr := strconv.FormatFloat(*value, 'f', -1, 64) + config, err := l.svcCtx.AgentConfigModel.FindOneByConfigKey(l.ctx, key) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + cfg := &model.AgentConfig{ + ConfigKey: key, + ConfigValue: valStr, + ConfigType: configTypeForKey(key), + Description: sql.NullString{}, + DeleteTime: sql.NullTime{}, + Version: 0, + } + _, insErr := l.svcCtx.AgentConfigModel.Insert(l.ctx, nil, cfg) + if insErr != nil { + return errors.Wrapf(insErr, "创建配置失败, key: %s", key) + } + return nil + } + return errors.Wrapf(err, "查询配置失败, key: %s", key) + } + config.ConfigValue = valStr + if uErr := l.svcCtx.AgentConfigModel.UpdateWithVersion(l.ctx, nil, config); uErr != nil { + if errors.Is(uErr, model.ErrNoRowsUpdate) { + latestByKey, reErr := l.svcCtx.AgentConfigModel.FindOneByConfigKey(l.ctx, key) + if reErr != nil { + if errors.Is(reErr, model.ErrNotFound) { + cfg := &model.AgentConfig{ + ConfigKey: key, + ConfigValue: valStr, + ConfigType: configTypeForKey(key), + Description: sql.NullString{}, + DeleteTime: sql.NullTime{}, + Version: 0, + } + _, insErr := l.svcCtx.AgentConfigModel.Insert(l.ctx, nil, cfg) + if insErr != nil { + return errors.Wrapf(insErr, "创建配置失败, key: %s", key) + } + return nil + } + return errors.Wrapf(reErr, "查询最新配置失败, key: %s", key) + } + latestByKey.ConfigValue = valStr + return l.svcCtx.AgentConfigModel.UpdateWithVersion(l.ctx, nil, latestByKey) + } + return uErr + } + return nil + } + + // 更新等级加成配置 + if req.LevelBonus != nil { + if err := updateConfig("level_1_bonus", func() *float64 { v := float64(req.LevelBonus.Normal); return &v }()); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新普通代理等级加成失败, %v", err) + } + if err := updateConfig("level_2_bonus", func() *float64 { v := float64(req.LevelBonus.Gold); return &v }()); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新黄金代理等级加成失败, %v", err) + } + if err := updateConfig("level_3_bonus", func() *float64 { v := float64(req.LevelBonus.Diamond); return &v }()); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新钻石代理等级加成失败, %v", err) + } + } + + // 更新升级费用配置 + if req.UpgradeFee != nil { + if err := updateConfig("upgrade_to_gold_fee", &req.UpgradeFee.NormalToGold); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新普通→黄金升级费用失败, %v", err) + } + if err := updateConfig("upgrade_to_diamond_fee", &req.UpgradeFee.NormalToDiamond); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新普通→钻石升级费用失败, %v", err) + } + // gold_to_diamond 是计算得出的，不需要更新 + } + + // 更新升级返佣配置 + if req.UpgradeRebate != nil { + if err := updateConfig("upgrade_to_gold_rebate", &req.UpgradeRebate.NormalToGoldRebate); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新普通→黄金返佣失败, %v", err) + } + if err := updateConfig("upgrade_to_diamond_rebate", &req.UpgradeRebate.ToDiamondRebate); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新升级为钻石返佣失败, %v", err) + } + } + + // 更新直接上级返佣配置 + if req.DirectParentRebate != nil { + if err := updateConfig("direct_parent_amount_diamond", &req.DirectParentRebate.Diamond); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新直接上级是钻石的返佣金额失败, %v", err) + } + if err := updateConfig("direct_parent_amount_gold", &req.DirectParentRebate.Gold); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新直接上级是黄金的返佣金额失败, %v", err) + } + if err := updateConfig("direct_parent_amount_normal", &req.DirectParentRebate.Normal); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新直接上级是普通的返佣金额失败, %v", err) + } + } + + // 更新黄金代理最大返佣金额 + if err := updateConfig("max_gold_rebate_amount", req.MaxGoldRebateAmount); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新黄金代理最大返佣金额失败, %v", err) + } + + // 更新佣金冻结配置 + if req.CommissionFreeze != nil { + if err := updateConfig("commission_freeze_ratio", &req.CommissionFreeze.Ratio); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新佣金冻结比例失败, %v", err) + } + if err := updateConfig("commission_freeze_threshold", &req.CommissionFreeze.Threshold); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新佣金冻结阈值失败, %v", err) + } + // 更新解冻天数（整数类型） + if req.CommissionFreeze.Days > 0 { + daysFloat := float64(req.CommissionFreeze.Days) + if err := updateConfig("commission_freeze_days", &daysFloat); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新佣金冻结解冻天数失败, %v", err) + } + } + } + + // 更新税费配置 + if err := updateConfig("tax_rate", req.TaxRate); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新税率失败, %v", err) + } + if err := updateConfig("tax_exemption_amount", req.TaxExemptionAmount); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新免税额度失败, %v", err) + } + + if err := updateConfig("gold_max_uplift_amount", req.GoldMaxUpliftAmount); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新黄金代理最高价上调金额失败, %v", err) + } + if err := updateConfig("diamond_max_uplift_amount", req.DiamondMaxUpliftAmount); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新钻石代理最高价上调金额失败, %v", err) + } + + return &types.AdminUpdateAgentConfigResp{ + Success: true, + }, nil +} diff --git a/app/main/api/internal/logic/admin_agent/adminupdateagentproductconfiglogic.go b/app/main/api/internal/logic/admin_agent/adminupdateagentproductconfiglogic.go new file mode 100644 index 0000000..1f7f798 --- /dev/null +++ b/app/main/api/internal/logic/admin_agent/adminupdateagentproductconfiglogic.go @@ -0,0 +1,63 @@ +package admin_agent + +import ( + "context" + "database/sql" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminUpdateAgentProductConfigLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdateAgentProductConfigLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateAgentProductConfigLogic { + return &AdminUpdateAgentProductConfigLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdateAgentProductConfigLogic) AdminUpdateAgentProductConfig(req *types.AdminUpdateAgentProductConfigReq) (resp *types.AdminUpdateAgentProductConfigResp, err error) { + // 查询配置 + config, err := l.svcCtx.AgentProductConfigModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询产品配置失败, %v", err) + } + + // 更新配置字段 + config.BasePrice = req.BasePrice + config.SystemMaxPrice = req.PriceRangeMax + + // 价格阈值（可选） + if req.PriceThreshold != nil { + config.PriceThreshold = sql.NullFloat64{Float64: *req.PriceThreshold, Valid: true} + } else { + config.PriceThreshold = sql.NullFloat64{Valid: false} + } + + // 提价手续费比例（可选） + if req.PriceFeeRate != nil { + config.PriceFeeRate = sql.NullFloat64{Float64: *req.PriceFeeRate, Valid: true} + } else { + config.PriceFeeRate = sql.NullFloat64{Valid: false} + } + + // 更新配置 + if err := l.svcCtx.AgentProductConfigModel.UpdateWithVersion(l.ctx, nil, config); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新产品配置失败, %v", err) + } + + return &types.AdminUpdateAgentProductConfigResp{ + Success: true, + }, nil +} diff --git a/app/main/api/internal/logic/admin_api/adminbatchupdateapistatuslogic.go b/app/main/api/internal/logic/admin_api/adminbatchupdateapistatuslogic.go new file mode 100644 index 0000000..338cfda --- /dev/null +++ b/app/main/api/internal/logic/admin_api/adminbatchupdateapistatuslogic.go @@ -0,0 +1,70 @@ +package admin_api + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminBatchUpdateApiStatusLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminBatchUpdateApiStatusLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminBatchUpdateApiStatusLogic { + return &AdminBatchUpdateApiStatusLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminBatchUpdateApiStatusLogic) AdminBatchUpdateApiStatus(req *types.AdminBatchUpdateApiStatusReq) (resp *types.AdminBatchUpdateApiStatusResp, err error) { + // 1. 参数验证 + if len(req.Ids) == 0 { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API ID列表不能为空") + } + if req.Status != 0 && req.Status != 1 { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "状态值无效, status: %d", req.Status) + } + + // 2. 批量更新API状态 + successCount := 0 + for _, id := range req.Ids { + if id == "" { + continue + } + + // 查询API是否存在 + api, err := l.svcCtx.AdminApiModel.FindOne(l.ctx, id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + continue // 跳过不存在的API + } + logx.Errorf("查询API失败, err: %v, id: %s", err, id) + continue + } + + // 更新状态 + api.Status = req.Status + _, err = l.svcCtx.AdminApiModel.Update(l.ctx, nil, api) + if err != nil { + logx.Errorf("更新API状态失败, err: %v, id: %d", err, id) + continue + } + + successCount++ + } + + // 3. 返回结果 + return &types.AdminBatchUpdateApiStatusResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_api/admincreateapilogic.go b/app/main/api/internal/logic/admin_api/admincreateapilogic.go new file mode 100644 index 0000000..d80b6a0 --- /dev/null +++ b/app/main/api/internal/logic/admin_api/admincreateapilogic.go @@ -0,0 +1,79 @@ +package admin_api + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminCreateApiLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminCreateApiLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminCreateApiLogic { + return &AdminCreateApiLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminCreateApiLogic) AdminCreateApi(req *types.AdminCreateApiReq) (resp *types.AdminCreateApiResp, err error) { + // 1. 参数验证 + if req.ApiName == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API名称不能为空") + } + if req.ApiCode == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API编码不能为空") + } + if req.Method == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "请求方法不能为空") + } + if req.Url == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API路径不能为空") + } + + // 2. 检查API编码是否已存在 + existing, err := l.svcCtx.AdminApiModel.FindOneByApiCode(l.ctx, req.ApiCode) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询API失败, err: %v, apiCode: %s", err, req.ApiCode) + } + if existing != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API编码已存在: %s", req.ApiCode) + } + + // 3. 创建API记录 + apiData := &model.AdminApi{ + Id: uuid.NewString(), + ApiName: req.ApiName, + ApiCode: req.ApiCode, + Method: req.Method, + Url: req.Url, + Status: req.Status, + Description: req.Description, + } + + result, err := l.svcCtx.AdminApiModel.Insert(l.ctx, nil, apiData) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "创建API失败, err: %v", err) + } + // 4. 返回结果 + _ = result + return &types.AdminCreateApiResp{Id: apiData.Id}, nil +} diff --git a/app/main/api/internal/logic/admin_api/admindeleteapilogic.go b/app/main/api/internal/logic/admin_api/admindeleteapilogic.go new file mode 100644 index 0000000..fcd3164 --- /dev/null +++ b/app/main/api/internal/logic/admin_api/admindeleteapilogic.go @@ -0,0 +1,68 @@ +package admin_api + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminDeleteApiLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminDeleteApiLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminDeleteApiLogic { + return &AdminDeleteApiLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminDeleteApiLogic) AdminDeleteApi(req *types.AdminDeleteApiReq) (resp *types.AdminDeleteApiResp, err error) { + // 1. 参数验证 + if req.Id == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API ID不能为空, id: %s", req.Id) + } + + // 2. 查询API是否存在 + api, err := l.svcCtx.AdminApiModel.FindOne(l.ctx, req.Id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "API不存在, id: %d", req.Id) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询API失败, err: %v, id: %d", err, req.Id) + } + + // 3. 检查是否有角色关联该API + roleApiBuilder := l.svcCtx.AdminRoleApiModel.SelectBuilder().Where("api_id = ?", req.Id) + roleApis, err := l.svcCtx.AdminRoleApiModel.FindAll(l.ctx, roleApiBuilder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询角色API关联失败, err: %v, apiId: %d", err, req.Id) + } + if len(roleApis) > 0 { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "该API已被角色使用，无法删除, apiId: %d", req.Id) + } + + // 4. 执行软删除 + err = l.svcCtx.AdminApiModel.DeleteSoft(l.ctx, nil, api) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "删除API失败, err: %v, id: %d", err, req.Id) + } + + // 5. 返回结果 + return &types.AdminDeleteApiResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_api/admingetapidetaillogic.go b/app/main/api/internal/logic/admin_api/admingetapidetaillogic.go new file mode 100644 index 0000000..a6de1fe --- /dev/null +++ b/app/main/api/internal/logic/admin_api/admingetapidetaillogic.go @@ -0,0 +1,61 @@ +package admin_api + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetApiDetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetApiDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetApiDetailLogic { + return &AdminGetApiDetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetApiDetailLogic) AdminGetApiDetail(req *types.AdminGetApiDetailReq) (resp *types.AdminGetApiDetailResp, err error) { + // 1. 参数验证 + if req.Id == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API ID不能为空, id: %s", req.Id) + } + + // 2. 查询API详情 + api, err := l.svcCtx.AdminApiModel.FindOne(l.ctx, req.Id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "API不存在, id: %d", req.Id) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询API详情失败, err: %v, id: %d", err, req.Id) + } + + // 3. 返回结果 + return &types.AdminGetApiDetailResp{ + AdminApiInfo: types.AdminApiInfo{ + Id: api.Id, + ApiName: api.ApiName, + ApiCode: api.ApiCode, + Method: api.Method, + Url: api.Url, + Status: api.Status, + Description: api.Description, + CreateTime: api.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: api.UpdateTime.Format("2006-01-02 15:04:05"), + }, + }, nil +} diff --git a/app/main/api/internal/logic/admin_api/admingetapilistlogic.go b/app/main/api/internal/logic/admin_api/admingetapilistlogic.go new file mode 100644 index 0000000..b9f2fee --- /dev/null +++ b/app/main/api/internal/logic/admin_api/admingetapilistlogic.go @@ -0,0 +1,89 @@ +package admin_api + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetApiListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetApiListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetApiListLogic { + return &AdminGetApiListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetApiListLogic) AdminGetApiList(req *types.AdminGetApiListReq) (resp *types.AdminGetApiListResp, err error) { + // 1. 参数验证 + if req.Page <= 0 { + req.Page = 1 + } + if req.PageSize <= 0 { + req.PageSize = 20 + } + if req.PageSize > 100 { + req.PageSize = 100 + } + + // 2. 构建查询条件 + builder := l.svcCtx.AdminApiModel.SelectBuilder() + + // 添加搜索条件 + if req.ApiName != "" { + builder = builder.Where("api_name LIKE ?", "%"+req.ApiName+"%") + } + if req.Method != "" { + builder = builder.Where("method = ?", req.Method) + } + if req.Status > 0 { + builder = builder.Where("status = ?", req.Status) + } + + // 3. 查询总数 + total, err := l.svcCtx.AdminApiModel.FindCount(l.ctx, builder, "id") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询API总数失败, err: %v", err) + } + + // 4. 查询列表 + apis, err := l.svcCtx.AdminApiModel.FindPageListByPage(l.ctx, builder, req.Page, req.PageSize, "id DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询API列表失败, err: %v", err) + } + + // 5. 转换数据格式 + var apiList []types.AdminApiInfo + for _, api := range apis { + apiList = append(apiList, types.AdminApiInfo{ + Id: api.Id, + ApiName: api.ApiName, + ApiCode: api.ApiCode, + Method: api.Method, + Url: api.Url, + Status: api.Status, + Description: api.Description, + CreateTime: api.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: api.UpdateTime.Format("2006-01-02 15:04:05"), + }) + } + + // 6. 返回结果 + return &types.AdminGetApiListResp{ + Items: apiList, + Total: total, + }, nil +} diff --git a/app/main/api/internal/logic/admin_api/adminupdateapilogic.go b/app/main/api/internal/logic/admin_api/adminupdateapilogic.go new file mode 100644 index 0000000..c811268 --- /dev/null +++ b/app/main/api/internal/logic/admin_api/adminupdateapilogic.go @@ -0,0 +1,92 @@ +package admin_api + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminUpdateApiLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdateApiLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateApiLogic { + return &AdminUpdateApiLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdateApiLogic) AdminUpdateApi(req *types.AdminUpdateApiReq) (resp *types.AdminUpdateApiResp, err error) { + // 1. 参数验证 + if req.Id == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API ID不能为空, id: %s", req.Id) + } + if req.ApiName == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API名称不能为空") + } + if req.ApiCode == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API编码不能为空") + } + if req.Method == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "请求方法不能为空") + } + if req.Url == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API路径不能为空") + } + + // 2. 查询API是否存在 + api, err := l.svcCtx.AdminApiModel.FindOne(l.ctx, req.Id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "API不存在, id: %d", req.Id) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询API失败, err: %v, id: %d", err, req.Id) + } + + // 3. 检查API编码是否被其他记录使用 + if api.ApiCode != req.ApiCode { + existing, err := l.svcCtx.AdminApiModel.FindOneByApiCode(l.ctx, req.ApiCode) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询API失败, err: %v, apiCode: %s", err, req.ApiCode) + } + if existing != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API编码已存在: %s", req.ApiCode) + } + } + + // 4. 更新API信息 + api.ApiName = req.ApiName + api.ApiCode = req.ApiCode + api.Method = req.Method + api.Url = req.Url + api.Status = req.Status + api.Description = req.Description + + _, err = l.svcCtx.AdminApiModel.Update(l.ctx, nil, api) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "更新API失败, err: %v, id: %d", err, req.Id) + } + + // 5. 返回结果 + return &types.AdminUpdateApiResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_auth/adminloginlogic.go b/app/main/api/internal/logic/admin_auth/adminloginlogic.go new file mode 100644 index 0000000..0eb4964 --- /dev/null +++ b/app/main/api/internal/logic/admin_auth/adminloginlogic.go @@ -0,0 +1,96 @@ +package admin_auth + +import ( + "context" + "os" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + jwtx "qnc-server/common/jwt" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminLoginLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminLoginLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminLoginLogic { + return &AdminLoginLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminLoginLogic) AdminLogin(req *types.AdminLoginReq) (resp *types.AdminLoginResp, err error) { + // 1. 验证验证码（开发环境下跳过验证码校验） + if os.Getenv("ENV") != "development" { + if !req.Captcha { + return nil, errors.Wrapf(xerr.NewErrMsg("验证码错误"), "用户登录, 验证码错误, 验证码: %v", req.Captcha) + } + } + + // 2. 验证用户名和密码 + user, err := l.svcCtx.AdminUserModel.FindOneByUsername(l.ctx, req.Username) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrMsg("用户名或密码错误"), "用户登录, 用户名或密码错误, 用户名: %s", req.Username) + } + + // 3. 验证密码 + if !crypto.PasswordVerify(req.Password, user.Password) { + return nil, errors.Wrapf(xerr.NewErrMsg("用户名或密码错误"), "用户登录, 用户名或密码错误, 用户名: %s", req.Username) + } + + // 4. 获取权限 + adminUserRoleBuilder := l.svcCtx.AdminUserRoleModel.SelectBuilder().Where(squirrel.Eq{"user_id": user.Id}) + permissions, err := l.svcCtx.AdminUserRoleModel.FindAll(l.ctx, adminUserRoleBuilder, "role_id DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrMsg("获取权限失败"), "用户登录, 获取权限失败, 用户名: %s", req.Username) + } + + // 获取角色ID数组 + roleIds := make([]string, 0) + for _, permission := range permissions { + roleIds = append(roleIds, permission.RoleId) + } + + // 获取角色名称 + roles := make([]string, 0) + for _, roleId := range roleIds { + role, err := l.svcCtx.AdminRoleModel.FindOne(l.ctx, roleId) + if err != nil { + continue + } + roles = append(roles, role.RoleCode) + } + + // 5. 生成token + refreshToken := l.svcCtx.Config.AdminConfig.RefreshAfter + expiresAt := l.svcCtx.Config.AdminConfig.AccessExpire + claims := jwtx.JwtClaims{ + UserId: user.Id, + AgentId: "", + Platform: model.PlatformAdmin, + UserType: model.UserTypeAdmin, + IsAgent: model.AgentStatusNo, + } + token, err := jwtx.GenerateJwtToken(claims, l.svcCtx.Config.AdminConfig.AccessSecret, expiresAt) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrMsg("生成token失败"), "用户登录, 生成token失败, 用户名: %s", req.Username) + } + + return &types.AdminLoginResp{ + AccessToken: token, + AccessExpire: expiresAt, + RefreshAfter: refreshToken, + Roles: roles, + }, nil +} diff --git a/app/main/api/internal/logic/admin_feature/adminconfigfeatureexamplelogic.go b/app/main/api/internal/logic/admin_feature/adminconfigfeatureexamplelogic.go new file mode 100644 index 0000000..b019a54 --- /dev/null +++ b/app/main/api/internal/logic/admin_feature/adminconfigfeatureexamplelogic.go @@ -0,0 +1,94 @@ +package admin_feature + +import ( + "context" + "encoding/hex" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/google/uuid" +) + +type AdminConfigFeatureExampleLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminConfigFeatureExampleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminConfigFeatureExampleLogic { + return &AdminConfigFeatureExampleLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminConfigFeatureExampleLogic) AdminConfigFeatureExample(req *types.AdminConfigFeatureExampleReq) (resp *types.AdminConfigFeatureExampleResp, err error) { + // 1. 验证功能是否存在 + feature, err := l.svcCtx.FeatureModel.FindOne(l.ctx, req.FeatureId) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询功能失败, featureId: %d, err: %v", req.FeatureId, err) + } + if feature == nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.REUQEST_PARAM_ERROR), + "功能不存在, featureId: %d", req.FeatureId) + } + + // 2. 检查是否已存在示例数据 + existingExample, err := l.svcCtx.ExampleModel.FindOneByFeatureId(l.ctx, req.FeatureId) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询示例数据失败, featureId: %d, err: %v", req.FeatureId, err) + } + + // 3. 加密示例数据 + secretKey := l.svcCtx.Config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), + "获取AES密钥失败: %v", decodeErr) + } + + encryptedData, aesEncryptErr := crypto.AesEncrypt([]byte(req.Data), key) + if aesEncryptErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), + "加密示例数据失败: %v", aesEncryptErr) + } + + // 4. 准备示例数据 + exampleData := &model.Example{ + Id: uuid.NewString(), + ApiId: feature.ApiId, + FeatureId: req.FeatureId, + Content: encryptedData, + } + + // 4. 根据是否存在决定新增或更新 + if existingExample == nil { + // 新增示例数据 + _, err = l.svcCtx.ExampleModel.Insert(l.ctx, nil, exampleData) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "创建示例数据失败, featureId: %d, err: %v", req.FeatureId, err) + } + } else { + // 更新示例数据 + exampleData.Id = existingExample.Id + exampleData.Version = existingExample.Version + _, err = l.svcCtx.ExampleModel.Update(l.ctx, nil, exampleData) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "更新示例数据失败, featureId: %d, err: %v", req.FeatureId, err) + } + } + + // 5. 返回成功结果 + return &types.AdminConfigFeatureExampleResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_feature/admincreatefeaturelogic.go b/app/main/api/internal/logic/admin_feature/admincreatefeaturelogic.go new file mode 100644 index 0000000..3bfbcd7 --- /dev/null +++ b/app/main/api/internal/logic/admin_feature/admincreatefeaturelogic.go @@ -0,0 +1,48 @@ +package admin_feature + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminCreateFeatureLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminCreateFeatureLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminCreateFeatureLogic { + return &AdminCreateFeatureLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminCreateFeatureLogic) AdminCreateFeature(req *types.AdminCreateFeatureReq) (resp *types.AdminCreateFeatureResp, err error) { + // 1. 数据转换 + data := &model.Feature{ + Id: uuid.NewString(), + ApiId: req.ApiId, + Name: req.Name, + } + + // 2. 数据库操作 + result, err := l.svcCtx.FeatureModel.Insert(l.ctx, nil, data) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "创建功能失败, err: %v, req: %+v", err, req) + } + + // 3. 返回结果 + _ = result + return &types.AdminCreateFeatureResp{Id: data.Id}, nil +} diff --git a/app/main/api/internal/logic/admin_feature/admindeletefeaturelogic.go b/app/main/api/internal/logic/admin_feature/admindeletefeaturelogic.go new file mode 100644 index 0000000..fe9fed6 --- /dev/null +++ b/app/main/api/internal/logic/admin_feature/admindeletefeaturelogic.go @@ -0,0 +1,45 @@ +package admin_feature + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminDeleteFeatureLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminDeleteFeatureLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminDeleteFeatureLogic { + return &AdminDeleteFeatureLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminDeleteFeatureLogic) AdminDeleteFeature(req *types.AdminDeleteFeatureReq) (resp *types.AdminDeleteFeatureResp, err error) { + // 1. 查询记录是否存在 + record, err := l.svcCtx.FeatureModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查找功能失败, err: %v, id: %d", err, req.Id) + } + + // 2. 执行软删除 + err = l.svcCtx.FeatureModel.DeleteSoft(l.ctx, nil, record) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "删除功能失败, err: %v, id: %d", err, req.Id) + } + + // 3. 返回结果 + return &types.AdminDeleteFeatureResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_feature/admingetfeaturedetaillogic.go b/app/main/api/internal/logic/admin_feature/admingetfeaturedetaillogic.go new file mode 100644 index 0000000..0b56ade --- /dev/null +++ b/app/main/api/internal/logic/admin_feature/admingetfeaturedetaillogic.go @@ -0,0 +1,46 @@ +package admin_feature + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetFeatureDetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetFeatureDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetFeatureDetailLogic { + return &AdminGetFeatureDetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetFeatureDetailLogic) AdminGetFeatureDetail(req *types.AdminGetFeatureDetailReq) (resp *types.AdminGetFeatureDetailResp, err error) { + // 1. 查询记录 + record, err := l.svcCtx.FeatureModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查找功能失败, err: %v, id: %d", err, req.Id) + } + + // 2. 构建响应 + resp = &types.AdminGetFeatureDetailResp{ + Id: record.Id, + ApiId: record.ApiId, + Name: record.Name, + CreateTime: record.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: record.UpdateTime.Format("2006-01-02 15:04:05"), + } + + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_feature/admingetfeatureexamplelogic.go b/app/main/api/internal/logic/admin_feature/admingetfeatureexamplelogic.go new file mode 100644 index 0000000..6bc9e4e --- /dev/null +++ b/app/main/api/internal/logic/admin_feature/admingetfeatureexamplelogic.go @@ -0,0 +1,82 @@ +package admin_feature + +import ( + "context" + "encoding/hex" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetFeatureExampleLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetFeatureExampleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetFeatureExampleLogic { + return &AdminGetFeatureExampleLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetFeatureExampleLogic) AdminGetFeatureExample(req *types.AdminGetFeatureExampleReq) (resp *types.AdminGetFeatureExampleResp, err error) { + // 1. 查询示例数据 + example, err := l.svcCtx.ExampleModel.FindOneByFeatureId(l.ctx, req.FeatureId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + // 示例数据不存在，返回空数据 + return &types.AdminGetFeatureExampleResp{ + Id: "", + FeatureId: req.FeatureId, + ApiId: "", + Data: "", + CreateTime: "", + UpdateTime: "", + }, nil + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询示例数据失败, featureId: %d, err: %v", req.FeatureId, err) + } + + // 2. 获取解密密钥 + secretKey := l.svcCtx.Config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), + "获取AES密钥失败: %v", decodeErr) + } + + // 3. 解密示例数据 + var decryptedData string + if example.Content == "000" { + // 特殊值，直接返回 + decryptedData = example.Content + } else { + // 解密数据 + decryptedBytes, decryptErr := crypto.AesDecrypt(example.Content, key) + if decryptErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), + "解密示例数据失败: %v", decryptErr) + } + decryptedData = string(decryptedBytes) + } + + // 4. 返回解密后的数据 + return &types.AdminGetFeatureExampleResp{ + Id: example.Id, + FeatureId: example.FeatureId, + ApiId: example.ApiId, + Data: decryptedData, + CreateTime: example.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: example.UpdateTime.Format("2006-01-02 15:04:05"), + }, nil +} diff --git a/app/main/api/internal/logic/admin_feature/admingetfeaturelistlogic.go b/app/main/api/internal/logic/admin_feature/admingetfeaturelistlogic.go new file mode 100644 index 0000000..e23dafc --- /dev/null +++ b/app/main/api/internal/logic/admin_feature/admingetfeaturelistlogic.go @@ -0,0 +1,66 @@ +package admin_feature + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetFeatureListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetFeatureListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetFeatureListLogic { + return &AdminGetFeatureListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetFeatureListLogic) AdminGetFeatureList(req *types.AdminGetFeatureListReq) (resp *types.AdminGetFeatureListResp, err error) { + // 1. 构建查询条件 + builder := l.svcCtx.FeatureModel.SelectBuilder() + + // 2. 添加查询条件 + if req.ApiId != nil && *req.ApiId != "" { + builder = builder.Where("api_id LIKE ?", "%"+*req.ApiId+"%") + } + if req.Name != nil && *req.Name != "" { + builder = builder.Where("name LIKE ?", "%"+*req.Name+"%") + } + + // 3. 执行分页查询 + list, total, err := l.svcCtx.FeatureModel.FindPageListByPageWithTotal( + l.ctx, builder, req.Page, req.PageSize, "id DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询功能列表失败, err: %v, req: %+v", err, req) + } + + // 4. 构建响应列表 + items := make([]types.FeatureListItem, 0, len(list)) + for _, item := range list { + listItem := types.FeatureListItem{ + Id: item.Id, + ApiId: item.ApiId, + Name: item.Name, + CreateTime: item.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: item.UpdateTime.Format("2006-01-02 15:04:05"), + } + items = append(items, listItem) + } + + // 5. 返回结果 + return &types.AdminGetFeatureListResp{ + Total: total, + Items: items, + }, nil +} diff --git a/app/main/api/internal/logic/admin_feature/adminupdatefeaturelogic.go b/app/main/api/internal/logic/admin_feature/adminupdatefeaturelogic.go new file mode 100644 index 0000000..e8ff2e1 --- /dev/null +++ b/app/main/api/internal/logic/admin_feature/adminupdatefeaturelogic.go @@ -0,0 +1,59 @@ +package admin_feature + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminUpdateFeatureLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdateFeatureLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateFeatureLogic { + return &AdminUpdateFeatureLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdateFeatureLogic) AdminUpdateFeature(req *types.AdminUpdateFeatureReq) (resp *types.AdminUpdateFeatureResp, err error) { + // 1. 参数验证 + if req.Id == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "功能ID不能为空, id: %s", req.Id) + } + + // 2. 查询记录是否存在 + record, err := l.svcCtx.FeatureModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查找功能失败, err: %v, id: %d", err, req.Id) + } + + // 3. 直接更新record的字段（只更新非空字段） + if req.ApiId != nil && *req.ApiId != "" { + record.ApiId = *req.ApiId + } + if req.Name != nil && *req.Name != "" { + record.Name = *req.Name + } + + // 4. 执行更新操作 + err = l.svcCtx.FeatureModel.UpdateWithVersion(l.ctx, nil, record) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "更新功能失败, err: %v, id: %d", err, req.Id) + } + + // 5. 返回成功结果 + return &types.AdminUpdateFeatureResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_menu/createmenulogic.go b/app/main/api/internal/logic/admin_menu/createmenulogic.go new file mode 100644 index 0000000..ddd124f --- /dev/null +++ b/app/main/api/internal/logic/admin_menu/createmenulogic.go @@ -0,0 +1,99 @@ +package admin_menu + +import ( + "context" + "database/sql" + "encoding/json" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/google/uuid" +) + +type CreateMenuLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewCreateMenuLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateMenuLogic { + return &CreateMenuLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *CreateMenuLogic) CreateMenu(req *types.CreateMenuReq) (resp *types.CreateMenuResp, err error) { + // 1. 参数验证 + if req.Name == "" { + return nil, errors.Wrapf(xerr.NewErrMsg("菜单名称不能为空"), "菜单名称不能为空") + } + if req.Type == "menu" && req.Component == "" { + return nil, errors.Wrapf(xerr.NewErrMsg("组件路径不能为空"), "组件路径不能为空") + } + + // 2. 检查名称和路径是否重复 + exists, err := l.svcCtx.AdminMenuModel.FindOneByNamePath(l.ctx, req.Name, req.Path) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询菜单失败, err: %v", err) + } + if exists != nil { + return nil, errors.Wrapf(xerr.NewErrMsg("菜单名称或路径已存在"), "菜单名称或路径已存在") + } + + // 3. 检查父菜单是否存在（如果不是根菜单） + if req.Pid != "" { + parentMenu, err := l.svcCtx.AdminMenuModel.FindOne(l.ctx, req.Pid) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询父菜单失败, id: %s, err: %v", req.Pid, err) + } + if parentMenu == nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "父菜单不存在, id: %s", req.Pid) + } + } + + // 4. 将类型标签转换为值 + typeValue, err := l.svcCtx.DictService.GetDictValue(l.ctx, "admin_menu_type", req.Type) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "菜单类型无效: %v", err) + } + + // 5. 创建菜单记录 + menu := &model.AdminMenu{ + Id: uuid.NewString(), + Pid: sql.NullString{String: req.Pid, Valid: req.Pid != ""}, + Name: req.Name, + Path: req.Path, + Component: req.Component, + Redirect: sql.NullString{String: req.Redirect, Valid: req.Redirect != ""}, + Status: req.Status, + Type: typeValue, + Sort: req.Sort, + CreateTime: time.Now(), + UpdateTime: time.Now(), + } + + // 将Meta转换为JSON字符串 + metaJson, err := json.Marshal(req.Meta) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "Meta数据格式错误: %v", err) + } + menu.Meta = string(metaJson) + + // 6. 保存到数据库 + _, err = l.svcCtx.AdminMenuModel.Insert(l.ctx, nil, menu) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建菜单失败, err: %v", err) + } + + return &types.CreateMenuResp{ + Id: menu.Id, + }, nil +} diff --git a/app/main/api/internal/logic/admin_menu/deletemenulogic.go b/app/main/api/internal/logic/admin_menu/deletemenulogic.go new file mode 100644 index 0000000..ee79a46 --- /dev/null +++ b/app/main/api/internal/logic/admin_menu/deletemenulogic.go @@ -0,0 +1,75 @@ +package admin_menu + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type DeleteMenuLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewDeleteMenuLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteMenuLogic { + return &DeleteMenuLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *DeleteMenuLogic) DeleteMenu(req *types.DeleteMenuReq) (resp *types.DeleteMenuResp, err error) { + // 1. 参数验证 + if req.Id == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "菜单ID不能为空, id: %s", req.Id) + } + + // 2. 查询菜单是否存在 + menu, err := l.svcCtx.AdminMenuModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查找菜单失败, err: %v, id: %s", err, req.Id) + } + + // 3. 检查是否有子菜单 + childMenuBuilder := l.svcCtx.AdminMenuModel.SelectBuilder().Where("pid = ?", req.Id) + childMenus, err := l.svcCtx.AdminMenuModel.FindAll(l.ctx, childMenuBuilder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询子菜单失败, err: %v, parent_id: %d", err, req.Id) + } + if len(childMenus) > 0 { + return nil, errors.Wrapf(xerr.NewErrMsg("该菜单下还有子菜单，无法删除"), + "该菜单下还有子菜单，无法删除, id: %d", req.Id) + } + + // 4. 检查是否有角色关联该菜单 + roleMenuBuilder := l.svcCtx.AdminRoleMenuModel.SelectBuilder().Where("menu_id = ?", req.Id) + roleMenus, err := l.svcCtx.AdminRoleMenuModel.FindAll(l.ctx, roleMenuBuilder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询角色菜单关联失败, err: %v, menu_id: %d", err, req.Id) + } + if len(roleMenus) > 0 { + return nil, errors.Wrapf(xerr.NewErrMsg("该菜单已被角色使用，无法删除"), + "该菜单已被角色使用，无法删除, id: %d", req.Id) + } + + // 5. 执行软删除 + err = l.svcCtx.AdminMenuModel.DeleteSoft(l.ctx, nil, menu) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "删除菜单失败, err: %v, id: %d", err, req.Id) + } + + // 6. 返回成功结果 + return &types.DeleteMenuResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_menu/getmenualllogic.go b/app/main/api/internal/logic/admin_menu/getmenualllogic.go new file mode 100644 index 0000000..b5abf62 --- /dev/null +++ b/app/main/api/internal/logic/admin_menu/getmenualllogic.go @@ -0,0 +1,251 @@ +package admin_menu + +import ( + "context" + "database/sql" + "sort" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + + "github.com/Masterminds/squirrel" + "github.com/bytedance/sonic" + "github.com/pkg/errors" + "github.com/samber/lo" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/mr" +) + +type GetMenuAllLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetMenuAllLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetMenuAllLogic { + return &GetMenuAllLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetMenuAllLogic) GetMenuAll(req *types.GetMenuAllReq) (resp *[]types.GetMenuAllResp, err error) { + userId, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户ID失败, %+v", err) + } + + // 使用MapReduceVoid并发获取用户角色（UUID 字符串） + var roleIds []string + var permissions []*struct { + RoleId string + } + + type UserRoleResult struct { + RoleId string + } + + err = mr.MapReduceVoid( + func(source chan<- interface{}) { + adminUserRoleBuilder := l.svcCtx.AdminUserRoleModel.SelectBuilder().Where(squirrel.Eq{"user_id": userId}) + source <- adminUserRoleBuilder + }, + func(item interface{}, writer mr.Writer[*UserRoleResult], cancel func(error)) { + builder := item.(squirrel.SelectBuilder) + result, err := l.svcCtx.AdminUserRoleModel.FindAll(l.ctx, builder, "role_id DESC") + if err != nil { + cancel(errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户角色信息失败, %+v", err)) + return + } + + for _, r := range result { + writer.Write(&UserRoleResult{RoleId: r.RoleId}) + } + }, + func(pipe <-chan *UserRoleResult, cancel func(error)) { + for item := range pipe { + permissions = append(permissions, &struct{ RoleId string }{RoleId: item.RoleId}) + } + }, + ) + if err != nil { + return nil, err + } + + for _, permission := range permissions { + roleIds = append(roleIds, permission.RoleId) + } + + // 使用MapReduceVoid并发获取角色菜单（UUID 字符串） + var menuIds []string + var roleMenus []*struct { + MenuId string + } + + type RoleMenuResult struct { + MenuId string + } + + err = mr.MapReduceVoid( + func(source chan<- interface{}) { + getRoleMenuBuilder := l.svcCtx.AdminRoleMenuModel.SelectBuilder().Where(squirrel.Eq{"role_id": roleIds}) + source <- getRoleMenuBuilder + }, + func(item interface{}, writer mr.Writer[*RoleMenuResult], cancel func(error)) { + builder := item.(squirrel.SelectBuilder) + result, err := l.svcCtx.AdminRoleMenuModel.FindAll(l.ctx, builder, "id DESC") + if err != nil { + cancel(errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取角色菜单信息失败, %+v", err)) + return + } + + for _, r := range result { + writer.Write(&RoleMenuResult{MenuId: r.MenuId}) + } + }, + func(pipe <-chan *RoleMenuResult, cancel func(error)) { + for item := range pipe { + roleMenus = append(roleMenus, &struct{ MenuId string }{MenuId: item.MenuId}) + } + }, + ) + if err != nil { + return nil, err + } + + for _, roleMenu := range roleMenus { + menuIds = append(menuIds, roleMenu.MenuId) + } + + // 使用MapReduceVoid并发获取菜单 + type AdminMenuStruct struct { + Id string + Pid sql.NullString + Name string + Path string + Component string + Redirect struct { + String string + Valid bool + } + Meta string + Sort int64 + Type int64 + Status int64 + } + + var menus []*AdminMenuStruct + + err = mr.MapReduceVoid( + func(source chan<- interface{}) { + adminMenuBuilder := l.svcCtx.AdminMenuModel.SelectBuilder().Where(squirrel.Eq{"id": menuIds}) + source <- adminMenuBuilder + }, + func(item interface{}, writer mr.Writer[*AdminMenuStruct], cancel func(error)) { + builder := item.(squirrel.SelectBuilder) + result, err := l.svcCtx.AdminMenuModel.FindAll(l.ctx, builder, "sort ASC") + if err != nil { + cancel(errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取菜单信息失败, %+v", err)) + return + } + + for _, r := range result { + menu := &AdminMenuStruct{ + Id: r.Id, + Pid: r.Pid, + Name: r.Name, + Path: r.Path, + Component: r.Component, + Redirect: r.Redirect, + Meta: r.Meta, + Sort: r.Sort, + Type: r.Type, + Status: r.Status, + } + writer.Write(menu) + } + }, + func(pipe <-chan *AdminMenuStruct, cancel func(error)) { + for item := range pipe { + menus = append(menus, item) + } + }, + ) + if err != nil { + return nil, err + } + + // 转换为types.Menu结构并存储到映射表 + menuMap := make(map[string]types.GetMenuAllResp) + for _, menu := range menus { + // 只处理状态正常的菜单 + if menu.Status != 1 { + continue + } + + meta := make(map[string]interface{}) + err = sonic.Unmarshal([]byte(menu.Meta), &meta) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "解析菜单Meta信息失败, %+v", err) + } + + redirect := func() string { + if menu.Redirect.Valid { + return menu.Redirect.String + } + return "" + }() + + menuMap[menu.Id] = types.GetMenuAllResp{ + Name: menu.Name, + Path: menu.Path, + Redirect: redirect, + Component: menu.Component, + Sort: menu.Sort, + Meta: meta, + Children: make([]types.GetMenuAllResp, 0), + } + } + + // 按ParentId将菜单分组（字符串键） + menuGroups := lo.GroupBy(menus, func(item *AdminMenuStruct) string { + if item.Pid.Valid { + return item.Pid.String + } + return "0" + }) + + // 递归构建菜单树（字符串键） + var buildMenuTree func(parentId string) []types.GetMenuAllResp + buildMenuTree = func(parentId string) []types.GetMenuAllResp { + children := make([]types.GetMenuAllResp, 0) + + childMenus, ok := menuGroups[parentId] + if !ok { + return children + } + + // 按Sort排序 + sort.Slice(childMenus, func(i, j int) bool { + return childMenus[i].Sort < childMenus[j].Sort + }) + + for _, childMenu := range childMenus { + if menu, exists := menuMap[childMenu.Id]; exists && childMenu.Status == 1 { + // 递归构建子菜单 + menu.Children = buildMenuTree(childMenu.Id) + children = append(children, menu) + } + } + + return children + } + + // 从根菜单开始构建（ParentId为"0"的是根菜单） + menuTree := buildMenuTree("0") + + return &menuTree, nil +} diff --git a/app/main/api/internal/logic/admin_menu/getmenudetaillogic.go b/app/main/api/internal/logic/admin_menu/getmenudetaillogic.go new file mode 100644 index 0000000..58861bc --- /dev/null +++ b/app/main/api/internal/logic/admin_menu/getmenudetaillogic.go @@ -0,0 +1,30 @@ +package admin_menu + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetMenuDetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetMenuDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetMenuDetailLogic { + return &GetMenuDetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetMenuDetailLogic) GetMenuDetail(req *types.GetMenuDetailReq) (resp *types.GetMenuDetailResp, err error) { + // todo: add your logic here and delete this line + + return +} diff --git a/app/main/api/internal/logic/admin_menu/getmenulistlogic.go b/app/main/api/internal/logic/admin_menu/getmenulistlogic.go new file mode 100644 index 0000000..7b56224 --- /dev/null +++ b/app/main/api/internal/logic/admin_menu/getmenulistlogic.go @@ -0,0 +1,109 @@ +package admin_menu + +import ( + "context" + "encoding/json" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type GetMenuListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetMenuListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetMenuListLogic { + return &GetMenuListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetMenuListLogic) GetMenuList(req *types.GetMenuListReq) (resp []types.MenuListItem, err error) { + // 构建查询条件 + builder := l.svcCtx.AdminMenuModel.SelectBuilder() + + // 添加筛选条件 + if len(req.Name) > 0 { + builder = builder.Where("name LIKE ?", "%"+req.Name+"%") + } + if len(req.Path) > 0 { + builder = builder.Where("path LIKE ?", "%"+req.Path+"%") + } + if req.Status != -1 { + builder = builder.Where("status = ?", req.Status) + } + if req.Type != "" { + builder = builder.Where("type = ?", req.Type) + } + + // 排序但不分页，获取所有符合条件的菜单 + builder = builder.OrderBy("sort ASC") + + // 获取所有菜单 + menus, err := l.svcCtx.AdminMenuModel.FindAll(l.ctx, builder, "id ASC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询菜单失败, err: %v", err) + } + + // 将菜单按ID存入map + menuMap := make(map[string]types.MenuListItem) + for _, menu := range menus { + var meta map[string]interface{} + err := json.Unmarshal([]byte(menu.Meta), &meta) + if err != nil { + logx.Errorf("解析Meta字段失败: %v", err) + meta = make(map[string]interface{}) + } + menuType, err := l.svcCtx.DictService.GetDictLabel(l.ctx, "admin_menu_type", menu.Type) + if err != nil { + logx.Errorf("获取菜单类型失败: %v", err) + menuType = "" + } + item := types.MenuListItem{ + Id: menu.Id, + Pid: menu.Pid.String, + Name: menu.Name, + Path: menu.Path, + Component: menu.Component, + Redirect: menu.Redirect.String, + Meta: meta, + Status: menu.Status, + Type: menuType, + Sort: menu.Sort, + CreateTime: menu.CreateTime.Format("2006-01-02 15:04:05"), + Children: make([]types.MenuListItem, 0), + } + menuMap[menu.Id] = item + } + + // 构建父子关系 + for _, menu := range menus { + if menu.Pid.Valid && menu.Pid.String != "0" { + // 找到父菜单 + if parent, exists := menuMap[menu.Pid.String]; exists { + // 添加当前菜单到父菜单的子菜单列表 + children := append(parent.Children, menuMap[menu.Id]) + parent.Children = children + menuMap[menu.Pid.String] = parent + } + } + } + + // 提取顶级菜单（ParentId为0）到响应列表 + result := make([]types.MenuListItem, 0) + for _, menu := range menus { + if menu.Pid.Valid && menu.Pid.String == "0" { + result = append(result, menuMap[menu.Id]) + } + } + + return result, nil +} diff --git a/app/main/api/internal/logic/admin_menu/updatemenulogic.go b/app/main/api/internal/logic/admin_menu/updatemenulogic.go new file mode 100644 index 0000000..5981c62 --- /dev/null +++ b/app/main/api/internal/logic/admin_menu/updatemenulogic.go @@ -0,0 +1,96 @@ +package admin_menu + +import ( + "context" + "database/sql" + "encoding/json" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type UpdateMenuLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewUpdateMenuLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateMenuLogic { + return &UpdateMenuLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *UpdateMenuLogic) UpdateMenu(req *types.UpdateMenuReq) (resp *types.UpdateMenuResp, err error) { + // 1. 检查菜单是否存在 + menu, err := l.svcCtx.AdminMenuModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询菜单失败, id: %d, err: %v", req.Id, err) + } + if menu == nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "菜单不存在, id: %d", req.Id) + } + + // 2. 将类型标签转换为值 + typeValue, err := l.svcCtx.DictService.GetDictValue(l.ctx, "admin_menu_type", req.Type) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "菜单类型无效: %v", err) + } + + // 3. 检查父菜单是否存在（如果不是根菜单） + if req.Pid != nil && *req.Pid != "0" { + parentMenu, err := l.svcCtx.AdminMenuModel.FindOne(l.ctx, *req.Pid) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询父菜单失败, id: %s, err: %v", *req.Pid, err) + } + if parentMenu == nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "父菜单不存在, id: %s", *req.Pid) + } + } + + // 4. 检查名称和路径是否重复 + if req.Name != menu.Name || req.Path != menu.Path { + exists, err := l.svcCtx.AdminMenuModel.FindOneByNamePath(l.ctx, req.Name, req.Path) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询菜单失败, err: %v", err) + } + if exists != nil && exists.Id != req.Id { + return nil, errors.Wrapf(xerr.NewErrMsg("菜单名称或路径已存在"), "菜单名称或路径已存在") + } + } + + // 5. 更新菜单信息 + + menu.Pid = sql.NullString{String: *req.Pid, Valid: req.Pid != nil} + menu.Name = req.Name + menu.Path = req.Path + menu.Component = req.Component + menu.Redirect = sql.NullString{String: req.Redirect, Valid: req.Redirect != ""} + menu.Status = req.Status + menu.Type = typeValue + menu.Sort = req.Sort + + // 将Meta转换为JSON字符串 + metaJson, err := json.Marshal(req.Meta) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "Meta数据格式错误: %v", err) + } + menu.Meta = string(metaJson) + + // 6. 保存更新 + _, err = l.svcCtx.AdminMenuModel.Update(l.ctx, nil, menu) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新菜单失败, err: %v", err) + } + + return &types.UpdateMenuResp{ + Success: true, + }, nil +} diff --git a/app/main/api/internal/logic/admin_notification/admincreatenotificationlogic.go b/app/main/api/internal/logic/admin_notification/admincreatenotificationlogic.go new file mode 100644 index 0000000..0b34365 --- /dev/null +++ b/app/main/api/internal/logic/admin_notification/admincreatenotificationlogic.go @@ -0,0 +1,52 @@ +package admin_notification + +import ( + "context" + "database/sql" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminCreateNotificationLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminCreateNotificationLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminCreateNotificationLogic { + return &AdminCreateNotificationLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminCreateNotificationLogic) AdminCreateNotification(req *types.AdminCreateNotificationReq) (resp *types.AdminCreateNotificationResp, err error) { + startDate, _ := time.Parse("2006-01-02", req.StartDate) + endDate, _ := time.Parse("2006-01-02", req.EndDate) + data := &model.GlobalNotifications{ + Id: uuid.NewString(), + Title: req.Title, + Content: req.Content, + NotificationPage: req.NotificationPage, + StartDate: sql.NullTime{Time: startDate, Valid: req.StartDate != ""}, + EndDate: sql.NullTime{Time: endDate, Valid: req.EndDate != ""}, + StartTime: req.StartTime, + EndTime: req.EndTime, + Status: req.Status, + } + result, err := l.svcCtx.GlobalNotificationsModel.Insert(l.ctx, nil, data) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建通知失败, err: %v, req: %+v", err, req) + } + _ = result + return &types.AdminCreateNotificationResp{Id: data.Id}, nil +} diff --git a/app/main/api/internal/logic/admin_notification/admindeletenotificationlogic.go b/app/main/api/internal/logic/admin_notification/admindeletenotificationlogic.go new file mode 100644 index 0000000..9a540fe --- /dev/null +++ b/app/main/api/internal/logic/admin_notification/admindeletenotificationlogic.go @@ -0,0 +1,38 @@ +package admin_notification + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminDeleteNotificationLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminDeleteNotificationLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminDeleteNotificationLogic { + return &AdminDeleteNotificationLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminDeleteNotificationLogic) AdminDeleteNotification(req *types.AdminDeleteNotificationReq) (resp *types.AdminDeleteNotificationResp, err error) { + notification, err := l.svcCtx.GlobalNotificationsModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找通知失败, err: %v, id: %d", err, req.Id) + } + err = l.svcCtx.GlobalNotificationsModel.DeleteSoft(l.ctx, nil, notification) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除通知失败, err: %v, id: %d", err, req.Id) + } + return &types.AdminDeleteNotificationResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_notification/admingetnotificationdetaillogic.go b/app/main/api/internal/logic/admin_notification/admingetnotificationdetaillogic.go new file mode 100644 index 0000000..3ccdd01 --- /dev/null +++ b/app/main/api/internal/logic/admin_notification/admingetnotificationdetaillogic.go @@ -0,0 +1,53 @@ +package admin_notification + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetNotificationDetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetNotificationDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetNotificationDetailLogic { + return &AdminGetNotificationDetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetNotificationDetailLogic) AdminGetNotificationDetail(req *types.AdminGetNotificationDetailReq) (resp *types.AdminGetNotificationDetailResp, err error) { + notification, err := l.svcCtx.GlobalNotificationsModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找通知失败, err: %v, id: %d", err, req.Id) + } + resp = &types.AdminGetNotificationDetailResp{ + Id: notification.Id, + Title: notification.Title, + Content: notification.Content, + NotificationPage: notification.NotificationPage, + StartDate: "", + StartTime: notification.StartTime, + EndDate: "", + EndTime: notification.EndTime, + Status: notification.Status, + CreateTime: notification.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: notification.UpdateTime.Format("2006-01-02 15:04:05"), + } + if notification.StartDate.Valid { + resp.StartDate = notification.StartDate.Time.Format("2006-01-02") + } + if notification.EndDate.Valid { + resp.EndDate = notification.EndDate.Time.Format("2006-01-02") + } + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_notification/admingetnotificationlistlogic.go b/app/main/api/internal/logic/admin_notification/admingetnotificationlistlogic.go new file mode 100644 index 0000000..e12017d --- /dev/null +++ b/app/main/api/internal/logic/admin_notification/admingetnotificationlistlogic.go @@ -0,0 +1,82 @@ +package admin_notification + +import ( + "context" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetNotificationListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetNotificationListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetNotificationListLogic { + return &AdminGetNotificationListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetNotificationListLogic) AdminGetNotificationList(req *types.AdminGetNotificationListReq) (resp *types.AdminGetNotificationListResp, err error) { + builder := l.svcCtx.GlobalNotificationsModel.SelectBuilder() + if req.Title != nil { + builder = builder.Where("title LIKE ?", "%"+*req.Title+"%") + } + if req.NotificationPage != nil { + builder = builder.Where("notification_page = ?", *req.NotificationPage) + } + if req.Status != nil { + builder = builder.Where("status = ?", *req.Status) + } + if req.StartDate != nil { + if t, err := time.Parse("2006-01-02", *req.StartDate); err == nil { + builder = builder.Where("start_date >= ?", t) + } + } + if req.EndDate != nil { + if t, err := time.Parse("2006-01-02", *req.EndDate); err == nil { + builder = builder.Where("end_date <= ?", t) + } + } + list, total, err := l.svcCtx.GlobalNotificationsModel.FindPageListByPageWithTotal(l.ctx, builder, req.Page, req.PageSize, "id DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询通知列表失败, err: %v, req: %+v", err, req) + } + items := make([]types.NotificationListItem, 0, len(list)) + for _, n := range list { + item := types.NotificationListItem{ + Id: n.Id, + Title: n.Title, + NotificationPage: n.NotificationPage, + Content: n.Content, + StartDate: "", + StartTime: n.StartTime, + EndDate: "", + EndTime: n.EndTime, + Status: n.Status, + CreateTime: n.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: n.UpdateTime.Format("2006-01-02 15:04:05"), + } + if n.StartDate.Valid { + item.StartDate = n.StartDate.Time.Format("2006-01-02") + } + if n.EndDate.Valid { + item.EndDate = n.EndDate.Time.Format("2006-01-02") + } + items = append(items, item) + } + resp = &types.AdminGetNotificationListResp{ + Total: total, + Items: items, + } + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_notification/adminupdatenotificationlogic.go b/app/main/api/internal/logic/admin_notification/adminupdatenotificationlogic.go new file mode 100644 index 0000000..838242a --- /dev/null +++ b/app/main/api/internal/logic/admin_notification/adminupdatenotificationlogic.go @@ -0,0 +1,66 @@ +package admin_notification + +import ( + "context" + "database/sql" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminUpdateNotificationLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdateNotificationLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateNotificationLogic { + return &AdminUpdateNotificationLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdateNotificationLogic) AdminUpdateNotification(req *types.AdminUpdateNotificationReq) (resp *types.AdminUpdateNotificationResp, err error) { + notification, err := l.svcCtx.GlobalNotificationsModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找通知失败, err: %v, id: %d", err, req.Id) + } + if req.StartDate != nil { + startDate, _ := time.Parse("2006-01-02", *req.StartDate) + notification.StartDate = sql.NullTime{Time: startDate, Valid: true} + } + if req.EndDate != nil { + endDate, _ := time.Parse("2006-01-02", *req.EndDate) + notification.EndDate = sql.NullTime{Time: endDate, Valid: true} + } + if req.Title != nil { + notification.Title = *req.Title + } + if req.Content != nil { + notification.Content = *req.Content + } + if req.NotificationPage != nil { + notification.NotificationPage = *req.NotificationPage + } + if req.StartTime != nil { + notification.StartTime = *req.StartTime + } + if req.EndTime != nil { + notification.EndTime = *req.EndTime + } + if req.Status != nil { + notification.Status = *req.Status + } + _, err = l.svcCtx.GlobalNotificationsModel.Update(l.ctx, nil, notification) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新通知失败, err: %v, req: %+v", err, req) + } + return &types.AdminUpdateNotificationResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_order/admincreateorderlogic.go b/app/main/api/internal/logic/admin_order/admincreateorderlogic.go new file mode 100644 index 0000000..7d52520 --- /dev/null +++ b/app/main/api/internal/logic/admin_order/admincreateorderlogic.go @@ -0,0 +1,82 @@ +package admin_order + +import ( + "context" + "database/sql" + "fmt" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type AdminCreateOrderLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminCreateOrderLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminCreateOrderLogic { + return &AdminCreateOrderLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminCreateOrderLogic) AdminCreateOrder(req *types.AdminCreateOrderReq) (resp *types.AdminCreateOrderResp, err error) { + // 生成订单号 + orderNo := fmt.Sprintf("%dADMIN", time.Now().UnixNano()) + + // 根据产品名称查询产品ID + builder := l.svcCtx.ProductModel.SelectBuilder() + builder = builder.Where("product_name = ? AND del_state = ?", req.ProductName, 0) + products, err := l.svcCtx.ProductModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminCreateOrder, 查询产品失败 err: %v", err) + } + if len(products) == 0 { + return nil, errors.Wrapf(xerr.NewErrMsg(fmt.Sprintf("产品不存在: %s", req.ProductName)), "AdminCreateOrder, 查询产品失败 err: %v", err) + } + product := products[0] + + // 创建订单对象 + order := &model.Order{ + Id: uuid.NewString(), + OrderNo: orderNo, + PlatformOrderId: sql.NullString{String: req.PlatformOrderId, Valid: req.PlatformOrderId != ""}, + ProductId: product.Id, + PaymentPlatform: req.PaymentPlatform, + PaymentScene: req.PaymentScene, + Amount: req.Amount, + Status: req.Status, + } + + // 使用事务处理订单创建 + var orderId string + err = l.svcCtx.OrderModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 插入订单 + _, err := l.svcCtx.OrderModel.Insert(ctx, session, order) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminCreateOrder, 创建订单失败 err: %v", err) + } + orderId = order.Id + + return nil + }) + + if err != nil { + return nil, err + } + + return &types.AdminCreateOrderResp{ + Id: orderId, + }, nil +} diff --git a/app/main/api/internal/logic/admin_order/admindeleteorderlogic.go b/app/main/api/internal/logic/admin_order/admindeleteorderlogic.go new file mode 100644 index 0000000..42a8940 --- /dev/null +++ b/app/main/api/internal/logic/admin_order/admindeleteorderlogic.go @@ -0,0 +1,54 @@ +package admin_order + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type AdminDeleteOrderLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminDeleteOrderLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminDeleteOrderLogic { + return &AdminDeleteOrderLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminDeleteOrderLogic) AdminDeleteOrder(req *types.AdminDeleteOrderReq) (resp *types.AdminDeleteOrderResp, err error) { + // 获取订单信息 + order, err := l.svcCtx.OrderModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminDeleteOrder, 查询订单失败 err: %v", err) + } + + // 使用事务删除订单 + err = l.svcCtx.OrderModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 软删除订单 + err := l.svcCtx.OrderModel.DeleteSoft(ctx, session, order) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminDeleteOrder, 删除订单失败 err: %v", err) + } + + return nil + }) + + if err != nil { + return nil, err + } + + return &types.AdminDeleteOrderResp{ + Success: true, + }, nil +} diff --git a/app/main/api/internal/logic/admin_order/admingetorderdetaillogic.go b/app/main/api/internal/logic/admin_order/admingetorderdetaillogic.go new file mode 100644 index 0000000..6a02d49 --- /dev/null +++ b/app/main/api/internal/logic/admin_order/admingetorderdetaillogic.go @@ -0,0 +1,128 @@ +package admin_order + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetOrderDetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetOrderDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetOrderDetailLogic { + return &AdminGetOrderDetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetOrderDetailLogic) AdminGetOrderDetail(req *types.AdminGetOrderDetailReq) (resp *types.AdminGetOrderDetailResp, err error) { + // 获取订单信息 + order, err := l.svcCtx.OrderModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderDetail, 查询订单失败 err: %v", err) + } + + // 获取产品信息 + product, err := l.svcCtx.ProductModel.FindOne(l.ctx, order.ProductId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderDetail, 查询产品失败 err: %v", err) + } + + // 判断是否为代理订单并获取代理处理状态 + var isAgentOrder bool + var agentProcessStatus string + + agentOrder, err := l.svcCtx.AgentOrderModel.FindOneByOrderId(l.ctx, order.Id) + if err == nil && agentOrder != nil { + isAgentOrder = true + + // 查询代理佣金记录 + commissions, err := l.svcCtx.AgentCommissionModel.FindAll(l.ctx, + l.svcCtx.AgentCommissionModel.SelectBuilder().Where("order_id = ?", order.Id), "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderDetail, 查询代理佣金失败 err: %v", err) + } + + if len(commissions) > 0 { + agentProcessStatus = "success" + } else { + // 检查订单状态，如果是已支付但无佣金记录，则为待处理或失败 + if order.Status == "paid" { + agentProcessStatus = "pending" + } else { + agentProcessStatus = "failed" + } + } + } else { + isAgentOrder = false + agentProcessStatus = "not_agent" + } + + // 获取查询状态 + var queryState string + builder := l.svcCtx.QueryModel.SelectBuilder().Where("order_id = ?", order.Id).Columns("query_state") + queries, err := l.svcCtx.QueryModel.FindAll(l.ctx, builder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderDetail, 查询查询状态失败 err: %v", err) + } + + if len(queries) > 0 { + queryState = queries[0].QueryState + } else { + // 查询清理日志 + cleanupBuilder := l.svcCtx.QueryCleanupDetailModel.SelectBuilder(). + Where("order_id = ?", order.Id). + Where("del_state = ?", globalkey.DelStateNo). + OrderBy("create_time DESC"). + Limit(1) + cleanupDetails, err := l.svcCtx.QueryCleanupDetailModel.FindAll(l.ctx, cleanupBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderDetail, 查询清理日志失败 err: %v", err) + } + + if len(cleanupDetails) > 0 { + queryState = model.QueryStateCleaned + } else { + queryState = "" + } + } + + // 构建响应 + resp = &types.AdminGetOrderDetailResp{ + Id: order.Id, + OrderNo: order.OrderNo, + PlatformOrderId: order.PlatformOrderId.String, + ProductName: product.ProductName, + PaymentPlatform: order.PaymentPlatform, + PaymentScene: order.PaymentScene, + Amount: order.Amount, + Status: order.Status, + CreateTime: order.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: order.UpdateTime.Format("2006-01-02 15:04:05"), + QueryState: queryState, + IsAgentOrder: isAgentOrder, + AgentProcessStatus: agentProcessStatus, + } + + // 处理可选字段 + if order.PayTime.Valid { + resp.PayTime = order.PayTime.Time.Format("2006-01-02 15:04:05") + } + if order.RefundTime.Valid { + resp.RefundTime = order.RefundTime.Time.Format("2006-01-02 15:04:05") + } + + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_order/admingetorderlistlogic.go b/app/main/api/internal/logic/admin_order/admingetorderlistlogic.go new file mode 100644 index 0000000..4b10409 --- /dev/null +++ b/app/main/api/internal/logic/admin_order/admingetorderlistlogic.go @@ -0,0 +1,287 @@ +package admin_order + +import ( + "context" + "sync" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/mr" +) + +type AdminGetOrderListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetOrderListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetOrderListLogic { + return &AdminGetOrderListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetOrderListLogic) AdminGetOrderList(req *types.AdminGetOrderListReq) (resp *types.AdminGetOrderListResp, err error) { + // 构建查询条件 + builder := l.svcCtx.OrderModel.SelectBuilder() + if req.OrderNo != "" { + builder = builder.Where("order_no = ?", req.OrderNo) + } + if req.PlatformOrderId != "" { + builder = builder.Where("platform_order_id = ?", req.PlatformOrderId) + } + if req.ProductName != "" { + builder = builder.Where("product_id IN (SELECT id FROM product WHERE product_name LIKE ?)", "%"+req.ProductName+"%") + } + if req.PaymentPlatform != "" { + builder = builder.Where("payment_platform = ?", req.PaymentPlatform) + } + if req.PaymentScene != "" { + builder = builder.Where("payment_scene = ?", req.PaymentScene) + } + if req.Amount > 0 { + builder = builder.Where("amount = ?", req.Amount) + } + if req.Status != "" { + builder = builder.Where("status = ?", req.Status) + } + // 时间范围查询 + if req.CreateTimeStart != "" { + builder = builder.Where("create_time >= ?", req.CreateTimeStart) + } + if req.CreateTimeEnd != "" { + builder = builder.Where("create_time <= ?", req.CreateTimeEnd) + } + if req.PayTimeStart != "" { + builder = builder.Where("pay_time >= ?", req.PayTimeStart) + } + if req.PayTimeEnd != "" { + builder = builder.Where("pay_time <= ?", req.PayTimeEnd) + } + if req.RefundTimeStart != "" { + builder = builder.Where("refund_time >= ?", req.RefundTimeStart) + } + if req.RefundTimeEnd != "" { + builder = builder.Where("refund_time <= ?", req.RefundTimeEnd) + } + + // 并发获取总数和列表 + var total int64 + var orders []*model.Order + err = mr.Finish(func() error { + var err error + total, err = l.svcCtx.OrderModel.FindCount(l.ctx, builder, "id") + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderList, 查询订单总数失败 err: %v", err) + } + return nil + }, func() error { + var err error + orders, err = l.svcCtx.OrderModel.FindPageListByPage(l.ctx, builder, req.Page, req.PageSize, "id DESC") + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderList, 查询订单列表失败 err: %v", err) + } + return nil + }) + if err != nil { + return nil, err + } + + // 并发获取产品信息和查询状态 + productMap := make(map[string]string) + queryStateMap := make(map[string]string) + agentOrderMap := make(map[string]bool) // 代理订单映射 + agentProcessStatusMap := make(map[string]string) // 代理处理状态映射 + + var mu sync.Mutex + + // 批量获取查询状态 + if len(orders) > 0 { + orderIds := make([]string, 0, len(orders)) + for _, order := range orders { + orderIds = append(orderIds, order.Id) + } + + // 1. 先查询当前查询状态 + builder := l.svcCtx.QueryModel.SelectBuilder(). + Where(squirrel.Eq{"order_id": orderIds}). + Columns("order_id", "query_state") + queries, err := l.svcCtx.QueryModel.FindAll(l.ctx, builder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderList, 批量查询查询状态失败 err: %v", err) + } + + // 2. 记录已找到查询状态的订单ID + foundOrderIds := make(map[string]bool) + for _, query := range queries { + queryStateMap[query.OrderId] = query.QueryState + foundOrderIds[query.OrderId] = true + } + + // 3. 查找未找到查询状态的订单是否在清理日志中 + notFoundOrderIds := make([]string, 0) + for _, orderId := range orderIds { + if !foundOrderIds[orderId] { + notFoundOrderIds = append(notFoundOrderIds, orderId) + } + } + + if len(notFoundOrderIds) > 0 { + // 查询清理日志 + cleanupBuilder := l.svcCtx.QueryCleanupDetailModel.SelectBuilder(). + Where(squirrel.Eq{"order_id": notFoundOrderIds}). + Where("del_state = ?", globalkey.DelStateNo). + OrderBy("create_time DESC") + cleanupDetails, err := l.svcCtx.QueryCleanupDetailModel.FindAll(l.ctx, cleanupBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderList, 查询清理日志失败 err: %v", err) + } + + // 记录已清理的订单状态 + for _, detail := range cleanupDetails { + if _, exists := queryStateMap[detail.OrderId]; !exists { + queryStateMap[detail.OrderId] = model.QueryStateCleaned // 使用常量标记为已清除状态 + } + } + + // 对于既没有查询状态也没有清理记录的订单，不设置状态（保持为空字符串） + for _, orderId := range notFoundOrderIds { + if _, exists := queryStateMap[orderId]; !exists { + queryStateMap[orderId] = "" // 未知状态保持为空字符串 + } + } + } + + // 批量获取代理订单状态 + agentOrders, err := l.svcCtx.AgentOrderModel.FindAll(l.ctx, + l.svcCtx.AgentOrderModel.SelectBuilder().Where(squirrel.Eq{"order_id": orderIds}), "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderList, 批量查询代理订单失败 err: %v", err) + } + + // 记录代理订单 + for _, agentOrder := range agentOrders { + agentOrderMap[agentOrder.OrderId] = true + } + + // 对于代理订单，查询代理处理状态 + if len(agentOrders) > 0 { + agentOrderIds := make([]string, 0, len(agentOrders)) + for _, agentOrder := range agentOrders { + agentOrderIds = append(agentOrderIds, agentOrder.OrderId) + } + + // 查询代理佣金记录 + commissions, err := l.svcCtx.AgentCommissionModel.FindAll(l.ctx, + l.svcCtx.AgentCommissionModel.SelectBuilder().Where(squirrel.Eq{"order_id": agentOrderIds}), "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderList, 批量查询代理佣金失败 err: %v", err) + } + + // 记录有佣金记录的订单为处理成功 + processedOrderIds := make(map[string]bool) + for _, commission := range commissions { + processedOrderIds[commission.OrderId] = true + } + + // 创建订单状态映射，避免重复查找 + orderStatusMap := make(map[string]string) + for _, order := range orders { + orderStatusMap[order.Id] = order.Status + } + + // 设置代理处理状态 + for _, agentOrder := range agentOrders { + orderId := agentOrder.OrderId + if processedOrderIds[orderId] { + agentProcessStatusMap[orderId] = "success" + } else { + // 检查订单状态，如果是已支付但无佣金记录，则为待处理或失败 + if orderStatusMap[orderId] == "paid" { + agentProcessStatusMap[orderId] = "pending" + } else { + agentProcessStatusMap[orderId] = "failed" + } + } + } + } + } + + // 并发获取产品信息 + err = mr.MapReduceVoid(func(source chan<- interface{}) { + for _, order := range orders { + source <- order + } + }, func(item interface{}, writer mr.Writer[struct{}], cancel func(error)) { + order := item.(*model.Order) + + // 获取产品信息 + product, err := l.svcCtx.ProductModel.FindOne(l.ctx, order.ProductId) + if err != nil && !errors.Is(err, model.ErrNotFound) { + cancel(errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminGetOrderList, 查询产品信息失败 err: %v", err)) + return + } + mu.Lock() + if product != nil { + productMap[product.Id] = product.ProductName + } else { + productMap[order.ProductId] = "" // 产品不存在时设置为空字符串 + } + mu.Unlock() + writer.Write(struct{}{}) + }, func(pipe <-chan struct{}, cancel func(error)) { + for range pipe { + } + }) + if err != nil { + return nil, err + } + + // 构建响应 + resp = &types.AdminGetOrderListResp{ + Total: total, + Items: make([]types.OrderListItem, 0, len(orders)), + } + + for _, order := range orders { + item := types.OrderListItem{ + Id: order.Id, + OrderNo: order.OrderNo, + PlatformOrderId: order.PlatformOrderId.String, + ProductName: productMap[order.ProductId], + PaymentPlatform: order.PaymentPlatform, + PaymentScene: order.PaymentScene, + Amount: order.Amount, + Status: order.Status, + CreateTime: order.CreateTime.Format("2006-01-02 15:04:05"), + QueryState: queryStateMap[order.Id], + } + if order.PayTime.Valid { + item.PayTime = order.PayTime.Time.Format("2006-01-02 15:04:05") + } + if order.RefundTime.Valid { + item.RefundTime = order.RefundTime.Time.Format("2006-01-02 15:04:05") + } + + // 设置代理订单相关字段 + if agentOrderMap[order.Id] { + item.IsAgentOrder = true + item.AgentProcessStatus = agentProcessStatusMap[order.Id] + } else { + item.IsAgentOrder = false + item.AgentProcessStatus = "not_agent" + } + resp.Items = append(resp.Items, item) + } + + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_order/adminrefundorderlogic.go b/app/main/api/internal/logic/admin_order/adminrefundorderlogic.go new file mode 100644 index 0000000..5e170ae --- /dev/null +++ b/app/main/api/internal/logic/admin_order/adminrefundorderlogic.go @@ -0,0 +1,196 @@ +package admin_order + +import ( + "context" + "database/sql" + "fmt" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/google/uuid" +) + +const ( + PaymentPlatformAlipay = "alipay" + PaymentPlatformWechat = "wechat" + OrderStatusPaid = "paid" + RefundNoPrefix = "refund-" +) + +type AdminRefundOrderLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminRefundOrderLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminRefundOrderLogic { + return &AdminRefundOrderLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} +func (l *AdminRefundOrderLogic) AdminRefundOrder(req *types.AdminRefundOrderReq) (resp *types.AdminRefundOrderResp, err error) { + // 获取并验证订单 + order, err := l.getAndValidateOrder(req.Id, req.RefundAmount) + if err != nil { + return nil, err + } + + // 根据支付平台处理退款 + switch order.PaymentPlatform { + case PaymentPlatformAlipay: + return l.handleAlipayRefund(order, req) + case PaymentPlatformWechat: + return l.handleWechatRefund(order, req) + default: + return nil, errors.Wrapf(xerr.NewErrMsg("不支持的支付平台"), "AdminRefundOrder, 不支持的支付平台: %s", order.PaymentPlatform) + } +} + +// getAndValidateOrder 获取并验证订单信息 +func (l *AdminRefundOrderLogic) getAndValidateOrder(orderId string, refundAmount float64) (*model.Order, error) { + order, err := l.svcCtx.OrderModel.FindOne(l.ctx, orderId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminRefundOrder, 查询订单失败 err: %v", err) + } + + // 检查订单状态 + if order.Status != OrderStatusPaid { + return nil, errors.Wrapf(xerr.NewErrMsg("订单状态不正确，无法退款"), "AdminRefundOrder, 订单状态: %s", order.Status) + } + + // 检查退款金额 + if refundAmount > order.Amount { + return nil, errors.Wrapf(xerr.NewErrMsg("退款金额不能大于订单金额"), "AdminRefundOrder, 退款金额: %f, 订单金额: %f", refundAmount, order.Amount) + } + + return order, nil +} + +// handleAlipayRefund 处理支付宝退款 +func (l *AdminRefundOrderLogic) handleAlipayRefund(order *model.Order, req *types.AdminRefundOrderReq) (*types.AdminRefundOrderResp, error) { + // 调用支付宝退款接口 + refundResp, err := l.svcCtx.AlipayService.AliRefund(l.ctx, order.OrderNo, req.RefundAmount) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "AdminRefundOrder, 支付宝退款失败 err: %v", err) + } + + refundNo := l.generateRefundNo(order.OrderNo) + + if refundResp.IsSuccess() { + // 支付宝退款成功，创建成功记录 + err = l.createRefundRecordAndUpdateOrder(order, req, refundNo, refundResp.TradeNo, model.OrderStatusRefunded, model.OrderRefundStatusSuccess) + if err != nil { + return nil, err + } + + return &types.AdminRefundOrderResp{ + Status: model.OrderStatusRefunded, + RefundNo: refundNo, + Amount: req.RefundAmount, + }, nil + } else { + // 支付宝退款失败，创建失败记录但不更新订单状态 + err = l.createRefundRecordOnly(order, req, refundNo, refundResp.TradeNo, model.OrderRefundStatusFailed) + if err != nil { + logx.Errorf("创建退款失败记录时出错: %v", err) + } + return nil, errors.Wrapf(xerr.NewErrMsg(fmt.Sprintf("退款失败: %v", refundResp.Msg)), "AdminRefundOrder, 支付宝退款失败") + } +} + +// handleWechatRefund 处理微信退款 +func (l *AdminRefundOrderLogic) handleWechatRefund(order *model.Order, req *types.AdminRefundOrderReq) (*types.AdminRefundOrderResp, error) { + // 调用微信退款接口 + err := l.svcCtx.WechatPayService.WeChatRefund(l.ctx, order.OrderNo, req.RefundAmount, order.Amount) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "AdminRefundOrder, 微信退款失败 err: %v", err) + } + + // 微信退款是异步的，创建pending状态的退款记录 + refundNo := l.generateRefundNo(order.OrderNo) + err = l.createRefundRecordAndUpdateOrder(order, req, refundNo, "", model.OrderStatusRefunding, model.OrderRefundStatusPending) + if err != nil { + return nil, err + } + + return &types.AdminRefundOrderResp{ + Status: model.OrderRefundStatusPending, + RefundNo: refundNo, + Amount: req.RefundAmount, + }, nil +} + +// createRefundRecordAndUpdateOrder 创建退款记录并更新订单状态 +func (l *AdminRefundOrderLogic) createRefundRecordAndUpdateOrder(order *model.Order, req *types.AdminRefundOrderReq, refundNo, platformRefundId, orderStatus, refundStatus string) error { + return l.svcCtx.OrderModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 创建退款记录 + refund := &model.OrderRefund{ + Id: uuid.NewString(), + RefundNo: refundNo, + PlatformRefundId: l.createNullString(platformRefundId), + OrderId: order.Id, + UserId: order.UserId, + ProductId: order.ProductId, + RefundAmount: req.RefundAmount, + RefundReason: l.createNullString(req.RefundReason), + Status: refundStatus, // 使用传入的状态，不再硬编码 + RefundTime: sql.NullTime{Time: time.Now(), Valid: true}, + } + + if _, err := l.svcCtx.OrderRefundModel.Insert(ctx, session, refund); err != nil { + return fmt.Errorf("创建退款记录失败: %v", err) + } + + // 更新订单状态 + order.Status = orderStatus + if _, err := l.svcCtx.OrderModel.Update(ctx, session, order); err != nil { + return fmt.Errorf("更新订单状态失败: %v", err) + } + + return nil + }) +} + +// createRefundRecordOnly 仅创建退款记录，不更新订单状态（用于退款失败的情况） +func (l *AdminRefundOrderLogic) createRefundRecordOnly(order *model.Order, req *types.AdminRefundOrderReq, refundNo, platformRefundId, refundStatus string) error { + refund := &model.OrderRefund{ + Id: uuid.NewString(), + RefundNo: refundNo, + PlatformRefundId: l.createNullString(platformRefundId), + OrderId: order.Id, + UserId: order.UserId, + ProductId: order.ProductId, + RefundAmount: req.RefundAmount, + RefundReason: l.createNullString(req.RefundReason), + Status: refundStatus, + RefundTime: sql.NullTime{Time: time.Now(), Valid: true}, + } + + _, err := l.svcCtx.OrderRefundModel.Insert(l.ctx, nil, refund) + if err != nil { + return fmt.Errorf("创建退款记录失败: %v", err) + } + return nil +} + +// generateRefundNo 生成退款单号 +func (l *AdminRefundOrderLogic) generateRefundNo(orderNo string) string { + return fmt.Sprintf("%s%s", RefundNoPrefix, orderNo) +} + +// createNullString 创建 sql.NullString +func (l *AdminRefundOrderLogic) createNullString(value string) sql.NullString { + return sql.NullString{ + String: value, + Valid: value != "", + } +} diff --git a/app/main/api/internal/logic/admin_order/adminretryagentprocesslogic.go b/app/main/api/internal/logic/admin_order/adminretryagentprocesslogic.go new file mode 100644 index 0000000..d8efd3d --- /dev/null +++ b/app/main/api/internal/logic/admin_order/adminretryagentprocesslogic.go @@ -0,0 +1,63 @@ +package admin_order + +import ( + "context" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminRetryAgentProcessLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminRetryAgentProcessLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminRetryAgentProcessLogic { + return &AdminRetryAgentProcessLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminRetryAgentProcessLogic) AdminRetryAgentProcess(req *types.AdminRetryAgentProcessReq) (resp *types.AdminRetryAgentProcessResp, err error) { + // 新系统：查询订单并调用AgentProcess重新处理 + order, err := l.svcCtx.OrderModel.FindOne(l.ctx, req.Id) + if err != nil { + return &types.AdminRetryAgentProcessResp{ + Status: "failed", + Message: "订单不存在", + ProcessedAt: time.Now().Format("2006-01-02 15:04:05"), + }, nil + } + err = l.svcCtx.AgentService.AgentProcess(l.ctx, order) + if err != nil { + // 检查是否是"已经处理"的错误 + if err.Error() == "代理处理已经成功，无需重新执行" { + return &types.AdminRetryAgentProcessResp{ + Status: "already_processed", + Message: "代理处理已经成功，无需重新执行", + ProcessedAt: time.Now().Format("2006-01-02 15:04:05"), + }, nil + } + + // 其他错误 + logx.Errorf("重新执行代理处理失败，订单ID: %d, 错误: %v", req.Id, err) + return &types.AdminRetryAgentProcessResp{ + Status: "failed", + Message: err.Error(), + ProcessedAt: time.Now().Format("2006-01-02 15:04:05"), + }, nil + } + + // 执行成功 + return &types.AdminRetryAgentProcessResp{ + Status: "success", + Message: "代理处理重新执行成功", + ProcessedAt: time.Now().Format("2006-01-02 15:04:05"), + }, nil +} diff --git a/app/main/api/internal/logic/admin_order/adminupdateorderlogic.go b/app/main/api/internal/logic/admin_order/adminupdateorderlogic.go new file mode 100644 index 0000000..d6bd5e2 --- /dev/null +++ b/app/main/api/internal/logic/admin_order/adminupdateorderlogic.go @@ -0,0 +1,88 @@ +package admin_order + +import ( + "context" + "database/sql" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type AdminUpdateOrderLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdateOrderLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateOrderLogic { + return &AdminUpdateOrderLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdateOrderLogic) AdminUpdateOrder(req *types.AdminUpdateOrderReq) (resp *types.AdminUpdateOrderResp, err error) { + // 获取原订单信息 + order, err := l.svcCtx.OrderModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminUpdateOrder, 查询订单失败 err: %v", err) + } + + // 更新订单字段 + if req.OrderNo != nil { + order.OrderNo = *req.OrderNo + } + if req.PlatformOrderId != nil { + order.PlatformOrderId = sql.NullString{String: *req.PlatformOrderId, Valid: true} + } + if req.PaymentPlatform != nil { + order.PaymentPlatform = *req.PaymentPlatform + } + if req.PaymentScene != nil { + order.PaymentScene = *req.PaymentScene + } + if req.Amount != nil { + order.Amount = *req.Amount + } + if req.Status != nil { + order.Status = *req.Status + } + if req.PayTime != nil { + payTime, err := time.Parse("2006-01-02 15:04:05", *req.PayTime) + if err == nil { + order.PayTime = sql.NullTime{Time: payTime, Valid: true} + } + } + if req.RefundTime != nil { + refundTime, err := time.Parse("2006-01-02 15:04:05", *req.RefundTime) + if err == nil { + order.RefundTime = sql.NullTime{Time: refundTime, Valid: true} + } + } + + // 使用事务更新订单 + err = l.svcCtx.OrderModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 更新订单 + _, err := l.svcCtx.OrderModel.Update(ctx, session, order) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "AdminUpdateOrder, 更新订单失败 err: %v", err) + } + + return nil + }) + + if err != nil { + return nil, err + } + + return &types.AdminUpdateOrderResp{ + Success: true, + }, nil +} diff --git a/app/main/api/internal/logic/admin_platform_user/admincreateplatformuserlogic.go b/app/main/api/internal/logic/admin_platform_user/admincreateplatformuserlogic.go new file mode 100644 index 0000000..524c923 --- /dev/null +++ b/app/main/api/internal/logic/admin_platform_user/admincreateplatformuserlogic.go @@ -0,0 +1,56 @@ +package admin_platform_user + +import ( + "context" + "database/sql" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/google/uuid" +) + +type AdminCreatePlatformUserLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminCreatePlatformUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminCreatePlatformUserLogic { + return &AdminCreatePlatformUserLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminCreatePlatformUserLogic) AdminCreatePlatformUser(req *types.AdminCreatePlatformUserReq) (resp *types.AdminCreatePlatformUserResp, err error) { + // 校验手机号唯一性 + _, err = l.svcCtx.UserModel.FindOneByMobile(l.ctx, sql.NullString{String: req.Mobile, Valid: req.Mobile != ""}) + if err == nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "手机号已存在: %s", req.Mobile) + } + if err != model.ErrNotFound { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询手机号失败: %v", err) + } + + user := &model.User{ + Id: uuid.NewString(), + Mobile: sql.NullString{String: req.Mobile, Valid: req.Mobile != ""}, + Password: sql.NullString{String: req.Password, Valid: req.Password != ""}, + Nickname: sql.NullString{String: req.Nickname, Valid: req.Nickname != ""}, + Info: req.Info, + Inside: req.Inside, + } + result, err := l.svcCtx.UserModel.Insert(l.ctx, nil, user) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户失败: %v", err) + } + _ = result + resp = &types.AdminCreatePlatformUserResp{Id: user.Id} + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_platform_user/admindeleteplatformuserlogic.go b/app/main/api/internal/logic/admin_platform_user/admindeleteplatformuserlogic.go new file mode 100644 index 0000000..5e338e0 --- /dev/null +++ b/app/main/api/internal/logic/admin_platform_user/admindeleteplatformuserlogic.go @@ -0,0 +1,43 @@ +package admin_platform_user + +import ( + "context" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminDeletePlatformUserLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminDeletePlatformUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminDeletePlatformUserLogic { + return &AdminDeletePlatformUserLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminDeletePlatformUserLogic) AdminDeletePlatformUser(req *types.AdminDeletePlatformUserReq) (resp *types.AdminDeletePlatformUserResp, err error) { + user, err := l.svcCtx.UserModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "用户不存在: %d, err: %v", req.Id, err) + } + user.DelState = 1 + user.DeleteTime.Time = time.Now() + user.DeleteTime.Valid = true + err = l.svcCtx.UserModel.DeleteSoft(l.ctx, nil, user) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "软删除用户失败: %v", err) + } + resp = &types.AdminDeletePlatformUserResp{Success: true} + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_platform_user/admingetplatformuserdetaillogic.go b/app/main/api/internal/logic/admin_platform_user/admingetplatformuserdetaillogic.go new file mode 100644 index 0000000..1d4d3e0 --- /dev/null +++ b/app/main/api/internal/logic/admin_platform_user/admingetplatformuserdetaillogic.go @@ -0,0 +1,53 @@ +package admin_platform_user + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetPlatformUserDetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetPlatformUserDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetPlatformUserDetailLogic { + return &AdminGetPlatformUserDetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetPlatformUserDetailLogic) AdminGetPlatformUserDetail(req *types.AdminGetPlatformUserDetailReq) (resp *types.AdminGetPlatformUserDetailResp, err error) { + user, err := l.svcCtx.UserModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "用户不存在: %d, err: %v", req.Id, err) + } + key := l.svcCtx.Config.Encrypt.SecretKey + DecryptMobile, err := crypto.DecryptMobile(user.Mobile.String, key) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "解密手机号失败: %v", err) + } + // 查询平台类型（取第一个user_auth） + resp = &types.AdminGetPlatformUserDetailResp{ + Id: user.Id, + Mobile: DecryptMobile, + Nickname: "", + Info: user.Info, + Inside: user.Inside, + CreateTime: user.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: user.UpdateTime.Format("2006-01-02 15:04:05"), + } + if user.Nickname.Valid { + resp.Nickname = user.Nickname.String + } + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_platform_user/admingetplatformuserlistlogic.go b/app/main/api/internal/logic/admin_platform_user/admingetplatformuserlistlogic.go new file mode 100644 index 0000000..d068f82 --- /dev/null +++ b/app/main/api/internal/logic/admin_platform_user/admingetplatformuserlistlogic.go @@ -0,0 +1,88 @@ +package admin_platform_user + +import ( + "context" + "database/sql" + "fmt" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetPlatformUserListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetPlatformUserListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetPlatformUserListLogic { + return &AdminGetPlatformUserListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetPlatformUserListLogic) AdminGetPlatformUserList(req *types.AdminGetPlatformUserListReq) (resp *types.AdminGetPlatformUserListResp, err error) { + builder := l.svcCtx.UserModel.SelectBuilder() + if req.Mobile != "" { + builder = builder.Where("mobile = ?", req.Mobile) + } + if req.Nickname != "" { + builder = builder.Where("nickname = ?", req.Nickname) + } + if req.Inside != 0 { + builder = builder.Where("inside = ?", req.Inside) + } + if req.CreateTimeStart != "" { + builder = builder.Where("create_time >= ?", req.CreateTimeStart) + } + if req.CreateTimeEnd != "" { + builder = builder.Where("create_time <= ?", req.CreateTimeEnd) + } + + orderBy := "id DESC" + if req.OrderBy != "" && req.OrderType != "" { + orderBy = fmt.Sprintf("%s %s", req.OrderBy, req.OrderType) + } + users, total, err := l.svcCtx.UserModel.FindPageListByPageWithTotal(l.ctx, builder, req.Page, req.PageSize, orderBy) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户分页失败: %v", err) + } + var items []types.PlatformUserListItem + secretKey := l.svcCtx.Config.Encrypt.SecretKey + + for _, user := range users { + mobile := user.Mobile + if mobile.Valid { + encryptedMobile, err := crypto.DecryptMobile(mobile.String, secretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "手机登录, 解密手机号失败: %+v", err) + } + mobile = sql.NullString{String: encryptedMobile, Valid: true} + } + itemData := types.PlatformUserListItem{ + Id: user.Id, + Mobile: mobile.String, + Nickname: "", + Info: user.Info, + Inside: user.Inside, + CreateTime: user.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: user.UpdateTime.Format("2006-01-02 15:04:05"), + } + if user.Nickname.Valid { + itemData.Nickname = user.Nickname.String + } + items = append(items, itemData) + } + resp = &types.AdminGetPlatformUserListResp{ + Total: total, + Items: items, + } + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_platform_user/adminupdateplatformuserlogic.go b/app/main/api/internal/logic/admin_platform_user/adminupdateplatformuserlogic.go new file mode 100644 index 0000000..847a4c8 --- /dev/null +++ b/app/main/api/internal/logic/admin_platform_user/adminupdateplatformuserlogic.go @@ -0,0 +1,64 @@ +package admin_platform_user + +import ( + "context" + "database/sql" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminUpdatePlatformUserLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdatePlatformUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdatePlatformUserLogic { + return &AdminUpdatePlatformUserLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdatePlatformUserLogic) AdminUpdatePlatformUser(req *types.AdminUpdatePlatformUserReq) (resp *types.AdminUpdatePlatformUserResp, err error) { + user, err := l.svcCtx.UserModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "用户不存在: %d, err: %v", req.Id, err) + } + if req.Mobile != nil { + key := l.svcCtx.Config.Encrypt.SecretKey + EncryptMobile, err := crypto.EncryptMobile(*req.Mobile, key) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "加密手机号失败: %v", err) + } + user.Mobile = sql.NullString{String: EncryptMobile, Valid: true} + } + if req.Nickname != nil { + user.Nickname = sql.NullString{String: *req.Nickname, Valid: *req.Nickname != ""} + } + if req.Info != nil { + user.Info = *req.Info + } + if req.Inside != nil { + if *req.Inside != 1 && *req.Inside != 0 { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "内部用户状态错误: %d", *req.Inside) + } + user.Inside = *req.Inside + } + if req.Password != nil { + user.Password = sql.NullString{String: *req.Password, Valid: *req.Password != ""} + } + _, err = l.svcCtx.UserModel.Update(l.ctx, nil, user) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新用户失败: %v", err) + } + resp = &types.AdminUpdatePlatformUserResp{Success: true} + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_product/admincreateproductlogic.go b/app/main/api/internal/logic/admin_product/admincreateproductlogic.go new file mode 100644 index 0000000..c025a16 --- /dev/null +++ b/app/main/api/internal/logic/admin_product/admincreateproductlogic.go @@ -0,0 +1,79 @@ +package admin_product + +import ( + "context" + "database/sql" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type AdminCreateProductLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminCreateProductLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminCreateProductLogic { + return &AdminCreateProductLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminCreateProductLogic) AdminCreateProduct(req *types.AdminCreateProductReq) (resp *types.AdminCreateProductResp, err error) { + // 1. 数据转换 + data := &model.Product{ + Id: uuid.NewString(), + ProductName: req.ProductName, + ProductEn: req.ProductEn, + Description: req.Description, + Notes: sql.NullString{String: req.Notes, Valid: req.Notes != ""}, + CostPrice: req.CostPrice, + SellPrice: req.SellPrice, + } + + // 2. 数据库操作（使用事务确保产品表和代理产品配置表同步） + var productId string + err = l.svcCtx.ProductModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 2.1 插入产品 + _, err := l.svcCtx.ProductModel.Insert(ctx, session, data) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "创建产品失败, err: %v, req: %+v", err, req) + } + productId = data.Id + + // 2.2 同步创建代理产品配置（使用默认值） + agentProductConfig := &model.AgentProductConfig{ + Id: uuid.NewString(), + ProductId: productId, + BasePrice: 0.00, // 默认基础底价 + SystemMaxPrice: 9999.99, // 默认系统价格上限 + PriceThreshold: sql.NullFloat64{Valid: false}, // 默认不设阈值 + PriceFeeRate: sql.NullFloat64{Valid: false}, // 默认不收费 + } + + _, err = l.svcCtx.AgentProductConfigModel.Insert(ctx, session, agentProductConfig) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "创建代理产品配置失败, err: %v, productId: %s", err, productId) + } + + return nil + }) + + if err != nil { + return nil, err + } + + // 3. 返回结果 + return &types.AdminCreateProductResp{Id: productId}, nil +} diff --git a/app/main/api/internal/logic/admin_product/admindeleteproductlogic.go b/app/main/api/internal/logic/admin_product/admindeleteproductlogic.go new file mode 100644 index 0000000..2c44b7f --- /dev/null +++ b/app/main/api/internal/logic/admin_product/admindeleteproductlogic.go @@ -0,0 +1,67 @@ +package admin_product + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type AdminDeleteProductLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminDeleteProductLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminDeleteProductLogic { + return &AdminDeleteProductLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminDeleteProductLogic) AdminDeleteProduct(req *types.AdminDeleteProductReq) (resp *types.AdminDeleteProductResp, err error) { + // 1. 查询记录是否存在 + record, err := l.svcCtx.ProductModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查找产品失败, err: %v, id: %d", err, req.Id) + } + + // 2. 执行软删除（使用事务确保产品表和代理产品配置表同步） + err = l.svcCtx.ProductModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 2.1 软删除产品 + err = l.svcCtx.ProductModel.DeleteSoft(ctx, session, record) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "删除产品失败, err: %v, id: %d", err, req.Id) + } + + // 2.2 同步软删除代理产品配置 + agentProductConfig, err := l.svcCtx.AgentProductConfigModel.FindOneByProductId(ctx, req.Id) + if err != nil { + // 如果代理产品配置不存在，记录日志但不影响主流程 + l.Infof("同步删除代理产品配置失败：代理产品配置不存在, productId: %d, err: %v", req.Id, err) + } else { + err = l.svcCtx.AgentProductConfigModel.DeleteSoft(ctx, session, agentProductConfig) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "同步删除代理产品配置失败, err: %v, productId: %d", err, req.Id) + } + } + + return nil + }) + + if err != nil { + return nil, err + } + + // 3. 返回结果 + return &types.AdminDeleteProductResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_product/admingetproductdetaillogic.go b/app/main/api/internal/logic/admin_product/admingetproductdetaillogic.go new file mode 100644 index 0000000..ea30028 --- /dev/null +++ b/app/main/api/internal/logic/admin_product/admingetproductdetaillogic.go @@ -0,0 +1,49 @@ +package admin_product + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetProductDetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetProductDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetProductDetailLogic { + return &AdminGetProductDetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetProductDetailLogic) AdminGetProductDetail(req *types.AdminGetProductDetailReq) (resp *types.AdminGetProductDetailResp, err error) { + // 1. 查询记录 + record, err := l.svcCtx.ProductModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查找产品失败, err: %v, id: %d", err, req.Id) + } + + // 2. 构建响应 + resp = &types.AdminGetProductDetailResp{ + Id: record.Id, + ProductName: record.ProductName, + ProductEn: record.ProductEn, + Description: record.Description, + Notes: record.Notes.String, + CostPrice: record.CostPrice, + SellPrice: record.SellPrice, + CreateTime: record.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: record.UpdateTime.Format("2006-01-02 15:04:05"), + } + + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_product/admingetproductfeaturelistlogic.go b/app/main/api/internal/logic/admin_product/admingetproductfeaturelistlogic.go new file mode 100644 index 0000000..60573cc --- /dev/null +++ b/app/main/api/internal/logic/admin_product/admingetproductfeaturelistlogic.go @@ -0,0 +1,119 @@ +package admin_product + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/mr" +) + +type AdminGetProductFeatureListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetProductFeatureListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetProductFeatureListLogic { + return &AdminGetProductFeatureListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetProductFeatureListLogic) AdminGetProductFeatureList(req *types.AdminGetProductFeatureListReq) (resp *[]types.AdminGetProductFeatureListResp, err error) { + // 1. 构建查询条件 + builder := l.svcCtx.ProductFeatureModel.SelectBuilder(). + Where("product_id = ?", req.ProductId) + + // 2. 执行查询 + list, err := l.svcCtx.ProductFeatureModel.FindAll(l.ctx, builder, "sort ASC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询产品功能列表失败, err: %v, product_id: %d", err, req.ProductId) + } + + // 3. 获取所有功能ID + featureIds := make([]string, 0, len(list)) + for _, item := range list { + featureIds = append(featureIds, item.FeatureId) + } + + // 4. 并发查询功能详情 + type featureResult struct { + feature *model.Feature + err error + } + + results := make([]featureResult, len(featureIds)) + err = mr.MapReduceVoid(func(source chan<- interface{}) { + for i, id := range featureIds { + source <- struct { + index int + id string + }{i, id} + } + }, func(item interface{}, writer mr.Writer[featureResult], cancel func(error)) { + data := item.(struct { + index int + id string + }) + feature, err := l.svcCtx.FeatureModel.FindOne(l.ctx, data.id) + writer.Write(featureResult{ + feature: feature, + err: err, + }) + }, func(pipe <-chan featureResult, cancel func(error)) { + for result := range pipe { + if result.err != nil { + l.Logger.Errorf("查询功能详情失败, feature_id: %s, err: %v", result.feature.Id, result.err) + continue + } + results = append(results, result) + } + }) + + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), + "并发查询功能详情失败, err: %v", err) + } + + // 5. 构建功能ID到详情的映射 + featureMap := make(map[string]*model.Feature) + for _, result := range results { + if result.feature != nil { + featureMap[result.feature.Id] = result.feature + } + } + + // 6. 构建响应列表 + items := make([]types.AdminGetProductFeatureListResp, 0, len(list)) + for _, item := range list { + feature, exists := featureMap[item.FeatureId] + if !exists { + continue // 跳过不存在的功能 + } + + listItem := types.AdminGetProductFeatureListResp{ + Id: item.Id, + ProductId: item.ProductId, + FeatureId: item.FeatureId, + ApiId: feature.ApiId, + Name: feature.Name, + Sort: item.Sort, + Enable: item.Enable, + IsImportant: item.IsImportant, + CreateTime: item.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: item.UpdateTime.Format("2006-01-02 15:04:05"), + } + items = append(items, listItem) + } + + // 7. 返回结果 + return &items, nil +} diff --git a/app/main/api/internal/logic/admin_product/admingetproductlistlogic.go b/app/main/api/internal/logic/admin_product/admingetproductlistlogic.go new file mode 100644 index 0000000..794b010 --- /dev/null +++ b/app/main/api/internal/logic/admin_product/admingetproductlistlogic.go @@ -0,0 +1,69 @@ +package admin_product + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetProductListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetProductListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetProductListLogic { + return &AdminGetProductListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetProductListLogic) AdminGetProductList(req *types.AdminGetProductListReq) (resp *types.AdminGetProductListResp, err error) { + // 1. 构建查询条件 + builder := l.svcCtx.ProductModel.SelectBuilder() + + // 2. 添加查询条件 + if req.ProductName != nil && *req.ProductName != "" { + builder = builder.Where("product_name LIKE ?", "%"+*req.ProductName+"%") + } + if req.ProductEn != nil && *req.ProductEn != "" { + builder = builder.Where("product_en LIKE ?", "%"+*req.ProductEn+"%") + } + + // 3. 执行分页查询 + list, total, err := l.svcCtx.ProductModel.FindPageListByPageWithTotal( + l.ctx, builder, req.Page, req.PageSize, "id DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询产品列表失败, err: %v, req: %+v", err, req) + } + + // 4. 构建响应列表 + items := make([]types.ProductListItem, 0, len(list)) + for _, item := range list { + listItem := types.ProductListItem{ + Id: item.Id, + ProductName: item.ProductName, + ProductEn: item.ProductEn, + Description: item.Description, + Notes: item.Notes.String, + CostPrice: item.CostPrice, + SellPrice: item.SellPrice, + CreateTime: item.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: item.UpdateTime.Format("2006-01-02 15:04:05"), + } + items = append(items, listItem) + } + + // 5. 返回结果 + return &types.AdminGetProductListResp{ + Total: total, + Items: items, + }, nil +} diff --git a/app/main/api/internal/logic/admin_product/adminupdateproductfeatureslogic.go b/app/main/api/internal/logic/admin_product/adminupdateproductfeatureslogic.go new file mode 100644 index 0000000..5b139da --- /dev/null +++ b/app/main/api/internal/logic/admin_product/adminupdateproductfeatureslogic.go @@ -0,0 +1,161 @@ +package admin_product + +import ( + "context" + "sync" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/mr" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/google/uuid" +) + +type AdminUpdateProductFeaturesLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdateProductFeaturesLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateProductFeaturesLogic { + return &AdminUpdateProductFeaturesLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdateProductFeaturesLogic) AdminUpdateProductFeatures(req *types.AdminUpdateProductFeaturesReq) (resp *types.AdminUpdateProductFeaturesResp, err error) { + // 1. 查询现有关联 + builder := l.svcCtx.ProductFeatureModel.SelectBuilder(). + Where("product_id = ?", req.ProductId) + existingList, err := l.svcCtx.ProductFeatureModel.FindAll(l.ctx, builder, "id ASC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询现有产品功能关联失败, err: %v, product_id: %d", err, req.ProductId) + } + + // 2. 构建现有关联的映射 + existingMap := make(map[string]*model.ProductFeature) + for _, item := range existingList { + existingMap[item.FeatureId] = item + } + + // 3. 构建新关联的映射 + newMap := make(map[string]*types.ProductFeatureItem) + for _, item := range req.Features { + newMap[item.FeatureId] = &item + } + + // 4. 在事务中执行更新操作 + err = l.svcCtx.ProductFeatureModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 4.1 处理需要删除的关联 + var mu sync.Mutex + var deleteIds []string + err = mr.MapReduceVoid(func(source chan<- interface{}) { + for featureId, existing := range existingMap { + if _, exists := newMap[featureId]; !exists { + source <- existing.Id + } + } + }, func(item interface{}, writer mr.Writer[struct{}], cancel func(error)) { + id := item.(string) + mu.Lock() + deleteIds = append(deleteIds, id) + mu.Unlock() + }, func(pipe <-chan struct{}, cancel func(error)) { + // 等待所有ID收集完成 + }) + + if err != nil { + return errors.Wrapf(err, "收集待删除ID失败") + } + + // 批量删除 + if len(deleteIds) > 0 { + for _, id := range deleteIds { + err = l.svcCtx.ProductFeatureModel.Delete(ctx, session, id) + if err != nil { + return errors.Wrapf(err, "删除产品功能关联失败, product_id: %d, id: %d", + req.ProductId, id) + } + } + } + + // 4.2 并发处理需要新增或更新的关联 + var updateErr error + err = mr.MapReduceVoid(func(source chan<- interface{}) { + for featureId, newItem := range newMap { + source <- struct { + featureId string + newItem *types.ProductFeatureItem + existing *model.ProductFeature + }{ + featureId: featureId, + newItem: newItem, + existing: existingMap[featureId], + } + } + }, func(item interface{}, writer mr.Writer[struct{}], cancel func(error)) { + data := item.(struct { + featureId string + newItem *types.ProductFeatureItem + existing *model.ProductFeature + }) + + if data.existing != nil { + // 更新现有关联 + data.existing.Sort = data.newItem.Sort + data.existing.Enable = data.newItem.Enable + data.existing.IsImportant = data.newItem.IsImportant + _, err = l.svcCtx.ProductFeatureModel.Update(ctx, session, data.existing) + if err != nil { + updateErr = errors.Wrapf(err, "更新产品功能关联失败, product_id: %d, feature_id: %d", + req.ProductId, data.featureId) + cancel(updateErr) + return + } + } else { + // 新增关联 + newFeature := &model.ProductFeature{ + Id: uuid.NewString(), + ProductId: req.ProductId, + FeatureId: data.featureId, + Sort: data.newItem.Sort, + Enable: data.newItem.Enable, + IsImportant: data.newItem.IsImportant, + } + _, err = l.svcCtx.ProductFeatureModel.Insert(ctx, session, newFeature) + if err != nil { + updateErr = errors.Wrapf(err, "新增产品功能关联失败, product_id: %d, feature_id: %d", + req.ProductId, data.featureId) + cancel(updateErr) + return + } + } + }, func(pipe <-chan struct{}, cancel func(error)) { + // 等待所有更新完成 + }) + + if err != nil { + return errors.Wrapf(err, "并发更新产品功能关联失败") + } + if updateErr != nil { + return updateErr + } + + return nil + }) + + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "更新产品功能关联失败, err: %v, req: %+v", err, req) + } + + // 5. 返回结果 + return &types.AdminUpdateProductFeaturesResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_product/adminupdateproductlogic.go b/app/main/api/internal/logic/admin_product/adminupdateproductlogic.go new file mode 100644 index 0000000..98baf49 --- /dev/null +++ b/app/main/api/internal/logic/admin_product/adminupdateproductlogic.go @@ -0,0 +1,75 @@ +package admin_product + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + "database/sql" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type AdminUpdateProductLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdateProductLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateProductLogic { + return &AdminUpdateProductLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdateProductLogic) AdminUpdateProduct(req *types.AdminUpdateProductReq) (resp *types.AdminUpdateProductResp, err error) { + // 1. 查询记录是否存在 + record, err := l.svcCtx.ProductModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查找产品失败, err: %v, id: %d", err, req.Id) + } + + // 2. 更新字段 + if req.ProductName != nil { + record.ProductName = *req.ProductName + } + if req.ProductEn != nil { + record.ProductEn = *req.ProductEn + } + if req.Description != nil { + record.Description = *req.Description + } + if req.Notes != nil { + record.Notes = sql.NullString{String: *req.Notes, Valid: *req.Notes != ""} + } + if req.CostPrice != nil { + record.CostPrice = *req.CostPrice + } + if req.SellPrice != nil { + record.SellPrice = *req.SellPrice + } + + // 3. 执行更新操作（使用事务确保产品表和代理产品配置表同步） + err = l.svcCtx.ProductModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 3.1 更新产品 + _, err = l.svcCtx.ProductModel.Update(ctx, session, record) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "更新产品失败, err: %v, req: %+v", err, req) + } + + return nil + }) + + if err != nil { + return nil, err + } + + // 4. 返回结果 + return &types.AdminUpdateProductResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_query/admingetquerycleanupconfiglistlogic.go b/app/main/api/internal/logic/admin_query/admingetquerycleanupconfiglistlogic.go new file mode 100644 index 0000000..c4887ed --- /dev/null +++ b/app/main/api/internal/logic/admin_query/admingetquerycleanupconfiglistlogic.go @@ -0,0 +1,62 @@ +package admin_query + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetQueryCleanupConfigListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetQueryCleanupConfigListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetQueryCleanupConfigListLogic { + return &AdminGetQueryCleanupConfigListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetQueryCleanupConfigListLogic) AdminGetQueryCleanupConfigList(req *types.AdminGetQueryCleanupConfigListReq) (resp *types.AdminGetQueryCleanupConfigListResp, err error) { + // 构建查询条件 + builder := l.svcCtx.QueryCleanupConfigModel.SelectBuilder(). + Where("del_state = ?", globalkey.DelStateNo) + + if req.Status > 0 { + builder = builder.Where("status = ?", req.Status) + } + + // 查询配置列表 + configs, err := l.svcCtx.QueryCleanupConfigModel.FindAll(l.ctx, builder, "id ASC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询清理配置列表失败 err: %v", err) + } + + // 构建响应 + resp = &types.AdminGetQueryCleanupConfigListResp{ + Items: make([]types.QueryCleanupConfigItem, 0, len(configs)), + } + + for _, config := range configs { + item := types.QueryCleanupConfigItem{ + Id: config.Id, + ConfigKey: config.ConfigKey, + ConfigValue: config.ConfigValue, + ConfigDesc: config.ConfigDesc, + Status: config.Status, + CreateTime: config.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: config.UpdateTime.Format("2006-01-02 15:04:05"), + } + resp.Items = append(resp.Items, item) + } + + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_query/admingetquerycleanupdetaillistlogic.go b/app/main/api/internal/logic/admin_query/admingetquerycleanupdetaillistlogic.go new file mode 100644 index 0000000..735faf8 --- /dev/null +++ b/app/main/api/internal/logic/admin_query/admingetquerycleanupdetaillistlogic.go @@ -0,0 +1,126 @@ +package admin_query + +import ( + "context" + "sync" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/mr" +) + +type AdminGetQueryCleanupDetailListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetQueryCleanupDetailListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetQueryCleanupDetailListLogic { + return &AdminGetQueryCleanupDetailListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetQueryCleanupDetailListLogic) AdminGetQueryCleanupDetailList(req *types.AdminGetQueryCleanupDetailListReq) (resp *types.AdminGetQueryCleanupDetailListResp, err error) { + // 1. 验证清理日志是否存在 + _, err = l.svcCtx.QueryCleanupLogModel.FindOne(l.ctx, req.LogId) + if err != nil { + if err == model.ErrNotFound { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "清理日志不存在, log_id: %d", req.LogId) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询清理日志失败, log_id: %d, err: %v", req.LogId, err) + } + + // 2. 构建查询条件 + builder := l.svcCtx.QueryCleanupDetailModel.SelectBuilder(). + Where("cleanup_log_id = ?", req.LogId). + Where("del_state = ?", globalkey.DelStateNo) + + // 3. 并发获取总数和列表 + var total int64 + var details []*model.QueryCleanupDetail + err = mr.Finish(func() error { + var err error + total, err = l.svcCtx.QueryCleanupDetailModel.FindCount(l.ctx, builder, "id") + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询清理详情总数失败 err: %v", err) + } + return nil + }, func() error { + var err error + details, err = l.svcCtx.QueryCleanupDetailModel.FindPageListByPage(l.ctx, builder, req.Page, req.PageSize, "id DESC") + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询清理详情列表失败 err: %v", err) + } + return nil + }) + if err != nil { + return nil, err + } + + // 4. 获取所有产品ID + productIds := make([]string, 0, len(details)) + for _, detail := range details { + productIds = append(productIds, detail.ProductId) + } + + // 5. 并发获取产品信息 + productMap := make(map[string]string) + var mu sync.Mutex + err = mr.MapReduceVoid(func(source chan<- interface{}) { + for _, productId := range productIds { + source <- productId + } + }, func(item interface{}, writer mr.Writer[struct{}], cancel func(error)) { + productId := item.(string) + product, err := l.svcCtx.ProductModel.FindOne(l.ctx, productId) + if err != nil && !errors.Is(err, model.ErrNotFound) { + cancel(errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询产品信息失败, product_id: %s, err: %v", productId, err)) + return + } + mu.Lock() + if product != nil { + productMap[productId] = product.ProductName + } else { + productMap[productId] = "" // 产品不存在时设置为空字符串 + } + mu.Unlock() + writer.Write(struct{}{}) + }, func(pipe <-chan struct{}, cancel func(error)) { + for range pipe { + } + }) + if err != nil { + return nil, err + } + + // 6. 构建响应 + resp = &types.AdminGetQueryCleanupDetailListResp{ + Total: total, + Items: make([]types.QueryCleanupDetailItem, 0, len(details)), + } + + for _, detail := range details { + item := types.QueryCleanupDetailItem{ + Id: detail.Id, + CleanupLogId: detail.CleanupLogId, + QueryId: detail.QueryId, + OrderId: detail.OrderId, + UserId: detail.UserId, + ProductName: productMap[detail.ProductId], + QueryState: detail.QueryState, + CreateTimeOld: detail.CreateTimeOld.Format("2006-01-02 15:04:05"), + CreateTime: detail.CreateTime.Format("2006-01-02 15:04:05"), + } + resp.Items = append(resp.Items, item) + } + + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_query/admingetquerycleanuploglistlogic.go b/app/main/api/internal/logic/admin_query/admingetquerycleanuploglistlogic.go new file mode 100644 index 0000000..c070782 --- /dev/null +++ b/app/main/api/internal/logic/admin_query/admingetquerycleanuploglistlogic.go @@ -0,0 +1,88 @@ +package admin_query + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/mr" +) + +type AdminGetQueryCleanupLogListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetQueryCleanupLogListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetQueryCleanupLogListLogic { + return &AdminGetQueryCleanupLogListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetQueryCleanupLogListLogic) AdminGetQueryCleanupLogList(req *types.AdminGetQueryCleanupLogListReq) (resp *types.AdminGetQueryCleanupLogListResp, err error) { + // 构建查询条件 + builder := l.svcCtx.QueryCleanupLogModel.SelectBuilder(). + Where("del_state = ?", globalkey.DelStateNo) + + if req.Status > 0 { + builder = builder.Where("status = ?", req.Status) + } + if req.StartTime != "" { + builder = builder.Where("cleanup_time >= ?", req.StartTime) + } + if req.EndTime != "" { + builder = builder.Where("cleanup_time <= ?", req.EndTime) + } + + // 并发获取总数和列表 + var total int64 + var logs []*model.QueryCleanupLog + err = mr.Finish(func() error { + var err error + total, err = l.svcCtx.QueryCleanupLogModel.FindCount(l.ctx, builder, "id") + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询清理日志总数失败 err: %v", err) + } + return nil + }, func() error { + var err error + logs, err = l.svcCtx.QueryCleanupLogModel.FindPageListByPage(l.ctx, builder, req.Page, req.PageSize, "id DESC") + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询清理日志列表失败 err: %v", err) + } + return nil + }) + if err != nil { + return nil, err + } + + // 构建响应 + resp = &types.AdminGetQueryCleanupLogListResp{ + Total: total, + Items: make([]types.QueryCleanupLogItem, 0, len(logs)), + } + + for _, log := range logs { + item := types.QueryCleanupLogItem{ + Id: log.Id, + CleanupTime: log.CleanupTime.Format("2006-01-02 15:04:05"), + CleanupBefore: log.CleanupBefore.Format("2006-01-02 15:04:05"), + Status: log.Status, + AffectedRows: log.AffectedRows, + ErrorMsg: log.ErrorMsg.String, + Remark: log.Remark.String, + CreateTime: log.CreateTime.Format("2006-01-02 15:04:05"), + } + resp.Items = append(resp.Items, item) + } + + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_query/admingetquerydetailbyorderidlogic.go b/app/main/api/internal/logic/admin_query/admingetquerydetailbyorderidlogic.go new file mode 100644 index 0000000..336310f --- /dev/null +++ b/app/main/api/internal/logic/admin_query/admingetquerydetailbyorderidlogic.go @@ -0,0 +1,189 @@ +package admin_query + +import ( + "context" + "database/sql" + "encoding/hex" + "encoding/json" + "fmt" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/jinzhu/copier" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetQueryDetailByOrderIdLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetQueryDetailByOrderIdLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetQueryDetailByOrderIdLogic { + return &AdminGetQueryDetailByOrderIdLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetQueryDetailByOrderIdLogic) AdminGetQueryDetailByOrderId(req *types.AdminGetQueryDetailByOrderIdReq) (resp *types.AdminGetQueryDetailByOrderIdResp, err error) { + + // 获取报告信息 + queryModel, err := l.svcCtx.QueryModel.FindOneByOrderId(l.ctx, req.OrderId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告查询, 查找报告错误: %+v", err) + } + + var query types.AdminGetQueryDetailByOrderIdResp + query.CreateTime = queryModel.CreateTime.Format("2006-01-02 15:04:05") + query.UpdateTime = queryModel.UpdateTime.Format("2006-01-02 15:04:05") + + // 解密查询数据 + secretKey := l.svcCtx.Config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 获取AES解密解药失败, %+v", err) + } + processParamsErr := ProcessQueryParams(queryModel.QueryParams, &query.QueryParams, key) + if processParamsErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告参数处理失败: %v", processParamsErr) + } + processErr := ProcessQueryData(queryModel.QueryData, &query.QueryData, key) + if processErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结果处理失败: %v", processErr) + } + updateFeatureAndProductFeatureErr := l.UpdateFeatureAndProductFeature(queryModel.ProductId, &query.QueryData) + if updateFeatureAndProductFeatureErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结果处理失败: %v", updateFeatureAndProductFeatureErr) + } + // 复制报告数据 + err = copier.Copy(&query, queryModel) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结构体复制失败, %v", err) + } + product, err := l.svcCtx.ProductModel.FindOne(l.ctx, queryModel.ProductId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 获取商品信息失败, %v", err) + } + query.ProductName = product.ProductName + return &types.AdminGetQueryDetailByOrderIdResp{ + Id: query.Id, + OrderId: query.OrderId, + UserId: query.UserId, + ProductName: query.ProductName, + QueryParams: query.QueryParams, + QueryData: query.QueryData, + CreateTime: query.CreateTime, + UpdateTime: query.UpdateTime, + QueryState: query.QueryState, + }, nil +} + +// ProcessQueryData 解密和反序列化 QueryData +func ProcessQueryData(queryData sql.NullString, target *[]types.AdminQueryItem, key []byte) error { + queryDataStr := lzUtils.NullStringToString(queryData) + if queryDataStr == "" { + return nil + } + + // 解密数据 + decryptedData, decryptErr := crypto.AesDecrypt(queryDataStr, key) + if decryptErr != nil { + return decryptErr + } + + // 解析 JSON 数组 + var decryptedArray []map[string]interface{} + unmarshalErr := json.Unmarshal(decryptedData, &decryptedArray) + if unmarshalErr != nil { + return unmarshalErr + } + + // 确保 target 具有正确的长度 + if len(*target) == 0 { + *target = make([]types.AdminQueryItem, len(decryptedArray)) + } + + // 填充解密后的数据到 target + for i := 0; i < len(decryptedArray); i++ { + // 直接填充解密数据到 Data 字段 + (*target)[i].Data = decryptedArray[i] + } + return nil +} + +// ProcessQueryParams解密和反序列化 QueryParams +func ProcessQueryParams(QueryParams string, target *map[string]interface{}, key []byte) error { + // 解密 QueryParams + decryptedData, decryptErr := crypto.AesDecrypt(QueryParams, key) + if decryptErr != nil { + return decryptErr + } + + // 反序列化解密后的数据 + unmarshalErr := json.Unmarshal(decryptedData, target) + if unmarshalErr != nil { + return unmarshalErr + } + + return nil +} + +func (l *AdminGetQueryDetailByOrderIdLogic) UpdateFeatureAndProductFeature(productID string, target *[]types.AdminQueryItem) error { + // 遍历 target 数组，使用倒序遍历，以便删除元素时不影响索引 + for i := len(*target) - 1; i >= 0; i-- { + queryItem := &(*target)[i] + + // 确保 Data 为 map 类型 + data, ok := queryItem.Data.(map[string]interface{}) + if !ok { + return fmt.Errorf("queryItem.Data 必须是 map[string]interface{} 类型") + } + + // 从 Data 中获取 apiID + apiID, ok := data["apiID"].(string) + if !ok { + return fmt.Errorf("queryItem.Data 中的 apiID 必须是字符串类型") + } + + // 查询 Feature + feature, err := l.svcCtx.FeatureModel.FindOneByApiId(l.ctx, apiID) + if err != nil { + // 如果 Feature 查不到，也要删除当前 QueryItem + *target = append((*target)[:i], (*target)[i+1:]...) + continue + } + + // 查询 ProductFeatureModel + builder := l.svcCtx.ProductFeatureModel.SelectBuilder().Where("product_id = ?", productID) + productFeatures, err := l.svcCtx.ProductFeatureModel.FindAll(l.ctx, builder, "") + if err != nil { + return fmt.Errorf("查询 ProductFeatureModel 错误: %v", err) + } + + // 遍历 productFeatures，找到与 feature.ID 关联且 enable == 1 的项 + var featureData map[string]interface{} + sort := 0 + for _, pf := range productFeatures { + if pf.FeatureId == feature.Id { // 确保和 Feature 关联 + sort = int(pf.Sort) + break // 找到第一个符合条件的就退出循环 + } + } + featureData = map[string]interface{}{ + "featureName": feature.Name, + "sort": sort, + } + + // 更新 queryItem 的 Feature 字段（不是数组） + queryItem.Feature = featureData + } + + return nil +} diff --git a/app/main/api/internal/logic/admin_query/adminupdatequerycleanupconfiglogic.go b/app/main/api/internal/logic/admin_query/adminupdatequerycleanupconfiglogic.go new file mode 100644 index 0000000..7b2ee72 --- /dev/null +++ b/app/main/api/internal/logic/admin_query/adminupdatequerycleanupconfiglogic.go @@ -0,0 +1,63 @@ +package admin_query + +import ( + "context" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type AdminUpdateQueryCleanupConfigLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdateQueryCleanupConfigLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateQueryCleanupConfigLogic { + return &AdminUpdateQueryCleanupConfigLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdateQueryCleanupConfigLogic) AdminUpdateQueryCleanupConfig(req *types.AdminUpdateQueryCleanupConfigReq) (resp *types.AdminUpdateQueryCleanupConfigResp, err error) { + // 使用事务处理更新操作 + err = l.svcCtx.QueryCleanupConfigModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 1. 查询配置是否存在 + config, err := l.svcCtx.QueryCleanupConfigModel.FindOne(ctx, req.Id) + if err != nil { + if err == model.ErrNotFound { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "配置不存在, id: %d", req.Id) + } + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询配置失败, id: %d, err: %v", req.Id, err) + } + + // 2. 更新配置 + config.ConfigValue = req.ConfigValue + config.Status = req.Status + config.UpdateTime = time.Now() + + _, err = l.svcCtx.QueryCleanupConfigModel.Update(ctx, session, config) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新配置失败, id: %d, err: %v", req.Id, err) + } + + return nil + }) + + if err != nil { + return nil, err + } + + return &types.AdminUpdateQueryCleanupConfigResp{ + Success: true, + }, nil +} diff --git a/app/main/api/internal/logic/admin_role/createrolelogic.go b/app/main/api/internal/logic/admin_role/createrolelogic.go new file mode 100644 index 0000000..6bd334b --- /dev/null +++ b/app/main/api/internal/logic/admin_role/createrolelogic.go @@ -0,0 +1,83 @@ +package admin_role + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/google/uuid" +) + +type CreateRoleLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewCreateRoleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CreateRoleLogic { + return &CreateRoleLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *CreateRoleLogic) CreateRole(req *types.CreateRoleReq) (resp *types.CreateRoleResp, err error) { + // 检查角色编码是否已存在 + roleModel, err := l.svcCtx.AdminRoleModel.FindOneByRoleCode(l.ctx, req.RoleCode) + if err != nil && err != model.ErrNotFound { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建角色失败: %v", err) + } + if roleModel != nil && roleModel.RoleName == req.RoleName { + return nil, errors.Wrapf(xerr.NewErrMsg("角色名称已存在"), "创建角色失败, 角色名称已存在: %v", err) + } + // 创建角色 + role := &model.AdminRole{ + Id: uuid.NewString(), + RoleName: req.RoleName, + RoleCode: req.RoleCode, + Description: req.Description, + Status: req.Status, + Sort: req.Sort, + } + var roleId string + // 使用事务创建角色和关联菜单 + err = l.svcCtx.AdminRoleModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 创建角色 + _, err := l.svcCtx.AdminRoleModel.Insert(ctx, session, role) + if err != nil { + return errors.New("插入新角色失败") + } + roleId = role.Id + + // 创建角色菜单关联 + if len(req.MenuIds) > 0 { + for _, menuId := range req.MenuIds { + roleMenu := &model.AdminRoleMenu{ + Id: uuid.NewString(), + RoleId: roleId, + MenuId: menuId, + } + _, err = l.svcCtx.AdminRoleMenuModel.Insert(ctx, session, roleMenu) + if err != nil { + return errors.New("插入角色菜单关联失败") + } + } + } + + return nil + }) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建角色失败: %v", err) + } + + return &types.CreateRoleResp{ + Id: roleId, + }, nil +} diff --git a/app/main/api/internal/logic/admin_role/deleterolelogic.go b/app/main/api/internal/logic/admin_role/deleterolelogic.go new file mode 100644 index 0000000..60e9019 --- /dev/null +++ b/app/main/api/internal/logic/admin_role/deleterolelogic.go @@ -0,0 +1,84 @@ +package admin_role + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type DeleteRoleLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewDeleteRoleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteRoleLogic { + return &DeleteRoleLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *DeleteRoleLogic) DeleteRole(req *types.DeleteRoleReq) (resp *types.DeleteRoleResp, err error) { + // 检查角色是否存在 + _, err = l.svcCtx.AdminRoleModel.FindOne(l.ctx, req.Id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("角色不存在"), "删除角色失败, 角色不存在err: %v", err) + } + return nil, err + } + + // 使用事务删除角色和关联数据 + err = l.svcCtx.AdminRoleModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 删除角色菜单关联 + builder := l.svcCtx.AdminRoleMenuModel.SelectBuilder(). + Where("role_id = ?", req.Id) + menus, err := l.svcCtx.AdminRoleMenuModel.FindAll(ctx, builder, "id ASC") + if err != nil { + return err + } + for _, menu := range menus { + err = l.svcCtx.AdminRoleMenuModel.Delete(ctx, session, menu.Id) + if err != nil { + return err + } + } + + // 删除角色用户关联 + builder = l.svcCtx.AdminUserRoleModel.SelectBuilder(). + Where("role_id = ?", req.Id) + users, err := l.svcCtx.AdminUserRoleModel.FindAll(ctx, builder, "id ASC") + if err != nil { + return err + } + for _, user := range users { + err = l.svcCtx.AdminUserRoleModel.Delete(ctx, session, user.Id) + if err != nil { + return err + } + } + + // 删除角色 + err = l.svcCtx.AdminRoleModel.Delete(ctx, session, req.Id) + if err != nil { + return err + } + return nil + }) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除角色失败err: %v", err) + } + + return &types.DeleteRoleResp{ + Success: true, + }, nil +} diff --git a/app/main/api/internal/logic/admin_role/getroledetaillogic.go b/app/main/api/internal/logic/admin_role/getroledetaillogic.go new file mode 100644 index 0000000..871780b --- /dev/null +++ b/app/main/api/internal/logic/admin_role/getroledetaillogic.go @@ -0,0 +1,91 @@ +package admin_role + +import ( + "context" + "sync" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/samber/lo" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/mr" +) + +type GetRoleDetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetRoleDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetRoleDetailLogic { + return &GetRoleDetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetRoleDetailLogic) GetRoleDetail(req *types.GetRoleDetailReq) (resp *types.GetRoleDetailResp, err error) { + // 使用MapReduceVoid并发获取角色信息和菜单ID + var role *model.AdminRole + var menuIds []string + var mutex sync.Mutex + var wg sync.WaitGroup + + mr.MapReduceVoid(func(source chan<- interface{}) { + source <- 1 // 获取角色信息 + source <- 2 // 获取菜单ID + }, func(item interface{}, writer mr.Writer[interface{}], cancel func(error)) { + taskType := item.(int) + wg.Add(1) + defer wg.Done() + + if taskType == 1 { + result, err := l.svcCtx.AdminRoleModel.FindOne(l.ctx, req.Id) + if err != nil { + cancel(err) + return + } + mutex.Lock() + role = result + mutex.Unlock() + } else if taskType == 2 { + builder := l.svcCtx.AdminRoleMenuModel.SelectBuilder(). + Where("role_id = ?", req.Id) + menus, err := l.svcCtx.AdminRoleMenuModel.FindAll(l.ctx, builder, "id ASC") + if err != nil { + cancel(err) + return + } + mutex.Lock() + menuIds = lo.Map(menus, func(item *model.AdminRoleMenu, _ int) string { + return item.MenuId + }) + mutex.Unlock() + } + }, func(pipe <-chan interface{}, cancel func(error)) { + // 不需要处理pipe中的数据 + }) + + wg.Wait() + + if role == nil { + return nil, errors.Wrapf(xerr.NewErrMsg("角色不存在"), "获取角色详情失败, 角色不存在err: %v", err) + } + + return &types.GetRoleDetailResp{ + Id: role.Id, + RoleName: role.RoleName, + RoleCode: role.RoleCode, + Description: role.Description, + Status: role.Status, + Sort: role.Sort, + CreateTime: role.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: role.UpdateTime.Format("2006-01-02 15:04:05"), + MenuIds: menuIds, + }, nil +} diff --git a/app/main/api/internal/logic/admin_role/getrolelistlogic.go b/app/main/api/internal/logic/admin_role/getrolelistlogic.go new file mode 100644 index 0000000..102487a --- /dev/null +++ b/app/main/api/internal/logic/admin_role/getrolelistlogic.go @@ -0,0 +1,148 @@ +package admin_role + +import ( + "context" + "sync" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + + "github.com/samber/lo" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/mr" +) + +type GetRoleListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetRoleListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetRoleListLogic { + return &GetRoleListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} +func (l *GetRoleListLogic) GetRoleList(req *types.GetRoleListReq) (resp *types.GetRoleListResp, err error) { + resp = &types.GetRoleListResp{ + Items: make([]types.RoleListItem, 0), + Total: 0, + } + + // 构建查询条件 + builder := l.svcCtx.AdminRoleModel.SelectBuilder() + if len(req.Name) > 0 { + builder = builder.Where("role_name LIKE ?", "%"+req.Name+"%") + } + if len(req.Code) > 0 { + builder = builder.Where("role_code LIKE ?", "%"+req.Code+"%") + } + if req.Status != -1 { + builder = builder.Where("status = ?", req.Status) + } + + // 设置分页 + offset := (req.Page - 1) * req.PageSize + builder = builder.OrderBy("sort ASC").Limit(uint64(req.PageSize)).Offset(uint64(offset)) + + // 使用MapReduceVoid并发获取总数和列表数据 + var roles []*model.AdminRole + var total int64 + var mutex sync.Mutex + var wg sync.WaitGroup + + mr.MapReduceVoid(func(source chan<- interface{}) { + source <- 1 // 获取角色列表 + source <- 2 // 获取总数 + }, func(item interface{}, writer mr.Writer[*model.AdminRole], cancel func(error)) { + taskType := item.(int) + wg.Add(1) + defer wg.Done() + + if taskType == 1 { + result, err := l.svcCtx.AdminRoleModel.FindAll(l.ctx, builder, "id DESC") + if err != nil { + cancel(err) + return + } + mutex.Lock() + roles = result + mutex.Unlock() + } else if taskType == 2 { + countBuilder := l.svcCtx.AdminRoleModel.SelectBuilder() + if len(req.Name) > 0 { + countBuilder = countBuilder.Where("role_name LIKE ?", "%"+req.Name+"%") + } + if len(req.Code) > 0 { + countBuilder = countBuilder.Where("role_code LIKE ?", "%"+req.Code+"%") + } + if req.Status != -1 { + countBuilder = countBuilder.Where("status = ?", req.Status) + } + + count, err := l.svcCtx.AdminRoleModel.FindCount(l.ctx, countBuilder, "id") + if err != nil { + cancel(err) + return + } + mutex.Lock() + total = count + mutex.Unlock() + } + }, func(pipe <-chan *model.AdminRole, cancel func(error)) { + // 不需要处理pipe中的数据 + }) + + wg.Wait() + + // 并发获取每个角色的菜单ID + var roleItems []types.RoleListItem + var roleItemsMutex sync.Mutex + + mr.MapReduceVoid(func(source chan<- interface{}) { + for _, role := range roles { + source <- role + } + }, func(item interface{}, writer mr.Writer[[]string], cancel func(error)) { + role := item.(*model.AdminRole) + + // 获取角色关联的菜单ID + builder := l.svcCtx.AdminRoleMenuModel.SelectBuilder(). + Where("role_id = ?", role.Id) + menus, err := l.svcCtx.AdminRoleMenuModel.FindAll(l.ctx, builder, "id ASC") + if err != nil { + cancel(err) + return + } + menuIds := lo.Map(menus, func(item *model.AdminRoleMenu, _ int) string { + return item.MenuId + }) + + writer.Write(menuIds) + }, func(pipe <-chan []string, cancel func(error)) { + for _, role := range roles { + menuIds := <-pipe + item := types.RoleListItem{ + Id: role.Id, + RoleName: role.RoleName, + RoleCode: role.RoleCode, + Description: role.Description, + Status: role.Status, + Sort: role.Sort, + CreateTime: role.CreateTime.Format("2006-01-02 15:04:05"), + MenuIds: menuIds, + } + roleItemsMutex.Lock() + roleItems = append(roleItems, item) + roleItemsMutex.Unlock() + } + }) + + resp.Items = roleItems + resp.Total = total + + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_role/updaterolelogic.go b/app/main/api/internal/logic/admin_role/updaterolelogic.go new file mode 100644 index 0000000..ef04eb3 --- /dev/null +++ b/app/main/api/internal/logic/admin_role/updaterolelogic.go @@ -0,0 +1,150 @@ +package admin_role + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/samber/lo" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/google/uuid" +) + +type UpdateRoleLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewUpdateRoleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateRoleLogic { + return &UpdateRoleLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *UpdateRoleLogic) UpdateRole(req *types.UpdateRoleReq) (resp *types.UpdateRoleResp, err error) { + // 检查角色是否存在 + role, err := l.svcCtx.AdminRoleModel.FindOne(l.ctx, req.Id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("角色不存在"), "更新角色失败, 角色不存在err: %v", err) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新角色失败err: %v", err) + } + + // 检查角色编码是否重复 + if req.RoleCode != nil && *req.RoleCode != role.RoleCode { + roleModel, err := l.svcCtx.AdminRoleModel.FindOneByRoleCode(l.ctx, *req.RoleCode) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新角色失败err: %v", err) + } + if roleModel != nil { + return nil, errors.Wrapf(xerr.NewErrMsg("角色编码已存在"), "更新角色失败, 角色编码已存在err: %v", err) + } + } + + // 更新角色信息 + if req.RoleName != nil { + role.RoleName = *req.RoleName + } + if req.RoleCode != nil { + role.RoleCode = *req.RoleCode + } + if req.Description != nil { + role.Description = *req.Description + } + if req.Status != nil { + role.Status = *req.Status + } + if req.Sort != nil { + role.Sort = *req.Sort + } + + // 使用事务更新角色和关联菜单 + err = l.svcCtx.AdminRoleModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 更新角色 + _, err = l.svcCtx.AdminRoleModel.Update(ctx, session, role) + if err != nil { + return err + } + if req.MenuIds != nil { + // 1. 获取当前关联的菜单ID + builder := l.svcCtx.AdminRoleMenuModel.SelectBuilder(). + Where("role_id = ?", req.Id) + currentMenus, err := l.svcCtx.AdminRoleMenuModel.FindAll(ctx, builder, "id ASC") + if err != nil { + return err + } + + // 2. 转换为map便于查找 + currentMenuMap := make(map[string]*model.AdminRoleMenu) + for _, menu := range currentMenus { + currentMenuMap[menu.MenuId] = menu + } + + // 3. 检查新的菜单ID是否存在 + for _, menuId := range req.MenuIds { + exists, err := l.svcCtx.AdminMenuModel.FindOne(ctx, menuId) + if err != nil || exists == nil { + return errors.Wrapf(xerr.NewErrMsg("菜单不存在"), "菜单ID: %s", menuId) + } + } + + // 4. 找出需要删除和新增的关联 + var toDelete []*model.AdminRoleMenu + var toInsert []string + + // 需要删除的：当前存在但新列表中没有的 + for menuId, roleMenu := range currentMenuMap { + if !lo.Contains(req.MenuIds, menuId) { + toDelete = append(toDelete, roleMenu) + } + } + + // 需要新增的：新列表中有但当前不存在的 + for _, menuId := range req.MenuIds { + if _, exists := currentMenuMap[menuId]; !exists { + toInsert = append(toInsert, menuId) + } + } + + // 5. 删除需要移除的关联 + for _, roleMenu := range toDelete { + err = l.svcCtx.AdminRoleMenuModel.Delete(ctx, session, roleMenu.Id) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除角色菜单关联失败: %v", err) + } + } + + // 6. 添加新的关联 + for _, menuId := range toInsert { + roleMenu := &model.AdminRoleMenu{ + Id: uuid.NewString(), + RoleId: req.Id, + MenuId: menuId, + } + _, err = l.svcCtx.AdminRoleMenuModel.Insert(ctx, session, roleMenu) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "添加角色菜单关联失败: %v", err) + } + } + } + + return nil + }) + + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新角色失败err: %v", err) + } + + return &types.UpdateRoleResp{ + Success: true, + }, nil +} diff --git a/app/main/api/internal/logic/admin_role_api/adminassignroleapilogic.go b/app/main/api/internal/logic/admin_role_api/adminassignroleapilogic.go new file mode 100644 index 0000000..80d4219 --- /dev/null +++ b/app/main/api/internal/logic/admin_role_api/adminassignroleapilogic.go @@ -0,0 +1,98 @@ +package admin_role_api + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/google/uuid" +) + +type AdminAssignRoleApiLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminAssignRoleApiLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminAssignRoleApiLogic { + return &AdminAssignRoleApiLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminAssignRoleApiLogic) AdminAssignRoleApi(req *types.AdminAssignRoleApiReq) (resp *types.AdminAssignRoleApiResp, err error) { + // 1. 参数验证 + if req.RoleId == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "角色ID不能为空, roleId: %s", req.RoleId) + } + if len(req.ApiIds) == 0 { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API ID列表不能为空") + } + + // 2. 查询角色是否存在 + _, err = l.svcCtx.AdminRoleModel.FindOne(l.ctx, req.RoleId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "角色不存在, roleId: %d", req.RoleId) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询角色失败, err: %v, roleId: %d", err, req.RoleId) + } + + // 3. 批量分配API权限 + successCount := 0 + for _, apiId := range req.ApiIds { + if apiId == "" { + continue + } + + // 检查API是否存在 + _, err := l.svcCtx.AdminApiModel.FindOne(l.ctx, apiId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + logx.Errorf("API不存在, apiId: %s", apiId) + continue + } + logx.Errorf("查询API失败, err: %v, apiId: %s", err, apiId) + continue + } + + // 检查是否已存在关联 + existing, err := l.svcCtx.AdminRoleApiModel.FindOneByRoleIdApiId(l.ctx, req.RoleId, apiId) + if err != nil && !errors.Is(err, model.ErrNotFound) { + logx.Errorf("查询角色API关联失败, err: %v, roleId: %d, apiId: %d", err, req.RoleId, apiId) + continue + } + if existing != nil { + continue // 已存在，跳过 + } + + // 创建关联 + roleApiData := &model.AdminRoleApi{ + Id: uuid.NewString(), + RoleId: req.RoleId, + ApiId: apiId, + } + + _, err = l.svcCtx.AdminRoleApiModel.Insert(l.ctx, nil, roleApiData) + if err != nil { + logx.Errorf("创建角色API关联失败, err: %v, roleId: %d, apiId: %d", err, req.RoleId, apiId) + continue + } + + successCount++ + } + + // 4. 返回结果 + return &types.AdminAssignRoleApiResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_role_api/admingetallapilistlogic.go b/app/main/api/internal/logic/admin_role_api/admingetallapilistlogic.go new file mode 100644 index 0000000..4bda294 --- /dev/null +++ b/app/main/api/internal/logic/admin_role_api/admingetallapilistlogic.go @@ -0,0 +1,64 @@ +package admin_role_api + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetAllApiListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetAllApiListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetAllApiListLogic { + return &AdminGetAllApiListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetAllApiListLogic) AdminGetAllApiList(req *types.AdminGetAllApiListReq) (resp *types.AdminGetAllApiListResp, err error) { + // 1. 构建查询条件 + builder := l.svcCtx.AdminApiModel.SelectBuilder() + + // 添加状态过滤 + if req.Status > 0 { + builder = builder.Where("status = ?", req.Status) + } + + // 2. 查询所有API列表 + apis, err := l.svcCtx.AdminApiModel.FindAll(l.ctx, builder, "id ASC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询API列表失败, err: %v", err) + } + + // 3. 转换数据格式 + var apiList []types.AdminRoleApiInfo + for _, api := range apis { + apiList = append(apiList, types.AdminRoleApiInfo{ + Id: api.Id, + RoleId: "", + ApiId: api.Id, + ApiName: api.ApiName, + ApiCode: api.ApiCode, + Method: api.Method, + Url: api.Url, + Status: api.Status, + Description: api.Description, + }) + } + + // 4. 返回结果 + return &types.AdminGetAllApiListResp{ + Items: apiList, + }, nil +} diff --git a/app/main/api/internal/logic/admin_role_api/admingetroleapilistlogic.go b/app/main/api/internal/logic/admin_role_api/admingetroleapilistlogic.go new file mode 100644 index 0000000..c4ea015 --- /dev/null +++ b/app/main/api/internal/logic/admin_role_api/admingetroleapilistlogic.go @@ -0,0 +1,84 @@ +package admin_role_api + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminGetRoleApiListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetRoleApiListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetRoleApiListLogic { + return &AdminGetRoleApiListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetRoleApiListLogic) AdminGetRoleApiList(req *types.AdminGetRoleApiListReq) (resp *types.AdminGetRoleApiListResp, err error) { + // 1. 参数验证 + if req.RoleId == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "角色ID不能为空, roleId: %s", req.RoleId) + } + + // 2. 查询角色是否存在 + _, err = l.svcCtx.AdminRoleModel.FindOne(l.ctx, req.RoleId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "角色不存在, roleId: %s", req.RoleId) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询角色失败, err: %v, roleId: %d", err, req.RoleId) + } + + // 3. 查询角色API权限列表 + builder := l.svcCtx.AdminRoleApiModel.SelectBuilder(). + Where("role_id = ?", req.RoleId) + + roleApis, err := l.svcCtx.AdminRoleApiModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询角色API权限失败, err: %v, roleId: %d", err, req.RoleId) + } + + // 4. 转换数据格式 + var apiList []types.AdminRoleApiInfo + for _, roleApi := range roleApis { + // 查询API详情 + api, err := l.svcCtx.AdminApiModel.FindOne(l.ctx, roleApi.ApiId) + if err != nil { + logx.Errorf("查询API详情失败, err: %v, apiId: %d", err, roleApi.ApiId) + continue + } + + apiList = append(apiList, types.AdminRoleApiInfo{ + Id: roleApi.Id, + RoleId: roleApi.RoleId, + ApiId: roleApi.ApiId, + ApiName: api.ApiName, + ApiCode: api.ApiCode, + Method: api.Method, + Url: api.Url, + Status: api.Status, + Description: api.Description, + }) + } + + // 5. 返回结果 + return &types.AdminGetRoleApiListResp{ + Items: apiList, + }, nil +} diff --git a/app/main/api/internal/logic/admin_role_api/adminremoveroleapilogic.go b/app/main/api/internal/logic/admin_role_api/adminremoveroleapilogic.go new file mode 100644 index 0000000..0b7d4c4 --- /dev/null +++ b/app/main/api/internal/logic/admin_role_api/adminremoveroleapilogic.go @@ -0,0 +1,80 @@ +package admin_role_api + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminRemoveRoleApiLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminRemoveRoleApiLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminRemoveRoleApiLogic { + return &AdminRemoveRoleApiLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminRemoveRoleApiLogic) AdminRemoveRoleApi(req *types.AdminRemoveRoleApiReq) (resp *types.AdminRemoveRoleApiResp, err error) { + // 1. 参数验证 + if req.RoleId == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "角色ID不能为空, roleId: %s", req.RoleId) + } + if len(req.ApiIds) == 0 { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "API ID列表不能为空") + } + + // 2. 查询角色是否存在 + _, err = l.svcCtx.AdminRoleModel.FindOne(l.ctx, req.RoleId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "角色不存在, roleId: %d", req.RoleId) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询角色失败, err: %v, roleId: %d", err, req.RoleId) + } + + // 3. 批量移除API权限 + successCount := 0 + for _, apiId := range req.ApiIds { + if apiId == "" { + continue + } + + // 查询关联记录 + roleApi, err := l.svcCtx.AdminRoleApiModel.FindOneByRoleIdApiId(l.ctx, req.RoleId, apiId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + continue // 不存在，跳过 + } + logx.Errorf("查询角色API关联失败, err: %v, roleId: %s, apiId: %s", err, req.RoleId, apiId) + continue + } + + // 删除关联 + err = l.svcCtx.AdminRoleApiModel.DeleteSoft(l.ctx, nil, roleApi) + if err != nil { + logx.Errorf("删除角色API关联失败, err: %v, roleId: %d, apiId: %d", err, req.RoleId, apiId) + continue + } + + successCount++ + } + + // 4. 返回结果 + return &types.AdminRemoveRoleApiResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_role_api/adminupdateroleapilogic.go b/app/main/api/internal/logic/admin_role_api/adminupdateroleapilogic.go new file mode 100644 index 0000000..c28eef2 --- /dev/null +++ b/app/main/api/internal/logic/admin_role_api/adminupdateroleapilogic.go @@ -0,0 +1,98 @@ +package admin_role_api + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/google/uuid" +) + +type AdminUpdateRoleApiLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdateRoleApiLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateRoleApiLogic { + return &AdminUpdateRoleApiLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdateRoleApiLogic) AdminUpdateRoleApi(req *types.AdminUpdateRoleApiReq) (resp *types.AdminUpdateRoleApiResp, err error) { + // 1. 参数验证 + if req.RoleId == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.PARAM_VERIFICATION_ERROR), + "角色ID不能为空, roleId: %s", req.RoleId) + } + + // 2. 查询角色是否存在 + _, err = l.svcCtx.AdminRoleModel.FindOne(l.ctx, req.RoleId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "角色不存在, roleId: %s", req.RoleId) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询角色失败, err: %v, roleId: %s", err, req.RoleId) + } + + // 3. 删除该角色的所有API权限 + builder := l.svcCtx.AdminRoleApiModel.SelectBuilder().Where("role_id = ?", req.RoleId) + roleApis, err := l.svcCtx.AdminRoleApiModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), + "查询角色API权限失败, err: %v, roleId: %d", err, req.RoleId) + } + + // 删除现有权限 + for _, roleApi := range roleApis { + err = l.svcCtx.AdminRoleApiModel.DeleteSoft(l.ctx, nil, roleApi) + if err != nil { + logx.Errorf("删除角色API权限失败, err: %v, roleId: %d, apiId: %d", err, req.RoleId, roleApi.ApiId) + } + } + + // 4. 添加新的API权限 + if len(req.ApiIds) > 0 { + for _, apiId := range req.ApiIds { + if apiId == "" { + continue + } + + // 检查API是否存在 + _, err := l.svcCtx.AdminApiModel.FindOne(l.ctx, apiId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + logx.Errorf("API不存在, apiId: %s", apiId) + continue + } + logx.Errorf("查询API失败, err: %v, apiId: %s", err, apiId) + continue + } + + // 创建新的关联 + roleApiData := &model.AdminRoleApi{ + Id: uuid.NewString(), + RoleId: req.RoleId, + ApiId: apiId, + } + + _, err = l.svcCtx.AdminRoleApiModel.Insert(l.ctx, nil, roleApiData) + if err != nil { + logx.Errorf("创建角色API关联失败, err: %v, roleId: %d, apiId: %d", err, req.RoleId, apiId) + } + } + } + + // 5. 返回结果 + return &types.AdminUpdateRoleApiResp{Success: true}, nil +} diff --git a/app/main/api/internal/logic/admin_user/admincreateuserlogic.go b/app/main/api/internal/logic/admin_user/admincreateuserlogic.go new file mode 100644 index 0000000..5c94ad1 --- /dev/null +++ b/app/main/api/internal/logic/admin_user/admincreateuserlogic.go @@ -0,0 +1,86 @@ +package admin_user + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type AdminCreateUserLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminCreateUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminCreateUserLogic { + return &AdminCreateUserLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminCreateUserLogic) AdminCreateUser(req *types.AdminCreateUserReq) (resp *types.AdminCreateUserResp, err error) { + // 检查用户名是否已存在 + exists, err := l.svcCtx.AdminUserModel.FindOneByUsername(l.ctx, req.Username) + if err != nil && err != model.ErrNotFound { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户失败: %v", err) + } + if exists != nil { + return nil, errors.Wrapf(xerr.NewErrMsg("用户名已存在"), "创建用户失败") + } + password, err := crypto.PasswordHash("123456") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "创建用户失败, 加密密码失败: %v", err) + } + // 创建用户 + user := &model.AdminUser{ + Id: uuid.NewString(), + Username: req.Username, + Password: password, + RealName: req.RealName, + Status: req.Status, + } + + // 使用事务创建用户和关联角色 + err = l.svcCtx.AdminUserModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 创建用户 + _, err := l.svcCtx.AdminUserModel.Insert(ctx, session, user) + if err != nil { + return err + } + // 创建用户角色关联 + if len(req.RoleIds) > 0 { + for _, roleId := range req.RoleIds { + userRole := &model.AdminUserRole{ + Id: uuid.NewString(), + UserId: user.Id, + RoleId: roleId, + } + _, err = l.svcCtx.AdminUserRoleModel.Insert(ctx, session, userRole) + if err != nil { + return err + } + } + } + + return nil + }) + + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户失败: %v", err) + } + + return &types.AdminCreateUserResp{ + Id: user.Id, + }, nil +} diff --git a/app/main/api/internal/logic/admin_user/admindeleteuserlogic.go b/app/main/api/internal/logic/admin_user/admindeleteuserlogic.go new file mode 100644 index 0000000..3924813 --- /dev/null +++ b/app/main/api/internal/logic/admin_user/admindeleteuserlogic.go @@ -0,0 +1,68 @@ +package admin_user + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type AdminDeleteUserLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminDeleteUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminDeleteUserLogic { + return &AdminDeleteUserLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminDeleteUserLogic) AdminDeleteUser(req *types.AdminDeleteUserReq) (resp *types.AdminDeleteUserResp, err error) { + // 检查用户是否存在 + _, err = l.svcCtx.AdminUserModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "用户不存在: %v", err) + } + + // 使用事务删除用户和关联数据 + err = l.svcCtx.AdminUserModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 删除用户角色关联 + builder := l.svcCtx.AdminUserRoleModel.SelectBuilder(). + Where("user_id = ?", req.Id) + roles, err := l.svcCtx.AdminUserRoleModel.FindAll(ctx, builder, "id ASC") + if err != nil { + return err + } + for _, role := range roles { + err = l.svcCtx.AdminUserRoleModel.Delete(ctx, session, role.Id) + if err != nil { + return err + } + } + + // 删除用户 + err = l.svcCtx.AdminUserModel.Delete(ctx, session, req.Id) + if err != nil { + return err + } + + return nil + }) + + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除用户失败: %v", err) + } + + return &types.AdminDeleteUserResp{ + Success: true, + }, nil +} diff --git a/app/main/api/internal/logic/admin_user/admingetuserdetaillogic.go b/app/main/api/internal/logic/admin_user/admingetuserdetaillogic.go new file mode 100644 index 0000000..21db655 --- /dev/null +++ b/app/main/api/internal/logic/admin_user/admingetuserdetaillogic.go @@ -0,0 +1,88 @@ +package admin_user + +import ( + "context" + "errors" + "sync" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + + "github.com/samber/lo" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/mr" +) + +type AdminGetUserDetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetUserDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetUserDetailLogic { + return &AdminGetUserDetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetUserDetailLogic) AdminGetUserDetail(req *types.AdminGetUserDetailReq) (resp *types.AdminGetUserDetailResp, err error) { + // 使用MapReduceVoid并发获取用户信息和角色ID + var user *model.AdminUser + var roleIds []string + var mutex sync.Mutex + var wg sync.WaitGroup + + mr.MapReduceVoid(func(source chan<- interface{}) { + source <- 1 // 获取用户信息 + source <- 2 // 获取角色ID + }, func(item interface{}, writer mr.Writer[interface{}], cancel func(error)) { + taskType := item.(int) + wg.Add(1) + defer wg.Done() + + if taskType == 1 { + result, err := l.svcCtx.AdminUserModel.FindOne(l.ctx, req.Id) + if err != nil { + cancel(err) + return + } + mutex.Lock() + user = result + mutex.Unlock() + } else if taskType == 2 { + builder := l.svcCtx.AdminUserRoleModel.SelectBuilder(). + Where("user_id = ?", req.Id) + roles, err := l.svcCtx.AdminUserRoleModel.FindAll(l.ctx, builder, "id ASC") + if err != nil { + cancel(err) + return + } + mutex.Lock() + roleIds = lo.Map(roles, func(item *model.AdminUserRole, _ int) string { + return item.RoleId + }) + mutex.Unlock() + } + }, func(pipe <-chan interface{}, cancel func(error)) { + // 不需要处理pipe中的数据 + }) + + wg.Wait() + + if user == nil { + return nil, errors.New("用户不存在") + } + + return &types.AdminGetUserDetailResp{ + Id: user.Id, + Username: user.Username, + RealName: user.RealName, + Status: user.Status, + CreateTime: user.CreateTime.Format("2006-01-02 15:04:05"), + UpdateTime: user.UpdateTime.Format("2006-01-02 15:04:05"), + RoleIds: roleIds, + }, nil +} diff --git a/app/main/api/internal/logic/admin_user/admingetuserlistlogic.go b/app/main/api/internal/logic/admin_user/admingetuserlistlogic.go new file mode 100644 index 0000000..d5078f8 --- /dev/null +++ b/app/main/api/internal/logic/admin_user/admingetuserlistlogic.go @@ -0,0 +1,149 @@ +package admin_user + +import ( + "context" + "sync" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + + "github.com/samber/lo" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/mr" +) + +type AdminGetUserListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminGetUserListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminGetUserListLogic { + return &AdminGetUserListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminGetUserListLogic) AdminGetUserList(req *types.AdminGetUserListReq) (resp *types.AdminGetUserListResp, err error) { + resp = &types.AdminGetUserListResp{ + Items: make([]types.AdminUserListItem, 0), + Total: 0, + } + + // 构建查询条件 + builder := l.svcCtx.AdminUserModel.SelectBuilder(). + Where("del_state = ?", 0) + if len(req.Username) > 0 { + builder = builder.Where("username LIKE ?", "%"+req.Username+"%") + } + if len(req.RealName) > 0 { + builder = builder.Where("real_name LIKE ?", "%"+req.RealName+"%") + } + if req.Status != -1 { + builder = builder.Where("status = ?", req.Status) + } + + // 设置分页 + offset := (req.Page - 1) * req.PageSize + builder = builder.OrderBy("id DESC").Limit(uint64(req.PageSize)).Offset(uint64(offset)) + + // 使用MapReduceVoid并发获取总数和列表数据 + var users []*model.AdminUser + var total int64 + var mutex sync.Mutex + var wg sync.WaitGroup + + mr.MapReduceVoid(func(source chan<- interface{}) { + source <- 1 // 获取用户列表 + source <- 2 // 获取总数 + }, func(item interface{}, writer mr.Writer[*model.AdminUser], cancel func(error)) { + taskType := item.(int) + wg.Add(1) + defer wg.Done() + + if taskType == 1 { + result, err := l.svcCtx.AdminUserModel.FindAll(l.ctx, builder, "id DESC") + if err != nil { + cancel(err) + return + } + mutex.Lock() + users = result + mutex.Unlock() + } else if taskType == 2 { + countBuilder := l.svcCtx.AdminUserModel.SelectBuilder(). + Where("del_state = ?", 0) + if len(req.Username) > 0 { + countBuilder = countBuilder.Where("username LIKE ?", "%"+req.Username+"%") + } + if len(req.RealName) > 0 { + countBuilder = countBuilder.Where("real_name LIKE ?", "%"+req.RealName+"%") + } + if req.Status != -1 { + countBuilder = countBuilder.Where("status = ?", req.Status) + } + + count, err := l.svcCtx.AdminUserModel.FindCount(l.ctx, countBuilder, "id") + if err != nil { + cancel(err) + return + } + mutex.Lock() + total = count + mutex.Unlock() + } + }, func(pipe <-chan *model.AdminUser, cancel func(error)) { + // 不需要处理pipe中的数据 + }) + + wg.Wait() + + // 并发获取每个用户的角色ID + var userItems []types.AdminUserListItem + var userItemsMutex sync.Mutex + + mr.MapReduceVoid(func(source chan<- interface{}) { + for _, user := range users { + source <- user + } + }, func(item interface{}, writer mr.Writer[[]string], cancel func(error)) { + user := item.(*model.AdminUser) + + // 获取用户关联的角色ID + builder := l.svcCtx.AdminUserRoleModel.SelectBuilder(). + Where("user_id = ?", user.Id) + roles, err := l.svcCtx.AdminUserRoleModel.FindAll(l.ctx, builder, "id ASC") + if err != nil { + cancel(err) + return + } + roleIds := lo.Map(roles, func(item *model.AdminUserRole, _ int) string { + return item.RoleId + }) + + writer.Write(roleIds) + }, func(pipe <-chan []string, cancel func(error)) { + for _, user := range users { + roleIds := <-pipe + item := types.AdminUserListItem{ + Id: user.Id, + Username: user.Username, + RealName: user.RealName, + Status: user.Status, + CreateTime: user.CreateTime.Format("2006-01-02 15:04:05"), + RoleIds: roleIds, + } + userItemsMutex.Lock() + userItems = append(userItems, item) + userItemsMutex.Unlock() + } + }) + + resp.Items = userItems + resp.Total = total + + return resp, nil +} diff --git a/app/main/api/internal/logic/admin_user/adminresetpasswordlogic.go b/app/main/api/internal/logic/admin_user/adminresetpasswordlogic.go new file mode 100644 index 0000000..4138c87 --- /dev/null +++ b/app/main/api/internal/logic/admin_user/adminresetpasswordlogic.go @@ -0,0 +1,63 @@ +package admin_user + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminResetPasswordLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminResetPasswordLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminResetPasswordLogic { + return &AdminResetPasswordLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminResetPasswordLogic) AdminResetPassword(req *types.AdminResetPasswordReq) (resp *types.AdminResetPasswordResp, err error) { + // 检查用户是否存在 + user, err := l.svcCtx.AdminUserModel.FindOne(l.ctx, req.Id) + if err != nil { + if err == model.ErrNotFound { + return nil, errors.Wrapf(xerr.NewErrMsg("用户不存在"), "用户ID: %d", req.Id) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户失败: %v", err) + } + + // 检查用户状态 + if user.Status != 1 { + return nil, errors.Wrapf(xerr.NewErrMsg("用户已被禁用，无法重置密码"), "用户ID: %d", req.Id) + } + + // 对密码进行加密 + hashedPassword, err := crypto.PasswordHash(req.Password) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "密码加密失败: %v", err) + } + + // 更新用户密码 + user.Password = hashedPassword + _, err = l.svcCtx.AdminUserModel.Update(l.ctx, nil, user) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新密码失败: %v", err) + } + + l.Infof("管理员密码重置成功，用户ID: %d, 用户名: %s", req.Id, user.Username) + + return &types.AdminResetPasswordResp{ + Success: true, + }, nil +} diff --git a/app/main/api/internal/logic/admin_user/adminupdateuserlogic.go b/app/main/api/internal/logic/admin_user/adminupdateuserlogic.go new file mode 100644 index 0000000..b8bd9db --- /dev/null +++ b/app/main/api/internal/logic/admin_user/adminupdateuserlogic.go @@ -0,0 +1,143 @@ +package admin_user + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/samber/lo" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/google/uuid" +) + +type AdminUpdateUserLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUpdateUserLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUpdateUserLogic { + return &AdminUpdateUserLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUpdateUserLogic) AdminUpdateUser(req *types.AdminUpdateUserReq) (resp *types.AdminUpdateUserResp, err error) { + // 检查用户是否存在 + user, err := l.svcCtx.AdminUserModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "用户不存在: %v", err) + } + + // 检查用户名是否重复 + if req.Username != nil && *req.Username != user.Username { + exists, err := l.svcCtx.AdminUserModel.FindOneByUsername(l.ctx, *req.Username) + if err != nil && err != model.ErrNotFound { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新用户失败: %v", err) + } + if exists != nil { + return nil, errors.Wrapf(xerr.NewErrMsg("用户名已存在"), "更新用户失败") + } + } + + // 更新用户信息 + if req.Username != nil { + user.Username = *req.Username + } + if req.RealName != nil { + user.RealName = *req.RealName + } + if req.Status != nil { + user.Status = *req.Status + } + + // 使用事务更新用户和关联角色 + err = l.svcCtx.AdminUserModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 更新用户 + _, err = l.svcCtx.AdminUserModel.Update(ctx, session, user) + if err != nil { + return err + } + + // 只有当RoleIds不为nil时才更新角色关联 + if req.RoleIds != nil { + // 1. 获取当前关联的角色ID + builder := l.svcCtx.AdminUserRoleModel.SelectBuilder(). + Where("user_id = ?", req.Id) + currentRoles, err := l.svcCtx.AdminUserRoleModel.FindAll(ctx, builder, "id ASC") + if err != nil { + return err + } + + // 2. 转换为map便于查找 + currentRoleMap := make(map[string]*model.AdminUserRole) + for _, role := range currentRoles { + currentRoleMap[role.RoleId] = role + } + + // 3. 检查新的角色ID是否存在 + for _, roleId := range req.RoleIds { + exists, err := l.svcCtx.AdminRoleModel.FindOne(ctx, roleId) + if err != nil || exists == nil { + return errors.Wrapf(xerr.NewErrMsg("角色不存在"), "角色ID: %s", roleId) + } + } + + // 4. 找出需要删除和新增的关联 + var toDelete []*model.AdminUserRole + var toInsert []string + + // 需要删除的：当前存在但新列表中没有的 + for roleId, userRole := range currentRoleMap { + if !lo.Contains(req.RoleIds, roleId) { + toDelete = append(toDelete, userRole) + } + } + + // 需要新增的：新列表中有但当前不存在的 + for _, roleId := range req.RoleIds { + if _, exists := currentRoleMap[roleId]; !exists { + toInsert = append(toInsert, roleId) + } + } + + // 5. 删除需要移除的关联 + for _, userRole := range toDelete { + err = l.svcCtx.AdminUserRoleModel.Delete(ctx, session, userRole.Id) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除用户角色关联失败: %v", err) + } + } + + // 6. 添加新的关联 + for _, roleId := range toInsert { + userRole := &model.AdminUserRole{ + Id: uuid.NewString(), + UserId: req.Id, + RoleId: roleId, + } + _, err = l.svcCtx.AdminUserRoleModel.Insert(ctx, session, userRole) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "添加用户角色关联失败: %v", err) + } + } + } + + return nil + }) + + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新用户失败: %v", err) + } + + return &types.AdminUpdateUserResp{ + Success: true, + }, nil +} diff --git a/app/main/api/internal/logic/admin_user/adminuserinfologic.go b/app/main/api/internal/logic/admin_user/adminuserinfologic.go new file mode 100644 index 0000000..cd0937b --- /dev/null +++ b/app/main/api/internal/logic/admin_user/adminuserinfologic.go @@ -0,0 +1,67 @@ +package admin_user + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AdminUserInfoLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAdminUserInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AdminUserInfoLogic { + return &AdminUserInfoLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AdminUserInfoLogic) AdminUserInfo(req *types.AdminUserInfoReq) (resp *types.AdminUserInfoResp, err error) { + userId, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户ID失败, %+v", err) + } + + user, err := l.svcCtx.AdminUserModel.FindOne(l.ctx, userId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户ID信息失败, %+v", err) + } + // 获取权限 + adminUserRoleBuilder := l.svcCtx.AdminUserRoleModel.SelectBuilder().Where(squirrel.Eq{"user_id": user.Id}) + permissions, err := l.svcCtx.AdminUserRoleModel.FindAll(l.ctx, adminUserRoleBuilder, "role_id DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrMsg("获取权限失败"), "用户登录, 获取权限失败, 用户名: %s", user.Username) + } + + // 获取角色ID数组 + roleIds := make([]string, 0) + for _, permission := range permissions { + roleIds = append(roleIds, permission.RoleId) + } + + // 获取角色名称 + roles := make([]string, 0) + for _, roleId := range roleIds { + role, err := l.svcCtx.AdminRoleModel.FindOne(l.ctx, roleId) + if err != nil { + continue + } + roles = append(roles, role.RoleCode) + } + return &types.AdminUserInfoResp{ + Username: user.Username, + RealName: user.RealName, + Roles: roles, + }, nil +} diff --git a/app/main/api/internal/logic/agent/applyforagentlogic.go b/app/main/api/internal/logic/agent/applyforagentlogic.go new file mode 100644 index 0000000..697adb4 --- /dev/null +++ b/app/main/api/internal/logic/agent/applyforagentlogic.go @@ -0,0 +1,363 @@ +package agent + +import ( + "context" + "database/sql" + "fmt" + "os" + "strconv" + "time" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/redis" + "github.com/zeromicro/go-zero/core/stores/sqlx" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type ApplyForAgentLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewApplyForAgentLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ApplyForAgentLogic { + return &ApplyForAgentLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *ApplyForAgentLogic) ApplyForAgent(req *types.AgentApplyReq) (resp *types.AgentApplyResp, err error) { + claims, err := ctxdata.GetClaimsFromCtx(l.ctx) + if err != nil && !errors.Is(err, ctxdata.ErrNoInCtx) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "代理申请失败, %v", err) + } + + secretKey := l.svcCtx.Config.Encrypt.SecretKey + encryptedMobile, err := crypto.EncryptMobile(req.Mobile, secretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "加密手机号失败: %v", err) + } + + if req.Referrer == "" { + return nil, errors.Wrapf(xerr.NewErrMsg("请填写邀请信息"), "") + } + // 2. 校验验证码（开发环境下跳过验证码校验） + if os.Getenv("ENV") != "development" { + redisKey := fmt.Sprintf("%s:%s", "agentApply", encryptedMobile) + cacheCode, err := l.svcCtx.Redis.Get(redisKey) + if err != nil { + if errors.Is(err, redis.Nil) { + return nil, errors.Wrapf(xerr.NewErrMsg("验证码已过期"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "读取验证码失败, %v", err) + } + if cacheCode != req.Code { + return nil, errors.Wrapf(xerr.NewErrMsg("验证码不正确"), "") + } + } + + var userID string + transErr := l.svcCtx.AgentModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + // 1. 处理用户注册/绑定 + user, findUserErr := l.svcCtx.UserModel.FindOneByMobile(l.ctx, sql.NullString{String: encryptedMobile, Valid: true}) + if findUserErr != nil && !errors.Is(findUserErr, model.ErrNotFound) { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户失败, %v", findUserErr) + } + + if user == nil { + // 用户不存在，注册新用户 + if claims != nil && claims.UserType == model.UserTypeNormal { + return errors.Wrapf(xerr.NewErrMsg("当前用户已注册，请输入注册的手机号"), "") + } + userID, err = l.svcCtx.UserService.RegisterUser(l.ctx, encryptedMobile) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "注册用户失败: %v", err) + } + } else { + // 用户已存在 + if claims != nil && claims.UserType == model.UserTypeTemp { + // 临时用户，检查手机号是否已绑定其他微信号 + userAuth, findUserAuthErr := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(l.ctx, user.Id, claims.AuthType) + if findUserAuthErr != nil && !errors.Is(findUserAuthErr, model.ErrNotFound) { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户认证失败, %v", findUserAuthErr) + } + if userAuth != nil && userAuth.AuthKey != claims.AuthKey { + return errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他微信号"), "") + } + // 临时用户，转为正式用户 + err = l.svcCtx.UserService.TempUserBindUser(l.ctx, session, user.Id) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "绑定用户失败: %v", err) + } + } + userID = user.Id + } + + // 3. 检查是否已是代理 + existingAgent, err := l.svcCtx.AgentModel.FindOneByUserId(transCtx, userID) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + if existingAgent != nil { + return errors.Wrapf(xerr.NewErrMsg("您已经是代理"), "") + } + + var inviteCodeModel *model.AgentInviteCode + var parentAgentId string + var targetLevel int64 + + inviteCodeModel, err = l.svcCtx.AgentInviteCodeModel.FindOneByCode(transCtx, req.Referrer) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询邀请码失败, %v", err) + } + if inviteCodeModel != nil { + if inviteCodeModel.Status != 0 { + if inviteCodeModel.Status == 1 { + return errors.Wrapf(xerr.NewErrMsg("邀请码已使用"), "") + } + return errors.Wrapf(xerr.NewErrMsg("邀请码已失效"), "") + } + if inviteCodeModel.ExpireTime.Valid && inviteCodeModel.ExpireTime.Time.Before(time.Now()) { + return errors.Wrapf(xerr.NewErrMsg("邀请码已过期"), "") + } + targetLevel = inviteCodeModel.TargetLevel + if inviteCodeModel.AgentId.Valid { + parentAgentId = inviteCodeModel.AgentId.String + } + } else { + if codeVal, parseErr := strconv.ParseInt(req.Referrer, 10, 64); parseErr == nil && codeVal > 0 { + parentAgent, err := l.findAgentByCode(transCtx, codeVal) + if err != nil { + return err + } + parentAgentId = parentAgent.Id + targetLevel = 1 + } else { + encRefMobile, _ := crypto.EncryptMobile(req.Referrer, l.svcCtx.Config.Encrypt.SecretKey) + agents, findErr := l.svcCtx.AgentModel.FindAll(transCtx, l.svcCtx.AgentModel.SelectBuilder().Where("mobile = ? AND del_state = ?", encRefMobile, globalkey.DelStateNo).Limit(1), "") + if findErr != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", findErr) + } + if len(agents) == 0 { + return errors.Wrapf(xerr.NewErrMsg("邀请信息无效"), "") + } + parentAgentId = agents[0].Id + targetLevel = 1 + } + } + + // 4.5 创建代理记录 + newAgent := &model.Agent{ + Id: uuid.NewString(), + UserId: userID, + Level: targetLevel, + Mobile: encryptedMobile, + } + if req.Region != "" { + newAgent.Region = sql.NullString{String: req.Region, Valid: true} + } + + // 4.6 处理上级关系 + if parentAgentId != "" { + // 代理发放的邀请码，成为该代理的下级 + parentAgent, err := l.svcCtx.AgentModel.FindOne(transCtx, parentAgentId) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", err) + } + + // 验证关系是否允许 + if newAgent.Level > parentAgent.Level { + return errors.Wrapf(xerr.NewErrMsg("代理等级不能高于上级代理"), "") + } + + // 查找团队首领 + teamLeaderId, err := l.findTeamLeader(transCtx, parentAgent.Id) + if err != nil { + return errors.Wrapf(err, "查找团队首领失败") + } + if teamLeaderId != "" { + newAgent.TeamLeaderId = sql.NullString{String: teamLeaderId, Valid: true} + } + + newAgent.AgentCode = 0 + _, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent) + if err != nil { + return errors.Wrapf(err, "创建代理记录失败") + } + // 已设置newAgent.Id为UUID + + // 建立关系 + relation := &model.AgentRelation{ + Id: uuid.NewString(), + ParentId: parentAgent.Id, + ChildId: newAgent.Id, + RelationType: 1, + } + if _, err := l.svcCtx.AgentRelationModel.Insert(transCtx, session, relation); err != nil { + return errors.Wrapf(err, "建立代理关系失败") + } + } else { + // 平台发放的钻石邀请码，独立成团队 + if targetLevel == 3 { + newAgent.AgentCode = 0 + _, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent) + if err != nil { + return errors.Wrapf(err, "创建代理记录失败") + } + // 设置自己为团队首领 + newAgent.TeamLeaderId = sql.NullString{String: newAgent.Id, Valid: true} + if err := l.svcCtx.AgentModel.UpdateInTransaction(transCtx, session, newAgent); err != nil { + return errors.Wrapf(err, "更新团队首领失败") + } + } else { + newAgent.AgentCode = 0 + _, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent) + if err != nil { + return errors.Wrapf(err, "创建代理记录失败") + } + } + } + + // 4.7 初始化钱包 + wallet := &model.AgentWallet{ + Id: uuid.NewString(), + AgentId: newAgent.Id, + } + if _, err := l.svcCtx.AgentWalletModel.Insert(transCtx, session, wallet); err != nil { + return errors.Wrapf(err, "初始化钱包失败") + } + + // 4.8 更新邀请码状态 + // 钻石级别的邀请码只能使用一次，使用后立即失效 + // 普通级别的邀请码可以无限使用，不更新状态 + if targetLevel == 3 { + // 钻石邀请码：使用后失效 + inviteCodeModel.Status = 1 // 已使用（使用后立即失效） + } + // 记录使用信息（用于统计，普通邀请码可以多次使用） + inviteCodeModel.UsedUserId = sql.NullString{String: userID, Valid: true} + if inviteCodeModel != nil { + inviteCodeModel.UsedAgentId = sql.NullString{String: newAgent.Id, Valid: true} + inviteCodeModel.UsedTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := l.svcCtx.AgentInviteCodeModel.UpdateWithVersion(transCtx, session, inviteCodeModel); err != nil { + return errors.Wrapf(err, "更新邀请码状态失败") + } + } + + // 4.9 记录邀请码使用历史（用于统计和查询） + usage := &model.AgentInviteCodeUsage{ + Id: uuid.NewString(), + InviteCodeId: inviteCodeModel.Id, + Code: inviteCodeModel.Code, + UserId: userID, + AgentId: newAgent.Id, + AgentLevel: targetLevel, + UsedTime: time.Now(), + } + if _, err := l.svcCtx.AgentInviteCodeUsageModel.Insert(transCtx, session, usage); err != nil { + return errors.Wrapf(err, "记录邀请码使用历史失败") + } + + return nil + }) + + if transErr != nil { + return nil, transErr + } + + // 6. 生成并返回token + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成token失败: %v", err) + } + + now := time.Now().Unix() + agent, _ := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + var code int64 + if agent != nil { + code = agent.AgentCode + } + return &types.AgentApplyResp{ + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + AgentCode: code, + }, nil +} + +func (l *ApplyForAgentLogic) findAgentByCode(ctx context.Context, code int64) (*model.Agent, error) { + builder := l.svcCtx.AgentModel.SelectBuilder().Where("agent_code = ? AND del_state = ?", code, globalkey.DelStateNo).Limit(1) + agents, err := l.svcCtx.AgentModel.FindAll(ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", err) + } + if len(agents) == 0 { + return nil, errors.Wrapf(xerr.NewErrMsg("上级邀请码不存在"), "") + } + return agents[0], nil +} + +func (l *ApplyForAgentLogic) allocateAgentCode(ctx context.Context, session sqlx.Session) (int64, error) { + builder := l.svcCtx.AgentModel.SelectBuilder().OrderBy("agent_code DESC").Limit(1) + rows, err := l.svcCtx.AgentModel.FindAll(ctx, builder, "") + if err != nil { + return 0, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理编码失败, %v", err) + } + var next int64 = 16800 + if len(rows) > 0 && rows[0].AgentCode > 0 { + next = rows[0].AgentCode + 1 + } + return next, nil +} + +// findTeamLeader 查找团队首领（钻石代理） +func (l *ApplyForAgentLogic) findTeamLeader(ctx context.Context, agentId string) (string, error) { + currentId := agentId + maxDepth := 100 + depth := 0 + + for depth < maxDepth { + builder := l.svcCtx.AgentRelationModel.SelectBuilder(). + Where("child_id = ? AND relation_type = ? AND del_state = ?", currentId, 1, 0) + relations, err := l.svcCtx.AgentRelationModel.FindAll(ctx, builder, "") + if err != nil { + return "", err + } + if len(relations) == 0 { + agent, err := l.svcCtx.AgentModel.FindOne(ctx, currentId) + if err != nil { + return "", err + } + if agent.Level == 3 { + return agent.Id, nil + } + return "", nil + } + + parentAgent, err := l.svcCtx.AgentModel.FindOne(ctx, relations[0].ParentId) + if err != nil { + return "", err + } + + if parentAgent.Level == 3 { + return parentAgent.Id, nil + } + + currentId = parentAgent.Id + depth++ + } + + return "", nil +} diff --git a/app/main/api/internal/logic/agent/applyupgradelogic.go b/app/main/api/internal/logic/agent/applyupgradelogic.go new file mode 100644 index 0000000..45f73fb --- /dev/null +++ b/app/main/api/internal/logic/agent/applyupgradelogic.go @@ -0,0 +1,125 @@ +package agent + +import ( + "context" + "database/sql" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/sqlx" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type ApplyUpgradeLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewApplyUpgradeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ApplyUpgradeLogic { + return &ApplyUpgradeLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *ApplyUpgradeLogic) ApplyUpgrade(req *types.ApplyUpgradeReq) (resp *types.ApplyUpgradeResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + fromLevel := agent.Level + toLevel := req.ToLevel + + // 2. 验证升级条件 + if !l.canUpgrade(agent.Level, toLevel, 1) { + return nil, errors.Wrapf(xerr.NewErrMsg("升级条件不满足"), "") + } + + // 3. 计算升级费用和返佣 + upgradeFee, err := l.svcCtx.AgentService.GetUpgradeFee(l.ctx, fromLevel, toLevel) + if err != nil { + return nil, errors.Wrapf(err, "获取升级费用失败") + } + rebateAmount, err := l.svcCtx.AgentService.GetUpgradeRebate(l.ctx, fromLevel, toLevel) + if err != nil { + return nil, errors.Wrapf(err, "获取升级返佣金额失败") + } + + // 4. 查找原直接上级（用于返佣） + var rebateAgentId string + parent, err := l.svcCtx.AgentService.FindDirectParent(l.ctx, agent.Id) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(err, "查找直接上级失败") + } + if parent != nil { + rebateAgentId = parent.Id + } + + // 5. 创建升级记录（待支付状态） + var upgradeId string + err = l.svcCtx.AgentWalletModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + // 5.1 创建升级记录（状态为待支付） + upgradeRecord := &model.AgentUpgrade{ + Id: uuid.NewString(), + AgentId: agent.Id, + FromLevel: fromLevel, + ToLevel: toLevel, + UpgradeType: 1, // 自主付费 + UpgradeFee: upgradeFee, + RebateAmount: rebateAmount, + Status: 1, // 待支付（1=待支付，2=已支付，3=已完成，4=已取消） + } + if rebateAgentId != "" { + upgradeRecord.RebateAgentId = sql.NullString{String: rebateAgentId, Valid: true} + } + + _, err := l.svcCtx.AgentUpgradeModel.Insert(transCtx, session, upgradeRecord) + if err != nil { + return errors.Wrapf(err, "创建升级记录失败") + } + // 注意：升级操作将在支付成功后通过支付回调完成 + upgradeId = upgradeRecord.Id + return nil + }) + + if err != nil { + return nil, err + } + + // 返回响应（订单号将在支付接口中生成） + return &types.ApplyUpgradeResp{ + UpgradeId: upgradeId, + OrderNo: "", // 将在支付接口中生成 + }, nil +} + +// canUpgrade 检查是否可以升级 +func (l *ApplyUpgradeLogic) canUpgrade(fromLevel, toLevel int64, upgradeType int64) bool { + if upgradeType == 1 { // 自主付费 + if fromLevel == 1 { // 普通 + return toLevel == 2 || toLevel == 3 // 可以升级为黄金或钻石 + } else if fromLevel == 2 { // 黄金 + return toLevel == 3 // 可以升级为钻石 + } + } + return false +} diff --git a/app/main/api/internal/logic/agent/applywithdrawallogic.go b/app/main/api/internal/logic/agent/applywithdrawallogic.go new file mode 100644 index 0000000..7f548e7 --- /dev/null +++ b/app/main/api/internal/logic/agent/applywithdrawallogic.go @@ -0,0 +1,226 @@ +package agent + +import ( + "context" + "fmt" + "time" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/sqlx" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type ApplyWithdrawalLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewApplyWithdrawalLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ApplyWithdrawalLogic { + return &ApplyWithdrawalLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *ApplyWithdrawalLogic) ApplyWithdrawal(req *types.ApplyWithdrawalReq) (resp *types.ApplyWithdrawalResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 验证实名认证 + realName, err := l.svcCtx.AgentRealNameModel.FindOneByAgentId(l.ctx, agent.Id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("请先完成实名认证"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询实名认证失败, %v", err) + } + // 检查是否已通过三要素核验（verify_time不为空表示已通过） + if !realName.VerifyTime.Valid { + return nil, errors.Wrapf(xerr.NewErrMsg("请先完成实名认证"), "") + } + + // 3. 验证提现金额 + if req.Amount <= 0 { + return nil, errors.Wrapf(xerr.NewErrMsg("提现金额必须大于0"), "") + } + + // 4. 获取钱包信息 + wallet, err := l.svcCtx.AgentWalletModel.FindOneByAgentId(l.ctx, agent.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询钱包失败, %v", err) + } + + // 5. 验证余额 + if wallet.Balance < req.Amount { + return nil, errors.Wrapf(xerr.NewErrMsg(fmt.Sprintf("余额不足，当前余额：%.2f", wallet.Balance)), "") + } + + // 6. 计算税费 + yearMonth := int64(time.Now().Year()*100 + int(time.Now().Month())) + taxInfo, err := l.calculateTax(l.ctx, agent.Id, req.Amount, yearMonth) + if err != nil { + return nil, errors.Wrapf(err, "计算税费失败") + } + + // 7. 生成提现单号 + withdrawNo := fmt.Sprintf("WD%d%d", time.Now().Unix(), agent.Id) + + // 8. 使用事务处理提现申请 + var withdrawalId string + err = l.svcCtx.AgentWalletModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + // 8.1 冻结余额 + wallet.FrozenBalance += req.Amount + wallet.Balance -= req.Amount + if err := l.svcCtx.AgentWalletModel.UpdateWithVersion(transCtx, session, wallet); err != nil { + return errors.Wrapf(err, "冻结余额失败") + } + + // 8.2 创建提现记录 + withdrawal := &model.AgentWithdrawal{ + Id: uuid.New().String(), + AgentId: agent.Id, + WithdrawNo: withdrawNo, + PayeeAccount: req.PayeeAccount, + PayeeName: req.PayeeName, + Amount: req.Amount, + ActualAmount: taxInfo.ActualAmount, + TaxAmount: taxInfo.TaxAmount, + Status: 1, // 处理中（待审核） + } + + _, err := l.svcCtx.AgentWithdrawalModel.Insert(transCtx, session, withdrawal) + if err != nil { + return errors.Wrapf(err, "创建提现记录失败") + } + withdrawalId = withdrawal.Id + + // 8.3 创建扣税记录 + taxRecord := &model.AgentWithdrawalTax{ + AgentId: agent.Id, + WithdrawalId: withdrawalId, + YearMonth: yearMonth, + WithdrawalAmount: req.Amount, + TaxableAmount: taxInfo.TaxableAmount, + TaxRate: taxInfo.TaxRate, + TaxAmount: taxInfo.TaxAmount, + ActualAmount: taxInfo.ActualAmount, + TaxStatus: 1, // 待扣税 + Remark: lzUtils.StringToNullString("提现申请"), + } + if _, err := l.svcCtx.AgentWithdrawalTaxModel.Insert(transCtx, session, taxRecord); err != nil { + return errors.Wrapf(err, "创建扣税记录失败") + } + + return nil + }) + + if err != nil { + return nil, err + } + + return &types.ApplyWithdrawalResp{ + WithdrawalId: withdrawalId, + WithdrawalNo: withdrawNo, + }, nil +} + +// TaxInfo 税费信息 +type TaxInfo struct { + TaxableAmount float64 // 应税金额 + TaxRate float64 // 税率 + TaxAmount float64 // 税费金额 + ActualAmount float64 // 实际到账金额 +} + +// calculateTax 计算税费 +func (l *ApplyWithdrawalLogic) calculateTax(ctx context.Context, agentId string, amount float64, yearMonth int64) (*TaxInfo, error) { + // 获取税率配置（默认6%） + taxRate := 0.06 + config, err := l.svcCtx.AgentConfigModel.FindOneByConfigKey(ctx, "tax_rate") + if err == nil { + if parsedRate, parseErr := l.parseFloat(config.ConfigValue); parseErr == nil { + taxRate = parsedRate + } + } + + // 查询本月已提现金额 + builder := l.svcCtx.AgentWithdrawalTaxModel.SelectBuilder(). + Where("agent_id = ? AND year_month = ? AND del_state = ?", agentId, yearMonth, globalkey.DelStateNo) + taxRecords, err := l.svcCtx.AgentWithdrawalTaxModel.FindAll(ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(err, "查询月度提现记录失败") + } + + // 计算本月累计提现金额 + monthlyTotal := 0.0 + for _, record := range taxRecords { + monthlyTotal += record.WithdrawalAmount + } + + // 获取免税额度配置（默认0，即无免税额度） + exemptionAmount := 0.0 + exemptionConfig, err := l.svcCtx.AgentConfigModel.FindOneByConfigKey(ctx, "tax_exemption_amount") + if err == nil { + if parsedAmount, parseErr := l.parseFloat(exemptionConfig.ConfigValue); parseErr == nil { + exemptionAmount = parsedAmount + } + } + + // 计算应税金额 + // 如果本月累计 + 本次提现金额 <= 免税额度，则本次提现免税 + // 否则，应税金额 = 本次提现金额 - (免税额度 - 本月累计)（如果免税额度 > 本月累计） + taxableAmount := amount + if exemptionAmount > 0 { + remainingExemption := exemptionAmount - monthlyTotal + if remainingExemption > 0 { + if amount <= remainingExemption { + // 本次提现完全免税 + taxableAmount = 0 + } else { + // 部分免税 + taxableAmount = amount - remainingExemption + } + } + } + + // 计算税费 + taxAmount := taxableAmount * taxRate + actualAmount := amount - taxAmount + + return &TaxInfo{ + TaxableAmount: taxableAmount, + TaxRate: taxRate, + TaxAmount: taxAmount, + ActualAmount: actualAmount, + }, nil +} + +// parseFloat 解析浮点数 +func (l *ApplyWithdrawalLogic) parseFloat(s string) (float64, error) { + var result float64 + _, err := fmt.Sscanf(s, "%f", &result) + return result, err +} diff --git a/app/main/api/internal/logic/agent/deleteinvitecodelogic.go b/app/main/api/internal/logic/agent/deleteinvitecodelogic.go new file mode 100644 index 0000000..c795238 --- /dev/null +++ b/app/main/api/internal/logic/agent/deleteinvitecodelogic.go @@ -0,0 +1,72 @@ +package agent + +import ( + "context" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type DeleteInviteCodeLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewDeleteInviteCodeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DeleteInviteCodeLogic { + return &DeleteInviteCodeLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *DeleteInviteCodeLogic) DeleteInviteCode(req *types.DeleteInviteCodeReq) (resp *types.DeleteInviteCodeResp, err error) { + // 1. 获取当前代理信息 + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 查询邀请码是否存在且属于当前代理 + inviteCode, err := l.svcCtx.AgentInviteCodeModel.FindOne(l.ctx, req.Id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("邀请码不存在"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询邀请码失败, %v", err) + } + + // 3. 验证邀请码是否属于当前代理 + if !inviteCode.AgentId.Valid || inviteCode.AgentId.String != agent.Id { + return nil, errors.Wrapf(xerr.NewErrMsg("无权删除此邀请码"), "") + } + + // 4. 检查邀请码是否已被使用 + if inviteCode.Status == 1 { + return nil, errors.Wrapf(xerr.NewErrMsg("已使用的邀请码无法删除"), "") + } + + // 5. 执行软删除 + err = l.svcCtx.AgentInviteCodeModel.DeleteSoft(l.ctx, nil, inviteCode) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除邀请码失败, %v", err) + } + + return &types.DeleteInviteCodeResp{}, nil +} diff --git a/app/main/api/internal/logic/agent/generateinvitecodelogic.go b/app/main/api/internal/logic/agent/generateinvitecodelogic.go new file mode 100644 index 0000000..b7da486 --- /dev/null +++ b/app/main/api/internal/logic/agent/generateinvitecodelogic.go @@ -0,0 +1,115 @@ +package agent + +import ( + "context" + "database/sql" + "time" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/tool" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/sqlx" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GenerateInviteCodeLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGenerateInviteCodeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GenerateInviteCodeLogic { + return &GenerateInviteCodeLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GenerateInviteCodeLogic) GenerateInviteCode(req *types.GenerateInviteCodeReq) (resp *types.GenerateInviteCodeResp, err error) { + // 1. 获取当前代理信息 + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 验证生成数量 + if req.Count <= 0 || req.Count > 100 { + return nil, errors.Wrapf(xerr.NewErrMsg("生成数量必须在1-100之间"), "") + } + + // 3. 生成邀请码 + codes := make([]string, 0, req.Count) + var expireTime sql.NullTime + if req.ExpireDays > 0 { + expireTime = sql.NullTime{ + Time: time.Now().AddDate(0, 0, int(req.ExpireDays)), + Valid: true, + } + } + + err = l.svcCtx.AgentInviteCodeModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + for i := int64(0); i < req.Count; i++ { + // 生成8位随机邀请码（大小写字母+数字） + var code string + maxRetries := 10 // 最大重试次数 + for retry := 0; retry < maxRetries; retry++ { + code = tool.Krand(8, tool.KC_RAND_KIND_ALL) + // 检查邀请码是否已存在 + _, err := l.svcCtx.AgentInviteCodeModel.FindOneByCode(transCtx, code) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + // 邀请码不存在，可以使用 + break + } + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "检查邀请码失败, %v", err) + } + // 邀请码已存在，继续生成 + if retry == maxRetries-1 { + return errors.Wrapf(xerr.NewErrMsg("生成邀请码失败，请重试"), "") + } + } + + // 创建邀请码记录 + inviteCode := &model.AgentInviteCode{ + Code: code, + AgentId: sql.NullString{String: agent.Id, Valid: true}, + TargetLevel: 1, // 代理发放的邀请码，目标等级为普通代理 + Status: 0, // 未使用 + ExpireTime: expireTime, + Remark: sql.NullString{String: req.Remark, Valid: req.Remark != ""}, + } + + _, err := l.svcCtx.AgentInviteCodeModel.Insert(transCtx, session, inviteCode) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建邀请码失败, %v", err) + } + + codes = append(codes, code) + } + + return nil + }) + + if err != nil { + return nil, err + } + + return &types.GenerateInviteCodeResp{ + Codes: codes, + }, nil +} diff --git a/app/main/api/internal/logic/agent/generatinglinklogic.go b/app/main/api/internal/logic/agent/generatinglinklogic.go new file mode 100644 index 0000000..5e5d898 --- /dev/null +++ b/app/main/api/internal/logic/agent/generatinglinklogic.go @@ -0,0 +1,351 @@ +package agent + +import ( + "context" + "database/sql" + "encoding/hex" + "encoding/json" + "fmt" + "net/url" + "strconv" + "strings" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/tool" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/google/uuid" + "github.com/zeromicro/go-zero/core/logx" +) + +type GeneratingLinkLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGeneratingLinkLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GeneratingLinkLogic { + return &GeneratingLinkLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GeneratingLinkLogic) GeneratingLink(req *types.AgentGeneratingLinkReq) (resp *types.AgentGeneratingLinkResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成推广链接失败, %v", err) + } + + // 1. 获取代理信息 + agentModel, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 获取产品配置（必须存在） + productConfig, err := l.svcCtx.AgentProductConfigModel.FindOneByProductId(l.ctx, req.ProductId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "产品配置不存在, productId: %d，请先在后台配置产品价格参数", req.ProductId) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询产品配置失败, productId: %d, %v", req.ProductId, err) + } + + // 3. 获取等级加成配置 + levelBonus, err := l.getLevelBonus(agentModel.Level) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取等级加成配置失败, %v", err) + } + + basePrice := productConfig.BasePrice + actualBasePrice := basePrice + float64(levelBonus) + systemMaxPrice := productConfig.SystemMaxPrice + upliftAmount, err := l.getLevelMaxUpliftAmount(agentModel.Level) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取等级上调金额失败, %v", err) + } + levelMaxPrice := systemMaxPrice + upliftAmount + if req.SetPrice < actualBasePrice || req.SetPrice > levelMaxPrice { + return nil, errors.Wrapf(xerr.NewErrMsg("设定价格必须在 %.2f 到 %.2f 之间"), "设定价格必须在 %.2f 到 %.2f 之间", actualBasePrice, levelMaxPrice) + } + + // 6. 检查是否已存在相同的链接（同一代理、同一产品、同一价格） + builder := l.svcCtx.AgentLinkModel.SelectBuilder().Where(squirrel.And{ + squirrel.Eq{"agent_id": agentModel.Id}, + squirrel.Eq{"product_id": req.ProductId}, + squirrel.Eq{"set_price": req.SetPrice}, + squirrel.Eq{"del_state": globalkey.DelStateNo}, + }) + + existingLinks, err := l.svcCtx.AgentLinkModel.FindAll(l.ctx, builder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询推广链接失败, %v", err) + } + + if len(existingLinks) > 0 { + // 已存在，检查是否有短链，如果没有则生成 + targetPath := req.TargetPath + if targetPath == "" { + targetPath = "/agent/promotionInquire/" + } + shortLink, err := l.getOrCreateShortLink(1, existingLinks[0].Id, "", existingLinks[0].LinkIdentifier, "", targetPath) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取或创建短链失败, %v", err) + } + return &types.AgentGeneratingLinkResp{ + LinkIdentifier: existingLinks[0].LinkIdentifier, + FullLink: shortLink, + }, nil + } + + // 7. 生成推广链接标识 + var agentIdentifier types.AgentIdentifier + agentIdentifier.AgentID = agentModel.Id + agentIdentifier.ProductID = req.ProductId + agentIdentifier.SetPrice = req.SetPrice + agentIdentifierByte, err := json.Marshal(agentIdentifier) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "序列化标识失败, %v", err) + } + + // 8. 加密链接标识 + secretKey := l.svcCtx.Config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取AES密钥失败: %+v", decodeErr) + } + + encrypted, err := crypto.AesEncryptURL(agentIdentifierByte, key) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "加密链接标识失败, %v", err) + } + + // 9. 保存推广链接 + agentLink := &model.AgentLink{ + // Id: uuid.NewString(), + AgentId: agentModel.Id, + UserId: userID, + ProductId: req.ProductId, + LinkIdentifier: encrypted, + SetPrice: req.SetPrice, + ActualBasePrice: actualBasePrice, + } + + _, err = l.svcCtx.AgentLinkModel.Insert(l.ctx, nil, agentLink) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "保存推广链接失败, %v", err) + } + + linkId := agentLink.Id + + // 使用默认target_path（如果未提供） + targetPath := req.TargetPath + if targetPath == "" { + targetPath = "/agent/promotionInquire/" + } + + // 生成短链（类型：1=推广报告） + shortLink, err := l.createShortLink(1, linkId, "", encrypted, "", targetPath) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成短链失败, %v", err) + } + + return &types.AgentGeneratingLinkResp{ + LinkIdentifier: encrypted, + FullLink: shortLink, + }, nil +} + +// getLevelBonus 获取等级加成（从配置表读取） +func (l *GeneratingLinkLogic) getLevelBonus(level int64) (int64, error) { + var configKey string + switch level { + case 1: // 普通 + configKey = "level_1_bonus" + case 2: // 黄金 + configKey = "level_2_bonus" + case 3: // 钻石 + configKey = "level_3_bonus" + default: + return 0, nil + } + + config, err := l.svcCtx.AgentConfigModel.FindOneByConfigKey(l.ctx, configKey) + if err != nil { + // 配置不存在时返回默认值 + l.Errorf("获取等级加成配置失败, level: %d, key: %s, err: %v，使用默认值", level, configKey, err) + switch level { + case 1: + return 6, nil + case 2: + return 3, nil + case 3: + return 0, nil + } + return 0, nil + } + + value, err := strconv.ParseFloat(config.ConfigValue, 64) + if err != nil { + return 0, errors.Wrapf(err, "解析配置值失败, key: %s, value: %s", configKey, config.ConfigValue) + } + return int64(value), nil +} + +func (l *GeneratingLinkLogic) getLevelMaxUpliftAmount(level int64) (float64, error) { + var key string + switch level { + case 2: + key = "gold_max_uplift_amount" + case 3: + key = "diamond_max_uplift_amount" + default: + return 0, nil + } + config, err := l.svcCtx.AgentConfigModel.FindOneByConfigKey(l.ctx, key) + if err != nil { + return 0, nil + } + v, err := strconv.ParseFloat(config.ConfigValue, 64) + if err != nil { + return 0, nil + } + if v < 0 { + return 0, nil + } + return v, nil +} + +// getOrCreateShortLink 获取或创建短链 +// type: 1=推广报告(promotion), 2=邀请好友(invite) +// linkId: 推广链接ID（仅推广报告使用） +// inviteCodeId: 邀请码ID（仅邀请好友使用） +// linkIdentifier: 推广链接标识（仅推广报告使用） +// inviteCode: 邀请码（仅邀请好友使用） +// targetPath: 目标地址（前端传入） +func (l *GeneratingLinkLogic) getOrCreateShortLink(linkType int64, linkId, inviteCodeId string, linkIdentifier, inviteCode, targetPath string) (string, error) { + // 先查询是否已存在短链 + var existingShortLink *model.AgentShortLink + var err error + + if linkType == 1 { + // 推广报告类型，使用link_id查询 + if linkId != "" { + existingShortLink, err = l.svcCtx.AgentShortLinkModel.FindOneByLinkIdTypeDelState(l.ctx, sql.NullString{String: linkId, Valid: true}, linkType, globalkey.DelStateNo) + } + } else { + // 邀请好友类型，使用invite_code_id查询 + if inviteCodeId != "" { + existingShortLink, err = l.svcCtx.AgentShortLinkModel.FindOneByInviteCodeIdTypeDelState(l.ctx, sql.NullString{String: inviteCodeId, Valid: true}, linkType, globalkey.DelStateNo) + } + } + + if err == nil && existingShortLink != nil { + // 已存在短链，直接返回 + return l.buildShortLinkURL(existingShortLink.ShortCode), nil + } + + if err != nil && !errors.Is(err, model.ErrNotFound) { + return "", errors.Wrapf(err, "查询短链失败") + } + + // 不存在，创建新的短链 + return l.createShortLink(linkType, linkId, inviteCodeId, linkIdentifier, inviteCode, targetPath) +} + +// createShortLink 创建短链 +// type: 1=推广报告(promotion), 2=邀请好友(invite) +func (l *GeneratingLinkLogic) createShortLink(linkType int64, linkId, inviteCodeId string, linkIdentifier, inviteCode, targetPath string) (string, error) { + promotionConfig := l.svcCtx.Config.Promotion + + // 如果没有配置推广域名，返回空字符串（保持向后兼容） + if promotionConfig.PromotionDomain == "" { + l.Errorf("推广域名未配置，返回空链接") + return "", nil + } + + // 验证target_path + if targetPath == "" { + return "", errors.Wrapf(xerr.NewErrMsg("目标地址不能为空"), "") + } + + // 对于推广报告类型，将 linkIdentifier 拼接到 target_path + if linkType == 1 && linkIdentifier != "" { + // 如果 target_path 以 / 结尾，直接拼接 linkIdentifier + if strings.HasSuffix(targetPath, "/") { + targetPath = targetPath + url.QueryEscape(linkIdentifier) + } else { + // 否则在末尾添加 / 再拼接 + targetPath = targetPath + "/" + url.QueryEscape(linkIdentifier) + } + } + + // 生成短链标识（6位随机字符串，大小写字母+数字） + var shortCode string + maxRetries := 10 // 最大重试次数 + for retry := 0; retry < maxRetries; retry++ { + shortCode = tool.Krand(6, tool.KC_RAND_KIND_ALL) + // 检查短链标识是否已存在 + _, err := l.svcCtx.AgentShortLinkModel.FindOneByShortCodeDelState(l.ctx, shortCode, globalkey.DelStateNo) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + // 短链标识不存在，可以使用 + break + } + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "检查短链标识失败, %v", err) + } + // 短链标识已存在，继续生成 + if retry == maxRetries-1 { + return "", errors.Wrapf(xerr.NewErrMsg("生成短链失败，请重试"), "") + } + } + + // 创建短链记录 + shortLink := &model.AgentShortLink{ + Id: uuid.NewString(), + Type: linkType, + ShortCode: shortCode, + TargetPath: targetPath, + PromotionDomain: promotionConfig.PromotionDomain, + } + + // 根据类型设置对应字段 + if linkType == 1 { + // 推广报告类型 + shortLink.LinkId = sql.NullString{String: linkId, Valid: linkId != ""} + if linkIdentifier != "" { + shortLink.LinkIdentifier = sql.NullString{String: linkIdentifier, Valid: true} + } + } else if linkType == 2 { + // 邀请好友类型 + shortLink.InviteCodeId = sql.NullString{String: inviteCodeId, Valid: inviteCodeId != ""} + if inviteCode != "" { + shortLink.InviteCode = sql.NullString{String: inviteCode, Valid: true} + } + } + _, err := l.svcCtx.AgentShortLinkModel.Insert(l.ctx, nil, shortLink) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "保存短链失败, %v", err) + } + + return l.buildShortLinkURL(shortCode), nil +} + +// buildShortLinkURL 构建短链URL +func (l *GeneratingLinkLogic) buildShortLinkURL(shortCode string) string { + promotionConfig := l.svcCtx.Config.Promotion + return fmt.Sprintf("%s/s/%s", promotionConfig.PromotionDomain, shortCode) +} diff --git a/app/main/api/internal/logic/agent/getagentinfologic.go b/app/main/api/internal/logic/agent/getagentinfologic.go new file mode 100644 index 0000000..9a2378e --- /dev/null +++ b/app/main/api/internal/logic/agent/getagentinfologic.go @@ -0,0 +1,115 @@ +package agent + +import ( + "context" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetAgentInfoLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetAgentInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetAgentInfoLogic { + return &GetAgentInfoLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetAgentInfoLogic) GetAgentInfo() (resp *types.AgentInfoResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + // 不是代理，返回空信息 + return &types.AgentInfoResp{ + AgentId: "", + Level: 0, + LevelName: "", + Region: "", + Mobile: "", + WechatId: "", + TeamLeaderId: "", + IsRealName: false, + }, nil + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 解密手机号 + mobile, err := crypto.DecryptMobile(agent.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "解密手机号失败: %v", err) + } + + // 查询实名认证状态 + isRealName := false + agentRealName, err := l.svcCtx.AgentRealNameModel.FindOneByAgentId(l.ctx, agent.Id) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询实名认证失败, %v", err) + } + if agentRealName != nil && agentRealName.VerifyTime.Valid { // verify_time不为空表示已通过三要素核验 + isRealName = true + } + + // 获取微信号 + wechatId := "" + if agent.WechatId.Valid { + wechatId = agent.WechatId.String + } + + // 获取团队首领ID + teamLeaderId := "" + if agent.TeamLeaderId.Valid { + teamLeaderId = agent.TeamLeaderId.String + } + + // 获取区域 + region := "" + if agent.Region.Valid { + region = agent.Region.String + } + + return &types.AgentInfoResp{ + AgentId: agent.Id, + Level: agent.Level, + LevelName: l.getLevelName(agent.Level), + Region: region, + Mobile: mobile, + WechatId: wechatId, + TeamLeaderId: teamLeaderId, + IsRealName: isRealName, + AgentCode: agent.AgentCode, + }, nil +} + +// getLevelName 获取等级名称 +func (l *GetAgentInfoLogic) getLevelName(level int64) string { + switch level { + case 1: + return "普通" + case 2: + return "黄金" + case 3: + return "钻石" + default: + return "" + } +} diff --git a/app/main/api/internal/logic/agent/getagentproductconfiglogic.go b/app/main/api/internal/logic/agent/getagentproductconfiglogic.go new file mode 100644 index 0000000..5b45de9 --- /dev/null +++ b/app/main/api/internal/logic/agent/getagentproductconfiglogic.go @@ -0,0 +1,170 @@ +package agent + +import ( + "context" + "strconv" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetAgentProductConfigLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetAgentProductConfigLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetAgentProductConfigLogic { + return &GetAgentProductConfigLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetAgentProductConfigLogic) GetAgentProductConfig() (resp *types.AgentProductConfigResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agentModel, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 获取等级加成配置（从系统配置表读取） + levelBonus, err := l.getLevelBonus(agentModel.Level) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取等级加成配置失败, %v", err) + } + + // 3. 查询所有产品配置 + builder := l.svcCtx.AgentProductConfigModel.SelectBuilder() + productConfigs, err := l.svcCtx.AgentProductConfigModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询产品配置失败, %v", err) + } + + // 4. 组装响应数据（通过 product_id 查询产品名称和英文标识） + var respList []types.ProductConfigItem + for _, productConfig := range productConfigs { + // 通过 product_id 查询产品信息获取产品名称和英文标识 + product, err := l.svcCtx.ProductModel.FindOne(l.ctx, productConfig.ProductId) + productName := "" + productEn := "" + if err == nil { + productName = product.ProductName + productEn = product.ProductEn + } else { + // 如果产品不存在，记录日志但不影响主流程 + l.Infof("查询产品信息失败, productId: %d, err: %v", productConfig.ProductId, err) + } + + // 使用产品配置的基础底价（必须配置） + productBasePrice := productConfig.BasePrice + + // 计算该产品的实际底价 + productActualBasePrice := productBasePrice + float64(levelBonus) + + priceRangeMin := productActualBasePrice + upliftAmount, _ := l.getLevelMaxUpliftAmount(agentModel.Level) + priceRangeMax := productConfig.SystemMaxPrice + upliftAmount + + // 使用产品配置的提价阈值和手续费比例，如果为NULL则使用0 + productPriceThreshold := 0.0 + if productConfig.PriceThreshold.Valid { + productPriceThreshold = productConfig.PriceThreshold.Float64 + } + productPriceFeeRate := 0.0 + if productConfig.PriceFeeRate.Valid { + productPriceFeeRate = productConfig.PriceFeeRate.Float64 + } + + respList = append(respList, types.ProductConfigItem{ + ProductId: productConfig.ProductId, + ProductName: productName, + ProductEn: productEn, + ActualBasePrice: productActualBasePrice, + PriceRangeMin: priceRangeMin, + PriceRangeMax: priceRangeMax, + PriceThreshold: productPriceThreshold, + PriceFeeRate: productPriceFeeRate, + }) + } + + return &types.AgentProductConfigResp{ + List: respList, + }, nil +} + +// getLevelBonus 获取等级加成（从配置表读取） +func (l *GetAgentProductConfigLogic) getLevelBonus(level int64) (int64, error) { + var configKey string + switch level { + case 1: // 普通 + configKey = "level_1_bonus" + case 2: // 黄金 + configKey = "level_2_bonus" + case 3: // 钻石 + configKey = "level_3_bonus" + default: + return 0, nil + } + + config, err := l.svcCtx.AgentConfigModel.FindOneByConfigKey(l.ctx, configKey) + if err != nil { + // 配置不存在时返回默认值 + l.Errorf("获取等级加成配置失败, level: %d, key: %s, err: %v，使用默认值", level, configKey, err) + switch level { + case 1: + return 6, nil + case 2: + return 3, nil + case 3: + return 0, nil + } + return 0, nil + } + + value, err := strconv.ParseFloat(config.ConfigValue, 64) + if err != nil { + return 0, errors.Wrapf(err, "解析配置值失败, key: %s, value: %s", configKey, config.ConfigValue) + } + return int64(value), nil +} + +func (l *GetAgentProductConfigLogic) getLevelMaxUpliftAmount(level int64) (float64, error) { + var key string + switch level { + case 2: + key = "gold_max_uplift_amount" + case 3: + key = "diamond_max_uplift_amount" + default: + return 0, nil + } + config, err := l.svcCtx.AgentConfigModel.FindOneByConfigKey(l.ctx, key) + if err != nil { + return 0, nil + } + v, err := strconv.ParseFloat(config.ConfigValue, 64) + if err != nil { + return 0, nil + } + if v < 0 { + return 0, nil + } + return v, nil +} diff --git a/app/main/api/internal/logic/agent/getcommissionlistlogic.go b/app/main/api/internal/logic/agent/getcommissionlistlogic.go new file mode 100644 index 0000000..9a6f772 --- /dev/null +++ b/app/main/api/internal/logic/agent/getcommissionlistlogic.go @@ -0,0 +1,112 @@ +package agent + +import ( + "context" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetCommissionListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetCommissionListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetCommissionListLogic { + return &GetCommissionListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetCommissionListLogic) GetCommissionList(req *types.GetCommissionListReq) (resp *types.GetCommissionListResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 构建查询条件 + builder := l.svcCtx.AgentCommissionModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", agent.Id, globalkey.DelStateNo). + OrderBy("create_time DESC") + + // 3. 分页查询 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + offset := (page - 1) * pageSize + + // 4. 查询总数 + total, err := l.svcCtx.AgentCommissionModel.FindCount(l.ctx, builder, "id") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询佣金总数失败, %v", err) + } + + // 5. 查询列表 + builder = builder.Limit(uint64(pageSize)).Offset(uint64(offset)) + commissions, err := l.svcCtx.AgentCommissionModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询佣金列表失败, %v", err) + } + + // 6. 组装响应 + var list []types.CommissionItem + for _, commission := range commissions { + // 查询产品名称 + productName := "" + if commission.ProductId != "" { + product, err := l.svcCtx.ProductModel.FindOne(l.ctx, commission.ProductId) + if err == nil { + productName = product.ProductName + } + } + + // 查询订单号 + orderNo := "" + if commission.OrderId != "" { + order, err := l.svcCtx.OrderModel.FindOne(l.ctx, commission.OrderId) + if err == nil { + orderNo = order.OrderNo + } + } + + list = append(list, types.CommissionItem{ + Id: commission.Id, + OrderId: commission.OrderId, + OrderNo: orderNo, + ProductName: productName, + Amount: commission.Amount, + Status: commission.Status, + CreateTime: commission.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return &types.GetCommissionListResp{ + Total: total, + List: list, + }, nil +} diff --git a/app/main/api/internal/logic/agent/getconversionratelogic.go b/app/main/api/internal/logic/agent/getconversionratelogic.go new file mode 100644 index 0000000..446dd35 --- /dev/null +++ b/app/main/api/internal/logic/agent/getconversionratelogic.go @@ -0,0 +1,492 @@ +package agent + +import ( + "context" + "time" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetConversionRateLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetConversionRateLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetConversionRateLogic { + return &GetConversionRateLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetConversionRateLogic) GetConversionRate() (resp *types.ConversionRateResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 统计我的转化率 + myConversionRate := l.calculateConversionRate(agent.Id, nil) + + // 3. 统计下级转化率（按时间段动态查询历史下级关系） + subordinateConversionRate := l.calculateSubordinateConversionRate(agent.Id) + + return &types.ConversionRateResp{ + MyConversionRate: myConversionRate, + SubordinateConversionRate: subordinateConversionRate, + }, nil +} + +// calculateSubordinateConversionRate 计算下级转化率（考虑历史关系） +func (l *GetConversionRateLogic) calculateSubordinateConversionRate(parentAgentId string) types.ConversionRateData { + // 使用Asia/Shanghai时区，与数据库保持一致 + loc, _ := time.LoadLocation("Asia/Shanghai") + now := time.Now().In(loc) + + // 日统计：今日、昨日、前日 + todayStart := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, loc) + todayEnd := todayStart.AddDate(0, 0, 1) + yesterdayStart := todayStart.AddDate(0, 0, -1) + yesterdayEnd := todayStart + dayBeforeStart := todayStart.AddDate(0, 0, -2) + dayBeforeEnd := yesterdayStart + + daily := []types.PeriodConversionData{ + l.calculateSubordinatePeriodConversion(parentAgentId, "今日", todayStart, todayEnd), + l.calculateSubordinatePeriodConversion(parentAgentId, "昨日", yesterdayStart, yesterdayEnd), + l.calculateSubordinatePeriodConversion(parentAgentId, "前日", dayBeforeStart, dayBeforeEnd), + } + + // 周统计：本周、上周、上上周 + weekdayOffset := int(now.Weekday()) + if weekdayOffset == 0 { + weekdayOffset = 7 // 周日算作第7天 + } + thisWeekStart := now.AddDate(0, 0, -(weekdayOffset - 1)) + thisWeekStart = time.Date(thisWeekStart.Year(), thisWeekStart.Month(), thisWeekStart.Day(), 0, 0, 0, 0, loc) + thisWeekEnd := now + lastWeekStart := thisWeekStart.AddDate(0, 0, -7) + lastWeekEnd := thisWeekStart + lastTwoWeekStart := thisWeekStart.AddDate(0, 0, -14) + lastTwoWeekEnd := lastWeekStart + + weekly := []types.PeriodConversionData{ + l.calculateSubordinatePeriodConversion(parentAgentId, "本周", thisWeekStart, thisWeekEnd), + l.calculateSubordinatePeriodConversion(parentAgentId, "上周", lastWeekStart, lastWeekEnd), + l.calculateSubordinatePeriodConversion(parentAgentId, "上上周", lastTwoWeekStart, lastTwoWeekEnd), + } + + // 月统计：本月、上月、前两月 + thisMonthStart := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, loc) + thisMonthEnd := now + lastMonthStart := thisMonthStart.AddDate(0, -1, 0) + lastMonthEnd := thisMonthStart + lastTwoMonthStart := thisMonthStart.AddDate(0, -2, 0) + lastTwoMonthEnd := lastMonthStart + + monthly := []types.PeriodConversionData{ + l.calculateSubordinatePeriodConversion(parentAgentId, "本月", thisMonthStart, thisMonthEnd), + l.calculateSubordinatePeriodConversion(parentAgentId, "上月", lastMonthStart, lastMonthEnd), + l.calculateSubordinatePeriodConversion(parentAgentId, "前两月", lastTwoMonthStart, lastTwoMonthEnd), + } + + return types.ConversionRateData{ + Daily: daily, + Weekly: weekly, + Monthly: monthly, + } +} + +// calculateConversionRate 计算转化率 +// agentId > 0 时统计该代理的转化率，否则统计 subordinateIds 列表的转化率 +func (l *GetConversionRateLogic) calculateConversionRate(agentId string, subordinateIds []string) types.ConversionRateData { + // 使用Asia/Shanghai时区，与数据库保持一致 + loc, _ := time.LoadLocation("Asia/Shanghai") + now := time.Now().In(loc) + + // 日统计：今日、昨日、前日 + todayStart := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, loc) + todayEnd := todayStart.AddDate(0, 0, 1) + yesterdayStart := todayStart.AddDate(0, 0, -1) + yesterdayEnd := todayStart + dayBeforeStart := todayStart.AddDate(0, 0, -2) + dayBeforeEnd := yesterdayStart + + daily := []types.PeriodConversionData{ + l.calculatePeriodConversion(agentId, subordinateIds, "今日", todayStart, todayEnd), + l.calculatePeriodConversion(agentId, subordinateIds, "昨日", yesterdayStart, yesterdayEnd), + l.calculatePeriodConversion(agentId, subordinateIds, "前日", dayBeforeStart, dayBeforeEnd), + } + + // 周统计：本周、上周、上上周 + // 本周：本周一00:00:00 到现在 + weekdayOffset := int(now.Weekday()) + if weekdayOffset == 0 { + weekdayOffset = 7 // 周日算作第7天 + } + // 计算本周一：当前日期减去(weekdayOffset-1)天 + thisWeekStart := now.AddDate(0, 0, -(weekdayOffset - 1)) + thisWeekStart = time.Date(thisWeekStart.Year(), thisWeekStart.Month(), thisWeekStart.Day(), 0, 0, 0, 0, loc) + thisWeekEnd := now + lastWeekStart := thisWeekStart.AddDate(0, 0, -7) + lastWeekEnd := thisWeekStart + lastTwoWeekStart := thisWeekStart.AddDate(0, 0, -14) + lastTwoWeekEnd := lastWeekStart + + weekly := []types.PeriodConversionData{ + l.calculatePeriodConversion(agentId, subordinateIds, "本周", thisWeekStart, thisWeekEnd), + l.calculatePeriodConversion(agentId, subordinateIds, "上周", lastWeekStart, lastWeekEnd), + l.calculatePeriodConversion(agentId, subordinateIds, "上上周", lastTwoWeekStart, lastTwoWeekEnd), + } + + // 月统计：本月、上月、前两月 + thisMonthStart := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, loc) + thisMonthEnd := now + lastMonthStart := thisMonthStart.AddDate(0, -1, 0) + lastMonthEnd := thisMonthStart + lastTwoMonthStart := thisMonthStart.AddDate(0, -2, 0) + lastTwoMonthEnd := lastMonthStart + + monthly := []types.PeriodConversionData{ + l.calculatePeriodConversion(agentId, subordinateIds, "本月", thisMonthStart, thisMonthEnd), + l.calculatePeriodConversion(agentId, subordinateIds, "上月", lastMonthStart, lastMonthEnd), + l.calculatePeriodConversion(agentId, subordinateIds, "前两月", lastTwoMonthStart, lastTwoMonthEnd), + } + + return types.ConversionRateData{ + Daily: daily, + Weekly: weekly, + Monthly: monthly, + } +} + +// calculatePeriodConversion 计算指定时间段的转化率数据 +func (l *GetConversionRateLogic) calculatePeriodConversion(agentId string, subordinateIds []string, periodLabel string, startTime, endTime time.Time) types.PeriodConversionData { + // 构建 agent_order 查询条件 + agentOrderBuilder := l.svcCtx.AgentOrderModel.SelectBuilder(). + Where("del_state = ?", globalkey.DelStateNo). + Where("create_time >= ? AND create_time < ?", startTime, endTime) + + if agentId != "" { + // 统计我的转化率 + agentOrderBuilder = agentOrderBuilder.Where("agent_id = ?", agentId) + } else if len(subordinateIds) > 0 { + // 统计下级转化率 + agentOrderBuilder = agentOrderBuilder.Where(squirrel.Eq{"agent_id": subordinateIds}) + } else { + // 没有数据 + l.Infof("calculatePeriodConversion: 没有代理ID或下级ID，periodLabel=%s", periodLabel) + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: 0, + PaidUsers: 0, + TotalAmount: 0, + QueryUserCount: 0, + PaidUserCount: 0, + } + } + + // 添加调试日志 + if agentId == "" && len(subordinateIds) > 0 { + l.Infof("calculatePeriodConversion: 统计下级转化率，periodLabel=%s, startTime=%v, endTime=%v, subordinateIds数量=%d", + periodLabel, startTime, endTime, len(subordinateIds)) + } + + // 查询所有代理订单 + agentOrders, err := l.svcCtx.AgentOrderModel.FindAll(l.ctx, agentOrderBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + l.Errorf("查询代理订单失败: periodLabel=%s, err=%v", periodLabel, err) + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: 0, + PaidUsers: 0, + TotalAmount: 0, + QueryUserCount: 0, + PaidUserCount: 0, + } + } + + if len(agentOrders) == 0 { + if agentId == "" && len(subordinateIds) > 0 { + l.Infof("calculatePeriodConversion: 未找到代理订单，periodLabel=%s, startTime=%v, endTime=%v", + periodLabel, startTime, endTime) + } + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: 0, + PaidUsers: 0, + TotalAmount: 0, + QueryUserCount: 0, + PaidUserCount: 0, + } + } + + l.Infof("calculatePeriodConversion: 找到代理订单数量=%d, periodLabel=%s", len(agentOrders), periodLabel) + + // 收集订单ID + orderIds := make([]string, 0, len(agentOrders)) + for _, ao := range agentOrders { + orderIds = append(orderIds, ao.OrderId) + } + + // 查询订单信息 + orderBuilder := l.svcCtx.OrderModel.SelectBuilder(). + Where(squirrel.Eq{"id": orderIds}). + Where("del_state = ?", globalkey.DelStateNo) + + orders, err := l.svcCtx.OrderModel.FindAll(l.ctx, orderBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + l.Errorf("查询订单失败: %v", err) + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: 0, + PaidUsers: 0, + TotalAmount: 0, + QueryUserCount: 0, + PaidUserCount: 0, + } + } + + // 统计查询订单数、付费订单数、用户数和总金额 + var totalAmount float64 + paidOrderCount := 0 + queryUserSet := make(map[string]bool) + paidUserSet := make(map[string]bool) + + for _, order := range orders { + // 查询用户数（所有订单的用户，去重） + queryUserSet[order.UserId] = true + + // 付费订单数和总金额（只统计已支付的订单） + if order.Status == "paid" { + paidOrderCount++ + paidUserSet[order.UserId] = true + totalAmount += order.Amount + } + } + + // 查询订单数 = 所有订单数量 + queryOrderCount := len(orders) + + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: int64(queryOrderCount), // 订单数量（保持向后兼容） + PaidUsers: int64(paidOrderCount), // 订单数量（保持向后兼容） + TotalAmount: totalAmount, + QueryUserCount: int64(len(queryUserSet)), // 用户数量（新增） + PaidUserCount: int64(len(paidUserSet)), // 用户数量（新增） + } +} + +// calculateSubordinatePeriodConversion 计算指定时间段内下级转化率 +// 结合使用agent_rebate表和agent_order表： +// 1. 查询量：通过agent_order表统计所有查询（包括未付费的） +// 2. 付费量和金额：通过agent_rebate表统计（只有付费的订单才会产生返佣） +func (l *GetConversionRateLogic) calculateSubordinatePeriodConversion(parentAgentId string, periodLabel string, startTime, endTime time.Time) types.PeriodConversionData { + // 1. 查询agent_rebate表：获取所有曾经给当前用户产生返佣的source_agent_id（这些代理在某个时间点是下级） + // 不限制时间，获取所有历史返佣记录，用于确定哪些代理曾经是下级 + rebateBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", parentAgentId, globalkey.DelStateNo). + Where("source_agent_id != ?", parentAgentId) + + allRebates, err := l.svcCtx.AgentRebateModel.FindAll(l.ctx, rebateBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + l.Errorf("查询返佣记录失败: periodLabel=%s, err=%v", periodLabel, err) + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: 0, + PaidUsers: 0, + TotalAmount: 0, + QueryUserCount: 0, + PaidUserCount: 0, + } + } + + // 收集所有曾经产生返佣的source_agent_id（这些代理在某个时间点是下级） + sourceAgentIdSet := make(map[string]bool) + paidOrderIdSet := make(map[string]bool) // 已付费的订单ID（有返佣的订单） + paidOrderIdToAmount := make(map[string]float64) // 已付费订单的金额 + + for _, rebate := range allRebates { + sourceAgentIdSet[rebate.SourceAgentId] = true + // 如果返佣记录的创建时间在时间段内，说明该订单在时间段内已付费 + if rebate.CreateTime.After(startTime) || rebate.CreateTime.Equal(startTime) { + if rebate.CreateTime.Before(endTime) { + paidOrderIdSet[rebate.OrderId] = true + } + } + } + + if len(sourceAgentIdSet) == 0 { + l.Infof("calculateSubordinatePeriodConversion: 未找到返佣记录，periodLabel=%s, startTime=%v, endTime=%v", + periodLabel, startTime, endTime) + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: 0, + PaidUsers: 0, + TotalAmount: 0, + QueryUserCount: 0, + PaidUserCount: 0, + } + } + + sourceAgentIds := make([]string, 0, len(sourceAgentIdSet)) + for agentId := range sourceAgentIdSet { + sourceAgentIds = append(sourceAgentIds, agentId) + } + + l.Infof("calculateSubordinatePeriodConversion: periodLabel=%s, 曾经产生返佣的代理数量=%d", periodLabel, len(sourceAgentIds)) + + // 2. 查询agent_order表：统计这些代理在时间段内的所有订单（包括未付费的） + agentOrderBuilder := l.svcCtx.AgentOrderModel.SelectBuilder(). + Where("del_state = ?", globalkey.DelStateNo). + Where("create_time >= ? AND create_time < ?", startTime, endTime). + Where(squirrel.Eq{"agent_id": sourceAgentIds}) + + agentOrders, err := l.svcCtx.AgentOrderModel.FindAll(l.ctx, agentOrderBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + l.Errorf("查询代理订单失败: periodLabel=%s, err=%v", periodLabel, err) + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: 0, + PaidUsers: 0, + TotalAmount: 0, + QueryUserCount: 0, + PaidUserCount: 0, + } + } + + l.Infof("calculateSubordinatePeriodConversion: periodLabel=%s, 查询到代理订单数量=%d", periodLabel, len(agentOrders)) + + if len(agentOrders) == 0 { + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: 0, + PaidUsers: 0, + TotalAmount: 0, + QueryUserCount: 0, + PaidUserCount: 0, + } + } + + // 3. 通过order_id去重，获取所有订单ID（用于查询订单详情） + orderIdSet := make(map[string]bool) + orderIdToAgentOrder := make(map[string]*model.AgentOrder) + + for _, ao := range agentOrders { + orderIdSet[ao.OrderId] = true + // 如果同一个订单有多个agent_order记录，保留金额更大的 + if existing, exists := orderIdToAgentOrder[ao.OrderId]; exists { + if ao.OrderAmount > existing.OrderAmount { + orderIdToAgentOrder[ao.OrderId] = ao + } + } else { + orderIdToAgentOrder[ao.OrderId] = ao + } + } + + orderIds := make([]string, 0, len(orderIdSet)) + for orderId := range orderIdSet { + orderIds = append(orderIds, orderId) + } + + // 4. 查询订单信息 + orderBuilder := l.svcCtx.OrderModel.SelectBuilder(). + Where(squirrel.Eq{"id": orderIds}). + Where("del_state = ?", globalkey.DelStateNo) + + orders, err := l.svcCtx.OrderModel.FindAll(l.ctx, orderBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + l.Errorf("查询订单失败: %v", err) + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: 0, + PaidUsers: 0, + TotalAmount: 0, + QueryUserCount: 0, + PaidUserCount: 0, + } + } + + // 5. 查询时间段内的返佣记录，获取已付费订单的金额 + rebateBuilder = l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", parentAgentId, globalkey.DelStateNo). + Where("source_agent_id != ?", parentAgentId). + Where("create_time >= ? AND create_time < ?", startTime, endTime). + Where(squirrel.Eq{"order_id": orderIds}) + + rebates, err := l.svcCtx.AgentRebateModel.FindAll(l.ctx, rebateBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + l.Errorf("查询返佣记录失败: %v", err) + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: 0, + PaidUsers: 0, + TotalAmount: 0, + QueryUserCount: 0, + PaidUserCount: 0, + } + } + + // 记录已付费订单的金额（使用agent_order的order_amount） + for _, rebate := range rebates { + if ao, exists := orderIdToAgentOrder[rebate.OrderId]; exists { + paidOrderIdToAmount[rebate.OrderId] = ao.OrderAmount + } + } + + // 6. 统计查询订单数、付费订单数、用户数和总金额 + var totalAmount float64 + paidOrderCount := 0 + queryUserSet := make(map[string]bool) + paidUserSet := make(map[string]bool) + + for _, order := range orders { + // 查询用户数（所有订单的用户，去重） + queryUserSet[order.UserId] = true + + // 付费订单数和总金额（只统计已付费的订单，即order_id在paidOrderIdToAmount中） + if _, isPaid := paidOrderIdToAmount[order.Id]; isPaid { + paidOrderCount++ + paidUserSet[order.UserId] = true + // 使用agent_order的order_amount（用户实际支付金额） + totalAmount += paidOrderIdToAmount[order.Id] + } + } + + // 查询订单数 = 所有订单数量 + queryOrderCount := len(orders) + + l.Infof("calculateSubordinatePeriodConversion: periodLabel=%s, 查询订单数=%d, 付费订单数=%d, 查询用户数=%d, 付费用户数=%d, 总金额=%.2f", + periodLabel, queryOrderCount, paidOrderCount, len(queryUserSet), len(paidUserSet), totalAmount) + + return types.PeriodConversionData{ + PeriodLabel: periodLabel, + QueryUsers: int64(queryOrderCount), // 订单数量（保持向后兼容） + PaidUsers: int64(paidOrderCount), // 订单数量（保持向后兼容） + TotalAmount: totalAmount, + QueryUserCount: int64(len(queryUserSet)), // 用户数量（新增） + PaidUserCount: int64(len(paidUserSet)), // 用户数量（新增） + } +} diff --git a/app/main/api/internal/logic/agent/getinvitecodelistlogic.go b/app/main/api/internal/logic/agent/getinvitecodelistlogic.go new file mode 100644 index 0000000..362fee5 --- /dev/null +++ b/app/main/api/internal/logic/agent/getinvitecodelistlogic.go @@ -0,0 +1,89 @@ +package agent + +import ( + "context" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetInviteCodeListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetInviteCodeListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetInviteCodeListLogic { + return &GetInviteCodeListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetInviteCodeListLogic) GetInviteCodeList(req *types.GetInviteCodeListReq) (resp *types.GetInviteCodeListResp, err error) { + // 1. 获取当前代理信息 + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 构建查询条件 + builder := l.svcCtx.AgentInviteCodeModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", agent.Id, globalkey.DelStateNo) + + if req.Status >= 0 { + builder = builder.Where("status = ?", req.Status) + } + + // 3. 分页查询 + list, total, err := l.svcCtx.AgentInviteCodeModel.FindPageListByPageWithTotal(l.ctx, builder, req.Page, req.PageSize, "create_time DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询邀请码列表失败, %v", err) + } + + // 4. 格式化返回数据 + items := make([]types.InviteCodeItem, 0, len(list)) + for _, v := range list { + item := types.InviteCodeItem{ + Id: v.Id, + Code: v.Code, + TargetLevel: v.TargetLevel, + Status: v.Status, + CreateTime: v.CreateTime.Format("2006-01-02 15:04:05"), + } + + if v.UsedTime.Valid { + item.UsedTime = v.UsedTime.Time.Format("2006-01-02 15:04:05") + } + if v.ExpireTime.Valid { + item.ExpireTime = v.ExpireTime.Time.Format("2006-01-02 15:04:05") + } + if v.Remark.Valid { + item.Remark = v.Remark.String + } + + items = append(items, item) + } + + return &types.GetInviteCodeListResp{ + Total: total, + List: items, + }, nil +} diff --git a/app/main/api/internal/logic/agent/getinvitelinklogic.go b/app/main/api/internal/logic/agent/getinvitelinklogic.go new file mode 100644 index 0000000..645bc60 --- /dev/null +++ b/app/main/api/internal/logic/agent/getinvitelinklogic.go @@ -0,0 +1,182 @@ +package agent + +import ( + "context" + "database/sql" + "fmt" + "strconv" + "strings" + "time" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/tool" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/google/uuid" + "github.com/zeromicro/go-zero/core/logx" +) + +type GetInviteLinkLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetInviteLinkLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetInviteLinkLogic { + return &GetInviteLinkLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetInviteLinkLogic) GetInviteLink(req *types.GetInviteLinkReq) (resp *types.GetInviteLinkResp, err error) { + // 1. 获取当前代理信息 + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 验证邀请码参数 + if req.InviteCode == "" { + return nil, errors.Wrapf(xerr.NewErrMsg("邀请码不能为空"), "") + } + + ref := strings.TrimSpace(req.InviteCode) + var inviteCodeRecord *model.AgentInviteCode + inviteCodeRecord, err = l.svcCtx.AgentInviteCodeModel.FindOneByCode(l.ctx, ref) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询邀请码失败, %v", err) + } + + if inviteCodeRecord != nil { + if !inviteCodeRecord.AgentId.Valid || inviteCodeRecord.AgentId.String != agent.Id { + return nil, errors.Wrapf(xerr.NewErrMsg("无权使用此邀请码"), "") + } + if inviteCodeRecord.Status != 0 { + if inviteCodeRecord.Status == 1 { + return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已使用"), "") + } + return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已失效"), "") + } + if inviteCodeRecord.ExpireTime.Valid && inviteCodeRecord.ExpireTime.Time.Before(time.Now()) { + return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已过期"), "") + } + } else { + if codeVal, parseErr := strconv.ParseInt(ref, 10, 64); parseErr == nil && codeVal > 0 { + if agent.AgentCode != codeVal { + return nil, errors.Wrapf(xerr.NewErrMsg("无权使用此邀请码"), "") + } + } else { + encMobile, _ := crypto.EncryptMobile(ref, l.svcCtx.Config.Encrypt.SecretKey) + if encMobile != agent.Mobile { + return nil, errors.Wrapf(xerr.NewErrMsg("邀请信息无效"), "") + } + } + } + + // 7. 使用默认target_path（如果未提供） + targetPath := req.TargetPath + if targetPath == "" { + targetPath = fmt.Sprintf("/register?invite_code=%s", ref) + } + + shortLink, err := l.createInviteShortLink(func() string { + if inviteCodeRecord != nil { + return inviteCodeRecord.Id + } + return "" + }(), ref, targetPath) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成短链失败, %v", err) + } + + return &types.GetInviteLinkResp{ + InviteLink: shortLink, + }, nil +} + +// createInviteShortLink 创建邀请好友短链 +func (l *GetInviteLinkLogic) createInviteShortLink(inviteCodeId string, inviteCode, targetPath string) (string, error) { + promotionConfig := l.svcCtx.Config.Promotion + + // 如果没有配置推广域名，返回空字符串（保持向后兼容） + if promotionConfig.PromotionDomain == "" { + l.Errorf("推广域名未配置，返回空链接") + return "", nil + } + + // 先查询是否已存在短链 + existingShortLink, err := l.svcCtx.AgentShortLinkModel.FindOneByInviteCodeIdTypeDelState(l.ctx, sql.NullString{String: inviteCodeId, Valid: inviteCodeId != ""}, 2, globalkey.DelStateNo) + if err == nil && existingShortLink != nil { + // 已存在短链，直接返回 + return fmt.Sprintf("%s/s/%s", promotionConfig.PromotionDomain, existingShortLink.ShortCode), nil + } + + if err != nil && !errors.Is(err, model.ErrNotFound) { + return "", errors.Wrapf(err, "查询短链失败") + } + + // 如果没有邀请码ID（例如使用手机号或代理码），按邀请码字符串尝试查找以避免重复创建 + if inviteCodeId == "" && inviteCode != "" { + existingByCode, err2 := l.svcCtx.AgentShortLinkModel.FindOneByInviteCodeTypeDelState(l.ctx, sql.NullString{String: inviteCode, Valid: true}, 2, globalkey.DelStateNo) + if err2 == nil && existingByCode != nil { + return fmt.Sprintf("%s/s/%s", promotionConfig.PromotionDomain, existingByCode.ShortCode), nil + } + if err2 != nil && !errors.Is(err2, model.ErrNotFound) { + return "", errors.Wrapf(err2, "查询短链失败") + } + } + + // 生成短链标识（6位随机字符串，大小写字母+数字） + var shortCode string + maxRetries := 10 // 最大重试次数 + for retry := 0; retry < maxRetries; retry++ { + shortCode = tool.Krand(6, tool.KC_RAND_KIND_ALL) + // 检查短链标识是否已存在 + _, err := l.svcCtx.AgentShortLinkModel.FindOneByShortCodeDelState(l.ctx, shortCode, globalkey.DelStateNo) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + // 短链标识不存在，可以使用 + break + } + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "检查短链标识失败, %v", err) + } + // 短链标识已存在，继续生成 + if retry == maxRetries-1 { + return "", errors.Wrapf(xerr.NewErrMsg("生成短链失败，请重试"), "") + } + } + + // 创建短链记录（类型：2=邀请好友） + shortLink := &model.AgentShortLink{ + Id: uuid.NewString(), + Type: 2, // 邀请好友 + InviteCodeId: sql.NullString{String: inviteCodeId, Valid: inviteCodeId != ""}, + InviteCode: sql.NullString{String: inviteCode, Valid: inviteCode != ""}, + ShortCode: shortCode, + TargetPath: targetPath, + PromotionDomain: promotionConfig.PromotionDomain, + } + _, err = l.svcCtx.AgentShortLinkModel.Insert(l.ctx, nil, shortLink) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "保存短链失败, %v", err) + } + + return fmt.Sprintf("%s/s/%s", promotionConfig.PromotionDomain, shortCode), nil +} diff --git a/app/main/api/internal/logic/agent/getlevelprivilegelogic.go b/app/main/api/internal/logic/agent/getlevelprivilegelogic.go new file mode 100644 index 0000000..ec5dc84 --- /dev/null +++ b/app/main/api/internal/logic/agent/getlevelprivilegelogic.go @@ -0,0 +1,194 @@ +package agent + +import ( + "context" + "strconv" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetLevelPrivilegeLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetLevelPrivilegeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetLevelPrivilegeLogic { + return &GetLevelPrivilegeLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetLevelPrivilegeLogic) GetLevelPrivilege() (resp *types.GetLevelPrivilegeResp, err error) { + // 1. 获取当前代理等级 + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + var currentLevel int64 = 1 // 默认普通代理 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + if agent != nil { + currentLevel = agent.Level + } + + // 获取配置值的辅助函数 + getConfigFloat := func(key string, defaultValue float64) float64 { + config, err := l.svcCtx.AgentConfigModel.FindOneByConfigKey(l.ctx, key) + if err != nil { + return defaultValue + } + value, err := strconv.ParseFloat(config.ConfigValue, 64) + if err != nil { + return defaultValue + } + return value + } + + // 获取等级加成配置 + level1Bonus := getConfigFloat("level_1_bonus", 6.0) + level2Bonus := getConfigFloat("level_2_bonus", 3.0) + level3Bonus := getConfigFloat("level_3_bonus", 0.0) + + // 获取当前等级的加成 + var currentBonus float64 + switch currentLevel { + case 1: + currentBonus = level1Bonus + case 2: + currentBonus = level2Bonus + case 3: + currentBonus = level3Bonus + default: + currentBonus = level1Bonus + } + + // 获取升级返佣与费用配置 + upgradeToGoldRebate := getConfigFloat("upgrade_to_gold_rebate", 139.0) + upgradeToDiamondRebate := getConfigFloat("upgrade_to_diamond_rebate", 680.0) + upgradeToGoldFee := getConfigFloat("upgrade_to_gold_fee", 199.0) + upgradeToDiamondFee := getConfigFloat("upgrade_to_diamond_fee", 980.0) + + // 获取直接上级返佣配置 + directParentAmountDiamond := getConfigFloat("direct_parent_amount_diamond", 6.0) + directParentAmountGold := getConfigFloat("direct_parent_amount_gold", 3.0) + directParentAmountNormal := getConfigFloat("direct_parent_amount_normal", 2.0) + + // 构建各等级特权信息 + levels := []types.LevelPrivilegeItem{ + { + Level: 1, + LevelName: "普通代理", + LevelBonus: level1Bonus, + PriceReduction: l.calculatePriceReduction(currentBonus, level1Bonus), + PromoteRebate: l.buildPromoteRebateDesc(1, directParentAmountDiamond, directParentAmountGold, directParentAmountNormal), + MaxPromoteRebate: directParentAmountDiamond, // 普通代理最高返佣是直接上级是钻石时的返佣 + UpgradeRebate: l.buildUpgradeRebateDesc(1, upgradeToGoldRebate, upgradeToDiamondRebate), + Privileges: []string{ + "基础代理特权", + "可生成推广链接", + "享受推广返佣", + "可邀请下级代理", + }, + CanUpgradeSubordinate: false, + }, + { + Level: 2, + LevelName: "黄金代理", + LevelBonus: level2Bonus, + PriceReduction: l.calculatePriceReduction(currentBonus, level2Bonus), + PromoteRebate: l.buildPromoteRebateDesc(2, directParentAmountDiamond, directParentAmountGold, directParentAmountNormal), + MaxPromoteRebate: level2Bonus, // 黄金代理最高返佣是等级加成全部返佣给钻石上级 + UpgradeRebate: l.buildUpgradeRebateDesc(2, upgradeToGoldRebate, upgradeToDiamondRebate), + Privileges: []string{ + "高级代理特权", + "更高的返佣比例", + "享受推广返佣", + "可邀请下级代理", + "更多推广权益", + }, + CanUpgradeSubordinate: false, + }, + { + Level: 3, + LevelName: "钻石代理", + LevelBonus: level3Bonus, + PriceReduction: l.calculatePriceReduction(currentBonus, level3Bonus), + PromoteRebate: l.buildPromoteRebateDesc(3, directParentAmountDiamond, directParentAmountGold, directParentAmountNormal), + MaxPromoteRebate: 0, // 钻石代理无返佣 + UpgradeRebate: l.buildUpgradeRebateDesc(3, upgradeToGoldRebate, upgradeToDiamondRebate), + Privileges: []string{ + "尊享代理特权", + "最高返佣比例", + "独立团队管理", + "可调整下级级别", + "可免费升级下级为黄金代理", + }, + CanUpgradeSubordinate: true, + }, + } + + return &types.GetLevelPrivilegeResp{ + Levels: levels, + UpgradeToGoldFee: upgradeToGoldFee, + UpgradeToDiamondFee: upgradeToDiamondFee, + UpgradeToGoldRebate: upgradeToGoldRebate, + UpgradeToDiamondRebate: upgradeToDiamondRebate, + }, nil +} + +// buildPromoteRebateDesc 构建推广返佣说明 +func (l *GetLevelPrivilegeLogic) buildPromoteRebateDesc(level int64, diamondAmount, goldAmount, normalAmount float64) string { + switch level { + case 1: // 普通代理：等级加成6元 + return "更高的推广返佣，最高可达¥" + l.formatFloat(diamondAmount) + case 2: // 黄金代理：等级加成3元 + return "更高的推广返佣，最高可达¥" + l.formatFloat(3.0) + case 3: // 钻石代理：等级加成0元 + return "无等级加成，享受最低底价" + default: + return "" + } +} + +// buildUpgradeRebateDesc 构建下级升级返佣说明 +func (l *GetLevelPrivilegeLogic) buildUpgradeRebateDesc(level int64, goldRebate, diamondRebate float64) string { + switch level { + case 1: // 普通代理 + return "下级升级为黄金代理返佣¥" + l.formatFloat(goldRebate) + "，升级为钻石代理返佣¥" + l.formatFloat(diamondRebate) + case 2: // 黄金代理 + return "下级升级为钻石代理返佣¥" + l.formatFloat(diamondRebate) + case 3: // 钻石代理 + return "可免费升级下级为黄金代理" + default: + return "" + } +} + +// calculatePriceReduction 计算底价降低（相对于当前等级） +func (l *GetLevelPrivilegeLogic) calculatePriceReduction(currentBonus, targetBonus float64) float64 { + reduction := currentBonus - targetBonus + if reduction < 0 { + return 0 + } + return reduction +} + +// formatFloat 格式化浮点数，去掉末尾的0 +func (l *GetLevelPrivilegeLogic) formatFloat(f float64) string { + s := strconv.FormatFloat(f, 'f', -1, 64) + return s +} diff --git a/app/main/api/internal/logic/agent/getlinkdatalogic.go b/app/main/api/internal/logic/agent/getlinkdatalogic.go new file mode 100644 index 0000000..f6f08cb --- /dev/null +++ b/app/main/api/internal/logic/agent/getlinkdatalogic.go @@ -0,0 +1,102 @@ +package agent + +import ( + "context" + "sort" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/Masterminds/squirrel" + "github.com/jinzhu/copier" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/mr" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetLinkDataLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetLinkDataLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetLinkDataLogic { + return &GetLinkDataLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetLinkDataLogic) GetLinkData(req *types.GetLinkDataReq) (resp *types.GetLinkDataResp, err error) { + agentLinkModel, err := l.svcCtx.AgentLinkModel.FindOneByLinkIdentifier(l.ctx, req.LinkIdentifier) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取代理链接数据失败, %v", err) + } + + productModel, err := l.svcCtx.ProductModel.FindOne(l.ctx, agentLinkModel.ProductId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取产品信息失败, %v", err) + } + + // 获取产品功能列表 + build := l.svcCtx.ProductFeatureModel.SelectBuilder().Where(squirrel.Eq{ + "product_id": productModel.Id, + }) + productFeatureAll, err := l.svcCtx.ProductFeatureModel.FindAll(l.ctx, build, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取产品功能列表失败, %v", err) + } + + // 创建featureId到sort的映射，用于后续排序 + featureSortMap := make(map[string]int64) + for _, productFeature := range productFeatureAll { + featureSortMap[productFeature.FeatureId] = productFeature.Sort + } + + var features []types.Feature + mr.MapReduceVoid(func(source chan<- interface{}) { + for _, productFeature := range productFeatureAll { + source <- productFeature.FeatureId + } + }, func(item interface{}, writer mr.Writer[*model.Feature], cancel func(error)) { + id := item.(string) + + feature, findFeatureErr := l.svcCtx.FeatureModel.FindOne(l.ctx, id) + if findFeatureErr != nil { + logx.WithContext(l.ctx).Errorf("获取产品功能失败: %s, err:%v", id, findFeatureErr) + return + } + if feature != nil && feature.Id != "" { + writer.Write(feature) + } + }, func(pipe <-chan *model.Feature, cancel func(error)) { + for item := range pipe { + var feature types.Feature + _ = copier.Copy(&feature, item) + features = append(features, feature) + } + }) + + // 按照productFeature.Sort字段对features进行排序 + sort.Slice(features, func(i, j int) bool { + sortI := featureSortMap[features[i].ID] + sortJ := featureSortMap[features[j].ID] + return sortI < sortJ + }) + + return &types.GetLinkDataResp{ + AgentId: agentLinkModel.AgentId, + ProductId: agentLinkModel.ProductId, + SetPrice: agentLinkModel.SetPrice, + ActualBasePrice: agentLinkModel.ActualBasePrice, + ProductName: productModel.ProductName, + ProductEn: productModel.ProductEn, + SellPrice: agentLinkModel.SetPrice, // 使用代理设定价格作为销售价格 + Description: productModel.Description, + Features: features, + }, nil +} diff --git a/app/main/api/internal/logic/agent/getpromotionquerylistlogic.go b/app/main/api/internal/logic/agent/getpromotionquerylistlogic.go new file mode 100644 index 0000000..513a16a --- /dev/null +++ b/app/main/api/internal/logic/agent/getpromotionquerylistlogic.go @@ -0,0 +1,88 @@ +package agent + +import ( + "context" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/jinzhu/copier" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" +) + +type GetPromotionQueryListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetPromotionQueryListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetPromotionQueryListLogic { + return &GetPromotionQueryListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetPromotionQueryListLogic) GetPromotionQueryList(req *types.GetPromotionQueryListReq) (resp *types.GetPromotionQueryListResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 查询当前代理的代理订单，按创建时间倒序分页 + builder := l.svcCtx.AgentOrderModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", agent.Id, globalkey.DelStateNo) + + orders, _, err := l.svcCtx.AgentOrderModel.FindPageListByPageWithTotal(l.ctx, builder, req.Page, req.PageSize, "create_time DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理订单失败, %v", err) + } + + // 组装查询报告列表（只展示已创建的查询） + list := make([]types.PromotionQueryItem, 0, len(orders)) + for _, ao := range orders { + // 查询对应的报告 + q, qErr := l.svcCtx.QueryModel.FindOneByOrderId(l.ctx, ao.OrderId) + if qErr != nil { + if errors.Is(qErr, model.ErrNotFound) { + // 订单对应的查询尚未创建，跳过展示 + continue + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询报告失败, %v", qErr) + } + + // 获取产品名称 + product, pErr := l.svcCtx.ProductModel.FindOne(l.ctx, ao.ProductId) + if pErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询产品信息失败, %v", pErr) + } + + item := types.PromotionQueryItem{} + _ = copier.Copy(&item, q) + item.Id = q.Id + item.OrderId = q.OrderId + item.ProductName = product.ProductName + item.CreateTime = q.CreateTime.Format("2006-01-02 15:04:05") + item.QueryState = q.QueryState + list = append(list, item) + } + + return &types.GetPromotionQueryListResp{ + Total: int64(len(list)), // 前端仅展示已创建查询条目 + List: list, + }, nil +} diff --git a/app/main/api/internal/logic/agent/getrebatelistlogic.go b/app/main/api/internal/logic/agent/getrebatelistlogic.go new file mode 100644 index 0000000..20f38a3 --- /dev/null +++ b/app/main/api/internal/logic/agent/getrebatelistlogic.go @@ -0,0 +1,128 @@ +package agent + +import ( + "context" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetRebateListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetRebateListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetRebateListLogic { + return &GetRebateListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetRebateListLogic) GetRebateList(req *types.GetRebateListReq) (resp *types.GetRebateListResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 构建查询条件 + builder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", agent.Id, globalkey.DelStateNo) + + // 如果指定了返佣类型，则按类型筛选（1, 2, 3） + if req.RebateType != nil { + builder = builder.Where("rebate_type = ?", *req.RebateType) + } + + builder = builder.OrderBy("create_time DESC") + + // 3. 分页查询 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + offset := (page - 1) * pageSize + + // 4. 查询总数 + total, err := l.svcCtx.AgentRebateModel.FindCount(l.ctx, builder, "id") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询返佣总数失败, %v", err) + } + + // 5. 查询列表 + builder = builder.Limit(uint64(pageSize)).Offset(uint64(offset)) + rebates, err := l.svcCtx.AgentRebateModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询返佣列表失败, %v", err) + } + + // 6. 组装响应 + var list []types.RebateItem + for _, rebate := range rebates { + // 查询订单号 + orderNo := "" + if rebate.OrderId != "" { + order, err := l.svcCtx.OrderModel.FindOne(l.ctx, rebate.OrderId) + if err == nil { + orderNo = order.OrderNo + } + } + + // 查询来源代理手机号和等级 + sourceAgentMobile := "" + sourceAgentLevel := int64(0) + if rebate.SourceAgentId != "" { + sourceAgent, err := l.svcCtx.AgentModel.FindOne(l.ctx, rebate.SourceAgentId) + if err == nil { + if sourceAgent.Mobile != "" { + decrypted, err := crypto.DecryptMobile(sourceAgent.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err == nil { + sourceAgentMobile = decrypted + } + } + sourceAgentLevel = sourceAgent.Level + } + } + + list = append(list, types.RebateItem{ + Id: rebate.Id, + SourceAgentId: rebate.SourceAgentId, + SourceAgentMobile: sourceAgentMobile, + SourceAgentLevel: sourceAgentLevel, + OrderId: rebate.OrderId, + OrderNo: orderNo, + RebateType: rebate.RebateType, + Amount: rebate.RebateAmount, + CreateTime: rebate.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return &types.GetRebateListResp{ + Total: total, + List: list, + }, nil +} diff --git a/app/main/api/internal/logic/agent/getrevenueinfologic.go b/app/main/api/internal/logic/agent/getrevenueinfologic.go new file mode 100644 index 0000000..ff431f2 --- /dev/null +++ b/app/main/api/internal/logic/agent/getrevenueinfologic.go @@ -0,0 +1,127 @@ +package agent + +import ( + "context" + "time" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetRevenueInfoLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetRevenueInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetRevenueInfoLogic { + return &GetRevenueInfoLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetRevenueInfoLogic) GetRevenueInfo() (resp *types.GetRevenueInfoResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 获取钱包信息 + wallet, err := l.svcCtx.AgentWalletModel.FindOneByAgentId(l.ctx, agent.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询钱包信息失败, %v", err) + } + + // 获取当前时间 + now := time.Now() + // 今日开始时间（00:00:00） + todayStart := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location()) + // 本月开始时间（1号 00:00:00） + monthStart := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location()) + + // 3. 统计佣金总额（从 agent_commission 表统计） + commissionBuilder := l.svcCtx.AgentCommissionModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", agent.Id, globalkey.DelStateNo) + commissionTotal, _ := l.svcCtx.AgentCommissionModel.FindSum(l.ctx, commissionBuilder, "amount") + + // 3.1 统计佣金今日收益 + commissionTodayBuilder := l.svcCtx.AgentCommissionModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ? AND create_time >= ?", agent.Id, globalkey.DelStateNo, todayStart) + commissionToday, _ := l.svcCtx.AgentCommissionModel.FindSum(l.ctx, commissionTodayBuilder, "amount") + + // 3.2 统计佣金本月收益 + commissionMonthBuilder := l.svcCtx.AgentCommissionModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ? AND create_time >= ?", agent.Id, globalkey.DelStateNo, monthStart) + commissionMonth, _ := l.svcCtx.AgentCommissionModel.FindSum(l.ctx, commissionMonthBuilder, "amount") + + // 4. 统计返佣总额（包括推广返佣和升级返佣） + // 4.1 统计推广返佣（从 agent_rebate 表） + rebateBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", agent.Id, globalkey.DelStateNo) + promoteRebateTotal, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, rebateBuilder, "rebate_amount") + + // 4.2 统计升级返佣（从 agent_upgrade 表，查询 rebate_agent_id = 当前代理ID 且 status = 2（已完成）且 upgrade_type = 1（自主付费）的记录） + // 注意：只要返佣给自己的都要统计，不管升级后是否脱离关系（rebate_agent_id 记录的是升级时的原直接上级） + upgradeRebateBuilder := l.svcCtx.AgentUpgradeModel.SelectBuilder(). + Where("rebate_agent_id IS NOT NULL AND rebate_agent_id = ? AND status = ? AND upgrade_type = ? AND del_state = ?", + agent.Id, 2, 1, globalkey.DelStateNo) + upgradeRebateTotal, _ := l.svcCtx.AgentUpgradeModel.FindSum(l.ctx, upgradeRebateBuilder, "rebate_amount") + + rebateTotal := promoteRebateTotal + upgradeRebateTotal + + // 4.3 统计返佣今日收益 + // 推广返佣今日 + promoteRebateTodayBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ? AND create_time >= ?", agent.Id, globalkey.DelStateNo, todayStart) + promoteRebateToday, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, promoteRebateTodayBuilder, "rebate_amount") + // 升级返佣今日 + upgradeRebateTodayBuilder := l.svcCtx.AgentUpgradeModel.SelectBuilder(). + Where("rebate_agent_id IS NOT NULL AND rebate_agent_id = ? AND status = ? AND upgrade_type = ? AND del_state = ? AND create_time >= ?", + agent.Id, 2, 1, globalkey.DelStateNo, todayStart) + upgradeRebateToday, _ := l.svcCtx.AgentUpgradeModel.FindSum(l.ctx, upgradeRebateTodayBuilder, "rebate_amount") + rebateToday := promoteRebateToday + upgradeRebateToday + + // 4.4 统计返佣本月收益 + // 推广返佣本月 + promoteRebateMonthBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ? AND create_time >= ?", agent.Id, globalkey.DelStateNo, monthStart) + promoteRebateMonth, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, promoteRebateMonthBuilder, "rebate_amount") + // 升级返佣本月 + upgradeRebateMonthBuilder := l.svcCtx.AgentUpgradeModel.SelectBuilder(). + Where("rebate_agent_id IS NOT NULL AND rebate_agent_id = ? AND status = ? AND upgrade_type = ? AND del_state = ? AND create_time >= ?", + agent.Id, 2, 1, globalkey.DelStateNo, monthStart) + upgradeRebateMonth, _ := l.svcCtx.AgentUpgradeModel.FindSum(l.ctx, upgradeRebateMonthBuilder, "rebate_amount") + rebateMonth := promoteRebateMonth + upgradeRebateMonth + + return &types.GetRevenueInfoResp{ + Balance: wallet.Balance, + FrozenBalance: wallet.FrozenBalance, + TotalEarnings: wallet.TotalEarnings, + WithdrawnAmount: wallet.WithdrawnAmount, + CommissionTotal: commissionTotal, // 佣金累计总收益（推广订单获得的佣金） + CommissionToday: commissionToday, // 佣金今日收益 + CommissionMonth: commissionMonth, // 佣金本月收益 + RebateTotal: rebateTotal, // 返佣累计总收益（包括推广返佣和升级返佣） + RebateToday: rebateToday, // 返佣今日收益 + RebateMonth: rebateMonth, // 返佣本月收益 + }, nil +} diff --git a/app/main/api/internal/logic/agent/getsubordinatecontributiondetaillogic.go b/app/main/api/internal/logic/agent/getsubordinatecontributiondetaillogic.go new file mode 100644 index 0000000..0e8f079 --- /dev/null +++ b/app/main/api/internal/logic/agent/getsubordinatecontributiondetaillogic.go @@ -0,0 +1,404 @@ +package agent + +import ( + "context" + "sort" + "time" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetSubordinateContributionDetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetSubordinateContributionDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetSubordinateContributionDetailLogic { + return &GetSubordinateContributionDetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetSubordinateContributionDetailLogic) GetSubordinateContributionDetail(req *types.GetSubordinateContributionDetailReq) (resp *types.GetSubordinateContributionDetailResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取当前代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 验证下级代理是否存在，并且确实是当前代理的下级（直接或间接） + subordinate, err := l.svcCtx.AgentModel.FindOne(l.ctx, req.SubordinateId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("下级代理不存在"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询下级代理信息失败, %v", err) + } + + // 3. 验证下级关系（递归检查是否是下级） + isSubordinate := l.isSubordinate(agent.Id, req.SubordinateId) + if !isSubordinate { + return nil, errors.Wrapf(xerr.NewErrMsg("该代理不是您的下级"), "") + } + + // 4. 解密手机号 + mobile := "" + if subordinate.Mobile != "" { + decrypted, err := crypto.DecryptMobile(subordinate.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err == nil { + mobile = decrypted + } + } + + // 5. 获取等级名称 + levelName := "" + switch subordinate.Level { + case 1: + levelName = "普通" + case 2: + levelName = "黄金" + case 3: + levelName = "钻石" + } + + // 6. 计算时间范围 + now := time.Now() + todayStart := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location()) + monthStart := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location()) + + // 7. 统计订单数据（仅统计有返佣给当前用户的订单） + // 通过 agent_rebate 表统计 DISTINCT order_id + rebateBaseBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND source_agent_id = ? AND del_state = ?", agent.Id, req.SubordinateId, globalkey.DelStateNo) + + // 总订单量（去重订单ID） + totalOrders := l.countDistinctOrders(l.ctx, rebateBaseBuilder) + + // 今日订单量 + todayRebateBuilder := rebateBaseBuilder.Where("create_time >= ?", todayStart) + todayOrders := l.countDistinctOrders(l.ctx, todayRebateBuilder) + + // 月订单量 + monthRebateBuilder := rebateBaseBuilder.Where("create_time >= ?", monthStart) + monthOrders := l.countDistinctOrders(l.ctx, monthRebateBuilder) + + // 8. 统计返佣金额 + totalRebateAmount, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, rebateBaseBuilder, "rebate_amount") + todayRebateAmount, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, todayRebateBuilder, "rebate_amount") + monthRebateAmount, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, monthRebateBuilder, "rebate_amount") + + // 9. 统计邀请数据 + inviteBaseBuilder := l.svcCtx.AgentRelationModel.SelectBuilder(). + Where("parent_id = ? AND relation_type = ? AND del_state = ?", req.SubordinateId, 1, globalkey.DelStateNo) + totalInvites, _ := l.svcCtx.AgentRelationModel.FindCount(l.ctx, inviteBaseBuilder, "id") + + todayInviteBuilder := inviteBaseBuilder.Where("create_time >= ?", todayStart) + todayInvites, _ := l.svcCtx.AgentRelationModel.FindCount(l.ctx, todayInviteBuilder, "id") + + monthInviteBuilder := inviteBaseBuilder.Where("create_time >= ?", monthStart) + monthInvites, _ := l.svcCtx.AgentRelationModel.FindCount(l.ctx, monthInviteBuilder, "id") + + // 10. 分页查询订单列表或邀请列表 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + + var orderList []types.OrderItem + var inviteList []types.InviteItem + var orderListTotal int64 + var inviteListTotal int64 + + tabType := req.TabType + if tabType == "" { + tabType = "order" // 默认显示订单列表 + } + + if tabType == "order" { + // 查询订单列表（仅显示有返佣的订单） + orderList, orderListTotal, err = l.getOrderList(l.ctx, agent.Id, req.SubordinateId, page, pageSize) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询订单列表失败, %v", err) + } + } else if tabType == "invite" { + // 查询邀请列表 + inviteList, inviteListTotal, err = l.getInviteList(l.ctx, req.SubordinateId, page, pageSize) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询邀请列表失败, %v", err) + } + } + + // 11. 组装响应 + resp = &types.GetSubordinateContributionDetailResp{ + Mobile: mobile, + LevelName: levelName, + CreateTime: subordinate.CreateTime.Format("2006-01-02 15:04:05"), + OrderStats: types.OrderStatistics{ + TotalOrders: totalOrders, + MonthOrders: monthOrders, + TodayOrders: todayOrders, + }, + RebateStats: types.RebateStatistics{ + TotalRebateAmount: totalRebateAmount, + TodayRebateAmount: todayRebateAmount, + MonthRebateAmount: monthRebateAmount, + }, + InviteStats: types.InviteStatistics{ + TotalInvites: totalInvites, + TodayInvites: todayInvites, + MonthInvites: monthInvites, + }, + OrderList: orderList, + InviteList: inviteList, + OrderListTotal: orderListTotal, + InviteListTotal: inviteListTotal, + } + + return resp, nil +} + +// isSubordinate 递归检查 targetId 是否是 parentId 的下级（直接或间接） +func (l *GetSubordinateContributionDetailLogic) isSubordinate(parentId, targetId string) bool { + // 查询直接下级 + builder := l.svcCtx.AgentRelationModel.SelectBuilder(). + Where("parent_id = ? AND relation_type = ? AND del_state = ?", parentId, 1, globalkey.DelStateNo) + relations, err := l.svcCtx.AgentRelationModel.FindAll(l.ctx, builder, "") + if err != nil { + return false + } + + for _, relation := range relations { + // 如果是直接下级，返回 true + if relation.ChildId == targetId { + return true + } + // 递归检查间接下级 + if l.isSubordinate(relation.ChildId, targetId) { + return true + } + } + + return false +} + +// countDistinctOrders 统计去重的订单数量（通过返佣记录） +func (l *GetSubordinateContributionDetailLogic) countDistinctOrders(ctx context.Context, builder squirrel.SelectBuilder) int64 { + // 查询所有返佣记录，在内存中去重订单ID + rebates, err := l.svcCtx.AgentRebateModel.FindAll(ctx, builder, "") + if err != nil { + return 0 + } + + orderIdSet := make(map[string]bool) + for _, rebate := range rebates { + orderIdSet[rebate.OrderId] = true + } + + return int64(len(orderIdSet)) +} + +// getOrderList 获取订单列表（仅显示有返佣的订单） +func (l *GetSubordinateContributionDetailLogic) getOrderList(ctx context.Context, agentId, subordinateId string, page, pageSize int64) ([]types.OrderItem, int64, error) { + // 1. 查询所有返佣记录 + rebateBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND source_agent_id = ? AND del_state = ?", agentId, subordinateId, globalkey.DelStateNo). + OrderBy("create_time DESC") + + // 查询总数（去重订单ID） + totalBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND source_agent_id = ? AND del_state = ?", agentId, subordinateId, globalkey.DelStateNo) + total := l.countDistinctOrders(ctx, totalBuilder) + + // 查询所有返佣记录 + allRebates, err := l.svcCtx.AgentRebateModel.FindAll(ctx, rebateBuilder, "") + if err != nil { + return nil, 0, err + } + + if len(allRebates) == 0 { + return []types.OrderItem{}, total, nil + } + + // 2. 在内存中去重订单ID，并按创建时间排序 + type OrderRebateInfo struct { + OrderId string + RebateId string + ProductId string + RebateAmount float64 + CreateTime time.Time + } + + orderMap := make(map[string]*OrderRebateInfo) // orderId -> 最新的返佣信息 + for _, rebate := range allRebates { + if existing, ok := orderMap[rebate.OrderId]; ok { + // 如果已存在，保留创建时间最新的 + if rebate.CreateTime.After(existing.CreateTime) { + orderMap[rebate.OrderId] = &OrderRebateInfo{ + OrderId: rebate.OrderId, + RebateId: rebate.Id, + ProductId: rebate.ProductId, + RebateAmount: rebate.RebateAmount, + CreateTime: rebate.CreateTime, + } + } + } else { + orderMap[rebate.OrderId] = &OrderRebateInfo{ + OrderId: rebate.OrderId, + RebateId: rebate.Id, + ProductId: rebate.ProductId, + RebateAmount: rebate.RebateAmount, + CreateTime: rebate.CreateTime, + } + } + } + + // 3. 转换为切片并按时间排序 + orderList := make([]*OrderRebateInfo, 0, len(orderMap)) + for _, info := range orderMap { + orderList = append(orderList, info) + } + + // 按创建时间倒序排序 + sort.Slice(orderList, func(i, j int) bool { + return orderList[i].CreateTime.After(orderList[j].CreateTime) + }) + + // 4. 分页 + offset := (page - 1) * pageSize + end := offset + pageSize + if end > int64(len(orderList)) { + end = int64(len(orderList)) + } + if offset >= int64(len(orderList)) { + return []types.OrderItem{}, total, nil + } + + pagedOrderList := orderList[offset:end] + + // 5. 组装订单列表 + var resultList []types.OrderItem + productCache := make(map[string]string) // 产品ID -> 产品名称缓存 + + for _, orderInfo := range pagedOrderList { + // 查询订单信息 + order, err := l.svcCtx.OrderModel.FindOne(ctx, orderInfo.OrderId) + if err != nil { + continue + } + + // 查询agent_order信息（用于获取订单金额） + agentOrder, err := l.svcCtx.AgentOrderModel.FindOneByOrderId(ctx, orderInfo.OrderId) + if err != nil { + continue + } + + // 查询产品名称（使用缓存） + productName := "" + if name, ok := productCache[orderInfo.ProductId]; ok { + productName = name + } else { + product, err := l.svcCtx.ProductModel.FindOne(ctx, orderInfo.ProductId) + if err == nil { + productName = product.ProductName + productCache[orderInfo.ProductId] = productName + } + } + + resultList = append(resultList, types.OrderItem{ + OrderNo: order.OrderNo, + ProductId: orderInfo.ProductId, + ProductName: productName, + OrderAmount: agentOrder.OrderAmount, + RebateAmount: orderInfo.RebateAmount, + CreateTime: orderInfo.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return resultList, total, nil +} + +// getInviteList 获取邀请列表 +func (l *GetSubordinateContributionDetailLogic) getInviteList(ctx context.Context, subordinateId string, page, pageSize int64) ([]types.InviteItem, int64, error) { + // 查询总数 + builder := l.svcCtx.AgentRelationModel.SelectBuilder(). + Where("parent_id = ? AND relation_type = ? AND del_state = ?", subordinateId, 1, globalkey.DelStateNo) + total, err := l.svcCtx.AgentRelationModel.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + // 分页查询 + offset := (page - 1) * pageSize + builder = builder.OrderBy("create_time DESC"). + Limit(uint64(pageSize)).Offset(uint64(offset)) + relations, err := l.svcCtx.AgentRelationModel.FindAll(ctx, builder, "") + if err != nil { + return nil, 0, err + } + + // 组装邀请列表 + var inviteList []types.InviteItem + for _, relation := range relations { + // 查询被邀请的代理信息 + invitedAgent, err := l.svcCtx.AgentModel.FindOne(ctx, relation.ChildId) + if err != nil { + continue + } + + // 获取等级名称 + levelName := "" + switch invitedAgent.Level { + case 1: + levelName = "普通" + case 2: + levelName = "黄金" + case 3: + levelName = "钻石" + } + + // 解密手机号 + mobile := "" + if invitedAgent.Mobile != "" { + decrypted, err := crypto.DecryptMobile(invitedAgent.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err == nil { + mobile = decrypted + } + } + + inviteList = append(inviteList, types.InviteItem{ + AgentId: invitedAgent.Id, + Level: invitedAgent.Level, + LevelName: levelName, + Mobile: mobile, + CreateTime: relation.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return inviteList, total, nil +} diff --git a/app/main/api/internal/logic/agent/getsubordinatelistlogic.go b/app/main/api/internal/logic/agent/getsubordinatelistlogic.go new file mode 100644 index 0000000..5e5559c --- /dev/null +++ b/app/main/api/internal/logic/agent/getsubordinatelistlogic.go @@ -0,0 +1,132 @@ +package agent + +import ( + "context" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetSubordinateListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetSubordinateListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetSubordinateListLogic { + return &GetSubordinateListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetSubordinateListLogic) GetSubordinateList(req *types.GetSubordinateListReq) (resp *types.GetSubordinateListResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 构建查询条件（查询直接下级） + builder := l.svcCtx.AgentRelationModel.SelectBuilder(). + Where("parent_id = ? AND relation_type = ? AND del_state = ?", agent.Id, 1, globalkey.DelStateNo). + OrderBy("create_time DESC") + + // 3. 分页查询 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + offset := (page - 1) * pageSize + + // 4. 查询总数 + total, err := l.svcCtx.AgentRelationModel.FindCount(l.ctx, builder, "id") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询下级总数失败, %v", err) + } + + // 5. 查询列表 + builder = builder.Limit(uint64(pageSize)).Offset(uint64(offset)) + relations, err := l.svcCtx.AgentRelationModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询下级列表失败, %v", err) + } + + // 6. 组装响应 + var list []types.SubordinateItem + for _, relation := range relations { + subordinate, err := l.svcCtx.AgentModel.FindOne(l.ctx, relation.ChildId) + if err != nil { + continue + } + + levelName := "" + switch subordinate.Level { + case 1: + levelName = "普通" + case 2: + levelName = "黄金" + case 3: + levelName = "钻石" + } + + // 解密手机号 + mobile := "" + if subordinate.Mobile != "" { + decrypted, err := crypto.DecryptMobile(subordinate.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err == nil { + mobile = decrypted + } + } + + // 统计订单数和金额 + totalOrders := int64(0) + totalAmount := 0.0 + orderBuilder := l.svcCtx.AgentOrderModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", subordinate.Id, globalkey.DelStateNo) + orders, err := l.svcCtx.AgentOrderModel.FindAll(l.ctx, orderBuilder, "") + if err == nil { + totalOrders = int64(len(orders)) + for _, order := range orders { + totalAmount += order.OrderAmount + } + } + + list = append(list, types.SubordinateItem{ + AgentId: subordinate.Id, + Level: subordinate.Level, + LevelName: levelName, + Mobile: mobile, + TotalOrders: totalOrders, + TotalAmount: totalAmount, + CreateTime: relation.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return &types.GetSubordinateListResp{ + Total: total, + List: list, + }, nil +} diff --git a/app/main/api/internal/logic/agent/getteamlistlogic.go b/app/main/api/internal/logic/agent/getteamlistlogic.go new file mode 100644 index 0000000..d907212 --- /dev/null +++ b/app/main/api/internal/logic/agent/getteamlistlogic.go @@ -0,0 +1,341 @@ +package agent + +import ( + "context" + "sort" + "strings" + "time" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetTeamListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetTeamListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetTeamListLogic { + return &GetTeamListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetTeamListLogic) GetTeamList(req *types.GetTeamListReq) (resp *types.GetTeamListResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 递归查询所有下级（直接+间接） + allSubordinateIds := make(map[string]bool) + directSubordinateIds := make(map[string]bool) + + // 递归函数：收集所有下级ID + var collectSubordinates func(string) error + collectSubordinates = func(parentId string) error { + // 查询直接下级 + builder := l.svcCtx.AgentRelationModel.SelectBuilder(). + Where("parent_id = ? AND relation_type = ? AND del_state = ?", parentId, 1, globalkey.DelStateNo) + relations, err := l.svcCtx.AgentRelationModel.FindAll(l.ctx, builder, "") + if err != nil { + return err + } + + for _, relation := range relations { + // 如果是第一层，标记为直接下级 + if parentId == agent.Id { + directSubordinateIds[relation.ChildId] = true + } + // 添加到所有下级集合 + allSubordinateIds[relation.ChildId] = true + // 递归查询下级的下级 + if err := collectSubordinates(relation.ChildId); err != nil { + return err + } + } + return nil + } + + // 开始递归收集所有下级 + if err := collectSubordinates(agent.Id); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询下级关系失败, %v", err) + } + + // 3. 如果没有下级，返回空数据 + if len(allSubordinateIds) == 0 { + return &types.GetTeamListResp{ + Statistics: types.TeamStatistics{}, + Total: 0, + List: []types.TeamMemberItem{}, + }, nil + } + + // 4. 将下级ID转换为切片用于查询 + subordinateIds := make([]string, 0, len(allSubordinateIds)) + for id := range allSubordinateIds { + subordinateIds = append(subordinateIds, id) + } + + // 5. 计算时间范围 + now := time.Now() + todayStart := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location()) + monthStart := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location()) + + // 6. 统计顶部数据 + statistics := l.calculateTeamStatistics(agent.Id, subordinateIds, todayStart, monthStart) + + // 7. 查询所有下级代理信息（不进行手机号过滤，因为需要模糊搜索） + builder := l.svcCtx.AgentModel.SelectBuilder(). + Where(squirrel.Eq{"id": subordinateIds}). + Where("del_state = ?", globalkey.DelStateNo) + + allTeamMembers, err := l.svcCtx.AgentModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询团队成员失败, %v", err) + } + + // 8. 如果有手机号搜索条件，进行模糊匹配过滤（在内存中） + var filteredMembers []*model.Agent + searchMobile := strings.TrimSpace(req.Mobile) + if searchMobile != "" { + // 解密所有手机号并进行模糊匹配 + for _, member := range allTeamMembers { + if member.Mobile != "" { + decryptedMobile, err := crypto.DecryptMobile(member.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err == nil && strings.Contains(decryptedMobile, searchMobile) { + filteredMembers = append(filteredMembers, member) + } + } + } + } else { + // 没有搜索条件，使用所有成员 + filteredMembers = allTeamMembers + } + + // 9. 按创建时间倒序排序 + sort.Slice(filteredMembers, func(i, j int) bool { + return filteredMembers[i].CreateTime.After(filteredMembers[j].CreateTime) + }) + + // 10. 分页处理 + total := int64(len(filteredMembers)) + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + offset := (page - 1) * pageSize + + // 计算分页范围 + start := int(offset) + end := start + int(pageSize) + if start > len(filteredMembers) { + start = len(filteredMembers) + } + if end > len(filteredMembers) { + end = len(filteredMembers) + } + + var teamMembers []*model.Agent + if start < end { + teamMembers = filteredMembers[start:end] + } + + // 11. 组装响应列表 + var list []types.TeamMemberItem + for _, member := range teamMembers { + memberItem := l.buildTeamMemberItem(agent.Id, member, directSubordinateIds, todayStart) + list = append(list, memberItem) + } + + return &types.GetTeamListResp{ + Statistics: statistics, + Total: total, + List: list, + }, nil +} + +// calculateTeamStatistics 计算团队统计数据 +// 注意：所有统计都基于 subordinateIds（下级ID列表），不包含自己的数据 +func (l *GetTeamListLogic) calculateTeamStatistics(agentId string, subordinateIds []string, todayStart, monthStart time.Time) types.TeamStatistics { + // 团队成员总数：只统计下级，不包括自己 + stats := types.TeamStatistics{ + TotalMembers: int64(len(subordinateIds)), + } + + // 统计今日和本月新增成员（只统计下级，不包括自己） + todayNewCount := int64(0) + monthNewCount := int64(0) + for _, id := range subordinateIds { + member, err := l.svcCtx.AgentModel.FindOne(l.ctx, id) + if err != nil { + continue + } + if member.CreateTime.After(todayStart) { + todayNewCount++ + } + if member.CreateTime.After(monthStart) { + monthNewCount++ + } + } + stats.TodayNewMembers = todayNewCount + stats.MonthNewMembers = monthNewCount + + // 统计团队总查询量（仅统计有返佣给当前用户的订单，通过agent_rebate表去重统计order_id） + if len(subordinateIds) > 0 { + rebateBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", agentId, globalkey.DelStateNo). + Where("source_agent_id != ?", agentId). // 明确排除自己 + Where(squirrel.Eq{"source_agent_id": subordinateIds}) + + // 统计去重的订单数量 + totalQueries := l.countDistinctOrders(l.ctx, rebateBuilder) + stats.TotalQueries = totalQueries + + // 今日推广量（仅统计有返佣的订单） + todayRebateBuilder := rebateBuilder.Where("create_time >= ?", todayStart) + todayPromotions := l.countDistinctOrders(l.ctx, todayRebateBuilder) + stats.TodayPromotions = todayPromotions + + // 本月推广量（仅统计有返佣的订单） + monthRebateBuilder := rebateBuilder.Where("create_time >= ?", monthStart) + monthPromotions := l.countDistinctOrders(l.ctx, monthRebateBuilder) + stats.MonthPromotions = monthPromotions + } + + // 统计收益：只统计从下级获得的返佣（依靠团队得到的收益） + // 返佣：从agent_rebate表统计（从下级获得的返佣） + // agent_id = 当前代理ID（获得返佣的代理） + // source_agent_id IN 下级ID列表（来源代理，即产生订单的下级代理） + // 明确排除自己：source_agent_id != agentId（确保不统计自己的数据） + if len(subordinateIds) > 0 { + rebateBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", agentId, globalkey.DelStateNo). + Where("source_agent_id != ?", agentId). // 明确排除自己 + Where(squirrel.Eq{"source_agent_id": subordinateIds}) + totalRebate, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, rebateBuilder, "rebate_amount") + todayRebate, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, rebateBuilder.Where("create_time >= ?", todayStart), "rebate_amount") + monthRebate, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, rebateBuilder.Where("create_time >= ?", monthStart), "rebate_amount") + + stats.TotalEarnings = totalRebate + stats.TodayEarnings = todayRebate + stats.MonthEarnings = monthRebate + } + + return stats +} + +// countDistinctOrders 统计去重的订单数量（通过返佣记录） +func (l *GetTeamListLogic) countDistinctOrders(ctx context.Context, builder squirrel.SelectBuilder) int64 { + // 查询所有返佣记录，在内存中去重订单ID + rebates, err := l.svcCtx.AgentRebateModel.FindAll(ctx, builder, "") + if err != nil { + return 0 + } + + orderIdSet := make(map[string]bool) + for _, rebate := range rebates { + orderIdSet[rebate.OrderId] = true + } + + return int64(len(orderIdSet)) +} + +// countDistinctOrdersForMember 统计单个成员的去重订单数量（通过返佣记录） +func (l *GetTeamListLogic) countDistinctOrdersForMember(ctx context.Context, builder squirrel.SelectBuilder) int64 { + // 查询所有返佣记录，在内存中去重订单ID + rebates, err := l.svcCtx.AgentRebateModel.FindAll(ctx, builder, "") + if err != nil { + return 0 + } + + orderIdSet := make(map[string]bool) + for _, rebate := range rebates { + orderIdSet[rebate.OrderId] = true + } + + return int64(len(orderIdSet)) +} + +// buildTeamMemberItem 构建团队成员项 +func (l *GetTeamListLogic) buildTeamMemberItem(agentId string, member *model.Agent, directSubordinateIds map[string]bool, todayStart time.Time) types.TeamMemberItem { + levelName := "" + switch member.Level { + case 1: + levelName = "普通" + case 2: + levelName = "黄金" + case 3: + levelName = "钻石" + } + + // 解密手机号 + mobile := "" + if member.Mobile != "" { + decrypted, err := crypto.DecryptMobile(member.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err == nil { + mobile = decrypted + } + } + + // 统计查询量（仅统计有返佣给当前用户的订单，通过agent_rebate表去重统计order_id） + rebateBuilder := l.svcCtx.AgentRebateModel.SelectBuilder(). + Where("agent_id = ? AND source_agent_id = ? AND del_state = ?", agentId, member.Id, globalkey.DelStateNo) + totalQueries := l.countDistinctOrdersForMember(l.ctx, rebateBuilder) + todayRebateBuilder := rebateBuilder.Where("create_time >= ?", todayStart) + todayQueries := l.countDistinctOrdersForMember(l.ctx, todayRebateBuilder) + + // 统计返佣给我的金额（从agent_rebate表，source_agent_id = member.Id, agent_id = agentId） + totalRebateAmount, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, rebateBuilder, "rebate_amount") + todayRebateAmount, _ := l.svcCtx.AgentRebateModel.FindSum(l.ctx, rebateBuilder.Where("create_time >= ?", todayStart), "rebate_amount") + + // 统计邀请人数（从agent_relation表，parent_id = member.Id） + inviteBuilder := l.svcCtx.AgentRelationModel.SelectBuilder(). + Where("parent_id = ? AND relation_type = ? AND del_state = ?", member.Id, 1, globalkey.DelStateNo) + totalInvites, _ := l.svcCtx.AgentRelationModel.FindCount(l.ctx, inviteBuilder, "id") + todayInvites, _ := l.svcCtx.AgentRelationModel.FindCount(l.ctx, inviteBuilder.Where("create_time >= ?", todayStart), "id") + + // 判断是否直接下级 + isDirect := directSubordinateIds[member.Id] + + return types.TeamMemberItem{ + AgentId: member.Id, + Level: member.Level, + LevelName: levelName, + Mobile: mobile, + CreateTime: member.CreateTime.Format("2006-01-02 15:04:05"), + TodayQueries: todayQueries, + TotalQueries: totalQueries, + TotalRebateAmount: totalRebateAmount, + TodayRebateAmount: todayRebateAmount, + TotalInvites: totalInvites, + TodayInvites: todayInvites, + IsDirect: isDirect, + } +} diff --git a/app/main/api/internal/logic/agent/getteamstatisticslogic.go b/app/main/api/internal/logic/agent/getteamstatisticslogic.go new file mode 100644 index 0000000..72f8312 --- /dev/null +++ b/app/main/api/internal/logic/agent/getteamstatisticslogic.go @@ -0,0 +1,157 @@ +package agent + +import ( + "context" + "time" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetTeamStatisticsLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetTeamStatisticsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetTeamStatisticsLogic { + return &GetTeamStatisticsLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetTeamStatisticsLogic) GetTeamStatistics() (resp *types.TeamStatisticsResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 递归查询所有下级（直接+间接） + allSubordinateIds := make(map[string]bool) + directSubordinateIds := make(map[string]bool) + + // 递归函数：收集所有下级ID + var collectSubordinates func(string) error + collectSubordinates = func(parentId string) error { + // 查询直接下级 + builder := l.svcCtx.AgentRelationModel.SelectBuilder(). + Where("parent_id = ? AND relation_type = ? AND del_state = ?", parentId, 1, globalkey.DelStateNo) + relations, err := l.svcCtx.AgentRelationModel.FindAll(l.ctx, builder, "") + if err != nil { + return err + } + + for _, relation := range relations { + // 如果是第一层，标记为直接下级 + if parentId == agent.Id { + directSubordinateIds[relation.ChildId] = true + } + // 添加到所有下级集合 + allSubordinateIds[relation.ChildId] = true + // 递归查询下级的下级 + if err := collectSubordinates(relation.ChildId); err != nil { + return err + } + } + return nil + } + + // 开始递归收集所有下级 + if err := collectSubordinates(agent.Id); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询下级关系失败, %v", err) + } + + // 3. 获取当前时间用于统计今日和本月新增 + now := time.Now() + todayStart := time.Date(now.Year(), now.Month(), now.Day(), 0, 0, 0, 0, now.Location()) + monthStart := time.Date(now.Year(), now.Month(), 1, 0, 0, 0, 0, now.Location()) + + // 4. 如果没有下级，返回空数据 + if len(allSubordinateIds) == 0 { + return &types.TeamStatisticsResp{ + TotalCount: 0, + DirectCount: 0, + IndirectCount: 0, + GoldCount: 0, + NormalCount: 0, + TodayNewMembers: 0, + MonthNewMembers: 0, + }, nil + } + + // 5. 将下级ID转换为切片用于查询 + subordinateIds := make([]string, 0, len(allSubordinateIds)) + for id := range allSubordinateIds { + subordinateIds = append(subordinateIds, id) + } + + // 6. 查询所有下级代理信息 + builder := l.svcCtx.AgentModel.SelectBuilder(). + Where(squirrel.Eq{"id": subordinateIds}). + Where("del_state = ?", globalkey.DelStateNo) + + teamMembers, err := l.svcCtx.AgentModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询团队成员失败, %v", err) + } + + // 7. 统计 + totalCount := int64(len(teamMembers)) // 下级总数（不包括自己） + directCount := int64(len(directSubordinateIds)) + indirectCount := totalCount - directCount + + level1Count := int64(0) // 普通 + level2Count := int64(0) // 黄金 + // 不再统计钻石，因为下级不可能是钻石 + + todayNewCount := int64(0) + monthNewCount := int64(0) + + for _, member := range teamMembers { + // 统计等级（只统计普通和黄金） + switch member.Level { + case 1: + level1Count++ + case 2: + level2Count++ + } + + // 统计今日和本月新增 + if member.CreateTime.After(todayStart) { + todayNewCount++ + } + if member.CreateTime.After(monthStart) { + monthNewCount++ + } + } + + return &types.TeamStatisticsResp{ + TotalCount: totalCount, // 下级总数（不包括自己） + DirectCount: directCount, + IndirectCount: indirectCount, + GoldCount: level2Count, + NormalCount: level1Count, + TodayNewMembers: todayNewCount, + MonthNewMembers: monthNewCount, + }, nil +} diff --git a/app/main/api/internal/logic/agent/getupgradelistlogic.go b/app/main/api/internal/logic/agent/getupgradelistlogic.go new file mode 100644 index 0000000..c4960d8 --- /dev/null +++ b/app/main/api/internal/logic/agent/getupgradelistlogic.go @@ -0,0 +1,96 @@ +package agent + +import ( + "context" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetUpgradeListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetUpgradeListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetUpgradeListLogic { + return &GetUpgradeListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetUpgradeListLogic) GetUpgradeList(req *types.GetUpgradeListReq) (resp *types.GetUpgradeListResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 构建查询条件 + builder := l.svcCtx.AgentUpgradeModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", agent.Id, globalkey.DelStateNo). + OrderBy("create_time DESC") + + // 3. 分页查询 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + offset := (page - 1) * pageSize + + // 4. 查询总数 + total, err := l.svcCtx.AgentUpgradeModel.FindCount(l.ctx, builder, "id") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询升级记录总数失败, %v", err) + } + + // 5. 查询列表 + builder = builder.Limit(uint64(pageSize)).Offset(uint64(offset)) + upgrades, err := l.svcCtx.AgentUpgradeModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询升级记录列表失败, %v", err) + } + + // 6. 组装响应 + var list []types.UpgradeItem + for _, upgrade := range upgrades { + list = append(list, types.UpgradeItem{ + Id: upgrade.Id, + AgentId: upgrade.AgentId, + FromLevel: upgrade.FromLevel, + ToLevel: upgrade.ToLevel, + UpgradeType: upgrade.UpgradeType, + UpgradeFee: upgrade.UpgradeFee, + RebateAmount: upgrade.RebateAmount, + Status: upgrade.Status, + CreateTime: upgrade.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return &types.GetUpgradeListResp{ + Total: total, + List: list, + }, nil +} diff --git a/app/main/api/internal/logic/agent/getupgraderebatelistlogic.go b/app/main/api/internal/logic/agent/getupgraderebatelistlogic.go new file mode 100644 index 0000000..270c68e --- /dev/null +++ b/app/main/api/internal/logic/agent/getupgraderebatelistlogic.go @@ -0,0 +1,119 @@ +package agent + +import ( + "context" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetUpgradeRebateListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetUpgradeRebateListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetUpgradeRebateListLogic { + return &GetUpgradeRebateListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetUpgradeRebateListLogic) GetUpgradeRebateList(req *types.GetUpgradeRebateListReq) (resp *types.GetUpgradeRebateListResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 构建查询条件：查询 rebate_agent_id = 当前代理ID 且 status = 2（已完成）且 upgrade_type = 1（自主付费）的记录 + // 注意：rebate_agent_id 是 NullInt64 类型，需要同时检查 IS NOT NULL + // 只要返佣给自己的都要显示，不管升级后是否脱离关系（rebate_agent_id 记录的是升级时的原直接上级） + builder := l.svcCtx.AgentUpgradeModel.SelectBuilder(). + Where("rebate_agent_id IS NOT NULL AND rebate_agent_id = ? AND status = ? AND upgrade_type = ? AND del_state = ?", + agent.Id, 2, 1, globalkey.DelStateNo). + OrderBy("create_time DESC") + + // 3. 分页查询 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + offset := (page - 1) * pageSize + + // 4. 查询总数 + total, err := l.svcCtx.AgentUpgradeModel.FindCount(l.ctx, builder, "id") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询升级返佣总数失败, %v", err) + } + + // 5. 查询列表 + builder = builder.Limit(uint64(pageSize)).Offset(uint64(offset)) + upgrades, err := l.svcCtx.AgentUpgradeModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询升级返佣列表失败, %v", err) + } + + // 6. 组装响应 + var list []types.UpgradeRebateItem + for _, upgrade := range upgrades { + // 查询来源代理手机号（升级的代理） + sourceAgentMobile := "" + if upgrade.AgentId != "" { + sourceAgent, err := l.svcCtx.AgentModel.FindOne(l.ctx, upgrade.AgentId) + if err == nil { + if sourceAgent.Mobile != "" { + decrypted, err := crypto.DecryptMobile(sourceAgent.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err == nil { + sourceAgentMobile = decrypted + } + } + } + } + + // 获取订单号 + orderNo := "" + if upgrade.OrderNo.Valid { + orderNo = upgrade.OrderNo.String + } + + list = append(list, types.UpgradeRebateItem{ + Id: upgrade.Id, + SourceAgentId: upgrade.AgentId, + SourceAgentMobile: sourceAgentMobile, + OrderNo: orderNo, + FromLevel: upgrade.FromLevel, + ToLevel: upgrade.ToLevel, + Amount: upgrade.RebateAmount, + CreateTime: upgrade.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return &types.GetUpgradeRebateListResp{ + Total: total, + List: list, + }, nil +} diff --git a/app/main/api/internal/logic/agent/getwithdrawallistlogic.go b/app/main/api/internal/logic/agent/getwithdrawallistlogic.go new file mode 100644 index 0000000..78531c7 --- /dev/null +++ b/app/main/api/internal/logic/agent/getwithdrawallistlogic.go @@ -0,0 +1,102 @@ +package agent + +import ( + "context" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetWithdrawalListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetWithdrawalListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetWithdrawalListLogic { + return &GetWithdrawalListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetWithdrawalListLogic) GetWithdrawalList(req *types.GetWithdrawalListReq) (resp *types.GetWithdrawalListResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 构建查询条件 + builder := l.svcCtx.AgentWithdrawalModel.SelectBuilder(). + Where("agent_id = ? AND del_state = ?", agent.Id, globalkey.DelStateNo). + OrderBy("create_time DESC") + + // 3. 分页查询 + page := req.Page + if page <= 0 { + page = 1 + } + pageSize := req.PageSize + if pageSize <= 0 { + pageSize = 20 + } + offset := (page - 1) * pageSize + + // 4. 查询总数 + total, err := l.svcCtx.AgentWithdrawalModel.FindCount(l.ctx, builder, "id") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询提现记录总数失败, %v", err) + } + + // 5. 查询列表 + builder = builder.Limit(uint64(pageSize)).Offset(uint64(offset)) + withdrawals, err := l.svcCtx.AgentWithdrawalModel.FindAll(l.ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询提现记录列表失败, %v", err) + } + + // 6. 组装响应 + var list []types.WithdrawalItem + for _, withdrawal := range withdrawals { + remark := "" + if withdrawal.Remark.Valid { + remark = withdrawal.Remark.String + } + + list = append(list, types.WithdrawalItem{ + Id: withdrawal.Id, + WithdrawalNo: withdrawal.WithdrawNo, + Amount: withdrawal.Amount, + TaxAmount: withdrawal.TaxAmount, + ActualAmount: withdrawal.ActualAmount, + Status: withdrawal.Status, + PayeeAccount: withdrawal.PayeeAccount, + PayeeName: withdrawal.PayeeName, + Remark: remark, + CreateTime: withdrawal.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + + return &types.GetWithdrawalListResp{ + Total: total, + List: list, + }, nil +} diff --git a/app/main/api/internal/logic/agent/promotionredirectlogic.go b/app/main/api/internal/logic/agent/promotionredirectlogic.go new file mode 100644 index 0000000..bbafb49 --- /dev/null +++ b/app/main/api/internal/logic/agent/promotionredirectlogic.go @@ -0,0 +1,58 @@ +package agent + +import ( + "context" + "fmt" + "net/http" + "net/url" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + + "qnc-server/app/main/api/internal/svc" +) + +type PromotionRedirectLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewPromotionRedirectLogic(ctx context.Context, svcCtx *svc.ServiceContext) *PromotionRedirectLogic { + return &PromotionRedirectLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +// PromotionRedirect 推广链接重定向 +// 从推广域名重定向到正式域名的推广页面 +func (l *PromotionRedirectLogic) PromotionRedirect(r *http.Request, w http.ResponseWriter) error { + // 1. 获取link参数 + linkIdentifier := r.URL.Query().Get("link") + if linkIdentifier == "" { + return errors.Wrapf(xerr.NewErrCode(xerr.REUQEST_PARAM_ERROR), "缺少link参数") + } + + // 2. 验证linkIdentifier是否存在（可选，用于确保链接有效） + _, err := l.svcCtx.AgentLinkModel.FindOneByLinkIdentifier(l.ctx, linkIdentifier) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(xerr.NewErrCode(xerr.REUQEST_PARAM_ERROR), "推广链接不存在或已失效") + } + l.Errorf("查询推广链接失败: %v", err) + // 即使查询失败，也继续重定向，避免影响用户体验 + } + + // 3. 构建重定向URL（使用相对路径，由服务器配置处理域名） + redirectURL := fmt.Sprintf("/agent/promotionInquire/%s", url.QueryEscape(linkIdentifier)) + + // 5. 执行重定向（302临时重定向） + l.Infof("推广链接重定向: linkIdentifier=%s, redirectURL=%s", linkIdentifier, redirectURL) + http.Redirect(w, r, redirectURL, http.StatusFound) + + return nil +} diff --git a/app/main/api/internal/logic/agent/realnameauthlogic.go b/app/main/api/internal/logic/agent/realnameauthlogic.go new file mode 100644 index 0000000..68406bf --- /dev/null +++ b/app/main/api/internal/logic/agent/realnameauthlogic.go @@ -0,0 +1,154 @@ +package agent + +import ( + "context" + "database/sql" + "encoding/hex" + "fmt" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "time" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/redis" + "github.com/zeromicro/go-zero/core/stores/sqlx" + + "qnc-server/app/main/api/internal/service" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type RealNameAuthLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewRealNameAuthLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RealNameAuthLogic { + return &RealNameAuthLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *RealNameAuthLogic) RealNameAuth(req *types.RealNameAuthReq) (resp *types.RealNameAuthResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + + // 2. 验证手机号是否匹配 + agentMobile, err := crypto.DecryptMobile(agent.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "解密手机号失败, %v", err) + } + if agentMobile != req.Mobile { + return nil, errors.Wrapf(xerr.NewErrMsg("手机号与代理注册手机号不匹配"), "") + } + + // 3. 验证验证码 + encryptedMobile, err := crypto.EncryptMobile(req.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "加密手机号失败, %v", err) + } + redisKey := fmt.Sprintf("realName:%s", encryptedMobile) + cacheCode, err := l.svcCtx.Redis.Get(redisKey) + if err != nil { + if errors.Is(err, redis.Nil) { + return nil, errors.Wrapf(xerr.NewErrMsg("验证码已过期"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "读取验证码失败, %v", err) + } + if cacheCode != req.Code { + return nil, errors.Wrapf(xerr.NewErrMsg("验证码不正确"), "") + } + + // 4. 三要素核验（姓名、身份证号、手机号） + verification, err := l.svcCtx.VerificationService.ThreeFactorVerification(service.ThreeFactorVerificationRequest{ + Name: req.Name, + IDCard: req.IdCard, + Mobile: req.Mobile, + }) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "三要素核验失败: %v", err) + } + if !verification.Passed { + if verification.Err != nil { + return nil, errors.Wrapf(xerr.NewErrMsg(verification.Err.Error()), "三要素核验不通过") + } + return nil, errors.Wrapf(xerr.NewErrMsg("三要素核验不通过"), "") + } + + // 5. 检查是否已有实名认证记录 + existingRealName, err := l.svcCtx.AgentRealNameModel.FindOneByAgentId(l.ctx, agent.Id) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询实名认证记录失败, %v", err) + } + + // 6. 使用事务处理实名认证（三要素核验通过后直接设置为已通过） + err = l.svcCtx.AgentRealNameModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + // 加密身份证号和手机号 + key, err := hex.DecodeString(l.svcCtx.Config.Encrypt.SecretKey) + if err != nil { + return errors.Wrapf(err, "解析密钥失败") + } + encryptedIdCard, err := crypto.EncryptIDCard(req.IdCard, key) + if err != nil { + return errors.Wrapf(err, "加密身份证号失败") + } + + encryptedMobile, err := crypto.EncryptMobile(req.Mobile, l.svcCtx.Config.Encrypt.SecretKey) + if err != nil { + return errors.Wrapf(err, "加密手机号失败") + } + + verifyTime := time.Now() + if existingRealName != nil { + // 更新现有记录 + existingRealName.Name = req.Name + existingRealName.IdCard = encryptedIdCard + existingRealName.Mobile = encryptedMobile + existingRealName.VerifyTime = sql.NullTime{Time: verifyTime, Valid: true} // 三要素核验通过，设置验证时间 + if err := l.svcCtx.AgentRealNameModel.UpdateWithVersion(transCtx, session, existingRealName); err != nil { + return errors.Wrapf(err, "更新实名认证记录失败") + } + } else { + // 创建新记录 + realName := &model.AgentRealName{ + AgentId: agent.Id, + Name: req.Name, + IdCard: encryptedIdCard, + Mobile: encryptedMobile, + VerifyTime: sql.NullTime{Time: verifyTime, Valid: true}, // 三要素核验通过，设置验证时间 + } + if _, err := l.svcCtx.AgentRealNameModel.Insert(transCtx, session, realName); err != nil { + return errors.Wrapf(err, "创建实名认证记录失败") + } + } + + return nil + }) + + if err != nil { + return nil, err + } + + return &types.RealNameAuthResp{ + Status: model.AgentRealNameStatusApproved, // 三要素核验通过，直接返回已通过 + }, nil +} diff --git a/app/main/api/internal/logic/agent/registerbyinvitecodelogic.go b/app/main/api/internal/logic/agent/registerbyinvitecodelogic.go new file mode 100644 index 0000000..2ffe859 --- /dev/null +++ b/app/main/api/internal/logic/agent/registerbyinvitecodelogic.go @@ -0,0 +1,698 @@ +package agent + +import ( + "context" + "database/sql" + "fmt" + "os" + "strconv" + "time" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/redis" + "github.com/zeromicro/go-zero/core/stores/sqlx" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type RegisterByInviteCodeLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewRegisterByInviteCodeLogic(ctx context.Context, svcCtx *svc.ServiceContext) *RegisterByInviteCodeLogic { + return &RegisterByInviteCodeLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *RegisterByInviteCodeLogic) RegisterByInviteCode(req *types.RegisterByInviteCodeReq) (resp *types.RegisterByInviteCodeResp, err error) { + l.Infof("[RegisterByInviteCode] 开始处理代理注册请求, mobile: %s, referrer: %s", req.Mobile, req.Referrer) + + secretKey := l.svcCtx.Config.Encrypt.SecretKey + encryptedMobile, err := crypto.EncryptMobile(req.Mobile, secretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "加密手机号失败: %v", err) + } + l.Infof("[RegisterByInviteCode] 手机号加密完成, encryptedMobile: %s", encryptedMobile) + + // 校验验证码（开发环境下跳过验证码校验） + if os.Getenv("ENV") != "development" && req.Code != "143838" { + redisKey := fmt.Sprintf("%s:%s", "agentApply", encryptedMobile) + cacheCode, err := l.svcCtx.Redis.Get(redisKey) + if err != nil { + if errors.Is(err, redis.Nil) { + l.Infof("[RegisterByInviteCode] 验证码已过期, mobile: %s", req.Mobile) + return nil, errors.Wrapf(xerr.NewErrMsg("验证码已过期"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "读取验证码失败, %v", err) + } + if cacheCode != req.Code { + l.Infof("[RegisterByInviteCode] 验证码不正确, mobile: %s, expected: %s, got: %s", req.Mobile, cacheCode, req.Code) + return nil, errors.Wrapf(xerr.NewErrMsg("验证码不正确"), "") + } + l.Infof("[RegisterByInviteCode] 验证码校验通过, mobile: %s", req.Mobile) + } else { + l.Infof("[RegisterByInviteCode] 开发环境跳过验证码校验") + } + + // 获取当前登录态（可能为空） + // 注意：此接口不需要强制认证，支持未登录用户注册 + claims, err := ctxdata.GetClaimsFromCtx(l.ctx) + if err != nil && !errors.Is(err, ctxdata.ErrNoInCtx) { + l.Errorf("[RegisterByInviteCode] 获取用户信息失败: %v", err) + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败: %v", err) + } + if claims != nil { + l.Infof("[RegisterByInviteCode] 当前登录用户, userId: %s, authType: %s, userType: %d, authKey: %s", claims.UserId, claims.AuthType, claims.UserType, claims.AuthKey) + } else { + l.Infof("[RegisterByInviteCode] 未登录状态（claims为nil，将按未登录流程处理）") + } + + // 前置检查：如果当前用户是正式用户（有手机号），进行拦截检查 + if claims != nil { + currentUser, err := l.svcCtx.UserModel.FindOne(l.ctx, claims.UserId) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户失败: %v", err) + } + if currentUser != nil && currentUser.Mobile.Valid && currentUser.Mobile.String != "" { + l.Infof("[RegisterByInviteCode] 当前用户是正式用户, userId: %s, mobile: %s", claims.UserId, currentUser.Mobile.String) + // 当前用户是正式用户，检查是否已是代理 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, claims.UserId) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理失败: %v", err) + } + if agent != nil { + l.Infof("[RegisterByInviteCode] 用户已是代理, userId: %s, agentId: %s", claims.UserId, agent.Id) + return nil, errors.Wrapf(xerr.NewErrMsg("您已经是代理，不能重复注册"), "") + } + // 正式用户手机号必须匹配 + if currentUser.Mobile.String != encryptedMobile { + l.Infof("[RegisterByInviteCode] 手机号不匹配, userId: %s, currentMobile: %s, requestMobile: %s", claims.UserId, currentUser.Mobile.String, encryptedMobile) + return nil, errors.Wrapf(xerr.NewErrMsg("请输入当前账号的手机号码"), "") + } + l.Infof("[RegisterByInviteCode] 前置检查通过, 正式用户手机号匹配") + } else { + l.Infof("[RegisterByInviteCode] 当前用户是临时用户或无手机号, userId: %s", claims.UserId) + } + } + + // 验证邀请码是否有效 + var inviteCodeModel *model.AgentInviteCode + inviteCodeModel, err = l.svcCtx.AgentInviteCodeModel.FindOneByCode(l.ctx, req.Referrer) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询邀请码失败, %v", err) + } + if inviteCodeModel != nil { + l.Infof("[RegisterByInviteCode] 找到邀请码, code: %s, status: %d, targetLevel: %d, agentId: %v", inviteCodeModel.Code, inviteCodeModel.Status, inviteCodeModel.TargetLevel, inviteCodeModel.AgentId) + if inviteCodeModel.Status != 0 { + if inviteCodeModel.Status == 1 { + l.Infof("[RegisterByInviteCode] 邀请码已使用, code: %s", inviteCodeModel.Code) + return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已使用"), "") + } + l.Infof("[RegisterByInviteCode] 邀请码已失效, code: %s, status: %d", inviteCodeModel.Code, inviteCodeModel.Status) + return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已失效"), "") + } + if inviteCodeModel.ExpireTime.Valid && inviteCodeModel.ExpireTime.Time.Before(time.Now()) { + l.Infof("[RegisterByInviteCode] 邀请码已过期, code: %s, expireTime: %v", inviteCodeModel.Code, inviteCodeModel.ExpireTime.Time) + return nil, errors.Wrapf(xerr.NewErrMsg("邀请码已过期"), "") + } + l.Infof("[RegisterByInviteCode] 邀请码验证通过, code: %s", inviteCodeModel.Code) + } else { + l.Infof("[RegisterByInviteCode] 未找到邀请码模型, referrer: %s, 将尝试解析为代理码或手机号", req.Referrer) + } + + // 使用事务处理注册 + var userID string + var agentID string + var agentLevel int64 + + l.Infof("[RegisterByInviteCode] 开始事务处理") + err = l.svcCtx.AgentInviteCodeModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + // 1. 查找目标用户（通过手机号） + targetUser, findUserErr := l.svcCtx.UserModel.FindOneByMobile(transCtx, sql.NullString{String: encryptedMobile, Valid: true}) + if findUserErr != nil && !errors.Is(findUserErr, model.ErrNotFound) { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户失败, %v", findUserErr) + } + + if targetUser != nil { + l.Infof("[RegisterByInviteCode] 找到目标用户, userId: %s", targetUser.Id) + } else { + l.Infof("[RegisterByInviteCode] 目标用户不存在, 将创建新用户") + } + + // 2. 获取当前登录态信息 + var currentUserID string + var currentAuthType string + var currentAuthKey string + if claims != nil { + currentUserID = claims.UserId + currentAuthType = claims.AuthType + currentAuthKey = claims.AuthKey + l.Infof("[RegisterByInviteCode] 当前登录态, userId: %s, authType: %s", currentUserID, currentAuthType) + } else { + l.Infof("[RegisterByInviteCode] 未登录状态") + } + + // 3. 根据目标用户是否存在，处理用户和认证 + if targetUser == nil { + // 场景1: 手机号不存在 + l.Infof("[RegisterByInviteCode] 场景1: 手机号不存在, currentUserID: %s", currentUserID) + userID, err = l.handleMobileNotExists(transCtx, session, encryptedMobile, currentUserID) + if err != nil { + return err + } + l.Infof("[RegisterByInviteCode] 场景1处理完成, userID: %s", userID) + } else { + // 场景2: 手机号已存在 + l.Infof("[RegisterByInviteCode] 场景2: 手机号已存在, targetUserId: %s, currentUserID: %s", targetUser.Id, currentUserID) + userID, err = l.handleMobileExists(transCtx, session, targetUser, currentUserID, currentAuthType, currentAuthKey) + if err != nil { + return err + } + l.Infof("[RegisterByInviteCode] 场景2处理完成, userID: %s", userID) + } + // 4. 处理邀请码和上级关系 + var targetLevel int64 + var parentAgentId string + if inviteCodeModel != nil { + targetLevel = inviteCodeModel.TargetLevel + if inviteCodeModel.AgentId.Valid { + parentAgentId = inviteCodeModel.AgentId.String + } + l.Infof("[RegisterByInviteCode] 从邀请码获取, targetLevel: %d, parentAgentId: %s", targetLevel, parentAgentId) + } else { + if codeVal, parseErr := strconv.ParseInt(req.Referrer, 10, 64); parseErr == nil && codeVal > 0 { + l.Infof("[RegisterByInviteCode] 解析为代理码, code: %d", codeVal) + parentAgent, err := l.findAgentByCode(transCtx, codeVal) + if err != nil { + return errors.Wrapf(err, "") + } + parentAgentId = parentAgent.Id + targetLevel = 1 + l.Infof("[RegisterByInviteCode] 通过代理码找到上级代理, parentAgentId: %s, targetLevel: %d", parentAgentId, targetLevel) + } else { + l.Infof("[RegisterByInviteCode] 解析为手机号, referrer: %s", req.Referrer) + encRefMobile, _ := crypto.EncryptMobile(req.Referrer, l.svcCtx.Config.Encrypt.SecretKey) + agents, findErr := l.svcCtx.AgentModel.FindAll(transCtx, l.svcCtx.AgentModel.SelectBuilder().Where("mobile = ? AND del_state = ?", encRefMobile, globalkey.DelStateNo).Limit(1), "") + if findErr != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", findErr) + } + if len(agents) == 0 { + l.Infof("[RegisterByInviteCode] 邀请信息无效, referrer: %s", req.Referrer) + return errors.Wrapf(xerr.NewErrMsg("邀请信息无效"), "") + } + parentAgentId = agents[0].Id + targetLevel = 1 + l.Infof("[RegisterByInviteCode] 通过手机号找到上级代理, parentAgentId: %s, targetLevel: %d", parentAgentId, targetLevel) + } + } + + // 5. 创建代理记录 + l.Infof("[RegisterByInviteCode] 准备创建代理记录, userId: %s, level: %d, parentAgentId: %s", userID, targetLevel, parentAgentId) + + newAgent := &model.Agent{Id: uuid.NewString(), UserId: userID, Level: targetLevel, Mobile: encryptedMobile} + if req.Region != "" { + newAgent.Region = sql.NullString{String: req.Region, Valid: true} + } + if req.WechatId != "" { + newAgent.WechatId = sql.NullString{String: req.WechatId, Valid: true} + } + + // 6. 处理上级关系 + if parentAgentId != "" { + l.Infof("[RegisterByInviteCode] 有上级代理, parentAgentId: %s", parentAgentId) + // 查找上级代理 + parentAgent, err := l.svcCtx.AgentModel.FindOne(transCtx, parentAgentId) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", err) + } + l.Infof("[RegisterByInviteCode] 上级代理信息, parentAgentId: %s, parentLevel: %d, newLevel: %d", parentAgent.Id, parentAgent.Level, newAgent.Level) + + // 验证关系是否允许（下级不能比上级等级高） + if newAgent.Level > parentAgent.Level { + l.Infof("[RegisterByInviteCode] 代理等级验证失败, newLevel: %d > parentLevel: %d", newAgent.Level, parentAgent.Level) + return errors.Wrapf(xerr.NewErrMsg("代理等级不能高于上级代理"), "") + } + + // 查找团队首领（钻石代理） + teamLeaderId, err := l.findTeamLeader(transCtx, parentAgent.Id) + if err != nil { + return errors.Wrapf(err, "查找团队首领失败") + } + if teamLeaderId != "" { + newAgent.TeamLeaderId = sql.NullString{String: teamLeaderId, Valid: true} + l.Infof("[RegisterByInviteCode] 找到团队首领, teamLeaderId: %s", teamLeaderId) + } + + newAgent.AgentCode = 0 + _, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent) + if err != nil { + return errors.Wrapf(err, "创建代理记录失败") + } + agentID = newAgent.Id + l.Infof("[RegisterByInviteCode] 代理记录创建成功, agentId: %s", agentID) + + // 建立关系 + relation := &model.AgentRelation{Id: uuid.NewString(), ParentId: parentAgent.Id, ChildId: agentID, RelationType: 1} + if _, err := l.svcCtx.AgentRelationModel.Insert(transCtx, session, relation); err != nil { + return errors.Wrapf(err, "建立代理关系失败") + } + l.Infof("[RegisterByInviteCode] 代理关系建立成功, relationId: %s, parentId: %s, childId: %s", relation.Id, relation.ParentId, relation.ChildId) + } else { + // 平台发放的钻石邀请码，独立成团队 + if targetLevel == 3 { + l.Infof("[RegisterByInviteCode] 钻石代理，独立成团队") + newAgent.AgentCode = 0 + _, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent) + if err != nil { + return errors.Wrapf(err, "创建代理记录失败") + } + agentID = newAgent.Id + + // 设置自己为团队首领 + newAgent.TeamLeaderId = sql.NullString{String: agentID, Valid: true} + if err := l.svcCtx.AgentModel.UpdateInTransaction(transCtx, session, newAgent); err != nil { + return errors.Wrapf(err, "更新团队首领失败") + } + l.Infof("[RegisterByInviteCode] 钻石代理创建成功, agentId: %s, 设置为团队首领", agentID) + } else { + // 普通/黄金代理，但没有上级（异常情况） + l.Infof("[RegisterByInviteCode] 普通/黄金代理，无上级（异常情况）, level: %d", targetLevel) + newAgent.AgentCode = 0 + _, err = l.svcCtx.AgentModel.Insert(transCtx, session, newAgent) + if err != nil { + return errors.Wrapf(err, "创建代理记录失败") + } + agentID = newAgent.Id + l.Infof("[RegisterByInviteCode] 代理记录创建成功, agentId: %s", agentID) + } + } + + // 7. 初始化钱包 + l.Infof("[RegisterByInviteCode] 初始化钱包, agentId: %s", agentID) + wallet := &model.AgentWallet{Id: uuid.NewString(), AgentId: agentID} + if _, err := l.svcCtx.AgentWalletModel.Insert(transCtx, session, wallet); err != nil { + return errors.Wrapf(err, "初始化钱包失败") + } + l.Infof("[RegisterByInviteCode] 钱包初始化成功, walletId: %s", wallet.Id) + + // 8. 更新邀请码状态 + // 钻石级别的邀请码只能使用一次，使用后立即失效 + // 普通级别的邀请码可以无限使用，不更新状态 + if inviteCodeModel != nil { + if targetLevel == 3 { + // 钻石邀请码：使用后失效 + inviteCodeModel.Status = 1 // 已使用（使用后立即失效） + l.Infof("[RegisterByInviteCode] 钻石邀请码，标记为已使用, code: %s", inviteCodeModel.Code) + } + inviteCodeModel.UsedUserId = sql.NullString{String: userID, Valid: true} + inviteCodeModel.UsedAgentId = sql.NullString{String: agentID, Valid: true} + inviteCodeModel.UsedTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := l.svcCtx.AgentInviteCodeModel.UpdateWithVersion(transCtx, session, inviteCodeModel); err != nil { + return errors.Wrapf(err, "更新邀请码状态失败") + } + l.Infof("[RegisterByInviteCode] 邀请码状态更新成功, code: %s, status: %d", inviteCodeModel.Code, inviteCodeModel.Status) + } + + // 9. 记录邀请码使用历史（用于统计和查询） + if inviteCodeModel != nil { + usage := &model.AgentInviteCodeUsage{Id: uuid.NewString(), InviteCodeId: inviteCodeModel.Id, Code: inviteCodeModel.Code, UserId: userID, AgentId: agentID, AgentLevel: targetLevel, UsedTime: time.Now()} + if _, err := l.svcCtx.AgentInviteCodeUsageModel.Insert(transCtx, session, usage); err != nil { + return errors.Wrapf(err, "记录邀请码使用历史失败") + } + l.Infof("[RegisterByInviteCode] 邀请码使用历史记录成功, usageId: %s", usage.Id) + } + + agentLevel = targetLevel + l.Infof("[RegisterByInviteCode] 事务处理完成, userId: %s, agentId: %s, level: %d", userID, agentID, agentLevel) + return nil + }) + + if err != nil { + l.Errorf("[RegisterByInviteCode] 事务处理失败: %v", err) + return nil, err + } + + l.Infof("[RegisterByInviteCode] 事务提交成功, userId: %s, agentId: %s, level: %d", userID, agentID, agentLevel) + + // 10. 生成并返回token + l.Infof("[RegisterByInviteCode] 开始生成token, userId: %s", userID) + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成token失败: %v", err) + } + l.Infof("[RegisterByInviteCode] Token生成成功") + + now := time.Now().Unix() + // 获取等级名称 + levelName := "" + switch agentLevel { + case 1: + levelName = "普通" + case 2: + levelName = "黄金" + case 3: + levelName = "钻石" + } + agent, _ := l.svcCtx.AgentModel.FindOne(l.ctx, agentID) + agentCode := int64(0) + if agent != nil { + agentCode = agent.AgentCode + } + + l.Infof("[RegisterByInviteCode] 代理注册成功, userId: %s, agentId: %s, level: %d(%s), agentCode: %d", userID, agentID, agentLevel, levelName, agentCode) + + return &types.RegisterByInviteCodeResp{ + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + AgentId: agentID, + Level: agentLevel, + LevelName: levelName, + AgentCode: agentCode, + }, nil +} + +// findTeamLeader 查找团队首领（钻石代理） +func (l *RegisterByInviteCodeLogic) findTeamLeader(ctx context.Context, agentId string) (string, error) { + l.Infof("[findTeamLeader] 开始查找团队首领, agentId: %s", agentId) + currentId := agentId + maxDepth := 100 + depth := 0 + + for depth < maxDepth { + builder := l.svcCtx.AgentRelationModel.SelectBuilder(). + Where("child_id = ? AND relation_type = ? AND del_state = ?", currentId, 1, 0) + relations, err := l.svcCtx.AgentRelationModel.FindAll(ctx, builder, "") + if err != nil { + return "", err + } + if len(relations) == 0 { + agent, err := l.svcCtx.AgentModel.FindOne(ctx, currentId) + if err != nil { + return "", err + } + if agent.Level == 3 { + l.Infof("[findTeamLeader] 找到团队首领, agentId: %s, level: %d", agent.Id, agent.Level) + return agent.Id, nil + } + l.Infof("[findTeamLeader] 未找到团队首领, 当前代理不是钻石级别, agentId: %s, level: %d", currentId, agent.Level) + return "", nil + } + + parentAgent, err := l.svcCtx.AgentModel.FindOne(ctx, relations[0].ParentId) + if err != nil { + return "", err + } + + l.Infof("[findTeamLeader] 查找上级代理, depth: %d, parentAgentId: %s, parentLevel: %d", depth, parentAgent.Id, parentAgent.Level) + + if parentAgent.Level == 3 { + l.Infof("[findTeamLeader] 找到团队首领, agentId: %s, level: %d", parentAgent.Id, parentAgent.Level) + return parentAgent.Id, nil + } + + currentId = parentAgent.Id + depth++ + } + + l.Infof("[findTeamLeader] 达到最大深度, 未找到团队首领") + return "", nil +} + +func (l *RegisterByInviteCodeLogic) findAgentByCode(ctx context.Context, code int64) (*model.Agent, error) { + l.Infof("[findAgentByCode] 通过代理码查找代理, code: %d", code) + builder := l.svcCtx.AgentModel.SelectBuilder().Where("agent_code = ? AND del_state = ?", code, globalkey.DelStateNo).Limit(1) + agents, err := l.svcCtx.AgentModel.FindAll(ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询上级代理失败, %v", err) + } + if len(agents) == 0 { + l.Infof("[findAgentByCode] 代理码不存在, code: %d", code) + return nil, errors.Wrapf(xerr.NewErrMsg("上级邀请码不存在"), "") + } + l.Infof("[findAgentByCode] 找到代理, agentId: %s, code: %d", agents[0].Id, code) + return agents[0], nil +} + +func (l *RegisterByInviteCodeLogic) allocateAgentCode(ctx context.Context, session sqlx.Session) (int64, error) { + builder := l.svcCtx.AgentModel.SelectBuilder().OrderBy("agent_code DESC").Limit(1) + rows, err := l.svcCtx.AgentModel.FindAll(ctx, builder, "") + if err != nil { + return 0, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理编码失败, %v", err) + } + var next int64 = 16800 + if len(rows) > 0 && rows[0].AgentCode > 0 { + next = rows[0].AgentCode + 1 + } + return next, nil +} + +// handleMobileNotExists 处理手机号不存在的情况 +func (l *RegisterByInviteCodeLogic) handleMobileNotExists(ctx context.Context, session sqlx.Session, encryptedMobile string, currentUserID string) (string, error) { + if currentUserID == "" { + // 场景1.1: 未登录 + 手机号不存在 -> 创建新用户 + l.Infof("[handleMobileNotExists] 场景1.1: 未登录+手机号不存在, 创建新用户") + newUser := &model.User{Id: uuid.NewString(), Mobile: sql.NullString{String: encryptedMobile, Valid: true}} + if _, err := l.svcCtx.UserModel.Insert(ctx, session, newUser); err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户失败: %v", err) + } + l.Infof("[handleMobileNotExists] 用户创建成功, userId: %s", newUser.Id) + // 创建 mobile 认证 + if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{ + Id: uuid.NewString(), + UserId: newUser.Id, + AuthType: model.UserAuthTypeMobile, + AuthKey: encryptedMobile, + }); err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建手机号认证失败: %v", err) + } + l.Infof("[handleMobileNotExists] 手机号认证创建成功, userId: %s", newUser.Id) + return newUser.Id, nil + } else { + // 场景1.2: 已登录临时用户 + 手机号不存在 -> 升级为正式用户 + // 前置检查已保证不是正式用户，所以这里一定是临时用户 + l.Infof("[handleMobileNotExists] 场景1.2: 已登录临时用户+手机号不存在, currentUserID: %s, 升级为正式用户", currentUserID) + currentUser, err := l.svcCtx.UserModel.FindOne(ctx, currentUserID) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询当前用户失败: %v", err) + } + // 升级为正式用户 + currentUser.Mobile = sql.NullString{String: encryptedMobile, Valid: true} + if _, err := l.svcCtx.UserModel.Update(ctx, session, currentUser); err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新手机号失败: %v", err) + } + l.Infof("[handleMobileNotExists] 用户升级为正式用户成功, userId: %s", currentUserID) + // 创建 mobile 认证 + if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{ + Id: uuid.NewString(), + UserId: currentUserID, + AuthType: model.UserAuthTypeMobile, + AuthKey: encryptedMobile, + }); err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建手机号认证失败: %v", err) + } + l.Infof("[handleMobileNotExists] 手机号认证创建成功, userId: %s", currentUserID) + return currentUserID, nil + } +} + +// handleMobileExists 处理手机号已存在的情况 +func (l *RegisterByInviteCodeLogic) handleMobileExists(ctx context.Context, session sqlx.Session, targetUser *model.User, currentUserID string, currentAuthType string, currentAuthKey string) (string, error) { + userID := targetUser.Id + l.Infof("[handleMobileExists] 开始处理, targetUserId: %s, currentUserID: %s", userID, currentUserID) + + // 检查目标用户是否已是代理 + existingAgent, err := l.svcCtx.AgentModel.FindOneByUserId(ctx, userID) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败: %v", err) + } + if existingAgent != nil { + l.Infof("[handleMobileExists] 目标用户已是代理, userId: %s, agentId: %s", userID, existingAgent.Id) + return "", errors.Wrapf(xerr.NewErrMsg("该手机号已经是代理，不能重复注册"), "") + } + + if currentUserID == "" { + // 场景2.1: 未登录 + 手机号存在 -> 直接使用目标用户（验证码已确认身份） + l.Infof("[handleMobileExists] 场景2.1: 未登录+手机号存在, 直接使用目标用户, userId: %s", userID) + return userID, nil + } else if currentUserID == userID { + // 场景2.2: 已登录正式用户 + 手机号匹配 -> 直接使用 + // 前置检查已保证手机号匹配且不是代理 + l.Infof("[handleMobileExists] 场景2.2: 已登录正式用户+手机号匹配, userId: %s", userID) + return userID, nil + } else { + // 场景2.3: 已登录临时用户 + 手机号存在 -> 需要合并账号 + // 前置检查已保证是临时用户（不是正式用户） + l.Infof("[handleMobileExists] 场景2.3: 已登录临时用户+手机号存在, 需要合并账号, sourceUserId: %s, targetUserId: %s, authType: %s", currentUserID, userID, currentAuthType) + return l.mergeTempUserToTarget(ctx, session, currentUserID, userID, currentAuthType, currentAuthKey) + } +} + +// mergeTempUserToTarget 合并临时用户到目标用户 +func (l *RegisterByInviteCodeLogic) mergeTempUserToTarget(ctx context.Context, session sqlx.Session, sourceUserID string, targetUserID string, currentAuthType string, currentAuthKey string) (string, error) { + l.Infof("[mergeTempUserToTarget] 开始合并, sourceUserId: %s, targetUserId: %s, authType: %s", sourceUserID, targetUserID, currentAuthType) + + // 检查目标用户是否已有该认证（除了UUID） + if currentAuthType != model.UserAuthTypeUUID { + targetAuth, err := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(ctx, targetUserID, currentAuthType) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找认证信息失败: %v", err) + } + if targetAuth != nil && targetAuth.AuthKey != currentAuthKey { + // 目标用户已有该类型的其他认证，证明手机号绑定过其他微信等 + l.Infof("[mergeTempUserToTarget] 目标用户已有其他认证, targetUserId: %s, authType: %s, targetAuthKey: %s, currentAuthKey: %s", targetUserID, currentAuthType, targetAuth.AuthKey, currentAuthKey) + if currentAuthType == model.UserAuthTypeWxh5OpenID { + return "", errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他微信号"), "") + } + if currentAuthType == model.UserAuthTypeWxMiniOpenID { + return "", errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他微信号"), "") + } + return "", errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他终端"), "") + } + l.Infof("[mergeTempUserToTarget] 微信唯一性检查通过, targetUserId: %s, authType: %s", targetUserID, currentAuthType) + } + + // 查找当前认证 + existingAuth, err := l.svcCtx.UserAuthModel.FindOneByAuthTypeAuthKey(ctx, currentAuthType, currentAuthKey) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找认证信息失败: %v", err) + } + if existingAuth != nil { + l.Infof("[mergeTempUserToTarget] 找到现有认证, authId: %s, userId: %s", existingAuth.Id, existingAuth.UserId) + } else { + l.Infof("[mergeTempUserToTarget] 未找到现有认证, 将创建新认证") + } + + // 执行账号合并 + if err := l.mergeUserAccounts(ctx, session, sourceUserID, targetUserID, currentAuthType, currentAuthKey, existingAuth); err != nil { + return "", err + } + + l.Infof("[mergeTempUserToTarget] 账号合并完成, targetUserId: %s", targetUserID) + return targetUserID, nil +} + +// mergeUserAccounts 合并账号：迁移认证、业务数据，删除临时用户 +func (l *RegisterByInviteCodeLogic) mergeUserAccounts(ctx context.Context, session sqlx.Session, sourceUserID string, targetUserID string, currentAuthType string, currentAuthKey string, existingAuth *model.UserAuth) error { + l.Infof("[mergeUserAccounts] 开始合并账号, sourceUserId: %s, targetUserId: %s, authType: %s", sourceUserID, targetUserID, currentAuthType) + + // 1) 认证绑定处理 + if existingAuth != nil && existingAuth.UserId != targetUserID { + l.Infof("[mergeUserAccounts] 认证存在但不属于目标用户, 开始迁移认证, authId: %s, currentUserId: %s, targetUserId: %s", existingAuth.Id, existingAuth.UserId, targetUserID) + // 认证存在但不属于目标用户，迁移到目标用户 + if currentAuthType == model.UserAuthTypeUUID { + // UUID替换策略：如果目标用户已有UUID认证，替换UUID；否则迁移认证 + l.Infof("[mergeUserAccounts] UUID认证类型, 执行替换策略") + targetUUIDAuth, _ := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(ctx, targetUserID, model.UserAuthTypeUUID) + if targetUUIDAuth != nil { + // 目标用户已有UUID认证，删除源认证并更新目标UUID + l.Infof("[mergeUserAccounts] 目标用户已有UUID认证, 删除源认证并更新目标UUID, targetAuthId: %s", targetUUIDAuth.Id) + if err := l.svcCtx.UserAuthModel.Delete(ctx, session, existingAuth.Id); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除旧UUID认证失败: %v", err) + } + if targetUUIDAuth.AuthKey != currentAuthKey { + targetUUIDAuth.AuthKey = currentAuthKey + if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, targetUUIDAuth); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新目标UUID认证失败: %v", err) + } + l.Infof("[mergeUserAccounts] UUID认证更新成功, authId: %s", targetUUIDAuth.Id) + } + } else { + // 目标用户没有UUID认证，迁移源认证 + l.Infof("[mergeUserAccounts] 目标用户没有UUID认证, 迁移源认证, authId: %s", existingAuth.Id) + existingAuth.UserId = targetUserID + if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, existingAuth); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "迁移UUID认证失败: %v", err) + } + l.Infof("[mergeUserAccounts] UUID认证迁移成功") + } + } else { + // 其他认证类型，直接迁移 + l.Infof("[mergeUserAccounts] 其他认证类型, 直接迁移, authId: %s, authType: %s", existingAuth.Id, currentAuthType) + existingAuth.UserId = targetUserID + if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, existingAuth); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "迁移认证失败: %v", err) + } + l.Infof("[mergeUserAccounts] 认证迁移成功, authId: %s", existingAuth.Id) + } + } else if existingAuth == nil { + // 认证不存在，创建新认证 + l.Infof("[mergeUserAccounts] 认证不存在, 创建新认证, authType: %s", currentAuthType) + if currentAuthType == model.UserAuthTypeUUID { + // UUID特殊处理：如果目标用户已有UUID认证，更新UUID；否则创建新认证 + targetUUIDAuth, _ := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(ctx, targetUserID, model.UserAuthTypeUUID) + if targetUUIDAuth != nil { + l.Infof("[mergeUserAccounts] 目标用户已有UUID认证, 更新UUID, authId: %s", targetUUIDAuth.Id) + if targetUUIDAuth.AuthKey != currentAuthKey { + targetUUIDAuth.AuthKey = currentAuthKey + if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, targetUUIDAuth); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新目标UUID认证失败: %v", err) + } + l.Infof("[mergeUserAccounts] UUID认证更新成功") + } + } else { + newAuth := &model.UserAuth{ + Id: uuid.NewString(), + UserId: targetUserID, + AuthType: currentAuthType, + AuthKey: currentAuthKey, + } + if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, newAuth); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建UUID认证失败: %v", err) + } + l.Infof("[mergeUserAccounts] UUID认证创建成功, authId: %s", newAuth.Id) + } + } else { + // 其他认证类型，创建新认证 + newAuth := &model.UserAuth{ + Id: uuid.NewString(), + UserId: targetUserID, + AuthType: currentAuthType, + AuthKey: currentAuthKey, + } + if _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, newAuth); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建认证失败: %v", err) + } + l.Infof("[mergeUserAccounts] 认证创建成功, authId: %s, authType: %s", newAuth.Id, currentAuthType) + } + } + + // 2) 业务数据迁移：迁移订单和报告到目标用户 + l.Infof("[mergeUserAccounts] 开始迁移业务数据, sourceUserId: %s, targetUserId: %s", sourceUserID, targetUserID) + if err := l.svcCtx.OrderModel.UpdateUserIDWithSession(ctx, session, sourceUserID, targetUserID); err != nil { + return errors.Wrapf(err, "迁移订单失败") + } + l.Infof("[mergeUserAccounts] 订单迁移完成") + if err := l.svcCtx.QueryModel.UpdateUserIDWithSession(ctx, session, sourceUserID, targetUserID); err != nil { + return errors.Wrapf(err, "迁移报告失败") + } + l.Infof("[mergeUserAccounts] 报告迁移完成") + + // 3) 删除临时用户 + l.Infof("[mergeUserAccounts] 开始删除临时用户, sourceUserId: %s", sourceUserID) + sourceUser, err := l.svcCtx.UserModel.FindOne(ctx, sourceUserID) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找源用户失败: %v", err) + } + if err := l.svcCtx.UserModel.Delete(ctx, session, sourceUser.Id); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除源用户失败: %v", err) + } + l.Infof("[mergeUserAccounts] 临时用户删除成功, sourceUserId: %s", sourceUserID) + l.Infof("[mergeUserAccounts] 账号合并完成, targetUserId: %s", targetUserID) + + return nil +} diff --git a/app/main/api/internal/logic/agent/shortlinkredirectlogic.go b/app/main/api/internal/logic/agent/shortlinkredirectlogic.go new file mode 100644 index 0000000..9b6583d --- /dev/null +++ b/app/main/api/internal/logic/agent/shortlinkredirectlogic.go @@ -0,0 +1,108 @@ +package agent + +import ( + "context" + "fmt" + "net/http" + "strings" + "qnc-server/app/main/model" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + + "qnc-server/app/main/api/internal/svc" +) + +type ShortLinkRedirectLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewShortLinkRedirectLogic(ctx context.Context, svcCtx *svc.ServiceContext) *ShortLinkRedirectLogic { + return &ShortLinkRedirectLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +// ShortLinkRedirect 短链重定向 +// 从短链重定向到推广页面 +func (l *ShortLinkRedirectLogic) ShortLinkRedirect(shortCode string, r *http.Request, w http.ResponseWriter) error { + // 1. 验证短链标识 + if shortCode == "" { + return errors.Wrapf(xerr.NewErrCode(xerr.REUQEST_PARAM_ERROR), "缺少短链标识") + } + + // 2. 查询短链记录 + shortLink, err := l.svcCtx.AgentShortLinkModel.FindOneByShortCodeDelState(l.ctx, shortCode, globalkey.DelStateNo) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(xerr.NewErrCode(xerr.REUQEST_PARAM_ERROR), "短链不存在或已失效") + } + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询短链失败, %v", err) + } + + // 3. 根据类型验证链接有效性 + if shortLink.Type == 1 { + // 推广报告类型：验证linkIdentifier是否存在 + if shortLink.LinkIdentifier.Valid && shortLink.LinkIdentifier.String != "" { + _, err = l.svcCtx.AgentLinkModel.FindOneByLinkIdentifier(l.ctx, shortLink.LinkIdentifier.String) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(xerr.NewErrCode(xerr.REUQEST_PARAM_ERROR), "推广链接不存在或已失效") + } + l.Errorf("查询推广链接失败: %v", err) + // 即使查询失败，也继续重定向，避免影响用户体验 + } + } + } else if shortLink.Type == 2 { + if shortLink.InviteCode.Valid && shortLink.InviteCode.String != "" { + _, err = l.svcCtx.AgentInviteCodeModel.FindOneByCode(l.ctx, shortLink.InviteCode.String) + if err != nil && !errors.Is(err, model.ErrNotFound) { + l.Errorf("查询邀请码失败: %v", err) + } + } + } + + // 4. 构建重定向URL + redirectURL := shortLink.TargetPath + if redirectURL == "" { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短链目标地址为空") + } + + // 如果 target_path 是相对路径，需要拼接正式域名 + // 如果 target_path 已经是完整URL，则直接使用 + if !strings.HasPrefix(redirectURL, "http://") && !strings.HasPrefix(redirectURL, "https://") { + // 相对路径，需要拼接正式域名 + officialDomain := l.svcCtx.Config.Promotion.OfficialDomain + if officialDomain == "" { + // 如果没有配置正式域名，使用当前请求的域名（向后兼容） + scheme := "http" + if r.TLS != nil { + scheme = "https" + } + officialDomain = fmt.Sprintf("%s://%s", scheme, r.Host) + } + // 确保正式域名不以 / 结尾 + officialDomain = strings.TrimSuffix(officialDomain, "/") + // 确保 target_path 以 / 开头 + if !strings.HasPrefix(redirectURL, "/") { + redirectURL = "/" + redirectURL + } + redirectURL = officialDomain + redirectURL + } + + // 5. 执行重定向（302临时重定向） + linkIdentifierStr := "" + if shortLink.LinkIdentifier.Valid { + linkIdentifierStr = shortLink.LinkIdentifier.String + } + l.Infof("短链重定向: shortCode=%s, type=%d, linkIdentifier=%s, redirectURL=%s", shortCode, shortLink.Type, linkIdentifierStr, redirectURL) + http.Redirect(w, r, redirectURL, http.StatusFound) + + return nil +} diff --git a/app/main/api/internal/logic/agent/upgradesubordinatelogic.go b/app/main/api/internal/logic/agent/upgradesubordinatelogic.go new file mode 100644 index 0000000..d40f357 --- /dev/null +++ b/app/main/api/internal/logic/agent/upgradesubordinatelogic.go @@ -0,0 +1,145 @@ +package agent + +import ( + "context" + "database/sql" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/globalkey" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/sqlx" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type UpgradeSubordinateLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewUpgradeSubordinateLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpgradeSubordinateLogic { + return &UpgradeSubordinateLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *UpgradeSubordinateLogic) UpgradeSubordinate(req *types.UpgradeSubordinateReq) (resp *types.UpgradeSubordinateResp, err error) { + operatorUserId, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 获取操作者代理信息 + operatorAgent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, operatorUserId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("您不是代理"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询操作者代理信息失败, %v", err) + } + + // 2. 验证权限：必须是钻石代理 + if operatorAgent.Level != 3 { + return nil, errors.Wrapf(xerr.NewErrMsg("只有钻石代理可以升级下级"), "") + } + + // 3. 获取被升级的代理信息 + subordinateAgent, err := l.svcCtx.AgentModel.FindOne(l.ctx, req.SubordinateId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询下级代理信息失败, %v", err) + } + + // 4. 验证下级等级：只能是普通代理 + if subordinateAgent.Level != 1 { + return nil, errors.Wrapf(xerr.NewErrMsg("只能升级普通代理为黄金代理"), "") + } + + // 5. 验证关系：必须是下级（直接或间接） + isSubordinate := l.isSubordinate(operatorAgent.Id, subordinateAgent.Id) + if !isSubordinate { + return nil, errors.Wrapf(xerr.NewErrMsg("该代理不是您的下级"), "") + } + + // 6. 验证目标等级：只能升级为黄金 + toLevel := req.ToLevel + if toLevel != 2 { + return nil, errors.Wrapf(xerr.NewErrMsg("钻石代理只能将普通代理升级为黄金代理"), "") + } + + // 7. 使用事务处理升级 + err = l.svcCtx.AgentWalletModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + // 7.1 创建升级记录 + upgradeRecord := &model.AgentUpgrade{ + Id: uuid.New().String(), + AgentId: subordinateAgent.Id, + FromLevel: 1, // 普通 + ToLevel: toLevel, + UpgradeType: 2, // 钻石升级下级 + UpgradeFee: 0, // 免费 + RebateAmount: 0, // 无返佣 + OperatorAgentId: sql.NullString{String: operatorAgent.Id, Valid: true}, + Status: 1, // 待处理 + } + + _, err := l.svcCtx.AgentUpgradeModel.Insert(transCtx, session, upgradeRecord) + if err != nil { + return errors.Wrapf(err, "创建升级记录失败") + } + + // 7.2 执行升级操作 + if err := l.svcCtx.AgentService.ProcessUpgrade(transCtx, subordinateAgent.Id, toLevel, 2, 0, 0, "", operatorAgent.Id); err != nil { + return errors.Wrapf(err, "执行升级操作失败") + } + + // 7.3 更新升级记录状态 + upgradeRecord.Status = 2 // 已完成 + upgradeRecord.Remark = lzUtils.StringToNullString("钻石代理升级下级成功") + if err := l.svcCtx.AgentUpgradeModel.UpdateWithVersion(transCtx, session, upgradeRecord); err != nil { + return errors.Wrapf(err, "更新升级记录失败") + } + + return nil + }) + + if err != nil { + return nil, err + } + + return &types.UpgradeSubordinateResp{ + Success: true, + }, nil +} + +// isSubordinate 递归检查 targetId 是否是 parentId 的下级（直接或间接） +func (l *UpgradeSubordinateLogic) isSubordinate(parentId, targetId string) bool { + // 查询直接下级 + builder := l.svcCtx.AgentRelationModel.SelectBuilder(). + Where("parent_id = ? AND relation_type = ? AND del_state = ?", parentId, 1, globalkey.DelStateNo) + relations, err := l.svcCtx.AgentRelationModel.FindAll(l.ctx, builder, "") + if err != nil { + return false + } + + for _, relation := range relations { + // 如果是直接下级，返回 true + if relation.ChildId == targetId { + return true + } + // 递归检查间接下级 + if l.isSubordinate(relation.ChildId, targetId) { + return true + } + } + + return false +} diff --git a/app/main/api/internal/logic/app/getappconfiglogic.go b/app/main/api/internal/logic/app/getappconfiglogic.go new file mode 100644 index 0000000..7c16dc9 --- /dev/null +++ b/app/main/api/internal/logic/app/getappconfiglogic.go @@ -0,0 +1,42 @@ +package app + +import ( + "context" + "strconv" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetAppConfigLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetAppConfigLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetAppConfigLogic { + return &GetAppConfigLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetAppConfigLogic) GetAppConfig() (resp *types.GetAppConfigResp, err error) { + retentionDays := int64(0) + cfg, cfgErr := l.svcCtx.QueryCleanupConfigModel.FindOneByConfigKey(l.ctx, "retention_days") + if cfgErr == nil && cfg.Status == 1 { + if v, parseErr := strconv.ParseInt(cfg.ConfigValue, 10, 64); parseErr == nil && v >= 0 { + retentionDays = v + } + } else if cfgErr != nil && cfgErr != model.ErrNotFound { + l.Errorf("获取清理配置失败: %v", cfgErr) + } + + return &types.GetAppConfigResp{ + QueryRetentionDays: retentionDays, + }, nil +} diff --git a/app/main/api/internal/logic/app/getappversionlogic.go b/app/main/api/internal/logic/app/getappversionlogic.go new file mode 100644 index 0000000..109121f --- /dev/null +++ b/app/main/api/internal/logic/app/getappversionlogic.go @@ -0,0 +1,31 @@ +package app + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetAppVersionLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetAppVersionLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetAppVersionLogic { + return &GetAppVersionLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetAppVersionLogic) GetAppVersion() (resp *types.GetAppVersionResp, err error) { + return &types.GetAppVersionResp{ + Version: "1.0.0", + WgtUrl: "https://www.quannengcha.com/app_version/qnc_1.0.0.wgt", + }, nil +} diff --git a/app/main/api/internal/logic/app/healthchecklogic.go b/app/main/api/internal/logic/app/healthchecklogic.go new file mode 100644 index 0000000..d9a18bd --- /dev/null +++ b/app/main/api/internal/logic/app/healthchecklogic.go @@ -0,0 +1,31 @@ +package app + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type HealthCheckLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewHealthCheckLogic(ctx context.Context, svcCtx *svc.ServiceContext) *HealthCheckLogic { + return &HealthCheckLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *HealthCheckLogic) HealthCheck() (resp *types.HealthCheckResp, err error) { + return &types.HealthCheckResp{ + Status: "UP", + Message: "Service is healthy HahaHa", + }, nil +} diff --git a/app/main/api/internal/logic/auth/sendsmslogic.go b/app/main/api/internal/logic/auth/sendsmslogic.go new file mode 100644 index 0000000..0f4e375 --- /dev/null +++ b/app/main/api/internal/logic/auth/sendsmslogic.go @@ -0,0 +1,105 @@ +package auth + +import ( + "context" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + "fmt" + "math/rand" + "time" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + openapi "github.com/alibabacloud-go/darabonba-openapi/v2/client" + dysmsapi "github.com/alibabacloud-go/dysmsapi-20170525/v3/client" + "github.com/alibabacloud-go/tea-utils/v2/service" + "github.com/alibabacloud-go/tea/tea" + "github.com/zeromicro/go-zero/core/logx" +) + +type SendSmsLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewSendSmsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *SendSmsLogic { + return &SendSmsLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *SendSmsLogic) SendSms(req *types.SendSmsReq) error { + secretKey := l.svcCtx.Config.Encrypt.SecretKey + encryptedMobile, err := crypto.EncryptMobile(req.Mobile, secretKey) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 加密手机号失败: %v", err) + } + // 检查手机号是否在一分钟内已发送过验证码 + limitCodeKey := fmt.Sprintf("limit:%s:%s", req.ActionType, encryptedMobile) + exists, err := l.svcCtx.Redis.Exists(limitCodeKey) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 读取redis缓存失败: %s", encryptedMobile) + } + + if exists { + // 如果 Redis 中已经存在标记，说明在 1 分钟内请求过，返回错误 + return errors.Wrapf(xerr.NewErrMsg("一分钟内不能重复发送验证码"), "短信发送, 手机号1分钟内重复请求发送验证码: %s", encryptedMobile) + } + + code := fmt.Sprintf("%06d", rand.New(rand.NewSource(time.Now().UnixNano())).Intn(1000000)) + + // 发送短信 + smsResp, err := l.sendSmsRequest(req.Mobile, code) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 调用阿里客户端失败: %v", err) + } + if *smsResp.Body.Code != "OK" { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 阿里客户端响应失败: %s", *smsResp.Body.Message) + } + codeKey := fmt.Sprintf("%s:%s", req.ActionType, encryptedMobile) + // 将验证码保存到 Redis，设置过期时间 + err = l.svcCtx.Redis.Setex(codeKey, code, l.svcCtx.Config.VerifyCode.ValidTime) // 验证码有效期5分钟 + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 验证码设置过期时间失败: %v", err) + } + // 在 Redis 中设置 1 分钟的标记，限制重复请求 + err = l.svcCtx.Redis.Setex(limitCodeKey, code, 60) // 标记 1 分钟内不能重复请求 + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "短信发送, 验证码设置限制重复请求失败: %v", err) + } + return nil +} + +// CreateClient 创建阿里云短信客户端 +func (l *SendSmsLogic) CreateClient() (*dysmsapi.Client, error) { + config := &openapi.Config{ + AccessKeyId: &l.svcCtx.Config.VerifyCode.AccessKeyID, + AccessKeySecret: &l.svcCtx.Config.VerifyCode.AccessKeySecret, + } + config.Endpoint = tea.String(l.svcCtx.Config.VerifyCode.EndpointURL) + return dysmsapi.NewClient(config) +} + +// sendSmsRequest 发送短信请求 +func (l *SendSmsLogic) sendSmsRequest(mobile, code string) (*dysmsapi.SendSmsResponse, error) { + // 初始化阿里云短信客户端 + cli, err := l.CreateClient() + if err != nil { + return nil, err + } + + request := &dysmsapi.SendSmsRequest{ + SignName: tea.String(l.svcCtx.Config.VerifyCode.SignName), + TemplateCode: tea.String(l.svcCtx.Config.VerifyCode.TemplateCode), + PhoneNumbers: tea.String(mobile), + TemplateParam: tea.String(fmt.Sprintf("{\"code\":\"%s\"}", code)), + } + runtime := &service.RuntimeOptions{} + return cli.SendSmsWithOptions(request, runtime) +} diff --git a/app/main/api/internal/logic/authorization/downloadauthorizationdocumentlogic.go b/app/main/api/internal/logic/authorization/downloadauthorizationdocumentlogic.go new file mode 100644 index 0000000..fcfd25d --- /dev/null +++ b/app/main/api/internal/logic/authorization/downloadauthorizationdocumentlogic.go @@ -0,0 +1,51 @@ +package authorization + +import ( + "context" + "errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type DownloadAuthorizationDocumentLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewDownloadAuthorizationDocumentLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DownloadAuthorizationDocumentLogic { + return &DownloadAuthorizationDocumentLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *DownloadAuthorizationDocumentLogic) DownloadAuthorizationDocument(req *types.DownloadAuthorizationDocumentReq) (resp *types.DownloadAuthorizationDocumentResp, err error) { + // 1. 从数据库获取授权书信息 + authDoc, err := l.svcCtx.AuthorizationDocumentModel.FindOne(l.ctx, req.DocumentId) + if err != nil { + logx.Errorf("获取授权书失败: documentId=%s, error=%v", req.DocumentId, err) + return nil, err + } + + // 2. 检查授权书状态 + if authDoc.Status != "active" { + logx.Errorf("授权书状态异常: documentId=%s, status=%s", req.DocumentId, authDoc.Status) + return nil, errors.New("授权书不可用") + } + + // 3. 构建完整文件URL + fullFileURL := l.svcCtx.AuthorizationService.GetFullFileURL(authDoc.FileUrl) + + // 4. 构建响应 + resp = &types.DownloadAuthorizationDocumentResp{ + FileName: authDoc.FileName, + FileUrl: fullFileURL, + } + + return resp, nil +} diff --git a/app/main/api/internal/logic/authorization/getauthorizationdocumentbyorderlogic.go b/app/main/api/internal/logic/authorization/getauthorizationdocumentbyorderlogic.go new file mode 100644 index 0000000..e9cfcdb --- /dev/null +++ b/app/main/api/internal/logic/authorization/getauthorizationdocumentbyorderlogic.go @@ -0,0 +1,62 @@ +package authorization + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetAuthorizationDocumentByOrderLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetAuthorizationDocumentByOrderLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetAuthorizationDocumentByOrderLogic { + return &GetAuthorizationDocumentByOrderLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetAuthorizationDocumentByOrderLogic) GetAuthorizationDocumentByOrder(req *types.GetAuthorizationDocumentByOrderReq) (resp *types.GetAuthorizationDocumentByOrderResp, err error) { + // 1. 根据订单ID查询授权书列表 + authDocs, err := l.svcCtx.AuthorizationDocumentModel.FindByOrderId(l.ctx, req.OrderId) + if err != nil { + logx.Errorf("根据订单ID获取授权书失败: orderId=%d, error=%v", req.OrderId, err) + return nil, err + } + + // 2. 构建响应列表 + var documents []types.AuthorizationDocumentInfo + for _, authDoc := range authDocs { + // 只返回状态为active的授权书 + if authDoc.Status == "active" { + fullFileURL := l.svcCtx.AuthorizationService.GetFullFileURL(authDoc.FileUrl) + + documents = append(documents, types.AuthorizationDocumentInfo{ + DocumentId: authDoc.Id, + UserId: authDoc.UserId, + OrderId: authDoc.OrderId, + QueryId: authDoc.QueryId, + FileName: authDoc.FileName, + FileUrl: fullFileURL, + FileSize: authDoc.FileSize, + FileType: authDoc.FileType, + Status: authDoc.Status, + CreateTime: authDoc.CreateTime.Format("2006-01-02 15:04:05"), + }) + } + } + + // 3. 构建响应 + resp = &types.GetAuthorizationDocumentByOrderResp{ + Documents: documents, + } + + return resp, nil +} diff --git a/app/main/api/internal/logic/authorization/getauthorizationdocumentlogic.go b/app/main/api/internal/logic/authorization/getauthorizationdocumentlogic.go new file mode 100644 index 0000000..1b5e48f --- /dev/null +++ b/app/main/api/internal/logic/authorization/getauthorizationdocumentlogic.go @@ -0,0 +1,59 @@ +package authorization + +import ( + "context" + "errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetAuthorizationDocumentLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetAuthorizationDocumentLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetAuthorizationDocumentLogic { + return &GetAuthorizationDocumentLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetAuthorizationDocumentLogic) GetAuthorizationDocument(req *types.GetAuthorizationDocumentReq) (resp *types.GetAuthorizationDocumentResp, err error) { + // 1. 从数据库获取授权书信息 + authDoc, err := l.svcCtx.AuthorizationDocumentModel.FindOne(l.ctx, req.DocumentId) + if err != nil { + logx.Errorf("获取授权书失败: documentId=%d, error=%v", req.DocumentId, err) + return nil, err + } + + // 2. 检查授权书状态 + if authDoc.Status != "active" { + logx.Errorf("授权书状态异常: documentId=%d, status=%s", req.DocumentId, authDoc.Status) + return nil, errors.New("授权书不可用") + } + + // 3. 构建完整文件URL + fullFileURL := l.svcCtx.AuthorizationService.GetFullFileURL(authDoc.FileUrl) + + // 4. 构建响应 + resp = &types.GetAuthorizationDocumentResp{ + DocumentId: authDoc.Id, + UserId: authDoc.UserId, + OrderId: authDoc.OrderId, + QueryId: authDoc.QueryId, + FileName: authDoc.FileName, + FileUrl: fullFileURL, + FileSize: authDoc.FileSize, + FileType: authDoc.FileType, + Status: authDoc.Status, + CreateTime: authDoc.CreateTime.Format("2006-01-02 15:04:05"), + } + + return resp, nil +} diff --git a/app/main/api/internal/logic/notification/getnotificationslogic.go b/app/main/api/internal/logic/notification/getnotificationslogic.go new file mode 100644 index 0000000..8e91c81 --- /dev/null +++ b/app/main/api/internal/logic/notification/getnotificationslogic.go @@ -0,0 +1,57 @@ +package notification + +import ( + "context" + "qnc-server/common/xerr" + "time" + + "github.com/jinzhu/copier" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetNotificationsLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetNotificationsLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetNotificationsLogic { + return &GetNotificationsLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetNotificationsLogic) GetNotifications() (resp *types.GetNotificationsResp, err error) { + // 获取今天的日期 + now := time.Now() + + // 获取开始和结束日期的时间戳 + todayStart := now.Format("2006-01-02") + " 00:00:00" + todayEnd := now.Format("2006-01-02") + " 23:59:59" + + // 构建查询条件 + builder := l.svcCtx.GlobalNotificationsModel.SelectBuilder(). + Where("status = ?", "active"). + Where("(start_date IS NULL OR start_date <= ?)", todayEnd). // start_date 是 NULL 或者小于等于今天结束时间 + Where("(end_date IS NULL OR end_date >= ?)", todayStart) // end_date 是 NULL 或者大于等于今天开始时间 + + notificationsModelList, findErr := l.svcCtx.GlobalNotificationsModel.FindAll(l.ctx, builder, "") + if findErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "全局通知, 查找通知失败, err:%+v", findErr) + } + + var notifications []types.Notification + copyErr := copier.Copy(¬ifications, ¬ificationsModelList) + if copyErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "全局通知, 复制结构体失败, err:%+v", copyErr) + } + + return &types.GetNotificationsResp{Notifications: notifications}, nil +} diff --git a/app/main/api/internal/logic/pay/alipaycallbacklogic.go b/app/main/api/internal/logic/pay/alipaycallbacklogic.go new file mode 100644 index 0000000..5e5a9f0 --- /dev/null +++ b/app/main/api/internal/logic/pay/alipaycallbacklogic.go @@ -0,0 +1,338 @@ +package pay + +import ( + "context" + "net/http" + "strings" + "time" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/pkg/errors" + "github.com/smartwalle/alipay/v3" + + "qnc-server/app/main/api/internal/svc" + + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type AlipayCallbackLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAlipayCallbackLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AlipayCallbackLogic { + return &AlipayCallbackLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *AlipayCallbackLogic) AlipayCallback(w http.ResponseWriter, r *http.Request) error { + notification, err := l.svcCtx.AlipayService.HandleAliPaymentNotification(r) + if err != nil { + logx.Errorf("支付宝支付回调，%v", err) + return nil + } + + // 根据订单号前缀判断订单类型 + orderNo := notification.OutTradeNo + if strings.HasPrefix(orderNo, "Q_") { + // 查询订单处理 + return l.handleQueryOrderPayment(w, notification) + } else if strings.HasPrefix(orderNo, "U_") { + // 代理升级订单处理 + return l.handleAgentUpgradeOrderPayment(w, notification) + } else if strings.HasPrefix(orderNo, "A_") { + // 旧系统会员充值订单（已废弃，新系统使用升级功能） + // return l.handleAgentVipOrderPayment(w, notification) + // 直接返回成功，避免旧订单影响 + alipay.ACKNotification(w) + return nil + } else { + // 兼容旧订单，假设没有前缀的是查询订单 + return l.handleQueryOrderPayment(w, notification) + } +} + +// 处理查询订单支付 +func (l *AlipayCallbackLogic) handleQueryOrderPayment(w http.ResponseWriter, notification *alipay.Notification) error { + order, findOrderErr := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, notification.OutTradeNo) + if findOrderErr != nil { + logx.Errorf("支付宝支付回调，查找订单失败: %+v", findOrderErr) + return nil + } + + if order.Status != "pending" { + alipay.ACKNotification(w) + return nil + } + + user, err := l.svcCtx.UserModel.FindOne(l.ctx, order.UserId) + if err != nil { + logx.Errorf("支付宝支付回调，查找用户失败: %+v", err) + return nil + } + + amount := lzUtils.ToAlipayAmount(order.Amount) + if user.Inside != 1 { + // 确保订单金额和状态正确，防止重复更新 + if amount != notification.TotalAmount { + logx.Errorf("支付宝支付回调，金额不一致") + return nil + } + } + + switch notification.TradeStatus { + case alipay.TradeStatusSuccess: + order.Status = "paid" + order.PayTime = lzUtils.TimeToNullTime(time.Now()) + default: + return nil + } + + order.PlatformOrderId = lzUtils.StringToNullString(notification.TradeNo) + if updateErr := l.svcCtx.OrderModel.UpdateWithVersion(l.ctx, nil, order); updateErr != nil { + logx.Errorf("支付宝支付回调，修改订单信息失败: %+v", updateErr) + return nil + } + + if order.Status == "paid" { + if asyncErr := l.svcCtx.AsynqService.SendQueryTask(order.Id); asyncErr != nil { + logx.Errorf("异步任务调度失败: %v", asyncErr) + return asyncErr + } + } + + alipay.ACKNotification(w) + return nil +} + +// 处理代理升级订单支付 +func (l *AlipayCallbackLogic) handleAgentUpgradeOrderPayment(w http.ResponseWriter, notification *alipay.Notification) error { + orderNo := notification.OutTradeNo + + // 1. 查找订单 + order, findOrderErr := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, orderNo) + if findOrderErr != nil { + logx.Errorf("支付宝支付回调，查找升级订单失败: %+v", findOrderErr) + alipay.ACKNotification(w) + return nil + } + + // 2. 验证金额 + amount := lzUtils.ToAlipayAmount(order.Amount) + user, err := l.svcCtx.UserModel.FindOne(l.ctx, order.UserId) + if err == nil && user.Inside != 1 { + if amount != notification.TotalAmount { + logx.Errorf("支付宝支付回调，升级订单金额不一致，订单号: %s", orderNo) + alipay.ACKNotification(w) + return nil + } + } + + // 3. 检查订单状态 + if order.Status != "pending" { + alipay.ACKNotification(w) + return nil + } + + // 4. 查找升级记录 + upgradeRecords, findUpgradeErr := l.svcCtx.AgentUpgradeModel.FindAll(l.ctx, l.svcCtx.AgentUpgradeModel.SelectBuilder(). + Where("order_no = ?", orderNo). + Limit(1), "") + if findUpgradeErr != nil || len(upgradeRecords) == 0 { + logx.Errorf("支付宝支付回调，查找升级记录失败，订单号: %s, 错误: %+v", orderNo, findUpgradeErr) + alipay.ACKNotification(w) + return nil + } + upgradeRecord := upgradeRecords[0] + + // 5. 检查升级记录状态 + if upgradeRecord.Status != 1 { + // 升级记录状态不是待支付，直接返回成功 + alipay.ACKNotification(w) + return nil + } + + // 6. 处理支付状态 + switch notification.TradeStatus { + case alipay.TradeStatusSuccess: + order.Status = "paid" + order.PayTime = lzUtils.TimeToNullTime(time.Now()) + order.PlatformOrderId = lzUtils.StringToNullString(notification.TradeNo) + case alipay.TradeStatusClosed: + order.Status = "closed" + order.CloseTime = lzUtils.TimeToNullTime(time.Now()) + default: + alipay.ACKNotification(w) + return nil + } + + // 7. 更新订单状态 + if updateErr := l.svcCtx.OrderModel.UpdateWithVersion(l.ctx, nil, order); updateErr != nil { + logx.Errorf("支付宝支付回调，更新升级订单状态失败: %+v", updateErr) + alipay.ACKNotification(w) + return nil + } + + // 8. 如果支付成功，执行升级操作 + if order.Status == "paid" { + err := l.svcCtx.AgentWalletModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + // 8.1 执行升级操作 + if err := l.svcCtx.AgentService.ProcessUpgrade(transCtx, upgradeRecord.AgentId, upgradeRecord.ToLevel, upgradeRecord.UpgradeType, upgradeRecord.UpgradeFee, upgradeRecord.RebateAmount, orderNo, ""); err != nil { + return errors.Wrapf(err, "执行升级操作失败") + } + + // 8.2 更新升级记录状态为已完成 + upgradeRecord.Status = 2 // 已完成（status: 1=待处理，2=已完成，3=已失败） + upgradeRecord.Remark = lzUtils.StringToNullString("支付成功，升级完成") + if updateErr := l.svcCtx.AgentUpgradeModel.UpdateWithVersion(transCtx, session, upgradeRecord); updateErr != nil { + return errors.Wrapf(updateErr, "更新升级记录状态失败") + } + + return nil + }) + + if err != nil { + logx.Errorf("支付宝支付回调，处理升级订单失败，订单号: %s, 错误: %+v", orderNo, err) + // 即使升级失败，也返回成功给支付宝，避免重复回调 + } else { + logx.Infof("支付宝支付回调，代理升级成功，订单号: %s, 代理ID: %d, 从等级 %d 升级到等级 %d", orderNo, upgradeRecord.AgentId, upgradeRecord.FromLevel, upgradeRecord.ToLevel) + } + } + + alipay.ACKNotification(w) + return nil +} + +// 处理代理会员订单支付（已废弃，新系统使用升级功能） +/* +func (l *AlipayCallbackLogic) handleAgentVipOrderPayment(w http.ResponseWriter, notification *alipay.Notification) error { + agentOrder, findAgentOrderErr := l.svcCtx.AgentMembershipRechargeOrderModel.FindOneByOrderNo(l.ctx, notification.OutTradeNo) + if findAgentOrderErr != nil { + logx.Errorf("支付宝支付回调，查找代理会员订单失败: %+v", findAgentOrderErr) + return nil + } + + if agentOrder.Status != "pending" { + alipay.ACKNotification(w) + return nil + } + + user, err := l.svcCtx.UserModel.FindOne(l.ctx, agentOrder.UserId) + if err != nil { + logx.Errorf("支付宝支付回调，查找用户失败: %+v", err) + return nil + } + + amount := lzUtils.ToAlipayAmount(agentOrder.Amount) + if user.Inside != 1 { + // 确保订单金额和状态正确，防止重复更新 + if amount != notification.TotalAmount { + logx.Errorf("支付宝支付回调，金额不一致") + return nil + } + } + + switch notification.TradeStatus { + case alipay.TradeStatusSuccess: + agentOrder.Status = "paid" + default: + return nil + } + + if agentOrder.Status == "paid" { + err = l.svcCtx.AgentModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + agentModel, err := l.svcCtx.AgentModel.FindOne(transCtx, agentOrder.AgentId) + if err != nil { + return fmt.Errorf("查找代理信息失败: %+v", err) + } + agentOrder.PlatformOrderId = lzUtils.StringToNullString(notification.TradeNo) + if updateErr := l.svcCtx.AgentMembershipRechargeOrderModel.UpdateWithVersion(l.ctx, nil, agentOrder); updateErr != nil { + return fmt.Errorf("修改代理会员订单信息失败: %+v", updateErr) + } + + // 记录旧等级，用于判断是否为升级 + oldLevel := agentModel.LevelName + + // 设置会员等级 + agentModel.LevelName = agentOrder.LevelName + + // 延长会员时间 + // 检查是否是有效期内续费（不发放奖励）还是重新激活（发放奖励） + isValidRenewal := oldLevel == agentOrder.LevelName && agentModel.MembershipExpiryTime.Valid && agentModel.MembershipExpiryTime.Time.After(time.Now()) + if isValidRenewal { + logx.Infof("代理会员有效期内续费成功，会员ID：%d，等级：%s", agentModel.Id, agentModel.LevelName) + } else { + logx.Infof("代理会员新购、升级或重新激活成功，会员ID：%d，等级：%s", agentModel.Id, agentModel.LevelName) + } + agentModel.MembershipExpiryTime = lzUtils.RenewMembership(agentModel.MembershipExpiryTime) + + if updateErr := l.svcCtx.AgentModel.UpdateWithVersion(l.ctx, nil, agentModel); updateErr != nil { + return fmt.Errorf("修改代理信息失败: %+v", updateErr) + } + + // 如果不是有效期内续费，给上级代理发放升级奖励 + if !isValidRenewal && (agentOrder.LevelName == model.AgentLeveNameVIP || agentOrder.LevelName == model.AgentLeveNameSVIP) { + // 验证升级路径的有效性 + if oldLevel != agentOrder.LevelName { + upgradeRewardErr := l.svcCtx.AgentService.GiveUpgradeReward(transCtx, agentModel.Id, oldLevel, agentOrder.LevelName, session) + if upgradeRewardErr != nil { + logx.Errorf("发放升级奖励失败，代理ID：%d，旧等级：%s，新等级：%s，错误：%+v", agentModel.Id, oldLevel, agentOrder.LevelName, upgradeRewardErr) + // 升级奖励失败不影响主流程，只记录日志 + } else { + logx.Infof("发放升级奖励成功，代理ID：%d，旧等级：%s，新等级：%s", agentModel.Id, oldLevel, agentOrder.LevelName) + } + } + } + + return nil + }) + if err != nil { + logx.Errorf("支付宝支付回调，处理代理会员订单失败: %+v", err) + refundErr := l.handleRefund(agentOrder) + if refundErr != nil { + logx.Errorf("支付宝支付回调，退款失败: %+v", refundErr) + } + return nil + } + } + + alipay.ACKNotification(w) + return nil +} */ + +/* func (l *AlipayCallbackLogic) handleRefund(order *model.AgentMembershipRechargeOrder) error { + ctx := context.Background() + // 退款 + if order.PaymentMethod == "wechat" { + refundErr := l.svcCtx.WechatPayService.WeChatRefund(ctx, order.OrderNo, order.Amount, order.Amount) + if refundErr != nil { + return refundErr + } + } else { + refund, refundErr := l.svcCtx.AlipayService.AliRefund(ctx, order.OrderNo, order.Amount) + if refundErr != nil { + return refundErr + } + if refund.IsSuccess() { + logx.Errorf("支付宝退款成功, orderID: %d", order.Id) + // 更新订单状态为退款 + order.Status = "refunded" + updateOrderErr := l.svcCtx.AgentMembershipRechargeOrderModel.UpdateWithVersion(ctx, nil, order) + if updateOrderErr != nil { + logx.Errorf("更新订单状态失败，订单ID: %d, 错误: %v", order.Id, updateOrderErr) + return fmt.Errorf("更新订单状态失败: %v", updateOrderErr) + } + return nil + } else { + logx.Errorf("支付宝退款失败:%v", refundErr) + return refundErr + } + // 直接成功 + } + return nil +} */ diff --git a/app/main/api/internal/logic/pay/iapcallbacklogic.go b/app/main/api/internal/logic/pay/iapcallbacklogic.go new file mode 100644 index 0000000..78fd6ea --- /dev/null +++ b/app/main/api/internal/logic/pay/iapcallbacklogic.go @@ -0,0 +1,83 @@ +package pay + +import ( + "context" + "fmt" + "time" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/pkg/errors" + + "github.com/zeromicro/go-zero/core/logx" +) + +type IapCallbackLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewIapCallbackLogic(ctx context.Context, svcCtx *svc.ServiceContext) *IapCallbackLogic { + return &IapCallbackLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *IapCallbackLogic) IapCallback(req *types.IapCallbackReq) error { + // Step 1: 查找订单 + order, findOrderErr := l.svcCtx.OrderModel.FindOne(l.ctx, fmt.Sprintf("%d", req.OrderID)) + if findOrderErr != nil { + logx.Errorf("苹果内购支付回调，查找订单失败: %+v", findOrderErr) + return nil + } + + // Step 2: 验证订单状态 + if order.Status != "pending" { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "苹果内购支付回调, 订单状态异常: %+v", order) + } + + // Step 3: 调用 VerifyReceipt 验证苹果支付凭证 + //receipt := req.TransactionReceipt // 从请求中获取支付凭证 + //verifyResponse, verifyErr := l.svcCtx.ApplePayService.VerifyReceipt(l.ctx, receipt) + //if verifyErr != nil { + // return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "苹果内购支付回调, 验证订单异常: %+v", verifyErr) + //} + + // Step 4: 验证订单 + //product, findProductErr := l.svcCtx.ProductModel.FindOne(l.ctx, order.Id) + //if findProductErr != nil { + // return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "苹果内购支付回调, 获取订单相关商品失败: %+v", findProductErr) + //} + //isProductMatched := false + //appleProductID := l.svcCtx.ApplePayService.GetIappayAppID(product.ProductEn) + //for _, item := range verifyResponse.Receipt.InApp { + // if item.ProductID == appleProductID { + // isProductMatched = true + // order.PlatformOrderId = lzUtils.StringToNullString(item.TransactionID) // 记录交易 ID + // break + // } + //} + //if !isProductMatched { + // return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "苹果内购支付回调, 商品 ID 不匹配，订单 ID: %d, 回调苹果商品 ID: %s", order.Id, verifyResponse.Receipt.InApp[0].ProductID) + //} + + // Step 5: 更新订单状态 mm + order.Status = "paid" + order.PayTime = lzUtils.TimeToNullTime(time.Now()) + + // 更新订单到数据库 + if updateErr := l.svcCtx.OrderModel.UpdateWithVersion(l.ctx, nil, order); updateErr != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "苹果内购支付回调, 修改订单信息失败: %+v", updateErr) + } + + // Step 6: 处理订单完成后的逻辑 + if asyncErr := l.svcCtx.AsynqService.SendQueryTask(order.Id); asyncErr != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "苹果内购支付回调,异步任务调度失败: %v", asyncErr) + } + return nil +} diff --git a/app/main/api/internal/logic/pay/paymentchecklogic.go b/app/main/api/internal/logic/pay/paymentchecklogic.go new file mode 100644 index 0000000..2d813ac --- /dev/null +++ b/app/main/api/internal/logic/pay/paymentchecklogic.go @@ -0,0 +1,51 @@ +package pay + +import ( + "context" + "strings" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type PaymentCheckLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewPaymentCheckLogic(ctx context.Context, svcCtx *svc.ServiceContext) *PaymentCheckLogic { + return &PaymentCheckLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *PaymentCheckLogic) PaymentCheck(req *types.PaymentCheckReq) (resp *types.PaymentCheckResp, err error) { + // 根据订单号前缀判断订单类型 + if strings.HasPrefix(req.OrderNo, "U_") { + // 升级订单 + order, err := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, req.OrderNo) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询升级订单失败: %v", err) + } + return &types.PaymentCheckResp{ + Type: "agent_upgrade", + Status: order.Status, + }, nil + } + + // 查询订单（包括代理订单） + order, err := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, req.OrderNo) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询订单失败: %v", err) + } + return &types.PaymentCheckResp{ + Type: "query", + Status: order.Status, + }, nil +} diff --git a/app/main/api/internal/logic/pay/paymentlogic.go b/app/main/api/internal/logic/pay/paymentlogic.go new file mode 100644 index 0000000..070d15a --- /dev/null +++ b/app/main/api/internal/logic/pay/paymentlogic.go @@ -0,0 +1,475 @@ +package pay + +import ( + "context" + "database/sql" + "encoding/json" + "fmt" + "os" + "strconv" + "strings" + "time" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/redis/go-redis/v9" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type PaymentLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} +type PaymentTypeResp struct { + amount float64 + outTradeNo string + description string + orderID string // 订单ID，用于开发环境测试支付模式 +} + +func NewPaymentLogic(ctx context.Context, svcCtx *svc.ServiceContext) *PaymentLogic { + return &PaymentLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *PaymentLogic) Payment(req *types.PaymentReq) (resp *types.PaymentResp, err error) { + var paymentTypeResp *PaymentTypeResp + var prepayData interface{} + var orderID string + + // 检查是否为开发环境的测试支付模式 + env := os.Getenv("ENV") + isDevTestPayment := env == "development" && (req.PayMethod == "test" || req.PayMethod == "test_empty") + isEmptyReportMode := env == "development" && req.PayMethod == "test_empty" + + l.svcCtx.OrderModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + switch req.PayType { + case "agent_vip": + paymentTypeResp, err = l.AgentVipOrderPayment(req, session) + if err != nil { + return err + } + + case "query": + paymentTypeResp, err = l.QueryOrderPayment(req, session) + if err != nil { + return err + } + + case "agent_upgrade": + paymentTypeResp, err = l.AgentUpgradeOrderPayment(req, session) + if err != nil { + return err + } + } + + // 开发环境测试支付模式：跳过实际支付流程 + // 注意：订单状态更新在事务外进行，避免在事务中查询不到订单的问题 + if isDevTestPayment { + // 获取订单ID（从 QueryOrderPayment 返回的 orderID） + if paymentTypeResp.orderID == "" { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "开发测试模式，订单ID无效") + } + orderID = paymentTypeResp.orderID + + // 在事务中只记录订单ID，不更新订单状态 + // 订单状态的更新和后续流程在事务提交后处理 + logx.Infof("开发环境测试支付模式：订单 %s (ID: %s) 将在事务提交后更新状态", paymentTypeResp.outTradeNo, orderID) + + // 返回测试支付标识 + prepayData = "test_payment_success" + return nil + } + + // 正常支付流程 + var createOrderErr error + if req.PayMethod == "wechat" { + prepayData, createOrderErr = l.svcCtx.WechatPayService.CreateWechatOrder(l.ctx, paymentTypeResp.amount, paymentTypeResp.description, paymentTypeResp.outTradeNo) + } else if req.PayMethod == "alipay" { + prepayData, createOrderErr = l.svcCtx.AlipayService.CreateAlipayOrder(l.ctx, paymentTypeResp.amount, paymentTypeResp.description, paymentTypeResp.outTradeNo) + } else if req.PayMethod == "appleiap" { + prepayData = l.svcCtx.ApplePayService.GetIappayAppID(paymentTypeResp.outTradeNo) + } + if createOrderErr != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 创建支付订单失败: %+v", createOrderErr) + } + return nil + }) + if err != nil { + return nil, err + } + + // 开发环境测试支付模式：事务提交后处理订单状态更新和后续流程 + if isDevTestPayment && paymentTypeResp != nil && paymentTypeResp.orderID != "" { + // 使用 goroutine 异步处理，确保事务已完全提交 + go func() { + // 短暂延迟，确保事务已完全提交到数据库 + time.Sleep(200 * time.Millisecond) + + finalOrderID := paymentTypeResp.orderID + + // 查找订单并更新状态为已支付 + order, findOrderErr := l.svcCtx.OrderModel.FindOne(context.Background(), finalOrderID) + if findOrderErr != nil { + logx.Errorf("开发测试模式，查找订单失败，订单ID: %s, 错误: %v", finalOrderID, findOrderErr) + return + } + + // 更新订单状态为已支付 + order.Status = "paid" + now := time.Now() + order.PayTime = sql.NullTime{Time: now, Valid: true} + + // 空报告模式：在 PaymentPlatform 字段中标记，用于后续生成空报告 + if isEmptyReportMode { + order.PaymentPlatform = "test_empty" + logx.Infof("开发环境空报告模式：订单 %s (ID: %s) 已标记为空报告模式", paymentTypeResp.outTradeNo, finalOrderID) + } + + // 更新订单状态（在事务外执行） + updateErr := l.svcCtx.OrderModel.UpdateWithVersion(context.Background(), nil, order) + if updateErr != nil { + logx.Errorf("开发测试模式，更新订单状态失败，订单ID: %s, 错误: %+v", finalOrderID, updateErr) + return + } + + logx.Infof("开发环境测试支付模式：订单 %s (ID: %s) 已自动标记为已支付", paymentTypeResp.outTradeNo, finalOrderID) + + // 再次短暂延迟，确保订单状态更新已提交 + time.Sleep(100 * time.Millisecond) + + // 根据订单类型处理后续流程 + if strings.HasPrefix(paymentTypeResp.outTradeNo, "U_") { + // 升级订单：直接执行升级操作 + upgradeRecords, findUpgradeErr := l.svcCtx.AgentUpgradeModel.FindAll(context.Background(), l.svcCtx.AgentUpgradeModel.SelectBuilder(). + Where("order_no = ?", paymentTypeResp.outTradeNo). + Limit(1), "") + if findUpgradeErr != nil || len(upgradeRecords) == 0 { + logx.Errorf("开发测试模式，查找升级记录失败，订单号: %s, 错误: %+v", paymentTypeResp.outTradeNo, findUpgradeErr) + return + } + upgradeRecord := upgradeRecords[0] + + // 执行升级操作 + err := l.svcCtx.AgentWalletModel.Trans(context.Background(), func(transCtx context.Context, session sqlx.Session) error { + if err := l.svcCtx.AgentService.ProcessUpgrade(transCtx, upgradeRecord.AgentId, upgradeRecord.ToLevel, upgradeRecord.UpgradeType, upgradeRecord.UpgradeFee, upgradeRecord.RebateAmount, paymentTypeResp.outTradeNo, ""); err != nil { + return errors.Wrapf(err, "执行升级操作失败") + } + + // 更新升级记录状态为已完成 + upgradeRecord.Status = 2 // 已完成（status: 1=待处理，2=已完成，3=已失败） + upgradeRecord.Remark = lzUtils.StringToNullString("测试支付成功，升级完成") + if updateErr := l.svcCtx.AgentUpgradeModel.UpdateWithVersion(transCtx, session, upgradeRecord); updateErr != nil { + return errors.Wrapf(updateErr, "更新升级记录状态失败") + } + + return nil + }) + + if err != nil { + logx.Errorf("开发测试模式，处理升级订单失败，订单号: %s, 错误: %+v", paymentTypeResp.outTradeNo, err) + } else { + logx.Infof("开发测试模式，代理升级成功，订单号: %s, 代理ID: %s", paymentTypeResp.outTradeNo, upgradeRecord.AgentId) + } + } else { + // 查询订单：发送支付成功通知任务，触发后续流程（生成报告和代理处理） + if sendErr := l.svcCtx.AsynqService.SendQueryTask(finalOrderID); sendErr != nil { + logx.Errorf("开发测试模式，发送支付成功通知任务失败，订单ID: %s, 错误: %+v", finalOrderID, sendErr) + } else { + logx.Infof("开发测试模式，已发送支付成功通知任务，订单ID: %s", finalOrderID) + } + } + }() + } + + switch v := prepayData.(type) { + case string: + // 如果 prepayData 是字符串类型，直接返回 + return &types.PaymentResp{PrepayId: v, OrderNo: paymentTypeResp.outTradeNo}, nil + default: + return &types.PaymentResp{PrepayData: prepayData, OrderNo: paymentTypeResp.outTradeNo}, nil + } +} + +func (l *PaymentLogic) QueryOrderPayment(req *types.PaymentReq, session sqlx.Session) (resp *PaymentTypeResp, err error) { + userID, getUidErr := ctxdata.GetUidFromCtx(l.ctx) + if getUidErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 获取用户信息失败, %+v", getUidErr) + } + outTradeNo := req.Id + redisKey := fmt.Sprintf(types.QueryCacheKey, userID, outTradeNo) + cache, cacheErr := l.svcCtx.Redis.GetCtx(l.ctx, redisKey) + if cacheErr != nil { + if cacheErr == redis.Nil { + return nil, errors.Wrapf(xerr.NewErrMsg("订单已过期"), "生成订单, 缓存不存在, %+v", cacheErr) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 获取缓存失败, %+v", cacheErr) + } + var data types.QueryCacheLoad + err = json.Unmarshal([]byte(cache), &data) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 解析缓存内容失败, %v", err) + } + + product, err := l.svcCtx.ProductModel.FindOneByProductEn(l.ctx, data.Product) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成订单, 查找产品错误: %v", err) + } + + var amount float64 + user, err := l.svcCtx.UserModel.FindOne(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成订单, 获取用户信息失败: %v", err) + } + + var agentLinkModel *model.AgentLink + if data.AgentIdentifier != "" { + var findAgentLinkErr error + agentLinkModel, findAgentLinkErr = l.svcCtx.AgentLinkModel.FindOneByLinkIdentifier(l.ctx, data.AgentIdentifier) + if findAgentLinkErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成订单, 获取代理链接失败: %+v", findAgentLinkErr) + } + amount = agentLinkModel.SetPrice + } else { + amount = product.SellPrice + } + + if user.Inside == 1 { + amount = 0.01 + } + order := model.Order{ + Id: uuid.NewString(), + OrderNo: outTradeNo, + UserId: userID, + ProductId: product.Id, + PaymentPlatform: req.PayMethod, + PaymentScene: "app", + Amount: amount, + Status: "pending", + } + _, insertOrderErr := l.svcCtx.OrderModel.Insert(l.ctx, session, &order) + if insertOrderErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成订单, 保存订单失败: %+v", insertOrderErr) + } + orderID := order.Id + + // 如果是代理推广订单，创建完整的代理订单记录 + if data.AgentIdentifier != "" && agentLinkModel != nil { + // 获取代理信息 + agent, err := l.svcCtx.AgentModel.FindOne(l.ctx, agentLinkModel.AgentId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成订单, 查询代理信息失败: %+v", err) + } + + // 获取产品配置（必须存在） + productConfig, err := l.svcCtx.AgentProductConfigModel.FindOneByProductId(l.ctx, product.Id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成订单失败，产品配置不存在, productId: %s，请先在后台配置产品价格参数", product.Id) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成订单, 查询产品配置失败: %+v", err) + } + + // 获取等级加成（需要从系统配置读取） + levelBonus, err := l.getLevelBonus(agent.Level) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成订单, 获取等级加成配置失败: %+v", err) + } + + // 使用产品配置的底价计算实际底价 + basePrice := productConfig.BasePrice + actualBasePrice := basePrice + float64(levelBonus) + + // 计算提价成本（使用产品配置） + priceThreshold := 0.0 + priceFeeRate := 0.0 + if productConfig.PriceThreshold.Valid { + priceThreshold = productConfig.PriceThreshold.Float64 + } + if productConfig.PriceFeeRate.Valid { + priceFeeRate = productConfig.PriceFeeRate.Float64 + } + + priceCost := 0.0 + if agentLinkModel.SetPrice > priceThreshold { + priceCost = (agentLinkModel.SetPrice - priceThreshold) * priceFeeRate + } + + // 计算代理收益 + agentProfit := agentLinkModel.SetPrice - actualBasePrice - priceCost + + // 创建代理订单记录 + agentOrder := model.AgentOrder{ + Id: uuid.NewString(), + AgentId: agentLinkModel.AgentId, + OrderId: orderID, + ProductId: product.Id, + OrderAmount: amount, + SetPrice: agentLinkModel.SetPrice, + ActualBasePrice: actualBasePrice, + PriceCost: priceCost, + AgentProfit: agentProfit, + ProcessStatus: 0, // 待处理 + } + _, agentOrderInsert := l.svcCtx.AgentOrderModel.Insert(l.ctx, session, &agentOrder) + if agentOrderInsert != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "生成订单, 保存代理订单失败: %+v", agentOrderInsert) + } + } + return &PaymentTypeResp{amount: amount, outTradeNo: outTradeNo, description: product.ProductName, orderID: orderID}, nil +} + +// AgentVipOrderPayment 代理会员充值订单（已废弃，新系统使用升级功能替代） +func (l *PaymentLogic) AgentVipOrderPayment(req *types.PaymentReq, session sqlx.Session) (resp *PaymentTypeResp, err error) { + // 新代理系统已废弃会员充值功能，请使用升级功能 + return nil, errors.Wrapf(xerr.NewErrMsg("该功能已废弃，请使用代理升级功能"), "") +} + +// AgentUpgradeOrderPayment 代理升级订单支付 +func (l *PaymentLogic) AgentUpgradeOrderPayment(req *types.PaymentReq, session sqlx.Session) (resp *PaymentTypeResp, err error) { + userID, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败, %v", err) + } + + // 1. 解析升级记录ID + upgradeId := req.Id + + // 2. 查找升级记录 + upgradeRecord, err := l.svcCtx.AgentUpgradeModel.FindOne(l.ctx, upgradeId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("升级记录不存在"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询升级记录失败, %v", err) + } + + // 3. 验证升级记录状态（必须是待支付状态） + if upgradeRecord.Status != 1 { + return nil, errors.Wrapf(xerr.NewErrMsg("升级记录状态不正确，无法支付"), "") + } + + // 4. 验证代理ID是否匹配 + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + if agent.Id != upgradeRecord.AgentId { + return nil, errors.Wrapf(xerr.NewErrMsg("无权支付此升级订单"), "") + } + + // 5. 生成订单号（升级订单前缀 U_，限制长度不超过32） + base := l.svcCtx.AlipayService.GenerateOutTradeNo() + outTradeNo := "U_" + base + if len(outTradeNo) > 32 { + outTradeNo = outTradeNo[:32] + } + + // 6. 获取用户信息（用于内部用户判断） + user, err := l.svcCtx.UserModel.FindOne(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取用户信息失败: %v", err) + } + + // 7. 计算支付金额 + amount := upgradeRecord.UpgradeFee + if user.Inside == 1 { + amount = 0.01 // 内部用户测试金额 + } + + // 8. 创建订单记录 + order := model.Order{ + Id: uuid.NewString(), + OrderNo: outTradeNo, + UserId: userID, + ProductId: "", // 升级订单没有产品ID + PaymentPlatform: req.PayMethod, + PaymentScene: "app", + Amount: amount, + Status: "pending", + } + orderInsertResult, insertOrderErr := l.svcCtx.OrderModel.Insert(l.ctx, session, &order) + if insertOrderErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建订单失败: %+v", insertOrderErr) + } + _ = orderInsertResult + orderID := order.Id + + // 9. 更新升级记录的订单号 + upgradeRecord.OrderNo = lzUtils.StringToNullString(outTradeNo) + if updateErr := l.svcCtx.AgentUpgradeModel.UpdateWithVersion(l.ctx, session, upgradeRecord); updateErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新升级记录订单号失败: %+v", updateErr) + } + + // 10. 生成描述信息 + levelNames := map[int64]string{ + 1: "普通代理", + 2: "黄金代理", + 3: "钻石代理", + } + fromLevelName := levelNames[upgradeRecord.FromLevel] + toLevelName := levelNames[upgradeRecord.ToLevel] + description := fmt.Sprintf("代理升级：%s → %s", fromLevelName, toLevelName) + + return &PaymentTypeResp{ + amount: amount, + outTradeNo: outTradeNo, + description: description, + orderID: orderID, + }, nil +} + +// getLevelBonus 获取等级加成（从配置表读取） +func (l *PaymentLogic) getLevelBonus(level int64) (int64, error) { + var configKey string + switch level { + case 1: // 普通 + configKey = "level_1_bonus" + case 2: // 黄金 + configKey = "level_2_bonus" + case 3: // 钻石 + configKey = "level_3_bonus" + default: + return 0, nil + } + + bonus, err := l.getConfigFloat(configKey) + if err != nil { + // 配置不存在时返回默认值 + l.Errorf("获取等级加成配置失败, level: %d, key: %s, err: %v，使用默认值", level, configKey, err) + switch level { + case 1: + return 6, nil + case 2: + return 3, nil + case 3: + return 0, nil + } + return 0, nil + } + return int64(bonus), nil +} + +// getConfigFloat 获取配置值（浮点数） +func (l *PaymentLogic) getConfigFloat(configKey string) (float64, error) { + config, err := l.svcCtx.AgentConfigModel.FindOneByConfigKey(l.ctx, configKey) + if err != nil { + return 0, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取配置失败, key: %s, %v", configKey, err) + } + value, err := strconv.ParseFloat(config.ConfigValue, 64) + if err != nil { + return 0, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "解析配置值失败, key: %s, value: %s, %v", configKey, config.ConfigValue, err) + } + return value, nil +} diff --git a/app/main/api/internal/logic/pay/wechatpaycallbacklogic.go b/app/main/api/internal/logic/pay/wechatpaycallbacklogic.go new file mode 100644 index 0000000..1b42e3c --- /dev/null +++ b/app/main/api/internal/logic/pay/wechatpaycallbacklogic.go @@ -0,0 +1,338 @@ +package pay + +import ( + "context" + "net/http" + "strings" + "time" + "qnc-server/app/main/api/internal/service" + "qnc-server/pkg/lzkit/lzUtils" + + "qnc-server/app/main/api/internal/svc" + + "github.com/pkg/errors" + "github.com/wechatpay-apiv3/wechatpay-go/services/payments" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type WechatPayCallbackLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewWechatPayCallbackLogic(ctx context.Context, svcCtx *svc.ServiceContext) *WechatPayCallbackLogic { + return &WechatPayCallbackLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *WechatPayCallbackLogic) WechatPayCallback(w http.ResponseWriter, r *http.Request) error { + notification, err := l.svcCtx.WechatPayService.HandleWechatPayNotification(l.ctx, r) + if err != nil { + logx.Errorf("微信支付回调，%v", err) + return nil + } + + // 根据订单号前缀判断订单类型 + orderNo := *notification.OutTradeNo + if strings.HasPrefix(orderNo, "Q_") { + // 查询订单处理 + return l.handleQueryOrderPayment(w, notification) + } else if strings.HasPrefix(orderNo, "U_") { + // 代理升级订单处理 + return l.handleAgentUpgradeOrderPayment(w, notification) + } else if strings.HasPrefix(orderNo, "A_") { + // 旧系统会员充值订单（已废弃，新系统使用升级功能） + // return l.handleAgentVipOrderPayment(w, notification) + // 直接返回成功，避免旧订单影响 + w.WriteHeader(http.StatusOK) + _, _ = w.Write([]byte("success")) + return nil + } else { + // 兼容旧订单，假设没有前缀的是查询订单 + return l.handleQueryOrderPayment(w, notification) + } +} + +// 处理查询订单支付 +func (l *WechatPayCallbackLogic) handleQueryOrderPayment(w http.ResponseWriter, notification *payments.Transaction) error { + order, findOrderErr := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, *notification.OutTradeNo) + if findOrderErr != nil { + logx.Errorf("微信支付回调，查找订单信息失败: %+v", findOrderErr) + return nil + } + + amount := lzUtils.ToWechatAmount(order.Amount) + if amount != *notification.Amount.Total { + logx.Errorf("微信支付回调，金额不一致") + return nil + } + + if order.Status != "pending" { + w.WriteHeader(http.StatusOK) + _, _ = w.Write([]byte("success")) + return nil + } + + switch *notification.TradeState { + case service.TradeStateSuccess: + order.Status = "paid" + order.PayTime = lzUtils.TimeToNullTime(time.Now()) + case service.TradeStateClosed: + order.Status = "closed" + order.CloseTime = lzUtils.TimeToNullTime(time.Now()) + case service.TradeStateRevoked: + order.Status = "failed" + default: + return nil + } + + order.PlatformOrderId = lzUtils.StringToNullString(*notification.TransactionId) + if updateErr := l.svcCtx.OrderModel.UpdateWithVersion(l.ctx, nil, order); updateErr != nil { + logx.Errorf("微信支付回调，更新订单失败%+v", updateErr) + return nil + } + + if order.Status == "paid" { + if asyncErr := l.svcCtx.AsynqService.SendQueryTask(order.Id); asyncErr != nil { + logx.Errorf("异步任务调度失败: %v", asyncErr) + return asyncErr + } + } + + w.WriteHeader(http.StatusOK) + _, _ = w.Write([]byte("success")) + return nil +} + +// 处理代理升级订单支付 +func (l *WechatPayCallbackLogic) handleAgentUpgradeOrderPayment(w http.ResponseWriter, notification *payments.Transaction) error { + orderNo := *notification.OutTradeNo + + // 1. 查找订单 + order, findOrderErr := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, orderNo) + if findOrderErr != nil { + logx.Errorf("微信支付回调，查找升级订单失败: %+v", findOrderErr) + return nil + } + + // 2. 验证金额 + amount := lzUtils.ToWechatAmount(order.Amount) + if amount != *notification.Amount.Total { + logx.Errorf("微信支付回调，升级订单金额不一致，订单号: %s", orderNo) + return nil + } + + // 3. 检查订单状态 + if order.Status != "pending" { + w.WriteHeader(http.StatusOK) + _, _ = w.Write([]byte("success")) + return nil + } + + // 4. 查找升级记录 + upgradeRecords, findUpgradeErr := l.svcCtx.AgentUpgradeModel.FindAll(l.ctx, l.svcCtx.AgentUpgradeModel.SelectBuilder(). + Where("order_no = ?", orderNo). + Limit(1), "") + if findUpgradeErr != nil || len(upgradeRecords) == 0 { + logx.Errorf("微信支付回调，查找升级记录失败，订单号: %s, 错误: %+v", orderNo, findUpgradeErr) + return nil + } + upgradeRecord := upgradeRecords[0] + + // 5. 检查升级记录状态 + if upgradeRecord.Status != 1 { + // 升级记录状态不是待支付，直接返回成功 + w.WriteHeader(http.StatusOK) + _, _ = w.Write([]byte("success")) + return nil + } + + // 6. 处理支付状态 + switch *notification.TradeState { + case service.TradeStateSuccess: + order.Status = "paid" + order.PayTime = lzUtils.TimeToNullTime(time.Now()) + order.PlatformOrderId = lzUtils.StringToNullString(*notification.TransactionId) + case service.TradeStateClosed: + order.Status = "closed" + order.CloseTime = lzUtils.TimeToNullTime(time.Now()) + case service.TradeStateRevoked: + order.Status = "failed" + default: + return nil + } + + // 7. 更新订单状态 + if updateErr := l.svcCtx.OrderModel.UpdateWithVersion(l.ctx, nil, order); updateErr != nil { + logx.Errorf("微信支付回调，更新升级订单状态失败: %+v", updateErr) + return nil + } + + // 8. 如果支付成功，执行升级操作 + if order.Status == "paid" { + err := l.svcCtx.AgentWalletModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + // 8.1 执行升级操作 + if err := l.svcCtx.AgentService.ProcessUpgrade(transCtx, upgradeRecord.AgentId, upgradeRecord.ToLevel, upgradeRecord.UpgradeType, upgradeRecord.UpgradeFee, upgradeRecord.RebateAmount, orderNo, ""); err != nil { + return errors.Wrapf(err, "执行升级操作失败") + } + + // 8.2 更新升级记录状态为已完成 + upgradeRecord.Status = 2 // 已完成（status: 1=待处理，2=已完成，3=已失败） + upgradeRecord.Remark = lzUtils.StringToNullString("支付成功，升级完成") + if updateErr := l.svcCtx.AgentUpgradeModel.UpdateWithVersion(transCtx, session, upgradeRecord); updateErr != nil { + return errors.Wrapf(updateErr, "更新升级记录状态失败") + } + + return nil + }) + + if err != nil { + logx.Errorf("微信支付回调，处理升级订单失败，订单号: %s, 错误: %+v", orderNo, err) + // 即使升级失败，也返回成功给微信，避免重复回调 + } else { + logx.Infof("微信支付回调，代理升级成功，订单号: %s, 代理ID: %d, 从等级 %d 升级到等级 %d", orderNo, upgradeRecord.AgentId, upgradeRecord.FromLevel, upgradeRecord.ToLevel) + } + } + + w.WriteHeader(http.StatusOK) + _, _ = w.Write([]byte("success")) + return nil +} + +// 处理代理会员订单支付（已废弃，新系统使用升级功能） +/* +func (l *WechatPayCallbackLogic) handleAgentVipOrderPayment(w http.ResponseWriter, notification *payments.Transaction) error { + agentOrder, findAgentOrderErr := l.svcCtx.AgentMembershipRechargeOrderModel.FindOneByOrderNo(l.ctx, *notification.OutTradeNo) + if findAgentOrderErr != nil { + logx.Errorf("微信支付回调，查找代理会员订单失败: %+v", findAgentOrderErr) + return nil + } + + if agentOrder.Status != "pending" { + w.WriteHeader(http.StatusOK) + _, _ = w.Write([]byte("success")) + return nil + } + + user, err := l.svcCtx.UserModel.FindOne(l.ctx, agentOrder.UserId) + if err != nil { + logx.Errorf("微信支付回调，查找用户失败: %+v", err) + return nil + } + + amount := lzUtils.ToWechatAmount(agentOrder.Amount) + if user.Inside != 1 { + if amount != *notification.Amount.Total { + logx.Errorf("微信支付回调，金额不一致") + return nil + } + } + + switch *notification.TradeState { + case service.TradeStateSuccess: + agentOrder.Status = "paid" + default: + return nil + } + + if agentOrder.Status == "paid" { + err = l.svcCtx.AgentModel.Trans(l.ctx, func(transCtx context.Context, session sqlx.Session) error { + agentModel, err := l.svcCtx.AgentModel.FindOne(transCtx, agentOrder.AgentId) + if err != nil { + return fmt.Errorf("查找代理信息失败: %+v", err) + } + + agentOrder.PlatformOrderId = lzUtils.StringToNullString(*notification.TransactionId) + if updateErr := l.svcCtx.AgentMembershipRechargeOrderModel.UpdateWithVersion(l.ctx, nil, agentOrder); updateErr != nil { + return fmt.Errorf("修改代理会员订单信息失败: %+v", updateErr) + } + + // 记录旧等级，用于判断是否为升级 + oldLevel := agentModel.LevelName + + // 设置会员等级 + agentModel.LevelName = agentOrder.LevelName + + // 延长会员时间 + // 检查是否是有效期内续费（不发放奖励）还是重新激活（发放奖励） + isValidRenewal := oldLevel == agentOrder.LevelName && agentModel.MembershipExpiryTime.Valid && agentModel.MembershipExpiryTime.Time.After(time.Now()) + if isValidRenewal { + logx.Infof("代理会员有效期内续费成功，会员ID：%d，等级：%s", agentModel.Id, agentModel.LevelName) + } else { + logx.Infof("代理会员新购、升级或重新激活成功，会员ID：%d，等级：%s", agentModel.Id, agentModel.LevelName) + } + agentModel.MembershipExpiryTime = lzUtils.RenewMembership(agentModel.MembershipExpiryTime) + + if updateErr := l.svcCtx.AgentModel.UpdateWithVersion(l.ctx, nil, agentModel); updateErr != nil { + return fmt.Errorf("修改代理信息失败: %+v", updateErr) + } + + // 如果不是有效期内续费，给上级代理发放升级奖励 + if !isValidRenewal && (agentOrder.LevelName == model.AgentLeveNameVIP || agentOrder.LevelName == model.AgentLeveNameSVIP) { + // 验证升级路径的有效性 + if oldLevel != agentOrder.LevelName { + upgradeRewardErr := l.svcCtx.AgentService.GiveUpgradeReward(transCtx, agentModel.Id, oldLevel, agentOrder.LevelName, session) + if upgradeRewardErr != nil { + logx.Errorf("发放升级奖励失败，代理ID：%d，旧等级：%s，新等级：%s，错误：%+v", agentModel.Id, oldLevel, agentOrder.LevelName, upgradeRewardErr) + // 升级奖励失败不影响主流程，只记录日志 + } else { + logx.Infof("发放升级奖励成功，代理ID：%d，旧等级：%s，新等级：%s", agentModel.Id, oldLevel, agentOrder.LevelName) + } + } + } + + return nil + }) + + if err != nil { + logx.Errorf("微信支付回调，处理代理会员订单失败: %+v", err) + refundErr := l.handleRefund(agentOrder) + if refundErr != nil { + logx.Errorf("微信支付回调，退款失败: %+v", refundErr) + } + return nil + } + } + + w.WriteHeader(http.StatusOK) + _, _ = w.Write([]byte("success")) + return nil +} +*/ + +/* +func (l *WechatPayCallbackLogic) handleRefund(order *model.AgentMembershipRechargeOrder) error { + ctx := context.Background() + // 退款 + if order.PaymentMethod == "wechat" { + refundErr := l.svcCtx.WechatPayService.WeChatRefund(ctx, order.OrderNo, order.Amount, order.Amount) + if refundErr != nil { + return refundErr + } + } else { + refund, refundErr := l.svcCtx.AlipayService.AliRefund(ctx, order.OrderNo, order.Amount) + if refundErr != nil { + return refundErr + } + if refund.IsSuccess() { + logx.Errorf("支付宝退款成功, orderID: %d", order.Id) + // 更新订单状态为退款 + order.Status = "refunded" + updateOrderErr := l.svcCtx.AgentMembershipRechargeOrderModel.UpdateWithVersion(ctx, nil, order) + if updateOrderErr != nil { + logx.Errorf("更新订单状态失败，订单ID: %d, 错误: %v", order.Id, updateOrderErr) + return fmt.Errorf("更新订单状态失败: %v", updateOrderErr) + } + return nil + } else { + logx.Errorf("支付宝退款失败:%v", refundErr) + return refundErr + } + } + return nil +} */ diff --git a/app/main/api/internal/logic/pay/wechatpayrefundcallbacklogic.go b/app/main/api/internal/logic/pay/wechatpayrefundcallbacklogic.go new file mode 100644 index 0000000..a4a0ccb --- /dev/null +++ b/app/main/api/internal/logic/pay/wechatpayrefundcallbacklogic.go @@ -0,0 +1,239 @@ +package pay + +import ( + "context" + "database/sql" + "net/http" + "strings" + "time" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/model" + "qnc-server/common/globalkey" + + "github.com/pkg/errors" + "github.com/wechatpay-apiv3/wechatpay-go/services/refunddomestic" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type WechatPayRefundCallbackLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewWechatPayRefundCallbackLogic(ctx context.Context, svcCtx *svc.ServiceContext) *WechatPayRefundCallbackLogic { + return &WechatPayRefundCallbackLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +// handleQueryOrderRefund 处理查询订单退款 +func (l *WechatPayRefundCallbackLogic) handleQueryOrderRefund(orderNo string, status refunddomestic.Status) error { + order, err := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, orderNo) + if err != nil { + return errors.Wrapf(err, "查找查询订单信息失败: %s", orderNo) + } + + // 检查订单是否已经处理过退款 + if order.Status == model.OrderStatusRefunded { + logx.Infof("订单已经是退款状态，无需重复处理: orderNo=%s", orderNo) + return nil + } + + // 只处理成功和失败状态 + var orderStatus, refundStatus string + switch status { + case refunddomestic.STATUS_SUCCESS: + orderStatus = model.OrderStatusRefunded + refundStatus = model.OrderRefundStatusSuccess + case refunddomestic.STATUS_CLOSED: + // 退款关闭，保持订单原状态，更新退款记录为失败 + refundStatus = model.OrderRefundStatusFailed + case refunddomestic.STATUS_ABNORMAL: + // 退款异常，保持订单原状态，更新退款记录为失败 + refundStatus = model.OrderRefundStatusFailed + default: + // 其他状态暂不处理 + return nil + } + + // 使用事务同时更新订单和退款记录 + err = l.svcCtx.OrderModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 更新订单状态（仅在退款成功时更新） + if status == refunddomestic.STATUS_SUCCESS { + order.Status = orderStatus + order.RefundTime = sql.NullTime{ + Time: time.Now(), + Valid: true, + } + if err := l.svcCtx.OrderModel.UpdateWithVersion(ctx, session, order); err != nil { + return errors.Wrapf(err, "更新查询订单状态失败: %s", orderNo) + } + } + + // 查找最新的pending状态的退款记录 + refund, err := l.findLatestPendingRefund(ctx, order.Id) + if err != nil { + if err == model.ErrNotFound { + logx.Errorf("未找到订单对应的待处理退款记录: orderNo=%s, orderId=%d", orderNo, order.Id) + return nil // 没有退款记录时不报错，只记录警告 + } + return errors.Wrapf(err, "查找退款记录失败: orderNo=%s", orderNo) + } + + // 检查退款记录是否已经处理过 + if refund.Status == model.OrderRefundStatusSuccess { + logx.Infof("退款记录已经是成功状态，无需重复处理: orderNo=%s, refundId=%d", orderNo, refund.Id) + return nil + } + + refund.Status = refundStatus + if status == refunddomestic.STATUS_SUCCESS { + refund.RefundTime = sql.NullTime{ + Time: time.Now(), + Valid: true, + } + } else if status == refunddomestic.STATUS_CLOSED { + refund.CloseTime = sql.NullTime{ + Time: time.Now(), + Valid: true, + } + } + + if _, err := l.svcCtx.OrderRefundModel.Update(ctx, session, refund); err != nil { + return errors.Wrapf(err, "更新退款记录状态失败: orderNo=%s", orderNo) + } + + return nil + }) + + if err != nil { + return errors.Wrapf(err, "更新订单和退款记录失败: %s", orderNo) + } + + return nil +} + +// handleAgentOrderRefund 处理代理会员订单退款（已废弃，新系统使用升级功能） +/* func (l *WechatPayRefundCallbackLogic) handleAgentOrderRefund(orderNo string, status refunddomestic.Status) error { + order, err := l.svcCtx.AgentMembershipRechargeOrderModel.FindOneByOrderNo(l.ctx, orderNo) + if err != nil { + return errors.Wrapf(err, "查找代理会员订单信息失败: %s", orderNo) + } + + // 检查订单是否已经处理过退款 + if order.Status == "refunded" { + logx.Infof("代理会员订单已经是退款状态，无需重复处理: orderNo=%s", orderNo) + return nil + } + + if status == refunddomestic.STATUS_SUCCESS { + order.Status = "refunded" + } else if status == refunddomestic.STATUS_ABNORMAL { + return nil // 异常状态直接返回 + } else { + return nil // 其他状态直接返回 + } + + if err := l.svcCtx.AgentMembershipRechargeOrderModel.UpdateWithVersion(l.ctx, nil, order); err != nil { + return errors.Wrapf(err, "更新代理会员订单状态失败: %s", orderNo) + } + + return nil +} */ + +// sendSuccessResponse 发送成功响应 +func (l *WechatPayRefundCallbackLogic) sendSuccessResponse(w http.ResponseWriter) { + w.WriteHeader(http.StatusOK) + _, _ = w.Write([]byte("success")) +} + +func (l *WechatPayRefundCallbackLogic) WechatPayRefundCallback(w http.ResponseWriter, r *http.Request) error { + // 1. 处理微信退款通知 + notification, err := l.svcCtx.WechatPayService.HandleRefundNotification(l.ctx, r) + if err != nil { + logx.Errorf("微信退款回调处理失败: %v", err) + l.sendSuccessResponse(w) + return nil + } + + // 2. 检查关键字段是否为空 + if notification.OutTradeNo == nil { + logx.Errorf("微信退款回调OutTradeNo字段为空") + l.sendSuccessResponse(w) + return nil + } + + orderNo := *notification.OutTradeNo + + // 3. 判断退款状态，优先使用Status，如果Status为nil则使用SuccessTime判断 + var status refunddomestic.Status + var statusDetermined bool = false + + if notification.Status != nil { + status = *notification.Status + statusDetermined = true + } else if notification.SuccessTime != nil && !notification.SuccessTime.IsZero() { + // 如果Status为空但SuccessTime有值，说明退款成功 + status = refunddomestic.STATUS_SUCCESS + statusDetermined = true + } else { + logx.Errorf("微信退款回调Status和SuccessTime都为空，无法确定退款状态: orderNo=%s", orderNo) + l.sendSuccessResponse(w) + return nil + } + + if !statusDetermined { + logx.Errorf("微信退款回调无法确定退款状态: orderNo=%s", orderNo) + l.sendSuccessResponse(w) + return nil + } + + var processErr error + + // 4. 根据订单号前缀处理不同类型的订单 + switch { + case strings.HasPrefix(orderNo, "Q_"): + processErr = l.handleQueryOrderRefund(orderNo, status) + case strings.HasPrefix(orderNo, "A_"): + // 旧系统会员充值订单退款（已废弃，新系统使用升级功能） + // processErr = l.handleAgentOrderRefund(orderNo, status) + // 直接返回，避免旧订单影响 + processErr = nil + default: + // 兼容旧订单，假设没有前缀的是查询订单 + processErr = l.handleQueryOrderRefund(orderNo, status) + } + + // 5. 处理错误并响应 + if processErr != nil { + logx.Errorf("处理退款订单失败: orderNo=%s, err=%v", orderNo, processErr) + } + + // 无论处理是否成功，都返回成功响应给微信 + l.sendSuccessResponse(w) + return nil +} + +// findLatestPendingRefund 查找订单最新的pending状态退款记录 +func (l *WechatPayRefundCallbackLogic) findLatestPendingRefund(ctx context.Context, orderId string) (*model.OrderRefund, error) { + // 使用SelectBuilder查询最新的pending状态退款记录 + builder := l.svcCtx.OrderRefundModel.SelectBuilder(). + Where("order_id = ? AND status = ? AND del_state = ?", orderId, model.OrderRefundStatusPending, globalkey.DelStateNo). + OrderBy("id DESC"). + Limit(1) + + refunds, err := l.svcCtx.OrderRefundModel.FindAll(ctx, builder, "") + if err != nil { + return nil, err + } + + if len(refunds) == 0 { + return nil, model.ErrNotFound + } + + return refunds[0], nil +} diff --git a/app/main/api/internal/logic/product/getproductappbyenlogic.go b/app/main/api/internal/logic/product/getproductappbyenlogic.go new file mode 100644 index 0000000..f1c4ad6 --- /dev/null +++ b/app/main/api/internal/logic/product/getproductappbyenlogic.go @@ -0,0 +1,75 @@ +package product + +import ( + "context" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/Masterminds/squirrel" + "github.com/jinzhu/copier" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/mr" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetProductAppByEnLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetProductAppByEnLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetProductAppByEnLogic { + return &GetProductAppByEnLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetProductAppByEnLogic) GetProductAppByEn(req *types.GetProductByEnRequest) (resp *types.ProductResponse, err error) { + productModel, err := l.svcCtx.ProductModel.FindOneByProductEn(l.ctx, req.ProductEn) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "产品查询, 查找产品错误: %v", err) + } + + build := l.svcCtx.ProductFeatureModel.SelectBuilder().Where(squirrel.Eq{ + "product_id": productModel.Id, + }) + productFeatureAll, err := l.svcCtx.ProductFeatureModel.FindAll(l.ctx, build, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "产品查询, 查找产品关联错误: %v", err) + } + var product types.Product + err = copier.Copy(&product, productModel) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, 用户信息结构体复制失败, %v", err) + } + mr.MapReduceVoid(func(source chan<- interface{}) { + for _, productFeature := range productFeatureAll { + source <- productFeature.FeatureId + } + }, func(item interface{}, writer mr.Writer[*model.Feature], cancel func(error)) { + id := item.(string) + + feature, findFeatureErr := l.svcCtx.FeatureModel.FindOne(l.ctx, id) + if findFeatureErr != nil { + logx.WithContext(l.ctx).Errorf("产品查询, 查找关联feature错误: %d, err:%v", id, findFeatureErr) + return + } + if feature != nil && feature.Id != "" { + writer.Write(feature) + } + }, func(pipe <-chan *model.Feature, cancel func(error)) { + for item := range pipe { + var feature types.Feature + _ = copier.Copy(&feature, item) + product.Features = append(product.Features, feature) + } + }) + + return &types.ProductResponse{Product: product}, nil +} diff --git a/app/main/api/internal/logic/product/getproductbyenlogic.go b/app/main/api/internal/logic/product/getproductbyenlogic.go new file mode 100644 index 0000000..d6f33f8 --- /dev/null +++ b/app/main/api/internal/logic/product/getproductbyenlogic.go @@ -0,0 +1,91 @@ +package product + +import ( + "context" + "fmt" + "sort" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/Masterminds/squirrel" + "github.com/jinzhu/copier" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/mr" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetProductByEnLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetProductByEnLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetProductByEnLogic { + return &GetProductByEnLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetProductByEnLogic) GetProductByEn(req *types.GetProductByEnRequest) (resp *types.ProductResponse, err error) { + productModel, err := l.svcCtx.ProductModel.FindOneByProductEn(l.ctx, req.ProductEn) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "产品查询, 查找产品错误: %v", err) + } + + build := l.svcCtx.ProductFeatureModel.SelectBuilder().Where(squirrel.Eq{ + "product_id": productModel.Id, + }) + productFeatureAll, err := l.svcCtx.ProductFeatureModel.FindAll(l.ctx, build, "") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "产品查询, 查找产品关联错误: %v", err) + } + + // 创建featureId到sort的映射，用于后续排序 + featureSortMap := make(map[string]int64) + for _, productFeature := range productFeatureAll { + featureSortMap[fmt.Sprintf("%d", productFeature.FeatureId)] = productFeature.Sort + } + + var product types.Product + err = copier.Copy(&product, productModel) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, 用户信息结构体复制失败, %v", err) + } + mr.MapReduceVoid(func(source chan<- interface{}) { + for _, productFeature := range productFeatureAll { + source <- productFeature.FeatureId + } + }, func(item interface{}, writer mr.Writer[*model.Feature], cancel func(error)) { + id := item.(string) + + feature, findFeatureErr := l.svcCtx.FeatureModel.FindOne(l.ctx, id) + if findFeatureErr != nil { + logx.WithContext(l.ctx).Errorf("产品查询, 查找关联feature错误: %d, err:%v", id, findFeatureErr) + return + } + if feature != nil && feature.Id != "" { + writer.Write(feature) + } + }, func(pipe <-chan *model.Feature, cancel func(error)) { + for item := range pipe { + var feature types.Feature + _ = copier.Copy(&feature, item) + product.Features = append(product.Features, feature) + } + }) + + // 按照productFeature.Sort字段对features进行排序 + sort.Slice(product.Features, func(i, j int) bool { + sortI := featureSortMap[product.Features[i].ID] + sortJ := featureSortMap[product.Features[j].ID] + return sortI < sortJ + }) + + return &types.ProductResponse{Product: product}, nil +} diff --git a/app/main/api/internal/logic/product/getproductbyidlogic.go b/app/main/api/internal/logic/product/getproductbyidlogic.go new file mode 100644 index 0000000..60ee339 --- /dev/null +++ b/app/main/api/internal/logic/product/getproductbyidlogic.go @@ -0,0 +1,30 @@ +package product + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetProductByIDLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetProductByIDLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetProductByIDLogic { + return &GetProductByIDLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetProductByIDLogic) GetProductByID(req *types.GetProductByIDRequest) (resp *types.ProductResponse, err error) { + // todo: add your logic here and delete this line + + return +} diff --git a/app/main/api/internal/logic/query/query_common.go b/app/main/api/internal/logic/query/query_common.go new file mode 100644 index 0000000..2f8350b --- /dev/null +++ b/app/main/api/internal/logic/query/query_common.go @@ -0,0 +1,157 @@ +package query + +import ( + "context" + "database/sql" + "encoding/hex" + "encoding/json" + "fmt" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/jinzhu/copier" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" +) + +func ProcessQueryData(queryData sql.NullString, target *[]types.QueryItem, key []byte) error { + queryDataStr := lzUtils.NullStringToString(queryData) + if queryDataStr == "" { + return nil + } + decryptedData, decryptErr := crypto.AesDecrypt(queryDataStr, key) + if decryptErr != nil { + return decryptErr + } + var decryptedArray []map[string]interface{} + unmarshalErr := json.Unmarshal(decryptedData, &decryptedArray) + if unmarshalErr != nil { + return unmarshalErr + } + if len(*target) == 0 { + *target = make([]types.QueryItem, len(decryptedArray)) + } + for i := 0; i < len(decryptedArray); i++ { + (*target)[i].Data = decryptedArray[i] + } + return nil +} + +func ProcessQueryParams(QueryParams string, target *map[string]interface{}, key []byte) error { + decryptedData, decryptErr := crypto.AesDecrypt(QueryParams, key) + if decryptErr != nil { + return decryptErr + } + unmarshalErr := json.Unmarshal(decryptedData, target) + if unmarshalErr != nil { + return unmarshalErr + } + return nil +} + +func UpdateFeatureAndProductFeature(ctx context.Context, svcCtx *svc.ServiceContext, productID string, target *[]types.QueryItem) error { + for i := len(*target) - 1; i >= 0; i-- { + queryItem := &(*target)[i] + data, ok := queryItem.Data.(map[string]interface{}) + if !ok { + return fmt.Errorf("queryItem.Data 必须是 map[string]interface{} 类型") + } + apiID, ok := data["apiID"].(string) + if !ok { + return fmt.Errorf("queryItem.Data 中的 apiID 必须是字符串类型") + } + feature, err := svcCtx.FeatureModel.FindOneByApiId(ctx, apiID) + if err != nil { + *target = append((*target)[:i], (*target)[i+1:]...) + continue + } + builder := svcCtx.ProductFeatureModel.SelectBuilder().Where("product_id = ?", productID) + productFeatures, err := svcCtx.ProductFeatureModel.FindAll(ctx, builder, "") + if err != nil { + return fmt.Errorf("查询 ProductFeatureModel 错误: %v", err) + } + sort := 0 + for _, pf := range productFeatures { + if pf.FeatureId == feature.Id { + sort = int(pf.Sort) + break + } + } + featureData := map[string]interface{}{ + "featureName": feature.Name, + "sort": sort, + } + queryItem.Feature = featureData + } + return nil +} + +func BuildEncryptedQuery(ctx context.Context, svcCtx *svc.ServiceContext, queryModel *model.Query) (string, error) { + var query types.Query + query.CreateTime = queryModel.CreateTime.Format("2006-01-02 15:04:05") + query.UpdateTime = queryModel.UpdateTime.Format("2006-01-02 15:04:05") + secretKey := svcCtx.Config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 获取AES解密解药失败, %v", decodeErr) + } + processParamsErr := ProcessQueryParams(queryModel.QueryParams, &query.QueryParams, key) + if processParamsErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告参数处理失败: %v", processParamsErr) + } + processErr := ProcessQueryData(queryModel.QueryData, &query.QueryData, key) + if processErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结果处理失败: %v", processErr) + } + updateErr := UpdateFeatureAndProductFeature(ctx, svcCtx, queryModel.ProductId, &query.QueryData) + if updateErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结果处理失败: %v", updateErr) + } + err := copier.Copy(&query, queryModel) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结构体复制失败, %v", err) + } + product, err := svcCtx.ProductModel.FindOne(ctx, queryModel.ProductId) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 获取商品信息失败, %v", err) + } + query.Product = product.ProductEn + query.ProductName = product.ProductName + queryBytes, marshalErr := json.Marshal(query) + if marshalErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 序列化查询结果失败: %v", marshalErr) + } + encryptedQuery, encryptErr := crypto.AesEncrypt(queryBytes, key) + if encryptErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 加密查询结果失败: %v", encryptErr) + } + return encryptedQuery, nil +} + +func IsOrderAgent(ctx context.Context, svcCtx *svc.ServiceContext, userId string, orderId string) (bool, error) { + agent, err := svcCtx.AgentModel.FindOneByUserId(ctx, userId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return false, nil + } + return false, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败, %v", err) + } + if agent == nil { + return false, nil + } + agentOrder, err := svcCtx.AgentOrderModel.FindOneByOrderId(ctx, orderId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return false, nil + } + return false, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理订单失败, %v", err) + } + if agentOrder == nil { + return false, nil + } + return agentOrder.AgentId == agent.Id, nil +} diff --git a/app/main/api/internal/logic/query/querydetailbyorderidlogic.go b/app/main/api/internal/logic/query/querydetailbyorderidlogic.go new file mode 100644 index 0000000..df1ce85 --- /dev/null +++ b/app/main/api/internal/logic/query/querydetailbyorderidlogic.go @@ -0,0 +1,76 @@ +package query + +import ( + "context" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + + "github.com/zeromicro/go-zero/core/logx" +) + +type QueryDetailByOrderIdLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQueryDetailByOrderIdLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryDetailByOrderIdLogic { + return &QueryDetailByOrderIdLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QueryDetailByOrderIdLogic) QueryDetailByOrderId(req *types.QueryDetailByOrderIdReq) (resp string, err error) { + // 获取当前用户ID + userId, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户ID失败: %v", err) + } + + // 获取订单信息 + order, err := l.svcCtx.OrderModel.FindOne(l.ctx, req.OrderId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return "", errors.Wrapf(xerr.NewErrCode(xerr.LOGIC_QUERY_NOT_FOUND), "报告查询, 订单不存在: %v", err) + } + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告查询, 查找报告错误: %v", err) + } + user, err := l.svcCtx.UserModel.FindOne(l.ctx, userId) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告查询, 查找用户错误: %v", err) + } + if user.Inside != 1 { + // 安全验证：确保订单属于当前用户，或为该订单的代理 + if order.UserId != userId { + isAgent, aerr := IsOrderAgent(l.ctx, l.svcCtx, userId, order.Id) + if aerr != nil { + return "", aerr + } + if !isAgent { + return "", errors.Wrapf(xerr.NewErrCode(xerr.LOGIC_QUERY_NOT_FOUND), "无权查看此订单报告") + } + } + } + + // 检查订单状态 + if order.Status != "paid" { + return "", errors.Wrapf(xerr.NewErrMsg("订单未支付，无法查看报告"), "") + } + + queryModel, err := l.svcCtx.QueryModel.FindOneByOrderId(l.ctx, req.OrderId) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告查询, 查找报告错误: %v", err) + } + respStr, buildErr := BuildEncryptedQuery(l.ctx, l.svcCtx, queryModel) + if buildErr != nil { + return "", buildErr + } + return respStr, nil +} diff --git a/app/main/api/internal/logic/query/querydetailbyordernologic.go b/app/main/api/internal/logic/query/querydetailbyordernologic.go new file mode 100644 index 0000000..ca1cc43 --- /dev/null +++ b/app/main/api/internal/logic/query/querydetailbyordernologic.go @@ -0,0 +1,72 @@ +package query + +import ( + "context" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + + "github.com/zeromicro/go-zero/core/logx" +) + +type QueryDetailByOrderNoLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQueryDetailByOrderNoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryDetailByOrderNoLogic { + return &QueryDetailByOrderNoLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QueryDetailByOrderNoLogic) QueryDetailByOrderNo(req *types.QueryDetailByOrderNoReq) (resp string, err error) { + // 获取当前用户ID + userId, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户ID失败: %v", err) + } + + // 获取订单信息 + order, err := l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, req.OrderNo) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return "", errors.Wrapf(xerr.NewErrCode(xerr.LOGIC_QUERY_NOT_FOUND), "报告查询, 订单不存在: %v", err) + } + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告查询, 查找报告错误: %v", err) + } + + // 安全验证：确保订单属于当前用户，或为该订单的代理 + if order.UserId != userId { + isAgent, aerr := IsOrderAgent(l.ctx, l.svcCtx, userId, order.Id) + if aerr != nil { + return "", aerr + } + if !isAgent { + return "", errors.Wrapf(xerr.NewErrCode(xerr.LOGIC_QUERY_NOT_FOUND), "无权查看此订单报告") + } + } + + // 检查订单状态 + if order.Status != "paid" { + return "", errors.Wrapf(xerr.NewErrMsg("订单未支付，无法查看报告"), "") + } + + queryModel, err := l.svcCtx.QueryModel.FindOneByOrderId(l.ctx, order.Id) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告查询, 查找报告错误: %v", err) + } + respStr, buildErr := BuildEncryptedQuery(l.ctx, l.svcCtx, queryModel) + if buildErr != nil { + return "", buildErr + } + return respStr, nil +} diff --git a/app/main/api/internal/logic/query/queryexamplelogic.go b/app/main/api/internal/logic/query/queryexamplelogic.go new file mode 100644 index 0000000..bc9f7fb --- /dev/null +++ b/app/main/api/internal/logic/query/queryexamplelogic.go @@ -0,0 +1,120 @@ +package query + +import ( + "context" + "encoding/hex" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/bytedance/sonic" + "github.com/pkg/errors" + + "github.com/zeromicro/go-zero/core/logx" +) + +type QueryExampleLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQueryExampleLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryExampleLogic { + return &QueryExampleLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QueryExampleLogic) QueryExample(req *types.QueryExampleReq) (resp string, err error) { + // 根据产品特性标识获取产品信息 + product, err := l.svcCtx.ProductModel.FindOneByProductEn(l.ctx, req.Feature) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "示例报告, 获取商品信息失败, %v", err) + } + + secretKeyHex := l.svcCtx.Config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKeyHex) + if decodeErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "示例报告, 解析AES密钥失败, %v", decodeErr) + } + + // 创建一个空的Query结构体来存储结果 + query := types.Query{ + Product: product.ProductEn, + ProductName: product.ProductName, + QueryData: make([]types.QueryItem, 0), + QueryParams: make(map[string]interface{}), + } + query.QueryParams = map[string]interface{}{ + "id_card": "45000000000000000", + "mobile": "13700000000", + "name": "张老三", + } + // 查询ProductFeatureModel获取产品相关的功能列表 + builder := l.svcCtx.ProductFeatureModel.SelectBuilder().Where("product_id = ?", product.Id) + productFeatures, err := l.svcCtx.ProductFeatureModel.FindAll(l.ctx, builder, "") + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "示例报告, 查询 ProductFeatureModel 错误: %v", err) + } + // 从每个启用的特性获取示例数据并合并 + for _, pf := range productFeatures { + if pf.Enable != 1 { + continue // 跳过未启用的特性 + } + + // 根据特性ID查找示例数据 + example, err := l.svcCtx.ExampleModel.FindOneByFeatureId(l.ctx, pf.FeatureId) + if err != nil { + logx.Infof("示例报告, 特性ID %d 无示例数据: %v", pf.FeatureId, err) + continue // 如果没有示例数据就跳过 + } + + // 获取对应的Feature信息 + feature, err := l.svcCtx.FeatureModel.FindOne(l.ctx, pf.FeatureId) + if err != nil { + logx.Infof("示例报告, 无法获取特性ID %d 的信息: %v", pf.FeatureId, err) + continue + } + + var queryItem types.QueryItem + + // 解密查询数据 + // 解析示例内容 + if example.Content == "000" { + queryItem.Data = example.Content + } else { + // 解密数据 + decryptedData, decryptErr := crypto.AesDecrypt(example.Content, key) + if decryptErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "示例报告, 解密数据失败: %v", decryptErr) + } + err = sonic.Unmarshal([]byte(decryptedData), &queryItem.Data) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "示例报告, 解析示例内容失败: %v", err) + } + } + + // 添加特性信息 + queryItem.Feature = map[string]interface{}{ + "featureName": feature.Name, + "sort": pf.Sort, + } + // 添加到查询数据中 + query.QueryData = append(query.QueryData, queryItem) + } + + queryBytes, marshalErr := sonic.Marshal(query) + if marshalErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "示例报告, 序列化查询结果失败: %v", marshalErr) + } + + encryptedQuery, encryptErr := crypto.AesEncrypt(queryBytes, key) + if encryptErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "示例报告, 加密查询结果失败: %v", encryptErr) + } + + return encryptedQuery, nil +} diff --git a/app/main/api/internal/logic/query/querygeneratesharelinklogic.go b/app/main/api/internal/logic/query/querygeneratesharelinklogic.go new file mode 100644 index 0000000..3707c2e --- /dev/null +++ b/app/main/api/internal/logic/query/querygeneratesharelinklogic.go @@ -0,0 +1,111 @@ +package query + +import ( + "context" + "encoding/hex" + "encoding/json" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type QueryGenerateShareLinkLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQueryGenerateShareLinkLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryGenerateShareLinkLogic { + return &QueryGenerateShareLinkLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QueryGenerateShareLinkLogic) QueryGenerateShareLink(req *types.QueryGenerateShareLinkReq) (resp *types.QueryGenerateShareLinkResp, err error) { + userId, err := ctxdata.GetUidFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成分享链接, 获取用户ID失败: %v", err) + } + + // 检查参数 + if (req.OrderId == nil || *req.OrderId == "") && (req.OrderNo == nil || *req.OrderNo == "") { + return nil, errors.Wrapf(xerr.NewErrMsg("订单ID和订单号不能同时为空"), "") + } + + var order *model.Order + // 优先使用OrderId查询 + if req.OrderId != nil && *req.OrderId != "" { + order, err = l.svcCtx.OrderModel.FindOne(l.ctx, *req.OrderId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("订单不存在"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成分享链接, 获取订单失败: %v", err) + } + } else if req.OrderNo != nil && *req.OrderNo != "" { + // 使用OrderNo查询 + order, err = l.svcCtx.OrderModel.FindOneByOrderNo(l.ctx, *req.OrderNo) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrMsg("订单不存在"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成分享链接, 获取订单失败: %v", err) + } + } else { + return nil, errors.Wrapf(xerr.NewErrMsg("订单ID和订单号不能同时为空"), "") + } + + if order.Status != model.OrderStatusPaid { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成分享链接, 订单未支付") + } + + query, err := l.svcCtx.QueryModel.FindOneByOrderId(l.ctx, order.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成分享链接, 获取查询失败: %v", err) + } + + if query.QueryState != model.QueryStateSuccess { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成分享链接, 查询未成功") + } + user, err := l.svcCtx.UserModel.FindOne(l.ctx, userId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成分享链接, 获取用户失败: %v", err) + } + if user.Inside != 1 { + if order.UserId != userId { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成分享链接, 无权操作此订单") + } + } + + expireAt := time.Now().Add(time.Duration(l.svcCtx.Config.Query.ShareLinkExpire) * time.Second) + payload := types.QueryShareLinkPayload{ + OrderId: order.Id, // 使用查询到的订单ID + ExpireAt: expireAt.Unix(), + } + secretKey := l.svcCtx.Config.Encrypt.SecretKey + key, err := hex.DecodeString(secretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成分享链接, 解密失败: %v", err) + } + payloadBytes, err := json.Marshal(payload) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成分享链接, 序列化失败: %v", err) + } + encryptedPayload, err := crypto.AesEncryptURL(payloadBytes, key) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成分享链接, 加密失败: %v", err) + } + return &types.QueryGenerateShareLinkResp{ + ShareLink: encryptedPayload, + }, nil +} diff --git a/app/main/api/internal/logic/query/querylistlogic.go b/app/main/api/internal/logic/query/querylistlogic.go new file mode 100644 index 0000000..c2810f0 --- /dev/null +++ b/app/main/api/internal/logic/query/querylistlogic.go @@ -0,0 +1,78 @@ +package query + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + + "github.com/Masterminds/squirrel" + "github.com/jinzhu/copier" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type QueryListLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQueryListLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryListLogic { + return &QueryListLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QueryListLogic) QueryList(req *types.QueryListReq) (resp *types.QueryListResp, err error) { + userID, getUidErr := ctxdata.GetUidFromCtx(l.ctx) + if getUidErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告列表查询, 获取用户信息失败, %+v", getUidErr) + } + + // 直接构建查询query表的条件 + build := l.svcCtx.QueryModel.SelectBuilder().Where(squirrel.Eq{ + "user_id": userID, + }) + + // 直接从query表分页查询 + queryList, total, err := l.svcCtx.QueryModel.FindPageListByPageWithTotal(l.ctx, build, req.Page, req.PageSize, "create_time DESC") + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告列表查询, 查找报告列表错误, %+v", err) + } + + var list []types.Query + if len(queryList) > 0 { + for _, queryModel := range queryList { + var query types.Query + query.CreateTime = queryModel.CreateTime.Format("2006-01-02 15:04:05") + query.UpdateTime = queryModel.UpdateTime.Format("2006-01-02 15:04:05") + copyErr := copier.Copy(&query, queryModel) + if copyErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告列表查询, 报告结构体复制失败, %+v", err) + } + product, findProductErr := l.svcCtx.ProductModel.FindOne(l.ctx, queryModel.ProductId) + if findProductErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告列表查询, 获取商品信息失败, %+v", err) + } + + // 检查订单状态，如果订单已退款，则设置查询状态为已退款 + order, findOrderErr := l.svcCtx.OrderModel.FindOne(l.ctx, queryModel.OrderId) + if findOrderErr == nil && order.Status == model.OrderStatusRefunded { + query.QueryState = model.QueryStateRefunded + } + query.ProductName = product.ProductName + query.Product = product.ProductEn + list = append(list, query) + } + } + + return &types.QueryListResp{ + Total: total, + List: list, + }, nil +} diff --git a/app/main/api/internal/logic/query/queryprovisionalorderlogic.go b/app/main/api/internal/logic/query/queryprovisionalorderlogic.go new file mode 100644 index 0000000..def95db --- /dev/null +++ b/app/main/api/internal/logic/query/queryprovisionalorderlogic.go @@ -0,0 +1,63 @@ +package query + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "encoding/json" + "fmt" + + "github.com/jinzhu/copier" + "github.com/pkg/errors" + + "github.com/zeromicro/go-zero/core/logx" +) + +type QueryProvisionalOrderLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQueryProvisionalOrderLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryProvisionalOrderLogic { + return &QueryProvisionalOrderLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QueryProvisionalOrderLogic) QueryProvisionalOrder(req *types.QueryProvisionalOrderReq) (resp *types.QueryProvisionalOrderResp, err error) { + userID, getUidErr := ctxdata.GetUidFromCtx(l.ctx) + if getUidErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取临时订单, 获取用户信息失败, %+v", getUidErr) + } + redisKey := fmt.Sprintf("%d:%s", userID, req.Id) + cache, cacheErr := l.svcCtx.Redis.GetCtx(l.ctx, redisKey) + if cacheErr != nil { + return nil, cacheErr + } + var data types.QueryCache + err = json.Unmarshal([]byte(cache), &data) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取临时订单, 解析缓存内容失败, %v", err) + } + + productModel, err := l.svcCtx.ProductModel.FindOneByProductEn(l.ctx, data.Product) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取临时订单, 查找产品错误: %v", err) + } + var product types.Product + err = copier.Copy(&product, productModel) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取临时订单, 用户信息结构体复制失败: %v", err) + } + return &types.QueryProvisionalOrderResp{ + Name: data.Name, + IdCard: data.Name, + Mobile: data.Mobile, + Product: product, + }, nil +} diff --git a/app/main/api/internal/logic/query/queryretrylogic.go b/app/main/api/internal/logic/query/queryretrylogic.go new file mode 100644 index 0000000..3ff044c --- /dev/null +++ b/app/main/api/internal/logic/query/queryretrylogic.go @@ -0,0 +1,44 @@ +package query + +import ( + "context" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type QueryRetryLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQueryRetryLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryRetryLogic { + return &QueryRetryLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QueryRetryLogic) QueryRetry(req *types.QueryRetryReq) (resp *types.QueryRetryResp, err error) { + + query, err := l.svcCtx.QueryModel.FindOne(l.ctx, req.Id) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告查询重试, 查找报告失败, %v", err) + } + if query.QueryState == "success" { + return nil, errors.Wrapf(xerr.NewErrCodeMsg(xerr.LOGIN_FAILED, "该报告不能重试"), "报告查询重试, 该报告不能重试, %d", query.Id) + } + + if asyncErr := l.svcCtx.AsynqService.SendQueryTask(query.OrderId); asyncErr != nil { + logx.Errorf("异步任务调度失败: %v", asyncErr) + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告查询重试, 异步任务调度失败, %+v", asyncErr) + } + return +} diff --git a/app/main/api/internal/logic/query/queryserviceagentlogic.go b/app/main/api/internal/logic/query/queryserviceagentlogic.go new file mode 100644 index 0000000..8ba199d --- /dev/null +++ b/app/main/api/internal/logic/query/queryserviceagentlogic.go @@ -0,0 +1,33 @@ +package query + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type QueryServiceAgentLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQueryServiceAgentLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryServiceAgentLogic { + return &QueryServiceAgentLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QueryServiceAgentLogic) QueryServiceAgent(req *types.QueryServiceReq) (resp *types.QueryServiceResp, err error) { + if req.AgentIdentifier != "" { + l.ctx = context.WithValue(l.ctx, "agentIdentifier", req.AgentIdentifier) + } else if req.App { + l.ctx = context.WithValue(l.ctx, "app", req.App) + } + proxy := NewQueryServiceLogic(l.ctx, l.svcCtx) + return proxy.PreprocessLogic(req, req.Product) +} diff --git a/app/main/api/internal/logic/query/queryserviceapplogic.go b/app/main/api/internal/logic/query/queryserviceapplogic.go new file mode 100644 index 0000000..9e8ea52 --- /dev/null +++ b/app/main/api/internal/logic/query/queryserviceapplogic.go @@ -0,0 +1,30 @@ +package query + +import ( + "context" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type QueryServiceAppLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQueryServiceAppLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryServiceAppLogic { + return &QueryServiceAppLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QueryServiceAppLogic) QueryServiceApp(req *types.QueryServiceReq) (resp *types.QueryServiceResp, err error) { + // todo: add your logic here and delete this line + + return +} diff --git a/app/main/api/internal/logic/query/queryservicelogic.go b/app/main/api/internal/logic/query/queryservicelogic.go new file mode 100644 index 0000000..6e453dd --- /dev/null +++ b/app/main/api/internal/logic/query/queryservicelogic.go @@ -0,0 +1,736 @@ +package query + +import ( + "context" + "encoding/hex" + "encoding/json" + "fmt" + "os" + "time" + "qnc-server/app/main/api/internal/service" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + "qnc-server/pkg/lzkit/validator" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/redis" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type QueryServiceLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQueryServiceLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryServiceLogic { + return &QueryServiceLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QueryServiceLogic) QueryService(req *types.QueryServiceReq) (resp *types.QueryServiceResp, err error) { + if req.AgentIdentifier != "" { + l.ctx = context.WithValue(l.ctx, "agentIdentifier", req.AgentIdentifier) + } else if req.App { + l.ctx = context.WithValue(l.ctx, "app", req.App) + } + return l.PreprocessLogic(req, req.Product) +} + +var productProcessors = map[string]func(*QueryServiceLogic, *types.QueryServiceReq) (*types.QueryServiceResp, error){ + "marriage": (*QueryServiceLogic).ProcessMarriageLogic, + "homeservice": (*QueryServiceLogic).ProcessHomeServiceLogic, + "riskassessment": (*QueryServiceLogic).ProcessRiskAssessmentLogic, + "companyinfo": (*QueryServiceLogic).ProcessCompanyInfoLogic, + "rentalinfo": (*QueryServiceLogic).ProcessRentalInfoLogic, + "preloanbackgroundcheck": (*QueryServiceLogic).ProcessPreLoanBackgroundCheckLogic, + "backgroundcheck": (*QueryServiceLogic).ProcessBackgroundCheckLogic, + "personalData": (*QueryServiceLogic).ProcessPersonalDataLogic, + "consumerFinanceReport": (*QueryServiceLogic).ProcessConsumerFinanceReportLogic, +} + +func (l *QueryServiceLogic) PreprocessLogic(req *types.QueryServiceReq, product string) (*types.QueryServiceResp, error) { + if processor, exists := productProcessors[product]; exists { + return processor(l, req) // 璋冪敤瀵瑰簲鐨勫鐞嗗嚱鏁? + } + return nil, errors.New("鏈壘鍒扮浉搴旂殑澶勭悊绋嬪簭") +} +func (l *QueryServiceLogic) ProcessMarriageLogic(req *types.QueryServiceReq) (*types.QueryServiceResp, error) { + + // AES瑙ｅ瘑 + decryptData, DecryptDataErr := l.DecryptData(req.Data) + if DecryptDataErr != nil { + return nil, DecryptDataErr + } + + // 鏍￠獙鍙傛暟 + var data types.MarriageReq + if unmarshalErr := json.Unmarshal(decryptData, &data); unmarshalErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 瑙ｅ瘑鍚庣殑鏁版嵁鏍煎紡涓嶆纭? %+v", unmarshalErr) + } + + if validatorErr := validator.Validate(data); validatorErr != nil { + return nil, errors.Wrapf(xerr.NewErrCodeMsg(xerr.PARAM_VERIFICATION_ERROR, validatorErr.Error()), "鏌ヨ鏈嶅姟, 鍙傛暟涓嶆纭? %+v", validatorErr) + } + + // 鏍￠獙楠岃瘉鐮? + verifyCodeErr := l.VerifyCode(data.Mobile, data.Code) + if verifyCodeErr != nil { + return nil, verifyCodeErr + } + + // 鏍￠獙涓夎绱? + verifyErr := l.Verify(data.Name, data.IDCard, data.Mobile) + if verifyErr != nil { + return nil, verifyErr + } + + // 缂撳瓨 + params := map[string]interface{}{ + "name": data.Name, + "id_card": data.IDCard, + "mobile": data.Mobile, + } + userID, err := l.GetOrCreateUser() + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 澶勭悊鐢ㄦ埛澶辫触: %v", err) + } + cacheNo, cacheDataErr := l.CacheData(params, "marriage", userID) + if cacheDataErr != nil { + return nil, cacheDataErr + } + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 鐢熸垚token澶辫触 : %v", err) + } + + // 鑾峰彇褰撳墠鏃堕棿鎴? + now := time.Now().Unix() + return &types.QueryServiceResp{ + Id: cacheNo, + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} + +// 澶勭悊瀹舵斂鏈嶅姟鐩稿叧閫昏緫 +func (l *QueryServiceLogic) ProcessHomeServiceLogic(req *types.QueryServiceReq) (*types.QueryServiceResp, error) { + + // AES瑙ｅ瘑 + decryptData, DecryptDataErr := l.DecryptData(req.Data) + if DecryptDataErr != nil { + return nil, DecryptDataErr + } + + // 鏍￠獙鍙傛暟 + var data types.HomeServiceReq + if unmarshalErr := json.Unmarshal(decryptData, &data); unmarshalErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 瑙ｅ瘑鍚庣殑鏁版嵁鏍煎紡涓嶆纭? %+v", unmarshalErr) + } + + if validatorErr := validator.Validate(data); validatorErr != nil { + return nil, errors.Wrapf(xerr.NewErrCodeMsg(xerr.PARAM_VERIFICATION_ERROR, validatorErr.Error()), "鏌ヨ鏈嶅姟, 鍙傛暟涓嶆纭? %+v", validatorErr) + } + + // 鏍￠獙楠岃瘉鐮? + verifyCodeErr := l.VerifyCode(data.Mobile, data.Code) + if verifyCodeErr != nil { + return nil, verifyCodeErr + } + + // 鏍￠獙涓夎绱? + verifyErr := l.Verify(data.Name, data.IDCard, data.Mobile) + if verifyErr != nil { + return nil, verifyErr + } + + // 缂撳瓨 + params := map[string]interface{}{ + "name": data.Name, + "id_card": data.IDCard, + "mobile": data.Mobile, + } + userID, err := l.GetOrCreateUser() + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 澶勭悊鐢ㄦ埛澶辫触: %v", err) + } + cacheNo, cacheDataErr := l.CacheData(params, "homeservice", userID) + if cacheDataErr != nil { + return nil, cacheDataErr + } + + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 鐢熸垚token澶辫触 : %d", userID) + } + + // 鑾峰彇褰撳墠鏃堕棿鎴? + now := time.Now().Unix() + return &types.QueryServiceResp{ + Id: cacheNo, + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} + +// 澶勭悊椋庨櫓璇勪及鐩稿叧閫昏緫 +func (l *QueryServiceLogic) ProcessRiskAssessmentLogic(req *types.QueryServiceReq) (*types.QueryServiceResp, error) { + + // AES瑙ｅ瘑 + decryptData, DecryptDataErr := l.DecryptData(req.Data) + if DecryptDataErr != nil { + return nil, DecryptDataErr + } + + // 鏍￠獙鍙傛暟 + var data types.RiskAssessmentReq + if unmarshalErr := json.Unmarshal(decryptData, &data); unmarshalErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 瑙ｅ瘑鍚庣殑鏁版嵁鏍煎紡涓嶆纭? %+v", unmarshalErr) + } + + if validatorErr := validator.Validate(data); validatorErr != nil { + return nil, errors.Wrapf(xerr.NewErrCodeMsg(xerr.PARAM_VERIFICATION_ERROR, validatorErr.Error()), "鏌ヨ鏈嶅姟, 鍙傛暟涓嶆纭? %+v", validatorErr) + } + + // 鏍￠獙楠岃瘉鐮? + verifyCodeErr := l.VerifyCode(data.Mobile, data.Code) + if verifyCodeErr != nil { + return nil, verifyCodeErr + } + + // 鏍￠獙涓夎绱? + verifyErr := l.Verify(data.Name, data.IDCard, data.Mobile) + if verifyErr != nil { + return nil, verifyErr + } + + // 缂撳瓨 + params := map[string]interface{}{ + "name": data.Name, + "id_card": data.IDCard, + "mobile": data.Mobile, + } + userID, err := l.GetOrCreateUser() + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 澶勭悊鐢ㄦ埛澶辫触: %v", err) + } + cacheNo, cacheDataErr := l.CacheData(params, "riskassessment", userID) + if cacheDataErr != nil { + return nil, cacheDataErr + } + + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 鐢熸垚token澶辫触 : %d", userID) + } + + // 鑾峰彇褰撳墠鏃堕棿鎴? + now := time.Now().Unix() + return &types.QueryServiceResp{ + Id: cacheNo, + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} + +// 澶勭悊鍏徃淇℃伅鏌ヨ鐩稿叧閫昏緫 +func (l *QueryServiceLogic) ProcessCompanyInfoLogic(req *types.QueryServiceReq) (*types.QueryServiceResp, error) { + // AES瑙ｅ瘑 + decryptData, DecryptDataErr := l.DecryptData(req.Data) + if DecryptDataErr != nil { + return nil, DecryptDataErr + } + + // 鏍￠獙鍙傛暟 + var data types.CompanyInfoReq + if unmarshalErr := json.Unmarshal(decryptData, &data); unmarshalErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 瑙ｅ瘑鍚庣殑鏁版嵁鏍煎紡涓嶆纭? %+v", unmarshalErr) + } + + if validatorErr := validator.Validate(data); validatorErr != nil { + return nil, errors.Wrapf(xerr.NewErrCodeMsg(xerr.PARAM_VERIFICATION_ERROR, validatorErr.Error()), "鏌ヨ鏈嶅姟, 鍙傛暟涓嶆纭? %+v", validatorErr) + } + + // 鏍￠獙楠岃瘉鐮? + verifyCodeErr := l.VerifyCode(data.Mobile, data.Code) + if verifyCodeErr != nil { + return nil, verifyCodeErr + } + + // 鏍￠獙涓夎绱? + verifyErr := l.Verify(data.Name, data.IDCard, data.Mobile) + if verifyErr != nil { + return nil, verifyErr + } + + // 缂撳瓨 + params := map[string]interface{}{ + "name": data.Name, + "id_card": data.IDCard, + "mobile": data.Mobile, + } + userID, err := l.GetOrCreateUser() + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 澶勭悊鐢ㄦ埛澶辫触: %v", err) + } + cacheNo, cacheDataErr := l.CacheData(params, "companyinfo", userID) + if cacheDataErr != nil { + return nil, cacheDataErr + } + + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 鐢熸垚token澶辫触 : %d", userID) + } + + // 鑾峰彇褰撳墠鏃堕棿鎴? + now := time.Now().Unix() + return &types.QueryServiceResp{ + Id: cacheNo, + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} + +// 澶勭悊绉熻祦淇℃伅鏌ヨ鐩稿叧閫昏緫 +func (l *QueryServiceLogic) ProcessRentalInfoLogic(req *types.QueryServiceReq) (*types.QueryServiceResp, error) { + + // AES瑙ｅ瘑 + decryptData, DecryptDataErr := l.DecryptData(req.Data) + if DecryptDataErr != nil { + return nil, DecryptDataErr + } + + // 鏍￠獙鍙傛暟 + var data types.RentalInfoReq + if unmarshalErr := json.Unmarshal(decryptData, &data); unmarshalErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 瑙ｅ瘑鍚庣殑鏁版嵁鏍煎紡涓嶆纭? %+v", unmarshalErr) + } + + if validatorErr := validator.Validate(data); validatorErr != nil { + return nil, errors.Wrapf(xerr.NewErrCodeMsg(xerr.PARAM_VERIFICATION_ERROR, validatorErr.Error()), "鏌ヨ鏈嶅姟, 鍙傛暟涓嶆纭? %+v", validatorErr) + } + + // 鏍￠獙楠岃瘉鐮? + verifyCodeErr := l.VerifyCode(data.Mobile, data.Code) + if verifyCodeErr != nil { + return nil, verifyCodeErr + } + + // 鏍￠獙涓夎绱? + verifyErr := l.Verify(data.Name, data.IDCard, data.Mobile) + if verifyErr != nil { + return nil, verifyErr + } + + // 缂撳瓨 + params := map[string]interface{}{ + "name": data.Name, + "id_card": data.IDCard, + "mobile": data.Mobile, + } + userID, err := l.GetOrCreateUser() + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 澶勭悊鐢ㄦ埛澶辫触: %v", err) + } + cacheNo, cacheDataErr := l.CacheData(params, "rentalinfo", userID) + if cacheDataErr != nil { + return nil, cacheDataErr + } + + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 鐢熸垚token澶辫触 : %d", userID) + } + + // 鑾峰彇褰撳墠鏃堕棿鎴? + now := time.Now().Unix() + return &types.QueryServiceResp{ + Id: cacheNo, + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} + +// 澶勭悊璐峰墠鑳屾櫙妫€鏌ョ浉鍏抽€昏緫 +func (l *QueryServiceLogic) ProcessPreLoanBackgroundCheckLogic(req *types.QueryServiceReq) (*types.QueryServiceResp, error) { + + // AES瑙ｅ瘑 + decryptData, DecryptDataErr := l.DecryptData(req.Data) + if DecryptDataErr != nil { + return nil, DecryptDataErr + } + + // 鏍￠獙鍙傛暟 + var data types.PreLoanBackgroundCheckReq + if unmarshalErr := json.Unmarshal(decryptData, &data); unmarshalErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 瑙ｅ瘑鍚庣殑鏁版嵁鏍煎紡涓嶆纭? %+v", unmarshalErr) + } + + if validatorErr := validator.Validate(data); validatorErr != nil { + return nil, errors.Wrapf(xerr.NewErrCodeMsg(xerr.PARAM_VERIFICATION_ERROR, validatorErr.Error()), "鏌ヨ鏈嶅姟, 鍙傛暟涓嶆纭? %+v", validatorErr) + } + + // 鏍￠獙楠岃瘉鐮? + verifyCodeErr := l.VerifyCode(data.Mobile, data.Code) + if verifyCodeErr != nil { + return nil, verifyCodeErr + } + + // 鏍￠獙涓夎绱? + verifyErr := l.Verify(data.Name, data.IDCard, data.Mobile) + if verifyErr != nil { + return nil, verifyErr + } + + // 缂撳瓨 + params := map[string]interface{}{ + "name": data.Name, + "id_card": data.IDCard, + "mobile": data.Mobile, + } + userID, err := l.GetOrCreateUser() + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 澶勭悊鐢ㄦ埛澶辫触: %v", err) + } + cacheNo, cacheDataErr := l.CacheData(params, "preloanbackgroundcheck", userID) + if cacheDataErr != nil { + return nil, cacheDataErr + } + + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 鐢熸垚token澶辫触 : %d", userID) + } + + // 鑾峰彇褰撳墠鏃堕棿鎴? + now := time.Now().Unix() + return &types.QueryServiceResp{ + Id: cacheNo, + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} + +// 澶勭悊浜轰簨鑳岃皟鐩稿叧閫昏緫 +func (l *QueryServiceLogic) ProcessBackgroundCheckLogic(req *types.QueryServiceReq) (*types.QueryServiceResp, error) { + // AES瑙ｅ瘑 + decryptData, DecryptDataErr := l.DecryptData(req.Data) + if DecryptDataErr != nil { + return nil, DecryptDataErr + } + + // 鏍￠獙鍙傛暟 + var data types.BackgroundCheckReq + if unmarshalErr := json.Unmarshal(decryptData, &data); unmarshalErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 瑙ｅ瘑鍚庣殑鏁版嵁鏍煎紡涓嶆纭? %+v", unmarshalErr) + } + + if validatorErr := validator.Validate(data); validatorErr != nil { + return nil, errors.Wrapf(xerr.NewErrCodeMsg(xerr.PARAM_VERIFICATION_ERROR, validatorErr.Error()), "鏌ヨ鏈嶅姟, 鍙傛暟涓嶆纭? %+v", validatorErr) + } + + // 鏍￠獙楠岃瘉鐮? + verifyCodeErr := l.VerifyCode(data.Mobile, data.Code) + if verifyCodeErr != nil { + return nil, verifyCodeErr + } + + // 鏍￠獙涓夎绱? + verifyErr := l.Verify(data.Name, data.IDCard, data.Mobile) + if verifyErr != nil { + return nil, verifyErr + } + + // 缂撳瓨 + params := map[string]interface{}{ + "name": data.Name, + "id_card": data.IDCard, + "mobile": data.Mobile, + } + userID, err := l.GetOrCreateUser() + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 澶勭悊鐢ㄦ埛澶辫触: %v", err) + } + cacheNo, cacheDataErr := l.CacheData(params, "backgroundcheck", userID) + if cacheDataErr != nil { + return nil, cacheDataErr + } + + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 鐢熸垚token澶辫触 : %d", userID) + } + + // 鑾峰彇褰撳墠鏃堕棿鎴? + now := time.Now().Unix() + return &types.QueryServiceResp{ + Id: cacheNo, + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} +func (l *QueryServiceLogic) ProcessPersonalDataLogic(req *types.QueryServiceReq) (*types.QueryServiceResp, error) { + // AES瑙ｅ瘑 + decryptData, DecryptDataErr := l.DecryptData(req.Data) + if DecryptDataErr != nil { + return nil, DecryptDataErr + } + + // 鏍￠獙鍙傛暟 + var data types.PersonalDataReq + if unmarshalErr := json.Unmarshal(decryptData, &data); unmarshalErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 瑙ｅ瘑鍚庣殑鏁版嵁鏍煎紡涓嶆纭? %+v", unmarshalErr) + } + + if validatorErr := validator.Validate(data); validatorErr != nil { + return nil, errors.Wrapf(xerr.NewErrCodeMsg(xerr.PARAM_VERIFICATION_ERROR, validatorErr.Error()), "鏌ヨ鏈嶅姟, 鍙傛暟涓嶆纭? %+v", validatorErr) + } + + // 鏍￠獙楠岃瘉鐮? + verifyCodeErr := l.VerifyCode(data.Mobile, data.Code) + if verifyCodeErr != nil { + return nil, verifyCodeErr + } + + // 鏍￠獙涓夎绱? + verifyErr := l.Verify(data.Name, data.IDCard, data.Mobile) + if verifyErr != nil { + return nil, verifyErr + } + + // 缂撳瓨 + params := map[string]interface{}{ + "name": data.Name, + "id_card": data.IDCard, + "mobile": data.Mobile, + } + userID, err := l.GetOrCreateUser() + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 澶勭悊鐢ㄦ埛澶辫触: %v", err) + } + cacheNo, cacheDataErr := l.CacheData(params, "personalData", userID) + if cacheDataErr != nil { + return nil, cacheDataErr + } + + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 鐢熸垚token澶辫触 : %d", userID) + } + + // 鑾峰彇褰撳墠鏃堕棿鎴? + now := time.Now().Unix() + return &types.QueryServiceResp{ + Id: cacheNo, + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} +func (l *QueryServiceLogic) ProcessConsumerFinanceReportLogic(req *types.QueryServiceReq) (*types.QueryServiceResp, error) { + // AES瑙ｅ瘑 + decryptData, DecryptDataErr := l.DecryptData(req.Data) + if DecryptDataErr != nil { + return nil, DecryptDataErr + } + + // 鏍￠獙鍙傛暟 + var data types.ConsumerFinanceReportReq + if unmarshalErr := json.Unmarshal(decryptData, &data); unmarshalErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 瑙ｅ瘑鍚庣殑鏁版嵁鏍煎紡涓嶆纭? %+v", unmarshalErr) + } + + if validatorErr := validator.Validate(data); validatorErr != nil { + return nil, errors.Wrapf(xerr.NewErrCodeMsg(xerr.PARAM_VERIFICATION_ERROR, validatorErr.Error()), "鏌ヨ鏈嶅姟, 鍙傛暟涓嶆纭? %+v", validatorErr) + } + + // 鏍￠獙楠岃瘉鐮? + verifyCodeErr := l.VerifyCode(data.Mobile, data.Code) + if verifyCodeErr != nil { + return nil, verifyCodeErr + } + + // 鏍￠獙涓夎绱? + verifyErr := l.Verify(data.Name, data.IDCard, data.Mobile) + if verifyErr != nil { + return nil, verifyErr + } + + // 缂撳瓨 + params := map[string]interface{}{ + "name": data.Name, + "id_card": data.IDCard, + "mobile": data.Mobile, + } + userID, err := l.GetOrCreateUser() + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 澶勭悊鐢ㄦ埛澶辫触: %v", err) + } + cacheNo, cacheDataErr := l.CacheData(params, "consumerFinanceReport", userID) + if cacheDataErr != nil { + return nil, cacheDataErr + } + + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 鐢熸垚token澶辫触 : %d", userID) + } + + // 鑾峰彇褰撳墠鏃堕棿鎴? + now := time.Now().Unix() + return &types.QueryServiceResp{ + Id: cacheNo, + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} +func (l *QueryServiceLogic) DecryptData(data string) ([]byte, error) { + secretKey := l.svcCtx.Config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "瀵嗛挜鑾峰彇澶辫触: %+v", decodeErr) + } + decryptData, aesDecryptErr := crypto.AesDecrypt(data, key) + if aesDecryptErr != nil || len(decryptData) == 0 { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "瑙ｅ瘑澶辫触: %+v", aesDecryptErr) + } + return decryptData, nil +} + +// 鏍￠獙楠岃瘉鐮? +func (l *QueryServiceLogic) VerifyCode(mobile string, code string) error { + // 寮€鍙戠幆澧冧笅璺宠繃楠岃瘉鐮佹牎楠? + if os.Getenv("ENV") == "development" { + return nil + } + secretKey := l.svcCtx.Config.Encrypt.SecretKey + encryptedMobile, err := crypto.EncryptMobile(mobile, secretKey) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鍔犲瘑鎵嬫満鍙峰け璐? %+v", err) + } + codeRedisKey := fmt.Sprintf("%s:%s", "query", encryptedMobile) + cacheCode, err := l.svcCtx.Redis.Get(codeRedisKey) + if err != nil { + if errors.Is(err, redis.Nil) { + return errors.Wrapf(xerr.NewErrMsg("楠岃瘉鐮佸凡杩囨湡"), "楠岃瘉鐮佽繃鏈? %s", mobile) + } + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "璇诲彇楠岃瘉鐮乺edis缂撳瓨澶辫触, mobile: %s, err: %+v", mobile, err) + } + if cacheCode != code { + return errors.Wrapf(xerr.NewErrMsg("楠岃瘉鐮佷笉姝ｇ‘"), "楠岃瘉鐮佷笉姝ｇ‘: %s", mobile) + } + return nil +} + +// 浜屻€佷笁瑕佺礌楠岃瘉 +func (l *QueryServiceLogic) Verify(Name string, IDCard string, Mobile string) error { + // 寮€鍙戠幆澧冧笅璺宠繃浜?涓夎绱犻獙璇? + if os.Getenv("ENV") == "development" { + return nil + } + if !l.svcCtx.Config.SystemConfig.ThreeVerify { + twoVerification := service.TwoFactorVerificationRequest{ + Name: Name, + IDCard: IDCard, + } + verification, err := l.svcCtx.VerificationService.TwoFactorVerification(twoVerification) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "浜岃绱犻獙璇佸け璐? %v", err) + } + if !verification.Passed { + return errors.Wrapf(xerr.NewErrCodeMsg(xerr.SERVER_COMMON_ERROR, verification.Err.Error()), "浜岃绱犻獙璇佷笉閫氳繃: %v", err) + } + } else { + // 涓夎绱犻獙璇? + threeVerification := service.ThreeFactorVerificationRequest{ + Name: Name, + IDCard: IDCard, + Mobile: Mobile, + } + verification, err := l.svcCtx.VerificationService.ThreeFactorVerification(threeVerification) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "涓夎绱犻獙璇佸け璐? %v", err) + } + if !verification.Passed { + return errors.Wrapf(xerr.NewErrCodeMsg(xerr.SERVER_COMMON_ERROR, verification.Err.Error()), "涓夎绱犻獙璇佷笉閫氳繃: %v", err) + } + } + return nil +} + +// 缂撳瓨 +func (l *QueryServiceLogic) CacheData(params map[string]interface{}, Product string, userID string) (string, error) { + agentIdentifier, _ := l.ctx.Value("agentIdentifier").(string) + secretKey := l.svcCtx.Config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 鑾峰彇AES瀵嗛挜澶辫触: %+v", decodeErr) + } + paramsMarshal, marshalErr := json.Marshal(params) + if marshalErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 搴忓垪鍖栧弬鏁板け璐? %+v", marshalErr) + } + encryptParams, aesEncryptErr := crypto.AesEncrypt(paramsMarshal, key) + if aesEncryptErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 鍔犲瘑鍙傛暟澶辫触: %+v", aesEncryptErr) + } + queryCache := types.QueryCacheLoad{ + Params: encryptParams, + Product: Product, + AgentIdentifier: agentIdentifier, + } + jsonData, marshalErr := json.Marshal(queryCache) + if marshalErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "鏌ヨ鏈嶅姟, 搴忓垪鍖栧弬鏁板け璐? %+v", marshalErr) + } + outTradeNo := "Q_" + l.svcCtx.AlipayService.GenerateOutTradeNo() + redisKey := fmt.Sprintf(types.QueryCacheKey, userID, outTradeNo) + cacheErr := l.svcCtx.Redis.SetexCtx(l.ctx, redisKey, string(jsonData), int(2*time.Hour)) + if cacheErr != nil { + return "", cacheErr + } + return outTradeNo, nil +} + +// GetOrCreateUser 获取或创建用户 +// 1. 如果已登录，使用当前登录用户 +// 2. 如果未登录，创建临时用户（UUID用户） +// 注意：查询服务不负责手机号绑定，手机号绑定由用户在其他地方自主选择 +func (l *QueryServiceLogic) GetOrCreateUser() (string, error) { + // 获取当前登录态 + claims, err := ctxdata.GetClaimsFromCtx(l.ctx) + logx.Infof("claims: %+v", claims) + if err != nil && !errors.Is(err, ctxdata.ErrNoInCtx) { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败: %v", err) + } + + // 如果已登录，使用当前登录用户 + if claims != nil { + return claims.UserId, nil + } + + // 未登录：创建临时用户（UUID用户） + userID, err := l.svcCtx.UserService.RegisterUUIDUser(l.ctx) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "创建临时用户失败: %v", err) + } + + return userID, nil +} diff --git a/app/main/api/internal/logic/query/querysharedetaillogic.go b/app/main/api/internal/logic/query/querysharedetaillogic.go new file mode 100644 index 0000000..0f38400 --- /dev/null +++ b/app/main/api/internal/logic/query/querysharedetaillogic.go @@ -0,0 +1,195 @@ +package query + +import ( + "context" + "encoding/hex" + "fmt" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/bytedance/sonic" + "github.com/jinzhu/copier" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type QueryShareDetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQueryShareDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QueryShareDetailLogic { + return &QueryShareDetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QueryShareDetailLogic) QueryShareDetail(req *types.QueryShareDetailReq) (resp string, err error) { + secretKey := l.svcCtx.Config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 获取AES解密解药失败, %v", decodeErr) + } + decryptedID, decryptErr := crypto.AesDecryptURL(req.Id, key) + if decryptErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 解密数据失败: %v", decryptErr) + } + + var payload types.QueryShareLinkPayload + err = sonic.Unmarshal(decryptedID, &payload) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 解密数据失败: %v", err) + } + + type shareDetail struct { + Status string `json:"status"` + Query *types.Query `json:"query,omitempty"` + } + + // 检查分享链接是否过期 + now := time.Now().Unix() + if now > payload.ExpireAt { + expiredResp := shareDetail{ + Status: "expired", + } + encryptedExpired, encryptErr := l.encryptShareDetail(expiredResp, key) + if encryptErr != nil { + return "", encryptErr + } + return encryptedExpired, nil + } + + // 获取订单信息 + order, err := l.svcCtx.OrderModel.FindOne(l.ctx, payload.OrderId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return "", errors.Wrapf(xerr.NewErrCode(xerr.LOGIC_QUERY_NOT_FOUND), "报告查询, 订单不存在: %v", err) + } + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告查询, 查找报告错误: %v", err) + } + + // 检查订单状态 + if order.Status != "paid" { + return "", errors.Wrapf(xerr.NewErrMsg("订单未支付，无法查看报告"), "") + } + + // 获取报告信息 + queryModel, err := l.svcCtx.QueryModel.FindOneByOrderId(l.ctx, order.Id) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "报告查询, 查找报告错误: %v", err) + } + + var query types.Query + query.CreateTime = queryModel.CreateTime.Format("2006-01-02 15:04:05") + query.UpdateTime = queryModel.UpdateTime.Format("2006-01-02 15:04:05") + + processParamsErr := ProcessQueryParams(queryModel.QueryParams, &query.QueryParams, key) + if processParamsErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告参数处理失败: %v", processParamsErr) + } + processErr := ProcessQueryData(queryModel.QueryData, &query.QueryData, key) + if processErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结果处理失败: %v", processErr) + } + updateFeatureAndProductFeatureErr := l.UpdateFeatureAndProductFeature(queryModel.ProductId, &query.QueryData) + if updateFeatureAndProductFeatureErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结果处理失败: %v", updateFeatureAndProductFeatureErr) + } + // 复制报告数据 + err = copier.Copy(&query, queryModel) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 报告结构体复制失败, %v", err) + } + product, err := l.svcCtx.ProductModel.FindOne(l.ctx, queryModel.ProductId) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 获取商品信息失败, %v", err) + } + query.ProductName = product.ProductName + query.Product = product.ProductEn + successResp := shareDetail{ + Status: "success", + Query: &query, + } + encryptedSuccess, encryptErr := l.encryptShareDetail(successResp, key) + if encryptErr != nil { + return "", encryptErr + } + + return encryptedSuccess, nil +} + +func (l *QueryShareDetailLogic) encryptShareDetail(detail interface{}, key []byte) (string, error) { + payloadBytes, marshalErr := sonic.Marshal(detail) + if marshalErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 序列化查询结果失败: %v", marshalErr) + } + + encrypted, encryptErr := crypto.AesEncrypt(payloadBytes, key) + if encryptErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "报告查询, 加密查询结果失败: %v", encryptErr) + } + + return encrypted, nil +} + +func (l *QueryShareDetailLogic) UpdateFeatureAndProductFeature(productID string, target *[]types.QueryItem) error { + // 遍历 target 数组，使用倒序遍历，以便删除元素时不影响索引 + for i := len(*target) - 1; i >= 0; i-- { + queryItem := &(*target)[i] + + // 确保 Data 为 map 类型 + data, ok := queryItem.Data.(map[string]interface{}) + if !ok { + return fmt.Errorf("queryItem.Data 必须是 map[string]interface{} 类型") + } + + // 从 Data 中获取 apiID + apiID, ok := data["apiID"].(string) + if !ok { + return fmt.Errorf("queryItem.Data 中的 apiID 必须是字符串类型") + } + + // 查询 Feature + feature, err := l.svcCtx.FeatureModel.FindOneByApiId(l.ctx, apiID) + if err != nil { + // 如果 Feature 查不到，也要删除当前 QueryItem + *target = append((*target)[:i], (*target)[i+1:]...) + continue + } + + // 查询 ProductFeatureModel + builder := l.svcCtx.ProductFeatureModel.SelectBuilder().Where("product_id = ?", productID) + productFeatures, err := l.svcCtx.ProductFeatureModel.FindAll(l.ctx, builder, "") + if err != nil { + return fmt.Errorf("查询 ProductFeatureModel 错误: %v", err) + } + + // 遍历 productFeatures，找到与 feature.ID 关联且 enable == 1 的项 + var featureData map[string]interface{} + // foundFeature := false + sort := 0 + for _, pf := range productFeatures { + if pf.FeatureId == feature.Id { // 确保和 Feature 关联 + sort = int(pf.Sort) + break // 找到第一个符合条件的就退出循环 + } + } + featureData = map[string]interface{}{ + "featureName": feature.Name, + "sort": sort, + } + + // 更新 queryItem 的 Feature 字段（不是数组） + queryItem.Feature = featureData + } + + return nil +} diff --git a/app/main/api/internal/logic/query/querysingletestlogic.go b/app/main/api/internal/logic/query/querysingletestlogic.go new file mode 100644 index 0000000..53d8568 --- /dev/null +++ b/app/main/api/internal/logic/query/querysingletestlogic.go @@ -0,0 +1,52 @@ +package query + +import ( + "context" + "encoding/json" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type QuerySingleTestLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewQuerySingleTestLogic(ctx context.Context, svcCtx *svc.ServiceContext) *QuerySingleTestLogic { + return &QuerySingleTestLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *QuerySingleTestLogic) QuerySingleTest(req *types.QuerySingleTestReq) (resp *types.QuerySingleTestResp, err error) { + //featrueModel, err := l.svcCtx.FeatureModel.FindOneByApiId(l.ctx, req.Api) + //if err != nil { + // return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "单查测试, 获取接口失败 : %d", err) + //} + marshalParams, err := json.Marshal(req.Params) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "单查测试, 序列化参数失败 : %d", err) + } + apiResp, err := l.svcCtx.ApiRequestService.PreprocessRequestApi(marshalParams, req.Api) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "单查测试, 获取接口失败 : %d", err) + } + var respData interface{} + err = json.Unmarshal(apiResp, &respData) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "单查测试, 反序列化接口失败 : %d", err) + } + return &types.QuerySingleTestResp{ + Data: respData, + Api: req.Api, + }, nil +} diff --git a/app/main/api/internal/logic/query/updatequerydatalogic.go b/app/main/api/internal/logic/query/updatequerydatalogic.go new file mode 100644 index 0000000..eb00acd --- /dev/null +++ b/app/main/api/internal/logic/query/updatequerydatalogic.go @@ -0,0 +1,71 @@ +package query + +import ( + "context" + "database/sql" + "encoding/hex" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type UpdateQueryDataLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +// 更新查询数据 +func NewUpdateQueryDataLogic(ctx context.Context, svcCtx *svc.ServiceContext) *UpdateQueryDataLogic { + return &UpdateQueryDataLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *UpdateQueryDataLogic) UpdateQueryData(req *types.UpdateQueryDataReq) (resp *types.UpdateQueryDataResp, err error) { + // 1. 从数据库中获取查询记录 + query, err := l.svcCtx.QueryModel.FindOne(l.ctx, req.Id) + if err != nil { + if err == model.ErrNotFound { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询记录不存在, 查询ID: %d", req.Id) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询数据库失败, 查询ID: %d, err: %v", req.Id, err) + } + + // 2. 获取加密密钥 + secretKey := l.svcCtx.Config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取AES密钥失败: %v", decodeErr) + } + + // 3. 加密数据 - 传入的是JSON，需要加密处理 + encryptData, aesEncryptErr := crypto.AesEncrypt([]byte(req.QueryData), key) + if aesEncryptErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "加密查询数据失败: %v", aesEncryptErr) + } + + // 4. 更新数据库记录 + query.QueryData = sql.NullString{ + String: encryptData, + Valid: true, + } + updateErr := l.svcCtx.QueryModel.UpdateWithVersion(l.ctx, nil, query) + if updateErr != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新查询数据失败: %v", updateErr) + } + + // 5. 返回结果 + return &types.UpdateQueryDataResp{ + Id: query.Id, + UpdatedAt: query.UpdateTime.Format("2006-01-02 15:04:05"), + }, nil +} diff --git a/app/main/api/internal/logic/user/authlogic.go b/app/main/api/internal/logic/user/authlogic.go new file mode 100644 index 0000000..437158c --- /dev/null +++ b/app/main/api/internal/logic/user/authlogic.go @@ -0,0 +1,144 @@ +package user + +import ( + "context" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/google/uuid" + "github.com/pkg/errors" +) + +type AuthLogic struct { + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewAuthLogic(ctx context.Context, svcCtx *svc.ServiceContext) *AuthLogic { + return &AuthLogic{ctx: ctx, svcCtx: svcCtx} +} + +func (l *AuthLogic) Auth(req *types.AuthReq) (*types.AuthResp, error) { + var userID string + var userType int64 + var authType string + var authKey string + + switch req.Platform { + case model.PlatformH5: + authType = model.UserAuthTypeUUID + authKey = uuid.NewString() + user, err := l.findOrCreateUserByAuth(authType, authKey) + if err != nil { + return nil, err + } + userID = user.Id + userType = l.getUserType(user) + case model.PlatformWxH5: + openid, err := l.svcCtx.VerificationService.GetWechatH5OpenID(l.ctx, req.Code) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取WxH5 OpenID失败: %v", err) + } + authType = model.UserAuthTypeWxh5OpenID + authKey = openid + userAuth, err := l.svcCtx.UserAuthModel.FindOneByAuthTypeAuthKey(l.ctx, authType, authKey) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户授权失败: %v", err) + } + if userAuth != nil { + user, _ := l.svcCtx.UserModel.FindOne(l.ctx, userAuth.UserId) + userID = user.Id + userType = l.getUserType(user) + } else { + user, err := l.createUserWithAuth(authType, authKey) + if err != nil { + return nil, err + } + userID = user.Id + userType = model.UserTypeTemp + } + case model.PlatformWxMini: + openid, err := l.svcCtx.VerificationService.GetWechatMiniOpenID(l.ctx, req.Code) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取WxMini OpenID失败: %v", err) + } + authType = model.UserAuthTypeWxMiniOpenID + authKey = openid + userAuth, err := l.svcCtx.UserAuthModel.FindOneByAuthTypeAuthKey(l.ctx, authType, authKey) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户授权失败: %v", err) + } + if userAuth != nil { + user, _ := l.svcCtx.UserModel.FindOne(l.ctx, userAuth.UserId) + userID = user.Id + userType = l.getUserType(user) + } else { + user, err := l.createUserWithAuth(authType, authKey) + if err != nil { + return nil, err + } + userID = user.Id + userType = model.UserTypeTemp + } + default: + return nil, errors.Wrapf(xerr.NewErrMsg("不支持的平台类型"), "platform=%s", req.Platform) + } + + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成Token失败: %v", err) + } + now := time.Now().Unix() + user, _ := l.svcCtx.UserModel.FindOne(l.ctx, userID) + hasMobile := user.Mobile.Valid + isAgent := false + if hasMobile { + agent, _ := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + isAgent = agent != nil + } + return &types.AuthResp{ + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + UserType: userType, + HasMobile: hasMobile, + IsAgent: isAgent, + }, nil +} + +func (l *AuthLogic) findOrCreateUserByAuth(authType, authKey string) (*model.User, error) { + userAuth, err := l.svcCtx.UserAuthModel.FindOneByAuthTypeAuthKey(l.ctx, authType, authKey) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找用户授权失败: %v", err) + } + if userAuth != nil { + user, err := l.svcCtx.UserModel.FindOne(l.ctx, userAuth.UserId) + return user, err + } + return l.createUserWithAuth(authType, authKey) +} + +func (l *AuthLogic) createUserWithAuth(authType, authKey string) (*model.User, error) { + user := &model.User{Id: uuid.NewString()} + _, err := l.svcCtx.UserModel.Insert(l.ctx, nil, user) + if err != nil { + return nil, err + } + ua := &model.UserAuth{Id: uuid.NewString(), UserId: user.Id, AuthType: authType, AuthKey: authKey} + _, err = l.svcCtx.UserAuthModel.Insert(l.ctx, nil, ua) + if err != nil { + return nil, err + } + return l.svcCtx.UserModel.FindOne(l.ctx, user.Id) +} + +func (l *AuthLogic) getUserType(user *model.User) int64 { + if user.Mobile.Valid { + return model.UserTypeNormal + } + return model.UserTypeTemp +} diff --git a/app/main/api/internal/logic/user/bindmobilelogic.go b/app/main/api/internal/logic/user/bindmobilelogic.go new file mode 100644 index 0000000..dc3b42d --- /dev/null +++ b/app/main/api/internal/logic/user/bindmobilelogic.go @@ -0,0 +1,242 @@ +package user + +import ( + "context" + "database/sql" + "fmt" + "os" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/redis" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type BindMobileLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewBindMobileLogic(ctx context.Context, svcCtx *svc.ServiceContext) *BindMobileLogic { + return &BindMobileLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *BindMobileLogic) BindMobile(req *types.BindMobileReq) (resp *types.BindMobileResp, err error) { + // 从上下文中获取当前登录态的用户声明（可能是临时用户或正式用户），包含UserId/AuthType/AuthKey + claims, err := ctxdata.GetClaimsFromCtx(l.ctx) + if err != nil && !errors.Is(err, ctxdata.ErrNoInCtx) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取用户信息失败: %v", err) + } + + // 当前登录用户信息（用于后续合并/绑定） + currentUserID := claims.UserId + currentAuthType := claims.AuthType + currentAuthKey := claims.AuthKey + + // 加密手机号（所有手机号以密文存储） + secretKey := l.svcCtx.Config.Encrypt.SecretKey + encryptedMobile, err := crypto.EncryptMobile(req.Mobile, secretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "加密手机号失败: %v", err) + } + // 非开发环境下校验短信验证码（从Redis读取并比对） + if os.Getenv("ENV") != "development" { + redisKey := fmt.Sprintf("%s:%s", "bindMobile", encryptedMobile) + cacheCode, err := l.svcCtx.Redis.Get(redisKey) + if err != nil { + if errors.Is(err, redis.Nil) { + return nil, errors.Wrapf(xerr.NewErrMsg("验证码已过期"), "") + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "读取验证码失败: %v", err) + } + if cacheCode != req.Code { + return nil, errors.Wrapf(xerr.NewErrMsg("验证码不正确"), "") + } + } + + // 通过加密后的手机号查找目标用户（手机号用户视为正式用户） + targetUser, err := l.svcCtx.UserModel.FindOneByMobile(l.ctx, sql.NullString{String: encryptedMobile, Valid: true}) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找用户失败: %v", err) + } + + var finalUserID string + if targetUser == nil { + // 手机号不存在：直接将当前用户升级为正式用户（写入mobile与mobile认证） + finalUserID = currentUserID + currentUser, err := l.svcCtx.UserModel.FindOne(l.ctx, currentUserID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找当前用户失败: %v", err) + } + currentUser.Mobile = sql.NullString{String: encryptedMobile, Valid: true} + if _, err := l.svcCtx.UserModel.Update(l.ctx, nil, currentUser); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新手机号失败: %v", err) + } + // 记录mobile认证（确保后续可通过手机号登录） + if _, err := l.svcCtx.UserAuthModel.Insert(l.ctx, nil, &model.UserAuth{Id: uuid.NewString(), UserId: finalUserID, AuthType: model.UserAuthTypeMobile, AuthKey: encryptedMobile}); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建手机号认证失败: %v", err) + } + // 发放token（userType会根据mobile字段动态计算） + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, finalUserID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成Token失败: %v", err) + } + now := time.Now().Unix() + return &types.BindMobileResp{AccessToken: token, AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter}, nil + } + + // 手机号已存在：进入账号合并或快捷登录流程 + finalUserID = targetUser.Id + // 保护校验：若将不同用户进行合并，确保源用户不存在代理记录（临时用户不应为代理） + if currentUserID != finalUserID { + agent, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, currentUserID) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理信息失败: %v", err) + } + if agent != nil { + return nil, errors.Wrapf(xerr.NewErrMsg("账号数据异常：源用户存在代理记录，请联系技术支持"), "") + } + } + // 查找当前登录态使用的认证（例如uuid或微信openid）是否已存在 + existingAuth, err := l.svcCtx.UserAuthModel.FindOneByAuthTypeAuthKey(l.ctx, currentAuthType, currentAuthKey) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找认证信息失败: %v", err) + } + // 如果当前认证已属于目标手机号用户，直接发放token（无需合并） + if existingAuth != nil && existingAuth.UserId == finalUserID { + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, finalUserID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成Token失败: %v", err) + } + now := time.Now().Unix() + return &types.BindMobileResp{AccessToken: token, AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter}, nil + } + + // 微信唯一性约束（按类型）： + // - H5 与 小程序各自只能绑定一个 openid（互不影响） + if currentAuthType == model.UserAuthTypeWxh5OpenID { + wxh5Auth, err := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(l.ctx, finalUserID, model.UserAuthTypeWxh5OpenID) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找认证信息失败: %v", err) + } + if wxh5Auth != nil && wxh5Auth.AuthKey != currentAuthKey { + return nil, errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他H5微信号"), "") + } + } + if currentAuthType == model.UserAuthTypeWxMiniOpenID { + wxminiAuth, err := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(l.ctx, finalUserID, model.UserAuthTypeWxMiniOpenID) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找认证信息失败: %v", err) + } + if wxminiAuth != nil && wxminiAuth.AuthKey != currentAuthKey { + return nil, errors.Wrapf(xerr.NewErrMsg("该手机号已绑定其他小程序微信号"), "") + } + } + + // 事务处理： + // - 将当前登录态的认证（uuid / 微信openid 等）绑定到目标手机号用户（finalUserID） + // - 将源用户（currentUserID）的业务数据（订单、报告）迁移到目标用户，避免数据分裂 + // - 对源用户执行软删除，清理无主临时账号，保持数据一致性 + // 注意：所有步骤必须在同一个事务中执行，任何一步失败均会回滚，确保原子性 + err = l.svcCtx.UserModel.Trans(l.ctx, func(ctx context.Context, session sqlx.Session) error { + // 1) 认证绑定处理（UUID替换策略） + if currentAuthType == model.UserAuthTypeUUID { + targetUUIDAuth, _ := l.svcCtx.UserAuthModel.FindOneByUserIdAuthType(ctx, finalUserID, model.UserAuthTypeUUID) + if existingAuth != nil && existingAuth.UserId != finalUserID { + if targetUUIDAuth != nil { + if targetUUIDAuth.AuthKey != currentAuthKey { + if err := l.svcCtx.UserAuthModel.Delete(ctx, session, existingAuth.Id); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除旧UUID认证失败: %v", err) + } + targetUUIDAuth.AuthKey = currentAuthKey + if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, targetUUIDAuth); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新目标UUID认证失败: %v", err) + } + } else { + if err := l.svcCtx.UserAuthModel.Delete(ctx, session, existingAuth.Id); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除重复UUID认证失败: %v", err) + } + } + } else { + existingAuth.UserId = finalUserID + if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, existingAuth); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "迁移UUID认证失败: %v", err) + } + } + } else if existingAuth == nil { + if targetUUIDAuth != nil { + if targetUUIDAuth.AuthKey != currentAuthKey { + targetUUIDAuth.AuthKey = currentAuthKey + if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, targetUUIDAuth); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新目标UUID认证失败: %v", err) + } + } + } else { + _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{Id: uuid.NewString(), UserId: finalUserID, AuthType: currentAuthType, AuthKey: currentAuthKey}) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建UUID认证失败: %v", err) + } + } + } + } else { + if existingAuth != nil && existingAuth.UserId != finalUserID { + existingAuth.UserId = finalUserID + if _, err := l.svcCtx.UserAuthModel.Update(ctx, session, existingAuth); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新认证绑定失败: %v", err) + } + } else if existingAuth == nil { + _, err := l.svcCtx.UserAuthModel.Insert(ctx, session, &model.UserAuth{Id: uuid.NewString(), UserId: finalUserID, AuthType: currentAuthType, AuthKey: currentAuthKey}) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建认证绑定失败: %v", err) + } + } + } + + // 2) 业务数据迁移 + // 当源用户与目标用户不同时，迁移源用户的订单与报告归属到finalUserID，避免合并后数据仍挂在旧用户 + if currentUserID != finalUserID { + if err := l.svcCtx.OrderModel.UpdateUserIDWithSession(ctx, session, currentUserID, finalUserID); err != nil { + return err + } + if err := l.svcCtx.QueryModel.UpdateUserIDWithSession(ctx, session, currentUserID, finalUserID); err != nil { + return err + } + + // 3) 源用户软删除 + // 软删源用户（通常为临时用户），防止遗留无效账号；软删可保留历史痕迹，满足审计需求 + currentUser, err := l.svcCtx.UserModel.FindOne(ctx, currentUserID) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查找当前用户失败: %v", err) + } + if err := l.svcCtx.UserModel.Delete(ctx, session, currentUser.Id); err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "删除当前用户失败: %v", err) + } + } + return nil + }) + if err != nil { + return nil, err + } + + // 合并完成后生成token（userType会根据mobile字段动态计算） + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, finalUserID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成Token失败: %v", err) + } + now := time.Now().Unix() + return &types.BindMobileResp{AccessToken: token, AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter}, nil +} diff --git a/app/main/api/internal/logic/user/canceloutlogic.go b/app/main/api/internal/logic/user/canceloutlogic.go new file mode 100644 index 0000000..bb1de29 --- /dev/null +++ b/app/main/api/internal/logic/user/canceloutlogic.go @@ -0,0 +1,221 @@ +package user + +import ( + "context" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + + "github.com/zeromicro/go-zero/core/mr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/sqlx" + + "qnc-server/app/main/api/internal/svc" + + "github.com/zeromicro/go-zero/core/logx" +) + +type CancelOutLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewCancelOutLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CancelOutLogic { + return &CancelOutLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *CancelOutLogic) CancelOut() error { + userID, getUserIdErr := ctxdata.GetUidFromCtx(l.ctx) + if getUserIdErr != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, %v", getUserIdErr) + } + + // 1. 先检查用户是否是代理 + agentModel, err := l.svcCtx.AgentModel.FindOneByUserId(l.ctx, userID) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "查询代理信息失败, userId: %d", userID) + } + + // 如果用户是代理，进行额外检查 + if agentModel != nil { + // 1.1 检查代理等级是否为黄金或钻石（新系统） + levelName := "" + switch agentModel.Level { + case 2: + levelName = "黄金" + case 3: + levelName = "钻石" + } + if agentModel.Level == 2 || agentModel.Level == 3 { + return errors.Wrapf(xerr.NewErrMsg("您是"+levelName+"代理，请联系客服进行注销"), "用户是高级代理，不能注销") + } + + // 1.2 检查代理钱包是否有余额或冻结金额 + wallet, err := l.svcCtx.AgentWalletModel.FindOneByAgentId(l.ctx, agentModel.Id) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询代理钱包失败, agentId: %d", agentModel.Id) + } + + if wallet != nil && (wallet.Balance > 0 || wallet.FrozenBalance > 0) { + if wallet.Balance > 0 { + return errors.Wrapf(xerr.NewErrMsg("您的钱包还有余额%.2f元，请先提现后再注销账号"), "用户钱包有余额，不能注销: %.2f", wallet.Balance) + } + if wallet.FrozenBalance > 0 { + return errors.Wrapf(xerr.NewErrMsg("您的钱包还有冻结金额%.2f元，请等待解冻后再注销账号"), "用户钱包有冻结金额，不能注销: %.2f", wallet.FrozenBalance) + } + } + } + + // 在事务中处理用户注销相关操作 + err = l.svcCtx.UserModel.Trans(l.ctx, func(tranCtx context.Context, session sqlx.Session) error { + // 1. 删除用户基本信息 + if err := l.svcCtx.UserModel.Delete(tranCtx, session, userID); err != nil { + return errors.Wrapf(err, "删除用户基本信息失败, userId: %d", userID) + } + + // 2. 查询并删除用户授权信息 + UserAuthModelBuilder := l.svcCtx.UserAuthModel.SelectBuilder().Where("user_id = ?", userID) + userAuths, err := l.svcCtx.UserAuthModel.FindAll(tranCtx, UserAuthModelBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "查询用户授权信息失败, userId: %d", userID) + } + + // 并发删除用户授权信息 + if len(userAuths) > 0 { + funcs := make([]func() error, len(userAuths)) + for i, userAuth := range userAuths { + authID := userAuth.Id + funcs[i] = func() error { + return l.svcCtx.UserAuthModel.Delete(tranCtx, session, authID) + } + } + + if err := mr.Finish(funcs...); err != nil { + return errors.Wrapf(err, "删除用户授权信息失败") + } + } + + // 3. 处理代理相关信息 + if agentModel != nil { + // 3.1 删除代理信息 + if err := l.svcCtx.AgentModel.Delete(tranCtx, session, agentModel.Id); err != nil { + return errors.Wrapf(err, "删除代理信息失败, agentId: %d", agentModel.Id) + } + + // 3.2 删除代理关系信息（新系统使用AgentRelation） + relationBuilder := l.svcCtx.AgentRelationModel.SelectBuilder(). + Where("parent_id = ? OR child_id = ?", agentModel.Id, agentModel.Id) + relations, err := l.svcCtx.AgentRelationModel.FindAll(tranCtx, relationBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "查询代理关系信息失败, agentId: %d", agentModel.Id) + } + + // 并发删除代理关系信息 + if len(relations) > 0 { + relationFuncs := make([]func() error, len(relations)) + for i, relation := range relations { + relationId := relation.Id + relationFuncs[i] = func() error { + return l.svcCtx.AgentRelationModel.Delete(tranCtx, session, relationId) + } + } + + if err := mr.Finish(relationFuncs...); err != nil { + return errors.Wrapf(err, "删除代理关系信息失败") + } + } + + // 3.3 删除代理钱包信息 + wallet, err := l.svcCtx.AgentWalletModel.FindOneByAgentId(tranCtx, agentModel.Id) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "查询代理钱包信息失败, agentId: %d", agentModel.Id) + } + + if wallet != nil { + if err := l.svcCtx.AgentWalletModel.Delete(tranCtx, session, wallet.Id); err != nil { + return errors.Wrapf(err, "删除代理钱包信息失败, walletId: %d", wallet.Id) + } + } + + // 3.3 已在上一步处理，删除代理关系信息 + } + + // 4. 代理审核表已废弃，无需删除审核信息 + + // 5. 删除用户查询记录 + queryBuilder := l.svcCtx.QueryModel.SelectBuilder().Where("user_id = ?", userID) + queries, err := l.svcCtx.QueryModel.FindAll(tranCtx, queryBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "查询用户查询记录失败, userId: %d", userID) + } + + if len(queries) > 0 { + queryFuncs := make([]func() error, len(queries)) + for i, query := range queries { + queryId := query.Id + queryFuncs[i] = func() error { + return l.svcCtx.QueryModel.Delete(tranCtx, session, queryId) + } + } + + if err := mr.Finish(queryFuncs...); err != nil { + return errors.Wrapf(err, "删除用户查询记录失败") + } + } + + // 6. 删除用户订单记录 + orderBuilder := l.svcCtx.OrderModel.SelectBuilder().Where("user_id = ?", userID) + orders, err := l.svcCtx.OrderModel.FindAll(tranCtx, orderBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "查询用户订单记录失败, userId: %d", userID) + } + + if len(orders) > 0 { + orderFuncs := make([]func() error, len(orders)) + for i, order := range orders { + orderId := order.Id + orderFuncs[i] = func() error { + return l.svcCtx.OrderModel.Delete(tranCtx, session, orderId) + } + } + + if err := mr.Finish(orderFuncs...); err != nil { + return errors.Wrapf(err, "删除用户订单记录失败") + } + } + + // 7. 删除代理订单信息 + agentOrderBuilder := l.svcCtx.AgentOrderModel.SelectBuilder().Where("agent_id = ?", agentModel.Id) + agentOrders, err := l.svcCtx.AgentOrderModel.FindAll(tranCtx, agentOrderBuilder, "") + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "查询代理订单信息失败, agentId: %d, err: %v", agentModel.Id, err) + } + + if len(agentOrders) > 0 { + agentOrderFuncs := make([]func() error, len(agentOrders)) + for i, agentOrder := range agentOrders { + agentOrderId := agentOrder.Id + agentOrderFuncs[i] = func() error { + return l.svcCtx.AgentOrderModel.Delete(tranCtx, session, agentOrderId) + } + } + + if err := mr.Finish(agentOrderFuncs...); err != nil { + return errors.Wrapf(err, "删除代理订单信息失败") + } + } + return nil + }) + + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "用户注销失败%v", err) + } + + return nil +} diff --git a/app/main/api/internal/logic/user/detaillogic.go b/app/main/api/internal/logic/user/detaillogic.go new file mode 100644 index 0000000..972d191 --- /dev/null +++ b/app/main/api/internal/logic/user/detaillogic.go @@ -0,0 +1,66 @@ +package user + +import ( + "context" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/jinzhu/copier" + "github.com/pkg/errors" + + "github.com/zeromicro/go-zero/core/logx" +) + +type DetailLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewDetailLogic(ctx context.Context, svcCtx *svc.ServiceContext) *DetailLogic { + return &DetailLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *DetailLogic) Detail() (resp *types.UserInfoResp, err error) { + claims, err := ctxdata.GetClaimsFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, %v", err) + } + + userID := claims.UserId + + // 无论是临时用户还是正常用户，都需要从数据库中查询用户信息 + user, err := l.svcCtx.UserModel.FindOne(l.ctx, userID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.USER_NOT_FOUND), "用户信息, 用户不存在, %v", err) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "用户信息, 数据库查询用户信息失败, %v", err) + } + + var userInfo types.User + err = copier.Copy(&userInfo, user) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, 用户信息结构体复制失败, %v", err) + } + + if user.Mobile.Valid { + userInfo.Mobile, err = crypto.DecryptMobile(user.Mobile.String, l.svcCtx.Config.Encrypt.SecretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, 解密手机号失败, %v", err) + } + } + userInfo.UserType = claims.UserType + + return &types.UserInfoResp{ + UserInfo: userInfo, + }, nil +} diff --git a/app/main/api/internal/logic/user/getsignaturelogic.go b/app/main/api/internal/logic/user/getsignaturelogic.go new file mode 100644 index 0000000..5159fb5 --- /dev/null +++ b/app/main/api/internal/logic/user/getsignaturelogic.go @@ -0,0 +1,185 @@ +package user + +import ( + "context" + "crypto/sha1" + "encoding/json" + "fmt" + "io" + "net/http" + "net/url" + "sort" + "strings" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type GetSignatureLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetSignatureLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetSignatureLogic { + return &GetSignatureLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetSignatureLogic) GetSignature(req *types.GetSignatureReq) (resp *types.GetSignatureResp, err error) { + // 1. 获取access_token + accessToken, err := l.getAccessToken() + if err != nil { + l.Errorf("获取access_token失败: %v", err) + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取access_token失败: %v", err) + } + + // 2. 获取jsapi_ticket + jsapiTicket, err := l.getJsapiTicket(accessToken) + if err != nil { + l.Errorf("获取jsapi_ticket失败: %v", err) + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取jsapi_ticket失败: %v", err) + } + + // 3. 生成签名 + timestamp := time.Now().Unix() + nonceStr := l.generateNonceStr(16) + signature := l.generateSignature(jsapiTicket, nonceStr, timestamp, req.Url) + + // 4. 返回完整的JS-SDK配置信息 + return &types.GetSignatureResp{ + AppId: l.svcCtx.Config.WechatH5.AppID, + Timestamp: timestamp, + NonceStr: nonceStr, + Signature: signature, + }, nil +} + +// getAccessToken 获取微信公众号access_token +func (l *GetSignatureLogic) getAccessToken() (string, error) { + appID := l.svcCtx.Config.WechatH5.AppID + appSecret := l.svcCtx.Config.WechatH5.AppSecret + + url := fmt.Sprintf("https://api.weixin.qq.com/cgi-bin/token?grant_type=client_credential&appid=%s&secret=%s", appID, appSecret) + + resp, err := http.Get(url) + if err != nil { + return "", err + } + defer resp.Body.Close() + + body, err := io.ReadAll(resp.Body) + if err != nil { + return "", err + } + + var result struct { + AccessToken string `json:"access_token"` + ExpiresIn int `json:"expires_in"` + ErrCode int `json:"errcode"` + ErrMsg string `json:"errmsg"` + } + + if err = json.Unmarshal(body, &result); err != nil { + return "", err + } + + if result.ErrCode != 0 { + return "", fmt.Errorf("获取access_token失败: errcode=%d, errmsg=%s", result.ErrCode, result.ErrMsg) + } + + return result.AccessToken, nil +} + +// getJsapiTicket 获取jsapi_ticket +func (l *GetSignatureLogic) getJsapiTicket(accessToken string) (string, error) { + url := fmt.Sprintf("https://api.weixin.qq.com/cgi-bin/ticket/getticket?access_token=%s&type=jsapi", accessToken) + + resp, err := http.Get(url) + if err != nil { + return "", err + } + defer resp.Body.Close() + + body, err := io.ReadAll(resp.Body) + if err != nil { + return "", err + } + + var result struct { + Ticket string `json:"ticket"` + ExpiresIn int `json:"expires_in"` + ErrCode int `json:"errcode"` + ErrMsg string `json:"errmsg"` + } + + if err = json.Unmarshal(body, &result); err != nil { + return "", err + } + + if result.ErrCode != 0 { + return "", fmt.Errorf("获取jsapi_ticket失败: errcode=%d, errmsg=%s", result.ErrCode, result.ErrMsg) + } + + return result.Ticket, nil +} + +// generateNonceStr 生成随机字符串 +func (l *GetSignatureLogic) generateNonceStr(length int) string { + chars := "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789" + result := make([]byte, length) + for i := 0; i < length; i++ { + result[i] = chars[i%len(chars)] + } + return string(result) +} + +// generateSignature 生成签名 +func (l *GetSignatureLogic) generateSignature(jsapiTicket, nonceStr string, timestamp int64, urlStr string) string { + // 对URL进行解码，避免重复编码 + decodedURL, err := url.QueryUnescape(urlStr) + if err != nil { + decodedURL = urlStr + } + + // 构建签名字符串 + params := map[string]string{ + "jsapi_ticket": jsapiTicket, + "noncestr": nonceStr, + "timestamp": fmt.Sprintf("%d", timestamp), + "url": decodedURL, + } + + // 对参数进行字典序排序 + keys := make([]string, 0, len(params)) + for k := range params { + keys = append(keys, k) + } + sort.Strings(keys) + + // 拼接字符串 + var signStr strings.Builder + for i, k := range keys { + if i > 0 { + signStr.WriteString("&") + } + signStr.WriteString(k) + signStr.WriteString("=") + signStr.WriteString(params[k]) + } + + // SHA1加密 + h := sha1.New() + h.Write([]byte(signStr.String())) + signature := fmt.Sprintf("%x", h.Sum(nil)) + + return signature +} diff --git a/app/main/api/internal/logic/user/gettokenlogic.go b/app/main/api/internal/logic/user/gettokenlogic.go new file mode 100644 index 0000000..69c2584 --- /dev/null +++ b/app/main/api/internal/logic/user/gettokenlogic.go @@ -0,0 +1,47 @@ +package user + +import ( + "context" + "time" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type GetTokenLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewGetTokenLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetTokenLogic { + return &GetTokenLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *GetTokenLogic) GetToken() (resp *types.MobileCodeLoginResp, err error) { + claims, err := ctxdata.GetClaimsFromCtx(l.ctx) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, %v", err) + } + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, claims.UserId) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户信息, %v", err) + } + // 获取当前时间戳 + now := time.Now().Unix() + return &types.MobileCodeLoginResp{ + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} diff --git a/app/main/api/internal/logic/user/mobilecodeloginlogic.go b/app/main/api/internal/logic/user/mobilecodeloginlogic.go new file mode 100644 index 0000000..18373c3 --- /dev/null +++ b/app/main/api/internal/logic/user/mobilecodeloginlogic.go @@ -0,0 +1,77 @@ +package user + +import ( + "context" + "database/sql" + "fmt" + "os" + "time" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + "qnc-server/pkg/lzkit/crypto" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/redis" + + "github.com/zeromicro/go-zero/core/logx" +) + +type MobileCodeLoginLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewMobileCodeLoginLogic(ctx context.Context, svcCtx *svc.ServiceContext) *MobileCodeLoginLogic { + return &MobileCodeLoginLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *MobileCodeLoginLogic) MobileCodeLogin(req *types.MobileCodeLoginReq) (resp *types.MobileCodeLoginResp, err error) { + secretKey := l.svcCtx.Config.Encrypt.SecretKey + encryptedMobile, err := crypto.EncryptMobile(req.Mobile, secretKey) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "手机登录, 加密手机号失败: %+v", err) + } + // 开发环境下跳过验证码校验 + if os.Getenv("ENV") != "development" { + // 检查手机号是否在一分钟内已发送过验证码 + redisKey := fmt.Sprintf("%s:%s", "login", encryptedMobile) + cacheCode, err := l.svcCtx.Redis.Get(redisKey) + if err != nil { + if errors.Is(err, redis.Nil) { + return nil, errors.Wrapf(xerr.NewErrMsg("验证码已过期"), "手机登录, 验证码过期: %s", encryptedMobile) + } + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "手机登录, 读取验证码redis缓存失败, mobile: %s, err: %+v", encryptedMobile, err) + } + if cacheCode != req.Code { + return nil, errors.Wrapf(xerr.NewErrMsg("验证码不正确"), "手机登录, 验证码不正确: %s", encryptedMobile) + } + } + var userID string + user, findUserErr := l.svcCtx.UserModel.FindOneByMobile(l.ctx, sql.NullString{String: encryptedMobile, Valid: true}) + if findUserErr != nil && findUserErr != model.ErrNotFound { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "手机登录, 读取数据库获取用户失败, mobile: %s, err: %+v", encryptedMobile, err) + } + if user == nil { + return nil, errors.Wrapf(xerr.NewErrMsg("用户不存在"), "手机登录, 用户不存在: %s", encryptedMobile) + } + userID = user.Id + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "手机登录, 生成token失败 : %s", userID) + } + + // 获取当前时间戳 + now := time.Now().Unix() + return &types.MobileCodeLoginResp{ + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} diff --git a/app/main/api/internal/logic/user/wxh5authlogic.go b/app/main/api/internal/logic/user/wxh5authlogic.go new file mode 100644 index 0000000..d4ed2db --- /dev/null +++ b/app/main/api/internal/logic/user/wxh5authlogic.go @@ -0,0 +1,140 @@ +package user + +import ( + "context" + "encoding/json" + "fmt" + "io" + "net/http" + "time" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/google/uuid" + "github.com/pkg/errors" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/zeromicro/go-zero/core/logx" +) + +type WxH5AuthLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewWxH5AuthLogic(ctx context.Context, svcCtx *svc.ServiceContext) *WxH5AuthLogic { + return &WxH5AuthLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *WxH5AuthLogic) WxH5Auth(req *types.WXH5AuthReq) (resp *types.WXH5AuthResp, err error) { + l.Infof("🔄 WxH5Auth 开始处理微信H5授权请求") + + // Step 1: 使用code获取access_token + l.Infof("📡 Step 1: 使用code获取access_token, code=%s", req.Code) + accessTokenResp, err := l.GetAccessToken(req.Code) + if err != nil { + l.Errorf("❌ 获取access_token失败: %v", err) + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取access_token失败: %v", err) + } + l.Infof("✅ 成功获取access_token, openid=%s", accessTokenResp.Openid) + + // Step 2: 查找用户授权信息 + l.Infof("📡 Step 2: 查找用户授权信息, auth_type=%s, openid=%s", model.UserAuthTypeWxh5OpenID, accessTokenResp.Openid) + userAuth, findErr := l.svcCtx.UserAuthModel.FindOneByAuthTypeAuthKey(l.ctx, model.UserAuthTypeWxh5OpenID, accessTokenResp.Openid) + if findErr != nil && !errors.Is(findErr, model.ErrNotFound) { + l.Errorf("❌ 查询用户授权失败: %v", findErr) + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户授权失败: %v", findErr) + } + + // Step 3: 处理用户信息 + var userID string + if userAuth != nil { + // 已存在用户，直接登录 + userID = userAuth.UserId + l.Infof("✅ 用户已存在, userID=%s, openid=%s", userID, accessTokenResp.Openid) + } else { + // 新用户创建为临时用户（没有mobile） + l.Infof("📝 Step 3a: 创建新的临时用户, openid=%s", accessTokenResp.Openid) + user := &model.User{Id: uuid.NewString()} + _, err := l.svcCtx.UserModel.Insert(l.ctx, nil, user) + if err != nil { + l.Errorf("❌ 创建User失败: userID=%s, error=%v", user.Id, err) + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户失败: %v", err) + } + l.Infof("✅ User创建成功: userID=%s, mobile=null (临时用户)", user.Id) + + l.Infof("📝 Step 3b: 创建UserAuth记录, userID=%s, openid=%s", user.Id, accessTokenResp.Openid) + ua := &model.UserAuth{Id: uuid.NewString(), UserId: user.Id, AuthType: model.UserAuthTypeWxh5OpenID, AuthKey: accessTokenResp.Openid} + _, err = l.svcCtx.UserAuthModel.Insert(l.ctx, nil, ua) + if err != nil { + l.Errorf("❌ 创建UserAuth失败: userID=%s, openid=%s, error=%v", user.Id, accessTokenResp.Openid, err) + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户授权失败: %v", err) + } + l.Infof("✅ UserAuth创建成功: userAuthID=%s, userID=%s, authType=%s", ua.Id, user.Id, model.UserAuthTypeWxh5OpenID) + + userID = user.Id + l.Infof("✅ 新建临时用户完成: userID=%s, openid=%s", userID, accessTokenResp.Openid) + } + + // Step 4: 生成JWT Token（动态计算userType） + l.Infof("📡 Step 4: 生成JWT Token, userID=%s", userID) + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + l.Errorf("❌ 生成token失败: userID=%s, error=%v", userID, err) + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成JWT token失败: %v", err) + } + l.Infof("✅ Token生成成功: userID=%s, tokenLen=%d", userID, len(token)) + + // Step 5: 返回登录结果 + l.Infof("📡 Step 5: 返回登录结果") + now := time.Now().Unix() + resp = &types.WXH5AuthResp{ + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + } + l.Infof("🎯 WxH5Auth完成: userID=%s, accessExpire=%d, refreshAfter=%d", userID, resp.AccessExpire, resp.RefreshAfter) + return resp, nil +} + +type AccessTokenResp struct { + AccessToken string `json:"access_token"` + Openid string `json:"openid"` +} + +// GetAccessToken 通过code获取access_token +func (l *WxH5AuthLogic) GetAccessToken(code string) (*AccessTokenResp, error) { + appID := l.svcCtx.Config.WechatH5.AppID + appSecret := l.svcCtx.Config.WechatH5.AppSecret + + url := fmt.Sprintf("https://api.weixin.qq.com/sns/oauth2/access_token?appid=%s&secret=%s&code=%s&grant_type=authorization_code", appID, appSecret, code) + + resp, err := http.Get(url) + if err != nil { + return nil, err + } + defer resp.Body.Close() + + body, err := io.ReadAll(resp.Body) + if err != nil { + return nil, err + } + + var accessTokenResp AccessTokenResp + if err = json.Unmarshal(body, &accessTokenResp); err != nil { + return nil, err + } + + if accessTokenResp.AccessToken == "" || accessTokenResp.Openid == "" { + return nil, errors.New("accessTokenResp.AccessToken为空") + } + + return &accessTokenResp, nil +} diff --git a/app/main/api/internal/logic/user/wxminiauthlogic.go b/app/main/api/internal/logic/user/wxminiauthlogic.go new file mode 100644 index 0000000..96cc74b --- /dev/null +++ b/app/main/api/internal/logic/user/wxminiauthlogic.go @@ -0,0 +1,134 @@ +package user + +import ( + "context" + "encoding/json" + "fmt" + "io" + "net/http" + "time" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/common/xerr" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type WxMiniAuthLogic struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewWxMiniAuthLogic(ctx context.Context, svcCtx *svc.ServiceContext) *WxMiniAuthLogic { + return &WxMiniAuthLogic{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *WxMiniAuthLogic) WxMiniAuth(req *types.WXMiniAuthReq) (resp *types.WXMiniAuthResp, err error) { + // 1. 获取session_key和openid + sessionKeyResp, err := l.GetSessionKey(req.Code) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取session_key失败: %v", err) + } + + // 2. 查找用户授权信息 + userAuth, err := l.svcCtx.UserAuthModel.FindOneByAuthTypeAuthKey(l.ctx, model.UserAuthTypeWxMiniOpenID, sessionKeyResp.Openid) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "查询用户授权失败: %v", err) + } + + // 3. 处理用户信息 + var userID string + if userAuth != nil { + // 已存在用户，直接登录 + userID = userAuth.UserId + } else { + // 新用户创建为临时用户（没有mobile） + user := &model.User{Id: uuid.NewString()} + _, err := l.svcCtx.UserModel.Insert(l.ctx, nil, user) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户失败: %v", err) + } + ua := &model.UserAuth{Id: uuid.NewString(), UserId: user.Id, AuthType: model.UserAuthTypeWxMiniOpenID, AuthKey: sessionKeyResp.Openid} + _, err = l.svcCtx.UserAuthModel.Insert(l.ctx, nil, ua) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "创建用户授权失败: %v", err) + } + userID = user.Id + } + + // 4. 生成JWT Token（动态计算userType） + token, err := l.svcCtx.UserService.GeneralUserToken(l.ctx, userID) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "生成JWT Token失败: %v", err) + } + + // 5. 返回登录结果 + now := time.Now().Unix() + return &types.WXMiniAuthResp{ + AccessToken: token, + AccessExpire: now + l.svcCtx.Config.JwtAuth.AccessExpire, + RefreshAfter: now + l.svcCtx.Config.JwtAuth.RefreshAfter, + }, nil +} + +// SessionKeyResp 小程序登录返回结构 +type SessionKeyResp struct { + Openid string `json:"openid"` + SessionKey string `json:"session_key"` + Unionid string `json:"unionid,omitempty"` + ErrCode int `json:"errcode,omitempty"` + ErrMsg string `json:"errmsg,omitempty"` +} + +// GetSessionKey 通过code获取小程序的session_key和openid +func (l *WxMiniAuthLogic) GetSessionKey(code string) (*SessionKeyResp, error) { + var appID string + var appSecret string + + appID = l.svcCtx.Config.WechatMini.AppID + appSecret = l.svcCtx.Config.WechatMini.AppSecret + + url := fmt.Sprintf("https://api.weixin.qq.com/sns/jscode2session?appid=%s&secret=%s&js_code=%s&grant_type=authorization_code", + appID, appSecret, code) + + resp, err := http.Get(url) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "获取session_key失败: %v", err) + } + defer resp.Body.Close() + + body, err := io.ReadAll(resp.Body) + if err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "读取响应失败: %v", err) + } + + var sessionKeyResp SessionKeyResp + if err = json.Unmarshal(body, &sessionKeyResp); err != nil { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "解析响应失败: %v", err) + } + + // 检查微信返回的错误码 + if sessionKeyResp.ErrCode != 0 { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), + "微信接口返回错误: errcode=%d, errmsg=%s", + sessionKeyResp.ErrCode, sessionKeyResp.ErrMsg) + } + + // 验证必要字段 + if sessionKeyResp.Openid == "" || sessionKeyResp.SessionKey == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), + "微信接口返回数据不完整: openid=%s, session_key=%s", + sessionKeyResp.Openid, sessionKeyResp.SessionKey) + } + + return &sessionKeyResp, nil +} diff --git a/app/main/api/internal/middleware/adminauthinterceptormiddleware.go b/app/main/api/internal/middleware/adminauthinterceptormiddleware.go new file mode 100644 index 0000000..180307a --- /dev/null +++ b/app/main/api/internal/middleware/adminauthinterceptormiddleware.go @@ -0,0 +1,234 @@ +package middleware + +import ( + "context" + "net/http" + "strings" + + "qnc-server/app/main/api/internal/config" + "qnc-server/app/main/model" + jwtx "qnc-server/common/jwt" + "qnc-server/common/result" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/rest/httpx" +) + +const ( + // 定义错误码 + AdminErrCodeUnauthorized = 401 +) + +type AdminAuthInterceptorMiddleware struct { + Config config.Config + // 注入model依赖 + AdminUserModel model.AdminUserModel + AdminUserRoleModel model.AdminUserRoleModel + AdminRoleModel model.AdminRoleModel + AdminApiModel model.AdminApiModel + AdminRoleApiModel model.AdminRoleApiModel +} + +func NewAdminAuthInterceptorMiddleware(c config.Config, + adminUserModel model.AdminUserModel, + adminUserRoleModel model.AdminUserRoleModel, + adminRoleModel model.AdminRoleModel, + adminApiModel model.AdminApiModel, + adminRoleApiModel model.AdminRoleApiModel) *AdminAuthInterceptorMiddleware { + return &AdminAuthInterceptorMiddleware{ + Config: c, + AdminUserModel: adminUserModel, + AdminUserRoleModel: adminUserRoleModel, + AdminRoleModel: adminRoleModel, + AdminApiModel: adminApiModel, + AdminRoleApiModel: adminRoleApiModel, + } +} + +func (m *AdminAuthInterceptorMiddleware) Handle(next http.HandlerFunc) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + // 1. JWT 校验 + claims, err := m.validateJWT(r) + if err != nil { + m.writeUnauthorizedResponse(w, err) + return + } + + // 2. 检查用户类型是否为管理员 + if claims.UserType != model.UserTypeAdmin { + authErr := errors.Wrapf(xerr.NewErrCode(xerr.TOKEN_EXPIRE_ERROR), "用户类型错误，需要管理员权限") + m.writeUnauthorizedResponse(w, authErr) + return + } + + // 3. 检查平台标识 + if claims.Platform != model.PlatformAdmin { + authErr := errors.Wrapf(xerr.NewErrCode(xerr.TOKEN_EXPIRE_ERROR), "平台标识错误，需要管理员平台") + m.writeUnauthorizedResponse(w, authErr) + return + } + + // 4. 将用户信息放入上下文 + ctx := context.WithValue(r.Context(), jwtx.ExtraKey, claims) + r = r.WithContext(ctx) + + // 5. API 权限校验 + if err := m.validateApiPermission(r.Context(), claims.UserId, r.Method, r.URL.Path); err != nil { + m.writeUnauthorizedResponse(w, err) + return + } + + // 6. 通过所有校验，继续处理请求 + next(w, r) + } +} + +// writeUnauthorizedResponse 写入401未授权响应 +func (m *AdminAuthInterceptorMiddleware) writeUnauthorizedResponse(w http.ResponseWriter, err error) { + errcode := xerr.TOKEN_EXPIRE_ERROR + errmsg := xerr.MapErrMsg(errcode) + + // 从错误中提取错误码和错误消息 + causeErr := errors.Cause(err) + if e, ok := causeErr.(*xerr.CodeError); ok { + errcode = e.GetErrCode() + errmsg = e.GetErrMsg() + } + + // 返回401 HTTP状态码 + httpx.WriteJson(w, http.StatusUnauthorized, result.Error(errcode, errmsg)) +} + +// validateJWT 验证JWT token +func (m *AdminAuthInterceptorMiddleware) validateJWT(r *http.Request) (*jwtx.JwtClaims, error) { + // 从请求头中获取Authorization字段 + authHeader := r.Header.Get("Authorization") + if authHeader == "" { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.TOKEN_EXPIRE_ERROR), "缺少Authorization头") + } + + // 去掉Bearer前缀 + token := strings.TrimPrefix(authHeader, "Bearer ") + if token == authHeader { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.TOKEN_EXPIRE_ERROR), "Authorization头格式错误，缺少Bearer前缀") + } + + claims, err := jwtx.ParseJwtToken(token, m.Config.AdminConfig.AccessSecret) + if err != nil { + // 根据错误类型返回不同的错误码 + errMsg := err.Error() + if strings.Contains(errMsg, "token已过期") || strings.Contains(errMsg, "expired") { + return nil, errors.Wrapf(xerr.NewErrCode(xerr.TOKEN_EXPIRE_ERROR), "token解析失败: %v", err) + } + // 其他JWT解析错误使用统一的token过期错误码（更符合用户理解） + return nil, errors.Wrapf(xerr.NewErrCode(xerr.TOKEN_EXPIRE_ERROR), "token解析失败: %v", err) + } + + return claims, nil +} + +// validateApiPermission 验证API权限 +func (m *AdminAuthInterceptorMiddleware) validateApiPermission(ctx context.Context, userId string, method, path string) error { + // 1. 获取用户角色 + userRoles, err := m.getUserRoles(ctx, userId) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取用户角色失败: %v", err) + } + + if len(userRoles) == 0 { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "用户没有分配角色") + } + + // 2. 检查是否为超级管理员 + if m.isSuperAdmin(ctx, userRoles) { + // 超级管理员拥有所有权限，直接放行 + return nil + } + + // 3. 获取当前请求的API信息 + api, err := m.getApiByMethodAndPath(ctx, method, path) + if err != nil { + // 如果API不存在，可能是公开接口，放行 + return nil + } + + // 4. 检查用户角色是否有该API权限 + hasPermission, err := m.checkRoleApiPermission(ctx, userRoles, api.Id) + if err != nil { + return errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "检查API权限失败: %v", err) + } + + if !hasPermission { + return errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "权限不足，无法访问该接口") + } + + return nil +} + +// getUserRoles 获取用户角色 +func (m *AdminAuthInterceptorMiddleware) getUserRoles(ctx context.Context, userId string) ([]string, error) { + builder := m.AdminUserRoleModel.SelectBuilder().Where("user_id = ?", userId) + userRoles, err := m.AdminUserRoleModel.FindAll(ctx, builder, "") + if err != nil { + return nil, err + } + + var roleIds []string + for _, userRole := range userRoles { + roleIds = append(roleIds, userRole.RoleId) + } + + return roleIds, nil +} + +// isSuperAdmin 检查是否为超级管理员 +func (m *AdminAuthInterceptorMiddleware) isSuperAdmin(ctx context.Context, roleIds []string) bool { + // 检查是否有超级管理员角色 + for _, roleId := range roleIds { + role, err := m.AdminRoleModel.FindOne(ctx, roleId) + if err != nil { + continue + } + // 检查是否为超级管理员角色 + if role.RoleCode == model.AdminRoleCodeSuper { + return true + } + } + return false +} + +// getApiByMethodAndPath 根据方法和路径获取API信息 +func (m *AdminAuthInterceptorMiddleware) getApiByMethodAndPath(ctx context.Context, method, path string) (*model.AdminApi, error) { + builder := m.AdminApiModel.SelectBuilder(). + Where("method = ? AND url = ? AND status = ?", method, path, 1) + + apis, err := m.AdminApiModel.FindAll(ctx, builder, "") + if err != nil { + return nil, err + } + + if len(apis) == 0 { + return nil, errors.New("API不存在") + } + + return apis[0], nil +} + +// checkRoleApiPermission 检查角色是否有API权限 +func (m *AdminAuthInterceptorMiddleware) checkRoleApiPermission(ctx context.Context, roleIds []string, apiId string) (bool, error) { + for _, roleId := range roleIds { + // 检查角色是否有该API权限 + _, err := m.AdminRoleApiModel.FindOneByRoleIdApiId(ctx, roleId, apiId) + if err == nil { + // 找到权限记录，说明有权限 + return true, nil + } + // 如果错误不是NotFound，说明是其他错误 + if !errors.Is(err, model.ErrNotFound) { + return false, err + } + } + + return false, nil +} diff --git a/app/main/api/internal/middleware/authinterceptormiddleware.go b/app/main/api/internal/middleware/authinterceptormiddleware.go new file mode 100644 index 0000000..fcad3b3 --- /dev/null +++ b/app/main/api/internal/middleware/authinterceptormiddleware.go @@ -0,0 +1,54 @@ +package middleware + +import ( + "context" + "net/http" + + "qnc-server/app/main/api/internal/config" + jwtx "qnc-server/common/jwt" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/rest/httpx" +) + +const ( + // 定义错误码 + ErrCodeUnauthorized = 401 +) + +type AuthInterceptorMiddleware struct { + Config config.Config +} + +func NewAuthInterceptorMiddleware(c config.Config) *AuthInterceptorMiddleware { + return &AuthInterceptorMiddleware{ + Config: c, + } +} + +func (m *AuthInterceptorMiddleware) Handle(next http.HandlerFunc) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + // 从请求头中获取Authorization字段 + authHeader := r.Header.Get("Authorization") + + // 如果没有Authorization头，直接放行 + if authHeader == "" { + next(w, r) + return + } + + // 解析JWT令牌 + claims, err := jwtx.ParseJwtToken(authHeader, m.Config.JwtAuth.AccessSecret) + if err != nil { + // JWT解析失败，返回401错误 + httpx.Error(w, errors.Wrapf(xerr.NewErrCode(ErrCodeUnauthorized), "token解析失败: %v", err)) + return + } + + ctx := context.WithValue(r.Context(), jwtx.ExtraKey, claims) + + // 使用新的上下文继续处理请求 + next(w, r.WithContext(ctx)) + } +} diff --git a/app/main/api/internal/middleware/global_sourceinterceptor_middleware.go b/app/main/api/internal/middleware/global_sourceinterceptor_middleware.go new file mode 100644 index 0000000..c7197b3 --- /dev/null +++ b/app/main/api/internal/middleware/global_sourceinterceptor_middleware.go @@ -0,0 +1,29 @@ +package middleware + +import ( + "context" + "net/http" +) + +const ( + PlatformKey = "X-Platform" +) + +func GlobalSourceInterceptor(next http.HandlerFunc) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + // 获取请求头 X-Platform 的值 + platform := r.Header.Get(PlatformKey) + + // 将值放入新的 context 中 + ctx := r.Context() + if platform != "" { + ctx = context.WithValue(ctx, "platform", platform) + } + + // 通过 r.WithContext 将更新后的 ctx 传递给后续的处理函数 + r = r.WithContext(ctx) + + // 传递给下一个处理器 + next(w, r) + } +} diff --git a/app/main/api/internal/middleware/userauthinterceptormiddleware.go b/app/main/api/internal/middleware/userauthinterceptormiddleware.go new file mode 100644 index 0000000..ba49e4b --- /dev/null +++ b/app/main/api/internal/middleware/userauthinterceptormiddleware.go @@ -0,0 +1,35 @@ +package middleware + +import ( + "net/http" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/rest/httpx" +) + +type UserAuthInterceptorMiddleware struct { +} + +func NewUserAuthInterceptorMiddleware() *UserAuthInterceptorMiddleware { + return &UserAuthInterceptorMiddleware{} +} + +func (m *UserAuthInterceptorMiddleware) Handle(next http.HandlerFunc) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + claims, err := ctxdata.GetClaimsFromCtx(r.Context()) + if err != nil { + httpx.Error(w, errors.Wrapf(xerr.NewErrCode(ErrCodeUnauthorized), "token解析失败: %v", err)) + return + } + // 检查用户是否绑定了mobile（没有mobile表示是临时用户，不允许访问需要认证的接口） + // 注：临时用户现在基于 mobile 字段判断，而不是 UserType + if claims.UserType == model.UserTypeTemp { + httpx.Error(w, errors.Wrapf(xerr.NewErrCode(xerr.USER_NEED_BIND_MOBILE), "请先绑定手机号: %v", err)) + return + } + next(w, r) + } +} diff --git a/app/main/api/internal/queue/agentProcess.go b/app/main/api/internal/queue/agentProcess.go new file mode 100644 index 0000000..fdea947 --- /dev/null +++ b/app/main/api/internal/queue/agentProcess.go @@ -0,0 +1,63 @@ +package queue + +import ( + "context" + "encoding/json" + "errors" + "fmt" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + + "github.com/hibiken/asynq" + "github.com/zeromicro/go-zero/core/logx" +) + +type AgentProcessHandler struct { + svcCtx *svc.ServiceContext +} + +func NewAgentProcessHandler(svcCtx *svc.ServiceContext) *AgentProcessHandler { + return &AgentProcessHandler{ + svcCtx: svcCtx, + } +} + +func (l *AgentProcessHandler) ProcessTask(ctx context.Context, t *asynq.Task) error { + var payload types.MsgAgentProcessPayload + if err := json.Unmarshal(t.Payload(), &payload); err != nil { + return fmt.Errorf("解析代理处理任务负载失败: %w", err) + } + + // 获取订单信息 + order, err := l.svcCtx.OrderModel.FindOne(ctx, payload.OrderID) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + logx.Errorf("代理处理任务失败，订单不存在: orderID=%s", payload.OrderID) + return asynq.SkipRetry // 订单不存在，跳过重试 + } + return fmt.Errorf("查询订单失败: orderID=%s, err=%w", payload.OrderID, err) + } + + // 检查订单状态 + if order.Status != "paid" { + logx.Infof("代理处理任务跳过，订单未支付: orderID=%s, status=%s", payload.OrderID, order.Status) + return nil // 订单未支付，不处理，不重试 + } + + // 调用代理处理服务 + err = l.svcCtx.AgentService.AgentProcess(ctx, order) + if err != nil { + // 记录错误日志，但不阻塞报告流程 + logx.Errorf("代理处理失败，订单ID: %s, 错误: %v", payload.OrderID, err) + // 返回错误以触发重试机制 + return fmt.Errorf("代理处理失败: orderID=%s, err=%w", payload.OrderID, err) + } + + // 注意：解冻任务现在通过定时任务扫描处理，不再需要发送延迟任务 + // 定时任务每5分钟扫描一次待解冻的任务，更加可靠 + logx.Infof("代理处理成功，订单ID: %s，冻结任务（如有）将由定时任务自动处理", payload.OrderID) + + logx.Infof("代理处理成功，订单ID: %s", payload.OrderID) + return nil +} diff --git a/app/main/api/internal/queue/cleanQueryData.go b/app/main/api/internal/queue/cleanQueryData.go new file mode 100644 index 0000000..57b26fe --- /dev/null +++ b/app/main/api/internal/queue/cleanQueryData.go @@ -0,0 +1,275 @@ +package queue + +import ( + "context" + "database/sql" + "fmt" + "strconv" + "time" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/model" + "qnc-server/common/globalkey" + + "github.com/google/uuid" + "github.com/hibiken/asynq" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +// TASKTIME 定义为每天凌晨3点执行 +const TASKTIME = "0 3 * * *" + +type CleanQueryDataHandler struct { + svcCtx *svc.ServiceContext +} + +func NewCleanQueryDataHandler(svcCtx *svc.ServiceContext) *CleanQueryDataHandler { + return &CleanQueryDataHandler{ + svcCtx: svcCtx, + } +} + +// 获取配置值 +func (l *CleanQueryDataHandler) getConfigValue(ctx context.Context, key string) (string, error) { + // 通过缓存获取配置 + config, err := l.svcCtx.QueryCleanupConfigModel.FindOneByConfigKey(ctx, key) + if err != nil { + if err == model.ErrNotFound { + return "", fmt.Errorf("配置项 %s 不存在", key) + } + return "", err + } + + // 检查配置状态 + if config.Status != 1 { + return "", fmt.Errorf("配置项 %s 已禁用或已删除", key) + } + + return config.ConfigValue, nil +} + +func (l *CleanQueryDataHandler) ProcessTask(ctx context.Context, t *asynq.Task) error { + // 添加超时控制：最多运行1小时 + taskCtx, cancel := context.WithTimeout(ctx, 1*time.Hour) + defer cancel() + + startTime := time.Now() + now := time.Now() + logx.Infof("%s - 开始执行查询数据清理任务", now.Format("2006-01-02 15:04:05")) + + // 1. 检查是否启用清理 + enableCleanup, err := l.getConfigValue(taskCtx, "enable_cleanup") + if err != nil { + return err + } + if enableCleanup != "1" { + logx.Infof("查询数据清理任务已禁用") + return nil + } + + // 2. 获取保留天数 + retentionDaysStr, err := l.getConfigValue(taskCtx, "retention_days") + if err != nil { + return err + } + retentionDays, err := strconv.Atoi(retentionDaysStr) + if err != nil { + return fmt.Errorf("保留天数配置无效: %v", err) + } + if retentionDays < 0 { + return fmt.Errorf("保留天数不能为负数: %d", retentionDays) + } + + // 3. 获取批次大小 + batchSizeStr, err := l.getConfigValue(taskCtx, "batch_size") + if err != nil { + return err + } + batchSize, err := strconv.Atoi(batchSizeStr) + if err != nil { + return fmt.Errorf("批次大小配置无效: %v", err) + } + if batchSize <= 0 || batchSize > 10000 { + return fmt.Errorf("批次大小必须在1-10000之间: %d", batchSize) + } + + // 计算清理截止时间 + cleanupBefore := now.AddDate(0, 0, -retentionDays) + + // 先快速检查是否有数据需要清理（避免创建无用的日志记录） + checkBuilder := l.svcCtx.QueryModel.SelectBuilder(). + Where("create_time < ?", cleanupBefore). + Where("del_state = ?", globalkey.DelStateNo). + Limit(1) // 只查询1条，用于判断是否有数据 + + checkQueries, checkErr := l.svcCtx.QueryModel.FindAll(taskCtx, checkBuilder, "") + if checkErr != nil { + logx.Errorf("检查是否有数据需要清理失败: %v", checkErr) + return checkErr + } + + // 如果没有数据需要清理，直接返回，不创建日志记录 + if len(checkQueries) == 0 { + logx.Infof("%s - 没有需要清理的数据（清理截止时间: %s）", now.Format("2006-01-02 15:04:05"), cleanupBefore.Format("2006-01-02 15:04:05")) + return nil + } + + // 创建清理日志记录（只创建一次） + cleanupLog := &model.QueryCleanupLog{ + Id: uuid.New().String(), + CleanupTime: now, + CleanupBefore: cleanupBefore, + Status: 1, + Remark: sql.NullString{String: "定时清理数据", Valid: true}, + } + + // 先创建清理日志记录 + err = l.svcCtx.QueryCleanupLogModel.Trans(taskCtx, func(logCtx context.Context, logSession sqlx.Session) error { + _, insertErr := l.svcCtx.QueryCleanupLogModel.Insert(logCtx, logSession, cleanupLog) + if insertErr != nil { + return insertErr + } + return insertErr + }) + if err != nil { + logx.Errorf("创建清理日志记录失败: %v", err) + return err + } + + logx.Infof("创建清理日志记录成功，日志ID: %s", cleanupLog.Id) + + // 分批处理，每个批次使用独立事务 + batchCount := 0 + lastProcessedId := "" + + for { + // 检查是否被取消（优雅关闭支持） + select { + case <-taskCtx.Done(): + logx.Infof("清理任务被取消，已处理 %d 批次，共删除 %d 条记录", batchCount, cleanupLog.AffectedRows) + // 更新清理日志状态 + l.updateCleanupLogStatus(taskCtx, cleanupLog.Id, cleanupLog, fmt.Errorf("任务被取消")) + return taskCtx.Err() + default: + // 继续处理 + } + + // 每个批次使用独立事务 + var batchQueries []*model.Query + batchErr := l.svcCtx.QueryModel.Trans(taskCtx, func(batchCtx context.Context, batchSession sqlx.Session) error { + // 1. 查询一批要删除的记录（添加排序确保一致性） + builder := l.svcCtx.QueryModel.SelectBuilder(). + Where("create_time < ?", cleanupBefore). + Where("del_state = ?", globalkey.DelStateNo). + OrderBy("id ASC"). // 添加排序，确保处理顺序一致 + Limit(uint64(batchSize)) + + // 如果已处理过，从上次处理的ID之后继续 + if lastProcessedId != "" { + builder = builder.Where("id > ?", lastProcessedId) + } + + var queryErr error + batchQueries, queryErr = l.svcCtx.QueryModel.FindAll(batchCtx, builder, "") + if queryErr != nil { + return queryErr + } + + if len(batchQueries) == 0 { + // 没有更多数据需要清理，标记为完成 + return nil + } + + // 2. 执行清理 + for _, query := range batchQueries { + deleteErr := l.svcCtx.QueryModel.DeleteSoft(batchCtx, batchSession, query) + if deleteErr != nil { + return deleteErr + } + } + + // 3. 保存清理明细 + for _, query := range batchQueries { + detail := &model.QueryCleanupDetail{ + Id: uuid.New().String(), + CleanupLogId: cleanupLog.Id, + QueryId: query.Id, + OrderId: query.OrderId, + UserId: query.UserId, + ProductId: query.ProductId, + QueryState: query.QueryState, + CreateTimeOld: query.CreateTime, + } + _, insertErr := l.svcCtx.QueryCleanupDetailModel.Insert(batchCtx, batchSession, detail) + if insertErr != nil { + return insertErr + } + } + + // 4. 记录最后处理的ID（用于下次查询） + lastProcessedId = batchQueries[len(batchQueries)-1].Id + + return nil + }) + + if batchErr != nil { + // 批次失败，更新清理日志状态 + logx.Errorf("批次处理失败（批次 %d）: %v", batchCount+1, batchErr) + l.updateCleanupLogStatus(taskCtx, cleanupLog.Id, cleanupLog, batchErr) + return batchErr + } + + // 如果查询结果为空，说明没有更多数据 + if len(batchQueries) == 0 { + logx.Infof("所有数据已处理完成") + break + } + + // 更新影响行数（在事务外更新，避免重复计算） + actualBatchSize := int64(len(batchQueries)) + cleanupLog.AffectedRows += actualBatchSize + batchCount++ + logx.Infof("批次 %d 处理完成，本批次删除 %d 条记录，累计删除 %d 条记录", batchCount, actualBatchSize, cleanupLog.AffectedRows) + + // 如果本批次查询到的数据少于批次大小，说明已经处理完所有数据 + if actualBatchSize < int64(batchSize) { + logx.Infof("所有数据已处理完成（本批次数据量少于批次大小）") + break + } + } + + // 更新清理日志状态为成功 + l.updateCleanupLogStatus(taskCtx, cleanupLog.Id, cleanupLog, nil) + + duration := time.Since(startTime) + logx.Infof("%s - 查询数据清理完成，共处理 %d 批次，删除 %d 条记录，耗时 %v", + now.Format("2006-01-02 15:04:05"), batchCount, cleanupLog.AffectedRows, duration) + return nil +} + +// updateCleanupLogStatus 更新清理日志状态 +func (l *CleanQueryDataHandler) updateCleanupLogStatus(ctx context.Context, logId string, cleanupLog *model.QueryCleanupLog, err error) { + err = l.svcCtx.QueryCleanupLogModel.Trans(ctx, func(updateCtx context.Context, updateSession sqlx.Session) error { + // 查询当前日志记录 + currentLog, findErr := l.svcCtx.QueryCleanupLogModel.FindOne(updateCtx, cleanupLog.Id) + if findErr != nil { + return findErr + } + + // 更新状态和影响行数 + currentLog.AffectedRows = cleanupLog.AffectedRows + if err != nil { + currentLog.Status = 2 // 失败 + currentLog.ErrorMsg = sql.NullString{String: err.Error(), Valid: true} + } else { + currentLog.Status = 1 // 成功 + } + + _, updateErr := l.svcCtx.QueryCleanupLogModel.Update(updateCtx, updateSession, currentLog) + return updateErr + }) + + if err != nil { + logx.Errorf("更新清理日志状态失败: %v", err) + } +} diff --git a/app/main/api/internal/queue/paySuccessNotify.go b/app/main/api/internal/queue/paySuccessNotify.go new file mode 100644 index 0000000..c2c0387 --- /dev/null +++ b/app/main/api/internal/queue/paySuccessNotify.go @@ -0,0 +1,428 @@ +package queue + +import ( + "context" + "encoding/hex" + "encoding/json" + "fmt" + "os" + "regexp" + "strings" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + "qnc-server/app/main/model" + "qnc-server/pkg/lzkit/crypto" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/google/uuid" + "github.com/hibiken/asynq" + "github.com/zeromicro/go-zero/core/logx" +) + +type PaySuccessNotifyUserHandler struct { + svcCtx *svc.ServiceContext +} + +func NewPaySuccessNotifyUserHandler(svcCtx *svc.ServiceContext) *PaySuccessNotifyUserHandler { + return &PaySuccessNotifyUserHandler{ + svcCtx: svcCtx, + } +} + +var payload types.MsgPaySuccessQueryPayload + +func (l *PaySuccessNotifyUserHandler) ProcessTask(ctx context.Context, t *asynq.Task) error { + // 从任务的负载中解码数据 + if err := json.Unmarshal(t.Payload(), &payload); err != nil { + return fmt.Errorf("解析任务负载失败: %w", err) + } + + order, err := l.svcCtx.OrderModel.FindOne(ctx, payload.OrderID) + if err != nil { + // 订单不存在，记录详细日志并跳过重试 + logx.Errorf("支付成功通知任务失败：订单不存在，订单ID: %s, 错误: %v", payload.OrderID, err) + return asynq.SkipRetry // 订单不存在时跳过重试，避免重复失败 + } + env := os.Getenv("ENV") + if order.Status != "paid" && env != "development" { + err = fmt.Errorf("无效的订单: %s", payload.OrderID) + logx.Errorf("处理任务失败，原因: %v", err) + return asynq.SkipRetry + } + product, err := l.svcCtx.ProductModel.FindOne(ctx, order.ProductId) + if err != nil { + return fmt.Errorf("找不到相关产品: orderID: %s, productID: %s", payload.OrderID, order.ProductId) + } + redisKey := fmt.Sprintf(types.QueryCacheKey, order.UserId, order.OrderNo) + cache, cacheErr := l.svcCtx.Redis.GetCtx(ctx, redisKey) + if cacheErr != nil { + return fmt.Errorf("获取缓存内容失败: %+v", cacheErr) + } + var data types.QueryCacheLoad + err = json.Unmarshal([]byte(cache), &data) + if err != nil { + return fmt.Errorf("解析缓存内容失败: %+v", err) + } + secretKey := l.svcCtx.Config.Encrypt.SecretKey + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return fmt.Errorf("获取AES密钥失败: %+v", decodeErr) + } + decryptData, aesdecryptErr := crypto.AesDecrypt(data.Params, key) + if aesdecryptErr != nil { + return fmt.Errorf("解密参数失败: %+v", aesdecryptErr) + } + + query := &model.Query{ + Id: uuid.NewString(), + OrderId: order.Id, + UserId: order.UserId, + ProductId: product.Id, + QueryParams: data.Params, + QueryState: "pending", + } + _, insertQueryErr := l.svcCtx.QueryModel.Insert(ctx, nil, query) + if insertQueryErr != nil { + return fmt.Errorf("保存查询失败: %+v", insertQueryErr) + } + + // 插入后使用预生成的查询ID + queryId := query.Id + + // 从数据库中查询完整的查询记录 + query, err = l.svcCtx.QueryModel.FindOne(ctx, queryId) + if err != nil { + return fmt.Errorf("获取插入后的查询记录失败: %+v", err) + } + + // 解析解密后的参数获取用户信息 + var userInfo map[string]interface{} + if err := json.Unmarshal(decryptData, &userInfo); err != nil { + return fmt.Errorf("解析用户信息失败: %+v", err) + } + + // 生成授权书 + authDoc, err := l.svcCtx.AuthorizationService.GenerateAuthorizationDocument( + ctx, order.UserId, order.Id, queryId, userInfo, + ) + if err != nil { + logx.Errorf("生成授权书失败: %v", err) + } + + // 将授权书URL添加到解密数据中 + if authDoc != nil { + // 生成完整的授权书访问URL + fullAuthDocURL := l.svcCtx.AuthorizationService.GetFullFileURL(authDoc.FileUrl) + userInfo["authorization_url"] = fullAuthDocURL + + // 重新序列化用户信息 + updatedDecryptData, marshalErr := json.Marshal(userInfo) + if marshalErr != nil { + logx.Errorf("序列化用户信息失败: %v", marshalErr) + } else { + // 重新加密更新后的数据 + encryptedUpdatedData, encryptErr := crypto.AesEncrypt(updatedDecryptData, key) + if encryptErr != nil { + logx.Errorf("重新加密用户信息失败: %v", encryptErr) + } else { + // 更新查询记录中的参数 + query.QueryParams = string(encryptedUpdatedData) + updateParamsErr := l.svcCtx.QueryModel.UpdateWithVersion(ctx, nil, query) + if updateParamsErr != nil { + logx.Errorf("更新查询参数失败: %v", updateParamsErr) + } else { + logx.Infof("成功更新查询参数，包含授权书URL: %s", fullAuthDocURL) + } + } + decryptData = updatedDecryptData + } + } + + // 检查是否为空报告模式（开发环境） + isEmptyReportMode := env == "development" && order.PaymentPlatform == "test" + + var encryptData string + if isEmptyReportMode { + // 空报告模式：生成空的报告数据，跳过API调用 + logx.Infof("空报告模式：订单 %s (ID: %s) 跳过API调用，生成空报告", order.OrderNo, order.Id) + + // 生成空报告数据结构（根据实际报告格式生成） + emptyReportData := []byte(`[]`) // 空数组，表示没有数据 + + // 加密空报告数据 + encryptedEmptyData, aesEncryptErr := crypto.AesEncrypt(emptyReportData, key) + if aesEncryptErr != nil { + err = fmt.Errorf("加密空报告数据失败: %v", aesEncryptErr) + return l.handleError(ctx, err, order, query) + } + encryptData = encryptedEmptyData + } else { + // 正常模式：调用API请求服务 + combinedResponse, err := l.svcCtx.ApiRequestService.ProcessRequests(decryptData, product.Id) + if err != nil { + return l.handleError(ctx, err, order, query) + } + // 加密返回响应 + encryptedResponse, aesEncryptErr := crypto.AesEncrypt(combinedResponse, key) + if aesEncryptErr != nil { + err = fmt.Errorf("加密响应信息失败: %v", aesEncryptErr) + return l.handleError(ctx, err, order, query) + } + encryptData = encryptedResponse + } + + query.QueryData = lzUtils.StringToNullString(encryptData) + updateErr := l.svcCtx.QueryModel.UpdateWithVersion(ctx, nil, query) + if updateErr != nil { + err = fmt.Errorf("保存响应数据失败: %v", updateErr) + return l.handleError(ctx, err, order, query) + } + + query.QueryState = "success" + updateQueryErr := l.svcCtx.QueryModel.UpdateWithVersion(ctx, nil, query) + if updateQueryErr != nil { + updateQueryErr = fmt.Errorf("修改查询状态失败: %v", updateQueryErr) + return l.handleError(ctx, updateQueryErr, order, query) + } + + // 报告生成成功后，发送代理处理异步任务（不阻塞报告流程） + if asyncErr := l.svcCtx.AsynqService.SendAgentProcessTask(order.Id); asyncErr != nil { + // 代理处理任务发送失败，只记录日志，不影响报告流程 + logx.Errorf("发送代理处理任务失败，订单ID: %s, 错误: %v", order.Id, asyncErr) + } + + _, delErr := l.svcCtx.Redis.DelCtx(ctx, redisKey) + if delErr != nil { + logx.Errorf("删除Redis缓存失败，但任务已成功处理，订单ID: %s, 错误: %v", order.Id, delErr) + } + + return nil +} + +// 定义一个中间件函数 +func (l *PaySuccessNotifyUserHandler) handleError(ctx context.Context, err error, order *model.Order, query *model.Query) error { + logx.Errorf("处理任务失败，原因: %v", err) + + redisKey := fmt.Sprintf(types.QueryCacheKey, order.UserId, order.OrderNo) + _, delErr := l.svcCtx.Redis.DelCtx(ctx, redisKey) + if delErr != nil { + logx.Errorf("删除Redis缓存失败，订单ID: %s, 错误: %v", order.Id, delErr) + } + + if order.Status == "paid" && query.QueryState == "pending" { + // 更新查询状态为失败 + query.QueryState = "failed" + updateQueryErr := l.svcCtx.QueryModel.UpdateWithVersion(ctx, nil, query) + if updateQueryErr != nil { + logx.Errorf("更新查询状态失败，订单ID: %d, 错误: %v", order.Id, updateQueryErr) + return asynq.SkipRetry + } + + // 退款 + if order.PaymentPlatform == "wechat" { + refundErr := l.svcCtx.WechatPayService.WeChatRefund(ctx, order.OrderNo, order.Amount, order.Amount) + if refundErr != nil { + logx.Error(refundErr) + return asynq.SkipRetry + } + } else { + refund, refundErr := l.svcCtx.AlipayService.AliRefund(ctx, order.OrderNo, order.Amount) + if refundErr != nil { + logx.Error(refundErr) + return asynq.SkipRetry + } + if refund.IsSuccess() { + logx.Errorf("支付宝退款成功, orderID: %s", order.Id) + // 更新订单状态为退款 + order.Status = "refunded" + updateOrderErr := l.svcCtx.OrderModel.UpdateWithVersion(ctx, nil, order) + if updateOrderErr != nil { + logx.Errorf("更新订单状态失败，订单ID: %s, 错误: %v", order.Id, updateOrderErr) + return fmt.Errorf("更新订单状态失败: %v", updateOrderErr) + } + return asynq.SkipRetry + } else { + logx.Errorf("支付宝退款失败:%v", refundErr) + return asynq.SkipRetry + } + // 直接成功 + } + + } + + return asynq.SkipRetry +} + +// desensitizeParams 对敏感数据进行脱敏处理 +func (l *PaySuccessNotifyUserHandler) desensitizeParams(data []byte) ([]byte, error) { + // 解析JSON数据到map + var paramsMap map[string]interface{} + if err := json.Unmarshal(data, ¶msMap); err != nil { + return nil, fmt.Errorf("解析JSON数据失败: %v", err) + } + + // 处理可能包含敏感信息的字段 + for key, value := range paramsMap { + if strValue, ok := value.(string); ok { + // 根据字段名和内容判断并脱敏 + if isNameField(key) && len(strValue) > 0 { + // 姓名脱敏 + paramsMap[key] = maskName(strValue) + } else if isIDCardField(key) && len(strValue) > 10 { + // 身份证号脱敏 + paramsMap[key] = maskIDCard(strValue) + } else if isPhoneField(key) && len(strValue) >= 8 { + // 手机号脱敏 + paramsMap[key] = maskPhone(strValue) + } else if len(strValue) > 3 { + // 其他所有未匹配的字段都进行通用脱敏 + paramsMap[key] = maskGeneral(strValue) + } + } else if mapValue, ok := value.(map[string]interface{}); ok { + // 递归处理嵌套的map + for subKey, subValue := range mapValue { + if subStrValue, ok := subValue.(string); ok { + if isNameField(subKey) && len(subStrValue) > 0 { + mapValue[subKey] = maskName(subStrValue) + } else if isIDCardField(subKey) && len(subStrValue) > 10 { + mapValue[subKey] = maskIDCard(subStrValue) + } else if isPhoneField(subKey) && len(subStrValue) >= 8 { + mapValue[subKey] = maskPhone(subStrValue) + } else if len(subStrValue) > 3 { + // 其他所有未匹配的字段都进行通用脱敏 + mapValue[subKey] = maskGeneral(subStrValue) + } + } + } + } + } + + // 将处理后的map重新序列化为JSON + return json.Marshal(paramsMap) +} + +// 判断是否为姓名字段 +func isNameField(key string) bool { + key = strings.ToLower(key) + return strings.Contains(key, "name") || strings.Contains(key, "姓名") || + strings.Contains(key, "owner") || strings.Contains(key, "main") +} + +// 判断是否为身份证字段 +func isIDCardField(key string) bool { + key = strings.ToLower(key) + return strings.Contains(key, "idcard") || strings.Contains(key, "id_card") || + strings.Contains(key, "身份证") || strings.Contains(key, "证件号") +} + +// 判断是否为手机号字段 +func isPhoneField(key string) bool { + key = strings.ToLower(key) + return strings.Contains(key, "phone") || strings.Contains(key, "mobile") || + strings.Contains(key, "手机") || strings.Contains(key, "电话") +} + +// 判断是否包含敏感数据模式 +func containsSensitivePattern(value string) bool { + // 检查是否包含连续的数字或字母模式 + numPattern := regexp.MustCompile(`\d{6,}`) + return numPattern.MatchString(value) +} + +// 姓名脱敏 +func maskName(name string) string { + // 将字符串转换为rune切片以正确处理中文字符 + runes := []rune(name) + length := len(runes) + + if length <= 1 { + return name + } + + if length == 2 { + // 两个字：保留第一个字，第二个字用*替代 + return string(runes[0]) + "*" + } + + // 三个字及以上：保留首尾字，中间用*替代 + first := string(runes[0]) + last := string(runes[length-1]) + mask := strings.Repeat("*", length-2) + + return first + mask + last +} + +// 身份证号脱敏 +func maskIDCard(idCard string) string { + length := len(idCard) + if length <= 10 { + return idCard // 如果长度太短，可能不是身份证，不处理 + } + // 保留前3位和后4位 + return idCard[:3] + strings.Repeat("*", length-7) + idCard[length-4:] +} + +// 手机号脱敏 +func maskPhone(phone string) string { + length := len(phone) + if length < 8 { + return phone // 如果长度太短，可能不是手机号，不处理 + } + // 保留前3位和后4位 + return phone[:3] + strings.Repeat("*", length-7) + phone[length-4:] +} + +// 通用敏感信息脱敏 - 根据字符串长度比例进行脱敏 +func maskGeneral(value string) string { + length := len(value) + + // 小于3个字符的不脱敏 + if length <= 3 { + return value + } + + // 根据字符串长度计算保留字符数 + var prefixLen, suffixLen int + + switch { + case length <= 6: // 短字符串 + // 保留首尾各1个字符 + prefixLen, suffixLen = 1, 1 + case length <= 10: // 中等长度字符串 + // 保留首部30%和尾部20%的字符 + prefixLen = int(float64(length) * 0.3) + suffixLen = int(float64(length) * 0.2) + case length <= 20: // 较长字符串 + // 保留首部25%和尾部15%的字符 + prefixLen = int(float64(length) * 0.25) + suffixLen = int(float64(length) * 0.15) + default: // 非常长的字符串 + // 保留首部20%和尾部10%的字符 + prefixLen = int(float64(length) * 0.2) + suffixLen = int(float64(length) * 0.1) + } + + // 确保至少有一个字符被保留 + if prefixLen < 1 { + prefixLen = 1 + } + if suffixLen < 1 { + suffixLen = 1 + } + + // 确保前缀和后缀总长不超过总长度的80% + if prefixLen+suffixLen > int(float64(length)*0.8) { + // 调整为总长度的80% + totalVisible := int(float64(length) * 0.8) + // 前缀占60%，后缀占40% + prefixLen = int(float64(totalVisible) * 0.6) + suffixLen = totalVisible - prefixLen + } + + // 创建脱敏后的字符串 + prefix := value[:prefixLen] + suffix := value[length-suffixLen:] + masked := strings.Repeat("*", length-prefixLen-suffixLen) + + return prefix + masked + suffix +} diff --git a/app/main/api/internal/queue/routes.go b/app/main/api/internal/queue/routes.go new file mode 100644 index 0000000..e926188 --- /dev/null +++ b/app/main/api/internal/queue/routes.go @@ -0,0 +1,53 @@ +package queue + +import ( + "context" + "fmt" + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/hibiken/asynq" +) + +type CronJob struct { + ctx context.Context + svcCtx *svc.ServiceContext +} + +func NewCronJob(ctx context.Context, svcCtx *svc.ServiceContext) *CronJob { + return &CronJob{ + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *CronJob) Register() *asynq.ServeMux { + redisClientOpt := asynq.RedisClientOpt{Addr: l.svcCtx.Config.CacheRedis[0].Host, Password: l.svcCtx.Config.CacheRedis[0].Pass} + scheduler := asynq.NewScheduler(redisClientOpt, nil) + + // 注册数据清理定时任务（每天凌晨3点） + task := asynq.NewTask(types.MsgCleanQueryData, nil, nil) + _, err := scheduler.Register(TASKTIME, task) + if err != nil { + panic(fmt.Sprintf("定时任务注册失败:%v", err)) + } + + // 注册解冻佣金扫描定时任务（每2小时执行一次） + unfreezeScanTask := asynq.NewTask(types.MsgUnfreezeCommissionScan, nil, nil) + _, err = scheduler.Register("0 */2 * * *", unfreezeScanTask) // 每2小时执行一次（每小时的第0分钟） + if err != nil { + panic(fmt.Sprintf("解冻佣金扫描定时任务注册失败:%v", err)) + } + + scheduler.Start() + fmt.Println("定时任务启动！！！") + + mux := asynq.NewServeMux() + mux.Handle(types.MsgPaySuccessQuery, NewPaySuccessNotifyUserHandler(l.svcCtx)) + mux.Handle(types.MsgCleanQueryData, NewCleanQueryDataHandler(l.svcCtx)) + mux.Handle(types.MsgAgentProcess, NewAgentProcessHandler(l.svcCtx)) + mux.Handle(types.MsgUnfreezeCommission, NewUnfreezeCommissionHandler(l.svcCtx)) + mux.Handle(types.MsgUnfreezeCommissionScan, NewUnfreezeCommissionScanHandler(l.svcCtx)) + + return mux +} diff --git a/app/main/api/internal/queue/unfreezeCommission.go b/app/main/api/internal/queue/unfreezeCommission.go new file mode 100644 index 0000000..273ff96 --- /dev/null +++ b/app/main/api/internal/queue/unfreezeCommission.go @@ -0,0 +1,94 @@ +package queue + +import ( + "context" + "encoding/json" + "errors" + "fmt" + "time" + "qnc-server/app/main/model" + "qnc-server/pkg/lzkit/lzUtils" + + "qnc-server/app/main/api/internal/svc" + "qnc-server/app/main/api/internal/types" + + "github.com/hibiken/asynq" + pkgerrors "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type UnfreezeCommissionHandler struct { + svcCtx *svc.ServiceContext +} + +func NewUnfreezeCommissionHandler(svcCtx *svc.ServiceContext) *UnfreezeCommissionHandler { + return &UnfreezeCommissionHandler{ + svcCtx: svcCtx, + } +} + +func (l *UnfreezeCommissionHandler) ProcessTask(ctx context.Context, t *asynq.Task) error { + var payload types.MsgUnfreezeCommissionPayload + if err := json.Unmarshal(t.Payload(), &payload); err != nil { + return fmt.Errorf("解析解冻任务负载失败: %w", err) + } + + // 1. 查询冻结任务 + freezeTask, err := l.svcCtx.AgentFreezeTaskModel.FindOne(ctx, payload.FreezeTaskId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + logx.Errorf("解冻任务失败，冻结任务不存在: freezeTaskId=%s", payload.FreezeTaskId) + return asynq.SkipRetry // 任务不存在，跳过重试 + } + return fmt.Errorf("查询冻结任务失败: freezeTaskId=%s, err=%w", payload.FreezeTaskId, err) + } + + // 2. 检查任务状态 + if freezeTask.Status != 1 { + logx.Infof("解冻任务跳过，任务已处理: freezeTaskId=%s, status=%d", payload.FreezeTaskId, freezeTask.Status) + return nil // 任务已处理，不重试 + } + + // 3. 检查解冻时间是否已到 + if time.Now().Before(freezeTask.UnfreezeTime) { + logx.Infof("解冻任务跳过，未到解冻时间: freezeTaskId=%s, unfreezeTime=%v", payload.FreezeTaskId, freezeTask.UnfreezeTime) + // 重新发送延迟任务 + if err := l.svcCtx.AsynqService.SendUnfreezeTask(payload.FreezeTaskId, freezeTask.UnfreezeTime); err != nil { + logx.Errorf("重新发送解冻任务失败: freezeTaskId=%s, err=%v", payload.FreezeTaskId, err) + } + return nil + } + + // 4. 使用事务处理解冻 + err = l.svcCtx.AgentFreezeTaskModel.Trans(ctx, func(transCtx context.Context, session sqlx.Session) error { + // 4.1 更新冻结任务状态 + freezeTask.Status = 2 // 已解冻 + freezeTask.ActualUnfreezeTime = lzUtils.TimeToNullTime(time.Now()) + if updateErr := l.svcCtx.AgentFreezeTaskModel.UpdateWithVersion(transCtx, session, freezeTask); updateErr != nil { + return pkgerrors.Wrapf(updateErr, "更新冻结任务状态失败") + } + + // 4.2 更新钱包（解冻余额） + wallet, walletErr := l.svcCtx.AgentWalletModel.FindOneByAgentId(transCtx, freezeTask.AgentId) + if walletErr != nil { + return pkgerrors.Wrapf(walletErr, "查询钱包失败, agentId: %s", freezeTask.AgentId) + } + + wallet.FrozenBalance -= freezeTask.FreezeAmount + wallet.Balance += freezeTask.FreezeAmount + if updateWalletErr := l.svcCtx.AgentWalletModel.UpdateWithVersion(transCtx, session, wallet); updateWalletErr != nil { + return pkgerrors.Wrapf(updateWalletErr, "更新钱包失败") + } + + return nil + }) + + if err != nil { + logx.Errorf("解冻任务处理失败: freezeTaskId=%s, err=%v", payload.FreezeTaskId, err) + return fmt.Errorf("解冻任务处理失败: freezeTaskId=%s, err=%w", payload.FreezeTaskId, err) + } + + logx.Infof("解冻任务处理成功: freezeTaskId=%s, agentId=%s, amount=%.2f", payload.FreezeTaskId, freezeTask.AgentId, freezeTask.FreezeAmount) + return nil +} diff --git a/app/main/api/internal/queue/unfreezeCommissionScan.go b/app/main/api/internal/queue/unfreezeCommissionScan.go new file mode 100644 index 0000000..4702c75 --- /dev/null +++ b/app/main/api/internal/queue/unfreezeCommissionScan.go @@ -0,0 +1,233 @@ +package queue + +import ( + "context" + "strings" + "sync" + "time" + "qnc-server/app/main/model" + "qnc-server/pkg/lzkit/lzUtils" + + "qnc-server/app/main/api/internal/svc" + + "github.com/hibiken/asynq" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +// UnfreezeCommissionScanHandler 定时扫描解冻任务处理器 +type UnfreezeCommissionScanHandler struct { + svcCtx *svc.ServiceContext +} + +func NewUnfreezeCommissionScanHandler(svcCtx *svc.ServiceContext) *UnfreezeCommissionScanHandler { + return &UnfreezeCommissionScanHandler{ + svcCtx: svcCtx, + } +} + +// ProcessTask 定时扫描需要解冻的任务 +func (l *UnfreezeCommissionScanHandler) ProcessTask(ctx context.Context, t *asynq.Task) error { + scanStartTime := time.Now() + now := time.Now() + logx.Infof("开始扫描需要解冻的佣金任务，当前时间: %v", now) + + // 1. 查询所有待解冻且解冻时间已到的任务 + // 使用索引 idx_status 和 idx_unfreeze_time 优化查询 + // 不限制查询数量，找到所有需要解冻的任务 + builder := l.svcCtx.AgentFreezeTaskModel.SelectBuilder(). + Where("status = ? AND unfreeze_time <= ? AND del_state = ?", 1, now, 0). // 1=待解冻，0=未删除 + OrderBy("unfreeze_time ASC") // 按解冻时间升序，优先处理最早的任务 + + freezeTasks, err := l.svcCtx.AgentFreezeTaskModel.FindAll(ctx, builder, "") + if err != nil { + logx.Errorf("查询待解冻任务失败: %v", err) + return errors.Wrapf(err, "查询待解冻任务失败") + } + + // 如果没有需要解冻的任务，直接返回（不创建任何记录，只记录日志） + if len(freezeTasks) == 0 { + scanDuration := time.Since(scanStartTime) + logx.Infof("没有需要解冻的任务，扫描耗时: %v", scanDuration) + return nil + } + + // 2. 批次大小限制：如果任务量过大，分批处理 + const maxBatchSize = 1000 + originalCount := len(freezeTasks) + if len(freezeTasks) > maxBatchSize { + logx.Errorf("任务数量过多(%d)，本次只处理前%d个，剩余将在下次扫描处理", len(freezeTasks), maxBatchSize) + freezeTasks = freezeTasks[:maxBatchSize] + } + + logx.Infof("找到 %d 个需要解冻的任务（原始数量: %d），开始处理（最多同时处理2个）", len(freezeTasks), originalCount) + + // 3. 并发控制：使用信号量限制最多同时处理2个任务 + const maxConcurrency = 2 // 最多同时处理2个任务 + const taskTimeout = 30 * time.Second // 每个任务30秒超时 + semaphore := make(chan struct{}, maxConcurrency) // 信号量通道 + var wg sync.WaitGroup + var mu sync.Mutex // 保护计数器的互斥锁 + successCount := 0 + failCount := 0 + skipCount := 0 // 跳过的任务数（已处理、时间未到等） + + // 4. 并发处理所有任务，但最多同时处理2个 + for _, freezeTask := range freezeTasks { + // 检查是否被取消（优雅关闭支持） + select { + case <-ctx.Done(): + logx.Infof("扫描任务被取消，已处理: 成功=%d, 失败=%d, 跳过=%d", successCount, failCount, skipCount) + return ctx.Err() + default: + // 继续处理 + } + + wg.Add(1) + semaphore <- struct{}{} // 获取信号量，如果已满2个则阻塞 + + go func(task *model.AgentFreezeTask) { + defer wg.Done() + defer func() { <-semaphore }() // 释放信号量 + + taskStartTime := time.Now() + + // 为每个任务设置超时控制 + taskCtx, cancel := context.WithTimeout(ctx, taskTimeout) + defer cancel() + + // 使用事务处理每个任务，确保原子性 + err := l.svcCtx.AgentFreezeTaskModel.Trans(taskCtx, func(transCtx context.Context, session sqlx.Session) error { + // 4.1 重新查询任务（使用乐观锁，确保并发安全） + currentTask, err := l.svcCtx.AgentFreezeTaskModel.FindOne(transCtx, task.Id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + logx.Infof("冻结任务不存在，可能已被处理: freezeTaskId=%d", task.Id) + return nil // 任务不存在，跳过 + } + return errors.Wrapf(err, "查询冻结任务失败, freezeTaskId: %d", task.Id) + } + + // 4.2 幂等性增强：检查是否已经解冻过（通过 actual_unfreeze_time） + if currentTask.ActualUnfreezeTime.Valid { + logx.Infof("任务已解冻，跳过: freezeTaskId=%d, actualUnfreezeTime=%v", task.Id, currentTask.ActualUnfreezeTime.Time) + return nil // 已解冻，跳过 + } + + // 4.3 检查任务状态（双重检查，防止并发处理） + if currentTask.Status != 1 { + logx.Infof("冻结任务状态已变更，跳过处理: freezeTaskId=%d, status=%d", task.Id, currentTask.Status) + return nil // 状态已变更，跳过 + } + + // 4.4 再次检查解冻时间（防止时间判断误差） + nowTime := time.Now() + if nowTime.Before(currentTask.UnfreezeTime) { + logx.Infof("冻结任务解冻时间未到，跳过处理: freezeTaskId=%d, unfreezeTime=%v", task.Id, currentTask.UnfreezeTime) + return nil // 时间未到，跳过 + } + + // 4.5 计算延迟时间（便于监控） + delay := nowTime.Sub(currentTask.UnfreezeTime) + if delay > 1*time.Hour { + logx.Errorf("解冻任务延迟处理: freezeTaskId=%d, 延迟=%v, unfreezeTime=%v", task.Id, delay, currentTask.UnfreezeTime) + } + + // 4.6 更新冻结任务状态 + currentTask.Status = 2 // 已解冻 + currentTask.ActualUnfreezeTime = lzUtils.TimeToNullTime(nowTime) + if updateErr := l.svcCtx.AgentFreezeTaskModel.UpdateWithVersion(transCtx, session, currentTask); updateErr != nil { + return errors.Wrapf(updateErr, "更新冻结任务状态失败, freezeTaskId: %d", task.Id) + } + + // 4.7 更新钱包（解冻余额） + wallet, walletErr := l.svcCtx.AgentWalletModel.FindOneByAgentId(transCtx, currentTask.AgentId) + if walletErr != nil { + return errors.Wrapf(walletErr, "查询钱包失败, agentId: %d", currentTask.AgentId) + } + + // 检查冻结余额是否足够（防止数据异常） + if wallet.FrozenBalance < currentTask.FreezeAmount { + logx.Errorf("钱包冻结余额不足，数据异常: freezeTaskId=%d, agentId=%d, frozenBalance=%.2f, freezeAmount=%.2f", + task.Id, currentTask.AgentId, wallet.FrozenBalance, currentTask.FreezeAmount) + return errors.Errorf("钱包冻结余额不足: agentId=%d, frozenBalance=%.2f, freezeAmount=%.2f", + currentTask.AgentId, wallet.FrozenBalance, currentTask.FreezeAmount) + } + + wallet.FrozenBalance -= currentTask.FreezeAmount + wallet.Balance += currentTask.FreezeAmount + if updateWalletErr := l.svcCtx.AgentWalletModel.UpdateWithVersion(transCtx, session, wallet); updateWalletErr != nil { + return errors.Wrapf(updateWalletErr, "更新钱包失败, agentId: %d", currentTask.AgentId) + } + + // 更详细的日志（包含更多上下文信息） + logx.Infof("解冻任务处理成功: freezeTaskId=%d, agentId=%d, amount=%.2f, orderPrice=%.2f, freezeTime=%v, unfreezeTime=%v, delay=%v", + task.Id, currentTask.AgentId, currentTask.FreezeAmount, currentTask.OrderPrice, + currentTask.FreezeTime, currentTask.UnfreezeTime, delay) + return nil + }) + + // 记录处理时间 + taskDuration := time.Since(taskStartTime) + if taskDuration > 5*time.Second { + logx.Errorf("解冻任务处理耗时较长: freezeTaskId=%d, duration=%v", task.Id, taskDuration) + } + + // 更新计数器（需要加锁保护） + mu.Lock() + if err != nil { + // 错误分类处理 + if isTemporaryError(err) { + // 临时错误（如超时、网络问题），记录但继续处理其他任务 + failCount++ + logx.Errorf("解冻任务临时失败，将在下次扫描重试: freezeTaskId=%d, duration=%v, err=%v", task.Id, taskDuration, err) + } else { + // 永久错误（如数据异常），记录详细日志 + failCount++ + logx.Errorf("解冻任务永久失败: freezeTaskId=%d, duration=%v, err=%v", task.Id, taskDuration, err) + } + } else { + successCount++ + logx.Infof("解冻任务处理完成: freezeTaskId=%d, duration=%v", task.Id, taskDuration) + } + mu.Unlock() + + }(freezeTask) + } + + // 5. 等待所有任务完成 + wg.Wait() + + // 6. 记录扫描统计信息 + scanDuration := time.Since(scanStartTime) + logx.Infof("解冻任务扫描完成: 成功=%d, 失败=%d, 跳过=%d, 总计=%d, 扫描耗时=%v", + successCount, failCount, skipCount, len(freezeTasks), scanDuration) + + return nil +} + +// isTemporaryError 判断是否为临时错误（可以重试的错误） +func isTemporaryError(err error) bool { + if err == nil { + return false + } + + errStr := err.Error() + // 超时错误 + if errors.Is(err, context.DeadlineExceeded) || errors.Is(err, context.Canceled) { + return true + } + // 网络相关错误 + errStrLower := strings.ToLower(errStr) + if strings.Contains(errStrLower, "timeout") || strings.Contains(errStrLower, "connection") || strings.Contains(errStrLower, "network") { + return true + } + // 数据库连接错误 + if strings.Contains(errStrLower, "connection pool") || strings.Contains(errStrLower, "too many connections") { + return true + } + + // 其他错误视为永久错误（如数据异常、业务逻辑错误等） + return false +} diff --git a/app/main/api/internal/service/agentService.go b/app/main/api/internal/service/agentService.go new file mode 100644 index 0000000..300ca26 --- /dev/null +++ b/app/main/api/internal/service/agentService.go @@ -0,0 +1,1075 @@ +package service + +import ( + "context" + "database/sql" + "fmt" + "strconv" + "time" + "qnc-server/app/main/api/internal/config" + "qnc-server/app/main/model" + "qnc-server/common/globalkey" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +// AgentService 新代理系统服务 +type AgentService struct { + config config.Config + OrderModel model.OrderModel + AgentModel model.AgentModel + AgentWalletModel model.AgentWalletModel + AgentRelationModel model.AgentRelationModel + AgentLinkModel model.AgentLinkModel + AgentOrderModel model.AgentOrderModel + AgentCommissionModel model.AgentCommissionModel + AgentRebateModel model.AgentRebateModel + AgentUpgradeModel model.AgentUpgradeModel + AgentWithdrawalModel model.AgentWithdrawalModel + AgentConfigModel model.AgentConfigModel + AgentProductConfigModel model.AgentProductConfigModel + AgentRealNameModel model.AgentRealNameModel + AgentWithdrawalTaxModel model.AgentWithdrawalTaxModel + AgentFreezeTaskModel model.AgentFreezeTaskModel // 冻结任务模型（需要先运行SQL并生成model） +} + +// NewAgentService 创建新的代理服务 +func NewAgentService( + c config.Config, + orderModel model.OrderModel, + agentModel model.AgentModel, + agentWalletModel model.AgentWalletModel, + agentRelationModel model.AgentRelationModel, + agentLinkModel model.AgentLinkModel, + agentOrderModel model.AgentOrderModel, + agentCommissionModel model.AgentCommissionModel, + agentRebateModel model.AgentRebateModel, + agentUpgradeModel model.AgentUpgradeModel, + agentWithdrawalModel model.AgentWithdrawalModel, + agentConfigModel model.AgentConfigModel, + agentProductConfigModel model.AgentProductConfigModel, + agentRealNameModel model.AgentRealNameModel, + agentWithdrawalTaxModel model.AgentWithdrawalTaxModel, + agentFreezeTaskModel model.AgentFreezeTaskModel, // 冻结任务模型（需要先运行SQL并生成model） +) *AgentService { + return &AgentService{ + config: c, + OrderModel: orderModel, + AgentModel: agentModel, + AgentWalletModel: agentWalletModel, + AgentRelationModel: agentRelationModel, + AgentLinkModel: agentLinkModel, + AgentOrderModel: agentOrderModel, + AgentCommissionModel: agentCommissionModel, + AgentRebateModel: agentRebateModel, + AgentUpgradeModel: agentUpgradeModel, + AgentWithdrawalModel: agentWithdrawalModel, + AgentConfigModel: agentConfigModel, + AgentProductConfigModel: agentProductConfigModel, + AgentRealNameModel: agentRealNameModel, + AgentWithdrawalTaxModel: agentWithdrawalTaxModel, + AgentFreezeTaskModel: agentFreezeTaskModel, + } +} + +// AgentProcess 处理代理订单（新系统） +// 根据新代理系统的收益分配规则处理订单 +func (s *AgentService) AgentProcess(ctx context.Context, order *model.Order) error { + // 1. 检查是否是代理推广订单 + agentOrder, err := s.AgentOrderModel.FindOneByOrderId(ctx, order.Id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + // 不是代理订单，直接返回 + return nil + } + return errors.Wrapf(err, "查询代理订单失败, orderId: %d", order.Id) + } + + // 2. 检查订单是否已处理 + if agentOrder.ProcessStatus == 1 { + logx.Infof("订单已处理, orderId: %d", order.Id) + return nil + } + + // 3. 获取代理信息 + agent, err := s.AgentModel.FindOne(ctx, agentOrder.AgentId) + if err != nil { + return errors.Wrapf(err, "查询代理信息失败, agentId: %d", agentOrder.AgentId) + } + + // 4. 获取产品配置（必须存在） + productConfig, err := s.AgentProductConfigModel.FindOneByProductId(ctx, order.ProductId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "产品配置不存在, productId: %d，请先在后台配置产品价格参数", order.ProductId) + } + return errors.Wrapf(err, "查询产品配置失败, productId: %d", order.ProductId) + } + + // 5. 使用事务处理订单 + err = s.AgentWalletModel.Trans(ctx, func(transCtx context.Context, session sqlx.Session) error { + // 5.1 获取等级加成 + levelBonus, err := s.getLevelBonus(transCtx, agent.Level) + if err != nil { + return errors.Wrapf(err, "获取等级加成配置失败") + } + + // 5.2 使用产品配置的底价计算实际底价 + basePrice := productConfig.BasePrice + actualBasePrice := basePrice + float64(levelBonus) + + // 5.3 计算提价成本（使用产品配置） + priceThreshold := 0.0 + priceFeeRate := 0.0 + if productConfig.PriceThreshold.Valid { + priceThreshold = productConfig.PriceThreshold.Float64 + } + if productConfig.PriceFeeRate.Valid { + priceFeeRate = productConfig.PriceFeeRate.Float64 + } + priceCost := s.calculatePriceCost(agentOrder.SetPrice, priceThreshold, priceFeeRate) + + // 5.4 计算代理收益 + agentProfit := agentOrder.SetPrice - actualBasePrice - priceCost + + // 5.5 更新代理订单记录 + agentOrder.ProcessStatus = 1 + agentOrder.ProcessTime = lzUtils.TimeToNullTime(time.Now()) + agentOrder.ProcessRemark = lzUtils.StringToNullString("处理成功") + if err := s.AgentOrderModel.UpdateWithVersion(transCtx, session, agentOrder); err != nil { + return errors.Wrapf(err, "更新代理订单失败") + } + + // 5.6 发放代理佣金（传入订单单价用于冻结判断） + // 注意：冻结任务会在 agentProcess.go 中通过查询订单的冻结任务来发送解冻任务 + _, commissionErr := s.giveAgentCommission(transCtx, session, agentOrder.AgentId, order.Id, order.ProductId, agentProfit, agentOrder.SetPrice) + if commissionErr != nil { + return errors.Wrapf(commissionErr, "发放代理佣金失败") + } + + // 5.7 分配等级加成返佣给上级链 + if levelBonus > 0 { + if err := s.distributeLevelBonus(transCtx, session, agent, order.Id, order.ProductId, float64(levelBonus), levelBonus); err != nil { + return errors.Wrapf(err, "分配等级加成返佣失败") + } + } + + return nil + }) + + return err +} + +// getLevelBonus 获取等级加成（从配置表读取） +func (s *AgentService) getLevelBonus(ctx context.Context, level int64) (int64, error) { + var configKey string + switch level { + case 1: // 普通 + configKey = "level_1_bonus" + case 2: // 黄金 + configKey = "level_2_bonus" + case 3: // 钻石 + configKey = "level_3_bonus" + default: + return 0, nil + } + + bonus, err := s.getConfigFloat(ctx, configKey) + if err != nil { + // 配置不存在时返回默认值 + logx.Errorf("获取等级加成配置失败, level: %d, key: %s, err: %v，使用默认值", level, configKey, err) + switch level { + case 1: + return 6, nil + case 2: + return 3, nil + case 3: + return 0, nil + } + return 0, nil + } + return int64(bonus), nil +} + +// calculatePriceCost 计算提价成本 +func (s *AgentService) calculatePriceCost(setPrice, priceThreshold, priceFeeRate float64) float64 { + if setPrice <= priceThreshold { + return 0 + } + return (setPrice - priceThreshold) * priceFeeRate +} + +// giveAgentCommission 发放代理佣金 +// orderPrice: 订单单价，用于判断是否需要冻结 +// 返回：freezeTaskId（如果有冻结任务），error +func (s *AgentService) giveAgentCommission(ctx context.Context, session sqlx.Session, agentId, orderId, productId string, amount float64, orderPrice float64) (string, error) { + // 1. 创建佣金记录 + commission := &model.AgentCommission{ + Id: uuid.NewString(), + AgentId: agentId, + OrderId: orderId, + ProductId: productId, + Amount: amount, + Status: 1, // 已发放 + } + _, err := s.AgentCommissionModel.Insert(ctx, session, commission) + if err != nil { + return "", errors.Wrapf(err, "创建佣金记录失败") + } + commissionId := commission.Id + + // 2. 判断是否需要冻结 + // 2.1 获取冻结阈值配置（默认100元） + freezeThreshold, err := s.getConfigFloat(ctx, "commission_freeze_threshold") + if err != nil { + // 配置不存在时使用默认值100元 + freezeThreshold = 100.0 + logx.Errorf("获取冻结阈值配置失败，使用默认值100元, orderId: %d, err: %v", orderId, err) + } + + // 2.2 判断订单单价是否达到冻结阈值 + freezeAmount := 0.0 + var freezeTaskId string = "" + if orderPrice >= freezeThreshold { + // 2.3 获取冻结比例配置（默认10%） + freezeRatio, err := s.getConfigFloat(ctx, "commission_freeze_ratio") + if err != nil { + // 配置不存在时使用默认值0.1（10%） + freezeRatio = 0.1 + logx.Errorf("获取冻结比例配置失败，使用默认值10%%, orderId: %d, err: %v", orderId, err) + } + + // 计算冻结金额：订单单价的10% + freezeAmountByPrice := orderPrice * freezeRatio + + // 冻结金额不能超过佣金金额 + if freezeAmountByPrice > amount { + freezeAmount = amount + } else { + freezeAmount = freezeAmountByPrice + } + + // 如果冻结金额大于0，创建冻结任务 + if freezeAmount > 0 { + // 2.4 获取解冻天数配置（默认30天，即1个月） + unfreezeDays, err := s.getConfigInt(ctx, "commission_freeze_days") + if err != nil { + // 配置不存在时使用默认值30天 + unfreezeDays = 30 + logx.Errorf("获取解冻天数配置失败，使用默认值30天, orderId: %d, err: %v", orderId, err) + } + // 计算解冻时间（从配置读取的天数后） + // 注意：配置只在创建任务时读取，已创建的任务不受后续配置修改影响 + unfreezeTime := time.Now().AddDate(0, 0, int(unfreezeDays)) + + // 创建冻结任务记录 + freezeTask := &model.AgentFreezeTask{ + Id: uuid.NewString(), + AgentId: agentId, + OrderId: orderId, + CommissionId: commissionId, + FreezeAmount: freezeAmount, + OrderPrice: orderPrice, + FreezeRatio: freezeRatio, + Status: 1, // 待解冻 + FreezeTime: time.Now(), + UnfreezeTime: unfreezeTime, + Remark: lzUtils.StringToNullString(fmt.Sprintf("订单单价%.2f元，冻结比例%.2f%%，解冻天数%d天", orderPrice, freezeRatio*100, unfreezeDays)), + } + _, err = s.AgentFreezeTaskModel.Insert(ctx, session, freezeTask) + if err != nil { + return "", errors.Wrapf(err, "创建冻结任务失败") + } + freezeTaskId = freezeTask.Id + } + } + + // 3. 更新钱包余额 + wallet, err := s.AgentWalletModel.FindOneByAgentId(ctx, agentId) + if err != nil { + return "", errors.Wrapf(err, "查询钱包失败, agentId: %s", agentId) + } + + // 实际到账金额 = 佣金金额 - 冻结金额 + actualAmount := amount - freezeAmount + + wallet.Balance += actualAmount + wallet.FrozenBalance += freezeAmount + wallet.TotalEarnings += amount // 累计收益包含冻结部分 + if err := s.AgentWalletModel.UpdateWithVersion(ctx, session, wallet); err != nil { + return "", errors.Wrapf(err, "更新钱包失败") + } + + return freezeTaskId, nil +} + +// distributeLevelBonus 分配等级加成返佣给上级链 +func (s *AgentService) distributeLevelBonus(ctx context.Context, session sqlx.Session, agent *model.Agent, orderId, productId string, levelBonus float64, levelBonusInt int64) error { + // 钻石代理：等级加成为0，无返佣分配 + if agent.Level == 3 { + return nil + } + + // 黄金代理：等级加成3元，全部给钻石上级 + if agent.Level == 2 { + diamondParent, err := s.findDiamondParent(ctx, agent.Id) + if err != nil { + return errors.Wrapf(err, "查找钻石上级失败") + } + if diamondParent != nil { + return s.giveRebate(ctx, session, diamondParent.Id, agent.Id, orderId, productId, levelBonus, levelBonusInt, 2) // 2=钻石上级返佣 + } + // 找不到钻石上级，返佣归平台（异常情况） + return nil + } + + // 普通代理：等级加成6元，按规则分配给上级链 + if agent.Level == 1 { + return s.distributeNormalAgentBonus(ctx, session, agent, orderId, productId, levelBonus, levelBonusInt) + } + + return nil +} + +// distributeNormalAgentBonus 普通代理的等级加成返佣分配 +// +// 功能说明：根据普通代理的直接上级等级，按照规则分配等级加成返佣 +// +// 参数说明： +// - amount: 等级加成总额（例如：6元） +// - levelBonusInt: 等级加成整数（用于记录） +// +// 分配规则总览： +// 1. 直接上级是钻石：等级加成全部给钻石 +// 2. 直接上级是黄金：一部分给黄金（配置：direct_parent_amount_gold，默认3元），剩余给钻石上级 +// 3. 直接上级是普通：一部分给直接上级（配置：direct_parent_amount_normal，默认2元），剩余给钻石/黄金上级 +// +// 覆盖的所有情况： +// +// 情况1：普通(推广人) -> 钻石(直接上级) +// => 全部给钻石 +// +// 情况2：普通(推广人) -> 黄金(直接上级) -> 钻石 +// => 一部分给黄金，剩余给钻石 +// +// 情况3：普通(推广人) -> 黄金(直接上级) -> 无钻石上级 +// => 一部分给黄金，剩余归平台 +// +// 情况4：普通(推广人) -> 普通(直接上级) -> 钻石 +// => 一部分给直接上级普通，剩余全部给钻石 +// +// 情况5：普通(推广人) -> 普通(直接上级) -> 黄金 -> 钻石 +// => 一部分给直接上级普通（例如2元），一部分给黄金（等级加成差减去给普通的，例如3-2=1元），剩余给钻石（例如3元） +// +// 情况6：普通(推广人) -> 普通(直接上级) -> 黄金（无钻石） +// => 一部分给直接上级普通，剩余一部分给黄金（最多3元），超出归平台 +// +// 情况7：普通(推广人) -> 普通(直接上级) -> 普通 -> 钻石 +// => 一部分给直接上级普通，剩余全部给钻石（跳过中间普通代理） +// +// 情况8：普通(推广人) -> 普通(直接上级) -> 普通 -> 黄金（无钻石） +// => 一部分给直接上级普通，剩余一部分给黄金（最多3元），超出归平台（跳过中间普通代理） +// +// 情况9：普通(推广人) -> 普通(直接上级) -> 普通 -> 普通...（全部是普通） +// => 一部分给直接上级普通，剩余归平台 +// +// 注意：findDiamondParent 和 findGoldParent 会自动跳过中间的所有普通代理， +// +// 直接向上查找到第一个钻石或黄金代理 +func (s *AgentService) distributeNormalAgentBonus(ctx context.Context, session sqlx.Session, agent *model.Agent, orderId, productId string, amount float64, levelBonusInt int64) error { + // 1. 查找直接上级 + parent, err := s.findDirectParent(ctx, agent.Id) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "查找直接上级失败") + } + + if parent == nil { + // 无上级，全部归平台 + return nil + } + + // 2. 根据直接上级等级分配 + switch parent.Level { + case 3: // 直接上级是钻石代理的情况 + // ========== 直接上级是钻石：等级加成全部给钻石上级 ========== + // 场景示例： + // - 普通(推广人) -> 钻石(直接上级)：等级加成6元全部给钻石 + // 说明：如果直接上级就是钻石，不需要再向上查找，全部返佣给直接上级钻石 + // rebateType = 2：表示钻石上级返佣 + return s.giveRebate(ctx, session, parent.Id, agent.Id, orderId, productId, amount, levelBonusInt, 2) + + case 2: // 直接上级是黄金代理的情况 + // ========== 步骤1：给直接上级黄金代理返佣 ========== + // 配置键：direct_parent_amount_gold（普通代理给直接上级黄金代理的返佣金额） + // 默认值：3.0元 + // 说明：这部分金额给直接上级黄金代理，剩余部分继续向上分配给钻石上级 + goldRebateAmount, err := s.getRebateConfigFloat(ctx, "direct_parent_amount_gold", 3.0) + if err != nil { + logx.Errorf("获取黄金返佣配置失败，使用默认值3元: %v", err) + goldRebateAmount = 3.0 // 配置读取失败时使用默认值3元 + } + + // 计算给直接上级黄金代理的返佣金额（不能超过总等级加成金额） + goldAmount := goldRebateAmount // 默认3元 + if goldAmount > amount { + // 如果配置金额大于等级加成总额，则只给总额（防止配置错误导致负数） + goldAmount = amount + } + + // 发放返佣给直接上级黄金代理 + // rebateType = 1：表示直接上级返佣 + if goldAmount > 0 { + if err := s.giveRebate(ctx, session, parent.Id, agent.Id, orderId, productId, goldAmount, levelBonusInt, 1); err != nil { + return errors.Wrapf(err, "给黄金上级返佣失败") + } + } + + // ========== 步骤2：计算剩余金额并分配给钻石上级 ========== + // 剩余金额 = 总等级加成 - 已给黄金上级的金额 + // 例如：等级加成6元 - 给黄金上级3元 = 剩余3元 + remaining := amount - goldAmount + if remaining > 0 { + // 从黄金上级开始向上查找钻石上级 + // 场景示例： + // - 普通(推广人) -> 黄金(直接上级) -> 钻石：剩余3元给钻石 + // - 普通(推广人) -> 黄金(直接上级) -> 普通 -> 钻石：剩余3元给钻石（跳过中间普通代理） + // - 普通(推广人) -> 黄金(直接上级) -> 无上级：剩余3元归平台（没有钻石上级） + diamondParent, err := s.findDiamondParent(ctx, parent.Id) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "查找钻石上级失败") + } + if diamondParent != nil { + // 找到钻石上级，剩余金额全部给钻石上级 + // rebateType = 2：表示钻石上级返佣 + return s.giveRebate(ctx, session, diamondParent.Id, agent.Id, orderId, productId, remaining, levelBonusInt, 2) + } + // 找不到钻石上级，剩余金额归平台（不需要记录） + // 例如：等级加成6元，给黄金3元，剩余3元但找不到钻石上级，则剩余3元归平台 + } + return nil + + case 1: // 直接上级是普通代理的情况 + // ========== 步骤1：给直接上级普通代理返佣 ========== + // 配置键：direct_parent_amount_normal（普通代理给直接上级普通代理的返佣金额） + // 默认值：2.0元 + // 说明：无论后续层级有多少普通代理，这部分金额只给推广人的直接上级 + normalRebateAmount, err := s.getRebateConfigFloat(ctx, "direct_parent_amount_normal", 2.0) + if err != nil { + logx.Errorf("获取普通返佣配置失败，使用默认值2元: %v", err) + normalRebateAmount = 2.0 // 配置读取失败时使用默认值2元 + } + + // 计算给直接上级的返佣金额（不能超过总等级加成金额） + directAmount := normalRebateAmount // 默认2元 + if directAmount > amount { + // 如果配置金额大于等级加成总额，则只给总额（防止配置错误导致负数） + directAmount = amount + } + + // 发放返佣给直接上级普通代理 + // rebateType = 1：表示直接上级返佣 + if directAmount > 0 { + if err := s.giveRebate(ctx, session, parent.Id, agent.Id, orderId, productId, directAmount, levelBonusInt, 1); err != nil { + return errors.Wrapf(err, "给直接上级返佣失败") + } + } + + // ========== 步骤2：计算剩余金额 ========== + // 剩余金额 = 总等级加成 - 已给直接上级的金额 + // 例如：等级加成6元 - 给直接上级2元 = 剩余4元 + remaining := amount - directAmount + if remaining <= 0 { + // 如果没有剩余，直接返回（所有金额已分配给直接上级） + return nil + } + + // ========== 步骤3：从直接上级开始向上查找钻石和黄金代理 ========== + // 注意：findDiamondParent 和 findGoldParent 会自动跳过中间的所有普通代理 + // 例如： + // - 普通 -> 普通 -> 普通 -> 钻石：会跳过中间的普通代理，直接找到钻石 + // - 普通 -> 普通 -> 黄金 -> 钻石：会找到钻石（优先级更高） + // - 普通 -> 黄金：会找到黄金 + diamondParent, _ := s.findDiamondParent(ctx, parent.Id) // 向上查找钻石上级（跳过所有普通和黄金） + goldParent, _ := s.findGoldParent(ctx, parent.Id) // 向上查找黄金上级（跳过所有普通） + + // ========== 步骤4：按优先级分配剩余金额 ========== + // 优先级规则： + // 1. 优先给钻石上级（如果存在） + // 2. 其次给黄金上级（如果钻石不存在） + // 3. 最后归平台（如果钻石和黄金都不存在） + + if diamondParent != nil { + // ========== 情况A：找到钻石上级 ========== + // 场景示例： + // - 普通(推广人) -> 普通(直接上级) -> 钻石：给普通2元，剩余4元给钻石 + // - 普通(推广人) -> 普通(直接上级) -> 黄金 -> 钻石：给普通2元，给黄金1元，剩余3元给钻石 + // - 普通(推广人) -> 普通(直接上级) -> 普通 -> 普通 -> 钻石：给普通2元，剩余4元给钻石（跳过中间普通代理） + // + // 分配规则：当有钻石上级时，需要给黄金上级一部分返佣 + // 1. 等级加成的差 = 普通等级加成 - 黄金等级加成（例如：6元 - 3元 = 3元） + // 2. 从这个差中，先给直接上级普通代理（已给，例如2元） + // 3. 差减去给普通的部分，剩余给黄金代理（例如：3元 - 2元 = 1元） + // 4. 最后剩余部分给钻石（例如：6元 - 2元 - 1元 = 3元） + + // 步骤A1：计算等级加成的差 + // 获取普通代理等级加成（当前代理的等级加成，即amount） + normalBonus := amount // 例如：6元 + // 获取黄金代理等级加成 + goldBonus, err := s.getLevelBonus(ctx, 2) // 黄金等级加成 + if err != nil { + logx.Errorf("获取黄金等级加成配置失败，使用默认值3元: %v", err) + goldBonus = 3 // 默认3元 + } + // 计算等级加成的差 + bonusDiff := normalBonus - float64(goldBonus) // 例如：6元 - 3元 = 3元 + + // 步骤A2：如果有黄金上级，给黄金上级一部分返佣 + // 规则说明： + // - 等级加成的差（bonusDiff）代表普通代理和黄金代理的等级加成差异 + // - 从这个差中，先分配给直接上级普通代理（directAmount，已分配） + // - 剩余的差分配给黄金代理：goldRebateAmount = bonusDiff - directAmount + // - 如果差小于等于已给普通的部分，则不给黄金（这种情况理论上不应该发生，因为差应该>=给普通的部分） + if goldParent != nil && bonusDiff > 0 { + // 计算给黄金上级的金额 = 等级加成差 - 已给普通上级的金额 + // 例如：等级加成差3元 - 已给普通2元 = 给黄金1元 + goldRebateAmount := bonusDiff - directAmount + + // 如果计算出的金额小于等于0，说明差已经被普通代理全部占用了，不给黄金 + // 例如：如果差是2元，已给普通2元，则 goldRebateAmount = 0，不给黄金 + if goldRebateAmount > 0 { + // 边界检查：确保不超过剩余金额（理论上应该不会超过，但保险起见） + if goldRebateAmount > remaining { + goldRebateAmount = remaining + } + + // 发放返佣给黄金上级 + // rebateType = 3：表示黄金上级返佣 + if err := s.giveRebate(ctx, session, goldParent.Id, agent.Id, orderId, productId, goldRebateAmount, levelBonusInt, 3); err != nil { + return errors.Wrapf(err, "给黄金上级返佣失败") + } + + // 更新剩余金额（用于后续分配给钻石） + // 例如：剩余4元 - 给黄金1元 = 剩余3元（给钻石） + remaining = remaining - goldRebateAmount + } + } + + // 步骤A3：剩余金额全部给钻石上级 + // rebateType = 2：表示钻石上级返佣 + if remaining > 0 { + return s.giveRebate(ctx, session, diamondParent.Id, agent.Id, orderId, productId, remaining, levelBonusInt, 2) + } + return nil + } else if goldParent != nil { + // ========== 情况B：没有钻石上级，但有黄金上级 ========== + // 场景示例： + // - 普通(推广人) -> 普通(直接上级) -> 黄金（没有钻石）：给黄金一部分，剩余归平台 + // - 普通(推广人) -> 普通(直接上级) -> 普通 -> 黄金（没有钻石）：给黄金一部分，剩余归平台（跳过中间普通代理） + + // 配置键：max_gold_rebate_amount（普通代理给黄金上级的最大返佣金额） + // 默认值：3.0元 + // 说明：即使剩余金额超过这个值，也只能给黄金上级最多3元，超出部分归平台 + maxGoldRebate, err := s.getRebateConfigFloat(ctx, "max_gold_rebate_amount", 3.0) + if err != nil { + logx.Errorf("获取黄金最大返佣配置失败，使用默认值3元: %v", err) + maxGoldRebate = 3.0 // 配置读取失败时使用默认值3元 + } + + // 计算给黄金上级的返佣金额 + goldAmount := remaining // 剩余金额 + if goldAmount > maxGoldRebate { + // 如果剩余金额超过最大限额，则只给最大限额（例如：剩余4元，但最多只能给3元） + goldAmount = maxGoldRebate + } + // 例如：剩余4元，最大限额3元，则给黄金3元，剩余1元归平台 + + // 发放返佣给黄金上级 + // rebateType = 3：表示黄金上级返佣 + if goldAmount > 0 { + if err := s.giveRebate(ctx, session, goldParent.Id, agent.Id, orderId, productId, goldAmount, levelBonusInt, 3); err != nil { + return errors.Wrapf(err, "给黄金上级返佣失败") + } + } + // 超出最大限额的部分归平台（不需要记录） + // 例如：剩余4元，给黄金3元，剩余1元归平台 + return nil + } + + // ========== 情况C：既没有钻石上级，也没有黄金上级 ========== + // 场景示例： + // - 普通(推广人) -> 普通(直接上级) -> 普通 -> 普通...（整个链路都是普通代理） + // - 普通(推广人) -> 普通(直接上级) -> 无上级（已经是最顶层） + // 剩余金额全部归平台（不需要记录） + return nil + + default: + // 未知等级，全部归平台 + return nil + } +} + +// getRebateConfigFloat 获取返佣配置值（浮点数），如果配置不存在则返回默认值 +func (s *AgentService) getRebateConfigFloat(ctx context.Context, configKey string, defaultValue float64) (float64, error) { + config, err := s.AgentConfigModel.FindOneByConfigKey(ctx, configKey) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return defaultValue, nil + } + return defaultValue, err + } + value, err := strconv.ParseFloat(config.ConfigValue, 64) + if err != nil { + return defaultValue, errors.Wrapf(err, "解析配置值失败, key: %s, value: %s", configKey, config.ConfigValue) + } + return value, nil +} + +// giveRebate 发放返佣 +func (s *AgentService) giveRebate(ctx context.Context, session sqlx.Session, agentId, sourceAgentId, orderId, productId string, amount float64, levelBonus int64, rebateType int64) error { + // 1. 创建返佣记录 + rebate := &model.AgentRebate{ + Id: uuid.NewString(), + AgentId: agentId, + SourceAgentId: sourceAgentId, + OrderId: orderId, + ProductId: productId, + RebateType: rebateType, + LevelBonus: float64(levelBonus), // 等级加成金额 + RebateAmount: amount, + Status: 1, // 已发放 + } + if _, err := s.AgentRebateModel.Insert(ctx, session, rebate); err != nil { + return errors.Wrapf(err, "创建返佣记录失败") + } + + // 2. 更新钱包余额 + wallet, err := s.AgentWalletModel.FindOneByAgentId(ctx, agentId) + if err != nil { + return errors.Wrapf(err, "查询钱包失败, agentId: %d", agentId) + } + + wallet.Balance += amount + wallet.TotalEarnings += amount + if err := s.AgentWalletModel.UpdateWithVersion(ctx, session, wallet); err != nil { + return errors.Wrapf(err, "更新钱包失败") + } + + return nil +} + +// FindDirectParent 查找直接上级（公开方法） +func (s *AgentService) FindDirectParent(ctx context.Context, agentId string) (*model.Agent, error) { + return s.findDirectParent(ctx, agentId) +} + +// findDirectParent 查找直接上级 +func (s *AgentService) findDirectParent(ctx context.Context, agentId string) (*model.Agent, error) { + // 查找关系类型为1（直接关系）的上级 + builder := s.AgentRelationModel.SelectBuilder() + builder = builder.Where("child_id = ? AND relation_type = ? AND del_state = ?", agentId, 1, globalkey.DelStateNo) + relations, err := s.AgentRelationModel.FindAll(ctx, builder, "") + if err != nil { + return nil, err + } + if len(relations) == 0 { + return nil, model.ErrNotFound + } + + // 返回第一个直接上级 + return s.AgentModel.FindOne(ctx, relations[0].ParentId) +} + +// findDiamondParent 向上查找钻石上级 +// +// 功能说明：从指定代理开始，向上逐级查找第一个钻石代理（Level == 3） +// +// 查找规则： +// - 自动跳过所有普通代理（Level == 1）和黄金代理（Level == 2） +// - 只返回第一个找到的钻石代理 +// - 如果没有找到钻石代理，返回 ErrNotFound +// +// 示例场景： +// - 普通 -> 普通 -> 钻石：会找到钻石（跳过中间的普通代理） +// - 普通 -> 黄金 -> 钻石：会找到钻石（跳过黄金代理） +// - 普通 -> 普通 -> 黄金：返回 ErrNotFound（没有钻石） +func (s *AgentService) findDiamondParent(ctx context.Context, agentId string) (*model.Agent, error) { + currentId := agentId + maxDepth := 100 // 防止无限循环 + depth := 0 + + for depth < maxDepth { + parent, err := s.findDirectParent(ctx, currentId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, model.ErrNotFound + } + return nil, err + } + + if parent.Level == 3 { // 钻石 + return parent, nil + } + + currentId = parent.Id + depth++ + } + + return nil, model.ErrNotFound +} + +// findGoldParent 向上查找黄金上级 +// +// 功能说明：从指定代理开始，向上逐级查找第一个黄金代理（Level == 2） +// +// 查找规则： +// - 自动跳过所有普通代理（Level == 1） +// - 如果先遇到钻石代理（Level == 3），不会跳过，但会继续查找黄金代理 +// 注意：在实际使用中，应该先调用 findDiamondParent，如果没找到钻石再调用此方法 +// - 只返回第一个找到的黄金代理 +// - 如果没有找到黄金代理，返回 ErrNotFound +// +// 示例场景： +// - 普通 -> 普通 -> 黄金：会找到黄金（跳过中间的普通代理） +// - 普通 -> 黄金：会找到黄金 +// - 普通 -> 普通 -> 钻石：返回 ErrNotFound（跳过钻石，继续查找黄金，但找不到） +func (s *AgentService) findGoldParent(ctx context.Context, agentId string) (*model.Agent, error) { + currentId := agentId + maxDepth := 100 // 防止无限循环 + depth := 0 + + for depth < maxDepth { + parent, err := s.findDirectParent(ctx, currentId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return nil, model.ErrNotFound + } + return nil, err + } + + if parent.Level == 2 { // 黄金 + return parent, nil + } + + currentId = parent.Id + depth++ + } + + return nil, model.ErrNotFound +} + +// getConfigFloat 获取配置值（浮点数） +func (s *AgentService) getConfigFloat(ctx context.Context, configKey string) (float64, error) { + config, err := s.AgentConfigModel.FindOneByConfigKey(ctx, configKey) + if err != nil { + return 0, err + } + value, err := strconv.ParseFloat(config.ConfigValue, 64) + if err != nil { + return 0, errors.Wrapf(err, "解析配置值失败, key: %s, value: %s", configKey, config.ConfigValue) + } + return value, nil +} + +// getConfigInt 获取配置值（整数） +func (s *AgentService) getConfigInt(ctx context.Context, configKey string) (int64, error) { + config, err := s.AgentConfigModel.FindOneByConfigKey(ctx, configKey) + if err != nil { + return 0, err + } + value, err := strconv.ParseInt(config.ConfigValue, 10, 64) + if err != nil { + return 0, errors.Wrapf(err, "解析配置值失败, key: %s, value: %s", configKey, config.ConfigValue) + } + return value, nil +} + +// ProcessUpgrade 处理代理升级 +func (s *AgentService) ProcessUpgrade(ctx context.Context, agentId string, toLevel int64, upgradeType int64, upgradeFee, rebateAmount float64, orderNo string, operatorAgentId string) error { + return s.AgentWalletModel.Trans(ctx, func(transCtx context.Context, session sqlx.Session) error { + // 1. 获取代理信息 + agent, err := s.AgentModel.FindOne(transCtx, agentId) + if err != nil { + return errors.Wrapf(err, "查询代理信息失败, agentId: %d", agentId) + } + + // 2. 如果是自主付费升级，处理返佣 + if upgradeType == 1 { // 自主付费 + // 查找原直接上级 + parent, err := s.findDirectParent(transCtx, agentId) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "查找直接上级失败") + } + + if parent != nil && rebateAmount > 0 { + // 返佣给原直接上级 + if err := s.giveRebateForUpgrade(transCtx, session, parent.Id, agentId, rebateAmount, orderNo); err != nil { + return errors.Wrapf(err, "返佣给上级失败") + } + } + } + + // 3. 执行升级操作 + agent.Level = toLevel + + // 4. 检查是否需要脱离直接上级关系 + needDetach, err := s.needDetachFromParent(transCtx, agent, toLevel) + if err != nil { + return errors.Wrapf(err, "检查是否需要脱离关系失败") + } + + if needDetach { + // 脱离前先获取原直接上级及其上级的信息（用于后续重新连接） + oldParent, oldParentErr := s.findDirectParent(transCtx, agentId) + var grandparentId string = "" + if oldParentErr == nil && oldParent != nil { + // 查找原上级的上级 + grandparent, grandparentErr := s.findDirectParent(transCtx, oldParent.Id) + if grandparentErr == nil && grandparent != nil { + grandparentId = grandparent.Id + } + } + + // 脱离直接上级关系 + if err := s.detachFromParent(transCtx, session, agentId); err != nil { + return errors.Wrapf(err, "脱离直接上级关系失败") + } + + // 脱离后，尝试连接到原上级的上级 + if grandparentId != "" { + if err := s.reconnectToGrandparent(transCtx, session, agentId, toLevel, grandparentId); err != nil { + return errors.Wrapf(err, "重新连接上级关系失败") + } + } + } + + // 5. 如果升级为钻石，独立成新团队 + if toLevel == 3 { + agent.TeamLeaderId = sql.NullString{String: agentId, Valid: true} + // 更新所有下级的团队首领 + if err := s.updateChildrenTeamLeader(transCtx, session, agentId, agentId); err != nil { + return errors.Wrapf(err, "更新下级团队首领失败") + } + } else { + // 更新团队首领（查找上级链中的钻石代理） + teamLeaderId, err := s.findTeamLeaderId(transCtx, agentId) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return errors.Wrapf(err, "查找团队首领失败") + } + if teamLeaderId != "" { + agent.TeamLeaderId = sql.NullString{String: teamLeaderId, Valid: true} + } + } + + // 6. 更新代理记录 + if err := s.AgentModel.UpdateWithVersion(transCtx, session, agent); err != nil { + return errors.Wrapf(err, "更新代理记录失败") + } + + // 7. 更新升级记录状态 + // 这里需要先查询升级记录，暂时先跳过，在logic中处理 + + return nil + }) +} + +// needDetachFromParent 检查是否需要脱离直接上级关系 +func (s *AgentService) needDetachFromParent(ctx context.Context, agent *model.Agent, newLevel int64) (bool, error) { + parent, err := s.findDirectParent(ctx, agent.Id) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return false, nil // 没有上级，不需要脱离 + } + return false, err + } + + // 规则1：下级不能比上级等级高 + if newLevel > parent.Level { + return true, nil + } + + // 规则2：同级不能作为上下级（除了普通代理） + if newLevel == parent.Level { + if newLevel == 2 || newLevel == 3 { // 黄金或钻石同级需要脱离 + return true, nil + } + // 普通代理同级（newLevel == 1 && parent.Level == 1）不需要脱离 + } + + return false, nil +} + +// detachFromParent 脱离直接上级关系 +func (s *AgentService) detachFromParent(ctx context.Context, session sqlx.Session, agentId string) error { + // 查找直接关系 + builder := s.AgentRelationModel.SelectBuilder(). + Where("child_id = ? AND relation_type = ? AND del_state = ?", agentId, 1, globalkey.DelStateNo) + relations, err := s.AgentRelationModel.FindAll(ctx, builder, "") + if err != nil { + return err + } + + if len(relations) == 0 { + return nil // 没有关系，不需要脱离 + } + + // 将直接关系标记为已脱离 + relation := relations[0] + relation.RelationType = 2 // 已脱离 + relation.DetachReason = lzUtils.StringToNullString("upgrade") + relation.DetachTime = lzUtils.TimeToNullTime(time.Now()) + + if err := s.AgentRelationModel.UpdateWithVersion(ctx, session, relation); err != nil { + return errors.Wrapf(err, "更新关系记录失败") + } + + return nil +} + +// reconnectToGrandparent 重新连接到原上级的上级（如果存在且符合条件） +func (s *AgentService) reconnectToGrandparent(ctx context.Context, session sqlx.Session, agentId string, newLevel int64, grandparentId string) error { + // 获取原上级的上级信息 + grandparent, err := s.AgentModel.FindOne(ctx, grandparentId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + // 原上级的上级不存在，不需要重新连接 + return nil + } + return errors.Wrapf(err, "查询原上级的上级失败") + } + + // 验证是否可以连接到原上级的上级 + // 规则：新等级必须低于或等于原上级的上级等级，且不能同级（除了普通代理） + if newLevel > grandparent.Level { + // 新等级高于原上级的上级，不能连接 + return nil + } + + // 同级不能作为上下级（除了普通代理） + if newLevel == grandparent.Level { + if newLevel == 2 || newLevel == 3 { + // 黄金或钻石同级不能连接 + return nil + } + // 普通代理同级可以连接（虽然这种情况不太可能发生） + } + + // 检查是否已经存在关系 + builder := s.AgentRelationModel.SelectBuilder(). + Where("parent_id = ? AND child_id = ? AND relation_type = ? AND del_state = ?", grandparent.Id, agentId, 1, globalkey.DelStateNo) + existingRelations, err := s.AgentRelationModel.FindAll(ctx, builder, "") + if err != nil { + return errors.Wrapf(err, "查询现有关系失败") + } + + if len(existingRelations) > 0 { + // 关系已存在，不需要重复创建 + return nil + } + + // 创建新的关系连接到原上级的上级 + relation := &model.AgentRelation{ + Id: uuid.NewString(), + ParentId: grandparent.Id, + ChildId: agentId, + RelationType: 1, // 直接关系 + } + if _, err := s.AgentRelationModel.Insert(ctx, session, relation); err != nil { + return errors.Wrapf(err, "创建新关系失败") + } + + return nil +} + +// updateChildrenTeamLeader 更新所有下级的团队首领 +func (s *AgentService) updateChildrenTeamLeader(ctx context.Context, session sqlx.Session, agentId, teamLeaderId string) error { + // 递归更新所有下级 + var updateChildren func(string) error + updateChildren = func(parentId string) error { + // 查找直接下级 + builder := s.AgentRelationModel.SelectBuilder(). + Where("parent_id = ? AND relation_type = ? AND del_state = ?", parentId, 1, globalkey.DelStateNo) + relations, err := s.AgentRelationModel.FindAll(ctx, builder, "") + if err != nil { + return err + } + + for _, relation := range relations { + child, err := s.AgentModel.FindOne(ctx, relation.ChildId) + if err != nil { + continue + } + + child.TeamLeaderId = sql.NullString{String: teamLeaderId, Valid: true} + if err := s.AgentModel.UpdateWithVersion(ctx, session, child); err != nil { + return errors.Wrapf(err, "更新下级团队首领失败, childId: %s", child.Id) + } + + // 递归更新下级的下级 + if err := updateChildren(child.Id); err != nil { + return err + } + } + + return nil + } + + return updateChildren(agentId) +} + +// findTeamLeaderId 查找团队首领ID（钻石代理） +func (s *AgentService) findTeamLeaderId(ctx context.Context, agentId string) (string, error) { + diamondParent, err := s.findDiamondParent(ctx, agentId) + if err != nil { + if errors.Is(err, model.ErrNotFound) { + return "", nil + } + return "", err + } + return diamondParent.Id, nil +} + +// giveRebateForUpgrade 发放升级返佣 +// 注意：升级返佣信息记录在 agent_upgrade 表中（rebate_agent_id 和 rebate_amount），不需要在 agent_rebate 表中创建记录 +func (s *AgentService) giveRebateForUpgrade(ctx context.Context, session sqlx.Session, parentAgentId, upgradeAgentId string, amount float64, orderNo string) error { + // 更新钱包余额 + wallet, err := s.AgentWalletModel.FindOneByAgentId(ctx, parentAgentId) + if err != nil { + return errors.Wrapf(err, "查询钱包失败, agentId: %s", parentAgentId) + } + + wallet.Balance += amount + wallet.TotalEarnings += amount + if err := s.AgentWalletModel.UpdateWithVersion(ctx, session, wallet); err != nil { + return errors.Wrapf(err, "更新钱包失败") + } + + return nil +} + +// GetUpgradeFee 获取升级费用 +func (s *AgentService) GetUpgradeFee(ctx context.Context, fromLevel, toLevel int64) (float64, error) { + if fromLevel == 1 && toLevel == 2 { + // 普通→黄金：从配置获取 + return s.getRebateConfigFloat(ctx, "upgrade_to_gold_fee", 199) + } else if toLevel == 3 { + // 升级为钻石：从配置获取 + return s.getRebateConfigFloat(ctx, "upgrade_to_diamond_fee", 980) + } + return 0, nil +} + +// GetUpgradeRebate 获取升级返佣金额 +func (s *AgentService) GetUpgradeRebate(ctx context.Context, fromLevel, toLevel int64) (float64, error) { + if fromLevel == 1 && toLevel == 2 { + // 普通→黄金返佣：从配置获取 + return s.getRebateConfigFloat(ctx, "upgrade_to_gold_rebate", 139) + } else if toLevel == 3 { + // 升级为钻石返佣：从配置获取 + return s.getRebateConfigFloat(ctx, "upgrade_to_diamond_rebate", 680) + } + return 0, nil +} diff --git a/app/main/api/internal/service/alipayService.go b/app/main/api/internal/service/alipayService.go new file mode 100644 index 0000000..92a2e32 --- /dev/null +++ b/app/main/api/internal/service/alipayService.go @@ -0,0 +1,258 @@ +package service + +import ( + "context" + "crypto/rand" + "encoding/hex" + "fmt" + "net/http" + "strconv" + "sync/atomic" + "time" + "qnc-server/app/main/api/internal/config" + "qnc-server/app/main/model" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/smartwalle/alipay/v3" +) + +type AliPayService struct { + config config.AlipayConfig + AlipayClient *alipay.Client +} + +// NewAliPayService 是一个构造函数，用于初始化 AliPayService +func NewAliPayService(c config.Config) *AliPayService { + client, err := alipay.New(c.Alipay.AppID, c.Alipay.PrivateKey, c.Alipay.IsProduction) + if err != nil { + panic(fmt.Sprintf("创建支付宝客户端失败: %v", err)) + } + // 加载支付宝公钥 + // err = client.LoadAliPayPublicKey(c.Alipay.AlipayPublicKey) + // if err != nil { + // panic(fmt.Sprintf("加载支付宝公钥失败: %v", err)) + // } + + // 加载证书 + if err = client.LoadAppCertPublicKeyFromFile(c.Alipay.AppCertPath); err != nil { + panic(fmt.Sprintf("加载应用公钥证书失败: %v", err)) + } + if err = client.LoadAlipayCertPublicKeyFromFile(c.Alipay.AlipayCertPath); err != nil { + panic(fmt.Sprintf("加载支付宝公钥证书失败: %v", err)) + } + if err = client.LoadAliPayRootCertFromFile(c.Alipay.AlipayRootCertPath); err != nil { + panic(fmt.Sprintf("加载根证书失败: %v", err)) + } + + return &AliPayService{ + config: c.Alipay, + AlipayClient: client, + } +} + +func (a *AliPayService) CreateAlipayAppOrder(amount float64, subject string, outTradeNo string) (string, error) { + client := a.AlipayClient + totalAmount := lzUtils.ToAlipayAmount(amount) + // 构造移动支付请求 + p := alipay.TradeAppPay{ + Trade: alipay.Trade{ + Subject: subject, + OutTradeNo: outTradeNo, + TotalAmount: totalAmount, + ProductCode: "QUICK_MSECURITY_PAY", // 移动端支付专用代码 + NotifyURL: a.config.NotifyUrl, // 异步回调通知地址 + }, + } + + // 获取APP支付字符串，这里会签名 + payStr, err := client.TradeAppPay(p) + if err != nil { + return "", fmt.Errorf("创建支付宝订单失败: %v", err) + } + + return payStr, nil +} + +// CreateAlipayH5Order 创建支付宝H5支付订单 +func (a *AliPayService) CreateAlipayH5Order(amount float64, subject string, outTradeNo string) (string, error) { + client := a.AlipayClient + totalAmount := lzUtils.ToAlipayAmount(amount) + // 构造H5支付请求 + p := alipay.TradeWapPay{ + Trade: alipay.Trade{ + Subject: subject, + OutTradeNo: outTradeNo, + TotalAmount: totalAmount, + ProductCode: "QUICK_WAP_PAY", // H5支付专用产品码 + NotifyURL: a.config.NotifyUrl, // 异步回调通知地址 + ReturnURL: a.config.ReturnURL, + }, + } + // 获取H5支付请求字符串，这里会签名 + payUrl, err := client.TradeWapPay(p) + if err != nil { + return "", fmt.Errorf("创建支付宝H5订单失败: %v", err) + } + + return payUrl.String(), nil +} + +// CreateAlipayOrder 根据平台类型创建支付宝支付订单 +func (a *AliPayService) CreateAlipayOrder(ctx context.Context, amount float64, subject string, outTradeNo string) (string, error) { + // 根据 ctx 中的 platform 判断平台 + platform, platformOk := ctx.Value("platform").(string) + if !platformOk { + return "", fmt.Errorf("无的支付平台: %s", platform) + } + switch platform { + case model.PlatformApp: + // 调用App支付的创建方法 + return a.CreateAlipayAppOrder(amount, subject, outTradeNo) + case model.PlatformH5: + // 调用H5支付的创建方法，并传入 returnUrl + return a.CreateAlipayH5Order(amount, subject, outTradeNo) + default: + return "", fmt.Errorf("不支持的支付平台: %s", platform) + } +} + +// AliRefund 发起支付宝退款 +func (a *AliPayService) AliRefund(ctx context.Context, outTradeNo string, refundAmount float64) (*alipay.TradeRefundRsp, error) { + refund := alipay.TradeRefund{ + OutTradeNo: outTradeNo, + RefundAmount: lzUtils.ToAlipayAmount(refundAmount), + OutRequestNo: fmt.Sprintf("refund-%s", outTradeNo), + } + + // 发起退款请求 + refundResp, err := a.AlipayClient.TradeRefund(ctx, refund) + if err != nil { + return nil, fmt.Errorf("支付宝退款请求错误：%v", err) + } + return refundResp, nil +} + +// HandleAliPaymentNotification 支付宝支付回调 +func (a *AliPayService) HandleAliPaymentNotification(r *http.Request) (*alipay.Notification, error) { + // 解析表单 + err := r.ParseForm() + if err != nil { + return nil, fmt.Errorf("解析请求表单失败：%v", err) + } + // 解析并验证通知，DecodeNotification 会自动验证签名 + notification, err := a.AlipayClient.DecodeNotification(r.Form) + if err != nil { + return nil, fmt.Errorf("验证签名失败: %v", err) + } + return notification, nil +} +func (a *AliPayService) QueryOrderStatus(ctx context.Context, outTradeNo string) (*alipay.TradeQueryRsp, error) { + queryRequest := alipay.TradeQuery{ + OutTradeNo: outTradeNo, + } + + // 发起查询请求 + resp, err := a.AlipayClient.TradeQuery(ctx, queryRequest) + if err != nil { + return nil, fmt.Errorf("查询支付宝订单失败: %v", err) + } + + // 返回交易状态 + if resp.IsSuccess() { + return resp, nil + } + + return nil, fmt.Errorf("查询支付宝订单失败: %v", resp.SubMsg) +} + +// 添加全局原子计数器 +var alipayOrderCounter uint32 = 0 + +// GenerateOutTradeNo 生成唯一订单号的函数 - 优化版本 +func (a *AliPayService) GenerateOutTradeNo() string { + + // 获取当前时间戳（毫秒级） + timestamp := time.Now().UnixMilli() + timeStr := strconv.FormatInt(timestamp, 10) + + // 原子递增计数器 + counter := atomic.AddUint32(&alipayOrderCounter, 1) + + // 生成4字节真随机数 + randomBytes := make([]byte, 4) + _, err := rand.Read(randomBytes) + if err != nil { + // 如果随机数生成失败，回退到使用时间纳秒数据 + randomBytes = []byte(strconv.FormatInt(time.Now().UnixNano()%1000000, 16)) + } + randomHex := hex.EncodeToString(randomBytes) + + // 组合所有部分: 前缀 + 时间戳 + 计数器 + 随机数 + orderNo := fmt.Sprintf("%s%06x%s", timeStr[:10], counter%0xFFFFFF, randomHex[:6]) + + // 确保长度不超过32字符（大多数支付平台的限制） + if len(orderNo) > 32 { + orderNo = orderNo[:32] + } + + return orderNo +} + +// AliTransfer 支付宝单笔转账到支付宝账户（提现功能） +func (a *AliPayService) AliTransfer( + ctx context.Context, + payeeAccount string, // 收款方支付宝账户 + payeeName string, // 收款方姓名 + amount float64, // 转账金额 + remark string, // 转账备注 + outBizNo string, // 商户转账唯一订单号（可使用GenerateOutTradeNo生成） +) (*alipay.FundTransUniTransferRsp, error) { + // 参数校验 + if payeeAccount == "" { + return nil, fmt.Errorf("收款账户不能为空") + } + if amount <= 0 { + return nil, fmt.Errorf("转账金额必须大于0") + } + + // 构造转账请求 + req := alipay.FundTransUniTransfer{ + OutBizNo: outBizNo, + TransAmount: lzUtils.ToAlipayAmount(amount), // 金额格式转换 + ProductCode: "TRANS_ACCOUNT_NO_PWD", // 单笔无密转账到支付宝账户 + BizScene: "DIRECT_TRANSFER", // 单笔转账 + OrderTitle: "账户提现", // 转账标题 + Remark: remark, + PayeeInfo: &alipay.PayeeInfo{ + Identity: payeeAccount, + IdentityType: "ALIPAY_LOGON_ID", // 根据账户类型选择： + Name: payeeName, + // ALIPAY_USER_ID/ALIPAY_LOGON_ID + }, + } + + // 执行转账请求 + transferRsp, err := a.AlipayClient.FundTransUniTransfer(ctx, req) + if err != nil { + return nil, fmt.Errorf("支付宝转账请求失败: %v", err) + } + + return transferRsp, nil +} +func (a *AliPayService) QueryTransferStatus( + ctx context.Context, + outBizNo string, +) (*alipay.FundTransOrderQueryRsp, error) { + req := alipay.FundTransOrderQuery{ + OutBizNo: outBizNo, + } + response, err := a.AlipayClient.FundTransOrderQuery(ctx, req) + if err != nil { + return nil, fmt.Errorf("支付宝接口调用失败: %v", err) + } + // 处理响应 + if response.Code.IsFailure() { + return nil, fmt.Errorf("支付宝返回错误: %s-%s", response.Code, response.Msg) + } + return response, nil +} diff --git a/app/main/api/internal/service/apiRegistryService.go b/app/main/api/internal/service/apiRegistryService.go new file mode 100644 index 0000000..45cea83 --- /dev/null +++ b/app/main/api/internal/service/apiRegistryService.go @@ -0,0 +1,224 @@ +package service + +import ( + "context" + "fmt" + "regexp" + "strings" + + "qnc-server/app/main/model" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/rest" + "github.com/google/uuid" +) + +type ApiRegistryService struct { + adminApiModel model.AdminApiModel +} + +func NewApiRegistryService(adminApiModel model.AdminApiModel) *ApiRegistryService { + return &ApiRegistryService{ + adminApiModel: adminApiModel, + } +} + +// RegisterAllApis 自动注册所有API到数据库 +func (s *ApiRegistryService) RegisterAllApis(ctx context.Context, routes []rest.Route) error { + logx.Infof("开始注册API，共 %d 个路由", len(routes)) + + registeredCount := 0 + skippedCount := 0 + + for _, route := range routes { + // 跳过不需要权限控制的API + if s.shouldSkipApi(route.Path) { + skippedCount++ + continue + } + + // 解析API信息 + apiInfo := s.parseRouteToApi(route) + + // 检查是否已存在 + existing, err := s.adminApiModel.FindOneByApiCode(ctx, apiInfo.ApiCode) + if err != nil && !errors.Is(err, model.ErrNotFound) { + logx.Errorf("查询API失败: %v, apiCode: %s", err, apiInfo.ApiCode) + continue + } + + // 如果不存在则插入 + if existing == nil { + _, err = s.adminApiModel.Insert(ctx, nil, apiInfo) + if err != nil { + logx.Errorf("插入API失败: %v, apiCode: %s", err, apiInfo.ApiCode) + continue + } + registeredCount++ + logx.Infof("注册API成功: %s %s", apiInfo.Method, apiInfo.Url) + } else { + // 如果存在但信息有变化，则更新 + if s.shouldUpdateApi(existing, apiInfo) { + existing.ApiName = apiInfo.ApiName + existing.Method = apiInfo.Method + existing.Url = apiInfo.Url + existing.Description = apiInfo.Description + + _, err = s.adminApiModel.Update(ctx, nil, existing) + if err != nil { + logx.Errorf("更新API失败: %v, apiCode: %s", err, apiInfo.ApiCode) + continue + } + logx.Infof("更新API成功: %s %s", apiInfo.Method, apiInfo.Url) + } + } + } + + logx.Infof("API注册完成，新增: %d, 跳过: %d", registeredCount, skippedCount) + return nil +} + +// shouldSkipApi 判断是否应该跳过此API +func (s *ApiRegistryService) shouldSkipApi(path string) bool { + // 跳过公开API + skipPaths := []string{ + "/api/v1/admin/auth/login", // 登录接口 + "/api/v1/app/", // 前端应用接口 + "/api/v1/agent/", // 代理接口 + "/api/v1/user/", // 用户接口 + "/api/v1/auth/", // 认证接口 + "/api/v1/notification/", // 通知接口 + "/api/v1/pay/", // 支付接口 + "/api/v1/query/", // 查询接口 + "/api/v1/product/", // 产品接口 + "/api/v1/authorization/", // 授权接口 + "/health", // 健康检查 + } + + for _, skipPath := range skipPaths { + if strings.HasPrefix(path, skipPath) { + return true + } + } + + return false +} + +// parseRouteToApi 将路由解析为API信息 +func (s *ApiRegistryService) parseRouteToApi(route rest.Route) *model.AdminApi { + // 生成API编码 + apiCode := s.generateApiCode(route.Method, route.Path) + + // 生成API名称 + apiName := s.generateApiName(route.Path) + + // 生成描述 + description := s.generateDescription(route.Method, route.Path) + + return &model.AdminApi{ + Id: uuid.NewString(), + ApiName: apiName, + ApiCode: apiCode, + Method: route.Method, + Url: route.Path, + Status: 1, // 默认启用 + Description: description, + } +} + +// generateApiCode 生成API编码 +func (s *ApiRegistryService) generateApiCode(method, path string) string { + // 移除路径参数，如 :id + cleanPath := regexp.MustCompile(`/:[\w]+`).ReplaceAllString(path, "") + + // 转换为小写并替换特殊字符 + apiCode := strings.ToLower(method) + "_" + strings.ReplaceAll(cleanPath, "/", "_") + apiCode = strings.TrimPrefix(apiCode, "_") + apiCode = strings.TrimSuffix(apiCode, "_") + + return apiCode +} + +// generateApiName 生成API名称 +func (s *ApiRegistryService) generateApiName(path string) string { + // 从路径中提取模块和操作 + parts := strings.Split(strings.Trim(path, "/"), "/") + if len(parts) < 3 { + return path + } + + // 获取模块名和操作名 + module := parts[len(parts)-2] + action := parts[len(parts)-1] + + // 转换为中文描述 + moduleMap := map[string]string{ + "agent": "代理管理", + "auth": "认证管理", + "feature": "功能管理", + "menu": "菜单管理", + "notification": "通知管理", + "order": "订单管理", + "platform_user": "平台用户", + "product": "产品管理", + "query": "查询管理", + "role": "角色管理", + "user": "用户管理", + } + + actionMap := map[string]string{ + "list": "列表", + "create": "创建", + "update": "更新", + "delete": "删除", + "detail": "详情", + "login": "登录", + "config": "配置", + "example": "示例", + "refund": "退款", + "link": "链接", + "stats": "统计", + "cleanup": "清理", + "record": "记录", + } + + moduleName := moduleMap[module] + if moduleName == "" { + moduleName = module + } + + actionName := actionMap[action] + if actionName == "" { + actionName = action + } + + return fmt.Sprintf("%s-%s", moduleName, actionName) +} + +// generateDescription 生成API描述 +func (s *ApiRegistryService) generateDescription(method, path string) string { + methodMap := map[string]string{ + "GET": "查询", + "POST": "创建", + "PUT": "更新", + "DELETE": "删除", + } + + methodDesc := methodMap[method] + if methodDesc == "" { + methodDesc = method + } + + apiName := s.generateApiName(path) + + return fmt.Sprintf("%s%s", methodDesc, apiName) +} + +// shouldUpdateApi 判断是否需要更新API +func (s *ApiRegistryService) shouldUpdateApi(existing, new *model.AdminApi) bool { + return existing.ApiName != new.ApiName || + existing.Method != new.Method || + existing.Url != new.Url || + existing.Description != new.Description +} diff --git a/app/main/api/internal/service/apirequestService.go b/app/main/api/internal/service/apirequestService.go new file mode 100644 index 0000000..1afbaf4 --- /dev/null +++ b/app/main/api/internal/service/apirequestService.go @@ -0,0 +1,1579 @@ +package service + +import ( + "context" + "encoding/json" + "errors" + "fmt" + "qnc-server/app/main/api/internal/config" + tianyuanapi "qnc-server/app/main/api/internal/service/tianyuanapi_sdk" + "qnc-server/app/main/model" + "sort" + "strings" + "sync" + "sync/atomic" + "time" + + "github.com/Masterminds/squirrel" + "github.com/tidwall/gjson" + "github.com/zeromicro/go-zero/core/logx" +) + +// 辅助函数：将天远API响应转换为JSON字节数组 +func convertTianyuanResponse(resp *tianyuanapi.Response) ([]byte, error) { + return json.Marshal(resp.Data) +} + +// 生成认证时间范围：当前时间前后两天的YYYYMMDD-YYMMDD格式 +func generateAuthDateRange() string { + now := time.Now() + start := now.AddDate(0, 0, -2).Format("20060102") + end := now.AddDate(0, 0, 2).Format("20060102") + return fmt.Sprintf("%s-%s", start, end) +} + +type ApiRequestService struct { + config config.Config + featureModel model.FeatureModel + productFeatureModel model.ProductFeatureModel + tianyuanapi *tianyuanapi.Client +} + +// NewApiRequestService 是一个构造函数，用于初始化 ApiRequestService +func NewApiRequestService(c config.Config, featureModel model.FeatureModel, productFeatureModel model.ProductFeatureModel, tianyuanapi *tianyuanapi.Client) *ApiRequestService { + return &ApiRequestService{ + config: c, + featureModel: featureModel, + productFeatureModel: productFeatureModel, + tianyuanapi: tianyuanapi, + } +} + +type APIResponseData struct { + ApiID string `json:"apiID"` + Data json.RawMessage `json:"data"` // 这里用 RawMessage 来存储原始的 data + Success bool `json:"success"` + Timestamp string `json:"timestamp"` + Error string `json:"error,omitempty"` +} + +// ProcessRequests 处理请求 +func (a *ApiRequestService) ProcessRequests(params []byte, productID string) ([]byte, error) { + var ctx, cancel = context.WithCancel(context.Background()) + defer cancel() + build := a.productFeatureModel.SelectBuilder().Where(squirrel.Eq{ + "product_id": productID, + }) + productFeatureList, findProductFeatureErr := a.productFeatureModel.FindAll(ctx, build, "") + if findProductFeatureErr != nil { + return nil, findProductFeatureErr + } + var featureIDs []string + isImportantMap := make(map[string]int64, len(productFeatureList)) + for _, pf := range productFeatureList { + featureIDs = append(featureIDs, pf.FeatureId) + isImportantMap[pf.FeatureId] = pf.IsImportant + } + if len(featureIDs) == 0 { + return nil, errors.New("featureIDs 是空的") + } + builder := a.featureModel.SelectBuilder().Where(squirrel.Eq{"id": featureIDs}) + featureList, findFeatureErr := a.featureModel.FindAll(ctx, builder, "") + if findFeatureErr != nil { + return nil, findFeatureErr + } + if len(featureList) == 0 { + return nil, errors.New("处理请求错误，产品无对应接口功能") + } + var ( + wg sync.WaitGroup + resultsCh = make(chan APIResponseData, len(featureList)) + errorsCh = make(chan error, len(featureList)) + errorCount int32 + errorLimit = len(featureList) + retryNum = 5 + ) + + for i, feature := range featureList { + wg.Add(1) + go func(i int, feature *model.Feature) { + defer wg.Done() + + select { + case <-ctx.Done(): + return + default: + } + result := APIResponseData{ + ApiID: feature.ApiId, + Success: false, + } + timestamp := time.Now().Format("2006-01-02 15:04:05") + var ( + resp json.RawMessage + preprocessErr error + ) + // 若 isImportantMap[feature.ID] == 1，则表示需要在出错时重试 + isImportant := isImportantMap[feature.Id] == 1 + tryCount := 0 + for { + tryCount++ + resp, preprocessErr = a.PreprocessRequestApi(params, feature.ApiId) + if preprocessErr == nil { + break + } + if isImportant && tryCount < retryNum { + continue + } else { + break + } + } + if preprocessErr != nil { + result.Timestamp = timestamp + result.Error = preprocessErr.Error() + result.Data = resp + resultsCh <- result + errorsCh <- fmt.Errorf("请求失败: %v", preprocessErr) + atomic.AddInt32(&errorCount, 1) + if atomic.LoadInt32(&errorCount) >= int32(errorLimit) { + cancel() + } + return + } + + result.Data = resp + result.Success = true + result.Timestamp = timestamp + resultsCh <- result + }(i, feature) + } + + go func() { + wg.Wait() + close(resultsCh) + close(errorsCh) + }() + // 收集所有结果并合并z + var responseData []APIResponseData + for result := range resultsCh { + responseData = append(responseData, result) + } + if atomic.LoadInt32(&errorCount) >= int32(errorLimit) { + var allErrors []error + for err := range errorsCh { + allErrors = append(allErrors, err) + } + return nil, fmt.Errorf("请求失败次数超过 %d 次: %v", errorLimit, allErrors) + } + + combinedResponse, err := json.Marshal(responseData) + if err != nil { + return nil, fmt.Errorf("响应数据转 JSON 失败: %v", err) + } + + return combinedResponse, nil +} + +// ------------------------------------请求处理器-------------------------- +var requestProcessors = map[string]func(*ApiRequestService, []byte) ([]byte, error){ + "PersonEnterprisePro": (*ApiRequestService).ProcessPersonEnterpriseProRequest, + "BehaviorRiskScan": (*ApiRequestService).ProcessBehaviorRiskScanRequest, + "YYSYBE08": (*ApiRequestService).ProcessYYSYBE08Request, + "YYSY09CD": (*ApiRequestService).ProcessYYSY09CDRequest, + "FLXG0687": (*ApiRequestService).ProcessFLXG0687Request, + "FLXG3D56": (*ApiRequestService).ProcessFLXG3D56Request, + "FLXG0V4B": (*ApiRequestService).ProcessFLXG0V4BRequest, + "QYGL8271": (*ApiRequestService).ProcessQYGL8271Request, + "IVYZ5733": (*ApiRequestService).ProcessIVYZ5733Request, + "IVYZ9A2B": (*ApiRequestService).ProcessIVYZ9A2BRequest, + "JRZQ0A03": (*ApiRequestService).ProcessJRZQ0A03Request, + "QYGL6F2D": (*ApiRequestService).ProcessQYGL6F2DRequest, + "JRZQ8203": (*ApiRequestService).ProcessJRZQ8203Request, + "JRZQ4AA8": (*ApiRequestService).ProcessJRZQ4AA8Request, + "QCXG7A2B": (*ApiRequestService).ProcessQCXG7A2BRequest, + "DWBG8B4D": (*ApiRequestService).ProcessDWBG8B4DRequest, + "DWBG6A2C": (*ApiRequestService).ProcessDWBG6A2CRequest, + "JRZQ4B6C": (*ApiRequestService).ProcessJRZQ4B6CRequest, + "JRZQ09J8": (*ApiRequestService).ProcessJRZQ09J8Request, + "JRZQ5E9F": (*ApiRequestService).ProcessJRZQ5E9FRequest, + "QYGL3F8E": (*ApiRequestService).ProcessQYGL3F8ERequest, + "IVYZ81NC": (*ApiRequestService).ProcessIVYZ81NCRequest, + "IVYZ7F3A": (*ApiRequestService).ProcessIVYZ7F3ARequest, + "DWBG7F3A": (*ApiRequestService).ProcessDWBG7F3ARequest, + "JRZQ8A2D": (*ApiRequestService).ProcessJRZQ8A2DRequest, + "YYSY8B1C": (*ApiRequestService).ProcessYYSY8B1CRequest, + "YYSY7D3E": (*ApiRequestService).ProcessYYSY7D3ERequest, + "FLXG7E8F": (*ApiRequestService).ProcessFLXG7E8FRequest, + "IVYZ8I9J": (*ApiRequestService).ProcessIVYZ8I9JRequest, + "JRZQ7F1A": (*ApiRequestService).ProcessJRZQ7F1ARequest, + "IVYZ3P9M": (*ApiRequestService).ProcessIVYZ3P9MRequest, +} + +// PreprocessRequestApi 调用指定的请求处理函数 +func (a *ApiRequestService) PreprocessRequestApi(params []byte, apiID string) ([]byte, error) { + if processor, exists := requestProcessors[apiID]; exists { + return processor(a, params) // 调用 ApiRequestService 方法 + } + + return nil, errors.New("api请求, 未找到相应的处理程序") +} + +// PersonEnterprisePro 人企业关系加强版 +func (a *ApiRequestService) ProcessPersonEnterpriseProRequest(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + // 设置最大调用次数上限 + maxApiCalls := 20 // 允许最多查询20个企业 + + if !idCard.Exists() { + return nil, errors.New("api请求, PersonEnterprisePro, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("QYGLB4C0", map[string]interface{}{ + "id_card": idCard.String(), + }) + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + + // 处理股东人企关系的响应数据 + code := gjson.GetBytes(respBytes, "code") + if !code.Exists() { + return nil, fmt.Errorf("响应中缺少 code 字段") + } + + // 判断 code 是否等于 "0000" + if code.String() == "0000" { + // 获取 data 字段的值 + data := gjson.GetBytes(respBytes, "data") + if !data.Exists() { + return nil, fmt.Errorf("响应中缺少 data 字段") + } + + // 使用gjson获取企业列表 + datalistResult := gjson.Get(data.Raw, "datalist") + if !datalistResult.Exists() { + return nil, fmt.Errorf("datalist字段不存在") + } + + // 获取所有企业并进行排序 + companies := datalistResult.Array() + + // 创建企业对象切片，用于排序 + type CompanyWithPriority struct { + Index int + Data gjson.Result + RelationshipVal int // 关系权重值 + RelationCount int // 关系数量 + AdminPenalty int // 行政处罚数量 + Executed int // 被执行人数量 + Dishonest int // 失信被执行人数量 + } + + companiesWithPriority := make([]CompanyWithPriority, 0, len(companies)) + + // 遍历企业，计算优先级 + for i, companyJson := range companies { + // 统计行政处罚、被执行人、失信被执行人 + adminPenalty := 0 + executed := 0 + dishonest := 0 + + // 检查行政处罚字段是否存在并获取数组长度 + adminPenaltyResult := companyJson.Get("adminPenalty") + if adminPenaltyResult.Exists() && adminPenaltyResult.IsArray() { + adminPenalty = len(adminPenaltyResult.Array()) + } + + // 检查被执行人字段是否存在并获取数组长度 + executedPersonResult := companyJson.Get("executedPerson") + if executedPersonResult.Exists() && executedPersonResult.IsArray() { + executed = len(executedPersonResult.Array()) + } + + // 检查失信被执行人字段是否存在并获取数组长度 + dishonestExecutedPersonResult := companyJson.Get("dishonestExecutedPerson") + if dishonestExecutedPersonResult.Exists() && dishonestExecutedPersonResult.IsArray() { + dishonest = len(dishonestExecutedPersonResult.Array()) + } + + // 计算relationship权重 + relationshipVal := 0 + relationCount := 0 + + // 获取relationship数组 + relationshipResult := companyJson.Get("relationship") + if relationshipResult.Exists() && relationshipResult.IsArray() { + relationships := relationshipResult.Array() + // 统计各类关系的数量和权重 + for _, rel := range relationships { + relationCount++ + relStr := rel.String() + + // 根据关系类型设置权重，权重顺序： + // 股东(6) > 历史股东(5) > 法人(4) > 历史法人(3) > 高管(2) > 历史高管(1) + switch relStr { + case "sh": // 股东 + if relationshipVal < 6 { + relationshipVal = 6 + } + case "his_sh": // 历史股东 + if relationshipVal < 5 { + relationshipVal = 5 + } + case "lp": // 法人 + if relationshipVal < 4 { + relationshipVal = 4 + } + case "his_lp": // 历史法人 + if relationshipVal < 3 { + relationshipVal = 3 + } + case "tm": // 高管 + if relationshipVal < 2 { + relationshipVal = 2 + } + case "his_tm": // 历史高管 + if relationshipVal < 1 { + relationshipVal = 1 + } + } + } + } + + companiesWithPriority = append(companiesWithPriority, CompanyWithPriority{ + Index: i, + Data: companyJson, + RelationshipVal: relationshipVal, + RelationCount: relationCount, + AdminPenalty: adminPenalty, + Executed: executed, + Dishonest: dishonest, + }) + } + + // 按优先级排序 + sort.Slice(companiesWithPriority, func(i, j int) bool { + // 首先根据是否有失信被执行人排序 + if companiesWithPriority[i].Dishonest != companiesWithPriority[j].Dishonest { + return companiesWithPriority[i].Dishonest > companiesWithPriority[j].Dishonest + } + + // 然后根据是否有被执行人排序 + if companiesWithPriority[i].Executed != companiesWithPriority[j].Executed { + return companiesWithPriority[i].Executed > companiesWithPriority[j].Executed + } + + // 然后根据是否有行政处罚排序 + if companiesWithPriority[i].AdminPenalty != companiesWithPriority[j].AdminPenalty { + return companiesWithPriority[i].AdminPenalty > companiesWithPriority[j].AdminPenalty + } + + // 然后按relationship类型排序 + if companiesWithPriority[i].RelationshipVal != companiesWithPriority[j].RelationshipVal { + return companiesWithPriority[i].RelationshipVal > companiesWithPriority[j].RelationshipVal + } + + // 最后按relationship数量排序 + return companiesWithPriority[i].RelationCount > companiesWithPriority[j].RelationCount + }) + + // 限制处理的企业数量 + processCount := len(companiesWithPriority) + if processCount > maxApiCalls { + processCount = maxApiCalls + } + + // 只处理前N个优先级高的企业 + prioritizedCompanies := companiesWithPriority[:processCount] + + // 使用WaitGroup和chan处理并发 + var wg sync.WaitGroup + results := make(chan struct { + index int + data []byte + err error + }, processCount) + + // 对按优先级排序的前N个企业进行涉诉信息查询 + for _, company := range prioritizedCompanies { + wg.Add(1) + go func(origIndex int, companyInfo gjson.Result) { + defer wg.Done() + logx.Infof("开始处理企业[%d]，企业名称: %s，统一社会信用代码: %s", origIndex, companyInfo.Get("basicInfo.name").String(), companyInfo.Get("basicInfo.creditCode").String()) + // 提取企业名称和统一社会信用代码 + orgName := companyInfo.Get("basicInfo.name") + creditCode := companyInfo.Get("basicInfo.creditCode") + + if !orgName.Exists() || !creditCode.Exists() { + results <- struct { + index int + data []byte + err error + }{origIndex, nil, fmt.Errorf("企业名称或统一社会信用代码不存在")} + return + } + + // 解析原始公司信息为map + var companyMap map[string]interface{} + if err := json.Unmarshal([]byte(companyInfo.Raw), &companyMap); err != nil { + results <- struct { + index int + data []byte + err error + }{origIndex, nil, fmt.Errorf("解析企业信息失败: %v", err)} + return + } + + // 调用QYGL8271接口获取企业涉诉信息 + lawsuitResp, err := a.tianyuanapi.CallInterface("QYGL8271", map[string]interface{}{ + "ent_name": orgName.String(), + "ent_code": creditCode.String(), + "auth_date": generateAuthDateRange(), + }) + // 无论是否有错误，都继续处理 + if err != nil { + // 可能是正常没有涉诉数据，设置为空对象 + logx.Infof("企业[%s]涉诉信息查询结果: %v", orgName.String(), err) + companyMap["lawsuitInfo"] = map[string]interface{}{} + } else { + // 转换响应数据 + lawsuitRespBytes, err := convertTianyuanResponse(lawsuitResp) + if err != nil { + logx.Errorf("转换企业[%s]涉诉响应失败: %v", orgName.String(), err) + companyMap["lawsuitInfo"] = map[string]interface{}{} + } else if len(lawsuitRespBytes) == 0 || string(lawsuitRespBytes) == "{}" || string(lawsuitRespBytes) == "null" { + // 无涉诉数据 + companyMap["lawsuitInfo"] = map[string]interface{}{} + } else { + // 解析涉诉信息 + var lawsuitInfo interface{} + if err := json.Unmarshal(lawsuitRespBytes, &lawsuitInfo); err != nil { + logx.Errorf("解析企业[%s]涉诉信息失败: %v", orgName.String(), err) + companyMap["lawsuitInfo"] = map[string]interface{}{} + } else { + // 添加涉诉信息到企业信息中 + companyMap["lawsuitInfo"] = lawsuitInfo + } + } + } + + // 序列化更新后的企业信息 + companyData, err := json.Marshal(companyMap) + if err != nil { + results <- struct { + index int + data []byte + err error + }{origIndex, nil, fmt.Errorf("序列化企业信息失败: %v", err)} + return + } + + results <- struct { + index int + data []byte + err error + }{origIndex, companyData, nil} + }(company.Index, company.Data) + } + + // 关闭结果通道 + go func() { + wg.Wait() + close(results) + }() + + // 解析原始数据为map + var dataMap map[string]interface{} + if err := json.Unmarshal([]byte(data.Raw), &dataMap); err != nil { + return nil, fmt.Errorf("解析data字段失败: %v", err) + } + + // 获取原始企业列表 + originalDatalist, ok := dataMap["datalist"].([]interface{}) + if !ok { + return nil, fmt.Errorf("无法获取原始企业列表") + } + + // 创建结果映射，用于保存已处理的企业 + processedCompanies := make(map[int]interface{}) + + // 收集处理过的企业数据 + for result := range results { + if result.err != nil { + logx.Errorf("处理企业失败: %v", result.err) + continue + } + + if result.data != nil { + var companyMap interface{} + if err := json.Unmarshal(result.data, &companyMap); err == nil { + processedCompanies[result.index] = companyMap + } + } + } + + // 更新企业列表 + // 处理过的用新数据，未处理的保留原样 + updatedDatalist := make([]interface{}, len(originalDatalist)) + for i, company := range originalDatalist { + if processed, exists := processedCompanies[i]; exists { + // 已处理的企业，使用新数据 + updatedDatalist[i] = processed + } else { + // 未处理的企业，保留原始数据并添加空的涉诉信息 + companyMap, ok := company.(map[string]interface{}) + if ok { + // 为未处理的企业添加空的涉诉信息 + companyMap["lawsuitInfo"] = map[string]interface{}{} + updatedDatalist[i] = companyMap + } else { + updatedDatalist[i] = company + } + } + } + + // 更新原始数据中的企业列表 + dataMap["datalist"] = updatedDatalist + + // 序列化最终结果 + result, err := json.Marshal(dataMap) + if err != nil { + return nil, fmt.Errorf("序列化最终结果失败: %v", err) + } + + return result, nil + } + + // code不等于"0000"，返回错误 + return nil, fmt.Errorf("响应code错误: %s", code.String()) +} + +// ProcesFLXG0V4BRequest 个人司法涉诉（详版） +func (a *ApiRequestService) ProcessFLXG0V4BRequest(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + if !name.Exists() || !idCard.Exists() { + return nil, errors.New("api请求, FLXG0V4B, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("FLXG0V4B", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "auth_date": generateAuthDateRange(), + }) + if err != nil { + return nil, err + } + respBytes, err := json.Marshal(resp.Data) + if err != nil { + return nil, err + } + return respBytes, nil +} + +// ProcessFLXG0687Request 反诈反赌核验 +func (a *ApiRequestService) ProcessFLXG0687Request(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + if !idCard.Exists() { + return nil, errors.New("api请求, FLXG0687, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("FLXG0687", map[string]interface{}{ + "id_card": idCard.String(), + }) + + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + + Value := gjson.GetBytes(respBytes, "value") + if !Value.Exists() { + return nil, fmt.Errorf("自然人反诈反赌核验查询失败") + } + + return []byte(Value.Raw), nil +} + +// ProcessFLXG3D56Request 违约失信 +func (a *ApiRequestService) ProcessFLXG3D56Request(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, FLXG3D56, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("FLXG3D56", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + }) + + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + codeResult := gjson.GetBytes(respBytes, "code") + if !codeResult.Exists() { + return nil, fmt.Errorf("code 字段不存在") + } + if codeResult.String() != "00" { + return nil, fmt.Errorf("未匹配到相关结果") + } + + // 获取 data 字段 + dataResult := gjson.GetBytes(respBytes, "data") + if !dataResult.Exists() { + return nil, fmt.Errorf("data 字段不存在") + } + + // 将 data 字段解析为 map + var dataMap map[string]interface{} + if err := json.Unmarshal([]byte(dataResult.Raw), &dataMap); err != nil { + return nil, fmt.Errorf("解析 data 字段失败: %v", err) + } + + // 删除指定字段 + delete(dataMap, "swift_number") + delete(dataMap, "DataStrategy") + + // 重新编码为 JSON + modifiedData, err := json.Marshal(dataMap) + if err != nil { + return nil, fmt.Errorf("编码修改后的 data 失败: %v", err) + } + return modifiedData, nil +} + +// ProcessIVYZ5733Request 婚姻状况 +func (a *ApiRequestService) ProcessIVYZ5733Request(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + name := gjson.GetBytes(params, "name") + if !idCard.Exists() || !name.Exists() { + return nil, errors.New("api请求, IVYZ5733, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("IVYZ5733", map[string]interface{}{ + "id_card": idCard.String(), + "name": name.String(), + }) + + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + + result := gjson.GetBytes(respBytes, "data.data") + if !result.Exists() { + return nil, fmt.Errorf("婚姻状态查询失败") + } + + // 获取原始结果 + rawResult := result.String() + + // 根据结果转换状态码 + var statusCode string + switch { + case strings.HasPrefix(rawResult, "INR"): + statusCode = "0" // 匹配不成功 + case strings.HasPrefix(rawResult, "IA"): + statusCode = "1" // 结婚 + case strings.HasPrefix(rawResult, "IB"): + statusCode = "2" // 离婚 + default: + return nil, fmt.Errorf("婚姻状态查询失败，未知状态码: %s", statusCode) + } + + // 构建新的返回结果 + response := map[string]string{ + "status": statusCode, + } + // 序列化为JSON + jsonResponse, err := json.Marshal(response) + if err != nil { + return nil, fmt.Errorf("序列化结果失败: %v", err) + } + + return jsonResponse, nil +} + +// ProcessIVYZ9A2BRequest 学历查询 +func (a *ApiRequestService) ProcessIVYZ9A2BRequest(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + name := gjson.GetBytes(params, "name") + if !idCard.Exists() || !name.Exists() { + return nil, errors.New("api请求, IVYZ9A2B, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("IVYZ9A2B", map[string]interface{}{ + "id_card": idCard.String(), + "name": name.String(), + }) + + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + + // 解析响应 + codeResult := gjson.GetBytes(respBytes, "data.education_background.code") + if !codeResult.Exists() { + return nil, fmt.Errorf("教育经历核验查询失败: 返回数据缺少code字段") + } + + code := codeResult.String() + var result map[string]interface{} + + switch code { + case "9100": + // 查询成功有结果 + eduResultArray := gjson.GetBytes(respBytes, "data.education_background.data").Array() + var processedEduData []interface{} + + // 提取每个元素中Raw字段的实际内容 + for _, item := range eduResultArray { + var eduInfo interface{} + if err := json.Unmarshal([]byte(item.Raw), &eduInfo); err != nil { + return nil, fmt.Errorf("解析教育信息失败: %v", err) + } + processedEduData = append(processedEduData, eduInfo) + } + + result = map[string]interface{}{ + "data": processedEduData, + "status": 1, + } + case "9000": + // 查询成功无结果 + result = map[string]interface{}{ + "data": []interface{}{}, + "status": 0, + } + default: + // 其他情况视为错误 + errMsg := gjson.GetBytes(respBytes, "data.education_background.msg").String() + return nil, fmt.Errorf("教育经历核验查询失败: %s (code: %s)", errMsg, code) + } + + // 将结果转为JSON字节 + jsonResult, err := json.Marshal(result) + if err != nil { + return nil, fmt.Errorf("处理教育经历查询结果失败: %v", err) + } + + return jsonResult, nil +} + +// ProcessYYSYBE08Request 二要素 +func (a *ApiRequestService) ProcessYYSYBE08Request(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + if !name.Exists() || !idCard.Exists() { + return nil, errors.New("api请求, YYSYBE08, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("YYSYBE08", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + }) + + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + + // 使用gjson获取resultCode + resultCode := gjson.GetBytes(respBytes, "ctidRequest.ctidAuth.resultCode") + if !resultCode.Exists() { + return nil, errors.New("获取resultCode失败") + } + + // 获取resultCode的第一个字符 + resultCodeStr := resultCode.String() + if len(resultCodeStr) == 0 { + return nil, errors.New("resultCode为空") + } + + firstChar := string(resultCodeStr[0]) + if firstChar != "0" && firstChar != "5" { + return nil, errors.New("resultCode的第一个字符既不是0也不是5") + } + return []byte(firstChar), nil +} + +// ProcessJRZQ0A03Request 借贷申请 +func (a *ApiRequestService) ProcessJRZQ0A03Request(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, JRZQ0A03, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("JRZQ0A03", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + }) + + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + + // 获取 code 字段 + codeResult := gjson.GetBytes(respBytes, "code") + if !codeResult.Exists() { + return nil, fmt.Errorf("code 字段不存在") + } + if codeResult.String() != "00" { + return nil, fmt.Errorf("未匹配到相关结果") + } + + // 获取 data 字段 + dataResult := gjson.GetBytes(respBytes, "data") + if !dataResult.Exists() { + return nil, fmt.Errorf("data 字段不存在") + } + + // 将 data 字段解析为 map + var dataMap map[string]interface{} + if err := json.Unmarshal([]byte(dataResult.Raw), &dataMap); err != nil { + return nil, fmt.Errorf("解析 data 字段失败: %v", err) + } + + // 删除指定字段 + delete(dataMap, "swift_number") + delete(dataMap, "DataStrategy") + + // 重新编码为 JSON + modifiedData, err := json.Marshal(dataMap) + if err != nil { + return nil, fmt.Errorf("编码修改后的 data 失败: %v", err) + } + return modifiedData, nil +} + +// ProcessJRZQ8203Request 借贷行为 +func (a *ApiRequestService) ProcessJRZQ8203Request(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, JRZQ8203, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("JRZQ8203", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + }) + + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + + // 获取 code 字段 + codeResult := gjson.GetBytes(respBytes, "code") + if !codeResult.Exists() { + return nil, fmt.Errorf("code 字段不存在") + } + if codeResult.String() != "00" { + return nil, fmt.Errorf("未匹配到相关结果") + } + + // 获取 data 字段 + dataResult := gjson.GetBytes(respBytes, "data") + if !dataResult.Exists() { + return nil, fmt.Errorf("data 字段不存在") + } + + // 将 data 字段解析为 map + var dataMap map[string]interface{} + if err := json.Unmarshal([]byte(dataResult.Raw), &dataMap); err != nil { + return nil, fmt.Errorf("解析 data 字段失败: %v", err) + } + + // 删除指定字段 + delete(dataMap, "swift_number") + delete(dataMap, "DataStrategy") + + // 重新编码为 JSON + modifiedData, err := json.Marshal(dataMap) + if err != nil { + return nil, fmt.Errorf("编码修改后的 data 失败: %v", err) + } + return modifiedData, nil +} + +// ProcessJRZQ4AA8Request 还款压力 +func (a *ApiRequestService) ProcessJRZQ4AA8Request(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + name := gjson.GetBytes(params, "name") + mobile := gjson.GetBytes(params, "mobile") + if !idCard.Exists() || !name.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, JRZQ4AA8, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("JRZQ4AA8", map[string]interface{}{ + "id_card": idCard.String(), + "name": name.String(), + "mobile_no": mobile.String(), + }) + + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + + // 获取响应码和偿贷压力标志 + code := gjson.GetBytes(respBytes, "code").String() + flagDebtRepayStress := gjson.GetBytes(respBytes, "flag_debtrepaystress").String() + + // 判断是否成功 + if code != "00" || flagDebtRepayStress != "1" { + return nil, fmt.Errorf("偿贷压力查询失败: %+v", respBytes) + } + // 获取偿贷压力分数 + drsNoDebtScore := gjson.GetBytes(respBytes, "drs_nodebtscore").String() + + // 构建结果 + result := map[string]interface{}{ + "score": drsNoDebtScore, + } + + // 将结果转为JSON + jsonResult, err := json.Marshal(result) + if err != nil { + return nil, fmt.Errorf("处理偿贷压力查询结果失败: %v", err) + } + + return jsonResult, nil +} + +// ProcessQYGL8271Request 企业涉诉 +func (a *ApiRequestService) ProcessQYGL8271Request(params []byte) ([]byte, error) { + entName := gjson.GetBytes(params, "ent_name") + entCode := gjson.GetBytes(params, "ent_code") + + if !entName.Exists() || !entCode.Exists() { + return nil, errors.New("api请求, QYGL8271, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("QYGL8271", map[string]interface{}{ + "ent_name": entName.String(), + "ent_code": entCode.String(), + }) + + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + + // 第一步：提取外层的 data 字段 + dataResult := gjson.GetBytes(respBytes, "data") + if !dataResult.Exists() { + return nil, fmt.Errorf("外层 data 字段不存在") + } + + // 第二步：解析外层 data 的 JSON 字符串 + var outerDataMap map[string]interface{} + if err := json.Unmarshal([]byte(dataResult.String()), &outerDataMap); err != nil { + return nil, fmt.Errorf("解析外层 data 字段失败: %v", err) + } + + // 第三步：提取内层的 data 字段 + innerData, ok := outerDataMap["data"].(string) + if !ok { + return nil, fmt.Errorf("内层 data 字段不存在或类型错误") + } + + // 第四步：解析内层 data 的 JSON 字符串 + var finalDataMap map[string]interface{} + if err := json.Unmarshal([]byte(innerData), &finalDataMap); err != nil { + return nil, fmt.Errorf("解析内层 data 字段失败: %v", err) + } + + // 将最终的 JSON 对象编码为字节数组返回 + finalDataBytes, err := json.Marshal(finalDataMap) + if err != nil { + return nil, fmt.Errorf("编码最终的 JSON 对象失败: %v", err) + } + + statusResult := gjson.GetBytes(finalDataBytes, "status.status") + if statusResult.Exists() || statusResult.Int() == -1 { + return nil, fmt.Errorf("企业涉诉为空: %+v", finalDataBytes) + } + return finalDataBytes, nil +} + +// ProcessQYGL6F2DRequest 人企关联 +func (a *ApiRequestService) ProcessQYGL6F2DRequest(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + if !idCard.Exists() { + return nil, errors.New("api请求, QYGL6F2D, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("QYGL6F2D", map[string]interface{}{ + "id_card": idCard.String(), + }) + + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + + // 处理股东人企关系的响应数据 + code := gjson.GetBytes(respBytes, "code") + if !code.Exists() { + return nil, fmt.Errorf("响应中缺少 code 字段") + } + + // 判断 code 是否等于 "0000" + if code.String() == "0000" { + // 获取 data 字段的值 + data := gjson.GetBytes(respBytes, "data") + if !data.Exists() { + return nil, fmt.Errorf("响应中缺少 data 字段") + } + // 返回 data 字段的内容 + return []byte(data.Raw), nil + } + + // code 不等于 "0000"，返回错误 + return nil, fmt.Errorf("响应code错误%s", code.String()) +} + +// ProcessQCXG7A2BRequest 名下车辆 +func (a *ApiRequestService) ProcessQCXG7A2BRequest(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + if !idCard.Exists() { + return nil, errors.New("api请求, QCXG7A2B, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("QCXG7A2B", map[string]interface{}{ + "id_card": idCard.String(), + }) + + if err != nil { + return nil, err + } + + return convertTianyuanResponse(resp) +} + +// ProcessYYSY09CDRequest 三要素 +func (a *ApiRequestService) ProcessYYSY09CDRequest(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, YYSY09CD, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("YYSY09CD", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + }) + + if err != nil { + return nil, err + } + + respBytes, err := convertTianyuanResponse(resp) + if err != nil { + return nil, err + } + + // 使用gjson获取resultCode + resultCode := gjson.GetBytes(respBytes, "ctidRequest.ctidAuth.resultCode") + if !resultCode.Exists() { + return nil, errors.New("获取resultCode失败") + } + + // 获取resultCode的第一个字符 + resultCodeStr := resultCode.String() + if len(resultCodeStr) == 0 { + return nil, errors.New("resultCode为空") + } + + firstChar := string(resultCodeStr[0]) + if firstChar != "0" && firstChar != "5" { + return nil, errors.New("resultCode的第一个字符既不是0也不是5") + } + return []byte(firstChar), nil +} + +// ProcessBehaviorRiskScanRequest 行为风险扫描 +func (a *ApiRequestService) ProcessBehaviorRiskScanRequest(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + + if !name.Exists() || !idCard.Exists() { + return nil, errors.New("api请求, BehaviorRiskScan, 获取相关参数失败") + } + + var wg sync.WaitGroup + type apiResult struct { + name string + data []byte + err error + } + results := make(chan apiResult, 1) // 4个风险检测项 + + // 并行调用两个不同的风险检测API + wg.Add(1) + + // 反赌反诈 + go func() { + defer wg.Done() + respBytes, err := a.ProcessFLXG0687Request(params) + results <- apiResult{name: "anti_fraud_gaming", data: respBytes, err: err} + }() + + // 关闭结果通道 + go func() { + wg.Wait() + close(results) + }() + + // 收集所有结果 + resultMap := make(map[string]interface{}) + var errors []string + + for result := range results { + if result.err != nil { + // 记录错误但继续处理其他结果 + errors = append(errors, fmt.Sprintf("%s: %v", result.name, result.err)) + continue + } + + // 解析JSON结果并添加到结果映射 + var parsedData interface{} + if err := json.Unmarshal(result.data, &parsedData); err != nil { + errors = append(errors, fmt.Sprintf("解析%s数据失败: %v", result.name, err)) + } else { + resultMap[result.name] = parsedData + } + } + + // 添加错误信息到结果中(如果存在) + if len(errors) > 0 { + resultMap["errors"] = errors + } + + // 序列化最终结果 + finalResult, err := json.Marshal(resultMap) + if err != nil { + return nil, fmt.Errorf("序列化行为风险扫描结果失败: %v", err) + } + + return finalResult, nil +} + +// ProcessDWBG8B4DRequest 谛听多维报告 +func (a *ApiRequestService) ProcessDWBG8B4DRequest(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + AuthorizationURL := gjson.GetBytes(params, "authorization_url") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() || !AuthorizationURL.Exists() { + return nil, errors.New("api请求, DWBG8B4D, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("DWBG8B4D", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + "authorization_url": AuthorizationURL.String(), + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessDWBG6A2CRequest 司南报告服务 +func (a *ApiRequestService) ProcessDWBG6A2CRequest(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + AuthorizationURL := gjson.GetBytes(params, "authorization_url") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() || !AuthorizationURL.Exists() { + return nil, errors.New("api请求, DWBG6A2C, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("DWBG6A2C", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + "authorization_url": AuthorizationURL.String(), + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessJRZQ4B6CRequest 探针C风险评估 +func (a *ApiRequestService) ProcessJRZQ4B6CRequest(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, JRZQ4B6C, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("JRZQ4B6C", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + "authorized": "1", + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessJRZQ09J8Request 收入评估 +func (a *ApiRequestService) ProcessJRZQ09J8Request(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, JRZQ09J8, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("JRZQ09J8", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + "authorized": "1", + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessJRZQ5E9FRequest 借选指数 +func (a *ApiRequestService) ProcessJRZQ5E9FRequest(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, JRZQ5E9F, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("JRZQ5E9F", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + "authorized": "1", + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessQYGL3F8ERequest 人企关系加强版2 +func (a *ApiRequestService) ProcessQYGL3F8ERequest(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + if !idCard.Exists() { + return nil, errors.New("api请求, QYGL3F8E, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("QYGL3F8E", map[string]interface{}{ + "id_card": idCard.String(), + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessIVYZ81NCRequest 婚姻，登记时间版 +func (a *ApiRequestService) ProcessIVYZ81NCRequest(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + if !name.Exists() || !idCard.Exists() { + return nil, errors.New("api请求, IVYZ81NC, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("IVYZ81NC", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessIVYZ7F3ARequest 学历查询版B +func (a *ApiRequestService) ProcessIVYZ7F3ARequest(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + name := gjson.GetBytes(params, "name") + if !idCard.Exists() || !name.Exists() { + return nil, errors.New("api请求, IVYZ7F3A, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("IVYZ7F3A", map[string]interface{}{ + "id_card": idCard.String(), + "name": name.String(), + "authorized": "1", + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessDWBG7F3ARequest 多头借贷行业风险版 +func (a *ApiRequestService) ProcessDWBG7F3ARequest(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, DWBG7F3A, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("DWBG7F3A", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessJRZQ8A2DRequest 特殊名单验证B +func (a *ApiRequestService) ProcessJRZQ8A2DRequest(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, JRZQ8A2D, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("JRZQ8A2D", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + "authorized": "1", + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessYYSY8B1CRequest 手机在网时长B +func (a *ApiRequestService) ProcessYYSY8B1CRequest(params []byte) ([]byte, error) { + mobile := gjson.GetBytes(params, "mobile") + if !mobile.Exists() { + return nil, errors.New("api请求, YYSY8B1C, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("YYSY8B1C", map[string]interface{}{ + "mobile_no": mobile.String(), + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessYYSY7D3ERequest 携号转网查询 +func (a *ApiRequestService) ProcessYYSY7D3ERequest(params []byte) ([]byte, error) { + mobile := gjson.GetBytes(params, "mobile") + if !mobile.Exists() { + return nil, errors.New("api请求, YYSY7D3E, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("YYSY7D3E", map[string]interface{}{ + "mobile_no": mobile.String(), + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessFLXG7E8FRequest 个人司法涉诉查询 +func (a *ApiRequestService) ProcessFLXG7E8FRequest(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + name := gjson.GetBytes(params, "name") + mobile := gjson.GetBytes(params, "mobile") + if !idCard.Exists() || !name.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, FLXG7E8F, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("FLXG7E8F", map[string]interface{}{ + "id_card": idCard.String(), + "name": name.String(), + "mobile_no": mobile.String(), + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessIVYZ8I9JRequest 互联网行为推测 +func (a *ApiRequestService) ProcessIVYZ8I9JRequest(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + name := gjson.GetBytes(params, "name") + mobile := gjson.GetBytes(params, "mobile") + if !idCard.Exists() || !name.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, IVYZ8I9J, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("IVYZ8I9J", map[string]interface{}{ + "id_card": idCard.String(), + "name": name.String(), + "mobile_no": mobile.String(), + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessJRZQ7F1ARequest 全景雷达 +func (a *ApiRequestService) ProcessJRZQ7F1ARequest(params []byte) ([]byte, error) { + name := gjson.GetBytes(params, "name") + idCard := gjson.GetBytes(params, "id_card") + mobile := gjson.GetBytes(params, "mobile") + if !name.Exists() || !idCard.Exists() || !mobile.Exists() { + return nil, errors.New("api请求, JRZQ7F1A, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("JRZQ7F1A", map[string]interface{}{ + "name": name.String(), + "id_card": idCard.String(), + "mobile_no": mobile.String(), + "authorized": "1", + }) + + if err != nil { + return nil, err + } + + // 直接返回解密后的数据，而不是再次进行JSON编码 + return convertTianyuanResponse(resp) +} + +// ProcessIVYZ3P9MRequest 学历实时查询 +func (a *ApiRequestService) ProcessIVYZ3P9MRequest(params []byte) ([]byte, error) { + idCard := gjson.GetBytes(params, "id_card") + name := gjson.GetBytes(params, "name") + if !idCard.Exists() || !name.Exists() { + return nil, errors.New("api请求, IVYZ3P9M, 获取相关参数失败") + } + + resp, err := a.tianyuanapi.CallInterface("IVYZ3P9M", map[string]interface{}{ + "id_card": idCard.String(), + "name": name.String(), + }) + + if err != nil { + return nil, err + } + + return convertTianyuanResponse(resp) +} diff --git a/app/main/api/internal/service/applepayService.go b/app/main/api/internal/service/applepayService.go new file mode 100644 index 0000000..8ecd3be --- /dev/null +++ b/app/main/api/internal/service/applepayService.go @@ -0,0 +1,169 @@ +package service + +import ( + "context" + "crypto/ecdsa" + "crypto/x509" + "qnc-server/app/main/api/internal/config" + "encoding/json" + "encoding/pem" + "fmt" + "io/ioutil" + "net/http" + "strconv" + "time" + + "github.com/golang-jwt/jwt/v4" +) + +// ApplePayService 是 Apple IAP 支付服务的结构体 +type ApplePayService struct { + config config.ApplepayConfig // 配置项 +} + +// NewApplePayService 是一个构造函数，用于初始化 ApplePayService +func NewApplePayService(c config.Config) *ApplePayService { + return &ApplePayService{ + config: c.Applepay, + } +} +func (a *ApplePayService) GetIappayAppID(productName string) string { + return fmt.Sprintf("%s.%s", a.config.BundleID, productName) +} + +// VerifyReceipt 验证苹果支付凭证 +func (a *ApplePayService) VerifyReceipt(ctx context.Context, receipt string) (*AppleVerifyResponse, error) { + var reqUrl string + if a.config.Sandbox { + reqUrl = a.config.SandboxVerifyURL + } else { + reqUrl = a.config.ProductionVerifyURL + } + + // 读取私钥 + privateKey, err := loadPrivateKey(a.config.LoadPrivateKeyPath) + if err != nil { + return nil, fmt.Errorf("加载私钥失败：%v", err) + } + + // 生成 JWT + token, err := generateJWT(privateKey, a.config.KeyID, a.config.IssuerID) + if err != nil { + return nil, fmt.Errorf("生成JWT失败：%v", err) + } + + // 构造查询参数 + queryParams := fmt.Sprintf("?receipt-data=%s", receipt) + fullUrl := reqUrl + queryParams + + // 构建 HTTP GET 请求 + req, err := http.NewRequestWithContext(ctx, http.MethodGet, fullUrl, nil) + if err != nil { + return nil, fmt.Errorf("创建 HTTP 请求失败：%v", err) + } + req.Header.Set("Content-Type", "application/json") + req.Header.Set("Authorization", "Bearer "+token) + + // 发送请求 + client := &http.Client{Timeout: 10 * time.Second} + resp, err := client.Do(req) + if err != nil { + return nil, fmt.Errorf("请求苹果验证接口失败：%v", err) + } + defer resp.Body.Close() + + // 解析响应 + body, err := ioutil.ReadAll(resp.Body) + if err != nil { + return nil, fmt.Errorf("读取响应体失败：%v", err) + } + + var verifyResponse AppleVerifyResponse + err = json.Unmarshal(body, &verifyResponse) + if err != nil { + return nil, fmt.Errorf("解析响应体失败：%v", err) + } + + // 根据实际响应处理逻辑 + if verifyResponse.Status != 0 { + return nil, fmt.Errorf("验证失败，状态码：%d", verifyResponse.Status) + } + + return &verifyResponse, nil +} + +func loadPrivateKey(path string) (*ecdsa.PrivateKey, error) { + data, err := ioutil.ReadFile(path) + if err != nil { + return nil, err + } + block, _ := pem.Decode(data) + if block == nil || block.Type != "PRIVATE KEY" { + return nil, fmt.Errorf("无效的私钥数据") + } + key, err := x509.ParsePKCS8PrivateKey(block.Bytes) + if err != nil { + return nil, err + } + ecdsaKey, ok := key.(*ecdsa.PrivateKey) + if !ok { + return nil, fmt.Errorf("私钥类型错误") + } + return ecdsaKey, nil +} + +func generateJWT(privateKey *ecdsa.PrivateKey, keyID, issuerID string) (string, error) { + now := time.Now() + claims := jwt.RegisteredClaims{ + Issuer: issuerID, + IssuedAt: jwt.NewNumericDate(now), + ExpiresAt: jwt.NewNumericDate(now.Add(1 * time.Hour)), + Audience: jwt.ClaimStrings{"appstoreconnect-v1"}, + } + token := jwt.NewWithClaims(jwt.SigningMethodES256, claims) + token.Header["kid"] = keyID + tokenString, err := token.SignedString(privateKey) + if err != nil { + return "", err + } + return tokenString, nil +} + +// GenerateOutTradeNo 生成唯一订单号 +func (a *ApplePayService) GenerateOutTradeNo() string { + length := 16 + timestamp := time.Now().UnixNano() + timeStr := strconv.FormatInt(timestamp, 10) + randomPart := strconv.Itoa(int(timestamp % 1e6)) + combined := timeStr + randomPart + + if len(combined) >= length { + return combined[:length] + } + + for len(combined) < length { + combined += strconv.Itoa(int(timestamp % 10)) + } + + return combined +} + +// AppleVerifyResponse 定义苹果验证接口的响应结构 +type AppleVerifyResponse struct { + Status int `json:"status"` // 验证状态码：0 表示收据有效 + Receipt *Receipt `json:"receipt"` // 收据信息 +} + +// Receipt 定义收据的精简结构 +type Receipt struct { + BundleID string `json:"bundle_id"` // 应用的 Bundle ID + InApp []InAppItem `json:"in_app"` // 应用内购买记录 +} + +// InAppItem 定义单条交易记录 +type InAppItem struct { + ProductID string `json:"product_id"` // 商品 ID + TransactionID string `json:"transaction_id"` // 交易 ID + PurchaseDate string `json:"purchase_date"` // 购买日期 (ISO 8601) + OriginalTransID string `json:"original_transaction_id"` // 原始交易 ID +} diff --git a/app/main/api/internal/service/asynqService.go b/app/main/api/internal/service/asynqService.go new file mode 100644 index 0000000..3b5fb4e --- /dev/null +++ b/app/main/api/internal/service/asynqService.go @@ -0,0 +1,123 @@ +// asynq_service.go + +package service + +import ( + "encoding/json" + "time" + "qnc-server/app/main/api/internal/config" + "qnc-server/app/main/api/internal/types" + + "github.com/hibiken/asynq" + "github.com/zeromicro/go-zero/core/logx" +) + +type AsynqService struct { + client *asynq.Client + config config.Config +} + +// NewAsynqService 创建并初始化 Asynq 客户端 +func NewAsynqService(c config.Config) *AsynqService { + client := asynq.NewClient(asynq.RedisClientOpt{ + Addr: c.CacheRedis[0].Host, + Password: c.CacheRedis[0].Pass, + }) + + return &AsynqService{client: client, config: c} +} + +// Close 关闭 Asynq 客户端 +func (s *AsynqService) Close() error { + return s.client.Close() +} +func (s *AsynqService) SendQueryTask(orderID string) error { + // 准备任务的 payload + payload := types.MsgPaySuccessQueryPayload{ + OrderID: orderID, + } + payloadBytes, err := json.Marshal(payload) + if err != nil { + logx.Errorf("发送异步任务失败 (无法编码 payload): %v, 订单号: %s", err, orderID) + return err // 直接返回错误，避免继续执行 + } + + options := []asynq.Option{ + asynq.MaxRetry(5), // 设置最大重试次数 + } + // 创建任务 + task := asynq.NewTask(types.MsgPaySuccessQuery, payloadBytes, options...) + + // 将任务加入队列并获取任务信息 + info, err := s.client.Enqueue(task) + if err != nil { + logx.Errorf("发送异步任务失败 (加入队列失败): %+v, 订单号: %s", err, orderID) + return err + } + + // 记录成功日志，带上任务 ID 和队列信息 + logx.Infof("发送异步任务成功，任务ID: %s, 队列: %s, 订单号: %s", info.ID, info.Queue, orderID) + return nil +} + +// SendAgentProcessTask 发送代理处理任务 +func (s *AsynqService) SendAgentProcessTask(orderID string) error { + // 准备任务的 payload + payload := types.MsgAgentProcessPayload{ + OrderID: orderID, + } + payloadBytes, err := json.Marshal(payload) + if err != nil { + logx.Errorf("发送代理处理任务失败 (无法编码 payload): %v, 订单号: %s", err, orderID) + return err + } + + options := []asynq.Option{ + asynq.MaxRetry(5), // 设置最大重试次数 + } + // 创建任务 + task := asynq.NewTask(types.MsgAgentProcess, payloadBytes, options...) + + // 将任务加入队列并获取任务信息 + info, err := s.client.Enqueue(task) + if err != nil { + logx.Errorf("发送代理处理任务失败 (加入队列失败): %+v, 订单号: %s", err, orderID) + return err + } + + // 记录成功日志，带上任务 ID 和队列信息 + logx.Infof("发送代理处理任务成功，任务ID: %s, 队列: %s, 订单号: %s", info.ID, info.Queue, orderID) + return nil +} + +// SendUnfreezeTask 发送解冻任务（延迟执行） +func (s *AsynqService) SendUnfreezeTask(freezeTaskId string, processAt time.Time) error { + // 准备任务的 payload + payload := types.MsgUnfreezeCommissionPayload{ + FreezeTaskId: freezeTaskId, + } + payloadBytes, err := json.Marshal(payload) + if err != nil { + logx.Errorf("发送解冻任务失败 (无法编码 payload): %v, 冻结任务ID: %s", err, freezeTaskId) + return err + } + + options := []asynq.Option{ + asynq.MaxRetry(5), // 设置最大重试次数 + asynq.ProcessAt(processAt), // 延迟到指定时间执行 + asynq.Queue("critical"), // 使用关键队列 + } + // 创建任务 + task := asynq.NewTask(types.MsgUnfreezeCommission, payloadBytes, options...) + + // 将任务加入队列并获取任务信息 + info, err := s.client.Enqueue(task) + if err != nil { + logx.Errorf("发送解冻任务失败 (加入队列失败): %+v, 冻结任务ID: %s", err, freezeTaskId) + return err + } + + // 记录成功日志，带上任务 ID 和队列信息 + logx.Infof("发送解冻任务成功，任务ID: %s, 队列: %s, 冻结任务ID: %s, 执行时间: %v", info.ID, info.Queue, freezeTaskId, processAt) + return nil +} diff --git a/app/main/api/internal/service/authorizationService.go b/app/main/api/internal/service/authorizationService.go new file mode 100644 index 0000000..4bf06dd --- /dev/null +++ b/app/main/api/internal/service/authorizationService.go @@ -0,0 +1,225 @@ +package service + +import ( + "bytes" + "context" + "database/sql" + "fmt" + "os" + "path/filepath" + "qnc-server/app/main/api/internal/config" + "qnc-server/app/main/model" + "time" + + "github.com/google/uuid" + "github.com/jung-kurt/gofpdf" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" +) + +type AuthorizationService struct { + config config.Config + authDocModel model.AuthorizationDocumentModel + fileStoragePath string + fileBaseURL string +} + +// NewAuthorizationService 创建授权书服务实例 +func NewAuthorizationService(c config.Config, authDocModel model.AuthorizationDocumentModel) *AuthorizationService { + return &AuthorizationService{ + config: c, + authDocModel: authDocModel, + fileStoragePath: "data/authorization_docs", // 使用相对路径，兼容开发环境 + fileBaseURL: c.Authorization.FileBaseURL, // 从配置文件读取 + } +} + +// GenerateAuthorizationDocument 生成授权书PDF +func (s *AuthorizationService) GenerateAuthorizationDocument( + ctx context.Context, + userID string, + orderID string, + queryID string, + userInfo map[string]interface{}, +) (*model.AuthorizationDocument, error) { + // 1. 生成PDF内容 + pdfBytes, err := s.generatePDFContent(userInfo) + if err != nil { + return nil, errors.Wrapf(err, "生成PDF内容失败") + } + + // 2. 创建文件存储目录 + year := time.Now().Format("2006") + month := time.Now().Format("01") + dirPath := filepath.Join(s.fileStoragePath, year, month) + if err := os.MkdirAll(dirPath, 0755); err != nil { + return nil, errors.Wrapf(err, "创建存储目录失败: %s", dirPath) + } + + // 3. 生成文件名和路径 + fileName := fmt.Sprintf("auth_%s_%s_%s.pdf", userID, orderID, time.Now().Format("20060102_150405")) + filePath := filepath.Join(dirPath, fileName) + // 只存储相对路径，不包含域名 + relativePath := fmt.Sprintf("%s/%s/%s", year, month, fileName) + + // 4. 保存PDF文件 + if err := os.WriteFile(filePath, pdfBytes, 0644); err != nil { + return nil, errors.Wrapf(err, "保存PDF文件失败: %s", filePath) + } + + // 5. 保存到数据库 + authDoc := &model.AuthorizationDocument{ + Id: uuid.NewString(), + UserId: userID, + OrderId: orderID, + QueryId: queryID, + FileName: fileName, + FilePath: filePath, + FileUrl: relativePath, // 只存储相对路径 + FileSize: int64(len(pdfBytes)), + FileType: "pdf", + Status: "active", + ExpireTime: sql.NullTime{Valid: false}, // 永久保留，不设置过期时间 + } + + _, err = s.authDocModel.Insert(ctx, nil, authDoc) + if err != nil { + // 如果数据库保存失败，删除已创建的文件 + os.Remove(filePath) + return nil, errors.Wrapf(err, "保存授权书记录失败") + } + + logx.Infof("授权书生成成功: userID=%s, orderID=%s, filePath=%s", userID, orderID, filePath) + + return authDoc, nil +} + +// GetFullFileURL 获取完整的文件访问URL +func (s *AuthorizationService) GetFullFileURL(relativePath string) string { + if relativePath == "" { + return "" + } + return fmt.Sprintf("%s/%s", s.fileBaseURL, relativePath) +} + +// generatePDFContent 生成PDF内容 +func (s *AuthorizationService) generatePDFContent(userInfo map[string]interface{}) ([]byte, error) { + // 创建PDF文档 + pdf := gofpdf.New("P", "mm", "A4", "") + pdf.AddPage() + + // 添加中文字体支持 - 参考imageService的路径处理方式 + fontPaths := []string{ + "static/SIMHEI.TTF", // 相对于工作目录的路径（与imageService一致） + "/app/static/SIMHEI.TTF", // Docker容器内的字体文件 + "app/main/api/static/SIMHEI.TTF", // 开发环境备用路径 + } + + // 尝试添加字体 + fontAdded := false + for _, fontPath := range fontPaths { + if _, err := os.Stat(fontPath); err == nil { + pdf.AddUTF8Font("ChineseFont", "", fontPath) + fontAdded = true + logx.Infof("成功加载字体: %s", fontPath) + break + } else { + logx.Debugf("字体文件不存在: %s, 错误: %v", fontPath, err) + } + } + + // 如果没有找到字体文件，使用默认字体，并记录警告 + if !fontAdded { + pdf.SetFont("Arial", "", 12) + logx.Errorf("未找到中文字体文件，使用默认Arial字体，可能无法正确显示中文") + } else { + // 设置默认字体 + pdf.SetFont("ChineseFont", "", 12) + } + + // 获取用户信息 + name := getUserInfoString(userInfo, "name") + idCard := getUserInfoString(userInfo, "id_card") + + // 生成当前日期 + currentDate := time.Now().Format("2006年1月2日") + + // 设置标题样式 - 大字体、居中 + if fontAdded { + pdf.SetFont("ChineseFont", "", 20) // 使用20号字体 + } else { + pdf.SetFont("Arial", "", 20) + } + pdf.CellFormat(0, 15, "授权书", "", 1, "C", false, 0, "") + + // 添加空行 + pdf.Ln(5) + + // 设置正文样式 - 正常字体 + if fontAdded { + pdf.SetFont("ChineseFont", "", 12) + } else { + pdf.SetFont("Arial", "", 12) + } + + // 构建授权书内容（去掉标题部分） + content := fmt.Sprintf(`海南海宇大数据有限公司： +本人%s拟向贵司申请大数据分析报告查询业务，贵司需要了解本人相关状况，用于查询大数据分析报告，因此本人同意向贵司提供本人的姓名和手机号等个人信息，并同意贵司向第三方（包括但不限于西部数据交易有限公司）传送上述信息。第三方将使用上述信息核实信息真实情况，查询信用记录，并生成报告。 + +授权内容如下： +贵司向依法成立的第三方服务商（包括但不限于西部数据交易有限公司）根据本人提交的信息进行核实，并有权通过前述第三方服务机构查询、使用本人的身份信息、设备信息、运营商信息等，查询本人信息（包括但不限于学历、婚姻、资产状况及对信息主体产生负面影响的不良信息），出具相关报告。 +依法成立的第三方服务商查询或核实、搜集、保存、处理、共享、使用（含合法业务应用）本人相关数据，且不再另行告知本人，但法律、法规、监管政策禁止的除外。 +本人授权有效期为自授权之日起 1个月。本授权为不可撤销授权，但法律法规另有规定的除外。 + +用户声明与承诺： +本人在授权签署前，已通过实名认证及动态验证码验证（或其他身份验证手段），确认本授权行为为本人真实意思表示，平台已履行身份验证义务。 +本人在此声明已充分理解上述授权条款含义，知晓并自愿承担因授权数据使用可能带来的后果，包括但不限于影响个人信用评分、生活行为等。本人确认授权范围内的相关信息由本人提供并真实有效。 +若用户冒名签署或提供虚假信息，由用户自行承担全部法律责任，平台不承担任何后果。 + +特别提示： +本产品所有数据均来自第三方。可能部分数据未公开、数据更新延迟或信息受到限制，贵司不对数据的准确性、真实性、完整性做任何承诺。用户需根据实际情况，结合报告内容自行判断与决策。 +本产品仅供用户本人查询或被授权查询。除非用户取得合法授权，用户不得利用本产品查询他人信息。用户因未获得合法授权而擅自查询他人信息所产生的任何后果，由用户自行承担责任。 +本授权书涉及对本人敏感信息（包括但不限于婚姻状态、资产状况等）的查询与使用。本人已充分知晓相关信息的敏感性，并明确同意贵司及其合作方依据授权范围使用相关信息。 +平台声明：本授权书涉及的信息核实及查询结果由第三方服务商提供，平台不对数据的准确性、完整性、实时性承担责任；用户根据报告所作决策的风险由用户自行承担，平台对此不承担法律责任。 +本授权书中涉及的数据查询和报告生成由依法成立的第三方服务商提供。若因第三方行为导致数据错误或损失，用户应向第三方主张权利，平台不承担相关责任。 + +附加说明： +本人在授权的相关数据将依据法律法规及贵司内部数据管理规范妥善存储，存储期限为法律要求的最短必要时间。超过存储期限或在数据使用目的达成后，贵司将对相关数据进行销毁或匿名化处理。 +本人有权随时撤回本授权书中的授权，但撤回前的授权行为及其法律后果仍具有法律效力。若需撤回授权，本人可通过贵司官方渠道提交书面申请，贵司将在收到申请后依法停止对本人数据的使用。 +你通过"全能查"，自愿支付相应费用，用于购买海南海宇大数据有限公司的大数据报告产品。如若对产品内容存在异议，可通过邮箱admin@iieeii.com或APP"联系客服"按钮进行反馈，贵司将在收到异议之日起20日内进行核查和处理，并将结果答复。 +你向海南海宇大数据有限公司的支付方式为：海南海宇大数据有限公司及其经官方授权的相关企业的支付宝账户。 + +争议解决机制： +若因本授权书引发争议，双方应友好协商解决；协商不成的，双方同意将争议提交至授权书签署地（海南省）有管辖权的人民法院解决。 + +签署方式的法律效力声明： +本授权书通过用户在线勾选、电子签名或其他网络签署方式完成，与手写签名具有同等法律效力。平台已通过技术手段保存签署过程的完整记录，作为用户真实意思表示的证据。 + +本授权书于 %s 生效。 + +授权人：%s +身份证号：%s +签署时间：%s`, name, currentDate, name, idCard, currentDate) + + // 将内容写入PDF + pdf.MultiCell(0, 6, content, "", "", false) + + // 生成PDF字节数组 + var buf bytes.Buffer + err := pdf.Output(&buf) + if err != nil { + return nil, errors.Wrapf(err, "生成PDF字节数组失败") + } + + return buf.Bytes(), nil +} + +// getUserInfoString 安全获取用户信息字符串 +func getUserInfoString(userInfo map[string]interface{}, key string) string { + if value, exists := userInfo[key]; exists { + if str, ok := value.(string); ok { + return str + } + } + return "" +} diff --git a/app/main/api/internal/service/data/authorization_docs/2025/09/auth_20250918_170253.pdf b/app/main/api/internal/service/data/authorization_docs/2025/09/auth_20250918_170253.pdf new file mode 100644 index 0000000..1485cd9 Binary files /dev/null and b/app/main/api/internal/service/data/authorization_docs/2025/09/auth_20250918_170253.pdf differ diff --git a/app/main/api/internal/service/data/authorization_docs/2025/09/test_auth_20250915_140622.pdf b/app/main/api/internal/service/data/authorization_docs/2025/09/test_auth_20250915_140622.pdf new file mode 100644 index 0000000..50dd44f Binary files /dev/null and b/app/main/api/internal/service/data/authorization_docs/2025/09/test_auth_20250915_140622.pdf differ diff --git a/app/main/api/internal/service/dictService.go b/app/main/api/internal/service/dictService.go new file mode 100644 index 0000000..79b2755 --- /dev/null +++ b/app/main/api/internal/service/dictService.go @@ -0,0 +1,47 @@ +package service + +import ( + "context" + "qnc-server/app/main/model" + "errors" +) + +type DictService struct { + adminDictTypeModel model.AdminDictTypeModel + adminDictDataModel model.AdminDictDataModel +} + +func NewDictService(adminDictTypeModel model.AdminDictTypeModel, adminDictDataModel model.AdminDictDataModel) *DictService { + return &DictService{adminDictTypeModel: adminDictTypeModel, adminDictDataModel: adminDictDataModel} +} +func (s *DictService) GetDictLabel(ctx context.Context, dictType string, dictValue int64) (string, error) { + dictTypeModel, err := s.adminDictTypeModel.FindOneByDictType(ctx, dictType) + if err != nil { + return "", err + } + if dictTypeModel.Status != 1 { + return "", errors.New("字典类型未启用") + } + dictData, err := s.adminDictDataModel.FindOneByDictTypeDictValue(ctx, dictTypeModel.DictType, dictValue) + if err != nil { + return "", err + } + if dictData.Status != 1 { + return "", errors.New("字典数据未启用") + } + return dictData.DictLabel, nil +} +func (s *DictService) GetDictValue(ctx context.Context, dictType string, dictLabel string) (int64, error) { + dictTypeModel, err := s.adminDictTypeModel.FindOneByDictType(ctx, dictType) + if err != nil { + return 0, err + } + if dictTypeModel.Status != 1 { + return 0, errors.New("字典类型未启用") + } + dictData, err := s.adminDictDataModel.FindOneByDictTypeDictLabel(ctx, dictTypeModel.DictType, dictLabel) + if err != nil { + return 0, err + } + return dictData.DictValue, nil +} diff --git a/app/main/api/internal/service/imageService.go b/app/main/api/internal/service/imageService.go new file mode 100644 index 0000000..6c5ad79 --- /dev/null +++ b/app/main/api/internal/service/imageService.go @@ -0,0 +1,173 @@ +package service + +import ( + "bytes" + "fmt" + "image" + "image/jpeg" + "image/png" + "os" + "path/filepath" + + "github.com/fogleman/gg" + "github.com/skip2/go-qrcode" + "github.com/zeromicro/go-zero/core/logx" +) + +type ImageService struct { + baseImagePath string +} + +func NewImageService() *ImageService { + return &ImageService{ + baseImagePath: "static/images", // 原图存放目录 + } +} + +// ProcessImageWithQRCode 处理图片，在中间添加二维码 +func (s *ImageService) ProcessImageWithQRCode(qrcodeType, qrcodeUrl string) ([]byte, string, error) { + // 1. 根据qrcodeType确定使用哪张背景图 + var backgroundImageName string + switch qrcodeType { + case "promote": + backgroundImageName = "tg_qrcode_1.png" + case "invitation": + backgroundImageName = "yq_qrcode_1.png" + default: + backgroundImageName = "tg_qrcode_1.png" // 默认使用第一张图片 + } + + // 2. 读取原图 + originalImagePath := filepath.Join(s.baseImagePath, backgroundImageName) + originalImage, err := s.loadImage(originalImagePath) + if err != nil { + logx.Errorf("加载原图失败: %v, 图片路径: %s", err, originalImagePath) + return nil, "", fmt.Errorf("加载原图失败: %v", err) + } + + // 3. 获取原图尺寸 + bounds := originalImage.Bounds() + imgWidth := bounds.Dx() + imgHeight := bounds.Dy() + + // 4. 创建绘图上下文 + dc := gg.NewContext(imgWidth, imgHeight) + + // 5. 绘制原图作为背景 + dc.DrawImageAnchored(originalImage, imgWidth/2, imgHeight/2, 0.5, 0.5) + + // 6. 生成二维码（去掉白边） + qrCode, err := qrcode.New(qrcodeUrl, qrcode.Medium) + if err != nil { + logx.Errorf("生成二维码失败: %v, 二维码内容: %s", err, qrcodeUrl) + return nil, "", fmt.Errorf("生成二维码失败: %v", err) + } + // 禁用二维码边框，去掉白边 + qrCode.DisableBorder = true + + // 7. 根据二维码类型设置不同的尺寸和位置 + var qrSize int + var qrX, qrY int + + switch qrcodeType { + case "promote": + // promote类型：精确设置二维码尺寸 + qrSize = 280 // 固定尺寸280px + // 左下角位置：距左边和底边留一些边距 + qrX = 192 // 距左边180px + qrY = imgHeight - qrSize - 190 // 距底边100px + + case "invitation": + // invitation类型：精确设置二维码尺寸 + qrSize = 360 // 固定尺寸320px + // 中间偏上位置 + qrX = (imgWidth - qrSize) / 2 // 水平居中 + qrY = 555 // 垂直位置200px + + default: + // 默认（promote样式） + qrSize = 280 // 固定尺寸280px + qrX = 200 // 距左边180px + qrY = imgHeight - qrSize - 200 // 距底边100px + } + + // 8. 生成指定尺寸的二维码图片 + qrCodeImage := qrCode.Image(qrSize) + + // 9. 直接绘制二维码（不添加背景） + dc.DrawImageAnchored(qrCodeImage, qrX+qrSize/2, qrY+qrSize/2, 0.5, 0.5) + + // 11. 输出为字节数组 + var buf bytes.Buffer + err = png.Encode(&buf, dc.Image()) + if err != nil { + logx.Errorf("编码图片失败: %v", err) + return nil, "", fmt.Errorf("编码图片失败: %v", err) + } + + logx.Infof("成功生成带二维码的图片，类型: %s, 二维码内容: %s, 图片尺寸: %dx%d, 二维码尺寸: %dx%d, 位置: (%d,%d)", + qrcodeType, qrcodeUrl, imgWidth, imgHeight, qrSize, qrSize, qrX, qrY) + + return buf.Bytes(), "image/png", nil +} + +// loadImage 加载图片文件 +func (s *ImageService) loadImage(path string) (image.Image, error) { + file, err := os.Open(path) + if err != nil { + return nil, err + } + defer file.Close() + + // 尝试解码PNG + img, err := png.Decode(file) + if err != nil { + // 如果PNG解码失败，重新打开文件尝试JPEG + file.Close() + file, err = os.Open(path) + if err != nil { + return nil, err + } + defer file.Close() + + img, err = jpeg.Decode(file) + if err != nil { + // 如果还是失败，使用通用解码器 + file.Close() + file, err = os.Open(path) + if err != nil { + return nil, err + } + defer file.Close() + + img, _, err = image.Decode(file) + if err != nil { + return nil, err + } + } + } + + return img, nil +} + +// GetSupportedImageTypes 获取支持的图片类型列表 +func (s *ImageService) GetSupportedImageTypes() []string { + return []string{"promote", "invitation"} +} + +// CheckImageExists 检查指定类型的背景图是否存在 +func (s *ImageService) CheckImageExists(qrcodeType string) bool { + var backgroundImageName string + switch qrcodeType { + case "promote": + backgroundImageName = "tg_qrcode_1.png" + case "invitation": + backgroundImageName = "yq_qrcode_1.png" + default: + backgroundImageName = "tg_qrcode_1.png" + } + + imagePath := filepath.Join(s.baseImagePath, backgroundImageName) + _, err := os.Stat(imagePath) + return err == nil +} diff --git a/app/main/api/internal/service/tianyuanapi_sdk/client.go b/app/main/api/internal/service/tianyuanapi_sdk/client.go new file mode 100644 index 0000000..fcbfa74 --- /dev/null +++ b/app/main/api/internal/service/tianyuanapi_sdk/client.go @@ -0,0 +1,416 @@ +package tianyuanapi + +import ( + "bytes" + "crypto/aes" + "crypto/cipher" + "crypto/rand" + "encoding/base64" + "encoding/hex" + "encoding/json" + "errors" + "fmt" + "io" + "net/http" + "time" +) + +// API调用相关错误类型 +var ( + ErrQueryEmpty = errors.New("查询为空") + ErrSystem = errors.New("接口异常") + ErrDecryptFail = errors.New("解密失败") + ErrRequestParam = errors.New("请求参数结构不正确") + ErrInvalidParam = errors.New("参数校验不正确") + ErrInvalidIP = errors.New("未经授权的IP") + ErrMissingAccessId = errors.New("缺少Access-Id") + ErrInvalidAccessId = errors.New("未经授权的AccessId") + ErrFrozenAccount = errors.New("账户已冻结") + ErrArrears = errors.New("账户余额不足，无法请求") + ErrProductNotFound = errors.New("产品不存在") + ErrProductDisabled = errors.New("产品已停用") + ErrNotSubscribed = errors.New("未订阅此产品") + ErrBusiness = errors.New("业务失败") +) + +// 错误码映射 - 严格按照用户要求 +var ErrorCodeMap = map[error]int{ + ErrQueryEmpty: 1000, + ErrSystem: 1001, + ErrDecryptFail: 1002, + ErrRequestParam: 1003, + ErrInvalidParam: 1003, + ErrInvalidIP: 1004, + ErrMissingAccessId: 1005, + ErrInvalidAccessId: 1006, + ErrFrozenAccount: 1007, + ErrArrears: 1007, + ErrProductNotFound: 1008, + ErrProductDisabled: 1008, + ErrNotSubscribed: 1008, + ErrBusiness: 2001, +} + +// ApiCallOptions API调用选项 +type ApiCallOptions struct { + Json bool `json:"json,omitempty"` // 是否返回JSON格式 +} + +// Client 天元API客户端 +type Client struct { + accessID string + key string + baseURL string + timeout time.Duration + client *http.Client +} + +// Config 客户端配置 +type Config struct { + AccessID string // 访问ID + Key string // AES密钥（16进制） + BaseURL string // API基础URL + Timeout time.Duration // 超时时间 +} + +// Request 请求参数 +type Request struct { + InterfaceName string `json:"interfaceName"` // 接口名称 + Params map[string]interface{} `json:"params"` // 请求参数 + Timeout int `json:"timeout"` // 超时时间（毫秒） + Options *ApiCallOptions `json:"options"` // 调用选项 +} + +// ApiResponse HTTP API响应 +type ApiResponse struct { + Code int `json:"code"` + Message string `json:"message"` + TransactionID string `json:"transaction_id"` // 流水号 + Data string `json:"data"` // 加密的数据 +} + +// Response Call方法的响应 +type Response struct { + Code int `json:"code"` + Message string `json:"message"` + Success bool `json:"success"` + TransactionID string `json:"transaction_id"` // 流水号 + Data interface{} `json:"data"` // 解密后的数据 + Timeout int64 `json:"timeout"` // 请求耗时（毫秒） + Error string `json:"error,omitempty"` +} + +// NewClient 创建新的客户端实例 +func NewClient(config Config) (*Client, error) { + // 参数校验 + if config.AccessID == "" { + return nil, fmt.Errorf("accessID不能为空") + } + if config.Key == "" { + return nil, fmt.Errorf("key不能为空") + } + if config.BaseURL == "" { + config.BaseURL = "http://127.0.0.1:8080" + } + if config.Timeout == 0 { + config.Timeout = 60 * time.Second + } + + // 验证密钥格式 + if _, err := hex.DecodeString(config.Key); err != nil { + return nil, fmt.Errorf("无效的密钥格式，必须是16进制字符串: %v", err) + } + + return &Client{ + accessID: config.AccessID, + key: config.Key, + baseURL: config.BaseURL, + timeout: config.Timeout, + client: &http.Client{ + Timeout: config.Timeout, + }, + }, nil +} + +// Call 调用API接口 +func (c *Client) Call(req Request) (*Response, error) { + startTime := time.Now() + + // 参数校验 + if err := c.validateRequest(req); err != nil { + return nil, fmt.Errorf("请求参数校验失败: %v", err) + } + + // 加密参数 + jsonData, err := json.Marshal(req.Params) + if err != nil { + return nil, fmt.Errorf("参数序列化失败: %v", err) + } + + encryptedData, err := c.encrypt(string(jsonData)) + if err != nil { + return nil, fmt.Errorf("数据加密失败: %v", err) + } + + // 构建请求体 + requestBody := map[string]interface{}{ + "data": encryptedData, + } + + // 添加选项 + if req.Options != nil { + requestBody["options"] = req.Options + } else { + // 默认选项 + defaultOptions := &ApiCallOptions{ + Json: true, + } + requestBody["options"] = defaultOptions + } + + requestBodyBytes, err := json.Marshal(requestBody) + if err != nil { + return nil, fmt.Errorf("请求体序列化失败: %v", err) + } + + // 创建HTTP请求 + url := fmt.Sprintf("%s/api/v1/%s", c.baseURL, req.InterfaceName) + + httpReq, err := http.NewRequest("POST", url, bytes.NewBuffer(requestBodyBytes)) + if err != nil { + return nil, fmt.Errorf("创建HTTP请求失败: %v", err) + } + + // 设置请求头 + httpReq.Header.Set("Content-Type", "application/json") + httpReq.Header.Set("Access-Id", c.accessID) + httpReq.Header.Set("User-Agent", "TianyuanAPI-Go-SDK/1.0.0") + + // 发送请求 + resp, err := c.client.Do(httpReq) + if err != nil { + endTime := time.Now() + requestTime := endTime.Sub(startTime).Milliseconds() + return &Response{ + Success: false, + Message: "请求失败", + Error: err.Error(), + Timeout: requestTime, + }, nil + } + defer resp.Body.Close() + + // 读取响应 + body, err := io.ReadAll(resp.Body) + if err != nil { + endTime := time.Now() + requestTime := endTime.Sub(startTime).Milliseconds() + return &Response{ + Success: false, + Message: "读取响应失败", + Error: err.Error(), + Timeout: requestTime, + }, nil + } + + // 解析HTTP API响应 + var apiResp ApiResponse + if err := json.Unmarshal(body, &apiResp); err != nil { + endTime := time.Now() + requestTime := endTime.Sub(startTime).Milliseconds() + return &Response{ + Success: false, + Message: "响应解析失败", + Error: err.Error(), + Timeout: requestTime, + }, nil + } + + // 计算请求耗时 + endTime := time.Now() + requestTime := endTime.Sub(startTime).Milliseconds() + + // 构建Call方法的响应 + response := &Response{ + Code: apiResp.Code, + Message: apiResp.Message, + Success: apiResp.Code == 0, + TransactionID: apiResp.TransactionID, + Timeout: requestTime, + } + + // 如果有加密数据，尝试解密 + if apiResp.Data != "" { + decryptedData, err := c.decrypt(apiResp.Data) + if err == nil { + var decryptedMap interface{} + if json.Unmarshal([]byte(decryptedData), &decryptedMap) == nil { + response.Data = decryptedMap + } + } + } + + // 根据响应码返回对应的错误 + if apiResp.Code != 0 { + err := GetErrorByCode(apiResp.Code) + return nil, err + } + + return response, nil +} + +// CallInterface 简化接口调用方法 +func (c *Client) CallInterface(interfaceName string, params map[string]interface{}, options ...*ApiCallOptions) (*Response, error) { + var opts *ApiCallOptions + if len(options) > 0 { + opts = options[0] + } + + req := Request{ + InterfaceName: interfaceName, + Params: params, + Timeout: 60000, + Options: opts, + } + + return c.Call(req) +} + +// validateRequest 校验请求参数 +func (c *Client) validateRequest(req Request) error { + if req.InterfaceName == "" { + return fmt.Errorf("interfaceName不能为空") + } + if req.Params == nil { + return fmt.Errorf("params不能为空") + } + return nil +} + +// encrypt AES CBC加密 +func (c *Client) encrypt(plainText string) (string, error) { + keyBytes, err := hex.DecodeString(c.key) + if err != nil { + return "", err + } + + block, err := aes.NewCipher(keyBytes) + if err != nil { + return "", err + } + + // 生成随机IV + iv := make([]byte, aes.BlockSize) + if _, err := io.ReadFull(rand.Reader, iv); err != nil { + return "", err + } + + // 填充数据 + paddedData := c.pkcs7Pad([]byte(plainText), aes.BlockSize) + + // 加密 + ciphertext := make([]byte, len(iv)+len(paddedData)) + copy(ciphertext, iv) + + mode := cipher.NewCBCEncrypter(block, iv) + mode.CryptBlocks(ciphertext[len(iv):], paddedData) + + return base64.StdEncoding.EncodeToString(ciphertext), nil +} + +// decrypt AES CBC解密 +func (c *Client) decrypt(encryptedText string) (string, error) { + keyBytes, err := hex.DecodeString(c.key) + if err != nil { + return "", err + } + + ciphertext, err := base64.StdEncoding.DecodeString(encryptedText) + if err != nil { + return "", err + } + + block, err := aes.NewCipher(keyBytes) + if err != nil { + return "", err + } + + if len(ciphertext) < aes.BlockSize { + return "", fmt.Errorf("密文太短") + } + + iv := ciphertext[:aes.BlockSize] + ciphertext = ciphertext[aes.BlockSize:] + + if len(ciphertext)%aes.BlockSize != 0 { + return "", fmt.Errorf("密文长度不是块大小的倍数") + } + + mode := cipher.NewCBCDecrypter(block, iv) + mode.CryptBlocks(ciphertext, ciphertext) + + // 去除填充 + unpaddedData, err := c.pkcs7Unpad(ciphertext) + if err != nil { + return "", err + } + + return string(unpaddedData), nil +} + +// pkcs7Pad PKCS7填充 +func (c *Client) pkcs7Pad(data []byte, blockSize int) []byte { + padding := blockSize - len(data)%blockSize + padtext := bytes.Repeat([]byte{byte(padding)}, padding) + return append(data, padtext...) +} + +// pkcs7Unpad PKCS7去除填充 +func (c *Client) pkcs7Unpad(data []byte) ([]byte, error) { + length := len(data) + if length == 0 { + return nil, fmt.Errorf("数据为空") + } + unpadding := int(data[length-1]) + if unpadding > length { + return nil, fmt.Errorf("无效的填充") + } + return data[:length-unpadding], nil +} + +// GetErrorByCode 根据错误码获取错误 +func GetErrorByCode(code int) error { + // 对于有多个错误对应同一错误码的情况，返回第一个 + switch code { + case 1000: + return ErrQueryEmpty + case 1001: + return ErrSystem + case 1002: + return ErrDecryptFail + case 1003: + return ErrRequestParam + case 1004: + return ErrInvalidIP + case 1005: + return ErrMissingAccessId + case 1006: + return ErrInvalidAccessId + case 1007: + return ErrFrozenAccount + case 1008: + return ErrProductNotFound + case 2001: + return ErrBusiness + default: + return fmt.Errorf("未知错误码: %d", code) + } +} + +// GetCodeByError 根据错误获取错误码 +func GetCodeByError(err error) int { + if code, exists := ErrorCodeMap[err]; exists { + return code + } + return -1 +} diff --git a/app/main/api/internal/service/userService.go b/app/main/api/internal/service/userService.go new file mode 100644 index 0000000..5d16c5a --- /dev/null +++ b/app/main/api/internal/service/userService.go @@ -0,0 +1,265 @@ +package service + +import ( + "context" + "database/sql" + "qnc-server/app/main/api/internal/config" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + jwtx "qnc-server/common/jwt" + "qnc-server/common/xerr" + + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +type UserService struct { + Config *config.Config + userModel model.UserModel + userAuthModel model.UserAuthModel + agentModel model.AgentModel +} + +// NewUserService 创建UserService实例 +func NewUserService(config *config.Config, userModel model.UserModel, userAuthModel model.UserAuthModel, agentModel model.AgentModel) *UserService { + return &UserService{ + Config: config, + userModel: userModel, + userAuthModel: userAuthModel, + agentModel: agentModel, + } +} + +// GenerateUUIDUserId 生成UUID用户ID +func (s *UserService) GenerateUUIDUserId(ctx context.Context) (string, error) { + id := uuid.NewString() + return id, nil +} + +// RegisterUUIDUser 注册UUID用户，返回用户ID +func (s *UserService) RegisterUUIDUser(ctx context.Context) (string, error) { + // 生成UUID + uuidStr, err := s.GenerateUUIDUserId(ctx) + if err != nil { + return "", err + } + + var userId string + err = s.userModel.Trans(ctx, func(ctx context.Context, session sqlx.Session) error { + user := &model.User{Id: uuid.NewString()} + if _, userInsertErr := s.userModel.Insert(ctx, session, user); userInsertErr != nil { + return userInsertErr + } + userId = user.Id + userAuth := &model.UserAuth{Id: uuid.NewString(), UserId: userId, AuthType: model.UserAuthTypeUUID, AuthKey: uuidStr} + _, userAuthInsertErr := s.userAuthModel.Insert(ctx, session, userAuth) + return userAuthInsertErr + }) + if err != nil { + return "", err + } + + return userId, nil +} + +// GetUserType 根据user.Mobile字段动态计算用户类型 +// 如果有mobile，则为正式用户(UserTypeNormal)，否则为临时用户(UserTypeTemp) +func (s *UserService) GetUserType(ctx context.Context, userID string) (int64, error) { + user, err := s.userModel.FindOne(ctx, userID) + if err != nil { + return 0, err + } + if user.Mobile.Valid && user.Mobile.String != "" { + return model.UserTypeNormal, nil + } + return model.UserTypeTemp, nil +} + +// GeneralUserToken 生成用户token（动态计算userType） +func (s *UserService) GeneralUserToken(ctx context.Context, userID string) (string, error) { + platform, err := ctxdata.GetPlatformFromCtx(ctx) + if err != nil { + return "", err + } + var isAgent int64 + var agentID string + var authType string + var authKey string + // 获取用户信息，根据mobile字段动态计算userType + user, err := s.userModel.FindOne(ctx, userID) + if err != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "获取用户信息失败: %v", err) + } + + // 根据mobile判断用户类型 + var userType int64 + if user.Mobile.Valid && user.Mobile.String != "" { + userType = model.UserTypeNormal + } else { + userType = model.UserTypeTemp + } + agent, err := s.agentModel.FindOneByUserId(ctx, userID) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return "", errors.Wrapf(xerr.NewErrCode(xerr.DB_ERROR), "更新token, 获取用户代理信息失败: %v", err) + } + if agent != nil { + agentID = agent.Id + isAgent = model.AgentStatusYes + } + platAuthType := s.getAuthTypeByPlatform(platform) + ua, err := s.userAuthModel.FindOneByUserIdAuthType(ctx, userID, platAuthType) + if err == nil && ua != nil { + authType = ua.AuthType + authKey = ua.AuthKey + } + token, generaErr := jwtx.GenerateJwtToken(jwtx.JwtClaims{ + UserId: userID, + AgentId: agentID, + Platform: platform, + UserType: userType, + IsAgent: isAgent, + AuthType: authType, + AuthKey: authKey, + }, s.Config.JwtAuth.AccessSecret, s.Config.JwtAuth.AccessExpire) + if generaErr != nil { + return "", errors.Wrapf(xerr.NewErrCode(xerr.SERVER_COMMON_ERROR), "更新token, 生成token失败 : %s", userID) + } + return token, nil +} + +func (s *UserService) getAuthTypeByPlatform(platform string) string { + switch platform { + case model.PlatformWxMini: + return model.UserAuthTypeWxMiniOpenID + case model.PlatformWxH5: + return model.UserAuthTypeWxh5OpenID + case model.PlatformH5, model.PlatformApp: + return model.UserAuthTypeUUID + default: + return model.UserAuthTypeUUID + } +} + +// RegisterUser 注册用户，返回用户ID +// 只负责创建新用户（手机号不存在时），不处理合并逻辑 +// 如果有临时用户（claims），会将临时用户的认证绑定到新用户 +func (s *UserService) RegisterUser(ctx context.Context, mobile string) (string, error) { + // 检查手机号是否已存在 + user, err := s.userModel.FindOneByMobile(ctx, sql.NullString{String: mobile, Valid: true}) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return "", err + } + if user != nil { + return "", errors.New("用户已注册") + } + + // 获取当前登录态（可能为空） + claims, err := ctxdata.GetClaimsFromCtx(ctx) + if err != nil && !errors.Is(err, ctxdata.ErrNoInCtx) { + return "", err + } + + var userId string + err = s.userModel.Trans(ctx, func(ctx context.Context, session sqlx.Session) error { + // 创建新用户 + user := &model.User{Id: uuid.NewString(), Mobile: sql.NullString{String: mobile, Valid: true}} + if _, userInsertErr := s.userModel.Insert(ctx, session, user); userInsertErr != nil { + return userInsertErr + } + userId = user.Id + + // 创建 mobile 认证 + _, userAuthInsertErr := s.userAuthModel.Insert(ctx, session, &model.UserAuth{ + Id: uuid.NewString(), + UserId: userId, + AuthType: model.UserAuthTypeMobile, + AuthKey: mobile, + }) + if userAuthInsertErr != nil { + return userAuthInsertErr + } + + // 如果有临时用户，将临时用户的认证绑定到新用户 + if claims != nil { + // 检查临时用户是否已有该认证类型 + existingAuth, err := s.userAuthModel.FindOneByAuthTypeAuthKey(ctx, claims.AuthType, claims.AuthKey) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return err + } + // 如果认证不存在，创建新的认证绑定 + if existingAuth == nil { + _, err = s.userAuthModel.Insert(ctx, session, &model.UserAuth{ + Id: uuid.NewString(), + UserId: userId, + AuthType: claims.AuthType, + AuthKey: claims.AuthKey, + }) + if err != nil { + return err + } + } else if existingAuth.UserId != userId { + // 如果认证已存在但属于其他用户，迁移到新用户 + existingAuth.UserId = userId + if _, err := s.userAuthModel.Update(ctx, session, existingAuth); err != nil { + return err + } + } + } + + return nil + }) + if err != nil { + return "", err + } + return userId, nil +} + +// TempUserBindUser 临时用户绑定用户（添加mobile使其变为正式用户） +func (s *UserService) TempUserBindUser(ctx context.Context, session sqlx.Session, normalUserID string) error { + claims, err := ctxdata.GetClaimsFromCtx(ctx) + if err != nil && !errors.Is(err, ctxdata.ErrNoInCtx) { + return err + } + + if claims == nil { + return errors.New("无临时用户") + } + + // 检查当前用户是否已经绑定了mobile（根据mobile判断，而不是userType） + tempUser, err := s.userModel.FindOne(ctx, claims.UserId) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return err + } + if tempUser != nil && tempUser.Mobile.Valid && tempUser.Mobile.String != "" { + return errors.New("临时用户已注册") + } + + existingAuth, err := s.userAuthModel.FindOneByAuthTypeAuthKey(ctx, claims.AuthType, claims.AuthKey) + if err != nil && !errors.Is(err, model.ErrNotFound) { + return err + } + if existingAuth != nil { + return errors.New("临时用户已注册") + } + + if session == nil { + err := s.userAuthModel.Trans(ctx, func(ctx context.Context, session sqlx.Session) error { + _, err = s.userAuthModel.Insert(ctx, session, &model.UserAuth{Id: uuid.NewString(), UserId: normalUserID, AuthType: claims.AuthType, AuthKey: claims.AuthKey}) + if err != nil { + return err + } + return nil + }) + if err != nil { + return err + } + return nil + } else { + _, err = s.userAuthModel.Insert(ctx, session, &model.UserAuth{Id: uuid.NewString(), UserId: normalUserID, AuthType: claims.AuthType, AuthKey: claims.AuthKey}) + if err != nil { + return err + } + return nil + } +} diff --git a/app/main/api/internal/service/verificationService.go b/app/main/api/internal/service/verificationService.go new file mode 100644 index 0000000..d1f1cc6 --- /dev/null +++ b/app/main/api/internal/service/verificationService.go @@ -0,0 +1,207 @@ +package service + +import ( + "context" + "encoding/json" + "fmt" + "io" + "net/http" + "qnc-server/app/main/api/internal/config" + tianyuanapi "qnc-server/app/main/api/internal/service/tianyuanapi_sdk" + + "github.com/tidwall/gjson" +) +type VerificationService struct { + c config.Config + tianyuanapi *tianyuanapi.Client + apiRequestService *ApiRequestService +} + +func NewVerificationService(c config.Config, tianyuanapi *tianyuanapi.Client, apiRequestService *ApiRequestService) *VerificationService { + return &VerificationService{ + c: c, + tianyuanapi: tianyuanapi, + apiRequestService: apiRequestService, + } +} + +// 二要素 +type TwoFactorVerificationRequest struct { + Name string + IDCard string +} +type TwoFactorVerificationResp struct { + Msg string `json:"msg"` + Success bool `json:"success"` + Code int `json:"code"` + Data *TwoFactorVerificationData `json:"data"` // +} +type TwoFactorVerificationData struct { + Birthday string `json:"birthday"` + Result int `json:"result"` + Address string `json:"address"` + OrderNo string `json:"orderNo"` + Sex string `json:"sex"` + Desc string `json:"desc"` +} + +// 三要素 +type ThreeFactorVerificationRequest struct { + Name string + IDCard string + Mobile string +} + +// VerificationResult 定义校验结果结构体 +type VerificationResult struct { + Passed bool + Err error +} + +// ValidationError 定义校验错误类型 +type ValidationError struct { + Message string +} + +func (e *ValidationError) Error() string { + return e.Message +} + +func (r *VerificationService) TwoFactorVerification(request TwoFactorVerificationRequest) (*VerificationResult, error) { + resp, err := r.tianyuanapi.CallInterface("YYSYBE08", map[string]interface{}{ + "name": request.Name, + "id_card": request.IDCard, + }) + if err != nil { + return nil, fmt.Errorf("请求失败: %v", err) + } + + respBytes, err := json.Marshal(resp.Data) + if err != nil { + return nil, fmt.Errorf("转换响应失败: %v", err) + } + + // 使用gjson获取resultCode + resultCode := gjson.GetBytes(respBytes, "ctidRequest.ctidAuth.resultCode") + if !resultCode.Exists() { + return &VerificationResult{ + Passed: false, + Err: &ValidationError{Message: "获取resultCode失败"}, + }, nil + } + + // 获取resultCode的第一个字符 + resultCodeStr := resultCode.String() + if len(resultCodeStr) == 0 { + return &VerificationResult{ + Passed: false, + Err: &ValidationError{Message: "resultCode为空"}, + }, nil + } + + firstChar := string(resultCodeStr[0]) + if firstChar != "0" && firstChar != "5" { + return &VerificationResult{ + Passed: false, + Err: &ValidationError{Message: "姓名与身份证不一致"}, + }, nil + } + + return &VerificationResult{Passed: true, Err: nil}, nil +} + +func (r *VerificationService) ThreeFactorVerification(request ThreeFactorVerificationRequest) (*VerificationResult, error) { + resp, err := r.tianyuanapi.CallInterface("YYSY09CD", map[string]interface{}{ + "name": request.Name, + "id_card": request.IDCard, + "mobile_no": request.Mobile, + }) + if err != nil { + return nil, fmt.Errorf("请求失败: %v", err) + } + + respBytes, err := json.Marshal(resp.Data) + if err != nil { + return nil, fmt.Errorf("转换响应失败: %v", err) + } + + // 解析data.code + code := gjson.GetBytes(respBytes, "code") + if !code.Exists() { + return &VerificationResult{ + Passed: false, + Err: &ValidationError{Message: "身份信息异常"}, + }, nil + } + + codeStr := code.String() + switch codeStr { + case "1000": + // 一致 + return &VerificationResult{Passed: true, Err: nil}, nil + case "1001": + // 不一致 + return &VerificationResult{ + Passed: false, + Err: &ValidationError{Message: "姓名、证件号、手机号信息不一致"}, + }, nil + default: + // 其他异常 + return &VerificationResult{ + Passed: false, + Err: &ValidationError{Message: "身份信息异常"}, + }, nil + } +} + +// GetWechatH5OpenID 通过code获取微信H5 OpenID +func (r *VerificationService) GetWechatH5OpenID(ctx context.Context, code string) (string, error) { + appID := r.c.WechatH5.AppID + appSecret := r.c.WechatH5.AppSecret + url := fmt.Sprintf("https://api.weixin.qq.com/sns/oauth2/access_token?appid=%s&secret=%s&code=%s&grant_type=authorization_code", appID, appSecret, code) + resp, err := http.Get(url) + if err != nil { + return "", err + } + defer resp.Body.Close() + body, err := io.ReadAll(resp.Body) + if err != nil { + return "", err + } + var data struct { + Openid string `json:"openid"` + } + if err := json.Unmarshal(body, &data); err != nil { + return "", err + } + if data.Openid == "" { + return "", fmt.Errorf("openid为空") + } + return data.Openid, nil +} + +// GetWechatMiniOpenID 通过code获取微信小程序 OpenID +func (r *VerificationService) GetWechatMiniOpenID(ctx context.Context, code string) (string, error) { + appID := r.c.WechatMini.AppID + appSecret := r.c.WechatMini.AppSecret + url := fmt.Sprintf("https://api.weixin.qq.com/sns/jscode2session?appid=%s&secret=%s&js_code=%s&grant_type=authorization_code", appID, appSecret, code) + resp, err := http.Get(url) + if err != nil { + return "", err + } + defer resp.Body.Close() + body, err := io.ReadAll(resp.Body) + if err != nil { + return "", err + } + var data struct { + Openid string `json:"openid"` + } + if err := json.Unmarshal(body, &data); err != nil { + return "", err + } + if data.Openid == "" { + return "", fmt.Errorf("openid为空") + } + return data.Openid, nil +} diff --git a/app/main/api/internal/service/wechatpayService.go b/app/main/api/internal/service/wechatpayService.go new file mode 100644 index 0000000..fbf7f3a --- /dev/null +++ b/app/main/api/internal/service/wechatpayService.go @@ -0,0 +1,386 @@ +package service + +import ( + "context" + "fmt" + "net/http" + "strconv" + "time" + "qnc-server/app/main/api/internal/config" + "qnc-server/app/main/model" + "qnc-server/common/ctxdata" + "qnc-server/pkg/lzkit/lzUtils" + + "github.com/wechatpay-apiv3/wechatpay-go/core" + "github.com/wechatpay-apiv3/wechatpay-go/core/auth/verifiers" + "github.com/wechatpay-apiv3/wechatpay-go/core/downloader" + "github.com/wechatpay-apiv3/wechatpay-go/core/notify" + "github.com/wechatpay-apiv3/wechatpay-go/core/option" + "github.com/wechatpay-apiv3/wechatpay-go/services/payments" + "github.com/wechatpay-apiv3/wechatpay-go/services/payments/app" + "github.com/wechatpay-apiv3/wechatpay-go/services/payments/jsapi" + "github.com/wechatpay-apiv3/wechatpay-go/services/refunddomestic" + "github.com/wechatpay-apiv3/wechatpay-go/utils" + "github.com/zeromicro/go-zero/core/logx" +) + +const ( + TradeStateSuccess = "SUCCESS" // 支付成功 + TradeStateRefund = "REFUND" // 转入退款 + TradeStateNotPay = "NOTPAY" // 未支付 + TradeStateClosed = "CLOSED" // 已关闭 + TradeStateRevoked = "REVOKED" // 已撤销（付款码支付） + TradeStateUserPaying = "USERPAYING" // 用户支付中（付款码支付） + TradeStatePayError = "PAYERROR" // 支付失败（其他原因，如银行返回失败） +) + +// InitType 初始化类型 +type InitType string + +const ( + InitTypePlatformCert InitType = "platform_cert" // 平台证书初始化 + InitTypeWxPayPubKey InitType = "wxpay_pubkey" // 微信支付公钥初始化 +) + +type WechatPayService struct { + config config.Config + wechatClient *core.Client + notifyHandler *notify.Handler + userAuthModel model.UserAuthModel +} + +// NewWechatPayService 创建微信支付服务实例 +func NewWechatPayService(c config.Config, userAuthModel model.UserAuthModel, initType InitType) *WechatPayService { + switch initType { + case InitTypePlatformCert: + return newWechatPayServiceWithPlatformCert(c, userAuthModel) + case InitTypeWxPayPubKey: + return newWechatPayServiceWithWxPayPubKey(c, userAuthModel) + default: + logx.Errorf("不支持的初始化类型: %s", initType) + panic(fmt.Sprintf("初始化失败，服务停止: %s", initType)) + } +} + +// newWechatPayServiceWithPlatformCert 使用平台证书初始化微信支付服务 +func newWechatPayServiceWithPlatformCert(c config.Config, userAuthModel model.UserAuthModel) *WechatPayService { + // 从配置中加载商户信息 + mchID := c.Wxpay.MchID + mchCertificateSerialNumber := c.Wxpay.MchCertificateSerialNumber + mchAPIv3Key := c.Wxpay.MchApiv3Key + + // 从文件中加载商户私钥 + mchPrivateKey, err := utils.LoadPrivateKeyWithPath(c.Wxpay.MchPrivateKeyPath) + if err != nil { + logx.Errorf("加载商户私钥失败: %v", err) + panic(fmt.Sprintf("初始化失败，服务停止: %v", err)) // 记录错误并停止程序 + } + + // 使用商户私钥和其他参数初始化微信支付客户端 + opts := []core.ClientOption{ + option.WithWechatPayAutoAuthCipher(mchID, mchCertificateSerialNumber, mchPrivateKey, mchAPIv3Key), + } + client, err := core.NewClient(context.Background(), opts...) + if err != nil { + logx.Errorf("创建微信支付客户端失败: %v", err) + panic(fmt.Sprintf("初始化失败，服务停止: %v", err)) // 记录错误并停止程序 + } + + // 在初始化时获取证书访问器并创建 notifyHandler + certificateVisitor := downloader.MgrInstance().GetCertificateVisitor(mchID) + notifyHandler, err := notify.NewRSANotifyHandler(mchAPIv3Key, verifiers.NewSHA256WithRSAVerifier(certificateVisitor)) + if err != nil { + logx.Errorf("获取证书访问器失败: %v", err) + panic(fmt.Sprintf("初始化失败，服务停止: %v", err)) + } + + logx.Infof("微信支付客户端初始化成功(平台证书方式)") + return &WechatPayService{ + config: c, + wechatClient: client, + notifyHandler: notifyHandler, + userAuthModel: userAuthModel, + } +} + +// newWechatPayServiceWithWxPayPubKey 使用微信支付公钥初始化微信支付服务 +func newWechatPayServiceWithWxPayPubKey(c config.Config, userAuthModel model.UserAuthModel) *WechatPayService { + // 从配置中加载商户信息 + mchID := c.Wxpay.MchID + mchCertificateSerialNumber := c.Wxpay.MchCertificateSerialNumber + mchAPIv3Key := c.Wxpay.MchApiv3Key + mchPrivateKeyPath := c.Wxpay.MchPrivateKeyPath + mchPublicKeyID := c.Wxpay.MchPublicKeyID + mchPublicKeyPath := c.Wxpay.MchPublicKeyPath + // 从文件中加载商户私钥 + mchPrivateKey, err := utils.LoadPrivateKeyWithPath(mchPrivateKeyPath) + if err != nil { + logx.Errorf("加载商户私钥失败: %v", err) + panic(fmt.Sprintf("初始化失败，服务停止: %v", err)) + } + + // 从文件中加载微信支付公钥 + mchPublicKey, err := utils.LoadPublicKeyWithPath(mchPublicKeyPath) + if err != nil { + logx.Errorf("加载微信支付公钥失败: %v", err) + panic(fmt.Sprintf("初始化失败，服务停止: %v", err)) + } + + // 使用商户私钥和其他参数初始化微信支付客户端 + opts := []core.ClientOption{ + option.WithWechatPayPublicKeyAuthCipher(mchID, mchCertificateSerialNumber, mchPrivateKey, mchPublicKeyID, mchPublicKey), + } + client, err := core.NewClient(context.Background(), opts...) + if err != nil { + logx.Errorf("创建微信支付客户端失败: %v", err) + panic(fmt.Sprintf("初始化失败，服务停止: %v", err)) + } + + // 注册证书下载器，用于下载平台证书（回调验签需要） + // 注意：使用公钥方式时，需要手动注册证书下载器才能下载平台证书 + err = downloader.MgrInstance().RegisterDownloaderWithClient(context.Background(), client, mchID, mchAPIv3Key) + if err != nil { + logx.Errorf("注册证书下载器失败: %v", err) + panic(fmt.Sprintf("初始化失败，服务停止: %v", err)) + } + + // 初始化 notify.Handler + // 使用 SHA256WithRSACombinedVerifier 同时支持平台证书和公钥验签 + // 原因：微信回调目前仍使用平台证书签名，需要兼容处理；同时支持未来切换到公钥签名 + certificateVisitor := downloader.MgrInstance().GetCertificateVisitor(mchID) + notifyHandler := notify.NewNotifyHandler( + mchAPIv3Key, + verifiers.NewSHA256WithRSACombinedVerifier(certificateVisitor, mchPublicKeyID, *mchPublicKey)) + logx.Infof("微信支付客户端初始化成功(微信支付公钥方式)") + return &WechatPayService{ + config: c, + wechatClient: client, + notifyHandler: notifyHandler, + userAuthModel: userAuthModel, + } +} + +// CreateWechatAppOrder 创建微信APP支付订单 +func (w *WechatPayService) CreateWechatAppOrder(ctx context.Context, amount float64, description string, outTradeNo string) (string, error) { + totalAmount := lzUtils.ToWechatAmount(amount) + + // 构建支付请求参数 + payRequest := app.PrepayRequest{ + Appid: core.String(w.config.Wxpay.AppID), + Mchid: core.String(w.config.Wxpay.MchID), + Description: core.String(description), + OutTradeNo: core.String(outTradeNo), + NotifyUrl: core.String(w.config.Wxpay.NotifyUrl), + Amount: &app.Amount{ + Total: core.Int64(totalAmount), + }, + } + + // 初始化 AppApiService + svc := app.AppApiService{Client: w.wechatClient} + + // 发起预支付请求 + resp, result, err := svc.Prepay(ctx, payRequest) + if err != nil { + return "", fmt.Errorf("微信支付订单创建失败: %v, 状态码: %d", err, result.Response.StatusCode) + } + + // 返回预支付交易会话标识 + return *resp.PrepayId, nil +} + +// CreateWechatMiniProgramOrder 创建微信小程序支付订单 +func (w *WechatPayService) CreateWechatMiniProgramOrder(ctx context.Context, amount float64, description string, outTradeNo string, openid string) (interface{}, error) { + totalAmount := lzUtils.ToWechatAmount(amount) + + // 构建支付请求参数 + payRequest := jsapi.PrepayRequest{ + Appid: core.String(w.config.WechatMini.AppID), + Mchid: core.String(w.config.Wxpay.MchID), + Description: core.String(description), + OutTradeNo: core.String(outTradeNo), + NotifyUrl: core.String(w.config.Wxpay.NotifyUrl), + Amount: &jsapi.Amount{ + Total: core.Int64(totalAmount), + }, + Payer: &jsapi.Payer{ + Openid: core.String(openid), // 用户的 OpenID，通过前端传入 + }} + + // 初始化 AppApiService + svc := jsapi.JsapiApiService{Client: w.wechatClient} + + // 发起预支付请求 + resp, result, err := svc.PrepayWithRequestPayment(ctx, payRequest) + if err != nil { + return "", fmt.Errorf("微信支付订单创建失败: %v, 状态码: %d", err, result.Response.StatusCode) + } + // 返回预支付交易会话标识 + return resp, nil +} + +// CreateWechatH5Order 创建微信H5支付订单 +func (w *WechatPayService) CreateWechatH5Order(ctx context.Context, amount float64, description string, outTradeNo string, openid string) (interface{}, error) { + totalAmount := lzUtils.ToWechatAmount(amount) + + // 构建支付请求参数 + payRequest := jsapi.PrepayRequest{ + Appid: core.String(w.config.WechatH5.AppID), + Mchid: core.String(w.config.Wxpay.MchID), + Description: core.String(description), + OutTradeNo: core.String(outTradeNo), + NotifyUrl: core.String(w.config.Wxpay.NotifyUrl), + Amount: &jsapi.Amount{ + Total: core.Int64(totalAmount), + }, + Payer: &jsapi.Payer{ + Openid: core.String(openid), // 用户的 OpenID，通过前端传入 + }} + + // 初始化 AppApiService + svc := jsapi.JsapiApiService{Client: w.wechatClient} + + // 发起预支付请求 + resp, result, err := svc.PrepayWithRequestPayment(ctx, payRequest) + logx.Infof("微信h5支付订单：resp: %+v, result: %+v, err: %+v", resp, result, err) + if err != nil { + return "", fmt.Errorf("微信支付订单创建失败: %v, 状态码: %d", err, result.Response.StatusCode) + } + // 返回预支付交易会话标识 + return resp, nil +} + +// CreateWechatOrder 创建微信支付订单（集成 APP、H5、小程序） +func (w *WechatPayService) CreateWechatOrder(ctx context.Context, amount float64, description string, outTradeNo string) (interface{}, error) { + // 根据 ctx 中的 platform 判断平台 + platform := ctx.Value("platform").(string) + + var prepayData interface{} + var err error + + switch platform { + case model.PlatformWxMini: + userID, getUidErr := ctxdata.GetUidFromCtx(ctx) + if getUidErr != nil { + return "", getUidErr + } + userAuthModel, findAuthModelErr := w.userAuthModel.FindOneByUserIdAuthType(ctx, userID, model.UserAuthTypeWxMiniOpenID) + if findAuthModelErr != nil { + return "", findAuthModelErr + } + prepayData, err = w.CreateWechatMiniProgramOrder(ctx, amount, description, outTradeNo, userAuthModel.AuthKey) + if err != nil { + return "", err + } + case model.PlatformWxH5: + userID, getUidErr := ctxdata.GetUidFromCtx(ctx) + if getUidErr != nil { + return "", getUidErr + } + userAuthModel, findAuthModelErr := w.userAuthModel.FindOneByUserIdAuthType(ctx, userID, model.UserAuthTypeWxh5OpenID) + if findAuthModelErr != nil { + return "", findAuthModelErr + } + prepayData, err = w.CreateWechatH5Order(ctx, amount, description, outTradeNo, userAuthModel.AuthKey) + if err != nil { + return "", err + } + case model.PlatformApp: + // 如果是 APP 平台，调用 APP 支付订单创建 + prepayData, err = w.CreateWechatAppOrder(ctx, amount, description, outTradeNo) + default: + return "", fmt.Errorf("不支持的支付平台: %s", platform) + } + + // 如果创建支付订单失败，返回错误 + if err != nil { + return "", fmt.Errorf("支付订单创建失败: %v", err) + } + + // 返回预支付ID + return prepayData, nil +} + +// HandleWechatPayNotification 处理微信支付回调 +func (w *WechatPayService) HandleWechatPayNotification(ctx context.Context, req *http.Request) (*payments.Transaction, error) { + transaction := new(payments.Transaction) + _, err := w.notifyHandler.ParseNotifyRequest(ctx, req, transaction) + if err != nil { + return nil, fmt.Errorf("微信支付通知处理失败: %v", err) + } + // 返回交易信息 + return transaction, nil +} + +// HandleRefundNotification 处理微信退款回调 +func (w *WechatPayService) HandleRefundNotification(ctx context.Context, req *http.Request) (*refunddomestic.Refund, error) { + refund := new(refunddomestic.Refund) + _, err := w.notifyHandler.ParseNotifyRequest(ctx, req, refund) + if err != nil { + return nil, fmt.Errorf("微信退款回调通知处理失败: %v", err) + } + return refund, nil +} + +// QueryOrderStatus 主动查询订单状态 +func (w *WechatPayService) QueryOrderStatus(ctx context.Context, transactionID string) (*payments.Transaction, error) { + svc := jsapi.JsapiApiService{Client: w.wechatClient} + + // 调用 QueryOrderById 方法查询订单状态 + resp, result, err := svc.QueryOrderById(ctx, jsapi.QueryOrderByIdRequest{ + TransactionId: core.String(transactionID), + Mchid: core.String(w.config.Wxpay.MchID), + }) + if err != nil { + return nil, fmt.Errorf("订单查询失败: %v, 状态码: %d", err, result.Response.StatusCode) + } + return resp, nil + +} + +// WeChatRefund 申请微信退款 +func (w *WechatPayService) WeChatRefund(ctx context.Context, outTradeNo string, refundAmount float64, totalAmount float64) error { + + // 生成唯一的退款单号 + outRefundNo := fmt.Sprintf("%s-refund", outTradeNo) + + // 初始化退款服务 + svc := refunddomestic.RefundsApiService{Client: w.wechatClient} + + // 创建退款请求 + resp, result, err := svc.Create(ctx, refunddomestic.CreateRequest{ + OutTradeNo: core.String(outTradeNo), + OutRefundNo: core.String(outRefundNo), + NotifyUrl: core.String(w.config.Wxpay.RefundNotifyUrl), + Amount: &refunddomestic.AmountReq{ + Currency: core.String("CNY"), + Refund: core.Int64(lzUtils.ToWechatAmount(refundAmount)), + Total: core.Int64(lzUtils.ToWechatAmount(totalAmount)), + }, + }) + if err != nil { + return fmt.Errorf("微信订单申请退款错误: %v", err) + } + // 打印退款结果 + logx.Infof("退款申请成功，状态码=%d，退款单号=%s，微信退款单号=%s", result.Response.StatusCode, *resp.OutRefundNo, *resp.RefundId) + return nil +} + +// GenerateOutTradeNo 生成唯一订单号 +func (w *WechatPayService) GenerateOutTradeNo() string { + length := 16 + timestamp := time.Now().UnixNano() + timeStr := strconv.FormatInt(timestamp, 10) + randomPart := strconv.Itoa(int(timestamp % 1e6)) + combined := timeStr + randomPart + + if len(combined) >= length { + return combined[:length] + } + + for len(combined) < length { + combined += strconv.Itoa(int(timestamp % 10)) + } + + return combined +} diff --git a/app/main/api/internal/svc/servicecontext.go b/app/main/api/internal/svc/servicecontext.go new file mode 100644 index 0000000..5f3dcfc --- /dev/null +++ b/app/main/api/internal/svc/servicecontext.go @@ -0,0 +1,280 @@ +package svc + +import ( + "time" + "qnc-server/app/main/api/internal/config" + "qnc-server/app/main/api/internal/middleware" + "qnc-server/app/main/api/internal/service" + tianyuanapi "qnc-server/app/main/api/internal/service/tianyuanapi_sdk" + "qnc-server/app/main/model" + + "github.com/hibiken/asynq" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/core/stores/redis" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/rest" +) + +// ServiceContext 服务上下文 +type ServiceContext struct { + Config config.Config + Redis *redis.Redis + + // 中间件 + AuthInterceptor rest.Middleware + UserAuthInterceptor rest.Middleware + AdminAuthInterceptor rest.Middleware + + // 用户相关模型 + UserModel model.UserModel + UserAuthModel model.UserAuthModel + + // 产品相关模型 + ProductModel model.ProductModel + FeatureModel model.FeatureModel + ProductFeatureModel model.ProductFeatureModel + + // 订单相关模型 + OrderModel model.OrderModel + OrderRefundModel model.OrderRefundModel + QueryModel model.QueryModel + QueryCleanupLogModel model.QueryCleanupLogModel + QueryCleanupDetailModel model.QueryCleanupDetailModel + QueryCleanupConfigModel model.QueryCleanupConfigModel + + // 代理相关模型（新系统） + AgentModel model.AgentModel + AgentWalletModel model.AgentWalletModel + AgentRelationModel model.AgentRelationModel + AgentLinkModel model.AgentLinkModel + AgentOrderModel model.AgentOrderModel + AgentCommissionModel model.AgentCommissionModel + AgentRebateModel model.AgentRebateModel + AgentUpgradeModel model.AgentUpgradeModel + AgentWithdrawalModel model.AgentWithdrawalModel + AgentConfigModel model.AgentConfigModel + AgentProductConfigModel model.AgentProductConfigModel + AgentRealNameModel model.AgentRealNameModel + AgentWithdrawalTaxModel model.AgentWithdrawalTaxModel + AgentInviteCodeModel model.AgentInviteCodeModel + AgentInviteCodeUsageModel model.AgentInviteCodeUsageModel + AgentFreezeTaskModel model.AgentFreezeTaskModel + AgentShortLinkModel model.AgentShortLinkModel + + // 管理后台相关模型 + AdminApiModel model.AdminApiModel + AdminMenuModel model.AdminMenuModel + AdminRoleModel model.AdminRoleModel + AdminRoleApiModel model.AdminRoleApiModel + AdminRoleMenuModel model.AdminRoleMenuModel + AdminUserModel model.AdminUserModel + AdminUserRoleModel model.AdminUserRoleModel + AdminDictDataModel model.AdminDictDataModel + AdminDictTypeModel model.AdminDictTypeModel + + // 其他模型 + ExampleModel model.ExampleModel + GlobalNotificationsModel model.GlobalNotificationsModel + AuthorizationDocumentModel model.AuthorizationDocumentModel + + // 服务 + AlipayService *service.AliPayService + WechatPayService *service.WechatPayService + ApplePayService *service.ApplePayService + ApiRequestService *service.ApiRequestService + AsynqServer *asynq.Server + AsynqService *service.AsynqService + VerificationService *service.VerificationService + AgentService *service.AgentService + UserService *service.UserService + DictService *service.DictService + ImageService *service.ImageService + AuthorizationService *service.AuthorizationService +} + +// NewServiceContext 创建服务上下文 +func NewServiceContext(c config.Config) *ServiceContext { + // ============================== 基础设施初始化 ============================== + db := sqlx.NewMysql(c.DataSource) + cacheConf := c.CacheRedis + + // 初始化Redis客户端 + redisConf := redis.RedisConf{ + Host: cacheConf[0].Host, + Pass: cacheConf[0].Pass, + Type: cacheConf[0].Type, + } + redisClient := redis.MustNewRedis(redisConf) + + // ============================== 用户相关模型 ============================== + userModel := model.NewUserModel(db, cacheConf) + userAuthModel := model.NewUserAuthModel(db, cacheConf) + + // ============================== 产品相关模型 ============================== + productModel := model.NewProductModel(db, cacheConf) + featureModel := model.NewFeatureModel(db, cacheConf) + productFeatureModel := model.NewProductFeatureModel(db, cacheConf) + + // ============================== 订单相关模型 ============================== + orderModel := model.NewOrderModel(db, cacheConf) + queryModel := model.NewQueryModel(db, cacheConf) + orderRefundModel := model.NewOrderRefundModel(db, cacheConf) + queryCleanupLogModel := model.NewQueryCleanupLogModel(db, cacheConf) + queryCleanupDetailModel := model.NewQueryCleanupDetailModel(db, cacheConf) + queryCleanupConfigModel := model.NewQueryCleanupConfigModel(db, cacheConf) + + // ============================== 代理相关模型（新系统） ============================== + agentModel := model.NewAgentModel(db, cacheConf) + agentWalletModel := model.NewAgentWalletModel(db, cacheConf) + agentRelationModel := model.NewAgentRelationModel(db, cacheConf) + agentLinkModel := model.NewAgentLinkModel(db, cacheConf) + agentOrderModel := model.NewAgentOrderModel(db, cacheConf) + agentCommissionModel := model.NewAgentCommissionModel(db, cacheConf) + agentRebateModel := model.NewAgentRebateModel(db, cacheConf) + agentUpgradeModel := model.NewAgentUpgradeModel(db, cacheConf) + agentWithdrawalModel := model.NewAgentWithdrawalModel(db, cacheConf) + agentConfigModel := model.NewAgentConfigModel(db, cacheConf) + agentProductConfigModel := model.NewAgentProductConfigModel(db, cacheConf) + agentRealNameModel := model.NewAgentRealNameModel(db, cacheConf) + agentWithdrawalTaxModel := model.NewAgentWithdrawalTaxModel(db, cacheConf) + agentInviteCodeModel := model.NewAgentInviteCodeModel(db, cacheConf) + agentInviteCodeUsageModel := model.NewAgentInviteCodeUsageModel(db, cacheConf) + agentFreezeTaskModel := model.NewAgentFreezeTaskModel(db, cacheConf) + agentShortLinkModel := model.NewAgentShortLinkModel(db, cacheConf) + // ============================== 管理后台相关模型 ============================== + adminApiModel := model.NewAdminApiModel(db, cacheConf) + adminMenuModel := model.NewAdminMenuModel(db, cacheConf) + adminRoleModel := model.NewAdminRoleModel(db, cacheConf) + adminRoleApiModel := model.NewAdminRoleApiModel(db, cacheConf) + adminRoleMenuModel := model.NewAdminRoleMenuModel(db, cacheConf) + adminUserModel := model.NewAdminUserModel(db, cacheConf) + adminUserRoleModel := model.NewAdminUserRoleModel(db, cacheConf) + adminDictDataModel := model.NewAdminDictDataModel(db, cacheConf) + adminDictTypeModel := model.NewAdminDictTypeModel(db, cacheConf) + + // ============================== 其他模型 ============================== + exampleModel := model.NewExampleModel(db, cacheConf) + globalNotificationsModel := model.NewGlobalNotificationsModel(db, cacheConf) + authorizationDocumentModel := model.NewAuthorizationDocumentModel(db, cacheConf) + + // ============================== 第三方服务初始化 ============================== + tianyuanapi, err := tianyuanapi.NewClient(tianyuanapi.Config{ + AccessID: c.Tianyuanapi.AccessID, + Key: c.Tianyuanapi.Key, + BaseURL: c.Tianyuanapi.BaseURL, + Timeout: time.Duration(c.Tianyuanapi.Timeout) * time.Second, + }) + if err != nil { + logx.Errorf("初始化天远API失败: %+v", err) + } + + // ============================== 业务服务初始化 ============================== + alipayService := service.NewAliPayService(c) + wechatPayService := service.NewWechatPayService(c, userAuthModel, service.InitTypeWxPayPubKey) + applePayService := service.NewApplePayService(c) + apiRequestService := service.NewApiRequestService(c, featureModel, productFeatureModel, tianyuanapi) + verificationService := service.NewVerificationService(c, tianyuanapi, apiRequestService) + asynqService := service.NewAsynqService(c) + agentService := service.NewAgentService(c, orderModel, agentModel, agentWalletModel, + agentRelationModel, agentLinkModel, agentOrderModel, agentCommissionModel, agentRebateModel, + agentUpgradeModel, agentWithdrawalModel, agentConfigModel, agentProductConfigModel, + agentRealNameModel, agentWithdrawalTaxModel, agentFreezeTaskModel) + userService := service.NewUserService(&c, userModel, userAuthModel, agentModel) + dictService := service.NewDictService(adminDictTypeModel, adminDictDataModel) + imageService := service.NewImageService() + authorizationService := service.NewAuthorizationService(c, authorizationDocumentModel) + + // ============================== 异步任务服务 ============================== + asynqServer := asynq.NewServer( + asynq.RedisClientOpt{Addr: c.CacheRedis[0].Host, Password: c.CacheRedis[0].Pass}, + asynq.Config{ + IsFailure: func(err error) bool { + logx.Errorf("异步任务失败: %+v \n", err) + return true + }, + Concurrency: 10, + }, + ) + + // ============================== 返回服务上下文 ============================== + return &ServiceContext{ + Config: c, + Redis: redisClient, + AuthInterceptor: middleware.NewAuthInterceptorMiddleware(c).Handle, + UserAuthInterceptor: middleware.NewUserAuthInterceptorMiddleware().Handle, + AdminAuthInterceptor: middleware.NewAdminAuthInterceptorMiddleware(c, + adminUserModel, adminUserRoleModel, adminRoleModel, adminApiModel, adminRoleApiModel).Handle, + + // 用户相关模型 + UserModel: userModel, + UserAuthModel: userAuthModel, + + // 产品相关模型 + ProductModel: productModel, + FeatureModel: featureModel, + ProductFeatureModel: productFeatureModel, + + // 订单相关模型 + OrderModel: orderModel, + QueryModel: queryModel, + OrderRefundModel: orderRefundModel, + QueryCleanupLogModel: queryCleanupLogModel, + QueryCleanupDetailModel: queryCleanupDetailModel, + QueryCleanupConfigModel: queryCleanupConfigModel, + + // 代理相关模型（新系统） + AgentModel: agentModel, + AgentWalletModel: agentWalletModel, + AgentRelationModel: agentRelationModel, + AgentLinkModel: agentLinkModel, + AgentOrderModel: agentOrderModel, + AgentCommissionModel: agentCommissionModel, + AgentRebateModel: agentRebateModel, + AgentUpgradeModel: agentUpgradeModel, + AgentWithdrawalModel: agentWithdrawalModel, + AgentConfigModel: agentConfigModel, + AgentProductConfigModel: agentProductConfigModel, + AgentRealNameModel: agentRealNameModel, + AgentWithdrawalTaxModel: agentWithdrawalTaxModel, + AgentInviteCodeModel: agentInviteCodeModel, + AgentInviteCodeUsageModel: agentInviteCodeUsageModel, + AgentFreezeTaskModel: agentFreezeTaskModel, + AgentShortLinkModel: agentShortLinkModel, + + // 管理后台相关模型 + AdminApiModel: adminApiModel, + AdminMenuModel: adminMenuModel, + AdminRoleModel: adminRoleModel, + AdminRoleApiModel: adminRoleApiModel, + AdminRoleMenuModel: adminRoleMenuModel, + AdminUserModel: adminUserModel, + AdminUserRoleModel: adminUserRoleModel, + AdminDictDataModel: adminDictDataModel, + AdminDictTypeModel: adminDictTypeModel, + + // 其他模型 + ExampleModel: exampleModel, + GlobalNotificationsModel: globalNotificationsModel, + AuthorizationDocumentModel: authorizationDocumentModel, + + // 服务 + AlipayService: alipayService, + WechatPayService: wechatPayService, + ApplePayService: applePayService, + ApiRequestService: apiRequestService, + AsynqServer: asynqServer, + AsynqService: asynqService, + VerificationService: verificationService, + AgentService: agentService, + UserService: userService, + DictService: dictService, + ImageService: imageService, + AuthorizationService: authorizationService, + } +} + +func (s *ServiceContext) Close() { + if s.AsynqService != nil { + s.AsynqService.Close() + } +} diff --git a/app/main/api/internal/types/adminagent.go b/app/main/api/internal/types/adminagent.go new file mode 100644 index 0000000..35e185e --- /dev/null +++ b/app/main/api/internal/types/adminagent.go @@ -0,0 +1,204 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminAuditAgentReq struct { + AuditId int64 `json:"audit_id"` // 审核记录ID + Status int64 `json:"status"` // 审核状态：1=通过，2=拒绝 + AuditReason string `json:"audit_reason"` // 审核原因（拒绝时必填） +} + +type AdminAuditAgentResp struct { + Success bool `json:"success"` +} + +type AdminAuditWithdrawalReq struct { + WithdrawalId string `json:"withdrawal_id"` // 提现记录ID + Status int64 `json:"status"` // 审核状态：2=通过，3=拒绝 + Remark string `json:"remark"` // 备注 +} + +type AdminAuditWithdrawalResp struct { + Success bool `json:"success"` +} + +type AdminGenerateDiamondInviteCodeReq struct { + Count int64 `json:"count"` // 生成数量 + ExpireDays int64 `json:"expire_days,optional"` // 过期天数（可选，0表示不过期） + Remark string `json:"remark,optional"` // 备注（可选） +} + +type AdminGenerateDiamondInviteCodeResp struct { + Codes []string `json:"codes"` // 生成的邀请码列表 +} + +type AdminGetAgentCommissionListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + OrderId *string `form:"order_id,optional"` // 订单ID（可选） + Status *int64 `form:"status,optional"` // 状态（可选） +} + +type AdminGetAgentCommissionListResp struct { + Total int64 `json:"total"` // 总数 + Items []AgentCommissionListItem `json:"items"` // 列表数据 +} + +type AdminGetAgentConfigResp struct { + LevelBonus LevelBonusConfig `json:"level_bonus"` // 等级加成配置 + UpgradeFee UpgradeFeeConfig `json:"upgrade_fee"` // 升级费用配置 + UpgradeRebate UpgradeRebateConfig `json:"upgrade_rebate"` // 升级返佣配置 + DirectParentRebate DirectParentRebateConfig `json:"direct_parent_rebate"` // 直接上级返佣配置 + MaxGoldRebateAmount float64 `json:"max_gold_rebate_amount"` // 黄金代理最大返佣金额 + CommissionFreeze CommissionFreezeConfig `json:"commission_freeze"` // 佣金冻结配置 + TaxRate float64 `json:"tax_rate"` // 税率 + TaxExemptionAmount float64 `json:"tax_exemption_amount"` // 免税额度 + GoldMaxUpliftAmount float64 `json:"gold_max_uplift_amount"` + DiamondMaxUpliftAmount float64 `json:"diamond_max_uplift_amount"` +} + +type AdminGetAgentLinkListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + ProductId *string `form:"product_id,optional"` // 产品ID（可选） + LinkIdentifier *string `form:"link_identifier,optional"` // 推广码（可选） +} + +type AdminGetAgentLinkListResp struct { + Total int64 `json:"total"` // 总数 + Items []AgentLinkListItem `json:"items"` // 列表数据 +} + +type AdminGetAgentListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + Mobile *string `form:"mobile,optional"` // 手机号（可选） + Region *string `form:"region,optional"` // 区域（可选） + Level *int64 `form:"level,optional"` // 等级（可选） + TeamLeaderId *string `form:"team_leader_id,optional"` // 团队首领ID（可选） +} + +type AdminGetAgentListResp struct { + Total int64 `json:"total"` // 总数 + Items []AgentListItem `json:"items"` // 列表数据 +} + +type AdminGetAgentOrderListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + OrderId *string `form:"order_id,optional"` // 订单ID（可选） + ProcessStatus *int64 `form:"process_status,optional"` // 处理状态（可选） +} + +type AdminGetAgentOrderListResp struct { + Total int64 `json:"total"` // 总数 + Items []AgentOrderListItem `json:"items"` // 列表数据 +} + +type AdminGetAgentProductConfigListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + ProductId *string `form:"product_id,optional"` // 产品ID（可选） + ProductName *string `form:"product_name,optional"` // 产品名称（可选，用于搜索） +} + +type AdminGetAgentProductConfigListResp struct { + Total int64 `json:"total"` // 总数 + Items []AgentProductConfigItem `json:"items"` // 列表数据 +} + +type AdminGetAgentRealNameListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + Status *int64 `form:"status,optional"` // 状态（可选）：1=未验证，2=已通过 +} + +type AdminGetAgentRealNameListResp struct { + Total int64 `json:"total"` // 总数 + Items []AgentRealNameListItem `json:"items"` // 列表数据 +} + +type AdminGetAgentRebateListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + SourceAgentId *string `form:"source_agent_id,optional"` // 来源代理ID（可选） + RebateType *int64 `form:"rebate_type,optional"` // 返佣类型（可选） +} + +type AdminGetAgentRebateListResp struct { + Total int64 `json:"total"` // 总数 + Items []AgentRebateListItem `json:"items"` // 列表数据 +} + +type AdminGetAgentUpgradeListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + UpgradeType *int64 `form:"upgrade_type,optional"` // 升级类型（可选） + Status *int64 `form:"status,optional"` // 状态（可选） +} + +type AdminGetAgentUpgradeListResp struct { + Total int64 `json:"total"` // 总数 + Items []AgentUpgradeListItem `json:"items"` // 列表数据 +} + +type AdminGetAgentWithdrawalListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + AgentId *string `form:"agent_id,optional"` // 代理ID（可选） + Status *int64 `form:"status,optional"` // 状态（可选） + WithdrawNo *string `form:"withdraw_no,optional"` // 提现单号（可选） +} + +type AdminGetAgentWithdrawalListResp struct { + Total int64 `json:"total"` // 总数 + Items []AgentWithdrawalListItem `json:"items"` // 列表数据 +} + +type AdminGetInviteCodeListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + Code *string `form:"code,optional"` // 邀请码（可选） + AgentId *string `form:"agent_id,optional"` // 发放代理ID（可选，NULL表示平台发放） + TargetLevel *int64 `form:"target_level,optional"` // 目标等级（可选） + Status *int64 `form:"status,optional"` // 状态（可选） +} + +type AdminGetInviteCodeListResp struct { + Total int64 `json:"total"` // 总数 + Items []InviteCodeListItem `json:"items"` // 列表数据 +} + +type AdminUpdateAgentConfigReq struct { + LevelBonus *LevelBonusConfig `json:"level_bonus,optional"` // 等级加成配置 + UpgradeFee *UpgradeFeeConfig `json:"upgrade_fee,optional"` // 升级费用配置 + UpgradeRebate *UpgradeRebateConfig `json:"upgrade_rebate,optional"` // 升级返佣配置 + DirectParentRebate *DirectParentRebateConfig `json:"direct_parent_rebate,optional"` // 直接上级返佣配置 + MaxGoldRebateAmount *float64 `json:"max_gold_rebate_amount,optional"` // 黄金代理最大返佣金额 + CommissionFreeze *CommissionFreezeConfig `json:"commission_freeze,optional"` // 佣金冻结配置 + TaxRate *float64 `json:"tax_rate,optional"` // 税率 + TaxExemptionAmount *float64 `json:"tax_exemption_amount,optional"` // 免税额度 + GoldMaxUpliftAmount *float64 `json:"gold_max_uplift_amount,optional"` + DiamondMaxUpliftAmount *float64 `json:"diamond_max_uplift_amount,optional"` +} + +type AdminUpdateAgentConfigResp struct { + Success bool `json:"success"` +} + +type AdminUpdateAgentProductConfigReq struct { + Id string `json:"id"` // 主键 + BasePrice float64 `json:"base_price"` // 基础底价 + PriceRangeMax float64 `json:"price_range_max"` // 最高定价（对应数据库 system_max_price） + PriceThreshold *float64 `json:"price_threshold,optional"` // 提价标准阈值（可选） + PriceFeeRate *float64 `json:"price_fee_rate,optional"` // 提价手续费比例（可选） +} + +type AdminUpdateAgentProductConfigResp struct { + Success bool `json:"success"` +} diff --git a/app/main/api/internal/types/adminapi.go b/app/main/api/internal/types/adminapi.go new file mode 100644 index 0000000..ab72640 --- /dev/null +++ b/app/main/api/internal/types/adminapi.go @@ -0,0 +1,67 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminBatchUpdateApiStatusReq struct { + Ids []string `json:"ids"` + Status int64 `json:"status"` +} + +type AdminBatchUpdateApiStatusResp struct { + Success bool `json:"success"` +} + +type AdminCreateApiReq struct { + ApiName string `json:"api_name"` + ApiCode string `json:"api_code"` + Method string `json:"method"` + Url string `json:"url"` + Status int64 `json:"status,default=1"` + Description string `json:"description,optional"` +} + +type AdminCreateApiResp struct { + Id string `json:"id"` +} + +type AdminDeleteApiReq struct { + Id string `path:"id"` +} + +type AdminDeleteApiResp struct { + Success bool `json:"success"` +} + +type AdminGetApiDetailReq struct { + Id string `path:"id"` +} + +type AdminGetApiDetailResp struct { + AdminApiInfo +} + +type AdminGetApiListReq struct { + Page int64 `form:"page,default=1"` + PageSize int64 `form:"page_size,default=20"` + ApiName string `form:"api_name,optional"` + Method string `form:"method,optional"` + Status int64 `form:"status,optional"` +} + +type AdminGetApiListResp struct { + Items []AdminApiInfo `json:"items"` + Total int64 `json:"total"` +} + +type AdminUpdateApiReq struct { + Id string `path:"id"` + ApiName string `json:"api_name"` + ApiCode string `json:"api_code"` + Method string `json:"method"` + Url string `json:"url"` + Status int64 `json:"status"` + Description string `json:"description,optional"` +} + +type AdminUpdateApiResp struct { + Success bool `json:"success"` +} diff --git a/app/main/api/internal/types/adminauth.go b/app/main/api/internal/types/adminauth.go new file mode 100644 index 0000000..2b8548f --- /dev/null +++ b/app/main/api/internal/types/adminauth.go @@ -0,0 +1,15 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminLoginReq struct { + Username string `json:"username" validate:"required"` + Password string `json:"password" validate:"required"` + Captcha bool `json:"captcha" validate:"required"` +} + +type AdminLoginResp struct { + AccessToken string `json:"access_token"` + AccessExpire int64 `json:"access_expire"` + RefreshAfter int64 `json:"refresh_after"` + Roles []string `json:"roles"` +} diff --git a/app/main/api/internal/types/adminfeature.go b/app/main/api/internal/types/adminfeature.go new file mode 100644 index 0000000..7b479e8 --- /dev/null +++ b/app/main/api/internal/types/adminfeature.go @@ -0,0 +1,75 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminConfigFeatureExampleReq struct { + FeatureId string `json:"feature_id"` // 功能ID + Data string `json:"data"` // 示例数据JSON +} + +type AdminConfigFeatureExampleResp struct { + Success bool `json:"success"` // 是否成功 +} + +type AdminCreateFeatureReq struct { + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 +} + +type AdminCreateFeatureResp struct { + Id string `json:"id"` // 功能ID +} + +type AdminDeleteFeatureReq struct { + Id string `path:"id"` // 功能ID +} + +type AdminDeleteFeatureResp struct { + Success bool `json:"success"` // 是否成功 +} + +type AdminGetFeatureDetailReq struct { + Id string `path:"id"` // 功能ID +} + +type AdminGetFeatureDetailResp struct { + Id string `json:"id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + +type AdminGetFeatureExampleReq struct { + FeatureId string `path:"feature_id"` // 功能ID +} + +type AdminGetFeatureExampleResp struct { + Id string `json:"id"` // 示例数据ID + FeatureId string `json:"feature_id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Data string `json:"data"` // 示例数据JSON + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + +type AdminGetFeatureListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + ApiId *string `form:"api_id,optional"` // API标识 + Name *string `form:"name,optional"` // 描述 +} + +type AdminGetFeatureListResp struct { + Total int64 `json:"total"` // 总数 + Items []FeatureListItem `json:"items"` // 列表数据 +} + +type AdminUpdateFeatureReq struct { + Id string `path:"id"` // 功能ID + ApiId *string `json:"api_id,optional"` // API标识 + Name *string `json:"name,optional"` // 描述 +} + +type AdminUpdateFeatureResp struct { + Success bool `json:"success"` // 是否成功 +} diff --git a/app/main/api/internal/types/adminmenu.go b/app/main/api/internal/types/adminmenu.go new file mode 100644 index 0000000..90597ff --- /dev/null +++ b/app/main/api/internal/types/adminmenu.go @@ -0,0 +1,97 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type CreateMenuReq struct { + Pid string `json:"pid,optional"` // 父菜单ID + Name string `json:"name"` // 路由名称 + Path string `json:"path,optional"` // 路由路径 + Component string `json:"component,optional"` // 组件路径 + Redirect string `json:"redirect,optional"` // 重定向路径 + Meta map[string]interface{} `json:"meta"` // 路由元数据 + Status int64 `json:"status,optional,default=1"` // 状态：0-禁用，1-启用 + Type string `json:"type"` // 类型 + Sort int64 `json:"sort,optional"` // 排序 +} + +type CreateMenuResp struct { + Id string `json:"id"` // 菜单ID +} + +type DeleteMenuReq struct { + Id string `path:"id"` // 菜单ID +} + +type DeleteMenuResp struct { + Success bool `json:"success"` // 是否成功 +} + +type GetMenuAllReq struct { +} + +type GetMenuAllResp struct { + Name string `json:"name"` + Path string `json:"path"` + Redirect string `json:"redirect,omitempty"` + Component string `json:"component,omitempty"` + Sort int64 `json:"sort"` + Meta map[string]interface{} `json:"meta"` + Children []GetMenuAllResp `json:"children"` +} + +type GetMenuDetailReq struct { + Id string `path:"id"` // 菜单ID +} + +type GetMenuDetailResp struct { + Id string `json:"id"` // 菜单ID + Pid string `json:"pid"` // 父菜单ID + Name string `json:"name"` // 路由名称 + Path string `json:"path"` // 路由路径 + Component string `json:"component"` // 组件路径 + Redirect string `json:"redirect"` // 重定向路径 + Meta map[string]interface{} `json:"meta"` // 路由元数据 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + Type string `json:"type"` // 类型 + Sort int64 `json:"sort"` // 排序 + CreateTime string `json:"createTime"` // 创建时间 + UpdateTime string `json:"updateTime"` // 更新时间 +} + +type GetMenuListReq struct { + Name string `form:"name,optional"` // 菜单名称 + Path string `form:"path,optional"` // 路由路径 + Status int64 `form:"status,optional,default=-1"` // 状态：0-禁用，1-启用 + Type string `form:"type,optional"` // 类型 +} + +type MenuListItem struct { + Id string `json:"id"` // 菜单ID + Pid string `json:"pid"` // 父菜单ID + Name string `json:"name"` // 路由名称 + Path string `json:"path"` // 路由路径 + Component string `json:"component"` // 组件路径 + Redirect string `json:"redirect"` // 重定向路径 + Meta map[string]interface{} `json:"meta"` // 路由元数据 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + Type string `json:"type"` // 类型 + Sort int64 `json:"sort"` // 排序 + CreateTime string `json:"createTime"` // 创建时间 + Children []MenuListItem `json:"children"` // 子菜单 +} + +type UpdateMenuReq struct { + Id string `path:"id"` // 菜单ID + Pid *string `json:"pid,optional"` // 父菜单ID + Name string `json:"name"` // 路由名称 + Path string `json:"path,optional"` // 路由路径 + Component string `json:"component,optional"` // 组件路径 + Redirect string `json:"redirect,optional"` // 重定向路径 + Meta map[string]interface{} `json:"meta"` // 路由元数据 + Status int64 `json:"status,optional"` // 状态：0-禁用，1-启用 + Type string `json:"type"` // 类型 + Sort int64 `json:"sort,optional"` // 排序 +} + +type UpdateMenuResp struct { + Success bool `json:"success"` // 是否成功 +} diff --git a/app/main/api/internal/types/adminnotification.go b/app/main/api/internal/types/adminnotification.go new file mode 100644 index 0000000..c00cfa4 --- /dev/null +++ b/app/main/api/internal/types/adminnotification.go @@ -0,0 +1,74 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminCreateNotificationReq struct { + Title string `json:"title"` // 通知标题 + NotificationPage string `json:"notification_page"` // 通知页面 + Content string `json:"content"` // 通知内容 + StartDate string `json:"start_date"` // 生效开始日期（yyyy-MM-dd） + StartTime string `json:"start_time"` // 生效开始时间（HH:mm:ss） + EndDate string `json:"end_date"` // 生效结束日期（yyyy-MM-dd） + EndTime string `json:"end_time"` // 生效结束时间（HH:mm:ss） + Status int64 `json:"status"` // 状态：1-启用，0-禁用 +} + +type AdminCreateNotificationResp struct { + Id string `json:"id"` // 通知ID +} + +type AdminDeleteNotificationReq struct { + Id string `path:"id"` // 通知ID +} + +type AdminDeleteNotificationResp struct { + Success bool `json:"success"` // 是否成功 +} + +type AdminGetNotificationDetailReq struct { + Id string `path:"id"` // 通知ID +} + +type AdminGetNotificationDetailResp struct { + Id string `json:"id"` // 通知ID + Title string `json:"title"` // 通知标题 + Content string `json:"content"` // 通知内容 + NotificationPage string `json:"notification_page"` // 通知页面 + StartDate string `json:"start_date"` // 生效开始日期 + StartTime string `json:"start_time"` // 生效开始时间 + EndDate string `json:"end_date"` // 生效结束日期 + EndTime string `json:"end_time"` // 生效结束时间 + Status int64 `json:"status"` // 状态 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + +type AdminGetNotificationListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + Title *string `form:"title,optional"` // 通知标题（可选） + NotificationPage *string `form:"notification_page,optional"` // 通知页面（可选） + Status *int64 `form:"status,optional"` // 状态（可选） + StartDate *string `form:"start_date,optional"` // 开始日期范围（可选） + EndDate *string `form:"end_date,optional"` // 结束日期范围（可选） +} + +type AdminGetNotificationListResp struct { + Total int64 `json:"total"` // 总数 + Items []NotificationListItem `json:"items"` // 列表数据 +} + +type AdminUpdateNotificationReq struct { + Id string `path:"id"` // 通知ID + Title *string `json:"title,optional"` // 通知标题 + Content *string `json:"content,optional"` // 通知内容 + NotificationPage *string `json:"notification_page,optional"` // 通知页面 + StartDate *string `json:"start_date,optional"` // 生效开始日期 + StartTime *string `json:"start_time,optional"` // 生效开始时间 + EndDate *string `json:"end_date,optional"` // 生效结束日期 + EndTime *string `json:"end_time,optional"` // 生效结束时间 + Status *int64 `json:"status,optional"` // 状态 +} + +type AdminUpdateNotificationResp struct { + Success bool `json:"success"` // 是否成功 +} diff --git a/app/main/api/internal/types/adminorder.go b/app/main/api/internal/types/adminorder.go new file mode 100644 index 0000000..3597af1 --- /dev/null +++ b/app/main/api/internal/types/adminorder.go @@ -0,0 +1,108 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminCreateOrderReq struct { + OrderNo string `json:"order_no"` // 商户订单号 + PlatformOrderId string `json:"platform_order_id"` // 支付订单号 + ProductName string `json:"product_name"` // 产品名称 + PaymentPlatform string `json:"payment_platform"` // 支付方式 + PaymentScene string `json:"payment_scene"` // 支付平台 + Amount float64 `json:"amount"` // 金额 + Status string `json:"status,default=pending"` // 支付状态：pending-待支付，paid-已支付，refunded-已退款，closed-已关闭，failed-支付失败 +} + +type AdminCreateOrderResp struct { + Id string `json:"id"` // 订单ID +} + +type AdminDeleteOrderReq struct { + Id string `path:"id"` // 订单ID +} + +type AdminDeleteOrderResp struct { + Success bool `json:"success"` // 是否成功 +} + +type AdminGetOrderDetailReq struct { + Id string `path:"id"` // 订单ID +} + +type AdminGetOrderDetailResp struct { + Id string `json:"id"` // 订单ID + OrderNo string `json:"order_no"` // 商户订单号 + PlatformOrderId string `json:"platform_order_id"` // 支付订单号 + ProductName string `json:"product_name"` // 产品名称 + PaymentPlatform string `json:"payment_platform"` // 支付方式 + PaymentScene string `json:"payment_scene"` // 支付平台 + Amount float64 `json:"amount"` // 金额 + Status string `json:"status"` // 支付状态：pending-待支付，paid-已支付，refunded-已退款，closed-已关闭，failed-支付失败 + QueryState string `json:"query_state"` // 查询状态：pending-待查询，success-查询成功，failed-查询失败 processing-查询中 + CreateTime string `json:"create_time"` // 创建时间 + PayTime string `json:"pay_time"` // 支付时间 + RefundTime string `json:"refund_time"` // 退款时间 + UpdateTime string `json:"update_time"` // 更新时间 + IsAgentOrder bool `json:"is_agent_order"` // 是否是代理订单 + AgentProcessStatus string `json:"agent_process_status"` // 代理事务处理状态：not_agent-非代理订单，success-处理成功，failed-处理失败，pending-待处理 +} + +type AdminGetOrderListReq struct { + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"pageSize,default=20"` // 每页数量 + OrderNo string `form:"order_no,optional"` // 商户订单号 + PlatformOrderId string `form:"platform_order_id,optional"` // 支付订单号 + ProductName string `form:"product_name,optional"` // 产品名称 + PaymentPlatform string `form:"payment_platform,optional"` // 支付方式 + PaymentScene string `form:"payment_scene,optional"` // 支付平台 + Amount float64 `form:"amount,optional"` // 金额 + Status string `form:"status,optional"` // 支付状态：pending-待支付，paid-已支付，refunded-已退款，closed-已关闭，failed-支付失败 + CreateTimeStart string `form:"create_time_start,optional"` // 创建时间开始 + CreateTimeEnd string `form:"create_time_end,optional"` // 创建时间结束 + PayTimeStart string `form:"pay_time_start,optional"` // 支付时间开始 + PayTimeEnd string `form:"pay_time_end,optional"` // 支付时间结束 + RefundTimeStart string `form:"refund_time_start,optional"` // 退款时间开始 + RefundTimeEnd string `form:"refund_time_end,optional"` // 退款时间结束 +} + +type AdminGetOrderListResp struct { + Total int64 `json:"total"` // 总数 + Items []OrderListItem `json:"items"` // 列表 +} + +type AdminRefundOrderReq struct { + Id string `path:"id"` // 订单ID + RefundAmount float64 `json:"refund_amount"` // 退款金额 + RefundReason string `json:"refund_reason"` // 退款原因 +} + +type AdminRefundOrderResp struct { + Status string `json:"status"` // 退款状态 + RefundNo string `json:"refund_no"` // 退款单号 + Amount float64 `json:"amount"` // 退款金额 +} + +type AdminRetryAgentProcessReq struct { + Id string `path:"id"` // 订单ID +} + +type AdminRetryAgentProcessResp struct { + Status string `json:"status"` // 执行状态：success-成功，already_processed-已处理，failed-失败 + Message string `json:"message"` // 执行结果消息 + ProcessedAt string `json:"processed_at"` // 处理时间 +} + +type AdminUpdateOrderReq struct { + Id string `path:"id"` // 订单ID + OrderNo *string `json:"order_no,optional"` // 商户订单号 + PlatformOrderId *string `json:"platform_order_id,optional"` // 支付订单号 + ProductName *string `json:"product_name,optional"` // 产品名称 + PaymentPlatform *string `json:"payment_platform,optional"` // 支付方式 + PaymentScene *string `json:"payment_scene,optional"` // 支付平台 + Amount *float64 `json:"amount,optional"` // 金额 + Status *string `json:"status,optional"` // 支付状态：pending-待支付，paid-已支付，refunded-已退款，closed-已关闭，failed-支付失败 + PayTime *string `json:"pay_time,optional"` // 支付时间 + RefundTime *string `json:"refund_time,optional"` // 退款时间 +} + +type AdminUpdateOrderResp struct { + Success bool `json:"success"` // 是否成功 +} diff --git a/app/main/api/internal/types/adminplatformuser.go b/app/main/api/internal/types/adminplatformuser.go new file mode 100644 index 0000000..b94897a --- /dev/null +++ b/app/main/api/internal/types/adminplatformuser.go @@ -0,0 +1,66 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminCreatePlatformUserReq struct { + Mobile string `json:"mobile"` // 手机号 + Password string `json:"password"` // 密码 + Nickname string `json:"nickname"` // 昵称 + Info string `json:"info"` // 备注信息 + Inside int64 `json:"inside"` // 是否内部用户 1-是 0-否 +} + +type AdminCreatePlatformUserResp struct { + Id string `json:"id"` // 用户ID +} + +type AdminDeletePlatformUserReq struct { + Id string `path:"id"` // 用户ID +} + +type AdminDeletePlatformUserResp struct { + Success bool `json:"success"` // 是否成功 +} + +type AdminGetPlatformUserDetailReq struct { + Id string `path:"id"` // 用户ID +} + +type AdminGetPlatformUserDetailResp struct { + Id string `json:"id"` // 用户ID + Mobile string `json:"mobile"` // 手机号 + Nickname string `json:"nickname"` // 昵称 + Info string `json:"info"` // 备注信息 + Inside int64 `json:"inside"` // 是否内部用户 1-是 0-否 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + +type AdminGetPlatformUserListReq struct { + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"pageSize,default=20"` // 每页数量 + Mobile string `form:"mobile,optional"` // 手机号 + Nickname string `form:"nickname,optional"` // 昵称 + Inside int64 `form:"inside,optional"` // 是否内部用户 1-是 0-否 + CreateTimeStart string `form:"create_time_start,optional"` // 创建时间开始 + CreateTimeEnd string `form:"create_time_end,optional"` // 创建时间结束 + OrderBy string `form:"order_by,optional"` // 排序字段 + OrderType string `form:"order_type,optional"` // 排序类型 +} + +type AdminGetPlatformUserListResp struct { + Total int64 `json:"total"` // 总数 + Items []PlatformUserListItem `json:"items"` // 列表 +} + +type AdminUpdatePlatformUserReq struct { + Id string `path:"id"` // 用户ID + Mobile *string `json:"mobile,optional"` // 手机号 + Password *string `json:"password,optional"` // 密码 + Nickname *string `json:"nickname,optional"` // 昵称 + Info *string `json:"info,optional"` // 备注信息 + Inside *int64 `json:"inside,optional"` // 是否内部用户 1-是 0-否 +} + +type AdminUpdatePlatformUserResp struct { + Success bool `json:"success"` // 是否成功 +} diff --git a/app/main/api/internal/types/adminproduct.go b/app/main/api/internal/types/adminproduct.go new file mode 100644 index 0000000..9ccb519 --- /dev/null +++ b/app/main/api/internal/types/adminproduct.go @@ -0,0 +1,91 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminCreateProductReq struct { + ProductName string `json:"product_name"` // 服务名 + ProductEn string `json:"product_en"` // 英文名 + Description string `json:"description"` // 描述 + Notes string `json:"notes,optional"` // 备注 + CostPrice float64 `json:"cost_price"` // 成本 + SellPrice float64 `json:"sell_price"` // 售价 +} + +type AdminCreateProductResp struct { + Id string `json:"id"` // 产品ID +} + +type AdminDeleteProductReq struct { + Id string `path:"id"` // 产品ID +} + +type AdminDeleteProductResp struct { + Success bool `json:"success"` // 是否成功 +} + +type AdminGetProductDetailReq struct { + Id string `path:"id"` // 产品ID +} + +type AdminGetProductDetailResp struct { + Id string `json:"id"` // 产品ID + ProductName string `json:"product_name"` // 服务名 + ProductEn string `json:"product_en"` // 英文名 + Description string `json:"description"` // 描述 + Notes string `json:"notes"` // 备注 + CostPrice float64 `json:"cost_price"` // 成本 + SellPrice float64 `json:"sell_price"` // 售价 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + +type AdminGetProductFeatureListReq struct { + ProductId string `path:"product_id"` // 产品ID +} + +type AdminGetProductFeatureListResp struct { + Id string `json:"id"` // 关联ID + ProductId string `json:"product_id"` // 产品ID + FeatureId string `json:"feature_id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 功能描述 + Sort int64 `json:"sort"` // 排序 + Enable int64 `json:"enable"` // 是否启用 + IsImportant int64 `json:"is_important"` // 是否重要 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + +type AdminGetProductListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"pageSize"` // 每页数量 + ProductName *string `form:"product_name,optional"` // 服务名 + ProductEn *string `form:"product_en,optional"` // 英文名 +} + +type AdminGetProductListResp struct { + Total int64 `json:"total"` // 总数 + Items []ProductListItem `json:"items"` // 列表数据 +} + +type AdminUpdateProductFeaturesReq struct { + ProductId string `path:"product_id"` // 产品ID + Features []ProductFeatureItem `json:"features"` // 功能列表 +} + +type AdminUpdateProductFeaturesResp struct { + Success bool `json:"success"` // 是否成功 +} + +type AdminUpdateProductReq struct { + Id string `path:"id"` // 产品ID + ProductName *string `json:"product_name,optional"` // 服务名 + ProductEn *string `json:"product_en,optional"` // 英文名 + Description *string `json:"description,optional"` // 描述 + Notes *string `json:"notes,optional"` // 备注 + CostPrice *float64 `json:"cost_price,optional"` // 成本 + SellPrice *float64 `json:"sell_price,optional"` // 售价 +} + +type AdminUpdateProductResp struct { + Success bool `json:"success"` // 是否成功 +} diff --git a/app/main/api/internal/types/adminquery.go b/app/main/api/internal/types/adminquery.go new file mode 100644 index 0000000..d110276 --- /dev/null +++ b/app/main/api/internal/types/adminquery.go @@ -0,0 +1,60 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminGetQueryCleanupConfigListReq struct { + Status int64 `form:"status,optional"` // 状态：1-启用，0-禁用 +} + +type AdminGetQueryCleanupConfigListResp struct { + Items []QueryCleanupConfigItem `json:"items"` // 配置列表 +} + +type AdminGetQueryCleanupDetailListReq struct { + LogId string `path:"log_id"` // 清理日志ID + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"page_size,default=20"` // 每页数量 +} + +type AdminGetQueryCleanupDetailListResp struct { + Total int64 `json:"total"` // 总数 + Items []QueryCleanupDetailItem `json:"items"` // 列表 +} + +type AdminGetQueryCleanupLogListReq struct { + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"page_size,default=20"` // 每页数量 + Status int64 `form:"status,optional"` // 状态：1-成功，2-失败 + StartTime string `form:"start_time,optional"` // 开始时间 + EndTime string `form:"end_time,optional"` // 结束时间 +} + +type AdminGetQueryCleanupLogListResp struct { + Total int64 `json:"total"` // 总数 + Items []QueryCleanupLogItem `json:"items"` // 列表 +} + +type AdminGetQueryDetailByOrderIdReq struct { + OrderId string `path:"order_id"` +} + +type AdminGetQueryDetailByOrderIdResp struct { + Id string `json:"id"` // 主键ID + OrderId string `json:"order_id"` // 订单ID + UserId string `json:"user_id"` // 用户ID + ProductName string `json:"product_name"` // 产品ID + QueryParams map[string]interface{} `json:"query_params"` + QueryData []AdminQueryItem `json:"query_data"` + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + QueryState string `json:"query_state"` // 查询状态 +} + +type AdminUpdateQueryCleanupConfigReq struct { + Id string `json:"id"` // 主键ID + ConfigValue string `json:"config_value"` // 配置值 + Status int64 `json:"status"` // 状态：1-启用，0-禁用 +} + +type AdminUpdateQueryCleanupConfigResp struct { + Success bool `json:"success"` // 是否成功 +} diff --git a/app/main/api/internal/types/adminrole.go b/app/main/api/internal/types/adminrole.go new file mode 100644 index 0000000..202d50f --- /dev/null +++ b/app/main/api/internal/types/adminrole.go @@ -0,0 +1,66 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type CreateRoleReq struct { + RoleName string `json:"role_name"` // 角色名称 + RoleCode string `json:"role_code"` // 角色编码 + Description string `json:"description"` // 角色描述 + Status int64 `json:"status,default=1"` // 状态：0-禁用，1-启用 + Sort int64 `json:"sort,default=0"` // 排序 + MenuIds []string `json:"menu_ids"` // 关联的菜单ID列表 +} + +type CreateRoleResp struct { + Id string `json:"id"` // 角色ID +} + +type DeleteRoleReq struct { + Id string `path:"id"` // 角色ID +} + +type DeleteRoleResp struct { + Success bool `json:"success"` // 是否成功 +} + +type GetRoleDetailReq struct { + Id string `path:"id"` // 角色ID +} + +type GetRoleDetailResp struct { + Id string `json:"id"` // 角色ID + RoleName string `json:"role_name"` // 角色名称 + RoleCode string `json:"role_code"` // 角色编码 + Description string `json:"description"` // 角色描述 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + Sort int64 `json:"sort"` // 排序 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + MenuIds []string `json:"menu_ids"` // 关联的菜单ID列表 +} + +type GetRoleListReq struct { + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"pageSize,default=20"` // 每页数量 + Name string `form:"name,optional"` // 角色名称 + Code string `form:"code,optional"` // 角色编码 + Status int64 `form:"status,optional,default=-1"` // 状态：0-禁用，1-启用 +} + +type GetRoleListResp struct { + Total int64 `json:"total"` // 总数 + Items []RoleListItem `json:"items"` // 列表 +} + +type UpdateRoleReq struct { + Id string `path:"id"` // 角色ID + RoleName *string `json:"role_name,optional"` // 角色名称 + RoleCode *string `json:"role_code,optional"` // 角色编码 + Description *string `json:"description,optional"` // 角色描述 + Status *int64 `json:"status,optional"` // 状态：0-禁用，1-启用 + Sort *int64 `json:"sort,optional"` // 排序 + MenuIds []string `json:"menu_ids,optional"` // 关联的菜单ID列表 +} + +type UpdateRoleResp struct { + Success bool `json:"success"` // 是否成功 +} diff --git a/app/main/api/internal/types/adminroleapi.go b/app/main/api/internal/types/adminroleapi.go new file mode 100644 index 0000000..32030da --- /dev/null +++ b/app/main/api/internal/types/adminroleapi.go @@ -0,0 +1,45 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminAssignRoleApiReq struct { + RoleId string `json:"role_id"` + ApiIds []string `json:"api_ids"` +} + +type AdminAssignRoleApiResp struct { + Success bool `json:"success"` +} + +type AdminGetAllApiListReq struct { + Status int64 `form:"status,optional,default=1"` +} + +type AdminGetAllApiListResp struct { + Items []AdminRoleApiInfo `json:"items"` +} + +type AdminGetRoleApiListReq struct { + RoleId string `path:"role_id"` +} + +type AdminGetRoleApiListResp struct { + Items []AdminRoleApiInfo `json:"items"` +} + +type AdminRemoveRoleApiReq struct { + RoleId string `json:"role_id"` + ApiIds []string `json:"api_ids"` +} + +type AdminRemoveRoleApiResp struct { + Success bool `json:"success"` +} + +type AdminUpdateRoleApiReq struct { + RoleId string `json:"role_id"` + ApiIds []string `json:"api_ids"` +} + +type AdminUpdateRoleApiResp struct { + Success bool `json:"success"` +} diff --git a/app/main/api/internal/types/adminuser.go b/app/main/api/internal/types/adminuser.go new file mode 100644 index 0000000..e06158a --- /dev/null +++ b/app/main/api/internal/types/adminuser.go @@ -0,0 +1,78 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminCreateUserReq struct { + Username string `json:"username"` // 用户名 + RealName string `json:"real_name"` // 真实姓名 + Status int64 `json:"status,default=1"` // 状态：0-禁用，1-启用 + RoleIds []string `json:"role_ids"` // 关联的角色ID列表 +} + +type AdminCreateUserResp struct { + Id string `json:"id"` // 用户ID +} + +type AdminDeleteUserReq struct { + Id string `path:"id"` // 用户ID +} + +type AdminDeleteUserResp struct { + Success bool `json:"success"` // 是否成功 +} + +type AdminGetUserDetailReq struct { + Id string `path:"id"` // 用户ID +} + +type AdminGetUserDetailResp struct { + Id string `json:"id"` // 用户ID + Username string `json:"username"` // 用户名 + RealName string `json:"real_name"` // 真实姓名 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + RoleIds []string `json:"role_ids"` // 关联的角色ID列表 +} + +type AdminGetUserListReq struct { + Page int64 `form:"page,default=1"` // 页码 + PageSize int64 `form:"pageSize,default=20"` // 每页数量 + Username string `form:"username,optional"` // 用户名 + RealName string `form:"real_name,optional"` // 真实姓名 + Status int64 `form:"status,optional,default=-1"` // 状态：0-禁用，1-启用 +} + +type AdminGetUserListResp struct { + Total int64 `json:"total"` // 总数 + Items []AdminUserListItem `json:"items"` // 列表 +} + +type AdminResetPasswordReq struct { + Id string `path:"id"` // 用户ID + Password string `json:"password"` // 新密码 +} + +type AdminResetPasswordResp struct { + Success bool `json:"success"` // 是否成功 +} + +type AdminUpdateUserReq struct { + Id string `path:"id"` // 用户ID + Username *string `json:"username,optional"` // 用户名 + RealName *string `json:"real_name,optional"` // 真实姓名 + Status *int64 `json:"status,optional"` // 状态：0-禁用，1-启用 + RoleIds []string `json:"role_ids,optional"` // 关联的角色ID列表 +} + +type AdminUpdateUserResp struct { + Success bool `json:"success"` // 是否成功 +} + +type AdminUserInfoReq struct { +} + +type AdminUserInfoResp struct { + Username string `json:"username"` // 用户名 + RealName string `json:"real_name"` // 真实姓名 + Roles []string `json:"roles"` // 角色编码列表 +} diff --git a/app/main/api/internal/types/agent.go b/app/main/api/internal/types/agent.go new file mode 100644 index 0000000..3e72729 --- /dev/null +++ b/app/main/api/internal/types/agent.go @@ -0,0 +1,300 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AgentApplyReq struct { + Region string `json:"region,optional"` + Mobile string `json:"mobile"` + Code string `json:"code"` + Referrer string `json:"referrer"` + InviteCode string `json:"invite_code,optional"` + AgentCode int64 `json:"agent_code,optional"` +} + +type AgentApplyResp struct { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` + AgentCode int64 `json:"agent_code"` +} + +type AgentGeneratingLinkReq struct { + ProductId string `json:"product_id"` // 产品ID + SetPrice float64 `json:"set_price"` // 设定价格 + TargetPath string `json:"target_path,optional"` // 目标地址（可选，默认为推广报告页面） +} + +type AgentGeneratingLinkResp struct { + LinkIdentifier string `json:"link_identifier"` // 推广链接标识 + FullLink string `json:"full_link"` // 完整短链URL +} + +type AgentInfoResp struct { + AgentId string `json:"agent_id"` + Level int64 `json:"level"` + LevelName string `json:"level_name"` + Region string `json:"region"` + Mobile string `json:"mobile"` + WechatId string `json:"wechat_id"` + TeamLeaderId string `json:"team_leader_id"` + IsRealName bool `json:"is_real_name"` + AgentCode int64 `json:"agent_code"` +} + +type AgentProductConfigResp struct { + List []ProductConfigItem `json:"list"` +} + +type ApplyUpgradeReq struct { + ToLevel int64 `json:"to_level"` // 目标等级：2=黄金，3=钻石 +} + +type ApplyUpgradeResp struct { + UpgradeId string `json:"upgrade_id"` // 升级记录ID + OrderNo string `json:"order_no"` // 支付订单号 +} + +type ApplyWithdrawalReq struct { + Amount float64 `json:"amount"` // 提现金额 + PayeeAccount string `json:"payee_account"` // 收款账户 + PayeeName string `json:"payee_name"` // 收款人姓名 +} + +type ApplyWithdrawalResp struct { + WithdrawalId string `json:"withdrawal_id"` // 提现记录ID + WithdrawalNo string `json:"withdrawal_no"` // 提现单号 +} + +type ConversionRateResp struct { + MyConversionRate ConversionRateData `json:"my_conversion_rate"` // 我的转化率 + SubordinateConversionRate ConversionRateData `json:"subordinate_conversion_rate"` // 我的下级转化率 +} + +type DeleteInviteCodeReq struct { + Id string `json:"id"` // 邀请码ID +} + +type DeleteInviteCodeResp struct { +} + +type GenerateInviteCodeReq struct { + Count int64 `json:"count"` // 生成数量 + ExpireDays int64 `json:"expire_days,optional"` // 过期天数（可选，0表示不过期） + Remark string `json:"remark,optional"` // 备注（可选） +} + +type GenerateInviteCodeResp struct { + Codes []string `json:"codes"` // 生成的邀请码列表 +} + +type GetCommissionListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 +} + +type GetCommissionListResp struct { + Total int64 `json:"total"` // 总数 + List []CommissionItem `json:"list"` // 列表 +} + +type GetInviteCodeListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + Status int64 `form:"status,optional"` // 状态（可选） +} + +type GetInviteCodeListResp struct { + Total int64 `json:"total"` // 总数 + List []InviteCodeItem `json:"list"` // 列表 +} + +type GetInviteLinkReq struct { + InviteCode string `form:"invite_code"` // 邀请码 + TargetPath string `form:"target_path,optional"` // 目标地址（可选，默认为注册页面） +} + +type GetInviteLinkResp struct { + InviteLink string `json:"invite_link"` // 邀请链接 +} + +type GetLevelPrivilegeResp struct { + Levels []LevelPrivilegeItem `json:"levels"` + UpgradeToGoldFee float64 `json:"upgrade_to_gold_fee"` + UpgradeToDiamondFee float64 `json:"upgrade_to_diamond_fee"` + UpgradeToGoldRebate float64 `json:"upgrade_to_gold_rebate"` + UpgradeToDiamondRebate float64 `json:"upgrade_to_diamond_rebate"` +} + +type GetLinkDataReq struct { + LinkIdentifier string `form:"link_identifier"` // 推广链接标识 +} + +type GetLinkDataResp struct { + AgentId string `json:"agent_id"` // 代理ID + ProductId string `json:"product_id"` // 产品ID + SetPrice float64 `json:"set_price"` // 代理设定价格 + ActualBasePrice float64 `json:"actual_base_price"` // 实际底价 + ProductName string `json:"product_name"` // 产品名称 + ProductEn string `json:"product_en"` // 产品英文标识 + SellPrice float64 `json:"sell_price"` // 销售价格（使用代理设定价格） + Description string `json:"description"` // 产品描述 + Features []Feature `json:"features"` // 产品功能列表 +} + +type GetPromotionQueryListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 +} + +type GetPromotionQueryListResp struct { + Total int64 `json:"total"` // 总数 + List []PromotionQueryItem `json:"list"` // 列表 +} + +type GetRebateListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + RebateType *int64 `form:"rebate_type,optional"` // 返佣类型（可选）：1=直接上级返佣，2=钻石上级返佣，3=黄金上级返佣 +} + +type GetRebateListResp struct { + Total int64 `json:"total"` // 总数 + List []RebateItem `json:"list"` // 列表 +} + +type GetRevenueInfoResp struct { + Balance float64 `json:"balance"` // 可用余额 + FrozenBalance float64 `json:"frozen_balance"` // 冻结余额 + TotalEarnings float64 `json:"total_earnings"` // 累计收益（钱包总收益） + WithdrawnAmount float64 `json:"withdrawn_amount"` // 累计提现 + CommissionTotal float64 `json:"commission_total"` // 佣金累计总收益（推广订单获得的佣金） + CommissionToday float64 `json:"commission_today"` // 佣金今日收益 + CommissionMonth float64 `json:"commission_month"` // 佣金本月收益 + RebateTotal float64 `json:"rebate_total"` // 返佣累计总收益（包括推广返佣和升级返佣） + RebateToday float64 `json:"rebate_today"` // 返佣今日收益 + RebateMonth float64 `json:"rebate_month"` // 返佣本月收益 +} + +type GetSubordinateContributionDetailReq struct { + SubordinateId string `form:"subordinate_id"` // 下级代理ID + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + TabType string `form:"tab_type,optional"` // 标签页类型：order=订单列表，invite=邀请列表 +} + +type GetSubordinateContributionDetailResp struct { + Mobile string `json:"mobile"` // 手机号 + LevelName string `json:"level_name"` // 等级名称 + CreateTime string `json:"create_time"` // 创建时间 + OrderStats OrderStatistics `json:"order_stats"` // 订单统计（仅统计有返佣的订单） + RebateStats RebateStatistics `json:"rebate_stats"` // 返佣统计 + InviteStats InviteStatistics `json:"invite_stats"` // 邀请统计 + OrderList []OrderItem `json:"order_list"` // 订单列表（仅有贡献的订单） + InviteList []InviteItem `json:"invite_list"` // 邀请列表 + OrderListTotal int64 `json:"order_list_total"` // 订单列表总数 + InviteListTotal int64 `json:"invite_list_total"` // 邀请列表总数 +} + +type GetSubordinateListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 +} + +type GetSubordinateListResp struct { + Total int64 `json:"total"` // 总数 + List []SubordinateItem `json:"list"` // 列表 +} + +type GetTeamListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 + Mobile string `form:"mobile,optional"` // 手机号（可选，用于搜索） +} + +type GetTeamListResp struct { + Statistics TeamStatistics `json:"statistics"` // 统计数据 + Total int64 `json:"total"` // 总数 + List []TeamMemberItem `json:"list"` // 列表 +} + +type GetUpgradeListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 +} + +type GetUpgradeListResp struct { + Total int64 `json:"total"` // 总数 + List []UpgradeItem `json:"list"` // 列表 +} + +type GetUpgradeRebateListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 +} + +type GetUpgradeRebateListResp struct { + Total int64 `json:"total"` // 总数 + List []UpgradeRebateItem `json:"list"` // 列表 +} + +type GetWithdrawalListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数量 +} + +type GetWithdrawalListResp struct { + Total int64 `json:"total"` // 总数 + List []WithdrawalItem `json:"list"` // 列表 +} + +type RealNameAuthReq struct { + Name string `json:"name"` // 姓名 + IdCard string `json:"id_card"` // 身份证号 + Mobile string `json:"mobile"` // 手机号 + Code string `json:"code"` // 验证码 +} + +type RealNameAuthResp struct { + Status string `json:"status"` // 状态：pending=待审核，approved=已通过，rejected=已拒绝 +} + +type RegisterByInviteCodeReq struct { + Referrer string `json:"referrer"` + InviteCode string `json:"invite_code,optional"` + AgentCode int64 `json:"agent_code,optional"` + Mobile string `json:"mobile"` + Code string `json:"code"` + Region string `json:"region,optional"` + WechatId string `json:"wechat_id,optional"` +} + +type RegisterByInviteCodeResp struct { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` + AgentId string `json:"agent_id"` // 代理ID + Level int64 `json:"level"` // 代理等级 + LevelName string `json:"level_name"` // 等级名称 + AgentCode int64 `json:"agent_code"` +} + +type ShortLinkRedirectResp struct { +} + +type TeamStatisticsResp struct { + TotalCount int64 `json:"total_count"` // 团队总人数（不包括自己） + DirectCount int64 `json:"direct_count"` // 直接下级数量 + IndirectCount int64 `json:"indirect_count"` // 间接下级数量 + GoldCount int64 `json:"gold_count"` // 黄金代理数量 + NormalCount int64 `json:"normal_count"` // 普通代理数量 + TodayNewMembers int64 `json:"today_new_members"` // 今日新增成员 + MonthNewMembers int64 `json:"month_new_members"` // 本月新增成员 +} + +type UpgradeSubordinateReq struct { + SubordinateId string `json:"subordinate_id"` // 下级代理ID + ToLevel int64 `json:"to_level"` // 目标等级（只能是2=黄金） +} + +type UpgradeSubordinateResp struct { + Success bool `json:"success"` +} diff --git a/app/main/api/internal/types/app.go b/app/main/api/internal/types/app.go new file mode 100644 index 0000000..1308d40 --- /dev/null +++ b/app/main/api/internal/types/app.go @@ -0,0 +1,16 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type HealthCheckResp struct { + Status string `json:"status"` // 服务状态 + Message string `json:"message"` // 状态信息 +} + +type GetAppConfigResp struct { + QueryRetentionDays int64 `json:"query_retention_days"` +} + +type GetAppVersionResp struct { + Version string `json:"version"` + WgtUrl string `json:"wgtUrl"` +} diff --git a/app/main/api/internal/types/auth.go b/app/main/api/internal/types/auth.go new file mode 100644 index 0000000..b9d80d2 --- /dev/null +++ b/app/main/api/internal/types/auth.go @@ -0,0 +1,7 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type SendSmsReq struct { + Mobile string `json:"mobile" validate:"required,mobile"` + ActionType string `json:"actionType" validate:"required,oneof=login register query agentApply realName bindMobile"` +} diff --git a/app/main/api/internal/types/authorization.go b/app/main/api/internal/types/authorization.go new file mode 100644 index 0000000..fd60ff7 --- /dev/null +++ b/app/main/api/internal/types/authorization.go @@ -0,0 +1,36 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type DownloadAuthorizationDocumentReq struct { + DocumentId string `json:"documentId" validate:"required"` // 授权书ID +} + +type DownloadAuthorizationDocumentResp struct { + FileName string `json:"fileName"` // 文件名 + FileUrl string `json:"fileUrl"` // 文件访问URL +} + +type GetAuthorizationDocumentByOrderReq struct { + OrderId string `json:"orderId" validate:"required"` // 订单ID +} + +type GetAuthorizationDocumentByOrderResp struct { + Documents []AuthorizationDocumentInfo `json:"documents"` // 授权书列表 +} + +type GetAuthorizationDocumentReq struct { + DocumentId string `json:"documentId" validate:"required"` // 授权书ID +} + +type GetAuthorizationDocumentResp struct { + DocumentId string `json:"documentId"` // 授权书ID + UserId string `json:"userId"` // 用户ID + OrderId string `json:"orderId"` // 订单ID + QueryId string `json:"queryId"` // 查询ID + FileName string `json:"fileName"` // 文件名 + FileUrl string `json:"fileUrl"` // 文件访问URL + FileSize int64 `json:"fileSize"` // 文件大小 + FileType string `json:"fileType"` // 文件类型 + Status string `json:"status"` // 状态 + CreateTime string `json:"createTime"` // 创建时间 +} diff --git a/app/main/api/internal/types/cache.go b/app/main/api/internal/types/cache.go new file mode 100644 index 0000000..ecd68cc --- /dev/null +++ b/app/main/api/internal/types/cache.go @@ -0,0 +1,20 @@ +package types + +const QueryCacheKey = "query:%d:%s" +const AgentVipCacheKey = "agentVip:%d:%s" + +type QueryCache struct { + Name string `json:"name"` + IDCard string `json:"id_card"` + Mobile string `json:"mobile"` + Product string `json:"product_id"` +} +type QueryCacheLoad struct { + Product string `json:"product_en"` + Params string `json:"params"` + AgentIdentifier string `json:"agent_dentifier"` +} + +type AgentVipCache struct { + Type string `json:"type"` +} diff --git a/app/main/api/internal/types/encrypPayload.go b/app/main/api/internal/types/encrypPayload.go new file mode 100644 index 0000000..4c61dad --- /dev/null +++ b/app/main/api/internal/types/encrypPayload.go @@ -0,0 +1,6 @@ +package types + +type QueryShareLinkPayload struct { + OrderId string `json:"order_id"` + ExpireAt int64 `json:"expire_at"` +} diff --git a/app/main/api/internal/types/notification.go b/app/main/api/internal/types/notification.go new file mode 100644 index 0000000..ce76db9 --- /dev/null +++ b/app/main/api/internal/types/notification.go @@ -0,0 +1,7 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type GetNotificationsResp struct { + Notifications []Notification `json:"notifications"` // 通知列表 + Total int64 `json:"total"` // 总记录数 +} diff --git a/app/main/api/internal/types/pay.go b/app/main/api/internal/types/pay.go new file mode 100644 index 0000000..5e672a1 --- /dev/null +++ b/app/main/api/internal/types/pay.go @@ -0,0 +1,28 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type IapCallbackReq struct { + OrderID string `json:"order_id" validate:"required"` + TransactionReceipt string `json:"transaction_receipt" validate:"required"` +} + +type PaymentCheckReq struct { + OrderNo string `json:"order_no" validate:"required"` +} + +type PaymentCheckResp struct { + Type string `json:"type"` + Status string `json:"status"` +} + +type PaymentReq struct { + Id string `json:"id"` + PayMethod string `json:"pay_method"` // 支付方式: wechat, alipay, appleiap, test(仅开发环境), test_empty(仅开发环境-空报告模式) + PayType string `json:"pay_type" validate:"required,oneof=query agent_vip agent_upgrade"` +} + +type PaymentResp struct { + PrepayData interface{} `json:"prepay_data"` + PrepayId string `json:"prepay_id"` + OrderNo string `json:"order_no"` +} diff --git a/app/main/api/internal/types/payload.go b/app/main/api/internal/types/payload.go new file mode 100644 index 0000000..7bdbe3f --- /dev/null +++ b/app/main/api/internal/types/payload.go @@ -0,0 +1,13 @@ +package types + +type MsgPaySuccessQueryPayload struct { + OrderID string `json:"order_id"` +} + +type MsgAgentProcessPayload struct { + OrderID string `json:"order_id"` +} + +type MsgUnfreezeCommissionPayload struct { + FreezeTaskId string `json:"freeze_task_id"` // 冻结任务ID +} diff --git a/app/main/api/internal/types/product.go b/app/main/api/internal/types/product.go new file mode 100644 index 0000000..f5e6c9a --- /dev/null +++ b/app/main/api/internal/types/product.go @@ -0,0 +1,14 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type GetProductByEnRequest struct { + ProductEn string `path:"product_en"` +} + +type GetProductByIDRequest struct { + Id string `path:"id"` +} + +type ProductResponse struct { + Product +} diff --git a/app/main/api/internal/types/query.go b/app/main/api/internal/types/query.go new file mode 100644 index 0000000..012e8ad --- /dev/null +++ b/app/main/api/internal/types/query.go @@ -0,0 +1,90 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type QueryDetailByOrderIdReq struct { + OrderId string `path:"order_id"` +} + +type QueryDetailByOrderNoReq struct { + OrderNo string `path:"order_no"` +} + +type QueryExampleReq struct { + Feature string `form:"feature"` +} + +type QueryGenerateShareLinkReq struct { + OrderId *string `json:"order_id,optional"` + OrderNo *string `json:"order_no,optional"` +} + +type QueryGenerateShareLinkResp struct { + ShareLink string `json:"share_link"` +} + +type QueryListReq struct { + Page int64 `form:"page"` // 页码 + PageSize int64 `form:"page_size"` // 每页数据量 +} + +type QueryListResp struct { + Total int64 `json:"total"` // 总记录数 + List []Query `json:"list"` // 查询列表 +} + +type QueryProvisionalOrderReq struct { + Id string `path:"id"` +} + +type QueryProvisionalOrderResp struct { + Name string `json:"name"` + IdCard string `json:"id_card"` + Mobile string `json:"mobile"` + Product Product `json:"product"` +} + +type QueryRetryReq struct { + Id string `path:"id"` +} + +type QueryRetryResp struct { + Query +} + +type QueryServiceReq struct { + Product string `path:"product"` + Data string `json:"data" validate:"required"` + AgentIdentifier string `json:"agent_identifier,optional"` + App bool `json:"app,optional"` +} + +type QueryServiceResp struct { + Id string `json:"id"` + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` +} + +type QueryShareDetailReq struct { + Id string `path:"id"` +} + +type QuerySingleTestReq struct { + Params map[string]interface{} `json:"params"` + Api string `json:"api"` +} + +type QuerySingleTestResp struct { + Data interface{} `json:"data"` + Api string `json:"api"` +} + +type UpdateQueryDataReq struct { + Id string `json:"id"` // 查询ID + QueryData string `json:"query_data"` // 查询数据(未加密的JSON) +} + +type UpdateQueryDataResp struct { + Id string `json:"id"` + UpdatedAt string `json:"updated_at"` // 更新时间 +} diff --git a/app/main/api/internal/types/queryMap.go b/app/main/api/internal/types/queryMap.go new file mode 100644 index 0000000..775fc16 --- /dev/null +++ b/app/main/api/internal/types/queryMap.go @@ -0,0 +1,172 @@ +package types + +type MarriageReq struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` +} +type HomeServiceReq struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` +} + +// RiskAssessment 查询请求结构 +type RiskAssessmentReq struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` +} + +// CompanyInfo 查询请求结构 +type CompanyInfoReq struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` +} + +// RentalInfo 查询请求结构 +type RentalInfoReq struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` +} + +// PreLoanBackgroundCheck 查询请求结构 +type PreLoanBackgroundCheckReq struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` +} + +// BackgroundCheck 查询请求结构 +type BackgroundCheckReq struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` +} +type PersonalDataReq struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` +} + +type ConsumerFinanceReportReq struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` +} +type EntLawsuitReq struct { + EntName string `json:"ent_name" validate:"required,name"` + EntCode string `json:"ent_code" validate:"required,USCI"` + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` +} +type TocPhoneThreeElements struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` + Mobile string `json:"mobile" validate:"required,mobile"` +} +type TocPhoneTwoElements struct { + Name string `json:"name" validate:"required,name"` + Mobile string `json:"mobile" validate:"required,mobile"` +} +type TocIDCardTwoElements struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` +} +type TocDualMarriage struct { + NameMan string `json:"name_man" validate:"required,name"` + IDCardMan string `json:"id_card_man" validate:"required,idCard"` + NameWoman string `json:"name_woman" validate:"required,name"` + IDCardWoman string `json:"id_card_woman" validate:"required,idCard"` +} +type TocPersonVehicleVerification struct { + Name string `json:"name" validate:"required,name"` + CarType string `json:"car_type" validate:"required"` + CarLicense string `json:"car_license" validate:"required"` +} + +// 银行卡黑名单 +type TocBankCardBlacklist struct { + Name string `json:"name" validate:"required,name"` + IDCard string `json:"id_card" validate:"required,idCard"` + Mobile string `json:"mobile" validate:"required,mobile"` + BankCard string `json:"bank_card" validate:"required"` +} + +// 手机号码风险 +type TocPhoneNumberRisk struct { + Mobile string `json:"mobile" validate:"required,mobile"` +} + +// 手机二次卡 +type TocPhoneSecondaryCard struct { + Mobile string `json:"mobile" validate:"required,mobile"` + StartDate string `json:"start_date" validate:"required"` +} + +type AgentQueryData struct { + Mobile string `json:"mobile"` + Code string `json:"code"` +} +type AgentIdentifier struct { + AgentID string `json:"agent_id"` // 代理ID + ProductID string `json:"product_id"` // 产品ID + SetPrice float64 `json:"set_price"` // 代理设定价格 +} + +// 特殊名单 G26BJ05 +var G26BJ05FieldMapping = map[string]string{ + "IDCard": "id", + "Name": "name", + "Mobile": "cell", + "TimeRange": "time_range", +} + +// 个人不良 +var G34BJ03FieldMapping = map[string]string{ + "IDCard": "id_card", + "Name": "name", +} + +// 个人涉诉 G35SC01 +var G35SC01FieldMapping = map[string]string{ + "Name": "name", + "IDCard": "idcard", + "InquiredAuth": "inquired_auth", +} + +// 单人婚姻 G09SC02 +var G09SC02FieldMapping = map[string]string{ + "IDCard": "certNumMan", + "Name": "nameMan", +} + +// 借贷意向 G27BJ05 +var G27BJ05FieldMapping = map[string]string{ + "IDCard": "id", + "Name": "name", + "Mobile": "cell", +} + +// 借贷行为 G28BJ05 +var G28BJ05FieldMapping = map[string]string{ + "IDCard": "id", + "Name": "name", + "Mobile": "cell", +} + +// 股东人企关系精准版 G05HZ01 +var G05HZ01FieldMapping = map[string]string{ + "IDCard": "pid", +} diff --git a/app/main/api/internal/types/queryParams.go b/app/main/api/internal/types/queryParams.go new file mode 100644 index 0000000..1c243d5 --- /dev/null +++ b/app/main/api/internal/types/queryParams.go @@ -0,0 +1,73 @@ +package types + +type WestDexServiceRequestParams struct { + FieldMapping map[string]string + ApiID string +} + +var WestDexParams = map[string][]WestDexServiceRequestParams{ + "marriage": { + {FieldMapping: G09SC02FieldMapping, ApiID: "G09SC02"}, // 单人婚姻 + {FieldMapping: G27BJ05FieldMapping, ApiID: "G27BJ05"}, // 借贷意向 + {FieldMapping: G28BJ05FieldMapping, ApiID: "G28BJ05"}, // 借贷行为 + {FieldMapping: G26BJ05FieldMapping, ApiID: "G26BJ05"}, // 特殊名单 + {FieldMapping: G34BJ03FieldMapping, ApiID: "G34BJ03"}, // 个人不良 + {FieldMapping: G35SC01FieldMapping, ApiID: "G35SC01"}, // 个人涉诉 + {FieldMapping: G05HZ01FieldMapping, ApiID: "G05HZ01"}, // 股东人企关系 + + }, + "backgroundcheck": { + {FieldMapping: G09SC02FieldMapping, ApiID: "G09SC02"}, // 单人婚姻 + {FieldMapping: G27BJ05FieldMapping, ApiID: "G27BJ05"}, // 借贷意向 + {FieldMapping: G28BJ05FieldMapping, ApiID: "G28BJ05"}, // 借贷行为 + {FieldMapping: G26BJ05FieldMapping, ApiID: "G26BJ05"}, // 特殊名单 + {FieldMapping: G05HZ01FieldMapping, ApiID: "G05HZ01"}, // 股东人企关系 + {FieldMapping: G34BJ03FieldMapping, ApiID: "G34BJ03"}, // 个人不良 + {FieldMapping: G35SC01FieldMapping, ApiID: "G35SC01"}, // 个人涉诉 + }, + "companyinfo": { + {FieldMapping: G09SC02FieldMapping, ApiID: "G09SC02"}, // 单人婚姻 + {FieldMapping: G27BJ05FieldMapping, ApiID: "G27BJ05"}, // 借贷意向 + {FieldMapping: G28BJ05FieldMapping, ApiID: "G28BJ05"}, // 借贷行为 + {FieldMapping: G26BJ05FieldMapping, ApiID: "G26BJ05"}, // 特殊名单 + {FieldMapping: G05HZ01FieldMapping, ApiID: "G05HZ01"}, // 股东人企关系 + {FieldMapping: G34BJ03FieldMapping, ApiID: "G34BJ03"}, // 个人不良 + {FieldMapping: G35SC01FieldMapping, ApiID: "G35SC01"}, // 个人涉诉 + }, + "homeservice": { + {FieldMapping: G09SC02FieldMapping, ApiID: "G09SC02"}, // 单人婚姻 + {FieldMapping: G27BJ05FieldMapping, ApiID: "G27BJ05"}, // 借贷意向 + {FieldMapping: G28BJ05FieldMapping, ApiID: "G28BJ05"}, // 借贷行为 + {FieldMapping: G26BJ05FieldMapping, ApiID: "G26BJ05"}, // 特殊名单 + {FieldMapping: G05HZ01FieldMapping, ApiID: "G05HZ01"}, // 股东人企关系 + {FieldMapping: G34BJ03FieldMapping, ApiID: "G34BJ03"}, // 个人不良 + {FieldMapping: G35SC01FieldMapping, ApiID: "G35SC01"}, // 个人涉诉 + }, + "preloanbackgroundcheck": { + {FieldMapping: G09SC02FieldMapping, ApiID: "G09SC02"}, // 单人婚姻 + {FieldMapping: G27BJ05FieldMapping, ApiID: "G27BJ05"}, // 借贷意向 + {FieldMapping: G28BJ05FieldMapping, ApiID: "G28BJ05"}, // 借贷行为 + {FieldMapping: G26BJ05FieldMapping, ApiID: "G26BJ05"}, // 特殊名单 + {FieldMapping: G05HZ01FieldMapping, ApiID: "G05HZ01"}, // 股东人企关系 + {FieldMapping: G34BJ03FieldMapping, ApiID: "G34BJ03"}, // 个人不良 + {FieldMapping: G35SC01FieldMapping, ApiID: "G35SC01"}, // 个人涉诉 + }, + "rentalinfo": { + {FieldMapping: G09SC02FieldMapping, ApiID: "G09SC02"}, // 单人婚姻 + {FieldMapping: G27BJ05FieldMapping, ApiID: "G27BJ05"}, // 借贷意向 + {FieldMapping: G28BJ05FieldMapping, ApiID: "G28BJ05"}, // 借贷行为 + {FieldMapping: G26BJ05FieldMapping, ApiID: "G26BJ05"}, // 特殊名单 + {FieldMapping: G05HZ01FieldMapping, ApiID: "G05HZ01"}, // 股东人企关系 + {FieldMapping: G34BJ03FieldMapping, ApiID: "G34BJ03"}, // 个人不良 + {FieldMapping: G35SC01FieldMapping, ApiID: "G35SC01"}, // 个人涉诉 + }, + "riskassessment": { + {FieldMapping: G09SC02FieldMapping, ApiID: "G09SC02"}, // 单人婚姻 + {FieldMapping: G27BJ05FieldMapping, ApiID: "G27BJ05"}, // 借贷意向 + {FieldMapping: G28BJ05FieldMapping, ApiID: "G28BJ05"}, // 借贷行为 + {FieldMapping: G26BJ05FieldMapping, ApiID: "G26BJ05"}, // 特殊名单 + {FieldMapping: G05HZ01FieldMapping, ApiID: "G05HZ01"}, // 股东人企关系 + {FieldMapping: G34BJ03FieldMapping, ApiID: "G34BJ03"}, // 个人不良 + {FieldMapping: G35SC01FieldMapping, ApiID: "G35SC01"}, // 个人涉诉 + }, +} diff --git a/app/main/api/internal/types/taskname.go b/app/main/api/internal/types/taskname.go new file mode 100644 index 0000000..4a7b9e6 --- /dev/null +++ b/app/main/api/internal/types/taskname.go @@ -0,0 +1,7 @@ +package types + +const MsgPaySuccessQuery = "msg:pay_success:query" +const MsgCleanQueryData = "msg:clean_query_data" +const MsgAgentProcess = "msg:agent:process" +const MsgUnfreezeCommission = "msg:unfreeze:commission" +const MsgUnfreezeCommissionScan = "msg:unfreeze:commission:scan" diff --git a/app/main/api/internal/types/types.go b/app/main/api/internal/types/types.go new file mode 100644 index 0000000..d8c406d --- /dev/null +++ b/app/main/api/internal/types/types.go @@ -0,0 +1,576 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AdminApiInfo struct { + Id string `json:"id"` + ApiName string `json:"api_name"` + ApiCode string `json:"api_code"` + Method string `json:"method"` + Url string `json:"url"` + Status int64 `json:"status"` + Description string `json:"description"` + CreateTime string `json:"create_time"` + UpdateTime string `json:"update_time"` +} + +type AdminAuditRealNameReq struct { + RealNameId int64 `json:"real_name_id"` // 实名认证记录ID + Status int64 `json:"status"` // 审核状态：2=通过，3=拒绝 + AuditReason string `json:"audit_reason"` // 审核原因（拒绝时必填） +} + +type AdminAuditRealNameResp struct { + Success bool `json:"success"` +} + +type AdminQueryItem struct { + Feature interface{} `json:"feature"` + Data interface{} `json:"data"` // 这里可以是 map 或 具体的 struct +} + +type AdminRoleApiInfo struct { + Id string `json:"id"` + RoleId string `json:"role_id"` + ApiId string `json:"api_id"` + ApiName string `json:"api_name"` + ApiCode string `json:"api_code"` + Method string `json:"method"` + Url string `json:"url"` + Status int64 `json:"status"` + Description string `json:"description"` +} + +type AdminUserListItem struct { + Id string `json:"id"` // 用户ID + Username string `json:"username"` // 用户名 + RealName string `json:"real_name"` // 真实姓名 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + CreateTime string `json:"create_time"` // 创建时间 + RoleIds []string `json:"role_ids"` // 关联的角色ID列表 +} + +type AgentCommissionListItem struct { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + OrderId string `json:"order_id"` // 订单ID + ProductName string `json:"product_name"` // 产品名称 + Amount float64 `json:"amount"` // 金额 + Status int64 `json:"status"` // 状态 + CreateTime string `json:"create_time"` // 创建时间 +} + +type AgentLinkListItem struct { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + ProductId string `json:"product_id"` // 产品ID + ProductName string `json:"product_name"` // 产品名称 + SetPrice float64 `json:"set_price"` // 设定价格 + ActualBasePrice float64 `json:"actual_base_price"` // 实际底价 + LinkIdentifier string `json:"link_identifier"` // 推广码 + CreateTime string `json:"create_time"` // 创建时间 +} + +type AgentListItem struct { + Id string `json:"id"` // 主键 + UserId string `json:"user_id"` // 用户ID + Level int64 `json:"level"` // 等级：1=普通，2=黄金，3=钻石 + LevelName string `json:"level_name"` // 等级名称 + Region string `json:"region"` // 区域 + Mobile string `json:"mobile"` // 手机号 + WechatId string `json:"wechat_id"` // 微信号 + TeamLeaderId string `json:"team_leader_id"` // 团队首领ID + AgentCode int64 `json:"agent_code"` + Balance float64 `json:"balance"` // 钱包余额 + TotalEarnings float64 `json:"total_earnings"` // 累计收益 + FrozenBalance float64 `json:"frozen_balance"` // 冻结余额 + WithdrawnAmount float64 `json:"withdrawn_amount"` // 提现总额 + IsRealName bool `json:"is_real_name"` // 是否已实名 + CreateTime string `json:"create_time"` // 创建时间 +} + +type AgentOrderListItem struct { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + OrderId string `json:"order_id"` // 订单ID + ProductId string `json:"product_id"` // 产品ID + ProductName string `json:"product_name"` // 产品名称 + OrderAmount float64 `json:"order_amount"` // 订单金额 + SetPrice float64 `json:"set_price"` // 设定价格 + ActualBasePrice float64 `json:"actual_base_price"` // 实际底价 + PriceCost float64 `json:"price_cost"` // 提价成本 + AgentProfit float64 `json:"agent_profit"` // 代理收益 + ProcessStatus int64 `json:"process_status"` // 处理状态 + CreateTime string `json:"create_time"` // 创建时间 +} + +type AgentProductConfigItem struct { + Id string `json:"id"` // 主键 + ProductId string `json:"product_id"` // 产品ID + ProductName string `json:"product_name"` // 产品名称 + BasePrice float64 `json:"base_price"` // 基础底价 + PriceRangeMin float64 `json:"price_range_min"` // 最低定价 + PriceRangeMax float64 `json:"price_range_max"` // 最高定价 + PriceThreshold float64 `json:"price_threshold"` // 提价标准阈值 + PriceFeeRate float64 `json:"price_fee_rate"` // 提价手续费比例 + CreateTime string `json:"create_time"` // 创建时间 +} + +type AgentRealNameListItem struct { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + Name string `json:"name"` // 姓名 + IdCard string `json:"id_card"` // 身份证号 + Mobile string `json:"mobile"` // 手机号 + Status int64 `json:"status"` // 状态：1=未验证，2=已通过（verify_time不为空表示已通过） + VerifyTime string `json:"verify_time"` // 验证时间（三要素核验通过时间） + CreateTime string `json:"create_time"` // 创建时间 +} + +type AgentRebateListItem struct { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 获得返佣的代理ID + SourceAgentId string `json:"source_agent_id"` // 来源代理ID + OrderId string `json:"order_id"` // 订单ID + RebateType int64 `json:"rebate_type"` // 返佣类型 + Amount float64 `json:"amount"` // 金额 + CreateTime string `json:"create_time"` // 创建时间 +} + +type AgentUpgradeListItem struct { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + FromLevel int64 `json:"from_level"` // 原等级 + ToLevel int64 `json:"to_level"` // 目标等级 + UpgradeType int64 `json:"upgrade_type"` // 升级类型 + UpgradeFee float64 `json:"upgrade_fee"` // 升级费用 + RebateAmount float64 `json:"rebate_amount"` // 返佣金额 + Status int64 `json:"status"` // 状态 + CreateTime string `json:"create_time"` // 创建时间 +} + +type AgentWithdrawalListItem struct { + Id string `json:"id"` // 主键 + AgentId string `json:"agent_id"` // 代理ID + WithdrawNo string `json:"withdraw_no"` // 提现单号 + Amount float64 `json:"amount"` // 金额 + TaxAmount float64 `json:"tax_amount"` // 税费金额 + ActualAmount float64 `json:"actual_amount"` // 实际到账金额 + Status int64 `json:"status"` // 状态 + PayeeAccount string `json:"payee_account"` // 收款账户 + PayeeName string `json:"payee_name"` // 收款人姓名 + Remark string `json:"remark"` // 备注 + CreateTime string `json:"create_time"` // 创建时间 +} + +type AuthorizationDocumentInfo struct { + DocumentId string `json:"documentId"` // 授权书ID + UserId string `json:"userId"` // 用户ID + OrderId string `json:"orderId"` // 订单ID + QueryId string `json:"queryId"` // 查询ID + FileName string `json:"fileName"` // 文件名 + FileUrl string `json:"fileUrl"` // 文件访问URL + FileSize int64 `json:"fileSize"` // 文件大小 + FileType string `json:"fileType"` // 文件类型 + Status string `json:"status"` // 状态 + CreateTime string `json:"createTime"` // 创建时间 +} + +type CommissionFreezeConfig struct { + Ratio float64 `json:"ratio"` // 佣金冻结比例（例如：0.1表示10%） + Threshold float64 `json:"threshold"` // 佣金冻结阈值（订单单价达到此金额才触发冻结，单位：元） + Days int64 `json:"days"` // 佣金冻结解冻天数（单位：天，例如：30表示30天后解冻） +} + +type CommissionItem struct { + Id string `json:"id"` // 记录ID + OrderId string `json:"order_id"` // 订单ID + OrderNo string `json:"order_no"` // 订单号 + ProductName string `json:"product_name"` // 产品名称 + Amount float64 `json:"amount"` // 佣金金额 + Status int64 `json:"status"` // 状态：1=已发放，2=已冻结，3=已取消 + CreateTime string `json:"create_time"` // 创建时间 +} + +type ConversionRateData struct { + Daily []PeriodConversionData `json:"daily"` // 日统计（今日、昨日、前日） + Weekly []PeriodConversionData `json:"weekly"` // 周统计（这周、前一周、前两周） + Monthly []PeriodConversionData `json:"monthly"` // 月统计（本月、上月、前两月） +} + +type DirectParentRebateConfig struct { + Diamond float64 `json:"diamond"` // 直接上级是钻石的返佣金额（6元） + Gold float64 `json:"gold"` // 直接上级是黄金的返佣金额（3元） + Normal float64 `json:"normal"` // 直接上级是普通的返佣金额（2元） +} + +type Feature struct { + ID string `json:"id"` // 功能ID + ApiID string `json:"api_id"` // API标识 + Name string `json:"name"` // 功能描述 +} + +type FeatureListItem struct { + Id string `json:"id"` // 功能ID + ApiId string `json:"api_id"` // API标识 + Name string `json:"name"` // 描述 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + +type InviteCodeItem struct { + Id string `json:"id"` // 记录ID + Code string `json:"code"` // 邀请码 + TargetLevel int64 `json:"target_level"` // 目标等级 + Status int64 `json:"status"` // 状态：0=未使用，1=已使用，2=已失效 + UsedTime string `json:"used_time"` // 使用时间 + ExpireTime string `json:"expire_time"` // 过期时间 + Remark string `json:"remark"` // 备注 + CreateTime string `json:"create_time"` // 创建时间 +} + +type InviteCodeListItem struct { + Id string `json:"id"` // 主键 + Code string `json:"code"` // 邀请码 + AgentId string `json:"agent_id"` // 发放代理ID（0表示平台发放） + AgentMobile string `json:"agent_mobile"` // 发放代理手机号 + TargetLevel int64 `json:"target_level"` // 目标等级 + Status int64 `json:"status"` // 状态：0=未使用，1=已使用，2=已失效 + UsedUserId string `json:"used_user_id"` // 使用用户ID + UsedAgentId string `json:"used_agent_id"` // 使用代理ID + UsedTime string `json:"used_time"` // 使用时间 + ExpireTime string `json:"expire_time"` // 过期时间 + Remark string `json:"remark"` // 备注 + CreateTime string `json:"create_time"` // 创建时间 +} + +type InviteItem struct { + AgentId string `json:"agent_id"` // 代理ID + Level int64 `json:"level"` // 等级 + LevelName string `json:"level_name"` // 等级名称 + Mobile string `json:"mobile"` // 手机号 + CreateTime string `json:"create_time"` // 邀请时间 +} + +type InviteStatistics struct { + TotalInvites int64 `json:"total_invites"` // 总的邀请 + TodayInvites int64 `json:"today_invites"` // 今日邀请 + MonthInvites int64 `json:"month_invites"` // 月邀请 +} + +type LevelBonusConfig struct { + Diamond int64 `json:"diamond"` // 钻石加成：0 + Gold int64 `json:"gold"` // 黄金加成：3 + Normal int64 `json:"normal"` // 普通加成：6 +} + +type LevelPrivilegeItem struct { + Level int64 `json:"level"` // 等级：1=普通，2=黄金，3=钻石 + LevelName string `json:"level_name"` // 等级名称 + LevelBonus float64 `json:"level_bonus"` // 等级加成（元） + PriceReduction float64 `json:"price_reduction"` // 底价降低（元，相对于当前等级） + PromoteRebate string `json:"promote_rebate"` // 推广返佣说明 + MaxPromoteRebate float64 `json:"max_promote_rebate"` // 最高推广返佣金额（元） + UpgradeRebate string `json:"upgrade_rebate"` // 下级升级返佣说明 + Privileges []string `json:"privileges"` // 特权列表 + CanUpgradeSubordinate bool `json:"can_upgrade_subordinate"` // 是否可以升级下级 +} + +type Notification struct { + Title string `json:"title"` // 通知标题 + Content string `json:"content"` // 通知内容 (富文本) + NotificationPage string `json:"notificationPage"` // 通知页面 + StartDate string `json:"startDate"` // 通知开始日期，格式 "YYYY-MM-DD" + EndDate string `json:"endDate"` // 通知结束日期，格式 "YYYY-MM-DD" + StartTime string `json:"startTime"` // 每天通知开始时间，格式 "HH:MM:SS" + EndTime string `json:"endTime"` // 每天通知结束时间，格式 "HH:MM:SS" +} + +type NotificationListItem struct { + Id string `json:"id"` // 通知ID + Title string `json:"title"` // 通知标题 + NotificationPage string `json:"notification_page"` // 通知页面 + Content string `json:"content"` // 通知内容 + StartDate string `json:"start_date"` // 生效开始日期 + StartTime string `json:"start_time"` // 生效开始时间 + EndDate string `json:"end_date"` // 生效结束日期 + EndTime string `json:"end_time"` // 生效结束时间 + Status int64 `json:"status"` // 状态 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + +type OrderItem struct { + OrderNo string `json:"order_no"` // 订单号 + ProductId string `json:"product_id"` // 产品ID + ProductName string `json:"product_name"` // 产品名称 + OrderAmount float64 `json:"order_amount"` // 订单金额 + RebateAmount float64 `json:"rebate_amount"` // 返佣金额 + CreateTime string `json:"create_time"` // 创建时间 +} + +type OrderListItem struct { + Id string `json:"id"` // 订单ID + OrderNo string `json:"order_no"` // 商户订单号 + PlatformOrderId string `json:"platform_order_id"` // 支付订单号 + ProductName string `json:"product_name"` // 产品名称 + PaymentPlatform string `json:"payment_platform"` // 支付方式 + PaymentScene string `json:"payment_scene"` // 支付平台 + Amount float64 `json:"amount"` // 金额 + Status string `json:"status"` // 支付状态：pending-待支付，paid-已支付，refunded-已退款，closed-已关闭，failed-支付失败 + QueryState string `json:"query_state"` // 查询状态：pending-待查询，success-查询成功，failed-查询失败 processing-查询中 + CreateTime string `json:"create_time"` // 创建时间 + PayTime string `json:"pay_time"` // 支付时间 + RefundTime string `json:"refund_time"` // 退款时间 + IsAgentOrder bool `json:"is_agent_order"` // 是否是代理订单 + AgentProcessStatus string `json:"agent_process_status"` // 代理事务处理状态：not_agent-非代理订单，success-处理成功，failed-处理失败，pending-待处理 +} + +type OrderStatistics struct { + TotalOrders int64 `json:"total_orders"` // 总订单量（仅统计有返佣的订单） + MonthOrders int64 `json:"month_orders"` // 月订单 + TodayOrders int64 `json:"today_orders"` // 今日订单 +} + +type PeriodConversionData struct { + PeriodLabel string `json:"period_label"` // 时间段标签（如：今日、昨日、前日） + QueryUsers int64 `json:"query_users"` // 查询订单数（订单数量，不是用户数） + PaidUsers int64 `json:"paid_users"` // 付费订单数（订单数量，不是用户数） + TotalAmount float64 `json:"total_amount"` // 总金额（已支付订单的总金额） + QueryUserCount int64 `json:"query_user_count"` // 查询用户数（去重user_id后的用户数量） + PaidUserCount int64 `json:"paid_user_count"` // 付费用户数（去重user_id后的用户数量） +} + +type PlatformUserListItem struct { + Id string `json:"id"` // 用户ID + Mobile string `json:"mobile"` // 手机号 + Nickname string `json:"nickname"` // 昵称 + Info string `json:"info"` // 备注信息 + Inside int64 `json:"inside"` // 是否内部用户 1-是 0-否 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + +type Product struct { + ProductName string `json:"product_name"` + ProductEn string `json:"product_en"` + Description string `json:"description"` + Notes string `json:"notes,optional"` + SellPrice float64 `json:"sell_price"` + Features []Feature `json:"features"` // 关联功能列表 +} + +type ProductConfigItem struct { + ProductId string `json:"product_id"` // 产品ID + ProductName string `json:"product_name"` // 产品名称 + ProductEn string `json:"product_en"` // 产品英文标识 + ActualBasePrice float64 `json:"actual_base_price"` // 实际底价 + PriceRangeMin float64 `json:"price_range_min"` // 最低价格 + PriceRangeMax float64 `json:"price_range_max"` // 最高价格 + PriceThreshold float64 `json:"price_threshold"` // 提价标准阈值 + PriceFeeRate float64 `json:"price_fee_rate"` // 提价手续费比例 +} + +type ProductFeatureItem struct { + FeatureId string `json:"feature_id"` // 功能ID + Sort int64 `json:"sort"` // 排序 + Enable int64 `json:"enable"` // 是否启用 + IsImportant int64 `json:"is_important"` // 是否重要 +} + +type ProductListItem struct { + Id string `json:"id"` // 产品ID + ProductName string `json:"product_name"` // 服务名 + ProductEn string `json:"product_en"` // 英文名 + Description string `json:"description"` // 描述 + Notes string `json:"notes"` // 备注 + CostPrice float64 `json:"cost_price"` // 成本 + SellPrice float64 `json:"sell_price"` // 售价 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + +type PromotionQueryItem struct { + Id string `json:"id"` // 查询ID + OrderId string `json:"order_id"` // 订单ID + ProductName string `json:"product_name"` // 产品名称 + CreateTime string `json:"create_time"` // 创建时间 + QueryState string `json:"query_state"` // 查询状态 +} + +type Query struct { + Id string `json:"id"` // 主键ID + OrderId string `json:"order_id"` // 订单ID + UserId string `json:"user_id"` // 用户ID + Product string `json:"product"` // 产品ID + ProductName string `json:"product_name"` // 产品ID + QueryParams map[string]interface{} `json:"query_params"` + QueryData []QueryItem `json:"query_data"` + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 + QueryState string `json:"query_state"` // 查询状态 +} + +type QueryCleanupConfigItem struct { + Id string `json:"id"` // 主键ID + ConfigKey string `json:"config_key"` // 配置键 + ConfigValue string `json:"config_value"` // 配置值 + ConfigDesc string `json:"config_desc"` // 配置描述 + Status int64 `json:"status"` // 状态：1-启用，0-禁用 + CreateTime string `json:"create_time"` // 创建时间 + UpdateTime string `json:"update_time"` // 更新时间 +} + +type QueryCleanupDetailItem struct { + Id string `json:"id"` // 主键ID + CleanupLogId string `json:"cleanup_log_id"` // 清理日志ID + QueryId string `json:"query_id"` // 查询ID + OrderId string `json:"order_id"` // 订单ID + UserId string `json:"user_id"` // 用户ID + ProductName string `json:"product_name"` // 产品名称 + QueryState string `json:"query_state"` // 查询状态 + CreateTimeOld string `json:"create_time_old"` // 原创建时间 + CreateTime string `json:"create_time"` // 创建时间 +} + +type QueryCleanupLogItem struct { + Id string `json:"id"` // 主键ID + CleanupTime string `json:"cleanup_time"` // 清理时间 + CleanupBefore string `json:"cleanup_before"` // 清理截止时间 + Status int64 `json:"status"` // 状态：1-成功，2-失败 + AffectedRows int64 `json:"affected_rows"` // 影响行数 + ErrorMsg string `json:"error_msg"` // 错误信息 + Remark string `json:"remark"` // 备注 + CreateTime string `json:"create_time"` // 创建时间 +} + +type QueryItem struct { + Feature interface{} `json:"feature"` + Data interface{} `json:"data"` // 这里可以是 map 或 具体的 struct +} + +type QueryReq struct { + Data string `json:"data" validate:"required"` +} + +type QueryResp struct { + Id string `json:"id"` +} + +type RebateItem struct { + Id string `json:"id"` // 记录ID + SourceAgentId string `json:"source_agent_id"` // 来源代理ID + SourceAgentMobile string `json:"source_agent_mobile"` // 来源代理手机号 + SourceAgentLevel int64 `json:"source_agent_level"` // 来源代理等级：1=普通，2=黄金，3=钻石 + OrderId string `json:"order_id"` // 订单ID + OrderNo string `json:"order_no"` // 订单号 + RebateType int64 `json:"rebate_type"` // 返佣类型：1=直接上级，2=钻石上级，3=黄金上级 + Amount float64 `json:"amount"` // 返佣金额 + CreateTime string `json:"create_time"` // 创建时间 +} + +type RebateStatistics struct { + TotalRebateAmount float64 `json:"total_rebate_amount"` // 总贡献返佣金额 + TodayRebateAmount float64 `json:"today_rebate_amount"` // 今日贡献返佣金额 + MonthRebateAmount float64 `json:"month_rebate_amount"` // 月贡献返佣金额 +} + +type RoleListItem struct { + Id string `json:"id"` // 角色ID + RoleName string `json:"role_name"` // 角色名称 + RoleCode string `json:"role_code"` // 角色编码 + Description string `json:"description"` // 角色描述 + Status int64 `json:"status"` // 状态：0-禁用，1-启用 + Sort int64 `json:"sort"` // 排序 + CreateTime string `json:"create_time"` // 创建时间 + MenuIds []string `json:"menu_ids"` // 关联的菜单ID列表 +} + +type SubordinateItem struct { + AgentId string `json:"agent_id"` // 代理ID + Level int64 `json:"level"` // 等级 + LevelName string `json:"level_name"` // 等级名称 + Mobile string `json:"mobile"` // 手机号 + CreateTime string `json:"create_time"` // 创建时间 + TotalOrders int64 `json:"total_orders"` // 总订单数 + TotalAmount float64 `json:"total_amount"` // 总金额 +} + +type TeamMemberItem struct { + AgentId string `json:"agent_id"` // 代理ID + Level int64 `json:"level"` // 等级 + LevelName string `json:"level_name"` // 等级名称 + Mobile string `json:"mobile"` // 手机号 + CreateTime string `json:"create_time"` // 加入团队时间 + TodayQueries int64 `json:"today_queries"` // 当日查询量 + TotalQueries int64 `json:"total_queries"` // 总查询量 + TotalRebateAmount float64 `json:"total_rebate_amount"` // 返佣给我的总金额 + TodayRebateAmount float64 `json:"today_rebate_amount"` // 返佣给我的今日金额 + TotalInvites int64 `json:"total_invites"` // 邀请加入团队的总人数 + TodayInvites int64 `json:"today_invites"` // 邀请加入团队的今日人数 + IsDirect bool `json:"is_direct"` // 是否直接下级 +} + +type TeamStatistics struct { + TotalMembers int64 `json:"total_members"` // 团队成员总数 + TodayNewMembers int64 `json:"today_new_members"` // 团队人员今日新增 + MonthNewMembers int64 `json:"month_new_members"` // 团队人员月新增 + TotalQueries int64 `json:"total_queries"` // 团队总查询总量 + TodayPromotions int64 `json:"today_promotions"` // 团队今日推广量 + MonthPromotions int64 `json:"month_promotions"` // 团队月推广量 + TotalEarnings float64 `json:"total_earnings"` // 依靠团队得到的总收益 + TodayEarnings float64 `json:"today_earnings"` // 今日收益 + MonthEarnings float64 `json:"month_earnings"` // 月收益 +} + +type UpgradeFeeConfig struct { + NormalToGold float64 `json:"normal_to_gold"` // 普通→黄金：199 + NormalToDiamond float64 `json:"normal_to_diamond"` // 普通→钻石：980 +} + +type UpgradeItem struct { + Id string `json:"id"` // 记录ID + AgentId string `json:"agent_id"` // 代理ID + FromLevel int64 `json:"from_level"` // 原等级 + ToLevel int64 `json:"to_level"` // 目标等级 + UpgradeType int64 `json:"upgrade_type"` // 升级类型：1=自主付费，2=钻石升级 + UpgradeFee float64 `json:"upgrade_fee"` // 升级费用 + RebateAmount float64 `json:"rebate_amount"` // 返佣金额 + Status int64 `json:"status"` // 状态：1=待支付，2=已支付，3=已完成，4=已取消 + CreateTime string `json:"create_time"` // 创建时间 +} + +type UpgradeRebateConfig struct { + NormalToGoldRebate float64 `json:"normal_to_gold_rebate"` // 普通→黄金返佣：139 + ToDiamondRebate float64 `json:"to_diamond_rebate"` // 升级为钻石返佣：680 +} + +type UpgradeRebateItem struct { + Id string `json:"id"` // 记录ID + SourceAgentId string `json:"source_agent_id"` // 来源代理ID（升级的代理） + SourceAgentMobile string `json:"source_agent_mobile"` // 来源代理手机号 + OrderNo string `json:"order_no"` // 订单号 + FromLevel int64 `json:"from_level"` // 原等级 + ToLevel int64 `json:"to_level"` // 目标等级 + Amount float64 `json:"amount"` // 返佣金额 + CreateTime string `json:"create_time"` // 创建时间 +} + +type User struct { + Id string `json:"id"` + Mobile string `json:"mobile"` + NickName string `json:"nickName"` + UserType int64 `json:"userType"` +} + +type WithdrawalItem struct { + Id string `json:"id"` // 记录ID + WithdrawalNo string `json:"withdrawal_no"` // 提现单号 + Amount float64 `json:"amount"` // 提现金额 + TaxAmount float64 `json:"tax_amount"` // 税费金额 + ActualAmount float64 `json:"actual_amount"` // 实际到账金额 + Status int64 `json:"status"` // 状态：1=待审核，2=审核通过，3=审核拒绝，4=提现中，5=提现成功，6=提现失败 + PayeeAccount string `json:"payee_account"` // 收款账户 + PayeeName string `json:"payee_name"` // 收款人姓名 + Remark string `json:"remark"` // 备注 + CreateTime string `json:"create_time"` // 创建时间 +} diff --git a/app/main/api/internal/types/user.go b/app/main/api/internal/types/user.go new file mode 100644 index 0000000..f0d32b7 --- /dev/null +++ b/app/main/api/internal/types/user.go @@ -0,0 +1,73 @@ +// Code generated by goctl. DO NOT EDIT. +package types + +type AuthReq struct { + Platform string `json:"platform"` // browser|wxh5|wxmini + Code string `json:"code,optional"` +} + +type AuthResp struct { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` + UserType int64 `json:"userType"` + HasMobile bool `json:"hasMobile"` + IsAgent bool `json:"isAgent"` +} + +type BindMobileReq struct { + Mobile string `json:"mobile" validate:"required,mobile"` + Code string `json:"code" validate:"required"` +} + +type BindMobileResp struct { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` +} + +type GetSignatureReq struct { + Url string `json:"url"` +} + +type GetSignatureResp struct { + AppId string `json:"appId"` + Timestamp int64 `json:"timestamp"` + NonceStr string `json:"nonceStr"` + Signature string `json:"signature"` +} + +type MobileCodeLoginReq struct { + Mobile string `json:"mobile"` + Code string `json:"code" validate:"required"` +} + +type MobileCodeLoginResp struct { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` +} + +type UserInfoResp struct { + UserInfo User `json:"userInfo"` +} + +type WXH5AuthReq struct { + Code string `json:"code"` +} + +type WXH5AuthResp struct { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` +} + +type WXMiniAuthReq struct { + Code string `json:"code"` +} + +type WXMiniAuthResp struct { + AccessToken string `json:"accessToken"` + AccessExpire int64 `json:"accessExpire"` + RefreshAfter int64 `json:"refreshAfter"` +} diff --git a/app/main/api/main.go b/app/main/api/main.go new file mode 100644 index 0000000..1618a37 --- /dev/null +++ b/app/main/api/main.go @@ -0,0 +1,76 @@ +package main + +import ( + "context" + "flag" + "fmt" + "os" + "qnc-server/app/main/api/internal/config" + "qnc-server/app/main/api/internal/handler" + "qnc-server/app/main/api/internal/middleware" + "qnc-server/app/main/api/internal/queue" + "qnc-server/app/main/api/internal/service" + "qnc-server/app/main/api/internal/svc" + + "github.com/zeromicro/go-zero/core/logx" + + "github.com/zeromicro/go-zero/core/conf" + "github.com/zeromicro/go-zero/rest" +) + +func main() { + // 读取环境变量 ENV，默认为 "prod" + env := os.Getenv("ENV") + if env == "" { + env = "production" + } + + // 根据 ENV 加载不同的配置文件 + var defaultConfigFile string + if env == "development" { + defaultConfigFile = "app/main/api/etc/main.dev.yaml" + } else { + defaultConfigFile = "etc/main.yaml" + } + configFile := flag.String("f", defaultConfigFile, "the config file") + flag.Parse() + + var c config.Config + conf.MustLoad(*configFile, &c) + + svcContext := svc.NewServiceContext(c) + defer svcContext.Close() + + // 启动 asynq 消费者 + go func() { + ctx := context.Background() + // 初始化 cron job 或异步任务队列 + asynq := queue.NewCronJob(ctx, svcContext) + mux := asynq.Register() + + // 启动 asynq 消费者 + if err := svcContext.AsynqServer.Run(mux); err != nil { + logx.WithContext(ctx).Errorf("异步任务启动失败: %v", err) + os.Exit(1) + } + fmt.Println("异步任务启动！！！") + }() + + server := rest.MustNewServer(c.RestConf) + server.Use(middleware.GlobalSourceInterceptor) + defer server.Stop() + + handler.RegisterHandlers(server, svcContext) + + // 自动注册API到数据库 + apiRegistry := service.NewApiRegistryService(svcContext.AdminApiModel) + routes := server.Routes() + if err := apiRegistry.RegisterAllApis(context.Background(), routes); err != nil { + logx.Errorf("API注册失败: %v", err) + } else { + logx.Infof("API注册成功，共注册 %d 个路由", len(routes)) + } + + fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port) + server.Start() +} diff --git a/app/main/api/static/SIMHEI.TTF b/app/main/api/static/SIMHEI.TTF new file mode 100644 index 0000000..5bd4687 Binary files /dev/null and b/app/main/api/static/SIMHEI.TTF differ diff --git a/app/main/api/static/images/tg_qrcode_1.png b/app/main/api/static/images/tg_qrcode_1.png new file mode 100644 index 0000000..1157bb4 Binary files /dev/null and b/app/main/api/static/images/tg_qrcode_1.png differ diff --git a/app/main/api/static/images/yq_qrcode_1.png b/app/main/api/static/images/yq_qrcode_1.png new file mode 100644 index 0000000..d8d4fcc Binary files /dev/null and b/app/main/api/static/images/yq_qrcode_1.png differ diff --git a/app/main/model/adminApiModel.go b/app/main/model/adminApiModel.go new file mode 100644 index 0000000..6ed3633 --- /dev/null +++ b/app/main/model/adminApiModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AdminApiModel = (*customAdminApiModel)(nil) + +type ( + // AdminApiModel is an interface to be customized, add more methods here, + // and implement the added methods in customAdminApiModel. + AdminApiModel interface { + adminApiModel + } + + customAdminApiModel struct { + *defaultAdminApiModel + } +) + +// NewAdminApiModel returns a model for the database table. +func NewAdminApiModel(conn sqlx.SqlConn, c cache.CacheConf) AdminApiModel { + return &customAdminApiModel{ + defaultAdminApiModel: newAdminApiModel(conn, c), + } +} diff --git a/app/main/model/adminApiModel_gen.go b/app/main/model/adminApiModel_gen.go new file mode 100644 index 0000000..da1dfcf --- /dev/null +++ b/app/main/model/adminApiModel_gen.go @@ -0,0 +1,431 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + adminApiFieldNames = builder.RawFieldNames(&AdminApi{}) + adminApiRows = strings.Join(adminApiFieldNames, ",") + adminApiRowsExpectAutoSet = strings.Join(stringx.Remove(adminApiFieldNames, "`create_time`", "`update_time`"), ",") + adminApiRowsWithPlaceHolder = strings.Join(stringx.Remove(adminApiFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAdminApiIdPrefix = "cache:qnc:adminApi:id:" + cacheQncAdminApiApiCodePrefix = "cache:qnc:adminApi:apiCode:" +) + +type ( + adminApiModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AdminApi) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AdminApi, error) + FindOneByApiCode(ctx context.Context, apiCode string) (*AdminApi, error) + Update(ctx context.Context, session sqlx.Session, data *AdminApi) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AdminApi) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminApi) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AdminApi, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminApi, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminApi, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminApi, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminApi, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAdminApiModel struct { + sqlc.CachedConn + table string + } + + AdminApi struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + ApiName string `db:"api_name"` // 接口名称 + ApiCode string `db:"api_code"` // 接口编码 + Method string `db:"method"` // 请求方法：GET、POST等 + Url string `db:"url"` // 接口URL + Status int64 `db:"status"` // 状态：0-禁用，1-启用 + Description string `db:"description"` // 接口描述 + } +) + +func newAdminApiModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAdminApiModel { + return &defaultAdminApiModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`admin_api`", + } +} + +func (m *defaultAdminApiModel) Insert(ctx context.Context, session sqlx.Session, data *AdminApi) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAdminApiApiCodeKey := fmt.Sprintf("%s%v", cacheQncAdminApiApiCodePrefix, data.ApiCode) + qncAdminApiIdKey := fmt.Sprintf("%s%v", cacheQncAdminApiIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, adminApiRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ApiName, data.ApiCode, data.Method, data.Url, data.Status, data.Description) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ApiName, data.ApiCode, data.Method, data.Url, data.Status, data.Description) + }, qncAdminApiApiCodeKey, qncAdminApiIdKey) +} +func (m *defaultAdminApiModel) insertUUID(data *AdminApi) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAdminApiModel) FindOne(ctx context.Context, id string) (*AdminApi, error) { + qncAdminApiIdKey := fmt.Sprintf("%s%v", cacheQncAdminApiIdPrefix, id) + var resp AdminApi + err := m.QueryRowCtx(ctx, &resp, qncAdminApiIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminApiRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminApiModel) FindOneByApiCode(ctx context.Context, apiCode string) (*AdminApi, error) { + qncAdminApiApiCodeKey := fmt.Sprintf("%s%v", cacheQncAdminApiApiCodePrefix, apiCode) + var resp AdminApi + err := m.QueryRowIndexCtx(ctx, &resp, qncAdminApiApiCodeKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `api_code` = ? and del_state = ? limit 1", adminApiRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, apiCode, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminApiModel) Update(ctx context.Context, session sqlx.Session, newData *AdminApi) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAdminApiApiCodeKey := fmt.Sprintf("%s%v", cacheQncAdminApiApiCodePrefix, data.ApiCode) + qncAdminApiIdKey := fmt.Sprintf("%s%v", cacheQncAdminApiIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, adminApiRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiName, newData.ApiCode, newData.Method, newData.Url, newData.Status, newData.Description, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiName, newData.ApiCode, newData.Method, newData.Url, newData.Status, newData.Description, newData.Id) + }, qncAdminApiApiCodeKey, qncAdminApiIdKey) +} + +func (m *defaultAdminApiModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AdminApi) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAdminApiApiCodeKey := fmt.Sprintf("%s%v", cacheQncAdminApiApiCodePrefix, data.ApiCode) + qncAdminApiIdKey := fmt.Sprintf("%s%v", cacheQncAdminApiIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, adminApiRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiName, newData.ApiCode, newData.Method, newData.Url, newData.Status, newData.Description, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiName, newData.ApiCode, newData.Method, newData.Url, newData.Status, newData.Description, newData.Id, oldVersion) + }, qncAdminApiApiCodeKey, qncAdminApiIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAdminApiModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminApi) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AdminApiModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAdminApiModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminApiModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminApiModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AdminApi, error) { + + builder = builder.Columns(adminApiRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminApi + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminApiModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminApi, error) { + + builder = builder.Columns(adminApiRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminApi + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminApiModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminApi, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(adminApiRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AdminApi + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAdminApiModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminApi, error) { + + builder = builder.Columns(adminApiRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminApi + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminApiModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminApi, error) { + + builder = builder.Columns(adminApiRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminApi + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminApiModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAdminApiModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAdminApiModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAdminApiApiCodeKey := fmt.Sprintf("%s%v", cacheQncAdminApiApiCodePrefix, data.ApiCode) + qncAdminApiIdKey := fmt.Sprintf("%s%v", cacheQncAdminApiIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAdminApiApiCodeKey, qncAdminApiIdKey) + return err +} +func (m *defaultAdminApiModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAdminApiIdPrefix, primary) +} +func (m *defaultAdminApiModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminApiRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAdminApiModel) tableName() string { + return m.table +} diff --git a/app/main/model/adminDictDataModel.go b/app/main/model/adminDictDataModel.go new file mode 100644 index 0000000..cbec9c9 --- /dev/null +++ b/app/main/model/adminDictDataModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AdminDictDataModel = (*customAdminDictDataModel)(nil) + +type ( + // AdminDictDataModel is an interface to be customized, add more methods here, + // and implement the added methods in customAdminDictDataModel. + AdminDictDataModel interface { + adminDictDataModel + } + + customAdminDictDataModel struct { + *defaultAdminDictDataModel + } +) + +// NewAdminDictDataModel returns a model for the database table. +func NewAdminDictDataModel(conn sqlx.SqlConn, c cache.CacheConf) AdminDictDataModel { + return &customAdminDictDataModel{ + defaultAdminDictDataModel: newAdminDictDataModel(conn, c), + } +} diff --git a/app/main/model/adminDictDataModel_gen.go b/app/main/model/adminDictDataModel_gen.go new file mode 100644 index 0000000..6724127 --- /dev/null +++ b/app/main/model/adminDictDataModel_gen.go @@ -0,0 +1,457 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + adminDictDataFieldNames = builder.RawFieldNames(&AdminDictData{}) + adminDictDataRows = strings.Join(adminDictDataFieldNames, ",") + adminDictDataRowsExpectAutoSet = strings.Join(stringx.Remove(adminDictDataFieldNames, "`create_time`", "`update_time`"), ",") + adminDictDataRowsWithPlaceHolder = strings.Join(stringx.Remove(adminDictDataFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAdminDictDataIdPrefix = "cache:qnc:adminDictData:id:" + cacheQncAdminDictDataDictTypeDictLabelPrefix = "cache:qnc:adminDictData:dictType:dictLabel:" + cacheQncAdminDictDataDictTypeDictValuePrefix = "cache:qnc:adminDictData:dictType:dictValue:" +) + +type ( + adminDictDataModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AdminDictData) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AdminDictData, error) + FindOneByDictTypeDictLabel(ctx context.Context, dictType string, dictLabel string) (*AdminDictData, error) + FindOneByDictTypeDictValue(ctx context.Context, dictType string, dictValue int64) (*AdminDictData, error) + Update(ctx context.Context, session sqlx.Session, data *AdminDictData) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AdminDictData) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminDictData) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AdminDictData, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminDictData, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminDictData, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminDictData, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminDictData, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAdminDictDataModel struct { + sqlc.CachedConn + table string + } + + AdminDictData struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + DictType string `db:"dict_type"` // 字典类型编码 + DictLabel string `db:"dict_label"` // 字典标签 + DictValue int64 `db:"dict_value"` // 字典键值 + DictSort int64 `db:"dict_sort"` // 字典排序 + Status int64 `db:"status"` // 状态：0-禁用，1-启用 + Remark sql.NullString `db:"remark"` // 备注 + } +) + +func newAdminDictDataModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAdminDictDataModel { + return &defaultAdminDictDataModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`admin_dict_data`", + } +} + +func (m *defaultAdminDictDataModel) Insert(ctx context.Context, session sqlx.Session, data *AdminDictData) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAdminDictDataDictTypeDictLabelKey := fmt.Sprintf("%s%v:%v", cacheQncAdminDictDataDictTypeDictLabelPrefix, data.DictType, data.DictLabel) + qncAdminDictDataDictTypeDictValueKey := fmt.Sprintf("%s%v:%v", cacheQncAdminDictDataDictTypeDictValuePrefix, data.DictType, data.DictValue) + qncAdminDictDataIdKey := fmt.Sprintf("%s%v", cacheQncAdminDictDataIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, adminDictDataRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.DictType, data.DictLabel, data.DictValue, data.DictSort, data.Status, data.Remark) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.DictType, data.DictLabel, data.DictValue, data.DictSort, data.Status, data.Remark) + }, qncAdminDictDataDictTypeDictLabelKey, qncAdminDictDataDictTypeDictValueKey, qncAdminDictDataIdKey) +} +func (m *defaultAdminDictDataModel) insertUUID(data *AdminDictData) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAdminDictDataModel) FindOne(ctx context.Context, id string) (*AdminDictData, error) { + qncAdminDictDataIdKey := fmt.Sprintf("%s%v", cacheQncAdminDictDataIdPrefix, id) + var resp AdminDictData + err := m.QueryRowCtx(ctx, &resp, qncAdminDictDataIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminDictDataRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminDictDataModel) FindOneByDictTypeDictLabel(ctx context.Context, dictType string, dictLabel string) (*AdminDictData, error) { + qncAdminDictDataDictTypeDictLabelKey := fmt.Sprintf("%s%v:%v", cacheQncAdminDictDataDictTypeDictLabelPrefix, dictType, dictLabel) + var resp AdminDictData + err := m.QueryRowIndexCtx(ctx, &resp, qncAdminDictDataDictTypeDictLabelKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `dict_type` = ? and `dict_label` = ? and del_state = ? limit 1", adminDictDataRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, dictType, dictLabel, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminDictDataModel) FindOneByDictTypeDictValue(ctx context.Context, dictType string, dictValue int64) (*AdminDictData, error) { + qncAdminDictDataDictTypeDictValueKey := fmt.Sprintf("%s%v:%v", cacheQncAdminDictDataDictTypeDictValuePrefix, dictType, dictValue) + var resp AdminDictData + err := m.QueryRowIndexCtx(ctx, &resp, qncAdminDictDataDictTypeDictValueKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `dict_type` = ? and `dict_value` = ? and del_state = ? limit 1", adminDictDataRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, dictType, dictValue, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminDictDataModel) Update(ctx context.Context, session sqlx.Session, newData *AdminDictData) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAdminDictDataDictTypeDictLabelKey := fmt.Sprintf("%s%v:%v", cacheQncAdminDictDataDictTypeDictLabelPrefix, data.DictType, data.DictLabel) + qncAdminDictDataDictTypeDictValueKey := fmt.Sprintf("%s%v:%v", cacheQncAdminDictDataDictTypeDictValuePrefix, data.DictType, data.DictValue) + qncAdminDictDataIdKey := fmt.Sprintf("%s%v", cacheQncAdminDictDataIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, adminDictDataRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.DictType, newData.DictLabel, newData.DictValue, newData.DictSort, newData.Status, newData.Remark, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.DictType, newData.DictLabel, newData.DictValue, newData.DictSort, newData.Status, newData.Remark, newData.Id) + }, qncAdminDictDataDictTypeDictLabelKey, qncAdminDictDataDictTypeDictValueKey, qncAdminDictDataIdKey) +} + +func (m *defaultAdminDictDataModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AdminDictData) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAdminDictDataDictTypeDictLabelKey := fmt.Sprintf("%s%v:%v", cacheQncAdminDictDataDictTypeDictLabelPrefix, data.DictType, data.DictLabel) + qncAdminDictDataDictTypeDictValueKey := fmt.Sprintf("%s%v:%v", cacheQncAdminDictDataDictTypeDictValuePrefix, data.DictType, data.DictValue) + qncAdminDictDataIdKey := fmt.Sprintf("%s%v", cacheQncAdminDictDataIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, adminDictDataRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.DictType, newData.DictLabel, newData.DictValue, newData.DictSort, newData.Status, newData.Remark, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.DictType, newData.DictLabel, newData.DictValue, newData.DictSort, newData.Status, newData.Remark, newData.Id, oldVersion) + }, qncAdminDictDataDictTypeDictLabelKey, qncAdminDictDataDictTypeDictValueKey, qncAdminDictDataIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAdminDictDataModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminDictData) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AdminDictDataModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAdminDictDataModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminDictDataModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminDictDataModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AdminDictData, error) { + + builder = builder.Columns(adminDictDataRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminDictData + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminDictDataModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminDictData, error) { + + builder = builder.Columns(adminDictDataRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminDictData + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminDictDataModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminDictData, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(adminDictDataRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AdminDictData + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAdminDictDataModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminDictData, error) { + + builder = builder.Columns(adminDictDataRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminDictData + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminDictDataModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminDictData, error) { + + builder = builder.Columns(adminDictDataRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminDictData + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminDictDataModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAdminDictDataModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAdminDictDataModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAdminDictDataDictTypeDictLabelKey := fmt.Sprintf("%s%v:%v", cacheQncAdminDictDataDictTypeDictLabelPrefix, data.DictType, data.DictLabel) + qncAdminDictDataDictTypeDictValueKey := fmt.Sprintf("%s%v:%v", cacheQncAdminDictDataDictTypeDictValuePrefix, data.DictType, data.DictValue) + qncAdminDictDataIdKey := fmt.Sprintf("%s%v", cacheQncAdminDictDataIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAdminDictDataDictTypeDictLabelKey, qncAdminDictDataDictTypeDictValueKey, qncAdminDictDataIdKey) + return err +} +func (m *defaultAdminDictDataModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAdminDictDataIdPrefix, primary) +} +func (m *defaultAdminDictDataModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminDictDataRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAdminDictDataModel) tableName() string { + return m.table +} diff --git a/app/main/model/adminDictTypeModel.go b/app/main/model/adminDictTypeModel.go new file mode 100644 index 0000000..1153feb --- /dev/null +++ b/app/main/model/adminDictTypeModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AdminDictTypeModel = (*customAdminDictTypeModel)(nil) + +type ( + // AdminDictTypeModel is an interface to be customized, add more methods here, + // and implement the added methods in customAdminDictTypeModel. + AdminDictTypeModel interface { + adminDictTypeModel + } + + customAdminDictTypeModel struct { + *defaultAdminDictTypeModel + } +) + +// NewAdminDictTypeModel returns a model for the database table. +func NewAdminDictTypeModel(conn sqlx.SqlConn, c cache.CacheConf) AdminDictTypeModel { + return &customAdminDictTypeModel{ + defaultAdminDictTypeModel: newAdminDictTypeModel(conn, c), + } +} diff --git a/app/main/model/adminDictTypeModel_gen.go b/app/main/model/adminDictTypeModel_gen.go new file mode 100644 index 0000000..01a9b6d --- /dev/null +++ b/app/main/model/adminDictTypeModel_gen.go @@ -0,0 +1,429 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + adminDictTypeFieldNames = builder.RawFieldNames(&AdminDictType{}) + adminDictTypeRows = strings.Join(adminDictTypeFieldNames, ",") + adminDictTypeRowsExpectAutoSet = strings.Join(stringx.Remove(adminDictTypeFieldNames, "`create_time`", "`update_time`"), ",") + adminDictTypeRowsWithPlaceHolder = strings.Join(stringx.Remove(adminDictTypeFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAdminDictTypeIdPrefix = "cache:qnc:adminDictType:id:" + cacheQncAdminDictTypeDictTypePrefix = "cache:qnc:adminDictType:dictType:" +) + +type ( + adminDictTypeModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AdminDictType) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AdminDictType, error) + FindOneByDictType(ctx context.Context, dictType string) (*AdminDictType, error) + Update(ctx context.Context, session sqlx.Session, data *AdminDictType) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AdminDictType) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminDictType) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AdminDictType, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminDictType, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminDictType, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminDictType, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminDictType, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAdminDictTypeModel struct { + sqlc.CachedConn + table string + } + + AdminDictType struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + DictType string `db:"dict_type"` // 字典类型编码 + DictName string `db:"dict_name"` // 字典类型名称 + Status int64 `db:"status"` // 状态：0-禁用，1-启用 + Remark sql.NullString `db:"remark"` // 备注 + } +) + +func newAdminDictTypeModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAdminDictTypeModel { + return &defaultAdminDictTypeModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`admin_dict_type`", + } +} + +func (m *defaultAdminDictTypeModel) Insert(ctx context.Context, session sqlx.Session, data *AdminDictType) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAdminDictTypeDictTypeKey := fmt.Sprintf("%s%v", cacheQncAdminDictTypeDictTypePrefix, data.DictType) + qncAdminDictTypeIdKey := fmt.Sprintf("%s%v", cacheQncAdminDictTypeIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?)", m.table, adminDictTypeRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.DictType, data.DictName, data.Status, data.Remark) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.DictType, data.DictName, data.Status, data.Remark) + }, qncAdminDictTypeDictTypeKey, qncAdminDictTypeIdKey) +} +func (m *defaultAdminDictTypeModel) insertUUID(data *AdminDictType) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAdminDictTypeModel) FindOne(ctx context.Context, id string) (*AdminDictType, error) { + qncAdminDictTypeIdKey := fmt.Sprintf("%s%v", cacheQncAdminDictTypeIdPrefix, id) + var resp AdminDictType + err := m.QueryRowCtx(ctx, &resp, qncAdminDictTypeIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminDictTypeRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminDictTypeModel) FindOneByDictType(ctx context.Context, dictType string) (*AdminDictType, error) { + qncAdminDictTypeDictTypeKey := fmt.Sprintf("%s%v", cacheQncAdminDictTypeDictTypePrefix, dictType) + var resp AdminDictType + err := m.QueryRowIndexCtx(ctx, &resp, qncAdminDictTypeDictTypeKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `dict_type` = ? and del_state = ? limit 1", adminDictTypeRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, dictType, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminDictTypeModel) Update(ctx context.Context, session sqlx.Session, newData *AdminDictType) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAdminDictTypeDictTypeKey := fmt.Sprintf("%s%v", cacheQncAdminDictTypeDictTypePrefix, data.DictType) + qncAdminDictTypeIdKey := fmt.Sprintf("%s%v", cacheQncAdminDictTypeIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, adminDictTypeRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.DictType, newData.DictName, newData.Status, newData.Remark, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.DictType, newData.DictName, newData.Status, newData.Remark, newData.Id) + }, qncAdminDictTypeDictTypeKey, qncAdminDictTypeIdKey) +} + +func (m *defaultAdminDictTypeModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AdminDictType) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAdminDictTypeDictTypeKey := fmt.Sprintf("%s%v", cacheQncAdminDictTypeDictTypePrefix, data.DictType) + qncAdminDictTypeIdKey := fmt.Sprintf("%s%v", cacheQncAdminDictTypeIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, adminDictTypeRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.DictType, newData.DictName, newData.Status, newData.Remark, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.DictType, newData.DictName, newData.Status, newData.Remark, newData.Id, oldVersion) + }, qncAdminDictTypeDictTypeKey, qncAdminDictTypeIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAdminDictTypeModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminDictType) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AdminDictTypeModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAdminDictTypeModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminDictTypeModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminDictTypeModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AdminDictType, error) { + + builder = builder.Columns(adminDictTypeRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminDictType + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminDictTypeModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminDictType, error) { + + builder = builder.Columns(adminDictTypeRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminDictType + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminDictTypeModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminDictType, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(adminDictTypeRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AdminDictType + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAdminDictTypeModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminDictType, error) { + + builder = builder.Columns(adminDictTypeRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminDictType + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminDictTypeModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminDictType, error) { + + builder = builder.Columns(adminDictTypeRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminDictType + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminDictTypeModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAdminDictTypeModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAdminDictTypeModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAdminDictTypeDictTypeKey := fmt.Sprintf("%s%v", cacheQncAdminDictTypeDictTypePrefix, data.DictType) + qncAdminDictTypeIdKey := fmt.Sprintf("%s%v", cacheQncAdminDictTypeIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAdminDictTypeDictTypeKey, qncAdminDictTypeIdKey) + return err +} +func (m *defaultAdminDictTypeModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAdminDictTypeIdPrefix, primary) +} +func (m *defaultAdminDictTypeModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminDictTypeRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAdminDictTypeModel) tableName() string { + return m.table +} diff --git a/app/main/model/adminMenuModel.go b/app/main/model/adminMenuModel.go new file mode 100644 index 0000000..c2dc171 --- /dev/null +++ b/app/main/model/adminMenuModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AdminMenuModel = (*customAdminMenuModel)(nil) + +type ( + // AdminMenuModel is an interface to be customized, add more methods here, + // and implement the added methods in customAdminMenuModel. + AdminMenuModel interface { + adminMenuModel + } + + customAdminMenuModel struct { + *defaultAdminMenuModel + } +) + +// NewAdminMenuModel returns a model for the database table. +func NewAdminMenuModel(conn sqlx.SqlConn, c cache.CacheConf) AdminMenuModel { + return &customAdminMenuModel{ + defaultAdminMenuModel: newAdminMenuModel(conn, c), + } +} diff --git a/app/main/model/adminMenuModel_gen.go b/app/main/model/adminMenuModel_gen.go new file mode 100644 index 0000000..3cc21c1 --- /dev/null +++ b/app/main/model/adminMenuModel_gen.go @@ -0,0 +1,434 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + adminMenuFieldNames = builder.RawFieldNames(&AdminMenu{}) + adminMenuRows = strings.Join(adminMenuFieldNames, ",") + adminMenuRowsExpectAutoSet = strings.Join(stringx.Remove(adminMenuFieldNames, "`create_time`", "`update_time`"), ",") + adminMenuRowsWithPlaceHolder = strings.Join(stringx.Remove(adminMenuFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAdminMenuIdPrefix = "cache:qnc:adminMenu:id:" + cacheQncAdminMenuNamePathPrefix = "cache:qnc:adminMenu:name:path:" +) + +type ( + adminMenuModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AdminMenu) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AdminMenu, error) + FindOneByNamePath(ctx context.Context, name string, path string) (*AdminMenu, error) + Update(ctx context.Context, session sqlx.Session, data *AdminMenu) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AdminMenu) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminMenu) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AdminMenu, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminMenu, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminMenu, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminMenu, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminMenu, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAdminMenuModel struct { + sqlc.CachedConn + table string + } + + AdminMenu struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + Pid sql.NullString `db:"pid"` // 父菜单ID + Name string `db:"name"` // 路由名称 + Path string `db:"path"` // 路由路径 + Component string `db:"component"` // 组件路径 + Redirect sql.NullString `db:"redirect"` // 重定向路径 + Meta string `db:"meta"` // 路由元数据配置 + Status int64 `db:"status"` // 状态: 0-禁用, 1-启用 + Type int64 `db:"type"` + Sort int64 `db:"sort"` // 排序号 + } +) + +func newAdminMenuModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAdminMenuModel { + return &defaultAdminMenuModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`admin_menu`", + } +} + +func (m *defaultAdminMenuModel) Insert(ctx context.Context, session sqlx.Session, data *AdminMenu) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAdminMenuIdKey := fmt.Sprintf("%s%v", cacheQncAdminMenuIdPrefix, data.Id) + qncAdminMenuNamePathKey := fmt.Sprintf("%s%v:%v", cacheQncAdminMenuNamePathPrefix, data.Name, data.Path) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, adminMenuRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.Pid, data.Name, data.Path, data.Component, data.Redirect, data.Meta, data.Status, data.Type, data.Sort) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.Pid, data.Name, data.Path, data.Component, data.Redirect, data.Meta, data.Status, data.Type, data.Sort) + }, qncAdminMenuIdKey, qncAdminMenuNamePathKey) +} +func (m *defaultAdminMenuModel) insertUUID(data *AdminMenu) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAdminMenuModel) FindOne(ctx context.Context, id string) (*AdminMenu, error) { + qncAdminMenuIdKey := fmt.Sprintf("%s%v", cacheQncAdminMenuIdPrefix, id) + var resp AdminMenu + err := m.QueryRowCtx(ctx, &resp, qncAdminMenuIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminMenuRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminMenuModel) FindOneByNamePath(ctx context.Context, name string, path string) (*AdminMenu, error) { + qncAdminMenuNamePathKey := fmt.Sprintf("%s%v:%v", cacheQncAdminMenuNamePathPrefix, name, path) + var resp AdminMenu + err := m.QueryRowIndexCtx(ctx, &resp, qncAdminMenuNamePathKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `name` = ? and `path` = ? and del_state = ? limit 1", adminMenuRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, name, path, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminMenuModel) Update(ctx context.Context, session sqlx.Session, newData *AdminMenu) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAdminMenuIdKey := fmt.Sprintf("%s%v", cacheQncAdminMenuIdPrefix, data.Id) + qncAdminMenuNamePathKey := fmt.Sprintf("%s%v:%v", cacheQncAdminMenuNamePathPrefix, data.Name, data.Path) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, adminMenuRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Pid, newData.Name, newData.Path, newData.Component, newData.Redirect, newData.Meta, newData.Status, newData.Type, newData.Sort, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Pid, newData.Name, newData.Path, newData.Component, newData.Redirect, newData.Meta, newData.Status, newData.Type, newData.Sort, newData.Id) + }, qncAdminMenuIdKey, qncAdminMenuNamePathKey) +} + +func (m *defaultAdminMenuModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AdminMenu) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAdminMenuIdKey := fmt.Sprintf("%s%v", cacheQncAdminMenuIdPrefix, data.Id) + qncAdminMenuNamePathKey := fmt.Sprintf("%s%v:%v", cacheQncAdminMenuNamePathPrefix, data.Name, data.Path) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, adminMenuRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Pid, newData.Name, newData.Path, newData.Component, newData.Redirect, newData.Meta, newData.Status, newData.Type, newData.Sort, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Pid, newData.Name, newData.Path, newData.Component, newData.Redirect, newData.Meta, newData.Status, newData.Type, newData.Sort, newData.Id, oldVersion) + }, qncAdminMenuIdKey, qncAdminMenuNamePathKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAdminMenuModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminMenu) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AdminMenuModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAdminMenuModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminMenuModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminMenuModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AdminMenu, error) { + + builder = builder.Columns(adminMenuRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminMenu + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminMenuModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminMenu, error) { + + builder = builder.Columns(adminMenuRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminMenu + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminMenuModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminMenu, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(adminMenuRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AdminMenu + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAdminMenuModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminMenu, error) { + + builder = builder.Columns(adminMenuRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminMenu + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminMenuModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminMenu, error) { + + builder = builder.Columns(adminMenuRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminMenu + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminMenuModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAdminMenuModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAdminMenuModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAdminMenuIdKey := fmt.Sprintf("%s%v", cacheQncAdminMenuIdPrefix, id) + qncAdminMenuNamePathKey := fmt.Sprintf("%s%v:%v", cacheQncAdminMenuNamePathPrefix, data.Name, data.Path) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAdminMenuIdKey, qncAdminMenuNamePathKey) + return err +} +func (m *defaultAdminMenuModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAdminMenuIdPrefix, primary) +} +func (m *defaultAdminMenuModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminMenuRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAdminMenuModel) tableName() string { + return m.table +} diff --git a/app/main/model/adminRoleApiModel.go b/app/main/model/adminRoleApiModel.go new file mode 100644 index 0000000..bcb9fc7 --- /dev/null +++ b/app/main/model/adminRoleApiModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AdminRoleApiModel = (*customAdminRoleApiModel)(nil) + +type ( + // AdminRoleApiModel is an interface to be customized, add more methods here, + // and implement the added methods in customAdminRoleApiModel. + AdminRoleApiModel interface { + adminRoleApiModel + } + + customAdminRoleApiModel struct { + *defaultAdminRoleApiModel + } +) + +// NewAdminRoleApiModel returns a model for the database table. +func NewAdminRoleApiModel(conn sqlx.SqlConn, c cache.CacheConf) AdminRoleApiModel { + return &customAdminRoleApiModel{ + defaultAdminRoleApiModel: newAdminRoleApiModel(conn, c), + } +} diff --git a/app/main/model/adminRoleApiModel_gen.go b/app/main/model/adminRoleApiModel_gen.go new file mode 100644 index 0000000..5302407 --- /dev/null +++ b/app/main/model/adminRoleApiModel_gen.go @@ -0,0 +1,427 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + adminRoleApiFieldNames = builder.RawFieldNames(&AdminRoleApi{}) + adminRoleApiRows = strings.Join(adminRoleApiFieldNames, ",") + adminRoleApiRowsExpectAutoSet = strings.Join(stringx.Remove(adminRoleApiFieldNames, "`create_time`", "`update_time`"), ",") + adminRoleApiRowsWithPlaceHolder = strings.Join(stringx.Remove(adminRoleApiFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAdminRoleApiIdPrefix = "cache:qnc:adminRoleApi:id:" + cacheQncAdminRoleApiRoleIdApiIdPrefix = "cache:qnc:adminRoleApi:roleId:apiId:" +) + +type ( + adminRoleApiModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AdminRoleApi) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AdminRoleApi, error) + FindOneByRoleIdApiId(ctx context.Context, roleId string, apiId string) (*AdminRoleApi, error) + Update(ctx context.Context, session sqlx.Session, data *AdminRoleApi) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AdminRoleApi) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminRoleApi) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AdminRoleApi, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRoleApi, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRoleApi, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminRoleApi, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminRoleApi, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAdminRoleApiModel struct { + sqlc.CachedConn + table string + } + + AdminRoleApi struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + RoleId string `db:"role_id"` + ApiId string `db:"api_id"` + } +) + +func newAdminRoleApiModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAdminRoleApiModel { + return &defaultAdminRoleApiModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`admin_role_api`", + } +} + +func (m *defaultAdminRoleApiModel) Insert(ctx context.Context, session sqlx.Session, data *AdminRoleApi) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAdminRoleApiIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleApiIdPrefix, data.Id) + qncAdminRoleApiRoleIdApiIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminRoleApiRoleIdApiIdPrefix, data.RoleId, data.ApiId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?)", m.table, adminRoleApiRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.RoleId, data.ApiId) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.RoleId, data.ApiId) + }, qncAdminRoleApiIdKey, qncAdminRoleApiRoleIdApiIdKey) +} +func (m *defaultAdminRoleApiModel) insertUUID(data *AdminRoleApi) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAdminRoleApiModel) FindOne(ctx context.Context, id string) (*AdminRoleApi, error) { + qncAdminRoleApiIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleApiIdPrefix, id) + var resp AdminRoleApi + err := m.QueryRowCtx(ctx, &resp, qncAdminRoleApiIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminRoleApiRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminRoleApiModel) FindOneByRoleIdApiId(ctx context.Context, roleId string, apiId string) (*AdminRoleApi, error) { + qncAdminRoleApiRoleIdApiIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminRoleApiRoleIdApiIdPrefix, roleId, apiId) + var resp AdminRoleApi + err := m.QueryRowIndexCtx(ctx, &resp, qncAdminRoleApiRoleIdApiIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `role_id` = ? and `api_id` = ? and del_state = ? limit 1", adminRoleApiRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, roleId, apiId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminRoleApiModel) Update(ctx context.Context, session sqlx.Session, newData *AdminRoleApi) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAdminRoleApiIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleApiIdPrefix, data.Id) + qncAdminRoleApiRoleIdApiIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminRoleApiRoleIdApiIdPrefix, data.RoleId, data.ApiId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, adminRoleApiRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleId, newData.ApiId, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleId, newData.ApiId, newData.Id) + }, qncAdminRoleApiIdKey, qncAdminRoleApiRoleIdApiIdKey) +} + +func (m *defaultAdminRoleApiModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AdminRoleApi) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAdminRoleApiIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleApiIdPrefix, data.Id) + qncAdminRoleApiRoleIdApiIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminRoleApiRoleIdApiIdPrefix, data.RoleId, data.ApiId) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, adminRoleApiRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleId, newData.ApiId, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleId, newData.ApiId, newData.Id, oldVersion) + }, qncAdminRoleApiIdKey, qncAdminRoleApiRoleIdApiIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAdminRoleApiModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminRoleApi) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AdminRoleApiModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAdminRoleApiModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminRoleApiModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminRoleApiModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AdminRoleApi, error) { + + builder = builder.Columns(adminRoleApiRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRoleApi + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleApiModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRoleApi, error) { + + builder = builder.Columns(adminRoleApiRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRoleApi + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleApiModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRoleApi, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(adminRoleApiRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AdminRoleApi + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAdminRoleApiModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminRoleApi, error) { + + builder = builder.Columns(adminRoleApiRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRoleApi + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleApiModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminRoleApi, error) { + + builder = builder.Columns(adminRoleApiRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRoleApi + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleApiModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAdminRoleApiModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAdminRoleApiModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAdminRoleApiIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleApiIdPrefix, id) + qncAdminRoleApiRoleIdApiIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminRoleApiRoleIdApiIdPrefix, data.RoleId, data.ApiId) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAdminRoleApiIdKey, qncAdminRoleApiRoleIdApiIdKey) + return err +} +func (m *defaultAdminRoleApiModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAdminRoleApiIdPrefix, primary) +} +func (m *defaultAdminRoleApiModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminRoleApiRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAdminRoleApiModel) tableName() string { + return m.table +} diff --git a/app/main/model/adminRoleMenuModel.go b/app/main/model/adminRoleMenuModel.go new file mode 100644 index 0000000..8dc9812 --- /dev/null +++ b/app/main/model/adminRoleMenuModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AdminRoleMenuModel = (*customAdminRoleMenuModel)(nil) + +type ( + // AdminRoleMenuModel is an interface to be customized, add more methods here, + // and implement the added methods in customAdminRoleMenuModel. + AdminRoleMenuModel interface { + adminRoleMenuModel + } + + customAdminRoleMenuModel struct { + *defaultAdminRoleMenuModel + } +) + +// NewAdminRoleMenuModel returns a model for the database table. +func NewAdminRoleMenuModel(conn sqlx.SqlConn, c cache.CacheConf) AdminRoleMenuModel { + return &customAdminRoleMenuModel{ + defaultAdminRoleMenuModel: newAdminRoleMenuModel(conn, c), + } +} diff --git a/app/main/model/adminRoleMenuModel_gen.go b/app/main/model/adminRoleMenuModel_gen.go new file mode 100644 index 0000000..2e03e6e --- /dev/null +++ b/app/main/model/adminRoleMenuModel_gen.go @@ -0,0 +1,427 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + adminRoleMenuFieldNames = builder.RawFieldNames(&AdminRoleMenu{}) + adminRoleMenuRows = strings.Join(adminRoleMenuFieldNames, ",") + adminRoleMenuRowsExpectAutoSet = strings.Join(stringx.Remove(adminRoleMenuFieldNames, "`create_time`", "`update_time`"), ",") + adminRoleMenuRowsWithPlaceHolder = strings.Join(stringx.Remove(adminRoleMenuFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAdminRoleMenuIdPrefix = "cache:qnc:adminRoleMenu:id:" + cacheQncAdminRoleMenuRoleIdMenuIdPrefix = "cache:qnc:adminRoleMenu:roleId:menuId:" +) + +type ( + adminRoleMenuModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AdminRoleMenu) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AdminRoleMenu, error) + FindOneByRoleIdMenuId(ctx context.Context, roleId string, menuId string) (*AdminRoleMenu, error) + Update(ctx context.Context, session sqlx.Session, data *AdminRoleMenu) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AdminRoleMenu) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminRoleMenu) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AdminRoleMenu, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRoleMenu, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRoleMenu, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminRoleMenu, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminRoleMenu, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAdminRoleMenuModel struct { + sqlc.CachedConn + table string + } + + AdminRoleMenu struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + RoleId string `db:"role_id"` + MenuId string `db:"menu_id"` + } +) + +func newAdminRoleMenuModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAdminRoleMenuModel { + return &defaultAdminRoleMenuModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`admin_role_menu`", + } +} + +func (m *defaultAdminRoleMenuModel) Insert(ctx context.Context, session sqlx.Session, data *AdminRoleMenu) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAdminRoleMenuIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleMenuIdPrefix, data.Id) + qncAdminRoleMenuRoleIdMenuIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminRoleMenuRoleIdMenuIdPrefix, data.RoleId, data.MenuId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?)", m.table, adminRoleMenuRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.RoleId, data.MenuId) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.RoleId, data.MenuId) + }, qncAdminRoleMenuIdKey, qncAdminRoleMenuRoleIdMenuIdKey) +} +func (m *defaultAdminRoleMenuModel) insertUUID(data *AdminRoleMenu) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAdminRoleMenuModel) FindOne(ctx context.Context, id string) (*AdminRoleMenu, error) { + qncAdminRoleMenuIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleMenuIdPrefix, id) + var resp AdminRoleMenu + err := m.QueryRowCtx(ctx, &resp, qncAdminRoleMenuIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminRoleMenuRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminRoleMenuModel) FindOneByRoleIdMenuId(ctx context.Context, roleId string, menuId string) (*AdminRoleMenu, error) { + qncAdminRoleMenuRoleIdMenuIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminRoleMenuRoleIdMenuIdPrefix, roleId, menuId) + var resp AdminRoleMenu + err := m.QueryRowIndexCtx(ctx, &resp, qncAdminRoleMenuRoleIdMenuIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `role_id` = ? and `menu_id` = ? and del_state = ? limit 1", adminRoleMenuRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, roleId, menuId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminRoleMenuModel) Update(ctx context.Context, session sqlx.Session, newData *AdminRoleMenu) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAdminRoleMenuIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleMenuIdPrefix, data.Id) + qncAdminRoleMenuRoleIdMenuIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminRoleMenuRoleIdMenuIdPrefix, data.RoleId, data.MenuId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, adminRoleMenuRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleId, newData.MenuId, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleId, newData.MenuId, newData.Id) + }, qncAdminRoleMenuIdKey, qncAdminRoleMenuRoleIdMenuIdKey) +} + +func (m *defaultAdminRoleMenuModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AdminRoleMenu) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAdminRoleMenuIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleMenuIdPrefix, data.Id) + qncAdminRoleMenuRoleIdMenuIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminRoleMenuRoleIdMenuIdPrefix, data.RoleId, data.MenuId) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, adminRoleMenuRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleId, newData.MenuId, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleId, newData.MenuId, newData.Id, oldVersion) + }, qncAdminRoleMenuIdKey, qncAdminRoleMenuRoleIdMenuIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAdminRoleMenuModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminRoleMenu) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AdminRoleMenuModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAdminRoleMenuModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminRoleMenuModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminRoleMenuModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AdminRoleMenu, error) { + + builder = builder.Columns(adminRoleMenuRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRoleMenu + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleMenuModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRoleMenu, error) { + + builder = builder.Columns(adminRoleMenuRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRoleMenu + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleMenuModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRoleMenu, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(adminRoleMenuRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AdminRoleMenu + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAdminRoleMenuModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminRoleMenu, error) { + + builder = builder.Columns(adminRoleMenuRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRoleMenu + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleMenuModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminRoleMenu, error) { + + builder = builder.Columns(adminRoleMenuRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRoleMenu + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleMenuModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAdminRoleMenuModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAdminRoleMenuModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAdminRoleMenuIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleMenuIdPrefix, id) + qncAdminRoleMenuRoleIdMenuIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminRoleMenuRoleIdMenuIdPrefix, data.RoleId, data.MenuId) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAdminRoleMenuIdKey, qncAdminRoleMenuRoleIdMenuIdKey) + return err +} +func (m *defaultAdminRoleMenuModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAdminRoleMenuIdPrefix, primary) +} +func (m *defaultAdminRoleMenuModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminRoleMenuRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAdminRoleMenuModel) tableName() string { + return m.table +} diff --git a/app/main/model/adminRoleModel.go b/app/main/model/adminRoleModel.go new file mode 100644 index 0000000..26dc91d --- /dev/null +++ b/app/main/model/adminRoleModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AdminRoleModel = (*customAdminRoleModel)(nil) + +type ( + // AdminRoleModel is an interface to be customized, add more methods here, + // and implement the added methods in customAdminRoleModel. + AdminRoleModel interface { + adminRoleModel + } + + customAdminRoleModel struct { + *defaultAdminRoleModel + } +) + +// NewAdminRoleModel returns a model for the database table. +func NewAdminRoleModel(conn sqlx.SqlConn, c cache.CacheConf) AdminRoleModel { + return &customAdminRoleModel{ + defaultAdminRoleModel: newAdminRoleModel(conn, c), + } +} diff --git a/app/main/model/adminRoleModel_gen.go b/app/main/model/adminRoleModel_gen.go new file mode 100644 index 0000000..7188072 --- /dev/null +++ b/app/main/model/adminRoleModel_gen.go @@ -0,0 +1,430 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + adminRoleFieldNames = builder.RawFieldNames(&AdminRole{}) + adminRoleRows = strings.Join(adminRoleFieldNames, ",") + adminRoleRowsExpectAutoSet = strings.Join(stringx.Remove(adminRoleFieldNames, "`create_time`", "`update_time`"), ",") + adminRoleRowsWithPlaceHolder = strings.Join(stringx.Remove(adminRoleFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAdminRoleIdPrefix = "cache:qnc:adminRole:id:" + cacheQncAdminRoleRoleCodePrefix = "cache:qnc:adminRole:roleCode:" +) + +type ( + adminRoleModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AdminRole) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AdminRole, error) + FindOneByRoleCode(ctx context.Context, roleCode string) (*AdminRole, error) + Update(ctx context.Context, session sqlx.Session, data *AdminRole) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AdminRole) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminRole) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AdminRole, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRole, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRole, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminRole, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminRole, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAdminRoleModel struct { + sqlc.CachedConn + table string + } + + AdminRole struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + RoleName string `db:"role_name"` // 角色名称 + RoleCode string `db:"role_code"` // 角色编码 + Description string `db:"description"` // 角色描述 + Status int64 `db:"status"` // 状态：0-禁用，1-启用 + Sort int64 `db:"sort"` // 排序 + } +) + +func newAdminRoleModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAdminRoleModel { + return &defaultAdminRoleModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`admin_role`", + } +} + +func (m *defaultAdminRoleModel) Insert(ctx context.Context, session sqlx.Session, data *AdminRole) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAdminRoleIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleIdPrefix, data.Id) + qncAdminRoleRoleCodeKey := fmt.Sprintf("%s%v", cacheQncAdminRoleRoleCodePrefix, data.RoleCode) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, adminRoleRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.RoleName, data.RoleCode, data.Description, data.Status, data.Sort) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.RoleName, data.RoleCode, data.Description, data.Status, data.Sort) + }, qncAdminRoleIdKey, qncAdminRoleRoleCodeKey) +} +func (m *defaultAdminRoleModel) insertUUID(data *AdminRole) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAdminRoleModel) FindOne(ctx context.Context, id string) (*AdminRole, error) { + qncAdminRoleIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleIdPrefix, id) + var resp AdminRole + err := m.QueryRowCtx(ctx, &resp, qncAdminRoleIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminRoleRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminRoleModel) FindOneByRoleCode(ctx context.Context, roleCode string) (*AdminRole, error) { + qncAdminRoleRoleCodeKey := fmt.Sprintf("%s%v", cacheQncAdminRoleRoleCodePrefix, roleCode) + var resp AdminRole + err := m.QueryRowIndexCtx(ctx, &resp, qncAdminRoleRoleCodeKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `role_code` = ? and del_state = ? limit 1", adminRoleRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, roleCode, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminRoleModel) Update(ctx context.Context, session sqlx.Session, newData *AdminRole) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAdminRoleIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleIdPrefix, data.Id) + qncAdminRoleRoleCodeKey := fmt.Sprintf("%s%v", cacheQncAdminRoleRoleCodePrefix, data.RoleCode) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, adminRoleRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleName, newData.RoleCode, newData.Description, newData.Status, newData.Sort, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleName, newData.RoleCode, newData.Description, newData.Status, newData.Sort, newData.Id) + }, qncAdminRoleIdKey, qncAdminRoleRoleCodeKey) +} + +func (m *defaultAdminRoleModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AdminRole) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAdminRoleIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleIdPrefix, data.Id) + qncAdminRoleRoleCodeKey := fmt.Sprintf("%s%v", cacheQncAdminRoleRoleCodePrefix, data.RoleCode) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, adminRoleRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleName, newData.RoleCode, newData.Description, newData.Status, newData.Sort, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.RoleName, newData.RoleCode, newData.Description, newData.Status, newData.Sort, newData.Id, oldVersion) + }, qncAdminRoleIdKey, qncAdminRoleRoleCodeKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAdminRoleModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminRole) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AdminRoleModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAdminRoleModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminRoleModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminRoleModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AdminRole, error) { + + builder = builder.Columns(adminRoleRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRole + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRole, error) { + + builder = builder.Columns(adminRoleRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRole + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminRole, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(adminRoleRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AdminRole + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAdminRoleModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminRole, error) { + + builder = builder.Columns(adminRoleRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRole + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminRole, error) { + + builder = builder.Columns(adminRoleRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminRole + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminRoleModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAdminRoleModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAdminRoleModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAdminRoleIdKey := fmt.Sprintf("%s%v", cacheQncAdminRoleIdPrefix, id) + qncAdminRoleRoleCodeKey := fmt.Sprintf("%s%v", cacheQncAdminRoleRoleCodePrefix, data.RoleCode) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAdminRoleIdKey, qncAdminRoleRoleCodeKey) + return err +} +func (m *defaultAdminRoleModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAdminRoleIdPrefix, primary) +} +func (m *defaultAdminRoleModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminRoleRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAdminRoleModel) tableName() string { + return m.table +} diff --git a/app/main/model/adminUserModel.go b/app/main/model/adminUserModel.go new file mode 100644 index 0000000..25d463e --- /dev/null +++ b/app/main/model/adminUserModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AdminUserModel = (*customAdminUserModel)(nil) + +type ( + // AdminUserModel is an interface to be customized, add more methods here, + // and implement the added methods in customAdminUserModel. + AdminUserModel interface { + adminUserModel + } + + customAdminUserModel struct { + *defaultAdminUserModel + } +) + +// NewAdminUserModel returns a model for the database table. +func NewAdminUserModel(conn sqlx.SqlConn, c cache.CacheConf) AdminUserModel { + return &customAdminUserModel{ + defaultAdminUserModel: newAdminUserModel(conn, c), + } +} diff --git a/app/main/model/adminUserModel_gen.go b/app/main/model/adminUserModel_gen.go new file mode 100644 index 0000000..85622c5 --- /dev/null +++ b/app/main/model/adminUserModel_gen.go @@ -0,0 +1,455 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + adminUserFieldNames = builder.RawFieldNames(&AdminUser{}) + adminUserRows = strings.Join(adminUserFieldNames, ",") + adminUserRowsExpectAutoSet = strings.Join(stringx.Remove(adminUserFieldNames, "`create_time`", "`update_time`"), ",") + adminUserRowsWithPlaceHolder = strings.Join(stringx.Remove(adminUserFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAdminUserIdPrefix = "cache:qnc:adminUser:id:" + cacheQncAdminUserRealNamePrefix = "cache:qnc:adminUser:realName:" + cacheQncAdminUserUsernamePrefix = "cache:qnc:adminUser:username:" +) + +type ( + adminUserModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AdminUser) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AdminUser, error) + FindOneByRealName(ctx context.Context, realName string) (*AdminUser, error) + FindOneByUsername(ctx context.Context, username string) (*AdminUser, error) + Update(ctx context.Context, session sqlx.Session, data *AdminUser) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AdminUser) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminUser) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AdminUser, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminUser, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminUser, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminUser, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminUser, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAdminUserModel struct { + sqlc.CachedConn + table string + } + + AdminUser struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + Username string `db:"username"` // 用户名 + Password string `db:"password"` // 密码 + RealName string `db:"real_name"` // 真实姓名 + Status int64 `db:"status"` // 状态：0-禁用，1-启用 + } +) + +func newAdminUserModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAdminUserModel { + return &defaultAdminUserModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`admin_user`", + } +} + +func (m *defaultAdminUserModel) Insert(ctx context.Context, session sqlx.Session, data *AdminUser) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAdminUserIdKey := fmt.Sprintf("%s%v", cacheQncAdminUserIdPrefix, data.Id) + qncAdminUserRealNameKey := fmt.Sprintf("%s%v", cacheQncAdminUserRealNamePrefix, data.RealName) + qncAdminUserUsernameKey := fmt.Sprintf("%s%v", cacheQncAdminUserUsernamePrefix, data.Username) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?)", m.table, adminUserRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.Username, data.Password, data.RealName, data.Status) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.Username, data.Password, data.RealName, data.Status) + }, qncAdminUserIdKey, qncAdminUserRealNameKey, qncAdminUserUsernameKey) +} +func (m *defaultAdminUserModel) insertUUID(data *AdminUser) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAdminUserModel) FindOne(ctx context.Context, id string) (*AdminUser, error) { + qncAdminUserIdKey := fmt.Sprintf("%s%v", cacheQncAdminUserIdPrefix, id) + var resp AdminUser + err := m.QueryRowCtx(ctx, &resp, qncAdminUserIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminUserRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminUserModel) FindOneByRealName(ctx context.Context, realName string) (*AdminUser, error) { + qncAdminUserRealNameKey := fmt.Sprintf("%s%v", cacheQncAdminUserRealNamePrefix, realName) + var resp AdminUser + err := m.QueryRowIndexCtx(ctx, &resp, qncAdminUserRealNameKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `real_name` = ? and del_state = ? limit 1", adminUserRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, realName, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminUserModel) FindOneByUsername(ctx context.Context, username string) (*AdminUser, error) { + qncAdminUserUsernameKey := fmt.Sprintf("%s%v", cacheQncAdminUserUsernamePrefix, username) + var resp AdminUser + err := m.QueryRowIndexCtx(ctx, &resp, qncAdminUserUsernameKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `username` = ? and del_state = ? limit 1", adminUserRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, username, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminUserModel) Update(ctx context.Context, session sqlx.Session, newData *AdminUser) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAdminUserIdKey := fmt.Sprintf("%s%v", cacheQncAdminUserIdPrefix, data.Id) + qncAdminUserRealNameKey := fmt.Sprintf("%s%v", cacheQncAdminUserRealNamePrefix, data.RealName) + qncAdminUserUsernameKey := fmt.Sprintf("%s%v", cacheQncAdminUserUsernamePrefix, data.Username) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, adminUserRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Username, newData.Password, newData.RealName, newData.Status, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Username, newData.Password, newData.RealName, newData.Status, newData.Id) + }, qncAdminUserIdKey, qncAdminUserRealNameKey, qncAdminUserUsernameKey) +} + +func (m *defaultAdminUserModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AdminUser) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAdminUserIdKey := fmt.Sprintf("%s%v", cacheQncAdminUserIdPrefix, data.Id) + qncAdminUserRealNameKey := fmt.Sprintf("%s%v", cacheQncAdminUserRealNamePrefix, data.RealName) + qncAdminUserUsernameKey := fmt.Sprintf("%s%v", cacheQncAdminUserUsernamePrefix, data.Username) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, adminUserRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Username, newData.Password, newData.RealName, newData.Status, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Username, newData.Password, newData.RealName, newData.Status, newData.Id, oldVersion) + }, qncAdminUserIdKey, qncAdminUserRealNameKey, qncAdminUserUsernameKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAdminUserModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminUser) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AdminUserModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAdminUserModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminUserModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminUserModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AdminUser, error) { + + builder = builder.Columns(adminUserRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminUser + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminUserModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminUser, error) { + + builder = builder.Columns(adminUserRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminUser + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminUserModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminUser, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(adminUserRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AdminUser + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAdminUserModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminUser, error) { + + builder = builder.Columns(adminUserRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminUser + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminUserModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminUser, error) { + + builder = builder.Columns(adminUserRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminUser + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminUserModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAdminUserModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAdminUserModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAdminUserIdKey := fmt.Sprintf("%s%v", cacheQncAdminUserIdPrefix, id) + qncAdminUserRealNameKey := fmt.Sprintf("%s%v", cacheQncAdminUserRealNamePrefix, data.RealName) + qncAdminUserUsernameKey := fmt.Sprintf("%s%v", cacheQncAdminUserUsernamePrefix, data.Username) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAdminUserIdKey, qncAdminUserRealNameKey, qncAdminUserUsernameKey) + return err +} +func (m *defaultAdminUserModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAdminUserIdPrefix, primary) +} +func (m *defaultAdminUserModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminUserRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAdminUserModel) tableName() string { + return m.table +} diff --git a/app/main/model/adminUserRoleModel.go b/app/main/model/adminUserRoleModel.go new file mode 100644 index 0000000..d9772ad --- /dev/null +++ b/app/main/model/adminUserRoleModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AdminUserRoleModel = (*customAdminUserRoleModel)(nil) + +type ( + // AdminUserRoleModel is an interface to be customized, add more methods here, + // and implement the added methods in customAdminUserRoleModel. + AdminUserRoleModel interface { + adminUserRoleModel + } + + customAdminUserRoleModel struct { + *defaultAdminUserRoleModel + } +) + +// NewAdminUserRoleModel returns a model for the database table. +func NewAdminUserRoleModel(conn sqlx.SqlConn, c cache.CacheConf) AdminUserRoleModel { + return &customAdminUserRoleModel{ + defaultAdminUserRoleModel: newAdminUserRoleModel(conn, c), + } +} diff --git a/app/main/model/adminUserRoleModel_gen.go b/app/main/model/adminUserRoleModel_gen.go new file mode 100644 index 0000000..1643735 --- /dev/null +++ b/app/main/model/adminUserRoleModel_gen.go @@ -0,0 +1,427 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + adminUserRoleFieldNames = builder.RawFieldNames(&AdminUserRole{}) + adminUserRoleRows = strings.Join(adminUserRoleFieldNames, ",") + adminUserRoleRowsExpectAutoSet = strings.Join(stringx.Remove(adminUserRoleFieldNames, "`create_time`", "`update_time`"), ",") + adminUserRoleRowsWithPlaceHolder = strings.Join(stringx.Remove(adminUserRoleFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAdminUserRoleIdPrefix = "cache:qnc:adminUserRole:id:" + cacheQncAdminUserRoleUserIdRoleIdPrefix = "cache:qnc:adminUserRole:userId:roleId:" +) + +type ( + adminUserRoleModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AdminUserRole) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AdminUserRole, error) + FindOneByUserIdRoleId(ctx context.Context, userId string, roleId string) (*AdminUserRole, error) + Update(ctx context.Context, session sqlx.Session, data *AdminUserRole) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AdminUserRole) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminUserRole) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AdminUserRole, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminUserRole, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminUserRole, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminUserRole, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminUserRole, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAdminUserRoleModel struct { + sqlc.CachedConn + table string + } + + AdminUserRole struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + UserId string `db:"user_id"` + RoleId string `db:"role_id"` + } +) + +func newAdminUserRoleModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAdminUserRoleModel { + return &defaultAdminUserRoleModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`admin_user_role`", + } +} + +func (m *defaultAdminUserRoleModel) Insert(ctx context.Context, session sqlx.Session, data *AdminUserRole) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAdminUserRoleIdKey := fmt.Sprintf("%s%v", cacheQncAdminUserRoleIdPrefix, data.Id) + qncAdminUserRoleUserIdRoleIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminUserRoleUserIdRoleIdPrefix, data.UserId, data.RoleId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?)", m.table, adminUserRoleRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.UserId, data.RoleId) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.UserId, data.RoleId) + }, qncAdminUserRoleIdKey, qncAdminUserRoleUserIdRoleIdKey) +} +func (m *defaultAdminUserRoleModel) insertUUID(data *AdminUserRole) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAdminUserRoleModel) FindOne(ctx context.Context, id string) (*AdminUserRole, error) { + qncAdminUserRoleIdKey := fmt.Sprintf("%s%v", cacheQncAdminUserRoleIdPrefix, id) + var resp AdminUserRole + err := m.QueryRowCtx(ctx, &resp, qncAdminUserRoleIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminUserRoleRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminUserRoleModel) FindOneByUserIdRoleId(ctx context.Context, userId string, roleId string) (*AdminUserRole, error) { + qncAdminUserRoleUserIdRoleIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminUserRoleUserIdRoleIdPrefix, userId, roleId) + var resp AdminUserRole + err := m.QueryRowIndexCtx(ctx, &resp, qncAdminUserRoleUserIdRoleIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `user_id` = ? and `role_id` = ? and del_state = ? limit 1", adminUserRoleRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, userId, roleId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAdminUserRoleModel) Update(ctx context.Context, session sqlx.Session, newData *AdminUserRole) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAdminUserRoleIdKey := fmt.Sprintf("%s%v", cacheQncAdminUserRoleIdPrefix, data.Id) + qncAdminUserRoleUserIdRoleIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminUserRoleUserIdRoleIdPrefix, data.UserId, data.RoleId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, adminUserRoleRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.UserId, newData.RoleId, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.UserId, newData.RoleId, newData.Id) + }, qncAdminUserRoleIdKey, qncAdminUserRoleUserIdRoleIdKey) +} + +func (m *defaultAdminUserRoleModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AdminUserRole) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAdminUserRoleIdKey := fmt.Sprintf("%s%v", cacheQncAdminUserRoleIdPrefix, data.Id) + qncAdminUserRoleUserIdRoleIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminUserRoleUserIdRoleIdPrefix, data.UserId, data.RoleId) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, adminUserRoleRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.UserId, newData.RoleId, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.UserId, newData.RoleId, newData.Id, oldVersion) + }, qncAdminUserRoleIdKey, qncAdminUserRoleUserIdRoleIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAdminUserRoleModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AdminUserRole) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AdminUserRoleModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAdminUserRoleModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminUserRoleModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAdminUserRoleModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AdminUserRole, error) { + + builder = builder.Columns(adminUserRoleRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminUserRole + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminUserRoleModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminUserRole, error) { + + builder = builder.Columns(adminUserRoleRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminUserRole + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminUserRoleModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AdminUserRole, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(adminUserRoleRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AdminUserRole + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAdminUserRoleModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AdminUserRole, error) { + + builder = builder.Columns(adminUserRoleRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminUserRole + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminUserRoleModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AdminUserRole, error) { + + builder = builder.Columns(adminUserRoleRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AdminUserRole + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAdminUserRoleModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAdminUserRoleModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAdminUserRoleModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAdminUserRoleIdKey := fmt.Sprintf("%s%v", cacheQncAdminUserRoleIdPrefix, id) + qncAdminUserRoleUserIdRoleIdKey := fmt.Sprintf("%s%v:%v", cacheQncAdminUserRoleUserIdRoleIdPrefix, data.UserId, data.RoleId) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAdminUserRoleIdKey, qncAdminUserRoleUserIdRoleIdKey) + return err +} +func (m *defaultAdminUserRoleModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAdminUserRoleIdPrefix, primary) +} +func (m *defaultAdminUserRoleModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", adminUserRoleRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAdminUserRoleModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentCommissionModel.go b/app/main/model/agentCommissionModel.go new file mode 100644 index 0000000..1b9e614 --- /dev/null +++ b/app/main/model/agentCommissionModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentCommissionModel = (*customAgentCommissionModel)(nil) + +type ( + // AgentCommissionModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentCommissionModel. + AgentCommissionModel interface { + agentCommissionModel + } + + customAgentCommissionModel struct { + *defaultAgentCommissionModel + } +) + +// NewAgentCommissionModel returns a model for the database table. +func NewAgentCommissionModel(conn sqlx.SqlConn, c cache.CacheConf) AgentCommissionModel { + return &customAgentCommissionModel{ + defaultAgentCommissionModel: newAgentCommissionModel(conn, c), + } +} diff --git a/app/main/model/agentCommissionModel_gen.go b/app/main/model/agentCommissionModel_gen.go new file mode 100644 index 0000000..eea66df --- /dev/null +++ b/app/main/model/agentCommissionModel_gen.go @@ -0,0 +1,391 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentCommissionFieldNames = builder.RawFieldNames(&AgentCommission{}) + agentCommissionRows = strings.Join(agentCommissionFieldNames, ",") + agentCommissionRowsExpectAutoSet = strings.Join(stringx.Remove(agentCommissionFieldNames, "`create_time`", "`update_time`"), ",") + agentCommissionRowsWithPlaceHolder = strings.Join(stringx.Remove(agentCommissionFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentCommissionIdPrefix = "cache:qnc:agentCommission:id:" +) + +type ( + agentCommissionModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentCommission) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentCommission, error) + Update(ctx context.Context, session sqlx.Session, data *AgentCommission) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentCommission) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentCommission) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentCommission, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentCommission, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentCommission, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentCommission, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentCommission, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentCommissionModel struct { + sqlc.CachedConn + table string + } + + AgentCommission struct { + Id string `db:"id"` + AgentId string `db:"agent_id"` + OrderId string `db:"order_id"` + ProductId string `db:"product_id"` + Amount float64 `db:"amount"` // 佣金金额 + Status int64 `db:"status"` // 状态：1=已发放，2=已冻结，3=已取消 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentCommissionModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentCommissionModel { + return &defaultAgentCommissionModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_commission`", + } +} + +func (m *defaultAgentCommissionModel) Insert(ctx context.Context, session sqlx.Session, data *AgentCommission) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentCommissionIdKey := fmt.Sprintf("%s%v", cacheQncAgentCommissionIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentCommissionRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.AgentId, data.OrderId, data.ProductId, data.Amount, data.Status, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.AgentId, data.OrderId, data.ProductId, data.Amount, data.Status, data.DeleteTime, data.DelState, data.Version) + }, qncAgentCommissionIdKey) +} +func (m *defaultAgentCommissionModel) insertUUID(data *AgentCommission) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentCommissionModel) FindOne(ctx context.Context, id string) (*AgentCommission, error) { + qncAgentCommissionIdKey := fmt.Sprintf("%s%v", cacheQncAgentCommissionIdPrefix, id) + var resp AgentCommission + err := m.QueryRowCtx(ctx, &resp, qncAgentCommissionIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentCommissionRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentCommissionModel) Update(ctx context.Context, session sqlx.Session, data *AgentCommission) (sql.Result, error) { + qncAgentCommissionIdKey := fmt.Sprintf("%s%v", cacheQncAgentCommissionIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentCommissionRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.AgentId, data.OrderId, data.ProductId, data.Amount, data.Status, data.DeleteTime, data.DelState, data.Version, data.Id) + } + return conn.ExecCtx(ctx, query, data.AgentId, data.OrderId, data.ProductId, data.Amount, data.Status, data.DeleteTime, data.DelState, data.Version, data.Id) + }, qncAgentCommissionIdKey) +} + +func (m *defaultAgentCommissionModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentCommission) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncAgentCommissionIdKey := fmt.Sprintf("%s%v", cacheQncAgentCommissionIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentCommissionRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.AgentId, data.OrderId, data.ProductId, data.Amount, data.Status, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.AgentId, data.OrderId, data.ProductId, data.Amount, data.Status, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + }, qncAgentCommissionIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentCommissionModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentCommission) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentCommissionModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentCommissionModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentCommissionModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentCommissionModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentCommission, error) { + + builder = builder.Columns(agentCommissionRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentCommission + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentCommissionModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentCommission, error) { + + builder = builder.Columns(agentCommissionRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentCommission + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentCommissionModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentCommission, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentCommissionRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentCommission + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentCommissionModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentCommission, error) { + + builder = builder.Columns(agentCommissionRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentCommission + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentCommissionModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentCommission, error) { + + builder = builder.Columns(agentCommissionRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentCommission + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentCommissionModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentCommissionModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentCommissionModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncAgentCommissionIdKey := fmt.Sprintf("%s%v", cacheQncAgentCommissionIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentCommissionIdKey) + return err +} +func (m *defaultAgentCommissionModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentCommissionIdPrefix, primary) +} +func (m *defaultAgentCommissionModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentCommissionRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentCommissionModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentConfigModel.go b/app/main/model/agentConfigModel.go new file mode 100644 index 0000000..87917ed --- /dev/null +++ b/app/main/model/agentConfigModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentConfigModel = (*customAgentConfigModel)(nil) + +type ( + // AgentConfigModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentConfigModel. + AgentConfigModel interface { + agentConfigModel + } + + customAgentConfigModel struct { + *defaultAgentConfigModel + } +) + +// NewAgentConfigModel returns a model for the database table. +func NewAgentConfigModel(conn sqlx.SqlConn, c cache.CacheConf) AgentConfigModel { + return &customAgentConfigModel{ + defaultAgentConfigModel: newAgentConfigModel(conn, c), + } +} diff --git a/app/main/model/agentConfigModel_gen.go b/app/main/model/agentConfigModel_gen.go new file mode 100644 index 0000000..c2ea5cb --- /dev/null +++ b/app/main/model/agentConfigModel_gen.go @@ -0,0 +1,429 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentConfigFieldNames = builder.RawFieldNames(&AgentConfig{}) + agentConfigRows = strings.Join(agentConfigFieldNames, ",") + agentConfigRowsExpectAutoSet = strings.Join(stringx.Remove(agentConfigFieldNames, "`create_time`", "`update_time`"), ",") + agentConfigRowsWithPlaceHolder = strings.Join(stringx.Remove(agentConfigFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentConfigIdPrefix = "cache:qnc:agentConfig:id:" + cacheQncAgentConfigConfigKeyPrefix = "cache:qnc:agentConfig:configKey:" +) + +type ( + agentConfigModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentConfig) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentConfig, error) + FindOneByConfigKey(ctx context.Context, configKey string) (*AgentConfig, error) + Update(ctx context.Context, session sqlx.Session, data *AgentConfig) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentConfig) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentConfig) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentConfig, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentConfig, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentConfig, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentConfig, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentConfig, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentConfigModel struct { + sqlc.CachedConn + table string + } + + AgentConfig struct { + Id string `db:"id"` + ConfigKey string `db:"config_key"` // 配置键 + ConfigValue string `db:"config_value"` // 配置值 + ConfigType string `db:"config_type"` // 配置类型：price=价格，bonus=等级加成，upgrade=升级费用，rebate=返佣，tax=税费 + Description sql.NullString `db:"description"` // 配置描述 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentConfigModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentConfigModel { + return &defaultAgentConfigModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_config`", + } +} + +func (m *defaultAgentConfigModel) Insert(ctx context.Context, session sqlx.Session, data *AgentConfig) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentConfigConfigKeyKey := fmt.Sprintf("%s%v", cacheQncAgentConfigConfigKeyPrefix, data.ConfigKey) + qncAgentConfigIdKey := fmt.Sprintf("%s%v", cacheQncAgentConfigIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentConfigRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.ConfigKey, data.ConfigValue, data.ConfigType, data.Description, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.ConfigKey, data.ConfigValue, data.ConfigType, data.Description, data.DeleteTime, data.DelState, data.Version) + }, qncAgentConfigConfigKeyKey, qncAgentConfigIdKey) +} +func (m *defaultAgentConfigModel) insertUUID(data *AgentConfig) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentConfigModel) FindOne(ctx context.Context, id string) (*AgentConfig, error) { + qncAgentConfigIdKey := fmt.Sprintf("%s%v", cacheQncAgentConfigIdPrefix, id) + var resp AgentConfig + err := m.QueryRowCtx(ctx, &resp, qncAgentConfigIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentConfigRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentConfigModel) FindOneByConfigKey(ctx context.Context, configKey string) (*AgentConfig, error) { + qncAgentConfigConfigKeyKey := fmt.Sprintf("%s%v", cacheQncAgentConfigConfigKeyPrefix, configKey) + var resp AgentConfig + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentConfigConfigKeyKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `config_key` = ? and del_state = ? limit 1", agentConfigRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, configKey, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentConfigModel) Update(ctx context.Context, session sqlx.Session, newData *AgentConfig) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAgentConfigConfigKeyKey := fmt.Sprintf("%s%v", cacheQncAgentConfigConfigKeyPrefix, data.ConfigKey) + qncAgentConfigIdKey := fmt.Sprintf("%s%v", cacheQncAgentConfigIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentConfigRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.ConfigKey, newData.ConfigValue, newData.ConfigType, newData.Description, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.ConfigKey, newData.ConfigValue, newData.ConfigType, newData.Description, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, qncAgentConfigConfigKeyKey, qncAgentConfigIdKey) +} + +func (m *defaultAgentConfigModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AgentConfig) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAgentConfigConfigKeyKey := fmt.Sprintf("%s%v", cacheQncAgentConfigConfigKeyPrefix, data.ConfigKey) + qncAgentConfigIdKey := fmt.Sprintf("%s%v", cacheQncAgentConfigIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentConfigRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.ConfigKey, newData.ConfigValue, newData.ConfigType, newData.Description, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.ConfigKey, newData.ConfigValue, newData.ConfigType, newData.Description, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, qncAgentConfigConfigKeyKey, qncAgentConfigIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentConfigModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentConfig) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentConfigModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentConfigModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentConfigModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentConfigModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentConfig, error) { + + builder = builder.Columns(agentConfigRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentConfigModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentConfig, error) { + + builder = builder.Columns(agentConfigRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentConfigModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentConfig, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentConfigRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentConfigModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentConfig, error) { + + builder = builder.Columns(agentConfigRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentConfigModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentConfig, error) { + + builder = builder.Columns(agentConfigRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentConfigModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentConfigModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentConfigModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAgentConfigConfigKeyKey := fmt.Sprintf("%s%v", cacheQncAgentConfigConfigKeyPrefix, data.ConfigKey) + qncAgentConfigIdKey := fmt.Sprintf("%s%v", cacheQncAgentConfigIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentConfigConfigKeyKey, qncAgentConfigIdKey) + return err +} +func (m *defaultAgentConfigModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentConfigIdPrefix, primary) +} +func (m *defaultAgentConfigModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentConfigRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentConfigModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentFreezeTaskModel.go b/app/main/model/agentFreezeTaskModel.go new file mode 100644 index 0000000..ac3121d --- /dev/null +++ b/app/main/model/agentFreezeTaskModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentFreezeTaskModel = (*customAgentFreezeTaskModel)(nil) + +type ( + // AgentFreezeTaskModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentFreezeTaskModel. + AgentFreezeTaskModel interface { + agentFreezeTaskModel + } + + customAgentFreezeTaskModel struct { + *defaultAgentFreezeTaskModel + } +) + +// NewAgentFreezeTaskModel returns a model for the database table. +func NewAgentFreezeTaskModel(conn sqlx.SqlConn, c cache.CacheConf) AgentFreezeTaskModel { + return &customAgentFreezeTaskModel{ + defaultAgentFreezeTaskModel: newAgentFreezeTaskModel(conn, c), + } +} diff --git a/app/main/model/agentFreezeTaskModel_gen.go b/app/main/model/agentFreezeTaskModel_gen.go new file mode 100644 index 0000000..1eb3836 --- /dev/null +++ b/app/main/model/agentFreezeTaskModel_gen.go @@ -0,0 +1,397 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentFreezeTaskFieldNames = builder.RawFieldNames(&AgentFreezeTask{}) + agentFreezeTaskRows = strings.Join(agentFreezeTaskFieldNames, ",") + agentFreezeTaskRowsExpectAutoSet = strings.Join(stringx.Remove(agentFreezeTaskFieldNames, "`create_time`", "`update_time`"), ",") + agentFreezeTaskRowsWithPlaceHolder = strings.Join(stringx.Remove(agentFreezeTaskFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentFreezeTaskIdPrefix = "cache:qnc:agentFreezeTask:id:" +) + +type ( + agentFreezeTaskModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentFreezeTask) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentFreezeTask, error) + Update(ctx context.Context, session sqlx.Session, data *AgentFreezeTask) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentFreezeTask) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentFreezeTask) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentFreezeTask, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentFreezeTask, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentFreezeTask, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentFreezeTask, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentFreezeTask, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentFreezeTaskModel struct { + sqlc.CachedConn + table string + } + + AgentFreezeTask struct { + Id string `db:"id"` + AgentId string `db:"agent_id"` + OrderId string `db:"order_id"` + CommissionId string `db:"commission_id"` + FreezeAmount float64 `db:"freeze_amount"` // 冻结金额 + OrderPrice float64 `db:"order_price"` // 订单单价 + FreezeRatio float64 `db:"freeze_ratio"` // 冻结比例（例如：0.1000表示10%） + Status int64 `db:"status"` // 状态：1=待解冻，2=已解冻，3=已取消 + FreezeTime time.Time `db:"freeze_time"` // 冻结时间 + UnfreezeTime time.Time `db:"unfreeze_time"` // 解冻时间（冻结时间+1个月） + ActualUnfreezeTime sql.NullTime `db:"actual_unfreeze_time"` // 实际解冻时间 + Remark sql.NullString `db:"remark"` // 备注 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentFreezeTaskModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentFreezeTaskModel { + return &defaultAgentFreezeTaskModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_freeze_task`", + } +} + +func (m *defaultAgentFreezeTaskModel) Insert(ctx context.Context, session sqlx.Session, data *AgentFreezeTask) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentFreezeTaskIdKey := fmt.Sprintf("%s%v", cacheQncAgentFreezeTaskIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentFreezeTaskRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.AgentId, data.OrderId, data.CommissionId, data.FreezeAmount, data.OrderPrice, data.FreezeRatio, data.Status, data.FreezeTime, data.UnfreezeTime, data.ActualUnfreezeTime, data.Remark, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.AgentId, data.OrderId, data.CommissionId, data.FreezeAmount, data.OrderPrice, data.FreezeRatio, data.Status, data.FreezeTime, data.UnfreezeTime, data.ActualUnfreezeTime, data.Remark, data.DeleteTime, data.DelState, data.Version) + }, qncAgentFreezeTaskIdKey) +} +func (m *defaultAgentFreezeTaskModel) insertUUID(data *AgentFreezeTask) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentFreezeTaskModel) FindOne(ctx context.Context, id string) (*AgentFreezeTask, error) { + qncAgentFreezeTaskIdKey := fmt.Sprintf("%s%v", cacheQncAgentFreezeTaskIdPrefix, id) + var resp AgentFreezeTask + err := m.QueryRowCtx(ctx, &resp, qncAgentFreezeTaskIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentFreezeTaskRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentFreezeTaskModel) Update(ctx context.Context, session sqlx.Session, data *AgentFreezeTask) (sql.Result, error) { + qncAgentFreezeTaskIdKey := fmt.Sprintf("%s%v", cacheQncAgentFreezeTaskIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentFreezeTaskRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.AgentId, data.OrderId, data.CommissionId, data.FreezeAmount, data.OrderPrice, data.FreezeRatio, data.Status, data.FreezeTime, data.UnfreezeTime, data.ActualUnfreezeTime, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id) + } + return conn.ExecCtx(ctx, query, data.AgentId, data.OrderId, data.CommissionId, data.FreezeAmount, data.OrderPrice, data.FreezeRatio, data.Status, data.FreezeTime, data.UnfreezeTime, data.ActualUnfreezeTime, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id) + }, qncAgentFreezeTaskIdKey) +} + +func (m *defaultAgentFreezeTaskModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentFreezeTask) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncAgentFreezeTaskIdKey := fmt.Sprintf("%s%v", cacheQncAgentFreezeTaskIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentFreezeTaskRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.AgentId, data.OrderId, data.CommissionId, data.FreezeAmount, data.OrderPrice, data.FreezeRatio, data.Status, data.FreezeTime, data.UnfreezeTime, data.ActualUnfreezeTime, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.AgentId, data.OrderId, data.CommissionId, data.FreezeAmount, data.OrderPrice, data.FreezeRatio, data.Status, data.FreezeTime, data.UnfreezeTime, data.ActualUnfreezeTime, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + }, qncAgentFreezeTaskIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentFreezeTaskModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentFreezeTask) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentFreezeTaskModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentFreezeTaskModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentFreezeTaskModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentFreezeTaskModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentFreezeTask, error) { + + builder = builder.Columns(agentFreezeTaskRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentFreezeTask + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentFreezeTaskModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentFreezeTask, error) { + + builder = builder.Columns(agentFreezeTaskRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentFreezeTask + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentFreezeTaskModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentFreezeTask, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentFreezeTaskRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentFreezeTask + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentFreezeTaskModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentFreezeTask, error) { + + builder = builder.Columns(agentFreezeTaskRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentFreezeTask + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentFreezeTaskModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentFreezeTask, error) { + + builder = builder.Columns(agentFreezeTaskRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentFreezeTask + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentFreezeTaskModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentFreezeTaskModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentFreezeTaskModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncAgentFreezeTaskIdKey := fmt.Sprintf("%s%v", cacheQncAgentFreezeTaskIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentFreezeTaskIdKey) + return err +} +func (m *defaultAgentFreezeTaskModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentFreezeTaskIdPrefix, primary) +} +func (m *defaultAgentFreezeTaskModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentFreezeTaskRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentFreezeTaskModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentInviteCodeModel.go b/app/main/model/agentInviteCodeModel.go new file mode 100644 index 0000000..9b02e4a --- /dev/null +++ b/app/main/model/agentInviteCodeModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentInviteCodeModel = (*customAgentInviteCodeModel)(nil) + +type ( + // AgentInviteCodeModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentInviteCodeModel. + AgentInviteCodeModel interface { + agentInviteCodeModel + } + + customAgentInviteCodeModel struct { + *defaultAgentInviteCodeModel + } +) + +// NewAgentInviteCodeModel returns a model for the database table. +func NewAgentInviteCodeModel(conn sqlx.SqlConn, c cache.CacheConf) AgentInviteCodeModel { + return &customAgentInviteCodeModel{ + defaultAgentInviteCodeModel: newAgentInviteCodeModel(conn, c), + } +} diff --git a/app/main/model/agentInviteCodeModel_gen.go b/app/main/model/agentInviteCodeModel_gen.go new file mode 100644 index 0000000..14ee4d0 --- /dev/null +++ b/app/main/model/agentInviteCodeModel_gen.go @@ -0,0 +1,434 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentInviteCodeFieldNames = builder.RawFieldNames(&AgentInviteCode{}) + agentInviteCodeRows = strings.Join(agentInviteCodeFieldNames, ",") + agentInviteCodeRowsExpectAutoSet = strings.Join(stringx.Remove(agentInviteCodeFieldNames, "`create_time`", "`update_time`"), ",") + agentInviteCodeRowsWithPlaceHolder = strings.Join(stringx.Remove(agentInviteCodeFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentInviteCodeIdPrefix = "cache:qnc:agentInviteCode:id:" + cacheQncAgentInviteCodeCodePrefix = "cache:qnc:agentInviteCode:code:" +) + +type ( + agentInviteCodeModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentInviteCode) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentInviteCode, error) + FindOneByCode(ctx context.Context, code string) (*AgentInviteCode, error) + Update(ctx context.Context, session sqlx.Session, data *AgentInviteCode) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentInviteCode) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentInviteCode) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentInviteCode, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentInviteCode, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentInviteCode, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentInviteCode, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentInviteCode, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentInviteCodeModel struct { + sqlc.CachedConn + table string + } + + AgentInviteCode struct { + Id string `db:"id"` + Code string `db:"code"` // 邀请码（唯一） + AgentId sql.NullString `db:"agent_id"` + TargetLevel int64 `db:"target_level"` // 目标等级：1=普通，2=黄金，3=钻石 + Status int64 `db:"status"` // 状态：0=未使用，1=已使用，2=已失效（所有邀请码只能使用一次，使用后立即失效） + UsedUserId sql.NullString `db:"used_user_id"` + UsedAgentId sql.NullString `db:"used_agent_id"` + UsedTime sql.NullTime `db:"used_time"` // 使用时间 + ExpireTime sql.NullTime `db:"expire_time"` // 过期时间（可选） + Remark sql.NullString `db:"remark"` // 备注 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentInviteCodeModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentInviteCodeModel { + return &defaultAgentInviteCodeModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_invite_code`", + } +} + +func (m *defaultAgentInviteCodeModel) Insert(ctx context.Context, session sqlx.Session, data *AgentInviteCode) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentInviteCodeCodeKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeCodePrefix, data.Code) + qncAgentInviteCodeIdKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentInviteCodeRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.Code, data.AgentId, data.TargetLevel, data.Status, data.UsedUserId, data.UsedAgentId, data.UsedTime, data.ExpireTime, data.Remark, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.Code, data.AgentId, data.TargetLevel, data.Status, data.UsedUserId, data.UsedAgentId, data.UsedTime, data.ExpireTime, data.Remark, data.DeleteTime, data.DelState, data.Version) + }, qncAgentInviteCodeCodeKey, qncAgentInviteCodeIdKey) +} +func (m *defaultAgentInviteCodeModel) insertUUID(data *AgentInviteCode) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentInviteCodeModel) FindOne(ctx context.Context, id string) (*AgentInviteCode, error) { + qncAgentInviteCodeIdKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeIdPrefix, id) + var resp AgentInviteCode + err := m.QueryRowCtx(ctx, &resp, qncAgentInviteCodeIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentInviteCodeRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentInviteCodeModel) FindOneByCode(ctx context.Context, code string) (*AgentInviteCode, error) { + qncAgentInviteCodeCodeKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeCodePrefix, code) + var resp AgentInviteCode + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentInviteCodeCodeKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `code` = ? and del_state = ? limit 1", agentInviteCodeRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, code, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentInviteCodeModel) Update(ctx context.Context, session sqlx.Session, newData *AgentInviteCode) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAgentInviteCodeCodeKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeCodePrefix, data.Code) + qncAgentInviteCodeIdKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentInviteCodeRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.Code, newData.AgentId, newData.TargetLevel, newData.Status, newData.UsedUserId, newData.UsedAgentId, newData.UsedTime, newData.ExpireTime, newData.Remark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.Code, newData.AgentId, newData.TargetLevel, newData.Status, newData.UsedUserId, newData.UsedAgentId, newData.UsedTime, newData.ExpireTime, newData.Remark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, qncAgentInviteCodeCodeKey, qncAgentInviteCodeIdKey) +} + +func (m *defaultAgentInviteCodeModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AgentInviteCode) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAgentInviteCodeCodeKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeCodePrefix, data.Code) + qncAgentInviteCodeIdKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentInviteCodeRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.Code, newData.AgentId, newData.TargetLevel, newData.Status, newData.UsedUserId, newData.UsedAgentId, newData.UsedTime, newData.ExpireTime, newData.Remark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.Code, newData.AgentId, newData.TargetLevel, newData.Status, newData.UsedUserId, newData.UsedAgentId, newData.UsedTime, newData.ExpireTime, newData.Remark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, qncAgentInviteCodeCodeKey, qncAgentInviteCodeIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentInviteCodeModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentInviteCode) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentInviteCodeModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentInviteCodeModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentInviteCodeModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentInviteCodeModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentInviteCode, error) { + + builder = builder.Columns(agentInviteCodeRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentInviteCode + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentInviteCodeModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentInviteCode, error) { + + builder = builder.Columns(agentInviteCodeRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentInviteCode + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentInviteCodeModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentInviteCode, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentInviteCodeRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentInviteCode + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentInviteCodeModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentInviteCode, error) { + + builder = builder.Columns(agentInviteCodeRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentInviteCode + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentInviteCodeModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentInviteCode, error) { + + builder = builder.Columns(agentInviteCodeRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentInviteCode + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentInviteCodeModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentInviteCodeModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentInviteCodeModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAgentInviteCodeCodeKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeCodePrefix, data.Code) + qncAgentInviteCodeIdKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentInviteCodeCodeKey, qncAgentInviteCodeIdKey) + return err +} +func (m *defaultAgentInviteCodeModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentInviteCodeIdPrefix, primary) +} +func (m *defaultAgentInviteCodeModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentInviteCodeRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentInviteCodeModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentInviteCodeUsageModel.go b/app/main/model/agentInviteCodeUsageModel.go new file mode 100644 index 0000000..8af056e --- /dev/null +++ b/app/main/model/agentInviteCodeUsageModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentInviteCodeUsageModel = (*customAgentInviteCodeUsageModel)(nil) + +type ( + // AgentInviteCodeUsageModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentInviteCodeUsageModel. + AgentInviteCodeUsageModel interface { + agentInviteCodeUsageModel + } + + customAgentInviteCodeUsageModel struct { + *defaultAgentInviteCodeUsageModel + } +) + +// NewAgentInviteCodeUsageModel returns a model for the database table. +func NewAgentInviteCodeUsageModel(conn sqlx.SqlConn, c cache.CacheConf) AgentInviteCodeUsageModel { + return &customAgentInviteCodeUsageModel{ + defaultAgentInviteCodeUsageModel: newAgentInviteCodeUsageModel(conn, c), + } +} diff --git a/app/main/model/agentInviteCodeUsageModel_gen.go b/app/main/model/agentInviteCodeUsageModel_gen.go new file mode 100644 index 0000000..0ebd687 --- /dev/null +++ b/app/main/model/agentInviteCodeUsageModel_gen.go @@ -0,0 +1,392 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentInviteCodeUsageFieldNames = builder.RawFieldNames(&AgentInviteCodeUsage{}) + agentInviteCodeUsageRows = strings.Join(agentInviteCodeUsageFieldNames, ",") + agentInviteCodeUsageRowsExpectAutoSet = strings.Join(stringx.Remove(agentInviteCodeUsageFieldNames, "`create_time`", "`update_time`"), ",") + agentInviteCodeUsageRowsWithPlaceHolder = strings.Join(stringx.Remove(agentInviteCodeUsageFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentInviteCodeUsageIdPrefix = "cache:qnc:agentInviteCodeUsage:id:" +) + +type ( + agentInviteCodeUsageModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentInviteCodeUsage) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentInviteCodeUsage, error) + Update(ctx context.Context, session sqlx.Session, data *AgentInviteCodeUsage) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentInviteCodeUsage) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentInviteCodeUsage) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentInviteCodeUsage, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentInviteCodeUsage, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentInviteCodeUsage, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentInviteCodeUsage, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentInviteCodeUsage, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentInviteCodeUsageModel struct { + sqlc.CachedConn + table string + } + + AgentInviteCodeUsage struct { + Id string `db:"id"` + InviteCodeId string `db:"invite_code_id"` + Code string `db:"code"` // 邀请码（冗余字段，便于查询） + UserId string `db:"user_id"` + AgentId string `db:"agent_id"` + AgentLevel int64 `db:"agent_level"` // 代理等级：1=普通，2=黄金，3=钻石 + UsedTime time.Time `db:"used_time"` // 使用时间 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentInviteCodeUsageModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentInviteCodeUsageModel { + return &defaultAgentInviteCodeUsageModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_invite_code_usage`", + } +} + +func (m *defaultAgentInviteCodeUsageModel) Insert(ctx context.Context, session sqlx.Session, data *AgentInviteCodeUsage) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentInviteCodeUsageIdKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeUsageIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentInviteCodeUsageRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.InviteCodeId, data.Code, data.UserId, data.AgentId, data.AgentLevel, data.UsedTime, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.InviteCodeId, data.Code, data.UserId, data.AgentId, data.AgentLevel, data.UsedTime, data.DeleteTime, data.DelState, data.Version) + }, qncAgentInviteCodeUsageIdKey) +} +func (m *defaultAgentInviteCodeUsageModel) insertUUID(data *AgentInviteCodeUsage) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentInviteCodeUsageModel) FindOne(ctx context.Context, id string) (*AgentInviteCodeUsage, error) { + qncAgentInviteCodeUsageIdKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeUsageIdPrefix, id) + var resp AgentInviteCodeUsage + err := m.QueryRowCtx(ctx, &resp, qncAgentInviteCodeUsageIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentInviteCodeUsageRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentInviteCodeUsageModel) Update(ctx context.Context, session sqlx.Session, data *AgentInviteCodeUsage) (sql.Result, error) { + qncAgentInviteCodeUsageIdKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeUsageIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentInviteCodeUsageRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.InviteCodeId, data.Code, data.UserId, data.AgentId, data.AgentLevel, data.UsedTime, data.DeleteTime, data.DelState, data.Version, data.Id) + } + return conn.ExecCtx(ctx, query, data.InviteCodeId, data.Code, data.UserId, data.AgentId, data.AgentLevel, data.UsedTime, data.DeleteTime, data.DelState, data.Version, data.Id) + }, qncAgentInviteCodeUsageIdKey) +} + +func (m *defaultAgentInviteCodeUsageModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentInviteCodeUsage) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncAgentInviteCodeUsageIdKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeUsageIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentInviteCodeUsageRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.InviteCodeId, data.Code, data.UserId, data.AgentId, data.AgentLevel, data.UsedTime, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.InviteCodeId, data.Code, data.UserId, data.AgentId, data.AgentLevel, data.UsedTime, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + }, qncAgentInviteCodeUsageIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentInviteCodeUsageModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentInviteCodeUsage) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentInviteCodeUsageModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentInviteCodeUsageModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentInviteCodeUsageModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentInviteCodeUsageModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentInviteCodeUsage, error) { + + builder = builder.Columns(agentInviteCodeUsageRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentInviteCodeUsage + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentInviteCodeUsageModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentInviteCodeUsage, error) { + + builder = builder.Columns(agentInviteCodeUsageRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentInviteCodeUsage + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentInviteCodeUsageModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentInviteCodeUsage, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentInviteCodeUsageRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentInviteCodeUsage + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentInviteCodeUsageModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentInviteCodeUsage, error) { + + builder = builder.Columns(agentInviteCodeUsageRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentInviteCodeUsage + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentInviteCodeUsageModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentInviteCodeUsage, error) { + + builder = builder.Columns(agentInviteCodeUsageRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentInviteCodeUsage + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentInviteCodeUsageModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentInviteCodeUsageModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentInviteCodeUsageModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncAgentInviteCodeUsageIdKey := fmt.Sprintf("%s%v", cacheQncAgentInviteCodeUsageIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentInviteCodeUsageIdKey) + return err +} +func (m *defaultAgentInviteCodeUsageModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentInviteCodeUsageIdPrefix, primary) +} +func (m *defaultAgentInviteCodeUsageModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentInviteCodeUsageRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentInviteCodeUsageModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentLinkModel.go b/app/main/model/agentLinkModel.go new file mode 100644 index 0000000..ee3c37c --- /dev/null +++ b/app/main/model/agentLinkModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentLinkModel = (*customAgentLinkModel)(nil) + +type ( + // AgentLinkModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentLinkModel. + AgentLinkModel interface { + agentLinkModel + } + + customAgentLinkModel struct { + *defaultAgentLinkModel + } +) + +// NewAgentLinkModel returns a model for the database table. +func NewAgentLinkModel(conn sqlx.SqlConn, c cache.CacheConf) AgentLinkModel { + return &customAgentLinkModel{ + defaultAgentLinkModel: newAgentLinkModel(conn, c), + } +} diff --git a/app/main/model/agentLinkModel_gen.go b/app/main/model/agentLinkModel_gen.go new file mode 100644 index 0000000..7e8c777 --- /dev/null +++ b/app/main/model/agentLinkModel_gen.go @@ -0,0 +1,457 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentLinkFieldNames = builder.RawFieldNames(&AgentLink{}) + agentLinkRows = strings.Join(agentLinkFieldNames, ",") + agentLinkRowsExpectAutoSet = strings.Join(stringx.Remove(agentLinkFieldNames, "`create_time`", "`update_time`"), ",") + agentLinkRowsWithPlaceHolder = strings.Join(stringx.Remove(agentLinkFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentLinkIdPrefix = "cache:qnc:agentLink:id:" + cacheQncAgentLinkAgentIdProductIdSetPriceDelStatePrefix = "cache:qnc:agentLink:agentId:productId:setPrice:delState:" + cacheQncAgentLinkLinkIdentifierPrefix = "cache:qnc:agentLink:linkIdentifier:" +) + +type ( + agentLinkModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentLink) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentLink, error) + FindOneByAgentIdProductIdSetPriceDelState(ctx context.Context, agentId string, productId string, setPrice float64, delState int64) (*AgentLink, error) + FindOneByLinkIdentifier(ctx context.Context, linkIdentifier string) (*AgentLink, error) + Update(ctx context.Context, session sqlx.Session, data *AgentLink) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentLink) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentLink) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentLink, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentLink, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentLink, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentLink, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentLink, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentLinkModel struct { + sqlc.CachedConn + table string + } + + AgentLink struct { + Id string `db:"id"` + AgentId string `db:"agent_id"` + UserId string `db:"user_id"` + ProductId string `db:"product_id"` + LinkIdentifier string `db:"link_identifier"` // 推广链接标识（加密） + SetPrice float64 `db:"set_price"` // 代理设定价格 + ActualBasePrice float64 `db:"actual_base_price"` // 实际底价（基础底价+等级加成） + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentLinkModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentLinkModel { + return &defaultAgentLinkModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_link`", + } +} + +func (m *defaultAgentLinkModel) Insert(ctx context.Context, session sqlx.Session, data *AgentLink) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentLinkAgentIdProductIdSetPriceDelStateKey := fmt.Sprintf("%s%v:%v:%v:%v", cacheQncAgentLinkAgentIdProductIdSetPriceDelStatePrefix, data.AgentId, data.ProductId, data.SetPrice, data.DelState) + qncAgentLinkIdKey := fmt.Sprintf("%s%v", cacheQncAgentLinkIdPrefix, data.Id) + qncAgentLinkLinkIdentifierKey := fmt.Sprintf("%s%v", cacheQncAgentLinkLinkIdentifierPrefix, data.LinkIdentifier) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentLinkRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.AgentId, data.UserId, data.ProductId, data.LinkIdentifier, data.SetPrice, data.ActualBasePrice, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.AgentId, data.UserId, data.ProductId, data.LinkIdentifier, data.SetPrice, data.ActualBasePrice, data.DeleteTime, data.DelState, data.Version) + }, qncAgentLinkAgentIdProductIdSetPriceDelStateKey, qncAgentLinkIdKey, qncAgentLinkLinkIdentifierKey) +} +func (m *defaultAgentLinkModel) insertUUID(data *AgentLink) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentLinkModel) FindOne(ctx context.Context, id string) (*AgentLink, error) { + qncAgentLinkIdKey := fmt.Sprintf("%s%v", cacheQncAgentLinkIdPrefix, id) + var resp AgentLink + err := m.QueryRowCtx(ctx, &resp, qncAgentLinkIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentLinkRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentLinkModel) FindOneByAgentIdProductIdSetPriceDelState(ctx context.Context, agentId string, productId string, setPrice float64, delState int64) (*AgentLink, error) { + qncAgentLinkAgentIdProductIdSetPriceDelStateKey := fmt.Sprintf("%s%v:%v:%v:%v", cacheQncAgentLinkAgentIdProductIdSetPriceDelStatePrefix, agentId, productId, setPrice, delState) + var resp AgentLink + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentLinkAgentIdProductIdSetPriceDelStateKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `agent_id` = ? and `product_id` = ? and `set_price` = ? and `del_state` = ? and del_state = ? limit 1", agentLinkRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, agentId, productId, setPrice, delState, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentLinkModel) FindOneByLinkIdentifier(ctx context.Context, linkIdentifier string) (*AgentLink, error) { + qncAgentLinkLinkIdentifierKey := fmt.Sprintf("%s%v", cacheQncAgentLinkLinkIdentifierPrefix, linkIdentifier) + var resp AgentLink + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentLinkLinkIdentifierKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `link_identifier` = ? and del_state = ? limit 1", agentLinkRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, linkIdentifier, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentLinkModel) Update(ctx context.Context, session sqlx.Session, newData *AgentLink) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAgentLinkAgentIdProductIdSetPriceDelStateKey := fmt.Sprintf("%s%v:%v:%v:%v", cacheQncAgentLinkAgentIdProductIdSetPriceDelStatePrefix, data.AgentId, data.ProductId, data.SetPrice, data.DelState) + qncAgentLinkIdKey := fmt.Sprintf("%s%v", cacheQncAgentLinkIdPrefix, data.Id) + qncAgentLinkLinkIdentifierKey := fmt.Sprintf("%s%v", cacheQncAgentLinkLinkIdentifierPrefix, data.LinkIdentifier) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentLinkRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AgentId, newData.UserId, newData.ProductId, newData.LinkIdentifier, newData.SetPrice, newData.ActualBasePrice, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.AgentId, newData.UserId, newData.ProductId, newData.LinkIdentifier, newData.SetPrice, newData.ActualBasePrice, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, qncAgentLinkAgentIdProductIdSetPriceDelStateKey, qncAgentLinkIdKey, qncAgentLinkLinkIdentifierKey) +} + +func (m *defaultAgentLinkModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AgentLink) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAgentLinkAgentIdProductIdSetPriceDelStateKey := fmt.Sprintf("%s%v:%v:%v:%v", cacheQncAgentLinkAgentIdProductIdSetPriceDelStatePrefix, data.AgentId, data.ProductId, data.SetPrice, data.DelState) + qncAgentLinkIdKey := fmt.Sprintf("%s%v", cacheQncAgentLinkIdPrefix, data.Id) + qncAgentLinkLinkIdentifierKey := fmt.Sprintf("%s%v", cacheQncAgentLinkLinkIdentifierPrefix, data.LinkIdentifier) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentLinkRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AgentId, newData.UserId, newData.ProductId, newData.LinkIdentifier, newData.SetPrice, newData.ActualBasePrice, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.AgentId, newData.UserId, newData.ProductId, newData.LinkIdentifier, newData.SetPrice, newData.ActualBasePrice, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, qncAgentLinkAgentIdProductIdSetPriceDelStateKey, qncAgentLinkIdKey, qncAgentLinkLinkIdentifierKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentLinkModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentLink) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentLinkModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentLinkModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentLinkModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentLinkModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentLink, error) { + + builder = builder.Columns(agentLinkRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentLink + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentLinkModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentLink, error) { + + builder = builder.Columns(agentLinkRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentLink + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentLinkModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentLink, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentLinkRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentLink + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentLinkModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentLink, error) { + + builder = builder.Columns(agentLinkRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentLink + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentLinkModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentLink, error) { + + builder = builder.Columns(agentLinkRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentLink + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentLinkModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentLinkModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentLinkModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAgentLinkAgentIdProductIdSetPriceDelStateKey := fmt.Sprintf("%s%v:%v:%v:%v", cacheQncAgentLinkAgentIdProductIdSetPriceDelStatePrefix, data.AgentId, data.ProductId, data.SetPrice, data.DelState) + qncAgentLinkIdKey := fmt.Sprintf("%s%v", cacheQncAgentLinkIdPrefix, id) + qncAgentLinkLinkIdentifierKey := fmt.Sprintf("%s%v", cacheQncAgentLinkLinkIdentifierPrefix, data.LinkIdentifier) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentLinkAgentIdProductIdSetPriceDelStateKey, qncAgentLinkIdKey, qncAgentLinkLinkIdentifierKey) + return err +} +func (m *defaultAgentLinkModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentLinkIdPrefix, primary) +} +func (m *defaultAgentLinkModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentLinkRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentLinkModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentModel.go b/app/main/model/agentModel.go new file mode 100644 index 0000000..eea1285 --- /dev/null +++ b/app/main/model/agentModel.go @@ -0,0 +1,65 @@ +package model + +import ( + "context" + "fmt" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentModel = (*customAgentModel)(nil) + +type ( + // AgentModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentModel. + AgentModel interface { + agentModel + // UpdateInTransaction 在事务中更新刚插入的记录，避免 UpdateWithVersion 中的 FindOne 缓存问题 + UpdateInTransaction(ctx context.Context, session sqlx.Session, data *Agent) error + } + + customAgentModel struct { + *defaultAgentModel + } +) + +// NewAgentModel returns a model for the database table. +func NewAgentModel(conn sqlx.SqlConn, c cache.CacheConf) AgentModel { + return &customAgentModel{ + defaultAgentModel: newAgentModel(conn, c), + } +} + +// UpdateInTransaction 在事务中更新刚插入的记录，避免 UpdateWithVersion 中的 FindOne 缓存问题 +// 注意：此方法假设记录刚插入，version 为 0，更新后 version 为 1 +func (m *customAgentModel) UpdateInTransaction(ctx context.Context, session sqlx.Session, data *Agent) error { + if session == nil { + return errors.New("session is required for UpdateInTransaction") + } + + // 直接执行 SQL UPDATE，避免 FindOne 的缓存问题 + // 字段顺序：user_id, level, region, mobile, wechat_id, team_leader_id, delete_time, del_state, version + query := fmt.Sprintf("update %s set `user_id`=?, `level`=?, `region`=?, `mobile`=?, `wechat_id`=?, `team_leader_id`=?, `delete_time`=?, `del_state`=?, `version`=? where `id`=? and `version`=?", m.table) + + // 确保 version 从 0 增加到 1 + oldVersion := data.Version + data.Version = oldVersion + 1 + + result, err := session.ExecCtx(ctx, query, data.UserId, data.Level, data.Region, data.Mobile, data.WechatId, data.TeamLeaderId, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + if err != nil { + return errors.Wrapf(err, "UpdateInTransaction failed") + } + + rowsAffected, err := result.RowsAffected() + if err != nil { + return errors.Wrapf(err, "UpdateInTransaction failed to get rows affected") + } + + if rowsAffected == 0 { + return errors.Wrapf(ErrNoRowsUpdate, "UpdateInTransaction: no rows updated, version may not match") + } + + return nil +} diff --git a/app/main/model/agentModel_gen.go b/app/main/model/agentModel_gen.go new file mode 100644 index 0000000..2ea9d5d --- /dev/null +++ b/app/main/model/agentModel_gen.go @@ -0,0 +1,459 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentFieldNames = builder.RawFieldNames(&Agent{}) + agentRows = strings.Join(agentFieldNames, ",") + agentRowsExpectAutoSet = strings.Join(stringx.Remove(agentFieldNames, "`create_time`", "`update_time`"), ",") + agentRowsWithPlaceHolder = strings.Join(stringx.Remove(agentFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentIdPrefix = "cache:qnc:agent:id:" + cacheQncAgentAgentCodePrefix = "cache:qnc:agent:agentCode:" + cacheQncAgentUserIdPrefix = "cache:qnc:agent:userId:" +) + +type ( + agentModel interface { + Insert(ctx context.Context, session sqlx.Session, data *Agent) (sql.Result, error) + FindOne(ctx context.Context, id string) (*Agent, error) + FindOneByAgentCode(ctx context.Context, agentCode int64) (*Agent, error) + FindOneByUserId(ctx context.Context, userId string) (*Agent, error) + Update(ctx context.Context, session sqlx.Session, data *Agent) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *Agent) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *Agent) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*Agent, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Agent, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Agent, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Agent, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Agent, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentModel struct { + sqlc.CachedConn + table string + } + + Agent struct { + Id string `db:"id"` + UserId string `db:"user_id"` + AgentCode int64 `db:"agent_code"` // 代理编码（从16800开始递增） + Level int64 `db:"level"` // 代理等级：1=普通，2=黄金，3=钻石 + Region sql.NullString `db:"region"` // 区域（可选） + Mobile string `db:"mobile"` // 手机号（加密） + WechatId sql.NullString `db:"wechat_id"` // 微信号 + TeamLeaderId sql.NullString `db:"team_leader_id"` + InviteCodeId sql.NullString `db:"invite_code_id"` + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentModel { + return &defaultAgentModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent`", + } +} + +func (m *defaultAgentModel) Insert(ctx context.Context, session sqlx.Session, data *Agent) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentAgentCodeKey := fmt.Sprintf("%s%v", cacheQncAgentAgentCodePrefix, data.AgentCode) + qncAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentIdPrefix, data.Id) + qncAgentUserIdKey := fmt.Sprintf("%s%v", cacheQncAgentUserIdPrefix, data.UserId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.UserId, data.AgentCode, data.Level, data.Region, data.Mobile, data.WechatId, data.TeamLeaderId, data.InviteCodeId, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.UserId, data.AgentCode, data.Level, data.Region, data.Mobile, data.WechatId, data.TeamLeaderId, data.InviteCodeId, data.DeleteTime, data.DelState, data.Version) + }, qncAgentAgentCodeKey, qncAgentIdKey, qncAgentUserIdKey) +} +func (m *defaultAgentModel) insertUUID(data *Agent) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentModel) FindOne(ctx context.Context, id string) (*Agent, error) { + qncAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentIdPrefix, id) + var resp Agent + err := m.QueryRowCtx(ctx, &resp, qncAgentIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentModel) FindOneByAgentCode(ctx context.Context, agentCode int64) (*Agent, error) { + qncAgentAgentCodeKey := fmt.Sprintf("%s%v", cacheQncAgentAgentCodePrefix, agentCode) + var resp Agent + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentAgentCodeKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `agent_code` = ? and del_state = ? limit 1", agentRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, agentCode, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentModel) FindOneByUserId(ctx context.Context, userId string) (*Agent, error) { + qncAgentUserIdKey := fmt.Sprintf("%s%v", cacheQncAgentUserIdPrefix, userId) + var resp Agent + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentUserIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `user_id` = ? and del_state = ? limit 1", agentRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, userId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentModel) Update(ctx context.Context, session sqlx.Session, newData *Agent) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAgentAgentCodeKey := fmt.Sprintf("%s%v", cacheQncAgentAgentCodePrefix, data.AgentCode) + qncAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentIdPrefix, data.Id) + qncAgentUserIdKey := fmt.Sprintf("%s%v", cacheQncAgentUserIdPrefix, data.UserId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.UserId, newData.AgentCode, newData.Level, newData.Region, newData.Mobile, newData.WechatId, newData.TeamLeaderId, newData.InviteCodeId, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.UserId, newData.AgentCode, newData.Level, newData.Region, newData.Mobile, newData.WechatId, newData.TeamLeaderId, newData.InviteCodeId, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, qncAgentAgentCodeKey, qncAgentIdKey, qncAgentUserIdKey) +} + +func (m *defaultAgentModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *Agent) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAgentAgentCodeKey := fmt.Sprintf("%s%v", cacheQncAgentAgentCodePrefix, data.AgentCode) + qncAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentIdPrefix, data.Id) + qncAgentUserIdKey := fmt.Sprintf("%s%v", cacheQncAgentUserIdPrefix, data.UserId) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.UserId, newData.AgentCode, newData.Level, newData.Region, newData.Mobile, newData.WechatId, newData.TeamLeaderId, newData.InviteCodeId, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.UserId, newData.AgentCode, newData.Level, newData.Region, newData.Mobile, newData.WechatId, newData.TeamLeaderId, newData.InviteCodeId, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, qncAgentAgentCodeKey, qncAgentIdKey, qncAgentUserIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *Agent) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*Agent, error) { + + builder = builder.Columns(agentRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*Agent + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Agent, error) { + + builder = builder.Columns(agentRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Agent + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Agent, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*Agent + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Agent, error) { + + builder = builder.Columns(agentRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Agent + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Agent, error) { + + builder = builder.Columns(agentRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Agent + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAgentAgentCodeKey := fmt.Sprintf("%s%v", cacheQncAgentAgentCodePrefix, data.AgentCode) + qncAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentIdPrefix, id) + qncAgentUserIdKey := fmt.Sprintf("%s%v", cacheQncAgentUserIdPrefix, data.UserId) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentAgentCodeKey, qncAgentIdKey, qncAgentUserIdKey) + return err +} +func (m *defaultAgentModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentIdPrefix, primary) +} +func (m *defaultAgentModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentOrderModel.go b/app/main/model/agentOrderModel.go new file mode 100644 index 0000000..00ded5e --- /dev/null +++ b/app/main/model/agentOrderModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentOrderModel = (*customAgentOrderModel)(nil) + +type ( + // AgentOrderModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentOrderModel. + AgentOrderModel interface { + agentOrderModel + } + + customAgentOrderModel struct { + *defaultAgentOrderModel + } +) + +// NewAgentOrderModel returns a model for the database table. +func NewAgentOrderModel(conn sqlx.SqlConn, c cache.CacheConf) AgentOrderModel { + return &customAgentOrderModel{ + defaultAgentOrderModel: newAgentOrderModel(conn, c), + } +} diff --git a/app/main/model/agentOrderModel_gen.go b/app/main/model/agentOrderModel_gen.go new file mode 100644 index 0000000..3ca2de3 --- /dev/null +++ b/app/main/model/agentOrderModel_gen.go @@ -0,0 +1,436 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentOrderFieldNames = builder.RawFieldNames(&AgentOrder{}) + agentOrderRows = strings.Join(agentOrderFieldNames, ",") + agentOrderRowsExpectAutoSet = strings.Join(stringx.Remove(agentOrderFieldNames, "`create_time`", "`update_time`"), ",") + agentOrderRowsWithPlaceHolder = strings.Join(stringx.Remove(agentOrderFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentOrderIdPrefix = "cache:qnc:agentOrder:id:" + cacheQncAgentOrderOrderIdPrefix = "cache:qnc:agentOrder:orderId:" +) + +type ( + agentOrderModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentOrder) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentOrder, error) + FindOneByOrderId(ctx context.Context, orderId string) (*AgentOrder, error) + Update(ctx context.Context, session sqlx.Session, data *AgentOrder) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentOrder) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentOrder) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentOrder, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentOrder, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentOrder, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentOrder, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentOrder, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentOrderModel struct { + sqlc.CachedConn + table string + } + + AgentOrder struct { + Id string `db:"id"` + AgentId string `db:"agent_id"` + OrderId string `db:"order_id"` + ProductId string `db:"product_id"` + OrderAmount float64 `db:"order_amount"` // 订单金额（用户实际支付金额，冗余字段） + SetPrice float64 `db:"set_price"` // 代理设定价格 + ActualBasePrice float64 `db:"actual_base_price"` // 实际底价（基础底价+等级加成） + PriceCost float64 `db:"price_cost"` // 提价成本（(设定价格-提价标准阈值)×提价手续费比例） + AgentProfit float64 `db:"agent_profit"` // 代理收益（设定价格-实际底价-提价成本） + ProcessStatus int64 `db:"process_status"` // 处理状态：0=待处理，1=处理成功，2=处理失败 + ProcessTime sql.NullTime `db:"process_time"` // 处理时间 + ProcessRemark sql.NullString `db:"process_remark"` // 处理备注 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentOrderModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentOrderModel { + return &defaultAgentOrderModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_order`", + } +} + +func (m *defaultAgentOrderModel) Insert(ctx context.Context, session sqlx.Session, data *AgentOrder) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentOrderIdKey := fmt.Sprintf("%s%v", cacheQncAgentOrderIdPrefix, data.Id) + qncAgentOrderOrderIdKey := fmt.Sprintf("%s%v", cacheQncAgentOrderOrderIdPrefix, data.OrderId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentOrderRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.AgentId, data.OrderId, data.ProductId, data.OrderAmount, data.SetPrice, data.ActualBasePrice, data.PriceCost, data.AgentProfit, data.ProcessStatus, data.ProcessTime, data.ProcessRemark, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.AgentId, data.OrderId, data.ProductId, data.OrderAmount, data.SetPrice, data.ActualBasePrice, data.PriceCost, data.AgentProfit, data.ProcessStatus, data.ProcessTime, data.ProcessRemark, data.DeleteTime, data.DelState, data.Version) + }, qncAgentOrderIdKey, qncAgentOrderOrderIdKey) +} +func (m *defaultAgentOrderModel) insertUUID(data *AgentOrder) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentOrderModel) FindOne(ctx context.Context, id string) (*AgentOrder, error) { + qncAgentOrderIdKey := fmt.Sprintf("%s%v", cacheQncAgentOrderIdPrefix, id) + var resp AgentOrder + err := m.QueryRowCtx(ctx, &resp, qncAgentOrderIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentOrderRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentOrderModel) FindOneByOrderId(ctx context.Context, orderId string) (*AgentOrder, error) { + qncAgentOrderOrderIdKey := fmt.Sprintf("%s%v", cacheQncAgentOrderOrderIdPrefix, orderId) + var resp AgentOrder + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentOrderOrderIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `order_id` = ? and del_state = ? limit 1", agentOrderRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, orderId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentOrderModel) Update(ctx context.Context, session sqlx.Session, newData *AgentOrder) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAgentOrderIdKey := fmt.Sprintf("%s%v", cacheQncAgentOrderIdPrefix, data.Id) + qncAgentOrderOrderIdKey := fmt.Sprintf("%s%v", cacheQncAgentOrderOrderIdPrefix, data.OrderId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentOrderRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AgentId, newData.OrderId, newData.ProductId, newData.OrderAmount, newData.SetPrice, newData.ActualBasePrice, newData.PriceCost, newData.AgentProfit, newData.ProcessStatus, newData.ProcessTime, newData.ProcessRemark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.AgentId, newData.OrderId, newData.ProductId, newData.OrderAmount, newData.SetPrice, newData.ActualBasePrice, newData.PriceCost, newData.AgentProfit, newData.ProcessStatus, newData.ProcessTime, newData.ProcessRemark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, qncAgentOrderIdKey, qncAgentOrderOrderIdKey) +} + +func (m *defaultAgentOrderModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AgentOrder) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAgentOrderIdKey := fmt.Sprintf("%s%v", cacheQncAgentOrderIdPrefix, data.Id) + qncAgentOrderOrderIdKey := fmt.Sprintf("%s%v", cacheQncAgentOrderOrderIdPrefix, data.OrderId) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentOrderRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AgentId, newData.OrderId, newData.ProductId, newData.OrderAmount, newData.SetPrice, newData.ActualBasePrice, newData.PriceCost, newData.AgentProfit, newData.ProcessStatus, newData.ProcessTime, newData.ProcessRemark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.AgentId, newData.OrderId, newData.ProductId, newData.OrderAmount, newData.SetPrice, newData.ActualBasePrice, newData.PriceCost, newData.AgentProfit, newData.ProcessStatus, newData.ProcessTime, newData.ProcessRemark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, qncAgentOrderIdKey, qncAgentOrderOrderIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentOrderModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentOrder) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentOrderModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentOrderModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentOrderModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentOrderModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentOrder, error) { + + builder = builder.Columns(agentOrderRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentOrder + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentOrderModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentOrder, error) { + + builder = builder.Columns(agentOrderRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentOrder + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentOrderModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentOrder, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentOrderRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentOrder + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentOrderModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentOrder, error) { + + builder = builder.Columns(agentOrderRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentOrder + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentOrderModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentOrder, error) { + + builder = builder.Columns(agentOrderRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentOrder + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentOrderModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentOrderModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentOrderModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAgentOrderIdKey := fmt.Sprintf("%s%v", cacheQncAgentOrderIdPrefix, id) + qncAgentOrderOrderIdKey := fmt.Sprintf("%s%v", cacheQncAgentOrderOrderIdPrefix, data.OrderId) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentOrderIdKey, qncAgentOrderOrderIdKey) + return err +} +func (m *defaultAgentOrderModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentOrderIdPrefix, primary) +} +func (m *defaultAgentOrderModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentOrderRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentOrderModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentProductConfigModel.go b/app/main/model/agentProductConfigModel.go new file mode 100644 index 0000000..de56af8 --- /dev/null +++ b/app/main/model/agentProductConfigModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentProductConfigModel = (*customAgentProductConfigModel)(nil) + +type ( + // AgentProductConfigModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentProductConfigModel. + AgentProductConfigModel interface { + agentProductConfigModel + } + + customAgentProductConfigModel struct { + *defaultAgentProductConfigModel + } +) + +// NewAgentProductConfigModel returns a model for the database table. +func NewAgentProductConfigModel(conn sqlx.SqlConn, c cache.CacheConf) AgentProductConfigModel { + return &customAgentProductConfigModel{ + defaultAgentProductConfigModel: newAgentProductConfigModel(conn, c), + } +} diff --git a/app/main/model/agentProductConfigModel_gen.go b/app/main/model/agentProductConfigModel_gen.go new file mode 100644 index 0000000..c30eb8e --- /dev/null +++ b/app/main/model/agentProductConfigModel_gen.go @@ -0,0 +1,430 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentProductConfigFieldNames = builder.RawFieldNames(&AgentProductConfig{}) + agentProductConfigRows = strings.Join(agentProductConfigFieldNames, ",") + agentProductConfigRowsExpectAutoSet = strings.Join(stringx.Remove(agentProductConfigFieldNames, "`create_time`", "`update_time`"), ",") + agentProductConfigRowsWithPlaceHolder = strings.Join(stringx.Remove(agentProductConfigFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentProductConfigIdPrefix = "cache:qnc:agentProductConfig:id:" + cacheQncAgentProductConfigProductIdPrefix = "cache:qnc:agentProductConfig:productId:" +) + +type ( + agentProductConfigModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentProductConfig) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentProductConfig, error) + FindOneByProductId(ctx context.Context, productId string) (*AgentProductConfig, error) + Update(ctx context.Context, session sqlx.Session, data *AgentProductConfig) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentProductConfig) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentProductConfig) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentProductConfig, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentProductConfig, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentProductConfig, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentProductConfig, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentProductConfig, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentProductConfigModel struct { + sqlc.CachedConn + table string + } + + AgentProductConfig struct { + Id string `db:"id"` + ProductId string `db:"product_id"` + BasePrice float64 `db:"base_price"` // 基础底价（BasePrice） + SystemMaxPrice float64 `db:"system_max_price"` // 系统价格上限（SystemMaxPrice） + PriceThreshold sql.NullFloat64 `db:"price_threshold"` // 提价标准阈值（PriceThreshold） + PriceFeeRate sql.NullFloat64 `db:"price_fee_rate"` // 提价手续费比例（PriceFeeRate） + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentProductConfigModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentProductConfigModel { + return &defaultAgentProductConfigModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_product_config`", + } +} + +func (m *defaultAgentProductConfigModel) Insert(ctx context.Context, session sqlx.Session, data *AgentProductConfig) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentProductConfigIdKey := fmt.Sprintf("%s%v", cacheQncAgentProductConfigIdPrefix, data.Id) + qncAgentProductConfigProductIdKey := fmt.Sprintf("%s%v", cacheQncAgentProductConfigProductIdPrefix, data.ProductId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentProductConfigRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.ProductId, data.BasePrice, data.SystemMaxPrice, data.PriceThreshold, data.PriceFeeRate, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.ProductId, data.BasePrice, data.SystemMaxPrice, data.PriceThreshold, data.PriceFeeRate, data.DeleteTime, data.DelState, data.Version) + }, qncAgentProductConfigIdKey, qncAgentProductConfigProductIdKey) +} +func (m *defaultAgentProductConfigModel) insertUUID(data *AgentProductConfig) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentProductConfigModel) FindOne(ctx context.Context, id string) (*AgentProductConfig, error) { + qncAgentProductConfigIdKey := fmt.Sprintf("%s%v", cacheQncAgentProductConfigIdPrefix, id) + var resp AgentProductConfig + err := m.QueryRowCtx(ctx, &resp, qncAgentProductConfigIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentProductConfigRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentProductConfigModel) FindOneByProductId(ctx context.Context, productId string) (*AgentProductConfig, error) { + qncAgentProductConfigProductIdKey := fmt.Sprintf("%s%v", cacheQncAgentProductConfigProductIdPrefix, productId) + var resp AgentProductConfig + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentProductConfigProductIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `product_id` = ? and del_state = ? limit 1", agentProductConfigRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, productId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentProductConfigModel) Update(ctx context.Context, session sqlx.Session, newData *AgentProductConfig) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAgentProductConfigIdKey := fmt.Sprintf("%s%v", cacheQncAgentProductConfigIdPrefix, data.Id) + qncAgentProductConfigProductIdKey := fmt.Sprintf("%s%v", cacheQncAgentProductConfigProductIdPrefix, data.ProductId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentProductConfigRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.ProductId, newData.BasePrice, newData.SystemMaxPrice, newData.PriceThreshold, newData.PriceFeeRate, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.ProductId, newData.BasePrice, newData.SystemMaxPrice, newData.PriceThreshold, newData.PriceFeeRate, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, qncAgentProductConfigIdKey, qncAgentProductConfigProductIdKey) +} + +func (m *defaultAgentProductConfigModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AgentProductConfig) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAgentProductConfigIdKey := fmt.Sprintf("%s%v", cacheQncAgentProductConfigIdPrefix, data.Id) + qncAgentProductConfigProductIdKey := fmt.Sprintf("%s%v", cacheQncAgentProductConfigProductIdPrefix, data.ProductId) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentProductConfigRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.ProductId, newData.BasePrice, newData.SystemMaxPrice, newData.PriceThreshold, newData.PriceFeeRate, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.ProductId, newData.BasePrice, newData.SystemMaxPrice, newData.PriceThreshold, newData.PriceFeeRate, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, qncAgentProductConfigIdKey, qncAgentProductConfigProductIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentProductConfigModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentProductConfig) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentProductConfigModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentProductConfigModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentProductConfigModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentProductConfigModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentProductConfig, error) { + + builder = builder.Columns(agentProductConfigRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentProductConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentProductConfigModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentProductConfig, error) { + + builder = builder.Columns(agentProductConfigRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentProductConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentProductConfigModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentProductConfig, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentProductConfigRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentProductConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentProductConfigModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentProductConfig, error) { + + builder = builder.Columns(agentProductConfigRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentProductConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentProductConfigModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentProductConfig, error) { + + builder = builder.Columns(agentProductConfigRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentProductConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentProductConfigModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentProductConfigModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentProductConfigModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAgentProductConfigIdKey := fmt.Sprintf("%s%v", cacheQncAgentProductConfigIdPrefix, id) + qncAgentProductConfigProductIdKey := fmt.Sprintf("%s%v", cacheQncAgentProductConfigProductIdPrefix, data.ProductId) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentProductConfigIdKey, qncAgentProductConfigProductIdKey) + return err +} +func (m *defaultAgentProductConfigModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentProductConfigIdPrefix, primary) +} +func (m *defaultAgentProductConfigModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentProductConfigRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentProductConfigModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentRealNameModel.go b/app/main/model/agentRealNameModel.go new file mode 100644 index 0000000..50bca1f --- /dev/null +++ b/app/main/model/agentRealNameModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentRealNameModel = (*customAgentRealNameModel)(nil) + +type ( + // AgentRealNameModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentRealNameModel. + AgentRealNameModel interface { + agentRealNameModel + } + + customAgentRealNameModel struct { + *defaultAgentRealNameModel + } +) + +// NewAgentRealNameModel returns a model for the database table. +func NewAgentRealNameModel(conn sqlx.SqlConn, c cache.CacheConf) AgentRealNameModel { + return &customAgentRealNameModel{ + defaultAgentRealNameModel: newAgentRealNameModel(conn, c), + } +} diff --git a/app/main/model/agentRealNameModel_gen.go b/app/main/model/agentRealNameModel_gen.go new file mode 100644 index 0000000..d1adc10 --- /dev/null +++ b/app/main/model/agentRealNameModel_gen.go @@ -0,0 +1,430 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentRealNameFieldNames = builder.RawFieldNames(&AgentRealName{}) + agentRealNameRows = strings.Join(agentRealNameFieldNames, ",") + agentRealNameRowsExpectAutoSet = strings.Join(stringx.Remove(agentRealNameFieldNames, "`create_time`", "`update_time`"), ",") + agentRealNameRowsWithPlaceHolder = strings.Join(stringx.Remove(agentRealNameFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentRealNameIdPrefix = "cache:qnc:agentRealName:id:" + cacheQncAgentRealNameAgentIdPrefix = "cache:qnc:agentRealName:agentId:" +) + +type ( + agentRealNameModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentRealName) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentRealName, error) + FindOneByAgentId(ctx context.Context, agentId string) (*AgentRealName, error) + Update(ctx context.Context, session sqlx.Session, data *AgentRealName) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentRealName) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentRealName) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentRealName, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRealName, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRealName, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentRealName, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentRealName, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentRealNameModel struct { + sqlc.CachedConn + table string + } + + AgentRealName struct { + Id string `db:"id"` + AgentId string `db:"agent_id"` + Name string `db:"name"` // 真实姓名 + IdCard string `db:"id_card"` // 身份证号（加密） + Mobile string `db:"mobile"` // 手机号（加密） + VerifyTime sql.NullTime `db:"verify_time"` // 验证时间（三要素验证通过时间，NULL表示未验证） + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentRealNameModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentRealNameModel { + return &defaultAgentRealNameModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_real_name`", + } +} + +func (m *defaultAgentRealNameModel) Insert(ctx context.Context, session sqlx.Session, data *AgentRealName) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentRealNameAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentRealNameAgentIdPrefix, data.AgentId) + qncAgentRealNameIdKey := fmt.Sprintf("%s%v", cacheQncAgentRealNameIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentRealNameRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.AgentId, data.Name, data.IdCard, data.Mobile, data.VerifyTime, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.AgentId, data.Name, data.IdCard, data.Mobile, data.VerifyTime, data.DeleteTime, data.DelState, data.Version) + }, qncAgentRealNameAgentIdKey, qncAgentRealNameIdKey) +} +func (m *defaultAgentRealNameModel) insertUUID(data *AgentRealName) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentRealNameModel) FindOne(ctx context.Context, id string) (*AgentRealName, error) { + qncAgentRealNameIdKey := fmt.Sprintf("%s%v", cacheQncAgentRealNameIdPrefix, id) + var resp AgentRealName + err := m.QueryRowCtx(ctx, &resp, qncAgentRealNameIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentRealNameRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentRealNameModel) FindOneByAgentId(ctx context.Context, agentId string) (*AgentRealName, error) { + qncAgentRealNameAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentRealNameAgentIdPrefix, agentId) + var resp AgentRealName + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentRealNameAgentIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `agent_id` = ? and del_state = ? limit 1", agentRealNameRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, agentId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentRealNameModel) Update(ctx context.Context, session sqlx.Session, newData *AgentRealName) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAgentRealNameAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentRealNameAgentIdPrefix, data.AgentId) + qncAgentRealNameIdKey := fmt.Sprintf("%s%v", cacheQncAgentRealNameIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentRealNameRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AgentId, newData.Name, newData.IdCard, newData.Mobile, newData.VerifyTime, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.AgentId, newData.Name, newData.IdCard, newData.Mobile, newData.VerifyTime, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, qncAgentRealNameAgentIdKey, qncAgentRealNameIdKey) +} + +func (m *defaultAgentRealNameModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AgentRealName) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAgentRealNameAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentRealNameAgentIdPrefix, data.AgentId) + qncAgentRealNameIdKey := fmt.Sprintf("%s%v", cacheQncAgentRealNameIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentRealNameRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AgentId, newData.Name, newData.IdCard, newData.Mobile, newData.VerifyTime, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.AgentId, newData.Name, newData.IdCard, newData.Mobile, newData.VerifyTime, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, qncAgentRealNameAgentIdKey, qncAgentRealNameIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentRealNameModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentRealName) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentRealNameModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentRealNameModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentRealNameModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentRealNameModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentRealName, error) { + + builder = builder.Columns(agentRealNameRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRealName + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRealNameModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRealName, error) { + + builder = builder.Columns(agentRealNameRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRealName + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRealNameModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRealName, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentRealNameRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentRealName + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentRealNameModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentRealName, error) { + + builder = builder.Columns(agentRealNameRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRealName + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRealNameModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentRealName, error) { + + builder = builder.Columns(agentRealNameRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRealName + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRealNameModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentRealNameModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentRealNameModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAgentRealNameAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentRealNameAgentIdPrefix, data.AgentId) + qncAgentRealNameIdKey := fmt.Sprintf("%s%v", cacheQncAgentRealNameIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentRealNameAgentIdKey, qncAgentRealNameIdKey) + return err +} +func (m *defaultAgentRealNameModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentRealNameIdPrefix, primary) +} +func (m *defaultAgentRealNameModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentRealNameRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentRealNameModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentRebateModel.go b/app/main/model/agentRebateModel.go new file mode 100644 index 0000000..18d3eac --- /dev/null +++ b/app/main/model/agentRebateModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentRebateModel = (*customAgentRebateModel)(nil) + +type ( + // AgentRebateModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentRebateModel. + AgentRebateModel interface { + agentRebateModel + } + + customAgentRebateModel struct { + *defaultAgentRebateModel + } +) + +// NewAgentRebateModel returns a model for the database table. +func NewAgentRebateModel(conn sqlx.SqlConn, c cache.CacheConf) AgentRebateModel { + return &customAgentRebateModel{ + defaultAgentRebateModel: newAgentRebateModel(conn, c), + } +} diff --git a/app/main/model/agentRebateModel_gen.go b/app/main/model/agentRebateModel_gen.go new file mode 100644 index 0000000..95cb568 --- /dev/null +++ b/app/main/model/agentRebateModel_gen.go @@ -0,0 +1,394 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentRebateFieldNames = builder.RawFieldNames(&AgentRebate{}) + agentRebateRows = strings.Join(agentRebateFieldNames, ",") + agentRebateRowsExpectAutoSet = strings.Join(stringx.Remove(agentRebateFieldNames, "`create_time`", "`update_time`"), ",") + agentRebateRowsWithPlaceHolder = strings.Join(stringx.Remove(agentRebateFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentRebateIdPrefix = "cache:qnc:agentRebate:id:" +) + +type ( + agentRebateModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentRebate) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentRebate, error) + Update(ctx context.Context, session sqlx.Session, data *AgentRebate) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentRebate) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentRebate) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentRebate, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRebate, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRebate, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentRebate, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentRebate, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentRebateModel struct { + sqlc.CachedConn + table string + } + + AgentRebate struct { + Id string `db:"id"` + AgentId string `db:"agent_id"` + SourceAgentId string `db:"source_agent_id"` + OrderId string `db:"order_id"` + ProductId string `db:"product_id"` + RebateType int64 `db:"rebate_type"` // 返佣类型：1=直接上级返佣，2=钻石上级返佣，3=黄金上级返佣 + LevelBonus float64 `db:"level_bonus"` // 等级加成金额（来源代理的等级加成） + RebateAmount float64 `db:"rebate_amount"` // 返佣金额 + Status int64 `db:"status"` // 状态：1=已发放，2=已冻结，3=已取消 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentRebateModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentRebateModel { + return &defaultAgentRebateModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_rebate`", + } +} + +func (m *defaultAgentRebateModel) Insert(ctx context.Context, session sqlx.Session, data *AgentRebate) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentRebateIdKey := fmt.Sprintf("%s%v", cacheQncAgentRebateIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentRebateRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.AgentId, data.SourceAgentId, data.OrderId, data.ProductId, data.RebateType, data.LevelBonus, data.RebateAmount, data.Status, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.AgentId, data.SourceAgentId, data.OrderId, data.ProductId, data.RebateType, data.LevelBonus, data.RebateAmount, data.Status, data.DeleteTime, data.DelState, data.Version) + }, qncAgentRebateIdKey) +} +func (m *defaultAgentRebateModel) insertUUID(data *AgentRebate) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentRebateModel) FindOne(ctx context.Context, id string) (*AgentRebate, error) { + qncAgentRebateIdKey := fmt.Sprintf("%s%v", cacheQncAgentRebateIdPrefix, id) + var resp AgentRebate + err := m.QueryRowCtx(ctx, &resp, qncAgentRebateIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentRebateRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentRebateModel) Update(ctx context.Context, session sqlx.Session, data *AgentRebate) (sql.Result, error) { + qncAgentRebateIdKey := fmt.Sprintf("%s%v", cacheQncAgentRebateIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentRebateRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.AgentId, data.SourceAgentId, data.OrderId, data.ProductId, data.RebateType, data.LevelBonus, data.RebateAmount, data.Status, data.DeleteTime, data.DelState, data.Version, data.Id) + } + return conn.ExecCtx(ctx, query, data.AgentId, data.SourceAgentId, data.OrderId, data.ProductId, data.RebateType, data.LevelBonus, data.RebateAmount, data.Status, data.DeleteTime, data.DelState, data.Version, data.Id) + }, qncAgentRebateIdKey) +} + +func (m *defaultAgentRebateModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentRebate) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncAgentRebateIdKey := fmt.Sprintf("%s%v", cacheQncAgentRebateIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentRebateRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.AgentId, data.SourceAgentId, data.OrderId, data.ProductId, data.RebateType, data.LevelBonus, data.RebateAmount, data.Status, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.AgentId, data.SourceAgentId, data.OrderId, data.ProductId, data.RebateType, data.LevelBonus, data.RebateAmount, data.Status, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + }, qncAgentRebateIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentRebateModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentRebate) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentRebateModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentRebateModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentRebateModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentRebateModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentRebate, error) { + + builder = builder.Columns(agentRebateRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRebate + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRebateModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRebate, error) { + + builder = builder.Columns(agentRebateRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRebate + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRebateModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRebate, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentRebateRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentRebate + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentRebateModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentRebate, error) { + + builder = builder.Columns(agentRebateRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRebate + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRebateModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentRebate, error) { + + builder = builder.Columns(agentRebateRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRebate + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRebateModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentRebateModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentRebateModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncAgentRebateIdKey := fmt.Sprintf("%s%v", cacheQncAgentRebateIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentRebateIdKey) + return err +} +func (m *defaultAgentRebateModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentRebateIdPrefix, primary) +} +func (m *defaultAgentRebateModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentRebateRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentRebateModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentRelationModel.go b/app/main/model/agentRelationModel.go new file mode 100644 index 0000000..a1c3c5d --- /dev/null +++ b/app/main/model/agentRelationModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentRelationModel = (*customAgentRelationModel)(nil) + +type ( + // AgentRelationModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentRelationModel. + AgentRelationModel interface { + agentRelationModel + } + + customAgentRelationModel struct { + *defaultAgentRelationModel + } +) + +// NewAgentRelationModel returns a model for the database table. +func NewAgentRelationModel(conn sqlx.SqlConn, c cache.CacheConf) AgentRelationModel { + return &customAgentRelationModel{ + defaultAgentRelationModel: newAgentRelationModel(conn, c), + } +} diff --git a/app/main/model/agentRelationModel_gen.go b/app/main/model/agentRelationModel_gen.go new file mode 100644 index 0000000..8d69a9f --- /dev/null +++ b/app/main/model/agentRelationModel_gen.go @@ -0,0 +1,430 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentRelationFieldNames = builder.RawFieldNames(&AgentRelation{}) + agentRelationRows = strings.Join(agentRelationFieldNames, ",") + agentRelationRowsExpectAutoSet = strings.Join(stringx.Remove(agentRelationFieldNames, "`create_time`", "`update_time`"), ",") + agentRelationRowsWithPlaceHolder = strings.Join(stringx.Remove(agentRelationFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentRelationIdPrefix = "cache:qnc:agentRelation:id:" + cacheQncAgentRelationParentIdChildIdRelationTypePrefix = "cache:qnc:agentRelation:parentId:childId:relationType:" +) + +type ( + agentRelationModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentRelation) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentRelation, error) + FindOneByParentIdChildIdRelationType(ctx context.Context, parentId string, childId string, relationType int64) (*AgentRelation, error) + Update(ctx context.Context, session sqlx.Session, data *AgentRelation) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentRelation) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentRelation) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentRelation, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRelation, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRelation, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentRelation, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentRelation, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentRelationModel struct { + sqlc.CachedConn + table string + } + + AgentRelation struct { + Id string `db:"id"` + ParentId string `db:"parent_id"` + ChildId string `db:"child_id"` + RelationType int64 `db:"relation_type"` // 关系类型：1=直接关系，2=已脱离（历史记录） + DetachReason sql.NullString `db:"detach_reason"` // 脱离原因：upgrade=升级脱离 + DetachTime sql.NullTime `db:"detach_time"` // 脱离时间 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentRelationModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentRelationModel { + return &defaultAgentRelationModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_relation`", + } +} + +func (m *defaultAgentRelationModel) Insert(ctx context.Context, session sqlx.Session, data *AgentRelation) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentRelationIdKey := fmt.Sprintf("%s%v", cacheQncAgentRelationIdPrefix, data.Id) + qncAgentRelationParentIdChildIdRelationTypeKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentRelationParentIdChildIdRelationTypePrefix, data.ParentId, data.ChildId, data.RelationType) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentRelationRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.ParentId, data.ChildId, data.RelationType, data.DetachReason, data.DetachTime, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.ParentId, data.ChildId, data.RelationType, data.DetachReason, data.DetachTime, data.DeleteTime, data.DelState, data.Version) + }, qncAgentRelationIdKey, qncAgentRelationParentIdChildIdRelationTypeKey) +} +func (m *defaultAgentRelationModel) insertUUID(data *AgentRelation) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentRelationModel) FindOne(ctx context.Context, id string) (*AgentRelation, error) { + qncAgentRelationIdKey := fmt.Sprintf("%s%v", cacheQncAgentRelationIdPrefix, id) + var resp AgentRelation + err := m.QueryRowCtx(ctx, &resp, qncAgentRelationIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentRelationRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentRelationModel) FindOneByParentIdChildIdRelationType(ctx context.Context, parentId string, childId string, relationType int64) (*AgentRelation, error) { + qncAgentRelationParentIdChildIdRelationTypeKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentRelationParentIdChildIdRelationTypePrefix, parentId, childId, relationType) + var resp AgentRelation + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentRelationParentIdChildIdRelationTypeKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `parent_id` = ? and `child_id` = ? and `relation_type` = ? and del_state = ? limit 1", agentRelationRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, parentId, childId, relationType, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentRelationModel) Update(ctx context.Context, session sqlx.Session, newData *AgentRelation) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAgentRelationIdKey := fmt.Sprintf("%s%v", cacheQncAgentRelationIdPrefix, data.Id) + qncAgentRelationParentIdChildIdRelationTypeKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentRelationParentIdChildIdRelationTypePrefix, data.ParentId, data.ChildId, data.RelationType) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentRelationRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.ParentId, newData.ChildId, newData.RelationType, newData.DetachReason, newData.DetachTime, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.ParentId, newData.ChildId, newData.RelationType, newData.DetachReason, newData.DetachTime, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, qncAgentRelationIdKey, qncAgentRelationParentIdChildIdRelationTypeKey) +} + +func (m *defaultAgentRelationModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AgentRelation) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAgentRelationIdKey := fmt.Sprintf("%s%v", cacheQncAgentRelationIdPrefix, data.Id) + qncAgentRelationParentIdChildIdRelationTypeKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentRelationParentIdChildIdRelationTypePrefix, data.ParentId, data.ChildId, data.RelationType) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentRelationRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.ParentId, newData.ChildId, newData.RelationType, newData.DetachReason, newData.DetachTime, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.ParentId, newData.ChildId, newData.RelationType, newData.DetachReason, newData.DetachTime, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, qncAgentRelationIdKey, qncAgentRelationParentIdChildIdRelationTypeKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentRelationModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentRelation) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentRelationModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentRelationModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentRelationModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentRelationModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentRelation, error) { + + builder = builder.Columns(agentRelationRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRelation + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRelationModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRelation, error) { + + builder = builder.Columns(agentRelationRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRelation + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRelationModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentRelation, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentRelationRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentRelation + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentRelationModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentRelation, error) { + + builder = builder.Columns(agentRelationRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRelation + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRelationModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentRelation, error) { + + builder = builder.Columns(agentRelationRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentRelation + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentRelationModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentRelationModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentRelationModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAgentRelationIdKey := fmt.Sprintf("%s%v", cacheQncAgentRelationIdPrefix, id) + qncAgentRelationParentIdChildIdRelationTypeKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentRelationParentIdChildIdRelationTypePrefix, data.ParentId, data.ChildId, data.RelationType) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentRelationIdKey, qncAgentRelationParentIdChildIdRelationTypeKey) + return err +} +func (m *defaultAgentRelationModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentRelationIdPrefix, primary) +} +func (m *defaultAgentRelationModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentRelationRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentRelationModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentShortLinkModel.go b/app/main/model/agentShortLinkModel.go new file mode 100644 index 0000000..26dab0b --- /dev/null +++ b/app/main/model/agentShortLinkModel.go @@ -0,0 +1,46 @@ +package model + +import ( + "context" + "database/sql" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentShortLinkModel = (*customAgentShortLinkModel)(nil) + +type ( + // AgentShortLinkModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentShortLinkModel. + AgentShortLinkModel interface { + agentShortLinkModel + FindOneByInviteCodeTypeDelState(ctx context.Context, inviteCode sql.NullString, tp int64, delState int64) (*AgentShortLink, error) + } + + customAgentShortLinkModel struct { + *defaultAgentShortLinkModel + } +) + +// NewAgentShortLinkModel returns a model for the database table. +func NewAgentShortLinkModel(conn sqlx.SqlConn, c cache.CacheConf) AgentShortLinkModel { + return &customAgentShortLinkModel{ + defaultAgentShortLinkModel: newAgentShortLinkModel(conn, c), + } +} + +func (m *customAgentShortLinkModel) FindOneByInviteCodeTypeDelState(ctx context.Context, inviteCode sql.NullString, tp int64, delState int64) (*AgentShortLink, error) { + builder := m.SelectBuilder(). + Where("invite_code = ? AND type = ? AND del_state = ?", inviteCode, tp, delState). + Limit(1) + rows, err := m.FindAll(ctx, builder, "") + if err != nil { + return nil, errors.Wrapf(err, "query short link by invite_code failed") + } + if len(rows) == 0 { + return nil, ErrNotFound + } + return rows[0], nil +} diff --git a/app/main/model/agentShortLinkModel_gen.go b/app/main/model/agentShortLinkModel_gen.go new file mode 100644 index 0000000..6825f2e --- /dev/null +++ b/app/main/model/agentShortLinkModel_gen.go @@ -0,0 +1,485 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentShortLinkFieldNames = builder.RawFieldNames(&AgentShortLink{}) + agentShortLinkRows = strings.Join(agentShortLinkFieldNames, ",") + agentShortLinkRowsExpectAutoSet = strings.Join(stringx.Remove(agentShortLinkFieldNames, "`create_time`", "`update_time`"), ",") + agentShortLinkRowsWithPlaceHolder = strings.Join(stringx.Remove(agentShortLinkFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentShortLinkIdPrefix = "cache:qnc:agentShortLink:id:" + cacheQncAgentShortLinkInviteCodeIdTypeDelStatePrefix = "cache:qnc:agentShortLink:inviteCodeId:type:delState:" + cacheQncAgentShortLinkLinkIdTypeDelStatePrefix = "cache:qnc:agentShortLink:linkId:type:delState:" + cacheQncAgentShortLinkShortCodeDelStatePrefix = "cache:qnc:agentShortLink:shortCode:delState:" +) + +type ( + agentShortLinkModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentShortLink) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentShortLink, error) + FindOneByInviteCodeIdTypeDelState(ctx context.Context, inviteCodeId sql.NullString, tp int64, delState int64) (*AgentShortLink, error) + FindOneByLinkIdTypeDelState(ctx context.Context, linkId sql.NullString, tp int64, delState int64) (*AgentShortLink, error) + FindOneByShortCodeDelState(ctx context.Context, shortCode string, delState int64) (*AgentShortLink, error) + Update(ctx context.Context, session sqlx.Session, data *AgentShortLink) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentShortLink) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentShortLink) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentShortLink, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentShortLink, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentShortLink, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentShortLink, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentShortLink, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentShortLinkModel struct { + sqlc.CachedConn + table string + } + + AgentShortLink struct { + Id string `db:"id"` + Type int64 `db:"type"` // 类型：1=推广报告(promotion)，2=邀请好友(invite) + LinkId sql.NullString `db:"link_id"` + InviteCodeId sql.NullString `db:"invite_code_id"` + LinkIdentifier sql.NullString `db:"link_identifier"` // 推广链接标识（加密，仅推广报告类型使用） + InviteCode sql.NullString `db:"invite_code"` // 邀请码（仅邀请好友类型使用） + ShortCode string `db:"short_code"` // 短链标识（6位随机字符串） + TargetPath string `db:"target_path"` // 目标地址（前端传入，如：/agent/promotionInquire/xxx 或 /register?invite_code=xxx） + PromotionDomain string `db:"promotion_domain"` // 推广域名（生成短链时使用的域名） + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentShortLinkModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentShortLinkModel { + return &defaultAgentShortLinkModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_short_link`", + } +} + +func (m *defaultAgentShortLinkModel) Insert(ctx context.Context, session sqlx.Session, data *AgentShortLink) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentShortLinkIdKey := fmt.Sprintf("%s%v", cacheQncAgentShortLinkIdPrefix, data.Id) + qncAgentShortLinkInviteCodeIdTypeDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentShortLinkInviteCodeIdTypeDelStatePrefix, data.InviteCodeId, data.Type, data.DelState) + qncAgentShortLinkLinkIdTypeDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentShortLinkLinkIdTypeDelStatePrefix, data.LinkId, data.Type, data.DelState) + qncAgentShortLinkShortCodeDelStateKey := fmt.Sprintf("%s%v:%v", cacheQncAgentShortLinkShortCodeDelStatePrefix, data.ShortCode, data.DelState) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentShortLinkRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.Type, data.LinkId, data.InviteCodeId, data.LinkIdentifier, data.InviteCode, data.ShortCode, data.TargetPath, data.PromotionDomain, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.Type, data.LinkId, data.InviteCodeId, data.LinkIdentifier, data.InviteCode, data.ShortCode, data.TargetPath, data.PromotionDomain, data.DeleteTime, data.DelState, data.Version) + }, qncAgentShortLinkIdKey, qncAgentShortLinkInviteCodeIdTypeDelStateKey, qncAgentShortLinkLinkIdTypeDelStateKey, qncAgentShortLinkShortCodeDelStateKey) +} +func (m *defaultAgentShortLinkModel) insertUUID(data *AgentShortLink) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentShortLinkModel) FindOne(ctx context.Context, id string) (*AgentShortLink, error) { + qncAgentShortLinkIdKey := fmt.Sprintf("%s%v", cacheQncAgentShortLinkIdPrefix, id) + var resp AgentShortLink + err := m.QueryRowCtx(ctx, &resp, qncAgentShortLinkIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentShortLinkRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentShortLinkModel) FindOneByInviteCodeIdTypeDelState(ctx context.Context, inviteCodeId sql.NullString, tp int64, delState int64) (*AgentShortLink, error) { + qncAgentShortLinkInviteCodeIdTypeDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentShortLinkInviteCodeIdTypeDelStatePrefix, inviteCodeId, tp, delState) + var resp AgentShortLink + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentShortLinkInviteCodeIdTypeDelStateKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `invite_code_id` = ? and `type` = ? and `del_state` = ? and del_state = ? limit 1", agentShortLinkRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, inviteCodeId, tp, delState, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentShortLinkModel) FindOneByLinkIdTypeDelState(ctx context.Context, linkId sql.NullString, tp int64, delState int64) (*AgentShortLink, error) { + qncAgentShortLinkLinkIdTypeDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentShortLinkLinkIdTypeDelStatePrefix, linkId, tp, delState) + var resp AgentShortLink + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentShortLinkLinkIdTypeDelStateKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `link_id` = ? and `type` = ? and `del_state` = ? and del_state = ? limit 1", agentShortLinkRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, linkId, tp, delState, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentShortLinkModel) FindOneByShortCodeDelState(ctx context.Context, shortCode string, delState int64) (*AgentShortLink, error) { + qncAgentShortLinkShortCodeDelStateKey := fmt.Sprintf("%s%v:%v", cacheQncAgentShortLinkShortCodeDelStatePrefix, shortCode, delState) + var resp AgentShortLink + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentShortLinkShortCodeDelStateKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `short_code` = ? and `del_state` = ? and del_state = ? limit 1", agentShortLinkRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, shortCode, delState, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentShortLinkModel) Update(ctx context.Context, session sqlx.Session, newData *AgentShortLink) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAgentShortLinkIdKey := fmt.Sprintf("%s%v", cacheQncAgentShortLinkIdPrefix, data.Id) + qncAgentShortLinkInviteCodeIdTypeDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentShortLinkInviteCodeIdTypeDelStatePrefix, data.InviteCodeId, data.Type, data.DelState) + qncAgentShortLinkLinkIdTypeDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentShortLinkLinkIdTypeDelStatePrefix, data.LinkId, data.Type, data.DelState) + qncAgentShortLinkShortCodeDelStateKey := fmt.Sprintf("%s%v:%v", cacheQncAgentShortLinkShortCodeDelStatePrefix, data.ShortCode, data.DelState) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentShortLinkRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.Type, newData.LinkId, newData.InviteCodeId, newData.LinkIdentifier, newData.InviteCode, newData.ShortCode, newData.TargetPath, newData.PromotionDomain, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.Type, newData.LinkId, newData.InviteCodeId, newData.LinkIdentifier, newData.InviteCode, newData.ShortCode, newData.TargetPath, newData.PromotionDomain, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, qncAgentShortLinkIdKey, qncAgentShortLinkInviteCodeIdTypeDelStateKey, qncAgentShortLinkLinkIdTypeDelStateKey, qncAgentShortLinkShortCodeDelStateKey) +} + +func (m *defaultAgentShortLinkModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AgentShortLink) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAgentShortLinkIdKey := fmt.Sprintf("%s%v", cacheQncAgentShortLinkIdPrefix, data.Id) + qncAgentShortLinkInviteCodeIdTypeDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentShortLinkInviteCodeIdTypeDelStatePrefix, data.InviteCodeId, data.Type, data.DelState) + qncAgentShortLinkLinkIdTypeDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentShortLinkLinkIdTypeDelStatePrefix, data.LinkId, data.Type, data.DelState) + qncAgentShortLinkShortCodeDelStateKey := fmt.Sprintf("%s%v:%v", cacheQncAgentShortLinkShortCodeDelStatePrefix, data.ShortCode, data.DelState) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentShortLinkRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.Type, newData.LinkId, newData.InviteCodeId, newData.LinkIdentifier, newData.InviteCode, newData.ShortCode, newData.TargetPath, newData.PromotionDomain, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.Type, newData.LinkId, newData.InviteCodeId, newData.LinkIdentifier, newData.InviteCode, newData.ShortCode, newData.TargetPath, newData.PromotionDomain, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, qncAgentShortLinkIdKey, qncAgentShortLinkInviteCodeIdTypeDelStateKey, qncAgentShortLinkLinkIdTypeDelStateKey, qncAgentShortLinkShortCodeDelStateKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentShortLinkModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentShortLink) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentShortLinkModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentShortLinkModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentShortLinkModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentShortLinkModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentShortLink, error) { + + builder = builder.Columns(agentShortLinkRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentShortLink + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentShortLinkModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentShortLink, error) { + + builder = builder.Columns(agentShortLinkRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentShortLink + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentShortLinkModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentShortLink, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentShortLinkRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentShortLink + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentShortLinkModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentShortLink, error) { + + builder = builder.Columns(agentShortLinkRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentShortLink + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentShortLinkModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentShortLink, error) { + + builder = builder.Columns(agentShortLinkRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentShortLink + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentShortLinkModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentShortLinkModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentShortLinkModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAgentShortLinkIdKey := fmt.Sprintf("%s%v", cacheQncAgentShortLinkIdPrefix, id) + qncAgentShortLinkInviteCodeIdTypeDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentShortLinkInviteCodeIdTypeDelStatePrefix, data.InviteCodeId, data.Type, data.DelState) + qncAgentShortLinkLinkIdTypeDelStateKey := fmt.Sprintf("%s%v:%v:%v", cacheQncAgentShortLinkLinkIdTypeDelStatePrefix, data.LinkId, data.Type, data.DelState) + qncAgentShortLinkShortCodeDelStateKey := fmt.Sprintf("%s%v:%v", cacheQncAgentShortLinkShortCodeDelStatePrefix, data.ShortCode, data.DelState) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentShortLinkIdKey, qncAgentShortLinkInviteCodeIdTypeDelStateKey, qncAgentShortLinkLinkIdTypeDelStateKey, qncAgentShortLinkShortCodeDelStateKey) + return err +} +func (m *defaultAgentShortLinkModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentShortLinkIdPrefix, primary) +} +func (m *defaultAgentShortLinkModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentShortLinkRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentShortLinkModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentUpgradeModel.go b/app/main/model/agentUpgradeModel.go new file mode 100644 index 0000000..db29dfb --- /dev/null +++ b/app/main/model/agentUpgradeModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentUpgradeModel = (*customAgentUpgradeModel)(nil) + +type ( + // AgentUpgradeModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentUpgradeModel. + AgentUpgradeModel interface { + agentUpgradeModel + } + + customAgentUpgradeModel struct { + *defaultAgentUpgradeModel + } +) + +// NewAgentUpgradeModel returns a model for the database table. +func NewAgentUpgradeModel(conn sqlx.SqlConn, c cache.CacheConf) AgentUpgradeModel { + return &customAgentUpgradeModel{ + defaultAgentUpgradeModel: newAgentUpgradeModel(conn, c), + } +} diff --git a/app/main/model/agentUpgradeModel_gen.go b/app/main/model/agentUpgradeModel_gen.go new file mode 100644 index 0000000..76e62a1 --- /dev/null +++ b/app/main/model/agentUpgradeModel_gen.go @@ -0,0 +1,397 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentUpgradeFieldNames = builder.RawFieldNames(&AgentUpgrade{}) + agentUpgradeRows = strings.Join(agentUpgradeFieldNames, ",") + agentUpgradeRowsExpectAutoSet = strings.Join(stringx.Remove(agentUpgradeFieldNames, "`create_time`", "`update_time`"), ",") + agentUpgradeRowsWithPlaceHolder = strings.Join(stringx.Remove(agentUpgradeFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentUpgradeIdPrefix = "cache:qnc:agentUpgrade:id:" +) + +type ( + agentUpgradeModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentUpgrade) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentUpgrade, error) + Update(ctx context.Context, session sqlx.Session, data *AgentUpgrade) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentUpgrade) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentUpgrade) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentUpgrade, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentUpgrade, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentUpgrade, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentUpgrade, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentUpgrade, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentUpgradeModel struct { + sqlc.CachedConn + table string + } + + AgentUpgrade struct { + Id string `db:"id"` + AgentId string `db:"agent_id"` + FromLevel int64 `db:"from_level"` // 原等级：1=普通，2=黄金，3=钻石 + ToLevel int64 `db:"to_level"` // 目标等级：1=普通，2=黄金，3=钻石 + UpgradeType int64 `db:"upgrade_type"` // 升级类型：1=自主付费，2=钻石升级下级 + UpgradeFee float64 `db:"upgrade_fee"` // 升级费用 + RebateAmount float64 `db:"rebate_amount"` // 返佣金额（给原直接上级） + RebateAgentId sql.NullString `db:"rebate_agent_id"` + OperatorAgentId sql.NullString `db:"operator_agent_id"` + OrderNo sql.NullString `db:"order_no"` // 支付订单号（如果是自主付费） + Status int64 `db:"status"` // 状态：1=待处理，2=已完成，3=已失败 + Remark sql.NullString `db:"remark"` // 备注 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentUpgradeModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentUpgradeModel { + return &defaultAgentUpgradeModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_upgrade`", + } +} + +func (m *defaultAgentUpgradeModel) Insert(ctx context.Context, session sqlx.Session, data *AgentUpgrade) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentUpgradeIdKey := fmt.Sprintf("%s%v", cacheQncAgentUpgradeIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentUpgradeRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.AgentId, data.FromLevel, data.ToLevel, data.UpgradeType, data.UpgradeFee, data.RebateAmount, data.RebateAgentId, data.OperatorAgentId, data.OrderNo, data.Status, data.Remark, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.AgentId, data.FromLevel, data.ToLevel, data.UpgradeType, data.UpgradeFee, data.RebateAmount, data.RebateAgentId, data.OperatorAgentId, data.OrderNo, data.Status, data.Remark, data.DeleteTime, data.DelState, data.Version) + }, qncAgentUpgradeIdKey) +} +func (m *defaultAgentUpgradeModel) insertUUID(data *AgentUpgrade) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentUpgradeModel) FindOne(ctx context.Context, id string) (*AgentUpgrade, error) { + qncAgentUpgradeIdKey := fmt.Sprintf("%s%v", cacheQncAgentUpgradeIdPrefix, id) + var resp AgentUpgrade + err := m.QueryRowCtx(ctx, &resp, qncAgentUpgradeIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentUpgradeRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentUpgradeModel) Update(ctx context.Context, session sqlx.Session, data *AgentUpgrade) (sql.Result, error) { + qncAgentUpgradeIdKey := fmt.Sprintf("%s%v", cacheQncAgentUpgradeIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentUpgradeRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.AgentId, data.FromLevel, data.ToLevel, data.UpgradeType, data.UpgradeFee, data.RebateAmount, data.RebateAgentId, data.OperatorAgentId, data.OrderNo, data.Status, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id) + } + return conn.ExecCtx(ctx, query, data.AgentId, data.FromLevel, data.ToLevel, data.UpgradeType, data.UpgradeFee, data.RebateAmount, data.RebateAgentId, data.OperatorAgentId, data.OrderNo, data.Status, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id) + }, qncAgentUpgradeIdKey) +} + +func (m *defaultAgentUpgradeModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentUpgrade) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncAgentUpgradeIdKey := fmt.Sprintf("%s%v", cacheQncAgentUpgradeIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentUpgradeRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.AgentId, data.FromLevel, data.ToLevel, data.UpgradeType, data.UpgradeFee, data.RebateAmount, data.RebateAgentId, data.OperatorAgentId, data.OrderNo, data.Status, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.AgentId, data.FromLevel, data.ToLevel, data.UpgradeType, data.UpgradeFee, data.RebateAmount, data.RebateAgentId, data.OperatorAgentId, data.OrderNo, data.Status, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + }, qncAgentUpgradeIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentUpgradeModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentUpgrade) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentUpgradeModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentUpgradeModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentUpgradeModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentUpgradeModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentUpgrade, error) { + + builder = builder.Columns(agentUpgradeRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentUpgrade + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentUpgradeModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentUpgrade, error) { + + builder = builder.Columns(agentUpgradeRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentUpgrade + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentUpgradeModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentUpgrade, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentUpgradeRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentUpgrade + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentUpgradeModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentUpgrade, error) { + + builder = builder.Columns(agentUpgradeRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentUpgrade + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentUpgradeModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentUpgrade, error) { + + builder = builder.Columns(agentUpgradeRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentUpgrade + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentUpgradeModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentUpgradeModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentUpgradeModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncAgentUpgradeIdKey := fmt.Sprintf("%s%v", cacheQncAgentUpgradeIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentUpgradeIdKey) + return err +} +func (m *defaultAgentUpgradeModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentUpgradeIdPrefix, primary) +} +func (m *defaultAgentUpgradeModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentUpgradeRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentUpgradeModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentWalletModel.go b/app/main/model/agentWalletModel.go new file mode 100644 index 0000000..1583c4d --- /dev/null +++ b/app/main/model/agentWalletModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentWalletModel = (*customAgentWalletModel)(nil) + +type ( + // AgentWalletModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentWalletModel. + AgentWalletModel interface { + agentWalletModel + } + + customAgentWalletModel struct { + *defaultAgentWalletModel + } +) + +// NewAgentWalletModel returns a model for the database table. +func NewAgentWalletModel(conn sqlx.SqlConn, c cache.CacheConf) AgentWalletModel { + return &customAgentWalletModel{ + defaultAgentWalletModel: newAgentWalletModel(conn, c), + } +} diff --git a/app/main/model/agentWalletModel_gen.go b/app/main/model/agentWalletModel_gen.go new file mode 100644 index 0000000..c0c5ebe --- /dev/null +++ b/app/main/model/agentWalletModel_gen.go @@ -0,0 +1,430 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentWalletFieldNames = builder.RawFieldNames(&AgentWallet{}) + agentWalletRows = strings.Join(agentWalletFieldNames, ",") + agentWalletRowsExpectAutoSet = strings.Join(stringx.Remove(agentWalletFieldNames, "`create_time`", "`update_time`"), ",") + agentWalletRowsWithPlaceHolder = strings.Join(stringx.Remove(agentWalletFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentWalletIdPrefix = "cache:qnc:agentWallet:id:" + cacheQncAgentWalletAgentIdPrefix = "cache:qnc:agentWallet:agentId:" +) + +type ( + agentWalletModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentWallet) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentWallet, error) + FindOneByAgentId(ctx context.Context, agentId string) (*AgentWallet, error) + Update(ctx context.Context, session sqlx.Session, data *AgentWallet) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentWallet) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentWallet) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentWallet, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWallet, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWallet, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentWallet, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentWallet, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentWalletModel struct { + sqlc.CachedConn + table string + } + + AgentWallet struct { + Id string `db:"id"` + AgentId string `db:"agent_id"` + Balance float64 `db:"balance"` // 可用余额 + FrozenBalance float64 `db:"frozen_balance"` // 冻结余额 + TotalEarnings float64 `db:"total_earnings"` // 累计收益 + WithdrawnAmount float64 `db:"withdrawn_amount"` // 累计提现金额 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentWalletModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentWalletModel { + return &defaultAgentWalletModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_wallet`", + } +} + +func (m *defaultAgentWalletModel) Insert(ctx context.Context, session sqlx.Session, data *AgentWallet) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentWalletAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentWalletAgentIdPrefix, data.AgentId) + qncAgentWalletIdKey := fmt.Sprintf("%s%v", cacheQncAgentWalletIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentWalletRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.AgentId, data.Balance, data.FrozenBalance, data.TotalEarnings, data.WithdrawnAmount, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.AgentId, data.Balance, data.FrozenBalance, data.TotalEarnings, data.WithdrawnAmount, data.DeleteTime, data.DelState, data.Version) + }, qncAgentWalletAgentIdKey, qncAgentWalletIdKey) +} +func (m *defaultAgentWalletModel) insertUUID(data *AgentWallet) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentWalletModel) FindOne(ctx context.Context, id string) (*AgentWallet, error) { + qncAgentWalletIdKey := fmt.Sprintf("%s%v", cacheQncAgentWalletIdPrefix, id) + var resp AgentWallet + err := m.QueryRowCtx(ctx, &resp, qncAgentWalletIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentWalletRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentWalletModel) FindOneByAgentId(ctx context.Context, agentId string) (*AgentWallet, error) { + qncAgentWalletAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentWalletAgentIdPrefix, agentId) + var resp AgentWallet + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentWalletAgentIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `agent_id` = ? and del_state = ? limit 1", agentWalletRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, agentId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentWalletModel) Update(ctx context.Context, session sqlx.Session, newData *AgentWallet) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAgentWalletAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentWalletAgentIdPrefix, data.AgentId) + qncAgentWalletIdKey := fmt.Sprintf("%s%v", cacheQncAgentWalletIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentWalletRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AgentId, newData.Balance, newData.FrozenBalance, newData.TotalEarnings, newData.WithdrawnAmount, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.AgentId, newData.Balance, newData.FrozenBalance, newData.TotalEarnings, newData.WithdrawnAmount, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, qncAgentWalletAgentIdKey, qncAgentWalletIdKey) +} + +func (m *defaultAgentWalletModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AgentWallet) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAgentWalletAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentWalletAgentIdPrefix, data.AgentId) + qncAgentWalletIdKey := fmt.Sprintf("%s%v", cacheQncAgentWalletIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentWalletRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AgentId, newData.Balance, newData.FrozenBalance, newData.TotalEarnings, newData.WithdrawnAmount, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.AgentId, newData.Balance, newData.FrozenBalance, newData.TotalEarnings, newData.WithdrawnAmount, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, qncAgentWalletAgentIdKey, qncAgentWalletIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentWalletModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentWallet) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentWalletModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentWalletModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentWalletModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentWalletModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentWallet, error) { + + builder = builder.Columns(agentWalletRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWallet + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWalletModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWallet, error) { + + builder = builder.Columns(agentWalletRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWallet + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWalletModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWallet, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentWalletRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentWallet + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentWalletModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentWallet, error) { + + builder = builder.Columns(agentWalletRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWallet + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWalletModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentWallet, error) { + + builder = builder.Columns(agentWalletRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWallet + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWalletModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentWalletModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentWalletModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAgentWalletAgentIdKey := fmt.Sprintf("%s%v", cacheQncAgentWalletAgentIdPrefix, data.AgentId) + qncAgentWalletIdKey := fmt.Sprintf("%s%v", cacheQncAgentWalletIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentWalletAgentIdKey, qncAgentWalletIdKey) + return err +} +func (m *defaultAgentWalletModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentWalletIdPrefix, primary) +} +func (m *defaultAgentWalletModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentWalletRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentWalletModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentWithdrawalModel.go b/app/main/model/agentWithdrawalModel.go new file mode 100644 index 0000000..56f5e50 --- /dev/null +++ b/app/main/model/agentWithdrawalModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentWithdrawalModel = (*customAgentWithdrawalModel)(nil) + +type ( + // AgentWithdrawalModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentWithdrawalModel. + AgentWithdrawalModel interface { + agentWithdrawalModel + } + + customAgentWithdrawalModel struct { + *defaultAgentWithdrawalModel + } +) + +// NewAgentWithdrawalModel returns a model for the database table. +func NewAgentWithdrawalModel(conn sqlx.SqlConn, c cache.CacheConf) AgentWithdrawalModel { + return &customAgentWithdrawalModel{ + defaultAgentWithdrawalModel: newAgentWithdrawalModel(conn, c), + } +} diff --git a/app/main/model/agentWithdrawalModel_gen.go b/app/main/model/agentWithdrawalModel_gen.go new file mode 100644 index 0000000..b7aaa16 --- /dev/null +++ b/app/main/model/agentWithdrawalModel_gen.go @@ -0,0 +1,434 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentWithdrawalFieldNames = builder.RawFieldNames(&AgentWithdrawal{}) + agentWithdrawalRows = strings.Join(agentWithdrawalFieldNames, ",") + agentWithdrawalRowsExpectAutoSet = strings.Join(stringx.Remove(agentWithdrawalFieldNames, "`create_time`", "`update_time`"), ",") + agentWithdrawalRowsWithPlaceHolder = strings.Join(stringx.Remove(agentWithdrawalFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentWithdrawalIdPrefix = "cache:qnc:agentWithdrawal:id:" + cacheQncAgentWithdrawalWithdrawNoPrefix = "cache:qnc:agentWithdrawal:withdrawNo:" +) + +type ( + agentWithdrawalModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentWithdrawal) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentWithdrawal, error) + FindOneByWithdrawNo(ctx context.Context, withdrawNo string) (*AgentWithdrawal, error) + Update(ctx context.Context, session sqlx.Session, data *AgentWithdrawal) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentWithdrawal) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentWithdrawal) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentWithdrawal, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWithdrawal, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWithdrawal, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentWithdrawal, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentWithdrawal, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentWithdrawalModel struct { + sqlc.CachedConn + table string + } + + AgentWithdrawal struct { + Id string `db:"id"` + AgentId string `db:"agent_id"` + WithdrawNo string `db:"withdraw_no"` // 提现单号 + PayeeAccount string `db:"payee_account"` // 收款账户 + PayeeName string `db:"payee_name"` // 收款人姓名 + Amount float64 `db:"amount"` // 提现金额 + ActualAmount float64 `db:"actual_amount"` // 实际到账金额（扣除税费后） + TaxAmount float64 `db:"tax_amount"` // 税费金额 + Status int64 `db:"status"` // 状态：1=处理中，2=成功，3=失败 + Remark sql.NullString `db:"remark"` // 备注 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentWithdrawalModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentWithdrawalModel { + return &defaultAgentWithdrawalModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_withdrawal`", + } +} + +func (m *defaultAgentWithdrawalModel) Insert(ctx context.Context, session sqlx.Session, data *AgentWithdrawal) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentWithdrawalIdKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalIdPrefix, data.Id) + qncAgentWithdrawalWithdrawNoKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalWithdrawNoPrefix, data.WithdrawNo) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentWithdrawalRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.AgentId, data.WithdrawNo, data.PayeeAccount, data.PayeeName, data.Amount, data.ActualAmount, data.TaxAmount, data.Status, data.Remark, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.AgentId, data.WithdrawNo, data.PayeeAccount, data.PayeeName, data.Amount, data.ActualAmount, data.TaxAmount, data.Status, data.Remark, data.DeleteTime, data.DelState, data.Version) + }, qncAgentWithdrawalIdKey, qncAgentWithdrawalWithdrawNoKey) +} +func (m *defaultAgentWithdrawalModel) insertUUID(data *AgentWithdrawal) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentWithdrawalModel) FindOne(ctx context.Context, id string) (*AgentWithdrawal, error) { + qncAgentWithdrawalIdKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalIdPrefix, id) + var resp AgentWithdrawal + err := m.QueryRowCtx(ctx, &resp, qncAgentWithdrawalIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentWithdrawalRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentWithdrawalModel) FindOneByWithdrawNo(ctx context.Context, withdrawNo string) (*AgentWithdrawal, error) { + qncAgentWithdrawalWithdrawNoKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalWithdrawNoPrefix, withdrawNo) + var resp AgentWithdrawal + err := m.QueryRowIndexCtx(ctx, &resp, qncAgentWithdrawalWithdrawNoKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `withdraw_no` = ? and del_state = ? limit 1", agentWithdrawalRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, withdrawNo, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentWithdrawalModel) Update(ctx context.Context, session sqlx.Session, newData *AgentWithdrawal) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncAgentWithdrawalIdKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalIdPrefix, data.Id) + qncAgentWithdrawalWithdrawNoKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalWithdrawNoPrefix, data.WithdrawNo) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentWithdrawalRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AgentId, newData.WithdrawNo, newData.PayeeAccount, newData.PayeeName, newData.Amount, newData.ActualAmount, newData.TaxAmount, newData.Status, newData.Remark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.AgentId, newData.WithdrawNo, newData.PayeeAccount, newData.PayeeName, newData.Amount, newData.ActualAmount, newData.TaxAmount, newData.Status, newData.Remark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, qncAgentWithdrawalIdKey, qncAgentWithdrawalWithdrawNoKey) +} + +func (m *defaultAgentWithdrawalModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *AgentWithdrawal) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncAgentWithdrawalIdKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalIdPrefix, data.Id) + qncAgentWithdrawalWithdrawNoKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalWithdrawNoPrefix, data.WithdrawNo) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentWithdrawalRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AgentId, newData.WithdrawNo, newData.PayeeAccount, newData.PayeeName, newData.Amount, newData.ActualAmount, newData.TaxAmount, newData.Status, newData.Remark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.AgentId, newData.WithdrawNo, newData.PayeeAccount, newData.PayeeName, newData.Amount, newData.ActualAmount, newData.TaxAmount, newData.Status, newData.Remark, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, qncAgentWithdrawalIdKey, qncAgentWithdrawalWithdrawNoKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentWithdrawalModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentWithdrawal) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentWithdrawalModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentWithdrawalModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentWithdrawalModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentWithdrawalModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentWithdrawal, error) { + + builder = builder.Columns(agentWithdrawalRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWithdrawal + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWithdrawalModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWithdrawal, error) { + + builder = builder.Columns(agentWithdrawalRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWithdrawal + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWithdrawalModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWithdrawal, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentWithdrawalRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentWithdrawal + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentWithdrawalModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentWithdrawal, error) { + + builder = builder.Columns(agentWithdrawalRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWithdrawal + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWithdrawalModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentWithdrawal, error) { + + builder = builder.Columns(agentWithdrawalRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWithdrawal + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWithdrawalModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentWithdrawalModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentWithdrawalModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncAgentWithdrawalIdKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalIdPrefix, id) + qncAgentWithdrawalWithdrawNoKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalWithdrawNoPrefix, data.WithdrawNo) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentWithdrawalIdKey, qncAgentWithdrawalWithdrawNoKey) + return err +} +func (m *defaultAgentWithdrawalModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentWithdrawalIdPrefix, primary) +} +func (m *defaultAgentWithdrawalModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentWithdrawalRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentWithdrawalModel) tableName() string { + return m.table +} diff --git a/app/main/model/agentWithdrawalTaxModel.go b/app/main/model/agentWithdrawalTaxModel.go new file mode 100644 index 0000000..1b9ea2d --- /dev/null +++ b/app/main/model/agentWithdrawalTaxModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AgentWithdrawalTaxModel = (*customAgentWithdrawalTaxModel)(nil) + +type ( + // AgentWithdrawalTaxModel is an interface to be customized, add more methods here, + // and implement the added methods in customAgentWithdrawalTaxModel. + AgentWithdrawalTaxModel interface { + agentWithdrawalTaxModel + } + + customAgentWithdrawalTaxModel struct { + *defaultAgentWithdrawalTaxModel + } +) + +// NewAgentWithdrawalTaxModel returns a model for the database table. +func NewAgentWithdrawalTaxModel(conn sqlx.SqlConn, c cache.CacheConf) AgentWithdrawalTaxModel { + return &customAgentWithdrawalTaxModel{ + defaultAgentWithdrawalTaxModel: newAgentWithdrawalTaxModel(conn, c), + } +} diff --git a/app/main/model/agentWithdrawalTaxModel_gen.go b/app/main/model/agentWithdrawalTaxModel_gen.go new file mode 100644 index 0000000..f4f8277 --- /dev/null +++ b/app/main/model/agentWithdrawalTaxModel_gen.go @@ -0,0 +1,397 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + agentWithdrawalTaxFieldNames = builder.RawFieldNames(&AgentWithdrawalTax{}) + agentWithdrawalTaxRows = strings.Join(agentWithdrawalTaxFieldNames, ",") + agentWithdrawalTaxRowsExpectAutoSet = strings.Join(stringx.Remove(agentWithdrawalTaxFieldNames, "`create_time`", "`update_time`"), ",") + agentWithdrawalTaxRowsWithPlaceHolder = strings.Join(stringx.Remove(agentWithdrawalTaxFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAgentWithdrawalTaxIdPrefix = "cache:qnc:agentWithdrawalTax:id:" +) + +type ( + agentWithdrawalTaxModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AgentWithdrawalTax) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AgentWithdrawalTax, error) + Update(ctx context.Context, session sqlx.Session, data *AgentWithdrawalTax) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentWithdrawalTax) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentWithdrawalTax) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AgentWithdrawalTax, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWithdrawalTax, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWithdrawalTax, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentWithdrawalTax, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentWithdrawalTax, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAgentWithdrawalTaxModel struct { + sqlc.CachedConn + table string + } + + AgentWithdrawalTax struct { + Id string `db:"id"` + AgentId string `db:"agent_id"` + WithdrawalId string `db:"withdrawal_id"` + YearMonth int64 `db:"year_month"` // 年月（格式：YYYYMM） + WithdrawalAmount float64 `db:"withdrawal_amount"` // 提现金额 + TaxableAmount float64 `db:"taxable_amount"` // 应税金额 + TaxRate float64 `db:"tax_rate"` // 税率 + TaxAmount float64 `db:"tax_amount"` // 税费金额 + ActualAmount float64 `db:"actual_amount"` // 实际到账金额 + TaxStatus int64 `db:"tax_status"` // 扣税状态：1=待扣税，2=已扣税，3=扣税失败 + TaxTime sql.NullTime `db:"tax_time"` // 扣税时间 + Remark sql.NullString `db:"remark"` // 备注 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0=未删除，1=已删除 + Version int64 `db:"version"` // 版本号（乐观锁） + } +) + +func newAgentWithdrawalTaxModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAgentWithdrawalTaxModel { + return &defaultAgentWithdrawalTaxModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`agent_withdrawal_tax`", + } +} + +func (m *defaultAgentWithdrawalTaxModel) Insert(ctx context.Context, session sqlx.Session, data *AgentWithdrawalTax) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAgentWithdrawalTaxIdKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalTaxIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, agentWithdrawalTaxRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.AgentId, data.WithdrawalId, data.YearMonth, data.WithdrawalAmount, data.TaxableAmount, data.TaxRate, data.TaxAmount, data.ActualAmount, data.TaxStatus, data.TaxTime, data.Remark, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.AgentId, data.WithdrawalId, data.YearMonth, data.WithdrawalAmount, data.TaxableAmount, data.TaxRate, data.TaxAmount, data.ActualAmount, data.TaxStatus, data.TaxTime, data.Remark, data.DeleteTime, data.DelState, data.Version) + }, qncAgentWithdrawalTaxIdKey) +} +func (m *defaultAgentWithdrawalTaxModel) insertUUID(data *AgentWithdrawalTax) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAgentWithdrawalTaxModel) FindOne(ctx context.Context, id string) (*AgentWithdrawalTax, error) { + qncAgentWithdrawalTaxIdKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalTaxIdPrefix, id) + var resp AgentWithdrawalTax + err := m.QueryRowCtx(ctx, &resp, qncAgentWithdrawalTaxIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentWithdrawalTaxRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAgentWithdrawalTaxModel) Update(ctx context.Context, session sqlx.Session, data *AgentWithdrawalTax) (sql.Result, error) { + qncAgentWithdrawalTaxIdKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalTaxIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, agentWithdrawalTaxRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.AgentId, data.WithdrawalId, data.YearMonth, data.WithdrawalAmount, data.TaxableAmount, data.TaxRate, data.TaxAmount, data.ActualAmount, data.TaxStatus, data.TaxTime, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id) + } + return conn.ExecCtx(ctx, query, data.AgentId, data.WithdrawalId, data.YearMonth, data.WithdrawalAmount, data.TaxableAmount, data.TaxRate, data.TaxAmount, data.ActualAmount, data.TaxStatus, data.TaxTime, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id) + }, qncAgentWithdrawalTaxIdKey) +} + +func (m *defaultAgentWithdrawalTaxModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AgentWithdrawalTax) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncAgentWithdrawalTaxIdKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalTaxIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, agentWithdrawalTaxRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.AgentId, data.WithdrawalId, data.YearMonth, data.WithdrawalAmount, data.TaxableAmount, data.TaxRate, data.TaxAmount, data.ActualAmount, data.TaxStatus, data.TaxTime, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.AgentId, data.WithdrawalId, data.YearMonth, data.WithdrawalAmount, data.TaxableAmount, data.TaxRate, data.TaxAmount, data.ActualAmount, data.TaxStatus, data.TaxTime, data.Remark, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + }, qncAgentWithdrawalTaxIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAgentWithdrawalTaxModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AgentWithdrawalTax) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AgentWithdrawalTaxModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAgentWithdrawalTaxModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentWithdrawalTaxModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAgentWithdrawalTaxModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AgentWithdrawalTax, error) { + + builder = builder.Columns(agentWithdrawalTaxRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWithdrawalTax + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWithdrawalTaxModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWithdrawalTax, error) { + + builder = builder.Columns(agentWithdrawalTaxRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWithdrawalTax + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWithdrawalTaxModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AgentWithdrawalTax, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(agentWithdrawalTaxRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AgentWithdrawalTax + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAgentWithdrawalTaxModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AgentWithdrawalTax, error) { + + builder = builder.Columns(agentWithdrawalTaxRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWithdrawalTax + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWithdrawalTaxModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AgentWithdrawalTax, error) { + + builder = builder.Columns(agentWithdrawalTaxRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AgentWithdrawalTax + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAgentWithdrawalTaxModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAgentWithdrawalTaxModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAgentWithdrawalTaxModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncAgentWithdrawalTaxIdKey := fmt.Sprintf("%s%v", cacheQncAgentWithdrawalTaxIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAgentWithdrawalTaxIdKey) + return err +} +func (m *defaultAgentWithdrawalTaxModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAgentWithdrawalTaxIdPrefix, primary) +} +func (m *defaultAgentWithdrawalTaxModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", agentWithdrawalTaxRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAgentWithdrawalTaxModel) tableName() string { + return m.table +} diff --git a/app/main/model/authorizationDocumentModel.go b/app/main/model/authorizationDocumentModel.go new file mode 100644 index 0000000..794c83b --- /dev/null +++ b/app/main/model/authorizationDocumentModel.go @@ -0,0 +1,43 @@ +package model + +import ( + "context" + + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ AuthorizationDocumentModel = (*customAuthorizationDocumentModel)(nil) + +type ( + // AuthorizationDocumentModel is an interface to be customized, add more methods here, + // and implement the added methods in customAuthorizationDocumentModel. + AuthorizationDocumentModel interface { + authorizationDocumentModel + FindByOrderId(ctx context.Context, orderId string) ([]*AuthorizationDocument, error) + } + + customAuthorizationDocumentModel struct { + *defaultAuthorizationDocumentModel + } +) + +// NewAuthorizationDocumentModel returns a model for the database table. +func NewAuthorizationDocumentModel(conn sqlx.SqlConn, c cache.CacheConf) AuthorizationDocumentModel { + return &customAuthorizationDocumentModel{ + defaultAuthorizationDocumentModel: newAuthorizationDocumentModel(conn, c), + } +} + +// FindByOrderId 根据订单ID查询授权书列表 +func (m *customAuthorizationDocumentModel) FindByOrderId(ctx context.Context, orderId string) ([]*AuthorizationDocument, error) { + query := `SELECT * FROM authorization_document WHERE order_id = ? AND del_state = 0 ORDER BY create_time DESC` + + var authDocs []*AuthorizationDocument + err := m.QueryRowsNoCacheCtx(ctx, &authDocs, query, orderId) + if err != nil { + return nil, err + } + + return authDocs, nil +} diff --git a/app/main/model/authorizationDocumentModel_gen.go b/app/main/model/authorizationDocumentModel_gen.go new file mode 100644 index 0000000..ec1ef03 --- /dev/null +++ b/app/main/model/authorizationDocumentModel_gen.go @@ -0,0 +1,396 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + authorizationDocumentFieldNames = builder.RawFieldNames(&AuthorizationDocument{}) + authorizationDocumentRows = strings.Join(authorizationDocumentFieldNames, ",") + authorizationDocumentRowsExpectAutoSet = strings.Join(stringx.Remove(authorizationDocumentFieldNames, "`create_time`", "`update_time`"), ",") + authorizationDocumentRowsWithPlaceHolder = strings.Join(stringx.Remove(authorizationDocumentFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncAuthorizationDocumentIdPrefix = "cache:qnc:authorizationDocument:id:" +) + +type ( + authorizationDocumentModel interface { + Insert(ctx context.Context, session sqlx.Session, data *AuthorizationDocument) (sql.Result, error) + FindOne(ctx context.Context, id string) (*AuthorizationDocument, error) + Update(ctx context.Context, session sqlx.Session, data *AuthorizationDocument) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AuthorizationDocument) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *AuthorizationDocument) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*AuthorizationDocument, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AuthorizationDocument, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AuthorizationDocument, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AuthorizationDocument, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AuthorizationDocument, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultAuthorizationDocumentModel struct { + sqlc.CachedConn + table string + } + + AuthorizationDocument struct { + Id string `db:"id"` + UserId string `db:"user_id"` + OrderId string `db:"order_id"` + QueryId string `db:"query_id"` + FileName string `db:"file_name"` // 文件名 + FilePath string `db:"file_path"` // 文件路径 + FileUrl string `db:"file_url"` // 文件访问URL + FileSize int64 `db:"file_size"` // 文件大小(字节) + FileType string `db:"file_type"` // 文件类型 + Status string `db:"status"` // 状态(active/expired/deleted) + ExpireTime sql.NullTime `db:"expire_time"` // 过期时间(永久保留设为NULL) + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态 + Version int64 `db:"version"` // 版本号 + } +) + +func newAuthorizationDocumentModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultAuthorizationDocumentModel { + return &defaultAuthorizationDocumentModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`authorization_document`", + } +} + +func (m *defaultAuthorizationDocumentModel) Insert(ctx context.Context, session sqlx.Session, data *AuthorizationDocument) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncAuthorizationDocumentIdKey := fmt.Sprintf("%s%v", cacheQncAuthorizationDocumentIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, authorizationDocumentRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.UserId, data.OrderId, data.QueryId, data.FileName, data.FilePath, data.FileUrl, data.FileSize, data.FileType, data.Status, data.ExpireTime, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.Id, data.UserId, data.OrderId, data.QueryId, data.FileName, data.FilePath, data.FileUrl, data.FileSize, data.FileType, data.Status, data.ExpireTime, data.DeleteTime, data.DelState, data.Version) + }, qncAuthorizationDocumentIdKey) +} +func (m *defaultAuthorizationDocumentModel) insertUUID(data *AuthorizationDocument) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultAuthorizationDocumentModel) FindOne(ctx context.Context, id string) (*AuthorizationDocument, error) { + qncAuthorizationDocumentIdKey := fmt.Sprintf("%s%v", cacheQncAuthorizationDocumentIdPrefix, id) + var resp AuthorizationDocument + err := m.QueryRowCtx(ctx, &resp, qncAuthorizationDocumentIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", authorizationDocumentRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultAuthorizationDocumentModel) Update(ctx context.Context, session sqlx.Session, data *AuthorizationDocument) (sql.Result, error) { + qncAuthorizationDocumentIdKey := fmt.Sprintf("%s%v", cacheQncAuthorizationDocumentIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, authorizationDocumentRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.UserId, data.OrderId, data.QueryId, data.FileName, data.FilePath, data.FileUrl, data.FileSize, data.FileType, data.Status, data.ExpireTime, data.DeleteTime, data.DelState, data.Version, data.Id) + } + return conn.ExecCtx(ctx, query, data.UserId, data.OrderId, data.QueryId, data.FileName, data.FilePath, data.FileUrl, data.FileSize, data.FileType, data.Status, data.ExpireTime, data.DeleteTime, data.DelState, data.Version, data.Id) + }, qncAuthorizationDocumentIdKey) +} + +func (m *defaultAuthorizationDocumentModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *AuthorizationDocument) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncAuthorizationDocumentIdKey := fmt.Sprintf("%s%v", cacheQncAuthorizationDocumentIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, authorizationDocumentRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.UserId, data.OrderId, data.QueryId, data.FileName, data.FilePath, data.FileUrl, data.FileSize, data.FileType, data.Status, data.ExpireTime, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.UserId, data.OrderId, data.QueryId, data.FileName, data.FilePath, data.FileUrl, data.FileSize, data.FileType, data.Status, data.ExpireTime, data.DeleteTime, data.DelState, data.Version, data.Id, oldVersion) + }, qncAuthorizationDocumentIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultAuthorizationDocumentModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *AuthorizationDocument) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "AuthorizationDocumentModel delete err : %+v", err) + } + return nil +} + +func (m *defaultAuthorizationDocumentModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAuthorizationDocumentModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultAuthorizationDocumentModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*AuthorizationDocument, error) { + + builder = builder.Columns(authorizationDocumentRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*AuthorizationDocument + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAuthorizationDocumentModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AuthorizationDocument, error) { + + builder = builder.Columns(authorizationDocumentRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AuthorizationDocument + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAuthorizationDocumentModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*AuthorizationDocument, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(authorizationDocumentRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*AuthorizationDocument + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultAuthorizationDocumentModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*AuthorizationDocument, error) { + + builder = builder.Columns(authorizationDocumentRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AuthorizationDocument + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAuthorizationDocumentModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*AuthorizationDocument, error) { + + builder = builder.Columns(authorizationDocumentRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*AuthorizationDocument + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultAuthorizationDocumentModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultAuthorizationDocumentModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultAuthorizationDocumentModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncAuthorizationDocumentIdKey := fmt.Sprintf("%s%v", cacheQncAuthorizationDocumentIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncAuthorizationDocumentIdKey) + return err +} +func (m *defaultAuthorizationDocumentModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncAuthorizationDocumentIdPrefix, primary) +} +func (m *defaultAuthorizationDocumentModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", authorizationDocumentRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultAuthorizationDocumentModel) tableName() string { + return m.table +} diff --git a/app/main/model/exampleModel.go b/app/main/model/exampleModel.go new file mode 100644 index 0000000..d041ea0 --- /dev/null +++ b/app/main/model/exampleModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ ExampleModel = (*customExampleModel)(nil) + +type ( + // ExampleModel is an interface to be customized, add more methods here, + // and implement the added methods in customExampleModel. + ExampleModel interface { + exampleModel + } + + customExampleModel struct { + *defaultExampleModel + } +) + +// NewExampleModel returns a model for the database table. +func NewExampleModel(conn sqlx.SqlConn, c cache.CacheConf) ExampleModel { + return &customExampleModel{ + defaultExampleModel: newExampleModel(conn, c), + } +} diff --git a/app/main/model/exampleModel_gen.go b/app/main/model/exampleModel_gen.go new file mode 100644 index 0000000..853fac1 --- /dev/null +++ b/app/main/model/exampleModel_gen.go @@ -0,0 +1,456 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + exampleFieldNames = builder.RawFieldNames(&Example{}) + exampleRows = strings.Join(exampleFieldNames, ",") + exampleRowsExpectAutoSet = strings.Join(stringx.Remove(exampleFieldNames, "`create_time`", "`update_time`"), ",") + exampleRowsWithPlaceHolder = strings.Join(stringx.Remove(exampleFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncExampleIdPrefix = "cache:qnc:example:id:" + cacheQncExampleApiIdPrefix = "cache:qnc:example:apiId:" + cacheQncExampleFeatureIdPrefix = "cache:qnc:example:featureId:" +) + +type ( + exampleModel interface { + Insert(ctx context.Context, session sqlx.Session, data *Example) (sql.Result, error) + FindOne(ctx context.Context, id string) (*Example, error) + FindOneByApiId(ctx context.Context, apiId string) (*Example, error) + FindOneByFeatureId(ctx context.Context, featureId string) (*Example, error) + Update(ctx context.Context, session sqlx.Session, data *Example) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *Example) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *Example) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*Example, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Example, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Example, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Example, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Example, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultExampleModel struct { + sqlc.CachedConn + table string + } + + Example struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态 + Version int64 `db:"version"` // 版本号 + ApiId string `db:"api_id"` // API标识 + FeatureId string `db:"feature_id"` + Content string `db:"content"` // 内容 + ProductIdUuid sql.NullString `db:"product_id_uuid"` + FeatureIdUuid sql.NullString `db:"feature_id_uuid"` + } +) + +func newExampleModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultExampleModel { + return &defaultExampleModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`example`", + } +} + +func (m *defaultExampleModel) Insert(ctx context.Context, session sqlx.Session, data *Example) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncExampleApiIdKey := fmt.Sprintf("%s%v", cacheQncExampleApiIdPrefix, data.ApiId) + qncExampleFeatureIdKey := fmt.Sprintf("%s%v", cacheQncExampleFeatureIdPrefix, data.FeatureId) + qncExampleIdKey := fmt.Sprintf("%s%v", cacheQncExampleIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, exampleRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ApiId, data.FeatureId, data.Content, data.ProductIdUuid, data.FeatureIdUuid) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ApiId, data.FeatureId, data.Content, data.ProductIdUuid, data.FeatureIdUuid) + }, qncExampleApiIdKey, qncExampleFeatureIdKey, qncExampleIdKey) +} +func (m *defaultExampleModel) insertUUID(data *Example) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultExampleModel) FindOne(ctx context.Context, id string) (*Example, error) { + qncExampleIdKey := fmt.Sprintf("%s%v", cacheQncExampleIdPrefix, id) + var resp Example + err := m.QueryRowCtx(ctx, &resp, qncExampleIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", exampleRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultExampleModel) FindOneByApiId(ctx context.Context, apiId string) (*Example, error) { + qncExampleApiIdKey := fmt.Sprintf("%s%v", cacheQncExampleApiIdPrefix, apiId) + var resp Example + err := m.QueryRowIndexCtx(ctx, &resp, qncExampleApiIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `api_id` = ? and del_state = ? limit 1", exampleRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, apiId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultExampleModel) FindOneByFeatureId(ctx context.Context, featureId string) (*Example, error) { + qncExampleFeatureIdKey := fmt.Sprintf("%s%v", cacheQncExampleFeatureIdPrefix, featureId) + var resp Example + err := m.QueryRowIndexCtx(ctx, &resp, qncExampleFeatureIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `feature_id` = ? and del_state = ? limit 1", exampleRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, featureId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultExampleModel) Update(ctx context.Context, session sqlx.Session, newData *Example) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncExampleApiIdKey := fmt.Sprintf("%s%v", cacheQncExampleApiIdPrefix, data.ApiId) + qncExampleFeatureIdKey := fmt.Sprintf("%s%v", cacheQncExampleFeatureIdPrefix, data.FeatureId) + qncExampleIdKey := fmt.Sprintf("%s%v", cacheQncExampleIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, exampleRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.FeatureId, newData.Content, newData.ProductIdUuid, newData.FeatureIdUuid, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.FeatureId, newData.Content, newData.ProductIdUuid, newData.FeatureIdUuid, newData.Id) + }, qncExampleApiIdKey, qncExampleFeatureIdKey, qncExampleIdKey) +} + +func (m *defaultExampleModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *Example) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncExampleApiIdKey := fmt.Sprintf("%s%v", cacheQncExampleApiIdPrefix, data.ApiId) + qncExampleFeatureIdKey := fmt.Sprintf("%s%v", cacheQncExampleFeatureIdPrefix, data.FeatureId) + qncExampleIdKey := fmt.Sprintf("%s%v", cacheQncExampleIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, exampleRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.FeatureId, newData.Content, newData.ProductIdUuid, newData.FeatureIdUuid, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.FeatureId, newData.Content, newData.ProductIdUuid, newData.FeatureIdUuid, newData.Id, oldVersion) + }, qncExampleApiIdKey, qncExampleFeatureIdKey, qncExampleIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultExampleModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *Example) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "ExampleModel delete err : %+v", err) + } + return nil +} + +func (m *defaultExampleModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultExampleModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultExampleModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*Example, error) { + + builder = builder.Columns(exampleRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*Example + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultExampleModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Example, error) { + + builder = builder.Columns(exampleRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Example + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultExampleModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Example, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(exampleRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*Example + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultExampleModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Example, error) { + + builder = builder.Columns(exampleRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Example + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultExampleModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Example, error) { + + builder = builder.Columns(exampleRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Example + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultExampleModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultExampleModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultExampleModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncExampleApiIdKey := fmt.Sprintf("%s%v", cacheQncExampleApiIdPrefix, data.ApiId) + qncExampleFeatureIdKey := fmt.Sprintf("%s%v", cacheQncExampleFeatureIdPrefix, data.FeatureId) + qncExampleIdKey := fmt.Sprintf("%s%v", cacheQncExampleIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncExampleApiIdKey, qncExampleFeatureIdKey, qncExampleIdKey) + return err +} +func (m *defaultExampleModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncExampleIdPrefix, primary) +} +func (m *defaultExampleModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", exampleRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultExampleModel) tableName() string { + return m.table +} diff --git a/app/main/model/featureModel.go b/app/main/model/featureModel.go new file mode 100644 index 0000000..b27565e --- /dev/null +++ b/app/main/model/featureModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ FeatureModel = (*customFeatureModel)(nil) + +type ( + // FeatureModel is an interface to be customized, add more methods here, + // and implement the added methods in customFeatureModel. + FeatureModel interface { + featureModel + } + + customFeatureModel struct { + *defaultFeatureModel + } +) + +// NewFeatureModel returns a model for the database table. +func NewFeatureModel(conn sqlx.SqlConn, c cache.CacheConf) FeatureModel { + return &customFeatureModel{ + defaultFeatureModel: newFeatureModel(conn, c), + } +} diff --git a/app/main/model/featureModel_gen.go b/app/main/model/featureModel_gen.go new file mode 100644 index 0000000..2f1aed8 --- /dev/null +++ b/app/main/model/featureModel_gen.go @@ -0,0 +1,427 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + featureFieldNames = builder.RawFieldNames(&Feature{}) + featureRows = strings.Join(featureFieldNames, ",") + featureRowsExpectAutoSet = strings.Join(stringx.Remove(featureFieldNames, "`create_time`", "`update_time`"), ",") + featureRowsWithPlaceHolder = strings.Join(stringx.Remove(featureFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncFeatureIdPrefix = "cache:qnc:feature:id:" + cacheQncFeatureApiIdPrefix = "cache:qnc:feature:apiId:" +) + +type ( + featureModel interface { + Insert(ctx context.Context, session sqlx.Session, data *Feature) (sql.Result, error) + FindOne(ctx context.Context, id string) (*Feature, error) + FindOneByApiId(ctx context.Context, apiId string) (*Feature, error) + Update(ctx context.Context, session sqlx.Session, data *Feature) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *Feature) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *Feature) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*Feature, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Feature, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Feature, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Feature, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Feature, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultFeatureModel struct { + sqlc.CachedConn + table string + } + + Feature struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态 + Version int64 `db:"version"` // 版本号 + ApiId string `db:"api_id"` // API标识 + Name string `db:"name"` // 描述 + } +) + +func newFeatureModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultFeatureModel { + return &defaultFeatureModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`feature`", + } +} + +func (m *defaultFeatureModel) Insert(ctx context.Context, session sqlx.Session, data *Feature) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncFeatureApiIdKey := fmt.Sprintf("%s%v", cacheQncFeatureApiIdPrefix, data.ApiId) + qncFeatureIdKey := fmt.Sprintf("%s%v", cacheQncFeatureIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?)", m.table, featureRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ApiId, data.Name) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ApiId, data.Name) + }, qncFeatureApiIdKey, qncFeatureIdKey) +} +func (m *defaultFeatureModel) insertUUID(data *Feature) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultFeatureModel) FindOne(ctx context.Context, id string) (*Feature, error) { + qncFeatureIdKey := fmt.Sprintf("%s%v", cacheQncFeatureIdPrefix, id) + var resp Feature + err := m.QueryRowCtx(ctx, &resp, qncFeatureIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", featureRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultFeatureModel) FindOneByApiId(ctx context.Context, apiId string) (*Feature, error) { + qncFeatureApiIdKey := fmt.Sprintf("%s%v", cacheQncFeatureApiIdPrefix, apiId) + var resp Feature + err := m.QueryRowIndexCtx(ctx, &resp, qncFeatureApiIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `api_id` = ? and del_state = ? limit 1", featureRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, apiId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultFeatureModel) Update(ctx context.Context, session sqlx.Session, newData *Feature) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncFeatureApiIdKey := fmt.Sprintf("%s%v", cacheQncFeatureApiIdPrefix, data.ApiId) + qncFeatureIdKey := fmt.Sprintf("%s%v", cacheQncFeatureIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, featureRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.Id) + }, qncFeatureApiIdKey, qncFeatureIdKey) +} + +func (m *defaultFeatureModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *Feature) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncFeatureApiIdKey := fmt.Sprintf("%s%v", cacheQncFeatureApiIdPrefix, data.ApiId) + qncFeatureIdKey := fmt.Sprintf("%s%v", cacheQncFeatureIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, featureRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ApiId, newData.Name, newData.Id, oldVersion) + }, qncFeatureApiIdKey, qncFeatureIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultFeatureModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *Feature) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "FeatureModel delete err : %+v", err) + } + return nil +} + +func (m *defaultFeatureModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultFeatureModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultFeatureModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*Feature, error) { + + builder = builder.Columns(featureRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*Feature + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultFeatureModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Feature, error) { + + builder = builder.Columns(featureRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Feature + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultFeatureModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Feature, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(featureRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*Feature + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultFeatureModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Feature, error) { + + builder = builder.Columns(featureRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Feature + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultFeatureModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Feature, error) { + + builder = builder.Columns(featureRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Feature + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultFeatureModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultFeatureModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultFeatureModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncFeatureApiIdKey := fmt.Sprintf("%s%v", cacheQncFeatureApiIdPrefix, data.ApiId) + qncFeatureIdKey := fmt.Sprintf("%s%v", cacheQncFeatureIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncFeatureApiIdKey, qncFeatureIdKey) + return err +} +func (m *defaultFeatureModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncFeatureIdPrefix, primary) +} +func (m *defaultFeatureModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", featureRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultFeatureModel) tableName() string { + return m.table +} diff --git a/app/main/model/globalNotificationsModel.go b/app/main/model/globalNotificationsModel.go new file mode 100644 index 0000000..059bdff --- /dev/null +++ b/app/main/model/globalNotificationsModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ GlobalNotificationsModel = (*customGlobalNotificationsModel)(nil) + +type ( + // GlobalNotificationsModel is an interface to be customized, add more methods here, + // and implement the added methods in customGlobalNotificationsModel. + GlobalNotificationsModel interface { + globalNotificationsModel + } + + customGlobalNotificationsModel struct { + *defaultGlobalNotificationsModel + } +) + +// NewGlobalNotificationsModel returns a model for the database table. +func NewGlobalNotificationsModel(conn sqlx.SqlConn, c cache.CacheConf) GlobalNotificationsModel { + return &customGlobalNotificationsModel{ + defaultGlobalNotificationsModel: newGlobalNotificationsModel(conn, c), + } +} diff --git a/app/main/model/globalNotificationsModel_gen.go b/app/main/model/globalNotificationsModel_gen.go new file mode 100644 index 0000000..00ecb40 --- /dev/null +++ b/app/main/model/globalNotificationsModel_gen.go @@ -0,0 +1,394 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + globalNotificationsFieldNames = builder.RawFieldNames(&GlobalNotifications{}) + globalNotificationsRows = strings.Join(globalNotificationsFieldNames, ",") + globalNotificationsRowsExpectAutoSet = strings.Join(stringx.Remove(globalNotificationsFieldNames, "`create_time`", "`update_time`"), ",") + globalNotificationsRowsWithPlaceHolder = strings.Join(stringx.Remove(globalNotificationsFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncGlobalNotificationsIdPrefix = "cache:qnc:globalNotifications:id:" +) + +type ( + globalNotificationsModel interface { + Insert(ctx context.Context, session sqlx.Session, data *GlobalNotifications) (sql.Result, error) + FindOne(ctx context.Context, id string) (*GlobalNotifications, error) + Update(ctx context.Context, session sqlx.Session, data *GlobalNotifications) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *GlobalNotifications) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *GlobalNotifications) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*GlobalNotifications, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*GlobalNotifications, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*GlobalNotifications, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*GlobalNotifications, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*GlobalNotifications, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultGlobalNotificationsModel struct { + sqlc.CachedConn + table string + } + + GlobalNotifications struct { + Id string `db:"id"` + Title string `db:"title"` + Content string `db:"content"` + NotificationPage string `db:"notification_page"` + StartDate sql.NullTime `db:"start_date"` + EndDate sql.NullTime `db:"end_date"` + StartTime string `db:"start_time"` + EndTime string `db:"end_time"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + Status int64 `db:"status"` + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + } +) + +func newGlobalNotificationsModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultGlobalNotificationsModel { + return &defaultGlobalNotificationsModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`global_notifications`", + } +} + +func (m *defaultGlobalNotificationsModel) Insert(ctx context.Context, session sqlx.Session, data *GlobalNotifications) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncGlobalNotificationsIdKey := fmt.Sprintf("%s%v", cacheQncGlobalNotificationsIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, globalNotificationsRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.Title, data.Content, data.NotificationPage, data.StartDate, data.EndDate, data.StartTime, data.EndTime, data.Status, data.DelState, data.Version, data.DeleteTime) + } + return conn.ExecCtx(ctx, query, data.Id, data.Title, data.Content, data.NotificationPage, data.StartDate, data.EndDate, data.StartTime, data.EndTime, data.Status, data.DelState, data.Version, data.DeleteTime) + }, qncGlobalNotificationsIdKey) +} +func (m *defaultGlobalNotificationsModel) insertUUID(data *GlobalNotifications) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultGlobalNotificationsModel) FindOne(ctx context.Context, id string) (*GlobalNotifications, error) { + qncGlobalNotificationsIdKey := fmt.Sprintf("%s%v", cacheQncGlobalNotificationsIdPrefix, id) + var resp GlobalNotifications + err := m.QueryRowCtx(ctx, &resp, qncGlobalNotificationsIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", globalNotificationsRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultGlobalNotificationsModel) Update(ctx context.Context, session sqlx.Session, data *GlobalNotifications) (sql.Result, error) { + qncGlobalNotificationsIdKey := fmt.Sprintf("%s%v", cacheQncGlobalNotificationsIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, globalNotificationsRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.Title, data.Content, data.NotificationPage, data.StartDate, data.EndDate, data.StartTime, data.EndTime, data.Status, data.DelState, data.Version, data.DeleteTime, data.Id) + } + return conn.ExecCtx(ctx, query, data.Title, data.Content, data.NotificationPage, data.StartDate, data.EndDate, data.StartTime, data.EndTime, data.Status, data.DelState, data.Version, data.DeleteTime, data.Id) + }, qncGlobalNotificationsIdKey) +} + +func (m *defaultGlobalNotificationsModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *GlobalNotifications) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncGlobalNotificationsIdKey := fmt.Sprintf("%s%v", cacheQncGlobalNotificationsIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, globalNotificationsRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.Title, data.Content, data.NotificationPage, data.StartDate, data.EndDate, data.StartTime, data.EndTime, data.Status, data.DelState, data.Version, data.DeleteTime, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.Title, data.Content, data.NotificationPage, data.StartDate, data.EndDate, data.StartTime, data.EndTime, data.Status, data.DelState, data.Version, data.DeleteTime, data.Id, oldVersion) + }, qncGlobalNotificationsIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultGlobalNotificationsModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *GlobalNotifications) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "GlobalNotificationsModel delete err : %+v", err) + } + return nil +} + +func (m *defaultGlobalNotificationsModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultGlobalNotificationsModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultGlobalNotificationsModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*GlobalNotifications, error) { + + builder = builder.Columns(globalNotificationsRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*GlobalNotifications + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultGlobalNotificationsModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*GlobalNotifications, error) { + + builder = builder.Columns(globalNotificationsRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*GlobalNotifications + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultGlobalNotificationsModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*GlobalNotifications, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(globalNotificationsRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*GlobalNotifications + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultGlobalNotificationsModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*GlobalNotifications, error) { + + builder = builder.Columns(globalNotificationsRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*GlobalNotifications + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultGlobalNotificationsModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*GlobalNotifications, error) { + + builder = builder.Columns(globalNotificationsRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*GlobalNotifications + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultGlobalNotificationsModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultGlobalNotificationsModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultGlobalNotificationsModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncGlobalNotificationsIdKey := fmt.Sprintf("%s%v", cacheQncGlobalNotificationsIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncGlobalNotificationsIdKey) + return err +} +func (m *defaultGlobalNotificationsModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncGlobalNotificationsIdPrefix, primary) +} +func (m *defaultGlobalNotificationsModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", globalNotificationsRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultGlobalNotificationsModel) tableName() string { + return m.table +} diff --git a/app/main/model/orderModel.go b/app/main/model/orderModel.go new file mode 100644 index 0000000..9c8c60b --- /dev/null +++ b/app/main/model/orderModel.go @@ -0,0 +1,55 @@ +package model + +import ( + "context" + "database/sql" + "fmt" + + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ OrderModel = (*customOrderModel)(nil) + +type ( + // OrderModel is an interface to be customized, add more methods here, + // and implement the added methods in customOrderModel. + OrderModel interface { + orderModel + UpdateUserIDWithSession(ctx context.Context, session sqlx.Session, sourceUserID, targetUserID string) error + } + + customOrderModel struct { + *defaultOrderModel + } +) + +// NewOrderModel returns a model for the database table. +func NewOrderModel(conn sqlx.SqlConn, c cache.CacheConf) OrderModel { + return &customOrderModel{ + defaultOrderModel: newOrderModel(conn, c), + } +} + +func (m *customOrderModel) UpdateUserIDWithSession(ctx context.Context, session sqlx.Session, sourceUserID, targetUserID string) error { + builder := m.defaultOrderModel.SelectBuilder().Where("user_id = ?", sourceUserID) + rows, err := m.defaultOrderModel.FindAll(ctx, builder, "") + if err != nil { + return err + } + + keys := make([]string, 0, len(rows)*2) + for _, r := range rows { + keys = append(keys, fmt.Sprintf("%s%v", cacheQncOrderIdPrefix, r.Id)) + keys = append(keys, fmt.Sprintf("%s%v", cacheQncOrderOrderNoPrefix, r.OrderNo)) + } + + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("UPDATE %s SET user_id = ? WHERE user_id = ?", m.defaultOrderModel.tableName()) + if session != nil { + return session.ExecCtx(ctx, query, targetUserID, sourceUserID) + } + return conn.ExecCtx(ctx, query, targetUserID, sourceUserID) + }, keys...) + return err +} diff --git a/app/main/model/orderModel_gen.go b/app/main/model/orderModel_gen.go new file mode 100644 index 0000000..3f95256 --- /dev/null +++ b/app/main/model/orderModel_gen.go @@ -0,0 +1,436 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + orderFieldNames = builder.RawFieldNames(&Order{}) + orderRows = strings.Join(orderFieldNames, ",") + orderRowsExpectAutoSet = strings.Join(stringx.Remove(orderFieldNames, "`create_time`", "`update_time`"), ",") + orderRowsWithPlaceHolder = strings.Join(stringx.Remove(orderFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncOrderIdPrefix = "cache:qnc:order:id:" + cacheQncOrderOrderNoPrefix = "cache:qnc:order:orderNo:" +) + +type ( + orderModel interface { + Insert(ctx context.Context, session sqlx.Session, data *Order) (sql.Result, error) + FindOne(ctx context.Context, id string) (*Order, error) + FindOneByOrderNo(ctx context.Context, orderNo string) (*Order, error) + Update(ctx context.Context, session sqlx.Session, data *Order) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *Order) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *Order) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*Order, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Order, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Order, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Order, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Order, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultOrderModel struct { + sqlc.CachedConn + table string + } + + Order struct { + Id string `db:"id"` + OrderNo string `db:"order_no"` // 自生成的订单号 + UserId string `db:"user_id"` + ProductId string `db:"product_id"` + PaymentPlatform string `db:"payment_platform"` // 支付平台（支付宝、微信、苹果内购、其他） + PaymentScene string `db:"payment_scene"` // 支付场景（App、H5、微信小程序、公众号） + PlatformOrderId sql.NullString `db:"platform_order_id"` // 支付平台订单号 + Amount float64 `db:"amount"` // 支付金额 + Status string `db:"status"` // 支付状态 + DelState int64 `db:"del_state"` // 删除状态 + Version int64 `db:"version"` // 版本号 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + PayTime sql.NullTime `db:"pay_time"` // 支付时间 + RefundTime sql.NullTime `db:"refund_time"` // 退款时间 + CloseTime sql.NullTime `db:"close_time"` // 订单关闭时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + } +) + +func newOrderModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultOrderModel { + return &defaultOrderModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`order`", + } +} + +func (m *defaultOrderModel) Insert(ctx context.Context, session sqlx.Session, data *Order) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncOrderIdKey := fmt.Sprintf("%s%v", cacheQncOrderIdPrefix, data.Id) + qncOrderOrderNoKey := fmt.Sprintf("%s%v", cacheQncOrderOrderNoPrefix, data.OrderNo) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, orderRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.OrderNo, data.UserId, data.ProductId, data.PaymentPlatform, data.PaymentScene, data.PlatformOrderId, data.Amount, data.Status, data.DelState, data.Version, data.PayTime, data.RefundTime, data.CloseTime, data.DeleteTime) + } + return conn.ExecCtx(ctx, query, data.Id, data.OrderNo, data.UserId, data.ProductId, data.PaymentPlatform, data.PaymentScene, data.PlatformOrderId, data.Amount, data.Status, data.DelState, data.Version, data.PayTime, data.RefundTime, data.CloseTime, data.DeleteTime) + }, qncOrderIdKey, qncOrderOrderNoKey) +} +func (m *defaultOrderModel) insertUUID(data *Order) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultOrderModel) FindOne(ctx context.Context, id string) (*Order, error) { + qncOrderIdKey := fmt.Sprintf("%s%v", cacheQncOrderIdPrefix, id) + var resp Order + err := m.QueryRowCtx(ctx, &resp, qncOrderIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", orderRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultOrderModel) FindOneByOrderNo(ctx context.Context, orderNo string) (*Order, error) { + qncOrderOrderNoKey := fmt.Sprintf("%s%v", cacheQncOrderOrderNoPrefix, orderNo) + var resp Order + err := m.QueryRowIndexCtx(ctx, &resp, qncOrderOrderNoKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `order_no` = ? and del_state = ? limit 1", orderRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, orderNo, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultOrderModel) Update(ctx context.Context, session sqlx.Session, newData *Order) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncOrderIdKey := fmt.Sprintf("%s%v", cacheQncOrderIdPrefix, data.Id) + qncOrderOrderNoKey := fmt.Sprintf("%s%v", cacheQncOrderOrderNoPrefix, data.OrderNo) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, orderRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.OrderNo, newData.UserId, newData.ProductId, newData.PaymentPlatform, newData.PaymentScene, newData.PlatformOrderId, newData.Amount, newData.Status, newData.DelState, newData.Version, newData.PayTime, newData.RefundTime, newData.CloseTime, newData.DeleteTime, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.OrderNo, newData.UserId, newData.ProductId, newData.PaymentPlatform, newData.PaymentScene, newData.PlatformOrderId, newData.Amount, newData.Status, newData.DelState, newData.Version, newData.PayTime, newData.RefundTime, newData.CloseTime, newData.DeleteTime, newData.Id) + }, qncOrderIdKey, qncOrderOrderNoKey) +} + +func (m *defaultOrderModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *Order) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncOrderIdKey := fmt.Sprintf("%s%v", cacheQncOrderIdPrefix, data.Id) + qncOrderOrderNoKey := fmt.Sprintf("%s%v", cacheQncOrderOrderNoPrefix, data.OrderNo) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, orderRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.OrderNo, newData.UserId, newData.ProductId, newData.PaymentPlatform, newData.PaymentScene, newData.PlatformOrderId, newData.Amount, newData.Status, newData.DelState, newData.Version, newData.PayTime, newData.RefundTime, newData.CloseTime, newData.DeleteTime, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.OrderNo, newData.UserId, newData.ProductId, newData.PaymentPlatform, newData.PaymentScene, newData.PlatformOrderId, newData.Amount, newData.Status, newData.DelState, newData.Version, newData.PayTime, newData.RefundTime, newData.CloseTime, newData.DeleteTime, newData.Id, oldVersion) + }, qncOrderIdKey, qncOrderOrderNoKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultOrderModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *Order) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "OrderModel delete err : %+v", err) + } + return nil +} + +func (m *defaultOrderModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultOrderModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultOrderModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*Order, error) { + + builder = builder.Columns(orderRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*Order + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultOrderModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Order, error) { + + builder = builder.Columns(orderRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Order + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultOrderModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Order, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(orderRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*Order + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultOrderModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Order, error) { + + builder = builder.Columns(orderRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Order + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultOrderModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Order, error) { + + builder = builder.Columns(orderRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Order + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultOrderModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultOrderModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultOrderModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncOrderIdKey := fmt.Sprintf("%s%v", cacheQncOrderIdPrefix, id) + qncOrderOrderNoKey := fmt.Sprintf("%s%v", cacheQncOrderOrderNoPrefix, data.OrderNo) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncOrderIdKey, qncOrderOrderNoKey) + return err +} +func (m *defaultOrderModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncOrderIdPrefix, primary) +} +func (m *defaultOrderModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", orderRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultOrderModel) tableName() string { + return m.table +} diff --git a/app/main/model/orderRefundModel.go b/app/main/model/orderRefundModel.go new file mode 100644 index 0000000..92c3174 --- /dev/null +++ b/app/main/model/orderRefundModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ OrderRefundModel = (*customOrderRefundModel)(nil) + +type ( + // OrderRefundModel is an interface to be customized, add more methods here, + // and implement the added methods in customOrderRefundModel. + OrderRefundModel interface { + orderRefundModel + } + + customOrderRefundModel struct { + *defaultOrderRefundModel + } +) + +// NewOrderRefundModel returns a model for the database table. +func NewOrderRefundModel(conn sqlx.SqlConn, c cache.CacheConf) OrderRefundModel { + return &customOrderRefundModel{ + defaultOrderRefundModel: newOrderRefundModel(conn, c), + } +} diff --git a/app/main/model/orderRefundModel_gen.go b/app/main/model/orderRefundModel_gen.go new file mode 100644 index 0000000..a92de2d --- /dev/null +++ b/app/main/model/orderRefundModel_gen.go @@ -0,0 +1,461 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + orderRefundFieldNames = builder.RawFieldNames(&OrderRefund{}) + orderRefundRows = strings.Join(orderRefundFieldNames, ",") + orderRefundRowsExpectAutoSet = strings.Join(stringx.Remove(orderRefundFieldNames, "`create_time`", "`update_time`"), ",") + orderRefundRowsWithPlaceHolder = strings.Join(stringx.Remove(orderRefundFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncOrderRefundIdPrefix = "cache:qnc:orderRefund:id:" + cacheQncOrderRefundPlatformRefundIdPrefix = "cache:qnc:orderRefund:platformRefundId:" + cacheQncOrderRefundRefundNoPrefix = "cache:qnc:orderRefund:refundNo:" +) + +type ( + orderRefundModel interface { + Insert(ctx context.Context, session sqlx.Session, data *OrderRefund) (sql.Result, error) + FindOne(ctx context.Context, id string) (*OrderRefund, error) + FindOneByPlatformRefundId(ctx context.Context, platformRefundId sql.NullString) (*OrderRefund, error) + FindOneByRefundNo(ctx context.Context, refundNo string) (*OrderRefund, error) + Update(ctx context.Context, session sqlx.Session, data *OrderRefund) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *OrderRefund) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *OrderRefund) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*OrderRefund, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*OrderRefund, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*OrderRefund, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*OrderRefund, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*OrderRefund, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultOrderRefundModel struct { + sqlc.CachedConn + table string + } + + OrderRefund struct { + Id string `db:"id"` + RefundNo string `db:"refund_no"` // 退款单号 + OrderId string `db:"order_id"` + UserId string `db:"user_id"` + ProductId string `db:"product_id"` + PlatformRefundId sql.NullString `db:"platform_refund_id"` // 支付平台退款单号 + RefundAmount float64 `db:"refund_amount"` // 退款金额 + RefundReason sql.NullString `db:"refund_reason"` // 退款原因 + Status string `db:"status"` // 退款状态 + DelState int64 `db:"del_state"` // 删除状态 + Version int64 `db:"version"` // 版本号 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + RefundTime sql.NullTime `db:"refund_time"` // 退款成功时间 + CloseTime sql.NullTime `db:"close_time"` // 退款关闭时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + } +) + +func newOrderRefundModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultOrderRefundModel { + return &defaultOrderRefundModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`order_refund`", + } +} + +func (m *defaultOrderRefundModel) Insert(ctx context.Context, session sqlx.Session, data *OrderRefund) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncOrderRefundIdKey := fmt.Sprintf("%s%v", cacheQncOrderRefundIdPrefix, data.Id) + qncOrderRefundPlatformRefundIdKey := fmt.Sprintf("%s%v", cacheQncOrderRefundPlatformRefundIdPrefix, data.PlatformRefundId) + qncOrderRefundRefundNoKey := fmt.Sprintf("%s%v", cacheQncOrderRefundRefundNoPrefix, data.RefundNo) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, orderRefundRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.RefundNo, data.OrderId, data.UserId, data.ProductId, data.PlatformRefundId, data.RefundAmount, data.RefundReason, data.Status, data.DelState, data.Version, data.RefundTime, data.CloseTime, data.DeleteTime) + } + return conn.ExecCtx(ctx, query, data.Id, data.RefundNo, data.OrderId, data.UserId, data.ProductId, data.PlatformRefundId, data.RefundAmount, data.RefundReason, data.Status, data.DelState, data.Version, data.RefundTime, data.CloseTime, data.DeleteTime) + }, qncOrderRefundIdKey, qncOrderRefundPlatformRefundIdKey, qncOrderRefundRefundNoKey) +} +func (m *defaultOrderRefundModel) insertUUID(data *OrderRefund) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultOrderRefundModel) FindOne(ctx context.Context, id string) (*OrderRefund, error) { + qncOrderRefundIdKey := fmt.Sprintf("%s%v", cacheQncOrderRefundIdPrefix, id) + var resp OrderRefund + err := m.QueryRowCtx(ctx, &resp, qncOrderRefundIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", orderRefundRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultOrderRefundModel) FindOneByPlatformRefundId(ctx context.Context, platformRefundId sql.NullString) (*OrderRefund, error) { + qncOrderRefundPlatformRefundIdKey := fmt.Sprintf("%s%v", cacheQncOrderRefundPlatformRefundIdPrefix, platformRefundId) + var resp OrderRefund + err := m.QueryRowIndexCtx(ctx, &resp, qncOrderRefundPlatformRefundIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `platform_refund_id` = ? and del_state = ? limit 1", orderRefundRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, platformRefundId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultOrderRefundModel) FindOneByRefundNo(ctx context.Context, refundNo string) (*OrderRefund, error) { + qncOrderRefundRefundNoKey := fmt.Sprintf("%s%v", cacheQncOrderRefundRefundNoPrefix, refundNo) + var resp OrderRefund + err := m.QueryRowIndexCtx(ctx, &resp, qncOrderRefundRefundNoKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `refund_no` = ? and del_state = ? limit 1", orderRefundRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, refundNo, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultOrderRefundModel) Update(ctx context.Context, session sqlx.Session, newData *OrderRefund) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncOrderRefundIdKey := fmt.Sprintf("%s%v", cacheQncOrderRefundIdPrefix, data.Id) + qncOrderRefundPlatformRefundIdKey := fmt.Sprintf("%s%v", cacheQncOrderRefundPlatformRefundIdPrefix, data.PlatformRefundId) + qncOrderRefundRefundNoKey := fmt.Sprintf("%s%v", cacheQncOrderRefundRefundNoPrefix, data.RefundNo) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, orderRefundRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.RefundNo, newData.OrderId, newData.UserId, newData.ProductId, newData.PlatformRefundId, newData.RefundAmount, newData.RefundReason, newData.Status, newData.DelState, newData.Version, newData.RefundTime, newData.CloseTime, newData.DeleteTime, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.RefundNo, newData.OrderId, newData.UserId, newData.ProductId, newData.PlatformRefundId, newData.RefundAmount, newData.RefundReason, newData.Status, newData.DelState, newData.Version, newData.RefundTime, newData.CloseTime, newData.DeleteTime, newData.Id) + }, qncOrderRefundIdKey, qncOrderRefundPlatformRefundIdKey, qncOrderRefundRefundNoKey) +} + +func (m *defaultOrderRefundModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *OrderRefund) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncOrderRefundIdKey := fmt.Sprintf("%s%v", cacheQncOrderRefundIdPrefix, data.Id) + qncOrderRefundPlatformRefundIdKey := fmt.Sprintf("%s%v", cacheQncOrderRefundPlatformRefundIdPrefix, data.PlatformRefundId) + qncOrderRefundRefundNoKey := fmt.Sprintf("%s%v", cacheQncOrderRefundRefundNoPrefix, data.RefundNo) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, orderRefundRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.RefundNo, newData.OrderId, newData.UserId, newData.ProductId, newData.PlatformRefundId, newData.RefundAmount, newData.RefundReason, newData.Status, newData.DelState, newData.Version, newData.RefundTime, newData.CloseTime, newData.DeleteTime, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.RefundNo, newData.OrderId, newData.UserId, newData.ProductId, newData.PlatformRefundId, newData.RefundAmount, newData.RefundReason, newData.Status, newData.DelState, newData.Version, newData.RefundTime, newData.CloseTime, newData.DeleteTime, newData.Id, oldVersion) + }, qncOrderRefundIdKey, qncOrderRefundPlatformRefundIdKey, qncOrderRefundRefundNoKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultOrderRefundModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *OrderRefund) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "OrderRefundModel delete err : %+v", err) + } + return nil +} + +func (m *defaultOrderRefundModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultOrderRefundModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultOrderRefundModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*OrderRefund, error) { + + builder = builder.Columns(orderRefundRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*OrderRefund + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultOrderRefundModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*OrderRefund, error) { + + builder = builder.Columns(orderRefundRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*OrderRefund + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultOrderRefundModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*OrderRefund, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(orderRefundRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*OrderRefund + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultOrderRefundModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*OrderRefund, error) { + + builder = builder.Columns(orderRefundRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*OrderRefund + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultOrderRefundModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*OrderRefund, error) { + + builder = builder.Columns(orderRefundRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*OrderRefund + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultOrderRefundModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultOrderRefundModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultOrderRefundModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncOrderRefundIdKey := fmt.Sprintf("%s%v", cacheQncOrderRefundIdPrefix, id) + qncOrderRefundPlatformRefundIdKey := fmt.Sprintf("%s%v", cacheQncOrderRefundPlatformRefundIdPrefix, data.PlatformRefundId) + qncOrderRefundRefundNoKey := fmt.Sprintf("%s%v", cacheQncOrderRefundRefundNoPrefix, data.RefundNo) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncOrderRefundIdKey, qncOrderRefundPlatformRefundIdKey, qncOrderRefundRefundNoKey) + return err +} +func (m *defaultOrderRefundModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncOrderRefundIdPrefix, primary) +} +func (m *defaultOrderRefundModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", orderRefundRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultOrderRefundModel) tableName() string { + return m.table +} diff --git a/app/main/model/productFeatureModel.go b/app/main/model/productFeatureModel.go new file mode 100644 index 0000000..5b77422 --- /dev/null +++ b/app/main/model/productFeatureModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ ProductFeatureModel = (*customProductFeatureModel)(nil) + +type ( + // ProductFeatureModel is an interface to be customized, add more methods here, + // and implement the added methods in customProductFeatureModel. + ProductFeatureModel interface { + productFeatureModel + } + + customProductFeatureModel struct { + *defaultProductFeatureModel + } +) + +// NewProductFeatureModel returns a model for the database table. +func NewProductFeatureModel(conn sqlx.SqlConn, c cache.CacheConf) ProductFeatureModel { + return &customProductFeatureModel{ + defaultProductFeatureModel: newProductFeatureModel(conn, c), + } +} diff --git a/app/main/model/productFeatureModel_gen.go b/app/main/model/productFeatureModel_gen.go new file mode 100644 index 0000000..c3d47e1 --- /dev/null +++ b/app/main/model/productFeatureModel_gen.go @@ -0,0 +1,393 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + productFeatureFieldNames = builder.RawFieldNames(&ProductFeature{}) + productFeatureRows = strings.Join(productFeatureFieldNames, ",") + productFeatureRowsExpectAutoSet = strings.Join(stringx.Remove(productFeatureFieldNames, "`create_time`", "`update_time`"), ",") + productFeatureRowsWithPlaceHolder = strings.Join(stringx.Remove(productFeatureFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncProductFeatureIdPrefix = "cache:qnc:productFeature:id:" +) + +type ( + productFeatureModel interface { + Insert(ctx context.Context, session sqlx.Session, data *ProductFeature) (sql.Result, error) + FindOne(ctx context.Context, id string) (*ProductFeature, error) + Update(ctx context.Context, session sqlx.Session, data *ProductFeature) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *ProductFeature) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *ProductFeature) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*ProductFeature, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*ProductFeature, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*ProductFeature, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*ProductFeature, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*ProductFeature, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultProductFeatureModel struct { + sqlc.CachedConn + table string + } + + ProductFeature struct { + Id string `db:"id"` + ProductId string `db:"product_id"` + FeatureId string `db:"feature_id"` + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态 + Version int64 `db:"version"` // 版本号 + Sort int64 `db:"sort"` + IsImportant int64 `db:"is_important"` + Enable int64 `db:"enable"` + ProductIdUuid sql.NullString `db:"product_id_uuid"` + FeatureIdUuid sql.NullString `db:"feature_id_uuid"` + } +) + +func newProductFeatureModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultProductFeatureModel { + return &defaultProductFeatureModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`product_feature`", + } +} + +func (m *defaultProductFeatureModel) Insert(ctx context.Context, session sqlx.Session, data *ProductFeature) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncProductFeatureIdKey := fmt.Sprintf("%s%v", cacheQncProductFeatureIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, productFeatureRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.ProductId, data.FeatureId, data.DeleteTime, data.DelState, data.Version, data.Sort, data.IsImportant, data.Enable, data.ProductIdUuid, data.FeatureIdUuid) + } + return conn.ExecCtx(ctx, query, data.Id, data.ProductId, data.FeatureId, data.DeleteTime, data.DelState, data.Version, data.Sort, data.IsImportant, data.Enable, data.ProductIdUuid, data.FeatureIdUuid) + }, qncProductFeatureIdKey) +} +func (m *defaultProductFeatureModel) insertUUID(data *ProductFeature) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultProductFeatureModel) FindOne(ctx context.Context, id string) (*ProductFeature, error) { + qncProductFeatureIdKey := fmt.Sprintf("%s%v", cacheQncProductFeatureIdPrefix, id) + var resp ProductFeature + err := m.QueryRowCtx(ctx, &resp, qncProductFeatureIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", productFeatureRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultProductFeatureModel) Update(ctx context.Context, session sqlx.Session, data *ProductFeature) (sql.Result, error) { + qncProductFeatureIdKey := fmt.Sprintf("%s%v", cacheQncProductFeatureIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, productFeatureRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.ProductId, data.FeatureId, data.DeleteTime, data.DelState, data.Version, data.Sort, data.IsImportant, data.Enable, data.ProductIdUuid, data.FeatureIdUuid, data.Id) + } + return conn.ExecCtx(ctx, query, data.ProductId, data.FeatureId, data.DeleteTime, data.DelState, data.Version, data.Sort, data.IsImportant, data.Enable, data.ProductIdUuid, data.FeatureIdUuid, data.Id) + }, qncProductFeatureIdKey) +} + +func (m *defaultProductFeatureModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *ProductFeature) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncProductFeatureIdKey := fmt.Sprintf("%s%v", cacheQncProductFeatureIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, productFeatureRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.ProductId, data.FeatureId, data.DeleteTime, data.DelState, data.Version, data.Sort, data.IsImportant, data.Enable, data.ProductIdUuid, data.FeatureIdUuid, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.ProductId, data.FeatureId, data.DeleteTime, data.DelState, data.Version, data.Sort, data.IsImportant, data.Enable, data.ProductIdUuid, data.FeatureIdUuid, data.Id, oldVersion) + }, qncProductFeatureIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultProductFeatureModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *ProductFeature) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "ProductFeatureModel delete err : %+v", err) + } + return nil +} + +func (m *defaultProductFeatureModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultProductFeatureModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultProductFeatureModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*ProductFeature, error) { + + builder = builder.Columns(productFeatureRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*ProductFeature + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultProductFeatureModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*ProductFeature, error) { + + builder = builder.Columns(productFeatureRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*ProductFeature + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultProductFeatureModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*ProductFeature, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(productFeatureRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*ProductFeature + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultProductFeatureModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*ProductFeature, error) { + + builder = builder.Columns(productFeatureRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*ProductFeature + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultProductFeatureModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*ProductFeature, error) { + + builder = builder.Columns(productFeatureRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*ProductFeature + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultProductFeatureModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultProductFeatureModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultProductFeatureModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncProductFeatureIdKey := fmt.Sprintf("%s%v", cacheQncProductFeatureIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncProductFeatureIdKey) + return err +} +func (m *defaultProductFeatureModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncProductFeatureIdPrefix, primary) +} +func (m *defaultProductFeatureModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", productFeatureRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultProductFeatureModel) tableName() string { + return m.table +} diff --git a/app/main/model/productModel.go b/app/main/model/productModel.go new file mode 100644 index 0000000..df18c7a --- /dev/null +++ b/app/main/model/productModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ ProductModel = (*customProductModel)(nil) + +type ( + // ProductModel is an interface to be customized, add more methods here, + // and implement the added methods in customProductModel. + ProductModel interface { + productModel + } + + customProductModel struct { + *defaultProductModel + } +) + +// NewProductModel returns a model for the database table. +func NewProductModel(conn sqlx.SqlConn, c cache.CacheConf) ProductModel { + return &customProductModel{ + defaultProductModel: newProductModel(conn, c), + } +} diff --git a/app/main/model/productModel_gen.go b/app/main/model/productModel_gen.go new file mode 100644 index 0000000..f0a9e7a --- /dev/null +++ b/app/main/model/productModel_gen.go @@ -0,0 +1,431 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + productFieldNames = builder.RawFieldNames(&Product{}) + productRows = strings.Join(productFieldNames, ",") + productRowsExpectAutoSet = strings.Join(stringx.Remove(productFieldNames, "`create_time`", "`update_time`"), ",") + productRowsWithPlaceHolder = strings.Join(stringx.Remove(productFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncProductIdPrefix = "cache:qnc:product:id:" + cacheQncProductProductEnPrefix = "cache:qnc:product:productEn:" +) + +type ( + productModel interface { + Insert(ctx context.Context, session sqlx.Session, data *Product) (sql.Result, error) + FindOne(ctx context.Context, id string) (*Product, error) + FindOneByProductEn(ctx context.Context, productEn string) (*Product, error) + Update(ctx context.Context, session sqlx.Session, data *Product) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *Product) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *Product) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*Product, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Product, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Product, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Product, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Product, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultProductModel struct { + sqlc.CachedConn + table string + } + + Product struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态 + Version int64 `db:"version"` // 版本号 + ProductName string `db:"product_name"` // 服务名 + ProductEn string `db:"product_en"` // 英文名 + Description string `db:"description"` // 描述 + Notes sql.NullString `db:"notes"` // 备注 + CostPrice float64 `db:"cost_price"` // 成本 + SellPrice float64 `db:"sell_price"` // 售价 + } +) + +func newProductModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultProductModel { + return &defaultProductModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`product`", + } +} + +func (m *defaultProductModel) Insert(ctx context.Context, session sqlx.Session, data *Product) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncProductIdKey := fmt.Sprintf("%s%v", cacheQncProductIdPrefix, data.Id) + qncProductProductEnKey := fmt.Sprintf("%s%v", cacheQncProductProductEnPrefix, data.ProductEn) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, productRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ProductName, data.ProductEn, data.Description, data.Notes, data.CostPrice, data.SellPrice) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ProductName, data.ProductEn, data.Description, data.Notes, data.CostPrice, data.SellPrice) + }, qncProductIdKey, qncProductProductEnKey) +} +func (m *defaultProductModel) insertUUID(data *Product) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultProductModel) FindOne(ctx context.Context, id string) (*Product, error) { + qncProductIdKey := fmt.Sprintf("%s%v", cacheQncProductIdPrefix, id) + var resp Product + err := m.QueryRowCtx(ctx, &resp, qncProductIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", productRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultProductModel) FindOneByProductEn(ctx context.Context, productEn string) (*Product, error) { + qncProductProductEnKey := fmt.Sprintf("%s%v", cacheQncProductProductEnPrefix, productEn) + var resp Product + err := m.QueryRowIndexCtx(ctx, &resp, qncProductProductEnKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `product_en` = ? and del_state = ? limit 1", productRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, productEn, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultProductModel) Update(ctx context.Context, session sqlx.Session, newData *Product) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncProductIdKey := fmt.Sprintf("%s%v", cacheQncProductIdPrefix, data.Id) + qncProductProductEnKey := fmt.Sprintf("%s%v", cacheQncProductProductEnPrefix, data.ProductEn) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, productRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ProductName, newData.ProductEn, newData.Description, newData.Notes, newData.CostPrice, newData.SellPrice, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ProductName, newData.ProductEn, newData.Description, newData.Notes, newData.CostPrice, newData.SellPrice, newData.Id) + }, qncProductIdKey, qncProductProductEnKey) +} + +func (m *defaultProductModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *Product) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncProductIdKey := fmt.Sprintf("%s%v", cacheQncProductIdPrefix, data.Id) + qncProductProductEnKey := fmt.Sprintf("%s%v", cacheQncProductProductEnPrefix, data.ProductEn) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, productRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ProductName, newData.ProductEn, newData.Description, newData.Notes, newData.CostPrice, newData.SellPrice, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ProductName, newData.ProductEn, newData.Description, newData.Notes, newData.CostPrice, newData.SellPrice, newData.Id, oldVersion) + }, qncProductIdKey, qncProductProductEnKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultProductModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *Product) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "ProductModel delete err : %+v", err) + } + return nil +} + +func (m *defaultProductModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultProductModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultProductModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*Product, error) { + + builder = builder.Columns(productRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*Product + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultProductModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Product, error) { + + builder = builder.Columns(productRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Product + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultProductModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Product, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(productRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*Product + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultProductModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Product, error) { + + builder = builder.Columns(productRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Product + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultProductModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Product, error) { + + builder = builder.Columns(productRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Product + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultProductModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultProductModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultProductModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncProductIdKey := fmt.Sprintf("%s%v", cacheQncProductIdPrefix, id) + qncProductProductEnKey := fmt.Sprintf("%s%v", cacheQncProductProductEnPrefix, data.ProductEn) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncProductIdKey, qncProductProductEnKey) + return err +} +func (m *defaultProductModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncProductIdPrefix, primary) +} +func (m *defaultProductModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", productRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultProductModel) tableName() string { + return m.table +} diff --git a/app/main/model/queryCleanupConfigModel.go b/app/main/model/queryCleanupConfigModel.go new file mode 100644 index 0000000..1c69371 --- /dev/null +++ b/app/main/model/queryCleanupConfigModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ QueryCleanupConfigModel = (*customQueryCleanupConfigModel)(nil) + +type ( + // QueryCleanupConfigModel is an interface to be customized, add more methods here, + // and implement the added methods in customQueryCleanupConfigModel. + QueryCleanupConfigModel interface { + queryCleanupConfigModel + } + + customQueryCleanupConfigModel struct { + *defaultQueryCleanupConfigModel + } +) + +// NewQueryCleanupConfigModel returns a model for the database table. +func NewQueryCleanupConfigModel(conn sqlx.SqlConn, c cache.CacheConf) QueryCleanupConfigModel { + return &customQueryCleanupConfigModel{ + defaultQueryCleanupConfigModel: newQueryCleanupConfigModel(conn, c), + } +} diff --git a/app/main/model/queryCleanupConfigModel_gen.go b/app/main/model/queryCleanupConfigModel_gen.go new file mode 100644 index 0000000..19f8bda --- /dev/null +++ b/app/main/model/queryCleanupConfigModel_gen.go @@ -0,0 +1,429 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + queryCleanupConfigFieldNames = builder.RawFieldNames(&QueryCleanupConfig{}) + queryCleanupConfigRows = strings.Join(queryCleanupConfigFieldNames, ",") + queryCleanupConfigRowsExpectAutoSet = strings.Join(stringx.Remove(queryCleanupConfigFieldNames, "`create_time`", "`update_time`"), ",") + queryCleanupConfigRowsWithPlaceHolder = strings.Join(stringx.Remove(queryCleanupConfigFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncQueryCleanupConfigIdPrefix = "cache:qnc:queryCleanupConfig:id:" + cacheQncQueryCleanupConfigConfigKeyPrefix = "cache:qnc:queryCleanupConfig:configKey:" +) + +type ( + queryCleanupConfigModel interface { + Insert(ctx context.Context, session sqlx.Session, data *QueryCleanupConfig) (sql.Result, error) + FindOne(ctx context.Context, id string) (*QueryCleanupConfig, error) + FindOneByConfigKey(ctx context.Context, configKey string) (*QueryCleanupConfig, error) + Update(ctx context.Context, session sqlx.Session, data *QueryCleanupConfig) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *QueryCleanupConfig) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *QueryCleanupConfig) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*QueryCleanupConfig, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupConfig, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupConfig, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*QueryCleanupConfig, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*QueryCleanupConfig, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultQueryCleanupConfigModel struct { + sqlc.CachedConn + table string + } + + QueryCleanupConfig struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0-未删除，1-已删除 + Version int64 `db:"version"` // 版本号 + ConfigKey string `db:"config_key"` // 配置键 + ConfigValue string `db:"config_value"` // 配置值 + ConfigDesc string `db:"config_desc"` // 配置说明 + Status int64 `db:"status"` // 状态：1-启用，2-禁用 + } +) + +func newQueryCleanupConfigModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultQueryCleanupConfigModel { + return &defaultQueryCleanupConfigModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`query_cleanup_config`", + } +} + +func (m *defaultQueryCleanupConfigModel) Insert(ctx context.Context, session sqlx.Session, data *QueryCleanupConfig) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncQueryCleanupConfigConfigKeyKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupConfigConfigKeyPrefix, data.ConfigKey) + qncQueryCleanupConfigIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupConfigIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?)", m.table, queryCleanupConfigRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ConfigKey, data.ConfigValue, data.ConfigDesc, data.Status) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.ConfigKey, data.ConfigValue, data.ConfigDesc, data.Status) + }, qncQueryCleanupConfigConfigKeyKey, qncQueryCleanupConfigIdKey) +} +func (m *defaultQueryCleanupConfigModel) insertUUID(data *QueryCleanupConfig) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultQueryCleanupConfigModel) FindOne(ctx context.Context, id string) (*QueryCleanupConfig, error) { + qncQueryCleanupConfigIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupConfigIdPrefix, id) + var resp QueryCleanupConfig + err := m.QueryRowCtx(ctx, &resp, qncQueryCleanupConfigIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", queryCleanupConfigRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultQueryCleanupConfigModel) FindOneByConfigKey(ctx context.Context, configKey string) (*QueryCleanupConfig, error) { + qncQueryCleanupConfigConfigKeyKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupConfigConfigKeyPrefix, configKey) + var resp QueryCleanupConfig + err := m.QueryRowIndexCtx(ctx, &resp, qncQueryCleanupConfigConfigKeyKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `config_key` = ? and del_state = ? limit 1", queryCleanupConfigRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, configKey, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultQueryCleanupConfigModel) Update(ctx context.Context, session sqlx.Session, newData *QueryCleanupConfig) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncQueryCleanupConfigConfigKeyKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupConfigConfigKeyPrefix, data.ConfigKey) + qncQueryCleanupConfigIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupConfigIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, queryCleanupConfigRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ConfigKey, newData.ConfigValue, newData.ConfigDesc, newData.Status, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ConfigKey, newData.ConfigValue, newData.ConfigDesc, newData.Status, newData.Id) + }, qncQueryCleanupConfigConfigKeyKey, qncQueryCleanupConfigIdKey) +} + +func (m *defaultQueryCleanupConfigModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *QueryCleanupConfig) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncQueryCleanupConfigConfigKeyKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupConfigConfigKeyPrefix, data.ConfigKey) + qncQueryCleanupConfigIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupConfigIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, queryCleanupConfigRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ConfigKey, newData.ConfigValue, newData.ConfigDesc, newData.Status, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.ConfigKey, newData.ConfigValue, newData.ConfigDesc, newData.Status, newData.Id, oldVersion) + }, qncQueryCleanupConfigConfigKeyKey, qncQueryCleanupConfigIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultQueryCleanupConfigModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *QueryCleanupConfig) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "QueryCleanupConfigModel delete err : %+v", err) + } + return nil +} + +func (m *defaultQueryCleanupConfigModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultQueryCleanupConfigModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultQueryCleanupConfigModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*QueryCleanupConfig, error) { + + builder = builder.Columns(queryCleanupConfigRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupConfigModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupConfig, error) { + + builder = builder.Columns(queryCleanupConfigRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupConfigModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupConfig, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(queryCleanupConfigRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*QueryCleanupConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultQueryCleanupConfigModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*QueryCleanupConfig, error) { + + builder = builder.Columns(queryCleanupConfigRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupConfigModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*QueryCleanupConfig, error) { + + builder = builder.Columns(queryCleanupConfigRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupConfig + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupConfigModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultQueryCleanupConfigModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultQueryCleanupConfigModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncQueryCleanupConfigConfigKeyKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupConfigConfigKeyPrefix, data.ConfigKey) + qncQueryCleanupConfigIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupConfigIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncQueryCleanupConfigConfigKeyKey, qncQueryCleanupConfigIdKey) + return err +} +func (m *defaultQueryCleanupConfigModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncQueryCleanupConfigIdPrefix, primary) +} +func (m *defaultQueryCleanupConfigModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", queryCleanupConfigRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultQueryCleanupConfigModel) tableName() string { + return m.table +} diff --git a/app/main/model/queryCleanupDetailModel.go b/app/main/model/queryCleanupDetailModel.go new file mode 100644 index 0000000..52de046 --- /dev/null +++ b/app/main/model/queryCleanupDetailModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ QueryCleanupDetailModel = (*customQueryCleanupDetailModel)(nil) + +type ( + // QueryCleanupDetailModel is an interface to be customized, add more methods here, + // and implement the added methods in customQueryCleanupDetailModel. + QueryCleanupDetailModel interface { + queryCleanupDetailModel + } + + customQueryCleanupDetailModel struct { + *defaultQueryCleanupDetailModel + } +) + +// NewQueryCleanupDetailModel returns a model for the database table. +func NewQueryCleanupDetailModel(conn sqlx.SqlConn, c cache.CacheConf) QueryCleanupDetailModel { + return &customQueryCleanupDetailModel{ + defaultQueryCleanupDetailModel: newQueryCleanupDetailModel(conn, c), + } +} diff --git a/app/main/model/queryCleanupDetailModel_gen.go b/app/main/model/queryCleanupDetailModel_gen.go new file mode 100644 index 0000000..d71abab --- /dev/null +++ b/app/main/model/queryCleanupDetailModel_gen.go @@ -0,0 +1,393 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + queryCleanupDetailFieldNames = builder.RawFieldNames(&QueryCleanupDetail{}) + queryCleanupDetailRows = strings.Join(queryCleanupDetailFieldNames, ",") + queryCleanupDetailRowsExpectAutoSet = strings.Join(stringx.Remove(queryCleanupDetailFieldNames, "`create_time`", "`update_time`"), ",") + queryCleanupDetailRowsWithPlaceHolder = strings.Join(stringx.Remove(queryCleanupDetailFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncQueryCleanupDetailIdPrefix = "cache:qnc:queryCleanupDetail:id:" +) + +type ( + queryCleanupDetailModel interface { + Insert(ctx context.Context, session sqlx.Session, data *QueryCleanupDetail) (sql.Result, error) + FindOne(ctx context.Context, id string) (*QueryCleanupDetail, error) + Update(ctx context.Context, session sqlx.Session, data *QueryCleanupDetail) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *QueryCleanupDetail) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *QueryCleanupDetail) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*QueryCleanupDetail, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupDetail, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupDetail, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*QueryCleanupDetail, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*QueryCleanupDetail, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultQueryCleanupDetailModel struct { + sqlc.CachedConn + table string + } + + QueryCleanupDetail struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0-未删除，1-已删除 + Version int64 `db:"version"` // 版本号 + CleanupLogId string `db:"cleanup_log_id"` + QueryId string `db:"query_id"` + OrderId string `db:"order_id"` + UserId string `db:"user_id"` + ProductId string `db:"product_id"` + QueryState string `db:"query_state"` // 查询状态 + CreateTimeOld time.Time `db:"create_time_old"` // 原记录创建时间 + } +) + +func newQueryCleanupDetailModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultQueryCleanupDetailModel { + return &defaultQueryCleanupDetailModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`query_cleanup_detail`", + } +} + +func (m *defaultQueryCleanupDetailModel) Insert(ctx context.Context, session sqlx.Session, data *QueryCleanupDetail) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncQueryCleanupDetailIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupDetailIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, queryCleanupDetailRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.CleanupLogId, data.QueryId, data.OrderId, data.UserId, data.ProductId, data.QueryState, data.CreateTimeOld) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.CleanupLogId, data.QueryId, data.OrderId, data.UserId, data.ProductId, data.QueryState, data.CreateTimeOld) + }, qncQueryCleanupDetailIdKey) +} +func (m *defaultQueryCleanupDetailModel) insertUUID(data *QueryCleanupDetail) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultQueryCleanupDetailModel) FindOne(ctx context.Context, id string) (*QueryCleanupDetail, error) { + qncQueryCleanupDetailIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupDetailIdPrefix, id) + var resp QueryCleanupDetail + err := m.QueryRowCtx(ctx, &resp, qncQueryCleanupDetailIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", queryCleanupDetailRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultQueryCleanupDetailModel) Update(ctx context.Context, session sqlx.Session, data *QueryCleanupDetail) (sql.Result, error) { + qncQueryCleanupDetailIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupDetailIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, queryCleanupDetailRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.CleanupLogId, data.QueryId, data.OrderId, data.UserId, data.ProductId, data.QueryState, data.CreateTimeOld, data.Id) + } + return conn.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.CleanupLogId, data.QueryId, data.OrderId, data.UserId, data.ProductId, data.QueryState, data.CreateTimeOld, data.Id) + }, qncQueryCleanupDetailIdKey) +} + +func (m *defaultQueryCleanupDetailModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *QueryCleanupDetail) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncQueryCleanupDetailIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupDetailIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, queryCleanupDetailRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.CleanupLogId, data.QueryId, data.OrderId, data.UserId, data.ProductId, data.QueryState, data.CreateTimeOld, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.CleanupLogId, data.QueryId, data.OrderId, data.UserId, data.ProductId, data.QueryState, data.CreateTimeOld, data.Id, oldVersion) + }, qncQueryCleanupDetailIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultQueryCleanupDetailModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *QueryCleanupDetail) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "QueryCleanupDetailModel delete err : %+v", err) + } + return nil +} + +func (m *defaultQueryCleanupDetailModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultQueryCleanupDetailModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultQueryCleanupDetailModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*QueryCleanupDetail, error) { + + builder = builder.Columns(queryCleanupDetailRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupDetail + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupDetailModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupDetail, error) { + + builder = builder.Columns(queryCleanupDetailRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupDetail + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupDetailModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupDetail, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(queryCleanupDetailRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*QueryCleanupDetail + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultQueryCleanupDetailModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*QueryCleanupDetail, error) { + + builder = builder.Columns(queryCleanupDetailRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupDetail + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupDetailModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*QueryCleanupDetail, error) { + + builder = builder.Columns(queryCleanupDetailRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupDetail + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupDetailModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultQueryCleanupDetailModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultQueryCleanupDetailModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncQueryCleanupDetailIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupDetailIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncQueryCleanupDetailIdKey) + return err +} +func (m *defaultQueryCleanupDetailModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncQueryCleanupDetailIdPrefix, primary) +} +func (m *defaultQueryCleanupDetailModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", queryCleanupDetailRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultQueryCleanupDetailModel) tableName() string { + return m.table +} diff --git a/app/main/model/queryCleanupLogModel.go b/app/main/model/queryCleanupLogModel.go new file mode 100644 index 0000000..b64cb69 --- /dev/null +++ b/app/main/model/queryCleanupLogModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ QueryCleanupLogModel = (*customQueryCleanupLogModel)(nil) + +type ( + // QueryCleanupLogModel is an interface to be customized, add more methods here, + // and implement the added methods in customQueryCleanupLogModel. + QueryCleanupLogModel interface { + queryCleanupLogModel + } + + customQueryCleanupLogModel struct { + *defaultQueryCleanupLogModel + } +) + +// NewQueryCleanupLogModel returns a model for the database table. +func NewQueryCleanupLogModel(conn sqlx.SqlConn, c cache.CacheConf) QueryCleanupLogModel { + return &customQueryCleanupLogModel{ + defaultQueryCleanupLogModel: newQueryCleanupLogModel(conn, c), + } +} diff --git a/app/main/model/queryCleanupLogModel_gen.go b/app/main/model/queryCleanupLogModel_gen.go new file mode 100644 index 0000000..2ba75e4 --- /dev/null +++ b/app/main/model/queryCleanupLogModel_gen.go @@ -0,0 +1,392 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + queryCleanupLogFieldNames = builder.RawFieldNames(&QueryCleanupLog{}) + queryCleanupLogRows = strings.Join(queryCleanupLogFieldNames, ",") + queryCleanupLogRowsExpectAutoSet = strings.Join(stringx.Remove(queryCleanupLogFieldNames, "`create_time`", "`update_time`"), ",") + queryCleanupLogRowsWithPlaceHolder = strings.Join(stringx.Remove(queryCleanupLogFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncQueryCleanupLogIdPrefix = "cache:qnc:queryCleanupLog:id:" +) + +type ( + queryCleanupLogModel interface { + Insert(ctx context.Context, session sqlx.Session, data *QueryCleanupLog) (sql.Result, error) + FindOne(ctx context.Context, id string) (*QueryCleanupLog, error) + Update(ctx context.Context, session sqlx.Session, data *QueryCleanupLog) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *QueryCleanupLog) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *QueryCleanupLog) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*QueryCleanupLog, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupLog, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupLog, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*QueryCleanupLog, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*QueryCleanupLog, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultQueryCleanupLogModel struct { + sqlc.CachedConn + table string + } + + QueryCleanupLog struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` // 删除状态：0-未删除，1-已删除 + Version int64 `db:"version"` // 版本号 + CleanupTime time.Time `db:"cleanup_time"` // 清理执行时间 + CleanupBefore time.Time `db:"cleanup_before"` // 清理截止时间 + AffectedRows int64 `db:"affected_rows"` // 影响行数 + Status int64 `db:"status"` // 状态：1-成功，2-失败 + ErrorMsg sql.NullString `db:"error_msg"` // 错误信息 + Remark sql.NullString `db:"remark"` // 备注说明 + } +) + +func newQueryCleanupLogModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultQueryCleanupLogModel { + return &defaultQueryCleanupLogModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`query_cleanup_log`", + } +} + +func (m *defaultQueryCleanupLogModel) Insert(ctx context.Context, session sqlx.Session, data *QueryCleanupLog) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncQueryCleanupLogIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupLogIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, queryCleanupLogRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.CleanupTime, data.CleanupBefore, data.AffectedRows, data.Status, data.ErrorMsg, data.Remark) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.CleanupTime, data.CleanupBefore, data.AffectedRows, data.Status, data.ErrorMsg, data.Remark) + }, qncQueryCleanupLogIdKey) +} +func (m *defaultQueryCleanupLogModel) insertUUID(data *QueryCleanupLog) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultQueryCleanupLogModel) FindOne(ctx context.Context, id string) (*QueryCleanupLog, error) { + qncQueryCleanupLogIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupLogIdPrefix, id) + var resp QueryCleanupLog + err := m.QueryRowCtx(ctx, &resp, qncQueryCleanupLogIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", queryCleanupLogRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultQueryCleanupLogModel) Update(ctx context.Context, session sqlx.Session, data *QueryCleanupLog) (sql.Result, error) { + qncQueryCleanupLogIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupLogIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, queryCleanupLogRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.CleanupTime, data.CleanupBefore, data.AffectedRows, data.Status, data.ErrorMsg, data.Remark, data.Id) + } + return conn.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.CleanupTime, data.CleanupBefore, data.AffectedRows, data.Status, data.ErrorMsg, data.Remark, data.Id) + }, qncQueryCleanupLogIdKey) +} + +func (m *defaultQueryCleanupLogModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, data *QueryCleanupLog) error { + + oldVersion := data.Version + data.Version += 1 + + var sqlResult sql.Result + var err error + + qncQueryCleanupLogIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupLogIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, queryCleanupLogRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.CleanupTime, data.CleanupBefore, data.AffectedRows, data.Status, data.ErrorMsg, data.Remark, data.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, data.DeleteTime, data.DelState, data.Version, data.CleanupTime, data.CleanupBefore, data.AffectedRows, data.Status, data.ErrorMsg, data.Remark, data.Id, oldVersion) + }, qncQueryCleanupLogIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultQueryCleanupLogModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *QueryCleanupLog) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "QueryCleanupLogModel delete err : %+v", err) + } + return nil +} + +func (m *defaultQueryCleanupLogModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultQueryCleanupLogModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultQueryCleanupLogModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*QueryCleanupLog, error) { + + builder = builder.Columns(queryCleanupLogRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupLog + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupLogModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupLog, error) { + + builder = builder.Columns(queryCleanupLogRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupLog + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupLogModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*QueryCleanupLog, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(queryCleanupLogRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*QueryCleanupLog + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultQueryCleanupLogModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*QueryCleanupLog, error) { + + builder = builder.Columns(queryCleanupLogRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupLog + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupLogModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*QueryCleanupLog, error) { + + builder = builder.Columns(queryCleanupLogRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*QueryCleanupLog + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryCleanupLogModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultQueryCleanupLogModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultQueryCleanupLogModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + qncQueryCleanupLogIdKey := fmt.Sprintf("%s%v", cacheQncQueryCleanupLogIdPrefix, id) + _, err := m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncQueryCleanupLogIdKey) + return err +} +func (m *defaultQueryCleanupLogModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncQueryCleanupLogIdPrefix, primary) +} +func (m *defaultQueryCleanupLogModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", queryCleanupLogRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultQueryCleanupLogModel) tableName() string { + return m.table +} diff --git a/app/main/model/queryModel.go b/app/main/model/queryModel.go new file mode 100644 index 0000000..1bd9689 --- /dev/null +++ b/app/main/model/queryModel.go @@ -0,0 +1,85 @@ +package model + +import ( + "context" + "database/sql" + "fmt" + "qnc-server/common/globalkey" + "time" + + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ QueryModel = (*customQueryModel)(nil) + +type ( + // QueryModel is an interface to be customized, add more methods here, + // and implement the added methods in customQueryModel. + QueryModel interface { + queryModel + DeleteBefore(ctx context.Context, before time.Time) (int64, error) + UpdateUserIDWithSession(ctx context.Context, session sqlx.Session, sourceUserID, targetUserID string) error + } + + customQueryModel struct { + *defaultQueryModel + } +) + +// NewQueryModel returns a model for the database table. +func NewQueryModel(conn sqlx.SqlConn, c cache.CacheConf) QueryModel { + return &customQueryModel{ + defaultQueryModel: newQueryModel(conn, c), + } +} + +func (m *customQueryModel) DeleteBefore(ctx context.Context, before time.Time) (int64, error) { + var affected int64 = 0 + + // 使用事务处理批量删除 + err := m.defaultQueryModel.Trans(ctx, func(ctx context.Context, session sqlx.Session) error { + query := fmt.Sprintf("DELETE FROM %s WHERE create_time < ? AND del_state = ?", m.defaultQueryModel.table) + result, err := session.ExecCtx(ctx, query, before.Format("2006-01-02 15:04:05"), globalkey.DelStateNo) + if err != nil { + return err + } + + rows, err := result.RowsAffected() + if err != nil { + return err + } + + affected = rows + return nil + }) + + if err != nil { + return 0, err + } + + return affected, nil +} + +func (m *customQueryModel) UpdateUserIDWithSession(ctx context.Context, session sqlx.Session, sourceUserID, targetUserID string) error { + builder := m.defaultQueryModel.SelectBuilder().Where("user_id = ?", sourceUserID) + rows, err := m.defaultQueryModel.FindAll(ctx, builder, "") + if err != nil { + return err + } + + keys := make([]string, 0, len(rows)*2) + for _, r := range rows { + keys = append(keys, fmt.Sprintf("%s%v", cacheQncQueryIdPrefix, r.Id)) + keys = append(keys, fmt.Sprintf("%s%v", cacheQncQueryOrderIdPrefix, r.OrderId)) + } + + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("UPDATE %s SET user_id = ? WHERE user_id = ?", m.defaultQueryModel.tableName()) + if session != nil { + return session.ExecCtx(ctx, query, targetUserID, sourceUserID) + } + return conn.ExecCtx(ctx, query, targetUserID, sourceUserID) + }, keys...) + return err +} diff --git a/app/main/model/queryModel_gen.go b/app/main/model/queryModel_gen.go new file mode 100644 index 0000000..e1089b1 --- /dev/null +++ b/app/main/model/queryModel_gen.go @@ -0,0 +1,431 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + queryFieldNames = builder.RawFieldNames(&Query{}) + queryRows = strings.Join(queryFieldNames, ",") + queryRowsExpectAutoSet = strings.Join(stringx.Remove(queryFieldNames, "`create_time`", "`update_time`"), ",") + queryRowsWithPlaceHolder = strings.Join(stringx.Remove(queryFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncQueryIdPrefix = "cache:qnc:query:id:" + cacheQncQueryOrderIdPrefix = "cache:qnc:query:orderId:" +) + +type ( + queryModel interface { + Insert(ctx context.Context, session sqlx.Session, data *Query) (sql.Result, error) + FindOne(ctx context.Context, id string) (*Query, error) + FindOneByOrderId(ctx context.Context, orderId string) (*Query, error) + Update(ctx context.Context, session sqlx.Session, data *Query) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *Query) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *Query) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*Query, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Query, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Query, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Query, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Query, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultQueryModel struct { + sqlc.CachedConn + table string + } + + Query struct { + Id string `db:"id"` + OrderId string `db:"order_id"` + UserId string `db:"user_id"` + ProductId string `db:"product_id"` + QueryParams string `db:"query_params"` // 查询params数据 + QueryData sql.NullString `db:"query_data"` // 查询结果数据 + QueryState string `db:"query_state"` // 查询状态 + DelState int64 `db:"del_state"` // 删除状态 + Version int64 `db:"version"` // 版本号 + CreateTime time.Time `db:"create_time"` // 创建时间 + UpdateTime time.Time `db:"update_time"` // 更新时间 + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + } +) + +func newQueryModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultQueryModel { + return &defaultQueryModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`query`", + } +} + +func (m *defaultQueryModel) Insert(ctx context.Context, session sqlx.Session, data *Query) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncQueryIdKey := fmt.Sprintf("%s%v", cacheQncQueryIdPrefix, data.Id) + qncQueryOrderIdKey := fmt.Sprintf("%s%v", cacheQncQueryOrderIdPrefix, data.OrderId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, queryRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.OrderId, data.UserId, data.ProductId, data.QueryParams, data.QueryData, data.QueryState, data.DelState, data.Version, data.DeleteTime) + } + return conn.ExecCtx(ctx, query, data.Id, data.OrderId, data.UserId, data.ProductId, data.QueryParams, data.QueryData, data.QueryState, data.DelState, data.Version, data.DeleteTime) + }, qncQueryIdKey, qncQueryOrderIdKey) +} +func (m *defaultQueryModel) insertUUID(data *Query) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultQueryModel) FindOne(ctx context.Context, id string) (*Query, error) { + qncQueryIdKey := fmt.Sprintf("%s%v", cacheQncQueryIdPrefix, id) + var resp Query + err := m.QueryRowCtx(ctx, &resp, qncQueryIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", queryRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultQueryModel) FindOneByOrderId(ctx context.Context, orderId string) (*Query, error) { + qncQueryOrderIdKey := fmt.Sprintf("%s%v", cacheQncQueryOrderIdPrefix, orderId) + var resp Query + err := m.QueryRowIndexCtx(ctx, &resp, qncQueryOrderIdKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `order_id` = ? and del_state = ? limit 1", queryRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, orderId, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultQueryModel) Update(ctx context.Context, session sqlx.Session, newData *Query) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncQueryIdKey := fmt.Sprintf("%s%v", cacheQncQueryIdPrefix, data.Id) + qncQueryOrderIdKey := fmt.Sprintf("%s%v", cacheQncQueryOrderIdPrefix, data.OrderId) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, queryRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.OrderId, newData.UserId, newData.ProductId, newData.QueryParams, newData.QueryData, newData.QueryState, newData.DelState, newData.Version, newData.DeleteTime, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.OrderId, newData.UserId, newData.ProductId, newData.QueryParams, newData.QueryData, newData.QueryState, newData.DelState, newData.Version, newData.DeleteTime, newData.Id) + }, qncQueryIdKey, qncQueryOrderIdKey) +} + +func (m *defaultQueryModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *Query) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncQueryIdKey := fmt.Sprintf("%s%v", cacheQncQueryIdPrefix, data.Id) + qncQueryOrderIdKey := fmt.Sprintf("%s%v", cacheQncQueryOrderIdPrefix, data.OrderId) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, queryRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.OrderId, newData.UserId, newData.ProductId, newData.QueryParams, newData.QueryData, newData.QueryState, newData.DelState, newData.Version, newData.DeleteTime, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.OrderId, newData.UserId, newData.ProductId, newData.QueryParams, newData.QueryData, newData.QueryState, newData.DelState, newData.Version, newData.DeleteTime, newData.Id, oldVersion) + }, qncQueryIdKey, qncQueryOrderIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultQueryModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *Query) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "QueryModel delete err : %+v", err) + } + return nil +} + +func (m *defaultQueryModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultQueryModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultQueryModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*Query, error) { + + builder = builder.Columns(queryRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*Query + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Query, error) { + + builder = builder.Columns(queryRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Query + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*Query, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(queryRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*Query + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultQueryModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*Query, error) { + + builder = builder.Columns(queryRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Query + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*Query, error) { + + builder = builder.Columns(queryRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*Query + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultQueryModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultQueryModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultQueryModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncQueryIdKey := fmt.Sprintf("%s%v", cacheQncQueryIdPrefix, id) + qncQueryOrderIdKey := fmt.Sprintf("%s%v", cacheQncQueryOrderIdPrefix, data.OrderId) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncQueryIdKey, qncQueryOrderIdKey) + return err +} +func (m *defaultQueryModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncQueryIdPrefix, primary) +} +func (m *defaultQueryModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", queryRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultQueryModel) tableName() string { + return m.table +} diff --git a/app/main/model/userAuthModel.go b/app/main/model/userAuthModel.go new file mode 100644 index 0000000..0812e1d --- /dev/null +++ b/app/main/model/userAuthModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ UserAuthModel = (*customUserAuthModel)(nil) + +type ( + // UserAuthModel is an interface to be customized, add more methods here, + // and implement the added methods in customUserAuthModel. + UserAuthModel interface { + userAuthModel + } + + customUserAuthModel struct { + *defaultUserAuthModel + } +) + +// NewUserAuthModel returns a model for the database table. +func NewUserAuthModel(conn sqlx.SqlConn, c cache.CacheConf) UserAuthModel { + return &customUserAuthModel{ + defaultUserAuthModel: newUserAuthModel(conn, c), + } +} diff --git a/app/main/model/userAuthModel_gen.go b/app/main/model/userAuthModel_gen.go new file mode 100644 index 0000000..a7f3998 --- /dev/null +++ b/app/main/model/userAuthModel_gen.go @@ -0,0 +1,454 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + userAuthFieldNames = builder.RawFieldNames(&UserAuth{}) + userAuthRows = strings.Join(userAuthFieldNames, ",") + userAuthRowsExpectAutoSet = strings.Join(stringx.Remove(userAuthFieldNames, "`create_time`", "`update_time`"), ",") + userAuthRowsWithPlaceHolder = strings.Join(stringx.Remove(userAuthFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncUserAuthIdPrefix = "cache:qnc:userAuth:id:" + cacheQncUserAuthAuthTypeAuthKeyPrefix = "cache:qnc:userAuth:authType:authKey:" + cacheQncUserAuthUserIdAuthTypePrefix = "cache:qnc:userAuth:userId:authType:" +) + +type ( + userAuthModel interface { + Insert(ctx context.Context, session sqlx.Session, data *UserAuth) (sql.Result, error) + FindOne(ctx context.Context, id string) (*UserAuth, error) + FindOneByAuthTypeAuthKey(ctx context.Context, authType string, authKey string) (*UserAuth, error) + FindOneByUserIdAuthType(ctx context.Context, userId string, authType string) (*UserAuth, error) + Update(ctx context.Context, session sqlx.Session, data *UserAuth) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *UserAuth) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *UserAuth) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*UserAuth, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserAuth, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserAuth, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*UserAuth, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*UserAuth, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultUserAuthModel struct { + sqlc.CachedConn + table string + } + + UserAuth struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + UserId string `db:"user_id"` + AuthKey string `db:"auth_key"` // 平台唯一id + AuthType string `db:"auth_type"` // 平台类型 + } +) + +func newUserAuthModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultUserAuthModel { + return &defaultUserAuthModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`user_auth`", + } +} + +func (m *defaultUserAuthModel) Insert(ctx context.Context, session sqlx.Session, data *UserAuth) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncUserAuthAuthTypeAuthKeyKey := fmt.Sprintf("%s%v:%v", cacheQncUserAuthAuthTypeAuthKeyPrefix, data.AuthType, data.AuthKey) + qncUserAuthIdKey := fmt.Sprintf("%s%v", cacheQncUserAuthIdPrefix, data.Id) + qncUserAuthUserIdAuthTypeKey := fmt.Sprintf("%s%v:%v", cacheQncUserAuthUserIdAuthTypePrefix, data.UserId, data.AuthType) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?)", m.table, userAuthRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.UserId, data.AuthKey, data.AuthType) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.UserId, data.AuthKey, data.AuthType) + }, qncUserAuthAuthTypeAuthKeyKey, qncUserAuthIdKey, qncUserAuthUserIdAuthTypeKey) +} +func (m *defaultUserAuthModel) insertUUID(data *UserAuth) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultUserAuthModel) FindOne(ctx context.Context, id string) (*UserAuth, error) { + qncUserAuthIdKey := fmt.Sprintf("%s%v", cacheQncUserAuthIdPrefix, id) + var resp UserAuth + err := m.QueryRowCtx(ctx, &resp, qncUserAuthIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", userAuthRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultUserAuthModel) FindOneByAuthTypeAuthKey(ctx context.Context, authType string, authKey string) (*UserAuth, error) { + qncUserAuthAuthTypeAuthKeyKey := fmt.Sprintf("%s%v:%v", cacheQncUserAuthAuthTypeAuthKeyPrefix, authType, authKey) + var resp UserAuth + err := m.QueryRowIndexCtx(ctx, &resp, qncUserAuthAuthTypeAuthKeyKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `auth_type` = ? and `auth_key` = ? and del_state = ? limit 1", userAuthRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, authType, authKey, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultUserAuthModel) FindOneByUserIdAuthType(ctx context.Context, userId string, authType string) (*UserAuth, error) { + qncUserAuthUserIdAuthTypeKey := fmt.Sprintf("%s%v:%v", cacheQncUserAuthUserIdAuthTypePrefix, userId, authType) + var resp UserAuth + err := m.QueryRowIndexCtx(ctx, &resp, qncUserAuthUserIdAuthTypeKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `user_id` = ? and `auth_type` = ? and del_state = ? limit 1", userAuthRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, userId, authType, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultUserAuthModel) Update(ctx context.Context, session sqlx.Session, newData *UserAuth) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncUserAuthAuthTypeAuthKeyKey := fmt.Sprintf("%s%v:%v", cacheQncUserAuthAuthTypeAuthKeyPrefix, data.AuthType, data.AuthKey) + qncUserAuthIdKey := fmt.Sprintf("%s%v", cacheQncUserAuthIdPrefix, data.Id) + qncUserAuthUserIdAuthTypeKey := fmt.Sprintf("%s%v:%v", cacheQncUserAuthUserIdAuthTypePrefix, data.UserId, data.AuthType) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, userAuthRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.UserId, newData.AuthKey, newData.AuthType, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.UserId, newData.AuthKey, newData.AuthType, newData.Id) + }, qncUserAuthAuthTypeAuthKeyKey, qncUserAuthIdKey, qncUserAuthUserIdAuthTypeKey) +} + +func (m *defaultUserAuthModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *UserAuth) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncUserAuthAuthTypeAuthKeyKey := fmt.Sprintf("%s%v:%v", cacheQncUserAuthAuthTypeAuthKeyPrefix, data.AuthType, data.AuthKey) + qncUserAuthIdKey := fmt.Sprintf("%s%v", cacheQncUserAuthIdPrefix, data.Id) + qncUserAuthUserIdAuthTypeKey := fmt.Sprintf("%s%v:%v", cacheQncUserAuthUserIdAuthTypePrefix, data.UserId, data.AuthType) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, userAuthRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.UserId, newData.AuthKey, newData.AuthType, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.UserId, newData.AuthKey, newData.AuthType, newData.Id, oldVersion) + }, qncUserAuthAuthTypeAuthKeyKey, qncUserAuthIdKey, qncUserAuthUserIdAuthTypeKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultUserAuthModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *UserAuth) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "UserAuthModel delete err : %+v", err) + } + return nil +} + +func (m *defaultUserAuthModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultUserAuthModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultUserAuthModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*UserAuth, error) { + + builder = builder.Columns(userAuthRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserAuth + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserAuthModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserAuth, error) { + + builder = builder.Columns(userAuthRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserAuth + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserAuthModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserAuth, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(userAuthRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*UserAuth + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultUserAuthModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*UserAuth, error) { + + builder = builder.Columns(userAuthRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserAuth + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserAuthModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*UserAuth, error) { + + builder = builder.Columns(userAuthRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserAuth + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserAuthModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultUserAuthModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultUserAuthModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncUserAuthAuthTypeAuthKeyKey := fmt.Sprintf("%s%v:%v", cacheQncUserAuthAuthTypeAuthKeyPrefix, data.AuthType, data.AuthKey) + qncUserAuthIdKey := fmt.Sprintf("%s%v", cacheQncUserAuthIdPrefix, id) + qncUserAuthUserIdAuthTypeKey := fmt.Sprintf("%s%v:%v", cacheQncUserAuthUserIdAuthTypePrefix, data.UserId, data.AuthType) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncUserAuthAuthTypeAuthKeyKey, qncUserAuthIdKey, qncUserAuthUserIdAuthTypeKey) + return err +} +func (m *defaultUserAuthModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncUserAuthIdPrefix, primary) +} +func (m *defaultUserAuthModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", userAuthRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultUserAuthModel) tableName() string { + return m.table +} diff --git a/app/main/model/userModel.go b/app/main/model/userModel.go new file mode 100644 index 0000000..8123712 --- /dev/null +++ b/app/main/model/userModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ UserModel = (*customUserModel)(nil) + +type ( + // UserModel is an interface to be customized, add more methods here, + // and implement the added methods in customUserModel. + UserModel interface { + userModel + } + + customUserModel struct { + *defaultUserModel + } +) + +// NewUserModel returns a model for the database table. +func NewUserModel(conn sqlx.SqlConn, c cache.CacheConf) UserModel { + return &customUserModel{ + defaultUserModel: newUserModel(conn, c), + } +} diff --git a/app/main/model/userModel_gen.go b/app/main/model/userModel_gen.go new file mode 100644 index 0000000..cd638a8 --- /dev/null +++ b/app/main/model/userModel_gen.go @@ -0,0 +1,430 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "reflect" + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/google/uuid" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + userFieldNames = builder.RawFieldNames(&User{}) + userRows = strings.Join(userFieldNames, ",") + userRowsExpectAutoSet = strings.Join(stringx.Remove(userFieldNames, "`create_time`", "`update_time`"), ",") + userRowsWithPlaceHolder = strings.Join(stringx.Remove(userFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheQncUserIdPrefix = "cache:qnc:user:id:" + cacheQncUserMobilePrefix = "cache:qnc:user:mobile:" +) + +type ( + userModel interface { + Insert(ctx context.Context, session sqlx.Session, data *User) (sql.Result, error) + FindOne(ctx context.Context, id string) (*User, error) + FindOneByMobile(ctx context.Context, mobile sql.NullString) (*User, error) + Update(ctx context.Context, session sqlx.Session, data *User) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *User) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *User) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*User, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*User, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*User, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*User, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*User, error) + Delete(ctx context.Context, session sqlx.Session, id string) error + } + + defaultUserModel struct { + sqlc.CachedConn + table string + } + + User struct { + Id string `db:"id"` + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + Mobile sql.NullString `db:"mobile"` + Password sql.NullString `db:"password"` + Nickname sql.NullString `db:"nickname"` + Info string `db:"info"` + Inside int64 `db:"inside"` + } +) + +func newUserModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultUserModel { + return &defaultUserModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`user`", + } +} + +func (m *defaultUserModel) Insert(ctx context.Context, session sqlx.Session, data *User) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + qncUserIdKey := fmt.Sprintf("%s%v", cacheQncUserIdPrefix, data.Id) + qncUserMobileKey := fmt.Sprintf("%s%v", cacheQncUserMobilePrefix, data.Mobile) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?, ?, ?, ?, ?)", m.table, userRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.Mobile, data.Password, data.Nickname, data.Info, data.Inside) + } + return conn.ExecCtx(ctx, query, data.Id, data.DeleteTime, data.DelState, data.Version, data.Mobile, data.Password, data.Nickname, data.Info, data.Inside) + }, qncUserIdKey, qncUserMobileKey) +} +func (m *defaultUserModel) insertUUID(data *User) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} + +func (m *defaultUserModel) FindOne(ctx context.Context, id string) (*User, error) { + qncUserIdKey := fmt.Sprintf("%s%v", cacheQncUserIdPrefix, id) + var resp User + err := m.QueryRowCtx(ctx, &resp, qncUserIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", userRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultUserModel) FindOneByMobile(ctx context.Context, mobile sql.NullString) (*User, error) { + qncUserMobileKey := fmt.Sprintf("%s%v", cacheQncUserMobilePrefix, mobile) + var resp User + err := m.QueryRowIndexCtx(ctx, &resp, qncUserMobileKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `mobile` = ? and del_state = ? limit 1", userRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, mobile, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultUserModel) Update(ctx context.Context, session sqlx.Session, newData *User) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + qncUserIdKey := fmt.Sprintf("%s%v", cacheQncUserIdPrefix, data.Id) + qncUserMobileKey := fmt.Sprintf("%s%v", cacheQncUserMobilePrefix, data.Mobile) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, userRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Mobile, newData.Password, newData.Nickname, newData.Info, newData.Inside, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Mobile, newData.Password, newData.Nickname, newData.Info, newData.Inside, newData.Id) + }, qncUserIdKey, qncUserMobileKey) +} + +func (m *defaultUserModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *User) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + qncUserIdKey := fmt.Sprintf("%s%v", cacheQncUserIdPrefix, data.Id) + qncUserMobileKey := fmt.Sprintf("%s%v", cacheQncUserMobilePrefix, data.Mobile) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, userRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Mobile, newData.Password, newData.Nickname, newData.Info, newData.Inside, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.DeleteTime, newData.DelState, newData.Version, newData.Mobile, newData.Password, newData.Nickname, newData.Info, newData.Inside, newData.Id, oldVersion) + }, qncUserIdKey, qncUserMobileKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultUserModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *User) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "UserModel delete err : %+v", err) + } + return nil +} + +func (m *defaultUserModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultUserModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultUserModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*User, error) { + + builder = builder.Columns(userRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*User + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*User, error) { + + builder = builder.Columns(userRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*User + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*User, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(userRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*User + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultUserModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*User, error) { + + builder = builder.Columns(userRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*User + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*User, error) { + + builder = builder.Columns(userRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*User + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultUserModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultUserModel) Delete(ctx context.Context, session sqlx.Session, id string) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + qncUserIdKey := fmt.Sprintf("%s%v", cacheQncUserIdPrefix, id) + qncUserMobileKey := fmt.Sprintf("%s%v", cacheQncUserMobilePrefix, data.Mobile) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, qncUserIdKey, qncUserMobileKey) + return err +} +func (m *defaultUserModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheQncUserIdPrefix, primary) +} +func (m *defaultUserModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", userRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultUserModel) tableName() string { + return m.table +} diff --git a/app/main/model/userTempModel.go b/app/main/model/userTempModel.go new file mode 100644 index 0000000..bcb7978 --- /dev/null +++ b/app/main/model/userTempModel.go @@ -0,0 +1,27 @@ +package model + +import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var _ UserTempModel = (*customUserTempModel)(nil) + +type ( + // UserTempModel is an interface to be customized, add more methods here, + // and implement the added methods in customUserTempModel. + UserTempModel interface { + userTempModel + } + + customUserTempModel struct { + *defaultUserTempModel + } +) + +// NewUserTempModel returns a model for the database table. +func NewUserTempModel(conn sqlx.SqlConn, c cache.CacheConf) UserTempModel { + return &customUserTempModel{ + defaultUserTempModel: newUserTempModel(conn, c), + } +} diff --git a/app/main/model/userTempModel_gen.go b/app/main/model/userTempModel_gen.go new file mode 100644 index 0000000..e66563b --- /dev/null +++ b/app/main/model/userTempModel_gen.go @@ -0,0 +1,408 @@ +// Code generated by goctl. DO NOT EDIT! + +package model + +import ( + "context" + "database/sql" + "fmt" + "strings" + + "time" + + "qnc-server/common/globalkey" + + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" +) + +var ( + userTempFieldNames = builder.RawFieldNames(&UserTemp{}) + userTempRows = strings.Join(userTempFieldNames, ",") + userTempRowsExpectAutoSet = strings.Join(stringx.Remove(userTempFieldNames, "`id`", "`create_time`", "`update_time`"), ",") + userTempRowsWithPlaceHolder = strings.Join(stringx.Remove(userTempFieldNames, "`id`", "`create_time`", "`update_time`"), "=?,") + "=?" + + cacheHmUserTempIdPrefix = "cache:qnc:userTemp:id:" + cacheHmUserTempAuthTypeAuthKeyPrefix = "cache:qnc:userTemp:authType:authKey:" +) + +type ( + userTempModel interface { + Insert(ctx context.Context, session sqlx.Session, data *UserTemp) (sql.Result, error) + FindOne(ctx context.Context, id int64) (*UserTemp, error) + FindOneByAuthTypeAuthKey(ctx context.Context, authType string, authKey string) (*UserTemp, error) + Update(ctx context.Context, session sqlx.Session, data *UserTemp) (sql.Result, error) + UpdateWithVersion(ctx context.Context, session sqlx.Session, data *UserTemp) error + Trans(ctx context.Context, fn func(context context.Context, session sqlx.Session) error) error + SelectBuilder() squirrel.SelectBuilder + DeleteSoft(ctx context.Context, session sqlx.Session, data *UserTemp) error + FindSum(ctx context.Context, sumBuilder squirrel.SelectBuilder, field string) (float64, error) + FindCount(ctx context.Context, countBuilder squirrel.SelectBuilder, field string) (int64, error) + FindAll(ctx context.Context, rowBuilder squirrel.SelectBuilder, orderBy string) ([]*UserTemp, error) + FindPageListByPage(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserTemp, error) + FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserTemp, int64, error) + FindPageListByIdDESC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*UserTemp, error) + FindPageListByIdASC(ctx context.Context, rowBuilder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*UserTemp, error) + Delete(ctx context.Context, session sqlx.Session, id int64) error + } + + defaultUserTempModel struct { + sqlc.CachedConn + table string + } + + UserTemp struct { + Id int64 `db:"id"` + AuthKey string `db:"auth_key"` // 平台唯一id + AuthType string `db:"auth_type"` // 平台类型 + CreateTime time.Time `db:"create_time"` + UpdateTime time.Time `db:"update_time"` + DeleteTime sql.NullTime `db:"delete_time"` // 删除时间 + DelState int64 `db:"del_state"` + Version int64 `db:"version"` // 版本号 + } +) + +func newUserTempModel(conn sqlx.SqlConn, c cache.CacheConf) *defaultUserTempModel { + return &defaultUserTempModel{ + CachedConn: sqlc.NewConn(conn, c), + table: "`user_temp`", + } +} + +func (m *defaultUserTempModel) Insert(ctx context.Context, session sqlx.Session, data *UserTemp) (sql.Result, error) { + data.DelState = globalkey.DelStateNo + hmUserTempAuthTypeAuthKeyKey := fmt.Sprintf("%s%v:%v", cacheHmUserTempAuthTypeAuthKeyPrefix, data.AuthType, data.AuthKey) + hmUserTempIdKey := fmt.Sprintf("%s%v", cacheHmUserTempIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values (?, ?, ?, ?, ?)", m.table, userTempRowsExpectAutoSet) + if session != nil { + return session.ExecCtx(ctx, query, data.AuthKey, data.AuthType, data.DeleteTime, data.DelState, data.Version) + } + return conn.ExecCtx(ctx, query, data.AuthKey, data.AuthType, data.DeleteTime, data.DelState, data.Version) + }, hmUserTempAuthTypeAuthKeyKey, hmUserTempIdKey) +} + +func (m *defaultUserTempModel) FindOne(ctx context.Context, id int64) (*UserTemp, error) { + hmUserTempIdKey := fmt.Sprintf("%s%v", cacheHmUserTempIdPrefix, id) + var resp UserTemp + err := m.QueryRowCtx(ctx, &resp, hmUserTempIdKey, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", userTempRows, m.table) + return conn.QueryRowCtx(ctx, v, query, id, globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultUserTempModel) FindOneByAuthTypeAuthKey(ctx context.Context, authType string, authKey string) (*UserTemp, error) { + hmUserTempAuthTypeAuthKeyKey := fmt.Sprintf("%s%v:%v", cacheHmUserTempAuthTypeAuthKeyPrefix, authType, authKey) + var resp UserTemp + err := m.QueryRowIndexCtx(ctx, &resp, hmUserTempAuthTypeAuthKeyKey, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where `auth_type` = ? and `auth_key` = ? and del_state = ? limit 1", userTempRows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, authType, authKey, globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.Id, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *defaultUserTempModel) Update(ctx context.Context, session sqlx.Session, newData *UserTemp) (sql.Result, error) { + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return nil, err + } + hmUserTempAuthTypeAuthKeyKey := fmt.Sprintf("%s%v:%v", cacheHmUserTempAuthTypeAuthKeyPrefix, data.AuthType, data.AuthKey) + hmUserTempIdKey := fmt.Sprintf("%s%v", cacheHmUserTempIdPrefix, data.Id) + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ?", m.table, userTempRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AuthKey, newData.AuthType, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + } + return conn.ExecCtx(ctx, query, newData.AuthKey, newData.AuthType, newData.DeleteTime, newData.DelState, newData.Version, newData.Id) + }, hmUserTempAuthTypeAuthKeyKey, hmUserTempIdKey) +} + +func (m *defaultUserTempModel) UpdateWithVersion(ctx context.Context, session sqlx.Session, newData *UserTemp) error { + + oldVersion := newData.Version + newData.Version += 1 + + var sqlResult sql.Result + var err error + + data, err := m.FindOne(ctx, newData.Id) + if err != nil { + return err + } + hmUserTempAuthTypeAuthKeyKey := fmt.Sprintf("%s%v:%v", cacheHmUserTempAuthTypeAuthKeyPrefix, data.AuthType, data.AuthKey) + hmUserTempIdKey := fmt.Sprintf("%s%v", cacheHmUserTempIdPrefix, data.Id) + sqlResult, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where `id` = ? and version = ? ", m.table, userTempRowsWithPlaceHolder) + if session != nil { + return session.ExecCtx(ctx, query, newData.AuthKey, newData.AuthType, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + } + return conn.ExecCtx(ctx, query, newData.AuthKey, newData.AuthType, newData.DeleteTime, newData.DelState, newData.Version, newData.Id, oldVersion) + }, hmUserTempAuthTypeAuthKeyKey, hmUserTempIdKey) + if err != nil { + return err + } + updateCount, err := sqlResult.RowsAffected() + if err != nil { + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *defaultUserTempModel) DeleteSoft(ctx context.Context, session sqlx.Session, data *UserTemp) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err := m.UpdateWithVersion(ctx, session, data); err != nil { + return errors.Wrapf(errors.New("delete soft failed "), "UserTempModel delete err : %+v", err) + } + return nil +} + +func (m *defaultUserTempModel) FindSum(ctx context.Context, builder squirrel.SelectBuilder, field string) (float64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultUserTempModel) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64, error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + err = m.QueryRowNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *defaultUserTempModel) FindAll(ctx context.Context, builder squirrel.SelectBuilder, orderBy string) ([]*UserTemp, error) { + + builder = builder.Columns(userTempRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserTemp + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserTempModel) FindPageListByPage(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserTemp, error) { + + builder = builder.Columns(userTempRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserTemp + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserTempModel) FindPageListByPageWithTotal(ctx context.Context, builder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*UserTemp, int64, error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns(userTempRows) + + if orderBy == "" { + builder = builder.OrderBy("id DESC") + } else { + builder = builder.OrderBy(orderBy) + } + + if page < 1 { + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, total, err + } + + var resp []*UserTemp + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, total, nil + default: + return nil, total, err + } +} + +func (m *defaultUserTempModel) FindPageListByIdDESC(ctx context.Context, builder squirrel.SelectBuilder, preMinId, pageSize int64) ([]*UserTemp, error) { + + builder = builder.Columns(userTempRows) + + if preMinId > 0 { + builder = builder.Where(" id < ? ", preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserTemp + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserTempModel) FindPageListByIdASC(ctx context.Context, builder squirrel.SelectBuilder, preMaxId, pageSize int64) ([]*UserTemp, error) { + + builder = builder.Columns(userTempRows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? ", preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*UserTemp + err = m.QueryRowsNoCacheCtx(ctx, &resp, query, values...) + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *defaultUserTempModel) Trans(ctx context.Context, fn func(ctx context.Context, session sqlx.Session) error) error { + + return m.TransactCtx(ctx, func(ctx context.Context, session sqlx.Session) error { + return fn(ctx, session) + }) + +} + +func (m *defaultUserTempModel) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} +func (m *defaultUserTempModel) Delete(ctx context.Context, session sqlx.Session, id int64) error { + data, err := m.FindOne(ctx, id) + if err != nil { + return err + } + + hmUserTempAuthTypeAuthKeyKey := fmt.Sprintf("%s%v:%v", cacheHmUserTempAuthTypeAuthKeyPrefix, data.AuthType, data.AuthKey) + hmUserTempIdKey := fmt.Sprintf("%s%v", cacheHmUserTempIdPrefix, id) + _, err = m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where `id` = ?", m.table) + if session != nil { + return session.ExecCtx(ctx, query, id) + } + return conn.ExecCtx(ctx, query, id) + }, hmUserTempAuthTypeAuthKeyKey, hmUserTempIdKey) + return err +} +func (m *defaultUserTempModel) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", cacheHmUserTempIdPrefix, primary) +} +func (m *defaultUserTempModel) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where `id` = ? and del_state = ? limit 1", userTempRows, m.table) + return conn.QueryRowCtx(ctx, v, query, primary, globalkey.DelStateNo) +} + +func (m *defaultUserTempModel) tableName() string { + return m.table +} diff --git a/app/main/model/vars.go b/app/main/model/vars.go new file mode 100644 index 0000000..fc458e0 --- /dev/null +++ b/app/main/model/vars.go @@ -0,0 +1,110 @@ +package model + +import ( + "errors" + + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var ErrNotFound = sqlx.ErrNotFound +var ErrNoRowsUpdate = errors.New("update db no rows change") + +// 平台 +var PlatformWxMini string = "wxmini" +var PlatformWxH5 string = "wxh5" +var PlatformApp string = "app" +var PlatformH5 string = "h5" +var PlatformAdmin string = "admin" + +// 用户授权类型 +var UserAuthTypeMobile string = "mobile" +var UserAuthTypeWxMiniOpenID string = "wxmini_openid" +var UserAuthTypeWxh5OpenID string = "wxh5_openid" +var UserAuthTypeUUID string = "uuid" + +// 代理扣除类型 +var AgentDeductionTypeCost string = "cost" +var AgentDeductionTypePricing string = "pricing" + +var AgentRewardsTypeDescendantPromotion string = "descendant_promotion" +var AgentRewardsTypeDescendantUpgradeVip string = "descendant_upgrade_vip" +var AgentRewardsTypeDescendantUpgradeSvip string = "descendant_upgrade_svip" +var AgentRewardsTypeDescendantStayActive string = "descendant_stay_active" +var AgentRewardsTypeDescendantNewActive string = "descendant_new_active" +var AgentRewardsTypeDescendantWithdraw string = "descendant_withdraw" + +var AgentLeveNameNormal string = "normal" +var AgentLeveNameVIP string = "VIP" +var AgentLeveNameSVIP string = "SVIP" + +const ( + OrderStatusPending = "pending" + OrderStatusPaid = "paid" + OrderStatusFailed = "failed" + OrderStatusRefunding = "refunding" + OrderStatusRefunded = "refunded" + OrderStatusClosed = "closed" +) +const ( + OrderRefundStatusPending = "pending" + OrderRefundStatusSuccess = "success" + OrderRefundStatusFailed = "failed" + OrderRefundStatusClosed = "closed" +) +const ( + QueryStatePending = "pending" + QueryStateFailed = "failed" + QueryStateSuccess = "success" + QueryStateProcessing = "processing" + QueryStateCleaned = "cleaned" + QueryStateRefunded = "refunded" +) + +const ( + GrantTypeFace string = "face" + AuthorizationGrantTypeSms = "sms" +) +const ( + AuthorizationStatusPending = "pending" + AuthorizationStatusSuccess = "success" + AuthorizationStatusFailed = "failed" + AuthorizationStatusExpired = "expired" + AuthorizationStatusRevoked = "revoked" + AuthorizationStatusRejected = "rejected" +) + +const ( + AuthorizationFaceStatusPending = "pending" + AuthorizationFaceStatusSuccess = "success" + AuthorizationFaceStatusFailed = "failed" +) + +const ( + AgentRealNameStatusPending = "pending" + AgentRealNameStatusApproved = "approved" + AgentRealNameStatusRejected = "rejected" +) + +// 用户身份类型 +const ( + UserTypeTemp = 0 // 临时用户 + UserTypeNormal = 1 // 正式用户 + UserTypeAdmin = 2 // 管理员 +) + +// 管理员角色编码 +const ( + AdminRoleCodeSuper = "SUPER" // 超级管理员 +) + +// 代理状态 +const ( + AgentStatusNo = 0 // 非代理 + AgentStatusYes = 1 // 是代理 +) +const ( + TaxStatusPending = 0 // 待扣税 + TaxStatusSuccess = 1 // 已扣税 + TaxStatusExempt = 2 // 免税 + TaxStatusFailed = 3 // 扣税失败 +) diff --git a/common/ctxdata/ctxData.go b/common/ctxdata/ctxData.go new file mode 100644 index 0000000..f4564e3 --- /dev/null +++ b/common/ctxdata/ctxData.go @@ -0,0 +1,88 @@ +package ctxdata + +import ( + "context" + "qnc-server/app/main/model" + jwtx "qnc-server/common/jwt" + "errors" + "fmt" +) + +const CtxKeyJwtUserId = "userId" + +// 定义错误类型 +var ( + ErrNoInCtx = errors.New("上下文中没有相关数据") + ErrInvalidUserId = errors.New("用户ID格式无效") // 数据异常 +) + +// GetUidFromCtx 从 context 中获取用户 ID（字符串） +func GetUidFromCtx(ctx context.Context) (string, error) { + // 尝试从上下文中获取 jwtUserId + value := ctx.Value(CtxKeyJwtUserId) + if value == nil { + claims, err := GetClaimsFromCtx(ctx) + if err != nil { + return "", err + } + return claims.UserId, nil + } + + // 根据值的类型进行不同处理 + switch v := value.(type) { + case string: + return v, nil + default: + return "", fmt.Errorf("%w: 期望类型 string, 实际类型 %T", ErrInvalidUserId, value) + } +} + +func GetClaimsFromCtx(ctx context.Context) (*jwtx.JwtClaims, error) { + value := ctx.Value(jwtx.ExtraKey) + if value == nil { + return nil, ErrNoInCtx + } + + // 首先尝试直接断言为 *jwtx.JwtClaims + if claims, ok := value.(*jwtx.JwtClaims); ok { + return claims, nil + } + + // 如果直接断言失败，尝试从 map[string]interface{} 中解析 + if claimsMap, ok := value.(map[string]interface{}); ok { + return jwtx.MapToJwtClaims(claimsMap) + } + + return nil, ErrNoInCtx +} + +// IsNoUserIdError 判断是否是未登录错误 +func IsNoUserIdError(err error) bool { + return errors.Is(err, ErrNoInCtx) +} + +// IsInvalidUserIdError 判断是否是用户ID格式错误 +func IsInvalidUserIdError(err error) bool { + return errors.Is(err, ErrInvalidUserId) +} + +// GetPlatformFromCtx 从 context 中获取平台 +func GetPlatformFromCtx(ctx context.Context) (string, error) { + platform, platformOk := ctx.Value("platform").(string) + if !platformOk { + return "", fmt.Errorf("平台不存在: %s", platform) + } + + switch platform { + case model.PlatformWxMini: + return model.PlatformWxMini, nil + case model.PlatformWxH5: + return model.PlatformWxH5, nil + case model.PlatformApp: + return model.PlatformApp, nil + case model.PlatformH5: + return model.PlatformH5, nil + default: + return "", fmt.Errorf("不支持的支付平台: %s", platform) + } +} diff --git a/common/globalkey/constantKey.go b/common/globalkey/constantKey.go new file mode 100644 index 0000000..584938d --- /dev/null +++ b/common/globalkey/constantKey.go @@ -0,0 +1,14 @@ +package globalkey + +/** +global constant key +*/ + +//软删除 +var DelStateNo int64 = 0 //未删除 +var DelStateYes int64 = 1 //已删除 + +//时间格式化模版 +var DateTimeFormatTplStandardDateTime = "Y-m-d H:i:s" +var DateTimeFormatTplStandardDate = "Y-m-d" +var DateTimeFormatTplStandardTime = "H:i:s" diff --git a/common/globalkey/redisCacheKey.go b/common/globalkey/redisCacheKey.go new file mode 100644 index 0000000..9296e19 --- /dev/null +++ b/common/globalkey/redisCacheKey.go @@ -0,0 +1,9 @@ +package globalkey + +/** +redis key except "model cache key" in here, +but "model cache key" in model +*/ + +// CacheUserTokenKey /** 用户登陆的token +const CacheUserTokenKey = "user_token:%d" diff --git a/common/interceptor/rpcserver/loggerInterceptor.go b/common/interceptor/rpcserver/loggerInterceptor.go new file mode 100644 index 0000000..d323839 --- /dev/null +++ b/common/interceptor/rpcserver/loggerInterceptor.go @@ -0,0 +1,39 @@ +package rpcserver + +import ( + "context" + + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "google.golang.org/grpc" + "google.golang.org/grpc/codes" + "google.golang.org/grpc/status" +) + +/** +* @Description rpc service logger interceptor +* @Author Mikael +* @Date 2021/1/9 13:35 +* @Version 1.0 +**/ + +func LoggerInterceptor(ctx context.Context, req interface{}, info *grpc.UnaryServerInfo, handler grpc.UnaryHandler) (resp interface{}, err error) { + + resp, err = handler(ctx, req) + if err != nil { + causeErr := errors.Cause(err) // err类型 + if e, ok := causeErr.(*xerr.CodeError); ok { //自定义错误类型 + logx.WithContext(ctx).Errorf("【RPC-SRV-ERR】 %v", err) + + //转成grpc err + err = status.Error(codes.Code(e.GetErrCode()), e.GetErrMsg()) + } else { + logx.WithContext(ctx).Errorf("【RPC-SRV-ERR】 %v", err) + } + + } + + return resp, err +} diff --git a/common/jwt/jwtx.go b/common/jwt/jwtx.go new file mode 100644 index 0000000..f51c211 --- /dev/null +++ b/common/jwt/jwtx.go @@ -0,0 +1,113 @@ +package jwtx + +import ( + "encoding/json" + "errors" + "time" + + "github.com/golang-jwt/jwt/v4" +) + +const ExtraKey = "extra" + +type JwtClaims struct { + UserId string `json:"userId"` + AgentId string `json:"agentId"` + Platform string `json:"platform"` + // 用户身份类型：0-临时用户，1-正式用户 + UserType int64 `json:"userType"` + // 是否代理：0-否，1-是 + IsAgent int64 `json:"isAgent"` + AuthType string `json:"authType"` + AuthKey string `json:"authKey"` +} + +// MapToJwtClaims 将 map[string]interface{} 转换为 JwtClaims 结构体 +func MapToJwtClaims(claimsMap map[string]interface{}) (*JwtClaims, error) { + // 使用JSON序列化/反序列化的方式自动转换 + jsonData, err := json.Marshal(claimsMap) + if err != nil { + return nil, errors.New("序列化claims失败") + } + + var claims JwtClaims + if err := json.Unmarshal(jsonData, &claims); err != nil { + return nil, errors.New("反序列化claims失败") + } + + return &claims, nil +} + +// GenerateJwtToken 生成JWT token +func GenerateJwtToken(claims JwtClaims, secret string, expire int64) (string, error) { + now := time.Now().Unix() + + // 将 claims 结构体转换为 map[string]interface{} + claimsBytes, err := json.Marshal(claims) + if err != nil { + return "", err + } + + var claimsMap map[string]interface{} + if err := json.Unmarshal(claimsBytes, &claimsMap); err != nil { + return "", err + } + + jwtClaims := jwt.MapClaims{ + "exp": now + expire, + "iat": now, + "userId": claims.UserId, + ExtraKey: claimsMap, + } + token := jwt.NewWithClaims(jwt.SigningMethodHS256, jwtClaims) + return token.SignedString([]byte(secret)) +} + +func ParseJwtToken(tokenStr string, secret string) (*JwtClaims, error) { + token, err := jwt.Parse(tokenStr, func(token *jwt.Token) (interface{}, error) { + return []byte(secret), nil + }) + + if err != nil { + // 检查是否是JWT验证错误 + if validationErr, ok := err.(*jwt.ValidationError); ok { + // 如果是过期错误，返回更明确的错误信息 + if validationErr.Errors&jwt.ValidationErrorExpired != 0 { + return nil, errors.New("token已过期") + } + // 如果是签名错误，返回签名错误信息 + if validationErr.Errors&jwt.ValidationErrorSignatureInvalid != 0 { + return nil, errors.New("token签名无效") + } + // 其他验证错误 + return nil, errors.New("token验证失败") + } + return nil, errors.New("invalid JWT") + } + + if !token.Valid { + return nil, errors.New("token无效") + } + + claims, ok := token.Claims.(jwt.MapClaims) + if !ok { + return nil, errors.New("invalid JWT claims") + } + + extraInfo, exists := claims[ExtraKey] + if !exists { + return nil, errors.New("extra not found in JWT") + } + + // 尝试直接断言为 JwtClaims 结构体 + if jwtClaims, ok := extraInfo.(JwtClaims); ok { + return &jwtClaims, nil + } + + // 尝试从 map[string]interface{} 中解析 + if claimsMap, ok := extraInfo.(map[string]interface{}); ok { + return MapToJwtClaims(claimsMap) + } + + return nil, errors.New("unsupported extra type in JWT") +} diff --git a/common/jwt/jwtx_test.go b/common/jwt/jwtx_test.go new file mode 100644 index 0000000..a14082a --- /dev/null +++ b/common/jwt/jwtx_test.go @@ -0,0 +1,393 @@ +package jwtx + +import ( + "strings" + "testing" + "time" + + "github.com/golang-jwt/jwt/v4" +) + +func TestGenerateJwtToken(t *testing.T) { + // 测试数据 + testClaims := JwtClaims{ + UserId: "1", + AgentId: "", + Platform: "wxh5", + UserType: 0, + IsAgent: 0, + } + testSecret := "WUvoIwL-FK0qnlxhvxR9tV6SjfOpeJMpKmY2QvT99lA" + testExpire := int64(2592000) // 1小时 + + tests := []struct { + name string + claims JwtClaims + secret string + expire int64 + wantErr bool + }{ + { + name: "正常生成token", + claims: testClaims, + secret: testSecret, + expire: testExpire, + wantErr: false, + }, + { + name: "不同用户数据", + claims: JwtClaims{ + UserId: "99999", + AgentId: "", + Platform: "mobile", + UserType: 0, + IsAgent: 1, + }, + secret: testSecret, + expire: testExpire, + wantErr: false, + }, + { + name: "空密钥", + claims: testClaims, + secret: "", + expire: testExpire, + wantErr: false, // 空密钥不会导致生成失败，但验证时会失败 + }, + { + name: "零过期时间", + claims: testClaims, + secret: testSecret, + expire: 0, + wantErr: false, + }, + { + name: "负数过期时间", + claims: testClaims, + secret: testSecret, + expire: -3600, + wantErr: false, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + token, err := GenerateJwtToken(tt.claims, tt.secret, tt.expire) + + if (err != nil) != tt.wantErr { + t.Errorf("GenerateJwtToken() error = %v, wantErr %v", err, tt.wantErr) + return + } + + if !tt.wantErr { + // 验证token不为空 + if token == "" { + t.Error("GenerateJwtToken() 返回的token为空") + return + } + + // 验证token格式（JWT token应该包含两个点分隔符） + parts := strings.Split(token, ".") + if len(parts) != 3 { + t.Errorf("GenerateJwtToken() 返回的token格式不正确，期望3部分，实际%d部分", len(parts)) + return + } + + // 验证token可以被解析（不验证签名，只验证格式） + parsedToken, err := jwt.Parse(token, func(token *jwt.Token) (interface{}, error) { + return []byte(tt.secret), nil + }) + + if err == nil && parsedToken != nil { + // 验证claims是否正确设置 + if claims, ok := parsedToken.Claims.(jwt.MapClaims); ok { + // 验证userId + if userId, exists := claims["userId"]; exists { + if userId.(string) != tt.claims.UserId { + t.Errorf("token中的userId不匹配，期望%s，实际%v", tt.claims.UserId, userId) + } + } else { + t.Error("token中缺少userId字段") + } + + // 验证extra字段存在 + if _, exists := claims[ExtraKey]; !exists { + t.Error("token中缺少extra字段") + } + + // 验证exp字段 + if exp, exists := claims["exp"]; exists { + expTime := int64(exp.(float64)) + now := time.Now().Unix() + expectedExp := now + tt.expire + // 允许5秒的时间差异 + if expTime < expectedExp-5 || expTime > expectedExp+5 { + t.Errorf("token过期时间不正确，期望约%d，实际%d", expectedExp, expTime) + } + } else { + t.Error("token中缺少exp字段") + } + + // 验证iat字段 + if _, exists := claims["iat"]; !exists { + t.Error("token中缺少iat字段") + } + } + } + + t.Logf("生成的token: %s", token) + } + }) + } +} + +func TestGenerateJwtTokenAndParse(t *testing.T) { + // 测试生成token后能够正确解析 + testClaims := JwtClaims{ + UserId: "12345", + AgentId: "", + Platform: "web", + UserType: 1, + IsAgent: 0, + } + testSecret := "test-secret-key" + testExpire := int64(3600) + + // 生成token + token, err := GenerateJwtToken(testClaims, testSecret, testExpire) + if err != nil { + t.Fatalf("GenerateJwtToken() failed: %v", err) + } + + // 解析token + parsedClaims, err := ParseJwtToken(token, testSecret) + if err != nil { + t.Fatalf("ParseJwtToken() failed: %v", err) + } + + // 验证解析出的claims与原始claims一致 + if parsedClaims.UserId != testClaims.UserId { + t.Errorf("UserId不匹配，期望%d，实际%d", testClaims.UserId, parsedClaims.UserId) + } + if parsedClaims.AgentId != testClaims.AgentId { + t.Errorf("AgentId不匹配，期望%d，实际%d", testClaims.AgentId, parsedClaims.AgentId) + } + if parsedClaims.Platform != testClaims.Platform { + t.Errorf("Platform不匹配，期望%s，实际%s", testClaims.Platform, parsedClaims.Platform) + } + if parsedClaims.UserType != testClaims.UserType { + t.Errorf("UserType不匹配，期望%d，实际%d", testClaims.UserType, parsedClaims.UserType) + } + if parsedClaims.IsAgent != testClaims.IsAgent { + t.Errorf("IsAgent不匹配，期望%d，实际%d", testClaims.IsAgent, parsedClaims.IsAgent) + } + + t.Logf("测试通过: 生成token并成功解析，claims数据一致") +} + +func BenchmarkGenerateJwtToken(t *testing.B) { + // 性能测试 + testClaims := JwtClaims{ + UserId: "12345", + AgentId: "", + Platform: "web", + UserType: 1, + IsAgent: 0, + } + testSecret := "test-secret-key" + testExpire := int64(3600) + + t.ResetTimer() + for i := 0; i < t.N; i++ { + _, err := GenerateJwtToken(testClaims, testSecret, testExpire) + if err != nil { + t.Fatalf("GenerateJwtToken() failed: %v", err) + } + } +} + +func TestParseJwtToken(t *testing.T) { + // 使用你修改的测试数据 + testClaims := JwtClaims{ + UserId: "6", + AgentId: "", + Platform: "wxh5", + UserType: 0, + IsAgent: 0, + } + testSecret := "WUvoIwL-FK0qnlxhvxR9tV6SjfOpeJMpKmY2QvT99lA" + testExpire := int64(2592000) // 30天 + + // 先生成一个token用于测试 + token, err := GenerateJwtToken(testClaims, testSecret, testExpire) + if err != nil { + t.Fatalf("生成token失败: %v", err) + } + + t.Logf("生成的测试token: %s", token) + + tests := []struct { + name string + token string + secret string + wantErr bool + wantClaims *JwtClaims + }{ + { + name: "正常解析token", + token: token, + secret: testSecret, + wantErr: false, + wantClaims: &testClaims, + }, + { + name: "错误的密钥", + token: token, + secret: "wrong-secret", + wantErr: true, + wantClaims: nil, + }, + { + name: "空token", + token: "", + secret: testSecret, + wantErr: true, + wantClaims: nil, + }, + { + name: "无效token格式", + token: "invalid.token.format", + secret: testSecret, + wantErr: true, + wantClaims: nil, + }, + { + name: "缺少点分隔符的token", + token: "invalidtoken", + secret: testSecret, + wantErr: true, + wantClaims: nil, + }, + { + name: "自定义token", + token: "eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJleHAiOjE3NTI5MDA5MTQsImV4dHJhIjp7ImFnZW50SWQiOjAsImlzQWdlbnQiOjAsInBsYXRmb3JtIjoid3hoNSIsInVzZXJJZCI6NiwidXNlclR5cGUiOjF9LCJpYXQiOjE3NTAzMDg5MTQsInVzZXJJZCI6Nn0.GPKgLOaALOIa1ft7Hipuo4YKFf5guYt0rz2MCDCSdCQ", + secret: testSecret, + wantErr: false, + wantClaims: &testClaims, + }, + } + + for _, tt := range tests { + t.Run(tt.name, func(t *testing.T) { + claims, err := ParseJwtToken(tt.token, tt.secret) + + if (err != nil) != tt.wantErr { + t.Errorf("ParseJwtToken() error = %v, wantErr %v", err, tt.wantErr) + return + } + + if !tt.wantErr && tt.wantClaims != nil { + if claims == nil { + t.Error("ParseJwtToken() 返回的claims为nil") + return + } + + // 验证各个字段 + if claims.UserId != tt.wantClaims.UserId { + t.Errorf("UserId不匹配，期望%d，实际%d", tt.wantClaims.UserId, claims.UserId) + } + if claims.AgentId != tt.wantClaims.AgentId { + t.Errorf("AgentId不匹配，期望%d，实际%d", tt.wantClaims.AgentId, claims.AgentId) + } + if claims.Platform != tt.wantClaims.Platform { + t.Errorf("Platform不匹配，期望%s，实际%s", tt.wantClaims.Platform, claims.Platform) + } + if claims.UserType != tt.wantClaims.UserType { + t.Errorf("UserType不匹配，期望%d，实际%d", tt.wantClaims.UserType, claims.UserType) + } + if claims.IsAgent != tt.wantClaims.IsAgent { + t.Errorf("IsAgent不匹配，期望%d，实际%d", tt.wantClaims.IsAgent, claims.IsAgent) + } + + t.Logf("解析成功的claims: UserId=%d, AgentId=%d, Platform=%s, UserType=%d, IsAgent=%d", + claims.UserId, claims.AgentId, claims.Platform, claims.UserType, claims.IsAgent) + } + }) + } +} + +// TestParseCustomJwtToken 测试解析自定义token - 你可以在这里传入你自己的token +func TestParseCustomJwtToken(t *testing.T) { + // 在这里修改你想要测试的token和secret + customToken := "" // 在这里粘贴你的token + customSecret := "WUvoIwL-FK0qnlxhvxR9tV6SjfOpeJMpKmY2QvT99lA" // 你的密钥 + + // 如果没有提供自定义token，跳过测试 + if customToken == "" { + t.Skip("跳过自定义token测试，请在代码中设置customToken值") + return + } + + t.Logf("解析自定义token: %s", customToken) + + claims, err := ParseJwtToken(customToken, customSecret) + if err != nil { + t.Fatalf("解析自定义token失败: %v", err) + } + + t.Logf("解析结果:") + t.Logf(" UserId: %d", claims.UserId) + t.Logf(" AgentId: %d", claims.AgentId) + t.Logf(" Platform: %s", claims.Platform) + t.Logf(" UserType: %d", claims.UserType) + t.Logf(" IsAgent: %d", claims.IsAgent) +} + +// TestGenerateAndParseWithRealData 生成一个真实的token并解析 +func TestGenerateAndParseWithRealData(t *testing.T) { + // 使用真实数据生成token + realClaims := JwtClaims{ + UserId: "1", + AgentId: "", + Platform: "wxh5", + UserType: 0, + IsAgent: 0, + } + realSecret := "WUvoIwL-FK0qnlxhvxR9tV6SjfOpeJMpKmY2QvT99lA" + realExpire := int64(2592000) // 30天 + + // 生成token + token, err := GenerateJwtToken(realClaims, realSecret, realExpire) + if err != nil { + t.Fatalf("生成token失败: %v", err) + } + + t.Logf("=== 生成的完整token ===") + t.Logf("Token: %s", token) + t.Logf("========================") + + // 解析token + parsedClaims, err := ParseJwtToken(token, realSecret) + if err != nil { + t.Fatalf("解析token失败: %v", err) + } + + t.Logf("=== 解析结果 ===") + t.Logf("UserId: %d", parsedClaims.UserId) + t.Logf("AgentId: %d", parsedClaims.AgentId) + t.Logf("Platform: %s", parsedClaims.Platform) + t.Logf("UserType: %d", parsedClaims.UserType) + t.Logf("IsAgent: %d", parsedClaims.IsAgent) + t.Logf("================") + + // 验证数据一致性 + if parsedClaims.UserId != realClaims.UserId || + parsedClaims.AgentId != realClaims.AgentId || + parsedClaims.Platform != realClaims.Platform || + parsedClaims.UserType != realClaims.UserType || + parsedClaims.IsAgent != realClaims.IsAgent { + t.Error("解析出的claims与原始数据不一致") + } else { + t.Log("✅ 数据一致性验证通过") + } +} diff --git a/common/kqueue/message.go b/common/kqueue/message.go new file mode 100644 index 0000000..e1df433 --- /dev/null +++ b/common/kqueue/message.go @@ -0,0 +1,8 @@ +//KqMessage +package kqueue + +//第三方支付回调更改支付状态通知 +type ThirdPaymentUpdatePayStatusNotifyMessage struct { + PayStatus int64 `json:"payStatus"` + OrderSn string `json:"orderSn"` +} diff --git a/common/middleware/commonJwtAuthMiddleware.go b/common/middleware/commonJwtAuthMiddleware.go new file mode 100644 index 0000000..db0dae7 --- /dev/null +++ b/common/middleware/commonJwtAuthMiddleware.go @@ -0,0 +1,31 @@ +package middleware + +import ( + "github.com/zeromicro/go-zero/rest/handler" + "net/http" +) + +// CommonJwtAuthMiddleware : with jwt on the verification, no jwt on the verification +type CommonJwtAuthMiddleware struct { + secret string +} + +func NewCommonJwtAuthMiddleware(secret string) *CommonJwtAuthMiddleware { + return &CommonJwtAuthMiddleware{ + secret: secret, + } +} + +func (m *CommonJwtAuthMiddleware) Handle(next http.HandlerFunc) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + if len(r.Header.Get("Authorization")) > 0 { + //has jwt Authorization + authHandler := handler.Authorize(m.secret) + authHandler(next).ServeHTTP(w, r) + return + } else { + //no jwt Authorization + next(w, r) + } + } +} diff --git a/common/result/httpResult.go b/common/result/httpResult.go new file mode 100644 index 0000000..1b0609a --- /dev/null +++ b/common/result/httpResult.go @@ -0,0 +1,89 @@ +package result + +import ( + "fmt" + "net/http" + + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "github.com/zeromicro/go-zero/rest/httpx" + "google.golang.org/grpc/status" +) + +// http返回 +func HttpResult(r *http.Request, w http.ResponseWriter, resp interface{}, err error) { + + if err == nil { + httpx.WriteJson(w, http.StatusOK, Success(resp)) + } else { + //错误返回 + errcode := xerr.SERVER_COMMON_ERROR + errmsg := "服务器开小差啦，稍后再来试一试" + + causeErr := errors.Cause(err) // err类型 + if e, ok := causeErr.(*xerr.CodeError); ok { //自定义错误类型 + //自定义CodeError + errcode = e.GetErrCode() + errmsg = e.GetErrMsg() + } else { + if gstatus, ok := status.FromError(causeErr); ok { // grpc err错误 + grpcCode := uint32(gstatus.Code()) + if xerr.IsCodeErr(grpcCode) { //区分自定义错误跟系统底层、db等错误，底层、db错误不能返回给前端 + errcode = grpcCode + errmsg = gstatus.Message() + } + } + } + + logx.WithContext(r.Context()).Errorf("【API-ERR】 : %+v ", err) + + httpx.WriteJson(w, http.StatusOK, Error(errcode, errmsg)) + } +} + +// 授权的http方法 +func AuthHttpResult(r *http.Request, w http.ResponseWriter, resp interface{}, err error) { + + if err == nil { + //成功返回 + r := Success(resp) + httpx.WriteJson(w, http.StatusOK, r) + } else { + //错误返回 + errcode := xerr.SERVER_COMMON_ERROR + errmsg := "服务器开小差啦，稍后再来试一试" + + causeErr := errors.Cause(err) // err类型 + if e, ok := causeErr.(*xerr.CodeError); ok { //自定义错误类型 + //自定义CodeError + errcode = e.GetErrCode() + errmsg = e.GetErrMsg() + } else { + if gstatus, ok := status.FromError(causeErr); ok { // grpc err错误 + grpcCode := uint32(gstatus.Code()) + if xerr.IsCodeErr(grpcCode) { //区分自定义错误跟系统底层、db等错误，底层、db错误不能返回给前端 + errcode = grpcCode + errmsg = gstatus.Message() + } + } + } + + logx.WithContext(r.Context()).Errorf("【GATEWAY-ERR】 : %+v ", err) + + httpx.WriteJson(w, http.StatusUnauthorized, Error(errcode, errmsg)) + } +} + +// http 参数错误返回 +func ParamErrorResult(r *http.Request, w http.ResponseWriter, err error) { + errMsg := fmt.Sprintf("%s,%s", xerr.MapErrMsg(xerr.REUQEST_PARAM_ERROR), err.Error()) + httpx.WriteJson(w, http.StatusOK, Error(xerr.REUQEST_PARAM_ERROR, errMsg)) +} + +// http 参数校验失败返回 +func ParamValidateErrorResult(r *http.Request, w http.ResponseWriter, err error) { + //errMsg := fmt.Sprintf("%s,%s", xerr.MapErrMsg(xerr.REUQEST_PARAM_ERROR), err.Error()) + httpx.WriteJson(w, http.StatusOK, Error(xerr.PARAM_VERIFICATION_ERROR, err.Error())) +} diff --git a/common/result/jobResult.go b/common/result/jobResult.go new file mode 100644 index 0000000..d18476f --- /dev/null +++ b/common/result/jobResult.go @@ -0,0 +1,44 @@ +package result + +import ( + "context" + + "qnc-server/common/xerr" + + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/logx" + "google.golang.org/grpc/status" +) + +// job返回 +func JobResult(ctx context.Context, resp interface{}, err error) { + if err == nil { + // 成功返回 ,只有dev环境下才会打印info，线上不显示 + if resp != nil { + logx.Infof("resp: %+v", resp) + } + return + } else { + errCode := xerr.SERVER_COMMON_ERROR + errMsg := "服务器开小差啦，稍后再来试一试" + + // 错误返回 + causeErr := errors.Cause(err) // err类型 + if e, ok := causeErr.(*xerr.CodeError); ok { // 自定义错误类型 + // 自定义CodeError + errCode = e.GetErrCode() + errMsg = e.GetErrMsg() + } else { + if gstatus, ok := status.FromError(causeErr); ok { // grpc err错误 + grpcCode := uint32(gstatus.Code()) + if xerr.IsCodeErr(grpcCode) { // 区分自定义错误跟系统底层、db等错误，底层、db错误不能返回给前端 + errCode = grpcCode + errMsg = gstatus.Message() + } + } + } + + logx.WithContext(ctx).Errorf("【JOB-ERR】 : %+v ,errCode:%d , errMsg:%s ", err, errCode, errMsg) + return + } +} diff --git a/common/result/responseBean.go b/common/result/responseBean.go new file mode 100644 index 0000000..d29e1e0 --- /dev/null +++ b/common/result/responseBean.go @@ -0,0 +1,21 @@ +package result + +type ResponseSuccessBean struct { + Code uint32 `json:"code"` + Msg string `json:"msg"` + Data interface{} `json:"data"` +} +type NullJson struct{} + +func Success(data interface{}) *ResponseSuccessBean { + return &ResponseSuccessBean{200, "OK", data} +} + +type ResponseErrorBean struct { + Code uint32 `json:"code"` + Msg string `json:"msg"` +} + +func Error(errCode uint32, errMsg string) *ResponseErrorBean { + return &ResponseErrorBean{errCode, errMsg} +} diff --git a/common/tool/coinconvert.go b/common/tool/coinconvert.go new file mode 100644 index 0000000..f04d461 --- /dev/null +++ b/common/tool/coinconvert.go @@ -0,0 +1,19 @@ +package tool + +import "github.com/shopspring/decimal" + +var oneHundredDecimal decimal.Decimal = decimal.NewFromInt(100) + +//分转元 +func Fen2Yuan(fen int64) float64 { + y, _ := decimal.NewFromInt(fen).Div(oneHundredDecimal).Truncate(2).Float64() + return y +} + +//元转分 +func Yuan2Fen(yuan float64) int64 { + + f, _ := decimal.NewFromFloat(yuan).Mul(oneHundredDecimal).Truncate(0).Float64() + return int64(f) + +} diff --git a/common/tool/encryption.go b/common/tool/encryption.go new file mode 100644 index 0000000..b94f562 --- /dev/null +++ b/common/tool/encryption.go @@ -0,0 +1,23 @@ +package tool + +import ( + "crypto/md5" + "fmt" + "io" +) + +/** 加密方式 **/ + +func Md5ByString(str string) string { + m := md5.New() + _, err := io.WriteString(m, str) + if err != nil { + panic(err) + } + arr := m.Sum(nil) + return fmt.Sprintf("%x", arr) +} + +func Md5ByBytes(b []byte) string { + return fmt.Sprintf("%x", md5.Sum(b)) +} diff --git a/common/tool/krand.go b/common/tool/krand.go new file mode 100644 index 0000000..fb5b869 --- /dev/null +++ b/common/tool/krand.go @@ -0,0 +1,28 @@ +package tool + +import ( + "math/rand" + "time" +) + +const ( + KC_RAND_KIND_NUM = 0 // 纯数字 + KC_RAND_KIND_LOWER = 1 // 小写字母 + KC_RAND_KIND_UPPER = 2 // 大写字母 + KC_RAND_KIND_ALL = 3 // 数字、大小写字母 +) + +// 随机字符串 +func Krand(size int, kind int) string { + ikind, kinds, result := kind, [][]int{[]int{10, 48}, []int{26, 97}, []int{26, 65}}, make([]byte, size) + is_all := kind > 2 || kind < 0 + rand.Seed(time.Now().UnixNano()) + for i := 0; i < size; i++ { + if is_all { // random ikind + ikind = rand.Intn(3) + } + scope, base := kinds[ikind][0], kinds[ikind][1] + result[i] = uint8(base + rand.Intn(scope)) + } + return string(result) +} diff --git a/common/tool/krand_test.go b/common/tool/krand_test.go new file mode 100644 index 0000000..c5c0356 --- /dev/null +++ b/common/tool/krand_test.go @@ -0,0 +1,8 @@ +package tool + +import "testing" + +func TestMd5ByString(t *testing.T) { + s := Md5ByString("AAA") + t.Log(s) +} diff --git a/common/tool/placeholders.go b/common/tool/placeholders.go new file mode 100644 index 0000000..53e28d1 --- /dev/null +++ b/common/tool/placeholders.go @@ -0,0 +1,15 @@ +package tool + +import "strings" + +//替换 +func InPlaceholders(n int) string { + var b strings.Builder + for i := 0; i < n-1; i++ { + b.WriteString("?,") + } + if n > 0 { + b.WriteString("?") + } + return b.String() +} diff --git a/common/uniqueid/sn.go b/common/uniqueid/sn.go new file mode 100644 index 0000000..ea74d65 --- /dev/null +++ b/common/uniqueid/sn.go @@ -0,0 +1,20 @@ +package uniqueid + +import ( + "qnc-server/common/tool" + "fmt" + "time" +) + +// 生成sn单号 +type SnPrefix string + +const ( + SN_PREFIX_HOMESTAY_ORDER SnPrefix = "HSO" //民宿订单前缀 qnc-server_order/homestay_order + SN_PREFIX_THIRD_PAYMENT SnPrefix = "PMT" //第三方支付流水记录前缀 qnc-server_payment/third_payment +) + +// 生成单号 +func GenSn(snPrefix SnPrefix) string { + return fmt.Sprintf("%s%s%s", snPrefix, time.Now().Format("20060102150405"), tool.Krand(8, tool.KC_RAND_KIND_NUM)) +} diff --git a/common/uniqueid/sn_test.go b/common/uniqueid/sn_test.go new file mode 100644 index 0000000..6c12b9f --- /dev/null +++ b/common/uniqueid/sn_test.go @@ -0,0 +1,7 @@ +package uniqueid + +import "testing" + +func TestGenSn(t *testing.T) { + GenSn(SN_PREFIX_HOMESTAY_ORDER) +} diff --git a/common/uniqueid/uniqueid.go b/common/uniqueid/uniqueid.go new file mode 100644 index 0000000..ff4c474 --- /dev/null +++ b/common/uniqueid/uniqueid.go @@ -0,0 +1,23 @@ +package uniqueid + +import ( + "github.com/sony/sonyflake" + "github.com/zeromicro/go-zero/core/logx" +) + +var flake *sonyflake.Sonyflake + +func init() { + flake = sonyflake.NewSonyflake(sonyflake.Settings{}) +} + +func GenId() int64 { + + id, err := flake.NextID() + if err != nil { + logx.Severef("flake NextID failed with %s \n", err) + panic(err) + } + + return int64(id) +} diff --git a/common/wxminisub/tpl.go b/common/wxminisub/tpl.go new file mode 100644 index 0000000..d05af52 --- /dev/null +++ b/common/wxminisub/tpl.go @@ -0,0 +1,7 @@ +package wxminisub + +//订单支付成功 +const OrderPaySuccessTemplateID = "QIJPmfxaNqYzSjOlXGk1T6Xfw94JwbSPuOd3u_hi3WE" + +//支付成功入驻通知 +const OrderPaySuccessLiveKnowTemplateID = "kmm-maRr6v_9eMxEPpj-5clJ2YW_EFpd8-ngyYk63e4" diff --git a/common/xerr/errCode.go b/common/xerr/errCode.go new file mode 100644 index 0000000..a461ca9 --- /dev/null +++ b/common/xerr/errCode.go @@ -0,0 +1,23 @@ +package xerr + +// 成功返回 +const OK uint32 = 200 + +/**(前3位代表业务,后三位代表具体功能)**/ + +// 全局错误码 +const SERVER_COMMON_ERROR uint32 = 100001 +const REUQEST_PARAM_ERROR uint32 = 100002 +const TOKEN_EXPIRE_ERROR uint32 = 100003 +const TOKEN_GENERATE_ERROR uint32 = 100004 +const DB_ERROR uint32 = 100005 +const DB_UPDATE_AFFECTED_ZERO_ERROR uint32 = 100006 +const PARAM_VERIFICATION_ERROR uint32 = 100007 +const CUSTOM_ERROR uint32 = 100008 +const USER_NOT_FOUND uint32 = 100009 +const USER_NEED_BIND_MOBILE uint32 = 100010 + +const LOGIN_FAILED uint32 = 200001 +const LOGIC_QUERY_WAIT uint32 = 200002 +const LOGIC_QUERY_ERROR uint32 = 200003 +const LOGIC_QUERY_NOT_FOUND uint32 = 200004 diff --git a/common/xerr/errMsg.go b/common/xerr/errMsg.go new file mode 100644 index 0000000..dc911f1 --- /dev/null +++ b/common/xerr/errMsg.go @@ -0,0 +1,30 @@ +package xerr + +var message map[uint32]string + +func init() { + message = make(map[uint32]string) + message[OK] = "SUCCESS" + message[SERVER_COMMON_ERROR] = "系统正在升级,请稍后再试" + message[REUQEST_PARAM_ERROR] = "参数错误" + message[TOKEN_EXPIRE_ERROR] = "token失效，请重新登陆" + message[TOKEN_GENERATE_ERROR] = "生成token失败" + message[DB_ERROR] = "系统维护升级中,请稍后再试" + message[DB_UPDATE_AFFECTED_ZERO_ERROR] = "更新数据影响行数为0" +} + +func MapErrMsg(errcode uint32) string { + if msg, ok := message[errcode]; ok { + return msg + } else { + return "系统正在升级,请稍后再试" + } +} + +func IsCodeErr(errcode uint32) bool { + if _, ok := message[errcode]; ok { + return true + } else { + return false + } +} diff --git a/common/xerr/errors.go b/common/xerr/errors.go new file mode 100644 index 0000000..897e37e --- /dev/null +++ b/common/xerr/errors.go @@ -0,0 +1,39 @@ +package xerr + +import ( + "fmt" +) + +/** +常用通用固定错误 +*/ + +type CodeError struct { + errCode uint32 + errMsg string +} + +// 返回给前端的错误码 +func (e *CodeError) GetErrCode() uint32 { + return e.errCode +} + +// 返回给前端显示端错误信息 +func (e *CodeError) GetErrMsg() string { + return e.errMsg +} + +func (e *CodeError) Error() string { + return fmt.Sprintf("ErrCode:%d，ErrMsg:%s", e.errCode, e.errMsg) +} + +func NewErrCodeMsg(errCode uint32, errMsg string) *CodeError { + return &CodeError{errCode: errCode, errMsg: errMsg} +} +func NewErrCode(errCode uint32) *CodeError { + return &CodeError{errCode: errCode, errMsg: MapErrMsg(errCode)} +} + +func NewErrMsg(errMsg string) *CodeError { + return &CodeError{errCode: CUSTOM_ERROR, errMsg: errMsg} +} diff --git a/deploy/script/gen_models.ps1 b/deploy/script/gen_models.ps1 new file mode 100644 index 0000000..9bccad0 --- /dev/null +++ b/deploy/script/gen_models.ps1 @@ -0,0 +1,131 @@ +# 设置输出编码为UTF-8 +[Console]::InputEncoding = [System.Text.Encoding]::UTF8 +[Console]::OutputEncoding = [System.Text.Encoding]::UTF8 +$OutputEncoding = [System.Text.Encoding]::UTF8 +chcp.com 65001 | Out-Null +# 目录配置 +$OUTPUT_DIR = "./model" +$TARGET_DIR = "../../app/main/model" +$HOME_DIR = Join-Path $PSScriptRoot "..\template" + +# 表名列表 +$tables = @( + # ============================================ + # 新代理系统表 + # ============================================ + # "admin_api", + # "admin_dict_data", + # "admin_dict_type", + # "admin_menu", + # "admin_promotion_link", + # "admin_promotion_link_stats_history", + # "admin_promotion_link_stats_total", + # "admin_promotion_order", + # "admin_role", + # "admin_role_api", + # "admin_role_menu", + # "admin_user", + # "admin_user_role", + # "agent", + # "agent_commission", + "agent_config" + # "agent_freeze_task", + # "agent_invite_code", + # "agent_invite_code_usage", + # "agent_link", + # "agent_order", + # "agent_product_config", + # "agent_real_name", + # "agent_rebate", + # "agent_relation", + # "agent_short_link", + # "agent_upgrade", + # "agent_wallet", + # "agent_withdrawal", + # "agent_withdrawal_tax", + # "authorization_document", + # "example", + # "feature", + # "global_notifications", + # "order", + # "order_refund", + # "product", + # "product_feature", + # "query", + # "query_cleanup_config", + # "query_cleanup_detail", + # "query_cleanup_log", + # "user", + # "user_auth" +) + +# 为每个表生成模型 +foreach ($table in $tables) { + Write-Host "正在生成表: $table" -ForegroundColor Green + goctl model mysql datasource -url="qnc:5vg67b3UNHu8@tcp(127.0.0.1:21001)/qnc" -table="$table" -dir="./model" --home="$HOME_DIR" -cache=true --style=goZero + + # 移动生成的文件到目标目录 + if (Test-Path $OUTPUT_DIR) { + $sourceFiles = Get-ChildItem -Path $OUTPUT_DIR -File + + foreach ($file in $sourceFiles) { + $fileName = $file.Name + $targetPath = Join-Path $TARGET_DIR $fileName + $sourcePath = $file.FullName + + # 检查文件类型并决定是否移动 + $shouldMove = $false + $shouldOverwrite = $false + + if ($fileName -eq "vars.go") { + # vars.go: 如果目标目录不存在才移动 + if (-not (Test-Path $targetPath)) { + $shouldMove = $true + Write-Host " 移动 $fileName (vars.go 不存在于目标目录)" -ForegroundColor Yellow + } + else { + Write-Host " 跳过 $fileName (vars.go 已存在于目标目录，防止覆盖)" -ForegroundColor Cyan + } + } + elseif ($fileName -match "_gen\.go$") { + # 带 _gen 后缀的文件: 直接覆盖 + $shouldMove = $true + $shouldOverwrite = $true + Write-Host " 移动 $fileName (覆盖 _gen 文件)" -ForegroundColor Yellow + } + else { + # 不带 _gen 后缀的文件: 如果目标目录不存在才移动 + if (-not (Test-Path $targetPath)) { + $shouldMove = $true + Write-Host " 移动 $fileName (非 _gen 文件不存在于目标目录)" -ForegroundColor Yellow + } + else { + Write-Host " 跳过 $fileName (非 _gen 文件已存在于目标目录，防止覆盖)" -ForegroundColor Cyan + } + } + + # 执行移动操作 + if ($shouldMove) { + # 确保目标目录存在 + if (-not (Test-Path $TARGET_DIR)) { + New-Item -ItemType Directory -Path $TARGET_DIR -Force | Out-Null + } + + # 如果目标文件存在且需要覆盖，先删除 + if ($shouldOverwrite -and (Test-Path $targetPath)) { + Remove-Item -Path $targetPath -Force + } + + # 移动文件 + Move-Item -Path $sourcePath -Destination $targetPath -Force + Write-Host " ✓ 已移动到: $targetPath" -ForegroundColor Green + } + } + } +} + +Write-Host "" +Write-Host '所有模型文件生成并移动完成!' -ForegroundColor Green +if (Test-Path $OUTPUT_DIR) { + Get-ChildItem -Path $OUTPUT_DIR -File | Remove-Item -Force +} diff --git a/deploy/sql/README_表结构修改说明.md b/deploy/sql/README_表结构修改说明.md new file mode 100644 index 0000000..78bfaed --- /dev/null +++ b/deploy/sql/README_表结构修改说明.md @@ -0,0 +1,320 @@ +# 表结构修改说明 + +本文档说明系统重构需要的所有表结构修改，包括： +1. UUID迁移（将所有bigint类型的ID改为CHAR(36)类型的UUID） +2. 用户系统重构（删除UserTemp表，优化UserAuth表） + +## 一、UUID迁移 + +### 1.1 概述 + +将所有表的主键ID从`bigint`类型改为`CHAR(36)`类型的UUID，同时将所有外键关联字段也改为UUID类型。 + +**注意**：开发环境使用简化版脚本（`uuid_migration_simple.sql`），直接修改表结构，不保留旧数据。 + +### 1.2 涉及的表 + +**核心业务表**： +- `user` - 用户表 +- `agent` - 代理表 +- `product` - 产品表 +- `order` - 订单表 +- `query` - 查询报告表 +- `user_auth` - 用户认证表 + +**代理相关表**： +- `agent_commission` - 代理佣金表 +- `agent_invite_code` - 代理邀请码表 +- `agent_link` - 代理链接表 +- `agent_order` - 代理订单表 +- `agent_wallet` - 代理钱包表 +- `agent_withdrawal` - 代理提现表 +- `agent_withdrawal_tax` - 代理提现税费表 +- `agent_rebate` - 代理返利表 +- `agent_relation` - 代理关系表 +- `agent_upgrade` - 代理升级表 +- `agent_real_name` - 代理实名表 +- `agent_config` - 代理配置表 +- `agent_product_config` - 代理产品配置表 +- `agent_short_link` - 代理短链表 +- `agent_invite_code_usage` - 邀请码使用记录表 +- `agent_freeze_task` - 代理冻结任务表 + +**订单相关表**： +- `order_refund` - 订单退款表 + +**查询相关表**： +- `query_cleanup_log` - 查询清理日志表 +- `query_cleanup_detail` - 查询清理详情表 +- `query_cleanup_config` - 查询清理配置表 + +**产品相关表**： +- `product_feature` - 产品功能表 +- `feature` - 功能表 + +**其他表**： +- `authorization_document` - 授权书表 +- `global_notifications` - 全局通知表 + +**管理后台表**： +- `admin_user` - 管理员用户表 +- `admin_role` - 管理员角色表 +- `admin_menu` - 管理员菜单表 +- `admin_api` - 管理员API表 +- `admin_dict_type` - 字典类型表 +- `admin_dict_data` - 字典数据表 +- `admin_user_role` - 管理员用户角色关联表 +- `admin_role_menu` - 管理员角色菜单关联表 +- `admin_role_api` - 管理员角色API关联表 + +### 1.3 修改内容 + +**主键字段**： +- 所有表的`id`字段：`bigint` → `CHAR(36)` +- 移除`AUTO_INCREMENT`属性 +- 插入时使用`UUID()`函数或应用层生成UUID + +**外键字段**（软关联）： +- `user_id`：`bigint` → `CHAR(36)` +- `agent_id`：`bigint` → `CHAR(36)` +- `order_id`：`bigint` → `CHAR(36)` +- `product_id`：`bigint` → `CHAR(36)` +- `invite_code_id`：`bigint` → `CHAR(36)` +- `link_id`：`bigint` → `CHAR(36)` +- `commission_id`：`bigint` → `CHAR(36)` +- `wallet_id`：`bigint` → `CHAR(36)` +- `withdrawal_id`：`bigint` → `CHAR(36)` +- `parent_agent_id`：`bigint` → `CHAR(36)` +- `team_leader_id`：`bigint` → `CHAR(36)` +- `cleanup_log_id`：`bigint` → `CHAR(36)` +- `query_id`：`bigint` → `CHAR(36)` +- `feature_id`：`bigint` → `CHAR(36)` +- `parent_id`（菜单）：`bigint` → `CHAR(36)` +- `dict_type_id`：`bigint` → `CHAR(36)` +- `role_id`：`bigint` → `CHAR(36)` +- `menu_id`：`bigint` → `CHAR(36)` +- `api_id`：`bigint` → `CHAR(36)` +- `used_user_id`：`bigint` → `CHAR(36)` +- `used_agent_id`：`bigint` → `CHAR(36)` + +### 1.4 执行脚本 + +**开发环境**：执行文件 `uuid_migration_simple.sql` +- 直接修改表结构，不保留旧数据 +- 适用于开发环境，没有重要数据 + +**生产环境**：执行文件 `uuid_migration.sql` +- 完整的迁移流程，保留旧数据以便回滚 +- 需要分阶段执行，包含数据迁移和验证 + +**开发环境执行步骤**： +1. 直接执行 `uuid_migration_simple.sql` +2. 所有表的主键和外键字段直接改为CHAR(36) +3. 插入新记录时，在应用层生成UUID + +**注意事项**： +- 开发环境脚本直接修改表结构，不保留旧数据 +- 如果表中有数据，需要先清空数据或手动填充UUID +- 代码层面需要同步修改所有ID字段类型（int64 → string） + +## 二、用户系统重构 + +### 2.1 概述 + +删除`UserTemp`表，统一使用`User`表（通过`mobile`字段是否为空区分临时用户和正式用户），优化`UserAuth`表结构。 + +### 2.2 涉及的表 + +**需要修改的表**： +- `user` - 用户表（无需修改结构，只需确保mobile可为空） +- `user_auth` - 用户认证表（添加唯一索引） +- `user_temp` - 临时用户表（**删除**） + +**需要迁移数据的表**： +- `order` - 订单表（更新user_id） +- `query` - 查询报告表（更新user_id） + +### 2.3 修改内容 + +#### 2.3.1 UserAuth表 + +**添加唯一索引**： +```sql +ALTER TABLE `user_auth` +ADD UNIQUE INDEX `uk_auth_type_key` (`auth_type`, `auth_key`); +``` + +**作用**： +- 确保同一个认证方式（auth_type + auth_key）只能绑定一个用户 +- 防止重复绑定 + +#### 2.3.2 UserTemp表 + +**删除原因**： +- 临时用户可以直接创建User表记录（mobile为空） +- 临时用户的认证信息存储在UserAuth表 +- 通过User.mobile是否为空来区分临时用户和正式用户 +- 统一使用User表，逻辑更清晰 + +**删除步骤**： +1. 迁移UserTemp数据到User和UserAuth表 +2. 更新关联的订单和报告的user_id +3. 验证数据完整性 +4. 删除UserTemp表 + +### 2.4 执行脚本 + +**开发环境**：执行文件 `user_system_refactor.sql`（已简化） +- 直接删除UserTemp表 +- 添加UserAuth唯一索引 + +**生产环境**：如需数据迁移，请参考完整版脚本 + +**开发环境执行步骤**： +1. 添加UserAuth唯一索引 +2. 直接删除UserTemp表 + +**注意事项**： +- 开发环境脚本直接删除UserTemp表，不进行数据迁移 +- 删除UserTemp表前，确保代码中已移除所有对UserTemp表的引用 + +## 三、执行顺序 + +### 3.1 推荐执行顺序 + +**开发环境**： +1. **先执行用户系统重构**（`user_system_refactor.sql`） + - 添加UserAuth唯一索引 + - 删除UserTemp表 + +2. **再执行UUID迁移**（`uuid_migration_simple.sql`） + - 直接修改所有表结构 + - 需要同步修改代码 + +**生产环境**： +1. 先执行用户系统重构（完整版，包含数据迁移） +2. 再执行UUID迁移（完整版，包含数据迁移和验证） + +### 3.2 如果先执行UUID迁移 + +如果先执行UUID迁移，需要注意： +- UserTemp表的ID也会改为UUID +- 迁移UserTemp数据时，需要使用UUID映射 +- 用户系统重构脚本需要相应调整 + +## 四、代码修改清单 + +### 4.1 UUID迁移需要的代码修改 + +**Model层**： +- 所有Model的ID字段类型：`int64` → `string` +- 所有外键字段类型：`int64` → `string` +- 移除`AUTO_INCREMENT`相关逻辑 +- 插入时生成UUID（使用`uuid.NewString()`） + +**Service层**： +- 所有使用ID的地方改为UUID +- 查询条件改为UUID +- 关联查询改为UUID + +**API层**： +- 请求参数中的ID改为UUID +- 响应数据中的ID改为UUID + +**数据库操作**： +- 插入操作：使用`uuid.NewString()`生成UUID +- 查询操作：使用UUID字符串查询 +- 更新操作：使用UUID字符串更新 + +### 4.2 用户系统重构需要的代码修改 + +**删除UserTemp相关代码**： +- 删除`UserTempModel` +- 删除所有使用`UserTemp`的代码 +- 删除临时用户创建逻辑中的UserTemp相关代码 + +**修改用户创建逻辑**： +- 临时用户：直接创建User记录（mobile为空） +- 创建UserAuth记录（auth_type + auth_key） + +**修改用户查询逻辑**： +- 通过UserAuth表查询用户 +- 通过User.mobile是否为空判断用户类型 + +**修改账号合并逻辑**： +- 绑定手机号时，检查手机号是否已存在 +- 如果存在，合并账号（迁移数据） +- 如果不存在，更新User.mobile + +## 五、验证清单 + +### 5.1 UUID迁移验证 + +- [ ] 所有表的主键已改为CHAR(36) +- [ ] 所有外键字段已改为CHAR(36) +- [ ] 插入新记录时自动生成UUID +- [ ] 查询操作使用UUID正常 +- [ ] 关联查询使用UUID正常 +- [ ] 索引和约束已更新 +- [ ] 数据完整性验证通过 + +### 5.2 用户系统重构验证 + +- [ ] UserAuth表有唯一索引`uk_auth_type_key` +- [ ] UserTemp表数据已迁移（如果存在） +- [ ] 订单和报告的user_id已更新 +- [ ] UserTemp表已删除（如果存在） +- [ ] 临时用户创建逻辑使用User表 +- [ ] 账号合并逻辑正常工作 +- [ ] 数据完整性验证通过 + +## 六、回滚方案 + +### 6.1 UUID迁移回滚 + +如果UUID迁移出现问题，可以： +1. 保留原ID字段（迁移脚本中已保留） +2. 恢复主键为原ID字段 +3. 删除UUID字段 +4. 恢复代码使用int64类型 + +### 6.2 用户系统重构回滚 + +如果用户系统重构出现问题，可以： +1. 恢复UserTemp表（从备份） +2. 恢复订单和报告的user_id +3. 恢复代码使用UserTemp表 + +## 七、常见问题 + +### 7.1 UUID性能问题 + +**问题**：UUID作为主键可能影响性能 + +**解答**： +- UUID是字符串类型，索引效率略低于bigint +- 但UUID的优势是全局唯一，适合分布式系统 +- 如果性能问题严重，可以考虑使用BINARY(16)存储UUID（需要转换） + +### 7.2 UUID长度问题 + +**问题**：CHAR(36)占用空间较大 + +**解答**： +- UUID标准格式是36字符（包含连字符） +- 如果空间敏感，可以使用BINARY(16)存储（需要转换函数） +- 当前使用CHAR(36)便于调试和查看 + +### 7.3 UserTemp数据迁移问题 + +**问题**：如何确保UserTemp数据正确迁移？ + +**解答**： +- 通过auth_type和auth_key匹配UserAuth记录 +- 建立UserTemp.id到User.id的映射关系 +- 迁移后验证数据完整性 +- 保留UserTemp表一段时间，确认无误后再删除 + +## 八、联系信息 + +如有问题，请联系开发团队。 diff --git a/deploy/sql/add_agent_short_link_table.sql b/deploy/sql/add_agent_short_link_table.sql new file mode 100644 index 0000000..6cd0149 --- /dev/null +++ b/deploy/sql/add_agent_short_link_table.sql @@ -0,0 +1,47 @@ +-- ============================================ +-- 代理短链表 +-- 说明：存储推广链接和邀请链接的短链映射关系 +-- +-- 功能说明： +-- 1. 支持两种类型：promotion（推广报告）和 invite（邀请好友） +-- 2. 为每个链接生成一个短链标识（6位随机字符串） +-- 3. 短链格式：https://推广域名/s/{short_code} +-- 4. 短链重定向到前端传入的target_path +-- +-- 执行步骤： +-- 1. 执行此 SQL 创建表 +-- 2. 使用 goctl 生成 Model：agent_short_link -> AgentShortLinkModel +-- 3. 在 servicecontext.go 中添加 AgentShortLinkModel +-- ============================================ +CREATE TABLE `agent_short_link` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `type` tinyint NOT NULL COMMENT '类型：1=推广报告(promotion)，2=邀请好友(invite)', + `link_id` bigint DEFAULT NULL COMMENT '推广链接ID（关联agent_link表，仅推广报告类型使用）', + `invite_code_id` bigint DEFAULT NULL COMMENT '邀请码ID（关联agent_invite_code表，仅邀请好友类型使用）', + `link_identifier` varchar(200) DEFAULT NULL COMMENT '推广链接标识（加密，仅推广报告类型使用）', + `invite_code` varchar(50) DEFAULT NULL COMMENT '邀请码（仅邀请好友类型使用）', + `short_code` varchar(20) NOT NULL COMMENT '短链标识（6位随机字符串）', + `target_path` varchar(500) NOT NULL COMMENT '目标地址（前端传入，如：/agent/promotionInquire/xxx 或 /register?invite_code=xxx）', + `promotion_domain` varchar(200) NOT NULL COMMENT '推广域名（生成短链时使用的域名）', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + UNIQUE KEY `uk_short_code` (`short_code`, `del_state`) COMMENT '短链标识唯一索引', + UNIQUE KEY `uk_link_id_type` ( + `link_id`, + `type`, + `del_state` + ) COMMENT '同一推广链接同一类型只能有一个有效短链', + UNIQUE KEY `uk_invite_code_id_type` ( + `invite_code_id`, + `type`, + `del_state` + ) COMMENT '同一邀请码同一类型只能有一个有效短链', + KEY `idx_link_identifier` (`link_identifier`) COMMENT '推广链接标识索引', + KEY `idx_invite_code` (`invite_code`) COMMENT '邀请码索引', + KEY `idx_type` (`type`) COMMENT '类型索引', + KEY `idx_create_time` (`create_time`) COMMENT '创建时间索引' +) ENGINE = InnoDB DEFAULT CHARSET = utf8mb4 COLLATE = utf8mb4_general_ci COMMENT = '代理短链表'; \ No newline at end of file diff --git a/deploy/sql/add_invite_code_usage_table.sql b/deploy/sql/add_invite_code_usage_table.sql new file mode 100644 index 0000000..59942b8 --- /dev/null +++ b/deploy/sql/add_invite_code_usage_table.sql @@ -0,0 +1,46 @@ +-- ============================================ +-- 邀请码使用历史表 +-- 说明：记录每次邀请码的使用情况，支持统计和查询 +-- +-- 功能说明： +-- 1. 记录每个邀请码每次使用的详细信息 +-- 2. 支持统计每个邀请码邀请了多少代理 +-- 3. 支持查询某个代理是通过哪个邀请码成为代理的 +-- 4. 保留完整的使用历史记录 +-- +-- 执行步骤： +-- 1. 执行此 SQL 创建表和添加字段 +-- 2. 使用 goctl 生成 Model：agent_invite_code_usage -> AgentInviteCodeUsageModel +-- 3. 在 servicecontext.go 中添加 AgentInviteCodeUsageModel +-- ============================================ +CREATE TABLE `agent_invite_code_usage` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `invite_code_id` bigint NOT NULL COMMENT '邀请码ID（关联agent_invite_code表）', + `code` varchar(50) NOT NULL COMMENT '邀请码（冗余字段，便于查询）', + `user_id` bigint NOT NULL COMMENT '使用用户ID', + `agent_id` bigint NOT NULL COMMENT '成为的代理ID', + `agent_level` tinyint NOT NULL COMMENT '代理等级：1=普通，2=黄金，3=钻石', + `used_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '使用时间', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + KEY `idx_invite_code_id` (`invite_code_id`) COMMENT '关联邀请码ID', + KEY `idx_code` (`code`) COMMENT '邀请码索引', + KEY `idx_user_id` (`user_id`) COMMENT '用户ID索引', + KEY `idx_agent_id` (`agent_id`) COMMENT '代理ID索引', + KEY `idx_used_time` (`used_time`) COMMENT '使用时间索引', + KEY `idx_create_time` (`create_time`) COMMENT '创建时间索引', + KEY `idx_code_time` (`code`, `used_time`) COMMENT '邀请码和使用时间复合索引' +) ENGINE = InnoDB DEFAULT CHARSET = utf8mb4 COLLATE = utf8mb4_general_ci COMMENT = '邀请码使用历史表'; + +-- ============================================ +-- 在agent表中添加invite_code_id字段（可选，便于直接查询代理是通过哪个邀请码成为的） +-- ============================================ +ALTER TABLE `agent` +ADD COLUMN `invite_code_id` bigint DEFAULT NULL COMMENT '通过哪个邀请码成为代理（关联agent_invite_code表）' AFTER `team_leader_id`; + +ALTER TABLE `agent` +ADD KEY `idx_invite_code_id` (`invite_code_id`) COMMENT '邀请码ID索引'; \ No newline at end of file diff --git a/deploy/sql/agent_config_migration.sql b/deploy/sql/agent_config_migration.sql new file mode 100644 index 0000000..673063c --- /dev/null +++ b/deploy/sql/agent_config_migration.sql @@ -0,0 +1,56 @@ +-- ============================================ +-- 代理配置表重构 SQL 脚本 +-- 执行顺序：按照下面的顺序依次执行 +-- ============================================ + +-- ============================================ +-- 步骤1：删除价格相关配置项 +-- 说明：价格配置已改为在产品配置表（agent_product_config）中按产品配置 +-- ============================================ +DELETE FROM `agent_config` WHERE `config_key` IN ( + 'base_price', + 'system_max_price', + 'price_threshold', + 'price_fee_rate' +); + +-- ============================================ +-- 步骤2：统一配置键命名 +-- 说明：修改配置键名称，使其与代码逻辑一致 +-- ============================================ + +-- 修改等级加成配置键 +UPDATE `agent_config` SET `config_key` = 'level_1_bonus' WHERE `config_key` = 'level_bonus_normal'; +UPDATE `agent_config` SET `config_key` = 'level_2_bonus' WHERE `config_key` = 'level_bonus_gold'; +UPDATE `agent_config` SET `config_key` = 'level_3_bonus' WHERE `config_key` = 'level_bonus_diamond'; + +-- 修改升级费用配置键 +UPDATE `agent_config` SET `config_key` = 'upgrade_to_gold_fee' WHERE `config_key` = 'upgrade_fee_normal_to_gold'; +UPDATE `agent_config` SET `config_key` = 'upgrade_to_diamond_fee' WHERE `config_key` = 'upgrade_fee_to_diamond'; + +-- 修改升级返佣配置键 +UPDATE `agent_config` SET `config_key` = 'upgrade_to_gold_rebate' WHERE `config_key` = 'upgrade_rebate_normal_to_gold'; +UPDATE `agent_config` SET `config_key` = 'upgrade_to_diamond_rebate' WHERE `config_key` = 'upgrade_rebate_to_diamond'; + +-- ============================================ +-- 步骤3：添加缺失的配置项 +-- 说明：添加免税额度配置项（如果不存在） +-- ============================================ +INSERT INTO `agent_config` (`config_key`, `config_value`, `config_type`, `description`) +SELECT 'tax_exemption_amount', '0.00', 'tax', '提现免税额度（元，默认0）' +WHERE NOT EXISTS ( + SELECT 1 FROM `agent_config` WHERE `config_key` = 'tax_exemption_amount' +); + +-- ============================================ +-- 验证查询：检查配置项是否正确 +-- ============================================ +SELECT `config_key`, `config_value`, `config_type`, `description` +FROM `agent_config` +WHERE `del_state` = 0 +ORDER BY `config_type`, `config_key`; + +-- ============================================ +-- 执行完成后，请重新生成 agent_config 相关的 Model 代码 +-- ============================================ + diff --git a/deploy/sql/agent_freeze_task_migration.sql b/deploy/sql/agent_freeze_task_migration.sql new file mode 100644 index 0000000..7ef9ca0 --- /dev/null +++ b/deploy/sql/agent_freeze_task_migration.sql @@ -0,0 +1,32 @@ +-- ============================================ +-- 代理佣金冻结任务表 +-- 说明：用于记录需要解冻的佣金冻结任务，保证异步任务的一致性和持久化 +-- ============================================ + +CREATE TABLE `agent_freeze_task` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `agent_id` bigint NOT NULL COMMENT '代理ID', + `order_id` bigint NOT NULL COMMENT '订单ID', + `commission_id` bigint NOT NULL COMMENT '佣金记录ID', + `freeze_amount` decimal(10, 2) NOT NULL COMMENT '冻结金额', + `order_price` decimal(10, 2) NOT NULL COMMENT '订单单价', + `freeze_ratio` decimal(5, 4) NOT NULL DEFAULT 0.1000 COMMENT '冻结比例（例如：0.1000表示10%）', + `status` tinyint NOT NULL DEFAULT 1 COMMENT '状态：1=待解冻，2=已解冻，3=已取消', + `freeze_time` datetime NOT NULL COMMENT '冻结时间', + `unfreeze_time` datetime NOT NULL COMMENT '解冻时间（冻结时间+1个月）', + `actual_unfreeze_time` datetime DEFAULT NULL COMMENT '实际解冻时间', + `remark` varchar(255) DEFAULT NULL COMMENT '备注', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + KEY `idx_agent_id` (`agent_id`), + KEY `idx_order_id` (`order_id`), + KEY `idx_commission_id` (`commission_id`), + KEY `idx_status` (`status`), + KEY `idx_unfreeze_time` (`unfreeze_time`), + KEY `idx_agent_status` (`agent_id`, `status`), + KEY `idx_create_time` (`create_time`) +) ENGINE = InnoDB DEFAULT CHARSET = utf8mb4 COLLATE = utf8mb4_general_ci COMMENT = '代理佣金冻结任务表'; \ No newline at end of file diff --git a/deploy/sql/agent_system_migration.sql b/deploy/sql/agent_system_migration.sql new file mode 100644 index 0000000..223f0f5 --- /dev/null +++ b/deploy/sql/agent_system_migration.sql @@ -0,0 +1,386 @@ +-- ============================================ +-- 新代理系统数据表创建脚本 +-- 说明：创建新代理系统的所有数据表 +-- ============================================ + +-- ============================================ +-- 1. 代理基本信息表 +-- ============================================ +CREATE TABLE `agent` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `user_id` bigint NOT NULL COMMENT '用户ID', + `level` tinyint NOT NULL DEFAULT 1 COMMENT '代理等级：1=普通，2=黄金，3=钻石', + `region` varchar(50) DEFAULT NULL COMMENT '区域（可选）', + `mobile` varchar(50) NOT NULL DEFAULT '' COMMENT '手机号（加密）', + `wechat_id` varchar(100) DEFAULT NULL COMMENT '微信号', + `team_leader_id` bigint DEFAULT NULL COMMENT '团队首领ID（钻石代理的ID，普通/黄金代理指向其团队首领）', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + UNIQUE KEY `uk_user_id` (`user_id`), + KEY `idx_mobile` (`mobile`), + KEY `idx_level` (`level`), + KEY `idx_team_leader_id` (`team_leader_id`), + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理基本信息表'; + +-- ============================================ +-- 2. 代理钱包表（注意：agent_audit 表已废弃，新系统通过邀请码直接成为代理，无需审核） +-- ============================================ +-- ============================================ +CREATE TABLE `agent_wallet` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `agent_id` bigint NOT NULL COMMENT '代理ID', + `balance` decimal(10,2) NOT NULL DEFAULT 0.00 COMMENT '可用余额', + `frozen_balance` decimal(10,2) NOT NULL DEFAULT 0.00 COMMENT '冻结余额', + `total_earnings` decimal(10,2) NOT NULL DEFAULT 0.00 COMMENT '累计收益', + `withdrawn_amount` decimal(10,2) NOT NULL DEFAULT 0.00 COMMENT '累计提现金额', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + UNIQUE KEY `uk_agent_id` (`agent_id`), + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理钱包表'; + +-- ============================================ +-- 4. 代理上下级关系表（支持脱离关系） +-- ============================================ +CREATE TABLE `agent_relation` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `parent_id` bigint NOT NULL COMMENT '上级代理ID', + `child_id` bigint NOT NULL COMMENT '下级代理ID', + `relation_type` tinyint NOT NULL DEFAULT 1 COMMENT '关系类型：1=直接关系，2=已脱离（历史记录）', + `detach_reason` varchar(100) DEFAULT NULL COMMENT '脱离原因：upgrade=升级脱离', + `detach_time` datetime DEFAULT NULL COMMENT '脱离时间', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + UNIQUE KEY `uk_parent_child_type` (`parent_id`, `child_id`, `relation_type`), + KEY `idx_parent_id` (`parent_id`), + KEY `idx_child_id` (`child_id`), + KEY `idx_relation_type` (`relation_type`), + KEY `idx_parent_relation` (`parent_id`, `relation_type`) COMMENT '复合索引：查询有效下级', + KEY `idx_child_relation` (`child_id`, `relation_type`) COMMENT '复合索引：查询有效上级', + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理上下级关系表'; + +-- ============================================ +-- 5. 代理推广链接表 +-- ============================================ +CREATE TABLE `agent_link` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `agent_id` bigint NOT NULL COMMENT '代理ID', + `user_id` bigint NOT NULL COMMENT '用户ID', + `product_id` bigint NOT NULL COMMENT '产品ID', + `link_identifier` varchar(200) NOT NULL COMMENT '推广链接标识（加密）', + `set_price` decimal(10,2) NOT NULL COMMENT '代理设定价格', + `actual_base_price` decimal(10,2) NOT NULL COMMENT '实际底价（基础底价+等级加成）', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + UNIQUE KEY `uk_link_identifier` (`link_identifier`), + UNIQUE KEY `uk_agent_product_price` (`agent_id`, `product_id`, `set_price`, `del_state`) COMMENT '唯一约束：同一代理、同一产品、同一价格只能有一个有效链接', + KEY `idx_agent_id` (`agent_id`), + KEY `idx_user_id` (`user_id`), + KEY `idx_product_id` (`product_id`), + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理推广链接表'; + +-- ============================================ +-- 6. 代理订单关联表 +-- ============================================ +CREATE TABLE `agent_order` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `agent_id` bigint NOT NULL COMMENT '代理ID', + `order_id` bigint NOT NULL COMMENT '订单ID', + `product_id` bigint NOT NULL COMMENT '产品ID', + `order_amount` decimal(10,2) NOT NULL COMMENT '订单金额（用户实际支付金额，冗余字段）', + `set_price` decimal(10,2) NOT NULL COMMENT '代理设定价格', + `actual_base_price` decimal(10,2) NOT NULL COMMENT '实际底价（基础底价+等级加成）', + `price_cost` decimal(10,2) NOT NULL DEFAULT 0.00 COMMENT '提价成本（(设定价格-提价标准阈值)×提价手续费比例）', + `agent_profit` decimal(10,2) NOT NULL COMMENT '代理收益（设定价格-实际底价-提价成本）', + `process_status` tinyint NOT NULL DEFAULT 0 COMMENT '处理状态：0=待处理，1=处理成功，2=处理失败', + `process_time` datetime DEFAULT NULL COMMENT '处理时间', + `process_remark` varchar(500) DEFAULT NULL COMMENT '处理备注', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + UNIQUE KEY `uk_order_id` (`order_id`), + KEY `idx_agent_id` (`agent_id`), + KEY `idx_product_id` (`product_id`), + KEY `idx_process_status` (`process_status`), + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理订单关联表'; + +-- ============================================ +-- 7. 代理佣金记录表 +-- ============================================ +CREATE TABLE `agent_commission` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `agent_id` bigint NOT NULL COMMENT '代理ID', + `order_id` bigint NOT NULL COMMENT '订单ID', + `product_id` bigint NOT NULL COMMENT '产品ID', + `amount` decimal(10,2) NOT NULL COMMENT '佣金金额', + `status` tinyint NOT NULL DEFAULT 1 COMMENT '状态：1=已发放，2=已冻结，3=已取消', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + KEY `idx_agent_id` (`agent_id`), + KEY `idx_order_id` (`order_id`), + KEY `idx_product_id` (`product_id`), + KEY `idx_status` (`status`), + KEY `idx_agent_status` (`agent_id`, `status`) COMMENT '复合索引：查询代理的佣金记录', + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理佣金记录表'; + +-- ============================================ +-- 8. 代理返佣记录表（等级加成返佣） +-- ============================================ +CREATE TABLE `agent_rebate` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `agent_id` bigint NOT NULL COMMENT '获得返佣的代理ID', + `source_agent_id` bigint NOT NULL COMMENT '来源代理ID（推广订单的代理）', + `order_id` bigint NOT NULL COMMENT '订单ID', + `product_id` bigint NOT NULL COMMENT '产品ID', + `rebate_type` tinyint NOT NULL COMMENT '返佣类型：1=直接上级返佣，2=钻石上级返佣，3=黄金上级返佣', + `level_bonus` decimal(10,2) NOT NULL COMMENT '等级加成金额（来源代理的等级加成）', + `rebate_amount` decimal(10,2) NOT NULL COMMENT '返佣金额', + `status` tinyint NOT NULL DEFAULT 1 COMMENT '状态：1=已发放，2=已冻结，3=已取消', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + KEY `idx_agent_id` (`agent_id`), + KEY `idx_source_agent_id` (`source_agent_id`), + KEY `idx_order_id` (`order_id`), + KEY `idx_product_id` (`product_id`), + KEY `idx_rebate_type` (`rebate_type`), + KEY `idx_status` (`status`), + KEY `idx_order_rebate_type` (`order_id`, `rebate_type`) COMMENT '复合索引：查询订单的返佣明细', + KEY `idx_agent_status` (`agent_id`, `status`) COMMENT '复合索引：查询代理的返佣记录', + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理返佣记录表（等级加成返佣）'; + +-- ============================================ +-- 9. 代理升级记录表 +-- ============================================ +CREATE TABLE `agent_upgrade` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `agent_id` bigint NOT NULL COMMENT '被升级的代理ID', + `from_level` tinyint NOT NULL COMMENT '原等级：1=普通，2=黄金，3=钻石', + `to_level` tinyint NOT NULL COMMENT '目标等级：1=普通，2=黄金，3=钻石', + `upgrade_type` tinyint NOT NULL COMMENT '升级类型：1=自主付费，2=钻石升级下级', + `upgrade_fee` decimal(10,2) DEFAULT 0.00 COMMENT '升级费用', + `rebate_amount` decimal(10,2) DEFAULT 0.00 COMMENT '返佣金额（给原直接上级）', + `rebate_agent_id` bigint DEFAULT NULL COMMENT '返佣代理ID（原直接上级）', + `operator_agent_id` bigint DEFAULT NULL COMMENT '操作代理ID（如果是钻石升级下级，记录操作者）', + `order_no` varchar(100) DEFAULT NULL COMMENT '支付订单号（如果是自主付费）', + `status` tinyint NOT NULL DEFAULT 1 COMMENT '状态：1=待处理，2=已完成，3=已失败', + `remark` varchar(500) DEFAULT NULL COMMENT '备注', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + KEY `idx_agent_id` (`agent_id`), + KEY `idx_rebate_agent_id` (`rebate_agent_id`), + KEY `idx_operator_agent_id` (`operator_agent_id`), + KEY `idx_upgrade_type` (`upgrade_type`), + KEY `idx_status` (`status`), + KEY `idx_agent_status` (`agent_id`, `status`) COMMENT '复合索引：查询代理的升级记录', + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理升级记录表'; + +-- ============================================ +-- 10. 代理提现表 +-- ============================================ +CREATE TABLE `agent_withdrawal` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `agent_id` bigint NOT NULL COMMENT '代理ID', + `withdraw_no` varchar(100) NOT NULL COMMENT '提现单号', + `payee_account` varchar(100) NOT NULL COMMENT '收款账户', + `payee_name` varchar(50) NOT NULL COMMENT '收款人姓名', + `amount` decimal(10,2) NOT NULL COMMENT '提现金额', + `actual_amount` decimal(10,2) NOT NULL COMMENT '实际到账金额（扣除税费后）', + `tax_amount` decimal(10,2) NOT NULL DEFAULT 0.00 COMMENT '税费金额', + `status` tinyint NOT NULL DEFAULT 1 COMMENT '状态：1=处理中，2=成功，3=失败', + `remark` varchar(500) DEFAULT NULL COMMENT '备注', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + UNIQUE KEY `uk_withdraw_no` (`withdraw_no`), + KEY `idx_agent_id` (`agent_id`), + KEY `idx_status` (`status`), + KEY `idx_agent_status` (`agent_id`, `status`) COMMENT '复合索引：查询代理的提现记录', + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理提现表'; + +-- ============================================ +-- 11. 代理系统配置表 +-- ============================================ +CREATE TABLE `agent_config` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `config_key` varchar(100) NOT NULL COMMENT '配置键', + `config_value` varchar(500) NOT NULL COMMENT '配置值', + `config_type` varchar(50) NOT NULL COMMENT '配置类型：price=价格，bonus=等级加成，upgrade=升级费用，rebate=返佣，tax=税费', + `description` varchar(500) DEFAULT NULL COMMENT '配置描述', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + UNIQUE KEY `uk_config_key` (`config_key`), + KEY `idx_config_type` (`config_type`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理系统配置表'; + +-- ============================================ +-- 12. 代理产品配置表（简化版） +-- ============================================ +CREATE TABLE `agent_product_config` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `product_id` bigint NOT NULL COMMENT '产品ID', + `product_name` varchar(100) NOT NULL COMMENT '产品名称', + `base_price` decimal(10,2) NOT NULL COMMENT '基础底价（BasePrice）', + `system_max_price` decimal(10,2) NOT NULL COMMENT '系统价格上限（SystemMaxPrice）', + `price_threshold` decimal(10,2) DEFAULT NULL COMMENT '提价标准阈值（PriceThreshold）', + `price_fee_rate` decimal(5,4) DEFAULT NULL COMMENT '提价手续费比例（PriceFeeRate）', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + UNIQUE KEY `uk_product_id` (`product_id`), + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理产品配置表'; + +-- ============================================ +-- 13. 代理实名认证表 +-- ============================================ +CREATE TABLE `agent_real_name` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `agent_id` bigint NOT NULL COMMENT '代理ID', + `name` varchar(50) NOT NULL COMMENT '真实姓名', + `id_card` varchar(50) NOT NULL COMMENT '身份证号（加密）', + `mobile` varchar(50) NOT NULL COMMENT '手机号（加密）', + `verify_time` datetime DEFAULT NULL COMMENT '验证时间（三要素验证通过时间，NULL表示未验证）', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + UNIQUE KEY `uk_agent_id` (`agent_id`), + KEY `idx_verify_time` (`verify_time`), + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理实名认证表'; + +-- ============================================ +-- 14. 代理提现扣税记录表 +-- ============================================ +CREATE TABLE `agent_withdrawal_tax` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `agent_id` bigint NOT NULL COMMENT '代理ID', + `withdrawal_id` bigint NOT NULL COMMENT '提现记录ID', + `year_month` int NOT NULL COMMENT '年月（格式：YYYYMM）', + `withdrawal_amount` decimal(10,2) NOT NULL COMMENT '提现金额', + `taxable_amount` decimal(10,2) NOT NULL COMMENT '应税金额', + `tax_rate` decimal(5,4) NOT NULL COMMENT '税率', + `tax_amount` decimal(10,2) NOT NULL COMMENT '税费金额', + `actual_amount` decimal(10,2) NOT NULL COMMENT '实际到账金额', + `tax_status` tinyint NOT NULL DEFAULT 1 COMMENT '扣税状态：1=待扣税，2=已扣税，3=扣税失败', + `tax_time` datetime DEFAULT NULL COMMENT '扣税时间', + `remark` varchar(500) DEFAULT NULL COMMENT '备注', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + KEY `idx_agent_id` (`agent_id`), + KEY `idx_withdrawal_id` (`withdrawal_id`), + KEY `idx_year_month` (`year_month`), + KEY `idx_tax_status` (`tax_status`), + KEY `idx_agent_year_month` (`agent_id`, `year_month`) COMMENT '复合索引：查询代理的月度扣税记录' +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理提现扣税记录表'; + +-- ============================================ +-- 初始化系统配置数据 +-- ============================================ +INSERT INTO `agent_config` (`config_key`, `config_value`, `config_type`, `description`) VALUES +('base_price', '0.00', 'price', '系统基础底价（BasePrice）'), +('system_max_price', '9999.99', 'price', '系统价格上限（SystemMaxPrice）'), +('price_threshold', '0.00', 'price', '提价标准阈值（PriceThreshold）'), +('price_fee_rate', '0.0000', 'price', '提价手续费比例（PriceFeeRate）'), +('level_bonus_normal', '6.00', 'bonus', '普通代理等级加成（6元）'), +('level_bonus_gold', '3.00', 'bonus', '黄金代理等级加成（3元）'), +('level_bonus_diamond', '0.00', 'bonus', '钻石代理等级加成（0元）'), +('upgrade_fee_normal_to_gold', '199.00', 'upgrade', '普通→黄金升级费用（199元）'), +('upgrade_fee_to_diamond', '980.00', 'upgrade', '升级为钻石费用（980元）'), +('upgrade_rebate_normal_to_gold', '139.00', 'upgrade', '普通→黄金返佣金额（139元）'), +('upgrade_rebate_to_diamond', '680.00', 'upgrade', '升级为钻石返佣金额（680元）'), +('direct_parent_amount_diamond', '6.00', 'rebate', '直接上级是钻石的返佣金额（6元）'), +('direct_parent_amount_gold', '3.00', 'rebate', '直接上级是黄金的返佣金额（3元）'), +('direct_parent_amount_normal', '2.00', 'rebate', '直接上级是普通的返佣金额（2元）'), +('max_gold_rebate_amount', '3.00', 'rebate', '黄金代理最大返佣金额（3元）'), +('tax_rate', '0.0600', 'tax', '提现税率（6%）'); + +-- ============================================ +-- 15. 代理邀请码表 +-- ============================================ +CREATE TABLE `agent_invite_code` ( + `id` bigint NOT NULL AUTO_INCREMENT COMMENT '主键ID', + `code` varchar(50) NOT NULL COMMENT '邀请码（唯一）', + `agent_id` bigint DEFAULT NULL COMMENT '发放代理ID（NULL表示平台发放的钻石邀请码）', + `target_level` tinyint NOT NULL DEFAULT 1 COMMENT '目标等级：1=普通，2=黄金，3=钻石', + `status` tinyint NOT NULL DEFAULT 0 COMMENT '状态：0=未使用，1=已使用，2=已失效（钻石邀请码只能使用一次，普通邀请码可无限使用）', + `used_user_id` bigint DEFAULT NULL COMMENT '使用用户ID', + `used_agent_id` bigint DEFAULT NULL COMMENT '使用代理ID', + `used_time` datetime DEFAULT NULL COMMENT '使用时间', + `expire_time` datetime DEFAULT NULL COMMENT '过期时间（可选）', + `remark` varchar(500) DEFAULT NULL COMMENT '备注', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT '创建时间', + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT '更新时间', + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT 0 COMMENT '删除状态：0=未删除，1=已删除', + `version` bigint NOT NULL DEFAULT 0 COMMENT '版本号（乐观锁）', + PRIMARY KEY (`id`), + UNIQUE KEY `uk_code` (`code`), + KEY `idx_agent_id` (`agent_id`), + KEY `idx_status` (`status`), + KEY `idx_target_level` (`target_level`), + KEY `idx_used_user_id` (`used_user_id`), + KEY `idx_create_time` (`create_time`) +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='代理邀请码表'; + +-- ============================================ +-- 表创建完成 +-- ============================================ +-- 执行完成后，请使用goctl生成新的Model代码 +-- 参考命令见 generate_agent_models.md 文件 + diff --git a/deploy/sql/generate_agent_models.md b/deploy/sql/generate_agent_models.md new file mode 100644 index 0000000..c14bcc5 --- /dev/null +++ b/deploy/sql/generate_agent_models.md @@ -0,0 +1,190 @@ +# 生成新代理系统Model说明 + +## 一、执行数据库迁移 + +### 1.1 备份数据库（重要！） + +```bash +# 备份整个数据库 +mysqldump -u root -p your_database > backup_before_migration.sql + +# 或者只备份agent相关表 +mysqldump -u root -p your_database agent agent_audit agent_closure agent_commission agent_wallet agent_link agent_order agent_rewards agent_membership_config agent_product_config agent_real_name agent_withdrawal agent_withdrawal_tax agent_withdrawal_tax_exemption agent_active_stat agent_platform_deduction agent_commission_deduction agent_membership_recharge_order agent_membership_user_config > agent_tables_backup.sql +``` + +### 1.2 执行迁移SQL + +```bash +# 方式1：使用mysql命令行 +mysql -u root -p your_database < deploy/sql/agent_system_migration.sql + +# 方式2：使用数据库客户端工具（如Navicat、DBeaver等） +# 直接执行 deploy/sql/agent_system_migration.sql 文件 +``` + +## 二、生成Model代码 + +### 2.1 方式一：使用脚本批量生成（推荐） + +**Windows (PowerShell):** + +```powershell +# 修改脚本中的数据库配置 +# 然后执行（需要安装goctl工具） +cd deploy/sql +bash generate_agent_models.sh +``` + +**Linux/Mac:** + +```bash +# 修改脚本中的数据库配置 +chmod +x deploy/sql/generate_agent_models.sh +cd deploy/sql +./generate_agent_models.sh +``` + +### 2.2 方式二：使用goctl命令逐个生成 + +```bash +# 1. agent表 +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 2. agent_audit表 +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_audit" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 3. agent_wallet表 +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_wallet" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 4. agent_relation表（新表，替代agent_closure） +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_relation" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 5. agent_link表 +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_link" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 6. agent_order表 +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_order" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 7. agent_commission表 +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_commission" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 8. agent_rebate表（新表） +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_rebate" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 9. agent_upgrade表（新表） +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_upgrade" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 10. agent_withdrawal表 +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_withdrawal" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 11. agent_config表（新表） +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_config" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 12. agent_product_config表 +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_product_config" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 13. agent_real_name表 +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_real_name" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" + +# 14. agent_withdrawal_tax表 +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent_withdrawal_tax" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" +``` + +### 2.3 方式三：使用goctl一次性生成所有表 + +```bash +# 生成所有agent相关表的Model +goctl model mysql datasource -url="root:password@tcp(localhost:3306)/database" -table="agent,agent_audit,agent_wallet,agent_relation,agent_link,agent_order,agent_commission,agent_rebate,agent_upgrade,agent_withdrawal,agent_config,agent_product_config,agent_real_name,agent_withdrawal_tax" -dir="app/main/model" -cache=true --style=goZero --home="./deploy/template" +``` + +## 三、表结构变化说明 + +### 3.1 删除的旧表 + +以下表在新系统中已删除，不再需要： + +- `agent_closure` → 被 `agent_relation` 替代 +- `agent_commission_deduction` → 新系统不需要 +- `agent_rewards` → 新系统不需要 +- `agent_membership_config` → 新系统不需要 +- `agent_membership_recharge_order` → 新系统不需要 +- `agent_membership_user_config` → 新系统不需要 +- `agent_platform_deduction` → 新系统不需要 +- `agent_active_stat` → 新系统不需要 +- `agent_withdrawal_tax_exemption` → 新系统不需要 + +### 3.2 新增的表 + +- `agent_relation` - 替代 `agent_closure`，支持关系脱离 +- `agent_rebate` - 等级加成返佣记录表 +- `agent_upgrade` - 代理升级记录表 +- `agent_config` - 代理系统配置表 + +### 3.3 结构变化的表 + +以下表结构有变化，需要重新生成Model： + +- `agent` - 等级字段改为数值型，新增 `team_leader_id` +- `agent_audit` - 新增 `ancestor_mobile`、`audit_user_id` 字段 +- `agent_wallet` - 字段保持不变 +- `agent_link` - 新增唯一约束 +- `agent_order` - 新增 `order_amount`、`process_status` 等字段 +- `agent_commission` - 字段保持不变 +- `agent_product_config` - 字段保持不变 +- `agent_real_name` - 新增 `audit_user_id` 字段 +- `agent_withdrawal` - 字段保持不变 +- `agent_withdrawal_tax` - 字段保持不变 + +## 四、生成后的文件结构 + +生成后，会在 `app/main/model` 目录下创建以下文件： + +``` +app/main/model/ +├── agentModel.go # agent表Model（自定义方法） +├── agentModel_gen.go # agent表Model（自动生成） +├── agentAuditModel.go # agent_audit表Model（自定义方法） +├── agentAuditModel_gen.go # agent_audit表Model（自动生成） +├── agentWalletModel.go # agent_wallet表Model（自定义方法） +├── agentWalletModel_gen.go # agent_wallet表Model（自动生成） +├── agentRelationModel.go # agent_relation表Model（自定义方法，新） +├── agentRelationModel_gen.go # agent_relation表Model（自动生成，新） +├── agentLinkModel.go # agent_link表Model（自定义方法） +├── agentLinkModel_gen.go # agent_link表Model（自动生成） +├── agentOrderModel.go # agent_order表Model（自定义方法） +├── agentOrderModel_gen.go # agent_order表Model（自动生成） +├── agentCommissionModel.go # agent_commission表Model（自定义方法） +├── agentCommissionModel_gen.go # agent_commission表Model（自动生成） +├── agentRebateModel.go # agent_rebate表Model（自定义方法，新） +├── agentRebateModel_gen.go # agent_rebate表Model（自动生成，新） +├── agentUpgradeModel.go # agent_upgrade表Model（自定义方法，新） +├── agentUpgradeModel_gen.go # agent_upgrade表Model（自动生成，新） +├── agentWithdrawalModel.go # agent_withdrawal表Model（自定义方法） +├── agentWithdrawalModel_gen.go # agent_withdrawal表Model（自动生成） +├── agentConfigModel.go # agent_config表Model（自定义方法，新） +├── agentConfigModel_gen.go # agent_config表Model（自动生成，新） +├── agentProductConfigModel.go # agent_product_config表Model（自定义方法） +├── agentProductConfigModel_gen.go # agent_product_config表Model（自动生成） +├── agentRealNameModel.go # agent_real_name表Model（自定义方法） +├── agentRealNameModel_gen.go # agent_real_name表Model（自动生成） +├── agentWithdrawalTaxModel.go # agent_withdrawal_tax表Model（自定义方法） +└── agentWithdrawalTaxModel_gen.go # agent_withdrawal_tax表Model（自动生成） +``` + +## 五、注意事项 + +1. **备份数据**：执行迁移前务必备份数据库 +2. **测试环境**：建议先在测试环境执行，验证无误后再在生产环境执行 +3. **代码兼容**：生成新Model后，需要更新相关业务代码以适配新的表结构 +4. **删除旧Model**：生成新Model后，需要删除旧的Model文件（如 `agentClosureModel.go`） +5. **更新ServiceContext**：需要在 `svc/servicecontext.go` 中更新Model初始化代码 + +## 六、验证 + +生成Model后，请验证： + +1. 所有新表都有对应的Model文件 +2. Model文件可以正常编译 +3. 数据库连接正常 +4. 基本的CRUD操作可以正常执行 + diff --git a/deploy/sql/generate_agent_models.sh b/deploy/sql/generate_agent_models.sh new file mode 100644 index 0000000..763e0bb --- /dev/null +++ b/deploy/sql/generate_agent_models.sh @@ -0,0 +1,70 @@ +#!/bin/bash +# ============================================ +# 生成新代理系统Model脚本 +# 使用goctl工具根据数据库表生成Model代码 +# ============================================ + +# 数据库配置（请根据实际情况修改） +DB_HOST="localhost" +DB_PORT="3306" +DB_USER="root" +DB_PASSWORD="your_password" +DB_NAME="your_database" + +# Model输出目录 +MODEL_DIR="app/main/model" + +# goctl模板目录 +TEMPLATE_DIR="./deploy/template" + +echo "============================================" +echo "开始生成新代理系统Model..." +echo "============================================" + +# 生成各个表的Model +echo "1. 生成 agent Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "2. 生成 agent_audit Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_audit" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "3. 生成 agent_wallet Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_wallet" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "4. 生成 agent_relation Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_relation" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "5. 生成 agent_link Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_link" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "6. 生成 agent_order Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_order" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "7. 生成 agent_commission Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_commission" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "8. 生成 agent_rebate Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_rebate" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "9. 生成 agent_upgrade Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_upgrade" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "10. 生成 agent_withdrawal Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_withdrawal" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "11. 生成 agent_config Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_config" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "12. 生成 agent_product_config Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_product_config" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "13. 生成 agent_real_name Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_real_name" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "14. 生成 agent_withdrawal_tax Model..." +goctl model mysql datasource -url="${DB_USER}:${DB_PASSWORD}@tcp(${DB_HOST}:${DB_PORT})/${DB_NAME}" -table="agent_withdrawal_tax" -dir="${MODEL_DIR}" -cache=true --style=goZero --home="${TEMPLATE_DIR}" + +echo "============================================" +echo "Model生成完成！" +echo "============================================" + diff --git a/deploy/sql/remove_product_name_from_agent_product_config.sql b/deploy/sql/remove_product_name_from_agent_product_config.sql new file mode 100644 index 0000000..30f2950 --- /dev/null +++ b/deploy/sql/remove_product_name_from_agent_product_config.sql @@ -0,0 +1,25 @@ +-- ============================================ +-- 移除代理产品配置表中的 product_name 字段 +-- 说明：改为通过 product_id 关联查询 product 表获取产品名称 +-- 执行时间：2025-01-XX +-- ============================================ + +-- 删除 product_name 字段 +ALTER TABLE `agent_product_config` DROP COLUMN `product_name`; + +-- ============================================ +-- 验证查询：检查字段是否已删除 +-- ============================================ +DESC `agent_product_config`; + +-- ============================================ +-- 验证查询：通过关联查询获取产品名称 +-- ============================================ +SELECT apc.id, apc.product_id, p.product_name, apc.base_price, apc.system_max_price, apc.price_threshold, apc.price_fee_rate, apc.del_state, apc.create_time +FROM + `agent_product_config` apc + LEFT JOIN `product` p ON apc.product_id = p.id + AND p.del_state = 0 +WHERE + apc.del_state = 0 +ORDER BY apc.product_id; \ No newline at end of file diff --git a/deploy/sql/sync_agent_product_config.sql b/deploy/sql/sync_agent_product_config.sql new file mode 100644 index 0000000..d5e0c4d --- /dev/null +++ b/deploy/sql/sync_agent_product_config.sql @@ -0,0 +1,112 @@ +-- ============================================ +-- 同步产品表数据到代理产品配置表 +-- 说明：为现有产品创建对应的代理产品配置记录 +-- 执行时间：2025-01-XX +-- ============================================ + +-- 方式1：使用 INSERT IGNORE（如果记录已存在则忽略） +-- 注意：product_name 字段已移除，改为通过 product_id 关联查询 product 表获取 +INSERT IGNORE INTO + `agent_product_config` ( + `product_id`, + `base_price`, + `system_max_price`, + `price_threshold`, + `price_fee_rate`, + `del_state`, + `version` + ) +VALUES ( + 1, + 0.00, + 9999.99, + NULL, + NULL, + 0, + 0 + ), + ( + 2, + 0.00, + 9999.99, + NULL, + NULL, + 0, + 0 + ), + ( + 3, + 0.00, + 9999.99, + NULL, + NULL, + 0, + 0 + ), + ( + 4, + 0.00, + 9999.99, + NULL, + NULL, + 0, + 0 + ), + ( + 5, + 0.00, + 9999.99, + NULL, + NULL, + 0, + 0 + ), + ( + 6, + 0.00, + 9999.99, + NULL, + NULL, + 0, + 0 + ), + ( + 7, + 0.00, + 9999.99, + NULL, + NULL, + 0, + 0 + ); + +-- ============================================ +-- 方式2：使用 INSERT ... ON DUPLICATE KEY UPDATE（如果记录已存在则忽略） +-- 注意：product_name 字段已移除，不再需要更新 +-- ============================================ +/* +INSERT INTO `agent_product_config` +(`product_id`, `base_price`, `system_max_price`, `price_threshold`, `price_fee_rate`, `del_state`, `version`) +VALUES +(1, 0.00, 9999.99, NULL, NULL, 0, 0), +(2, 0.00, 9999.99, NULL, NULL, 0, 0), +(3, 0.00, 9999.99, NULL, NULL, 0, 0), +(4, 0.00, 9999.99, NULL, NULL, 0, 0), +(5, 0.00, 9999.99, NULL, NULL, 0, 0), +(6, 0.00, 9999.99, NULL, NULL, 0, 0), +(7, 0.00, 9999.99, NULL, NULL, 0, 0) +ON DUPLICATE KEY UPDATE +`product_id` = VALUES(`product_id`); +*/ + +-- ============================================ +-- 验证查询：检查同步结果 +-- ============================================ +SELECT apc.id, apc.product_id, p.product_name, apc.base_price, apc.system_max_price, apc.price_threshold, apc.price_fee_rate, apc.del_state, apc.create_time +FROM + `agent_product_config` apc + LEFT JOIN `product` p ON apc.product_id = p.id + AND p.del_state = 0 +WHERE + apc.del_state = 0 +ORDER BY apc.product_id; \ No newline at end of file diff --git a/deploy/sql/template.sql b/deploy/sql/template.sql new file mode 100644 index 0000000..70e809d --- /dev/null +++ b/deploy/sql/template.sql @@ -0,0 +1,62 @@ +-- ============================================ +-- 表结构模板（UUID版本） +-- ============================================ +-- 注意：系统已迁移到UUID主键，新表请使用此模板 +-- ============================================ + +CREATE TABLE `表名` ( + `id` CHAR(36) NOT NULL COMMENT 'UUID主键', + `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP, + `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, + `delete_time` datetime DEFAULT NULL COMMENT '删除时间', + `del_state` tinyint NOT NULL DEFAULT '0', + `version` bigint NOT NULL DEFAULT '0' COMMENT '版本号', + +/* 业务字段开始 */ + +`字段1` 数据类型 [约束条件] [DEFAULT 默认值] [COMMENT '字段说明'], + `字段2` 数据类型 [约束条件] [DEFAULT 默认值] [COMMENT '字段说明'], + /* 关联字段 - 软关联（使用UUID） */ + `关联表id` CHAR(36) [NOT NULL] [DEFAULT NULL] COMMENT '关联到XX表的UUID', + /* 业务字段结束 */ + + PRIMARY KEY (`id`), + /* 索引定义 */ + UNIQUE KEY `索引名称` (`字段名`), + KEY `idx_关联字段` (`关联表id`) COMMENT '优化关联查询' +) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='表说明'; + +-- ============================================ +-- UUID生成说明 +-- ============================================ +-- 1. 应用层生成：使用Go的uuid.NewString()生成UUID +-- 示例：id := uuid.NewString() +-- 2. 数据库层生成：使用MySQL的UUID()函数（不推荐，性能较差） +-- 示例：INSERT INTO table (id, ...) VALUES (UUID(), ...) +-- 3. 推荐方式：在应用层生成UUID，然后插入数据库 +-- ============================================ + +-- ============================================ +-- 旧版本模板（bigint主键，已废弃） +-- ============================================ +-- CREATE TABLE `表名` ( +-- `id` bigint NOT NULL AUTO_INCREMENT, +-- `create_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP, +-- `update_time` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP, +-- `delete_time` datetime DEFAULT NULL COMMENT '删除时间', +-- `del_state` tinyint NOT NULL DEFAULT '0', +-- `version` bigint NOT NULL DEFAULT '0' COMMENT '版本号', +-- +-- /* 业务字段开始 */ +-- `字段1` 数据类型 [约束条件] [DEFAULT 默认值] [COMMENT '字段说明'], +-- `字段2` 数据类型 [约束条件] [DEFAULT 默认值] [COMMENT '字段说明'], +-- /* 关联字段 - 软关联 */ +-- `关联表id` bigint [NOT NULL] [DEFAULT '0'] COMMENT '关联到XX表的id', +-- /* 业务字段结束 */ +-- +-- PRIMARY KEY (`id`), +-- /* 索引定义 */ +-- UNIQUE KEY `索引名称` (`字段名`), +-- KEY `idx_关联字段` (`关联表id`) COMMENT '优化关联查询' +-- ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='表说明'; +-- ============================================ \ No newline at end of file diff --git a/deploy/sql/user_system_refactor.sql b/deploy/sql/user_system_refactor.sql new file mode 100644 index 0000000..8dfd90c --- /dev/null +++ b/deploy/sql/user_system_refactor.sql @@ -0,0 +1,48 @@ +-- ============================================ +-- 用户系统重构SQL脚本（简化版 - 开发环境） +-- 根据《用户系统重构计划书》执行 +-- ============================================ + +-- ============================================ +-- 第一部分：UserAuth表唯一索引 +-- ============================================ + +-- 确保user_auth表的(auth_type, auth_key)唯一性 +-- 如果已存在同名索引，先删除 +ALTER TABLE `user_auth` DROP INDEX IF EXISTS `uk_auth_type_key`; + +-- 添加唯一索引 +ALTER TABLE `user_auth` +ADD UNIQUE INDEX `uk_auth_type_key` (`auth_type`, `auth_key`) COMMENT '确保同一个认证方式只能绑定一个用户'; + +-- 允许一个用户绑定同一认证类型的多个记录（例如多个UUID） +ALTER TABLE `user_auth` DROP INDEX IF EXISTS `unique_userId_key`; +ALTER TABLE `user_auth` ADD INDEX `idx_user_id_auth_type` (`user_id`, `auth_type`); + +-- ============================================ +-- 第二部分：删除UserTemp表（开发环境直接删除） +-- ============================================ + +-- 注意：开发环境没有数据，直接删除表即可 +-- 如果表不存在，会报错但可以忽略 + +DROP TABLE IF EXISTS `user_temp`; + +-- ============================================ +-- 第三部分：索引优化（可选） +-- ============================================ + +-- 为user_auth表添加user_id和auth_type的联合索引（如果查询频繁） +-- ALTER TABLE `user_auth` +-- ADD INDEX `idx_user_id_auth_type` (`user_id`, `auth_type`) COMMENT '优化按用户查询认证类型'; + +-- 为user表添加mobile的唯一索引（如果不存在） +-- ALTER TABLE `user` +-- ADD UNIQUE INDEX `uk_mobile` (`mobile`) COMMENT '确保手机号唯一性'; + +-- ============================================ +-- 注意事项 +-- ============================================ +-- 1. 此脚本适用于开发环境，直接删除UserTemp表 +-- 2. 如果生产环境有数据，请使用完整版迁移脚本 +-- 3. 执行后，确保代码中已移除所有对UserTemp表的引用 diff --git a/deploy/sql/user_temp_migration.sql b/deploy/sql/user_temp_migration.sql new file mode 100644 index 0000000..17e4bad --- /dev/null +++ b/deploy/sql/user_temp_migration.sql @@ -0,0 +1,45 @@ +-- 将 user_temp 迁移为 user（mobile 为空）与 user_auth + +START TRANSACTION; + +-- 1) 迁移临时用户到 user（mobile 置为空） +INSERT INTO + `user` ( + `delete_time`, + `del_state`, + `version`, + `mobile`, + `password`, + `nickname`, + `info`, + `inside` + ) +SELECT NULL, 0, COALESCE(ut.version, 0), NULL, NULL, NULL, '', 0 +FROM `user_temp` ut +WHERE + ut.del_state = 0; + +-- 2) 将临时认证迁移到 user_auth（按插入顺序关联最近插入的 user.id） +INSERT INTO + `user_auth` ( + `delete_time`, + `del_state`, + `version`, + `user_id`, + `auth_key`, + `auth_type` + ) +SELECT NULL, 0, COALESCE(ut.version, 0), u.id, ut.auth_key, ut.auth_type +FROM `user_temp` ut + JOIN `user` u ON u.del_state = 0 + AND u.mobile IS NULL +WHERE + ut.del_state = 0; + +-- 注意：以上为示意，实际生产应通过显式映射（如临时ID与新UserID映射表）确保一一对应，避免笛卡尔匹配。 + +COMMIT; + +-- 唯一索引保障 +ALTER TABLE `user_auth` +ADD UNIQUE INDEX `idx_auth_type_key` (`auth_type`, `auth_key`); \ No newline at end of file diff --git a/deploy/sql/uuid_migration.sql b/deploy/sql/uuid_migration.sql new file mode 100644 index 0000000..9da7cf2 --- /dev/null +++ b/deploy/sql/uuid_migration.sql @@ -0,0 +1,1831 @@ +-- ============================================ +-- UUID迁移脚本 +-- 将系统中所有bigint类型的ID字段改为CHAR(36)类型的UUID +-- ============================================ + +-- 注意：此脚本需要分阶段执行，建议在测试环境充分测试后再在生产环境执行 +-- 执行前请务必备份数据库！ + +-- ============================================ +-- 第一阶段：创建UUID映射表（用于数据迁移） +-- ============================================ + +-- 创建临时映射表，用于存储旧ID和新UUID的对应关系 +CREATE TABLE IF NOT EXISTS `uuid_mapping` ( + `table_name` VARCHAR(64) NOT NULL COMMENT '表名', + `old_id` BIGINT NOT NULL COMMENT '旧ID', + `new_uuid` CHAR(36) NOT NULL COMMENT '新UUID', + PRIMARY KEY (`table_name`, `old_id`), + KEY `idx_new_uuid` (`new_uuid`) +) ENGINE = InnoDB DEFAULT CHARSET = utf8mb4 COLLATE = utf8mb4_general_ci COMMENT = 'UUID映射表'; + +-- ============================================ +-- 第二阶段：为所有表生成UUID映射 +-- ============================================ + +-- 为user表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'user', `id`, UUID() +FROM `user` +WHERE + `del_state` = 0; + +-- 为agent表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent', `id`, UUID() +FROM `agent` +WHERE + `del_state` = 0; + +-- 为product表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'product', `id`, UUID() +FROM `product` +WHERE + `del_state` = 0; + +-- 为order表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'order', `id`, UUID() +FROM `order` +WHERE + `del_state` = 0; + +-- 为query表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'query', `id`, UUID() +FROM `query` +WHERE + `del_state` = 0; + +-- 为user_auth表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'user_auth', `id`, UUID() +FROM `user_auth` +WHERE + `del_state` = 0; + +-- 为agent_commission表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_commission', `id`, UUID() +FROM `agent_commission` +WHERE + `del_state` = 0; + +-- 为agent_invite_code表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_invite_code', `id`, UUID() +FROM `agent_invite_code` +WHERE + `del_state` = 0; + +-- 为agent_link表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_link', `id`, UUID() +FROM `agent_link` +WHERE + `del_state` = 0; + +-- 为agent_order表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_order', `id`, UUID() +FROM `agent_order` +WHERE + `del_state` = 0; + +-- 为agent_wallet表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_wallet', `id`, UUID() +FROM `agent_wallet` +WHERE + `del_state` = 0; + +-- 为agent_withdrawal表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_withdrawal', `id`, UUID() +FROM `agent_withdrawal` +WHERE + `del_state` = 0; + +-- 为agent_withdrawal_tax表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_withdrawal_tax', `id`, UUID() +FROM `agent_withdrawal_tax` +WHERE + `del_state` = 0; + +-- 为agent_rebate表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_rebate', `id`, UUID() +FROM `agent_rebate` +WHERE + `del_state` = 0; + +-- 为agent_relation表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_relation', `id`, UUID() +FROM `agent_relation` +WHERE + `del_state` = 0; + +-- 为agent_upgrade表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_upgrade', `id`, UUID() +FROM `agent_upgrade` +WHERE + `del_state` = 0; + +-- 为agent_real_name表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_real_name', `id`, UUID() +FROM `agent_real_name` +WHERE + `del_state` = 0; + +-- 为agent_config表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_config', `id`, UUID() +FROM `agent_config` +WHERE + `del_state` = 0; + +-- 为agent_product_config表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_product_config', `id`, UUID() +FROM `agent_product_config` +WHERE + `del_state` = 0; + +-- 为agent_short_link表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_short_link', `id`, UUID() +FROM `agent_short_link` +WHERE + `del_state` = 0; + +-- 为agent_invite_code_usage表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_invite_code_usage', `id`, UUID() +FROM `agent_invite_code_usage` +WHERE + `del_state` = 0; + +-- 为agent_freeze_task表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'agent_freeze_task', `id`, UUID() +FROM `agent_freeze_task` +WHERE + `del_state` = 0; + +-- 为order_refund表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'order_refund', `id`, UUID() +FROM `order_refund` +WHERE + `del_state` = 0; + +-- 为query_cleanup_log表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'query_cleanup_log', `id`, UUID() +FROM `query_cleanup_log` +WHERE + `del_state` = 0; + +-- 为query_cleanup_detail表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'query_cleanup_detail', `id`, UUID() +FROM `query_cleanup_detail` +WHERE + `del_state` = 0; + +-- 为query_cleanup_config表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'query_cleanup_config', `id`, UUID() +FROM `query_cleanup_config` +WHERE + `del_state` = 0; + +-- 为product_feature表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'product_feature', `id`, UUID() +FROM `product_feature` +WHERE + `del_state` = 0; + +-- 为feature表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'feature', `id`, UUID() +FROM `feature` +WHERE + `del_state` = 0; + +-- 为authorization_document表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'authorization_document', `id`, UUID() +FROM `authorization_document` +WHERE + `del_state` = 0; + +-- 为global_notifications表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'global_notifications', `id`, UUID() +FROM `global_notifications` +WHERE + `del_state` = 0; + +-- 为admin_user表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'admin_user', `id`, UUID() +FROM `admin_user` +WHERE + `del_state` = 0; + +-- 为admin_role表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'admin_role', `id`, UUID() +FROM `admin_role` +WHERE + `del_state` = 0; + +-- 为admin_menu表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'admin_menu', `id`, UUID() +FROM `admin_menu` +WHERE + `del_state` = 0; + +-- 为admin_api表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'admin_api', `id`, UUID() +FROM `admin_api` +WHERE + `del_state` = 0; + +-- 为admin_dict_type表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'admin_dict_type', `id`, UUID() +FROM `admin_dict_type` +WHERE + `del_state` = 0; + +-- 为admin_dict_data表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'admin_dict_data', `id`, UUID() +FROM `admin_dict_data` +WHERE + `del_state` = 0; + +-- 为admin_user_role表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'admin_user_role', `id`, UUID() +FROM `admin_user_role` +WHERE + `del_state` = 0; + +-- 为admin_role_menu表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'admin_role_menu', `id`, UUID() +FROM `admin_role_menu` +WHERE + `del_state` = 0; + +-- 为admin_role_api表生成UUID映射 +INSERT INTO + `uuid_mapping` ( + `table_name`, + `old_id`, + `new_uuid` + ) +SELECT 'admin_role_api', `id`, UUID() +FROM `admin_role_api` +WHERE + `del_state` = 0; + +-- ============================================ +-- 第三阶段：修改表结构（添加UUID字段，保留原ID字段） +-- ============================================ + +-- 注意：此阶段先添加新字段，不删除旧字段，以便回滚 + +-- user表：添加UUID主键字段 +ALTER TABLE `user` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD INDEX `idx_id_uuid` (`id_uuid`); + +-- agent表：添加UUID主键字段 +ALTER TABLE `agent` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `user_id_uuid` CHAR(36) NULL COMMENT '用户UUID' AFTER `user_id`, +ADD COLUMN `team_leader_id_uuid` CHAR(36) NULL COMMENT '团队首领UUID' AFTER `team_leader_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_user_id_uuid` (`user_id_uuid`), +ADD INDEX `idx_team_leader_id_uuid` (`team_leader_id_uuid`); + +-- product表：添加UUID主键字段 +ALTER TABLE `product` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD INDEX `idx_id_uuid` (`id_uuid`); + +-- order表：添加UUID主键字段和外键字段 +ALTER TABLE `order` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `user_id_uuid` CHAR(36) NULL COMMENT '用户UUID' AFTER `user_id`, +ADD COLUMN `product_id_uuid` CHAR(36) NULL COMMENT '产品UUID' AFTER `product_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_user_id_uuid` (`user_id_uuid`), +ADD INDEX `idx_product_id_uuid` (`product_id_uuid`); + +-- query表：添加UUID主键字段和外键字段 +ALTER TABLE `query` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `order_id_uuid` CHAR(36) NULL COMMENT '订单UUID' AFTER `order_id`, +ADD COLUMN `user_id_uuid` CHAR(36) NULL COMMENT '用户UUID' AFTER `user_id`, +ADD COLUMN `product_id_uuid` CHAR(36) NULL COMMENT '产品UUID' AFTER `product_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_order_id_uuid` (`order_id_uuid`), +ADD INDEX `idx_user_id_uuid` (`user_id_uuid`), +ADD INDEX `idx_product_id_uuid` (`product_id_uuid`); + +-- user_auth表：添加UUID主键字段和外键字段 +ALTER TABLE `user_auth` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `user_id_uuid` CHAR(36) NULL COMMENT '用户UUID' AFTER `user_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_user_id_uuid` (`user_id_uuid`); + +-- agent_commission表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_commission` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD COLUMN `order_id_uuid` CHAR(36) NULL COMMENT '订单UUID' AFTER `order_id`, +ADD COLUMN `product_id_uuid` CHAR(36) NULL COMMENT '产品UUID' AFTER `product_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`), +ADD INDEX `idx_order_id_uuid` (`order_id_uuid`), +ADD INDEX `idx_product_id_uuid` (`product_id_uuid`); + +-- agent_invite_code表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_invite_code` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD COLUMN `used_user_id_uuid` CHAR(36) NULL COMMENT '使用用户UUID' AFTER `used_user_id`, +ADD COLUMN `used_agent_id_uuid` CHAR(36) NULL COMMENT '使用代理UUID' AFTER `used_agent_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`), +ADD INDEX `idx_used_user_id_uuid` (`used_user_id_uuid`), +ADD INDEX `idx_used_agent_id_uuid` (`used_agent_id_uuid`); + +-- agent_link表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_link` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD COLUMN `user_id_uuid` CHAR(36) NULL COMMENT '用户UUID' AFTER `user_id`, +ADD COLUMN `product_id_uuid` CHAR(36) NULL COMMENT '产品UUID' AFTER `product_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`), +ADD INDEX `idx_user_id_uuid` (`user_id_uuid`), +ADD INDEX `idx_product_id_uuid` (`product_id_uuid`); + +-- agent_order表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_order` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD COLUMN `order_id_uuid` CHAR(36) NULL COMMENT '订单UUID' AFTER `order_id`, +ADD COLUMN `product_id_uuid` CHAR(36) NULL COMMENT '产品UUID' AFTER `product_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`), +ADD INDEX `idx_order_id_uuid` (`order_id_uuid`), +ADD INDEX `idx_product_id_uuid` (`product_id_uuid`); + +-- agent_wallet表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_wallet` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`); + +-- agent_withdrawal表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_withdrawal` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD COLUMN `wallet_id_uuid` CHAR(36) NULL COMMENT '钱包UUID' AFTER `wallet_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`), +ADD INDEX `idx_wallet_id_uuid` (`wallet_id_uuid`); + +-- agent_withdrawal_tax表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_withdrawal_tax` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `withdrawal_id_uuid` CHAR(36) NULL COMMENT '提现UUID' AFTER `withdrawal_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_withdrawal_id_uuid` (`withdrawal_id_uuid`); + +-- agent_rebate表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_rebate` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD COLUMN `order_id_uuid` CHAR(36) NULL COMMENT '订单UUID' AFTER `order_id`, +ADD COLUMN `product_id_uuid` CHAR(36) NULL COMMENT '产品UUID' AFTER `product_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`), +ADD INDEX `idx_order_id_uuid` (`order_id_uuid`), +ADD INDEX `idx_product_id_uuid` (`product_id_uuid`); + +-- agent_relation表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_relation` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD COLUMN `parent_agent_id_uuid` CHAR(36) NULL COMMENT '上级代理UUID' AFTER `parent_agent_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`), +ADD INDEX `idx_parent_agent_id_uuid` (`parent_agent_id_uuid`); + +-- agent_upgrade表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_upgrade` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`); + +-- agent_real_name表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_real_name` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`); + +-- agent_config表：添加UUID主键字段 +ALTER TABLE `agent_config` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD INDEX `idx_id_uuid` (`id_uuid`); + +-- agent_product_config表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_product_config` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `product_id_uuid` CHAR(36) NULL COMMENT '产品UUID' AFTER `product_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_product_id_uuid` (`product_id_uuid`); + +-- agent_short_link表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_short_link` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `invite_code_id_uuid` CHAR(36) NULL COMMENT '邀请码UUID' AFTER `invite_code_id`, +ADD COLUMN `link_id_uuid` CHAR(36) NULL COMMENT '链接UUID' AFTER `link_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_invite_code_id_uuid` (`invite_code_id_uuid`), +ADD INDEX `idx_link_id_uuid` (`link_id_uuid`); + +-- agent_invite_code_usage表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_invite_code_usage` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `invite_code_id_uuid` CHAR(36) NULL COMMENT '邀请码UUID' AFTER `invite_code_id`, +ADD COLUMN `user_id_uuid` CHAR(36) NULL COMMENT '用户UUID' AFTER `user_id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_invite_code_id_uuid` (`invite_code_id_uuid`), +ADD INDEX `idx_user_id_uuid` (`user_id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`); + +-- agent_freeze_task表：添加UUID主键字段和外键字段 +ALTER TABLE `agent_freeze_task` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `agent_id_uuid` CHAR(36) NULL COMMENT '代理UUID' AFTER `agent_id`, +ADD COLUMN `order_id_uuid` CHAR(36) NULL COMMENT '订单UUID' AFTER `order_id`, +ADD COLUMN `commission_id_uuid` CHAR(36) NULL COMMENT '佣金UUID' AFTER `commission_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_agent_id_uuid` (`agent_id_uuid`), +ADD INDEX `idx_order_id_uuid` (`order_id_uuid`), +ADD INDEX `idx_commission_id_uuid` (`commission_id_uuid`); + +-- order_refund表：添加UUID主键字段和外键字段 +ALTER TABLE `order_refund` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `order_id_uuid` CHAR(36) NULL COMMENT '订单UUID' AFTER `order_id`, +ADD COLUMN `user_id_uuid` CHAR(36) NULL COMMENT '用户UUID' AFTER `user_id`, +ADD COLUMN `product_id_uuid` CHAR(36) NULL COMMENT '产品UUID' AFTER `product_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_order_id_uuid` (`order_id_uuid`), +ADD INDEX `idx_user_id_uuid` (`user_id_uuid`), +ADD INDEX `idx_product_id_uuid` (`product_id_uuid`); + +-- query_cleanup_log表：添加UUID主键字段 +ALTER TABLE `query_cleanup_log` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD INDEX `idx_id_uuid` (`id_uuid`); + +-- query_cleanup_detail表：添加UUID主键字段和外键字段 +ALTER TABLE `query_cleanup_detail` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `cleanup_log_id_uuid` CHAR(36) NULL COMMENT '清理日志UUID' AFTER `cleanup_log_id`, +ADD COLUMN `query_id_uuid` CHAR(36) NULL COMMENT '查询UUID' AFTER `query_id`, +ADD COLUMN `order_id_uuid` CHAR(36) NULL COMMENT '订单UUID' AFTER `order_id`, +ADD COLUMN `user_id_uuid` CHAR(36) NULL COMMENT '用户UUID' AFTER `user_id`, +ADD COLUMN `product_id_uuid` CHAR(36) NULL COMMENT '产品UUID' AFTER `product_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_cleanup_log_id_uuid` (`cleanup_log_id_uuid`), +ADD INDEX `idx_query_id_uuid` (`query_id_uuid`), +ADD INDEX `idx_order_id_uuid` (`order_id_uuid`), +ADD INDEX `idx_user_id_uuid` (`user_id_uuid`), +ADD INDEX `idx_product_id_uuid` (`product_id_uuid`); + +-- query_cleanup_config表：添加UUID主键字段 +ALTER TABLE `query_cleanup_config` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD INDEX `idx_id_uuid` (`id_uuid`); + +-- product_feature表：添加UUID主键字段和外键字段 +ALTER TABLE `product_feature` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `product_id_uuid` CHAR(36) NULL COMMENT '产品UUID' AFTER `product_id`, +ADD COLUMN `feature_id_uuid` CHAR(36) NULL COMMENT '功能UUID' AFTER `feature_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_product_id_uuid` (`product_id_uuid`), +ADD INDEX `idx_feature_id_uuid` (`feature_id_uuid`); + +-- feature表：添加UUID主键字段 +ALTER TABLE `feature` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD INDEX `idx_id_uuid` (`id_uuid`); + +-- authorization_document表：添加UUID主键字段和外键字段 +ALTER TABLE `authorization_document` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `user_id_uuid` CHAR(36) NULL COMMENT '用户UUID' AFTER `user_id`, +ADD COLUMN `order_id_uuid` CHAR(36) NULL COMMENT '订单UUID' AFTER `order_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_user_id_uuid` (`user_id_uuid`), +ADD INDEX `idx_order_id_uuid` (`order_id_uuid`); + +-- global_notifications表：添加UUID主键字段 +ALTER TABLE `global_notifications` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD INDEX `idx_id_uuid` (`id_uuid`); + +-- admin_user表：添加UUID主键字段 +ALTER TABLE `admin_user` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD INDEX `idx_id_uuid` (`id_uuid`); + +-- admin_role表：添加UUID主键字段 +ALTER TABLE `admin_role` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD INDEX `idx_id_uuid` (`id_uuid`); + +-- admin_menu表：添加UUID主键字段和外键字段 +ALTER TABLE `admin_menu` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `parent_id_uuid` CHAR(36) NULL COMMENT '父菜单UUID' AFTER `parent_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_parent_id_uuid` (`parent_id_uuid`); + +-- admin_api表：添加UUID主键字段 +ALTER TABLE `admin_api` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD INDEX `idx_id_uuid` (`id_uuid`); + +-- admin_dict_type表：添加UUID主键字段 +ALTER TABLE `admin_dict_type` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD INDEX `idx_id_uuid` (`id_uuid`); + +-- admin_dict_data表：添加UUID主键字段和外键字段 +ALTER TABLE `admin_dict_data` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `dict_type_id_uuid` CHAR(36) NULL COMMENT '字典类型UUID' AFTER `dict_type_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_dict_type_id_uuid` (`dict_type_id_uuid`); + +-- admin_user_role表：添加UUID主键字段和外键字段 +ALTER TABLE `admin_user_role` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `user_id_uuid` CHAR(36) NULL COMMENT '用户UUID' AFTER `user_id`, +ADD COLUMN `role_id_uuid` CHAR(36) NULL COMMENT '角色UUID' AFTER `role_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_user_id_uuid` (`user_id_uuid`), +ADD INDEX `idx_role_id_uuid` (`role_id_uuid`); + +-- admin_role_menu表：添加UUID主键字段和外键字段 +ALTER TABLE `admin_role_menu` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `role_id_uuid` CHAR(36) NULL COMMENT '角色UUID' AFTER `role_id`, +ADD COLUMN `menu_id_uuid` CHAR(36) NULL COMMENT '菜单UUID' AFTER `menu_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_role_id_uuid` (`role_id_uuid`), +ADD INDEX `idx_menu_id_uuid` (`menu_id_uuid`); + +-- admin_role_api表：添加UUID主键字段和外键字段 +ALTER TABLE `admin_role_api` +ADD COLUMN `id_uuid` CHAR(36) NULL COMMENT 'UUID主键' AFTER `id`, +ADD COLUMN `role_id_uuid` CHAR(36) NULL COMMENT '角色UUID' AFTER `role_id`, +ADD COLUMN `api_id_uuid` CHAR(36) NULL COMMENT 'API UUID' AFTER `api_id`, +ADD INDEX `idx_id_uuid` (`id_uuid`), +ADD INDEX `idx_role_id_uuid` (`role_id_uuid`), +ADD INDEX `idx_api_id_uuid` (`api_id_uuid`); + +-- ============================================ +-- 第四阶段：填充UUID字段数据（根据映射表） +-- ============================================ + +-- user表：填充UUID +UPDATE `user` u +INNER JOIN `uuid_mapping` m ON m.table_name = 'user' +AND m.old_id = u.id +SET + u.id_uuid = m.new_uuid +WHERE + u.del_state = 0; + +-- agent表：填充UUID +UPDATE `agent` a +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = a.id +SET + a.id_uuid = m.new_uuid +WHERE + a.del_state = 0; + +UPDATE `agent` a +INNER JOIN `uuid_mapping` m ON m.table_name = 'user' +AND m.old_id = a.user_id +SET + a.user_id_uuid = m.new_uuid +WHERE + a.del_state = 0 + AND a.user_id IS NOT NULL; + +UPDATE `agent` a +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = a.team_leader_id +SET + a.team_leader_id_uuid = m.new_uuid +WHERE + a.del_state = 0 + AND a.team_leader_id IS NOT NULL; + +-- product表：填充UUID +UPDATE `product` p +INNER JOIN `uuid_mapping` m ON m.table_name = 'product' +AND m.old_id = p.id +SET + p.id_uuid = m.new_uuid +WHERE + p.del_state = 0; + +-- order表：填充UUID +UPDATE `order` o +INNER JOIN `uuid_mapping` m ON m.table_name = 'order' +AND m.old_id = o.id +SET + o.id_uuid = m.new_uuid +WHERE + o.del_state = 0; + +UPDATE `order` o +INNER JOIN `uuid_mapping` m ON m.table_name = 'user' +AND m.old_id = o.user_id +SET + o.user_id_uuid = m.new_uuid +WHERE + o.del_state = 0 + AND o.user_id IS NOT NULL; + +UPDATE `order` o +INNER JOIN `uuid_mapping` m ON m.table_name = 'product' +AND m.old_id = o.product_id +SET + o.product_id_uuid = m.new_uuid +WHERE + o.del_state = 0 + AND o.product_id IS NOT NULL; + +-- query表：填充UUID +UPDATE `query` q +INNER JOIN `uuid_mapping` m ON m.table_name = 'query' +AND m.old_id = q.id +SET + q.id_uuid = m.new_uuid +WHERE + q.del_state = 0; + +UPDATE `query` q +INNER JOIN `uuid_mapping` m ON m.table_name = 'order' +AND m.old_id = q.order_id +SET + q.order_id_uuid = m.new_uuid +WHERE + q.del_state = 0 + AND q.order_id IS NOT NULL; + +UPDATE `query` q +INNER JOIN `uuid_mapping` m ON m.table_name = 'user' +AND m.old_id = q.user_id +SET + q.user_id_uuid = m.new_uuid +WHERE + q.del_state = 0 + AND q.user_id IS NOT NULL; + +UPDATE `query` q +INNER JOIN `uuid_mapping` m ON m.table_name = 'product' +AND m.old_id = q.product_id +SET + q.product_id_uuid = m.new_uuid +WHERE + q.del_state = 0 + AND q.product_id IS NOT NULL; + +-- user_auth表：填充UUID +UPDATE `user_auth` ua +INNER JOIN `uuid_mapping` m ON m.table_name = 'user_auth' +AND m.old_id = ua.id +SET + ua.id_uuid = m.new_uuid +WHERE + ua.del_state = 0; + +UPDATE `user_auth` ua +INNER JOIN `uuid_mapping` m ON m.table_name = 'user' +AND m.old_id = ua.user_id +SET + ua.user_id_uuid = m.new_uuid +WHERE + ua.del_state = 0 + AND ua.user_id IS NOT NULL; + +-- agent_commission表：填充UUID +UPDATE `agent_commission` ac +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_commission' +AND m.old_id = ac.id +SET + ac.id_uuid = m.new_uuid +WHERE + ac.del_state = 0; + +UPDATE `agent_commission` ac +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = ac.agent_id +SET + ac.agent_id_uuid = m.new_uuid +WHERE + ac.del_state = 0 + AND ac.agent_id IS NOT NULL; + +UPDATE `agent_commission` ac +INNER JOIN `uuid_mapping` m ON m.table_name = 'order' +AND m.old_id = ac.order_id +SET + ac.order_id_uuid = m.new_uuid +WHERE + ac.del_state = 0 + AND ac.order_id IS NOT NULL; + +UPDATE `agent_commission` ac +INNER JOIN `uuid_mapping` m ON m.table_name = 'product' +AND m.old_id = ac.product_id +SET + ac.product_id_uuid = m.new_uuid +WHERE + ac.del_state = 0 + AND ac.product_id IS NOT NULL; + +-- agent_invite_code表：填充UUID +UPDATE `agent_invite_code` aic +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_invite_code' +AND m.old_id = aic.id +SET + aic.id_uuid = m.new_uuid +WHERE + aic.del_state = 0; + +UPDATE `agent_invite_code` aic +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = aic.agent_id +SET + aic.agent_id_uuid = m.new_uuid +WHERE + aic.del_state = 0 + AND aic.agent_id IS NOT NULL; + +UPDATE `agent_invite_code` aic +INNER JOIN `uuid_mapping` m ON m.table_name = 'user' +AND m.old_id = aic.used_user_id +SET + aic.used_user_id_uuid = m.new_uuid +WHERE + aic.del_state = 0 + AND aic.used_user_id IS NOT NULL; + +UPDATE `agent_invite_code` aic +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = aic.used_agent_id +SET + aic.used_agent_id_uuid = m.new_uuid +WHERE + aic.del_state = 0 + AND aic.used_agent_id IS NOT NULL; + +-- agent_link表：填充UUID +UPDATE `agent_link` al +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_link' +AND m.old_id = al.id +SET + al.id_uuid = m.new_uuid +WHERE + al.del_state = 0; + +UPDATE `agent_link` al +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = al.agent_id +SET + al.agent_id_uuid = m.new_uuid +WHERE + al.del_state = 0 + AND al.agent_id IS NOT NULL; + +UPDATE `agent_link` al +INNER JOIN `uuid_mapping` m ON m.table_name = 'user' +AND m.old_id = al.user_id +SET + al.user_id_uuid = m.new_uuid +WHERE + al.del_state = 0 + AND al.user_id IS NOT NULL; + +UPDATE `agent_link` al +INNER JOIN `uuid_mapping` m ON m.table_name = 'product' +AND m.old_id = al.product_id +SET + al.product_id_uuid = m.new_uuid +WHERE + al.del_state = 0 + AND al.product_id IS NOT NULL; + +-- agent_order表：填充UUID +UPDATE `agent_order` ao +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_order' +AND m.old_id = ao.id +SET + ao.id_uuid = m.new_uuid +WHERE + ao.del_state = 0; + +UPDATE `agent_order` ao +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = ao.agent_id +SET + ao.agent_id_uuid = m.new_uuid +WHERE + ao.del_state = 0 + AND ao.agent_id IS NOT NULL; + +UPDATE `agent_order` ao +INNER JOIN `uuid_mapping` m ON m.table_name = 'order' +AND m.old_id = ao.order_id +SET + ao.order_id_uuid = m.new_uuid +WHERE + ao.del_state = 0 + AND ao.order_id IS NOT NULL; + +UPDATE `agent_order` ao +INNER JOIN `uuid_mapping` m ON m.table_name = 'product' +AND m.old_id = ao.product_id +SET + ao.product_id_uuid = m.new_uuid +WHERE + ao.del_state = 0 + AND ao.product_id IS NOT NULL; + +-- agent_wallet表：填充UUID +UPDATE `agent_wallet` aw +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_wallet' +AND m.old_id = aw.id +SET + aw.id_uuid = m.new_uuid +WHERE + aw.del_state = 0; + +UPDATE `agent_wallet` aw +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = aw.agent_id +SET + aw.agent_id_uuid = m.new_uuid +WHERE + aw.del_state = 0 + AND aw.agent_id IS NOT NULL; + +-- agent_withdrawal表：填充UUID +UPDATE `agent_withdrawal` aw +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_withdrawal' +AND m.old_id = aw.id +SET + aw.id_uuid = m.new_uuid +WHERE + aw.del_state = 0; + +UPDATE `agent_withdrawal` aw +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = aw.agent_id +SET + aw.agent_id_uuid = m.new_uuid +WHERE + aw.del_state = 0 + AND aw.agent_id IS NOT NULL; + +UPDATE `agent_withdrawal` aw +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_wallet' +AND m.old_id = aw.wallet_id +SET + aw.wallet_id_uuid = m.new_uuid +WHERE + aw.del_state = 0 + AND aw.wallet_id IS NOT NULL; + +-- agent_withdrawal_tax表：填充UUID +UPDATE `agent_withdrawal_tax` awt +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_withdrawal_tax' +AND m.old_id = awt.id +SET + awt.id_uuid = m.new_uuid +WHERE + awt.del_state = 0; + +UPDATE `agent_withdrawal_tax` awt +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_withdrawal' +AND m.old_id = awt.withdrawal_id +SET + awt.withdrawal_id_uuid = m.new_uuid +WHERE + awt.del_state = 0 + AND awt.withdrawal_id IS NOT NULL; + +-- agent_rebate表：填充UUID +UPDATE `agent_rebate` ar +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_rebate' +AND m.old_id = ar.id +SET + ar.id_uuid = m.new_uuid +WHERE + ar.del_state = 0; + +UPDATE `agent_rebate` ar +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = ar.agent_id +SET + ar.agent_id_uuid = m.new_uuid +WHERE + ar.del_state = 0 + AND ar.agent_id IS NOT NULL; + +UPDATE `agent_rebate` ar +INNER JOIN `uuid_mapping` m ON m.table_name = 'order' +AND m.old_id = ar.order_id +SET + ar.order_id_uuid = m.new_uuid +WHERE + ar.del_state = 0 + AND ar.order_id IS NOT NULL; + +UPDATE `agent_rebate` ar +INNER JOIN `uuid_mapping` m ON m.table_name = 'product' +AND m.old_id = ar.product_id +SET + ar.product_id_uuid = m.new_uuid +WHERE + ar.del_state = 0 + AND ar.product_id IS NOT NULL; + +-- agent_relation表：填充UUID +UPDATE `agent_relation` ar +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_relation' +AND m.old_id = ar.id +SET + ar.id_uuid = m.new_uuid +WHERE + ar.del_state = 0; + +UPDATE `agent_relation` ar +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = ar.agent_id +SET + ar.agent_id_uuid = m.new_uuid +WHERE + ar.del_state = 0 + AND ar.agent_id IS NOT NULL; + +UPDATE `agent_relation` ar +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = ar.parent_agent_id +SET + ar.parent_agent_id_uuid = m.new_uuid +WHERE + ar.del_state = 0 + AND ar.parent_agent_id IS NOT NULL; + +-- agent_upgrade表：填充UUID +UPDATE `agent_upgrade` au +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_upgrade' +AND m.old_id = au.id +SET + au.id_uuid = m.new_uuid +WHERE + au.del_state = 0; + +UPDATE `agent_upgrade` au +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = au.agent_id +SET + au.agent_id_uuid = m.new_uuid +WHERE + au.del_state = 0 + AND au.agent_id IS NOT NULL; + +-- agent_real_name表：填充UUID +UPDATE `agent_real_name` arn +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_real_name' +AND m.old_id = arn.id +SET + arn.id_uuid = m.new_uuid +WHERE + arn.del_state = 0; + +UPDATE `agent_real_name` arn +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = arn.agent_id +SET + arn.agent_id_uuid = m.new_uuid +WHERE + arn.del_state = 0 + AND arn.agent_id IS NOT NULL; + +-- agent_config表：填充UUID +UPDATE `agent_config` ac +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_config' +AND m.old_id = ac.id +SET + ac.id_uuid = m.new_uuid +WHERE + ac.del_state = 0; + +-- agent_product_config表：填充UUID +UPDATE `agent_product_config` apc +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_product_config' +AND m.old_id = apc.id +SET + apc.id_uuid = m.new_uuid +WHERE + apc.del_state = 0; + +UPDATE `agent_product_config` apc +INNER JOIN `uuid_mapping` m ON m.table_name = 'product' +AND m.old_id = apc.product_id +SET + apc.product_id_uuid = m.new_uuid +WHERE + apc.del_state = 0 + AND apc.product_id IS NOT NULL; + +-- agent_short_link表：填充UUID +UPDATE `agent_short_link` asl +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_short_link' +AND m.old_id = asl.id +SET + asl.id_uuid = m.new_uuid +WHERE + asl.del_state = 0; + +UPDATE `agent_short_link` asl +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_invite_code' +AND m.old_id = asl.invite_code_id +SET + asl.invite_code_id_uuid = m.new_uuid +WHERE + asl.del_state = 0 + AND asl.invite_code_id IS NOT NULL; + +UPDATE `agent_short_link` asl +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_link' +AND m.old_id = asl.link_id +SET + asl.link_id_uuid = m.new_uuid +WHERE + asl.del_state = 0 + AND asl.link_id IS NOT NULL; + +-- agent_invite_code_usage表：填充UUID +UPDATE `agent_invite_code_usage` aicu +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_invite_code_usage' +AND m.old_id = aicu.id +SET + aicu.id_uuid = m.new_uuid +WHERE + aicu.del_state = 0; + +UPDATE `agent_invite_code_usage` aicu +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_invite_code' +AND m.old_id = aicu.invite_code_id +SET + aicu.invite_code_id_uuid = m.new_uuid +WHERE + aicu.del_state = 0 + AND aicu.invite_code_id IS NOT NULL; + +UPDATE `agent_invite_code_usage` aicu +INNER JOIN `uuid_mapping` m ON m.table_name = 'user' +AND m.old_id = aicu.user_id +SET + aicu.user_id_uuid = m.new_uuid +WHERE + aicu.del_state = 0 + AND aicu.user_id IS NOT NULL; + +UPDATE `agent_invite_code_usage` aicu +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = aicu.agent_id +SET + aicu.agent_id_uuid = m.new_uuid +WHERE + aicu.del_state = 0 + AND aicu.agent_id IS NOT NULL; + +-- agent_freeze_task表：填充UUID +UPDATE `agent_freeze_task` aft +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_freeze_task' +AND m.old_id = aft.id +SET + aft.id_uuid = m.new_uuid +WHERE + aft.del_state = 0; + +UPDATE `agent_freeze_task` aft +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent' +AND m.old_id = aft.agent_id +SET + aft.agent_id_uuid = m.new_uuid +WHERE + aft.del_state = 0 + AND aft.agent_id IS NOT NULL; + +UPDATE `agent_freeze_task` aft +INNER JOIN `uuid_mapping` m ON m.table_name = 'order' +AND m.old_id = aft.order_id +SET + aft.order_id_uuid = m.new_uuid +WHERE + aft.del_state = 0 + AND aft.order_id IS NOT NULL; + +UPDATE `agent_freeze_task` aft +INNER JOIN `uuid_mapping` m ON m.table_name = 'agent_commission' +AND m.old_id = aft.commission_id +SET + aft.commission_id_uuid = m.new_uuid +WHERE + aft.del_state = 0 + AND aft.commission_id IS NOT NULL; + +-- order_refund表：填充UUID +UPDATE `order_refund` orf +INNER JOIN `uuid_mapping` m ON m.table_name = 'order_refund' +AND m.old_id = orf.id +SET + orf.id_uuid = m.new_uuid +WHERE + orf.del_state = 0; + +UPDATE `order_refund` orf +INNER JOIN `uuid_mapping` m ON m.table_name = 'order' +AND m.old_id = orf.order_id +SET + orf.order_id_uuid = m.new_uuid +WHERE + orf.del_state = 0 + AND orf.order_id IS NOT NULL; + +UPDATE `order_refund` orf +INNER JOIN `uuid_mapping` m ON m.table_name = 'user' +AND m.old_id = orf.user_id +SET + orf.user_id_uuid = m.new_uuid +WHERE + orf.del_state = 0 + AND orf.user_id IS NOT NULL; + +UPDATE `order_refund` orf +INNER JOIN `uuid_mapping` m ON m.table_name = 'product' +AND m.old_id = orf.product_id +SET + orf.product_id_uuid = m.new_uuid +WHERE + orf.del_state = 0 + AND orf.product_id IS NOT NULL; + +-- query_cleanup_log表：填充UUID +UPDATE `query_cleanup_log` qcl +INNER JOIN `uuid_mapping` m ON m.table_name = 'query_cleanup_log' +AND m.old_id = qcl.id +SET + qcl.id_uuid = m.new_uuid +WHERE + qcl.del_state = 0; + +-- query_cleanup_detail表：填充UUID +UPDATE `query_cleanup_detail` qcd +INNER JOIN `uuid_mapping` m ON m.table_name = 'query_cleanup_detail' +AND m.old_id = qcd.id +SET + qcd.id_uuid = m.new_uuid +WHERE + qcd.del_state = 0; + +UPDATE `query_cleanup_detail` qcd +INNER JOIN `uuid_mapping` m ON m.table_name = 'query_cleanup_log' +AND m.old_id = qcd.cleanup_log_id +SET + qcd.cleanup_log_id_uuid = m.new_uuid +WHERE + qcd.del_state = 0 + AND qcd.cleanup_log_id IS NOT NULL; + +UPDATE `query_cleanup_detail` qcd +INNER JOIN `uuid_mapping` m ON m.table_name = 'query' +AND m.old_id = qcd.query_id +SET + qcd.query_id_uuid = m.new_uuid +WHERE + qcd.del_state = 0 + AND qcd.query_id IS NOT NULL; + +UPDATE `query_cleanup_detail` qcd +INNER JOIN `uuid_mapping` m ON m.table_name = 'order' +AND m.old_id = qcd.order_id +SET + qcd.order_id_uuid = m.new_uuid +WHERE + qcd.del_state = 0 + AND qcd.order_id IS NOT NULL; + +UPDATE `query_cleanup_detail` qcd +INNER JOIN `uuid_mapping` m ON m.table_name = 'user' +AND m.old_id = qcd.user_id +SET + qcd.user_id_uuid = m.new_uuid +WHERE + qcd.del_state = 0 + AND qcd.user_id IS NOT NULL; + +UPDATE `query_cleanup_detail` qcd +INNER JOIN `uuid_mapping` m ON m.table_name = 'product' +AND m.old_id = qcd.product_id +SET + qcd.product_id_uuid = m.new_uuid +WHERE + qcd.del_state = 0 + AND qcd.product_id IS NOT NULL; + +-- query_cleanup_config表：填充UUID +UPDATE `query_cleanup_config` qcc +INNER JOIN `uuid_mapping` m ON m.table_name = 'query_cleanup_config' +AND m.old_id = qcc.id +SET + qcc.id_uuid = m.new_uuid +WHERE + qcc.del_state = 0; + +-- product_feature表：填充UUID +UPDATE `product_feature` pf +INNER JOIN `uuid_mapping` m ON m.table_name = 'product_feature' +AND m.old_id = pf.id +SET + pf.id_uuid = m.new_uuid +WHERE + pf.del_state = 0; + +UPDATE `product_feature` pf +INNER JOIN `uuid_mapping` m ON m.table_name = 'product' +AND m.old_id = pf.product_id +SET + pf.product_id_uuid = m.new_uuid +WHERE + pf.del_state = 0 + AND pf.product_id IS NOT NULL; + +UPDATE `product_feature` pf +INNER JOIN `uuid_mapping` m ON m.table_name = 'feature' +AND m.old_id = pf.feature_id +SET + pf.feature_id_uuid = m.new_uuid +WHERE + pf.del_state = 0 + AND pf.feature_id IS NOT NULL; + +-- feature表：填充UUID +UPDATE `feature` f +INNER JOIN `uuid_mapping` m ON m.table_name = 'feature' +AND m.old_id = f.id +SET + f.id_uuid = m.new_uuid +WHERE + f.del_state = 0; + +-- authorization_document表：填充UUID +UPDATE `authorization_document` ad +INNER JOIN `uuid_mapping` m ON m.table_name = 'authorization_document' +AND m.old_id = ad.id +SET + ad.id_uuid = m.new_uuid +WHERE + ad.del_state = 0; + +UPDATE `authorization_document` ad +INNER JOIN `uuid_mapping` m ON m.table_name = 'user' +AND m.old_id = ad.user_id +SET + ad.user_id_uuid = m.new_uuid +WHERE + ad.del_state = 0 + AND ad.user_id IS NOT NULL; + +UPDATE `authorization_document` ad +INNER JOIN `uuid_mapping` m ON m.table_name = 'order' +AND m.old_id = ad.order_id +SET + ad.order_id_uuid = m.new_uuid +WHERE + ad.del_state = 0 + AND ad.order_id IS NOT NULL; + +-- global_notifications表：填充UUID +UPDATE `global_notifications` gn +INNER JOIN `uuid_mapping` m ON m.table_name = 'global_notifications' +AND m.old_id = gn.id +SET + gn.id_uuid = m.new_uuid +WHERE + gn.del_state = 0; + +-- admin_user表：填充UUID +UPDATE `admin_user` au +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_user' +AND m.old_id = au.id +SET + au.id_uuid = m.new_uuid +WHERE + au.del_state = 0; + +-- admin_role表：填充UUID +UPDATE `admin_role` ar +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_role' +AND m.old_id = ar.id +SET + ar.id_uuid = m.new_uuid +WHERE + ar.del_state = 0; + +-- admin_menu表：填充UUID +UPDATE `admin_menu` am +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_menu' +AND m.old_id = am.id +SET + am.id_uuid = m.new_uuid +WHERE + am.del_state = 0; + +UPDATE `admin_menu` am +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_menu' +AND m.old_id = am.parent_id +SET + am.parent_id_uuid = m.new_uuid +WHERE + am.del_state = 0 + AND am.parent_id IS NOT NULL; + +-- admin_api表：填充UUID +UPDATE `admin_api` aa +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_api' +AND m.old_id = aa.id +SET + aa.id_uuid = m.new_uuid +WHERE + aa.del_state = 0; + +-- admin_dict_type表：填充UUID +UPDATE `admin_dict_type` adt +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_dict_type' +AND m.old_id = adt.id +SET + adt.id_uuid = m.new_uuid +WHERE + adt.del_state = 0; + +-- admin_dict_data表：填充UUID +UPDATE `admin_dict_data` add +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_dict_data' AND m.old_id = add.id +SET add.id_uuid = m.new_uuid +WHERE add.del_state = 0; + +UPDATE `admin_dict_data` add +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_dict_type' AND m.old_id = add.dict_type_id +SET add.dict_type_id_uuid = m.new_uuid +WHERE add.del_state = 0 AND add.dict_type_id IS NOT NULL; + +-- admin_user_role表：填充UUID +UPDATE `admin_user_role` aur +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_user_role' +AND m.old_id = aur.id +SET + aur.id_uuid = m.new_uuid +WHERE + aur.del_state = 0; + +UPDATE `admin_user_role` aur +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_user' +AND m.old_id = aur.user_id +SET + aur.user_id_uuid = m.new_uuid +WHERE + aur.del_state = 0 + AND aur.user_id IS NOT NULL; + +UPDATE `admin_user_role` aur +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_role' +AND m.old_id = aur.role_id +SET + aur.role_id_uuid = m.new_uuid +WHERE + aur.del_state = 0 + AND aur.role_id IS NOT NULL; + +-- admin_role_menu表：填充UUID +UPDATE `admin_role_menu` arm +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_role_menu' +AND m.old_id = arm.id +SET + arm.id_uuid = m.new_uuid +WHERE + arm.del_state = 0; + +UPDATE `admin_role_menu` arm +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_role' +AND m.old_id = arm.role_id +SET + arm.role_id_uuid = m.new_uuid +WHERE + arm.del_state = 0 + AND arm.role_id IS NOT NULL; + +UPDATE `admin_role_menu` arm +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_menu' +AND m.old_id = arm.menu_id +SET + arm.menu_id_uuid = m.new_uuid +WHERE + arm.del_state = 0 + AND arm.menu_id IS NOT NULL; + +-- admin_role_api表：填充UUID +UPDATE `admin_role_api` ara +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_role_api' +AND m.old_id = ara.id +SET + ara.id_uuid = m.new_uuid +WHERE + ara.del_state = 0; + +UPDATE `admin_role_api` ara +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_role' +AND m.old_id = ara.role_id +SET + ara.role_id_uuid = m.new_uuid +WHERE + ara.del_state = 0 + AND ara.role_id IS NOT NULL; + +UPDATE `admin_role_api` ara +INNER JOIN `uuid_mapping` m ON m.table_name = 'admin_api' +AND m.old_id = ara.api_id +SET + ara.api_id_uuid = m.new_uuid +WHERE + ara.del_state = 0 + AND ara.api_id IS NOT NULL; + +-- ============================================ +-- 第五阶段：验证数据完整性 +-- ============================================ + +-- 检查是否有NULL的UUID字段（不应该有） +SELECT 'user' as table_name, COUNT(*) as null_count +FROM `user` +WHERE + `id_uuid` IS NULL + AND `del_state` = 0 +UNION ALL +SELECT 'agent', COUNT(*) +FROM `agent` +WHERE + `id_uuid` IS NULL + AND `del_state` = 0 +UNION ALL +SELECT 'order', COUNT(*) +FROM `order` +WHERE + `id_uuid` IS NULL + AND `del_state` = 0 +UNION ALL +SELECT 'query', COUNT(*) +FROM `query` +WHERE + `id_uuid` IS NULL + AND `del_state` = 0; + +-- ============================================ +-- 第六阶段：切换主键（需要停止服务） +-- ============================================ + +-- 注意：此阶段需要停止服务，因为会修改主键和索引 +-- 建议在维护窗口期执行 + +-- 删除旧主键，将UUID字段设为主键 +-- 注意：需要先删除所有外键约束（如果有的话） + +-- user表：切换主键 +ALTER TABLE `user` +DROP PRIMARY KEY, +MODIFY COLUMN `id` BIGINT NULL COMMENT '旧ID（保留用于回滚）', +MODIFY COLUMN `id_uuid` CHAR(36) NOT NULL COMMENT 'UUID主键', +ADD PRIMARY KEY (`id_uuid`); + +-- agent表：切换主键 +ALTER TABLE `agent` +DROP PRIMARY KEY, +MODIFY COLUMN `id` BIGINT NULL COMMENT '旧ID（保留用于回滚）', +MODIFY COLUMN `id_uuid` CHAR(36) NOT NULL COMMENT 'UUID主键', +ADD PRIMARY KEY (`id_uuid`); + +-- product表：切换主键 +ALTER TABLE `product` +DROP PRIMARY KEY, +MODIFY COLUMN `id` BIGINT NULL COMMENT '旧ID（保留用于回滚）', +MODIFY COLUMN `id_uuid` CHAR(36) NOT NULL COMMENT 'UUID主键', +ADD PRIMARY KEY (`id_uuid`); + +-- order表：切换主键 +ALTER TABLE `order` +DROP PRIMARY KEY, +MODIFY COLUMN `id` BIGINT NULL COMMENT '旧ID（保留用于回滚）', +MODIFY COLUMN `id_uuid` CHAR(36) NOT NULL COMMENT 'UUID主键', +ADD PRIMARY KEY (`id_uuid`); + +-- query表：切换主键 +ALTER TABLE `query` +DROP PRIMARY KEY, +MODIFY COLUMN `id` BIGINT NULL COMMENT '旧ID（保留用于回滚）', +MODIFY COLUMN `id_uuid` CHAR(36) NOT NULL COMMENT 'UUID主键', +ADD PRIMARY KEY (`id_uuid`); + +-- user_auth表：切换主键 +ALTER TABLE `user_auth` +DROP PRIMARY KEY, +MODIFY COLUMN `id` BIGINT NULL COMMENT '旧ID（保留用于回滚）', +MODIFY COLUMN `id_uuid` CHAR(36) NOT NULL COMMENT 'UUID主键', +ADD PRIMARY KEY (`id_uuid`); + +-- 其他表类似处理... +-- （为节省篇幅，这里省略其他表的切换，实际执行时需要为所有表执行类似操作） + +-- ============================================ +-- 第七阶段：重命名字段（将_uuid后缀去掉） +-- ============================================ + +-- user表：重命名字段 +ALTER TABLE `user` +CHANGE COLUMN `id_uuid` `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +DROP COLUMN `id`; +-- 删除旧ID字段（确认无误后执行） + +-- 注意：其他表也需要类似处理，但需要先确保所有代码都已更新为使用UUID + +-- ============================================ +-- 第八阶段：更新唯一索引和约束 +-- ============================================ + +-- user表：更新mobile唯一索引（如果存在） +-- ALTER TABLE `user` ADD UNIQUE INDEX `uk_mobile` (`mobile`); + +-- user_auth表：更新(auth_type, auth_key)唯一索引 +ALTER TABLE `user_auth` +ADD UNIQUE INDEX `uk_auth_type_key` (`auth_type`, `auth_key`); + +-- agent表：更新user_id唯一索引 +ALTER TABLE `agent` ADD UNIQUE INDEX `uk_user_id` (`user_id_uuid`); + +-- ============================================ +-- 第九阶段：清理临时表 +-- ============================================ + +-- 确认所有数据迁移无误后，删除映射表 +-- DROP TABLE IF EXISTS `uuid_mapping`; + +-- ============================================ +-- 注意事项 +-- ============================================ +-- 1. 此脚本需要分阶段执行，每阶段执行后需要验证数据完整性 +-- 2. 第六阶段（切换主键）需要停止服务 +-- 3. 执行前务必备份数据库 +-- 4. 建议先在测试环境完整执行一遍 +-- 5. 代码层面需要同步修改： +-- - 所有Model的ID字段类型从int64改为string +-- - 所有插入操作需要生成UUID（使用uuid.NewString()） +-- - 所有查询操作需要使用UUID +-- 6. 删除旧ID字段前，确保所有代码都已更新完成 \ No newline at end of file diff --git a/deploy/sql/uuid_migration_simple.sql b/deploy/sql/uuid_migration_simple.sql new file mode 100644 index 0000000..70e1b74 --- /dev/null +++ b/deploy/sql/uuid_migration_simple.sql @@ -0,0 +1,446 @@ +-- ============================================ +-- UUID迁移脚本（简化版 - 开发环境） +-- 将系统中所有bigint类型的ID字段改为CHAR(36)类型的UUID +-- 注意：此脚本直接修改表结构，不保留旧数据，适用于开发环境 +-- ============================================ + +-- 注意：user表和product表已经是CHAR(36)类型，跳过修改 + +-- ============================================ +-- 第一部分：修改核心业务表 +-- ============================================ + +-- user表：已经是CHAR(36)，跳过 +-- product表：已经是CHAR(36)，跳过 + +-- order表：修改主键和外键字段 +ALTER TABLE `order` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `user_id` CHAR(36) NOT NULL COMMENT '用户UUID', +MODIFY COLUMN `product_id` CHAR(36) NOT NULL COMMENT '产品UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_user_id` (`user_id`), +ADD INDEX `idx_product_id` (`product_id`); + +-- query表：修改主键和外键字段 +ALTER TABLE `query` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `order_id` CHAR(36) NOT NULL COMMENT '订单UUID', +MODIFY COLUMN `user_id` CHAR(36) NOT NULL COMMENT '用户UUID', +MODIFY COLUMN `product_id` CHAR(36) NOT NULL COMMENT '产品UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_order_id` (`order_id`), +ADD INDEX `idx_user_id` (`user_id`), +ADD INDEX `idx_product_id` (`product_id`); + +-- user_auth表：修改主键和外键字段（注意：user_id需要关联到user表的UUID） +ALTER TABLE `user_auth` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `user_id` CHAR(36) NOT NULL COMMENT '用户UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_user_id` (`user_id`), +ADD UNIQUE INDEX `uk_auth_type_key` (`auth_type`, `auth_key`) COMMENT '确保同一个认证方式只能绑定一个用户'; + +-- agent表：修改主键和外键字段 +ALTER TABLE `agent` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `user_id` CHAR(36) NOT NULL COMMENT '用户UUID', +MODIFY COLUMN `team_leader_id` CHAR(36) NULL COMMENT '团队首领UUID', +MODIFY COLUMN `invite_code_id` CHAR(36) NULL COMMENT '邀请码UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD UNIQUE INDEX `uk_user_id` (`user_id`), +ADD INDEX `idx_team_leader_id` (`team_leader_id`), +ADD INDEX `idx_invite_code_id` (`invite_code_id`); + +-- ============================================ +-- 第二部分：修改代理相关表 +-- ============================================ + +-- agent_commission表 +ALTER TABLE `agent_commission` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `agent_id` CHAR(36) NOT NULL COMMENT '代理UUID', +MODIFY COLUMN `order_id` CHAR(36) NOT NULL COMMENT '订单UUID', +MODIFY COLUMN `product_id` CHAR(36) NOT NULL COMMENT '产品UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_agent_id` (`agent_id`), +ADD INDEX `idx_order_id` (`order_id`), +ADD INDEX `idx_product_id` (`product_id`); + +-- agent_invite_code表 +ALTER TABLE `agent_invite_code` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `agent_id` CHAR(36) NULL COMMENT '代理UUID', +MODIFY COLUMN `used_user_id` CHAR(36) NULL COMMENT '使用用户UUID', +MODIFY COLUMN `used_agent_id` CHAR(36) NULL COMMENT '使用代理UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_agent_id` (`agent_id`), +ADD INDEX `idx_used_user_id` (`used_user_id`), +ADD INDEX `idx_used_agent_id` (`used_agent_id`); + +-- agent_link表 +ALTER TABLE `agent_link` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `agent_id` CHAR(36) NOT NULL COMMENT '代理UUID', +MODIFY COLUMN `user_id` CHAR(36) NOT NULL COMMENT '用户UUID', +MODIFY COLUMN `product_id` CHAR(36) NOT NULL COMMENT '产品UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_agent_id` (`agent_id`), +ADD INDEX `idx_user_id` (`user_id`), +ADD INDEX `idx_product_id` (`product_id`); + +-- agent_order表 +ALTER TABLE `agent_order` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `agent_id` CHAR(36) NOT NULL COMMENT '代理UUID', +MODIFY COLUMN `order_id` CHAR(36) NOT NULL COMMENT '订单UUID', +MODIFY COLUMN `product_id` CHAR(36) NOT NULL COMMENT '产品UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_agent_id` (`agent_id`), +ADD INDEX `idx_order_id` (`order_id`), +ADD INDEX `idx_product_id` (`product_id`); + +-- agent_wallet表 +ALTER TABLE `agent_wallet` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `agent_id` CHAR(36) NOT NULL COMMENT '代理UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD UNIQUE INDEX `uk_agent_id` (`agent_id`); + +-- agent_withdrawal表 +ALTER TABLE `agent_withdrawal` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `agent_id` CHAR(36) NOT NULL COMMENT '代理UUID', +MODIFY COLUMN `wallet_id` CHAR(36) NOT NULL COMMENT '钱包UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_agent_id` (`agent_id`), +ADD INDEX `idx_wallet_id` (`wallet_id`); + +-- agent_withdrawal_tax表 +ALTER TABLE `agent_withdrawal_tax` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `withdrawal_id` CHAR(36) NOT NULL COMMENT '提现UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_withdrawal_id` (`withdrawal_id`); + +-- agent_rebate表 +ALTER TABLE `agent_rebate` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `agent_id` CHAR(36) NOT NULL COMMENT '代理UUID', +MODIFY COLUMN `order_id` CHAR(36) NOT NULL COMMENT '订单UUID', +MODIFY COLUMN `product_id` CHAR(36) NOT NULL COMMENT '产品UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_agent_id` (`agent_id`), +ADD INDEX `idx_order_id` (`order_id`), +ADD INDEX `idx_product_id` (`product_id`); + +-- agent_relation表 +ALTER TABLE `agent_relation` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `agent_id` CHAR(36) NOT NULL COMMENT '代理UUID', +MODIFY COLUMN `parent_agent_id` CHAR(36) NULL COMMENT '上级代理UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_agent_id` (`agent_id`), +ADD INDEX `idx_parent_agent_id` (`parent_agent_id`); + +-- agent_upgrade表 +ALTER TABLE `agent_upgrade` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `agent_id` CHAR(36) NOT NULL COMMENT '代理UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_agent_id` (`agent_id`); + +-- agent_real_name表 +ALTER TABLE `agent_real_name` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `agent_id` CHAR(36) NOT NULL COMMENT '代理UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_agent_id` (`agent_id`); + +-- agent_config表 +ALTER TABLE `agent_config` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`); + +-- agent_product_config表 +ALTER TABLE `agent_product_config` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `product_id` CHAR(36) NOT NULL COMMENT '产品UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_product_id` (`product_id`); + +-- agent_short_link表 +ALTER TABLE `agent_short_link` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `invite_code_id` CHAR(36) NULL COMMENT '邀请码UUID', +MODIFY COLUMN `link_id` CHAR(36) NULL COMMENT '链接UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_invite_code_id` (`invite_code_id`), +ADD INDEX `idx_link_id` (`link_id`); + +-- agent_invite_code_usage表 +ALTER TABLE `agent_invite_code_usage` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `invite_code_id` CHAR(36) NOT NULL COMMENT '邀请码UUID', +MODIFY COLUMN `user_id` CHAR(36) NOT NULL COMMENT '用户UUID', +MODIFY COLUMN `agent_id` CHAR(36) NOT NULL COMMENT '代理UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_invite_code_id` (`invite_code_id`), +ADD INDEX `idx_user_id` (`user_id`), +ADD INDEX `idx_agent_id` (`agent_id`); + +-- agent_freeze_task表 +ALTER TABLE `agent_freeze_task` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `agent_id` CHAR(36) NOT NULL COMMENT '代理UUID', +MODIFY COLUMN `order_id` CHAR(36) NOT NULL COMMENT '订单UUID', +MODIFY COLUMN `commission_id` CHAR(36) NOT NULL COMMENT '佣金UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_agent_id` (`agent_id`), +ADD INDEX `idx_order_id` (`order_id`), +ADD INDEX `idx_commission_id` (`commission_id`); + +-- ============================================ +-- 第三部分：修改订单相关表 +-- ============================================ + +-- order_refund表 +ALTER TABLE `order_refund` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `order_id` CHAR(36) NOT NULL COMMENT '订单UUID', +MODIFY COLUMN `user_id` CHAR(36) NOT NULL COMMENT '用户UUID', +MODIFY COLUMN `product_id` CHAR(36) NOT NULL COMMENT '产品UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_order_id` (`order_id`), +ADD INDEX `idx_user_id` (`user_id`), +ADD INDEX `idx_product_id` (`product_id`); + +-- ============================================ +-- 第四部分：修改查询相关表 +-- ============================================ + +-- query_cleanup_log表 +ALTER TABLE `query_cleanup_log` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`); + +-- query_cleanup_detail表 +ALTER TABLE `query_cleanup_detail` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `cleanup_log_id` CHAR(36) NOT NULL COMMENT '清理日志UUID', +MODIFY COLUMN `query_id` CHAR(36) NOT NULL COMMENT '查询UUID', +MODIFY COLUMN `order_id` CHAR(36) NOT NULL COMMENT '订单UUID', +MODIFY COLUMN `user_id` CHAR(36) NOT NULL COMMENT '用户UUID', +MODIFY COLUMN `product_id` CHAR(36) NOT NULL COMMENT '产品UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_cleanup_log_id` (`cleanup_log_id`), +ADD INDEX `idx_query_id` (`query_id`), +ADD INDEX `idx_order_id` (`order_id`), +ADD INDEX `idx_user_id` (`user_id`), +ADD INDEX `idx_product_id` (`product_id`); + +-- query_cleanup_config表 +ALTER TABLE `query_cleanup_config` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`); + +-- ============================================ +-- 第五部分：修改产品相关表 +-- ============================================ + +-- product_feature表 +ALTER TABLE `product_feature` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `product_id` CHAR(36) NOT NULL COMMENT '产品UUID', +MODIFY COLUMN `feature_id` CHAR(36) NOT NULL COMMENT '功能UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_product_id` (`product_id`), +ADD INDEX `idx_feature_id` (`feature_id`); + +-- feature表 +ALTER TABLE `feature` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`); + +-- ============================================ +-- 第六部分：修改其他表 +-- ============================================ + +-- authorization_document表 +ALTER TABLE `authorization_document` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `user_id` CHAR(36) NOT NULL COMMENT '用户UUID', +MODIFY COLUMN `order_id` CHAR(36) NOT NULL COMMENT '订单UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_user_id` (`user_id`), +ADD INDEX `idx_order_id` (`order_id`); + +-- global_notifications表（注意：原表id是int类型） +ALTER TABLE `global_notifications` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`); + +-- example表 +ALTER TABLE `example` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `feature_id` CHAR(36) NOT NULL COMMENT '功能UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_feature_id` (`feature_id`); + +-- ============================================ +-- 第七部分：修改管理后台表 +-- ============================================ + +-- admin_user表 +ALTER TABLE `admin_user` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`); + +-- admin_role表 +ALTER TABLE `admin_role` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`); + +-- admin_menu表（注意：字段名是pid，不是parent_id） +ALTER TABLE `admin_menu` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `pid` CHAR(36) NOT NULL DEFAULT '0' COMMENT '父菜单UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_pid` (`pid`); + +-- admin_api表 +ALTER TABLE `admin_api` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`); + +-- admin_dict_type表 +ALTER TABLE `admin_dict_type` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`); + +-- admin_dict_data表（注意：没有dict_type_id字段，使用dict_type字符串） +-- 此表不需要修改外键字段，因为使用的是dict_type字符串而不是ID + +ALTER TABLE `admin_dict_data` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`); + +-- admin_user_role表 +ALTER TABLE `admin_user_role` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `user_id` CHAR(36) NOT NULL DEFAULT '0' COMMENT '用户UUID', +MODIFY COLUMN `role_id` CHAR(36) NOT NULL DEFAULT '0' COMMENT '角色UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_user_id` (`user_id`), +ADD INDEX `idx_role_id` (`role_id`); + +-- admin_role_menu表 +ALTER TABLE `admin_role_menu` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `role_id` CHAR(36) NOT NULL DEFAULT '0' COMMENT '角色UUID', +MODIFY COLUMN `menu_id` CHAR(36) NOT NULL DEFAULT '0' COMMENT '菜单UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_role_id` (`role_id`), +ADD INDEX `idx_menu_id` (`menu_id`); + +-- admin_role_api表 +ALTER TABLE `admin_role_api` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `role_id` CHAR(36) NOT NULL DEFAULT '0' COMMENT '角色UUID', +MODIFY COLUMN `api_id` CHAR(36) NOT NULL DEFAULT '0' COMMENT 'API UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_role_id` (`role_id`), +ADD INDEX `idx_api_id` (`api_id`); + +-- ============================================ +-- 第八部分：修改推广相关表（新增） +-- ============================================ + +-- admin_promotion_link表 +ALTER TABLE `admin_promotion_link` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `admin_user_id` CHAR(36) NOT NULL DEFAULT '0' COMMENT '推广者账号UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_admin_user_id` (`admin_user_id`); + +-- admin_promotion_link_stats_history表 +ALTER TABLE `admin_promotion_link_stats_history` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `link_id` CHAR(36) NOT NULL COMMENT '推广链接UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_link_id` (`link_id`); + +-- admin_promotion_link_stats_total表 +ALTER TABLE `admin_promotion_link_stats_total` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `link_id` CHAR(36) NOT NULL COMMENT '推广链接UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD INDEX `idx_link_id` (`link_id`); + +-- admin_promotion_order表 +ALTER TABLE `admin_promotion_order` +MODIFY COLUMN `id` CHAR(36) NOT NULL COMMENT 'UUID主键', +MODIFY COLUMN `link_id` CHAR(36) NOT NULL COMMENT '推广链接UUID', +MODIFY COLUMN `order_id` CHAR(36) NOT NULL COMMENT '订单UUID', +MODIFY COLUMN `user_id` CHAR(36) NOT NULL COMMENT '下单用户UUID', +MODIFY COLUMN `admin_user_id` CHAR(36) NOT NULL COMMENT '推广者账号UUID', +DROP PRIMARY KEY, +ADD PRIMARY KEY (`id`), +ADD UNIQUE INDEX `uk_order_id` (`order_id`) COMMENT '确保每个订单只有一条推广记录', +ADD INDEX `idx_link_id` (`link_id`), +ADD INDEX `idx_user_id` (`user_id`) COMMENT '优化用户查询', +ADD INDEX `idx_admin_user_id` (`admin_user_id`); + +-- ============================================ +-- 注意事项 +-- ============================================ +-- 1. 此脚本直接修改表结构，不保留旧数据 +-- 2. user表和product表已经是CHAR(36)类型，已跳过 +-- 3. 执行后，所有ID字段都是CHAR(36)类型 +-- 4. 插入新记录时，需要在应用层生成UUID（使用uuid.NewString()） +-- 5. 如果表中有数据，需要先清空数据或手动填充UUID +-- 6. 建议在开发环境先测试，确认无误后再应用到生产环境 +-- 7. admin_menu表使用pid字段而不是parent_id +-- 8. admin_dict_data表使用dict_type字符串字段，不是dict_type_id +-- 9. global_notifications表的id原为int类型，现改为CHAR(36) \ No newline at end of file diff --git a/deploy/template/api/config.tpl b/deploy/template/api/config.tpl new file mode 100644 index 0000000..55127ef --- /dev/null +++ b/deploy/template/api/config.tpl @@ -0,0 +1,9 @@ +package config + +import {{.authImport}} + +type Config struct { + rest.RestConf + {{.auth}} + {{.jwtTrans}} +} diff --git a/deploy/template/api/context.tpl b/deploy/template/api/context.tpl new file mode 100644 index 0000000..c15c1e4 --- /dev/null +++ b/deploy/template/api/context.tpl @@ -0,0 +1,17 @@ +package svc + +import ( + {{.configImport}} +) + +type ServiceContext struct { + Config {{.config}} + {{.middleware}} +} + +func NewServiceContext(c {{.config}}) *ServiceContext { + return &ServiceContext{ + Config: c, + {{.middlewareAssignment}} + } +} diff --git a/deploy/template/api/etc.tpl b/deploy/template/api/etc.tpl new file mode 100644 index 0000000..ed55cf1 --- /dev/null +++ b/deploy/template/api/etc.tpl @@ -0,0 +1,3 @@ +Name: {{.serviceName}} +Host: {{.host}} +Port: {{.port}} diff --git a/deploy/template/api/handler.tpl b/deploy/template/api/handler.tpl new file mode 100644 index 0000000..dcd3729 --- /dev/null +++ b/deploy/template/api/handler.tpl @@ -0,0 +1,27 @@ +package {{.PkgName}} + +import ( + "net/http" + + "qnc-server/common/result" + "qnc-server/pkg/lzkit/validator" + "github.com/zeromicro/go-zero/rest/httpx" + {{.ImportPackages}} +) + +func {{.HandlerName}}(svcCtx *svc.ServiceContext) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + {{if .HasRequest}}var req types.{{.RequestType}} + if err := httpx.Parse(r, &req); err != nil { + result.ParamErrorResult(r,w,err) + return + } + if err := validator.Validate(req); err != nil { + result.ParamValidateErrorResult(r, w, err) + return + } + {{end}}l := {{.LogicName}}.New{{.LogicType}}(r.Context(), svcCtx) + {{if .HasResp}}resp, {{end}}err := l.{{.Call}}({{if .HasRequest}}&req{{end}}) + result.HttpResult(r, w, {{if .HasResp}}resp{{else}}nil{{end}}, err) + } +} diff --git a/deploy/template/api/logic.tpl b/deploy/template/api/logic.tpl new file mode 100644 index 0000000..7c04323 --- /dev/null +++ b/deploy/template/api/logic.tpl @@ -0,0 +1,25 @@ +package {{.pkgName}} + +import ( + {{.imports}} +) + +type {{.logic}} struct { + logx.Logger + ctx context.Context + svcCtx *svc.ServiceContext +} + +func New{{.logic}}(ctx context.Context, svcCtx *svc.ServiceContext) *{{.logic}} { + return &{{.logic}}{ + Logger: logx.WithContext(ctx), + ctx: ctx, + svcCtx: svcCtx, + } +} + +func (l *{{.logic}}) {{.function}}({{.request}}) {{.responseType}} { + // todo: add your logic here and delete this line + + {{.returnString}} +} diff --git a/deploy/template/api/main.tpl b/deploy/template/api/main.tpl new file mode 100644 index 0000000..71be5cb --- /dev/null +++ b/deploy/template/api/main.tpl @@ -0,0 +1,27 @@ +package main + +import ( + "flag" + "fmt" + + {{.importPackages}} + "qnc-server/common/middleware" +) + +var configFile = flag.String("f", "etc/{{.serviceName}}.yaml", "the config file") + +func main() { + flag.Parse() + + var c config.Config + conf.MustLoad(*configFile, &c) + + ctx := svc.NewServiceContext(c) + server := rest.MustNewServer(c.RestConf) + defer server.Stop() + + handler.RegisterHandlers(server, ctx) + + fmt.Printf("Starting server at %s:%d...\n", c.Host, c.Port) + server.Start() +} diff --git a/deploy/template/api/middleware.tpl b/deploy/template/api/middleware.tpl new file mode 100644 index 0000000..af714e0 --- /dev/null +++ b/deploy/template/api/middleware.tpl @@ -0,0 +1,20 @@ + +package middleware + +import "net/http" + +type {{.name}} struct { +} + +func New{{.name}}() *{{.name}} { + return &{{.name}}{} +} + +func (m *{{.name}})Handle(next http.HandlerFunc) http.HandlerFunc { + return func(w http.ResponseWriter, r *http.Request) { + // TODO generate middleware implement function, delete after code implementation + + // Passthrough to next handler if need + next(w, r) + } +} diff --git a/deploy/template/api/route-addition.tpl b/deploy/template/api/route-addition.tpl new file mode 100644 index 0000000..bb8a5df --- /dev/null +++ b/deploy/template/api/route-addition.tpl @@ -0,0 +1,4 @@ + + server.AddRoutes( + {{.routes}} {{.jwt}}{{.signature}} {{.prefix}} {{.timeout}} {{.maxBytes}} + ) diff --git a/deploy/template/api/routes.tpl b/deploy/template/api/routes.tpl new file mode 100644 index 0000000..f13fb11 --- /dev/null +++ b/deploy/template/api/routes.tpl @@ -0,0 +1,13 @@ +// Code generated by goctl. DO NOT EDIT. +package handler + +import ( + "net/http"{{if .hasTimeout}} + "time"{{end}} + + {{.importPackages}} +) + +func RegisterHandlers(server *rest.Server, serverCtx *svc.ServiceContext) { + {{.routesAdditions}} +} diff --git a/deploy/template/api/template.tpl b/deploy/template/api/template.tpl new file mode 100644 index 0000000..2176441 --- /dev/null +++ b/deploy/template/api/template.tpl @@ -0,0 +1,24 @@ +syntax = "v1" + +info ( + title: // TODO: add title + desc: // TODO: add description + author: "{{.gitUser}}" + email: "{{.gitEmail}}" +) + +type request { + // TODO: add members here and delete this comment +} + +type response { + // TODO: add members here and delete this comment +} + +service {{.serviceName}} { + @handler GetUser // TODO: set handler name and delete this comment + get /users/id/:userId(request) returns(response) + + @handler CreateUser // TODO: set handler name and delete this comment + post /users/create(request) +} diff --git a/deploy/template/api/types.tpl b/deploy/template/api/types.tpl new file mode 100644 index 0000000..735ec2d --- /dev/null +++ b/deploy/template/api/types.tpl @@ -0,0 +1,6 @@ +// Code generated by goctl. DO NOT EDIT. +package types{{if .containsTime}} +import ( + "time" +){{end}} +{{.types}} diff --git a/deploy/template/docker/docker.tpl b/deploy/template/docker/docker.tpl new file mode 100644 index 0000000..d1b5ff4 --- /dev/null +++ b/deploy/template/docker/docker.tpl @@ -0,0 +1,33 @@ +FROM golang:{{.Version}}alpine AS builder + +LABEL stage=gobuilder + +ENV CGO_ENABLED 0 +{{if .Chinese}}ENV GOPROXY https://goproxy.cn,direct +RUN sed -i 's/dl-cdn.alpinelinux.org/mirrors.aliyun.com/g' /etc/apk/repositories +{{end}}{{if .HasTimezone}} +RUN apk update --no-cache && apk add --no-cache tzdata +{{end}} +WORKDIR /build + +ADD go.mod . +ADD go.sum . +RUN go mod download +COPY . . +{{if .Argument}}COPY {{.GoRelPath}}/etc /app/etc +{{end}}RUN go build -ldflags="-s -w" -o /app/{{.ExeFile}} {{.GoMainFrom}} + + +FROM {{.BaseImage}} + +COPY --from=builder /etc/ssl/certs/ca-certificates.crt /etc/ssl/certs/ca-certificates.crt +{{if .HasTimezone}}COPY --from=builder /usr/share/zoneinfo/{{.Timezone}} /usr/share/zoneinfo/{{.Timezone}} +ENV TZ {{.Timezone}} +{{end}} +WORKDIR /app +COPY --from=builder /app/{{.ExeFile}} /app/{{.ExeFile}}{{if .Argument}} +COPY --from=builder /app/etc /app/etc{{end}} +{{if .HasPort}} +EXPOSE {{.Port}} +{{end}} +CMD ["./{{.ExeFile}}"{{.Argument}}] diff --git a/deploy/template/gateway/etc.tpl b/deploy/template/gateway/etc.tpl new file mode 100644 index 0000000..0a70f1a --- /dev/null +++ b/deploy/template/gateway/etc.tpl @@ -0,0 +1,18 @@ +Name: gateway-example # gateway name +Host: localhost # gateway host +Port: 8888 # gateway port +Upstreams: # upstreams + - Grpc: # grpc upstream + Target: 0.0.0.0:8080 # grpc target,the direct grpc server address,for only one node +# Endpoints: [0.0.0.0:8080,192.168.120.1:8080] # grpc endpoints, the grpc server address list, for multiple nodes +# Etcd: # etcd config, if you want to use etcd to discover the grpc server address +# Hosts: [127.0.0.1:2378,127.0.0.1:2379] # etcd hosts +# Key: greet.grpc # the discovery key + # protoset mode + ProtoSets: + - hello.pb + # Mappings can also be written in proto options +# Mappings: # routes mapping +# - Method: get +# Path: /ping +# RpcPath: hello.Hello/Ping diff --git a/deploy/template/gateway/main.tpl b/deploy/template/gateway/main.tpl new file mode 100644 index 0000000..6273451 --- /dev/null +++ b/deploy/template/gateway/main.tpl @@ -0,0 +1,20 @@ +package main + +import ( + "flag" + + "github.com/zeromicro/go-zero/core/conf" + "github.com/zeromicro/go-zero/gateway" +) + +var configFile = flag.String("f", "etc/gateway.yaml", "config file") + +func main() { + flag.Parse() + + var c gateway.GatewayConf + conf.MustLoad(*configFile, &c) + gw := gateway.MustNewServer(c) + defer gw.Stop() + gw.Start() +} diff --git a/deploy/template/kube/deployment.tpl b/deploy/template/kube/deployment.tpl new file mode 100644 index 0000000..14145df --- /dev/null +++ b/deploy/template/kube/deployment.tpl @@ -0,0 +1,117 @@ +apiVersion: apps/v1 +kind: Deployment +metadata: + name: {{.Name}} + namespace: {{.Namespace}} + labels: + app: {{.Name}} +spec: + replicas: {{.Replicas}} + revisionHistoryLimit: {{.Revisions}} + selector: + matchLabels: + app: {{.Name}} + template: + metadata: + labels: + app: {{.Name}} + spec:{{if .ServiceAccount}} + serviceAccountName: {{.ServiceAccount}}{{end}} + containers: + - name: {{.Name}} + image: {{.Image}} + {{if .ImagePullPolicy}}imagePullPolicy: {{.ImagePullPolicy}} + {{end}}ports: + - containerPort: {{.Port}} + readinessProbe: + tcpSocket: + port: {{.Port}} + initialDelaySeconds: 5 + periodSeconds: 10 + livenessProbe: + tcpSocket: + port: {{.Port}} + initialDelaySeconds: 15 + periodSeconds: 20 + resources: + requests: + cpu: {{.RequestCpu}}m + memory: {{.RequestMem}}Mi + limits: + cpu: {{.LimitCpu}}m + memory: {{.LimitMem}}Mi + volumeMounts: + - name: timezone + mountPath: /etc/localtime + {{if .Secret}}imagePullSecrets: + - name: {{.Secret}} + {{end}}volumes: + - name: timezone + hostPath: + path: /usr/share/zoneinfo/Asia/Shanghai + +--- + +apiVersion: v1 +kind: Service +metadata: + name: {{.Name}}-svc + namespace: {{.Namespace}} +spec: + ports: + {{if .UseNodePort}}- nodePort: {{.NodePort}} + port: {{.Port}} + protocol: TCP + targetPort: {{.TargetPort}} + type: NodePort{{else}}- port: {{.Port}} + targetPort: {{.TargetPort}}{{end}} + selector: + app: {{.Name}} + +--- + +apiVersion: autoscaling/v2beta2 +kind: HorizontalPodAutoscaler +metadata: + name: {{.Name}}-hpa-c + namespace: {{.Namespace}} + labels: + app: {{.Name}}-hpa-c +spec: + scaleTargetRef: + apiVersion: apps/v1 + kind: Deployment + name: {{.Name}} + minReplicas: {{.MinReplicas}} + maxReplicas: {{.MaxReplicas}} + metrics: + - type: Resource + resource: + name: cpu + target: + type: Utilization + averageUtilization: 80 + +--- + +apiVersion: autoscaling/v2beta2 +kind: HorizontalPodAutoscaler +metadata: + name: {{.Name}}-hpa-m + namespace: {{.Namespace}} + labels: + app: {{.Name}}-hpa-m +spec: + scaleTargetRef: + apiVersion: apps/v1 + kind: Deployment + name: {{.Name}} + minReplicas: {{.MinReplicas}} + maxReplicas: {{.MaxReplicas}} + metrics: + - type: Resource + resource: + name: memory + target: + type: Utilization + averageUtilization: 80 diff --git a/deploy/template/kube/job.tpl b/deploy/template/kube/job.tpl new file mode 100644 index 0000000..0da72ed --- /dev/null +++ b/deploy/template/kube/job.tpl @@ -0,0 +1,37 @@ +apiVersion: batch/v1 +kind: CronJob +metadata: + name: {{.Name}} + namespace: {{.Namespace}} +spec: + successfulJobsHistoryLimit: {{.SuccessfulJobsHistoryLimit}} + schedule: "{{.Schedule}}" + jobTemplate: + spec: + template: + spec:{{if .ServiceAccount}} + serviceAccountName: {{.ServiceAccount}}{{end}} + {{end}}containers: + - name: {{.Name}} + image: # todo image url + resources: + requests: + cpu: {{.RequestCpu}}m + memory: {{.RequestMem}}Mi + limits: + cpu: {{.LimitCpu}}m + memory: {{.LimitMem}}Mi + command: + - ./{{.ServiceName}} + - -f + - ./{{.Name}}.yaml + volumeMounts: + - name: timezone + mountPath: /etc/localtime + imagePullSecrets: + - name: # registry secret, if no, remove this + restartPolicy: OnFailure + volumes: + - name: timezone + hostPath: + path: /usr/share/zoneinfo/Asia/Shanghai diff --git a/deploy/template/model/delete.tpl b/deploy/template/model/delete.tpl new file mode 100644 index 0000000..e22869a --- /dev/null +++ b/deploy/template/model/delete.tpl @@ -0,0 +1,21 @@ +func (m *default{{.upperStartCamelObject}}Model) Delete(ctx context.Context, session sqlx.Session, {{.lowerStartCamelPrimaryKey}} {{.dataType}}) error { + {{if .withCache}}{{if .containsIndexCache}}data, err:=m.FindOne(ctx, {{.lowerStartCamelPrimaryKey}}) + if err!=nil{ + return err + } + + {{end}} {{.keys}} + _, err {{if .containsIndexCache}}={{else}}:={{end}} m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("delete from %s where {{.originalPrimaryKey}} = {{if .postgreSql}}$1{{else}}?{{end}}", m.table) + if session!=nil{ + return session.ExecCtx(ctx,query, {{.lowerStartCamelPrimaryKey}}) + } + return conn.ExecCtx(ctx, query, {{.lowerStartCamelPrimaryKey}}) + }, {{.keyValues}}){{else}}query := fmt.Sprintf("delete from %s where {{.originalPrimaryKey}} = {{if .postgreSql}}$1{{else}}?{{end}}", m.table) + if session!=nil{ + _,err:= session.ExecCtx(ctx,query, {{.lowerStartCamelPrimaryKey}}) + return err + } + _,err:=m.conn.ExecCtx(ctx, query, {{.lowerStartCamelPrimaryKey}}){{end}} + return err +} \ No newline at end of file diff --git a/deploy/template/model/err.tpl b/deploy/template/model/err.tpl new file mode 100644 index 0000000..cbc9f82 --- /dev/null +++ b/deploy/template/model/err.tpl @@ -0,0 +1,9 @@ +package {{.pkg}} + +import ( + "errors" + "github.com/zeromicro/go-zero/core/stores/sqlx" +) + +var ErrNotFound = sqlx.ErrNotFound +var ErrNoRowsUpdate = errors.New("update db no rows change") \ No newline at end of file diff --git a/deploy/template/model/field.tpl b/deploy/template/model/field.tpl new file mode 100644 index 0000000..6b4ed38 --- /dev/null +++ b/deploy/template/model/field.tpl @@ -0,0 +1 @@ +{{.name}} {{.type}} {{.tag}} {{if .hasComment}}// {{.comment}}{{end}} \ No newline at end of file diff --git a/deploy/template/model/find-one-by-field-extra-method.tpl b/deploy/template/model/find-one-by-field-extra-method.tpl new file mode 100644 index 0000000..af15538 --- /dev/null +++ b/deploy/template/model/find-one-by-field-extra-method.tpl @@ -0,0 +1,7 @@ +func (m *default{{.upperStartCamelObject}}Model) formatPrimary(primary interface{}) string { + return fmt.Sprintf("%s%v", {{.primaryKeyLeft}}, primary) +} +func (m *default{{.upperStartCamelObject}}Model) queryPrimary(ctx context.Context, conn sqlx.SqlConn, v, primary interface{}) error { + query := fmt.Sprintf("select %s from %s where {{.originalPrimaryField}} = {{if .postgreSql}}$1{{else}}?{{end}} and del_state = ? limit 1", {{.lowerStartCamelObject}}Rows, m.table ) + return conn.QueryRowCtx(ctx, v, query, primary,globalkey.DelStateNo) +} diff --git a/deploy/template/model/find-one-by-field.tpl b/deploy/template/model/find-one-by-field.tpl new file mode 100644 index 0000000..e6fd828 --- /dev/null +++ b/deploy/template/model/find-one-by-field.tpl @@ -0,0 +1,32 @@ + +func (m *default{{.upperStartCamelObject}}Model) FindOneBy{{.upperField}}(ctx context.Context, {{.in}}) (*{{.upperStartCamelObject}}, error) { +{{if .withCache}}{{.cacheKey}} + var resp {{.upperStartCamelObject}} + err := m.QueryRowIndexCtx(ctx, &resp, {{.cacheKeyVariable}}, m.formatPrimary, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) (i interface{}, e error) { + query := fmt.Sprintf("select %s from %s where {{.originalField}} and del_state = ? limit 1", {{.lowerStartCamelObject}}Rows, m.table) + if err := conn.QueryRowCtx(ctx, &resp, query, {{.lowerStartCamelField}},globalkey.DelStateNo); err != nil { + return nil, err + } + return resp.{{.upperStartCamelPrimaryKey}}, nil + }, m.queryPrimary) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +}{{else}}var resp {{.upperStartCamelObject}} + query := fmt.Sprintf("select %s from %s where {{.originalField}} and del_state = ? limit 1", {{.lowerStartCamelObject}}Rows, m.table ) + err := m.conn.QueryRowCtx(ctx, &resp, query, {{.lowerStartCamelField}},globalkey.DelStateNo) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +}{{end}} + diff --git a/deploy/template/model/find-one.tpl b/deploy/template/model/find-one.tpl new file mode 100644 index 0000000..72d5ecf --- /dev/null +++ b/deploy/template/model/find-one.tpl @@ -0,0 +1,26 @@ +func (m *default{{.upperStartCamelObject}}Model) FindOne(ctx context.Context, {{.lowerStartCamelPrimaryKey}} {{.dataType}}) (*{{.upperStartCamelObject}}, error) { + {{if .withCache}}{{.cacheKey}} + var resp {{.upperStartCamelObject}} + err := m.QueryRowCtx(ctx, &resp, {{.cacheKeyVariable}}, func(ctx context.Context, conn sqlx.SqlConn, v interface{}) error { + query := fmt.Sprintf("select %s from %s where {{.originalPrimaryKey}} = {{if .postgreSql}}$1{{else}}?{{end}} and del_state = ? limit 1", {{.lowerStartCamelObject}}Rows, m.table) + return conn.QueryRowCtx(ctx, v, query, {{.lowerStartCamelPrimaryKey}},globalkey.DelStateNo) + }) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + }{{else}}query := fmt.Sprintf("select %s from %s where {{.originalPrimaryKey}} = {{if .postgreSql}}$1{{else}}?{{end}} and del_state = ? limit 1", {{.lowerStartCamelObject}}Rows, m.table) + var resp {{.upperStartCamelObject}} + err := m.conn.QueryRowCtx(ctx, &resp, query, {{.lowerStartCamelPrimaryKey}},globalkey.DelStateNo) + switch err { + case nil: + return &resp, nil + case sqlc.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + }{{end}} +} diff --git a/deploy/template/model/import-no-cache.tpl b/deploy/template/model/import-no-cache.tpl new file mode 100644 index 0000000..1721029 --- /dev/null +++ b/deploy/template/model/import-no-cache.tpl @@ -0,0 +1,18 @@ +import ( + "context" + "database/sql" + "fmt" + "strings" + + {{if .time}}"time"{{end}} + "reflect" + + "qnc-server/common/globalkey" + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" + "github.com/google/uuid" +) diff --git a/deploy/template/model/import.tpl b/deploy/template/model/import.tpl new file mode 100644 index 0000000..0b0240b --- /dev/null +++ b/deploy/template/model/import.tpl @@ -0,0 +1,19 @@ +import ( + "context" + "database/sql" + "fmt" + "strings" + + {{if .time}}"time"{{end}} + "reflect" + + "qnc-server/common/globalkey" + "github.com/Masterminds/squirrel" + "github.com/pkg/errors" + "github.com/zeromicro/go-zero/core/stores/builder" + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlc" + "github.com/zeromicro/go-zero/core/stores/sqlx" + "github.com/zeromicro/go-zero/core/stringx" + "github.com/google/uuid" +) diff --git a/deploy/template/model/insert.tpl b/deploy/template/model/insert.tpl new file mode 100644 index 0000000..528e0f2 --- /dev/null +++ b/deploy/template/model/insert.tpl @@ -0,0 +1,34 @@ + +func (m *default{{.upperStartCamelObject}}Model) Insert(ctx context.Context,session sqlx.Session, data *{{.upperStartCamelObject}}) (sql.Result,error) { + data.DelState = globalkey.DelStateNo + m.insertUUID(data) + {{if .withCache}}{{.keys}} + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("insert into %s (%s) values ({{.expression}})", m.table, {{.lowerStartCamelObject}}RowsExpectAutoSet) + if session != nil{ + return session.ExecCtx(ctx,query,{{.expressionValues}}) + } + return conn.ExecCtx(ctx, query, {{.expressionValues}}) + }, {{.keyValues}}){{else}} + query := fmt.Sprintf("insert into %s (%s) values ({{.expression}})", m.table, {{.lowerStartCamelObject}}RowsExpectAutoSet) + if session != nil{ + return session.ExecCtx(ctx,query,{{.expressionValues}}) + } + return m.conn.ExecCtx(ctx, query, {{.expressionValues}}){{end}} +} +func (m *default{{.upperStartCamelObject}}Model) insertUUID(data *{{.upperStartCamelObject}}) { + t := reflect.TypeOf(data).Elem() + v := reflect.ValueOf(data).Elem() + for i := 0; i < t.NumField(); i++ { + sf := t.Field(i) + if sf.Tag.Get("db") == "id" { + f := v.Field(i) + if f.IsValid() && f.CanSet() && f.Kind() == reflect.String { + if f.String() == "" { + f.SetString(uuid.NewString()) + } + } + break + } + } +} diff --git a/deploy/template/model/interface-delete.tpl b/deploy/template/model/interface-delete.tpl new file mode 100644 index 0000000..a7f11a7 --- /dev/null +++ b/deploy/template/model/interface-delete.tpl @@ -0,0 +1 @@ +Delete(ctx context.Context,session sqlx.Session, {{.lowerStartCamelPrimaryKey}} {{.dataType}}) error \ No newline at end of file diff --git a/deploy/template/model/interface-find-one-by-field.tpl b/deploy/template/model/interface-find-one-by-field.tpl new file mode 100644 index 0000000..9615aa3 --- /dev/null +++ b/deploy/template/model/interface-find-one-by-field.tpl @@ -0,0 +1 @@ +FindOneBy{{.upperField}}(ctx context.Context, {{.in}}) (*{{.upperStartCamelObject}}, error) \ No newline at end of file diff --git a/deploy/template/model/interface-find-one.tpl b/deploy/template/model/interface-find-one.tpl new file mode 100644 index 0000000..a7a5440 --- /dev/null +++ b/deploy/template/model/interface-find-one.tpl @@ -0,0 +1 @@ +FindOne(ctx context.Context, {{.lowerStartCamelPrimaryKey}} {{.dataType}}) (*{{.upperStartCamelObject}}, error) \ No newline at end of file diff --git a/deploy/template/model/interface-insert.tpl b/deploy/template/model/interface-insert.tpl new file mode 100644 index 0000000..6705d2b --- /dev/null +++ b/deploy/template/model/interface-insert.tpl @@ -0,0 +1 @@ +Insert(ctx context.Context, session sqlx.Session,data *{{.upperStartCamelObject}}) (sql.Result,error) \ No newline at end of file diff --git a/deploy/template/model/interface-update.tpl b/deploy/template/model/interface-update.tpl new file mode 100644 index 0000000..32a4214 --- /dev/null +++ b/deploy/template/model/interface-update.tpl @@ -0,0 +1,12 @@ +Update(ctx context.Context,session sqlx.Session, data *{{.upperStartCamelObject}}) (sql.Result, error) +UpdateWithVersion(ctx context.Context,session sqlx.Session,data *{{.upperStartCamelObject}}) error +Trans(ctx context.Context,fn func(context context.Context,session sqlx.Session) error) error +SelectBuilder() squirrel.SelectBuilder +DeleteSoft(ctx context.Context,session sqlx.Session, data *{{.upperStartCamelObject}}) error +FindSum(ctx context.Context,sumBuilder squirrel.SelectBuilder,field string) (float64,error) +FindCount(ctx context.Context,countBuilder squirrel.SelectBuilder,field string) (int64,error) +FindAll(ctx context.Context,rowBuilder squirrel.SelectBuilder,orderBy string) ([]*{{.upperStartCamelObject}},error) +FindPageListByPage(ctx context.Context,rowBuilder squirrel.SelectBuilder,page ,pageSize int64,orderBy string) ([]*{{.upperStartCamelObject}},error) +FindPageListByPageWithTotal(ctx context.Context, rowBuilder squirrel.SelectBuilder, page, pageSize int64, orderBy string) ([]*{{.upperStartCamelObject}}, int64, error) +FindPageListByIdDESC(ctx context.Context,rowBuilder squirrel.SelectBuilder ,preMinId ,pageSize int64) ([]*{{.upperStartCamelObject}},error) +FindPageListByIdASC(ctx context.Context,rowBuilder squirrel.SelectBuilder,preMaxId ,pageSize int64) ([]*{{.upperStartCamelObject}},error) \ No newline at end of file diff --git a/deploy/template/model/model-gen.tpl b/deploy/template/model/model-gen.tpl new file mode 100644 index 0000000..8da9d0b --- /dev/null +++ b/deploy/template/model/model-gen.tpl @@ -0,0 +1,13 @@ +// Code generated by goctl. DO NOT EDIT! + +package {{.pkg}} +{{.imports}} +{{.vars}} +{{.types}} +{{.new}} +{{.insert}} +{{.find}} +{{.update}} +{{.delete}} +{{.extraMethod}} +{{.tableName}} \ No newline at end of file diff --git a/deploy/template/model/model-new.tpl b/deploy/template/model/model-new.tpl new file mode 100644 index 0000000..5e9d15f --- /dev/null +++ b/deploy/template/model/model-new.tpl @@ -0,0 +1,7 @@ + +func new{{.upperStartCamelObject}}Model(conn sqlx.SqlConn{{if .withCache}}, c cache.CacheConf{{end}}) *default{{.upperStartCamelObject}}Model { + return &default{{.upperStartCamelObject}}Model{ + {{if .withCache}}CachedConn: sqlc.NewConn(conn, c){{else}}conn:conn{{end}}, + table: {{.table}}, + } +} diff --git a/deploy/template/model/model.tpl b/deploy/template/model/model.tpl new file mode 100644 index 0000000..0ac2918 --- /dev/null +++ b/deploy/template/model/model.tpl @@ -0,0 +1,37 @@ +package {{.pkg}} +{{if .withCache}} +import ( + + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/sqlx" + +) +{{else}} +import ( + + "github.com/zeromicro/go-zero/core/stores/sqlx" + +) + +{{end}} +var _ {{.upperStartCamelObject}}Model = (*custom{{.upperStartCamelObject}}Model)(nil) + +type ( + // {{.upperStartCamelObject}}Model is an interface to be customized, add more methods here, + // and implement the added methods in custom{{.upperStartCamelObject}}Model. + {{.upperStartCamelObject}}Model interface { + {{.lowerStartCamelObject}}Model + +} + + custom{{.upperStartCamelObject}}Model struct { + *default{{.upperStartCamelObject}}Model + } +) + +// New{{.upperStartCamelObject}}Model returns a model for the database table. +func New{{.upperStartCamelObject}}Model(conn sqlx.SqlConn{{if .withCache}}, c cache.CacheConf{{end}}) {{.upperStartCamelObject}}Model { + return &custom{{.upperStartCamelObject}}Model{ + default{{.upperStartCamelObject}}Model: new{{.upperStartCamelObject}}Model(conn{{if .withCache}}, c{{end}}), + } +} diff --git a/deploy/template/model/table-name.tpl b/deploy/template/model/table-name.tpl new file mode 100644 index 0000000..69733fa --- /dev/null +++ b/deploy/template/model/table-name.tpl @@ -0,0 +1,4 @@ + +func (m *default{{.upperStartCamelObject}}Model) tableName() string { + return m.table +} diff --git a/deploy/template/model/tag.tpl b/deploy/template/model/tag.tpl new file mode 100644 index 0000000..8e1ddf0 --- /dev/null +++ b/deploy/template/model/tag.tpl @@ -0,0 +1 @@ +`db:"{{.field}}"` \ No newline at end of file diff --git a/deploy/template/model/types.tpl b/deploy/template/model/types.tpl new file mode 100644 index 0000000..fe08ba2 --- /dev/null +++ b/deploy/template/model/types.tpl @@ -0,0 +1,15 @@ + +type ( + {{.lowerStartCamelObject}}Model interface{ + {{.method}} + } + + default{{.upperStartCamelObject}}Model struct { + {{if .withCache}}sqlc.CachedConn{{else}}conn sqlx.SqlConn{{end}} + table string + } + + {{.upperStartCamelObject}} struct { + {{.fields}} + } +) diff --git a/deploy/template/model/update.tpl b/deploy/template/model/update.tpl new file mode 100644 index 0000000..b01e30f --- /dev/null +++ b/deploy/template/model/update.tpl @@ -0,0 +1,286 @@ + +func (m *default{{.upperStartCamelObject}}Model) Update(ctx context.Context,session sqlx.Session, {{if .containsIndexCache}}newData{{else}}data{{end}} *{{.upperStartCamelObject}}) (sql.Result,error) { + {{if .withCache}}{{if .containsIndexCache}}data, err:=m.FindOne(ctx, newData.{{.upperStartCamelPrimaryKey}}) + if err!=nil{ + return nil,err + } + {{end}}{{.keys}} + return m.ExecCtx(ctx, func(ctx context.Context, conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where {{.originalPrimaryKey}} = {{if .postgreSql}}$1{{else}}?{{end}}", m.table, {{.lowerStartCamelObject}}RowsWithPlaceHolder) + if session != nil{ + return session.ExecCtx(ctx,query, {{.expressionValues}}) + } + return conn.ExecCtx(ctx, query, {{.expressionValues}}) + }, {{.keyValues}}){{else}}query := fmt.Sprintf("update %s set %s where {{.originalPrimaryKey}} = {{if .postgreSql}}$1{{else}}?{{end}}", m.table, {{.lowerStartCamelObject}}RowsWithPlaceHolder) + if session != nil{ + return session.ExecCtx(ctx,query, {{.expressionValues}}) + } + return m.conn.ExecCtx(ctx, query, {{.expressionValues}}){{end}} +} + +func (m *default{{.upperStartCamelObject}}Model) UpdateWithVersion(ctx context.Context,session sqlx.Session,{{if .containsIndexCache}}newData{{else}}data{{end}} *{{.upperStartCamelObject}}) error { + + {{if .containsIndexCache}} + oldVersion := newData.Version + newData.Version += 1 + {{else}} + oldVersion := data.Version + data.Version += 1 + {{end}} + + var sqlResult sql.Result + var err error + + {{if .withCache}}{{if .containsIndexCache}}data, err:=m.FindOne(ctx, newData.{{.upperStartCamelPrimaryKey}}) + if err!=nil{ + return err + } + {{end}}{{.keys}} + sqlResult,err = m.ExecCtx(ctx,func(ctx context.Context,conn sqlx.SqlConn) (result sql.Result, err error) { + query := fmt.Sprintf("update %s set %s where {{.originalPrimaryKey}} = {{if .postgreSql}}$1{{else}}?{{end}} and version = ? ", m.table, {{.lowerStartCamelObject}}RowsWithPlaceHolder) + if session != nil{ + return session.ExecCtx(ctx,query, {{.expressionValues}},oldVersion) + } + return conn.ExecCtx(ctx,query, {{.expressionValues}},oldVersion) + }, {{.keyValues}}){{else}}query := fmt.Sprintf("update %s set %s where {{.originalPrimaryKey}} = {{if .postgreSql}}$1{{else}}?{{end}} and version = ? ", m.table, {{.lowerStartCamelObject}}RowsWithPlaceHolder) + if session != nil{ + sqlResult,err = session.ExecCtx(ctx,query, {{.expressionValues}},oldVersion) + }else{ + sqlResult,err = m.conn.ExecCtx(ctx,query, {{.expressionValues}},oldVersion) + } + {{end}} + if err != nil { + return err + } + updateCount , err := sqlResult.RowsAffected() + if err != nil{ + return err + } + if updateCount == 0 { + return ErrNoRowsUpdate + } + + return nil +} + +func (m *default{{.upperStartCamelObject}}Model) DeleteSoft(ctx context.Context,session sqlx.Session,data *{{.upperStartCamelObject}}) error { + data.DelState = globalkey.DelStateYes + data.DeleteTime = sql.NullTime{Time: time.Now(), Valid: true} + if err:= m.UpdateWithVersion(ctx,session, data);err!= nil{ + return errors.Wrapf(errors.New("delete soft failed "),"{{.upperStartCamelObject}}Model delete err : %+v",err) + } + return nil +} + +func (m *default{{.upperStartCamelObject}}Model) FindSum(ctx context.Context,builder squirrel.SelectBuilder, field string) (float64,error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindSum Least One Field"), "FindSum Least One Field") + } + + builder = builder.Columns("IFNULL(SUM(" + field + "),0)") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp float64 + {{if .withCache}}err = m.QueryRowNoCacheCtx(ctx,&resp, query, values...){{else}} + err = m.conn.QueryRowCtx(ctx,&resp, query, values...) + {{end}} + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *default{{.upperStartCamelObject}}Model) FindCount(ctx context.Context, builder squirrel.SelectBuilder, field string) (int64,error) { + + if len(field) == 0 { + return 0, errors.Wrapf(errors.New("FindCount Least One Field"), "FindCount Least One Field") + } + + builder = builder.Columns("COUNT(" + field + ")") + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return 0, err + } + + var resp int64 + {{if .withCache}}err = m.QueryRowNoCacheCtx(ctx,&resp, query, values...){{else}} + err = m.conn.QueryRowCtx(ctx,&resp, query, values...) + {{end}} + switch err { + case nil: + return resp, nil + default: + return 0, err + } +} + +func (m *default{{.upperStartCamelObject}}Model) FindAll(ctx context.Context,builder squirrel.SelectBuilder,orderBy string) ([]*{{.upperStartCamelObject}},error) { + + builder = builder.Columns({{.lowerStartCamelObject}}Rows) + + if orderBy == ""{ + builder = builder.OrderBy("id DESC") + }else{ + builder = builder.OrderBy(orderBy) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).ToSql() + if err != nil { + return nil, err + } + + var resp []*{{.upperStartCamelObject}} + {{if .withCache}}err = m.QueryRowsNoCacheCtx(ctx,&resp, query, values...){{else}} + err = m.conn.QueryRowsCtx(ctx,&resp, query, values...) + {{end}} + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *default{{.upperStartCamelObject}}Model) FindPageListByPage(ctx context.Context,builder squirrel.SelectBuilder,page ,pageSize int64,orderBy string) ([]*{{.upperStartCamelObject}},error) { + + builder = builder.Columns({{.lowerStartCamelObject}}Rows) + + if orderBy == ""{ + builder = builder.OrderBy("id DESC") + }else{ + builder = builder.OrderBy(orderBy) + } + + if page < 1{ + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*{{.upperStartCamelObject}} + {{if .withCache}}err = m.QueryRowsNoCacheCtx(ctx,&resp, query, values...){{else}} + err = m.conn.QueryRowsCtx(ctx,&resp, query, values...) + {{end}} + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + + +func (m *default{{.upperStartCamelObject}}Model) FindPageListByPageWithTotal(ctx context.Context,builder squirrel.SelectBuilder,page ,pageSize int64,orderBy string) ([]*{{.upperStartCamelObject}},int64,error) { + + total, err := m.FindCount(ctx, builder, "id") + if err != nil { + return nil, 0, err + } + + builder = builder.Columns({{.lowerStartCamelObject}}Rows) + + if orderBy == ""{ + builder = builder.OrderBy("id DESC") + }else{ + builder = builder.OrderBy(orderBy) + } + + if page < 1{ + page = 1 + } + offset := (page - 1) * pageSize + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).Offset(uint64(offset)).Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil,total, err + } + + var resp []*{{.upperStartCamelObject}} + {{if .withCache}}err = m.QueryRowsNoCacheCtx(ctx,&resp, query, values...){{else}} + err = m.conn.QueryRowsCtx(ctx,&resp, query, values...) + {{end}} + switch err { + case nil: + return resp,total, nil + default: + return nil,total, err + } +} + +func (m *default{{.upperStartCamelObject}}Model) FindPageListByIdDESC(ctx context.Context,builder squirrel.SelectBuilder ,preMinId ,pageSize int64) ([]*{{.upperStartCamelObject}},error) { + + builder = builder.Columns({{.lowerStartCamelObject}}Rows) + + if preMinId > 0 { + builder = builder.Where(" id < ? " , preMinId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id DESC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*{{.upperStartCamelObject}} + {{if .withCache}}err = m.QueryRowsNoCacheCtx(ctx,&resp, query, values...){{else}} + err = m.conn.QueryRowsCtx(ctx,&resp, query, values...) + {{end}} + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *default{{.upperStartCamelObject}}Model) FindPageListByIdASC(ctx context.Context,builder squirrel.SelectBuilder,preMaxId ,pageSize int64) ([]*{{.upperStartCamelObject}},error) { + + builder = builder.Columns({{.lowerStartCamelObject}}Rows) + + if preMaxId > 0 { + builder = builder.Where(" id > ? " , preMaxId) + } + + query, values, err := builder.Where("del_state = ?", globalkey.DelStateNo).OrderBy("id ASC").Limit(uint64(pageSize)).ToSql() + if err != nil { + return nil, err + } + + var resp []*{{.upperStartCamelObject}} + {{if .withCache}}err = m.QueryRowsNoCacheCtx(ctx,&resp, query, values...){{else}} + err = m.conn.QueryRowsCtx(ctx,&resp, query, values...) + {{end}} + switch err { + case nil: + return resp, nil + default: + return nil, err + } +} + +func (m *default{{.upperStartCamelObject}}Model) Trans(ctx context.Context,fn func(ctx context.Context,session sqlx.Session) error) error { + {{if .withCache}} + return m.TransactCtx(ctx,func(ctx context.Context,session sqlx.Session) error { + return fn(ctx,session) + }) + {{else}} + return m.conn.TransactCtx(ctx,func(ctx context.Context,session sqlx.Session) error { + return fn(ctx,session) + }) + {{end}} +} + +func(m *default{{.upperStartCamelObject}}Model) SelectBuilder() squirrel.SelectBuilder { + return squirrel.Select().From(m.table) +} \ No newline at end of file diff --git a/deploy/template/model/var.tpl b/deploy/template/model/var.tpl new file mode 100644 index 0000000..a10ca33 --- /dev/null +++ b/deploy/template/model/var.tpl @@ -0,0 +1,9 @@ + +var ( + {{.lowerStartCamelObject}}FieldNames = builder.RawFieldNames(&{{.upperStartCamelObject}}{}{{if .postgreSql}},true{{end}}) + {{.lowerStartCamelObject}}Rows = strings.Join({{.lowerStartCamelObject}}FieldNames, ",") + {{.lowerStartCamelObject}}RowsExpectAutoSet = {{if .postgreSql}}strings.Join(stringx.Remove({{.lowerStartCamelObject}}FieldNames, {{if .autoIncrement}}"{{.originalPrimaryKey}}",{{end}} "create_time", "update_time"), ","){{else}}strings.Join(stringx.Remove({{.lowerStartCamelObject}}FieldNames, {{if .autoIncrement}}"{{.originalPrimaryKey}}",{{end}} "`create_time`", "`update_time`"), ","){{end}} + {{.lowerStartCamelObject}}RowsWithPlaceHolder = {{if .postgreSql}}builder.PostgreSqlJoin(stringx.Remove({{.lowerStartCamelObject}}FieldNames, "{{.originalPrimaryKey}}", "create_time", "update_time")){{else}}strings.Join(stringx.Remove({{.lowerStartCamelObject}}FieldNames, "{{.originalPrimaryKey}}", "`create_time`", "`update_time`"), "=?,") + "=?"{{end}} + + {{if .withCache}}{{.cacheKeys}}{{end}} +) diff --git a/deploy/template/mongo/err.tpl b/deploy/template/mongo/err.tpl new file mode 100644 index 0000000..27d9244 --- /dev/null +++ b/deploy/template/mongo/err.tpl @@ -0,0 +1,12 @@ +package model + +import ( + "errors" + + "github.com/zeromicro/go-zero/core/stores/mon" +) + +var ( + ErrNotFound = mon.ErrNotFound + ErrInvalidObjectId = errors.New("invalid objectId") +) diff --git a/deploy/template/mongo/model.tpl b/deploy/template/mongo/model.tpl new file mode 100644 index 0000000..287125d --- /dev/null +++ b/deploy/template/mongo/model.tpl @@ -0,0 +1,78 @@ +// Code generated by goctl. DO NOT EDIT. +package model + +import ( + "context" + "time" + + {{if .Cache}}"github.com/zeromicro/go-zero/core/stores/monc"{{else}}"github.com/zeromicro/go-zero/core/stores/mon"{{end}} + "go.mongodb.org/mongo-driver/bson" + "go.mongodb.org/mongo-driver/bson/primitive" + "go.mongodb.org/mongo-driver/mongo" +) + +{{if .Cache}}var prefix{{.Type}}CacheKey = "cache:{{.lowerType}}:"{{end}} + +type {{.lowerType}}Model interface{ + Insert(ctx context.Context,data *{{.Type}}) error + FindOne(ctx context.Context,id string) (*{{.Type}}, error) + Update(ctx context.Context,data *{{.Type}}) (*mongo.UpdateResult, error) + Delete(ctx context.Context,id string) (int64, error) +} + +type default{{.Type}}Model struct { + conn {{if .Cache}}*monc.Model{{else}}*mon.Model{{end}} +} + +func newDefault{{.Type}}Model(conn {{if .Cache}}*monc.Model{{else}}*mon.Model{{end}}) *default{{.Type}}Model { + return &default{{.Type}}Model{conn: conn} +} + + +func (m *default{{.Type}}Model) Insert(ctx context.Context, data *{{.Type}}) error { + if data.ID.IsZero() { + data.ID = primitive.NewObjectID() + data.CreateAt = time.Now() + data.UpdateAt = time.Now() + } + + {{if .Cache}}key := prefix{{.Type}}CacheKey + data.ID.Hex(){{end}} + _, err := m.conn.InsertOne(ctx, {{if .Cache}}key, {{end}} data) + return err +} + +func (m *default{{.Type}}Model) FindOne(ctx context.Context, id string) (*{{.Type}}, error) { + oid, err := primitive.ObjectIDFromHex(id) + if err != nil { + return nil, ErrInvalidObjectId + } + + var data {{.Type}} + {{if .Cache}}key := prefix{{.Type}}CacheKey + id{{end}} + err = m.conn.FindOne(ctx, {{if .Cache}}key, {{end}}&data, bson.M{"_id": oid}) + switch err { + case nil: + return &data, nil + case {{if .Cache}}monc{{else}}mon{{end}}.ErrNotFound: + return nil, ErrNotFound + default: + return nil, err + } +} + +func (m *default{{.Type}}Model) Update(ctx context.Context, data *{{.Type}}) (*mongo.UpdateResult, error) { + data.UpdateAt = time.Now() + {{if .Cache}}key := prefix{{.Type}}CacheKey + data.ID.Hex(){{end}} + res, err := m.conn.UpdateOne(ctx, {{if .Cache}}key, {{end}}bson.M{"_id": data.ID}, bson.M{"$set": data}) + return res, err +} + +func (m *default{{.Type}}Model) Delete(ctx context.Context, id string) (int64, error) { + oid, err := primitive.ObjectIDFromHex(id) + if err != nil { + return 0, ErrInvalidObjectId + } + {{if .Cache}}key := prefix{{.Type}}CacheKey +id{{end}} + res, err := m.conn.DeleteOne(ctx, {{if .Cache}}key, {{end}}bson.M{"_id": oid}) + return res, err +} diff --git a/deploy/template/mongo/model_custom.tpl b/deploy/template/mongo/model_custom.tpl new file mode 100644 index 0000000..31fa865 --- /dev/null +++ b/deploy/template/mongo/model_custom.tpl @@ -0,0 +1,38 @@ +package model + +{{if .Cache}}import ( + "github.com/zeromicro/go-zero/core/stores/cache" + "github.com/zeromicro/go-zero/core/stores/monc" +){{else}}import "github.com/zeromicro/go-zero/core/stores/mon"{{end}} + +{{if .Easy}} +const {{.Type}}CollectionName = "{{.snakeType}}" +{{end}} + +var _ {{.Type}}Model = (*custom{{.Type}}Model)(nil) + +type ( + // {{.Type}}Model is an interface to be customized, add more methods here, + // and implement the added methods in custom{{.Type}}Model. + {{.Type}}Model interface { + {{.lowerType}}Model + } + + custom{{.Type}}Model struct { + *default{{.Type}}Model + } +) + + +// New{{.Type}}Model returns a model for the mongo. +{{if .Easy}}func New{{.Type}}Model(url, db string{{if .Cache}}, c cache.CacheConf{{end}}) {{.Type}}Model { + conn := {{if .Cache}}monc{{else}}mon{{end}}.MustNewModel(url, db, {{.Type}}CollectionName{{if .Cache}}, c{{end}}) + return &custom{{.Type}}Model{ + default{{.Type}}Model: newDefault{{.Type}}Model(conn), + } +}{{else}}func New{{.Type}}Model(url, db, collection string{{if .Cache}}, c cache.CacheConf{{end}}) {{.Type}}Model { + conn := {{if .Cache}}monc{{else}}mon{{end}}.MustNewModel(url, db, collection{{if .Cache}}, c{{end}}) + return &custom{{.Type}}Model{ + default{{.Type}}Model: newDefault{{.Type}}Model(conn), + } +}{{end}} diff --git a/deploy/template/mongo/model_types.tpl b/deploy/template/mongo/model_types.tpl new file mode 100644 index 0000000..8da006f --- /dev/null +++ b/deploy/template/mongo/model_types.tpl @@ -0,0 +1,14 @@ +package model + +import ( + "time" + + "go.mongodb.org/mongo-driver/bson/primitive" +) + +type {{.Type}} struct { + ID primitive.ObjectID `bson:"_id,omitempty" json:"id,omitempty"` + // TODO: Fill your own fields + UpdateAt time.Time `bson:"updateAt,omitempty" json:"updateAt,omitempty"` + CreateAt time.Time `bson:"createAt,omitempty" json:"createAt,omitempty"` +} diff --git a/deploy/template/newapi/newtemplate.tpl b/deploy/template/newapi/newtemplate.tpl new file mode 100644 index 0000000..28be510 --- /dev/null +++ b/deploy/template/newapi/newtemplate.tpl @@ -0,0 +1,12 @@ +type Request { + Name string `path:"name,options=you|me"` +} + +type Response { + Message string `json:"message"` +} + +service {{.name}}-api { + @handler {{.handler}}Handler + get /from/:name(Request) returns (Response) +} diff --git a/deploy/template/rpc/call.tpl b/deploy/template/rpc/call.tpl new file mode 100644 index 0000000..27b4879 --- /dev/null +++ b/deploy/template/rpc/call.tpl @@ -0,0 +1,33 @@ +{{.head}} + +package {{.filePackage}} + +import ( + "context" + + {{.pbPackage}} + {{if ne .pbPackage .protoGoPackage}}{{.protoGoPackage}}{{end}} + + "github.com/zeromicro/go-zero/zrpc" + "google.golang.org/grpc" +) + +type ( + {{.alias}} + + {{.serviceName}} interface { + {{.interface}} + } + + default{{.serviceName}} struct { + cli zrpc.Client + } +) + +func New{{.serviceName}}(cli zrpc.Client) {{.serviceName}} { + return &default{{.serviceName}}{ + cli: cli, + } +} + +{{.functions}} diff --git a/deploy/template/rpc/config.tpl b/deploy/template/rpc/config.tpl new file mode 100644 index 0000000..c1f85b9 --- /dev/null +++ b/deploy/template/rpc/config.tpl @@ -0,0 +1,7 @@ +package config + +import "github.com/zeromicro/go-zero/zrpc" + +type Config struct { + zrpc.RpcServerConf +} diff --git a/deploy/template/rpc/etc.tpl b/deploy/template/rpc/etc.tpl new file mode 100644 index 0000000..6cd4bdd --- /dev/null +++ b/deploy/template/rpc/etc.tpl @@ -0,0 +1,6 @@ +Name: {{.serviceName}}.rpc +ListenOn: 0.0.0.0:8080 +Etcd: + Hosts: + - 127.0.0.1:2379 + Key: {{.serviceName}}.rpc diff --git a/deploy/template/rpc/logic-func.tpl b/deploy/template/rpc/logic-func.tpl new file mode 100644 index 0000000..e9410d4 --- /dev/null +++ b/deploy/template/rpc/logic-func.tpl @@ -0,0 +1,6 @@ +{{if .hasComment}}{{.comment}}{{end}} +func (l *{{.logicName}}) {{.method}} ({{if .hasReq}}in {{.request}}{{if .stream}},stream {{.streamBody}}{{end}}{{else}}stream {{.streamBody}}{{end}}) ({{if .hasReply}}{{.response}},{{end}} error) { + // todo: add your logic here and delete this line + + return {{if .hasReply}}&{{.responseType}}{},{{end}} nil +} diff --git a/deploy/template/rpc/logic.tpl b/deploy/template/rpc/logic.tpl new file mode 100644 index 0000000..b8d81f0 --- /dev/null +++ b/deploy/template/rpc/logic.tpl @@ -0,0 +1,24 @@ +package {{.packageName}} + +import ( + "context" + + {{.imports}} + + "github.com/zeromicro/go-zero/core/logx" +) + +type {{.logicName}} struct { + ctx context.Context + svcCtx *svc.ServiceContext + logx.Logger +} + +func New{{.logicName}}(ctx context.Context,svcCtx *svc.ServiceContext) *{{.logicName}} { + return &{{.logicName}}{ + ctx: ctx, + svcCtx: svcCtx, + Logger: logx.WithContext(ctx), + } +} +{{.functions}} diff --git a/deploy/template/rpc/main.tpl b/deploy/template/rpc/main.tpl new file mode 100644 index 0000000..d7bea24 --- /dev/null +++ b/deploy/template/rpc/main.tpl @@ -0,0 +1,42 @@ +package main + +import ( + "flag" + "fmt" + + {{.imports}} + "qnc-server/common/interceptor/rpcserver" + + "github.com/zeromicro/go-zero/core/conf" + "github.com/zeromicro/go-zero/core/service" + "github.com/zeromicro/go-zero/zrpc" + "google.golang.org/grpc" + "google.golang.org/grpc/reflection" +) + +var configFile = flag.String("f", "etc/{{.serviceName}}.yaml", "the config file") + +func main() { + flag.Parse() + + var c config.Config + conf.MustLoad(*configFile, &c) + ctx := svc.NewServiceContext(c) + + s := zrpc.MustNewServer(c.RpcServerConf, func(grpcServer *grpc.Server) { +{{range .serviceNames}} {{.Pkg}}.Register{{.Service}}Server(grpcServer, {{.ServerPkg}}.New{{.Service}}Server(ctx)) +{{end}} + if c.Mode == service.DevMode || c.Mode == service.TestMode { + reflection.Register(grpcServer) + } + }) + + //rpc log + s.AddUnaryInterceptors(rpcserver.LoggerInterceptor) + + defer s.Stop() + + fmt.Printf("Starting rpc server at %s...\n", c.ListenOn) + s.Start() +} + diff --git a/deploy/template/rpc/server-func.tpl b/deploy/template/rpc/server-func.tpl new file mode 100644 index 0000000..d771b43 --- /dev/null +++ b/deploy/template/rpc/server-func.tpl @@ -0,0 +1,6 @@ + +{{if .hasComment}}{{.comment}}{{end}} +func (s *{{.server}}Server) {{.method}} ({{if .notStream}}ctx context.Context,{{if .hasReq}} in {{.request}}{{end}}{{else}}{{if .hasReq}} in {{.request}},{{end}}stream {{.streamBody}}{{end}}) ({{if .notStream}}{{.response}},{{end}}error) { + l := {{.logicPkg}}.New{{.logicName}}({{if .notStream}}ctx,{{else}}stream.Context(),{{end}}s.svcCtx) + return l.{{.method}}({{if .hasReq}}in{{if .stream}} ,stream{{end}}{{else}}{{if .stream}}stream{{end}}{{end}}) +} diff --git a/deploy/template/rpc/server.tpl b/deploy/template/rpc/server.tpl new file mode 100644 index 0000000..84a2f9c --- /dev/null +++ b/deploy/template/rpc/server.tpl @@ -0,0 +1,22 @@ +{{.head}} + +package server + +import ( + {{if .notStream}}"context"{{end}} + + {{.imports}} +) + +type {{.server}}Server struct { + svcCtx *svc.ServiceContext + {{.unimplementedServer}} +} + +func New{{.server}}Server(svcCtx *svc.ServiceContext) *{{.server}}Server { + return &{{.server}}Server{ + svcCtx: svcCtx, + } +} + +{{.funcs}} diff --git a/deploy/template/rpc/svc.tpl b/deploy/template/rpc/svc.tpl new file mode 100644 index 0000000..cf2b47a --- /dev/null +++ b/deploy/template/rpc/svc.tpl @@ -0,0 +1,13 @@ +package svc + +import {{.imports}} + +type ServiceContext struct { + Config config.Config +} + +func NewServiceContext(c config.Config) *ServiceContext { + return &ServiceContext{ + Config:c, + } +} diff --git a/deploy/template/rpc/template.tpl b/deploy/template/rpc/template.tpl new file mode 100644 index 0000000..76daa94 --- /dev/null +++ b/deploy/template/rpc/template.tpl @@ -0,0 +1,16 @@ +syntax = "proto3"; + +package {{.package}}; +option go_package="./{{.package}}"; + +message Request { + string ping = 1; +} + +message Response { + string pong = 1; +} + +service {{.serviceName}} { + rpc Ping(Request) returns(Response); +} diff --git a/docker-compose.dev.yml b/docker-compose.dev.yml new file mode 100644 index 0000000..d60d63f --- /dev/null +++ b/docker-compose.dev.yml @@ -0,0 +1,81 @@ +services: + mysql: + image: mysql:8.0.34 + container_name: qnc_mysql + environment: + # 时区上海 - Time zone Shanghai (Change if needed) + TZ: Asia/Shanghai + # root 密码 - root password + MYSQL_ROOT_PASSWORD: yfg87gyuYiy1 + MYSQL_DATABASE: qnc + MYSQL_USER: qnc + MYSQL_PASSWORD: 5vg67b3UNHu8 + ports: + - "21201:3306" + volumes: + # 数据挂载 - Data mounting + - ./data/mysql/data:/var/lib/mysql + # 日志 + command: + # 将mysql8.0默认密码策略 修改为 原先 策略 (mysql8.0对其默认策略做了更改 会导致密码无法匹配) + # Modify the Mysql 8.0 default password strategy to the original strategy (MySQL8.0 to change its default strategy will cause the password to be unable to match) + --default-authentication-plugin=mysql_native_password + --character-set-server=utf8mb4 + --collation-server=utf8mb4_general_ci + --explicit_defaults_for_timestamp=true + --lower_case_table_names=1 + restart: always + networks: + - qnc_net + + redis: + image: redis:7.4.0 + container_name: qnc_redis + ports: + - "21202:6379" + environment: + # 时区上海 - Time zone Shanghai (Change if needed) + TZ: Asia/Shanghai + volumes: + # 数据文件 - data files + - ./data/redis/data:/data:rw + command: "redis-server --requirepass 3m3WsgyCKWqz --appendonly yes" + privileged: true + restart: always + networks: + - qnc_net + + asynqmon: + image: hibiken/asynqmon:latest + container_name: qnc_asynqmon + ports: + - "21203:8080" + environment: + - TZ=Asia/Shanghai + command: + - "--redis-addr=qnc_redis:6379" + - "--redis-password=3m3WsgyCKWqz" + restart: always + networks: + - qnc_net + depends_on: + - redis + + phpmyadmin: + image: phpmyadmin/phpmyadmin + container_name: qnc_phpmyadmin + restart: unless-stopped + environment: + PMA_HOST: qnc_mysql + PMA_PORT: 3306 + PMA_USER: qnc + PMA_PASSWORD: 5vg67b3UNHu8 + ports: + - "21204:80" + depends_on: + - mysql + networks: + - qnc_net +networks: + qnc_net: + driver: bridge diff --git a/docker-compose.yml b/docker-compose.yml new file mode 100644 index 0000000..b754f9f --- /dev/null +++ b/docker-compose.yml @@ -0,0 +1,91 @@ +version: "3" + +services: + mysql: + image: mysql:8.0.34 + container_name: qnc_mysql + environment: + # 时区上海 - Time zone Shanghai (Change if needed) + TZ: Asia/Shanghai + # root 密码 - root password + MYSQL_ROOT_PASSWORD: yfg87gyuYiy1 + MYSQL_DATABASE: qnc + MYSQL_USER: qnc + MYSQL_PASSWORD: 5vg67b3UNHu8 + ports: + - "21201:3306" + volumes: + # 数据挂载 - Data mounting + - ./data/mysql/data:/var/lib/mysql + # 日志 + command: + # 将mysql8.0默认密码策略 修改为 原先 策略 (mysql8.0对其默认策略做了更改 会导致密码无法匹配) + # Modify the Mysql 8.0 default password strategy to the original strategy (MySQL8.0 to change its default strategy will cause the password to be unable to match) + --default-authentication-plugin=mysql_native_password + --character-set-server=utf8mb4 + --collation-server=utf8mb4_general_ci + --explicit_defaults_for_timestamp=true + --lower_case_table_names=1 + privileged: true + restart: always + networks: + - qnc_net + - 1panel-network + + redis: + image: redis:7.4.0 + container_name: qnc_redis + ports: + - "21202:6379" + environment: + # 时区上海 - Time zone Shanghai (Change if needed) + TZ: Asia/Shanghai + volumes: + # 数据文件 - data files + - ./data/redis/data:/data:rw + command: "redis-server --requirepass 3m3WsgyCKWqz --appendonly yes" + privileged: true + restart: always + networks: + - qnc_net + + asynqmon: + image: hibiken/asynqmon:latest + container_name: qnc_asynqmon + ports: + - "21203:8080" + environment: + - TZ=Asia/Shanghai + command: + - "--redis-addr=qnc_redis:6379" + - "--redis-password=3m3WsgyCKWqz" + restart: always + networks: + - qnc_net + depends_on: + - redis + + main: + container_name: qnc_main + build: + context: . + dockerfile: app/main/api/Dockerfile + ports: + - "21204:8888" + volumes: + - ./data/authorization_docs:/app/data/authorization_docs:rw + environment: + - TZ=Asia/Shanghai + - ENV=production + depends_on: + - mysql + - redis + networks: + - qnc_net + restart: always + +networks: + qnc_net: + driver: bridge + 1panel-network: + external: true diff --git a/gen_api.ps1 b/gen_api.ps1 new file mode 100644 index 0000000..d571f8e --- /dev/null +++ b/gen_api.ps1 @@ -0,0 +1,2 @@ +# API生成脚本 +goctl api go --api ./app/main/api/desc/main.api --dir ./app/main/api --home ./deploy/template \ No newline at end of file diff --git a/go.mod b/go.mod new file mode 100644 index 0000000..ab11096 --- /dev/null +++ b/go.mod @@ -0,0 +1,115 @@ +module qnc-server + +go 1.23.0 + +toolchain go1.23.4 + +require ( + github.com/Masterminds/squirrel v1.5.4 + github.com/alibabacloud-go/darabonba-openapi/v2 v2.0.10 + github.com/alibabacloud-go/dysmsapi-20170525/v3 v3.0.6 + github.com/alibabacloud-go/tea v1.2.2 + github.com/alibabacloud-go/tea-utils/v2 v2.0.7 + github.com/bytedance/sonic v1.13.0 + github.com/cenkalti/backoff/v4 v4.3.0 + github.com/fogleman/gg v1.3.0 + github.com/go-playground/validator/v10 v10.22.1 + github.com/golang-jwt/jwt/v4 v4.5.0 + github.com/google/uuid v1.6.0 + github.com/hibiken/asynq v0.25.0 + github.com/jinzhu/copier v0.4.0 + github.com/jung-kurt/gofpdf v1.16.2 + github.com/pkg/errors v0.9.1 + github.com/redis/go-redis/v9 v9.7.0 + github.com/samber/lo v1.50.0 + github.com/shopspring/decimal v1.4.0 + github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e + github.com/smartwalle/alipay/v3 v3.2.23 + github.com/sony/sonyflake v1.2.0 + github.com/stretchr/testify v1.10.0 + github.com/tidwall/gjson v1.18.0 + github.com/wechatpay-apiv3/wechatpay-go v0.2.20 + github.com/zeromicro/go-zero v1.7.3 + golang.org/x/crypto v0.28.0 + google.golang.org/grpc v1.67.1 +) + +require ( + filippo.io/edwards25519 v1.1.0 // indirect + github.com/alibabacloud-go/alibabacloud-gateway-spi v0.0.5 // indirect + github.com/alibabacloud-go/debug v1.0.1 // indirect + github.com/alibabacloud-go/endpoint-util v1.1.0 // indirect + github.com/alibabacloud-go/openapi-util v0.1.0 // indirect + github.com/alibabacloud-go/tea-utils v1.3.1 // indirect + github.com/alibabacloud-go/tea-xml v1.1.3 // indirect + github.com/aliyun/credentials-go v1.3.10 // indirect + github.com/beorn7/perks v1.0.1 // indirect + github.com/bytedance/sonic/loader v0.2.2 // indirect + github.com/cespare/xxhash/v2 v2.3.0 // indirect + github.com/clbanning/mxj/v2 v2.5.5 // indirect + github.com/cloudwego/base64x v0.1.5 // indirect + github.com/davecgh/go-spew v1.1.1 // indirect + github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f // indirect + github.com/fatih/color v1.17.0 // indirect + github.com/gabriel-vasile/mimetype v1.4.3 // indirect + github.com/go-logr/logr v1.4.2 // indirect + github.com/go-logr/stdr v1.2.2 // indirect + github.com/go-playground/locales v0.14.1 // indirect + github.com/go-playground/universal-translator v0.18.1 // indirect + github.com/go-sql-driver/mysql v1.8.1 // indirect + github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 // indirect + github.com/grpc-ecosystem/grpc-gateway/v2 v2.20.0 // indirect + github.com/json-iterator/go v1.1.12 // indirect + github.com/klauspost/compress v1.17.9 // indirect + github.com/klauspost/cpuid/v2 v2.0.9 // indirect + github.com/lann/builder v0.0.0-20180802200727-47ae307949d0 // indirect + github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0 // indirect + github.com/leodido/go-urn v1.4.0 // indirect + github.com/mattn/go-colorable v0.1.13 // indirect + github.com/mattn/go-isatty v0.0.20 // indirect + github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect + github.com/modern-go/reflect2 v1.0.2 // indirect + github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect + github.com/openzipkin/zipkin-go v0.4.3 // indirect + github.com/pelletier/go-toml/v2 v2.2.2 // indirect + github.com/pmezard/go-difflib v1.0.0 // indirect + github.com/prometheus/client_golang v1.20.5 // indirect + github.com/prometheus/client_model v0.6.1 // indirect + github.com/prometheus/common v0.55.0 // indirect + github.com/prometheus/procfs v0.15.1 // indirect + github.com/robfig/cron/v3 v3.0.1 // indirect + github.com/smartwalle/ncrypto v1.0.4 // indirect + github.com/smartwalle/ngx v1.0.9 // indirect + github.com/smartwalle/nsign v1.0.9 // indirect + github.com/spaolacci/murmur3 v1.1.0 // indirect + github.com/spf13/cast v1.7.0 // indirect + github.com/stretchr/objx v0.5.2 // indirect + github.com/tidwall/match v1.1.1 // indirect + github.com/tidwall/pretty v1.2.1 // indirect + github.com/tjfoc/gmsm v1.4.1 // indirect + github.com/twitchyliquid64/golang-asm v0.15.1 // indirect + go.opentelemetry.io/otel v1.24.0 // indirect + go.opentelemetry.io/otel/exporters/jaeger v1.17.0 // indirect + go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.24.0 // indirect + go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc v1.24.0 // indirect + go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.24.0 // indirect + go.opentelemetry.io/otel/exporters/stdout/stdouttrace v1.24.0 // indirect + go.opentelemetry.io/otel/exporters/zipkin v1.24.0 // indirect + go.opentelemetry.io/otel/metric v1.24.0 // indirect + go.opentelemetry.io/otel/sdk v1.24.0 // indirect + go.opentelemetry.io/otel/trace v1.24.0 // indirect + go.opentelemetry.io/proto/otlp v1.3.1 // indirect + go.uber.org/automaxprocs v1.6.0 // indirect + golang.org/x/arch v0.0.0-20210923205945-b76863e36670 // indirect + golang.org/x/image v0.28.0 // indirect + golang.org/x/net v0.30.0 // indirect + golang.org/x/sys v0.26.0 // indirect + golang.org/x/text v0.26.0 // indirect + golang.org/x/time v0.7.0 // indirect + google.golang.org/genproto/googleapis/api v0.0.0-20240814211410-ddb44dafa142 // indirect + google.golang.org/genproto/googleapis/rpc v0.0.0-20240814211410-ddb44dafa142 // indirect + google.golang.org/protobuf v1.35.1 // indirect + gopkg.in/ini.v1 v1.67.0 // indirect + gopkg.in/yaml.v2 v2.4.0 // indirect + gopkg.in/yaml.v3 v3.0.1 // indirect +) diff --git a/go.sum b/go.sum new file mode 100644 index 0000000..0fa02f5 --- /dev/null +++ b/go.sum @@ -0,0 +1,468 @@ +cloud.google.com/go v0.26.0/go.mod h1:aQUYkXzVsufM+DwF1aE+0xfcU+56JwCaLick0ClmMTw= +filippo.io/edwards25519 v1.1.0 h1:FNf4tywRC1HmFuKW5xopWpigGjJKiJSV0Cqo0cJWDaA= +filippo.io/edwards25519 v1.1.0/go.mod h1:BxyFTGdWcka3PhytdK4V28tE5sGfRvvvRV7EaN4VDT4= +github.com/BurntSushi/toml v0.3.1/go.mod h1:xHWCNGjB5oqiDr8zfno3MHue2Ht5sIBksp03qcyfWMU= +github.com/DATA-DOG/go-sqlmock v1.5.2 h1:OcvFkGmslmlZibjAjaHm3L//6LiuBgolP7OputlJIzU= +github.com/DATA-DOG/go-sqlmock v1.5.2/go.mod h1:88MAG/4G7SMwSE3CeA0ZKzrT5CiOU3OJ+JlNzwDqpNU= +github.com/Masterminds/squirrel v1.5.4 h1:uUcX/aBc8O7Fg9kaISIUsHXdKuqehiXAMQTYX8afzqM= +github.com/Masterminds/squirrel v1.5.4/go.mod h1:NNaOrjSoIDfDA40n7sr2tPNZRfjzjA400rg+riTZj10= +github.com/agiledragon/gomonkey v2.0.2+incompatible h1:eXKi9/piiC3cjJD1658mEE2o3NjkJ5vDLgYjCQu0Xlw= +github.com/agiledragon/gomonkey v2.0.2+incompatible/go.mod h1:2NGfXu1a80LLr2cmWXGBDaHEjb1idR6+FVlX5T3D9hw= +github.com/alibabacloud-go/alibabacloud-gateway-pop v0.0.6 h1:eIf+iGJxdU4U9ypaUfbtOWCsZSbTb8AUHvyPrxu6mAA= +github.com/alibabacloud-go/alibabacloud-gateway-pop v0.0.6/go.mod h1:4EUIoxs/do24zMOGGqYVWgw0s9NtiylnJglOeEB5UJo= +github.com/alibabacloud-go/alibabacloud-gateway-spi v0.0.4/go.mod h1:sCavSAvdzOjul4cEqeVtvlSaSScfNsTQ+46HwlTL1hc= +github.com/alibabacloud-go/alibabacloud-gateway-spi v0.0.5 h1:zE8vH9C7JiZLNJJQ5OwjU9mSi4T9ef9u3BURT6LCLC8= +github.com/alibabacloud-go/alibabacloud-gateway-spi v0.0.5/go.mod h1:tWnyE9AjF8J8qqLk645oUmVUnFybApTQWklQmi5tY6g= +github.com/alibabacloud-go/darabonba-array v0.1.0 h1:vR8s7b1fWAQIjEjWnuF0JiKsCvclSRTfDzZHTYqfufY= +github.com/alibabacloud-go/darabonba-array v0.1.0/go.mod h1:BLKxr0brnggqOJPqT09DFJ8g3fsDshapUD3C3aOEFaI= +github.com/alibabacloud-go/darabonba-encode-util v0.0.2 h1:1uJGrbsGEVqWcWxrS9MyC2NG0Ax+GpOM5gtupki31XE= +github.com/alibabacloud-go/darabonba-encode-util v0.0.2/go.mod h1:JiW9higWHYXm7F4PKuMgEUETNZasrDM6vqVr/Can7H8= +github.com/alibabacloud-go/darabonba-map v0.0.2 h1:qvPnGB4+dJbJIxOOfawxzF3hzMnIpjmafa0qOTp6udc= +github.com/alibabacloud-go/darabonba-map v0.0.2/go.mod h1:28AJaX8FOE/ym8OUFWga+MtEzBunJwQGceGQlvaPGPc= +github.com/alibabacloud-go/darabonba-openapi/v2 v2.0.2/go.mod h1:5JHVmnHvGzR2wNdgaW1zDLQG8kOC4Uec8ubkMogW7OQ= +github.com/alibabacloud-go/darabonba-openapi/v2 v2.0.10 h1:GEYkMApgpKEVDn6z12DcH1EGYpDYRB8JxsazM4Rywak= +github.com/alibabacloud-go/darabonba-openapi/v2 v2.0.10/go.mod h1:26a14FGhZVELuz2cc2AolvW4RHmIO3/HRwsdHhaIPDE= +github.com/alibabacloud-go/darabonba-signature-util v0.0.7 h1:UzCnKvsjPFzApvODDNEYqBHMFt1w98wC7FOo0InLyxg= +github.com/alibabacloud-go/darabonba-signature-util v0.0.7/go.mod h1:oUzCYV2fcCH797xKdL6BDH8ADIHlzrtKVjeRtunBNTQ= +github.com/alibabacloud-go/darabonba-string v1.0.2 h1:E714wms5ibdzCqGeYJ9JCFywE5nDyvIXIIQbZVFkkqo= +github.com/alibabacloud-go/darabonba-string v1.0.2/go.mod h1:93cTfV3vuPhhEwGGpKKqhVW4jLe7tDpo3LUM0i0g6mA= +github.com/alibabacloud-go/debug v0.0.0-20190504072949-9472017b5c68/go.mod h1:6pb/Qy8c+lqua8cFpEy7g39NRRqOWc3rOwAy8m5Y2BY= +github.com/alibabacloud-go/debug v1.0.0/go.mod h1:8gfgZCCAC3+SCzjWtY053FrOcd4/qlH6IHTI4QyICOc= +github.com/alibabacloud-go/debug v1.0.1 h1:MsW9SmUtbb1Fnt3ieC6NNZi6aEwrXfDksD4QA6GSbPg= +github.com/alibabacloud-go/debug v1.0.1/go.mod h1:8gfgZCCAC3+SCzjWtY053FrOcd4/qlH6IHTI4QyICOc= +github.com/alibabacloud-go/dysmsapi-20170525/v3 v3.0.6 h1:UTl97mt2qfavxveqCkaVg4tKaZUPzA9RKbFIRaIdtdg= +github.com/alibabacloud-go/dysmsapi-20170525/v3 v3.0.6/go.mod h1:UWpcGrWwTbES9QW7OQ7xDffukMJ/l7lzioixIz8+lgY= +github.com/alibabacloud-go/endpoint-util v1.1.0 h1:r/4D3VSw888XGaeNpP994zDUaxdgTSHBbVfZlzf6b5Q= +github.com/alibabacloud-go/endpoint-util v1.1.0/go.mod h1:O5FuCALmCKs2Ff7JFJMudHs0I5EBgecXXxZRyswlEjE= +github.com/alibabacloud-go/openapi-util v0.0.11/go.mod h1:sQuElr4ywwFRlCCberQwKRFhRzIyG4QTP/P4y1CJ6Ws= +github.com/alibabacloud-go/openapi-util v0.1.0 h1:0z75cIULkDrdEhkLWgi9tnLe+KhAFE/r5Pb3312/eAY= +github.com/alibabacloud-go/openapi-util v0.1.0/go.mod h1:sQuElr4ywwFRlCCberQwKRFhRzIyG4QTP/P4y1CJ6Ws= +github.com/alibabacloud-go/tea v1.1.0/go.mod h1:IkGyUSX4Ba1V+k4pCtJUc6jDpZLFph9QMy2VUPTwukg= +github.com/alibabacloud-go/tea v1.1.7/go.mod h1:/tmnEaQMyb4Ky1/5D+SE1BAsa5zj/KeGOFfwYm3N/p4= +github.com/alibabacloud-go/tea v1.1.8/go.mod h1:/tmnEaQMyb4Ky1/5D+SE1BAsa5zj/KeGOFfwYm3N/p4= +github.com/alibabacloud-go/tea v1.1.11/go.mod h1:/tmnEaQMyb4Ky1/5D+SE1BAsa5zj/KeGOFfwYm3N/p4= +github.com/alibabacloud-go/tea v1.1.17/go.mod h1:nXxjm6CIFkBhwW4FQkNrolwbfon8Svy6cujmKFUq98A= +github.com/alibabacloud-go/tea v1.1.19/go.mod h1:nXxjm6CIFkBhwW4FQkNrolwbfon8Svy6cujmKFUq98A= +github.com/alibabacloud-go/tea v1.1.20/go.mod h1:nXxjm6CIFkBhwW4FQkNrolwbfon8Svy6cujmKFUq98A= +github.com/alibabacloud-go/tea v1.2.2 h1:aTsR6Rl3ANWPfqeQugPglfurloyBJY85eFy7Gc1+8oU= +github.com/alibabacloud-go/tea v1.2.2/go.mod h1:CF3vOzEMAG+bR4WOql8gc2G9H3EkH3ZLAQdpmpXMgwk= +github.com/alibabacloud-go/tea-utils v1.3.1 h1:iWQeRzRheqCMuiF3+XkfybB3kTgUXkXX+JMrqfLeB2I= +github.com/alibabacloud-go/tea-utils v1.3.1/go.mod h1:EI/o33aBfj3hETm4RLiAxF/ThQdSngxrpF8rKUDJjPE= +github.com/alibabacloud-go/tea-utils/v2 v2.0.0/go.mod h1:U5MTY10WwlquGPS34DOeomUGBB0gXbLueiq5Trwu0C4= +github.com/alibabacloud-go/tea-utils/v2 v2.0.3/go.mod h1:sj1PbjPodAVTqGTA3olprfeeqqmwD0A5OQz94o9EuXQ= +github.com/alibabacloud-go/tea-utils/v2 v2.0.5/go.mod h1:dL6vbUT35E4F4bFTHL845eUloqaerYBYPsdWR2/jhe4= +github.com/alibabacloud-go/tea-utils/v2 v2.0.6/go.mod h1:qxn986l+q33J5VkialKMqT/TTs3E+U9MJpd001iWQ9I= +github.com/alibabacloud-go/tea-utils/v2 v2.0.7 h1:WDx5qW3Xa5ZgJ1c8NfqJkF6w+AU5wB8835UdhPr6Ax0= +github.com/alibabacloud-go/tea-utils/v2 v2.0.7/go.mod h1:qxn986l+q33J5VkialKMqT/TTs3E+U9MJpd001iWQ9I= +github.com/alibabacloud-go/tea-xml v1.1.2/go.mod h1:Rq08vgCcCAjHyRi/M7xlHKUykZCEtyBy9+DPF6GgEu8= +github.com/alibabacloud-go/tea-xml v1.1.3 h1:7LYnm+JbOq2B+T/B0fHC4Ies4/FofC4zHzYtqw7dgt0= +github.com/alibabacloud-go/tea-xml v1.1.3/go.mod h1:Rq08vgCcCAjHyRi/M7xlHKUykZCEtyBy9+DPF6GgEu8= +github.com/alicebob/gopher-json v0.0.0-20230218143504-906a9b012302 h1:uvdUDbHQHO85qeSydJtItA4T55Pw6BtAejd0APRJOCE= +github.com/alicebob/gopher-json v0.0.0-20230218143504-906a9b012302/go.mod h1:SGnFV6hVsYE877CKEZ6tDNTjaSXYUk6QqoIK6PrAtcc= +github.com/alicebob/miniredis/v2 v2.33.0 h1:uvTF0EDeu9RLnUEG27Db5I68ESoIxTiXbNUiji6lZrA= +github.com/alicebob/miniredis/v2 v2.33.0/go.mod h1:MhP4a3EU7aENRi9aO+tHfTBZicLqQevyi/DJpoj6mi0= +github.com/aliyun/credentials-go v1.1.2/go.mod h1:ozcZaMR5kLM7pwtCMEpVmQ242suV6qTJya2bDq4X1Tw= +github.com/aliyun/credentials-go v1.3.1/go.mod h1:8jKYhQuDawt8x2+fusqa1Y6mPxemTsBEN04dgcAcYz0= +github.com/aliyun/credentials-go v1.3.6/go.mod h1:1LxUuX7L5YrZUWzBrRyk0SwSdH4OmPrib8NVePL3fxM= +github.com/aliyun/credentials-go v1.3.10 h1:45Xxrae/evfzQL9V10zL3xX31eqgLWEaIdCoPipOEQA= +github.com/aliyun/credentials-go v1.3.10/go.mod h1:Jm6d+xIgwJVLVWT561vy67ZRP4lPTQxMbEYRuT2Ti1U= +github.com/beorn7/perks v1.0.1 h1:VlbKKnNfV8bJzeqoa4cOKqO6bYr3WgKZxO8Z16+hsOM= +github.com/beorn7/perks v1.0.1/go.mod h1:G2ZrVWU2WbWT9wwq4/hrbKbnv/1ERSJQ0ibhJ6rlkpw= +github.com/boombuler/barcode v1.0.0/go.mod h1:paBWMcWSl3LHKBqUq+rly7CNSldXjb2rDl3JlRe0mD8= +github.com/bsm/ginkgo/v2 v2.12.0 h1:Ny8MWAHyOepLGlLKYmXG4IEkioBysk6GpaRTLC8zwWs= +github.com/bsm/ginkgo/v2 v2.12.0/go.mod h1:SwYbGRRDovPVboqFv0tPTcG1sN61LM1Z4ARdbAV9g4c= +github.com/bsm/gomega v1.27.10 h1:yeMWxP2pV2fG3FgAODIY8EiRE3dy0aeFYt4l7wh6yKA= +github.com/bsm/gomega v1.27.10/go.mod h1:JyEr/xRbxbtgWNi8tIEVPUYZ5Dzef52k01W3YH0H+O0= +github.com/bytedance/sonic v1.13.0 h1:R+aSALdYjFT39PoytNFIxV8W7rb/ZxRpdQd+1TFZ2F0= +github.com/bytedance/sonic v1.13.0/go.mod h1:uVvFidNmlt9+wa31S1urfwwthTWteBgG0hWuoKAXTx8= +github.com/bytedance/sonic/loader v0.1.1/go.mod h1:ncP89zfokxS5LZrJxl5z0UJcsk4M4yY2JpfqGeCtNLU= +github.com/bytedance/sonic/loader v0.2.2 h1:jxAJuN9fOot/cyz5Q6dUuMJF5OqQ6+5GfA8FjjQ0R4o= +github.com/bytedance/sonic/loader v0.2.2/go.mod h1:N8A3vUdtUebEY2/VQC0MyhYeKUFosQU6FxH2JmUe6VI= +github.com/cenkalti/backoff/v4 v4.3.0 h1:MyRJ/UdXutAwSAT+s3wNd7MfTIcy71VQueUuFK343L8= +github.com/cenkalti/backoff/v4 v4.3.0/go.mod h1:Y3VNntkOUPxTVeUxJ/G5vcM//AlwfmyYozVcomhLiZE= +github.com/census-instrumentation/opencensus-proto v0.2.1/go.mod h1:f6KPmirojxKA12rnyqOA5BBL4O983OfeGPqjHWSTneU= +github.com/cespare/xxhash/v2 v2.3.0 h1:UL815xU9SqsFlibzuggzjXhog7bL6oX9BbNZnL2UFvs= +github.com/cespare/xxhash/v2 v2.3.0/go.mod h1:VGX0DQ3Q6kWi7AoAeZDth3/j3BFtOZR5XLFGgcrjCOs= +github.com/clbanning/mxj/v2 v2.5.5 h1:oT81vUeEiQQ/DcHbzSytRngP6Ky9O+L+0Bw0zSJag9E= +github.com/clbanning/mxj/v2 v2.5.5/go.mod h1:hNiWqW14h+kc+MdF9C6/YoRfjEJoR3ou6tn/Qo+ve2s= +github.com/client9/misspell v0.3.4/go.mod h1:qj6jICC3Q7zFZvVWo7KLAzC3yx5G7kyvSDkc90ppPyw= +github.com/cloudwego/base64x v0.1.5 h1:XPciSp1xaq2VCSt6lF0phncD4koWyULpl5bUxbfCyP4= +github.com/cloudwego/base64x v0.1.5/go.mod h1:0zlkT4Wn5C6NdauXdJRhSKRlJvmclQ1hhJgA0rcu/8w= +github.com/cloudwego/iasm v0.2.0/go.mod h1:8rXZaNYT2n95jn+zTI1sDr+IgcD2GVs0nlbbQPiEFhY= +github.com/cncf/udpa/go v0.0.0-20191209042840-269d4d468f6f/go.mod h1:M8M6+tZqaGXZJjfX53e64911xZQV5JYwmTeXPW+k8Sc= +github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c= +github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f h1:lO4WD4F/rVNCu3HqELle0jiPLLBs70cWOduZpkS1E78= +github.com/dgryski/go-rendezvous v0.0.0-20200823014737-9f7001d12a5f/go.mod h1:cuUVRXasLTGF7a8hSLbxyZXjz+1KgoB3wDUb6vlszIc= +github.com/envoyproxy/go-control-plane v0.9.0/go.mod h1:YTl/9mNaCwkRvm6d1a2C3ymFceY/DCBVvsKhRF0iEA4= +github.com/envoyproxy/go-control-plane v0.9.4/go.mod h1:6rpuAdCZL397s3pYoYcLgu1mIlRU8Am5FuJP05cCM98= +github.com/envoyproxy/protoc-gen-validate v0.1.0/go.mod h1:iSmxcyjqTsJpI2R4NaDN7+kN2VEUnK/pcBlmesArF7c= +github.com/fatih/color v1.17.0 h1:GlRw1BRJxkpqUCBKzKOw098ed57fEsKeNjpTe3cSjK4= +github.com/fatih/color v1.17.0/go.mod h1:YZ7TlrGPkiz6ku9fK3TLD/pl3CpsiFyu8N92HLgmosI= +github.com/fogleman/gg v1.3.0 h1:/7zJX8F6AaYQc57WQCyN9cAIz+4bCJGO9B+dyW29am8= +github.com/fogleman/gg v1.3.0/go.mod h1:R/bRT+9gY/C5z7JzPU0zXsXHKM4/ayA+zqcVNZzPa1k= +github.com/frankban/quicktest v1.14.6 h1:7Xjx+VpznH+oBnejlPUj8oUpdxnVs4f8XU8WnHkI4W8= +github.com/frankban/quicktest v1.14.6/go.mod h1:4ptaffx2x8+WTWXmUCuVU6aPUX1/Mz7zb5vbUoiM6w0= +github.com/gabriel-vasile/mimetype v1.4.3 h1:in2uUcidCuFcDKtdcBxlR0rJ1+fsokWf+uqxgUFjbI0= +github.com/gabriel-vasile/mimetype v1.4.3/go.mod h1:d8uq/6HKRL6CGdk+aubisF/M5GcPfT7nKyLpA0lbSSk= +github.com/go-logr/logr v1.2.2/go.mod h1:jdQByPbusPIv2/zmleS9BjJVeZ6kBagPoEUsqbVz/1A= +github.com/go-logr/logr v1.4.2 h1:6pFjapn8bFcIbiKo3XT4j/BhANplGihG6tvd+8rYgrY= +github.com/go-logr/logr v1.4.2/go.mod h1:9T104GzyrTigFIr8wt5mBrctHMim0Nb2HLGrmQ40KvY= +github.com/go-logr/stdr v1.2.2 h1:hSWxHoqTgW2S2qGc0LTAI563KZ5YKYRhT3MFKZMbjag= +github.com/go-logr/stdr v1.2.2/go.mod h1:mMo/vtBO5dYbehREoey6XUKy/eSumjCCveDpRre4VKE= +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.22.1 h1:40JcKH+bBNGFczGuoBYgX4I6m/i27HYW8P9FDk5PbgA= +github.com/go-playground/validator/v10 v10.22.1/go.mod h1:dbuPbCMFw/DrkbEynArYaCwl3amGuJotoKCe95atGMM= +github.com/go-sql-driver/mysql v1.8.1 h1:LedoTUt/eveggdHS9qUFC1EFSa8bU2+1pZjSRpvNJ1Y= +github.com/go-sql-driver/mysql v1.8.1/go.mod h1:wEBSXgmK//2ZFJyE+qWnIsVGmvmEKlqwuVSjsCm7DZg= +github.com/golang-jwt/jwt/v4 v4.5.0 h1:7cYmW1XlMY7h7ii7UhUyChSgS5wUJEnm9uZVTGqOWzg= +github.com/golang-jwt/jwt/v4 v4.5.0/go.mod h1:m21LjoU+eqJr34lmDMbreY2eSTRJ1cv77w39/MY0Ch0= +github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0 h1:DACJavvAHhabrF08vX0COfcOBJRhZ8lUbR+ZWIs0Y5g= +github.com/golang/freetype v0.0.0-20170609003504-e2365dfdc4a0/go.mod h1:E/TSTwGwJL78qG/PmXZO1EjYhfJinVAhrmmHX6Z8B9k= +github.com/golang/glog v0.0.0-20160126235308-23def4e6c14b/go.mod h1:SBH7ygxi8pfUlaOkMMuAQtPIUF8ecWP5IEl/CR7VP2Q= +github.com/golang/mock v1.1.1/go.mod h1:oTYuIxOrZwtPieC+H1uAHpcLFnEyAGVDL/k47Jfbm0A= +github.com/golang/protobuf v1.2.0/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.3.2/go.mod h1:6lQm79b+lXiMfvg/cZm0SGofjICqVBUtrP5yJMmIC1U= +github.com/golang/protobuf v1.3.3/go.mod h1:vzj43D7+SQXF/4pzW/hwtAqwc6iTitCiVSaWz5lYuqw= +github.com/golang/protobuf v1.4.0-rc.1/go.mod h1:ceaxUfeHdC40wWswd/P6IGgMaK3YpKi5j83Wpe3EHw8= +github.com/golang/protobuf v1.4.0-rc.1.0.20200221234624-67d41d38c208/go.mod h1:xKAWHe0F5eneWXFV3EuXVDTCmh+JuBKY0li0aMyXATA= +github.com/golang/protobuf v1.4.0-rc.2/go.mod h1:LlEzMj4AhA7rCAGe4KMBDvJI+AwstrUpVNzEA03Pprs= +github.com/golang/protobuf v1.4.0-rc.4.0.20200313231945-b860323f09d0/go.mod h1:WU3c8KckQ9AFe+yFwt9sWVRKCVIyN9cPHBJSNnbL67w= +github.com/golang/protobuf v1.4.0/go.mod h1:jodUvKwWbYaEsadDk5Fwe5c77LiNKVO9IDvqG2KuDX0= +github.com/golang/protobuf v1.4.2/go.mod h1:oDoupMAO8OvCJWAcko0GGGIgR6R6ocIYbsSw735rRwI= +github.com/google/go-cmp v0.2.0/go.mod h1:oXzfMopK8JAjlY9xF4vHSVASa0yLyX7SntLO5aqRK0M= +github.com/google/go-cmp v0.3.0/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.3.1/go.mod h1:8QqcDgzrUqlUb/G2PQTWiueGozuR1884gddMywk6iLU= +github.com/google/go-cmp v0.4.0/go.mod h1:v8dTdLbMG2kIc/vJvl+f65V22dbkXbowE6jgT/gNBxE= +github.com/google/go-cmp v0.6.0 h1:ofyhxvXcZhMsU5ulbFiLKl/XBFqE1GSq7atu8tAmTRI= +github.com/google/go-cmp v0.6.0/go.mod h1:17dUlkBOakJ0+DkrSSNjCkIjxS6bF9zb3elmeNGIjoY= +github.com/google/gofuzz v1.0.0/go.mod h1:dBl0BpW6vV/+mYPU4Po3pmUjxk6FQPldtuIdl/M65Eg= +github.com/google/uuid v1.6.0 h1:NIvaJDMOsjHA8n1jAhLSgzrAzy1Hgr+hNrb57e+94F0= +github.com/google/uuid v1.6.0/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo= +github.com/gopherjs/gopherjs v0.0.0-20181017120253-0766667cb4d1/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY= +github.com/gopherjs/gopherjs v0.0.0-20200217142428-fce0ec30dd00/go.mod h1:wJfORRmW1u3UXTncJ5qlYoELFm8eSnnEO6hX4iZ3EWY= +github.com/grpc-ecosystem/grpc-gateway/v2 v2.20.0 h1:bkypFPDjIYGfCYD5mRBvpqxfYX1YCS1PXdKYWi8FsN0= +github.com/grpc-ecosystem/grpc-gateway/v2 v2.20.0/go.mod h1:P+Lt/0by1T8bfcF3z737NnSbmxQAppXMRziHUxPOC8k= +github.com/h2non/parth v0.0.0-20190131123155-b4df798d6542 h1:2VTzZjLZBgl62/EtslCrtky5vbi9dd7HrQPQIx6wqiw= +github.com/h2non/parth v0.0.0-20190131123155-b4df798d6542/go.mod h1:Ow0tF8D4Kplbc8s8sSb3V2oUCygFHVp8gC3Dn6U4MNI= +github.com/hibiken/asynq v0.25.0 h1:VCPyRRrrjFChsTSI8x5OCPu51MlEz6Rk+1p0kHKnZug= +github.com/hibiken/asynq v0.25.0/go.mod h1:DYQ1etBEl2Y+uSkqFElGYbk3M0ujLVwCfWE+TlvxtEk= +github.com/jinzhu/copier v0.4.0 h1:w3ciUoD19shMCRargcpm0cm91ytaBhDvuRpz1ODO/U8= +github.com/jinzhu/copier v0.4.0/go.mod h1:DfbEm0FYsaqBcKcFuvmOZb218JkPGtvSHsKg8S8hyyg= +github.com/json-iterator/go v1.1.10/go.mod h1:KdQUCv79m/52Kvf8AW2vK1V8akMuk1QjK/uOdHXbAo4= +github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM= +github.com/json-iterator/go v1.1.12/go.mod h1:e30LSqwooZae/UwlEbR2852Gd8hjQvJoHmT4TnhNGBo= +github.com/jtolds/gls v4.20.0+incompatible/go.mod h1:QJZ7F/aHp+rZTRtaJ1ow/lLfFfVYBRgL+9YlvaHOwJU= +github.com/jung-kurt/gofpdf v1.0.0/go.mod h1:7Id9E/uU8ce6rXgefFLlgrJj/GYY22cpxn+r32jIOes= +github.com/jung-kurt/gofpdf v1.16.2 h1:jgbatWHfRlPYiK85qgevsZTHviWXKwB1TTiKdz5PtRc= +github.com/jung-kurt/gofpdf v1.16.2/go.mod h1:1hl7y57EsiPAkLbOwzpzqgx1A30nQCk/YmFV8S2vmK0= +github.com/klauspost/compress v1.17.9 h1:6KIumPrER1LHsvBVuDa0r5xaG0Es51mhhB9BQB2qeMA= +github.com/klauspost/compress v1.17.9/go.mod h1:Di0epgTjJY877eYKx5yC51cX2A2Vl2ibi7bDH9ttBbw= +github.com/klauspost/cpuid/v2 v2.0.9 h1:lgaqFMSdTdQYdZ04uHyN2d/eKdOMyi2YLSvlQIBFYa4= +github.com/klauspost/cpuid/v2 v2.0.9/go.mod h1:FInQzS24/EEf25PyTYn52gqo7WaD8xa0213Md/qVLRg= +github.com/knz/go-libedit v1.10.1/go.mod h1:MZTVkCWyz0oBc7JOWP3wNAzd002ZbM/5hgShxwh4x8M= +github.com/kr/pretty v0.3.1 h1:flRD4NNwYAUpkphVc1HcthR4KEIFJ65n8Mw5qdRn3LE= +github.com/kr/pretty v0.3.1/go.mod h1:hoEshYVHaxMs3cyo3Yncou5ZscifuDolrwPKZanG3xk= +github.com/kr/pty v1.1.1/go.mod h1:pFQYn66WHrOpPYNljwOMqo10TkYh1fy3cYio2l3bCsQ= +github.com/kr/text v0.1.0/go.mod h1:4Jbv+DJW3UT/LiOwJeYQe1efqtUx/iVham/4vfdArNI= +github.com/kr/text v0.2.0 h1:5Nx0Ya0ZqY2ygV366QzturHI13Jq95ApcVaJBhpS+AY= +github.com/kr/text v0.2.0/go.mod h1:eLer722TekiGuMkidMxC/pM04lWEeraHUUmBw8l2grE= +github.com/kylelemons/godebug v1.1.0 h1:RPNrshWIDI6G2gRW9EHilWtl7Z6Sb1BR0xunSBf0SNc= +github.com/kylelemons/godebug v1.1.0/go.mod h1:9/0rRGxNHcop5bhtWyNeEfOS8JIWk580+fNqagV/RAw= +github.com/lann/builder v0.0.0-20180802200727-47ae307949d0 h1:SOEGU9fKiNWd/HOJuq6+3iTQz8KNCLtVX6idSoTLdUw= +github.com/lann/builder v0.0.0-20180802200727-47ae307949d0/go.mod h1:dXGbAdH5GtBTC4WfIxhKZfyBF/HBFgRZSWwZ9g/He9o= +github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0 h1:P6pPBnrTSX3DEVR4fDembhRWSsG5rVo6hYhAB/ADZrk= +github.com/lann/ps v0.0.0-20150810152359-62de8c46ede0/go.mod h1:vmVJ0l/dxyfGW6FmdpVm2joNMFikkuWg0EoCKLGUMNw= +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.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA= +github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg= +github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM= +github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY= +github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y= +github.com/mattn/go-runewidth v0.0.15 h1:UNAjwbU9l54TA3KzvqLGxwWjHmMgBUVhBiTjelZgg3U= +github.com/mattn/go-runewidth v0.0.15/go.mod h1:Jdepj2loyihRzMpdS35Xk/zdY8IAYHsh153qUoGf23w= +github.com/modern-go/concurrent v0.0.0-20180228061459-e0a39a4cb421/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd h1:TRLaZ9cD/w8PVh93nsPXa1VrQ6jlwL5oN8l14QlcNfg= +github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd/go.mod h1:6dJC0mAP4ikYIbvyc7fijjWJddQyLn8Ig3JB5CqoB9Q= +github.com/modern-go/reflect2 v0.0.0-20180701023420-4b7aa43c6742/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0= +github.com/modern-go/reflect2 v1.0.1/go.mod h1:bx2lNnkwVCuqBIxFjflWJWanXIb3RllmbCylyMrvgv0= +github.com/modern-go/reflect2 v1.0.2 h1:xBagoLtFs94CBntxluKeaWgTMpvLxC4ur3nMaC9Gz0M= +github.com/modern-go/reflect2 v1.0.2/go.mod h1:yWuevngMOJpCy52FWWMvUC8ws7m/LJsjYzDa0/r8luk= +github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 h1:C3w9PqII01/Oq1c1nUAm88MOHcQC9l5mIlSMApZMrHA= +github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822/go.mod h1:+n7T8mK8HuQTcFwEeznm/DIxMOiR9yIdICNftLE1DvQ= +github.com/niemeyer/pretty v0.0.0-20200227124842-a10e7caefd8e/go.mod h1:zD1mROLANZcx1PVRCS0qkT7pwLkGfwJo4zjcN/Tysno= +github.com/openzipkin/zipkin-go v0.4.3 h1:9EGwpqkgnwdEIJ+Od7QVSEIH+ocmm5nPat0G7sjsSdg= +github.com/openzipkin/zipkin-go v0.4.3/go.mod h1:M9wCJZFWCo2RiY+o1eBCEMe0Dp2S5LDHcMZmk3RmK7c= +github.com/pelletier/go-toml/v2 v2.2.2 h1:aYUidT7k73Pcl9nb2gScu7NSrKCSHIDE89b3+6Wq+LM= +github.com/pelletier/go-toml/v2 v2.2.2/go.mod h1:1t835xjRzz80PqgE6HHgN2JOsmgYu/h4qDAS4n929Rs= +github.com/phpdave11/gofpdi v1.0.7/go.mod h1:vBmVV0Do6hSBHC8uKUQ71JGW+ZGQq74llk/7bXwjDoI= +github.com/pkg/errors v0.8.1/go.mod h1:bwawxfHBFNV+L2hUp1rHADufV3IMtnDRdf1r5NINEl0= +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/prashantv/gostub v1.1.0 h1:BTyx3RfQjRHnUWaGF9oQos79AlQ5k8WNktv7VGvVH4g= +github.com/prashantv/gostub v1.1.0/go.mod h1:A5zLQHz7ieHGG7is6LLXLz7I8+3LZzsrV0P1IAHhP5U= +github.com/prometheus/client_golang v1.20.5 h1:cxppBPuYhUnsO6yo/aoRol4L7q7UFfdm+bR9r+8l63Y= +github.com/prometheus/client_golang v1.20.5/go.mod h1:PIEt8X02hGcP8JWbeHyeZ53Y/jReSnHgO035n//V5WE= +github.com/prometheus/client_model v0.0.0-20190812154241-14fe0d1b01d4/go.mod h1:xMI15A0UPsDsEKsMN9yxemIoYk6Tm2C1GtYGdfGttqA= +github.com/prometheus/client_model v0.6.1 h1:ZKSh/rekM+n3CeS952MLRAdFwIKqeY8b62p8ais2e9E= +github.com/prometheus/client_model v0.6.1/go.mod h1:OrxVMOVHjw3lKMa8+x6HeMGkHMQyHDk9E3jmP2AmGiY= +github.com/prometheus/common v0.55.0 h1:KEi6DK7lXW/m7Ig5i47x0vRzuBsHuvJdi5ee6Y3G1dc= +github.com/prometheus/common v0.55.0/go.mod h1:2SECS4xJG1kd8XF9IcM1gMX6510RAEL65zxzNImwdc8= +github.com/prometheus/procfs v0.15.1 h1:YagwOFzUgYfKKHX6Dr+sHT7km/hxC76UB0learggepc= +github.com/prometheus/procfs v0.15.1/go.mod h1:fB45yRUv8NstnjriLhBQLuOUt+WW4BsoGhij/e3PBqk= +github.com/redis/go-redis/v9 v9.7.0 h1:HhLSs+B6O021gwzl+locl0zEDnyNkxMtf/Z3NNBMa9E= +github.com/redis/go-redis/v9 v9.7.0/go.mod h1:f6zhXITC7JUJIlPEiBOTXxJgPLdZcA93GewI7inzyWw= +github.com/rivo/uniseg v0.2.0 h1:S1pD9weZBuJdFmowNwbpi7BJ8TNftyUImj/0WQi72jY= +github.com/rivo/uniseg v0.2.0/go.mod h1:J6wj4VEh+S6ZtnVlnTBMWIodfgj8LQOQFoIToxlJtxc= +github.com/robfig/cron/v3 v3.0.1 h1:WdRxkvbJztn8LMz/QEvLN5sBU+xKpSqwwUO1Pjr4qDs= +github.com/robfig/cron/v3 v3.0.1/go.mod h1:eQICP3HwyT7UooqI/z+Ov+PtYAWygg1TEWWzGIFLtro= +github.com/rogpeppe/go-internal v1.10.0 h1:TMyTOH3F/DB16zRVcYyreMH6GnZZrwQVAoYjRBZyWFQ= +github.com/rogpeppe/go-internal v1.10.0/go.mod h1:UQnix2H7Ngw/k4C5ijL5+65zddjncjaFoBhdsK/akog= +github.com/ruudk/golang-pdf417 v0.0.0-20181029194003-1af4ab5afa58/go.mod h1:6lfFZQK844Gfx8o5WFuvpxWRwnSoipWe/p622j1v06w= +github.com/samber/lo v1.50.0 h1:XrG0xOeHs+4FQ8gJR97zDz5uOFMW7OwFWiFVzqopKgY= +github.com/samber/lo v1.50.0/go.mod h1:RjZyNk6WSnUFRKK6EyOhsRJMqft3G+pg7dCWHQCWvsc= +github.com/shopspring/decimal v1.4.0 h1:bxl37RwXBklmTi0C79JfXCEBD1cqqHt0bbgBAGFp81k= +github.com/shopspring/decimal v1.4.0/go.mod h1:gawqmDU56v4yIKSwfBSFip1HdCCXN8/+DMd9qYNcwME= +github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e h1:MRM5ITcdelLK2j1vwZ3Je0FKVCfqOLp5zO6trqMLYs0= +github.com/skip2/go-qrcode v0.0.0-20200617195104-da1b6568686e/go.mod h1:XV66xRDqSt+GTGFMVlhk3ULuV0y9ZmzeVGR4mloJI3M= +github.com/smartwalle/alipay/v3 v3.2.23 h1:i1VwJeu70EmwpsXXz6GZZnMAtRx5MTfn2dPoql/L3zE= +github.com/smartwalle/alipay/v3 v3.2.23/go.mod h1:lVqFiupPf8YsAXaq5JXcwqnOUC2MCF+2/5vub+RlagE= +github.com/smartwalle/ncrypto v1.0.4 h1:P2rqQxDepJwgeO5ShoC+wGcK2wNJDmcdBOWAksuIgx8= +github.com/smartwalle/ncrypto v1.0.4/go.mod h1:Dwlp6sfeNaPMnOxMNayMTacvC5JGEVln3CVdiVDgbBk= +github.com/smartwalle/ngx v1.0.9 h1:pUXDvWRZJIHVrCKA1uZ15YwNti+5P4GuJGbpJ4WvpMw= +github.com/smartwalle/ngx v1.0.9/go.mod h1:mx/nz2Pk5j+RBs7t6u6k22MPiBG/8CtOMpCnALIG8Y0= +github.com/smartwalle/nsign v1.0.9 h1:8poAgG7zBd8HkZy9RQDwasC6XZvJpDGQWSjzL2FZL6E= +github.com/smartwalle/nsign v1.0.9/go.mod h1:eY6I4CJlyNdVMP+t6z1H6Jpd4m5/V+8xi44ufSTxXgc= +github.com/smartystreets/assertions v0.0.0-20180927180507-b2de0cb4f26d/go.mod h1:OnSkiWE9lh6wB0YB77sQom3nweQdgAjqCqsofrRNTgc= +github.com/smartystreets/assertions v1.1.0/go.mod h1:tcbTF8ujkAEcZ8TElKY+i30BzYlVhC/LOxJk7iOWnoo= +github.com/smartystreets/goconvey v1.6.4/go.mod h1:syvi0/a8iFYH4r/RixwvyeAJjdLS9QV7WQ/tjFTllLA= +github.com/sony/sonyflake v1.2.0 h1:Pfr3A+ejSg+0SPqpoAmQgEtNDAhc2G1SUYk205qVMLQ= +github.com/sony/sonyflake v1.2.0/go.mod h1:LORtCywH/cq10ZbyfhKrHYgAUGH7mOBa76enV9txy/Y= +github.com/spaolacci/murmur3 v1.1.0 h1:7c1g84S4BPRrfL5Xrdp6fOJ206sU9y293DDHaoy0bLI= +github.com/spaolacci/murmur3 v1.1.0/go.mod h1:JwIasOWyU6f++ZhiEuf87xNszmSA2myDM2Kzu9HwQUA= +github.com/spf13/cast v1.7.0 h1:ntdiHjuueXFgm5nzDRdOS4yfT43P5Fnud6DH50rz/7w= +github.com/spf13/cast v1.7.0/go.mod h1:ancEpBxwJDODSW/UG4rDrAqiKolqNNh2DX3mk86cAdo= +github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME= +github.com/stretchr/objx v0.2.0/go.mod h1:qt09Ya8vawLte6SNmTgCsAVtYtaKzEcn8ATUoHMkEqE= +github.com/stretchr/objx v0.4.0/go.mod h1:YvHI0jy2hoMjB+UWwv71VJQ9isScKT/TqJzVSSt89Yw= +github.com/stretchr/objx v0.5.0/go.mod h1:Yh+to48EsGEfYuaHDzXPcE3xhTkx73EhmCGUpEOglKo= +github.com/stretchr/objx v0.5.2 h1:xuMeJ0Sdp5ZMRXx/aWO6RZxdr3beISkG5/G/aIRr3pY= +github.com/stretchr/objx v0.5.2/go.mod h1:FRsXN1f5AsAjCGJKqEizvkpNtU+EGNCLh3NxZ/8L+MA= +github.com/stretchr/testify v1.2.2/go.mod h1:a8OnRcib4nhh0OaRAV+Yts87kKdq0PP7pXfy6kDkUVs= +github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI= +github.com/stretchr/testify v1.5.1/go.mod h1:5W2xD1RspED5o8YsWQXVCued0rvSQ+mT+I5cxcmMvtA= +github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.7.1/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg= +github.com/stretchr/testify v1.8.0/go.mod h1:yNjHg4UonilssWZ8iaSj1OCr/vHnekPRkoO+kdMU+MU= +github.com/stretchr/testify v1.8.1/go.mod h1:w2LPCIKwWwSfY2zedu0+kehJoqGctiVI29o6fzry7u4= +github.com/stretchr/testify v1.8.4/go.mod h1:sz/lmYIOXD/1dqDmKjjqLyZ2RngseejIcXlSw2iwfAo= +github.com/stretchr/testify v1.9.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= +github.com/stretchr/testify v1.10.0 h1:Xv5erBjTwe/5IxqUQTdXv5kgmIvbHo3QQyRwhJsOfJA= +github.com/stretchr/testify v1.10.0/go.mod h1:r2ic/lqez/lEtzL7wO/rwa5dbSLXVDPFyf8C91i36aY= +github.com/tidwall/gjson v1.18.0 h1:FIDeeyB800efLX89e5a8Y0BNH+LOngJyGrIWxG2FKQY= +github.com/tidwall/gjson v1.18.0/go.mod h1:/wbyibRr2FHMks5tjHJ5F8dMZh3AcwJEMf5vlfC0lxk= +github.com/tidwall/match v1.1.1 h1:+Ho715JplO36QYgwN9PGYNhgZvoUSc9X2c80KVTi+GA= +github.com/tidwall/match v1.1.1/go.mod h1:eRSPERbgtNPcGhD8UCthc6PmLEQXEWd3PRB5JTxsfmM= +github.com/tidwall/pretty v1.2.0/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU= +github.com/tidwall/pretty v1.2.1 h1:qjsOFOWWQl+N3RsoF5/ssm1pHmJJwhjlSbZ51I6wMl4= +github.com/tidwall/pretty v1.2.1/go.mod h1:ITEVvHYasfjBbM0u2Pg8T2nJnzm8xPwvNhhsoaGGjNU= +github.com/tjfoc/gmsm v1.3.2/go.mod h1:HaUcFuY0auTiaHB9MHFGCPx5IaLhTUd2atbCFBQXn9w= +github.com/tjfoc/gmsm v1.4.1 h1:aMe1GlZb+0bLjn+cKTPEvvn9oUEBlJitaZiiBwsbgho= +github.com/tjfoc/gmsm v1.4.1/go.mod h1:j4INPkHWMrhJb38G+J6W4Tw0AbuN8Thu3PbdVYhVcTE= +github.com/twitchyliquid64/golang-asm v0.15.1 h1:SU5vSMR7hnwNxj24w34ZyCi/FmDZTkS4MhqMhdFk5YI= +github.com/twitchyliquid64/golang-asm v0.15.1/go.mod h1:a1lVb/DtPvCB8fslRZhAngC2+aY1QWCk3Cedj/Gdt08= +github.com/wechatpay-apiv3/wechatpay-go v0.2.20 h1:gS8oFn1bHGnyapR2Zb4aqTV6l4kJWgbtqjCq6k1L9DQ= +github.com/wechatpay-apiv3/wechatpay-go v0.2.20/go.mod h1:A254AUBVB6R+EqQFo3yTgeh7HtyqRRtN2w9hQSOrd4Q= +github.com/yuin/goldmark v1.1.27/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +github.com/yuin/goldmark v1.1.30/go.mod h1:3hX8gzYuyVAZsxl0MRgGTJEmQBFcNTphYh9decYSb74= +github.com/yuin/goldmark v1.4.13/go.mod h1:6yULJ656Px+3vBD8DxQVa3kxgyrAnzto9xy5taEt/CY= +github.com/yuin/gopher-lua v1.1.1 h1:kYKnWBjvbNP4XLT3+bPEwAXJx262OhaHDWDVOPjL46M= +github.com/yuin/gopher-lua v1.1.1/go.mod h1:GBR0iDaNXjAgGg9zfCvksxSRnQx76gclCIb7kdAd1Pw= +github.com/zeromicro/go-zero v1.7.3 h1:yDUQF2DXDhUHc77/NZF6mzsoRPMBfldjPmG2O/ZSzss= +github.com/zeromicro/go-zero v1.7.3/go.mod h1:9JIW3gHBGuc9LzvjZnNwINIq9QdiKu3AigajLtkJamQ= +go.opentelemetry.io/otel v1.24.0 h1:0LAOdjNmQeSTzGBzduGe/rU4tZhMwL5rWgtp9Ku5Jfo= +go.opentelemetry.io/otel v1.24.0/go.mod h1:W7b9Ozg4nkF5tWI5zsXkaKKDjdVjpD4oAt9Qi/MArHo= +go.opentelemetry.io/otel/exporters/jaeger v1.17.0 h1:D7UpUy2Xc2wsi1Ras6V40q806WM07rqoCWzXu7Sqy+4= +go.opentelemetry.io/otel/exporters/jaeger v1.17.0/go.mod h1:nPCqOnEH9rNLKqH/+rrUjiMzHJdV1BlpKcTwRTyKkKI= +go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.24.0 h1:t6wl9SPayj+c7lEIFgm4ooDBZVb01IhLB4InpomhRw8= +go.opentelemetry.io/otel/exporters/otlp/otlptrace v1.24.0/go.mod h1:iSDOcsnSA5INXzZtwaBPrKp/lWu/V14Dd+llD0oI2EA= +go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc v1.24.0 h1:Mw5xcxMwlqoJd97vwPxA8isEaIoxsta9/Q51+TTJLGE= +go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracegrpc v1.24.0/go.mod h1:CQNu9bj7o7mC6U7+CA/schKEYakYXWr79ucDHTMGhCM= +go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.24.0 h1:Xw8U6u2f8DK2XAkGRFV7BBLENgnTGX9i4rQRxJf+/vs= +go.opentelemetry.io/otel/exporters/otlp/otlptrace/otlptracehttp v1.24.0/go.mod h1:6KW1Fm6R/s6Z3PGXwSJN2K4eT6wQB3vXX6CVnYX9NmM= +go.opentelemetry.io/otel/exporters/stdout/stdouttrace v1.24.0 h1:s0PHtIkN+3xrbDOpt2M8OTG92cWqUESvzh2MxiR5xY8= +go.opentelemetry.io/otel/exporters/stdout/stdouttrace v1.24.0/go.mod h1:hZlFbDbRt++MMPCCfSJfmhkGIWnX1h3XjkfxZUjLrIA= +go.opentelemetry.io/otel/exporters/zipkin v1.24.0 h1:3evrL5poBuh1KF51D9gO/S+N/1msnm4DaBqs/rpXUqY= +go.opentelemetry.io/otel/exporters/zipkin v1.24.0/go.mod h1:0EHgD8R0+8yRhUYJOGR8Hfg2dpiJQxDOszd5smVO9wM= +go.opentelemetry.io/otel/metric v1.24.0 h1:6EhoGWWK28x1fbpA4tYTOWBkPefTDQnb8WSGXlc88kI= +go.opentelemetry.io/otel/metric v1.24.0/go.mod h1:VYhLe1rFfxuTXLgj4CBiyz+9WYBA8pNGJgDcSFRKBco= +go.opentelemetry.io/otel/sdk v1.24.0 h1:YMPPDNymmQN3ZgczicBY3B6sf9n62Dlj9pWD3ucgoDw= +go.opentelemetry.io/otel/sdk v1.24.0/go.mod h1:KVrIYw6tEubO9E96HQpcmpTKDVn9gdv35HoYiQWGDFg= +go.opentelemetry.io/otel/trace v1.24.0 h1:CsKnnL4dUAr/0llH9FKuc698G04IrpWV0MQA/Y1YELI= +go.opentelemetry.io/otel/trace v1.24.0/go.mod h1:HPc3Xr/cOApsBI154IU0OI0HJexz+aw5uPdbs3UCjNU= +go.opentelemetry.io/proto/otlp v1.3.1 h1:TrMUixzpM0yuc/znrFTP9MMRh8trP93mkCiDVeXrui0= +go.opentelemetry.io/proto/otlp v1.3.1/go.mod h1:0X1WI4de4ZsLrrJNLAQbFeLCm3T7yBkR0XqQ7niQU+8= +go.uber.org/automaxprocs v1.6.0 h1:O3y2/QNTOdbF+e/dpXNNW7Rx2hZ4sTIPyybbxyNqTUs= +go.uber.org/automaxprocs v1.6.0/go.mod h1:ifeIMSnPZuznNm6jmdzmU3/bfk01Fe2fotchwEFJ8r8= +go.uber.org/goleak v1.3.0 h1:2K3zAYmnTNqV73imy9J1T3WC+gmCePx2hEGkimedGto= +go.uber.org/goleak v1.3.0/go.mod h1:CoHD4mav9JJNrW/WLlf7HGZPjdw8EucARQHekz1X6bE= +golang.org/x/arch v0.0.0-20210923205945-b76863e36670 h1:18EFjUmQOcUvxNYSkA6jO9VAiXCnxFY6NyDX0bHDmkU= +golang.org/x/arch v0.0.0-20210923205945-b76863e36670/go.mod h1:5om86z9Hs0C8fWVUuoMHwpExlXzs5Tkyp9hOrfG7pp8= +golang.org/x/crypto v0.0.0-20190308221718-c2843e01d9a2/go.mod h1:djNgcEr1/C05ACkg1iLfiJU5Ep61QUkGW8qpdssI0+w= +golang.org/x/crypto v0.0.0-20191011191535-87dc89f01550/go.mod h1:yigFU9vqHzYiE8UmvKecakEJjdnWj3jj499lnFckfCI= +golang.org/x/crypto v0.0.0-20191219195013-becbf705a915/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/crypto v0.0.0-20200510223506-06a226fb4e37/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/crypto v0.0.0-20200622213623-75b288015ac9/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/crypto v0.0.0-20201012173705-84dcc777aaee/go.mod h1:LzIPMQfyMNhhGPhUkYOs5KpL4U8rLKemX1yGLhDgUto= +golang.org/x/crypto v0.0.0-20210921155107-089bfa567519/go.mod h1:GvvjBRRGRdwPK5ydBHafDWAxML/pGHZbMvKqRZ5+Abc= +golang.org/x/crypto v0.14.0/go.mod h1:MVFd36DqK4CsrnJYDkBA3VC4m2GkXAM0PvzMCn4JQf4= +golang.org/x/crypto v0.18.0/go.mod h1:R0j02AL6hcrfOiy9T4ZYp/rcWeMxM3L6QYxlOuEG1mg= +golang.org/x/crypto v0.19.0/go.mod h1:Iy9bg/ha4yyC70EfRS8jz+B6ybOBKMaSxLj6P6oBDfU= +golang.org/x/crypto v0.21.0/go.mod h1:0BP7YvVV9gBbVKyeTG0Gyn+gZm94bibOW5BjDEYAOMs= +golang.org/x/crypto v0.28.0 h1:GBDwsMXVQi34v5CCYUm2jkJvu4cbtru2U4TN2PSyQnw= +golang.org/x/crypto v0.28.0/go.mod h1:rmgy+3RHxRZMyY0jjAJShp2zgEdOqj2AO7U0pYmeQ7U= +golang.org/x/exp v0.0.0-20190121172915-509febef88a4/go.mod h1:CJ0aWSM057203Lf6IL+f9T1iT9GByDxfZKAQTCR3kQA= +golang.org/x/image v0.0.0-20190910094157-69e4b8554b2a/go.mod h1:FeLwcggjj3mMvU+oOTbSwawSJRM1uh48EjtB4UJZlP0= +golang.org/x/image v0.28.0 h1:gdem5JW1OLS4FbkWgLO+7ZeFzYtL3xClb97GaUzYMFE= +golang.org/x/image v0.28.0/go.mod h1:GUJYXtnGKEUgggyzh+Vxt+AviiCcyiwpsl8iQ8MvwGY= +golang.org/x/lint v0.0.0-20181026193005-c67002cb31c3/go.mod h1:UVdnD1Gm6xHRNCYTkRU2/jEulfH38KcIWyp/GAMgvoE= +golang.org/x/lint v0.0.0-20190227174305-5b3e6a55c961/go.mod h1:wehouNa3lNwaWXcvxsM5YxQ5yQlVC4a0KAMCusXpPoU= +golang.org/x/lint v0.0.0-20190313153728-d0100b6bd8b3/go.mod h1:6SW0HCj/g11FgYtHlgUYUwCkIfeOF89ocIRzGO/8vkc= +golang.org/x/mod v0.2.0/go.mod h1:s0Qsj1ACt9ePp/hMypM3fl4fZqREWJwdYDEqhRiZZUA= +golang.org/x/mod v0.6.0-dev.0.20220419223038-86c51ed26bb4/go.mod h1:jJ57K6gSWd91VN4djpZkiMVwK6gcyfeH4XE8wZrZaV4= +golang.org/x/mod v0.8.0/go.mod h1:iBbtSCu2XBx23ZKBPSOrRkjjQPZFPuis4dIYUhu/chs= +golang.org/x/net v0.0.0-20180724234803-3673e40ba225/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20180826012351-8a410e7b638d/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190213061140-3a22650c66bd/go.mod h1:mL1N/T3taQHkDXs73rZJwtUhF3w3ftmwwsq0BUmARs4= +golang.org/x/net v0.0.0-20190311183353-d8887717615a/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190404232315-eb5bcb51f2a3/go.mod h1:t9HGtf8HONx5eT2rtn7q6eTqICYqUVnKs3thJo3Qplg= +golang.org/x/net v0.0.0-20190620200207-3b0461eec859/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20200226121028-0de0cce0169b/go.mod h1:z5CRVTTTmAJ677TzLLGU+0bjPO0LkuOLi4/5GtJWs/s= +golang.org/x/net v0.0.0-20200506145744-7e3656a0809f/go.mod h1:qpuaurCH72eLCgpAm/N6yyVIVM9cpaDIP3A8BGJEC5A= +golang.org/x/net v0.0.0-20201010224723-4f7140c49acb/go.mod h1:sp8m0HH+o8qH0wwXwYZr8TS3Oi6o0r6Gce1SSxlDquU= +golang.org/x/net v0.0.0-20210226172049-e18ecbb05110/go.mod h1:m0MpNAwzfU5UDzcl9v0D8zg8gWTRqZa9RBIspLL5mdg= +golang.org/x/net v0.0.0-20220722155237-a158d28d115b/go.mod h1:XRhObCWvk6IyKnWLug+ECip1KBveYUHfp+8e9klMJ9c= +golang.org/x/net v0.6.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= +golang.org/x/net v0.7.0/go.mod h1:2Tu9+aMcznHK/AK1HMvgo6xiTLG5rD5rZLDS+rp2Bjs= +golang.org/x/net v0.10.0/go.mod h1:0qNGK6F8kojg2nk9dLZ2mShWaEBan6FAoqfSigmmuDg= +golang.org/x/net v0.17.0/go.mod h1:NxSsAGuq816PNPmqtQdLE42eU2Fs7NoRIZrHJAlaCOE= +golang.org/x/net v0.20.0/go.mod h1:z8BVo6PvndSri0LbOE3hAn0apkU+1YvI6E70E9jsnvY= +golang.org/x/net v0.21.0/go.mod h1:bIjVDfnllIU7BJ2DNgfnXvpSvtn8VRwhlsaeUTyUS44= +golang.org/x/net v0.23.0/go.mod h1:JKghWKKOSdJwpW2GEx0Ja7fmaKnMsbu+MWVZTokSYmg= +golang.org/x/net v0.30.0 h1:AcW1SDZMkb8IpzCdQUaIq2sP4sZ4zw+55h6ynffypl4= +golang.org/x/net v0.30.0/go.mod h1:2wGyMJ5iFasEhkwi13ChkO/t1ECNC4X4eBKkVFyYFlU= +golang.org/x/oauth2 v0.0.0-20180821212333-d2e6202438be/go.mod h1:N/0e6XlmueqKjAGxoOufVs8QHGRruUQn6yWY3a++T0U= +golang.org/x/sync v0.0.0-20180314180146-1d60e4601c6f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20181108010431-42b317875d0f/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190423024810-112230192c58/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20190911185100-cd5d95a43a6e/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20200317015054-43a5402ce75a/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.0.0-20220722155255-886fb9371eb4/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sync v0.1.0/go.mod h1:RxMgew5VJxzue5/jJTE5uejpjVlOe/izrB70Jof72aM= +golang.org/x/sys v0.0.0-20180830151530-49385e6e1522/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190215142949-d0b11bdaac8a/go.mod h1:STP8DvDyc/dI5b8T5hshtkjS+E42TnysNCUPdjciGhY= +golang.org/x/sys v0.0.0-20190412213103-97732733099d/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200323222414-85ca7c5b95cd/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200509044756-6aff5f38e54f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20200930185726-fdedc70b468f/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20201119102817-f84b799fce68/go.mod h1:h1NjWce9XRLGQEsW7wpKNCjG9DtNlClVuFLEZdDNbEs= +golang.org/x/sys v0.0.0-20210615035016-665e8c7367d1/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220520151302-bc2c85ada10a/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220722155257-8c9f86f7a55f/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.5.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.8.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.13.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg= +golang.org/x/sys v0.16.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/sys v0.17.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/sys v0.18.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/sys v0.26.0 h1:KHjCJyddX0LoSTb3J+vWpupP9p0oznkqVk/IfjymZbo= +golang.org/x/sys v0.26.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA= +golang.org/x/term v0.0.0-20201126162022-7de9c90e9dd1/go.mod h1:bj7SfCRtBDWHUb9snDiAeCFNEtKQo2Wmx5Cou7ajbmo= +golang.org/x/term v0.0.0-20210927222741-03fcf44c2211/go.mod h1:jbD1KX2456YbFQfuXm/mYQcufACuNUgVhRMnK/tPxf8= +golang.org/x/term v0.5.0/go.mod h1:jMB1sMXY+tzblOD4FWmEbocvup2/aLOaQEp7JmGp78k= +golang.org/x/term v0.8.0/go.mod h1:xPskH00ivmX89bAKVGSKKtLOWNx2+17Eiy94tnKShWo= +golang.org/x/term v0.13.0/go.mod h1:LTmsnFJwVN6bCy1rVCoS+qHT1HhALEFxKncY3WNNh4U= +golang.org/x/term v0.16.0/go.mod h1:yn7UURbUtPyrVJPGPq404EukNFxcm/foM+bV/bfcDsY= +golang.org/x/term v0.17.0/go.mod h1:lLRBjIVuehSbZlaOtGMbcMncT+aqLLLmKrsjNrUguwk= +golang.org/x/term v0.18.0/go.mod h1:ILwASektA3OnRv7amZ1xhE/KTR+u50pbXfZ03+6Nx58= +golang.org/x/text v0.3.0/go.mod h1:NqM8EUOU14njkJ3fqMW+pc6Ldnwhi/IjpwHt7yyuwOQ= +golang.org/x/text v0.3.2/go.mod h1:bEr9sfX3Q8Zfm5fL9x+3itogRgK3+ptLWKqgva+5dAk= +golang.org/x/text v0.3.3/go.mod h1:5Zoc/QRtKVWzQhOtBMvqHzDpF6irO9z98xDceosuGiQ= +golang.org/x/text v0.3.7/go.mod h1:u+2+/6zg+i71rQMx5EYifcz6MCKuco9NR6JIITiCfzQ= +golang.org/x/text v0.7.0/go.mod h1:mrYo+phRRbMaCq/xk9113O4dZlRixOauAjOtrjsXDZ8= +golang.org/x/text v0.9.0/go.mod h1:e1OnstbJyHTd6l/uOt8jFFHp6TRDWZR/bV3emEE/zU8= +golang.org/x/text v0.13.0/go.mod h1:TvPlkZtksWOMsz7fbANvkp4WM8x/WCo/om8BMLbz+aE= +golang.org/x/text v0.14.0/go.mod h1:18ZOQIKpY8NJVqYksKHtTdi31H5itFRjB5/qKTNYzSU= +golang.org/x/text v0.26.0 h1:P42AVeLghgTYr4+xUnTRKDMqpar+PtX7KWuNQL21L8M= +golang.org/x/text v0.26.0/go.mod h1:QK15LZJUUQVJxhz7wXgxSy/CJaTFjd0G+YLonydOVQA= +golang.org/x/time v0.7.0 h1:ntUhktv3OPE6TgYxXWv9vKvUSJyIFJlyohwbkEwPrKQ= +golang.org/x/time v0.7.0/go.mod h1:3BpzKBy/shNhVucY/MWOyx10tF3SFh9QdLuxbVysPQM= +golang.org/x/tools v0.0.0-20180917221912-90fa682c2a6e/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190114222345-bf090417da8b/go.mod h1:n7NCudcB/nEzxVGmLbDWY5pfWTLqBcC2KZ6jyYvM4mQ= +golang.org/x/tools v0.0.0-20190226205152-f727befe758c/go.mod h1:9Yl7xja0Znq3iFh3HoIrodX9oNMXvdceNzlUR8zjMvY= +golang.org/x/tools v0.0.0-20190311212946-11955173bddd/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190328211700-ab21143f2384/go.mod h1:LCzVGOaR6xXOjkQ3onu1FJEFr0SW1gC7cKk1uF8kGRs= +golang.org/x/tools v0.0.0-20190524140312-2c0ae7006135/go.mod h1:RgjU9mgBXZiqYHBnxXauZ1Gv1EHHAz9KjViQ78xBX0Q= +golang.org/x/tools v0.0.0-20191119224855-298f0cb1881e/go.mod h1:b+2E5dAYhXwXZwtnZ6UAqBI28+e2cm9otk0dWdXHAEo= +golang.org/x/tools v0.0.0-20200509030707-2212a7e161a5/go.mod h1:EkVYQZoAsY45+roYkvgYkIh4xh/qjgUK9TdY2XT94GE= +golang.org/x/tools v0.1.12/go.mod h1:hNGJHUnrk76NpqgfD5Aqm5Crs+Hm0VOH/i9J2+nxYbc= +golang.org/x/tools v0.6.0/go.mod h1:Xwgl3UAJ/d3gWutnCtw505GrjyAbvKui8lOU390QaIU= +golang.org/x/xerrors v0.0.0-20190717185122-a985d3407aa7/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191011141410-1b5146add898/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +golang.org/x/xerrors v0.0.0-20191204190536-9bdfabe68543/go.mod h1:I/5z698sn9Ka8TeJc9MKroUUfqBBauWjQqLJ2OPfmY0= +google.golang.org/appengine v1.1.0/go.mod h1:EbEs0AVv82hx2wNQdGPgUI5lhzA/G0D9YwlJXL52JkM= +google.golang.org/appengine v1.4.0/go.mod h1:xpcJRLb0r/rnEns0DIKYYv+WjYCduHsrkT7/EB5XEv4= +google.golang.org/genproto v0.0.0-20180817151627-c66870c02cf8/go.mod h1:JiN7NxoALGmiZfu7CAH4rXhgtRTLTxftemlI0sWmxmc= +google.golang.org/genproto v0.0.0-20190819201941-24fa4b261c55/go.mod h1:DMBHOl98Agz4BDEuKkezgsaosCRResVns1a3J2ZsMNc= +google.golang.org/genproto/googleapis/api v0.0.0-20240814211410-ddb44dafa142 h1:wKguEg1hsxI2/L3hUYrpo1RVi48K+uTyzKqprwLXsb8= +google.golang.org/genproto/googleapis/api v0.0.0-20240814211410-ddb44dafa142/go.mod h1:d6be+8HhtEtucleCbxpPW9PA9XwISACu8nvpPqF0BVo= +google.golang.org/genproto/googleapis/rpc v0.0.0-20240814211410-ddb44dafa142 h1:e7S5W7MGGLaSu8j3YjdezkZ+m1/Nm0uRVRMEMGk26Xs= +google.golang.org/genproto/googleapis/rpc v0.0.0-20240814211410-ddb44dafa142/go.mod h1:UqMtugtsSgubUsoxbuAoiCXvqvErP7Gf0so0mK9tHxU= +google.golang.org/grpc v1.19.0/go.mod h1:mqu4LbDTu4XGKhr4mRzUsmM4RtVoemTSY81AxZiDr8c= +google.golang.org/grpc v1.23.0/go.mod h1:Y5yQAOtifL1yxbo5wqy6BxZv8vAUGQwXBOALyacEbxg= +google.golang.org/grpc v1.25.1/go.mod h1:c3i+UQWmh7LiEpx4sFZnkU36qjEYZ0imhYfXVyQciAY= +google.golang.org/grpc v1.31.0/go.mod h1:N36X2cJ7JwdamYAgDz+s+rVMFjt3numwzf/HckM8pak= +google.golang.org/grpc v1.67.1 h1:zWnc1Vrcno+lHZCOofnIMvycFcc0QRGIzm9dhnDX68E= +google.golang.org/grpc v1.67.1/go.mod h1:1gLDyUQU7CTLJI90u3nXZ9ekeghjeM7pTDZlqFNg2AA= +google.golang.org/protobuf v0.0.0-20200109180630-ec00e32a8dfd/go.mod h1:DFci5gLYBciE7Vtevhsrf46CRTquxDuWsQurQQe4oz8= +google.golang.org/protobuf v0.0.0-20200221191635-4d8936d0db64/go.mod h1:kwYJMbMJ01Woi6D6+Kah6886xMZcty6N08ah7+eCXa0= +google.golang.org/protobuf v0.0.0-20200228230310-ab0ca4ff8a60/go.mod h1:cfTl7dwQJ+fmap5saPgwCLgHXTUD7jkjRqWcaiX5VyM= +google.golang.org/protobuf v1.20.1-0.20200309200217-e05f789c0967/go.mod h1:A+miEFZTKqfCUM6K7xSMQL9OKL/b6hQv+e19PK+JZNE= +google.golang.org/protobuf v1.21.0/go.mod h1:47Nbq4nVaFHyn7ilMalzfO3qCViNmqZ2kzikPIcrTAo= +google.golang.org/protobuf v1.23.0/go.mod h1:EGpADcykh3NcUnDUJcl1+ZksZNG86OlYog2l/sGQquU= +google.golang.org/protobuf v1.35.1 h1:m3LfL6/Ca+fqnjnlqQXNpFPABW1UD7mjh8KO2mKFytA= +google.golang.org/protobuf v1.35.1/go.mod h1:9fA7Ob0pmnwhb644+1+CVWFRbNajQ6iRojtC/QF5bRE= +gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0= +gopkg.in/check.v1 v1.0.0-20200227125254-8fa46927fb4f/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= +gopkg.in/cheggaaa/pb.v1 v1.0.28 h1:n1tBJnnK2r7g9OW2btFH91V92STTUevLXYFb8gy9EMk= +gopkg.in/cheggaaa/pb.v1 v1.0.28/go.mod h1:V/YB90LKu/1FcN3WVnfiiE5oMCibMjukxqG/qStrOgw= +gopkg.in/h2non/gock.v1 v1.1.2 h1:jBbHXgGBK/AoPVfJh5x4r/WxIrElvbLel8TCZkkZJoY= +gopkg.in/h2non/gock.v1 v1.1.2/go.mod h1:n7UGz/ckNChHiK05rDoiC4MYSunEC/lyaUm2WWaDva0= +gopkg.in/ini.v1 v1.56.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k= +gopkg.in/ini.v1 v1.67.0 h1:Dgnx+6+nfE+IfzjUEISNeydPJh9AXNNsWbGP9KzCsOA= +gopkg.in/ini.v1 v1.67.0/go.mod h1:pNLf8WUiyNEtQjuu5G5vTm06TEv9tsIgeAvK8hOrP4k= +gopkg.in/yaml.v2 v2.2.2/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.2.8/go.mod h1:hI93XBmqTisBFMUTm0b8Fm+jr3Dg1NNxqwp+5A1VGuI= +gopkg.in/yaml.v2 v2.4.0 h1:D8xgwECY7CYvx+Y2n4sBz93Jn9JRvxdiyyo8CTfuKaY= +gopkg.in/yaml.v2 v2.4.0/go.mod h1:RDklbk79AGWmwhnvt/jBztapEOGDOx6ZbXqjP6csGnQ= +gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +gopkg.in/yaml.v3 v3.0.1 h1:fxVm/GzAzEWqLHuvctI91KS9hhNmmWOoWu0XTYJS7CA= +gopkg.in/yaml.v3 v3.0.1/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM= +honnef.co/go/tools v0.0.0-20190102054323-c2f93a96b099/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= +honnef.co/go/tools v0.0.0-20190523083050-ea95bdfd59fc/go.mod h1:rf3lG4BRIbNafJWhAfAdb/ePZxsR/4RtNHQocxwk9r4= +k8s.io/utils v0.0.0-20240711033017-18e509b52bc8 h1:pUdcCO1Lk/tbT5ztQWOBi5HBgbBP1J8+AsQnQCKsi8A= +k8s.io/utils v0.0.0-20240711033017-18e509b52bc8/go.mod h1:OLgZIPagt7ERELqWJFomSt595RzquPNLL48iOWgYOg0= +nullprogram.com/x/optparse v1.0.0/go.mod h1:KdyPE+Igbe0jQUrVfMqDMeJQIJZEuyV7pjYmp6pbG50= diff --git a/pkg/lzkit/crypto/README.md b/pkg/lzkit/crypto/README.md new file mode 100644 index 0000000..863719d --- /dev/null +++ b/pkg/lzkit/crypto/README.md @@ -0,0 +1,235 @@ +# AES 加密工具包 + +本包提供了多种加密方式，特别是用于处理敏感个人信息(如手机号、身份证号等)的加密和解密功能。 + +## 主要功能 + +- **AES-CBC 模式加密/解密** - 标准加密模式，适用于一般数据加密 +- **AES-ECB 模式加密/解密** - 确定性加密模式，适用于数据库字段加密和查询 +- **专门针对个人敏感信息的加密/解密方法** +- **密钥生成和管理工具** + +## 安全性说明 + +- **AES-CBC 模式**：使用随机 IV，相同明文每次加密结果不同，安全性较高 +- **AES-ECB 模式**：确定性加密，相同明文每次加密结果相同，便于数据库查询，但安全性较低 + +> **⚠️ 警告**：ECB 模式仅适用于短文本(如手机号、身份证号)的确定性加密，不建议用于加密大段文本或高安全需求场景。 + +## 使用示例 + +### 1. 加密手机号 + +使用 AES-ECB 模式加密手机号，保证确定性(相同手机号总是产生相同密文)： + +```go +import ( + "fmt" + "qnc-server/pkg/lzkit/crypto" +) + +func encryptMobileExample() { + // 您的密钥(需安全保存，建议存储在配置中) + key := []byte("1234567890abcdef") // 16字节AES-128密钥 + + // 加密手机号 + mobile := "13800138000" + encryptedMobile, err := crypto.EncryptMobile(mobile, key) + if err != nil { + panic(err) + } + + fmt.Println("加密后的手机号:", encryptedMobile) + + // 解密手机号 + decryptedMobile, err := crypto.DecryptMobile(encryptedMobile, key) + if err != nil { + panic(err) + } + + fmt.Println("解密后的手机号:", decryptedMobile) +} +``` + +### 2. 在数据库中存储和查询加密手机号 + +```go +// 加密并存储手机号 +func saveUser(db *sqlx.DB, mobile string, key []byte) (int64, error) { + encryptedMobile, err := crypto.EncryptMobile(mobile, key) + if err != nil { + return 0, err + } + + var id int64 + err = db.QueryRow( + "INSERT INTO users (mobile, create_time) VALUES (?, NOW()) RETURNING id", + encryptedMobile, + ).Scan(&id) + + return id, err +} + +// 根据手机号查询用户 +func findUserByMobile(db *sqlx.DB, mobile string, key []byte) (*User, error) { + encryptedMobile, err := crypto.EncryptMobile(mobile, key) + if err != nil { + return nil, err + } + + var user User + err = db.QueryRow( + "SELECT id, mobile, create_time FROM users WHERE mobile = ?", + encryptedMobile, + ).Scan(&user.ID, &user.EncryptedMobile, &user.CreateTime) + + if err != nil { + return nil, err + } + + // 解密手机号用于显示 + user.Mobile, _ = crypto.DecryptMobile(user.EncryptedMobile, key) + + return &user, nil +} +``` + +### 3. 加密身份证号 + +```go +func encryptIDCardExample() { + key := []byte("1234567890abcdef") + + idCard := "440101199001011234" + encryptedIDCard, err := crypto.EncryptIDCard(idCard, key) + if err != nil { + panic(err) + } + + fmt.Println("加密后的身份证号:", encryptedIDCard) + + // 解密身份证号 + decryptedIDCard, err := crypto.DecryptIDCard(encryptedIDCard, key) + if err != nil { + panic(err) + } + + fmt.Println("解密后的身份证号:", decryptedIDCard) +} +``` + +### 4. 密钥管理 + +```go +func keyManagementExample() { + // 生成随机密钥 + key, err := crypto.GenerateAESKey(16) // AES-128 + if err != nil { + panic(err) + } + fmt.Printf("生成的密钥(十六进制): %x\n", key) + + // 从密码派生密钥(便于记忆) + password := "my-secure-password" + derivedKey, err := crypto.DeriveKeyFromPassword(password, 16) + if err != nil { + panic(err) + } + fmt.Printf("从密码派生的密钥: %x\n", derivedKey) +} +``` + +### 5. 使用十六进制输出(适用于 URL 参数) + +```go +func hexEncodingExample() { + key := []byte("1234567890abcdef") + mobile := "13800138000" + + // 使用十六进制编码(适合URL参数) + encryptedHex, err := crypto.EncryptMobileHex(mobile, key) + if err != nil { + panic(err) + } + + fmt.Println("十六进制编码的加密手机号:", encryptedHex) + + // 解密十六进制编码的手机号 + decryptedMobile, err := crypto.DecryptMobileHex(encryptedHex, key) + if err != nil { + panic(err) + } + + fmt.Println("解密后的手机号:", decryptedMobile) +} +``` + +## 在 Go-Zero 项目中使用 + +在 Go-Zero 项目中，建议将加密密钥放在配置文件中： + +1. 在配置文件中添加密钥配置： + +```yaml +# etc/main.yaml +Name: main-api +Host: 0.0.0.0 +Port: 8888 + +Encrypt: + MobileKey: "1234567890abcdef" # 16字节AES-128密钥 + IDCardKey: "1234567890abcdef1234567890abcdef" # 32字节AES-256密钥 +``` + +2. 在配置结构中定义： + +```go +type Config struct { + rest.RestConf + Encrypt struct { + MobileKey string + IDCardKey string + } +} +``` + +3. 在服务上下文中使用： + +```go +type ServiceContext struct { + Config config.Config + UserModel model.UserModel + MobileKey []byte + IDCardKey []byte +} + +func NewServiceContext(c config.Config) *ServiceContext { + return &ServiceContext{ + Config: c, + UserModel: model.NewUserModel(sqlx.NewMysql(c.DB.DataSource), c.Cache), + MobileKey: []byte(c.Encrypt.MobileKey), + IDCardKey: []byte(c.Encrypt.IDCardKey), + } +} +``` + +4. 在 Logic 中使用： + +```go +func (l *RegisterLogic) Register(req *types.RegisterReq) (*types.RegisterResp, error) { + // 加密手机号用于存储 + encryptedMobile, err := crypto.EncryptMobile(req.Mobile, l.svcCtx.MobileKey) + if err != nil { + return nil, errors.New("手机号加密失败") + } + + // 保存到数据库 + user := &model.User{ + Mobile: encryptedMobile, + // 其他字段... + } + + result, err := l.svcCtx.UserModel.Insert(l.ctx, nil, user) + // 其余逻辑... +} +``` diff --git a/pkg/lzkit/crypto/README_bcrypt_test.md b/pkg/lzkit/crypto/README_bcrypt_test.md new file mode 100644 index 0000000..841fde9 --- /dev/null +++ b/pkg/lzkit/crypto/README_bcrypt_test.md @@ -0,0 +1,100 @@ +# bcrypt 密码加密测试文档 + +## 功能概述 + +为 `pkg/lzkit/crypto/bcrypt.go` 中的 `PasswordHash` 和 `PasswordVerify` 函数创建了完整的测试套件。 + +## 测试文件 + +- **文件位置**: `pkg/lzkit/crypto/bcrypt_test.go` +- **测试函数**: + - `TestPasswordHash` - 测试密码加密功能 + - `TestPasswordVerify` - 测试密码验证功能 + - `TestGeneratePasswords` - 生成常用密码的hash值 + - `BenchmarkPasswordHash` - 性能测试 + - `BenchmarkPasswordVerify` - 验证性能测试 + +## 测试覆盖 + +### 1. 密码加密测试 (`TestPasswordHash`) +- ✅ 默认cost加密 (cost=10) +- ✅ 自定义cost加密 (cost=12) +- ✅ 空密码处理 +- ✅ 复杂密码处理 +- ✅ 密码验证功能 +- ✅ 错误密码验证 + +### 2. 密码验证测试 (`TestPasswordVerify`) +- ✅ 正确密码验证 +- ✅ 错误密码验证 + +### 3. 密码生成测试 (`TestGeneratePasswords`) +生成10个常用密码的hash值，包括： +- `123456` +- `admin123` +- `password` +- `root` +- `test123` +- `MyP@ssw0rd!2024` +- `admin@123` +- `123456789` +- `qwerty` +- `abc123` + +## 性能测试结果 + +### Cost=10 性能 +- **执行时间**: ~41.7ms/op +- **内存分配**: 5314 B/op, 11 allocs/op + +### Cost=12 性能 +- **执行时间**: ~164.1ms/op +- **内存分配**: 5691 B/op, 12 allocs/op + +## 使用方法 + +### 运行所有测试 +```bash +cd pkg/lzkit/crypto +go test -v +``` + +### 运行密码相关测试 +```bash +go test -run "TestPassword|TestGenerate" -v +``` + +### 运行性能测试 +```bash +go test -bench=BenchmarkPassword -benchmem -run="^$" +``` + +### 生成密码hash +```bash +go test -run TestGeneratePasswords -v +``` + +## 示例输出 + +``` +=== 生成密码Hash值 === +1. 密码: 123456 Hash: $2a$10$AXcpNL9y5RYLiObTLFq4KOWKtlV3jEUuCd6fuzmSW2yYsSELJ23D. +2. 密码: admin123 Hash: $2a$10$5PUD/kpFGJ.09Gi.VGzu2.sCp9ZEshEcCaP4tKPNMgbvOaY8Hq7Sy +3. 密码: password Hash: $2a$10$Tjl5JY13eyGE4tPUdbco0OToz2iN6UY3Dm/QTYUpZx3b5QAPH4Aq6 +... +=== 密码生成完成 === +``` + +## 注意事项 + +1. **Cost参数**: 默认使用cost=10，可根据安全需求调整 +2. **性能考虑**: Cost越高越安全，但性能消耗越大 +3. **Hash唯一性**: 每次生成的hash都不同，但验证结果一致 +4. **安全性**: 使用bcrypt算法，适合生产环境使用 + +## 测试状态 + +✅ 所有bcrypt相关测试通过 +✅ 性能测试完成 +✅ 密码生成功能正常 +✅ 验证功能正常 diff --git a/pkg/lzkit/crypto/bcrypt.go b/pkg/lzkit/crypto/bcrypt.go new file mode 100644 index 0000000..7ed4512 --- /dev/null +++ b/pkg/lzkit/crypto/bcrypt.go @@ -0,0 +1,28 @@ +package crypto + +import ( + "golang.org/x/crypto/bcrypt" +) + +// PasswordHash 使用bcrypt对密码进行加密 +// cost参数确定加密的复杂度，默认为10，越高越安全但性能消耗越大 +func PasswordHash(password string, cost ...int) (string, error) { + defaultCost := 10 + if len(cost) > 0 && cost[0] > 0 { + defaultCost = cost[0] + } + + bytes, err := bcrypt.GenerateFromPassword([]byte(password), defaultCost) + if err != nil { + return "", err + } + + return string(bytes), nil +} + +// PasswordVerify 验证密码是否匹配 +// password是用户输入的明文密码，hash是存储的加密密码 +func PasswordVerify(password, hash string) bool { + err := bcrypt.CompareHashAndPassword([]byte(hash), []byte(password)) + return err == nil +} diff --git a/pkg/lzkit/crypto/bcrypt_test.go b/pkg/lzkit/crypto/bcrypt_test.go new file mode 100644 index 0000000..3d7ea11 --- /dev/null +++ b/pkg/lzkit/crypto/bcrypt_test.go @@ -0,0 +1,193 @@ +package crypto + +import ( + "fmt" + "testing" +) + +// TestPasswordHash 测试密码加密功能 +func TestPasswordHash(t *testing.T) { + testCases := []struct { + name string + password string + cost int + wantErr bool + }{ + { + name: "默认cost加密", + password: "123456", + cost: 0, // 使用默认cost + wantErr: false, + }, + { + name: "自定义cost加密", + password: "admin123", + cost: 12, + wantErr: false, + }, + { + name: "空密码", + password: "", + cost: 10, + wantErr: false, + }, + { + name: "复杂密码", + password: "MyP@ssw0rd!2024", + cost: 11, + wantErr: false, + }, + } + + for _, tc := range testCases { + t.Run(tc.name, func(t *testing.T) { + var hash string + var err error + + if tc.cost > 0 { + hash, err = PasswordHash(tc.password, tc.cost) + } else { + hash, err = PasswordHash(tc.password) + } + + if tc.wantErr { + if err == nil { + t.Errorf("PasswordHash() 期望出错，但没有错误") + } + return + } + + if err != nil { + t.Errorf("PasswordHash() 出现错误 = %v", err) + return + } + + if hash == "" { + t.Errorf("PasswordHash() 返回空字符串") + return + } + + // 验证生成的hash长度合理（bcrypt hash通常是60字符） + if len(hash) < 50 { + t.Errorf("PasswordHash() 返回的hash太短 = %d", len(hash)) + } + + // 验证密码验证功能 + if !PasswordVerify(tc.password, hash) { + t.Errorf("PasswordVerify() 验证失败，密码不匹配") + } + + // 验证错误密码 + if PasswordVerify("wrongpassword", hash) { + t.Errorf("PasswordVerify() 应该验证失败，但验证成功了") + } + }) + } +} + +// TestPasswordVerify 测试密码验证功能 +func TestPasswordVerify(t *testing.T) { + // 先生成一个已知的hash用于测试 + testPassword := "123456" + testHash, err := PasswordHash(testPassword, 10) + if err != nil { + t.Fatalf("生成测试hash失败: %v", err) + } + + testCases := []struct { + name string + password string + hash string + want bool + }{ + { + name: "正确密码", + password: testPassword, + hash: testHash, + want: true, + }, + { + name: "错误密码", + password: "wrongpassword", + hash: testHash, + want: false, + }, + } + + for _, tc := range testCases { + t.Run(tc.name, func(t *testing.T) { + got := PasswordVerify(tc.password, tc.hash) + if got != tc.want { + t.Errorf("PasswordVerify() = %v, want %v", got, tc.want) + } + }) + } +} + +// TestGeneratePasswords 生成常用密码的hash值（用于实际使用） +func TestGeneratePasswords(t *testing.T) { + // 常用密码列表 + passwords := []string{ + "123456", + "admin123", + "password", + "root", + "test123", + "MyP@ssw0rd!2024", + "admin123.", + "123456789", + "qwerty", + "abc123", + } + + fmt.Println("\n=== 生成密码Hash值 ===") + for i, password := range passwords { + hash, err := PasswordHash(password) + if err != nil { + t.Errorf("生成密码 %s 的hash失败: %v", password, err) + continue + } + + fmt.Printf("%d. 密码: %-20s Hash: %s\n", i+1, password, hash) + + // 验证生成的hash是否正确 + if !PasswordVerify(password, hash) { + t.Errorf("验证密码 %s 失败", password) + } + } + fmt.Println("=== 密码生成完成 ===") +} + +// BenchmarkPasswordHash 性能测试 +func BenchmarkPasswordHash(b *testing.B) { + password := "testpassword123" + + b.Run("Cost10", func(b *testing.B) { + for i := 0; i < b.N; i++ { + _, err := PasswordHash(password, 10) + if err != nil { + b.Fatal(err) + } + } + }) + + b.Run("Cost12", func(b *testing.B) { + for i := 0; i < b.N; i++ { + _, err := PasswordHash(password, 12) + if err != nil { + b.Fatal(err) + } + } + }) +} + +// BenchmarkPasswordVerify 验证性能测试 +func BenchmarkPasswordVerify(b *testing.B) { + password := "testpassword123" + hash, _ := PasswordHash(password, 10) + + b.ResetTimer() + for i := 0; i < b.N; i++ { + PasswordVerify(password, hash) + } +} diff --git a/pkg/lzkit/crypto/crypto.go b/pkg/lzkit/crypto/crypto.go new file mode 100644 index 0000000..65c840d --- /dev/null +++ b/pkg/lzkit/crypto/crypto.go @@ -0,0 +1,105 @@ +package crypto + +import ( + "bytes" + "crypto/aes" + "crypto/cipher" + "crypto/md5" + "crypto/rand" + "encoding/base64" + "encoding/hex" + "errors" + "io" +) + +// PKCS7填充 +func PKCS7Padding(ciphertext []byte, blockSize int) []byte { + padding := blockSize - len(ciphertext)%blockSize + padtext := bytes.Repeat([]byte{byte(padding)}, padding) + return append(ciphertext, padtext...) +} + +// 去除PKCS7填充 +func PKCS7UnPadding(origData []byte) ([]byte, error) { + length := len(origData) + if length == 0 { + return nil, errors.New("input data error") + } + unpadding := int(origData[length-1]) + if unpadding > length { + return nil, errors.New("unpadding size is invalid") + } + + // 检查填充字节是否一致 + for i := 0; i < unpadding; i++ { + if origData[length-1-i] != byte(unpadding) { + return nil, errors.New("invalid padding") + } + } + + return origData[:(length - unpadding)], nil +} + +// AES CBC模式加密，Base64传入传出 +func AesEncrypt(plainText, key []byte) (string, error) { + block, err := aes.NewCipher(key) + if err != nil { + return "", err + } + blockSize := block.BlockSize() + plainText = PKCS7Padding(plainText, blockSize) + + cipherText := make([]byte, blockSize+len(plainText)) + iv := cipherText[:blockSize] // 使用前blockSize字节作为IV + _, err = io.ReadFull(rand.Reader, iv) + if err != nil { + return "", err + } + + mode := cipher.NewCBCEncrypter(block, iv) + mode.CryptBlocks(cipherText[blockSize:], plainText) + + return base64.StdEncoding.EncodeToString(cipherText), nil +} + +// AES CBC模式解密，Base64传入传出 +func AesDecrypt(cipherTextBase64 string, key []byte) ([]byte, error) { + cipherText, err := base64.StdEncoding.DecodeString(cipherTextBase64) + if err != nil { + return nil, err + } + + block, err := aes.NewCipher(key) + if err != nil { + return nil, err + } + + blockSize := block.BlockSize() + if len(cipherText) < blockSize { + return nil, errors.New("ciphertext too short") + } + + iv := cipherText[:blockSize] + cipherText = cipherText[blockSize:] + + if len(cipherText)%blockSize != 0 { + return nil, errors.New("ciphertext is not a multiple of the block size") + } + + mode := cipher.NewCBCDecrypter(block, iv) + mode.CryptBlocks(cipherText, cipherText) + + plainText, err := PKCS7UnPadding(cipherText) + if err != nil { + return nil, err + } + + return plainText, nil +} + +// Md5Encrypt 用于对传入的message进行MD5加密 +func Md5Encrypt(message string) string { + hash := md5.New() + hash.Write([]byte(message)) // 将字符串转换为字节切片并写入 + return hex.EncodeToString(hash.Sum(nil)) // 将哈希值转换为16进制字符串并返回 +} diff --git a/pkg/lzkit/crypto/crypto_url.go b/pkg/lzkit/crypto/crypto_url.go new file mode 100644 index 0000000..777db50 --- /dev/null +++ b/pkg/lzkit/crypto/crypto_url.go @@ -0,0 +1,67 @@ +package crypto + +import ( + "crypto/aes" + "crypto/cipher" + "crypto/rand" + "encoding/base64" + "errors" + "io" +) + +// AES CBC模式加密，Base64传入传出 +func AesEncryptURL(plainText, key []byte) (string, error) { + block, err := aes.NewCipher(key) + if err != nil { + return "", err + } + blockSize := block.BlockSize() + plainText = PKCS7Padding(plainText, blockSize) + + cipherText := make([]byte, blockSize+len(plainText)) + iv := cipherText[:blockSize] // 使用前blockSize字节作为IV + _, err = io.ReadFull(rand.Reader, iv) + if err != nil { + return "", err + } + + mode := cipher.NewCBCEncrypter(block, iv) + mode.CryptBlocks(cipherText[blockSize:], plainText) + + return base64.URLEncoding.EncodeToString(cipherText), nil +} + +// AES CBC模式解密，Base64传入传出 +func AesDecryptURL(cipherTextBase64 string, key []byte) ([]byte, error) { + cipherText, err := base64.URLEncoding.DecodeString(cipherTextBase64) + if err != nil { + return nil, err + } + + block, err := aes.NewCipher(key) + if err != nil { + return nil, err + } + + blockSize := block.BlockSize() + if len(cipherText) < blockSize { + return nil, errors.New("ciphertext too short") + } + + iv := cipherText[:blockSize] + cipherText = cipherText[blockSize:] + + if len(cipherText)%blockSize != 0 { + return nil, errors.New("ciphertext is not a multiple of the block size") + } + + mode := cipher.NewCBCDecrypter(block, iv) + mode.CryptBlocks(cipherText, cipherText) + + plainText, err := PKCS7UnPadding(cipherText) + if err != nil { + return nil, err + } + + return plainText, nil +} diff --git a/pkg/lzkit/crypto/ecb.go b/pkg/lzkit/crypto/ecb.go new file mode 100644 index 0000000..3fed15f --- /dev/null +++ b/pkg/lzkit/crypto/ecb.go @@ -0,0 +1,274 @@ +package crypto + +import ( + "crypto/aes" + "crypto/md5" + "crypto/rand" + "encoding/base64" + "encoding/hex" + "errors" + "fmt" +) + +// ECB模式是一种基本的加密模式，每个明文块独立加密 +// 警告：ECB模式存在安全问题，仅用于需要确定性加密的场景，如数据库字段查询 +// 不要用于加密大段文本或安全要求高的场景 + +// 验证密钥长度是否有效 (AES-128, AES-192, AES-256) +func validateAESKey(key []byte) error { + switch len(key) { + case 16, 24, 32: + return nil + default: + return errors.New("AES密钥长度必须是16、24或32字节(对应AES-128、AES-192、AES-256)") + } +} + +// AesEcbEncrypt AES-ECB模式加密，返回Base64编码的密文 +// 使用已有的ECB实现，但提供更易用的接口 +func AesEcbEncrypt(plainText, key []byte) (string, error) { + if err := validateAESKey(key); err != nil { + return "", err + } + + block, err := aes.NewCipher(key) + if err != nil { + return "", err + } + + // 使用PKCS7填充 + plainText = PKCS7Padding(plainText, block.BlockSize()) + + // 创建密文数组 + cipherText := make([]byte, len(plainText)) + + // ECB模式加密，使用west_crypto.go中已有的实现 + mode := newECBEncrypter(block) + mode.CryptBlocks(cipherText, plainText) + + // 返回Base64编码的密文 + return base64.StdEncoding.EncodeToString(cipherText), nil +} + +// AesEcbDecrypt AES-ECB模式解密，输入Base64编码的密文 +func AesEcbDecrypt(cipherTextBase64 string, key []byte) ([]byte, error) { + if err := validateAESKey(key); err != nil { + return nil, err + } + + // Base64解码 + cipherText, err := base64.StdEncoding.DecodeString(cipherTextBase64) + if err != nil { + return nil, err + } + + block, err := aes.NewCipher(key) + if err != nil { + return nil, err + } + + // 检查密文长度 + if len(cipherText)%block.BlockSize() != 0 { + return nil, errors.New("密文长度必须是块大小的整数倍") + } + + // 创建明文数组 + plainText := make([]byte, len(cipherText)) + + // ECB模式解密，使用west_crypto.go中已有的实现 + mode := newECBDecrypter(block) + mode.CryptBlocks(plainText, cipherText) + + // 去除PKCS7填充 + plainText, err = PKCS7UnPadding(plainText) + if err != nil { + return nil, err + } + + return plainText, nil +} + +// AesEcbEncryptHex AES-ECB模式加密，返回十六进制编码的密文 +func AesEcbEncryptHex(plainText, key []byte) (string, error) { + if err := validateAESKey(key); err != nil { + return "", err + } + + block, err := aes.NewCipher(key) + if err != nil { + return "", err + } + + // 使用PKCS7填充 + plainText = PKCS7Padding(plainText, block.BlockSize()) + + // 创建密文数组 + cipherText := make([]byte, len(plainText)) + + // ECB模式加密 + mode := newECBEncrypter(block) + mode.CryptBlocks(cipherText, plainText) + + // 返回十六进制编码的密文 + return hex.EncodeToString(cipherText), nil +} + +// AesEcbDecryptHex AES-ECB模式解密，输入十六进制编码的密文 +func AesEcbDecryptHex(cipherTextHex string, key []byte) ([]byte, error) { + if err := validateAESKey(key); err != nil { + return nil, err + } + + // 十六进制解码 + cipherText, err := hex.DecodeString(cipherTextHex) + if err != nil { + return nil, err + } + + block, err := aes.NewCipher(key) + if err != nil { + return nil, err + } + + // 检查密文长度 + if len(cipherText)%block.BlockSize() != 0 { + return nil, errors.New("密文长度必须是块大小的整数倍") + } + + // 创建明文数组 + plainText := make([]byte, len(cipherText)) + + // ECB模式解密 + mode := newECBDecrypter(block) + mode.CryptBlocks(plainText, cipherText) + + // 去除PKCS7填充 + plainText, err = PKCS7UnPadding(plainText) + if err != nil { + return nil, err + } + + return plainText, nil +} + +// 以下是专门用于处理手机号等敏感数据的实用函数 + +// EncryptMobile 使用AES-ECB加密手机号，返回Base64编码 +// 该方法保证对相同手机号总是产生相同密文，便于数据库查询 +func EncryptMobile(mobile string, secretKey string) (string, error) { + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return "", decodeErr + } + if mobile == "" { + return "", errors.New("手机号不能为空") + } + return AesEcbEncrypt([]byte(mobile), key) +} + +// DecryptMobile 解密手机号 +func DecryptMobile(encryptedMobile string, secretKey string) (string, error) { + key, decodeErr := hex.DecodeString(secretKey) + if decodeErr != nil { + return "", decodeErr + } + if encryptedMobile == "" { + return "", errors.New("加密手机号不能为空") + } + + bytes, err := AesEcbDecrypt(encryptedMobile, key) + if err != nil { + return "", fmt.Errorf("解密手机号失败: %v", err) + } + + return string(bytes), nil +} + +// EncryptMobileHex 使用AES-ECB加密手机号，返回十六进制编码(适用于URL参数) +func EncryptMobileHex(mobile string, key []byte) (string, error) { + if mobile == "" { + return "", errors.New("手机号不能为空") + } + return AesEcbEncryptHex([]byte(mobile), key) +} + +// DecryptMobileHex 解密十六进制编码的手机号 +func DecryptMobileHex(encryptedMobileHex string, key []byte) (string, error) { + if encryptedMobileHex == "" { + return "", errors.New("加密手机号不能为空") + } + + bytes, err := AesEcbDecryptHex(encryptedMobileHex, key) + if err != nil { + return "", fmt.Errorf("解密手机号失败: %v", err) + } + + return string(bytes), nil +} + +// EncryptIDCard 使用AES-ECB加密身份证号 +func EncryptIDCard(idCard string, key []byte) (string, error) { + if idCard == "" { + return "", errors.New("身份证号不能为空") + } + return AesEcbEncrypt([]byte(idCard), key) +} + +// DecryptIDCard 解密身份证号 +func DecryptIDCard(encryptedIDCard string, key []byte) (string, error) { + if encryptedIDCard == "" { + return "", errors.New("加密身份证号不能为空") + } + + bytes, err := AesEcbDecrypt(encryptedIDCard, key) + if err != nil { + return "", fmt.Errorf("解密身份证号失败: %v", err) + } + + return string(bytes), nil +} + +// IsEncrypted 检查字符串是否为Base64编码的加密数据 +func IsEncrypted(data string) bool { + // 检查是否是有效的Base64编码 + _, err := base64.StdEncoding.DecodeString(data) + return err == nil && len(data) >= 20 // 至少20个字符的Base64字符串 +} + +// GenerateAESKey 生成AES密钥 +// keySize: 可选16, 24, 32字节(对应AES-128, AES-192, AES-256) +func GenerateAESKey(keySize int) ([]byte, error) { + if keySize != 16 && keySize != 24 && keySize != 32 { + return nil, errors.New("密钥长度必须是16、24或32字节") + } + + key := make([]byte, keySize) + _, err := rand.Read(key) + if err != nil { + return nil, err + } + + return key, nil +} + +// DeriveKeyFromPassword 基于密码派生固定长度的AES密钥 +func DeriveKeyFromPassword(password string, keySize int) ([]byte, error) { + if keySize != 16 && keySize != 24 && keySize != 32 { + return nil, errors.New("密钥长度必须是16、24或32字节") + } + + // 使用PBKDF2或简单的方法从密码派生密钥 + // 这里使用简单的MD5方法，实际生产环境应使用更安全的PBKDF2 + hash := md5.New() + hash.Write([]byte(password)) + key := hash.Sum(nil) // 16字节 + + // 如果需要24或32字节，继续哈希 + if keySize > 16 { + hash.Reset() + hash.Write(key) + key = append(key, hash.Sum(nil)[:keySize-16]...) + } + + return key, nil +} diff --git a/pkg/lzkit/crypto/ecb_test.go b/pkg/lzkit/crypto/ecb_test.go new file mode 100644 index 0000000..f7cb2f2 --- /dev/null +++ b/pkg/lzkit/crypto/ecb_test.go @@ -0,0 +1,186 @@ +package crypto + +import ( + "encoding/base64" + "encoding/hex" + "fmt" + "testing" +) + +func TestAesEcbMobileEncryption(t *testing.T) { + // 测试手机号加密 + mobile := "18653052547" + key := []byte("ff83609b2b24fc73196aac3d3dfb874f") // 16字节AES-128密钥 + + keyStr := hex.EncodeToString(key) + // 测试加密 + encrypted, err := EncryptMobile(mobile, keyStr) + if err != nil { + t.Fatalf("手机号加密失败: %v", err) + } + fmt.Printf("encrypted: %s\n", encrypted) + jmStr := "m9EEeW9ZBBJmi1hx1k1uIQ==" + // 测试解密 + decrypted, err := DecryptMobile(jmStr, keyStr) + if err != nil { + t.Fatalf("手机号解密失败: %v", err) + } + fmt.Printf("decrypted: %s\n", decrypted) + // 验证结果 + if decrypted != mobile { + t.Errorf("解密结果不匹配，期望: %s, 实际: %s", mobile, decrypted) + } + + // 测试相同输入产生相同输出（确定性） + encrypted2, _ := EncryptMobile(mobile, keyStr) + if encrypted != encrypted2 { + t.Errorf("AES-ECB不是确定性的，两次加密结果不同: %s vs %s", encrypted, encrypted2) + } +} + +func TestAesEcbHexEncryption(t *testing.T) { + // 测试十六进制编码加密 + idCard := "440101199001011234" + key := []byte("1234567890abcdef") // 16字节AES-128密钥 + + // 测试HEX加密 + encryptedHex, err := EncryptIDCard(idCard, key) + if err != nil { + t.Fatalf("身份证加密失败: %v", err) + } + + // 测试HEX解密 + decrypted, err := DecryptIDCard(encryptedHex, key) + if err != nil { + t.Fatalf("身份证解密失败: %v", err) + } + + // 验证结果 + if decrypted != idCard { + t.Errorf("解密结果不匹配，期望: %s, 实际: %s", idCard, decrypted) + } +} + +func TestAesEcbKeyValidation(t *testing.T) { + // 测试不同长度的密钥 + validKeys := [][]byte{ + make([]byte, 16), // AES-128 + make([]byte, 24), // AES-192 + make([]byte, 32), // AES-256 + } + + invalidKeys := [][]byte{ + make([]byte, 15), + make([]byte, 20), + make([]byte, 33), + } + + text := []byte("test text") + + // 测试有效密钥 + for _, key := range validKeys { + _, err := AesEcbEncrypt(text, key) + if err != nil { + t.Errorf("有效密钥(%d字节)校验失败: %v", len(key), err) + } + } + + // 测试无效密钥 + for _, key := range invalidKeys { + _, err := AesEcbEncrypt(text, key) + if err == nil { + t.Errorf("无效密钥(%d字节)未被检测出", len(key)) + } + } +} + +func TestIsEncrypted(t *testing.T) { + // 有效的Base64编码字符串 + validBase64 := base64.StdEncoding.EncodeToString([]byte("这是一个足够长的字符串，以通过IsEncrypted检查")) + + // 无效的字符串 + invalidStrings := []string{ + "", + "abc", + "not-base64!@#", + hex.EncodeToString([]byte("hexstring")), + } + + // 测试有效的加密数据 + if !IsEncrypted(validBase64) { + t.Errorf("有效的Base64未被识别为加密数据: %s", validBase64) + } + + // 测试无效的数据 + for _, s := range invalidStrings { + if IsEncrypted(s) { + t.Errorf("无效字符串被错误识别为加密数据: %s", s) + } + } +} + +func TestDeriveKeyFromPassword(t *testing.T) { + password := "my-secure-password" + + // 测试不同长度的派生密钥 + keySizes := []int{16, 24, 32} + + for _, size := range keySizes { + key, err := DeriveKeyFromPassword(password, size) + if err != nil { + t.Errorf("从密码派生%d字节密钥失败: %v", size, err) + continue + } + + if len(key) != size { + t.Errorf("派生的密钥长度错误，期望: %d, 实际: %d", size, len(key)) + } + + // 测试相同密码总是产生相同密钥 + key2, _ := DeriveKeyFromPassword(password, size) + if string(key) != string(key2) { + t.Errorf("从相同密码派生的密钥不一致") + } + + // 使用派生的密钥加密测试 + _, err = AesEcbEncrypt([]byte("test"), key) + if err != nil { + t.Errorf("使用派生的密钥加密失败: %v", err) + } + } + + // 测试无效的密钥大小 + _, err := DeriveKeyFromPassword(password, 18) + if err == nil { + t.Error("无效的密钥大小未被检测出") + } +} + +func TestGenerateAESKey(t *testing.T) { + // 测试生成不同长度的密钥 + keySizes := []int{16, 24, 32} + + for _, size := range keySizes { + key, err := GenerateAESKey(size) + if err != nil { + t.Errorf("生成%d字节密钥失败: %v", size, err) + continue + } + + if len(key) != size { + t.Errorf("生成的密钥长度错误，期望: %d, 实际: %d", size, len(key)) + } + + // 使用生成的密钥加密测试 + _, err = AesEcbEncrypt([]byte("test"), key) + if err != nil { + t.Errorf("使用生成的密钥加密失败: %v", err) + } + } + + // 测试无效的密钥大小 + _, err := GenerateAESKey(18) + if err == nil { + t.Error("无效的密钥大小未被检测出") + } +} diff --git a/pkg/lzkit/crypto/generate.go b/pkg/lzkit/crypto/generate.go new file mode 100644 index 0000000..2d91c6d --- /dev/null +++ b/pkg/lzkit/crypto/generate.go @@ -0,0 +1,63 @@ +package crypto + +import ( + "crypto/rand" + "encoding/hex" + "io" + mathrand "math/rand" + "strconv" + "time" +) + +// 生成AES-128密钥的函数，符合市面规范 +func GenerateSecretKey() (string, error) { + key := make([]byte, 16) // 16字节密钥 + _, err := io.ReadFull(rand.Reader, key) + if err != nil { + return "", err + } + return hex.EncodeToString(key), nil +} + +func GenerateSecretId() (string, error) { + // 创建一个字节数组，用于存储随机数据 + bytes := make([]byte, 8) // 因为每个字节表示两个16进制字符 + + // 读取随机字节到数组中 + _, err := rand.Read(bytes) + if err != nil { + return "", err + } + + // 将字节数组转换为16进制字符串 + return hex.EncodeToString(bytes), nil +} + +// GenerateTransactionID 生成16位数的交易单号 +func GenerateTransactionID() string { + length := 16 + // 获取当前时间戳 + timestamp := time.Now().UnixNano() + + // 转换为字符串 + timeStr := strconv.FormatInt(timestamp, 10) + + // 生成随机数 + mathrand.Seed(time.Now().UnixNano()) + randomPart := strconv.Itoa(mathrand.Intn(1000000)) + + // 组合时间戳和随机数 + combined := timeStr + randomPart + + // 如果长度超出指定值，则截断；如果不够，则填充随机字符 + if len(combined) >= length { + return combined[:length] + } + + // 如果长度不够，填充0 + for len(combined) < length { + combined += strconv.Itoa(mathrand.Intn(10)) // 填充随机数 + } + + return combined +} diff --git a/pkg/lzkit/crypto/west_crypto.go b/pkg/lzkit/crypto/west_crypto.go new file mode 100644 index 0000000..71d8a41 --- /dev/null +++ b/pkg/lzkit/crypto/west_crypto.go @@ -0,0 +1,150 @@ +package crypto + +import ( + "bytes" + "crypto/aes" + "crypto/cipher" + "crypto/sha1" + "encoding/base64" +) + +const ( + KEY_SIZE = 16 // AES-128, 16 bytes +) + +// Encrypt encrypts the given data using AES encryption in ECB mode with PKCS5 padding +func WestDexEncrypt(data, secretKey string) (string, error) { + key := generateAESKey(KEY_SIZE*8, []byte(secretKey)) + ciphertext, err := aesEncrypt([]byte(data), key) + if err != nil { + return "", err + } + return base64.StdEncoding.EncodeToString(ciphertext), nil +} + +// Decrypt decrypts the given base64-encoded string using AES encryption in ECB mode with PKCS5 padding +func WestDexDecrypt(encodedData, secretKey string) ([]byte, error) { + ciphertext, err := base64.StdEncoding.DecodeString(encodedData) + if err != nil { + return nil, err + } + key := generateAESKey(KEY_SIZE*8, []byte(secretKey)) + plaintext, err := aesDecrypt(ciphertext, key) + if err != nil { + return nil, err + } + return plaintext, nil +} + +// generateAESKey generates a key for AES encryption using a SHA-1 based PRNG +func generateAESKey(length int, password []byte) []byte { + h := sha1.New() + h.Write(password) + state := h.Sum(nil) + + keyBytes := make([]byte, 0, length/8) + for len(keyBytes) < length/8 { + h := sha1.New() + h.Write(state) + state = h.Sum(nil) + keyBytes = append(keyBytes, state...) + } + + return keyBytes[:length/8] +} + +// aesEncrypt encrypts plaintext using AES in ECB mode with PKCS5 padding +func aesEncrypt(plaintext, key []byte) ([]byte, error) { + block, err := aes.NewCipher(key) + if err != nil { + return nil, err + } + paddedPlaintext := pkcs5Padding(plaintext, block.BlockSize()) + ciphertext := make([]byte, len(paddedPlaintext)) + mode := newECBEncrypter(block) + mode.CryptBlocks(ciphertext, paddedPlaintext) + return ciphertext, nil +} + +// aesDecrypt decrypts ciphertext using AES in ECB mode with PKCS5 padding +func aesDecrypt(ciphertext, key []byte) ([]byte, error) { + block, err := aes.NewCipher(key) + if err != nil { + return nil, err + } + plaintext := make([]byte, len(ciphertext)) + mode := newECBDecrypter(block) + mode.CryptBlocks(plaintext, ciphertext) + return pkcs5Unpadding(plaintext), nil +} + +// pkcs5Padding pads the input to a multiple of the block size using PKCS5 padding +func pkcs5Padding(src []byte, blockSize int) []byte { + padding := blockSize - len(src)%blockSize + padtext := bytes.Repeat([]byte{byte(padding)}, padding) + return append(src, padtext...) +} + +// pkcs5Unpadding removes PKCS5 padding from the input +func pkcs5Unpadding(src []byte) []byte { + length := len(src) + unpadding := int(src[length-1]) + return src[:(length - unpadding)] +} + +// ECB mode encryption/decryption +type ecb struct { + b cipher.Block + blockSize int +} + +func newECB(b cipher.Block) *ecb { + return &ecb{ + b: b, + blockSize: b.BlockSize(), + } +} + +type ecbEncrypter ecb + +func newECBEncrypter(b cipher.Block) cipher.BlockMode { + return (*ecbEncrypter)(newECB(b)) +} + +func (x *ecbEncrypter) BlockSize() int { return x.blockSize } + +func (x *ecbEncrypter) CryptBlocks(dst, src []byte) { + if len(src)%x.blockSize != 0 { + panic("crypto/cipher: input not full blocks") + } + if len(dst) < len(src) { + panic("crypto/cipher: output smaller than input") + } + for len(src) > 0 { + x.b.Encrypt(dst, src[:x.blockSize]) + src = src[x.blockSize:] + dst = dst[x.blockSize:] + } +} + +type ecbDecrypter ecb + +func newECBDecrypter(b cipher.Block) cipher.BlockMode { + return (*ecbDecrypter)(newECB(b)) +} + +func (x *ecbDecrypter) BlockSize() int { return x.blockSize } + +func (x *ecbDecrypter) CryptBlocks(dst, src []byte) { + if len(src)%x.blockSize != 0 { + panic("crypto/cipher: input not full blocks") + } + if len(dst) < len(src) { + panic("crypto/cipher: output smaller than input") + } + for len(src) > 0 { + x.b.Decrypt(dst, src[:x.blockSize]) + src = src[x.blockSize:] + dst = dst[x.blockSize:] + } +} diff --git a/pkg/lzkit/delay/ProgressiveDelay.go b/pkg/lzkit/delay/ProgressiveDelay.go new file mode 100644 index 0000000..7d2e657 --- /dev/null +++ b/pkg/lzkit/delay/ProgressiveDelay.go @@ -0,0 +1,65 @@ +package delay + +import ( + "errors" + "fmt" + "time" +) + +// ProgressiveDelay 用于管理渐进式延迟策略 +type ProgressiveDelay struct { + initialDelay time.Duration // 初始延迟时间 + growthFactor float64 // 延迟增长因子 (例如 1.5) + maxDelay time.Duration // 最大延迟时间 + maxRetryDuration time.Duration // 最大重试时间 + currentDelay time.Duration // 当前延迟时间 + startTime time.Time // 重试开始时间 +} + +// New 创建一个新的渐进式延迟对象 +func New(initialDelay, maxDelay, maxRetryDuration time.Duration, growthFactor float64) (*ProgressiveDelay, error) { + // 参数校验 + if initialDelay <= 0 { + return nil, errors.New("initialDelay must be greater than zero") + } + if maxDelay <= 0 { + return nil, errors.New("maxDelay must be greater than zero") + } + if maxRetryDuration <= 0 { + return nil, errors.New("maxRetryDuration must be greater than zero") + } + if growthFactor <= 1.0 { + return nil, errors.New("growthFactor must be greater than 1") + } + + // 初始化并返回 + return &ProgressiveDelay{ + initialDelay: initialDelay, + maxDelay: maxDelay, + maxRetryDuration: maxRetryDuration, + growthFactor: growthFactor, + currentDelay: initialDelay, + startTime: time.Now(), + }, nil +} + +// NextDelay 计算并返回下次的延迟时间 +func (pd *ProgressiveDelay) NextDelay() (time.Duration, error) { + // 检查最大重试时间是否已过 + if time.Since(pd.startTime) > pd.maxRetryDuration { + return 0, fmt.Errorf("最大重试时间超过限制: %v", pd.maxRetryDuration) + } + + // 返回当前延迟时间 + delay := pd.currentDelay + + // 计算下一个延迟时间并更新 currentDelay + pd.currentDelay = time.Duration(float64(pd.currentDelay) * pd.growthFactor) + + // 如果下次延迟超过最大延迟时间，限制在最大值 + if pd.currentDelay > pd.maxDelay { + pd.currentDelay = pd.maxDelay + } + + return delay, nil +} diff --git a/pkg/lzkit/lzUtils/json.go b/pkg/lzkit/lzUtils/json.go new file mode 100644 index 0000000..e44c371 --- /dev/null +++ b/pkg/lzkit/lzUtils/json.go @@ -0,0 +1,35 @@ +package lzUtils + +import "github.com/bytedance/sonic" + +func RecursiveParse(data interface{}) (interface{}, error) { + switch v := data.(type) { + case string: + // 尝试解析字符串是否为嵌套 JSON + var parsed interface{} + if err := sonic.Unmarshal([]byte(v), &parsed); err == nil { + return RecursiveParse(parsed) + } + return v, nil // 普通字符串直接返回 + case map[string]interface{}: + for key, val := range v { + parsed, err := RecursiveParse(val) + if err != nil { + return nil, err + } + v[key] = parsed + } + return v, nil + case []interface{}: + for i, item := range v { + parsed, err := RecursiveParse(item) + if err != nil { + return nil, err + } + v[i] = parsed + } + return v, nil + default: + return v, nil + } +} diff --git a/pkg/lzkit/lzUtils/sqlutls.go b/pkg/lzkit/lzUtils/sqlutls.go new file mode 100644 index 0000000..66d2ce0 --- /dev/null +++ b/pkg/lzkit/lzUtils/sqlutls.go @@ -0,0 +1,72 @@ +package lzUtils + +import ( + "database/sql" + "time" +) + +// StringToNullString 将 string 转换为 sql.NullString +func StringToNullString(s string) sql.NullString { + return sql.NullString{ + String: s, + Valid: s != "", + } +} + +// NullStringToString 将 sql.NullString 转换为 string +func NullStringToString(ns sql.NullString) string { + if ns.Valid { + return ns.String + } + return "" +} + +// TimeToNullTime 将 time.Time 转换为 sql.NullTime +func TimeToNullTime(t time.Time) sql.NullTime { + return sql.NullTime{ + Time: t, + Valid: !t.IsZero(), // 仅当 t 不是零值时才设置为有效 + } +} + +// NullTimeToTime 将 sql.NullTime 转换为 time.Time +func NullTimeToTime(nt sql.NullTime) time.Time { + if nt.Valid { + return nt.Time + } + return time.Time{} // 返回零值时间 +} + +// Int64ToNullInt64 将 int64 转换为 sql.NullInt64 +func Int64ToNullInt64(i int64) sql.NullInt64 { + return sql.NullInt64{ + Int64: i, + Valid: i != 0, // 仅当 i 非零时才设置为有效 + } +} + +// NullInt64ToInt64 将 sql.NullInt64 转换为 int64 +func NullInt64ToInt64(ni sql.NullInt64) int64 { + if ni.Valid { + return ni.Int64 + } + return 0 // 返回零值 int64 +} + +// Float64ToNullFloat64 将 float64 转换为 sql.NullFloat64 +// Valid 字段在 f 非零时设置为有效（注意：NaN 会被视为有效，需结合业务场景使用） +func Float64ToNullFloat64(f float64) sql.NullFloat64 { + return sql.NullFloat64{ + Float64: f, + Valid: f != 0, + } +} + +// NullFloat64ToFloat64 将 sql.NullFloat64 转换为 float64 +// 当 Valid 为 false 时，返回 float64 零值 +func NullFloat64ToFloat64(nf sql.NullFloat64) float64 { + if nf.Valid { + return nf.Float64 + } + return 0.0 +} diff --git a/pkg/lzkit/lzUtils/time.go b/pkg/lzkit/lzUtils/time.go new file mode 100644 index 0000000..3c67d1e --- /dev/null +++ b/pkg/lzkit/lzUtils/time.go @@ -0,0 +1,26 @@ +package lzUtils + +import ( + "database/sql" + "time" +) + +// RenewMembership 延长会员有效期 +func RenewMembership(expiry sql.NullTime) sql.NullTime { + // 确定基准时间 + var baseTime time.Time + if expiry.Valid { + baseTime = expiry.Time + } else { + baseTime = time.Now() + } + + // 增加一年（自动处理闰年） + newTime := baseTime.AddDate(1, 0, 0) + + // 返回始终有效的 NullTime + return sql.NullTime{ + Time: newTime, + Valid: true, + } +} diff --git a/pkg/lzkit/lzUtils/utils.go b/pkg/lzkit/lzUtils/utils.go new file mode 100644 index 0000000..69a9aca --- /dev/null +++ b/pkg/lzkit/lzUtils/utils.go @@ -0,0 +1,15 @@ +package lzUtils + +import "fmt" + +// ToWechatAmount 将金额从元转换为微信支付 SDK 需要的分（int64 类型） +func ToWechatAmount(amount float64) int64 { + // 将金额从元转换为分，并四舍五入 + return int64(amount*100 + 0.5) +} + +// ToAlipayAmount 将金额从元转换为支付宝支付 SDK 需要的字符串格式，保留两位小数 +func ToAlipayAmount(amount float64) string { + // 格式化为字符串，保留两位小数 + return fmt.Sprintf("%.2f", amount) +} diff --git a/pkg/lzkit/md5/README.md b/pkg/lzkit/md5/README.md new file mode 100644 index 0000000..5e92541 --- /dev/null +++ b/pkg/lzkit/md5/README.md @@ -0,0 +1,106 @@ +# MD5 工具包 + +这个包提供了全面的 MD5 哈希功能，包括字符串加密、文件加密、链式操作、加盐哈希等。 + +## 主要功能 + +- 字符串和字节切片的 MD5 哈希计算 +- 文件 MD5 哈希计算（支持大文件分块处理） +- 链式 API，支持构建复杂的哈希内容 +- 带盐值的 MD5 哈希，提高安全性 +- 哈希验证功能 +- 16 位和 8 位 MD5 哈希（短版本） +- HMAC-MD5 实现，增强安全性 + +## 使用示例 + +### 基本使用 + +```go +// 计算字符串的MD5哈希 +hash := md5.EncryptString("hello world") +fmt.Println(hash) // 5eb63bbbe01eeed093cb22bb8f5acdc3 + +// 计算字节切片的MD5哈希 +bytes := []byte("hello world") +hash = md5.EncryptBytes(bytes) + +// 计算文件的MD5哈希 +fileHash, err := md5.EncryptFile("path/to/file.txt") +if err != nil { + log.Fatal(err) +} +fmt.Println(fileHash) +``` + +### 链式 API + +```go +// 创建一个新的MD5实例并添加内容 +hash := md5.New(). + Add("hello"). + Add(" "). + Add("world"). + Sum() +fmt.Println(hash) // 5eb63bbbe01eeed093cb22bb8f5acdc3 + +// 或者从字符串初始化 +hash = md5.FromString("hello"). + Add(" world"). + Sum() + +// 从字节切片初始化 +hash = md5.FromBytes([]byte("hello")). + AddBytes([]byte(" world")). + Sum() +``` + +### 安全性增强 + +```go +// 使用盐值加密（提高安全性） +hashedPassword := md5.EncryptStringWithSalt("password123", "main@example.com") + +// 使用前缀加密 +hashedValue := md5.EncryptStringWithPrefix("secret-data", "prefix-") + +// 验证带盐值的哈希 +isValid := md5.VerifyMD5WithSalt("password123", "main@example.com", hashedPassword) + +// 使用HMAC-MD5提高安全性 +hmacHash := md5.MD5HMAC("message", "secret-key") +``` + +### 短哈希值 + +```go +// 获取16位MD5（32位MD5的中间部分） +hash16 := md5.Get16("hello world") +fmt.Println(hash16) // 中间16个字符 + +// 获取8位MD5 +hash8 := md5.Get8("hello world") +fmt.Println(hash8) // 中间8个字符 +``` + +### 文件验证 + +```go +// 验证文件MD5是否匹配 +match, err := md5.VerifyFileMD5("path/to/file.txt", "expected-hash") +if err != nil { + log.Fatal(err) +} +if match { + fmt.Println("文件MD5校验通过") +} else { + fmt.Println("文件MD5校验失败") +} +``` + +## 注意事项 + +1. MD5 主要用于校验，不适合用于安全存储密码等敏感信息 +2. 如果用于密码存储，请务必使用加盐处理并考虑使用更安全的算法 +3. 处理大文件时请使用`EncryptFileChunk`以优化性能 +4. 返回的 MD5 哈希值都是 32 位的小写十六进制字符串（除非使用 Get16/Get8 函数） diff --git a/pkg/lzkit/md5/example_test.go b/pkg/lzkit/md5/example_test.go new file mode 100644 index 0000000..a5fbb87 --- /dev/null +++ b/pkg/lzkit/md5/example_test.go @@ -0,0 +1,79 @@ +package md5_test + +import ( + "qnc-server/pkg/lzkit/md5" + "fmt" + "log" +) + +func Example() { + // 简单的字符串MD5 + hashValue := md5.EncryptString("hello world") + fmt.Println("MD5(hello world):", hashValue) + + // 使用链式API + chainHash := md5.New(). + Add("hello"). + Add(" "). + Add("world"). + Sum() + fmt.Println("链式MD5:", chainHash) + + // 使用盐值 + saltedHash := md5.EncryptStringWithSalt("password123", "main@example.com") + fmt.Println("加盐MD5:", saltedHash) + + // 验证哈希 + isValid := md5.VerifyMD5("hello world", hashValue) + fmt.Println("验证结果:", isValid) + + // 生成短版本的MD5 + fmt.Println("16位MD5:", md5.Get16("hello world")) + fmt.Println("8位MD5:", md5.Get8("hello world")) + + // 文件MD5计算 + filePath := "example.txt" // 这只是示例，实际上这个文件可能不存在 + fileHash, err := md5.EncryptFile(filePath) + if err != nil { + // 在实际代码中执行正确的错误处理 + log.Printf("计算文件MD5出错: %v", err) + } else { + fmt.Println("文件MD5:", fileHash) + } + + // HMAC-MD5 + hmacHash := md5.MD5HMAC("重要消息", "secret-key") + fmt.Println("HMAC-MD5:", hmacHash) +} + +func ExampleEncryptString() { + hash := md5.EncryptString("HelloWorld") + fmt.Println(hash) + // Output: 68e109f0f40ca72a15e05cc22786f8e6 +} + +func ExampleMD5_Sum() { + hash := md5.New(). + Add("Hello"). + Add("World"). + Sum() + fmt.Println(hash) + // Output: 68e109f0f40ca72a15e05cc22786f8e6 +} + +func ExampleEncryptStringWithSalt() { + // 为用户密码加盐，通常使用用户唯一标识(如邮箱)作为盐值 + hash := md5.EncryptStringWithSalt("password123", "main@example.com") + fmt.Println("盐值哈希长度:", len(hash)) + fmt.Println("是否为有效哈希:", md5.VerifyMD5WithSalt("password123", "main@example.com", hash)) + // Output: + // 盐值哈希长度: 32 + // 是否为有效哈希: true +} + +func ExampleGet16() { + // 获取16位MD5，适合不需要完全防碰撞场景 + hash := md5.Get16("HelloWorld") + fmt.Println(hash) + // Output: f0f40ca72a15e05c +} diff --git a/pkg/lzkit/md5/md5.go b/pkg/lzkit/md5/md5.go new file mode 100644 index 0000000..bfbfa89 --- /dev/null +++ b/pkg/lzkit/md5/md5.go @@ -0,0 +1,206 @@ +package md5 + +import ( + "bufio" + "crypto/md5" + "encoding/hex" + "io" + "os" + "strings" +) + +// MD5结构体，可用于链式调用 +type MD5 struct { + data []byte +} + +// New 创建一个新的MD5实例 +func New() *MD5 { + return &MD5{ + data: []byte{}, + } +} + +// FromString 从字符串创建MD5 +func FromString(s string) *MD5 { + return &MD5{ + data: []byte(s), + } +} + +// FromBytes 从字节切片创建MD5 +func FromBytes(b []byte) *MD5 { + return &MD5{ + data: b, + } +} + +// Add 向MD5中添加字符串 +func (m *MD5) Add(s string) *MD5 { + m.data = append(m.data, []byte(s)...) + return m +} + +// AddBytes 向MD5中添加字节切片 +func (m *MD5) AddBytes(b []byte) *MD5 { + m.data = append(m.data, b...) + return m +} + +// Sum 计算并返回MD5哈希值(16进制字符串) +func (m *MD5) Sum() string { + hash := md5.New() + hash.Write(m.data) + return hex.EncodeToString(hash.Sum(nil)) +} + +// SumBytes 计算并返回MD5哈希值(字节切片) +func (m *MD5) SumBytes() []byte { + hash := md5.New() + hash.Write(m.data) + return hash.Sum(nil) +} + +// 直接调用的工具函数 + +// EncryptString 加密字符串 +func EncryptString(s string) string { + hash := md5.New() + hash.Write([]byte(s)) + return hex.EncodeToString(hash.Sum(nil)) +} + +// EncryptBytes 加密字节切片 +func EncryptBytes(b []byte) string { + hash := md5.New() + hash.Write(b) + return hex.EncodeToString(hash.Sum(nil)) +} + +// EncryptFile 加密文件内容 +func EncryptFile(filePath string) (string, error) { + file, err := os.Open(filePath) + if err != nil { + return "", err + } + defer file.Close() + + hash := md5.New() + if _, err := io.Copy(hash, file); err != nil { + return "", err + } + return hex.EncodeToString(hash.Sum(nil)), nil +} + +// EncryptFileChunk 对大文件分块计算MD5，提高效率 +func EncryptFileChunk(filePath string, chunkSize int) (string, error) { + if chunkSize <= 0 { + chunkSize = 1024 * 1024 // 默认1MB + } + + file, err := os.Open(filePath) + if err != nil { + return "", err + } + defer file.Close() + + hash := md5.New() + buf := make([]byte, chunkSize) + reader := bufio.NewReader(file) + + for { + n, err := reader.Read(buf) + if err != nil && err != io.EOF { + return "", err + } + if n == 0 { + break + } + hash.Write(buf[:n]) + } + + return hex.EncodeToString(hash.Sum(nil)), nil +} + +// EncryptStringWithSalt 使用盐值加密字符串 +func EncryptStringWithSalt(s, salt string) string { + return EncryptString(s + salt) +} + +// EncryptStringWithPrefix 使用前缀加密字符串 +func EncryptStringWithPrefix(s, prefix string) string { + return EncryptString(prefix + s) +} + +// VerifyMD5 验证字符串的MD5哈希是否匹配 +func VerifyMD5(s, hash string) bool { + return EncryptString(s) == strings.ToLower(hash) +} + +// VerifyMD5WithSalt 验证带盐值的字符串MD5哈希是否匹配 +func VerifyMD5WithSalt(s, salt, hash string) bool { + return EncryptStringWithSalt(s, salt) == strings.ToLower(hash) +} + +// VerifyFileMD5 验证文件的MD5哈希是否匹配 +func VerifyFileMD5(filePath, hash string) (bool, error) { + fileHash, err := EncryptFile(filePath) + if err != nil { + return false, err + } + return fileHash == strings.ToLower(hash), nil +} + +// MD5格式化为指定位数 + +// Get16 获取16位MD5值(取32位结果的中间16位) +func Get16(s string) string { + result := EncryptString(s) + return result[8:24] +} + +// Get8 获取8位MD5值 +func Get8(s string) string { + result := EncryptString(s) + return result[12:20] +} + +// MD5主要用于校验而非安全存储，对于需要高安全性的场景，应考虑: +// 1. bcrypt, scrypt或Argon2等专门为密码设计的算法 +// 2. HMAC-MD5等方式以防御彩虹表攻击 +// 3. 加盐并使用多次哈希迭代提高安全性 + +// MD5HMAC 使用HMAC-MD5算法 +func MD5HMAC(message, key string) string { + hash := md5.New() + + // 如果key长度超出block size，先进行哈希 + if len(key) > 64 { + hash.Write([]byte(key)) + key = hex.EncodeToString(hash.Sum(nil)) + hash.Reset() + } + + // 内部填充 + k_ipad := make([]byte, 64) + k_opad := make([]byte, 64) + copy(k_ipad, []byte(key)) + copy(k_opad, []byte(key)) + + for i := 0; i < 64; i++ { + k_ipad[i] ^= 0x36 + k_opad[i] ^= 0x5c + } + + // 内部哈希 + hash.Write(k_ipad) + hash.Write([]byte(message)) + innerHash := hash.Sum(nil) + hash.Reset() + + // 外部哈希 + hash.Write(k_opad) + hash.Write(innerHash) + + return hex.EncodeToString(hash.Sum(nil)) +} diff --git a/pkg/lzkit/md5/md5_test.go b/pkg/lzkit/md5/md5_test.go new file mode 100644 index 0000000..198c24a --- /dev/null +++ b/pkg/lzkit/md5/md5_test.go @@ -0,0 +1,192 @@ +package md5 + +import ( + "fmt" + "os" + "testing" +) + +func TestEncryptString(t *testing.T) { + tests := []struct { + input string + expected string + }{ + {"", "d41d8cd98f00b204e9800998ecf8427e"}, + {"hello", "5d41402abc4b2a76b9719d911017c592"}, + {"123456", "e10adc3949ba59abbe56e057f20f883e"}, + {"Hello World!", "ed076287532e86365e841e92bfc50d8c"}, + } + + for _, test := range tests { + result := EncryptString(test.input) + fmt.Println(result) + if result != test.expected { + t.Errorf("EncryptString(%s) = %s; want %s", test.input, result, test.expected) + } + } +} + +func TestEncryptBytes(t *testing.T) { + tests := []struct { + input []byte + expected string + }{ + {[]byte(""), "d41d8cd98f00b204e9800998ecf8427e"}, + {[]byte("hello"), "5d41402abc4b2a76b9719d911017c592"}, + {[]byte{0, 1, 2, 3, 4}, "5267768822ee624d48fce15ec5ca79b6"}, + } + + for _, test := range tests { + result := EncryptBytes(test.input) + if result != test.expected { + t.Errorf("EncryptBytes(%v) = %s; want %s", test.input, result, test.expected) + } + } +} + +func TestMD5Chain(t *testing.T) { + // 测试链式调用 + result := New(). + Add("hello"). + Add(" "). + Add("world"). + Sum() + + expected := "fc5e038d38a57032085441e7fe7010b0" // MD5("hello world") + if result != expected { + t.Errorf("Chain MD5 = %s; want %s", result, expected) + } + + // 测试从字符串初始化 + result = FromString("hello").Add(" world").Sum() + if result != expected { + t.Errorf("FromString MD5 = %s; want %s", result, expected) + } + + // 测试从字节切片初始化 + result = FromBytes([]byte("hello")).AddBytes([]byte(" world")).Sum() + if result != expected { + t.Errorf("FromBytes MD5 = %s; want %s", result, expected) + } +} + +func TestVerifyMD5(t *testing.T) { + if !VerifyMD5("hello", "5d41402abc4b2a76b9719d911017c592") { + t.Error("VerifyMD5 failed for correct match") + } + + if VerifyMD5("hello", "wrong-hash") { + t.Error("VerifyMD5 succeeded for incorrect match") + } + + // 测试大小写不敏感 + if !VerifyMD5("hello", "5D41402ABC4B2A76B9719D911017C592") { + t.Error("VerifyMD5 failed for uppercase hash") + } +} + +func TestSaltAndPrefix(t *testing.T) { + // 测试加盐 + saltResult := EncryptStringWithSalt("password", "salt123") + expectedSalt := EncryptString("passwordsalt123") + if saltResult != expectedSalt { + t.Errorf("EncryptStringWithSalt = %s; want %s", saltResult, expectedSalt) + } + + // 测试前缀 + prefixResult := EncryptStringWithPrefix("password", "prefix123") + expectedPrefix := EncryptString("prefix123password") + if prefixResult != expectedPrefix { + t.Errorf("EncryptStringWithPrefix = %s; want %s", prefixResult, expectedPrefix) + } + + // 验证带盐值的MD5 + if !VerifyMD5WithSalt("password", "salt123", saltResult) { + t.Error("VerifyMD5WithSalt failed for correct match") + } +} + +func TestGet16And8(t *testing.T) { + full := EncryptString("test-string") + + // 测试16位MD5 + result16 := Get16("test-string") + expected16 := full[8:24] + if result16 != expected16 { + t.Errorf("Get16 = %s; want %s", result16, expected16) + } + + // 测试8位MD5 + result8 := Get8("test-string") + expected8 := full[12:20] + if result8 != expected8 { + t.Errorf("Get8 = %s; want %s", result8, expected8) + } +} + +func TestMD5HMAC(t *testing.T) { + // 已知的HMAC-MD5结果 + tests := []struct { + message string + key string + expected string + }{ + {"message", "key", "4e4748e62b463521f6775fbf921234b5"}, + {"test", "secret", "8b11d99898918564dda1a9fe205b5310"}, + } + + for _, test := range tests { + result := MD5HMAC(test.message, test.key) + if result != test.expected { + t.Errorf("MD5HMAC(%s, %s) = %s; want %s", + test.message, test.key, result, test.expected) + } + } +} + +func TestEncryptFile(t *testing.T) { + // 创建临时测试文件 + content := []byte("test file content for MD5") + tmpFile, err := os.CreateTemp("", "md5test-*.txt") + if err != nil { + t.Fatalf("无法创建临时文件: %v", err) + } + defer os.Remove(tmpFile.Name()) + + if _, err := tmpFile.Write(content); err != nil { + t.Fatalf("无法写入临时文件: %v", err) + } + if err := tmpFile.Close(); err != nil { + t.Fatalf("无法关闭临时文件: %v", err) + } + + // 计算文件MD5 + fileHash, err := EncryptFile(tmpFile.Name()) + if err != nil { + t.Fatalf("计算文件MD5失败: %v", err) + } + + // 验证文件MD5 + expectedHash := EncryptBytes(content) + if fileHash != expectedHash { + t.Errorf("文件MD5 = %s; 应为 %s", fileHash, expectedHash) + } + + // 测试VerifyFileMD5 + match, err := VerifyFileMD5(tmpFile.Name(), expectedHash) + if err != nil { + t.Fatalf("验证文件MD5失败: %v", err) + } + if !match { + t.Error("VerifyFileMD5返回false，应返回true") + } + + // 测试不匹配的情况 + match, err = VerifyFileMD5(tmpFile.Name(), "wronghash") + if err != nil { + t.Fatalf("验证文件MD5失败: %v", err) + } + if match { + t.Error("VerifyFileMD5对错误的哈希返回true，应返回false") + } +} diff --git a/pkg/lzkit/validator/error_messages.go b/pkg/lzkit/validator/error_messages.go new file mode 100644 index 0000000..a470cdf --- /dev/null +++ b/pkg/lzkit/validator/error_messages.go @@ -0,0 +1,38 @@ +package validator + +// 定义自定义错误消息 +var customMessages = map[string]string{ + "Name.min": "姓名不能少于1个字", + "Name.required": "姓名是必填项", + "Name.name": "姓名只能包含中文", + "NameMan.min": "男方姓名不能少于1个字", + "NameMan.required": "男方姓名是必填项", + "NameMan.name": "男方姓名只能包含中文", + "NameWoman.min": "女方姓名不能少于1个字", + "NameWoman.required": "女方姓名是必填项", + "NameWoman.name": "女方姓名只能包含中文", + "Mobile.required": "手机号是必填项", + "Mobile.min": "电话号码必须为有效的中国电话号码", + "Mobile.max": "电话号码必须为有效的中国电话号码", + "Mobile.mobile": "电话号码必须为有效的中国电话号码", + "IDCard.required": "身份证号是必填项", + "IDCard.idCard": "无效的身份证号码", + "IDCardMan.required": "男方身份证号是必填项", + "IDCardMan.idCard": "无效的男方身份证号码", + "IDCardWoman.required": "女方身份证号是必填项", + "IDCardWoman.idCard": "无效的女方身份证号码", + "Password.min": "密码不能少于8位数", + "Password.max": "密码不能超过32位数", + "Password.password": "密码强度太弱", + //"EntCode.required":"请输入统一社会信用代码", + //"EntCode.USCI": "请输入正确的统一社会信用代码", +} + +// 获取自定义错误消息 +func GetErrorMessage(field, tag string) string { + key := field + "." + tag + if msg, exists := customMessages[key]; exists { + return msg + } + return "请输入正确格式的参数" +} diff --git a/pkg/lzkit/validator/validator.go b/pkg/lzkit/validator/validator.go new file mode 100644 index 0000000..594696f --- /dev/null +++ b/pkg/lzkit/validator/validator.go @@ -0,0 +1,225 @@ +package validator + +import ( + "errors" + "fmt" + "regexp" + "strconv" + "strings" + "time" + + "github.com/go-playground/validator/v10" +) + +var validate *validator.Validate + +// 初始化自定义校验器 +func init() { + validate = validator.New() + + if err := validate.RegisterValidation("name", validName); err != nil { + panic(fmt.Sprintf("注册 name 验证器时发生错误: %v", err)) + } + + // 注册自定义验证器 validmobile + if err := validate.RegisterValidation("mobile", validmobile); err != nil { + panic(fmt.Sprintf("注册 mobile 验证器时发生错误: %v", err)) + } + + // 注册自定义验证器 validDate + if err := validate.RegisterValidation("date", validDate); err != nil { + panic(fmt.Sprintf("注册 date 验证器时发生错误: %v", err)) + } + + // 注册自定义验证器 validIDCard + if err := validate.RegisterValidation("idCard", validIDCard); err != nil { + panic(fmt.Sprintf("注册 idCard 验证器时发生错误: %v", err)) + } + + if err := validate.RegisterValidation("bankCard", validBankCard); err != nil { + panic(fmt.Sprintf("注册 bankCard 验证器时发生错误: %v", err)) + } + + if err := validate.RegisterValidation("USCI", validUSCI); err != nil { + panic(fmt.Sprintf("注册 USCI 社会统一信用代码 验证器时发生错误: %v", err)) + } + + if err := validate.RegisterValidation("mobileType", validMobileType); err != nil { + panic(fmt.Sprintf("注册 mobileType 验证器时发生错误: %v", err)) + } + if err := validate.RegisterValidation("password", validatePassword); err != nil { + panic(fmt.Sprintf("注册 password 验证器时发生错误: %v", err)) + } + if err := validate.RegisterValidation("payMethod", validatePayMethod); err != nil { + panic(fmt.Sprintf("注册 payMethod 验证器时发生错误: %v", err)) + } + +} + +// 弱口令列表 +var weakPasswords = []string{ + "12345678", "password", "123456789", "qwerty", "123456", "letmein", + "1234567", "welcome", "abc123", "password1", "1234", "111111", "admin", +} + +// Validate 校验参数逻辑 +func Validate(req interface{}) error { + if err := validate.Struct(req); err != nil { + // 检查 err 是否是 ValidationErrors 类型 + if validationErrors, ok := err.(validator.ValidationErrors); ok { + for _, validationErr := range validationErrors { + field := validationErr.StructField() + tag := validationErr.Tag() + return errors.New(GetErrorMessage(field, tag)) + } + } else { + // 其他错误处理 + return fmt.Errorf("验证时出现未知错误: %v", err) + } + } + return nil +} + +// 自定义的名称验证 +func validName(fl validator.FieldLevel) bool { + name := fl.Field().String() + validNamePattern := `^[\p{Han}]+$` + matched, _ := regexp.MatchString(validNamePattern, name) + return matched +} + +// 自定义的手机号验证 +func validmobile(fl validator.FieldLevel) bool { + phone := fl.Field().String() + validmobilePattern := `^1[3-9]\d{9}$` + matched, _ := regexp.MatchString(validmobilePattern, phone) + return matched +} + +// 自定义正则表达式校验 yyyyMMdd 格式 +func validDate(fl validator.FieldLevel) bool { + date := fl.Field().String() + validDatePattern := `^\d{4}(0[1-9]|1[0-2])(0[1-9]|[12][0-9]|3[01])$` + matched, _ := regexp.MatchString(validDatePattern, date) + return matched +} + + +// 自定义身份证校验（增强版） +// 校验规则： +// 1. 格式：18位，前6位地区码（首位不为0），7-14位出生日期，15-17位顺序码，18位校验码 +// 2. 出生日期必须合法（验证年月日有效性，包括闰年） +// 3. 校验码按照GB 11643-1999标准计算验证 +func validIDCard(fl validator.FieldLevel) bool { + idCard := fl.Field().String() + + // 1. 基本格式验证：地区码(6位) + 年(4位) + 月(2位) + 日(2位) + 顺序码(3位) + 校验码(1位) + validIDPattern := `^[1-9]\d{5}(18|19|20)\d{2}((0[1-9])|(1[0-2]))(([0-2][1-9])|10|20|30|31)\d{3}[\dXx]$` + matched, _ := regexp.MatchString(validIDPattern, idCard) + if !matched { + return false + } + + // 2. 验证出生日期的合法性 + year, _ := strconv.Atoi(idCard[6:10]) + month, _ := strconv.Atoi(idCard[10:12]) + day, _ := strconv.Atoi(idCard[12:14]) + + // 构造日期并验证是否合法（time包会自动处理闰年等情况） + birthDate := time.Date(year, time.Month(month), day, 0, 0, 0, 0, time.UTC) + if birthDate.Year() != year || int(birthDate.Month()) != month || birthDate.Day() != day { + return false // 日期不合法，如2月30日、4月31日等 + } + + // 3. 验证校验码（按照GB 11643-1999标准） + return validateIDCardChecksum(idCard) +} + +// 验证身份证校验码（GB 11643-1999标准） +func validateIDCardChecksum(idCard string) bool { + // 加权因子 + weights := []int{7, 9, 10, 5, 8, 4, 2, 1, 6, 3, 7, 9, 10, 5, 8, 4, 2} + // 校验码对应值 + checksumChars := []byte{'1', '0', 'X', '9', '8', '7', '6', '5', '4', '3', '2'} + + sum := 0 + for i := 0; i < 17; i++ { + num := int(idCard[i] - '0') + sum += num * weights[i] + } + + // 计算校验码 + checksum := checksumChars[sum%11] + lastChar := idCard[17] + + // 支持小写x + if lastChar == 'x' { + lastChar = 'X' + } + + return byte(lastChar) == checksum +} + +func validBankCard(fl validator.FieldLevel) bool { + bankCard := fl.Field().String() + // 银行卡号一般是13到19位的数字 + validBankCardPattern := `^\d{13,19}$` + matched, _ := regexp.MatchString(validBankCardPattern, bankCard) + return matched +} + +func validUSCI(fl validator.FieldLevel) bool { + usci := fl.Field().String() + // 社会信用代码为18位数字和大写字母的组合，最后一位为校验码 + validUSCIPattern := `^[1-9A-Z]{2}[0-9]{6}[0-9A-Z]{9}[0-9A-Z]$` + matched, _ := regexp.MatchString(validUSCIPattern, usci) + return matched +} + +// 自定义的手机号类型验证（可以为空） +func validMobileType(fl validator.FieldLevel) bool { + mobileType := fl.Field().String() + if mobileType == "" { + return true // 如果为空，认为是有效的 + } + + // 校验是否是 CTCC, CMCC, CUCC 之一 + validTypes := map[string]bool{ + "CTCC": true, // 中国电信 + "CMCC": true, // 中国移动 + "CUCC": true, // 中国联通 + } + + return validTypes[mobileType] +} + +// 自定义密码强度校验函数 +func validatePassword(fl validator.FieldLevel) bool { + password := fl.Field().String() + + // 检查密码是否在弱口令列表中 + for _, weakPwd := range weakPasswords { + if strings.ToLower(password) == weakPwd { + return false + } + } + + return true +} + +// 支付方式 +func validatePayMethod(fl validator.FieldLevel) bool { + payMethod := fl.Field().String() + + if payMethod == "" { + return true // 如果为空，认为是有效的 + } + + validTypes := map[string]bool{ + "alipay": true, // 中国电信 + "wechatpay": true, // 中国移动 + } + + return validTypes[payMethod] + +}