Compare commits
9 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
| a6350dd906 | |||
| 411ded8a0c | |||
| 2a998c90b8 | |||
| a6f0bfd4e8 | |||
| fd097729d7 | |||
| 1c0aeb152a | |||
| b4170b4863 | |||
| 1eab1b99c1 | |||
| aeaf6a04c2 |
+16
-16
@@ -46,25 +46,25 @@ ADMIN_API_TOKEN=replace-with-strong-random-token
|
|||||||
ADMIN_DEFAULT_USERNAME=admin
|
ADMIN_DEFAULT_USERNAME=admin
|
||||||
ADMIN_DEFAULT_PASSWORD=admin123
|
ADMIN_DEFAULT_PASSWORD=admin123
|
||||||
|
|
||||||
# 七牛直传配置(Kodo)
|
# 阿里云 OSS 直传配置
|
||||||
QINIU_ACCESS_KEY=replace-with-access-key
|
OSS_ACCESS_KEY=replace-with-access-key
|
||||||
QINIU_SECRET_KEY=replace-with-secret-key
|
OSS_SECRET_KEY=replace-with-secret-key
|
||||||
QINIU_BUCKET=replace-with-bucket
|
OSS_BUCKET=replace-with-bucket
|
||||||
# 上传地址:可保持默认(自动调度),也可以配置具体 Region 的 up 域名
|
# 上传地址(可选):OSS 时自动根据 endpoint 计算,留空即可
|
||||||
QINIU_UPLOAD_URL=https://upload.qiniup.com
|
OSS_UPLOAD_URL=
|
||||||
# CDN 域名(可选):用于拼接最终访问地址,例如 https://cdn.example.com
|
# CDN 域名:阿里云 OSS endpoint,例如 oss-cn-beijing.aliyuncs.com
|
||||||
QINIU_CDN_DOMAIN=
|
OSS_CDN_DOMAIN=
|
||||||
# 上传 key 前缀(可选)
|
# 上传 key 前缀(可选)
|
||||||
QINIU_KEY_PREFIX=uploads/
|
OSS_KEY_PREFIX=uploads/
|
||||||
# token 有效期(秒)
|
# 凭证有效期(秒)
|
||||||
QINIU_TOKEN_EXPIRE_SECONDS=300
|
OSS_TOKEN_EXPIRE_SECONDS=300
|
||||||
# 上传回调地址(可选):配置后,七牛上传成功会回调该地址
|
# 上传回调地址(可选):上传成功后回调该地址
|
||||||
# 示例: https://api.example.com/api/v1/common/upload/qiniu/callback
|
# 示例: https://api.example.com/api/v1/common/upload/oss/callback
|
||||||
QINIU_CALLBACK_URL=
|
OSS_CALLBACK_URL=
|
||||||
# 回调内容模板(可选)
|
# 回调内容模板(可选)
|
||||||
QINIU_CALLBACK_BODY=key=$(key)&hash=$(etag)&fsize=$(fsize)&mimeType=$(mimeType)
|
OSS_CALLBACK_BODY=key=$(key)&hash=$(etag)&fsize=$(fsize)&mimeType=$(mimeType)
|
||||||
# 回调内容类型(可选)
|
# 回调内容类型(可选)
|
||||||
QINIU_CALLBACK_BODY_TYPE=application/x-www-form-urlencoded
|
OSS_CALLBACK_BODY_TYPE=application/x-www-form-urlencoded
|
||||||
|
|
||||||
# 微信公众号(网页授权 OAuth2)
|
# 微信公众号(网页授权 OAuth2)
|
||||||
WECHAT_OA_APP_ID=replace-with-oa-appid
|
WECHAT_OA_APP_ID=replace-with-oa-appid
|
||||||
|
|||||||
@@ -36,3 +36,4 @@ go.work
|
|||||||
# Local build binary
|
# Local build binary
|
||||||
wx_service
|
wx_service
|
||||||
wx_service_api
|
wx_service_api
|
||||||
|
/api
|
||||||
|
|||||||
+23
-6
@@ -11,8 +11,8 @@ import (
|
|||||||
adminmodule "wx_service/internal/admin"
|
adminmodule "wx_service/internal/admin"
|
||||||
authhandler "wx_service/internal/common/auth/handler"
|
authhandler "wx_service/internal/common/auth/handler"
|
||||||
authservice "wx_service/internal/common/auth/service"
|
authservice "wx_service/internal/common/auth/service"
|
||||||
qiniuhandler "wx_service/internal/common/qiniu/handler"
|
uploadhandler "wx_service/internal/common/upload/handler"
|
||||||
qiniuservice "wx_service/internal/common/qiniu/service"
|
uploadservice "wx_service/internal/common/upload/service"
|
||||||
rediscache "wx_service/internal/common/redis/cache"
|
rediscache "wx_service/internal/common/redis/cache"
|
||||||
redisservice "wx_service/internal/common/redis/service"
|
redisservice "wx_service/internal/common/redis/service"
|
||||||
oahandler "wx_service/internal/common/wechat_official/handler"
|
oahandler "wx_service/internal/common/wechat_official/handler"
|
||||||
@@ -30,6 +30,7 @@ import (
|
|||||||
membershipservice "wx_service/internal/membership/service"
|
membershipservice "wx_service/internal/membership/service"
|
||||||
"wx_service/internal/model"
|
"wx_service/internal/model"
|
||||||
"wx_service/internal/observability"
|
"wx_service/internal/observability"
|
||||||
|
"wx_service/internal/achievement"
|
||||||
quitcheckinhandler "wx_service/internal/quitcheckin/handler"
|
quitcheckinhandler "wx_service/internal/quitcheckin/handler"
|
||||||
quitcheckinmodel "wx_service/internal/quitcheckin/model"
|
quitcheckinmodel "wx_service/internal/quitcheckin/model"
|
||||||
quitcheckinservice "wx_service/internal/quitcheckin/service"
|
quitcheckinservice "wx_service/internal/quitcheckin/service"
|
||||||
@@ -82,10 +83,15 @@ func main() {
|
|||||||
&marketingmodel.MarketingCategory{},
|
&marketingmodel.MarketingCategory{},
|
||||||
&marketingmodel.MarketingTemplate{},
|
&marketingmodel.MarketingTemplate{},
|
||||||
&marketingmodel.MarketingDownload{},
|
&marketingmodel.MarketingDownload{},
|
||||||
|
&marketingmodel.UserLogo{},
|
||||||
|
&marketingmodel.AdPlacement{},
|
||||||
&quitcheckinmodel.Profile{},
|
&quitcheckinmodel.Profile{},
|
||||||
&quitcheckinmodel.DailyStatus{},
|
&quitcheckinmodel.DailyStatus{},
|
||||||
&quitcheckinmodel.RelapseEvent{},
|
&quitcheckinmodel.RelapseEvent{},
|
||||||
&quitcheckinmodel.RewardGoal{},
|
&quitcheckinmodel.RewardGoal{},
|
||||||
|
&quitcheckinmodel.DreamPreset{},
|
||||||
|
&achievement.Theme{},
|
||||||
|
&achievement.Level{},
|
||||||
); err != nil {
|
); err != nil {
|
||||||
log.Fatalf("auto migrate failed: %v", err)
|
log.Fatalf("auto migrate failed: %v", err)
|
||||||
}
|
}
|
||||||
@@ -115,16 +121,20 @@ func main() {
|
|||||||
smokeAINextService := smokeservice.NewSmokeAINextSmokeService(database.DB, config.AppConfig.AI)
|
smokeAINextService := smokeservice.NewSmokeAINextSmokeService(database.DB, config.AppConfig.AI)
|
||||||
smokeShareService := smokeservice.NewSmokeShareService(database.DB)
|
smokeShareService := smokeservice.NewSmokeShareService(database.DB)
|
||||||
smokeQuitPlanService := smokeservice.NewSmokeQuitPlanService(database.DB, config.AppConfig.AI)
|
smokeQuitPlanService := smokeservice.NewSmokeQuitPlanService(database.DB, config.AppConfig.AI)
|
||||||
smokeHandler := smokehandler.NewSmokeHandler(smokeLogService, smokeAIAdviceService, smokeProfileService, smokeNextService, smokeAINextService, smokeShareService)
|
achievementService := achievement.NewService(database.DB)
|
||||||
quitPlanHandler := smokehandler.NewQuitPlanHandler(smokeQuitPlanService)
|
if err := achievementService.SeedDefaults(context.Background()); err != nil {
|
||||||
|
log.Printf("seed achievement defaults: %v", err)
|
||||||
|
}
|
||||||
quitCheckinService := quitcheckinservice.NewService(database.DB)
|
quitCheckinService := quitcheckinservice.NewService(database.DB)
|
||||||
|
smokeHandler := smokehandler.NewSmokeHandler(smokeLogService, smokeAIAdviceService, smokeProfileService, smokeNextService, smokeAINextService, smokeShareService, achievementService, quitCheckinService)
|
||||||
|
quitPlanHandler := smokehandler.NewQuitPlanHandler(smokeQuitPlanService)
|
||||||
quitCheckinHandler := quitcheckinhandler.NewHandler(quitCheckinService)
|
quitCheckinHandler := quitcheckinhandler.NewHandler(quitCheckinService)
|
||||||
|
|
||||||
redeemCodeService := membershipservice.NewRedeemCodeService(database.DB, config.AppConfig.Admin.Token)
|
redeemCodeService := membershipservice.NewRedeemCodeService(database.DB, config.AppConfig.Admin.Token)
|
||||||
redeemCodeHandler := membershiphandler.NewRedeemCodeHandler(redeemCodeService)
|
redeemCodeHandler := membershiphandler.NewRedeemCodeHandler(redeemCodeService)
|
||||||
|
|
||||||
qiniuService := qiniuservice.NewQiniuService(config.AppConfig.Qiniu)
|
uploadSvc := uploadservice.NewUploadService(config.AppConfig.OSS)
|
||||||
uploadHandler := qiniuhandler.NewUploadHandler(qiniuService)
|
uploadHandler := uploadhandler.NewUploadHandler(uploadSvc)
|
||||||
|
|
||||||
oaService := oaservice.NewWeChatOAService(config.AppConfig.WeChatOA)
|
oaService := oaservice.NewWeChatOAService(config.AppConfig.WeChatOA)
|
||||||
oaOAuthHandler := oahandler.NewOAuthHandler(oaService)
|
oaOAuthHandler := oahandler.NewOAuthHandler(oaService)
|
||||||
@@ -156,12 +166,17 @@ func main() {
|
|||||||
categoryRepo := marketingrepo.NewCategoryRepository(database.DB)
|
categoryRepo := marketingrepo.NewCategoryRepository(database.DB)
|
||||||
templateRepo := marketingrepo.NewTemplateRepository(database.DB)
|
templateRepo := marketingrepo.NewTemplateRepository(database.DB)
|
||||||
downloadRepo := marketingrepo.NewDownloadRepository(database.DB)
|
downloadRepo := marketingrepo.NewDownloadRepository(database.DB)
|
||||||
|
userLogoRepo := marketingrepo.NewUserLogoRepository(database.DB)
|
||||||
categorySvc := marketingservice.NewCategoryService(categoryRepo)
|
categorySvc := marketingservice.NewCategoryService(categoryRepo)
|
||||||
templateSvc := marketingservice.NewTemplateService(templateRepo)
|
templateSvc := marketingservice.NewTemplateService(templateRepo)
|
||||||
downloadSvc := marketingservice.NewDownloadService(downloadRepo, templateRepo)
|
downloadSvc := marketingservice.NewDownloadService(downloadRepo, templateRepo)
|
||||||
|
userLogoSvc := marketingservice.NewUserLogoService(userLogoRepo)
|
||||||
marketingCategoryHandler := marketinghandler.NewCategoryHandler(categorySvc)
|
marketingCategoryHandler := marketinghandler.NewCategoryHandler(categorySvc)
|
||||||
marketingTemplateHandler := marketinghandler.NewTemplateHandler(templateSvc)
|
marketingTemplateHandler := marketinghandler.NewTemplateHandler(templateSvc)
|
||||||
marketingDownloadHandler := marketinghandler.NewDownloadHandler(downloadSvc)
|
marketingDownloadHandler := marketinghandler.NewDownloadHandler(downloadSvc)
|
||||||
|
marketingUserLogoHandler := marketinghandler.NewUserLogoHandler(userLogoSvc)
|
||||||
|
adPlacementRepo := marketingrepo.NewAdPlacementRepository(database.DB)
|
||||||
|
marketingAdPlacementHandler := marketinghandler.NewAdPlacementHandler(adPlacementRepo)
|
||||||
|
|
||||||
adminService := adminmodule.NewService(
|
adminService := adminmodule.NewService(
|
||||||
database.DB,
|
database.DB,
|
||||||
@@ -194,6 +209,8 @@ func main() {
|
|||||||
marketingCategoryHandler,
|
marketingCategoryHandler,
|
||||||
marketingTemplateHandler,
|
marketingTemplateHandler,
|
||||||
marketingDownloadHandler,
|
marketingDownloadHandler,
|
||||||
|
marketingUserLogoHandler,
|
||||||
|
marketingAdPlacementHandler,
|
||||||
quitCheckinHandler,
|
quitCheckinHandler,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
+15
-15
@@ -17,7 +17,7 @@ type Config struct {
|
|||||||
ShortVideo ShortVideoConfig
|
ShortVideo ShortVideoConfig
|
||||||
AI AIConfig
|
AI AIConfig
|
||||||
Admin AdminConfig
|
Admin AdminConfig
|
||||||
Qiniu QiniuConfig
|
OSS OSSConfig
|
||||||
WeChatOA WeChatOfficialConfig
|
WeChatOA WeChatOfficialConfig
|
||||||
Redis RedisConfig
|
Redis RedisConfig
|
||||||
}
|
}
|
||||||
@@ -71,9 +71,9 @@ type AdminConfig struct {
|
|||||||
DefaultPassword string
|
DefaultPassword string
|
||||||
}
|
}
|
||||||
|
|
||||||
// QiniuConfig 用于七牛云(Kodo)直传相关配置。
|
// OSSConfig 用于阿里云 OSS 直传相关配置。
|
||||||
// 前端通常会向后端请求 upload token,然后直传文件到七牛。
|
// 前端通常会向后端请求上传凭证,然后直传文件到 OSS。
|
||||||
type QiniuConfig struct {
|
type OSSConfig struct {
|
||||||
AccessKey string
|
AccessKey string
|
||||||
SecretKey string
|
SecretKey string
|
||||||
Bucket string
|
Bucket string
|
||||||
@@ -151,17 +151,17 @@ func LoadConfig() {
|
|||||||
DefaultUsername: getEnv("ADMIN_DEFAULT_USERNAME", "admin"),
|
DefaultUsername: getEnv("ADMIN_DEFAULT_USERNAME", "admin"),
|
||||||
DefaultPassword: getEnv("ADMIN_DEFAULT_PASSWORD", "admin123"),
|
DefaultPassword: getEnv("ADMIN_DEFAULT_PASSWORD", "admin123"),
|
||||||
},
|
},
|
||||||
Qiniu: QiniuConfig{
|
OSS: OSSConfig{
|
||||||
AccessKey: getEnv("QINIU_ACCESS_KEY", ""),
|
AccessKey: getEnv("OSS_ACCESS_KEY", ""),
|
||||||
SecretKey: getEnv("QINIU_SECRET_KEY", ""),
|
SecretKey: getEnv("OSS_SECRET_KEY", ""),
|
||||||
Bucket: getEnv("QINIU_BUCKET", ""),
|
Bucket: getEnv("OSS_BUCKET", ""),
|
||||||
UploadURL: getEnv("QINIU_UPLOAD_URL", "https://upload.qiniup.com"),
|
UploadURL: getEnv("OSS_UPLOAD_URL", ""),
|
||||||
CDNDomain: getEnv("QINIU_CDN_DOMAIN", ""),
|
CDNDomain: getEnv("OSS_CDN_DOMAIN", ""),
|
||||||
KeyPrefix: getEnv("QINIU_KEY_PREFIX", "uploads/"),
|
KeyPrefix: getEnv("OSS_KEY_PREFIX", "uploads/"),
|
||||||
TokenExpireSeconds: getEnvAsInt("QINIU_TOKEN_EXPIRE_SECONDS", 300),
|
TokenExpireSeconds: getEnvAsInt("OSS_TOKEN_EXPIRE_SECONDS", 300),
|
||||||
CallbackURL: getEnv("QINIU_CALLBACK_URL", ""),
|
CallbackURL: getEnv("OSS_CALLBACK_URL", ""),
|
||||||
CallbackBody: getEnv("QINIU_CALLBACK_BODY", "key=$(key)&hash=$(etag)&fsize=$(fsize)&mimeType=$(mimeType)"),
|
CallbackBody: getEnv("OSS_CALLBACK_BODY", "key=$(key)&hash=$(etag)&fsize=$(fsize)&mimeType=$(mimeType)"),
|
||||||
CallbackBodyType: getEnv("QINIU_CALLBACK_BODY_TYPE", "application/x-www-form-urlencoded"),
|
CallbackBodyType: getEnv("OSS_CALLBACK_BODY_TYPE", "application/x-www-form-urlencoded"),
|
||||||
},
|
},
|
||||||
WeChatOA: WeChatOfficialConfig{
|
WeChatOA: WeChatOfficialConfig{
|
||||||
AppID: getEnv("WECHAT_OA_APP_ID", ""),
|
AppID: getEnv("WECHAT_OA_APP_ID", ""),
|
||||||
|
|||||||
@@ -23,7 +23,7 @@
|
|||||||
|
|
||||||
除登录接口外,其他接口都需要携带登录后返回的 `session_key`(见:`docs/common/auth.md`)。
|
除登录接口外,其他接口都需要携带登录后返回的 `session_key`(见:`docs/common/auth.md`)。
|
||||||
|
|
||||||
## 上传(七牛直传)
|
## 上传(阿里云 OSS 直传)
|
||||||
|
|
||||||
- `docs/common/upload_qiniu.md`
|
- `docs/common/upload_qiniu.md`
|
||||||
|
|
||||||
@@ -38,3 +38,7 @@
|
|||||||
## 自动化部署(非 Docker)
|
## 自动化部署(非 Docker)
|
||||||
|
|
||||||
- `docs/common/deploy_ci.md`
|
- `docs/common/deploy_ci.md`
|
||||||
|
|
||||||
|
## 后端开发规范
|
||||||
|
|
||||||
|
- `docs/common/backend_convention.md`(编码风格、模块分层、命名约定、接口规范、新增模块流程等)
|
||||||
|
|||||||
@@ -0,0 +1,477 @@
|
|||||||
|
# 后端开发规范 (wx_service)
|
||||||
|
|
||||||
|
本文档定义 `wx_service` 后端项目的编码规范、架构约定和开发流程,适用于所有后端贡献者。
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 1. 技术栈与环境
|
||||||
|
|
||||||
|
| 分类 | 技术 | 最低版本 |
|
||||||
|
|------|------|----------|
|
||||||
|
| 语言 | Go | 1.23 |
|
||||||
|
| Web 框架 | Gin | v1.11 |
|
||||||
|
| ORM | GORM | v1.31 |
|
||||||
|
| 数据库 | MySQL | 8.x |
|
||||||
|
| 缓存 | Redis (go-redis v9) | 可选 |
|
||||||
|
| 鉴权 | JWT (golang-jwt v5) | — |
|
||||||
|
| 配置 | godotenv | — |
|
||||||
|
|
||||||
|
**开发工具要求**:
|
||||||
|
- 使用 `gofmt` / `goimports` 格式化代码
|
||||||
|
- 推荐 `golangci-lint` 进行静态检查
|
||||||
|
- 提交前确保 `go vet ./...` 无警告
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 2. 项目结构
|
||||||
|
|
||||||
|
```
|
||||||
|
wx_service/
|
||||||
|
├── cmd/api/main.go # 程序入口(配置加载 → DB 初始化 → 依赖注入 → 路由注册 → 启动)
|
||||||
|
├── config/ # 配置结构体与加载逻辑
|
||||||
|
├── internal/ # 业务代码(不可被外部包导入)
|
||||||
|
│ ├── admin/ # 管理后台模块
|
||||||
|
│ ├── common/ # 公共能力
|
||||||
|
│ │ ├── auth/ # 登录认证
|
||||||
|
│ │ ├── upload/ # OSS 上传
|
||||||
|
│ │ ├── redis/ # Redis 缓存
|
||||||
|
│ │ └── wechat_official/ # 公众号 OAuth
|
||||||
|
│ ├── database/ # 数据库连接与迁移
|
||||||
|
│ ├── middleware/ # HTTP 中间件
|
||||||
|
│ ├── model/ # 公共数据模型(User、MiniProgram、Response)
|
||||||
|
│ ├── observability/ # 指标采集与日志
|
||||||
|
│ ├── routes/ # 路由注册
|
||||||
|
│ └── <module>/ # 业务模块(按域拆分)
|
||||||
|
├── docs/ # 文档
|
||||||
|
├── deploy/ # 部署配置
|
||||||
|
├── scripts/ # 运维脚本
|
||||||
|
├── .env.example # 环境变量模板
|
||||||
|
├── Dockerfile # 容器镜像构建
|
||||||
|
├── docker-compose.yml # 开发环境依赖
|
||||||
|
└── docker-compose.prod.yml # 生产环境编排
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 3. 模块分层架构
|
||||||
|
|
||||||
|
每个业务模块位于 `internal/<module>/`,采用以下分层结构:
|
||||||
|
|
||||||
|
```
|
||||||
|
internal/<module>/
|
||||||
|
├── handler/ # HTTP 层
|
||||||
|
├── service/ # 业务逻辑层
|
||||||
|
├── model/ # 数据模型
|
||||||
|
└── repository/ # 数据访问层(可选)
|
||||||
|
```
|
||||||
|
|
||||||
|
### 3.1 各层职责
|
||||||
|
|
||||||
|
| 层 | 职责 | 禁止事项 |
|
||||||
|
|----|------|----------|
|
||||||
|
| **handler** | 解析请求参数、调用 service、组装 HTTP 响应 | 不包含业务逻辑、不直接操作数据库 |
|
||||||
|
| **service** | 实现核心业务规则、编排多表操作、调用外部 API | 不感知 HTTP 层(不依赖 `gin.Context` 进行响应) |
|
||||||
|
| **model** | 定义 GORM struct、表名、数据校验 | 不包含业务逻辑 |
|
||||||
|
| **repository** | 封装数据库查询(可选,简单模块可由 service 直接操作 DB) | 不包含业务逻辑 |
|
||||||
|
|
||||||
|
### 3.2 依赖方向
|
||||||
|
|
||||||
|
```
|
||||||
|
handler → service → repository → model
|
||||||
|
→ model
|
||||||
|
```
|
||||||
|
|
||||||
|
handler 依赖 service,service 依赖 repository(或直接使用 `*gorm.DB`),所有层共享 model。禁止反向依赖。
|
||||||
|
|
||||||
|
### 3.3 依赖注入
|
||||||
|
|
||||||
|
采用构造函数注入,在 `cmd/api/main.go` 中完成装配:
|
||||||
|
|
||||||
|
```go
|
||||||
|
// service 接收 *gorm.DB 和配置
|
||||||
|
svc := someservice.NewSomeService(database.DB, config.AppConfig.SomeConfig)
|
||||||
|
|
||||||
|
// handler 接收 service
|
||||||
|
handler := somehandler.NewSomeHandler(svc)
|
||||||
|
|
||||||
|
// 路由注册接收 handler
|
||||||
|
routes.Register(router, handler, ...)
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 4. 命名规范
|
||||||
|
|
||||||
|
### 4.1 文件命名
|
||||||
|
|
||||||
|
- 使用 **snake_case**:`smoke_log_service.go`、`auth_handler.go`
|
||||||
|
- 测试文件以 `_test.go` 结尾:`smoke_log_service_test.go`
|
||||||
|
- 每个文件聚焦单一职责,避免大文件
|
||||||
|
|
||||||
|
### 4.2 包命名
|
||||||
|
|
||||||
|
- 全小写,不使用下划线:`handler`、`service`、`model`
|
||||||
|
- 导入别名使用有意义的缩写:`smokeservice`、`rmhandler`
|
||||||
|
|
||||||
|
### 4.3 结构体与函数
|
||||||
|
|
||||||
|
- 导出类型使用 PascalCase:`SmokeHandler`、`SmokeLogService`
|
||||||
|
- 构造函数统一使用 `New` 前缀:`NewSmokeHandler`
|
||||||
|
- 请求结构体使用小写开头(非导出):`createSmokeLogRequest`
|
||||||
|
- 常量使用 camelCase 或 PascalCase(视作用域定)
|
||||||
|
|
||||||
|
### 4.4 数据库相关
|
||||||
|
|
||||||
|
- 表名使用 **snake_case**:`fa_smoke_log`、`marketing_categories`
|
||||||
|
- 模型通过 `TableName()` 方法显式指定表名
|
||||||
|
- 字段标签使用 `gorm:"column:xxx"` 明确列名
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 5. HTTP 接口规范
|
||||||
|
|
||||||
|
### 5.1 响应格式
|
||||||
|
|
||||||
|
所有接口统一使用 `model.Response` 结构:
|
||||||
|
|
||||||
|
```go
|
||||||
|
// 成功
|
||||||
|
c.JSON(http.StatusOK, model.Success(data))
|
||||||
|
|
||||||
|
// 失败
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "参数错误描述"))
|
||||||
|
```
|
||||||
|
|
||||||
|
响应 JSON 结构:
|
||||||
|
|
||||||
|
```json
|
||||||
|
{
|
||||||
|
"code": 200,
|
||||||
|
"message": "success",
|
||||||
|
"data": {}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 5.2 HTTP 状态码使用
|
||||||
|
|
||||||
|
| 状态码 | 场景 |
|
||||||
|
|--------|------|
|
||||||
|
| 200 | 成功(包括空结果集) |
|
||||||
|
| 400 | 请求参数校验失败 |
|
||||||
|
| 401 | 未登录或 Token 过期 |
|
||||||
|
| 403 | 无权限或业务限制(如额度用尽) |
|
||||||
|
| 404 | 资源不存在 |
|
||||||
|
| 500 | 服务端内部错误 |
|
||||||
|
| 502 | 第三方服务调用失败 |
|
||||||
|
| 503 | 关键配置缺失导致服务不可用 |
|
||||||
|
|
||||||
|
### 5.3 路由规范
|
||||||
|
|
||||||
|
- RESTful 风格:`GET` 获取、`POST` 创建、`PUT` 更新、`DELETE` 删除
|
||||||
|
- 路由前缀:
|
||||||
|
- `/api/v1` — 主版本 API
|
||||||
|
- `/api/v2` — 新版本模块
|
||||||
|
- `/api/expiry` — 保质期独立域
|
||||||
|
- `/api/admin` — 管理后台
|
||||||
|
- 路由注册集中在 `internal/routes/` 目录下
|
||||||
|
- 每个模块创建独立的 `register<Module>Routes` 函数
|
||||||
|
|
||||||
|
### 5.4 认证
|
||||||
|
|
||||||
|
- 用户认证:`Bearer <session_key>`,通过 `AuthMiddleware` + `RequireUserMiddleware` 处理
|
||||||
|
- 管理后台:`X-Admin-Token` 头或 JWT
|
||||||
|
- 公开接口不挂载鉴权中间件
|
||||||
|
- handler 中通过 `middleware.MustCurrentUser(c)` 获取当前用户
|
||||||
|
|
||||||
|
### 5.5 分页
|
||||||
|
|
||||||
|
列表接口统一使用以下参数和返回结构:
|
||||||
|
|
||||||
|
```
|
||||||
|
// 请求参数
|
||||||
|
?page=1&page_size=20
|
||||||
|
|
||||||
|
// 响应
|
||||||
|
{
|
||||||
|
"code": 200,
|
||||||
|
"message": "success",
|
||||||
|
"data": {
|
||||||
|
"items": [...],
|
||||||
|
"total": 100,
|
||||||
|
"page": 1,
|
||||||
|
"page_size": 20
|
||||||
|
}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 6. 数据库规范
|
||||||
|
|
||||||
|
### 6.1 GORM 模型
|
||||||
|
|
||||||
|
```go
|
||||||
|
type SomeModel struct {
|
||||||
|
ID uint `gorm:"primaryKey" json:"id"`
|
||||||
|
UserID uint `gorm:"index;not null" json:"user_id"`
|
||||||
|
Name string `gorm:"size:100;not null" json:"name"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (SomeModel) TableName() string {
|
||||||
|
return "some_models"
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 6.2 迁移
|
||||||
|
|
||||||
|
- 开发阶段使用 `AutoMigrate`,在 `cmd/api/main.go` 中注册新模型
|
||||||
|
- 生产环境建议手动迁移,SQL 参考放在 `docs/sql/` 目录
|
||||||
|
|
||||||
|
### 6.3 查询
|
||||||
|
|
||||||
|
- 优先使用 GORM 链式查询,避免裸 SQL
|
||||||
|
- 复杂报表查询可使用 `db.Raw()`,但需添加注释说明
|
||||||
|
- 列表查询必须加分页限制(`Limit` + `Offset`)
|
||||||
|
- 敏感查询加 `context.Context` 传递超时控制
|
||||||
|
|
||||||
|
### 6.4 事务
|
||||||
|
|
||||||
|
涉及多表写入的操作使用事务:
|
||||||
|
|
||||||
|
```go
|
||||||
|
err := db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
|
||||||
|
// 多步操作
|
||||||
|
return nil
|
||||||
|
})
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 7. 错误处理
|
||||||
|
|
||||||
|
### 7.1 Service 层
|
||||||
|
|
||||||
|
- 定义领域错误变量:`var ErrSmokeLogNotFound = errors.New("smoke log not found")`
|
||||||
|
- Service 返回具体的错误类型,handler 据此决定 HTTP 状态码
|
||||||
|
- 内部错误使用 `fmt.Errorf("xxx: %w", err)` 包装
|
||||||
|
|
||||||
|
### 7.2 Handler 层
|
||||||
|
|
||||||
|
- 用 `errors.Is()` 判断 service 返回的错误类型
|
||||||
|
- 用户可见的错误信息使用中文
|
||||||
|
- 内部错误记录日志后返回通用提示:"xxx失败,请稍后重试"
|
||||||
|
|
||||||
|
```go
|
||||||
|
record, err := h.service.GetByID(ctx, userID, id)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, service.ErrNotFound) {
|
||||||
|
c.JSON(http.StatusNotFound, model.Error(http.StatusNotFound, "记录不存在"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
log.Printf("GetByID failed: %v", err)
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "查询失败,请稍后重试"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 8. 配置管理
|
||||||
|
|
||||||
|
### 8.1 环境变量
|
||||||
|
|
||||||
|
- 所有配置通过 `.env` 文件加载,使用 `godotenv`
|
||||||
|
- 新增配置项必须同步更新 `.env.example`
|
||||||
|
- 敏感信息(密钥、密码)不可提交到代码仓库
|
||||||
|
|
||||||
|
### 8.2 配置结构
|
||||||
|
|
||||||
|
配置统一定义在 `config/config.go` 的结构体中:
|
||||||
|
|
||||||
|
```go
|
||||||
|
type Config struct {
|
||||||
|
Server ServerConfig
|
||||||
|
Database DatabaseConfig
|
||||||
|
JWT JWTConfig
|
||||||
|
// 按需扩展...
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 8.3 可选功能降级
|
||||||
|
|
||||||
|
部分功能(如 Redis、额外数据库)可选启用。启动时检测配置是否存在,缺失则打印日志并禁用相关接口,不阻塞服务启动:
|
||||||
|
|
||||||
|
```go
|
||||||
|
if redisClient == nil {
|
||||||
|
log.Println("redis disabled: ...")
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 9. 测试规范
|
||||||
|
|
||||||
|
### 9.1 文件与组织
|
||||||
|
|
||||||
|
- 测试文件与被测文件同目录:`smoke_log_service.go` → `smoke_log_service_test.go`
|
||||||
|
- 使用 Go 标准测试框架 `testing`
|
||||||
|
- 表驱动测试(table-driven tests)优先
|
||||||
|
|
||||||
|
### 9.2 命名
|
||||||
|
|
||||||
|
- 测试函数:`TestFunctionName_Scenario`
|
||||||
|
- 子测试:`t.Run("scenario description", ...)`
|
||||||
|
|
||||||
|
### 9.3 运行
|
||||||
|
|
||||||
|
```bash
|
||||||
|
# 运行全部测试
|
||||||
|
go test ./...
|
||||||
|
|
||||||
|
# 运行指定模块测试
|
||||||
|
go test ./internal/smoke/service/...
|
||||||
|
|
||||||
|
# 带覆盖率
|
||||||
|
go test -cover ./...
|
||||||
|
```
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 10. 新增业务模块流程
|
||||||
|
|
||||||
|
以新增"xx功能"模块为例:
|
||||||
|
|
||||||
|
### 第一步:创建模块目录
|
||||||
|
|
||||||
|
```
|
||||||
|
internal/xx/
|
||||||
|
├── handler/
|
||||||
|
│ └── xx_handler.go
|
||||||
|
├── service/
|
||||||
|
│ └── xx_service.go
|
||||||
|
└── model/
|
||||||
|
└── xx.go
|
||||||
|
```
|
||||||
|
|
||||||
|
### 第二步:定义数据模型
|
||||||
|
|
||||||
|
```go
|
||||||
|
// internal/xx/model/xx.go
|
||||||
|
package model
|
||||||
|
|
||||||
|
type XX struct {
|
||||||
|
ID uint `gorm:"primaryKey" json:"id"`
|
||||||
|
UserID uint `gorm:"index;not null" json:"user_id"`
|
||||||
|
// ...
|
||||||
|
}
|
||||||
|
|
||||||
|
func (XX) TableName() string { return "xx_records" }
|
||||||
|
```
|
||||||
|
|
||||||
|
### 第三步:实现 Service
|
||||||
|
|
||||||
|
```go
|
||||||
|
// internal/xx/service/xx_service.go
|
||||||
|
package service
|
||||||
|
|
||||||
|
type XXService struct {
|
||||||
|
db *gorm.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewXXService(db *gorm.DB) *XXService {
|
||||||
|
return &XXService{db: db}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 第四步:实现 Handler
|
||||||
|
|
||||||
|
```go
|
||||||
|
// internal/xx/handler/xx_handler.go
|
||||||
|
package handler
|
||||||
|
|
||||||
|
type XXHandler struct {
|
||||||
|
service *xxservice.XXService
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewXXHandler(svc *xxservice.XXService) *XXHandler {
|
||||||
|
return &XXHandler{service: svc}
|
||||||
|
}
|
||||||
|
```
|
||||||
|
|
||||||
|
### 第五步:注册依赖和路由
|
||||||
|
|
||||||
|
1. 在 `cmd/api/main.go` 中:
|
||||||
|
- 导入新模块
|
||||||
|
- 初始化 service → handler
|
||||||
|
- 将新模型加入 `AutoMigrate`
|
||||||
|
2. 在 `internal/routes/` 中新建 `xx_routes.go`,实现 `registerXXRoutes`
|
||||||
|
3. 在 `routes.go` 的 `Register` 函数中调用
|
||||||
|
|
||||||
|
### 第六步:编写文档
|
||||||
|
|
||||||
|
在 `docs/xx/` 下创建:
|
||||||
|
- `README.md`:模块需求说明
|
||||||
|
- `API.md`:接口文档(方法、路径、参数、返回值)
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 11. Git 与提交规范
|
||||||
|
|
||||||
|
### 11.1 分支策略
|
||||||
|
|
||||||
|
- `main` / `master`:生产分支,推送自动触发部署
|
||||||
|
- 功能分支:`feature/<name>` 或 `fix/<name>`
|
||||||
|
|
||||||
|
### 11.2 提交信息格式
|
||||||
|
|
||||||
|
```
|
||||||
|
<type>: <简要描述>
|
||||||
|
|
||||||
|
<可选的详细说明>
|
||||||
|
```
|
||||||
|
|
||||||
|
类型约定:
|
||||||
|
|
||||||
|
| 类型 | 说明 |
|
||||||
|
|------|------|
|
||||||
|
| feat | 新功能 |
|
||||||
|
| fix | 修复 Bug |
|
||||||
|
| refactor | 重构(不影响功能) |
|
||||||
|
| docs | 文档变更 |
|
||||||
|
| test | 测试相关 |
|
||||||
|
| chore | 构建/运维/依赖等杂项 |
|
||||||
|
|
||||||
|
示例:`feat: add quit plan CRUD endpoints`
|
||||||
|
|
||||||
|
### 11.3 提交检查清单
|
||||||
|
|
||||||
|
- [ ] `go vet ./...` 无警告
|
||||||
|
- [ ] `gofmt` / `goimports` 已格式化
|
||||||
|
- [ ] 相关测试通过
|
||||||
|
- [ ] 新增接口已更新文档
|
||||||
|
- [ ] `.env.example` 已同步更新(如有新配置项)
|
||||||
|
|
||||||
|
---
|
||||||
|
|
||||||
|
## 12. 部署
|
||||||
|
|
||||||
|
### Docker 方式
|
||||||
|
|
||||||
|
```bash
|
||||||
|
docker compose -f docker-compose.prod.yml up -d
|
||||||
|
```
|
||||||
|
|
||||||
|
### 非 Docker(GitHub Actions)
|
||||||
|
|
||||||
|
推送 `main` 分支自动触发 `.github/workflows/deploy-prod.yml`。
|
||||||
|
|
||||||
|
### 健康检查
|
||||||
|
|
||||||
|
- `GET /healthz` → `{"status": "ok"}`
|
||||||
|
- Docker 健康检查间隔 30s
|
||||||
|
- Nginx 反代配置在 `deploy/nginx/` 下
|
||||||
|
|
||||||
|
详细部署文档:`docs/common/deploy_ci.md`
|
||||||
+34
-30
@@ -1,19 +1,19 @@
|
|||||||
# 七牛(Kodo)直传:获取上传凭证
|
# 阿里云 OSS 直传:获取上传凭证
|
||||||
|
|
||||||
用途:小程序/前端把文件直接上传到七牛(CDN 源站),后端只负责签发上传 token,减少带宽与压力。
|
用途:小程序/前端把文件直接上传到阿里云 OSS,后端只负责签发上传凭证,减少带宽与压力。
|
||||||
|
|
||||||
## 前置条件
|
## 前置条件
|
||||||
|
|
||||||
- 已完成登录并拿到 `session_key`(见:`docs/common/auth.md`)
|
- 已完成登录并拿到 `session_key`(见:`docs/common/auth.md`)
|
||||||
- 已配置 `.env` 中的七牛参数(见:`.env.example`)
|
- 已配置 `.env` 中的 OSS 参数(见:`.env.example`)
|
||||||
- 若需要上传成功回调,请额外配置:
|
- 若需要上传成功回调,请额外配置:
|
||||||
- `QINIU_CALLBACK_URL`(例如:`https://api.example.com/api/v1/common/upload/qiniu/callback`)
|
- `OSS_CALLBACK_URL`(例如:`https://api.example.com/api/v1/common/upload/oss/callback`)
|
||||||
- `QINIU_CALLBACK_BODY`
|
- `OSS_CALLBACK_BODY`
|
||||||
- `QINIU_CALLBACK_BODY_TYPE`
|
- `OSS_CALLBACK_BODY_TYPE`
|
||||||
|
|
||||||
## 接口
|
## 接口
|
||||||
|
|
||||||
`POST /api/v1/common/upload/qiniu/token`
|
`POST /api/v1/common/upload/oss/token`
|
||||||
|
|
||||||
Header:
|
Header:
|
||||||
|
|
||||||
@@ -33,66 +33,70 @@ Content-Type: application/json
|
|||||||
说明:
|
说明:
|
||||||
- `filename` 仅用于提取文件后缀(例如 `.png`),以便后端生成带后缀的 `key`;不传也可以。
|
- `filename` 仅用于提取文件后缀(例如 `.png`),以便后端生成带后缀的 `key`;不传也可以。
|
||||||
|
|
||||||
成功响应示例:
|
成功响应示例(OSS PostObject 凭证):
|
||||||
|
|
||||||
```json
|
```json
|
||||||
{
|
{
|
||||||
"code": 200,
|
"code": 200,
|
||||||
"message": "success",
|
"message": "success",
|
||||||
"data": {
|
"data": {
|
||||||
"token": "xxx",
|
"key": "uploads/mp_1/user_123/20251231/ab12cd34....png",
|
||||||
"key": "uploads/mp_1/user_123/20251231/ab12cd34ef....png",
|
"upload_url": "https://bucket.oss-cn-beijing.aliyuncs.com",
|
||||||
"upload_url": "https://upload.qiniup.com",
|
"cdn_domain": "https://bucket.oss-cn-beijing.aliyuncs.com",
|
||||||
"expire": 1767150000,
|
"oss_access_key_id": "LTAI5t...",
|
||||||
"cdn_domain": "https://cdn.example.com"
|
"oss_policy": "base64...",
|
||||||
|
"oss_signature": "sig..."
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
```
|
```
|
||||||
|
|
||||||
字段说明:
|
字段说明:
|
||||||
- `token`:七牛上传凭证(uptoken)
|
|
||||||
- `key`:后端生成的对象 key,上传时必须使用该 key
|
- `key`:后端生成的对象 key,上传时必须使用该 key
|
||||||
- `upload_url`:上传入口(表单上传 URL)
|
- `upload_url`:OSS PostObject 上传地址
|
||||||
- `expire`:token 过期时间(Unix 秒)
|
- `cdn_domain`:CDN 访问域名,可用 `cdn_domain + "/" + key` 拼出访问 URL
|
||||||
- `cdn_domain`:可选;如果配置了,可用 `cdn_domain + "/" + key` 拼出访问 URL
|
- `oss_access_key_id`:OSS AccessKeyId
|
||||||
|
- `oss_policy`:Base64 编码的 Policy
|
||||||
|
- `oss_signature`:HMAC-SHA1 签名
|
||||||
|
|
||||||
## 使用示例(curl 直传)
|
## 使用示例(curl 直传)
|
||||||
|
|
||||||
1) 先请求 token:
|
1) 先请求凭证:
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
curl -X POST 'http://127.0.0.1:8080/api/v1/common/upload/qiniu/token' \
|
curl -X POST 'http://127.0.0.1:8080/api/v1/common/upload/oss/token' \
|
||||||
-H 'Content-Type: application/json' \
|
-H 'Content-Type: application/json' \
|
||||||
-H 'Authorization: Bearer wx-session-key' \
|
-H 'Authorization: Bearer wx-session-key' \
|
||||||
-d '{"filename":"avatar.png"}'
|
-d '{"filename":"avatar.png"}'
|
||||||
```
|
```
|
||||||
|
|
||||||
2) 再把文件直传七牛(multipart/form-data):
|
2) 再把文件直传 OSS(multipart/form-data):
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
curl -X POST 'https://upload.qiniup.com' \
|
curl -X POST '<upload_url>' \
|
||||||
-F "token=上一步返回的 token" \
|
-F "key=<上一步返回的 key>" \
|
||||||
-F "key=上一步返回的 key" \
|
-F "policy=<oss_policy>" \
|
||||||
|
-F "OSSAccessKeyId=<oss_access_key_id>" \
|
||||||
|
-F "Signature=<oss_signature>" \
|
||||||
-F "file=@./avatar.png"
|
-F "file=@./avatar.png"
|
||||||
```
|
```
|
||||||
|
|
||||||
七牛成功时会返回 JSON(字段可能因配置不同略有差异),其中一般会包含 `key/hash`。
|
OSS 成功时返回 HTTP 204(无 body)。
|
||||||
|
|
||||||
---
|
---
|
||||||
|
|
||||||
## 上传回调(服务端)
|
## 上传回调(服务端)
|
||||||
|
|
||||||
当配置了 `QINIU_CALLBACK_URL` 后,后端签发的 putPolicy 会包含 callback 参数。七牛上传成功后会回调:
|
当配置了 `OSS_CALLBACK_URL` 后,上传成功后会回调:
|
||||||
|
|
||||||
`POST /api/v1/common/upload/qiniu/callback`
|
`POST /api/v1/common/upload/oss/callback`
|
||||||
|
|
||||||
说明:
|
说明:
|
||||||
- 该接口无需登录(由七牛服务端调用)。
|
- 该接口无需登录(由 OSS 服务端调用)。
|
||||||
- 服务端会校验 `Authorization: QBox ...` 签名。
|
- 服务端会校验 `Authorization` 签名。
|
||||||
- 验签失败返回 `401`,直接拒绝。
|
- 验签失败返回 `401`,直接拒绝。
|
||||||
- 当业务处理发生临时异常时可返回 `503`,利用七牛回调重试机制重试。
|
- 当业务处理发生临时异常时可返回 `503`,利用回调重试机制重试。
|
||||||
|
|
||||||
默认回调体(可通过 `QINIU_CALLBACK_BODY` 调整):
|
默认回调体(可通过 `OSS_CALLBACK_BODY` 调整):
|
||||||
|
|
||||||
```txt
|
```txt
|
||||||
key=$(key)&hash=$(etag)&fsize=$(fsize)&mimeType=$(mimeType)
|
key=$(key)&hash=$(etag)&fsize=$(fsize)&mimeType=$(mimeType)
|
||||||
|
|||||||
@@ -124,7 +124,7 @@ web/marketing/
|
|||||||
## 依赖的公共服务
|
## 依赖的公共服务
|
||||||
|
|
||||||
- **鉴权**:复用 `middleware.AuthMiddleware` + `middleware.RequireUserMiddleware`
|
- **鉴权**:复用 `middleware.AuthMiddleware` + `middleware.RequireUserMiddleware`
|
||||||
- **七牛上传**:模板图片和用户 Logo 均通过七牛直传,复用 `common/qiniu` 模块
|
- **OSS 上传**:模板图片和用户 Logo 均通过阿里云 OSS 直传,复用 `common/upload` 模块
|
||||||
- **Admin Token**:通过 `X-Admin-Token` 请求头鉴权,Token 来自 `.env` 的 `ADMIN_API_TOKEN`
|
- **Admin Token**:通过 `X-Admin-Token` 请求头鉴权,Token 来自 `.env` 的 `ADMIN_API_TOKEN`
|
||||||
|
|
||||||
## Web 管理后台
|
## Web 管理后台
|
||||||
|
|||||||
@@ -1,8 +1,6 @@
|
|||||||
module wx_service
|
module wx_service
|
||||||
|
|
||||||
go 1.23.0
|
go 1.25.0
|
||||||
|
|
||||||
toolchain go1.24.4
|
|
||||||
|
|
||||||
require (
|
require (
|
||||||
github.com/DATA-DOG/go-sqlmock v1.5.2
|
github.com/DATA-DOG/go-sqlmock v1.5.2
|
||||||
@@ -10,7 +8,8 @@ require (
|
|||||||
github.com/golang-jwt/jwt/v5 v5.3.1
|
github.com/golang-jwt/jwt/v5 v5.3.1
|
||||||
github.com/joho/godotenv v1.5.1
|
github.com/joho/godotenv v1.5.1
|
||||||
github.com/redis/go-redis/v9 v9.17.2
|
github.com/redis/go-redis/v9 v9.17.2
|
||||||
golang.org/x/crypto v0.40.0
|
golang.org/x/crypto v0.48.0
|
||||||
|
golang.org/x/image v0.38.0
|
||||||
gorm.io/driver/mysql v1.6.0
|
gorm.io/driver/mysql v1.6.0
|
||||||
gorm.io/driver/sqlite v1.6.0
|
gorm.io/driver/sqlite v1.6.0
|
||||||
gorm.io/gorm v1.31.1
|
gorm.io/gorm v1.31.1
|
||||||
@@ -47,11 +46,11 @@ require (
|
|||||||
github.com/ugorji/go/codec v1.3.0 // indirect
|
github.com/ugorji/go/codec v1.3.0 // indirect
|
||||||
go.uber.org/mock v0.5.0 // indirect
|
go.uber.org/mock v0.5.0 // indirect
|
||||||
golang.org/x/arch v0.20.0 // indirect
|
golang.org/x/arch v0.20.0 // indirect
|
||||||
golang.org/x/mod v0.25.0 // indirect
|
golang.org/x/mod v0.33.0 // indirect
|
||||||
golang.org/x/net v0.42.0 // indirect
|
golang.org/x/net v0.50.0 // indirect
|
||||||
golang.org/x/sync v0.16.0 // indirect
|
golang.org/x/sync v0.20.0 // indirect
|
||||||
golang.org/x/sys v0.35.0 // indirect
|
golang.org/x/sys v0.41.0 // indirect
|
||||||
golang.org/x/text v0.27.0 // indirect
|
golang.org/x/text v0.35.0 // indirect
|
||||||
golang.org/x/tools v0.34.0 // indirect
|
golang.org/x/tools v0.42.0 // indirect
|
||||||
google.golang.org/protobuf v1.36.9 // indirect
|
google.golang.org/protobuf v1.36.9 // indirect
|
||||||
)
|
)
|
||||||
|
|||||||
@@ -92,21 +92,23 @@ go.uber.org/mock v0.5.0 h1:KAMbZvZPyBPWgD14IrIQ38QCyjwpvVVV6K/bHl1IwQU=
|
|||||||
go.uber.org/mock v0.5.0/go.mod h1:ge71pBPLYDk7QIi1LupWxdAykm7KIEFchiOqd6z7qMM=
|
go.uber.org/mock v0.5.0/go.mod h1:ge71pBPLYDk7QIi1LupWxdAykm7KIEFchiOqd6z7qMM=
|
||||||
golang.org/x/arch v0.20.0 h1:dx1zTU0MAE98U+TQ8BLl7XsJbgze2WnNKF/8tGp/Q6c=
|
golang.org/x/arch v0.20.0 h1:dx1zTU0MAE98U+TQ8BLl7XsJbgze2WnNKF/8tGp/Q6c=
|
||||||
golang.org/x/arch v0.20.0/go.mod h1:bdwinDaKcfZUGpH09BB7ZmOfhalA8lQdzl62l8gGWsk=
|
golang.org/x/arch v0.20.0/go.mod h1:bdwinDaKcfZUGpH09BB7ZmOfhalA8lQdzl62l8gGWsk=
|
||||||
golang.org/x/crypto v0.40.0 h1:r4x+VvoG5Fm+eJcxMaY8CQM7Lb0l1lsmjGBQ6s8BfKM=
|
golang.org/x/crypto v0.48.0 h1:/VRzVqiRSggnhY7gNRxPauEQ5Drw9haKdM0jqfcCFts=
|
||||||
golang.org/x/crypto v0.40.0/go.mod h1:Qr1vMER5WyS2dfPHAlsOj01wgLbsyWtFn/aY+5+ZdxY=
|
golang.org/x/crypto v0.48.0/go.mod h1:r0kV5h3qnFPlQnBSrULhlsRfryS2pmewsg+XfMgkVos=
|
||||||
golang.org/x/mod v0.25.0 h1:n7a+ZbQKQA/Ysbyb0/6IbB1H/X41mKgbhfv7AfG/44w=
|
golang.org/x/image v0.38.0 h1:5l+q+Y9JDC7mBOMjo4/aPhMDcxEptsX+Tt3GgRQRPuE=
|
||||||
golang.org/x/mod v0.25.0/go.mod h1:IXM97Txy2VM4PJ3gI61r1YEk/gAj6zAHN3AdZt6S9Ww=
|
golang.org/x/image v0.38.0/go.mod h1:/3f6vaXC+6CEanU4KJxbcUZyEePbyKbaLoDOe4ehFYY=
|
||||||
golang.org/x/net v0.42.0 h1:jzkYrhi3YQWD6MLBJcsklgQsoAcw89EcZbJw8Z614hs=
|
golang.org/x/mod v0.33.0 h1:tHFzIWbBifEmbwtGz65eaWyGiGZatSrT9prnU8DbVL8=
|
||||||
golang.org/x/net v0.42.0/go.mod h1:FF1RA5d3u7nAYA4z2TkclSCKh68eSXtiFwcWQpPXdt8=
|
golang.org/x/mod v0.33.0/go.mod h1:swjeQEj+6r7fODbD2cqrnje9PnziFuw4bmLbBZFrQ5w=
|
||||||
golang.org/x/sync v0.16.0 h1:ycBJEhp9p4vXvUZNszeOq0kGTPghopOL8q0fq3vstxw=
|
golang.org/x/net v0.50.0 h1:ucWh9eiCGyDR3vtzso0WMQinm2Dnt8cFMuQa9K33J60=
|
||||||
golang.org/x/sync v0.16.0/go.mod h1:1dzgHSNfp02xaA81J2MS99Qcpr2w7fw1gpm99rleRqA=
|
golang.org/x/net v0.50.0/go.mod h1:UgoSli3F/pBgdJBHCTc+tp3gmrU4XswgGRgtnwWTfyM=
|
||||||
|
golang.org/x/sync v0.20.0 h1:e0PTpb7pjO8GAtTs2dQ6jYa5BWYlMuX047Dco/pItO4=
|
||||||
|
golang.org/x/sync v0.20.0/go.mod h1:9xrNwdLfx4jkKbNva9FpL6vEN7evnE43NNNJQ2LF3+0=
|
||||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.35.0 h1:vz1N37gP5bs89s7He8XuIYXpyY0+QlsKmzipCbUtyxI=
|
golang.org/x/sys v0.41.0 h1:Ivj+2Cp/ylzLiEU89QhWblYnOE9zerudt9Ftecq2C6k=
|
||||||
golang.org/x/sys v0.35.0/go.mod h1:BJP2sWEmIv4KK5OTEluFJCKSidICx8ciO85XgH3Ak8k=
|
golang.org/x/sys v0.41.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||||
golang.org/x/text v0.27.0 h1:4fGWRpyh641NLlecmyl4LOe6yDdfaYNrGb2zdfo4JV4=
|
golang.org/x/text v0.35.0 h1:JOVx6vVDFokkpaq1AEptVzLTpDe9KGpj5tR4/X+ybL8=
|
||||||
golang.org/x/text v0.27.0/go.mod h1:1D28KMCvyooCX9hBiosv5Tz/+YLxj0j7XhWjpSUF7CU=
|
golang.org/x/text v0.35.0/go.mod h1:khi/HExzZJ2pGnjenulevKNX1W67CUy0AsXcNubPGCA=
|
||||||
golang.org/x/tools v0.34.0 h1:qIpSLOxeCYGg9TrcJokLBG4KFA6d795g0xkBkiESGlo=
|
golang.org/x/tools v0.42.0 h1:uNgphsn75Tdz5Ji2q36v/nsFSfR/9BRFvqhGBaJGd5k=
|
||||||
golang.org/x/tools v0.34.0/go.mod h1:pAP9OwEaY1CAW3HOmg3hLZC5Z0CCmzjAF2UQMSqNARg=
|
golang.org/x/tools v0.42.0/go.mod h1:Ma6lCIwGZvHK6XtgbswSoWroEkhugApmsXyrUmBhfr0=
|
||||||
google.golang.org/protobuf v1.36.9 h1:w2gp2mA27hUeUzj9Ex9FBjsBm40zfaDtEWow293U7Iw=
|
google.golang.org/protobuf v1.36.9 h1:w2gp2mA27hUeUzj9Ex9FBjsBm40zfaDtEWow293U7Iw=
|
||||||
google.golang.org/protobuf v1.36.9/go.mod h1:fuxRtAxBytpl4zzqUh6/eyUujkJdNiuEkXntxiD/uRU=
|
google.golang.org/protobuf v1.36.9/go.mod h1:fuxRtAxBytpl4zzqUh6/eyUujkJdNiuEkXntxiD/uRU=
|
||||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||||
|
|||||||
@@ -0,0 +1,43 @@
|
|||||||
|
package achievement
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
type Theme struct {
|
||||||
|
ID uint `gorm:"primaryKey" json:"id"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
|
||||||
|
|
||||||
|
Name string `gorm:"size:50;not null;comment:主题名称" json:"name"`
|
||||||
|
Key string `gorm:"size:50;uniqueIndex;not null;comment:主题标识" json:"key"`
|
||||||
|
Icon string `gorm:"size:255;comment:主题图标" json:"icon"`
|
||||||
|
SortOrder int `gorm:"default:0;comment:排序" json:"sort_order"`
|
||||||
|
IsActive bool `gorm:"default:true;comment:是否启用" json:"is_active"`
|
||||||
|
|
||||||
|
Levels []Level `gorm:"foreignKey:ThemeID" json:"levels,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (Theme) TableName() string {
|
||||||
|
return "fa_achievement_theme"
|
||||||
|
}
|
||||||
|
|
||||||
|
type Level struct {
|
||||||
|
ID uint `gorm:"primaryKey" json:"id"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
DeletedAt gorm.DeletedAt `gorm:"index" json:"-"`
|
||||||
|
|
||||||
|
ThemeID uint `gorm:"index;not null;comment:主题ID" json:"theme_id"`
|
||||||
|
Name string `gorm:"size:50;not null;comment:等级名称" json:"name"`
|
||||||
|
Icon string `gorm:"size:255;comment:等级图标" json:"icon"`
|
||||||
|
RequiredDays int `gorm:"not null;default:0;comment:所需打卡天数" json:"required_days"`
|
||||||
|
SortOrder int `gorm:"default:0;comment:排序" json:"sort_order"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (Level) TableName() string {
|
||||||
|
return "fa_achievement_level"
|
||||||
|
}
|
||||||
@@ -0,0 +1,187 @@
|
|||||||
|
package achievement
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
var ErrThemeNotFound = errors.New("achievement theme not found")
|
||||||
|
|
||||||
|
type Service struct {
|
||||||
|
db *gorm.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewService(db *gorm.DB) *Service {
|
||||||
|
return &Service{db: db}
|
||||||
|
}
|
||||||
|
|
||||||
|
type UserAchievement struct {
|
||||||
|
ThemeID uint `json:"theme_id"`
|
||||||
|
ThemeName string `json:"theme_name"`
|
||||||
|
ThemeKey string `json:"theme_key"`
|
||||||
|
ThemeIcon string `json:"theme_icon"`
|
||||||
|
Days int `json:"days"`
|
||||||
|
Current *Level `json:"current"`
|
||||||
|
Next *Level `json:"next,omitempty"`
|
||||||
|
Progress float64 `json:"progress"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) ListActiveThemes(ctx context.Context) ([]Theme, error) {
|
||||||
|
var themes []Theme
|
||||||
|
err := s.db.WithContext(ctx).
|
||||||
|
Where("is_active = ?", true).
|
||||||
|
Preload("Levels", func(db *gorm.DB) *gorm.DB {
|
||||||
|
return db.Order("required_days ASC, sort_order ASC")
|
||||||
|
}).
|
||||||
|
Order("sort_order ASC, id ASC").
|
||||||
|
Find(&themes).Error
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("list active themes: %w", err)
|
||||||
|
}
|
||||||
|
return themes, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) GetUserAchievement(ctx context.Context, themeID uint, days int) (*UserAchievement, error) {
|
||||||
|
var theme Theme
|
||||||
|
err := s.db.WithContext(ctx).
|
||||||
|
Preload("Levels", func(db *gorm.DB) *gorm.DB {
|
||||||
|
return db.Order("required_days ASC, sort_order ASC")
|
||||||
|
}).
|
||||||
|
First(&theme, themeID).Error
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||||
|
return nil, ErrThemeNotFound
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("get theme: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if days < 0 {
|
||||||
|
days = 0
|
||||||
|
}
|
||||||
|
|
||||||
|
result := &UserAchievement{
|
||||||
|
ThemeID: theme.ID,
|
||||||
|
ThemeName: theme.Name,
|
||||||
|
ThemeKey: theme.Key,
|
||||||
|
ThemeIcon: theme.Icon,
|
||||||
|
Days: days,
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, level := range theme.Levels {
|
||||||
|
if days >= level.RequiredDays {
|
||||||
|
lvl := level
|
||||||
|
result.Current = &lvl
|
||||||
|
if i+1 < len(theme.Levels) {
|
||||||
|
next := theme.Levels[i+1]
|
||||||
|
result.Next = &next
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if result.Current != nil && result.Next != nil {
|
||||||
|
rangeTotal := result.Next.RequiredDays - result.Current.RequiredDays
|
||||||
|
rangeDone := days - result.Current.RequiredDays
|
||||||
|
if rangeTotal > 0 {
|
||||||
|
result.Progress = float64(rangeDone) / float64(rangeTotal)
|
||||||
|
}
|
||||||
|
} else if result.Current != nil && result.Next == nil {
|
||||||
|
result.Progress = 1.0
|
||||||
|
}
|
||||||
|
|
||||||
|
return result, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) SeedDefaults(ctx context.Context) error {
|
||||||
|
var count int64
|
||||||
|
if err := s.db.WithContext(ctx).Model(&Theme{}).Count(&count).Error; err != nil {
|
||||||
|
return fmt.Errorf("count themes: %w", err)
|
||||||
|
}
|
||||||
|
if count > 0 {
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
themes := []struct {
|
||||||
|
Name string
|
||||||
|
Key string
|
||||||
|
Icon string
|
||||||
|
Levels []struct {
|
||||||
|
Name string
|
||||||
|
RequiredDays int
|
||||||
|
}
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
Name: "修仙", Key: "xiuxian", Icon: "⚔️",
|
||||||
|
Levels: []struct {
|
||||||
|
Name string
|
||||||
|
RequiredDays int
|
||||||
|
}{
|
||||||
|
{"炼体", 0}, {"练气", 3}, {"筑基", 7}, {"金丹", 30}, {"元婴", 90}, {"化神", 365},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "军队", Key: "army", Icon: "🎖️",
|
||||||
|
Levels: []struct {
|
||||||
|
Name string
|
||||||
|
RequiredDays int
|
||||||
|
}{
|
||||||
|
{"新兵", 0}, {"排长", 3}, {"连长", 7}, {"营长", 30}, {"师长", 90}, {"大将", 365},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "魔法", Key: "magic", Icon: "🔮",
|
||||||
|
Levels: []struct {
|
||||||
|
Name string
|
||||||
|
RequiredDays int
|
||||||
|
}{
|
||||||
|
{"学徒", 0}, {"术士", 3}, {"大术士", 7}, {"元素师", 30}, {"大魔导师", 90}, {"圣导师", 365},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "武侠", Key: "wuxia", Icon: "🗡️",
|
||||||
|
Levels: []struct {
|
||||||
|
Name string
|
||||||
|
RequiredDays int
|
||||||
|
}{
|
||||||
|
{"徒弟", 0}, {"掌门弟子", 3}, {"门派高手", 7}, {"大侠", 30}, {"剑仙", 90}, {"武圣", 365},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "电竞", Key: "esports", Icon: "🏆",
|
||||||
|
Levels: []struct {
|
||||||
|
Name string
|
||||||
|
RequiredDays int
|
||||||
|
}{
|
||||||
|
{"青铜", 0}, {"白银", 3}, {"黄金", 7}, {"铂金", 30}, {"钻石", 90}, {"王者", 365},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
for i, t := range themes {
|
||||||
|
theme := Theme{
|
||||||
|
Name: t.Name,
|
||||||
|
Key: t.Key,
|
||||||
|
Icon: t.Icon,
|
||||||
|
SortOrder: i,
|
||||||
|
IsActive: true,
|
||||||
|
}
|
||||||
|
if err := s.db.WithContext(ctx).Create(&theme).Error; err != nil {
|
||||||
|
return fmt.Errorf("seed theme %s: %w", t.Key, err)
|
||||||
|
}
|
||||||
|
for j, l := range t.Levels {
|
||||||
|
level := Level{
|
||||||
|
ThemeID: theme.ID,
|
||||||
|
Name: l.Name,
|
||||||
|
RequiredDays: l.RequiredDays,
|
||||||
|
SortOrder: j,
|
||||||
|
}
|
||||||
|
if err := s.db.WithContext(ctx).Create(&level).Error; err != nil {
|
||||||
|
return fmt.Errorf("seed level %s/%s: %w", t.Key, l.Name, err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,162 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
|
"wx_service/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
type achievementThemeRequest struct {
|
||||||
|
Name string `json:"name"`
|
||||||
|
Key string `json:"key"`
|
||||||
|
Icon string `json:"icon"`
|
||||||
|
SortOrder int `json:"sort_order"`
|
||||||
|
IsActive *bool `json:"is_active"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type achievementLevelRequest struct {
|
||||||
|
ThemeID uint `json:"theme_id"`
|
||||||
|
Name string `json:"name"`
|
||||||
|
Icon string `json:"icon"`
|
||||||
|
RequiredDays int `json:"required_days"`
|
||||||
|
SortOrder int `json:"sort_order"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) ListAchievementThemes(c *gin.Context) {
|
||||||
|
themes, err := h.svc.ListAchievementThemes(c.Request.Context())
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取主题列表失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{"themes": themes}))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) GetAchievementTheme(c *gin.Context) {
|
||||||
|
id, err := parseUintID(c.Param("id"))
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "invalid id"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
theme, err := h.svc.GetAchievementTheme(c.Request.Context(), id)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusNotFound, model.Error(http.StatusNotFound, "主题不存在"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(theme))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) CreateAchievementTheme(c *gin.Context) {
|
||||||
|
var req achievementThemeRequest
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "请求参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if req.Name == "" || req.Key == "" {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "name 和 key 必填"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
theme, err := h.svc.CreateAchievementTheme(c.Request.Context(), req.Name, req.Key, req.Icon, req.SortOrder, req.IsActive)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "创建主题失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(theme))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) UpdateAchievementTheme(c *gin.Context) {
|
||||||
|
id, err := parseUintID(c.Param("id"))
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "invalid id"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var req achievementThemeRequest
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "请求参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
theme, err := h.svc.UpdateAchievementTheme(c.Request.Context(), id, req.Name, req.Key, req.Icon, req.SortOrder, req.IsActive)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "更新主题失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(theme))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) DeleteAchievementTheme(c *gin.Context) {
|
||||||
|
id, err := parseUintID(c.Param("id"))
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "invalid id"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.svc.DeleteAchievementTheme(c.Request.Context(), id); err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "删除主题失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{"deleted": true}))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) ListAchievementLevels(c *gin.Context) {
|
||||||
|
themeID, err := parseUintID(c.Param("id"))
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "invalid theme id"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
levels, err := h.svc.ListAchievementLevels(c.Request.Context(), themeID)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取等级列表失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{"levels": levels}))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) CreateAchievementLevel(c *gin.Context) {
|
||||||
|
var req achievementLevelRequest
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "请求参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if req.Name == "" || req.ThemeID == 0 {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "name 和 theme_id 必填"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
level, err := h.svc.CreateAchievementLevel(c.Request.Context(), req.ThemeID, req.Name, req.Icon, req.RequiredDays, req.SortOrder)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "创建等级失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(level))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) UpdateAchievementLevel(c *gin.Context) {
|
||||||
|
id, err := parseUintID(c.Param("id"))
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "invalid id"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var req achievementLevelRequest
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "请求参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
level, err := h.svc.UpdateAchievementLevel(c.Request.Context(), id, req.Name, req.Icon, req.RequiredDays, req.SortOrder)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "更新等级失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(level))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) DeleteAchievementLevel(c *gin.Context) {
|
||||||
|
id, err := parseUintID(c.Param("id"))
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "invalid id"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.svc.DeleteAchievementLevel(c.Request.Context(), id); err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "删除等级失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{"deleted": true}))
|
||||||
|
}
|
||||||
@@ -0,0 +1,83 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
|
"wx_service/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
type dreamPresetRequest struct {
|
||||||
|
Title string `json:"title"`
|
||||||
|
CoverImage string `json:"cover_image"`
|
||||||
|
SortOrder int `json:"sort_order"`
|
||||||
|
IsActive *bool `json:"is_active"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) ListDreamPresets(c *gin.Context) {
|
||||||
|
presets, err := h.svc.ListDreamPresets(c.Request.Context())
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取预设目标失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{"items": presets}))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) CreateDreamPreset(c *gin.Context) {
|
||||||
|
var req dreamPresetRequest
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if strings.TrimSpace(req.CoverImage) == "" {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "图标不能为空"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
preset, err := h.svc.CreateDreamPreset(c.Request.Context(), req.Title, req.CoverImage, req.SortOrder, req.IsActive)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "创建失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(preset))
|
||||||
|
}
|
||||||
|
|
||||||
|
type dreamPresetUpdateRequest struct {
|
||||||
|
Title *string `json:"title"`
|
||||||
|
CoverImage *string `json:"cover_image"`
|
||||||
|
SortOrder *int `json:"sort_order"`
|
||||||
|
IsActive *bool `json:"is_active"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) UpdateDreamPreset(c *gin.Context) {
|
||||||
|
id, err := parseUintID(c.Param("id"))
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "id 参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
var req dreamPresetUpdateRequest
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
preset, err := h.svc.UpdateDreamPreset(c.Request.Context(), id, req.Title, req.CoverImage, req.SortOrder, req.IsActive)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "更新失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(preset))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *Handler) DeleteDreamPreset(c *gin.Context) {
|
||||||
|
id, err := parseUintID(c.Param("id"))
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "id 参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if err := h.svc.DeleteDreamPreset(c.Request.Context(), id); err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "删除失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{"deleted": true}))
|
||||||
|
}
|
||||||
@@ -0,0 +1,92 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"net/http"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
|
adminservice "wx_service/internal/admin/service"
|
||||||
|
"wx_service/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
type quitDailyListQuery struct {
|
||||||
|
Page int `form:"page"`
|
||||||
|
PageSize int `form:"page_size"`
|
||||||
|
UID int `form:"uid"`
|
||||||
|
DateFrom string `form:"date_from"`
|
||||||
|
DateTo string `form:"date_to"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListQuitDailyStatuses GET /api/admin/quit-checkin/daily-statuses
|
||||||
|
func (h *Handler) ListQuitDailyStatuses(c *gin.Context) {
|
||||||
|
var query quitDailyListQuery
|
||||||
|
if err := c.ShouldBindQuery(&query); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "invalid query"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if query.Page == 0 {
|
||||||
|
query.Page = 1
|
||||||
|
}
|
||||||
|
if query.PageSize == 0 {
|
||||||
|
query.PageSize = 20
|
||||||
|
}
|
||||||
|
|
||||||
|
dateFrom, err := parseDateOnly(query.DateFrom)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "invalid date_from, expected YYYY-MM-DD"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
dateTo, err := parseDateOnly(query.DateTo)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "invalid date_to, expected YYYY-MM-DD"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := h.svc.ListQuitDailyStatuses(c.Request.Context(), adminservice.ListQuitDailyStatusesQuery{
|
||||||
|
Page: query.Page,
|
||||||
|
PageSize: query.PageSize,
|
||||||
|
UID: query.UID,
|
||||||
|
DateFrom: dateFrom,
|
||||||
|
DateTo: dateTo,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "load quit daily statuses failed"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(data))
|
||||||
|
}
|
||||||
|
|
||||||
|
type quitRewardGoalsListQuery struct {
|
||||||
|
Page int `form:"page"`
|
||||||
|
PageSize int `form:"page_size"`
|
||||||
|
UID int `form:"uid"`
|
||||||
|
Status string `form:"status"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListQuitRewardGoals GET /api/admin/quit-checkin/reward-goals
|
||||||
|
func (h *Handler) ListQuitRewardGoals(c *gin.Context) {
|
||||||
|
var query quitRewardGoalsListQuery
|
||||||
|
if err := c.ShouldBindQuery(&query); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "invalid query"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if query.Page == 0 {
|
||||||
|
query.Page = 1
|
||||||
|
}
|
||||||
|
if query.PageSize == 0 {
|
||||||
|
query.PageSize = 20
|
||||||
|
}
|
||||||
|
|
||||||
|
data, err := h.svc.ListQuitRewardGoals(c.Request.Context(), adminservice.ListQuitRewardGoalsQuery{
|
||||||
|
Page: query.Page,
|
||||||
|
PageSize: query.PageSize,
|
||||||
|
UID: query.UID,
|
||||||
|
Status: strings.TrimSpace(query.Status),
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "load reward goals failed"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(data))
|
||||||
|
}
|
||||||
@@ -0,0 +1,127 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"gorm.io/gorm"
|
||||||
|
|
||||||
|
"wx_service/internal/achievement"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (s *Service) ListAchievementThemes(ctx context.Context) ([]achievement.Theme, error) {
|
||||||
|
var themes []achievement.Theme
|
||||||
|
err := s.db.WithContext(ctx).
|
||||||
|
Preload("Levels", func(db *gorm.DB) *gorm.DB {
|
||||||
|
return db.Order("required_days ASC, sort_order ASC")
|
||||||
|
}).
|
||||||
|
Order("sort_order ASC, id ASC").
|
||||||
|
Find(&themes).Error
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("list themes: %w", err)
|
||||||
|
}
|
||||||
|
return themes, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) GetAchievementTheme(ctx context.Context, id uint) (*achievement.Theme, error) {
|
||||||
|
var theme achievement.Theme
|
||||||
|
err := s.db.WithContext(ctx).
|
||||||
|
Preload("Levels").
|
||||||
|
First(&theme, id).Error
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &theme, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) CreateAchievementTheme(ctx context.Context, name, key, icon string, sortOrder int, isActive *bool) (*achievement.Theme, error) {
|
||||||
|
active := true
|
||||||
|
if isActive != nil {
|
||||||
|
active = *isActive
|
||||||
|
}
|
||||||
|
theme := achievement.Theme{
|
||||||
|
Name: name,
|
||||||
|
Key: key,
|
||||||
|
Icon: icon,
|
||||||
|
SortOrder: sortOrder,
|
||||||
|
IsActive: active,
|
||||||
|
}
|
||||||
|
if err := s.db.WithContext(ctx).Create(&theme).Error; err != nil {
|
||||||
|
return nil, fmt.Errorf("create theme: %w", err)
|
||||||
|
}
|
||||||
|
return &theme, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) UpdateAchievementTheme(ctx context.Context, id uint, name, key, icon string, sortOrder int, isActive *bool) (*achievement.Theme, error) {
|
||||||
|
var theme achievement.Theme
|
||||||
|
if err := s.db.WithContext(ctx).First(&theme, id).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if name != "" {
|
||||||
|
theme.Name = name
|
||||||
|
}
|
||||||
|
if key != "" {
|
||||||
|
theme.Key = key
|
||||||
|
}
|
||||||
|
if icon != "" {
|
||||||
|
theme.Icon = icon
|
||||||
|
}
|
||||||
|
theme.SortOrder = sortOrder
|
||||||
|
if isActive != nil {
|
||||||
|
theme.IsActive = *isActive
|
||||||
|
}
|
||||||
|
if err := s.db.WithContext(ctx).Save(&theme).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &theme, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) DeleteAchievementTheme(ctx context.Context, id uint) error {
|
||||||
|
return s.db.WithContext(ctx).Delete(&achievement.Theme{}, id).Error
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) ListAchievementLevels(ctx context.Context, themeID uint) ([]achievement.Level, error) {
|
||||||
|
var levels []achievement.Level
|
||||||
|
err := s.db.WithContext(ctx).
|
||||||
|
Where("theme_id = ?", themeID).
|
||||||
|
Order("required_days ASC, sort_order ASC").
|
||||||
|
Find(&levels).Error
|
||||||
|
return levels, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) CreateAchievementLevel(ctx context.Context, themeID uint, name, icon string, requiredDays, sortOrder int) (*achievement.Level, error) {
|
||||||
|
level := achievement.Level{
|
||||||
|
ThemeID: themeID,
|
||||||
|
Name: name,
|
||||||
|
Icon: icon,
|
||||||
|
RequiredDays: requiredDays,
|
||||||
|
SortOrder: sortOrder,
|
||||||
|
}
|
||||||
|
if err := s.db.WithContext(ctx).Create(&level).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &level, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) UpdateAchievementLevel(ctx context.Context, id uint, name, icon string, requiredDays, sortOrder int) (*achievement.Level, error) {
|
||||||
|
var level achievement.Level
|
||||||
|
if err := s.db.WithContext(ctx).First(&level, id).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if name != "" {
|
||||||
|
level.Name = name
|
||||||
|
}
|
||||||
|
if icon != "" {
|
||||||
|
level.Icon = icon
|
||||||
|
}
|
||||||
|
level.RequiredDays = requiredDays
|
||||||
|
level.SortOrder = sortOrder
|
||||||
|
if err := s.db.WithContext(ctx).Save(&level).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &level, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) DeleteAchievementLevel(ctx context.Context, id uint) error {
|
||||||
|
return s.db.WithContext(ctx).Delete(&achievement.Level{}, id).Error
|
||||||
|
}
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
quitmodel "wx_service/internal/quitcheckin/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (s *Service) ListDreamPresets(ctx context.Context) ([]quitmodel.DreamPreset, error) {
|
||||||
|
var presets []quitmodel.DreamPreset
|
||||||
|
err := s.db.WithContext(ctx).
|
||||||
|
Order("sort_order ASC, id ASC").
|
||||||
|
Find(&presets).Error
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("list dream presets: %w", err)
|
||||||
|
}
|
||||||
|
return presets, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) CreateDreamPreset(ctx context.Context, title, coverImage string, sortOrder int, isActive *bool) (*quitmodel.DreamPreset, error) {
|
||||||
|
active := true
|
||||||
|
if isActive != nil {
|
||||||
|
active = *isActive
|
||||||
|
}
|
||||||
|
preset := quitmodel.DreamPreset{
|
||||||
|
Title: title,
|
||||||
|
CoverImage: coverImage,
|
||||||
|
SortOrder: sortOrder,
|
||||||
|
IsActive: active,
|
||||||
|
}
|
||||||
|
if err := s.db.WithContext(ctx).Create(&preset).Error; err != nil {
|
||||||
|
return nil, fmt.Errorf("create dream preset: %w", err)
|
||||||
|
}
|
||||||
|
return &preset, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) UpdateDreamPreset(ctx context.Context, id uint, title, coverImage *string, sortOrder *int, isActive *bool) (*quitmodel.DreamPreset, error) {
|
||||||
|
var preset quitmodel.DreamPreset
|
||||||
|
if err := s.db.WithContext(ctx).First(&preset, id).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if title != nil {
|
||||||
|
preset.Title = *title
|
||||||
|
}
|
||||||
|
if coverImage != nil {
|
||||||
|
preset.CoverImage = *coverImage
|
||||||
|
}
|
||||||
|
if sortOrder != nil {
|
||||||
|
preset.SortOrder = *sortOrder
|
||||||
|
}
|
||||||
|
if isActive != nil {
|
||||||
|
preset.IsActive = *isActive
|
||||||
|
}
|
||||||
|
if err := s.db.WithContext(ctx).Save(&preset).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return &preset, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *Service) DeleteDreamPreset(ctx context.Context, id uint) error {
|
||||||
|
return s.db.WithContext(ctx).Delete(&quitmodel.DreamPreset{}, id).Error
|
||||||
|
}
|
||||||
@@ -0,0 +1,179 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
quitmodel "wx_service/internal/quitcheckin/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
// ListQuitDailyStatusesQuery 戒烟打卡每日状态列表查询。
|
||||||
|
type ListQuitDailyStatusesQuery struct {
|
||||||
|
Page int
|
||||||
|
PageSize int
|
||||||
|
UID int
|
||||||
|
DateFrom *time.Time
|
||||||
|
DateTo *time.Time
|
||||||
|
}
|
||||||
|
|
||||||
|
// QuitDailyStatusItem 管理端展示用(含 uid)。
|
||||||
|
type QuitDailyStatusItem struct {
|
||||||
|
ID int `json:"id"`
|
||||||
|
UID int `json:"uid"`
|
||||||
|
Date string `json:"date"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
CheckInAt *time.Time `json:"check_in_at,omitempty"`
|
||||||
|
RelapsedAt *time.Time `json:"relapsed_at,omitempty"`
|
||||||
|
RelapseNum int `json:"relapse_num"`
|
||||||
|
Reason string `json:"reason,omitempty"`
|
||||||
|
Note string `json:"note,omitempty"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListQuitDailyStatusesResult 分页结果。
|
||||||
|
type ListQuitDailyStatusesResult struct {
|
||||||
|
List []QuitDailyStatusItem `json:"list"`
|
||||||
|
Total int64 `json:"total"`
|
||||||
|
Page int `json:"page"`
|
||||||
|
PageSize int `json:"page_size"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListQuitDailyStatuses 分页查询 fa_quit_checkin_daily_status。
|
||||||
|
func (s *Service) ListQuitDailyStatuses(ctx context.Context, q ListQuitDailyStatusesQuery) (*ListQuitDailyStatusesResult, error) {
|
||||||
|
q.Page, q.PageSize = normalizePage(q.Page, q.PageSize)
|
||||||
|
|
||||||
|
dbQuery := s.db.WithContext(ctx).Model(&quitmodel.DailyStatus{})
|
||||||
|
if q.UID > 0 {
|
||||||
|
dbQuery = dbQuery.Where("uid = ?", q.UID)
|
||||||
|
}
|
||||||
|
if q.DateFrom != nil {
|
||||||
|
dbQuery = dbQuery.Where("date >= ?", q.DateFrom.Format("2006-01-02"))
|
||||||
|
}
|
||||||
|
if q.DateTo != nil {
|
||||||
|
dbQuery = dbQuery.Where("date <= ?", q.DateTo.Format("2006-01-02"))
|
||||||
|
}
|
||||||
|
|
||||||
|
var total int64
|
||||||
|
if err := dbQuery.Count(&total).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var rows []quitmodel.DailyStatus
|
||||||
|
if total > 0 {
|
||||||
|
if err := dbQuery.Order("date DESC, id DESC").
|
||||||
|
Limit(q.PageSize).
|
||||||
|
Offset((q.Page - 1) * q.PageSize).
|
||||||
|
Find(&rows).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
list := make([]QuitDailyStatusItem, 0, len(rows))
|
||||||
|
for _, r := range rows {
|
||||||
|
dateStr := ""
|
||||||
|
if !r.Date.IsZero() {
|
||||||
|
dateStr = r.Date.Format("2006-01-02")
|
||||||
|
}
|
||||||
|
list = append(list, QuitDailyStatusItem{
|
||||||
|
ID: int(r.ID),
|
||||||
|
UID: r.UID,
|
||||||
|
Date: dateStr,
|
||||||
|
Status: r.Status,
|
||||||
|
CheckInAt: r.CheckInAt,
|
||||||
|
RelapsedAt: r.RelapsedAt,
|
||||||
|
RelapseNum: r.RelapseNum,
|
||||||
|
Reason: r.Reason,
|
||||||
|
Note: r.Note,
|
||||||
|
CreatedAt: r.CreatedAt,
|
||||||
|
UpdatedAt: r.UpdatedAt,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return &ListQuitDailyStatusesResult{
|
||||||
|
List: list,
|
||||||
|
Total: total,
|
||||||
|
Page: q.Page,
|
||||||
|
PageSize: q.PageSize,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListQuitRewardGoalsQuery 用户梦想目标列表查询。
|
||||||
|
type ListQuitRewardGoalsQuery struct {
|
||||||
|
Page int
|
||||||
|
PageSize int
|
||||||
|
UID int
|
||||||
|
Status string
|
||||||
|
}
|
||||||
|
|
||||||
|
// QuitRewardGoalItem 管理端展示用。
|
||||||
|
type QuitRewardGoalItem struct {
|
||||||
|
ID int `json:"id"`
|
||||||
|
UID int `json:"uid"`
|
||||||
|
Title string `json:"title"`
|
||||||
|
TargetAmountCent int `json:"target_amount_cent"`
|
||||||
|
CoverImage string `json:"cover_image,omitempty"`
|
||||||
|
Status string `json:"status"`
|
||||||
|
CompletedAt *time.Time `json:"completed_at,omitempty"`
|
||||||
|
CreatedAt time.Time `json:"created_at"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListQuitRewardGoalsResult 分页结果。
|
||||||
|
type ListQuitRewardGoalsResult struct {
|
||||||
|
List []QuitRewardGoalItem `json:"list"`
|
||||||
|
Total int64 `json:"total"`
|
||||||
|
Page int `json:"page"`
|
||||||
|
PageSize int `json:"page_size"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// ListQuitRewardGoals 分页查询 fa_quit_checkin_reward_goal。
|
||||||
|
func (s *Service) ListQuitRewardGoals(ctx context.Context, q ListQuitRewardGoalsQuery) (*ListQuitRewardGoalsResult, error) {
|
||||||
|
q.Page, q.PageSize = normalizePage(q.Page, q.PageSize)
|
||||||
|
|
||||||
|
dbQuery := s.db.WithContext(ctx).Model(&quitmodel.RewardGoal{})
|
||||||
|
if q.UID > 0 {
|
||||||
|
dbQuery = dbQuery.Where("uid = ?", q.UID)
|
||||||
|
}
|
||||||
|
if st := strings.TrimSpace(q.Status); st != "" && st != "all" {
|
||||||
|
dbQuery = dbQuery.Where("status = ?", st)
|
||||||
|
}
|
||||||
|
|
||||||
|
var total int64
|
||||||
|
if err := dbQuery.Count(&total).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
var rows []quitmodel.RewardGoal
|
||||||
|
if total > 0 {
|
||||||
|
if err := dbQuery.Order("id DESC").
|
||||||
|
Limit(q.PageSize).
|
||||||
|
Offset((q.Page - 1) * q.PageSize).
|
||||||
|
Find(&rows).Error; err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
list := make([]QuitRewardGoalItem, 0, len(rows))
|
||||||
|
for _, r := range rows {
|
||||||
|
list = append(list, QuitRewardGoalItem{
|
||||||
|
ID: int(r.ID),
|
||||||
|
UID: r.UID,
|
||||||
|
Title: r.Title,
|
||||||
|
TargetAmountCent: r.TargetAmountCent,
|
||||||
|
CoverImage: r.CoverImage,
|
||||||
|
Status: r.Status,
|
||||||
|
CompletedAt: r.CompletedAt,
|
||||||
|
CreatedAt: r.CreatedAt,
|
||||||
|
UpdatedAt: r.UpdatedAt,
|
||||||
|
})
|
||||||
|
}
|
||||||
|
|
||||||
|
return &ListQuitRewardGoalsResult{
|
||||||
|
List: list,
|
||||||
|
Total: total,
|
||||||
|
Page: q.Page,
|
||||||
|
PageSize: q.PageSize,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
@@ -247,6 +247,7 @@ type ListSmokeProfilesQuery struct {
|
|||||||
type SmokeProfileItem struct {
|
type SmokeProfileItem struct {
|
||||||
ID uint `json:"id"`
|
ID uint `json:"id"`
|
||||||
UID int `json:"uid"`
|
UID int `json:"uid"`
|
||||||
|
Mode string `json:"mode,omitempty"`
|
||||||
BaselineCigsPerDay int `json:"baseline_cigs_per_day"`
|
BaselineCigsPerDay int `json:"baseline_cigs_per_day"`
|
||||||
SmokingYears float64 `json:"smoking_years"`
|
SmokingYears float64 `json:"smoking_years"`
|
||||||
PackPriceCent int `json:"pack_price_cent"`
|
PackPriceCent int `json:"pack_price_cent"`
|
||||||
@@ -255,6 +256,7 @@ type SmokeProfileItem struct {
|
|||||||
WakeUpTime string `json:"wake_up_time"`
|
WakeUpTime string `json:"wake_up_time"`
|
||||||
SleepTime string `json:"sleep_time"`
|
SleepTime string `json:"sleep_time"`
|
||||||
QuitDate *time.Time `json:"quit_date,omitempty"`
|
QuitDate *time.Time `json:"quit_date,omitempty"`
|
||||||
|
AchievementThemeID *uint `json:"achievement_theme_id,omitempty"`
|
||||||
OnboardingCompletedAt *time.Time `json:"onboarding_completed_at,omitempty"`
|
OnboardingCompletedAt *time.Time `json:"onboarding_completed_at,omitempty"`
|
||||||
CreatedAt time.Time `json:"created_at"`
|
CreatedAt time.Time `json:"created_at"`
|
||||||
UpdatedAt time.Time `json:"updated_at"`
|
UpdatedAt time.Time `json:"updated_at"`
|
||||||
@@ -368,6 +370,7 @@ func convertSmokeProfile(row smokemodel.SmokeUserProfile) SmokeProfileItem {
|
|||||||
return SmokeProfileItem{
|
return SmokeProfileItem{
|
||||||
ID: row.ID,
|
ID: row.ID,
|
||||||
UID: row.UID,
|
UID: row.UID,
|
||||||
|
Mode: row.Mode,
|
||||||
BaselineCigsPerDay: row.BaselineCigsPerDay,
|
BaselineCigsPerDay: row.BaselineCigsPerDay,
|
||||||
SmokingYears: row.SmokingYears,
|
SmokingYears: row.SmokingYears,
|
||||||
PackPriceCent: row.PackPriceCent,
|
PackPriceCent: row.PackPriceCent,
|
||||||
@@ -376,6 +379,7 @@ func convertSmokeProfile(row smokemodel.SmokeUserProfile) SmokeProfileItem {
|
|||||||
WakeUpTime: row.WakeUpTime,
|
WakeUpTime: row.WakeUpTime,
|
||||||
SleepTime: row.SleepTime,
|
SleepTime: row.SleepTime,
|
||||||
QuitDate: row.QuitDate,
|
QuitDate: row.QuitDate,
|
||||||
|
AchievementThemeID: row.AchievementThemeID,
|
||||||
OnboardingCompletedAt: row.OnboardingCompletedAt,
|
OnboardingCompletedAt: row.OnboardingCompletedAt,
|
||||||
CreatedAt: row.CreatedAt,
|
CreatedAt: row.CreatedAt,
|
||||||
UpdatedAt: row.UpdatedAt,
|
UpdatedAt: row.UpdatedAt,
|
||||||
|
|||||||
@@ -2,11 +2,13 @@ package handler
|
|||||||
|
|
||||||
import (
|
import (
|
||||||
"errors"
|
"errors"
|
||||||
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
"wx_service/internal/common/auth/service"
|
"wx_service/internal/common/auth/service"
|
||||||
|
"wx_service/internal/middleware"
|
||||||
"wx_service/internal/model"
|
"wx_service/internal/model"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -107,3 +109,100 @@ func (h *AuthHandler) LoginWithWeChat(c *gin.Context) {
|
|||||||
"mini_program": miniProgramPayload,
|
"mini_program": miniProgramPayload,
|
||||||
}))
|
}))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type devLoginRequest struct {
|
||||||
|
MiniProgramID uint `json:"mini_program_id"`
|
||||||
|
}
|
||||||
|
|
||||||
|
// DevLogin 仅在非 release 模式下可用,用于 H5 开发调试。
|
||||||
|
func (h *AuthHandler) DevLogin(c *gin.Context) {
|
||||||
|
if gin.Mode() == gin.ReleaseMode {
|
||||||
|
c.JSON(http.StatusNotFound, model.Error(http.StatusNotFound, "not found"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var req devLoginRequest
|
||||||
|
_ = c.ShouldBindJSON(&req)
|
||||||
|
if req.MiniProgramID == 0 {
|
||||||
|
req.MiniProgramID = 3
|
||||||
|
}
|
||||||
|
|
||||||
|
result, err := h.authService.DevLogin(c.Request.Context(), req.MiniProgramID)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("[dev_login] error: %v", err)
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "dev login failed: "+err.Error()))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{
|
||||||
|
"user": gin.H{
|
||||||
|
"id": result.User.ID,
|
||||||
|
"mini_program_id": result.User.MiniProgramID,
|
||||||
|
"open_id": result.User.OpenID,
|
||||||
|
"nickname": result.User.NickName,
|
||||||
|
"avatar_url": result.User.AvatarURL,
|
||||||
|
},
|
||||||
|
"session_key": result.SessionKey,
|
||||||
|
"mini_program": gin.H{
|
||||||
|
"id": result.MiniProgram.ID,
|
||||||
|
"name": result.MiniProgram.Name,
|
||||||
|
},
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
|
||||||
|
type updateProfileRequest struct {
|
||||||
|
Nickname string `json:"nickname"`
|
||||||
|
AvatarURL string `json:"avatar_url"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type miniProgramCodeQuery struct {
|
||||||
|
Path string `form:"path"`
|
||||||
|
Width int `form:"width"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AuthHandler) UpdateProfile(c *gin.Context) {
|
||||||
|
user := middleware.MustCurrentUser(c)
|
||||||
|
|
||||||
|
var req updateProfileRequest
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if req.Nickname == "" && req.AvatarURL == "" {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "请提供昵称或头像"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
updated, err := h.authService.UpdateProfile(c.Request.Context(), user.ID, req.Nickname, req.AvatarURL)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("[update_profile] error: %v", err)
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "更新失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{
|
||||||
|
"id": updated.ID,
|
||||||
|
"nickname": updated.NickName,
|
||||||
|
"avatar_url": updated.AvatarURL,
|
||||||
|
}))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AuthHandler) GetMiniProgramTestCode(c *gin.Context) {
|
||||||
|
user := middleware.MustCurrentUser(c)
|
||||||
|
|
||||||
|
var query miniProgramCodeQuery
|
||||||
|
if err := c.ShouldBindQuery(&query); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
codeBytes, err := h.authService.GetMiniProgramTestCode(c.Request.Context(), user.ID, query.Path, query.Width)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("[get_mini_program_test_code] error: %v", err)
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取小程序码失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
c.Header("Cache-Control", "no-store")
|
||||||
|
c.Data(http.StatusOK, "image/png", codeBytes)
|
||||||
|
}
|
||||||
|
|||||||
@@ -147,6 +147,76 @@ func (s *AuthService) LoginWithCode(ctx context.Context, req LoginRequest) (*Log
|
|||||||
return result, nil
|
return result, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// DevLogin 开发模式专用:创建或查找测试用户,无需微信授权。
|
||||||
|
func (s *AuthService) DevLogin(ctx context.Context, miniProgramID uint) (*LoginResult, error) {
|
||||||
|
if miniProgramID == 0 {
|
||||||
|
return nil, ErrMiniProgramRequired
|
||||||
|
}
|
||||||
|
|
||||||
|
miniProgram, err := s.miniProgramSvc.GetByID(ctx, miniProgramID)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||||
|
return nil, ErrMiniProgramNotFound
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("load mini program: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
const devOpenID = "dev_test_user"
|
||||||
|
sessionKey := fmt.Sprintf("dev_session_%d", miniProgramID)
|
||||||
|
|
||||||
|
tx := s.db.WithContext(ctx)
|
||||||
|
var user model.User
|
||||||
|
err = tx.Where("mini_program_id = ? AND open_id = ?", miniProgramID, devOpenID).First(&user).Error
|
||||||
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||||
|
user = model.User{
|
||||||
|
MiniProgramID: miniProgramID,
|
||||||
|
OpenID: devOpenID,
|
||||||
|
NickName: "开发测试用户",
|
||||||
|
AvatarURL: defaultAvatarURL,
|
||||||
|
SessionKey: sessionKey,
|
||||||
|
}
|
||||||
|
if err := tx.Create(&user).Error; err != nil {
|
||||||
|
return nil, fmt.Errorf("create dev user: %w", err)
|
||||||
|
}
|
||||||
|
} else if err != nil {
|
||||||
|
return nil, fmt.Errorf("query dev user: %w", err)
|
||||||
|
} else {
|
||||||
|
tx.Model(&user).Update("session_key", sessionKey)
|
||||||
|
user.SessionKey = sessionKey
|
||||||
|
}
|
||||||
|
|
||||||
|
return &LoginResult{
|
||||||
|
User: &user,
|
||||||
|
SessionKey: sessionKey,
|
||||||
|
MiniProgram: miniProgram,
|
||||||
|
}, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// UpdateProfile 更新用户昵称和头像。
|
||||||
|
func (s *AuthService) UpdateProfile(ctx context.Context, userID uint, nickname, avatarURL string) (*model.User, error) {
|
||||||
|
tx := s.db.WithContext(ctx)
|
||||||
|
var user model.User
|
||||||
|
if err := tx.First(&user, userID).Error; err != nil {
|
||||||
|
return nil, fmt.Errorf("find user: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
updates := map[string]interface{}{}
|
||||||
|
if nickname != "" {
|
||||||
|
updates["nick_name"] = nickname
|
||||||
|
user.NickName = nickname
|
||||||
|
}
|
||||||
|
if avatarURL != "" {
|
||||||
|
updates["avatar_url"] = avatarURL
|
||||||
|
user.AvatarURL = avatarURL
|
||||||
|
}
|
||||||
|
if len(updates) > 0 {
|
||||||
|
if err := tx.Model(&user).Updates(updates).Error; err != nil {
|
||||||
|
return nil, fmt.Errorf("update profile: %w", err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
return &user, nil
|
||||||
|
}
|
||||||
|
|
||||||
func (s *AuthService) getSmokeMode(ctx context.Context, uid int) (string, error) {
|
func (s *AuthService) getSmokeMode(ctx context.Context, uid int) (string, error) {
|
||||||
var profile smokemodel.SmokeUserProfile
|
var profile smokemodel.SmokeUserProfile
|
||||||
err := s.db.WithContext(ctx).
|
err := s.db.WithContext(ctx).
|
||||||
@@ -195,3 +265,32 @@ func normalizeAvatarURL(raw string) string {
|
|||||||
}
|
}
|
||||||
return text
|
return text
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (s *AuthService) GetMiniProgramTestCode(ctx context.Context, userID uint, path string, width int) ([]byte, error) {
|
||||||
|
var user model.User
|
||||||
|
if err := s.db.WithContext(ctx).Select("id, mini_program_id").First(&user, userID).Error; err != nil {
|
||||||
|
return nil, fmt.Errorf("find user: %w", err)
|
||||||
|
}
|
||||||
|
if user.MiniProgramID == 0 {
|
||||||
|
return nil, fmt.Errorf("user mini program id missing")
|
||||||
|
}
|
||||||
|
|
||||||
|
miniProgram, err := s.miniProgramSvc.GetByID(ctx, user.MiniProgramID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("load mini program: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if strings.TrimSpace(path) == "" {
|
||||||
|
path = "pages/nsti/test?resume=0"
|
||||||
|
}
|
||||||
|
if width <= 0 {
|
||||||
|
width = 280
|
||||||
|
}
|
||||||
|
|
||||||
|
client := s.getWeChatClient(miniProgram)
|
||||||
|
codeBytes, err := client.GetWXACode(ctx, path, width)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("get mini program test code: %w", err)
|
||||||
|
}
|
||||||
|
return codeBytes, nil
|
||||||
|
}
|
||||||
|
|||||||
@@ -0,0 +1,66 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"context"
|
||||||
|
"testing"
|
||||||
|
|
||||||
|
"gorm.io/driver/sqlite"
|
||||||
|
"gorm.io/gorm"
|
||||||
|
|
||||||
|
"wx_service/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
func newAuthTestDB(t *testing.T) *gorm.DB {
|
||||||
|
t.Helper()
|
||||||
|
|
||||||
|
db, err := gorm.Open(sqlite.Open("file::memory:?cache=shared"), &gorm.Config{})
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("open sqlite: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := db.AutoMigrate(&model.User{}); err != nil {
|
||||||
|
t.Fatalf("auto migrate: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
return db
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestAuthServiceUpdateProfilePersistsUserFields(t *testing.T) {
|
||||||
|
db := newAuthTestDB(t)
|
||||||
|
svc := NewAuthService(db, nil)
|
||||||
|
|
||||||
|
user := model.User{
|
||||||
|
MiniProgramID: 1,
|
||||||
|
OpenID: "openid-1",
|
||||||
|
NickName: "旧昵称",
|
||||||
|
AvatarURL: "https://example.com/old.png",
|
||||||
|
SessionKey: "session-1",
|
||||||
|
}
|
||||||
|
if err := db.Create(&user).Error; err != nil {
|
||||||
|
t.Fatalf("seed user: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
updated, err := svc.UpdateProfile(context.Background(), user.ID, "新昵称", "https://example.com/new.png")
|
||||||
|
if err != nil {
|
||||||
|
t.Fatalf("update profile: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if updated.NickName != "新昵称" {
|
||||||
|
t.Fatalf("expected updated nickname, got %q", updated.NickName)
|
||||||
|
}
|
||||||
|
if updated.AvatarURL != "https://example.com/new.png" {
|
||||||
|
t.Fatalf("expected updated avatar, got %q", updated.AvatarURL)
|
||||||
|
}
|
||||||
|
|
||||||
|
var persisted model.User
|
||||||
|
if err := db.First(&persisted, user.ID).Error; err != nil {
|
||||||
|
t.Fatalf("reload user: %v", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if persisted.NickName != "新昵称" {
|
||||||
|
t.Fatalf("expected persisted nickname, got %q", persisted.NickName)
|
||||||
|
}
|
||||||
|
if persisted.AvatarURL != "https://example.com/new.png" {
|
||||||
|
t.Fatalf("expected persisted avatar, got %q", persisted.AvatarURL)
|
||||||
|
}
|
||||||
|
}
|
||||||
@@ -1,15 +1,19 @@
|
|||||||
package service
|
package service
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"context"
|
"context"
|
||||||
"encoding/json"
|
"encoding/json"
|
||||||
"fmt"
|
"fmt"
|
||||||
|
"io"
|
||||||
"net/http"
|
"net/http"
|
||||||
"net/url"
|
"net/url"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
const weChatCode2SessionURL = "https://api.weixin.qq.com/sns/jscode2session"
|
const weChatCode2SessionURL = "https://api.weixin.qq.com/sns/jscode2session"
|
||||||
|
const weChatAccessTokenURL = "https://api.weixin.qq.com/cgi-bin/token"
|
||||||
|
const weChatGetWXACodeURL = "https://api.weixin.qq.com/wxa/getwxacode"
|
||||||
|
|
||||||
// WeChatClient 调用微信接口获取 session/openid。
|
// WeChatClient 调用微信接口获取 session/openid。
|
||||||
type WeChatClient struct {
|
type WeChatClient struct {
|
||||||
@@ -30,6 +34,13 @@ type weChatSessionResponse struct {
|
|||||||
ErrMsg string `json:"errmsg"`
|
ErrMsg string `json:"errmsg"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
type weChatAccessTokenResponse struct {
|
||||||
|
AccessToken string `json:"access_token"`
|
||||||
|
ExpiresIn int `json:"expires_in"`
|
||||||
|
ErrCode int `json:"errcode"`
|
||||||
|
ErrMsg string `json:"errmsg"`
|
||||||
|
}
|
||||||
|
|
||||||
// WeChatError 表示微信接口级错误。
|
// WeChatError 表示微信接口级错误。
|
||||||
type WeChatError struct {
|
type WeChatError struct {
|
||||||
Code int
|
Code int
|
||||||
@@ -87,3 +98,98 @@ func (c *WeChatClient) Code2Session(ctx context.Context, code string) (*WeChatSe
|
|||||||
|
|
||||||
return &raw.WeChatSession, nil
|
return &raw.WeChatSession, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (c *WeChatClient) GetAccessToken(ctx context.Context) (string, error) {
|
||||||
|
query := url.Values{}
|
||||||
|
query.Set("grant_type", "client_credential")
|
||||||
|
query.Set("appid", c.appID)
|
||||||
|
query.Set("secret", c.appSecret)
|
||||||
|
|
||||||
|
req, err := http.NewRequestWithContext(ctx, http.MethodGet, fmt.Sprintf("%s?%s", weChatAccessTokenURL, query.Encode()), nil)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("build access token request: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
resp, err := c.client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("call wechat access token api: %w", err)
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
return "", fmt.Errorf("wechat access token api unexpected status: %s", resp.Status)
|
||||||
|
}
|
||||||
|
|
||||||
|
var raw weChatAccessTokenResponse
|
||||||
|
if err := json.NewDecoder(resp.Body).Decode(&raw); err != nil {
|
||||||
|
return "", fmt.Errorf("decode access token response: %w", err)
|
||||||
|
}
|
||||||
|
if raw.ErrCode != 0 {
|
||||||
|
return "", &WeChatError{Code: raw.ErrCode, Msg: raw.ErrMsg}
|
||||||
|
}
|
||||||
|
if raw.AccessToken == "" {
|
||||||
|
return "", fmt.Errorf("wechat access token missing")
|
||||||
|
}
|
||||||
|
return raw.AccessToken, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
type getWXACodeRequest struct {
|
||||||
|
Path string `json:"path"`
|
||||||
|
Width int `json:"width,omitempty"`
|
||||||
|
AutoColor bool `json:"auto_color"`
|
||||||
|
IsHyaline bool `json:"is_hyaline"`
|
||||||
|
}
|
||||||
|
|
||||||
|
type weChatAPIErrorResponse struct {
|
||||||
|
ErrCode int `json:"errcode"`
|
||||||
|
ErrMsg string `json:"errmsg"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c *WeChatClient) GetWXACode(ctx context.Context, path string, width int) ([]byte, error) {
|
||||||
|
accessToken, err := c.GetAccessToken(ctx)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
body, err := json.Marshal(getWXACodeRequest{
|
||||||
|
Path: path,
|
||||||
|
Width: width,
|
||||||
|
AutoColor: false,
|
||||||
|
IsHyaline: true,
|
||||||
|
})
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("marshal getwxacode request: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
req, err := http.NewRequestWithContext(
|
||||||
|
ctx,
|
||||||
|
http.MethodPost,
|
||||||
|
fmt.Sprintf("%s?access_token=%s", weChatGetWXACodeURL, url.QueryEscape(accessToken)),
|
||||||
|
bytes.NewReader(body),
|
||||||
|
)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("build getwxacode request: %w", err)
|
||||||
|
}
|
||||||
|
req.Header.Set("Content-Type", "application/json")
|
||||||
|
|
||||||
|
resp, err := c.client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("call wechat getwxacode api: %w", err)
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
payload, err := io.ReadAll(resp.Body)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("read getwxacode response: %w", err)
|
||||||
|
}
|
||||||
|
if resp.StatusCode != http.StatusOK {
|
||||||
|
return nil, fmt.Errorf("wechat getwxacode api unexpected status: %s", resp.Status)
|
||||||
|
}
|
||||||
|
|
||||||
|
var apiErr weChatAPIErrorResponse
|
||||||
|
if err := json.Unmarshal(payload, &apiErr); err == nil && apiErr.ErrCode != 0 {
|
||||||
|
return nil, &WeChatError{Code: apiErr.ErrCode, Msg: apiErr.ErrMsg}
|
||||||
|
}
|
||||||
|
|
||||||
|
return payload, nil
|
||||||
|
}
|
||||||
|
|||||||
@@ -0,0 +1,72 @@
|
|||||||
|
package imageutil
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"fmt"
|
||||||
|
"image"
|
||||||
|
"image/jpeg"
|
||||||
|
"image/png"
|
||||||
|
"io"
|
||||||
|
"strings"
|
||||||
|
|
||||||
|
"golang.org/x/image/draw"
|
||||||
|
)
|
||||||
|
|
||||||
|
type ResizeOptions struct {
|
||||||
|
MaxWidth int
|
||||||
|
Quality int // JPEG quality 1-100
|
||||||
|
}
|
||||||
|
|
||||||
|
func DefaultResizeOptions() ResizeOptions {
|
||||||
|
return ResizeOptions{MaxWidth: 800, Quality: 80}
|
||||||
|
}
|
||||||
|
|
||||||
|
// GenerateThumbnail reads image data, resizes if wider than MaxWidth, and
|
||||||
|
// re-encodes as JPEG. Returns the thumbnail bytes and whether the image
|
||||||
|
// was actually resized (false if already small enough but still re-encoded).
|
||||||
|
func GenerateThumbnail(data []byte, contentType string, opts ResizeOptions) ([]byte, error) {
|
||||||
|
if opts.MaxWidth <= 0 {
|
||||||
|
opts.MaxWidth = 800
|
||||||
|
}
|
||||||
|
if opts.Quality <= 0 || opts.Quality > 100 {
|
||||||
|
opts.Quality = 80
|
||||||
|
}
|
||||||
|
|
||||||
|
src, err := decodeImage(bytes.NewReader(data), contentType)
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("decode image: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
bounds := src.Bounds()
|
||||||
|
srcW := bounds.Dx()
|
||||||
|
srcH := bounds.Dy()
|
||||||
|
|
||||||
|
dstW := srcW
|
||||||
|
dstH := srcH
|
||||||
|
if srcW > opts.MaxWidth {
|
||||||
|
dstW = opts.MaxWidth
|
||||||
|
dstH = srcH * opts.MaxWidth / srcW
|
||||||
|
}
|
||||||
|
|
||||||
|
dst := image.NewRGBA(image.Rect(0, 0, dstW, dstH))
|
||||||
|
draw.BiLinear.Scale(dst, dst.Bounds(), src, bounds, draw.Over, nil)
|
||||||
|
|
||||||
|
var buf bytes.Buffer
|
||||||
|
if err := jpeg.Encode(&buf, dst, &jpeg.Options{Quality: opts.Quality}); err != nil {
|
||||||
|
return nil, fmt.Errorf("encode jpeg: %w", err)
|
||||||
|
}
|
||||||
|
return buf.Bytes(), nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func decodeImage(r io.Reader, contentType string) (image.Image, error) {
|
||||||
|
ct := strings.ToLower(contentType)
|
||||||
|
switch {
|
||||||
|
case strings.Contains(ct, "png"):
|
||||||
|
return png.Decode(r)
|
||||||
|
case strings.Contains(ct, "jpeg"), strings.Contains(ct, "jpg"):
|
||||||
|
return jpeg.Decode(r)
|
||||||
|
default:
|
||||||
|
img, _, err := image.Decode(r)
|
||||||
|
return img, err
|
||||||
|
}
|
||||||
|
}
|
||||||
+25
-31
@@ -18,25 +18,24 @@ import (
|
|||||||
|
|
||||||
"wx_service/config"
|
"wx_service/config"
|
||||||
oss "wx_service/internal/common/oss"
|
oss "wx_service/internal/common/oss"
|
||||||
qiniuservice "wx_service/internal/common/qiniu/service"
|
uploadservice "wx_service/internal/common/upload/service"
|
||||||
"wx_service/internal/middleware"
|
"wx_service/internal/middleware"
|
||||||
"wx_service/internal/model"
|
"wx_service/internal/model"
|
||||||
)
|
)
|
||||||
|
|
||||||
type UploadHandler struct {
|
type UploadHandler struct {
|
||||||
qiniuService *qiniuservice.QiniuService
|
uploadService *uploadservice.UploadService
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewUploadHandler(qiniuService *qiniuservice.QiniuService) *UploadHandler {
|
func NewUploadHandler(uploadService *uploadservice.UploadService) *UploadHandler {
|
||||||
return &UploadHandler{qiniuService: qiniuService}
|
return &UploadHandler{uploadService: uploadService}
|
||||||
}
|
}
|
||||||
|
|
||||||
type qiniuTokenRequest struct {
|
type uploadTokenRequest struct {
|
||||||
// filename 用于保留文件后缀(可选),例如:"a.png"、"video.mp4"
|
|
||||||
Filename string `json:"filename"`
|
Filename string `json:"filename"`
|
||||||
}
|
}
|
||||||
|
|
||||||
type qiniuCallbackPayload struct {
|
type callbackPayload struct {
|
||||||
Key string `json:"key"`
|
Key string `json:"key"`
|
||||||
Hash string `json:"hash"`
|
Hash string `json:"hash"`
|
||||||
Fsize int64 `json:"fsize"`
|
Fsize int64 `json:"fsize"`
|
||||||
@@ -56,15 +55,14 @@ type uploadTokenResponse struct {
|
|||||||
|
|
||||||
var extPattern = regexp.MustCompile(`^\.[a-z0-9]{1,10}$`)
|
var extPattern = regexp.MustCompile(`^\.[a-z0-9]{1,10}$`)
|
||||||
|
|
||||||
// QiniuToken 返回直传所需的 token/key/upload_url;CDN 为阿里云 OSS 时返回 OSS PostObject 凭证。
|
// GetUploadToken 返回直传所需的凭证;CDN 为阿里云 OSS 时返回 OSS PostObject 凭证。
|
||||||
// 建议放在鉴权后:用当前登录用户生成 key,避免前端写入任意路径。
|
func (h *UploadHandler) GetUploadToken(c *gin.Context) {
|
||||||
func (h *UploadHandler) QiniuToken(c *gin.Context) {
|
|
||||||
user := middleware.MustCurrentUser(c)
|
user := middleware.MustCurrentUser(c)
|
||||||
|
|
||||||
var req qiniuTokenRequest
|
var req uploadTokenRequest
|
||||||
_ = c.ShouldBindJSON(&req)
|
_ = c.ShouldBindJSON(&req)
|
||||||
|
|
||||||
cfg := config.AppConfig.Qiniu
|
cfg := config.AppConfig.OSS
|
||||||
cdnDomain := strings.TrimSpace(cfg.CDNDomain)
|
cdnDomain := strings.TrimSpace(cfg.CDNDomain)
|
||||||
|
|
||||||
if oss.IsOSSDomain(cdnDomain) && cfg.AccessKey != "" && cfg.SecretKey != "" && cfg.Bucket != "" {
|
if oss.IsOSSDomain(cdnDomain) && cfg.AccessKey != "" && cfg.SecretKey != "" && cfg.Bucket != "" {
|
||||||
@@ -99,10 +97,10 @@ func (h *UploadHandler) QiniuToken(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
token, err := h.qiniuService.CreateUploadToken(user.MiniProgramID, user.ID, req.Filename)
|
token, err := h.uploadService.CreateUploadToken(user.MiniProgramID, user.ID, req.Filename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if errors.Is(err, qiniuservice.ErrQiniuNotConfigured) {
|
if errors.Is(err, uploadservice.ErrUploadNotConfigured) {
|
||||||
c.JSON(http.StatusServiceUnavailable, model.Error(http.StatusServiceUnavailable, "未配置七牛上传服务,请联系管理员"))
|
c.JSON(http.StatusServiceUnavailable, model.Error(http.StatusServiceUnavailable, "未配置上传服务,请联系管理员"))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取上传凭证失败,请稍后重试"))
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取上传凭证失败,请稍后重试"))
|
||||||
@@ -111,28 +109,25 @@ func (h *UploadHandler) QiniuToken(c *gin.Context) {
|
|||||||
c.JSON(http.StatusOK, model.Success(token))
|
c.JSON(http.StatusOK, model.Success(token))
|
||||||
}
|
}
|
||||||
|
|
||||||
// QiniuCallback 处理七牛上传回调(无需登录),通过签名验签确保来源可信。
|
// UploadCallback 处理上传回调(无需登录),通过签名验签确保来源可信。
|
||||||
// 说明:
|
func (h *UploadHandler) UploadCallback(c *gin.Context) {
|
||||||
// - 验签失败返回 401(非可信请求,直接拒绝)
|
|
||||||
// - 业务处理临时失败返回 503(触发七牛重试)
|
|
||||||
func (h *UploadHandler) QiniuCallback(c *gin.Context) {
|
|
||||||
rawBody, err := io.ReadAll(c.Request.Body)
|
rawBody, err := io.ReadAll(c.Request.Body)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "读取回调内容失败"))
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "读取回调内容失败"))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := h.qiniuService.VerifyCallbackSignature(c.Request, rawBody); err != nil {
|
if err := h.uploadService.VerifyCallbackSignature(c.Request, rawBody); err != nil {
|
||||||
switch {
|
switch {
|
||||||
case errors.Is(err, qiniuservice.ErrQiniuNotConfigured):
|
case errors.Is(err, uploadservice.ErrUploadNotConfigured):
|
||||||
c.JSON(http.StatusServiceUnavailable, model.Error(http.StatusServiceUnavailable, "七牛服务未配置"))
|
c.JSON(http.StatusServiceUnavailable, model.Error(http.StatusServiceUnavailable, "上传服务未配置"))
|
||||||
default:
|
default:
|
||||||
c.JSON(http.StatusUnauthorized, model.Error(http.StatusUnauthorized, "回调验签失败"))
|
c.JSON(http.StatusUnauthorized, model.Error(http.StatusUnauthorized, "回调验签失败"))
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
payload, err := parseQiniuCallbackPayload(c.ContentType(), rawBody)
|
payload, err := parseCallbackPayload(c.ContentType(), rawBody)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "回调内容格式错误"))
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "回调内容格式错误"))
|
||||||
return
|
return
|
||||||
@@ -142,24 +137,23 @@ func (h *UploadHandler) QiniuCallback(c *gin.Context) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 当前阶段先记录日志。后续如接入 DB/任务系统,处理失败可保持 503 以触发七牛重试。
|
log.Printf("[upload_callback] key=%s hash=%s fsize=%d mimeType=%s", payload.Key, payload.Hash, payload.Fsize, payload.MimeType)
|
||||||
log.Printf("[qiniu_callback] key=%s hash=%s fsize=%d mimeType=%s", payload.Key, payload.Hash, payload.Fsize, payload.MimeType)
|
|
||||||
c.JSON(http.StatusOK, model.Success(gin.H{"ok": true}))
|
c.JSON(http.StatusOK, model.Success(gin.H{"ok": true}))
|
||||||
}
|
}
|
||||||
|
|
||||||
func parseQiniuCallbackPayload(contentType string, raw []byte) (qiniuCallbackPayload, error) {
|
func parseCallbackPayload(contentType string, raw []byte) (callbackPayload, error) {
|
||||||
var payload qiniuCallbackPayload
|
var payload callbackPayload
|
||||||
trimmed := strings.TrimSpace(strings.ToLower(contentType))
|
trimmed := strings.TrimSpace(strings.ToLower(contentType))
|
||||||
if strings.Contains(trimmed, "application/json") {
|
if strings.Contains(trimmed, "application/json") {
|
||||||
if err := json.Unmarshal(raw, &payload); err != nil {
|
if err := json.Unmarshal(raw, &payload); err != nil {
|
||||||
return qiniuCallbackPayload{}, err
|
return callbackPayload{}, err
|
||||||
}
|
}
|
||||||
return payload, nil
|
return payload, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
values, err := url.ParseQuery(string(raw))
|
values, err := url.ParseQuery(string(raw))
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return qiniuCallbackPayload{}, err
|
return callbackPayload{}, err
|
||||||
}
|
}
|
||||||
payload.Key = strings.TrimSpace(values.Get("key"))
|
payload.Key = strings.TrimSpace(values.Get("key"))
|
||||||
payload.Hash = strings.TrimSpace(values.Get("hash"))
|
payload.Hash = strings.TrimSpace(values.Get("hash"))
|
||||||
@@ -167,7 +161,7 @@ func parseQiniuCallbackPayload(contentType string, raw []byte) (qiniuCallbackPay
|
|||||||
if rawFsize := strings.TrimSpace(values.Get("fsize")); rawFsize != "" {
|
if rawFsize := strings.TrimSpace(values.Get("fsize")); rawFsize != "" {
|
||||||
fsize, parseErr := strconv.ParseInt(rawFsize, 10, 64)
|
fsize, parseErr := strconv.ParseInt(rawFsize, 10, 64)
|
||||||
if parseErr != nil {
|
if parseErr != nil {
|
||||||
return qiniuCallbackPayload{}, parseErr
|
return callbackPayload{}, parseErr
|
||||||
}
|
}
|
||||||
payload.Fsize = fsize
|
payload.Fsize = fsize
|
||||||
}
|
}
|
||||||
+22
-22
@@ -12,25 +12,25 @@ import (
|
|||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
"wx_service/config"
|
"wx_service/config"
|
||||||
qiniuservice "wx_service/internal/common/qiniu/service"
|
uploadservice "wx_service/internal/common/upload/service"
|
||||||
)
|
)
|
||||||
|
|
||||||
func TestQiniuCallbackSuccess(t *testing.T) {
|
func TestUploadCallbackSuccess(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
gin.SetMode(gin.TestMode)
|
gin.SetMode(gin.TestMode)
|
||||||
|
|
||||||
cfg := config.QiniuConfig{AccessKey: "ak-test", SecretKey: "sk-test"}
|
cfg := config.OSSConfig{AccessKey: "ak-test", SecretKey: "sk-test"}
|
||||||
h := NewUploadHandler(qiniuservice.NewQiniuService(cfg))
|
h := NewUploadHandler(uploadservice.NewUploadService(cfg))
|
||||||
|
|
||||||
body := "key=uploads/test.png&hash=abc&fsize=12&mimeType=image%2Fpng"
|
body := "key=uploads/test.png&hash=abc&fsize=12&mimeType=image%2Fpng"
|
||||||
req := httptest.NewRequest(http.MethodPost, "/api/v1/common/upload/qiniu/callback", strings.NewReader(body))
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/common/upload/oss/callback", strings.NewReader(body))
|
||||||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||||
req.Header.Set("Authorization", "QBox "+cfg.AccessKey+":"+signQiniu(req.URL.Path+"\n"+body, cfg.SecretKey))
|
req.Header.Set("Authorization", "QBox "+cfg.AccessKey+":"+signCallback(req.URL.Path+"\n"+body, cfg.SecretKey))
|
||||||
w := httptest.NewRecorder()
|
w := httptest.NewRecorder()
|
||||||
c, _ := gin.CreateTestContext(w)
|
c, _ := gin.CreateTestContext(w)
|
||||||
c.Request = req
|
c.Request = req
|
||||||
|
|
||||||
h.QiniuCallback(c)
|
h.UploadCallback(c)
|
||||||
|
|
||||||
if w.Code != http.StatusOK {
|
if w.Code != http.StatusOK {
|
||||||
t.Fatalf("status=%d, want=200, body=%s", w.Code, w.Body.String())
|
t.Fatalf("status=%d, want=200, body=%s", w.Code, w.Body.String())
|
||||||
@@ -40,63 +40,63 @@ func TestQiniuCallbackSuccess(t *testing.T) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestQiniuCallbackInvalidSignature(t *testing.T) {
|
func TestUploadCallbackInvalidSignature(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
gin.SetMode(gin.TestMode)
|
gin.SetMode(gin.TestMode)
|
||||||
|
|
||||||
cfg := config.QiniuConfig{AccessKey: "ak-test", SecretKey: "sk-test"}
|
cfg := config.OSSConfig{AccessKey: "ak-test", SecretKey: "sk-test"}
|
||||||
h := NewUploadHandler(qiniuservice.NewQiniuService(cfg))
|
h := NewUploadHandler(uploadservice.NewUploadService(cfg))
|
||||||
|
|
||||||
body := "key=uploads/test.png&hash=abc"
|
body := "key=uploads/test.png&hash=abc"
|
||||||
req := httptest.NewRequest(http.MethodPost, "/api/v1/common/upload/qiniu/callback", strings.NewReader(body))
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/common/upload/oss/callback", strings.NewReader(body))
|
||||||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||||
req.Header.Set("Authorization", "QBox ak-test:bad-sign")
|
req.Header.Set("Authorization", "QBox ak-test:bad-sign")
|
||||||
w := httptest.NewRecorder()
|
w := httptest.NewRecorder()
|
||||||
c, _ := gin.CreateTestContext(w)
|
c, _ := gin.CreateTestContext(w)
|
||||||
c.Request = req
|
c.Request = req
|
||||||
|
|
||||||
h.QiniuCallback(c)
|
h.UploadCallback(c)
|
||||||
|
|
||||||
if w.Code != http.StatusUnauthorized {
|
if w.Code != http.StatusUnauthorized {
|
||||||
t.Fatalf("status=%d, want=401, body=%s", w.Code, w.Body.String())
|
t.Fatalf("status=%d, want=401, body=%s", w.Code, w.Body.String())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestQiniuCallbackMissingKey(t *testing.T) {
|
func TestUploadCallbackMissingKey(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
gin.SetMode(gin.TestMode)
|
gin.SetMode(gin.TestMode)
|
||||||
|
|
||||||
cfg := config.QiniuConfig{AccessKey: "ak-test", SecretKey: "sk-test"}
|
cfg := config.OSSConfig{AccessKey: "ak-test", SecretKey: "sk-test"}
|
||||||
h := NewUploadHandler(qiniuservice.NewQiniuService(cfg))
|
h := NewUploadHandler(uploadservice.NewUploadService(cfg))
|
||||||
|
|
||||||
body := "hash=abc&fsize=12"
|
body := "hash=abc&fsize=12"
|
||||||
req := httptest.NewRequest(http.MethodPost, "/api/v1/common/upload/qiniu/callback", strings.NewReader(body))
|
req := httptest.NewRequest(http.MethodPost, "/api/v1/common/upload/oss/callback", strings.NewReader(body))
|
||||||
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
|
||||||
req.Header.Set("Authorization", "QBox "+cfg.AccessKey+":"+signQiniu(req.URL.Path+"\n"+body, cfg.SecretKey))
|
req.Header.Set("Authorization", "QBox "+cfg.AccessKey+":"+signCallback(req.URL.Path+"\n"+body, cfg.SecretKey))
|
||||||
w := httptest.NewRecorder()
|
w := httptest.NewRecorder()
|
||||||
c, _ := gin.CreateTestContext(w)
|
c, _ := gin.CreateTestContext(w)
|
||||||
c.Request = req
|
c.Request = req
|
||||||
|
|
||||||
h.QiniuCallback(c)
|
h.UploadCallback(c)
|
||||||
|
|
||||||
if w.Code != http.StatusBadRequest {
|
if w.Code != http.StatusBadRequest {
|
||||||
t.Fatalf("status=%d, want=400, body=%s", w.Code, w.Body.String())
|
t.Fatalf("status=%d, want=400, body=%s", w.Code, w.Body.String())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func TestParseQiniuCallbackPayloadJSON(t *testing.T) {
|
func TestParseCallbackPayloadJSON(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
raw := []byte(`{"key":"uploads/test.png","hash":"abc","fsize":321,"mimeType":"image/png"}`)
|
raw := []byte(`{"key":"uploads/test.png","hash":"abc","fsize":321,"mimeType":"image/png"}`)
|
||||||
got, err := parseQiniuCallbackPayload("application/json", raw)
|
got, err := parseCallbackPayload("application/json", raw)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
t.Fatalf("parseQiniuCallbackPayload: %v", err)
|
t.Fatalf("parseCallbackPayload: %v", err)
|
||||||
}
|
}
|
||||||
if got.Key != "uploads/test.png" || got.Hash != "abc" || got.Fsize != 321 {
|
if got.Key != "uploads/test.png" || got.Hash != "abc" || got.Fsize != 321 {
|
||||||
t.Fatalf("unexpected payload: %+v", got)
|
t.Fatalf("unexpected payload: %+v", got)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func signQiniu(signing, secret string) string {
|
func signCallback(signing, secret string) string {
|
||||||
mac := hmac.New(sha1.New, []byte(secret))
|
mac := hmac.New(sha1.New, []byte(secret))
|
||||||
_, _ = mac.Write([]byte(signing))
|
_, _ = mac.Write([]byte(signing))
|
||||||
return base64.URLEncoding.WithPadding(base64.NoPadding).EncodeToString(mac.Sum(nil))
|
return base64.URLEncoding.WithPadding(base64.NoPadding).EncodeToString(mac.Sum(nil))
|
||||||
+24
-29
@@ -19,20 +19,20 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
ErrQiniuNotConfigured = errors.New("qiniu is not configured")
|
ErrUploadNotConfigured = errors.New("oss upload is not configured")
|
||||||
ErrQiniuCallbackUnauthorized = errors.New("qiniu callback unauthorized")
|
ErrCallbackUnauthorized = errors.New("upload callback unauthorized")
|
||||||
ErrQiniuCallbackInvalidHeader = errors.New("qiniu callback authorization header is invalid")
|
ErrCallbackInvalidHeader = errors.New("upload callback authorization header is invalid")
|
||||||
)
|
)
|
||||||
|
|
||||||
type QiniuService struct {
|
type UploadService struct {
|
||||||
cfg config.QiniuConfig
|
cfg config.OSSConfig
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewQiniuService(cfg config.QiniuConfig) *QiniuService {
|
func NewUploadService(cfg config.OSSConfig) *UploadService {
|
||||||
return &QiniuService{cfg: cfg}
|
return &UploadService{cfg: cfg}
|
||||||
}
|
}
|
||||||
|
|
||||||
type QiniuUploadToken struct {
|
type UploadToken struct {
|
||||||
Token string `json:"token"`
|
Token string `json:"token"`
|
||||||
Key string `json:"key"`
|
Key string `json:"key"`
|
||||||
UploadURL string `json:"upload_url"`
|
UploadURL string `json:"upload_url"`
|
||||||
@@ -42,9 +42,9 @@ type QiniuUploadToken struct {
|
|||||||
|
|
||||||
var extPattern = regexp.MustCompile(`^\.[a-z0-9]{1,10}$`)
|
var extPattern = regexp.MustCompile(`^\.[a-z0-9]{1,10}$`)
|
||||||
|
|
||||||
func (s *QiniuService) CreateUploadToken(miniProgramID uint, userID uint, filename string) (QiniuUploadToken, error) {
|
func (s *UploadService) CreateUploadToken(miniProgramID uint, userID uint, filename string) (UploadToken, error) {
|
||||||
if s.cfg.AccessKey == "" || s.cfg.SecretKey == "" || s.cfg.Bucket == "" {
|
if s.cfg.AccessKey == "" || s.cfg.SecretKey == "" || s.cfg.Bucket == "" {
|
||||||
return QiniuUploadToken{}, ErrQiniuNotConfigured
|
return UploadToken{}, ErrUploadNotConfigured
|
||||||
}
|
}
|
||||||
|
|
||||||
expireSeconds := s.cfg.TokenExpireSeconds
|
expireSeconds := s.cfg.TokenExpireSeconds
|
||||||
@@ -60,11 +60,9 @@ func (s *QiniuService) CreateUploadToken(miniProgramID uint, userID uint, filena
|
|||||||
|
|
||||||
randomHex, err := randomHex(16)
|
randomHex, err := randomHex(16)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return QiniuUploadToken{}, fmt.Errorf("generate random key: %w", err)
|
return UploadToken{}, fmt.Errorf("generate random key: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
// 统一由后端生成 key,避免前端随意写入任意路径。
|
|
||||||
// 这里按“业务前缀/小程序/用户/日期/随机名”组织,便于后期排查与管理。
|
|
||||||
keyPrefix := strings.Trim(s.cfg.KeyPrefix, "/")
|
keyPrefix := strings.Trim(s.cfg.KeyPrefix, "/")
|
||||||
key := fmt.Sprintf("%s/mp_%d/user_%d/%s/%s%s",
|
key := fmt.Sprintf("%s/mp_%d/user_%d/%s/%s%s",
|
||||||
keyPrefix,
|
keyPrefix,
|
||||||
@@ -76,10 +74,8 @@ func (s *QiniuService) CreateUploadToken(miniProgramID uint, userID uint, filena
|
|||||||
)
|
)
|
||||||
|
|
||||||
putPolicy := map[string]interface{}{
|
putPolicy := map[string]interface{}{
|
||||||
// scope = "<bucket>:<key>" 表示只允许写入指定 key(更安全)
|
"scope": fmt.Sprintf("%s:%s", s.cfg.Bucket, key),
|
||||||
"scope": fmt.Sprintf("%s:%s", s.cfg.Bucket, key),
|
"deadline": expireAt,
|
||||||
"deadline": expireAt,
|
|
||||||
// 上传完成后返回给前端的 JSON(七牛会做变量替换)
|
|
||||||
"returnBody": `{"key":"$(key)","hash":"$(etag)","fsize":$(fsize),"mimeType":"$(mimeType)"}`,
|
"returnBody": `{"key":"$(key)","hash":"$(etag)","fsize":$(fsize),"mimeType":"$(mimeType)"}`,
|
||||||
}
|
}
|
||||||
if callbackURL := strings.TrimSpace(s.cfg.CallbackURL); callbackURL != "" {
|
if callbackURL := strings.TrimSpace(s.cfg.CallbackURL); callbackURL != "" {
|
||||||
@@ -98,7 +94,7 @@ func (s *QiniuService) CreateUploadToken(miniProgramID uint, userID uint, filena
|
|||||||
|
|
||||||
policyJSON, err := json.Marshal(putPolicy)
|
policyJSON, err := json.Marshal(putPolicy)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return QiniuUploadToken{}, fmt.Errorf("marshal put policy: %w", err)
|
return UploadToken{}, fmt.Errorf("marshal put policy: %w", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
encodedPolicy := urlSafeBase64NoPad(policyJSON)
|
encodedPolicy := urlSafeBase64NoPad(policyJSON)
|
||||||
@@ -107,7 +103,7 @@ func (s *QiniuService) CreateUploadToken(miniProgramID uint, userID uint, filena
|
|||||||
|
|
||||||
token := fmt.Sprintf("%s:%s:%s", s.cfg.AccessKey, encodedSign, encodedPolicy)
|
token := fmt.Sprintf("%s:%s:%s", s.cfg.AccessKey, encodedSign, encodedPolicy)
|
||||||
|
|
||||||
return QiniuUploadToken{
|
return UploadToken{
|
||||||
Token: token,
|
Token: token,
|
||||||
Key: key,
|
Key: key,
|
||||||
UploadURL: s.cfg.UploadURL,
|
UploadURL: s.cfg.UploadURL,
|
||||||
@@ -116,38 +112,37 @@ func (s *QiniuService) CreateUploadToken(miniProgramID uint, userID uint, filena
|
|||||||
}, nil
|
}, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *QiniuService) VerifyCallbackSignature(req *http.Request, rawBody []byte) error {
|
func (s *UploadService) VerifyCallbackSignature(req *http.Request, rawBody []byte) error {
|
||||||
if s.cfg.AccessKey == "" || s.cfg.SecretKey == "" {
|
if s.cfg.AccessKey == "" || s.cfg.SecretKey == "" {
|
||||||
return ErrQiniuNotConfigured
|
return ErrUploadNotConfigured
|
||||||
}
|
}
|
||||||
|
|
||||||
authHeader := strings.TrimSpace(req.Header.Get("Authorization"))
|
authHeader := strings.TrimSpace(req.Header.Get("Authorization"))
|
||||||
if authHeader == "" {
|
if authHeader == "" {
|
||||||
return ErrQiniuCallbackInvalidHeader
|
return ErrCallbackInvalidHeader
|
||||||
}
|
}
|
||||||
|
|
||||||
parts := strings.SplitN(authHeader, " ", 2)
|
parts := strings.SplitN(authHeader, " ", 2)
|
||||||
if len(parts) != 2 {
|
if len(parts) != 2 {
|
||||||
return ErrQiniuCallbackInvalidHeader
|
return ErrCallbackInvalidHeader
|
||||||
}
|
}
|
||||||
scheme := strings.TrimSpace(parts[0])
|
scheme := strings.TrimSpace(parts[0])
|
||||||
if !strings.EqualFold(scheme, "QBox") {
|
if !strings.EqualFold(scheme, "QBox") {
|
||||||
// 七牛上传回调使用 QBox;其它 scheme 视为非法。
|
return ErrCallbackInvalidHeader
|
||||||
return ErrQiniuCallbackInvalidHeader
|
|
||||||
}
|
}
|
||||||
|
|
||||||
token := strings.TrimSpace(parts[1])
|
token := strings.TrimSpace(parts[1])
|
||||||
tokenParts := strings.SplitN(token, ":", 2)
|
tokenParts := strings.SplitN(token, ":", 2)
|
||||||
if len(tokenParts) != 2 {
|
if len(tokenParts) != 2 {
|
||||||
return ErrQiniuCallbackInvalidHeader
|
return ErrCallbackInvalidHeader
|
||||||
}
|
}
|
||||||
accessKey := strings.TrimSpace(tokenParts[0])
|
accessKey := strings.TrimSpace(tokenParts[0])
|
||||||
providedSign := strings.TrimSpace(tokenParts[1])
|
providedSign := strings.TrimSpace(tokenParts[1])
|
||||||
if accessKey == "" || providedSign == "" {
|
if accessKey == "" || providedSign == "" {
|
||||||
return ErrQiniuCallbackInvalidHeader
|
return ErrCallbackInvalidHeader
|
||||||
}
|
}
|
||||||
if accessKey != s.cfg.AccessKey {
|
if accessKey != s.cfg.AccessKey {
|
||||||
return ErrQiniuCallbackUnauthorized
|
return ErrCallbackUnauthorized
|
||||||
}
|
}
|
||||||
|
|
||||||
signing := req.URL.Path
|
signing := req.URL.Path
|
||||||
@@ -159,7 +154,7 @@ func (s *QiniuService) VerifyCallbackSignature(req *http.Request, rawBody []byte
|
|||||||
|
|
||||||
expected := urlSafeBase64NoPad(hmacSHA1([]byte(s.cfg.SecretKey), []byte(signing)))
|
expected := urlSafeBase64NoPad(hmacSHA1([]byte(s.cfg.SecretKey), []byte(signing)))
|
||||||
if !hmac.Equal([]byte(providedSign), []byte(expected)) {
|
if !hmac.Equal([]byte(providedSign), []byte(expected)) {
|
||||||
return ErrQiniuCallbackUnauthorized
|
return ErrCallbackUnauthorized
|
||||||
}
|
}
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
+12
-12
@@ -13,13 +13,13 @@ import (
|
|||||||
func TestCreateUploadTokenWithCallbackPolicy(t *testing.T) {
|
func TestCreateUploadTokenWithCallbackPolicy(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
svc := NewQiniuService(config.QiniuConfig{
|
svc := NewUploadService(config.OSSConfig{
|
||||||
AccessKey: "ak-test",
|
AccessKey: "ak-test",
|
||||||
SecretKey: "sk-test",
|
SecretKey: "sk-test",
|
||||||
Bucket: "bucket-test",
|
Bucket: "bucket-test",
|
||||||
UploadURL: "https://upload.qiniup.com",
|
UploadURL: "https://bucket.oss-cn-beijing.aliyuncs.com",
|
||||||
KeyPrefix: "uploads/",
|
KeyPrefix: "uploads/",
|
||||||
CallbackURL: "https://api.example.com/api/v1/common/upload/qiniu/callback",
|
CallbackURL: "https://api.example.com/api/v1/common/upload/oss/callback",
|
||||||
CallbackBody: "key=$(key)&hash=$(etag)",
|
CallbackBody: "key=$(key)&hash=$(etag)",
|
||||||
CallbackBodyType: "application/x-www-form-urlencoded",
|
CallbackBodyType: "application/x-www-form-urlencoded",
|
||||||
})
|
})
|
||||||
@@ -44,7 +44,7 @@ func TestCreateUploadTokenWithCallbackPolicy(t *testing.T) {
|
|||||||
t.Fatalf("unmarshal policy: %v", err)
|
t.Fatalf("unmarshal policy: %v", err)
|
||||||
}
|
}
|
||||||
|
|
||||||
if got, _ := policy["callbackUrl"].(string); got != "https://api.example.com/api/v1/common/upload/qiniu/callback" {
|
if got, _ := policy["callbackUrl"].(string); got != "https://api.example.com/api/v1/common/upload/oss/callback" {
|
||||||
t.Fatalf("callbackUrl=%q, want callback endpoint", got)
|
t.Fatalf("callbackUrl=%q, want callback endpoint", got)
|
||||||
}
|
}
|
||||||
if got, _ := policy["callbackBody"].(string); got != "key=$(key)&hash=$(etag)" {
|
if got, _ := policy["callbackBody"].(string); got != "key=$(key)&hash=$(etag)" {
|
||||||
@@ -58,14 +58,14 @@ func TestCreateUploadTokenWithCallbackPolicy(t *testing.T) {
|
|||||||
func TestVerifyCallbackSignature(t *testing.T) {
|
func TestVerifyCallbackSignature(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
cfg := config.QiniuConfig{
|
cfg := config.OSSConfig{
|
||||||
AccessKey: "ak-test",
|
AccessKey: "ak-test",
|
||||||
SecretKey: "sk-test",
|
SecretKey: "sk-test",
|
||||||
}
|
}
|
||||||
svc := NewQiniuService(cfg)
|
svc := NewUploadService(cfg)
|
||||||
|
|
||||||
body := []byte("key=uploads/test.png&hash=abc")
|
body := []byte("key=uploads/test.png&hash=abc")
|
||||||
req := httptest.NewRequest("POST", "http://example.com/api/v1/common/upload/qiniu/callback?x=1", strings.NewReader(string(body)))
|
req := httptest.NewRequest("POST", "http://example.com/api/v1/common/upload/oss/callback?x=1", strings.NewReader(string(body)))
|
||||||
signing := req.URL.Path + "?" + req.URL.RawQuery + "\n" + string(body)
|
signing := req.URL.Path + "?" + req.URL.RawQuery + "\n" + string(body)
|
||||||
sign := urlSafeBase64NoPad(hmacSHA1([]byte(cfg.SecretKey), []byte(signing)))
|
sign := urlSafeBase64NoPad(hmacSHA1([]byte(cfg.SecretKey), []byte(signing)))
|
||||||
req.Header.Set("Authorization", "QBox "+cfg.AccessKey+":"+sign)
|
req.Header.Set("Authorization", "QBox "+cfg.AccessKey+":"+sign)
|
||||||
@@ -78,21 +78,21 @@ func TestVerifyCallbackSignature(t *testing.T) {
|
|||||||
func TestVerifyCallbackSignatureInvalid(t *testing.T) {
|
func TestVerifyCallbackSignatureInvalid(t *testing.T) {
|
||||||
t.Parallel()
|
t.Parallel()
|
||||||
|
|
||||||
cfg := config.QiniuConfig{
|
cfg := config.OSSConfig{
|
||||||
AccessKey: "ak-test",
|
AccessKey: "ak-test",
|
||||||
SecretKey: "sk-test",
|
SecretKey: "sk-test",
|
||||||
}
|
}
|
||||||
svc := NewQiniuService(cfg)
|
svc := NewUploadService(cfg)
|
||||||
|
|
||||||
body := []byte("key=uploads/test.png&hash=abc")
|
body := []byte("key=uploads/test.png&hash=abc")
|
||||||
req := httptest.NewRequest("POST", "http://example.com/api/v1/common/upload/qiniu/callback", strings.NewReader(string(body)))
|
req := httptest.NewRequest("POST", "http://example.com/api/v1/common/upload/oss/callback", strings.NewReader(string(body)))
|
||||||
req.Header.Set("Authorization", "QBox ak-test:bad-sign")
|
req.Header.Set("Authorization", "QBox ak-test:bad-sign")
|
||||||
|
|
||||||
err := svc.VerifyCallbackSignature(req, body)
|
err := svc.VerifyCallbackSignature(req, body)
|
||||||
if err == nil {
|
if err == nil {
|
||||||
t.Fatalf("expected signature verification error")
|
t.Fatalf("expected signature verification error")
|
||||||
}
|
}
|
||||||
if err != ErrQiniuCallbackUnauthorized {
|
if err != ErrCallbackUnauthorized {
|
||||||
t.Fatalf("error=%v, want=%v", err, ErrQiniuCallbackUnauthorized)
|
t.Fatalf("error=%v, want=%v", err, ErrCallbackUnauthorized)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@@ -0,0 +1,157 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
|
"wx_service/internal/marketing/model"
|
||||||
|
"wx_service/internal/marketing/repository"
|
||||||
|
commonmodel "wx_service/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
type AdPlacementHandler struct {
|
||||||
|
repo *repository.AdPlacementRepository
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewAdPlacementHandler(repo *repository.AdPlacementRepository) *AdPlacementHandler {
|
||||||
|
return &AdPlacementHandler{repo: repo}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdPlacementHandler) AdminList(c *gin.Context) {
|
||||||
|
list, err := h.repo.ListAll()
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, commonmodel.Error(http.StatusInternalServerError, "获取广告位列表失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, commonmodel.Success(list))
|
||||||
|
}
|
||||||
|
|
||||||
|
type adPlacementCreateReq struct {
|
||||||
|
MiniProgramID uint `json:"mini_program_id" binding:"required"`
|
||||||
|
Name string `json:"name" binding:"required"`
|
||||||
|
AdType string `json:"ad_type" binding:"required"`
|
||||||
|
AdUnitID string `json:"ad_unit_id"`
|
||||||
|
Status *int `json:"status"`
|
||||||
|
Description string `json:"description"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdPlacementHandler) AdminCreate(c *gin.Context) {
|
||||||
|
var req adPlacementCreateReq
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, commonmodel.Error(http.StatusBadRequest, "参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
p := &model.AdPlacement{
|
||||||
|
MiniProgramID: req.MiniProgramID,
|
||||||
|
Name: req.Name,
|
||||||
|
AdType: req.AdType,
|
||||||
|
AdUnitID: req.AdUnitID,
|
||||||
|
Description: req.Description,
|
||||||
|
Status: 1,
|
||||||
|
}
|
||||||
|
if req.Status != nil {
|
||||||
|
p.Status = *req.Status
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := h.repo.Create(p); err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, commonmodel.Error(http.StatusInternalServerError, "创建失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, commonmodel.Success(p))
|
||||||
|
}
|
||||||
|
|
||||||
|
type adPlacementUpdateReq struct {
|
||||||
|
Name *string `json:"name"`
|
||||||
|
AdUnitID *string `json:"ad_unit_id"`
|
||||||
|
Status *int `json:"status"`
|
||||||
|
Description *string `json:"description"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdPlacementHandler) AdminUpdate(c *gin.Context) {
|
||||||
|
id, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
||||||
|
if err != nil || id == 0 {
|
||||||
|
c.JSON(http.StatusBadRequest, commonmodel.Error(http.StatusBadRequest, "无效 ID"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
p, err := h.repo.FindByID(uint(id))
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, repository.ErrAdPlacementNotFound) {
|
||||||
|
c.JSON(http.StatusNotFound, commonmodel.Error(http.StatusNotFound, "广告位不存在"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusInternalServerError, commonmodel.Error(http.StatusInternalServerError, "查询失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var req adPlacementUpdateReq
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, commonmodel.Error(http.StatusBadRequest, "参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if req.Name != nil {
|
||||||
|
p.Name = *req.Name
|
||||||
|
}
|
||||||
|
if req.AdUnitID != nil {
|
||||||
|
p.AdUnitID = *req.AdUnitID
|
||||||
|
}
|
||||||
|
if req.Status != nil {
|
||||||
|
p.Status = *req.Status
|
||||||
|
}
|
||||||
|
if req.Description != nil {
|
||||||
|
p.Description = *req.Description
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := h.repo.Update(p); err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, commonmodel.Error(http.StatusInternalServerError, "更新失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, commonmodel.Success(p))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdPlacementHandler) AdminDelete(c *gin.Context) {
|
||||||
|
id, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
||||||
|
if err != nil || id == 0 {
|
||||||
|
c.JSON(http.StatusBadRequest, commonmodel.Error(http.StatusBadRequest, "无效 ID"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := h.repo.Delete(uint(id)); err != nil {
|
||||||
|
if errors.Is(err, repository.ErrAdPlacementNotFound) {
|
||||||
|
c.JSON(http.StatusNotFound, commonmodel.Error(http.StatusNotFound, "广告位不存在"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusInternalServerError, commonmodel.Error(http.StatusInternalServerError, "删除失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, commonmodel.Success(nil))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *AdPlacementHandler) GetAdConfig(c *gin.Context) {
|
||||||
|
miniProgramIDStr := c.Query("mini_program_id")
|
||||||
|
miniProgramID, _ := strconv.ParseUint(miniProgramIDStr, 10, 64)
|
||||||
|
if miniProgramID == 0 {
|
||||||
|
c.JSON(http.StatusBadRequest, commonmodel.Error(http.StatusBadRequest, "缺少 mini_program_id"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
p, err := h.repo.FindByMiniProgramAndType(uint(miniProgramID), "rewarded_video")
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, repository.ErrAdPlacementNotFound) {
|
||||||
|
c.JSON(http.StatusOK, commonmodel.Success(gin.H{"ad_unit_id": "", "enabled": false}))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusInternalServerError, commonmodel.Error(http.StatusInternalServerError, "查询失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(http.StatusOK, commonmodel.Success(gin.H{
|
||||||
|
"ad_unit_id": p.AdUnitID,
|
||||||
|
"enabled": p.Status == 1 && p.AdUnitID != "",
|
||||||
|
}))
|
||||||
|
}
|
||||||
@@ -1,18 +1,31 @@
|
|||||||
package handler
|
package handler
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"bytes"
|
||||||
"errors"
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"log"
|
||||||
|
"mime/multipart"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"path"
|
||||||
|
"regexp"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
"wx_service/config"
|
"wx_service/config"
|
||||||
qiniuservice "wx_service/internal/common/qiniu/service"
|
"wx_service/internal/common/imageutil"
|
||||||
|
oss "wx_service/internal/common/oss"
|
||||||
|
uploadservice "wx_service/internal/common/upload/service"
|
||||||
"wx_service/internal/marketing/service"
|
"wx_service/internal/marketing/service"
|
||||||
"wx_service/internal/middleware"
|
"wx_service/internal/middleware"
|
||||||
"wx_service/internal/model"
|
"wx_service/internal/model"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
var adminExtPattern = regexp.MustCompile(`^\.[a-z0-9]{1,10}$`)
|
||||||
|
|
||||||
type DownloadHandler struct {
|
type DownloadHandler struct {
|
||||||
svc *service.DownloadService
|
svc *service.DownloadService
|
||||||
}
|
}
|
||||||
@@ -93,19 +106,69 @@ func (h *DownloadHandler) AdminStats(c *gin.Context) {
|
|||||||
c.JSON(http.StatusOK, model.Success(stats))
|
c.JSON(http.StatusOK, model.Success(stats))
|
||||||
}
|
}
|
||||||
|
|
||||||
type adminQiniuTokenRequest struct {
|
type adminUploadTokenRequest struct {
|
||||||
Filename string `json:"filename"`
|
Filename string `json:"filename"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *DownloadHandler) AdminQiniuToken(c *gin.Context) {
|
type adminUploadTokenResponse struct {
|
||||||
var req adminQiniuTokenRequest
|
Token string `json:"token,omitempty"`
|
||||||
|
Key string `json:"key"`
|
||||||
|
UploadURL string `json:"upload_url"`
|
||||||
|
ExpireAt int64 `json:"expire,omitempty"`
|
||||||
|
CDNDomain string `json:"cdn_domain,omitempty"`
|
||||||
|
OSSAccessKey string `json:"oss_access_key_id,omitempty"`
|
||||||
|
OSSPolicy string `json:"oss_policy,omitempty"`
|
||||||
|
OSSSignature string `json:"oss_signature,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *DownloadHandler) AdminUploadToken(c *gin.Context) {
|
||||||
|
var req adminUploadTokenRequest
|
||||||
_ = c.ShouldBindJSON(&req)
|
_ = c.ShouldBindJSON(&req)
|
||||||
|
|
||||||
qiniuSvc := qiniuservice.NewQiniuService(config.AppConfig.Qiniu)
|
cfg := config.AppConfig.OSS
|
||||||
token, err := qiniuSvc.CreateUploadToken(0, 0, req.Filename)
|
if cfg.AccessKey == "" || cfg.SecretKey == "" || cfg.Bucket == "" {
|
||||||
|
c.JSON(http.StatusServiceUnavailable, model.Error(http.StatusServiceUnavailable, "未配置上传服务"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cdnDomain := strings.TrimSpace(cfg.CDNDomain)
|
||||||
|
|
||||||
|
if oss.IsOSSDomain(cdnDomain) {
|
||||||
|
ext := path.Ext(req.Filename)
|
||||||
|
if ext == "" || !adminExtPattern.MatchString(strings.ToLower(ext)) {
|
||||||
|
ext = ".jpg"
|
||||||
|
}
|
||||||
|
keyPrefix := strings.Trim(cfg.KeyPrefix, "/")
|
||||||
|
key := fmt.Sprintf("%s/admin/%s/%x%s",
|
||||||
|
keyPrefix, time.Now().Format("20060102"), time.Now().UnixNano()&0xffffffff, ext)
|
||||||
|
endpoint := oss.ParseOSSEndpoint(cdnDomain)
|
||||||
|
expireSeconds := cfg.TokenExpireSeconds
|
||||||
|
if expireSeconds <= 0 {
|
||||||
|
expireSeconds = 300
|
||||||
|
}
|
||||||
|
policy, signature, err := oss.PostPolicy(cfg.Bucket, endpoint, key, cfg.SecretKey, expireSeconds)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "生成上传凭证失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
uploadURL := oss.UploadHost(cfg.Bucket, endpoint)
|
||||||
|
cdnHost := "https://" + cfg.Bucket + "." + endpoint + ".aliyuncs.com"
|
||||||
|
c.JSON(http.StatusOK, model.Success(adminUploadTokenResponse{
|
||||||
|
Key: key,
|
||||||
|
UploadURL: uploadURL,
|
||||||
|
CDNDomain: cdnHost,
|
||||||
|
OSSAccessKey: cfg.AccessKey,
|
||||||
|
OSSPolicy: policy,
|
||||||
|
OSSSignature: signature,
|
||||||
|
}))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
uploadSvc := uploadservice.NewUploadService(cfg)
|
||||||
|
token, err := uploadSvc.CreateUploadToken(0, 0, req.Filename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if errors.Is(err, qiniuservice.ErrQiniuNotConfigured) {
|
if errors.Is(err, uploadservice.ErrUploadNotConfigured) {
|
||||||
c.JSON(http.StatusServiceUnavailable, model.Error(http.StatusServiceUnavailable, "未配置七牛上传服务"))
|
c.JSON(http.StatusServiceUnavailable, model.Error(http.StatusServiceUnavailable, "未配置上传服务"))
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取上传凭证失败"))
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取上传凭证失败"))
|
||||||
@@ -114,3 +177,124 @@ func (h *DownloadHandler) AdminQiniuToken(c *gin.Context) {
|
|||||||
|
|
||||||
c.JSON(http.StatusOK, model.Success(token))
|
c.JSON(http.StatusOK, model.Success(token))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// AdminUploadFile 接收管理后台上传的文件,服务端代理转发到 OSS,
|
||||||
|
// 同时自动生成一张压缩缩略图一并上传,减少 CDN 占用。
|
||||||
|
func (h *DownloadHandler) AdminUploadFile(c *gin.Context) {
|
||||||
|
file, header, err := c.Request.FormFile("file")
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "请选择要上传的文件"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
defer file.Close()
|
||||||
|
|
||||||
|
if header.Size > 10*1024*1024 {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "文件大小不能超过 10MB"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cfg := config.AppConfig.OSS
|
||||||
|
if cfg.AccessKey == "" || cfg.SecretKey == "" || cfg.Bucket == "" {
|
||||||
|
c.JSON(http.StatusServiceUnavailable, model.Error(http.StatusServiceUnavailable, "未配置上传服务"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
cdnDomain := strings.TrimSpace(cfg.CDNDomain)
|
||||||
|
if !oss.IsOSSDomain(cdnDomain) {
|
||||||
|
c.JSON(http.StatusServiceUnavailable, model.Error(http.StatusServiceUnavailable, "仅支持 OSS 上传"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
fileData, err := io.ReadAll(file)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "读取文件失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
ext := strings.ToLower(path.Ext(header.Filename))
|
||||||
|
if ext == "" || !adminExtPattern.MatchString(ext) {
|
||||||
|
ext = ".jpg"
|
||||||
|
}
|
||||||
|
keyPrefix := strings.Trim(cfg.KeyPrefix, "/")
|
||||||
|
ts := fmt.Sprintf("%x", time.Now().UnixNano()&0xffffffff)
|
||||||
|
origKey := fmt.Sprintf("%s/admin/%s/%s%s", keyPrefix, time.Now().Format("20060102"), ts, ext)
|
||||||
|
thumbKey := fmt.Sprintf("%s/admin/%s/%s_thumb.jpg", keyPrefix, time.Now().Format("20060102"), ts)
|
||||||
|
|
||||||
|
endpoint := oss.ParseOSSEndpoint(cdnDomain)
|
||||||
|
cdnHost := "https://" + cfg.Bucket + "." + endpoint + ".aliyuncs.com"
|
||||||
|
|
||||||
|
origURL, err := uploadToOSS(cfg, endpoint, origKey, header.Filename, fileData)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("[admin_upload] upload original error: %v", err)
|
||||||
|
c.JSON(http.StatusBadGateway, model.Error(http.StatusBadGateway, "上传原图到 OSS 失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
thumbURL := ""
|
||||||
|
thumbData, thumbErr := imageutil.GenerateThumbnail(fileData, header.Header.Get("Content-Type"), imageutil.DefaultResizeOptions())
|
||||||
|
if thumbErr != nil {
|
||||||
|
log.Printf("[admin_upload] thumbnail generation skipped: %v", thumbErr)
|
||||||
|
} else {
|
||||||
|
if url, err := uploadToOSS(cfg, endpoint, thumbKey, "thumb.jpg", thumbData); err != nil {
|
||||||
|
log.Printf("[admin_upload] upload thumbnail error: %v", err)
|
||||||
|
} else {
|
||||||
|
thumbURL = url
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if origURL == "" {
|
||||||
|
origURL = cdnHost + "/" + origKey
|
||||||
|
}
|
||||||
|
|
||||||
|
result := gin.H{"url": origURL}
|
||||||
|
if thumbURL != "" {
|
||||||
|
result["thumbnail_url"] = thumbURL
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(result))
|
||||||
|
}
|
||||||
|
|
||||||
|
func uploadToOSS(cfg config.OSSConfig, endpoint, key, filename string, data []byte) (string, error) {
|
||||||
|
expireSeconds := cfg.TokenExpireSeconds
|
||||||
|
if expireSeconds <= 0 {
|
||||||
|
expireSeconds = 300
|
||||||
|
}
|
||||||
|
policy, signature, err := oss.PostPolicy(cfg.Bucket, endpoint, key, cfg.SecretKey, expireSeconds)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("generate policy: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
uploadURL := oss.UploadHost(cfg.Bucket, endpoint)
|
||||||
|
var body bytes.Buffer
|
||||||
|
writer := multipart.NewWriter(&body)
|
||||||
|
writer.WriteField("key", key)
|
||||||
|
writer.WriteField("policy", policy)
|
||||||
|
writer.WriteField("OSSAccessKeyId", cfg.AccessKey)
|
||||||
|
writer.WriteField("Signature", signature)
|
||||||
|
fw, err := writer.CreateFormFile("file", filename)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("create form file: %w", err)
|
||||||
|
}
|
||||||
|
fw.Write(data)
|
||||||
|
writer.Close()
|
||||||
|
|
||||||
|
req, err := http.NewRequest(http.MethodPost, uploadURL, &body)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("new request: %w", err)
|
||||||
|
}
|
||||||
|
req.Header.Set("Content-Type", writer.FormDataContentType())
|
||||||
|
|
||||||
|
client := &http.Client{Timeout: 60 * time.Second}
|
||||||
|
resp, err := client.Do(req)
|
||||||
|
if err != nil {
|
||||||
|
return "", fmt.Errorf("do request: %w", err)
|
||||||
|
}
|
||||||
|
defer resp.Body.Close()
|
||||||
|
|
||||||
|
if resp.StatusCode < 200 || resp.StatusCode >= 300 {
|
||||||
|
respBody, _ := io.ReadAll(resp.Body)
|
||||||
|
return "", fmt.Errorf("oss status=%d body=%s", resp.StatusCode, string(respBody))
|
||||||
|
}
|
||||||
|
|
||||||
|
cdnHost := "https://" + cfg.Bucket + "." + endpoint + ".aliyuncs.com"
|
||||||
|
return cdnHost + "/" + key, nil
|
||||||
|
}
|
||||||
|
|||||||
@@ -0,0 +1,73 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"net/http"
|
||||||
|
"strconv"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
|
"wx_service/internal/marketing/repository"
|
||||||
|
"wx_service/internal/marketing/service"
|
||||||
|
"wx_service/internal/middleware"
|
||||||
|
"wx_service/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
type UserLogoHandler struct {
|
||||||
|
svc *service.UserLogoService
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewUserLogoHandler(svc *service.UserLogoService) *UserLogoHandler {
|
||||||
|
return &UserLogoHandler{svc: svc}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *UserLogoHandler) List(c *gin.Context) {
|
||||||
|
user := middleware.MustCurrentUser(c)
|
||||||
|
logos, err := h.svc.List(user.ID)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取 Logo 列表失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(logos))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *UserLogoHandler) Save(c *gin.Context) {
|
||||||
|
user := middleware.MustCurrentUser(c)
|
||||||
|
|
||||||
|
var req service.SaveLogoRequest
|
||||||
|
if err := c.ShouldBindJSON(&req); err != nil {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "请求参数错误"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
logo, err := h.svc.Save(user.ID, req)
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, service.ErrLogoLimitReached) || errors.Is(err, service.ErrLogoTooLarge) {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, err.Error()))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "保存 Logo 失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(logo))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *UserLogoHandler) Delete(c *gin.Context) {
|
||||||
|
user := middleware.MustCurrentUser(c)
|
||||||
|
|
||||||
|
id, err := strconv.ParseUint(c.Param("id"), 10, 64)
|
||||||
|
if err != nil || id == 0 {
|
||||||
|
c.JSON(http.StatusBadRequest, model.Error(http.StatusBadRequest, "无效的 ID"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
if err := h.svc.Delete(uint(id), user.ID); err != nil {
|
||||||
|
if errors.Is(err, repository.ErrUserLogoNotFound) {
|
||||||
|
c.JSON(http.StatusNotFound, model.Error(http.StatusNotFound, "Logo 不存在"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "删除失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(nil))
|
||||||
|
}
|
||||||
@@ -0,0 +1,19 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
type AdPlacement struct {
|
||||||
|
ID uint `json:"id" gorm:"primaryKey;comment:主键ID"`
|
||||||
|
MiniProgramID uint `json:"mini_program_id" gorm:"not null;index;comment:小程序ID"`
|
||||||
|
Name string `json:"name" gorm:"size:100;not null;comment:广告位名称"`
|
||||||
|
AdType string `json:"ad_type" gorm:"size:50;not null;default:rewarded_video;comment:广告类型(rewarded_video/banner/interstitial)"`
|
||||||
|
AdUnitID string `json:"ad_unit_id" gorm:"size:200;comment:微信广告单元ID"`
|
||||||
|
Status int `json:"status" gorm:"default:1;comment:状态(0禁用/1启用)"`
|
||||||
|
Description string `json:"description" gorm:"size:500;comment:备注说明"`
|
||||||
|
CreatedAt time.Time `json:"created_at" gorm:"comment:创建时间"`
|
||||||
|
UpdatedAt time.Time `json:"updated_at" gorm:"comment:更新时间"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (AdPlacement) TableName() string {
|
||||||
|
return "marketing_ad_placements"
|
||||||
|
}
|
||||||
@@ -0,0 +1,16 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import "time"
|
||||||
|
|
||||||
|
type UserLogo struct {
|
||||||
|
ID uint `json:"id" gorm:"primaryKey;comment:主键ID"`
|
||||||
|
UserID uint `json:"user_id" gorm:"not null;index;comment:用户ID"`
|
||||||
|
URL string `json:"url" gorm:"size:500;not null;comment:Logo CDN地址"`
|
||||||
|
Filename string `json:"filename" gorm:"size:255;comment:原始文件名"`
|
||||||
|
FileSize int64 `json:"file_size" gorm:"comment:文件大小(字节)"`
|
||||||
|
CreatedAt time.Time `json:"created_at" gorm:"comment:创建时间"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (UserLogo) TableName() string {
|
||||||
|
return "marketing_user_logos"
|
||||||
|
}
|
||||||
@@ -0,0 +1,76 @@
|
|||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"gorm.io/gorm"
|
||||||
|
|
||||||
|
"wx_service/internal/marketing/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
var ErrAdPlacementNotFound = errors.New("ad placement not found")
|
||||||
|
|
||||||
|
type AdPlacementRepository struct {
|
||||||
|
db *gorm.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewAdPlacementRepository(db *gorm.DB) *AdPlacementRepository {
|
||||||
|
return &AdPlacementRepository{db: db}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *AdPlacementRepository) Create(p *model.AdPlacement) error {
|
||||||
|
if err := r.db.Create(p).Error; err != nil {
|
||||||
|
return fmt.Errorf("create ad placement: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *AdPlacementRepository) Update(p *model.AdPlacement) error {
|
||||||
|
if err := r.db.Save(p).Error; err != nil {
|
||||||
|
return fmt.Errorf("update ad placement: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *AdPlacementRepository) Delete(id uint) error {
|
||||||
|
tx := r.db.Delete(&model.AdPlacement{}, id)
|
||||||
|
if tx.Error != nil {
|
||||||
|
return fmt.Errorf("delete ad placement: %w", tx.Error)
|
||||||
|
}
|
||||||
|
if tx.RowsAffected == 0 {
|
||||||
|
return ErrAdPlacementNotFound
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *AdPlacementRepository) FindByID(id uint) (*model.AdPlacement, error) {
|
||||||
|
var p model.AdPlacement
|
||||||
|
err := r.db.First(&p, id).Error
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||||
|
return nil, ErrAdPlacementNotFound
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("find ad placement: %w", err)
|
||||||
|
}
|
||||||
|
return &p, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *AdPlacementRepository) ListAll() ([]model.AdPlacement, error) {
|
||||||
|
var list []model.AdPlacement
|
||||||
|
err := r.db.Order("id DESC").Find(&list).Error
|
||||||
|
return list, err
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *AdPlacementRepository) FindByMiniProgramAndType(miniProgramID uint, adType string) (*model.AdPlacement, error) {
|
||||||
|
var p model.AdPlacement
|
||||||
|
err := r.db.Where("mini_program_id = ? AND ad_type = ? AND status = 1", miniProgramID, adType).
|
||||||
|
First(&p).Error
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||||
|
return nil, ErrAdPlacementNotFound
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("find ad placement by type: %w", err)
|
||||||
|
}
|
||||||
|
return &p, nil
|
||||||
|
}
|
||||||
@@ -0,0 +1,65 @@
|
|||||||
|
package repository
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
|
||||||
|
"gorm.io/gorm"
|
||||||
|
|
||||||
|
"wx_service/internal/marketing/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
var ErrUserLogoNotFound = errors.New("user logo not found")
|
||||||
|
|
||||||
|
type UserLogoRepository struct {
|
||||||
|
db *gorm.DB
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewUserLogoRepository(db *gorm.DB) *UserLogoRepository {
|
||||||
|
return &UserLogoRepository{db: db}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *UserLogoRepository) Create(logo *model.UserLogo) error {
|
||||||
|
if err := r.db.Create(logo).Error; err != nil {
|
||||||
|
return fmt.Errorf("create user logo: %w", err)
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *UserLogoRepository) FindByUser(userID uint) ([]model.UserLogo, error) {
|
||||||
|
var logos []model.UserLogo
|
||||||
|
err := r.db.Where("user_id = ?", userID).Order("id DESC").Find(&logos).Error
|
||||||
|
if err != nil {
|
||||||
|
return nil, fmt.Errorf("list user logos: %w", err)
|
||||||
|
}
|
||||||
|
return logos, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *UserLogoRepository) FindByID(id, userID uint) (*model.UserLogo, error) {
|
||||||
|
var logo model.UserLogo
|
||||||
|
err := r.db.Where("id = ? AND user_id = ?", id, userID).First(&logo).Error
|
||||||
|
if err != nil {
|
||||||
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
||||||
|
return nil, ErrUserLogoNotFound
|
||||||
|
}
|
||||||
|
return nil, fmt.Errorf("find user logo: %w", err)
|
||||||
|
}
|
||||||
|
return &logo, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *UserLogoRepository) Delete(id, userID uint) error {
|
||||||
|
tx := r.db.Where("id = ? AND user_id = ?", id, userID).Delete(&model.UserLogo{})
|
||||||
|
if tx.Error != nil {
|
||||||
|
return fmt.Errorf("delete user logo: %w", tx.Error)
|
||||||
|
}
|
||||||
|
if tx.RowsAffected == 0 {
|
||||||
|
return ErrUserLogoNotFound
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *UserLogoRepository) CountByUser(userID uint) (int64, error) {
|
||||||
|
var count int64
|
||||||
|
err := r.db.Model(&model.UserLogo{}).Where("user_id = ?", userID).Count(&count).Error
|
||||||
|
return count, err
|
||||||
|
}
|
||||||
@@ -0,0 +1,63 @@
|
|||||||
|
package service
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
|
||||||
|
"wx_service/internal/marketing/model"
|
||||||
|
"wx_service/internal/marketing/repository"
|
||||||
|
)
|
||||||
|
|
||||||
|
const MaxLogosPerUser = 3
|
||||||
|
const MaxLogoFileSize = 2 * 1024 * 1024 // 2MB
|
||||||
|
|
||||||
|
var (
|
||||||
|
ErrLogoLimitReached = errors.New("Logo 数量已达上限")
|
||||||
|
ErrLogoTooLarge = errors.New("Logo 文件不能超过 2MB")
|
||||||
|
)
|
||||||
|
|
||||||
|
type UserLogoService struct {
|
||||||
|
repo *repository.UserLogoRepository
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewUserLogoService(repo *repository.UserLogoRepository) *UserLogoService {
|
||||||
|
return &UserLogoService{repo: repo}
|
||||||
|
}
|
||||||
|
|
||||||
|
type SaveLogoRequest struct {
|
||||||
|
URL string `json:"url" binding:"required"`
|
||||||
|
Filename string `json:"filename"`
|
||||||
|
FileSize int64 `json:"file_size"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *UserLogoService) Save(userID uint, req SaveLogoRequest) (*model.UserLogo, error) {
|
||||||
|
if req.FileSize > MaxLogoFileSize {
|
||||||
|
return nil, ErrLogoTooLarge
|
||||||
|
}
|
||||||
|
|
||||||
|
count, err := s.repo.CountByUser(userID)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
if count >= MaxLogosPerUser {
|
||||||
|
return nil, ErrLogoLimitReached
|
||||||
|
}
|
||||||
|
|
||||||
|
logo := &model.UserLogo{
|
||||||
|
UserID: userID,
|
||||||
|
URL: req.URL,
|
||||||
|
Filename: req.Filename,
|
||||||
|
FileSize: req.FileSize,
|
||||||
|
}
|
||||||
|
if err := s.repo.Create(logo); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return logo, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *UserLogoService) List(userID uint) ([]model.UserLogo, error) {
|
||||||
|
return s.repo.FindByUser(userID)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (s *UserLogoService) Delete(id, userID uint) error {
|
||||||
|
return s.repo.Delete(id, userID)
|
||||||
|
}
|
||||||
@@ -330,6 +330,16 @@ func (h *Handler) ListRelapses(c *gin.Context) {
|
|||||||
c.JSON(http.StatusOK, model.Success(result))
|
c.JSON(http.StatusOK, model.Success(result))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ListDreamPresets 获取管理员预设的梦想目标列表。
|
||||||
|
func (h *Handler) ListDreamPresets(c *gin.Context) {
|
||||||
|
presets, err := h.service.ListDreamPresets(c.Request.Context())
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取预设目标失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{"items": presets}))
|
||||||
|
}
|
||||||
|
|
||||||
// ListRewardGoals 获取梦想目标列表。
|
// ListRewardGoals 获取梦想目标列表。
|
||||||
// @Summary 获取梦想目标列表
|
// @Summary 获取梦想目标列表
|
||||||
// @Description 按状态筛选梦想目标,用于梦想实验室页面展示。
|
// @Description 按状态筛选梦想目标,用于梦想实验室页面展示。
|
||||||
|
|||||||
@@ -0,0 +1,23 @@
|
|||||||
|
package model
|
||||||
|
|
||||||
|
import (
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"gorm.io/gorm"
|
||||||
|
)
|
||||||
|
|
||||||
|
type DreamPreset struct {
|
||||||
|
ID uint `gorm:"primaryKey;comment:主键" json:"id"`
|
||||||
|
CreatedAt time.Time `gorm:"comment:创建时间" json:"created_at"`
|
||||||
|
UpdatedAt time.Time `gorm:"comment:更新时间" json:"updated_at"`
|
||||||
|
DeletedAt gorm.DeletedAt `gorm:"index;comment:删除时间" json:"-"`
|
||||||
|
|
||||||
|
Title string `gorm:"column:title;size:64;comment:图标名称" json:"title"`
|
||||||
|
CoverImage string `gorm:"column:cover_image;size:500;comment:图标(emoji或图片URL)" json:"cover_image"`
|
||||||
|
IsActive bool `gorm:"column:is_active;default:true;comment:是否启用" json:"is_active"`
|
||||||
|
SortOrder int `gorm:"column:sort_order;default:0;comment:排序" json:"sort_order"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (DreamPreset) TableName() string {
|
||||||
|
return "fa_dream_goal_preset"
|
||||||
|
}
|
||||||
@@ -986,6 +986,9 @@ func (s *Service) computeSummary(ctx context.Context, uid int, profile quitmodel
|
|||||||
}
|
}
|
||||||
|
|
||||||
elapsedDays := daysBetween(normalizeDate(profile.QuitStartDate), today)
|
elapsedDays := daysBetween(normalizeDate(profile.QuitStartDate), today)
|
||||||
|
if currentStreak > elapsedDays {
|
||||||
|
elapsedDays = currentStreak
|
||||||
|
}
|
||||||
theoreticalCigs := elapsedDays * profile.BaselineCigsPerDay
|
theoreticalCigs := elapsedDays * profile.BaselineCigsPerDay
|
||||||
avoidedCigs := theoreticalCigs - totalRelapseNum
|
avoidedCigs := theoreticalCigs - totalRelapseNum
|
||||||
if avoidedCigs < 0 {
|
if avoidedCigs < 0 {
|
||||||
@@ -1129,6 +1132,18 @@ func SortRelapses(items []RelapseEventResult) {
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// ListDreamPresets 返回启用的预设梦想目标列表。
|
||||||
|
func (s *Service) ListDreamPresets(ctx context.Context) ([]quitmodel.DreamPreset, error) {
|
||||||
|
var presets []quitmodel.DreamPreset
|
||||||
|
if err := s.db.WithContext(ctx).
|
||||||
|
Where("is_active = ?", true).
|
||||||
|
Order("sort_order ASC, id ASC").
|
||||||
|
Find(&presets).Error; err != nil {
|
||||||
|
return nil, fmt.Errorf("list dream presets: %w", err)
|
||||||
|
}
|
||||||
|
return presets, nil
|
||||||
|
}
|
||||||
|
|
||||||
func normalizeShowFields(items []string) []string {
|
func normalizeShowFields(items []string) []string {
|
||||||
allowed := map[string]struct{}{
|
allowed := map[string]struct{}{
|
||||||
"streak_days": {},
|
"streak_days": {},
|
||||||
|
|||||||
@@ -13,6 +13,7 @@ func registerAdminRoutes(
|
|||||||
categoryHandler *marketinghandler.CategoryHandler,
|
categoryHandler *marketinghandler.CategoryHandler,
|
||||||
templateHandler *marketinghandler.TemplateHandler,
|
templateHandler *marketinghandler.TemplateHandler,
|
||||||
downloadHandler *marketinghandler.DownloadHandler,
|
downloadHandler *marketinghandler.DownloadHandler,
|
||||||
|
adPlacementHandler *marketinghandler.AdPlacementHandler,
|
||||||
) {
|
) {
|
||||||
if handler == nil {
|
if handler == nil {
|
||||||
return
|
return
|
||||||
@@ -89,6 +90,24 @@ func registerAdminRoutes(
|
|||||||
protected.PUT("/smoke/motivation-quotes/:id", handler.UpdateSmokeMotivation)
|
protected.PUT("/smoke/motivation-quotes/:id", handler.UpdateSmokeMotivation)
|
||||||
protected.DELETE("/smoke/motivation-quotes/:id", handler.DeleteSmokeMotivation)
|
protected.DELETE("/smoke/motivation-quotes/:id", handler.DeleteSmokeMotivation)
|
||||||
|
|
||||||
|
protected.GET("/achievement/themes", handler.ListAchievementThemes)
|
||||||
|
protected.GET("/achievement/themes/:id", handler.GetAchievementTheme)
|
||||||
|
protected.POST("/achievement/themes", handler.CreateAchievementTheme)
|
||||||
|
protected.PUT("/achievement/themes/:id", handler.UpdateAchievementTheme)
|
||||||
|
protected.DELETE("/achievement/themes/:id", handler.DeleteAchievementTheme)
|
||||||
|
protected.GET("/achievement/themes/:id/levels", handler.ListAchievementLevels)
|
||||||
|
protected.POST("/achievement/levels", handler.CreateAchievementLevel)
|
||||||
|
protected.PUT("/achievement/levels/:id", handler.UpdateAchievementLevel)
|
||||||
|
protected.DELETE("/achievement/levels/:id", handler.DeleteAchievementLevel)
|
||||||
|
|
||||||
|
protected.GET("/dream-presets", handler.ListDreamPresets)
|
||||||
|
protected.POST("/dream-presets", handler.CreateDreamPreset)
|
||||||
|
protected.PUT("/dream-presets/:id", handler.UpdateDreamPreset)
|
||||||
|
protected.DELETE("/dream-presets/:id", handler.DeleteDreamPreset)
|
||||||
|
|
||||||
|
protected.GET("/quit-checkin/daily-statuses", handler.ListQuitDailyStatuses)
|
||||||
|
protected.GET("/quit-checkin/reward-goals", handler.ListQuitRewardGoals)
|
||||||
|
|
||||||
protected.GET("/memberships/overview", handler.MembershipOverview)
|
protected.GET("/memberships/overview", handler.MembershipOverview)
|
||||||
protected.GET("/memberships/redeem-codes", handler.ListMembershipRedeemCodes)
|
protected.GET("/memberships/redeem-codes", handler.ListMembershipRedeemCodes)
|
||||||
protected.POST("/memberships/redeem-codes", handler.CreateMembershipRedeemCodes)
|
protected.POST("/memberships/redeem-codes", handler.CreateMembershipRedeemCodes)
|
||||||
@@ -107,10 +126,21 @@ func registerAdminRoutes(
|
|||||||
marketing.PUT("/templates/:id", templateHandler.AdminUpdate)
|
marketing.PUT("/templates/:id", templateHandler.AdminUpdate)
|
||||||
marketing.DELETE("/templates/:id", templateHandler.AdminDelete)
|
marketing.DELETE("/templates/:id", templateHandler.AdminDelete)
|
||||||
|
|
||||||
marketing.GET("/stats", downloadHandler.AdminStats)
|
marketing.GET("/stats", downloadHandler.AdminStats)
|
||||||
marketing.POST("/upload/qiniu/token", downloadHandler.AdminQiniuToken)
|
marketing.POST("/upload/oss/token", downloadHandler.AdminUploadToken)
|
||||||
|
marketing.POST("/upload", downloadHandler.AdminUploadFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
if adPlacementHandler != nil {
|
||||||
|
ads := protected.Group("/marketing/ad-placements")
|
||||||
|
{
|
||||||
|
ads.GET("", adPlacementHandler.AdminList)
|
||||||
|
ads.POST("", adPlacementHandler.AdminCreate)
|
||||||
|
ads.PUT("/:id", adPlacementHandler.AdminUpdate)
|
||||||
|
ads.DELETE("/:id", adPlacementHandler.AdminDelete)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
}
|
||||||
|
|||||||
@@ -3,19 +3,16 @@ package routes
|
|||||||
import (
|
import (
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
qiniuhandler "wx_service/internal/common/qiniu/handler"
|
uploadhandler "wx_service/internal/common/upload/handler"
|
||||||
)
|
)
|
||||||
|
|
||||||
func registerCommonPublicRoutes(api *gin.RouterGroup, uploadHandler *qiniuhandler.UploadHandler) {
|
func registerCommonPublicRoutes(api *gin.RouterGroup, uploadHandler *uploadhandler.UploadHandler) {
|
||||||
// 七牛上传回调:由七牛服务端调用,不能挂登录鉴权。
|
api.POST("/common/upload/oss/callback", uploadHandler.UploadCallback)
|
||||||
api.POST("/common/upload/qiniu/callback", uploadHandler.QiniuCallback)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func registerCommonRoutes(protected *gin.RouterGroup, uploadHandler *qiniuhandler.UploadHandler) {
|
func registerCommonRoutes(protected *gin.RouterGroup, uploadHandler *uploadhandler.UploadHandler) {
|
||||||
// 公共接口(所有小程序共用)
|
|
||||||
common := protected.Group("/common")
|
common := protected.Group("/common")
|
||||||
{
|
{
|
||||||
// 七牛直传凭证:前端先拿 token,再直传文件到七牛 upload_url
|
common.POST("/upload/oss/token", uploadHandler.GetUploadToken)
|
||||||
common.POST("/upload/qiniu/token", uploadHandler.QiniuToken)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -13,6 +13,8 @@ func registerMarketingRoutes(
|
|||||||
categoryHandler *marketinghandler.CategoryHandler,
|
categoryHandler *marketinghandler.CategoryHandler,
|
||||||
templateHandler *marketinghandler.TemplateHandler,
|
templateHandler *marketinghandler.TemplateHandler,
|
||||||
downloadHandler *marketinghandler.DownloadHandler,
|
downloadHandler *marketinghandler.DownloadHandler,
|
||||||
|
userLogoHandler *marketinghandler.UserLogoHandler,
|
||||||
|
adPlacementHandler *marketinghandler.AdPlacementHandler,
|
||||||
) {
|
) {
|
||||||
if categoryHandler == nil || templateHandler == nil || downloadHandler == nil {
|
if categoryHandler == nil || templateHandler == nil || downloadHandler == nil {
|
||||||
return
|
return
|
||||||
@@ -23,6 +25,10 @@ func registerMarketingRoutes(
|
|||||||
marketing.GET("/categories", categoryHandler.ListEnabled)
|
marketing.GET("/categories", categoryHandler.ListEnabled)
|
||||||
marketing.GET("/templates", templateHandler.ListEnabled)
|
marketing.GET("/templates", templateHandler.ListEnabled)
|
||||||
marketing.GET("/templates/:id", templateHandler.GetDetail)
|
marketing.GET("/templates/:id", templateHandler.GetDetail)
|
||||||
|
|
||||||
|
if adPlacementHandler != nil {
|
||||||
|
marketing.GET("/ad-config", adPlacementHandler.GetAdConfig)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
protectedMarketing := protected.Group("/marketing")
|
protectedMarketing := protected.Group("/marketing")
|
||||||
@@ -30,6 +36,12 @@ func registerMarketingRoutes(
|
|||||||
protectedMarketing.POST("/downloads", downloadHandler.Create)
|
protectedMarketing.POST("/downloads", downloadHandler.Create)
|
||||||
protectedMarketing.POST("/ad_callback", downloadHandler.AdCallback)
|
protectedMarketing.POST("/ad_callback", downloadHandler.AdCallback)
|
||||||
protectedMarketing.GET("/downloads", downloadHandler.ListByUser)
|
protectedMarketing.GET("/downloads", downloadHandler.ListByUser)
|
||||||
|
|
||||||
|
if userLogoHandler != nil {
|
||||||
|
protectedMarketing.GET("/logos", userLogoHandler.List)
|
||||||
|
protectedMarketing.POST("/logos", userLogoHandler.Save)
|
||||||
|
protectedMarketing.DELETE("/logos/:id", userLogoHandler.Delete)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
admin := api.Group("/admin/marketing")
|
admin := api.Group("/admin/marketing")
|
||||||
@@ -46,6 +58,7 @@ func registerMarketingRoutes(
|
|||||||
admin.DELETE("/templates/:id", templateHandler.AdminDelete)
|
admin.DELETE("/templates/:id", templateHandler.AdminDelete)
|
||||||
|
|
||||||
admin.GET("/stats", downloadHandler.AdminStats)
|
admin.GET("/stats", downloadHandler.AdminStats)
|
||||||
admin.POST("/upload/qiniu/token", downloadHandler.AdminQiniuToken)
|
admin.POST("/upload/oss/token", downloadHandler.AdminUploadToken)
|
||||||
|
admin.POST("/upload", downloadHandler.AdminUploadFile)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -21,6 +21,7 @@ func registerQuitCheckinRoutes(protected *gin.RouterGroup, handler *quitcheckinh
|
|||||||
v2.GET("/badges", handler.ListBadges)
|
v2.GET("/badges", handler.ListBadges)
|
||||||
v2.GET("/relapses", handler.ListRelapses)
|
v2.GET("/relapses", handler.ListRelapses)
|
||||||
|
|
||||||
|
v2.GET("/dream-presets", handler.ListDreamPresets)
|
||||||
v2.GET("/reward-goals", handler.ListRewardGoals)
|
v2.GET("/reward-goals", handler.ListRewardGoals)
|
||||||
v2.POST("/reward-goals", handler.CreateRewardGoal)
|
v2.POST("/reward-goals", handler.CreateRewardGoal)
|
||||||
v2.PUT("/reward-goals/:id", handler.UpdateRewardGoal)
|
v2.PUT("/reward-goals/:id", handler.UpdateRewardGoal)
|
||||||
|
|||||||
@@ -8,7 +8,7 @@ import (
|
|||||||
|
|
||||||
adminhandler "wx_service/internal/admin"
|
adminhandler "wx_service/internal/admin"
|
||||||
authhandler "wx_service/internal/common/auth/handler"
|
authhandler "wx_service/internal/common/auth/handler"
|
||||||
qiniuhandler "wx_service/internal/common/qiniu/handler"
|
uploadhandler "wx_service/internal/common/upload/handler"
|
||||||
rediscache "wx_service/internal/common/redis/cache"
|
rediscache "wx_service/internal/common/redis/cache"
|
||||||
oahandler "wx_service/internal/common/wechat_official/handler"
|
oahandler "wx_service/internal/common/wechat_official/handler"
|
||||||
expiryhandler "wx_service/internal/expiry"
|
expiryhandler "wx_service/internal/expiry"
|
||||||
@@ -29,7 +29,7 @@ func Register(
|
|||||||
smokeHandler *smokehandler.SmokeHandler,
|
smokeHandler *smokehandler.SmokeHandler,
|
||||||
quitPlanHandler *smokehandler.QuitPlanHandler,
|
quitPlanHandler *smokehandler.QuitPlanHandler,
|
||||||
redeemCodeHandler *membershiphandler.RedeemCodeHandler,
|
redeemCodeHandler *membershiphandler.RedeemCodeHandler,
|
||||||
uploadHandler *qiniuhandler.UploadHandler,
|
uploadHandler *uploadhandler.UploadHandler,
|
||||||
oaOAuthHandler *oahandler.OAuthHandler,
|
oaOAuthHandler *oahandler.OAuthHandler,
|
||||||
sessionCache *rediscache.SessionUserCache,
|
sessionCache *rediscache.SessionUserCache,
|
||||||
lawyerHandler *lawyerhandler.LawyerHandler,
|
lawyerHandler *lawyerhandler.LawyerHandler,
|
||||||
@@ -39,6 +39,8 @@ func Register(
|
|||||||
marketingCategoryHandler *marketinghandler.CategoryHandler,
|
marketingCategoryHandler *marketinghandler.CategoryHandler,
|
||||||
marketingTemplateHandler *marketinghandler.TemplateHandler,
|
marketingTemplateHandler *marketinghandler.TemplateHandler,
|
||||||
marketingDownloadHandler *marketinghandler.DownloadHandler,
|
marketingDownloadHandler *marketinghandler.DownloadHandler,
|
||||||
|
marketingUserLogoHandler *marketinghandler.UserLogoHandler,
|
||||||
|
marketingAdPlacementHandler *marketinghandler.AdPlacementHandler,
|
||||||
quitCheckinHandler *quitcheckinhandler.Handler,
|
quitCheckinHandler *quitcheckinhandler.Handler,
|
||||||
) {
|
) {
|
||||||
// Register 用来集中注册所有 HTTP 路由,便于工程结构更清晰:
|
// Register 用来集中注册所有 HTTP 路由,便于工程结构更清晰:
|
||||||
@@ -48,6 +50,8 @@ func Register(
|
|||||||
{
|
{
|
||||||
// 登录接口:用微信 code 换取/创建用户并返回 session_key(作为后续 Bearer Token)
|
// 登录接口:用微信 code 换取/创建用户并返回 session_key(作为后续 Bearer Token)
|
||||||
api.POST("/auth/login", authHandler.LoginWithWeChat)
|
api.POST("/auth/login", authHandler.LoginWithWeChat)
|
||||||
|
// H5 开发调试登录(仅 debug 模式可用)
|
||||||
|
api.POST("/auth/dev-login", authHandler.DevLogin)
|
||||||
|
|
||||||
// 公众号网页授权:不需要登录(code 本身来自微信授权回调)
|
// 公众号网页授权:不需要登录(code 本身来自微信授权回调)
|
||||||
registerWeChatOfficialRoutes(api, oaOAuthHandler)
|
registerWeChatOfficialRoutes(api, oaOAuthHandler)
|
||||||
@@ -62,13 +66,15 @@ func Register(
|
|||||||
protected.Use(middleware.AuthMiddleware(db, sessionCache))
|
protected.Use(middleware.AuthMiddleware(db, sessionCache))
|
||||||
protected.Use(middleware.RequireUserMiddleware())
|
protected.Use(middleware.RequireUserMiddleware())
|
||||||
{
|
{
|
||||||
|
protected.PUT("/auth/profile", authHandler.UpdateProfile)
|
||||||
|
protected.GET("/auth/mini-program-test-code", authHandler.GetMiniProgramTestCode)
|
||||||
registerCommonRoutes(protected, uploadHandler)
|
registerCommonRoutes(protected, uploadHandler)
|
||||||
registerRemoveWatermarkRoutes(api, protected, videoHandler)
|
registerRemoveWatermarkRoutes(api, protected, videoHandler)
|
||||||
registerMembershipRoutes(protected, redeemCodeHandler)
|
registerMembershipRoutes(protected, redeemCodeHandler)
|
||||||
registerSmokeRoutes(protected, smokeHandler, quitPlanHandler)
|
registerSmokeRoutes(protected, smokeHandler, quitPlanHandler)
|
||||||
}
|
}
|
||||||
|
|
||||||
registerMarketingRoutes(api, protected, adminToken, marketingCategoryHandler, marketingTemplateHandler, marketingDownloadHandler)
|
registerMarketingRoutes(api, protected, adminToken, marketingCategoryHandler, marketingTemplateHandler, marketingDownloadHandler, marketingUserLogoHandler, marketingAdPlacementHandler)
|
||||||
}
|
}
|
||||||
|
|
||||||
apiV2 := router.Group("/api/v2")
|
apiV2 := router.Group("/api/v2")
|
||||||
@@ -81,7 +87,7 @@ func Register(
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
registerAdminRoutes(router, adminHandler, marketingCategoryHandler, marketingTemplateHandler, marketingDownloadHandler)
|
registerAdminRoutes(router, adminHandler, marketingCategoryHandler, marketingTemplateHandler, marketingDownloadHandler, marketingAdPlacementHandler)
|
||||||
|
|
||||||
// 保质期提醒模块使用独立前缀 /api/expiry,与现有 /api/v1 并存。
|
// 保质期提醒模块使用独立前缀 /api/expiry,与现有 /api/v1 并存。
|
||||||
expiryAPI := router.Group("/api/expiry")
|
expiryAPI := router.Group("/api/expiry")
|
||||||
|
|||||||
@@ -50,5 +50,9 @@ func registerSmokeRoutes(protected *gin.RouterGroup, smokeHandler *smokehandler.
|
|||||||
smoke.GET("/quit-plan", quitPlanHandler.GetQuitPlan)
|
smoke.GET("/quit-plan", quitPlanHandler.GetQuitPlan)
|
||||||
smoke.GET("/quit-plan/days", quitPlanHandler.GetQuitPlanDays)
|
smoke.GET("/quit-plan/days", quitPlanHandler.GetQuitPlanDays)
|
||||||
smoke.POST("/quit-plan/reset", quitPlanHandler.ResetQuitPlan)
|
smoke.POST("/quit-plan/reset", quitPlanHandler.ResetQuitPlan)
|
||||||
|
|
||||||
|
// 成就系统
|
||||||
|
smoke.GET("/achievement/themes", smokeHandler.ListAchievementThemes)
|
||||||
|
smoke.GET("/achievement", smokeHandler.GetAchievement)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -0,0 +1,69 @@
|
|||||||
|
package handler
|
||||||
|
|
||||||
|
import (
|
||||||
|
"log"
|
||||||
|
"net/http"
|
||||||
|
"time"
|
||||||
|
|
||||||
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
|
"wx_service/internal/achievement"
|
||||||
|
"wx_service/internal/middleware"
|
||||||
|
"wx_service/internal/model"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (h *SmokeHandler) ListAchievementThemes(c *gin.Context) {
|
||||||
|
themes, err := h.achievementService.ListActiveThemes(c.Request.Context())
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取成就主题失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{"themes": themes}))
|
||||||
|
}
|
||||||
|
|
||||||
|
func (h *SmokeHandler) GetAchievement(c *gin.Context) {
|
||||||
|
user := middleware.MustCurrentUser(c)
|
||||||
|
ctx := c.Request.Context()
|
||||||
|
uid := int(user.ID)
|
||||||
|
now := time.Now()
|
||||||
|
|
||||||
|
profile, err := h.smokeProfileService.Get(ctx, uid)
|
||||||
|
if err != nil {
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取用户信息失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if profile == nil || profile.AchievementThemeID == nil {
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{"achievement": nil}))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
var days int
|
||||||
|
if profile.Mode == "quit" {
|
||||||
|
homeData, err := h.quitCheckinService.Home(ctx, uid, now)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("achievement: quitcheckin home err uid=%d: %v", uid, err)
|
||||||
|
days = 0
|
||||||
|
} else {
|
||||||
|
days = homeData.Summary.CurrentStreakDays
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
streakDays, err := h.smokeLogService.GetStreakDays(ctx, uid, now)
|
||||||
|
if err != nil {
|
||||||
|
log.Printf("achievement: smoke streak err uid=%d: %v", uid, err)
|
||||||
|
days = 0
|
||||||
|
}
|
||||||
|
days = streakDays
|
||||||
|
}
|
||||||
|
|
||||||
|
ach, err := h.achievementService.GetUserAchievement(ctx, *profile.AchievementThemeID, days)
|
||||||
|
if err != nil {
|
||||||
|
if err == achievement.ErrThemeNotFound {
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{"achievement": nil}))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
c.JSON(http.StatusInternalServerError, model.Error(http.StatusInternalServerError, "获取成就信息失败"))
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
c.JSON(http.StatusOK, model.Success(gin.H{"achievement": ach}))
|
||||||
|
}
|
||||||
@@ -10,8 +10,10 @@ import (
|
|||||||
|
|
||||||
"github.com/gin-gonic/gin"
|
"github.com/gin-gonic/gin"
|
||||||
|
|
||||||
|
"wx_service/internal/achievement"
|
||||||
"wx_service/internal/middleware"
|
"wx_service/internal/middleware"
|
||||||
"wx_service/internal/model"
|
"wx_service/internal/model"
|
||||||
|
quitcheckinservice "wx_service/internal/quitcheckin/service"
|
||||||
smokeservice "wx_service/internal/smoke/service"
|
smokeservice "wx_service/internal/smoke/service"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -22,6 +24,8 @@ type SmokeHandler struct {
|
|||||||
smokeNextService *smokeservice.SmokeNextService
|
smokeNextService *smokeservice.SmokeNextService
|
||||||
smokeAINextService *smokeservice.SmokeAINextSmokeService
|
smokeAINextService *smokeservice.SmokeAINextSmokeService
|
||||||
smokeShareService *smokeservice.SmokeShareService
|
smokeShareService *smokeservice.SmokeShareService
|
||||||
|
achievementService *achievement.Service
|
||||||
|
quitCheckinService *quitcheckinservice.Service
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewSmokeHandler(
|
func NewSmokeHandler(
|
||||||
@@ -31,6 +35,8 @@ func NewSmokeHandler(
|
|||||||
smokeNextService *smokeservice.SmokeNextService,
|
smokeNextService *smokeservice.SmokeNextService,
|
||||||
smokeAINextService *smokeservice.SmokeAINextSmokeService,
|
smokeAINextService *smokeservice.SmokeAINextSmokeService,
|
||||||
smokeShareService *smokeservice.SmokeShareService,
|
smokeShareService *smokeservice.SmokeShareService,
|
||||||
|
achievementService *achievement.Service,
|
||||||
|
quitCheckinService *quitcheckinservice.Service,
|
||||||
) *SmokeHandler {
|
) *SmokeHandler {
|
||||||
return &SmokeHandler{
|
return &SmokeHandler{
|
||||||
smokeLogService: smokeLogService,
|
smokeLogService: smokeLogService,
|
||||||
@@ -39,6 +45,8 @@ func NewSmokeHandler(
|
|||||||
smokeNextService: smokeNextService,
|
smokeNextService: smokeNextService,
|
||||||
smokeAINextService: smokeAINextService,
|
smokeAINextService: smokeAINextService,
|
||||||
smokeShareService: smokeShareService,
|
smokeShareService: smokeShareService,
|
||||||
|
achievementService: achievementService,
|
||||||
|
quitCheckinService: quitCheckinService,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -27,6 +27,8 @@ type upsertSmokeProfileRequest struct {
|
|||||||
SleepTime *string `json:"sleep_time"`
|
SleepTime *string `json:"sleep_time"`
|
||||||
|
|
||||||
QuitDate *string `json:"quit_date"`
|
QuitDate *string `json:"quit_date"`
|
||||||
|
|
||||||
|
AchievementThemeID *uint `json:"achievement_theme_id"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func (h *SmokeHandler) GetProfile(c *gin.Context) {
|
func (h *SmokeHandler) GetProfile(c *gin.Context) {
|
||||||
@@ -95,17 +97,21 @@ func (h *SmokeHandler) UpsertProfile(c *gin.Context) {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
achievementThemeIDProvided := req.AchievementThemeID != nil
|
||||||
|
|
||||||
view, err := h.smokeProfileService.Upsert(c.Request.Context(), int(user.ID), smokeservice.UpsertSmokeProfileRequest{
|
view, err := h.smokeProfileService.Upsert(c.Request.Context(), int(user.ID), smokeservice.UpsertSmokeProfileRequest{
|
||||||
BaselineCigsPerDay: req.BaselineCigsPerDay,
|
BaselineCigsPerDay: req.BaselineCigsPerDay,
|
||||||
SmokingYears: req.SmokingYears,
|
SmokingYears: req.SmokingYears,
|
||||||
PackPriceCent: req.PackPriceCent,
|
PackPriceCent: req.PackPriceCent,
|
||||||
Mode: req.Mode,
|
Mode: req.Mode,
|
||||||
SmokeMotivations: req.SmokeMotivations,
|
SmokeMotivations: req.SmokeMotivations,
|
||||||
QuitMotivations: req.QuitMotivations,
|
QuitMotivations: req.QuitMotivations,
|
||||||
WakeUpTime: req.WakeUpTime,
|
WakeUpTime: req.WakeUpTime,
|
||||||
SleepTime: req.SleepTime,
|
SleepTime: req.SleepTime,
|
||||||
QuitDateProvided: quitDateProvided,
|
QuitDateProvided: quitDateProvided,
|
||||||
QuitDate: quitDate,
|
QuitDate: quitDate,
|
||||||
|
AchievementThemeIDProvided: achievementThemeIDProvided,
|
||||||
|
AchievementThemeID: req.AchievementThemeID,
|
||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
if errors.Is(err, smokeservice.ErrSmokeProfileInvalidTime) {
|
if errors.Is(err, smokeservice.ErrSmokeProfileInvalidTime) {
|
||||||
|
|||||||
@@ -74,6 +74,8 @@ type SmokeUserProfile struct {
|
|||||||
|
|
||||||
QuitDate *time.Time `gorm:"column:quit_date;type:date;comment:目标戒烟日期" json:"quit_date,omitempty"`
|
QuitDate *time.Time `gorm:"column:quit_date;type:date;comment:目标戒烟日期" json:"quit_date,omitempty"`
|
||||||
|
|
||||||
|
AchievementThemeID *uint `gorm:"column:achievement_theme_id;comment:成就主题ID" json:"achievement_theme_id,omitempty"`
|
||||||
|
|
||||||
OnboardingCompletedAt *time.Time `gorm:"column:onboarding_completed_at;comment:首次补全完成时间" json:"onboarding_completed_at,omitempty"`
|
OnboardingCompletedAt *time.Time `gorm:"column:onboarding_completed_at;comment:首次补全完成时间" json:"onboarding_completed_at,omitempty"`
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -99,6 +99,9 @@ type UpsertSmokeProfileRequest struct {
|
|||||||
|
|
||||||
QuitDateProvided bool
|
QuitDateProvided bool
|
||||||
QuitDate *time.Time
|
QuitDate *time.Time
|
||||||
|
|
||||||
|
AchievementThemeIDProvided bool
|
||||||
|
AchievementThemeID *uint
|
||||||
}
|
}
|
||||||
|
|
||||||
func (s *SmokeProfileService) Upsert(ctx context.Context, uid int, req UpsertSmokeProfileRequest) (SmokeProfileView, error) {
|
func (s *SmokeProfileService) Upsert(ctx context.Context, uid int, req UpsertSmokeProfileRequest) (SmokeProfileView, error) {
|
||||||
@@ -166,6 +169,9 @@ func (s *SmokeProfileService) Upsert(ctx context.Context, uid int, req UpsertSmo
|
|||||||
if req.QuitDateProvided {
|
if req.QuitDateProvided {
|
||||||
profile.QuitDate = req.QuitDate
|
profile.QuitDate = req.QuitDate
|
||||||
}
|
}
|
||||||
|
if req.AchievementThemeIDProvided {
|
||||||
|
profile.AchievementThemeID = req.AchievementThemeID
|
||||||
|
}
|
||||||
|
|
||||||
now := time.Now()
|
now := time.Now()
|
||||||
profile.Mode = normalizedSmokeMode(profile.Mode)
|
profile.Mode = normalizedSmokeMode(profile.Mode)
|
||||||
|
|||||||
@@ -252,6 +252,10 @@ func (s *SmokeLogService) countResisted(ctx context.Context, uid int, start, end
|
|||||||
return int(count), nil
|
return int(count), nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func (s *SmokeLogService) GetStreakDays(ctx context.Context, uid int, asOf time.Time) (int, error) {
|
||||||
|
return s.computeStreakDays(ctx, uid, asOf)
|
||||||
|
}
|
||||||
|
|
||||||
func (s *SmokeLogService) computeStreakDays(ctx context.Context, uid int, asOf time.Time) (int, error) {
|
func (s *SmokeLogService) computeStreakDays(ctx context.Context, uid int, asOf time.Time) (int, error) {
|
||||||
asOf = dateOnly(asOf)
|
asOf = dateOnly(asOf)
|
||||||
start := asOf.AddDate(0, 0, -400)
|
start := asOf.AddDate(0, 0, -400)
|
||||||
|
|||||||
Reference in New Issue
Block a user