forked from hidevopsio/hiboot
-
Notifications
You must be signed in to change notification settings - Fork 0
/
health.go
74 lines (62 loc) · 1.96 KB
/
health.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
// Copyright 2018 John Deng (hi.devops.io@gmail.com).
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
// Package controller provide the controller for health check
package actuator
import (
"hidevops.io/hiboot/pkg/app"
"hidevops.io/hiboot/pkg/app/web/context"
"hidevops.io/hiboot/pkg/at"
"hidevops.io/hiboot/pkg/factory"
)
// HealthService is the interface for health check
type HealthService interface {
Name() string
Status() bool
}
// Health is the health check struct
type Health struct {
Status string `json:"status"`
}
type healthController struct {
at.RestController
configurableFactory factory.ConfigurableFactory
}
func init() {
app.Register(newHealthController)
}
func newHealthController(configurableFactory factory.ConfigurableFactory) *healthController {
return &healthController{configurableFactory: configurableFactory}
}
// GET /health
func (c *healthController) Get(ctx context.Context) {
healthServices := c.configurableFactory.GetInstances(new(at.HealthCheckService))
healthCheckProfiles := make(map[string]interface{})
healthCheckProfiles["status"] = "Up"
if healthServices != nil {
for _, md := range healthServices {
metaData := factory.CastMetaData(md)
if metaData.Instance != nil {
healthService := metaData.Instance.(HealthService)
status := "Down"
if healthService.Status() {
status = "Up"
}
healthCheckProfiles[healthService.Name()] = Health{
Status: status,
}
}
}
}
ctx.JSON(healthCheckProfiles)
}