injectlib/urldb
1
0
Fork 0
mirror of https://github.com/ctwj/urldb.git synced 2025-11-25 11:29:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

update: ui

Browse Source
This commit is contained in:
ctwj
2025-09-14 10:26:58 +08:00
parent 9690a63646
commit d23a6b26e4
24 changed files with 1887 additions and 1713 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
handlers/task_handler.go
View File

@@ -392,8 +392,9 @@ func (h *TaskHandler) DeleteTask(c *gin.Context) {
// CreateExpansionTask 创建扩容任务
func (h *TaskHandler) CreateExpansionTask(c *gin.Context) {
var req struct {
PanAccountID uint `json:"pan_account_id" binding:"required"`
Description string `json:"description"`
PanAccountID uint `json:"pan_account_id" binding:"required"`
Description string `json:"description"`
DataSource map[string]interface{} `json:"dataSource"`
}
if err := c.ShouldBindJSON(&req); err != nil {
@@ -420,10 +421,14 @@ func (h *TaskHandler) CreateExpansionTask(c *gin.Context) {
accountName = fmt.Sprintf("账号%d", cks.ID)
}
// 构建任务配置（存储账号ID）
// 构建任务配置（存储账号ID和数据源）
taskConfig := map[string]interface{}{
"pan_account_id": req.PanAccountID,
}
// 如果有数据源配置，添加到taskConfig中
if req.DataSource != nil && len(req.DataSource) > 0 {
taskConfig["data_source"] = req.DataSource
}
configJSON, _ := json.Marshal(taskConfig)
// 创建任务标题，包含账号名称
@@ -451,6 +456,10 @@ func (h *TaskHandler) CreateExpansionTask(c *gin.Context) {
expansionInput := task.ExpansionInput{
PanAccountID: req.PanAccountID,
}
// 如果有数据源配置，添加到输入数据中
if req.DataSource != nil && len(req.DataSource) > 0 {
expansionInput.DataSource = req.DataSource
}
inputJSON, _ := json.Marshal(expansionInput)