Ma/GoFilm
1
0
Fork 0
mirror of https://github.com/ProudMuBai/GoFilm.git synced 2026-02-14 05:54:40 +08:00
Code Issues Packages Projects Releases Wiki Activity

format Server API

Browse Source
This commit is contained in:
mubai
2024-02-26 22:59:04 +08:00
parent 93b9855f52
commit 6709720707
17 changed files with 167 additions and 430 deletions
Download Patch File Download Diff File Expand all files Collapse all files

82
server/controller/SpiderController.go
View File

@@ -3,7 +3,6 @@ package controller
import (
"fmt"
"github.com/gin-gonic/gin"
"net/http"
"server/logic"
"server/model/system"
"strconv"
@@ -14,32 +13,20 @@ func CollectFilm(c *gin.Context) {
id := c.DefaultQuery("id", "")
hourStr := c.DefaultQuery("h", "0")
if id == "" || hourStr == "0" {
c.JSON(http.StatusOK, gin.H{
"status": StatusFailed,
"message": "采集任务开启失败, 缺乏必要参数",
})
system.Failed("采集任务开启失败, 缺乏必要参数", c)
return
}
h, err := strconv.Atoi(hourStr)
if err != nil {
c.JSON(http.StatusOK, gin.H{
"status": StatusFailed,
"message": "采集任务开启失败, hour(时长)参数不符合规范",
})
system.Failed("采集任务开启失败, 采集(时长)不符合规范", c)
return
}
// 执行采集逻处理逻辑
if err = logic.SL.StartCollect(id, h); err != nil {
c.JSON(http.StatusOK, gin.H{
"status": StatusFailed,
"message": fmt.Sprint("采集任务开启失败: ", err.Error()),
})
system.Failed(fmt.Sprint("采集任务开启失败: ", err.Error()), c)
return
}
c.JSON(http.StatusOK, gin.H{
"status": StatusOk,
"message": "采集任务已成功开启!!!",
})
system.SuccessOnlyMsg("采集任务已成功开启!!!", c)
}
// StarSpider 开启并执行采集任务
@@ -47,81 +34,48 @@ func StarSpider(c *gin.Context) {
var cp system.CollectParams
// 获取请求参数
if err := c.ShouldBindJSON(&cp); err != nil {
c.JSON(http.StatusOK, gin.H{
"status": StatusFailed,
"message": "请求参数异常",
})
system.Failed("请求参数异常!!!", c)
return
}
if cp.Time == 0 {
c.JSON(http.StatusOK, gin.H{
"status": StatusFailed,
"message": "采集开启失败,采集时长不能为0",
})
system.Failed("采集开启失败,采集时长不能为0", c)
return
}
// 根据 Batch 执行对应的逻辑
if cp.Batch {
// 执行批量采集
if cp.Ids == nil || len(cp.Ids) <= 0 {
c.JSON(http.StatusOK, gin.H{
"status": StatusFailed,
"message": "批量采集开启失败, 关联的资源站信息为空",
})
system.Failed("批量采集开启失败, 关联的资源站信息为空", c)
return
}
// 执行批量采集
logic.SL.BatchCollect(cp.Time, cp.Ids)
} else {
if len(cp.Id) <= 0 {
c.JSON(http.StatusOK, gin.H{
"status": StatusFailed,
"message": "批量采集开启失败, 资源站Id获取失败",
})
system.Failed("批量采集开启失败, 资源站Id获取失败", c)
return
}
if err := logic.SL.StartCollect(cp.Id, cp.Time); err != nil {
c.JSON(http.StatusOK, gin.H{
"status": StatusFailed,
"message": fmt.Sprint("采集任务开启失败: ", err.Error()),
})
system.Failed(fmt.Sprint("采集任务开启失败: ", err.Error()), c)
return
}
}
// 返回成功执行的信息
c.JSON(http.StatusOK, gin.H{
"status": StatusOk,
"message": "采集任务已成功开启!!!",
})
system.SuccessOnlyMsg("采集任务已成功开启!!!", c)
}
// SpiderReset 重置影视数据, 清空库存, 从零开始
func SpiderReset(c *gin.Context) {
var cp system.CollectParams
// 获取请求参数
if err := c.ShouldBindJSON(&cp); err != nil {
c.JSON(http.StatusOK, gin.H{
"status": StatusFailed,
"message": "请求参数异常",
})
// 清空采集数据进行重新采集前校验输入的密码是否正确
key := c.DefaultQuery("accessKey", "")
// 如密码错误则不执行后续操作
if len(key) <= 0 || key != "Re0" {
system.Failed("重置失败, 密钥校验失败!!!", c)
return
}
if cp.Time == 0 {
c.JSON(http.StatusOK, gin.H{
"status": StatusFailed,
"message": "采集开启失败,采集时长不能为0",
})
return
}
// 后期加入一些前置验证
if len(cp.Id) <= 0 {
c.JSON(http.StatusOK, gin.H{
"status": StatusFailed,
"message": "SpiderReset Failed, 资源站Id获取失败",
})
return
}
logic.SL.ZeroCollect(cp.Time)
// 前置校验通过则清空采集数据并对已启用站点进行 全量采集
logic.SL.ZeroCollect(-1)
system.SuccessOnlyMsg("影视数据已重置, 请耐心等待采集完成!!!", c)
}
// CoverFilmClass 重置覆盖影片分类信息