-
Notifications
You must be signed in to change notification settings - Fork 0
/
run.go
124 lines (108 loc) · 3.22 KB
/
run.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
package main
import (
"encoding/json"
"fmt"
"math/rand"
"os"
"strconv"
"strings"
"time"
log "github.com/sirupsen/logrus"
"github.com/Zioyi/zocker/cgroups"
"github.com/Zioyi/zocker/cgroups/subsystems"
"github.com/Zioyi/zocker/container"
)
func Run(tty bool, cmdArray []string, res *subsystems.ResourceConfig, volume string, containerName string, imageName string, envVariables []string) {
containerID := randStringBytes(10)
if containerName == "" {
containerName = containerID
}
parent, writePipe := container.NewParentProcess(tty, volume, containerName, imageName, envVariables)
if parent == nil {
log.Errorf("New parent process error")
}
if err := parent.Start(); err != nil {
log.Error(err)
}
containerName, err := recordContainerInfo(parent.Process.Pid, cmdArray, containerName, containerID, volume)
if err != nil {
log.Errorf("Record container info error %v", err)
return
}
// 启动时限制
cgroupManager := cgroups.NewCgroupManger("zocker-cgroup")
defer cgroupManager.Destroy()
cgroupManager.Set(res)
cgroupManager.Apply(parent.Process.Pid)
sendInitCommand(cmdArray, writePipe)
log.Infof("[zocker] container name is %s\n", containerName)
if tty {
_ = parent.Wait()
container.DeleteWorkSpace(volume, containerName)
deleteContainerInfo(containerName)
}
os.Exit(0)
}
func sendInitCommand(cmdArray []string, writePipe *os.File) {
command := strings.Join(cmdArray, " ")
log.Infof("command all is %s", command)
writePipe.WriteString(command)
writePipe.Close()
}
func recordContainerInfo(containerPID int, commandArray []string, containerName string, id string, volume string) (string, error) {
var cstZone = time.FixedZone("CST", 8*3600)
createTime := time.Now().In(cstZone).Format("2006-01-02 15:04:05")
command := strings.Join(commandArray, " ")
containerInfo := &container.ContainerInfo{
Id: id,
Pid: strconv.Itoa(containerPID),
Command: command,
Status: container.Running,
Name: containerName,
CreateTime: createTime,
Volume: volume,
}
jsonBytes, err := json.Marshal(containerInfo)
if err != nil {
log.Errorf("Record contaier info error %v", err)
return "", err
}
jsonStr := string(jsonBytes)
dirUrl := fmt.Sprintf(container.DefaultInfoLocation, containerName)
if err := os.MkdirAll(dirUrl, 0622); err != nil {
log.Errorf("Mkdir %s error %v", dirUrl, err)
return "", err
}
fileName := dirUrl + "/" + container.ConfigName
file, err := os.Create(fileName)
defer func(file *os.File) {
err := file.Close()
if err != nil {
log.Errorf("Close %s error %v", fileName, err)
}
}(file)
if err != nil {
log.Errorf("Create file %s error %v", fileName, err)
return "", err
}
if _, err := file.WriteString(jsonStr); err != nil {
log.Errorf("File write string error %v", err)
return "", err
}
return containerName, nil
}
func deleteContainerInfo(containName string) {
dirURL := fmt.Sprintf(container.DefaultInfoLocation, containName)
if err := os.RemoveAll(dirURL); err != nil {
log.Errorf("Remove dir %s error %v", dirURL, err)
}
}
func randStringBytes(n int) string {
letterBytes := "1234567890"
rand.Seed(time.Now().UnixNano())
b := make([]byte, n)
for i := range b {
b[i] = letterBytes[rand.Intn(len(letterBytes))]
}
return string(b)
}