Added all required dependencies
This commit is contained in:
parent
78f86abba4
commit
1ebb35b988
660 changed files with 502447 additions and 0 deletions
83
vendor/github.com/go-macaron/toolbox/healthcheck.go
generated
vendored
Normal file
83
vendor/github.com/go-macaron/toolbox/healthcheck.go
generated
vendored
Normal file
|
@ -0,0 +1,83 @@
|
|||
// Copyright 2013 Beego Authors
|
||||
// Copyright 2014 The Macaron Authors
|
||||
//
|
||||
// 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 toolbox
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
)
|
||||
|
||||
// HealthChecker represents a health check instance.
|
||||
type HealthChecker interface {
|
||||
Desc() string
|
||||
Check() error
|
||||
}
|
||||
|
||||
// HealthCheckFunc represents a callable function for health check.
|
||||
type HealthCheckFunc func() error
|
||||
|
||||
// HealthCheckFunc represents a callable function for health check with description.
|
||||
type HealthCheckFuncDesc struct {
|
||||
Desc string
|
||||
Func HealthCheckFunc
|
||||
}
|
||||
|
||||
type healthCheck struct {
|
||||
desc string
|
||||
HealthChecker
|
||||
check HealthCheckFunc // Not nil if add job as a function.
|
||||
}
|
||||
|
||||
// AddHealthCheck adds new health check job.
|
||||
func (t *toolbox) AddHealthCheck(hc HealthChecker) {
|
||||
t.healthCheckJobs = append(t.healthCheckJobs, &healthCheck{
|
||||
HealthChecker: hc,
|
||||
})
|
||||
}
|
||||
|
||||
// AddHealthCheckFunc adds a function as a new health check job.
|
||||
func (t *toolbox) AddHealthCheckFunc(desc string, fn HealthCheckFunc) {
|
||||
t.healthCheckJobs = append(t.healthCheckJobs, &healthCheck{
|
||||
desc: desc,
|
||||
check: fn,
|
||||
})
|
||||
}
|
||||
|
||||
func (t *toolbox) handleHealthCheck() string {
|
||||
if len(t.healthCheckJobs) == 0 {
|
||||
return "no health check jobs"
|
||||
}
|
||||
|
||||
var buf bytes.Buffer
|
||||
var err error
|
||||
for _, job := range t.healthCheckJobs {
|
||||
buf.WriteString("* ")
|
||||
if job.check != nil {
|
||||
buf.WriteString(job.desc)
|
||||
err = job.check()
|
||||
} else {
|
||||
buf.WriteString(job.Desc())
|
||||
err = job.Check()
|
||||
}
|
||||
buf.WriteString(": ")
|
||||
if err == nil {
|
||||
buf.WriteString("OK")
|
||||
} else {
|
||||
buf.WriteString(err.Error())
|
||||
}
|
||||
buf.WriteString("\n")
|
||||
}
|
||||
return buf.String()
|
||||
}
|
Loading…
Add table
Add a link
Reference in a new issue