mirror of
https://github.com/eosswedenorg/antelope-api-healthcheck
synced 2026-06-19 05:10:02 +02:00
Fix code formatting
This commit is contained in:
parent
b0e5b455ca
commit
adb1ad3c6d
21 changed files with 884 additions and 950 deletions
|
|
@ -1,22 +1,25 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
"github.com/eosswedenorg/eosio-api-healthcheck/internal"
|
||||
"github.com/eosswedenorg/eosio-api-healthcheck/internal/utils"
|
||||
log "github.com/inconshreveable/log15"
|
||||
"github.com/eosswedenorg-go/pid"
|
||||
"github.com/pborman/getopt/v2"
|
||||
"fmt"
|
||||
"os"
|
||||
"os/signal"
|
||||
"syscall"
|
||||
|
||||
"github.com/eosswedenorg-go/pid"
|
||||
"github.com/eosswedenorg/eosio-api-healthcheck/internal"
|
||||
"github.com/eosswedenorg/eosio-api-healthcheck/internal/utils"
|
||||
log "github.com/inconshreveable/log15"
|
||||
"github.com/pborman/getopt/v2"
|
||||
)
|
||||
|
||||
// Command line flags
|
||||
// ---------------------------------------------------------
|
||||
|
||||
var logFile string
|
||||
var pidFile string
|
||||
var (
|
||||
logFile string
|
||||
pidFile string
|
||||
)
|
||||
|
||||
// Global variables
|
||||
// ---------------------------------------------------------
|
||||
|
|
@ -28,159 +31,160 @@ var VersionString string = "-"
|
|||
// File descriptor to the current log file.
|
||||
var logfd *os.File
|
||||
|
||||
var logfmt log.Format
|
||||
var logger log.Logger
|
||||
var (
|
||||
logfmt log.Format
|
||||
logger log.Logger
|
||||
)
|
||||
|
||||
// argv_listen_addr
|
||||
// Parse listen address from command line.
|
||||
// argv_listen_addr
|
||||
// Parse listen address from command line.
|
||||
//
|
||||
// ---------------------------------------------------------
|
||||
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, 0o644)
|
||||
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, logfmt))
|
||||
// Update variable and set log writer.
|
||||
logfd = fd
|
||||
logger.SetHandler(log.StreamHandler(logfd, logfmt))
|
||||
}
|
||||
|
||||
// signalEventLoop()
|
||||
// Initialize event channel for OS signals
|
||||
// and runs an event loop.
|
||||
// signalEventLoop()
|
||||
// Initialize event channel for OS signals
|
||||
// and runs an event loop.
|
||||
//
|
||||
// ---------------------------------------------------------
|
||||
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, syscall.SIGINT, syscall.SIGTERM)
|
||||
|
||||
// subscribe to SIGHUP signal.
|
||||
signal.Notify(sig_ch, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)
|
||||
// Event loop
|
||||
func() {
|
||||
var run bool = true
|
||||
for run {
|
||||
// Block until we get a signal.
|
||||
sig := <-sig_ch
|
||||
|
||||
// Event loop
|
||||
func() {
|
||||
var run bool = true
|
||||
for run {
|
||||
// Block until we get a signal.
|
||||
sig := <- sig_ch
|
||||
l := logger.New("signal", sig)
|
||||
|
||||
l := logger.New("signal", sig)
|
||||
switch sig {
|
||||
case syscall.SIGINT, syscall.SIGTERM:
|
||||
l.Info("Program was asked to terminate.")
|
||||
run = false
|
||||
// SIGHUP is sent when logfile is rotated.
|
||||
case syscall.SIGHUP:
|
||||
msg := "Logfile was rotated: "
|
||||
|
||||
switch sig {
|
||||
case syscall.SIGINT, syscall.SIGTERM :
|
||||
l.Info("Program was asked to terminate.")
|
||||
run = false
|
||||
// SIGHUP is sent when logfile is rotated.
|
||||
case syscall.SIGHUP :
|
||||
msg := "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)"
|
||||
}
|
||||
|
||||
l.Info(msg)
|
||||
default:
|
||||
l.Warn("Unknown signal")
|
||||
}
|
||||
}
|
||||
}()
|
||||
l.Info(msg)
|
||||
default:
|
||||
l.Warn("Unknown signal")
|
||||
}
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
// main
|
||||
// main
|
||||
//
|
||||
// ---------------------------------------------------------
|
||||
func main() {
|
||||
var version bool
|
||||
var usage bool
|
||||
var addr string
|
||||
var logFormatter *string
|
||||
|
||||
var version bool
|
||||
var usage bool
|
||||
var addr string
|
||||
var logFormatter *string
|
||||
logger = log.Root()
|
||||
|
||||
logger = log.Root()
|
||||
// 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")
|
||||
logFormatter = getopt.EnumLong("log-format", 0, []string{"term", "logfmt", "json", "json-pretty"}, "", "Log format to use: term,logfmt,json,json-pretty")
|
||||
|
||||
// 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")
|
||||
logFormatter = getopt.EnumLong("log-format", 0, []string{"term", "logfmt", "json", "json-pretty"}, "", "Log format to use: term,logfmt,json,json-pretty")
|
||||
getopt.Parse()
|
||||
|
||||
getopt.Parse()
|
||||
if usage {
|
||||
getopt.Usage()
|
||||
return
|
||||
}
|
||||
|
||||
if usage {
|
||||
getopt.Usage()
|
||||
return
|
||||
}
|
||||
if version {
|
||||
fmt.Printf("Version: %s\n", VersionString)
|
||||
return
|
||||
}
|
||||
|
||||
if version {
|
||||
fmt.Printf("Version: %s\n", VersionString)
|
||||
return;
|
||||
}
|
||||
logfmt = utils.ParseLogFormatter(*logFormatter)
|
||||
|
||||
logfmt = utils.ParseLogFormatter(*logFormatter)
|
||||
// Open logfile.
|
||||
if len(logFile) > 0 {
|
||||
setLogFile()
|
||||
} else {
|
||||
logger.SetHandler(log.StreamHandler(os.Stdout, logfmt))
|
||||
}
|
||||
|
||||
// Open logfile.
|
||||
if len(logFile) > 0 {
|
||||
setLogFile()
|
||||
} else {
|
||||
logger.SetHandler(log.StreamHandler(os.Stdout, logfmt))
|
||||
}
|
||||
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)
|
||||
}
|
||||
}
|
||||
addr = argv_listen_addr()
|
||||
|
||||
addr = argv_listen_addr()
|
||||
// Start listening to TCP Connections
|
||||
err := internal.SpawnTcpServer(addr)
|
||||
if err == nil {
|
||||
logger.Info("TCP Server started", "addr", addr)
|
||||
|
||||
// Start listening to TCP Connections
|
||||
err := internal.SpawnTcpServer(addr)
|
||||
if err == nil {
|
||||
logger.Info("TCP Server started", "addr", addr)
|
||||
// Run the signal event loop.
|
||||
signalEventLoop()
|
||||
} else {
|
||||
log.Error("Failed to start tcp server", "error", err)
|
||||
}
|
||||
|
||||
// Run the signal event loop.
|
||||
signalEventLoop()
|
||||
} else {
|
||||
log.Error("Failed to start tcp server", "error", err)
|
||||
}
|
||||
|
||||
logger.Info("Shutdown")
|
||||
logger.Info("Shutdown")
|
||||
}
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue