|
|
package monkey
|
|
|
|
|
|
import (
|
|
|
"fmt"
|
|
|
"os/exec"
|
|
|
"strings"
|
|
|
|
|
|
"github.com/spf13/cast"
|
|
|
)
|
|
|
|
|
|
type DockerContainer struct {
|
|
|
ID string `json:"id"`
|
|
|
Name string `json:"name"`
|
|
|
Device string `json:"device"`
|
|
|
}
|
|
|
|
|
|
// 根据任务id获取关联容器，由于后续可能支持一个任务绑定多个设备执行，所以返回数组
|
|
|
func GetTaskFromDocker(task_id int) []DockerContainer {
|
|
|
var list []DockerContainer
|
|
|
cmd := exec.Command("docker", "ps")
|
|
|
output, err := cmd.Output()
|
|
|
if err != nil {
|
|
|
fmt.Println(err)
|
|
|
return list
|
|
|
}
|
|
|
|
|
|
lines := strings.Split(string(output), "\n")
|
|
|
for i, line := range lines {
|
|
|
if i == 0 {
|
|
|
continue // skip header line
|
|
|
}
|
|
|
fields := strings.Fields(line)
|
|
|
if len(fields) >= 7 {
|
|
|
// fmt.Printf("Container ID: %s, Name: %s\n", fields[0], fields[len(fields)-1])
|
|
|
var c DockerContainer
|
|
|
c.ID = fields[0]
|
|
|
c.Name = fields[len(fields)-1]
|
|
|
_strs := strings.Split(c.Name, "_")
|
|
|
c.Device = _strs[len(_strs)-1]
|
|
|
if strings.Contains(c.Name, "monkey_task_"+cast.ToString(task_id)+"_") {
|
|
|
list = append(list, c)
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
return list
|
|
|
}
|
|
|
|
|
|
func StopDockerContainer(container_name string) bool {
|
|
|
cmd := exec.Command("docker", "stop", "-t", "0", container_name)
|
|
|
out, err := cmd.CombinedOutput()
|
|
|
if err != nil {
|
|
|
fmt.Println("[docker] Error stopping container:", err)
|
|
|
return true
|
|
|
}
|
|
|
fmt.Println("[docker] Container stopped successfully:", string(out))
|
|
|
return true
|
|
|
}
|