feat: add health check to telemetry server
Signed-off-by: Mark Sagi-Kazar <mark.sagikazar@gmail.com>
This commit is contained in:
29
storage/health.go
Normal file
29
storage/health.go
Normal file
@@ -0,0 +1,29 @@
|
||||
package storage
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"time"
|
||||
)
|
||||
|
||||
// NewCustomHealthCheckFunc returns a new health check function.
|
||||
func NewCustomHealthCheckFunc(s Storage, now func() time.Time) func() (details interface{}, err error) {
|
||||
return func() (details interface{}, err error) {
|
||||
a := AuthRequest{
|
||||
ID: NewID(),
|
||||
ClientID: NewID(),
|
||||
|
||||
// Set a short expiry so if the delete fails this will be cleaned up quickly by garbage collection.
|
||||
Expiry: now().Add(time.Minute),
|
||||
}
|
||||
|
||||
if err := s.CreateAuthRequest(a); err != nil {
|
||||
return nil, fmt.Errorf("create auth request: %v", err)
|
||||
}
|
||||
|
||||
if err := s.DeleteAuthRequest(a.ID); err != nil {
|
||||
return nil, fmt.Errorf("delete auth request: %v", err)
|
||||
}
|
||||
|
||||
return nil, nil
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user