func (b *Broker) CreateRpcTaskRequest(c *echo.Context) error {
args := struct {
Method string `json:"method"`
URL string `json:"url"`
Args string `json:"args"` //json Marshal后的字符串
StartTime int64 `json:"start_time,string"`
TimeInterval string `json:"time_interval"` //空格分隔各个参数
MaxRunTime int64 `json:"max_run_time,string"`
}{}
err := c.Bind(&args)
if err != nil {
return c.JSON(http.StatusForbidden, err.Error())
}
taskRequest := new(task.TaskRequest)
taskRequest.Uuid = uuid.New()
if len(args.URL) == 0 {
return c.JSON(http.StatusForbidden, errors.ErrInvalidArgument.Error())
}
taskRequest.BinName = args.URL
taskRequest.Args = args.Args
taskRequest.StartTime = args.StartTime
taskRequest.TimeInterval = args.TimeInterval
taskRequest.Index = 0
taskRequest.MaxRunTime = args.MaxRunTime
switch args.Method {
case "GET":
taskRequest.TaskType = task.RpcTaskGET
case "POST":
taskRequest.TaskType = task.RpcTaskPOST
case "PUT":
taskRequest.TaskType = task.RpcTaskPUT
case "DELETE":
taskRequest.TaskType = task.RpcTaskDELETE
default:
return c.JSON(http.StatusForbidden, errors.ErrInvalidArgument.Error())
}
err = b.HandleRequest(taskRequest)
if err != nil {
return c.JSON(http.StatusForbidden, err.Error())
}
golog.Info("Broker", "CreateRpcTaskRequest", "ok", 0,
"uuid", taskRequest.Uuid,
"bin_name", taskRequest.BinName,
"args", taskRequest.Args,
"start_time", taskRequest.StartTime,
"time_interval", taskRequest.TimeInterval,
"index", taskRequest.Index,
"max_run_time", taskRequest.MaxRunTime,
"task_type", taskRequest.TaskType,
)
return c.JSON(http.StatusOK, taskRequest.Uuid)
}
func (b *Broker) CreateScriptTaskRequest(c *echo.Context) error {
args := struct {
BinName string `json:"bin_name"`
Args string `json:"args"` //空格分隔各个参数
StartTime int64 `json:"start_time,string"`
TimeInterval string `json:"time_interval"` //空格分隔各个参数
MaxRunTime int64 `json:"max_run_time,string"`
}{}
err := c.Bind(&args)
if err != nil {
return c.JSON(http.StatusForbidden, err.Error())
}
taskRequest := new(task.TaskRequest)
taskRequest.Uuid = uuid.New()
if len(args.BinName) == 0 {
return c.JSON(http.StatusForbidden, errors.ErrInvalidArgument.Error())
}
taskRequest.BinName = args.BinName
taskRequest.Args = args.Args
taskRequest.StartTime = args.StartTime
taskRequest.TimeInterval = args.TimeInterval
taskRequest.Index = 0
taskRequest.MaxRunTime = args.MaxRunTime
taskRequest.TaskType = task.ScriptTask
err = b.HandleRequest(taskRequest)
if err != nil {
return c.JSON(http.StatusForbidden, err.Error())
}
golog.Info("Broker", "CreateScriptTaskRequest", "ok", 0,
"uuid", taskRequest.Uuid,
"bin_name", taskRequest.BinName,
"args", taskRequest.Args,
"start_time", taskRequest.StartTime,
"time_interval", taskRequest.TimeInterval,
"index", taskRequest.Index,
"max_run_time", taskRequest.MaxRunTime,
"task_type", taskRequest.TaskType,
)
return c.JSON(http.StatusOK, taskRequest.Uuid)
}
func (b *Broker) HandleRequest(request *task.TaskRequest) error {
var err error
now := time.Now().Unix()
if request.StartTime == 0 {
request.StartTime = now
}
if request.StartTime <= now {
err = b.AddRequestToRedis(request)
if err != nil {
return err
}
} else {
afterTime := time.Second * time.Duration(request.StartTime-now)
b.timer.NewTimer(afterTime, b.AddRequestToRedis, request)
}
return nil
}
