internal/server/server.go: Make OnTick() interval configurable.

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()

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),
+		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