// Package service 放业务逻辑。 // // 本包**不认识 *gin.Context**——这样才能不起服务器就写单元测试。 // handler 负责取参数,service 负责判断和编排,repository 负责读写数据库。 // // 骨架阶段这里只有函数签名和 TODO,实现按工单逐个补。 package service import ( "database/sql" "errors" "fmt" "strings" "time" "cmautobuy/admin/model" "cmautobuy/admin/repository" ) // ErrNotImplemented 表示该功能还没实现。 // 补完实现后要把对应的返回删掉,不要留着假装能用。 var ErrNotImplemented = errors.New("功能尚未实现") // ErrTaskNotVisible 同时表示任务不存在和不在当前账号范围,避免泄露他人任务。 var ErrTaskNotVisible = errors.New("任务不存在或不在当前账号可见范围") // ---------- 采集 ---------- // 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 } // ---------- 采购任务 ---------- // TaskCreateError 说明某一条为什么建不了任务。 // **不要静默跳过**,要把这些列出来告诉操作员缺什么。 type TaskCreateError struct { SybID string Reason string // 例如「该商品未填写 PDD 链接」 } // ---------- 客户端 ---------- // RegisterClient 登记或更新一台客户端。 // // explicit 的含义见 repository.UpsertClient 的说明: // true = 设置页点保存(会更新名称),false = claim 顺带(不更新名称)。 func RegisterClient(db repository.Execer, c model.Client, explicit bool) error { return repository.UpsertClient(db, c, explicit) } // RegisterClientProfile 处理设置页发起的**显式登记**。 // // `[必须]` 本函数**不读取、不领取、不修改任何任务**,也不返回任务。 // 这正是它存在的理由:设置页点"保存"不该顺带把一个任务领走—— // 领走了 Admin 就把任务标成 claimed 了,而保存动作没有义务 // 去可靠保存那个任务,任务就丢了。 func RegisterClientProfile(db *sql.DB, clientID string, p ClientProfileRequest) (string, error) { if clientID == "" { return "", fmt.Errorf("client_id 不能为空") } if err := p.Validate(); err != nil { return "", err } if err := RegisterClient(db, p.ToClient(clientID), true); err != nil { return "", err } return model.NowISO(), nil } // 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 PurchaseMode string AssignedUserID string AssignedUsername string } // ListClientViews 查客户端列表,并把在线状态算出来。 func ListClientViews(db *sql.DB, keyword string, threshold time.Duration) ([]ClientView, error) { return listClientViews(db, keyword, "", threshold) } // ListClientViewsForUser 按网页登录身份限制可见范围:管理员全量,采购员只看自己。 func ListClientViewsForUser(db *sql.DB, actor *model.User, keyword string, threshold time.Duration) ([]ClientView, error) { visibleUserID, err := visibleClientUserID(actor) if err != nil { return nil, err } return listClientViews(db, keyword, visibleUserID, threshold) } func visibleClientUserID(actor *model.User) (string, error) { if actor == nil { return "", ErrUnauthenticated } visibleUserID := "" if !actor.IsAdmin() { if actor.Role != model.RolePurchaser { return "", ErrAdminRequired } visibleUserID = actor.UserID } return visibleUserID, nil } // taskActorScope 校验可创建/管理任务的网页登录账号,并返回客户端/任务可见范围。 func taskActorScope(actor *model.User) (string, error) { if actor == nil { return "", ErrUnauthenticated } if actor.Status != model.UserActive { return "", fmt.Errorf("当前账号不是正常状态,不能操作任务") } return visibleClientUserID(actor) } func listClientViews(db *sql.DB, keyword, visibleUserID string, threshold time.Duration) ([]ClientView, error) { clients, err := repository.ListClientsForUser(db, keyword, visibleUserID) 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.Client, Status: c.StatusText(now, threshold), PurchaseMode: repository.ParseClientPurchaseMode(c.Capabilities), AssignedUserID: c.AssignedUserID, AssignedUsername: c.AssignedUsername, }) } return views, nil } // ClientListResult 是客户端页面的一页数据和同一权限范围内的完整统计。 type ClientListResult struct { Rows []ClientView Page int Total int TotalPages int Online int Offline int } // ListClientPageForUser 按网页登录身份、搜索词和统一页大小返回客户端列表。 func ListClientPageForUser(db *sql.DB, actor *model.User, keyword string, threshold time.Duration, requestedPage int) (*ClientListResult, error) { visibleUserID, err := visibleClientUserID(actor) if err != nil { return nil, err } now := time.Now().UTC() page := requestedPage if page < 1 { page = 1 } rows, total, err := repository.ListClientsForUserPage(db, keyword, visibleUserID, PageSize, (page-1)*PageSize) if err != nil { return nil, err } totalPages := TotalPages(total) clampedPage := ClampPage(page, totalPages) if clampedPage != page { page = clampedPage rows, _, err = repository.ListClientsForUserPage(db, keyword, visibleUserID, PageSize, (page-1)*PageSize) if err != nil { return nil, err } } online, err := repository.CountOnlineClientsForUser( db, keyword, visibleUserID, now.Add(-threshold).Format(model.TimeLayout)) if err != nil { return nil, err } views := make([]ClientView, 0, len(rows)) for _, client := range rows { views = append(views, ClientView{ Client: client.Client, Status: client.StatusText(now, threshold), PurchaseMode: repository.ParseClientPurchaseMode(client.Capabilities), AssignedUserID: client.AssignedUserID, AssignedUsername: client.AssignedUsername, }) } return &ClientListResult{ Rows: views, Page: page, Total: total, TotalPages: totalPages, Online: online, Offline: total - online, }, nil } // ListAssignableClients 返回当前用户在采购任务页面可选择的客户端。 // 当前创建页面尚未实现,本函数固定未来入口也必须沿用相同权限边界。 func ListAssignableClients(db *sql.DB, actor *model.User, threshold time.Duration) ([]ClientView, error) { return ListClientViewsForUser(db, actor, "", threshold) } // PurchaseClientOptions 是采购弹窗的客户端候选和可选数量。 // 非 live Client 也返回给页面显示禁用原因;创建时仍会在事务内复核能力。 type PurchaseClientOptions struct { Rows []ClientView SelectableCount int } // ListPurchaseClientOptions 返回当前用户可见的采购客户端。 // live Client 即使暂时离线也可提前指派,等它上线后再领取任务。 func ListPurchaseClientOptions(db *sql.DB, actor *model.User, threshold time.Duration) (*PurchaseClientOptions, error) { clients, err := ListClientViewsForUser(db, actor, "", threshold) if err != nil { return nil, err } result := &PurchaseClientOptions{Rows: clients} result.SelectableCount = len(clients) return result, nil } // ListActivePurchasers 返回管理员可选择的绑定目标。 func ListActivePurchasers(db *sql.DB, actor *model.User) ([]model.User, error) { if actor == nil || !actor.IsAdmin() { return nil, ErrAdminRequired } return repository.ListActivePurchasers(db) } // AssignClient 由管理员完成首次绑定或转交。 func AssignClient(db *sql.DB, actor *model.User, clientID, purchaserUserID string, now time.Time) (bool, bool, error) { if actor == nil || !actor.IsAdmin() { return false, false, ErrAdminRequired } clientID = strings.TrimSpace(clientID) purchaserUserID = strings.TrimSpace(purchaserUserID) if clientID == "" || purchaserUserID == "" { return false, false, invalidInput("客户端和采购员都不能为空") } id, err := randomID("CA-", 16) if err != nil { return false, false, err } at := now.UTC().Format(model.TimeLayout) return repository.AssignClient(db, model.ClientUserAssignment{ AssignmentID: id, ClientID: clientID, UserID: purchaserUserID, StartedAt: at, AssignedByUserID: actor.UserID, }) } // UnassignClient 由管理员结束当前归属,不改动任何任务。 func UnassignClient(db *sql.DB, actor *model.User, clientID string, now time.Time) error { if actor == nil || !actor.IsAdmin() { return ErrAdminRequired } clientID = strings.TrimSpace(clientID) if clientID == "" { return invalidInput("客户端不能为空") } return repository.UnassignClient(db, clientID, actor.UserID, now.UTC().Format(model.TimeLayout)) } // 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, purchaseModes ...string) (*model.Task, error) { return repository.ClaimNextTask(db, clientID, supportedTypes, purchaseModes...) }