forked from goodrain/rainbond
-
Notifications
You must be signed in to change notification settings - Fork 0
/
service_check.go
103 lines (91 loc) · 3.19 KB
/
service_check.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
// Copyright (C) 2014-2018 Goodrain Co., Ltd.
// RAINBOND, Application Management Platform
// This program is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version. For any non-GPL usage of Rainbond,
// one or multiple Commercial Licenses authorized by Goodrain Co., Ltd.
// must be obtained first.
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
// You should have received a copy of the GNU General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
package handler
import (
"context"
"fmt"
"strings"
"time"
"github.com/pquerna/ffjson/ffjson"
"github.com/Sirupsen/logrus"
api_model "github.com/goodrain/rainbond/api/model"
"github.com/goodrain/rainbond/api/util"
"github.com/goodrain/rainbond/builder/exector"
client "github.com/goodrain/rainbond/mq/client"
tutil "github.com/goodrain/rainbond/util"
"github.com/twinj/uuid"
)
//ServiceCheck check service build source
func (s *ServiceAction) ServiceCheck(scs *api_model.ServiceCheckStruct) (string, string, *util.APIHandleError) {
checkUUID := uuid.NewV4().String()
scs.Body.CheckUUID = checkUUID
if scs.Body.EventID == "" {
scs.Body.EventID = tutil.NewUUID()
}
topic := client.BuilderTopic
if tutil.StringArrayContains(s.conf.EnableFeature, "windows") {
if scs.Body.CheckOS == "windows" {
topic = client.WindowsBuilderTopic
}
if scs.Body.SourceType == "docker-run" || scs.Body.SourceType == "docker-compose" {
if maybeIsWindowsContainerImage(scs.Body.SourceBody) {
topic = client.WindowsBuilderTopic
}
}
}
err := s.MQClient.SendBuilderTopic(client.TaskStruct{
TaskType: "service_check",
TaskBody: scs.Body,
Topic: topic,
})
if err != nil {
logrus.Errorf("enqueue service check message to mq error, %v", err)
return "", "", util.CreateAPIHandleError(500, err)
}
return checkUUID, scs.Body.EventID, nil
}
var windowsKeywords = []string{"windows", "asp", "microsoft", "nanoserver"}
func maybeIsWindowsContainerImage(source string) bool {
for _, k := range windowsKeywords {
if strings.Contains(source, k) {
return true
}
}
return false
}
//GetServiceCheckInfo get application source detection information
func (s *ServiceAction) GetServiceCheckInfo(uuid string) (*exector.ServiceCheckResult, *util.APIHandleError) {
k := fmt.Sprintf("/servicecheck/%s", uuid)
var si exector.ServiceCheckResult
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
resp, err := s.EtcdCli.Get(ctx, k)
cancel()
if err != nil {
logrus.Errorf("get etcd k %s error, %v", k, err)
return nil, util.CreateAPIHandleError(500, err)
}
if resp.Count == 0 {
return &si, nil
}
v := resp.Kvs[0].Value
if err := ffjson.Unmarshal(v, &si); err != nil {
return nil, util.CreateAPIHandleError(500, err)
}
if si.CheckStatus == "" {
si.CheckStatus = "Checking"
logrus.Debugf("checking is %v", si)
}
return &si, nil
}