114 lines
2.2 KiB
Go
114 lines
2.2 KiB
Go
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
|
|
}
|