package taskrpc
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"log"
|
|
)
|
|
|
|
// 完成单次任务
|
|
func DoTask(dbname, siteid, userId, task_type string, url ...string) (bool, error) {
|
|
|
|
if dbname == "" || userId == "" || task_type == "" || siteid == "" {
|
|
return false, errors.New("参数不能为空")
|
|
}
|
|
|
|
conn, err := rpc_server_conn(url...)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
defer conn.Close()
|
|
|
|
data := DoTaskParam{}
|
|
data.Dbname = dbname
|
|
data.SiteId = siteid
|
|
data.UserId = userId
|
|
data.Type = task_type
|
|
|
|
req, err := SetReqData(data)
|
|
if err != nil {
|
|
log.Println("DoTask crypter error:", err)
|
|
return false, err
|
|
}
|
|
|
|
res := &Response{}
|
|
|
|
err = conn.DoTask(req, res)
|
|
|
|
if err != nil {
|
|
log.Println("taskrpc DoTask error:", err)
|
|
return false, err
|
|
}
|
|
|
|
res_data_de, err := GetResData(res)
|
|
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if res_data_de == "" {
|
|
return false, nil
|
|
}
|
|
|
|
var task_res bool
|
|
err = json.Unmarshal([]byte(res_data_de), &task_res)
|
|
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return task_res, nil
|
|
}
|
|
|
|
// 完成循环类任务
|
|
func DoCycleTask(dbname, siteid, userId, task_id, task_category string, url ...string) (bool, error) {
|
|
|
|
if dbname == "" || userId == "" || task_id == "" {
|
|
return false, errors.New("参数错误")
|
|
}
|
|
|
|
conn, err := rpc_server_conn(url...)
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
defer conn.Close()
|
|
|
|
data := DoCycleTaskParam{}
|
|
data.Dbname = dbname
|
|
data.SiteId = siteid
|
|
data.UserId = userId
|
|
data.TaskId = task_id
|
|
data.TaskCategroy = task_category
|
|
|
|
req, err := SetReqData(data)
|
|
if err != nil {
|
|
log.Println("DoCycleTask crypter error:", err)
|
|
return false, err
|
|
}
|
|
|
|
res := &Response{}
|
|
|
|
err = conn.DoCycleTask(req, res)
|
|
|
|
if err != nil {
|
|
log.Println("taskrpc DoCycleTask error:", err)
|
|
return false, err
|
|
}
|
|
|
|
res_data_de, err := GetResData(res)
|
|
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
if res_data_de == "" {
|
|
return false, nil
|
|
}
|
|
|
|
var task_res bool
|
|
err = json.Unmarshal([]byte(res_data_de), &task_res)
|
|
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
|
|
return task_res, nil
|
|
}
|