feat: 实现客户端注册与任务领取接口
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>
This commit is contained in:
@@ -9,9 +9,14 @@ package api
|
||||
|
||||
import (
|
||||
"database/sql"
|
||||
"encoding/json"
|
||||
"log"
|
||||
"net/http"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
|
||||
"cmautobuy/admin/model"
|
||||
"cmautobuy/admin/service"
|
||||
)
|
||||
|
||||
// Handler 持有接口共用的依赖。
|
||||
@@ -75,17 +80,81 @@ func (h *Handler) Claim(c *gin.Context) {
|
||||
return
|
||||
}
|
||||
|
||||
// TODO(骨架): 1. service.RegisterClient(h.db, clientID, req)
|
||||
// 新 client_id 就新增,已有就更新 device/capabilities/last_seen_at。
|
||||
// name 若已被人工改过,不要覆盖。
|
||||
// 1. 注册/更新客户端。**注册就在这里做,没有单独的注册接口。**
|
||||
// capabilities 原样存起来,将来查问题时能看到客户端当时声明了什么。
|
||||
caps, _ := json.Marshal(req.Capabilities)
|
||||
err := service.RegisterClient(h.db, model.Client{
|
||||
ClientID: clientID,
|
||||
Name: req.Client.Name, // 为空时 repository 会用 clientID 兜底
|
||||
DeviceAddress: req.Device.Address,
|
||||
Platform: req.Device.Platform,
|
||||
PddPackage: req.Device.PddPackage,
|
||||
Capabilities: string(caps),
|
||||
})
|
||||
if err != nil {
|
||||
log.Printf("client_register_failed client_id=%s err=%v", clientID, err)
|
||||
apiError(c, http.StatusInternalServerError, "CLIENT_REGISTER_FAILED",
|
||||
"登记客户端失败", true)
|
||||
return
|
||||
}
|
||||
|
||||
// TODO(骨架): 2. service.ClaimNextTask(h.db, clientID)
|
||||
// 只返回分配给这个客户端的任务,一次一条。
|
||||
// UPDATE ... WHERE task_id = ? AND status = 'assigned'
|
||||
// 检查影响行数,为 0 说明被抢先了,取下一条。
|
||||
// 2. 领取一个任务,只会拿到分配给这个客户端的
|
||||
task, err := service.ClaimNextTask(h.db, clientID, req.SupportedTypes)
|
||||
if err != nil {
|
||||
log.Printf("task_claim_failed client_id=%s err=%v", clientID, err)
|
||||
apiError(c, http.StatusInternalServerError, "TASK_CLAIM_FAILED",
|
||||
"领取任务失败", true)
|
||||
return
|
||||
}
|
||||
|
||||
// 新客户端第一次来必然拿不到任务(还没人给它分配),返回 204 是正常的。
|
||||
c.Status(http.StatusNoContent)
|
||||
// 3. 没有可领的任务 -> 204 No Content,**不是 200 加空对象**。
|
||||
// 新客户端第一次来必然走到这里(还没人给它分配),是正常的。
|
||||
if task == nil {
|
||||
c.Status(http.StatusNoContent)
|
||||
return
|
||||
}
|
||||
|
||||
log.Printf("task_claimed client_id=%s task_id=%s type=%s",
|
||||
clientID, task.TaskID, task.TaskType)
|
||||
c.JSON(http.StatusOK, gin.H{"task": taskPayload(task)})
|
||||
}
|
||||
|
||||
// taskPayload 把任务转成契约里的响应结构。
|
||||
//
|
||||
// **不含租约,也不含 Admin 侧状态**——Client 不关心这些,
|
||||
// 见 docs/client/04-admin-api-contract.md §5。
|
||||
func taskPayload(t *model.Task) gin.H {
|
||||
payload := gin.H{
|
||||
"goods_url": t.PddGoodsURL,
|
||||
}
|
||||
if t.PddGoodsID != "" {
|
||||
payload["goods_id"] = t.PddGoodsID
|
||||
}
|
||||
if t.PddOptions != "" {
|
||||
// pdd_options 存的是 JSON 字符串,要还原成对象再嵌进去,
|
||||
// 否则 Client 拿到的是一个字符串而不是 {"color":...}
|
||||
var options map[string]any
|
||||
if err := json.Unmarshal([]byte(t.PddOptions), &options); err == nil {
|
||||
payload["options"] = options
|
||||
}
|
||||
}
|
||||
// 采购任务必须带数量和价格上限,这是 Client 的价格保护
|
||||
if t.Quantity > 0 {
|
||||
payload["quantity"] = t.Quantity
|
||||
}
|
||||
if t.MaxPriceCent > 0 {
|
||||
payload["max_price_cent"] = t.MaxPriceCent
|
||||
}
|
||||
|
||||
return gin.H{
|
||||
"id": t.TaskID,
|
||||
"type": t.TaskType,
|
||||
"version": t.Version,
|
||||
"priority": t.Priority,
|
||||
"payload": payload,
|
||||
"created_at": t.CreatedAt,
|
||||
"updated_at": t.UpdatedAt,
|
||||
}
|
||||
}
|
||||
|
||||
// SubmitResult 接收成功结果。
|
||||
|
||||
Reference in New Issue
Block a user