extern/gatus
1
0
Fork 0
mirror of https://github.com/TwiN/gatus.git synced 2025-03-03 17:41:37 +01:00
Code Issues Packages Projects Releases Wiki Activity

add func to store for getting single service status

Browse Source 
and use that in the watchdog
This commit is contained in:
Chris Heppell 2020-12-31 20:33:35 +00:00
parent 5eb289c4d3
commit 4d24a4d647
2 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
storage/memory.go
View File

@ -41,6 +41,18 @@ func (ims *InMemoryStore) GetAll() map[string]*core.ServiceStatus {
return results
}
// GetServiceStatus returns the service status for a given service name in the given group
func (ims *InMemoryStore) GetServiceStatus(group, name string) *core.ServiceStatus {
key := fmt.Sprintf("%s_%s", group, name)
serviceResultsMutex.RLock()
serviceStatus, exists := serviceStatuses[key]
serviceResultsMutex.RUnlock()
if !exists {
return nil
}
return serviceStatus
}
// Insert inserts the observed result for the specified service into the in memory store
func (ims *InMemoryStore) Insert(service *core.Service, result *core.Result) {
key := fmt.Sprintf("%s_%s", service.Group, service.Name)

6
watchdog/watchdog.go
View File

@ -30,10 +30,8 @@ func GetServiceStatusesAsJSON() ([]byte, error) {
// GetUptimeByServiceGroupAndName returns the uptime of a service based on its group and name
func GetUptimeByServiceGroupAndName(group, name string) *core.Uptime {
key := fmt.Sprintf("%s_%s", group, name)
serviceStatuses := store.GetAll()
serviceStatus, exists := serviceStatuses[key]
if !exists {
serviceStatus := store.GetServiceStatus(group, name)
if serviceStatus == nil {
return nil
}
return serviceStatus.Uptime