|
|
package crontask
|
|
|
|
|
|
import (
|
|
|
"autogo/common"
|
|
|
"autogo/dbsql"
|
|
|
"autogo/models"
|
|
|
"autogo/monkey"
|
|
|
"strings"
|
|
|
)
|
|
|
|
|
|
func CheckMonkeyTasks() {
|
|
|
db, err := dbsql.GetConn(dbsql.DSN)
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
defer dbsql.Close(db)
|
|
|
|
|
|
var list []models.MonkeyTask
|
|
|
db.Model(models.MonkeyTask{}).Where("status = ? AND is_del = 0", "WAITTING").Find(&list)
|
|
|
|
|
|
for _, task := range list {
|
|
|
product_name := task.Product
|
|
|
if strings.Contains(task.Product, "-") {
|
|
|
product_name = strings.Split(task.Product, "-")[0]
|
|
|
}
|
|
|
var device models.Device
|
|
|
db.Table("device").Model(models.Device{}).Where("project = ? AND product_name = ? AND status = ?", task.Project, product_name, "online").First(&device)
|
|
|
if device.ID < 1 {
|
|
|
// 没有空闲设备
|
|
|
continue
|
|
|
}
|
|
|
db.Table("device").Model(models.Device{}).Where("udid = ?", device.Udid).Update("status", "busy")
|
|
|
go monkey.RunAndroidMonkeyCmd(task, device.Udid)
|
|
|
common.PushCorntaskLog("执行Monkey任务：" + task.Project + "-" + device.Udid)
|
|
|
}
|
|
|
db.Model(models.MonkeyTask{}).Where("status = ? AND is_del = 0", "RUNNING").Find(&list)
|
|
|
for _, task := range list {
|
|
|
var pids []models.MonkeyPid
|
|
|
db.Model(models.MonkeyPid{}).Where("task_id = ?", task.Id).Find(&pids)
|
|
|
if code := checkTaskPids(pids); code == 0 {
|
|
|
task.Status = "RUNNING"
|
|
|
} else if code == 1 {
|
|
|
task.Status = "FINISH"
|
|
|
db.Model(models.MonkeyTask{}).Where("id = ?", task.Id).Update("status", task.Status)
|
|
|
common.PushMonkeyResult(task)
|
|
|
}
|
|
|
}
|
|
|
}
|