155 lines
4.0 KiB
Go
155 lines
4.0 KiB
Go
package controllers
|
||
|
||
import (
|
||
"fmt"
|
||
"strconv"
|
||
"strings"
|
||
"time"
|
||
|
||
"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 对应的 Cursor Token。
|
||
// - 只读取记录，不更新 is_extracted、extracted_time、extracted_platform、is_used 等任何状态。
|
||
// - 不再跳过已提取的记录，输入什么 ID 就提取什么 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()
|
||
}
|
||
|
||
func (c *ApiCursorDetectController) setTokenUsableByID(isUsed int8) {
|
||
id, err := c.readStartID()
|
||
if err != nil {
|
||
c.cursorDetectJSONErr(400, 400, err.Error())
|
||
return
|
||
}
|
||
|
||
now := time.Now()
|
||
updated, err := models.Orm.QueryTable(new(models.PlatformAccountPoolCursor)).
|
||
Filter("id", id).
|
||
Filter("delete_time__isnull", true).
|
||
Update(orm.Params{
|
||
"is_used": isUsed,
|
||
"update_time": now,
|
||
})
|
||
if err != nil {
|
||
c.cursorDetectJSONErr(500, 500, "更新失败: "+err.Error())
|
||
return
|
||
}
|
||
if updated == 0 {
|
||
c.cursorDetectJSONErr(404, 404, "当前 ID 数据不存在或已删除")
|
||
return
|
||
}
|
||
|
||
c.Data["json"] = map[string]interface{}{
|
||
"code": 200,
|
||
"msg": "success",
|
||
"data": map[string]interface{}{
|
||
"id": id,
|
||
"is_used": isUsed,
|
||
"is_available": isUsed == 1,
|
||
"state_changed": true,
|
||
"update_time": now,
|
||
},
|
||
}
|
||
_ = c.ServeJSON()
|
||
}
|
||
|
||
// MarkTokenAvailable 将当前 ID 的 Cursor Token 标记为可用。
|
||
func (c *ApiCursorDetectController) MarkTokenAvailable() {
|
||
c.setTokenUsableByID(1)
|
||
}
|
||
|
||
// MarkTokenUnavailable 将当前 ID 的 Cursor Token 标记为不可用/用完。
|
||
func (c *ApiCursorDetectController) MarkTokenUnavailable() {
|
||
c.setTokenUsableByID(0)
|
||
}
|
||
|
||
// PeekToken 按 ID 顺序读取 Cursor Token，不改变任何状态。
|
||
func (c *ApiCursorDetectController) PeekToken() {
|
||
startID, err := c.readStartID()
|
||
if err != nil {
|
||
c.cursorDetectJSONErr(400, 400, err.Error())
|
||
return
|
||
}
|
||
|
||
var row models.PlatformAccountPoolCursor
|
||
qs := models.Orm.QueryTable(new(models.PlatformAccountPoolCursor)).
|
||
Filter("id", startID).
|
||
Filter("delete_time__isnull", true).
|
||
Exclude("token", "")
|
||
|
||
err = qs.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
|
||
}
|