112 lines
2.6 KiB
Go
112 lines
2.6 KiB
Go
package imitater
|
|
|
|
import (
|
|
"time"
|
|
|
|
"474420502.top/eson/structure/circular_linked"
|
|
|
|
"474420502.top/eson/crontabex"
|
|
"474420502.top/eson/curl2info"
|
|
"474420502.top/eson/requests"
|
|
)
|
|
|
|
// Task 任务
|
|
type Task struct {
|
|
ITask
|
|
|
|
name string
|
|
crontab *crontab.Crontab
|
|
curl *curl2info.CURL
|
|
workflow *requests.Workflow
|
|
session *requests.Session
|
|
proxies clinked.CircularLinked
|
|
}
|
|
|
|
// SetName 任务的名字
|
|
func (task *Task) SetName(name string) {
|
|
task.name = name
|
|
}
|
|
|
|
// GetName 获取任务的名字
|
|
func (task *Task) GetName() string {
|
|
return task.name
|
|
}
|
|
|
|
// SetCurl 设置任务的curl信息类
|
|
func (task *Task) SetCurl(curl *curl2info.CURL) {
|
|
task.curl = curl
|
|
}
|
|
|
|
// GetCurl 获取任务的curl信息类
|
|
func (task *Task) GetCurl() *curl2info.CURL {
|
|
return task.curl
|
|
}
|
|
|
|
// AddProxies 添加代理集合
|
|
func (task *Task) AddProxies(proxy string) {
|
|
task.proxies.Append(proxy)
|
|
}
|
|
|
|
// GetProxies 获取代理的字符串
|
|
func (task *Task) GetProxies() *clinked.CircularLinked {
|
|
return &task.proxies
|
|
}
|
|
|
|
// SetCrontab 设置crontab的控制规则字符串
|
|
func (task *Task) SetCrontab(cron string) {
|
|
task.crontab = crontab.NewCrontab(cron)
|
|
}
|
|
|
|
// SetLastStatus 设置上次执行的状态 成功true 失败false
|
|
func (task *Task) SetLastStatus(status bool) {
|
|
task.crontab.SetStatus(status)
|
|
}
|
|
|
|
// TimeUp 判断是否到了下次执行的时间点
|
|
func (task *Task) TimeUp() bool {
|
|
return task.crontab.TimeUp()
|
|
}
|
|
|
|
// NextTime 下次执行的时间点
|
|
func (task *Task) NextTime() time.Time {
|
|
return task.crontab.NextTime()
|
|
}
|
|
|
|
// Workflow 根据persistent 是否返回现有session(or 新建 session),创建Workflow的信息, 便于设置或者更改参数, Session持久化
|
|
func (task *Task) Workflow(persistent bool) *requests.Workflow {
|
|
|
|
if persistent {
|
|
task.workflow = task.curl.CreateWorkflow(task.Session())
|
|
return task.workflow
|
|
}
|
|
|
|
proxies := task.GetProxies()
|
|
ses := task.curl.CreateSession()
|
|
if proxies.Size() > 0 {
|
|
proxy := proxies.Cursor().GetValue().(string)
|
|
ses.SetConfig(requests.CProxy, proxy)
|
|
proxies.MoveNext()
|
|
}
|
|
|
|
return task.curl.CreateWorkflow(ses)
|
|
}
|
|
|
|
// Request 根据curl信息执行,没持久化
|
|
func (task *Task) Request() (*requests.Response, error) {
|
|
return task.Workflow(false).Execute()
|
|
}
|
|
|
|
// Session 获取Session的信息 只保留session的数据和url参数.
|
|
func (task *Task) Session() *requests.Session {
|
|
if task.session == nil {
|
|
task.session = task.curl.CreateSession()
|
|
}
|
|
if task.proxies.Size() > 0 {
|
|
proxy := task.proxies.Cursor().GetValue().(string)
|
|
task.session.SetConfig(requests.CProxy, proxy)
|
|
task.proxies.MoveNext()
|
|
}
|
|
|
|
return task.session
|
|
}
|