eosswedenorg/antelope-api-healthcheck
1
0
Fork 0
mirror of https://github.com/eosswedenorg/antelope-api-healthcheck synced 2026-06-16 04:44:55 +02:00
Code Releases Activity

internal/server/server.go: count number of connections and send a log message in OnTick()

Browse source
This commit is contained in:
Henrik Hautakoski 2023-01-04 13:53:29 +01:00
parent 1db04c0e76
commit 1f22435dfa
No known key found for this signature in database
GPG key ID: 217490840C18A5D9
1 changed files with 14 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

17
internal/server/server.go
View file

@ -4,6 +4,7 @@ import (
"context"
"fmt"
"strings"
"sync/atomic"
"time"
"github.com/eosswedenorg-go/haproxy/agentcheck"
@ -15,8 +16,9 @@ import (
type Server struct {
gnet.BuiltinEventEngine
addr string
eng gnet.Engine
addr string
eng gnet.Engine
num_conn uint64
}
func New(addr string) *Server {
@ -34,6 +36,11 @@ func (s *Server) OnBoot(eng gnet.Engine) gnet.Action {
return gnet.None
}
func (s *Server) OnOpen(c gnet.Conn) ([]byte, gnet.Action) {
atomic.AddUint64(&s.num_conn, 1)
return nil, gnet.None
}
// OnShutdown callback function
//
// ---------------------------------------------------------
@ -45,7 +52,11 @@ func (s *Server) OnShutdown(eng gnet.Engine) {
//
// ---------------------------------------------------------
func (s *Server) OnTick() (time.Duration, gnet.Action) {
log.Info("Server info", "connections", s.eng.CountConnections())
log.Info("Server info", log.Ctx{
"connections": atomic.LoadUint64(&s.num_conn),
"current_connections": s.eng.CountConnections(),
})
atomic.StoreUint64(&s.num_conn, 0)
return time.Second * 10, gnet.None
}