add first basic program structure
This commit is contained in:
parent
00eb993b17
commit
51206fb263
6 changed files with 106 additions and 0 deletions
7
Makefile
Normal file
7
Makefile
Normal file
|
@ -0,0 +1,7 @@
|
|||
.PHONY: build test
|
||||
|
||||
build:
|
||||
go build -o build/
|
||||
|
||||
test:
|
||||
go test -v ./...
|
BIN
build/heartbeat-server
Executable file
BIN
build/heartbeat-server
Executable file
Binary file not shown.
5
go.mod
Normal file
5
go.mod
Normal file
|
@ -0,0 +1,5 @@
|
|||
module git.faercol.me/monitoring/heartbeat-server
|
||||
|
||||
go 1.16
|
||||
|
||||
require github.com/sirupsen/logrus v1.9.0
|
15
go.sum
Normal file
15
go.sum
Normal file
|
@ -0,0 +1,15 @@
|
|||
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/davecgh/go-spew v1.1.1 h1:vj9j/u1bqnvCEfJOwUhtlOARqs3+rkHYY13jYWTU97c=
|
||||
github.com/davecgh/go-spew v1.1.1/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
|
||||
github.com/pmezard/go-difflib v1.0.0 h1:4DBwDE0NGyQoBHbLQYPwSUPoCMWR5BEzIk/f1lZbAQM=
|
||||
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
|
||||
github.com/sirupsen/logrus v1.9.0 h1:trlNQbNUG3OdDrDil03MCb1H2o9nJ1x4/5LYw7byDE0=
|
||||
github.com/sirupsen/logrus v1.9.0/go.mod h1:naHLuLoDiP4jHNo9R0sCBMtWGeIprob74mVsIT4qYEQ=
|
||||
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
|
||||
github.com/stretchr/testify v1.7.0 h1:nwc3DEeHmmLAfoZucVR881uASk0Mfjw8xYJ99tb5CcY=
|
||||
github.com/stretchr/testify v1.7.0/go.mod h1:6Fq8oRcR53rry900zMqJjRRixrwX3KX962/h/Wwjteg=
|
||||
golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8 h1:0A+M6Uqn+Eje4kHMK80dtF3JCXC4ykBgQG4Fe06QRhQ=
|
||||
golang.org/x/sys v0.0.0-20220715151400-c0bba94af5f8/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||
gopkg.in/check.v1 v0.0.0-20161208181325-20d25e280405/go.mod h1:Co6ibVJAznAaIkqp8huTwlJQCZ016jof/cbN4VW5Yz0=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c h1:dUUwHk2QECo/6vqA44rthZ8ie2QXMNeKRTHCNY2nXvo=
|
||||
gopkg.in/yaml.v3 v3.0.0-20200313102051-9f266ea9e77c/go.mod h1:K4uyk7z7BCEPqu6E+C64Yfv1cQ7kz7rIZviUmN+EgEM=
|
20
main.go
Normal file
20
main.go
Normal file
|
@ -0,0 +1,20 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"context"
|
||||
|
||||
"git.faercol.me/monitoring/heartbeat-server/server"
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
func main() {
|
||||
addr := "localhost:9000"
|
||||
logger := logrus.New()
|
||||
|
||||
srv, err := server.New(addr, logger)
|
||||
if err != nil {
|
||||
logger.Fatalf("Failed to create HTTP server: %s", err.Error())
|
||||
}
|
||||
|
||||
srv.Run(context.Background())
|
||||
}
|
59
server/server.go
Normal file
59
server/server.go
Normal file
|
@ -0,0 +1,59 @@
|
|||
package server
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"net"
|
||||
"net/http"
|
||||
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
type Server struct {
|
||||
ctx context.Context
|
||||
cancel context.CancelFunc
|
||||
httpSrv *http.Server
|
||||
listener net.Listener
|
||||
handler *http.ServeMux
|
||||
logger *logrus.Logger
|
||||
}
|
||||
|
||||
func (s *Server) initMux() {
|
||||
s.handler.HandleFunc("/", s.heartbeatHandler)
|
||||
}
|
||||
|
||||
func (s *Server) heartbeatHandler(w http.ResponseWriter, r *http.Request) {
|
||||
remoteHost, _, err := net.SplitHostPort(r.RemoteAddr)
|
||||
if err != nil {
|
||||
s.logger.Warnf("Unexpected remote address %s: %q", r.RemoteAddr, err.Error())
|
||||
} else {
|
||||
s.logger.Infof("Received a heartbeat query from %s", remoteHost)
|
||||
}
|
||||
w.WriteHeader(http.StatusNoContent)
|
||||
}
|
||||
|
||||
func (s *Server) Run(ctx context.Context) {
|
||||
s.ctx, s.cancel = context.WithCancel(ctx)
|
||||
s.initMux()
|
||||
if err := s.httpSrv.Serve(s.listener); err != nil {
|
||||
s.logger.Errorf("Failed to serve HTTP server: %s", err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
func New(addr string, logger *logrus.Logger) (*Server, error) {
|
||||
m := http.NewServeMux()
|
||||
l, err := net.Listen("tcp", addr)
|
||||
if err != nil {
|
||||
return nil, fmt.Errorf("failed to setup listener on addr %q: %w", addr, err)
|
||||
}
|
||||
|
||||
return &Server{
|
||||
ctx: context.TODO(),
|
||||
httpSrv: &http.Server{
|
||||
Handler: m,
|
||||
},
|
||||
handler: m,
|
||||
listener: l,
|
||||
logger: logger,
|
||||
}, nil
|
||||
}
|
Loading…
Reference in a new issue