Booking trace API

This commit is contained in:
mr 2024-08-21 15:46:16 +02:00
parent 9a6ab5c29c
commit 6ef8cde225
2 changed files with 34 additions and 1 deletions

View File

@ -33,7 +33,19 @@ type PeerCache struct {
} }
func (p *PeerCache) checkPeerStatus(peerID string) bool { func (p *PeerCache) checkPeerStatus(peerID string) bool {
return true api := tools.API{}
access := (&Peer{}).GetAccessor(nil)
res, code, _ := access.LoadOne(peerID)
if code != 200 {
return false
}
url := res.(*Peer).Url
if strings.Contains(url, "localhost") || strings.Contains(url, "127.0.0.1") {
url = strings.ReplaceAll(url, "localhost", utils.PEER.API())
}
state := api.CheckRemotePeer(url + ":8093")
return state != tools.DEAD
} }
func (p *PeerCache) GetAccessor(caller *tools.HTTPCaller) utils.Accessor { func (p *PeerCache) GetAccessor(caller *tools.HTTPCaller) utils.Accessor {

View File

@ -20,6 +20,15 @@ const (
DEAD DEAD
) )
func ToState(str string) State {
for _, s := range []State{ALIVE, REDUCED_SERVICE, UNPROCESSABLE_ENTITY, DB_FALLOUT, TEAPOT, DEAD} {
if s.String() == str {
return s
}
}
return DEAD
}
func (s State) String() string { func (s State) String() string {
return [...]string{"alive", "reduced service", "unprocessable entity", "database fallout", return [...]string{"alive", "reduced service", "unprocessable entity", "database fallout",
"some things boils in here, i'm probably a teapot", "dead"}[s] "some things boils in here, i'm probably a teapot", "dead"}[s]
@ -47,6 +56,18 @@ func (a *API) GetState() (State, int, error) {
return ALIVE, 200, nil return ALIVE, 200, nil
} }
func (a *API) CheckRemotePeer(url string) State {
// Check if the database is up
caller := NewHTTPCaller(map[string]map[METHOD]string{})
var resp APIStatusResponse
b, err := caller.CallGet(url, "/status")
if err != nil {
return DEAD
}
json.Unmarshal(b, &resp)
return ToState(resp.Data.State)
}
func (a *API) CheckRemoteAPIs(urls []string) (State, int, error) { func (a *API) CheckRemoteAPIs(urls []string) (State, int, error) {
// Check if the database is up // Check if the database is up
caller := NewHTTPCaller(map[string]map[METHOD]string{}) caller := NewHTTPCaller(map[string]map[METHOD]string{})