Add go backend
This commit is contained in:
66
api/models/db.go
Normal file
66
api/models/db.go
Normal file
@@ -0,0 +1,66 @@
|
||||
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
|
||||
}
|
||||
9
api/models/migrations.go
Normal file
9
api/models/migrations.go
Normal file
@@ -0,0 +1,9 @@
|
||||
package models
|
||||
|
||||
import "fmt"
|
||||
|
||||
func migrate() {
|
||||
DB().AutoMigrate(&User{})
|
||||
|
||||
fmt.Println("Migrated")
|
||||
}
|
||||
37
api/models/user.go
Normal file
37
api/models/user.go
Normal file
@@ -0,0 +1,37 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
type User struct {
|
||||
gorm.Model
|
||||
Username string `json:"username" gorm:"uniqueIndex, column: username"`
|
||||
Password string
|
||||
}
|
||||
|
||||
func (u User) Create() error {
|
||||
return DB().Create(&u).Error
|
||||
}
|
||||
|
||||
func (u *User) Read() error {
|
||||
return Read(u)
|
||||
}
|
||||
|
||||
func (u User) ReadAll(offset, limit int) ([]User, error) {
|
||||
var users []User
|
||||
err := ReadAll(&users, offset, limit)
|
||||
return users, err
|
||||
}
|
||||
|
||||
func (u *User) Update() error {
|
||||
return Update(u, "username", u.Username)
|
||||
}
|
||||
|
||||
func (u User) Delete() error {
|
||||
err := u.Read()
|
||||
if err == nil {
|
||||
err = Delete(&u)
|
||||
}
|
||||
return err
|
||||
}
|
||||
Reference in New Issue
Block a user