mirror of
https://github.com/eosswedenorg/antelope-api-healthcheck
synced 2026-08-27 20:58:13 +02:00
Convert hard tabs to soft.
This commit is contained in:
parent
1dc31db4e1
commit
fa9a943215
7 changed files with 259 additions and 259 deletions
178
src/main.go
178
src/main.go
|
|
@ -2,12 +2,12 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
log "github.com/inconshreveable/log15"
|
||||
"github.com/eosswedenorg-go/pid"
|
||||
"github.com/pborman/getopt/v2"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
log "github.com/inconshreveable/log15"
|
||||
"github.com/eosswedenorg-go/pid"
|
||||
"github.com/pborman/getopt/v2"
|
||||
)
|
||||
|
||||
// Command line flags
|
||||
|
|
@ -29,43 +29,43 @@ var logger log.Logger
|
|||
// ---------------------------------------------------------
|
||||
func argv_listen_addr() string {
|
||||
|
||||
var addr string
|
||||
var addr string
|
||||
|
||||
argv := getopt.Args()
|
||||
if len(argv) > 0 {
|
||||
addr = argv[0]
|
||||
} else {
|
||||
addr = "127.0.0.1"
|
||||
}
|
||||
argv := getopt.Args()
|
||||
if len(argv) > 0 {
|
||||
addr = argv[0]
|
||||
} else {
|
||||
addr = "127.0.0.1"
|
||||
}
|
||||
|
||||
addr += ":"
|
||||
if len(argv) > 1 {
|
||||
addr += argv[1]
|
||||
} else {
|
||||
addr += "1337"
|
||||
}
|
||||
addr += ":"
|
||||
if len(argv) > 1 {
|
||||
addr += argv[1]
|
||||
} else {
|
||||
addr += "1337"
|
||||
}
|
||||
|
||||
return addr
|
||||
return addr
|
||||
}
|
||||
|
||||
func setLogFile() {
|
||||
|
||||
// Open file
|
||||
fd, err := os.OpenFile(logFile, os.O_APPEND | os.O_CREATE | os.O_WRONLY, 0644)
|
||||
if err != nil {
|
||||
logger.Error(err.Error())
|
||||
}
|
||||
// Open file
|
||||
fd, err := os.OpenFile(logFile, os.O_APPEND | os.O_CREATE | os.O_WRONLY, 0644)
|
||||
if err != nil {
|
||||
logger.Error(err.Error())
|
||||
}
|
||||
|
||||
// Try close if old descriptor is defined.
|
||||
if logfd != nil {
|
||||
if err = logfd.Close(); err != nil {
|
||||
logger.Error(err.Error())
|
||||
}
|
||||
}
|
||||
// Try close if old descriptor is defined.
|
||||
if logfd != nil {
|
||||
if err = logfd.Close(); err != nil {
|
||||
logger.Error(err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
// Update variable and set log writer.
|
||||
logfd = fd
|
||||
logger.SetHandler(log.StreamHandler(logfd, log.LogfmtFormat()))
|
||||
// Update variable and set log writer.
|
||||
logfd = fd
|
||||
logger.SetHandler(log.StreamHandler(logfd, log.LogfmtFormat()))
|
||||
}
|
||||
|
||||
// signalEventLoop()
|
||||
|
|
@ -74,88 +74,88 @@ func setLogFile() {
|
|||
// ---------------------------------------------------------
|
||||
func signalEventLoop() {
|
||||
|
||||
// Setup a channel
|
||||
sig_ch := make(chan os.Signal, 1)
|
||||
// Setup a channel
|
||||
sig_ch := make(chan os.Signal, 1)
|
||||
|
||||
// subscribe to SIGHUP signal.
|
||||
signal.Notify(sig_ch, syscall.SIGHUP)
|
||||
// subscribe to SIGHUP signal.
|
||||
signal.Notify(sig_ch, syscall.SIGHUP)
|
||||
|
||||
// Event loop (runs in a seperate thread)
|
||||
go func() {
|
||||
for {
|
||||
// Block until we get a signal.
|
||||
sig := <- sig_ch
|
||||
// Event loop (runs in a seperate thread)
|
||||
go func() {
|
||||
for {
|
||||
// Block until we get a signal.
|
||||
sig := <- sig_ch
|
||||
|
||||
switch sig {
|
||||
// SIGHUP is sent when logfile is rotated.
|
||||
case syscall.SIGHUP :
|
||||
msg := "SIGHUP (Logfile was rotated): "
|
||||
switch sig {
|
||||
// SIGHUP is sent when logfile is rotated.
|
||||
case syscall.SIGHUP :
|
||||
msg := "SIGHUP (Logfile was rotated): "
|
||||
|
||||
if logfd != nil {
|
||||
setLogFile()
|
||||
msg += "Filedescriptor was updated"
|
||||
} else {
|
||||
msg += "No Filedescriptor to update (most likely uses standard out/err streams)"
|
||||
}
|
||||
if logfd != nil {
|
||||
setLogFile()
|
||||
msg += "Filedescriptor was updated"
|
||||
} else {
|
||||
msg += "No Filedescriptor to update (most likely uses standard out/err streams)"
|
||||
}
|
||||
|
||||
logger.Info(msg)
|
||||
default:
|
||||
logger.Warn("Unknown signal", "signal", sig)
|
||||
}
|
||||
}
|
||||
}()
|
||||
logger.Info(msg)
|
||||
default:
|
||||
logger.Warn("Unknown signal", "signal", sig)
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
// main
|
||||
// ---------------------------------------------------------
|
||||
func main() {
|
||||
|
||||
var version bool
|
||||
var version bool
|
||||
var usage bool
|
||||
var addr string;
|
||||
var addr string;
|
||||
|
||||
logger = log.New()
|
||||
logger = log.New()
|
||||
|
||||
// Command line parsing
|
||||
// Command line parsing
|
||||
getopt.SetParameters("[ip] [port]")
|
||||
getopt.FlagLong(&usage, "help", 'h', "Print this help text")
|
||||
getopt.FlagLong(&version, "version", 'v', "Print version")
|
||||
getopt.FlagLong(&logFile, "log", 'l', "Path to log file", "file")
|
||||
getopt.FlagLong(&pidFile, "pid", 'p', "Path to pid file", "file")
|
||||
getopt.Parse()
|
||||
getopt.FlagLong(&version, "version", 'v', "Print version")
|
||||
getopt.FlagLong(&logFile, "log", 'l', "Path to log file", "file")
|
||||
getopt.FlagLong(&pidFile, "pid", 'p', "Path to pid file", "file")
|
||||
getopt.Parse()
|
||||
|
||||
if usage {
|
||||
getopt.Usage()
|
||||
return
|
||||
}
|
||||
|
||||
if version {
|
||||
print("Version: v1.1\n")
|
||||
return;
|
||||
}
|
||||
if version {
|
||||
print("Version: v1.1\n")
|
||||
return;
|
||||
}
|
||||
|
||||
// Open logfile.
|
||||
if len(logFile) > 0 {
|
||||
setLogFile()
|
||||
}
|
||||
// Open logfile.
|
||||
if len(logFile) > 0 {
|
||||
setLogFile()
|
||||
}
|
||||
|
||||
logger.Info("Process is starting", "pid", pid.Get())
|
||||
logger.Info("Process is starting", "pid", pid.Get())
|
||||
|
||||
if len(pidFile) > 0 {
|
||||
logger.Info("Writing pidfile", "file", pidFile)
|
||||
err := pid.Save(pidFile)
|
||||
if err != nil {
|
||||
logger.Error("Failed to write pidfile", "msg", err)
|
||||
}
|
||||
}
|
||||
if len(pidFile) > 0 {
|
||||
logger.Info("Writing pidfile", "file", pidFile)
|
||||
err := pid.Save(pidFile)
|
||||
if err != nil {
|
||||
logger.Error("Failed to write pidfile", "msg", err)
|
||||
}
|
||||
}
|
||||
|
||||
// Run the signal event loop.
|
||||
signalEventLoop()
|
||||
// Run the signal event loop.
|
||||
signalEventLoop()
|
||||
|
||||
addr = argv_listen_addr()
|
||||
addr = argv_listen_addr()
|
||||
|
||||
logger.Info("TCP Server started", "addr", addr)
|
||||
logger.Info("TCP Server started", "addr", addr)
|
||||
|
||||
// Start listening to TCP Connections
|
||||
spawnTcpServer(addr);
|
||||
// Start listening to TCP Connections
|
||||
spawnTcpServer(addr);
|
||||
}
|
||||
|
|
|
|||
206
src/server.go
206
src/server.go
|
|
@ -2,12 +2,12 @@ package main
|
|||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
"strings"
|
||||
"strconv"
|
||||
"internal/utils"
|
||||
"github.com/eosswedenorg-go/eosapi"
|
||||
"github.com/eosswedenorg-go/haproxy"
|
||||
"time"
|
||||
"strings"
|
||||
"strconv"
|
||||
"internal/utils"
|
||||
"github.com/eosswedenorg-go/eosapi"
|
||||
"github.com/eosswedenorg-go/haproxy"
|
||||
"github.com/eosswedenorg-go/tcp_server"
|
||||
)
|
||||
|
||||
|
|
@ -15,30 +15,30 @@ import (
|
|||
// ---------------------------------------------------------
|
||||
func check_api(p eosapi.ReqParams, block_time float64) (haproxy.HealthCheckStatus, string) {
|
||||
|
||||
info, err := eosapi.GetInfo(p)
|
||||
if err != nil {
|
||||
msg := fmt.Sprintf("%s", err);
|
||||
return haproxy.HealthCheckFailed, msg
|
||||
}
|
||||
info, err := eosapi.GetInfo(p)
|
||||
if err != nil {
|
||||
msg := fmt.Sprintf("%s", err);
|
||||
return haproxy.HealthCheckFailed, msg
|
||||
}
|
||||
|
||||
// Check HTTP Status Code
|
||||
if info.HTTPStatusCode > 299 {
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because %v was received from backend", info.HTTPStatusCode)
|
||||
}
|
||||
// Check HTTP Status Code
|
||||
if info.HTTPStatusCode > 299 {
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because %v was received from backend", info.HTTPStatusCode)
|
||||
}
|
||||
|
||||
// Validate head block.
|
||||
now := time.Now().In(time.UTC)
|
||||
diff := now.Sub(info.HeadBlockTime).Seconds()
|
||||
// Validate head block.
|
||||
now := time.Now().In(time.UTC)
|
||||
diff := now.Sub(info.HeadBlockTime).Seconds()
|
||||
|
||||
if diff > block_time {
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because head block is lagging %.0f seconds", diff)
|
||||
} else if diff < -block_time {
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because head block is %.0f seconds into the future", diff)
|
||||
}
|
||||
return haproxy.HealthCheckUp, "OK"
|
||||
if diff > block_time {
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because head block is lagging %.0f seconds", diff)
|
||||
} else if diff < -block_time {
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because head block is %.0f seconds into the future", diff)
|
||||
}
|
||||
return haproxy.HealthCheckUp, "OK"
|
||||
}
|
||||
|
||||
// check_api_v2 (hyperion)
|
||||
|
|
@ -47,111 +47,111 @@ func check_api(p eosapi.ReqParams, block_time float64) (haproxy.HealthCheckStatu
|
|||
// ---------------------------------------------------------
|
||||
func check_api_v2(p eosapi.ReqParams, offset int64) (haproxy.HealthCheckStatus, string) {
|
||||
|
||||
health, err := eosapi.GetHealth(p)
|
||||
if err != nil {
|
||||
msg := fmt.Sprintf("%s", err);
|
||||
return haproxy.HealthCheckFailed, msg
|
||||
}
|
||||
health, err := eosapi.GetHealth(p)
|
||||
if err != nil {
|
||||
msg := fmt.Sprintf("%s", err);
|
||||
return haproxy.HealthCheckFailed, msg
|
||||
}
|
||||
|
||||
// Check HTTP Status Code
|
||||
if health.HTTPStatusCode > 299 {
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because %v was received from backend", health.HTTPStatusCode)
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because %v was received from backend", health.HTTPStatusCode)
|
||||
}
|
||||
|
||||
// Fetch elasticsearch and nodeos block numbers from json.
|
||||
var es_block int64 = 0
|
||||
var node_block int64 = 0
|
||||
// Fetch elasticsearch and nodeos block numbers from json.
|
||||
var es_block int64 = 0
|
||||
var node_block int64 = 0
|
||||
|
||||
for _, v := range health.Health {
|
||||
if v.Name == "Elasticsearch" {
|
||||
es_block = utils.JsonGetInt64(v.Data["last_indexed_block"])
|
||||
} else if v.Name == "NodeosRPC" {
|
||||
node_block = utils.JsonGetInt64(v.Data["head_block_num"])
|
||||
}
|
||||
}
|
||||
for _, v := range health.Health {
|
||||
if v.Name == "Elasticsearch" {
|
||||
es_block = utils.JsonGetInt64(v.Data["last_indexed_block"])
|
||||
} else if v.Name == "NodeosRPC" {
|
||||
node_block = utils.JsonGetInt64(v.Data["head_block_num"])
|
||||
}
|
||||
}
|
||||
|
||||
// Error out if ether or both are zero.
|
||||
if es_block == 0 || node_block == 0 {
|
||||
msg := fmt.Sprintf("Failed to get Elasticsearch and/or nodeos " +
|
||||
"block numbers (es: %d, eos: %d)", es_block, node_block)
|
||||
return haproxy.HealthCheckFailed, msg
|
||||
}
|
||||
// Error out if ether or both are zero.
|
||||
if es_block == 0 || node_block == 0 {
|
||||
msg := fmt.Sprintf("Failed to get Elasticsearch and/or nodeos " +
|
||||
"block numbers (es: %d, eos: %d)", es_block, node_block)
|
||||
return haproxy.HealthCheckFailed, msg
|
||||
}
|
||||
|
||||
// Check if ES is behind or in the future.
|
||||
diff := node_block - es_block;
|
||||
if diff > offset {
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because Elastic is %d blocks behind", diff)
|
||||
} else if diff < -offset {
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because Elastic is %d blocks into the future", -1 * diff)
|
||||
}
|
||||
return haproxy.HealthCheckUp, "OK"
|
||||
// Check if ES is behind or in the future.
|
||||
diff := node_block - es_block;
|
||||
if diff > offset {
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because Elastic is %d blocks behind", diff)
|
||||
} else if diff < -offset {
|
||||
return haproxy.HealthCheckDown,
|
||||
fmt.Sprintf("Taking offline because Elastic is %d blocks into the future", -1 * diff)
|
||||
}
|
||||
return haproxy.HealthCheckUp, "OK"
|
||||
}
|
||||
|
||||
// onTcpMessage callback function
|
||||
// ---------------------------------------------------------
|
||||
|
||||
func onTcpMessage(c *tcp_server.Client, args string) {
|
||||
params := eosapi.ReqParams{}
|
||||
var block_time int = 10
|
||||
var version string = "v1"
|
||||
params := eosapi.ReqParams{}
|
||||
var block_time int = 10
|
||||
var version string = "v1"
|
||||
|
||||
// Parse arguments.
|
||||
// -------------------
|
||||
split := strings.Split(strings.TrimSpace(args), "|")
|
||||
// Parse arguments.
|
||||
// -------------------
|
||||
split := strings.Split(strings.TrimSpace(args), "|")
|
||||
|
||||
// 1. url (scheme + ip/domain + port)
|
||||
params.Url = split[0]
|
||||
// 1. url (scheme + ip/domain + port)
|
||||
params.Url = split[0]
|
||||
|
||||
// 2. Block time.
|
||||
if len(split) > 1 {
|
||||
p, err := strconv.ParseInt(split[1], 10, 32)
|
||||
if err == nil {
|
||||
block_time = int(p)
|
||||
}
|
||||
}
|
||||
// 2. Block time.
|
||||
if len(split) > 1 {
|
||||
p, err := strconv.ParseInt(split[1], 10, 32)
|
||||
if err == nil {
|
||||
block_time = int(p)
|
||||
}
|
||||
}
|
||||
|
||||
// 3. Version
|
||||
if len(split) > 2 {
|
||||
version = split[2]
|
||||
}
|
||||
// 3. Version
|
||||
if len(split) > 2 {
|
||||
version = split[2]
|
||||
}
|
||||
|
||||
// 4. Host
|
||||
if len(split) > 3 {
|
||||
params.Host = split[3]
|
||||
}
|
||||
// 4. Host
|
||||
if len(split) > 3 {
|
||||
params.Host = split[3]
|
||||
}
|
||||
|
||||
// Check api.
|
||||
// -------------------
|
||||
var status haproxy.HealthCheckStatus
|
||||
var msg string
|
||||
// Check api.
|
||||
// -------------------
|
||||
var status haproxy.HealthCheckStatus
|
||||
var msg string
|
||||
|
||||
if version == "v2" {
|
||||
status, msg = check_api_v2(params, int64(block_time / 2))
|
||||
} else {
|
||||
version = "v1"
|
||||
status, msg = check_api(params, float64(block_time))
|
||||
}
|
||||
if version == "v2" {
|
||||
status, msg = check_api_v2(params, int64(block_time / 2))
|
||||
} else {
|
||||
version = "v1"
|
||||
status, msg = check_api(params, float64(block_time))
|
||||
}
|
||||
|
||||
logger.Info("API Check", "version", version, "url", params.Url,
|
||||
"block", block_time / 2, "status", status)
|
||||
logger.Info("API Check", "version", version, "url", params.Url,
|
||||
"block", block_time / 2, "status", status)
|
||||
|
||||
if status != haproxy.HealthCheckUp && len(msg) > 0 {
|
||||
logger.Warn("API Check Failed", "message", msg)
|
||||
}
|
||||
if status != haproxy.HealthCheckUp && len(msg) > 0 {
|
||||
logger.Warn("API Check Failed", "message", msg)
|
||||
}
|
||||
|
||||
// Report status to HAproxy
|
||||
c.WriteString(fmt.Sprintln(status))
|
||||
c.Close()
|
||||
// Report status to HAproxy
|
||||
c.WriteString(fmt.Sprintln(status))
|
||||
c.Close()
|
||||
}
|
||||
|
||||
// spawnTcpServer
|
||||
// ---------------------------------------------------------
|
||||
|
||||
func spawnTcpServer(addr string) {
|
||||
server := tcp_server.New(addr)
|
||||
server.OnMessage(onTcpMessage)
|
||||
server := tcp_server.New(addr)
|
||||
server.OnMessage(onTcpMessage)
|
||||
server.Listen()
|
||||
}
|
||||
|
|
|
|||
|
|
@ -9,9 +9,9 @@ package utils
|
|||
// ---------------------------------------------------------
|
||||
|
||||
func JsonGetInt64(input interface{}) (int64) {
|
||||
v, res := input.(float64)
|
||||
if res {
|
||||
return (int64) (v)
|
||||
}
|
||||
return 0
|
||||
v, res := input.(float64)
|
||||
if res {
|
||||
return (int64) (v)
|
||||
}
|
||||
return 0
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue