mirror of https://github.com/statping/statping
67 lines
1.1 KiB
Go
67 lines
1.1 KiB
Go
package groups
|
|
|
|
import (
|
|
"github.com/statping/statping/database"
|
|
"github.com/statping/statping/types/errors"
|
|
"sort"
|
|
)
|
|
|
|
var db database.Database
|
|
|
|
func SetDB(database database.Database) {
|
|
db = database.Model(&Group{})
|
|
}
|
|
|
|
func Find(id int64) (*Group, error) {
|
|
var group Group
|
|
q := db.Where("id = ?", id).Find(&group)
|
|
if q.Error() != nil {
|
|
return nil, errors.Missing(group, id)
|
|
}
|
|
return &group, q.Error()
|
|
}
|
|
|
|
func All() []*Group {
|
|
var groups []*Group
|
|
db.Find(&groups)
|
|
return groups
|
|
}
|
|
|
|
func (g *Group) Create() error {
|
|
q := db.Create(g)
|
|
return q.Error()
|
|
}
|
|
|
|
func (g *Group) Update() error {
|
|
q := db.Update(g)
|
|
return q.Error()
|
|
}
|
|
|
|
func (g *Group) Delete() error {
|
|
q := db.Delete(g)
|
|
return q.Error()
|
|
}
|
|
|
|
// SelectGroups returns all groups
|
|
func SelectGroups(includeAll bool, auth bool) []*Group {
|
|
var validGroups []*Group
|
|
|
|
all := All()
|
|
if includeAll {
|
|
sort.Sort(GroupOrder(all))
|
|
return all
|
|
}
|
|
|
|
for _, g := range all {
|
|
if !g.Public.Bool {
|
|
if auth {
|
|
validGroups = append(validGroups, g)
|
|
}
|
|
} else {
|
|
validGroups = append(validGroups, g)
|
|
}
|
|
}
|
|
sort.Sort(GroupOrder(validGroups))
|
|
return validGroups
|
|
}
|