Add session store fix crud func
This commit is contained in:
parent
5e83f17555
commit
1dcf4f691c
|
@ -5,11 +5,13 @@ import (
|
|||
"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) {
|
||||
|
@ -35,25 +37,30 @@ func DB() *gorm.DB {
|
|||
return db
|
||||
}
|
||||
|
||||
func create(d interface{}) error {
|
||||
return DB().Create(&d).Error
|
||||
func Store() *gormstore.Store {
|
||||
if store == nil {
|
||||
store = gormstore.NewOptions(DB(), gormstore.Options{}, []byte(os.Getenv("HASHKEY")), []byte("BLOCKKEY"))
|
||||
}
|
||||
return store
|
||||
}
|
||||
|
||||
func read(d *interface{}) error {
|
||||
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) ([]interface{}, error) {
|
||||
var data []interface{}
|
||||
err := DB().Offset(offset).Limit(limit).Where(&d).Find(&data).Error
|
||||
return data, err
|
||||
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 {
|
||||
func Update(d interface{}, where string, f interface{}) error {
|
||||
return DB().Model(d).Where(where+" = ?", f).Updates(d).Error
|
||||
}
|
||||
|
||||
func delete(d interface{}) error {
|
||||
func Delete(d interface{}) error {
|
||||
|
||||
return DB().Delete(d).Error
|
||||
}
|
||||
|
|
|
@ -0,0 +1,72 @@
|
|||
package models_test
|
||||
|
||||
import (
|
||||
"magmise/models"
|
||||
"testing"
|
||||
)
|
||||
|
||||
type Test struct {
|
||||
Name string
|
||||
}
|
||||
|
||||
func TestDB(t *testing.T) {
|
||||
// Create table
|
||||
models.DB().AutoMigrate(&models.User{})
|
||||
|
||||
user := models.User{Username: "test"}
|
||||
|
||||
//Create
|
||||
if err := models.Create(&user); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
t.Log("\033[33mShould error with unique contraint\033[0m")
|
||||
if err := models.Create(&user); err == nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
// Read
|
||||
if err := models.Read(&user); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
// Create more rows
|
||||
user = models.User{Username: "test2"}
|
||||
models.Create(&user)
|
||||
user = models.User{Username: "test3"}
|
||||
models.Create(&user)
|
||||
user = models.User{Username: "test4"}
|
||||
models.Create(&user)
|
||||
|
||||
// Get second and third row
|
||||
var users []models.User
|
||||
if err := models.ReadAll(&users, 1, 2); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if users[0].Username != "test2" {
|
||||
t.Fatal("Offset failed\n", users)
|
||||
}
|
||||
|
||||
if len(users) != 2 {
|
||||
t.Fatal("Limit failed\n", users)
|
||||
}
|
||||
|
||||
// Update
|
||||
user.Password = "b"
|
||||
if err := models.Update(&user, "username", user.Username); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
// Check Update
|
||||
if err := models.Read(&user); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if user.Password != "b" {
|
||||
t.Fatal("failed update")
|
||||
}
|
||||
|
||||
// Delete
|
||||
if err := models.Delete(&user); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
models.DB().Migrator().DropTable(&models.User{})
|
||||
}
|
Loading…
Reference in New Issue