package main import ( // External we "git.gibonuddevalla.se/go/wrappederror" // Standard "database/sql" "encoding/json" "time" ) type Problem struct { ID int Start time.Time End sql.NullTime Acknowledged bool TriggerID int `json:"trigger_id"` TriggerName string `json:"trigger_name"` AreaName string `json:"area_name"` SectionName string `json:"section_name"` } func ProblemsRetrieve() (problems []Problem, err error) { problems = []Problem{} row := service.Db.Conn.QueryRow(` SELECT jsonb_agg(p.*) FROM ( SELECT p.id, p.start, p.end, p.acknowledged, t.id AS trigger_id, t.name AS trigger_name, a.name AS area_name, s.name AS section_name FROM problem p INNER JOIN "trigger" t ON p.trigger_id = t.id INNER JOIN section s ON t.section_id = s.id INNER JOIN area a ON s.area_id = a.id WHERE p.end IS NULL ORDER BY p.start DESC ) p `) var jsonBody []byte err = row.Scan(&jsonBody) if err != nil { err = we.Wrap(err) return } if jsonBody == nil { return } err = json.Unmarshal(jsonBody, &problems) if err != nil { err = we.Wrap(err) } return } func ProblemStart(trigger Trigger) (err error) { row := service.Db.Conn.QueryRow(` SELECT COUNT(id) FROM problem WHERE trigger_id = $1 AND "end" IS NULL GROUP BY trigger_id `, trigger.ID, ) var openProblems int err = row.Scan(&openProblems) if err != nil && err != sql.ErrNoRows { err = we.Wrap(err).WithData(trigger.ID) return } // Open up a new problem if no open exists. if openProblems == 0 { _, err = service.Db.Conn.Exec(`INSERT INTO problem(trigger_id) VALUES($1)`, trigger.ID) if err != nil { err = we.Wrap(err).WithData(trigger) } } return } func ProblemClose(trigger Trigger) (err error) { _, err = service.Db.Conn.Exec(`UPDATE problem SET "end"=NOW() WHERE trigger_id=$1 AND "end" IS NULL`, trigger.ID) if err != nil { err = we.Wrap(err).WithData(trigger) return } return } func ProblemAcknowledge(id int, state bool) (err error) { _, err = service.Db.Conn.Exec(`UPDATE problem SET "acknowledged"=$2 WHERE id=$1`, id, state) if err != nil { err = we.Wrap(err).WithData(id) return } return }