mirror of https://github.com/statping/statping
114 lines
3.0 KiB
Go
114 lines
3.0 KiB
Go
package configs
|
|
|
|
import (
|
|
"github.com/go-yaml/yaml"
|
|
"github.com/hunterlong/statping/database"
|
|
"github.com/hunterlong/statping/types/checkins"
|
|
"github.com/hunterlong/statping/types/core"
|
|
"github.com/hunterlong/statping/types/failures"
|
|
"github.com/hunterlong/statping/types/groups"
|
|
"github.com/hunterlong/statping/types/hits"
|
|
"github.com/hunterlong/statping/types/incidents"
|
|
"github.com/hunterlong/statping/types/integrations"
|
|
"github.com/hunterlong/statping/types/messages"
|
|
"github.com/hunterlong/statping/types/notifications"
|
|
"github.com/hunterlong/statping/types/services"
|
|
"github.com/hunterlong/statping/types/users"
|
|
"github.com/hunterlong/statping/utils"
|
|
"os"
|
|
)
|
|
|
|
type Sampler interface {
|
|
Samples() []database.DbObject
|
|
}
|
|
|
|
func TriggerSamples() error {
|
|
return createSamples(
|
|
&services.Service{},
|
|
&users.User{},
|
|
&hits.Hit{},
|
|
&failures.Failure{},
|
|
&groups.Group{},
|
|
&checkins.Checkin{},
|
|
&checkins.CheckinHit{},
|
|
&incidents.Incident{},
|
|
&incidents.IncidentUpdate{},
|
|
)
|
|
}
|
|
|
|
func createSamples(sm ...Sampler) error {
|
|
for _, v := range sm {
|
|
for _, sample := range v.Samples() {
|
|
if err := sample.Create(); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (d *DbConfig) Connect() error {
|
|
|
|
return nil
|
|
}
|
|
|
|
func (d *DbConfig) Create() error {
|
|
|
|
return nil
|
|
}
|
|
|
|
// Migrate function
|
|
func (d *DbConfig) Update() error {
|
|
var err error
|
|
config, err := os.Create(utils.Directory + "/config.yml")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer config.Close()
|
|
|
|
data, err := yaml.Marshal(d)
|
|
if err != nil {
|
|
log.Errorln(err)
|
|
return err
|
|
}
|
|
config.WriteString(string(data))
|
|
return nil
|
|
}
|
|
|
|
// Save will initially create the config.yml file
|
|
func (d *DbConfig) Delete() error {
|
|
return os.Remove(d.filename)
|
|
}
|
|
|
|
// DropDatabase will DROP each table Statping created
|
|
func (d *DbConfig) DropDatabase() error {
|
|
var DbModels = []interface{}{&services.Service{}, &users.User{}, &hits.Hit{}, &failures.Failure{}, &messages.Message{}, &groups.Group{}, &checkins.Checkin{}, &checkins.CheckinHit{}, ¬ifications.Notification{}, &incidents.Incident{}, &incidents.IncidentUpdate{}, &integrations.Integration{}}
|
|
log.Infoln("Dropping Database Tables...")
|
|
for _, t := range DbModels {
|
|
if err := database.DB().DropTableIfExists(t); err != nil {
|
|
return err.Error()
|
|
}
|
|
log.Infof("Dropped table: %T\n", t)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// CreateDatabase will CREATE TABLES for each of the Statping elements
|
|
func CreateDatabase() error {
|
|
var err error
|
|
|
|
var DbModels = []interface{}{&services.Service{}, &users.User{}, &hits.Hit{}, &failures.Failure{}, &messages.Message{}, &groups.Group{}, &checkins.Checkin{}, &checkins.CheckinHit{}, ¬ifications.Notification{}, &incidents.Incident{}, &incidents.IncidentUpdate{}, &integrations.Integration{}}
|
|
|
|
log.Infoln("Creating Database Tables...")
|
|
for _, table := range DbModels {
|
|
if err := database.DB().CreateTable(table); err.Error() != nil {
|
|
return err.Error()
|
|
}
|
|
}
|
|
if err := database.DB().Table("core").CreateTable(&core.Core{}); err.Error() != nil {
|
|
return err.Error()
|
|
}
|
|
log.Infoln("Statping Database Created")
|
|
return err
|
|
}
|