chaged stuff

pull/1097/head
Rhythm 2022-01-04 16:21:41 +05:30
parent aa36f50011
commit b4995bcdab
3 changed files with 31 additions and 35 deletions

View File

@ -22,8 +22,6 @@ type serviceOrder struct {
Order int `json:"order"` Order int `json:"order"`
} }
var ( var (
zeroTime time.Time zeroTime time.Time
zeroBool bool zeroBool bool
@ -54,7 +52,7 @@ func ConvertToUnixTime(str string) (time.Time,error){
return tm, nil return tm, nil
} }
func findDowntimeByTime(t string,s services.Service) *downtimes.Downtime { func findAllDowntimes(t string) []downtimes.Downtime {
var timeVar time.Time var timeVar time.Time
if t == "" { if t == "" {
timeVar = time.Now() timeVar = time.Now()
@ -65,11 +63,10 @@ func findDowntimeByTime(t string,s services.Service) *downtimes.Downtime {
return nil return nil
} }
} }
downTime := downtimes.FindDowntime(s.Id,timeVar) downTime := downtimes.FindDowntime(timeVar)
return downTime return downTime
} }
func findPublicSubService(r *http.Request, service *services.Service) (*services.Service, error) { func findPublicSubService(r *http.Request, service *services.Service) (*services.Service, error) {
vars := mux.Vars(r) vars := mux.Vars(r)
id := utils.ToInt(vars["sub_id"]) id := utils.ToInt(vars["sub_id"])
@ -562,20 +559,23 @@ func apiAllServicesStatusHandler(r *http.Request) interface{} {
t = query.Get("time") t = query.Get("time")
} }
var srvs []services.ServiceWithDowntime var srvs []services.ServiceWithDowntime
dtime := findAllDowntimes(t)
m := make(map[int64]downtimes.Downtime)
for i := 0; i < len(dtime); i += 1 {
m[dtime[i].ServiceId] = dtime[i]
}
for _, v := range services.AllInOrder() { for _, v := range services.AllInOrder() {
if !v.Public.Bool && !IsUser(r) { if !v.Public.Bool && !IsUser(r) {
continue continue
} }
dtime :=findDowntimeByTime(t,v)// we get status of each service at time t
serviceJson, _ := json.Marshal(&v) serviceJson, _ := json.Marshal(&v)
var serviceDowntimeMap map[string]interface{} var serviceDowntimeMap map[string]interface{}
json.Unmarshal(serviceJson, &serviceDowntimeMap) json.Unmarshal(serviceJson, &serviceDowntimeMap)
jsonString, _ := json.Marshal(serviceDowntimeMap) jsonString, _ := json.Marshal(serviceDowntimeMap)
serviceDowntime := services.ServiceWithDowntime{} serviceDowntime := services.ServiceWithDowntime{}
json.Unmarshal(jsonString, &serviceDowntime) json.Unmarshal(jsonString, &serviceDowntime)
if dtime!=nil{ if vv, ok := m[v.Id]; ok == true {
serviceDowntime.Downtime = *dtime serviceDowntime.Downtime = vv
} }
srvs = append(srvs, serviceDowntime) srvs = append(srvs, serviceDowntime)
} }

View File

@ -58,17 +58,12 @@ func FindByService(service int64, start time.Time, end time.Time) (*[]Downtime,
return &downtime, q.Error() return &downtime, q.Error()
} }
func FindDowntime(service int64, timeVar time.Time) *Downtime { func FindDowntime(timeVar time.Time) []Downtime {
var downtime []Downtime var downtime []Downtime
q := db.Where("service = $1 and start BETWEEN $2 AND $3 ", service, time.Time{}, timeVar) q := db.Where("start <= ? and \"end\" >= ?", timeVar, timeVar)
q = q.Order("id ASC ").Find(&downtime) q = q.Order("id ASC ").Find(&downtime)
downtimeList := *(&downtime) return downtime
for _,dtime := range downtimeList{
if (*(dtime.End)).Unix() > timeVar.Unix(){
return &dtime
}
}
return nil
} }
func (c *Downtime) Create() error { func (c *Downtime) Create() error {

View File

@ -145,6 +145,7 @@ type ServiceWithDowntime struct {
ManualDowntime bool `gorm:"default:false;column:manual_downtime" json:"manual_downtime"` ManualDowntime bool `gorm:"default:false;column:manual_downtime" json:"manual_downtime"`
Downtime downtimes.Downtime `gorm:"-" json:"downtime" yaml:"-"` Downtime downtimes.Downtime `gorm:"-" json:"downtime" yaml:"-"`
} }
// ServiceOrder will reorder the services based on 'order_id' (Order) // ServiceOrder will reorder the services based on 'order_id' (Order)
type ServiceOrder []Service type ServiceOrder []Service