-
Notifications
You must be signed in to change notification settings - Fork 19
/
guest.go
93 lines (79 loc) · 1.92 KB
/
guest.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
package yavirt
import (
"context"
"encoding/json"
"fmt"
"sync"
"time"
"github.com/projecteru2/agent/utils"
"github.com/projecteru2/core/log"
)
// LabelMeta .
const LabelMeta = "ERU_META"
// HealthCheck .
type HealthCheck struct {
TCPPorts []string
HTTPPort string
HTTPURL string
HTTPCode int
}
type healthCheckMeta struct {
Publish []string
HealthCheck *HealthCheck
}
// Guest yavirt virtual machine
type Guest struct {
ID string
Status string
TransitStatus string
CreateTime int64
TransitTime int64
UpdateTime int64
CPU int
Mem int64
Storage int64
ImageID int64
ImageName string
ImageUser string
Networks map[string]string
Labels map[string]string
IPs []string
Hostname string
Running bool
HealthCheck *HealthCheck
once sync.Once
}
// CheckHealth returns if the guest is healthy
func (g *Guest) CheckHealth(ctx context.Context, timeout time.Duration) bool {
// init health check bridge
g.once.Do(func() {
if meta, ok := g.Labels[LabelMeta]; ok {
hcm := &healthCheckMeta{}
err := json.Unmarshal([]byte(meta), hcm)
if err != nil {
log.WithFunc("CheckHealth").Error(ctx, err, "invalid json format, guest %v, meta %v", g.ID, meta)
return
}
g.HealthCheck = hcm.HealthCheck
}
})
if g.HealthCheck == nil {
return true
}
var tcpChecker []string
var httpChecker []string
healthCheck := g.HealthCheck
for _, port := range healthCheck.TCPPorts {
for _, ip := range g.IPs {
tcpChecker = append(tcpChecker, fmt.Sprintf("%s:%s", ip, port))
}
}
if healthCheck.HTTPPort != "" {
for _, ip := range g.IPs {
httpChecker = append(httpChecker, fmt.Sprintf("http://%s:%s%s", ip, healthCheck.HTTPPort, healthCheck.HTTPURL))
}
}
f1 := utils.CheckHTTP(ctx, g.ID, httpChecker, healthCheck.HTTPCode, timeout)
f2 := utils.CheckTCP(ctx, g.ID, tcpChecker, timeout)
return f1 && f2
}