mirror of https://github.com/statping/statping
141 lines
3.0 KiB
Go
141 lines
3.0 KiB
Go
package downtimes
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/statping/statping/database"
|
|
"strconv"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
zeroInt64 int64
|
|
)
|
|
var db database.Database
|
|
var dbHits database.Database
|
|
|
|
func SetDB(database database.Database) {
|
|
db = database.Model(&Downtime{})
|
|
}
|
|
|
|
func Find(id int64) (*Downtime, error) {
|
|
var downtime Downtime
|
|
q := db.Where("id = ?", id).Find(&downtime)
|
|
if q.Error() != nil {
|
|
return nil, q.Error()
|
|
}
|
|
if q.RecordNotFound() {
|
|
return nil, fmt.Errorf(" Downtime record not found : %s", id)
|
|
}
|
|
return &downtime, q.Error()
|
|
}
|
|
|
|
func (c *Downtime) Validate() error {
|
|
if c.Type == "manual" {
|
|
if c.End != nil && c.End.After(time.Now()) || c.Start.After(time.Now()) {
|
|
return fmt.Errorf("Downtime cannot be in future")
|
|
}
|
|
if c.ServiceId == zeroInt64 {
|
|
return fmt.Errorf("Service ID cannot be null")
|
|
}
|
|
if c.SubStatus != "down" && c.SubStatus != "degraded" {
|
|
return fmt.Errorf("SubStatus can only be 'down' or 'degraded'")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (c *Downtime) BeforeCreate() error {
|
|
return c.Validate()
|
|
}
|
|
|
|
func (c *Downtime) BeforeUpdate() error {
|
|
return c.Validate()
|
|
}
|
|
|
|
func FindByService(service int64, start time.Time, end time.Time) (*[]Downtime, error) {
|
|
var downtime []Downtime
|
|
q := db.Where("service = ? and start BETWEEN ? AND ? ", service, start, end)
|
|
q = q.Order("id ASC ").Find(&downtime)
|
|
return &downtime, q.Error()
|
|
}
|
|
|
|
func ConvertToUnixTime(str string) (time.Time,error){
|
|
i, err := strconv.ParseInt(str, 10, 64)
|
|
var t time.Time
|
|
if err != nil {
|
|
return t,err
|
|
}
|
|
tm := time.Unix(i, 0)
|
|
return tm,nil
|
|
}
|
|
|
|
func FindAll(vars map[string]string ) (*[]Downtime, error) {
|
|
var downtime []Downtime
|
|
var start time.Time
|
|
var end time.Time
|
|
var err error
|
|
var count int64
|
|
var skip int64
|
|
st,err1 := vars["start"]
|
|
en,err2 := vars["end"]
|
|
if err1 && err2 && (en > st){
|
|
start,err = ConvertToUnixTime(vars["start"])
|
|
if err!=nil {
|
|
return &downtime,err
|
|
}
|
|
end,err = ConvertToUnixTime(vars["end"])
|
|
if err!=nil {
|
|
return &downtime,err
|
|
}
|
|
}else{
|
|
ninetyDaysAgo := time.Now().Add(time.Duration(-90*24) * time.Hour)
|
|
start = ninetyDaysAgo
|
|
end = time.Now()
|
|
}
|
|
q := db.Where("start BETWEEN ? AND ? ", start, end)
|
|
subStatus,err3 := vars["sub_status"]
|
|
if err3{
|
|
q = q.Where(" sub_status = ?", subStatus)
|
|
}
|
|
serviceId,err4 := vars["service_id"]
|
|
if err4{
|
|
q = q.Where(" service = ?", serviceId)
|
|
}
|
|
ty,err5 := vars["type"]
|
|
if err5{
|
|
q = q.Where(" type = ?", ty)
|
|
}
|
|
cnt,err5 := vars["count"]
|
|
if err5{
|
|
count,err = strconv.ParseInt(cnt,10,64)
|
|
if count > 100 {
|
|
count = 100
|
|
}
|
|
}else {
|
|
count = 20
|
|
}
|
|
skp,err6:=vars["skip"]
|
|
if err6{
|
|
skip,err = strconv.ParseInt(skp,10,64)
|
|
}else {
|
|
skip = 0
|
|
}
|
|
q = q.Order("id ASC ")
|
|
q = q.Limit((int)(count)).Offset((int)(skip)).Find(&downtime)
|
|
return &downtime, q.Error()
|
|
}
|
|
func (c *Downtime) Create() error {
|
|
q := db.Create(c)
|
|
return q.Error()
|
|
}
|
|
|
|
func (c *Downtime) Update() error {
|
|
q := db.Where(" id = ? ", c.Id).Updates(c)
|
|
return q.Error()
|
|
}
|
|
|
|
func (c *Downtime) Delete() error {
|
|
q := db.Model(&Downtime{}).Delete(c)
|
|
return q.Error()
|
|
}
|