feat: add new features such as database watchdog
This commit is contained in:
@@ -1,36 +1,65 @@
|
||||
package database
|
||||
|
||||
import (
|
||||
"eden-server/internal/runtime"
|
||||
"path/filepath"
|
||||
"time"
|
||||
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
"gorm.io/gorm/schema"
|
||||
)
|
||||
|
||||
func KickoffDatabase() (*gorm.DB, error) {
|
||||
db, err := gorm.Open(sqlite.Open("garden.db"), &gorm.Config{
|
||||
NamingStrategy: schema.NamingStrategy{
|
||||
SingularTable: true,
|
||||
},
|
||||
})
|
||||
//var watchdogStop = make(chan struct{})
|
||||
|
||||
func KickoffDatabase(workDir string) (*gorm.DB, error) {
|
||||
dbLoc := filepath.Join(workDir, "data", "garden.db")
|
||||
db, err := gorm.Open(sqlite.Open(dbLoc), &gorm.Config{})
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
if err := db.AutoMigrate(&AppState{}); err != nil {
|
||||
if err := db.AutoMigrate(&State{}); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := db.AutoMigrate(&Files{}); err != nil {
|
||||
if err := db.AutoMigrate(&File{}); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
// create the first row if it does not exist yet
|
||||
if err := db.FirstOrCreate(&AppState{}, AppState{
|
||||
ID: 1,
|
||||
Mode: "unspecified",
|
||||
Running: false,
|
||||
if err := db.FirstOrCreate(&State{}, State{
|
||||
ID: 0,
|
||||
Mode: "unspecified",
|
||||
}).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
||||
return db, nil
|
||||
}
|
||||
|
||||
func KickoffDatabaseWatchdog(env runtime.Environment, db *gorm.DB) {
|
||||
timeInterval := time.Second * time.Duration(env.WatchInterval)
|
||||
ticker := time.NewTicker(timeInterval)
|
||||
|
||||
go func() {
|
||||
defer ticker.Stop()
|
||||
|
||||
/*
|
||||
// Possible future mechanism to stop the watchdog
|
||||
// must be inside a non-conditional for loop
|
||||
select {
|
||||
case <-ticker.C: // ticker event
|
||||
watchdog(env.DataDirectory, db)
|
||||
case <-watchdogStop:
|
||||
return
|
||||
}
|
||||
*/
|
||||
|
||||
// run the watchdog function once to see if all is well.
|
||||
watchdog(env.DataDirectory, db)
|
||||
// then defer to a decoupled/disowned golang goroutine
|
||||
|
||||
for range ticker.C {
|
||||
watchdog(env.DataDirectory, db)
|
||||
}
|
||||
}()
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user