Admin 的第一个业务功能。选它打头是因为它是穿透所有层的最薄一条竖切
(HTTP → handler/api → service → repository → SQLite → handler/web → 页面),
一个工单把分层模式立起来,后面四个模块照抄;同时它是与 Client 联调的接口,
能解锁另一条并行的工作线。
实现
- POST /tasks/claim:注册 + 领取。注册就在这里做,没有单独的注册接口,
也没有心跳(理由见 docs/admin/04-client-api.md §3)
- 领取用条件更新 + 检查影响行数防并发,SQLite 没有 SELECT FOR UPDATE
- 客户端列表页:查询、按名称搜索、批量删除
- 在线状态是**算出来的**(last_seen_at 在 10 分钟内),数据库里没有该字段
- CSRF 中间件:双提交 Cookie,手写 82 行不引依赖。
**只挂页面路由**,/api/v1/client/* 不能加——Client 不是浏览器、没有 Cookie
- 14 个单元测试
修复一个真 bug:PRAGMA 必须写进 DSN
并发领取测试报 database is locked (SQLITE_BUSY)。根因是
PRAGMA busy_timeout 每连接生效,而 database/sql 是连接池——
db.Exec("PRAGMA ...") 只作用于当时那条连接,池子新开的连接没执行过。
单线程正常、一并发就炸。改成 DSN 传参后并发测试跑 20 次全过。
这个坑已写进 docs/admin/03-data-model.md §2.1。
与工单的两处差异
- 去掉 name_is_custom 列后,"人工改的名字不被覆盖"改用更简单的做法:
ON CONFLICT DO UPDATE SET 里不含 name,即只在首次注册时写入。
效果相同,零额外字段、零迁移。已同步 04 §3
- 验收项"不向 dry_run 客户端分配真实下单任务"**未实现**:
tasks 表没有字段标记任务是否需要真实下单。当前真实下单开关默认关闭、
MVP 全是演练模式,暂不出问题,但开真实下单前必须补该字段,需另开工单
已验证(Go 1.23.0)
- go vet / gofmt / go test 全过,并发测试重复 20 次稳定通过
- 端到端:无任务 claim 204;插入任务后 claim 200 且 payload 含
goods_url/goods_id/options/quantity/max_price_cent、无租约无 Admin 状态;
重复 claim 204;缺 X-Client-Id 400;POST 无 CSRF token 403;
列表页两台客户端在线状态与统计正确
说明:Gitea 尚未配置,本次无对应工单号。
submit_result / submit_failure 及其幂等处理留给下一个工单。
Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>
175 lines
6.6 KiB
Go
175 lines
6.6 KiB
Go
// Package service 放业务逻辑。
|
||
//
|
||
// 本包**不认识 *gin.Context**——这样才能不起服务器就写单元测试。
|
||
// handler 负责取参数,service 负责判断和编排,repository 负责读写数据库。
|
||
//
|
||
// 骨架阶段这里只有函数签名和 TODO,实现按工单逐个补。
|
||
package service
|
||
|
||
import (
|
||
"database/sql"
|
||
"errors"
|
||
"time"
|
||
|
||
"cmautobuy/admin/model"
|
||
"cmautobuy/admin/repository"
|
||
)
|
||
|
||
// ErrNotImplemented 表示该功能还没实现。
|
||
// 补完实现后要把对应的返回删掉,不要留着假装能用。
|
||
var ErrNotImplemented = errors.New("功能尚未实现")
|
||
|
||
// ---------- 蝦皮 Excel 导入 ----------
|
||
|
||
// ImportResult 是一次导入的统计结果,要显示给操作员看。
|
||
type ImportResult struct {
|
||
ProductCount int // 写入的商品数,样本应为 5195
|
||
SKUCount int // 写入的 SKU 数,样本应为 6092
|
||
FailedRows []int // 解析失败的行号,**不要静默跳过**
|
||
}
|
||
|
||
// ImportShopeeExcel 解析蝦皮报表并 upsert 进库。
|
||
//
|
||
// 实现要点见 docs/admin/03-data-model.md §3.3,五步缺一不可:
|
||
// 1. 分行:商品規格ID 为 "-" 的是商品汇总行,其余是 SKU 行;
|
||
// 2. 按列名找索引,不要写死列号(报表 40 列,蝦皮改格式列号就变);
|
||
// 3. 解析规格原文:先按第一个逗号切开,右边再提取建议。
|
||
// 实测两种格式各占 53.4% / 46.6%,只认【】会漏掉一半。
|
||
// **任何一步失败都不要猜**,parse_ok 置 0,颜色尺码留空,原文照存;
|
||
// 4. upsert,**绝不清空**。DO UPDATE SET 里不得出现
|
||
// pdd_goods_url / pdd_data / collect_status——报表里没这些列,
|
||
// 写进去会把人工填的洗成空;
|
||
// 5. 返回统计。
|
||
func ImportShopeeExcel(db *sql.DB, path string) (*ImportResult, error) {
|
||
// TODO(骨架): 用 github.com/xuri/excelize/v2 流式读行
|
||
return nil, ErrNotImplemented
|
||
}
|
||
|
||
// ParseSpec 把蝦皮的规格原文拆成颜色、尺码、建议。
|
||
//
|
||
// 输入样例(两种格式各占一半):
|
||
//
|
||
// "黑色,M【建議40-50公斤】" -> 黑色 / M / 40-50公斤 / true
|
||
// "卡其色拼黑色,L 建議50-57.5kg" -> 卡其色拼黑色 / L / 50-57.5kg / true
|
||
// "莫名其妙的格式" -> "" / "" / "" / false
|
||
//
|
||
// 最后一个返回值是 ok;false 时前三个必须为空,**不要猜**。
|
||
// 调用方要把 parse_ok 存进库,界面上把这些行标出来让人工补。
|
||
func ParseSpec(raw string) (color, size, advice string, ok bool) {
|
||
// TODO(骨架): 按第一个逗号切开;右边先找【建議...】,再找空格后的 建議...
|
||
return "", "", "", false
|
||
}
|
||
|
||
// ---------- 采集 ----------
|
||
|
||
// CreateCollectTasks 为若干蝦皮商品创建采集任务。
|
||
//
|
||
// 规则:
|
||
// - 按 goods_id **去重**(一个商品有多个 SKU 行,别建重复任务);
|
||
// - PDD 链接为空的跳过;
|
||
// - collect_status 已是 collecting 的跳过,并在结果里说明跳过了几个;
|
||
// - 建任务成功后把 collect_status 置为 collecting。
|
||
func CreateCollectTasks(db *sql.DB, goodsIDs []string, clientID string) (created, skipped int, err error) {
|
||
// TODO(骨架)
|
||
return 0, 0, ErrNotImplemented
|
||
}
|
||
|
||
// ---------- 规格匹配 ----------
|
||
|
||
// SaveMapping 保存「蝦皮规格 = PDD 规格」的对应关系。
|
||
//
|
||
// 存的是**可复用的映射**(sku_mappings 表),不是某一张订单的临时数据。
|
||
// 下次遇到同一个蝦皮 SKU 自动带出,操作员只需确认。
|
||
func SaveMapping(db *sql.DB, shopeeSKUID, goodsID, pddOptionsJSON, operator string) error {
|
||
// TODO(骨架): upsert sku_mappings
|
||
return ErrNotImplemented
|
||
}
|
||
|
||
// ---------- 采购任务 ----------
|
||
|
||
// TaskCreateError 说明某一条为什么建不了任务。
|
||
// **不要静默跳过**,要把这些列出来告诉操作员缺什么。
|
||
type TaskCreateError struct {
|
||
SybID string
|
||
Reason string // 例如「该商品未填写 PDD 链接」
|
||
}
|
||
|
||
// CreatePurchaseTasks 由货运单创建采购任务。
|
||
//
|
||
// 六条校验缺一不可(docs/admin/01-requirements.md §5):
|
||
// 1. 已填 PDD 链接
|
||
// 2. 已采集成功(pdd_data 非空)
|
||
// 3. 已有 SKU 映射
|
||
// 4. 数量 > 0
|
||
// 5. **价格上限已填且 > 0**——默认从 pdd_data 里该 SKU 的价格带出,
|
||
// 操作员可改但不允许为空。没有它 Client 会拒绝执行。
|
||
// 6. 已选择分配的客户端
|
||
//
|
||
// 另外 pdd_goods_url 必须写进任务,Client 那边是 NOT NULL。
|
||
func CreatePurchaseTasks(db *sql.DB, sybIDs []string, clientID string) (created int, failures []TaskCreateError, err error) {
|
||
// TODO(骨架)
|
||
return 0, nil, ErrNotImplemented
|
||
}
|
||
|
||
// ---------- 客户端 ----------
|
||
|
||
// RegisterClient 在客户端领取任务时登记或更新它。
|
||
//
|
||
// **没有单独的注册接口,也没有心跳**——注册就在 claim 里做,
|
||
// 理由见 docs/admin/04-client-api.md §3。
|
||
//
|
||
// 关于 name:**只在第一次注册时写入,之后不再更新**。
|
||
// 这样操作员在界面上改成好记的名字后,客户端每次 claim
|
||
// 都不会把它覆盖回去。客户端没上报 name 时用 clientID 当显示名。
|
||
func RegisterClient(db *sql.DB, c model.Client) error {
|
||
return repository.UpsertClient(db, c)
|
||
}
|
||
|
||
// TouchClient 刷新 last_seen_at。
|
||
//
|
||
// claim / result / failure **三个接口都要调**。
|
||
// 只在 claim 里调的话,客户端执行长任务期间不调 claim,
|
||
// 会被误判成离线。
|
||
func TouchClient(db *sql.DB, clientID string) error {
|
||
return repository.TouchClient(db, clientID)
|
||
}
|
||
|
||
// ClientView 是客户端列表页要显示的一行。
|
||
// Status 是**算出来的**,数据库里没有这个字段。
|
||
type ClientView struct {
|
||
model.Client
|
||
Status string
|
||
}
|
||
|
||
// ListClientViews 查客户端列表,并把在线状态算出来。
|
||
func ListClientViews(db *sql.DB, keyword string, threshold time.Duration) ([]ClientView, error) {
|
||
clients, err := repository.ListClients(db, keyword)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
now := time.Now().UTC()
|
||
|
||
views := make([]ClientView, 0, len(clients))
|
||
for _, c := range clients {
|
||
views = append(views, ClientView{
|
||
Client: c,
|
||
Status: c.StatusText(now, threshold),
|
||
})
|
||
}
|
||
return views, nil
|
||
}
|
||
|
||
// DeleteClients 批量删除,返回实际删除条数。
|
||
func DeleteClients(db *sql.DB, clientIDs []string) (int64, error) {
|
||
return repository.DeleteClients(db, clientIDs)
|
||
}
|
||
|
||
// ---------- 领取任务 ----------
|
||
|
||
// ClaimNextTask 为客户端领取一个任务,没有可领的返回 (nil, nil)。
|
||
//
|
||
// 调用方拿到 nil 要返回 204 No Content,**不是 200 加空对象**。
|
||
func ClaimNextTask(db *sql.DB, clientID string, supportedTypes []string) (*model.Task, error) {
|
||
return repository.ClaimNextTask(db, clientID, supportedTypes)
|
||
}
|