mirror of https://github.com/statping/statping
90 lines
1.7 KiB
Go
90 lines
1.7 KiB
Go
package services
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"github.com/hunterlong/statping/database"
|
|
"github.com/hunterlong/statping/utils"
|
|
"sort"
|
|
)
|
|
|
|
var log = utils.Log
|
|
|
|
func DB() database.Database {
|
|
return database.DB().Model(&Service{})
|
|
}
|
|
|
|
func Find(id int64) (*Service, error) {
|
|
srv := allServices[id]
|
|
if srv == nil {
|
|
return nil, errors.New("service not found")
|
|
}
|
|
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 {
|
|
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()
|
|
}
|
|
allServices[s.Id] = s
|
|
|
|
go ServiceCheckQueue(allServices[s.Id], true)
|
|
return nil
|
|
}
|
|
|
|
func (s *Service) Update() error {
|
|
db := DB().Update(s)
|
|
|
|
allServices[s.Id] = s
|
|
|
|
if !s.AllowNotifications.Bool {
|
|
//for _, n := range CoreApp.Notifications {
|
|
// notif := n.(notifier.Notifier).Select()
|
|
// notif.ResetUniqueQueue(fmt.Sprintf("service_%v", s.Id))
|
|
//}
|
|
}
|
|
s.Close()
|
|
s.SleepDuration = s.Duration()
|
|
go ServiceCheckQueue(allServices[s.Id], true)
|
|
|
|
//notifier.OnUpdatedService(s.Service)
|
|
|
|
return db.Error()
|
|
}
|
|
|
|
func (s *Service) Delete() error {
|
|
db := database.DB().Delete(s)
|
|
|
|
s.Close()
|
|
delete(allServices, s.Id)
|
|
//notifier.OnDeletedService(s.Service)
|
|
|
|
return db.Error()
|
|
}
|
|
|
|
func (s *Service) DeleteFailures() error {
|
|
query := database.DB().Exec(`DELETE FROM failures WHERE service = ?`, s.Id)
|
|
return query.Error()
|
|
}
|