63 lines
1.3 KiB
Go
63 lines
1.3 KiB
Go
package apiclient
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"net/http"
|
|
"time"
|
|
|
|
log "github.com/sirupsen/logrus"
|
|
tomb "gopkg.in/tomb.v2"
|
|
|
|
"github.com/crowdsecurity/go-cs-lib/trace"
|
|
)
|
|
|
|
type HeartBeatService service
|
|
|
|
func (h *HeartBeatService) Ping(ctx context.Context) (bool, *Response, error) {
|
|
|
|
u := fmt.Sprintf("%s/heartbeat", h.client.URLPrefix)
|
|
|
|
req, err := h.client.NewRequest(http.MethodGet, u, nil)
|
|
if err != nil {
|
|
return false, nil, err
|
|
}
|
|
|
|
resp, err := h.client.Do(ctx, req, nil)
|
|
if err != nil {
|
|
return false, resp, err
|
|
}
|
|
|
|
return true, resp, nil
|
|
}
|
|
|
|
func (h *HeartBeatService) StartHeartBeat(ctx context.Context, t *tomb.Tomb) {
|
|
t.Go(func() error {
|
|
defer trace.CatchPanic("crowdsec/apiClient/heartbeat")
|
|
hbTimer := time.NewTicker(1 * time.Minute)
|
|
for {
|
|
select {
|
|
case <-hbTimer.C:
|
|
log.Debug("heartbeat: sending heartbeat")
|
|
ok, resp, err := h.Ping(ctx)
|
|
if err != nil {
|
|
log.Errorf("heartbeat error : %s", err)
|
|
continue
|
|
}
|
|
resp.Response.Body.Close()
|
|
if resp.Response.StatusCode != http.StatusOK {
|
|
log.Errorf("heartbeat unexpected return code : %d", resp.Response.StatusCode)
|
|
continue
|
|
}
|
|
if !ok {
|
|
log.Errorf("heartbeat returned false")
|
|
continue
|
|
}
|
|
case <-t.Dying():
|
|
log.Debugf("heartbeat: stopping")
|
|
hbTimer.Stop()
|
|
return nil
|
|
}
|
|
}
|
|
})
|
|
}
|