gallery/api/models/db.go

67 lines
1.2 KiB
Go

package models
import (
"errors"
"log"
"os"
"github.com/wader/gormstore/v2"
"gorm.io/driver/sqlite"
"gorm.io/gorm"
)
var db *gorm.DB
var store *gormstore.Store
func Init() {
if _, err := os.Stat("/data/database.db"); errors.Is(err, os.ErrNotExist) {
log.Printf("Creating sqlite database")
_, err := os.Create("/data/database.db")
if err != nil {
log.Fatal(err)
}
}
var err error
db, err = gorm.Open(sqlite.Open("/data/database.db"), &gorm.Config{})
if err != nil {
log.Println("Failed to connect to db")
}
migrate()
}
func DB() *gorm.DB {
if db == nil {
Init()
}
return db
}
func Store() *gormstore.Store {
if store == nil {
store = gormstore.NewOptions(DB(), gormstore.Options{}, []byte(os.Getenv("HASHKEY")), []byte("BLOCKKEY"))
}
return store
}
func Create(d interface{}) error {
return DB().Create(d).Error
}
func Read(d interface{}) error {
return DB().First(d, d).Error
}
func ReadAll(d interface{}, offset, limit int) error {
return DB().Offset(offset).Limit(limit).Find(d).Error
}
func Update(d interface{}, where string, f interface{}) error {
return DB().Model(d).Where(where+" = ?", f).Updates(d).Error
}
func Delete(d interface{}) error {
return DB().Delete(d).Error
}