package controllers import ( "fmt" "strconv" "strings" "server/models" "github.com/beego/beego/v2/client/orm" beego "github.com/beego/beego/v2/server/web" ) // ApiCursorDetectController Cursor Token 顺序读取接口(不改变号池状态) // // 用途: // - 前端传入 start_id/current_id/id,从该 ID 开始按 id 从小到大读取 Cursor Token。 // - 只读取未提取记录,不更新 is_extracted、extracted_time、extracted_platform、is_used 等任何状态。 // - 如果传入 ID 对应记录已提取,会自动跳过,继续找下一条未提取记录。 // - 返回 next_id,前端下一次点击时传 next_id,即可实现 11 -> 12 -> 13 递增读取。 // // 示例: // // GET /api/cursor/token/peek?id=11 // GET /api/cursor/token/peek?start_id=11 // GET /api/cursor/token/peek?current_id=11 // // 可选参数: // - data_type=tk/account/account_tk,默认 tk type ApiCursorDetectController struct { beego.Controller } func (c *ApiCursorDetectController) cursorDetectJSONErr(httpStatus, code int, msg string) { c.Ctx.Output.SetStatus(httpStatus) c.Data["json"] = map[string]interface{}{ "code": code, "msg": msg, } _ = c.ServeJSON() } // PeekToken 按 ID 顺序读取 Cursor Token,不改变任何状态。 func (c *ApiCursorDetectController) PeekToken() { startID, err := c.readStartID() if err != nil { c.cursorDetectJSONErr(400, 400, err.Error()) return } dataType := strings.TrimSpace(c.GetString("data_type")) if dataType == "" { dataType = "tk" } if !isValidPoolType(dataType) { c.cursorDetectJSONErr(400, 400, "data_type 不合法,支持: account/tk/account_tk") return } var row models.PlatformAccountPoolCursor qs := models.Orm.QueryTable(new(models.PlatformAccountPoolCursor)). Filter("id__gte", startID). Filter("data_type", dataType). Filter("delete_time__isnull", true). Filter("extracted_time__isnull", true). Filter("is_extracted", 0). Exclude("token", "") err = qs.OrderBy("id").One(&row) if err != nil { if err == orm.ErrNoRows { c.cursorDetectJSONErr(404, 404, "从当前 ID 开始暂无未提取 Cursor Token") return } c.cursorDetectJSONErr(500, 500, "查询失败: "+err.Error()) return } c.Data["json"] = map[string]interface{}{ "code": 200, "msg": "success", "data": map[string]interface{}{ "id": row.ID, "next_id": row.ID + 1, "data_type": row.DataType, "account": row.Account, "password": row.Password, "token": row.Token, "remark": row.Remark, "is_used": row.IsUsed, "create_time": row.CreateTime, // 明确告诉前端:本接口只是读取,没有更新号池状态。 "state_changed": false, }, } _ = c.ServeJSON() } func (c *ApiCursorDetectController) readStartID() (uint64, error) { raw := strings.TrimSpace(c.GetString("id")) if raw == "" { raw = strings.TrimSpace(c.GetString("start_id")) } if raw == "" { raw = strings.TrimSpace(c.GetString("current_id")) } if raw == "" { return 0, fmt.Errorf("缺少参数 id/start_id/current_id") } id, err := strconv.ParseUint(raw, 10, 64) if err != nil || id == 0 { return 0, fmt.Errorf("id 必须是大于 0 的整数") } return id, nil }