mirror of https://github.com/statping/statping
163 lines
3.3 KiB
Go
163 lines
3.3 KiB
Go
package services
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/statping/statping/database"
|
|
"github.com/statping/statping/types/errors"
|
|
"github.com/statping/statping/types/metrics"
|
|
"github.com/statping/statping/utils"
|
|
"sort"
|
|
)
|
|
|
|
var (
|
|
db database.Database
|
|
log = utils.Log.WithField("type", "service")
|
|
allServices map[int64]*Service
|
|
)
|
|
|
|
func (s *Service) Validate() error {
|
|
if s.Name == "" {
|
|
return errors.New("missing service name")
|
|
} else if s.Domain == "" {
|
|
return errors.New("missing domain name")
|
|
} else if s.Type == "" {
|
|
return errors.New("missing service type")
|
|
} else if s.Interval == 0 {
|
|
return errors.New("missing check interval")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *Service) BeforeCreate() error {
|
|
return s.Validate()
|
|
}
|
|
|
|
func (s *Service) BeforeUpdate() error {
|
|
return s.Validate()
|
|
}
|
|
|
|
func (s *Service) AfterFind() {
|
|
db.Model(s).Related(&s.Incidents).Related(&s.Messages).Related(&s.Checkins).Related(&s.Incidents)
|
|
metrics.Query("service", "find")
|
|
}
|
|
|
|
func (s *Service) AfterCreate() error {
|
|
s.prevOnline = true
|
|
allServices[s.Id] = s
|
|
metrics.Query("service", "create")
|
|
return nil
|
|
}
|
|
|
|
func (s *Service) AfterUpdate() {
|
|
metrics.Query("service", "update")
|
|
}
|
|
|
|
func (s *Service) AfterDelete() {
|
|
metrics.Query("service", "delete")
|
|
}
|
|
|
|
func init() {
|
|
allServices = make(map[int64]*Service)
|
|
}
|
|
|
|
func Services() map[int64]*Service {
|
|
return allServices
|
|
}
|
|
|
|
func SetDB(database database.Database) {
|
|
db = database.Model(&Service{})
|
|
}
|
|
|
|
func Find(id int64) (*Service, error) {
|
|
srv := allServices[id]
|
|
if srv == nil {
|
|
return nil, errors.Missing(&Service{}, id)
|
|
}
|
|
db.First(&srv, id)
|
|
return srv, nil
|
|
}
|
|
|
|
func all() []*Service {
|
|
var services []*Service
|
|
db.Find(&services)
|
|
return services
|
|
}
|
|
|
|
func All() map[int64]*Service {
|
|
return allServices
|
|
}
|
|
|
|
func AllInOrder() []Service {
|
|
var services []Service
|
|
for _, service := range allServices {
|
|
service.UpdateStats()
|
|
services = append(services, *service)
|
|
}
|
|
sort.Sort(ServiceOrder(services))
|
|
return services
|
|
}
|
|
|
|
func (s *Service) Create() error {
|
|
err := db.Create(s)
|
|
if err.Error() != nil {
|
|
log.Errorln(fmt.Sprintf("Failed to create service %v #%v: %v", s.Name, s.Id, err))
|
|
return err.Error()
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (s *Service) Update() error {
|
|
q := db.Update(s)
|
|
s.Close()
|
|
allServices[s.Id] = s
|
|
s.SleepDuration = s.Duration()
|
|
go ServiceCheckQueue(allServices[s.Id], true)
|
|
return q.Error()
|
|
}
|
|
|
|
func (s *Service) Delete() error {
|
|
s.Close()
|
|
if err := s.AllFailures().DeleteAll(); err != nil {
|
|
return err
|
|
}
|
|
if err := s.AllHits().DeleteAll(); err != nil {
|
|
return err
|
|
}
|
|
if err := s.DeleteCheckins(); err != nil {
|
|
return err
|
|
}
|
|
db.Model(s).Association("Checkins").Clear()
|
|
if err := s.DeleteIncidents(); err != nil {
|
|
return err
|
|
}
|
|
db.Model(s).Association("Incidents").Clear()
|
|
if err := s.DeleteMessages(); err != nil {
|
|
return err
|
|
}
|
|
db.Model(s).Association("Messages").Clear()
|
|
|
|
delete(allServices, s.Id)
|
|
q := db.Model(&Service{}).Delete(s)
|
|
return q.Error()
|
|
}
|
|
|
|
func (s *Service) DeleteMessages() error {
|
|
for _, m := range s.Messages {
|
|
if err := m.Delete(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
db.Model(s).Association("messages").Clear()
|
|
return nil
|
|
}
|
|
|
|
func (s *Service) DeleteCheckins() error {
|
|
for _, c := range s.Checkins {
|
|
if err := c.Delete(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
db.Model(s).Association("checkins").Clear()
|
|
return nil
|
|
}
|