63 lines
1.1 KiB
Go
63 lines
1.1 KiB
Go
//go:build !windows
|
|
|
|
package cmd
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/evcc-io/evcc/server"
|
|
"github.com/spf13/cobra"
|
|
"github.com/tv42/httpunix"
|
|
)
|
|
|
|
const serviceName = "evcc"
|
|
|
|
// healthCmd represents the meter command
|
|
var healthCmd = &cobra.Command{
|
|
Use: "health",
|
|
Short: "Check application health",
|
|
Run: runHealth,
|
|
}
|
|
|
|
func init() {
|
|
rootCmd.AddCommand(healthCmd)
|
|
}
|
|
|
|
func runHealth(cmd *cobra.Command, args []string) {
|
|
// load config
|
|
if err := loadConfigFile(&conf); err != nil {
|
|
log.FATAL.Fatal(err)
|
|
}
|
|
|
|
u := &httpunix.Transport{
|
|
DialTimeout: 100 * time.Millisecond,
|
|
RequestTimeout: 1 * time.Second,
|
|
ResponseHeaderTimeout: 1 * time.Second,
|
|
}
|
|
|
|
u.RegisterLocation(serviceName, server.SocketPath(conf.Network.Port))
|
|
|
|
client := http.Client{
|
|
Transport: u,
|
|
}
|
|
|
|
var ok bool
|
|
resp, err := client.Get(fmt.Sprintf("http+unix://%s/health", serviceName))
|
|
|
|
if err == nil {
|
|
resp.Body.Close()
|
|
|
|
if resp.StatusCode == http.StatusOK {
|
|
log.INFO.Printf("health check ok")
|
|
ok = true
|
|
}
|
|
}
|
|
|
|
if !ok {
|
|
log.ERROR.Printf("health check failed")
|
|
os.Exit(1)
|
|
}
|
|
}
|