Add go backend

This commit is contained in:
2022-12-18 13:42:21 +13:00
parent 5719891df8
commit 99f88d6adf
15 changed files with 1179 additions and 0 deletions

66
api/models/db.go Normal file
View 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
View File

@@ -0,0 +1,9 @@
package models
import "fmt"
func migrate() {
DB().AutoMigrate(&User{})
fmt.Println("Migrated")
}

37
api/models/user.go Normal file
View 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
}