admin/wx_service
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Enhance AI and Redis integration for smoke logging features

Browse Source 
- Added AI configuration options to .env.example and config.go for OpenAI integration.
- Implemented Redis caching for session management in main.go and auth middleware.
- Updated smoke logging service to support real smoking time (`smoke_at`) and AI advice retrieval.
- Enhanced API routes to include endpoints for AI advice and unlock functionality for non-members.
- Improved database schema with new tables for AI advice and unlock records.
- Expanded documentation to cover new AI features and Redis caching implementation.
This commit is contained in:
nepiedg
2026-01-03 02:14:21 +00:00
parent 1c48fbdeaf
commit 16844d4a42
30 changed files with 1662 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
docs/README.md
+1
View File
@@ -9,6 +9,7 @@
- `docs/common/response.md`
- `docs/common/upload_qiniu.md`
- `docs/common/wechat_official.md`
- `docs/common/redis.md`
## 去水印小程序
docs/common/README.md
+4
View File
@@ -30,3 +30,7 @@
## 微信公众号
- `docs/common/wechat_official.md`
## Redis
- `docs/common/redis.md`
docs/common/redis.md
+22
View File
@@ -0,0 +1,22 @@
# Redis（可选）
当前项目引入 Redis 的主要目的：缓存 `session_key -> user`，减少每个受保护接口都查询 MySQL（尤其在并发增大时更明显）。
## 什么时候需要
- 访问量变大，MySQL 压力主要来自“鉴权查用户”这种高频读
- 需要做更复杂的能力：限流、计数、分布式锁、异步队列等
## 配置
见 `.env.example`：
- `REDIS_ADDR`
- `REDIS_PASSWORD`
- `REDIS_DB`
- `REDIS_KEY_PREFIX`
- `REDIS_SESSION_TTL_SECONDS`
说明：
- 如果不配置 `REDIS_ADDR`，程序会自动禁用 Redis，行为与之前一致。
docs/membership/API.md
+71
View File
@@ -0,0 +1,71 @@
# 会员兑换码 API
所有接口前缀：`/api/v1`
除登录外都需要：`Authorization: Bearer <session_key>`（见：`docs/common/auth.md`）
## 1) 生成兑换码（后台口令）
`POST /api/v1/membership/redeem_codes`
Header：
- `X-Admin-Token: <ADMIN_API_TOKEN>`
请求体：
```json
{
"count": 10,
"plan": "month",
"duration_days": 30,
"expires_at": "2026-12-31 23:59:59",
"max_uses": 1
}
```
说明：
- 需要在环境变量配置 `ADMIN_API_TOKEN`，否则接口会返回 503。
- 兑换码明文只在生成时返回；服务端只保存 `sha256(code)`。
成功响应（示例）：
```json
{
"code": 200,
"message": "success",
"data": {
"count": 10,
"codes": [
{ "code": "ABCD...1234", "plan": "month" }
]
}
}
```
## 2) 使用兑换码开通/延长会员
`POST /api/v1/membership/redeem`
请求体：
```json
{
"code": "ABCD...1234"
}
```
成功响应（示例）：
```json
{
"code": 200,
"message": "success",
"data": {
"plan": "month",
"starts_at": "2026-01-03T10:00:00+08:00",
"ends_at": "2026-02-02T10:00:00+08:00",
"extended": false,
"code_suffix": "X7K9Q2"
}
}
```
常见错误：
- `400`：兑换码无效/过期/已被使用/不可用
- `401`：未登录或 token 失效
docs/smoke/API.md
+54 -1
View File
@@ -12,6 +12,7 @@
```json
{
"smoke_time": "2025-12-31",
"smoke_at": "2025-12-31 08:30:00",
"remark": "压力大",
"level": 2,
"num": 3
@@ -20,6 +21,7 @@
说明：
- `smoke_time` 可选；不传则默认“当天”。
- `smoke_at` 可选；真实抽烟时间（格式 `YYYY-MM-DD HH:MM:SS`）。用于“按时间节点分析/AI 建议”；不传则可用 `createtime` 近似。
- `level/num` 可选；不传或传 `<=0` 时会按 `1` 处理。
curl 示例：
@@ -28,7 +30,7 @@ curl 示例：
curl -X POST 'http://127.0.0.1:8080/api/v1/smoke/logs' \
-H 'Content-Type: application/json' \
-H 'Authorization: Bearer wx-session-key' \
-d '{"smoke_time":"2025-12-31","remark":"压力大","level":2,"num":3}'
-d '{"smoke_time":"2025-12-31","smoke_at":"2025-12-31 08:30:00","remark":"压力大","level":2,"num":3}'
```
成功响应示例（字段以实际为准）：
@@ -40,6 +42,7 @@ curl -X POST 'http://127.0.0.1:8080/api/v1/smoke/logs' \
"data": {
"id": 5202,
"smoke_time": "2025-12-31T00:00:00+08:00",
"smoke_at": "2025-12-31T08:30:00+08:00",
"remark": "压力大",
"createtime": 1735600000,
"updatetime": 1735600000,
@@ -94,6 +97,7 @@ curl -X GET 'http://127.0.0.1:8080/api/v1/smoke/logs/5202' \
```json
{
"smoke_time": "2026-01-01",
"smoke_at": "2026-01-01 21:10:00",
"remark": "聚会",
"level": 3,
"num": 1
@@ -102,6 +106,7 @@ curl -X GET 'http://127.0.0.1:8080/api/v1/smoke/logs/5202' \
注意：
- 如果你想“清空 smoke_time”，请传空字符串：`{"smoke_time":""}`。
- 如果你想“清空 smoke_at”，请传空字符串：`{"smoke_at":""}`。
- 如果传 `null` 或者不传 `smoke_time`，后端会认为你没有修改该字段。
## 5) 删除记录（软删除）
@@ -120,3 +125,51 @@ curl -X GET 'http://127.0.0.1:8080/api/v1/smoke/logs/5202' \
}
```
## 6) 获取 AI 戒烟建议（会员 + 广告解锁并行）
`GET /api/v1/smoke/ai/advice?date=2026-01-02`
说明：
- `date` 可选，默认“昨天”（建议针对哪一天的数据）。
- 权限：会员用户直接可用；非会员需要先对该 `date` 完成“看广告解锁”（见下一个接口）。
- 建议结果会按 `uid + date + prompt_version` 缓存（表：`fa_smoke_ai_advice`）。
未满足权限时的建议响应（示例）：
```json
{
"code": 403,
"message": "需要会员或观看广告解锁后才可生成建议",
"data": {
"date": "2026-01-02",
"need": "vip_or_ad"
}
}
```
成功响应（示例）：
```json
{
"code": 200,
"message": "success",
"data": {
"date": "2026-01-02",
"advice": "..."
}
}
```
## 7) 看广告解锁（用于非会员）
`POST /api/v1/smoke/ai/advice_unlocks`
请求体：
```json
{
"date": "2026-01-02",
"ad_watched_at": "2026-01-03 09:00:00"
}
```
说明：
- 该接口用于记录“已完成观看广告”，落库到 `fa_smoke_ai_advice_unlocks`（`uid + unlock_date` 唯一）。
- `ad_watched_at` 可由后端取当前时间；如需审计/对账可保留前端上报并做校验。
docs/smoke/README.md
+32
View File
@@ -15,3 +15,35 @@
- 该表使用旧系统字段：`createtime/updatetime/deletetime`（秒级时间戳），并非 GORM 默认的 `created_at/updated_at/deleted_at`。
- 接口层通过 Token 识别用户，`uid` 由后端从登录用户推导，不允许前端传入。
### 真实抽烟时间（推荐使用 `smoke_at`）
为支持“按时间节点分析”（例如：昨天哪些时段更容易想抽），建议在 `fa_smoke_log` 中新增：
- `smoke_at`：真实抽烟时间（精确到时分秒，可补录）
数据约定建议：
- 前端若提供了真实时间：写入 `smoke_at`；并同步写 `smoke_time = date(smoke_at)`，方便沿用现有按天筛选。
- 前端未提供真实时间：`smoke_at=NULL`，时间节点可用 `createtime` 作为近似（但补录会造成偏差）。
### AI 戒烟建议（会员 + 广告解锁并行）
面向会员用户提供“昨日 AI 建议”；非会员用户在完成当日/指定日期的“看广告解锁”后也可生成建议。
涉及表（DDL 见：`docs/sql/smoke.sql`）：
- `fa_smoke_ai_advice`：按 `uid + advice_date + prompt_version` 缓存建议结果，避免重复调用 AI。
- `fa_smoke_ai_advice_unlocks`：非会员用户的“每日解锁”记录（按 `uid + unlock_date` 唯一）。
建议的权限判断顺序：
1) 若用户是会员：直接允许生成/获取建议。
2) 否则：查询 `fa_smoke_ai_advice_unlocks` 是否已对 `unlock_date=昨天（或请求 date）` 解锁；已解锁则允许。
3) 否则：拒绝并提示“开通会员或观看广告解锁”。
会员判断建议使用通用会员表：
- `user_memberships`（DDL 见：`docs/sql/users.sql`）
无支付系统时，可用兑换码开通会员：
- API：`docs/membership/API.md`
- DDL：`docs/sql/membership.sql`
给 AI 的输入（最小必需）建议包含：
- 昨日总量：`SUM(num)`
- 时间节点列表：按 `COALESCE(smoke_at, FROM_UNIXTIME(createtime))` 排序的多条记录（每条带 `num/level/remark`）
docs/sql/membership.sql
+45
View File
@@ -0,0 +1,45 @@
-- 会员兑换码（用于无支付系统时的会员开通）
CREATE TABLE IF NOT EXISTS `membership_redeem_codes` (
`id` bigint unsigned NOT NULL AUTO_INCREMENT,
`created_at` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
`updated_at` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
`deleted_at` datetime DEFAULT NULL,
`code_hash` varchar(64) NOT NULL COMMENT 'sha256(code) hex；不存明文',
`code_suffix` varchar(16) DEFAULT NULL COMMENT '展示/审计用后缀',
`plan` varchar(30) NOT NULL DEFAULT 'default',
`duration_days` int NOT NULL,
`expires_at` datetime DEFAULT NULL,
`max_uses` int NOT NULL DEFAULT 1,
`used_uses` int NOT NULL DEFAULT 0,
`status` varchar(20) NOT NULL DEFAULT 'active' COMMENT 'active/disabled',
PRIMARY KEY (`id`),
UNIQUE KEY `uniq_redeem_code_hash` (`code_hash`),
KEY `idx_redeem_code_suffix` (`code_suffix`),
KEY `idx_redeem_code_deleted` (`deleted_at`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;
CREATE TABLE IF NOT EXISTS `membership_redemptions` (
`id` bigint unsigned NOT NULL AUTO_INCREMENT,
`created_at` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
`updated_at` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
`deleted_at` datetime DEFAULT NULL,
`mini_program_id` bigint unsigned NOT NULL,
`user_id` bigint unsigned NOT NULL,
`redeem_code_id` bigint unsigned NOT NULL,
`code_suffix` varchar(16) DEFAULT NULL,
`membership_id` bigint unsigned NOT NULL,
`client_ip` varchar(64) DEFAULT NULL,
`user_agent` varchar(255) DEFAULT NULL,
PRIMARY KEY (`id`),
KEY `idx_redemption_user_date` (`mini_program_id`,`user_id`,`created_at`),
KEY `idx_redemption_code` (`redeem_code_id`),
KEY `idx_redemption_deleted` (`deleted_at`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;
docs/sql/smoke.sql
+41 -1
View File
@@ -5,12 +5,52 @@ CREATE TABLE `fa_smoke_log` (
`id` int(11) NOT NULL AUTO_INCREMENT,
`uid` int(11) NOT NULL COMMENT '用户ID',
`smoke_time` date DEFAULT NULL COMMENT '抽烟时间',
`smoke_at` datetime DEFAULT NULL COMMENT '真实抽烟时间（可补录，精确到时分秒；为空则可用 createtime 近似）',
`remark` text COMMENT '抽烟原因',
`createtime` int(11) DEFAULT NULL COMMENT '创建时间',
`updatetime` int(11) DEFAULT NULL COMMENT '修改时间',
`deletetime` int(11) DEFAULT NULL COMMENT '删除时间',
`level` bigint(2) DEFAULT '1' COMMENT '烟瘾程度',
`num` int(2) DEFAULT '1' COMMENT '抽烟数量',
PRIMARY KEY (`id`,`uid`)
PRIMARY KEY (`id`,`uid`),
KEY `idx_smoke_uid_date` (`uid`,`smoke_time`),
KEY `idx_smoke_uid_at` (`uid`,`smoke_at`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='抽烟记录';
-- AI 戒烟建议（按天缓存，避免重复调用；通常 date=昨天）
-- 说明：
-- - access gating：会员用户直接可用；非会员需要先完成“看广告解锁”（见 fa_smoke_ai_advice_unlocks）
-- - input_snapshot：建议将“昨日总量 + 时间节点列表”作为 JSON 落库，便于追溯与复现
CREATE TABLE IF NOT EXISTS `fa_smoke_ai_advice` (
`id` bigint unsigned NOT NULL AUTO_INCREMENT,
`uid` int(11) NOT NULL COMMENT '用户ID',
`advice_date` date NOT NULL COMMENT '建议针对的日期（通常=昨天）',
`prompt_version` varchar(30) NOT NULL DEFAULT 'v1' COMMENT '提示词版本',
`provider` varchar(30) DEFAULT NULL COMMENT 'AI 提供方（可选）',
`model` varchar(60) DEFAULT NULL COMMENT '模型名（可选）',
`input_snapshot` json NOT NULL COMMENT '当次输入快照（昨日总量+节点）',
`advice` mediumtext NOT NULL COMMENT 'AI 建议内容（文本/Markdown）',
`tokens_in` int DEFAULT NULL,
`tokens_out` int DEFAULT NULL,
`cost_cent` int DEFAULT NULL COMMENT '成本（分，可选）',
`createtime` int(11) DEFAULT NULL COMMENT '创建时间',
`updatetime` int(11) DEFAULT NULL COMMENT '修改时间',
`deletetime` int(11) DEFAULT NULL COMMENT '删除时间',
PRIMARY KEY (`id`),
UNIQUE KEY `uniq_smoke_ai_advice` (`uid`,`advice_date`,`prompt_version`),
KEY `idx_smoke_ai_advice_uid_date` (`uid`,`advice_date`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='每日AI戒烟建议';
-- 每日广告解锁记录（非会员：当天/指定日期解锁后即可生成建议）
CREATE TABLE IF NOT EXISTS `fa_smoke_ai_advice_unlocks` (
`id` bigint unsigned NOT NULL AUTO_INCREMENT,
`uid` int(11) NOT NULL COMMENT '用户ID',
`unlock_date` date NOT NULL COMMENT '解锁的建议日期（通常=昨天）',
`ad_watched_at` datetime NOT NULL COMMENT '完成观看时间',
`createtime` int(11) DEFAULT NULL COMMENT '创建时间',
`updatetime` int(11) DEFAULT NULL COMMENT '修改时间',
`deletetime` int(11) DEFAULT NULL COMMENT '删除时间',
PRIMARY KEY (`id`),
UNIQUE KEY `uniq_smoke_ai_unlock` (`uid`,`unlock_date`),
KEY `idx_smoke_ai_unlock_uid_date` (`uid`,`unlock_date`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COMMENT='AI戒烟建议-广告解锁';
docs/sql/users.sql
+20
View File
@@ -32,3 +32,23 @@ CREATE TABLE IF NOT EXISTS `users` (
KEY `idx_users_deleted_at` (`deleted_at`),
CONSTRAINT `fk_users_mini_program` FOREIGN KEY (`mini_program_id`) REFERENCES `mini_programs`(`id`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;
-- 会员订阅（通用能力：可给多个功能做“会员优先/免广告”策略）
-- 说明：
-- - 会员判断：存在 `status='active' AND ends_at > NOW()` 的记录即可视为会员
-- - 如果你不想引入该表，也可以选择在 users 表新增 `vip_expires_at` 字段（但扩展性较弱）
CREATE TABLE IF NOT EXISTS `user_memberships` (
`id` bigint unsigned NOT NULL AUTO_INCREMENT,
`created_at` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP,
`updated_at` datetime NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP,
`deleted_at` datetime DEFAULT NULL,
`mini_program_id` bigint unsigned NOT NULL,
`user_id` bigint unsigned NOT NULL,
`plan` varchar(30) NOT NULL COMMENT '例如：month/year/lifetime',
`status` varchar(20) NOT NULL COMMENT 'active/canceled/expired',
`starts_at` datetime NOT NULL,
`ends_at` datetime NOT NULL,
PRIMARY KEY (`id`),
KEY `idx_membership_user` (`mini_program_id`,`user_id`,`ends_at`),
KEY `idx_membership_status` (`mini_program_id`,`user_id`,`status`,`ends_at`)
) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_unicode_ci;