diff --git a/cmd/antelope-api-healtcheck/main.go b/cmd/antelope-api-healtcheck/main.go index 94d2ab9..f6fa811 100644 --- a/cmd/antelope-api-healtcheck/main.go +++ b/cmd/antelope-api-healtcheck/main.go @@ -5,6 +5,7 @@ import ( "os" "os/signal" "syscall" + "time" "github.com/eosswedenorg-go/pid" "github.com/eosswedenorg/antelope-api-healthcheck/internal/server" @@ -182,7 +183,7 @@ func main() { } // Create server - srv = server.New(argv_listen_addr()) + srv = server.New(argv_listen_addr(), time.Second*10) // Run signal event loop in its own goroutine go signalEventLoop() diff --git a/internal/server/server.go b/internal/server/server.go index 25af4ac..5402320 100644 --- a/internal/server/server.go +++ b/internal/server/server.go @@ -23,11 +23,15 @@ type Server struct { // Number of connections between each OnTick() num_conn uint64 + + // Time between each call to OnTick() + tick_interval time.Duration } -func New(addr string) *Server { +func New(addr string, tick_interval time.Duration) *Server { return &Server{ - addr: fmt.Sprintf("tcp://%s", addr), + addr: fmt.Sprintf("tcp://%s", addr), + tick_interval: tick_interval, } } @@ -61,7 +65,7 @@ func (s *Server) OnTick() (time.Duration, gnet.Action) { "current_connections": s.eng.CountConnections(), }) atomic.StoreUint64(&s.num_conn, 0) - return time.Second * 10, gnet.None + return s.tick_interval, gnet.None } // OnTraffic callback function