sql privider
parent
78f1cc8339
commit
892307185a
GPG Key ID:
91991C9808A18BB0
@ -0,0 +1,11 @@
|
||||
{
|
||||
"sqltools.connections": [
|
||||
{
|
||||
"previewLimit": 50,
|
||||
"driver": "SQLite",
|
||||
"name": "devdb",
|
||||
"database": "${workspaceFolder}/dev.db"
|
||||
}
|
||||
],
|
||||
"sqltools.useNodeRuntime": true
|
||||
}
|
||||
@ -0,0 +1,82 @@
|
||||
package implementations
|
||||
|
||||
import (
|
||||
"dymatrix.de/jspahl/todo/internal/models"
|
||||
"dymatrix.de/jspahl/todo/internal/types"
|
||||
)
|
||||
|
||||
type SqlProvider struct {
|
||||
}
|
||||
|
||||
func (prov SqlProvider) CreateItem(item types.IItem) error {
|
||||
storedItem := models.Item{Message: item.GetMessage()}
|
||||
result := models.DB.Create(storedItem)
|
||||
if result.Error != nil {
|
||||
return result.Error
|
||||
}
|
||||
return nil
|
||||
}
|
||||
func (prov SqlProvider) UpdateItem(item types.IItem) error {
|
||||
var dbItem models.Item
|
||||
result := models.DB.First(&dbItem, item.GetId())
|
||||
if result.Error != nil {
|
||||
return result.Error
|
||||
}
|
||||
dbItem.Message = item.GetMessage()
|
||||
result = models.DB.Save(&dbItem)
|
||||
if result.Error != nil {
|
||||
return result.Error
|
||||
}
|
||||
return nil
|
||||
}
|
||||
func (prov SqlProvider) GetItem(id int) (types.IItem, error) {
|
||||
var dbItem models.Item
|
||||
result := models.DB.First(&dbItem, id)
|
||||
if result.Error != nil {
|
||||
return nil, result.Error
|
||||
}
|
||||
return dbItem, nil
|
||||
}
|
||||
func (prov SqlProvider) GetItemByUser(user types.IUser) ([]types.IItem, error) {
|
||||
var dbUser models.User
|
||||
if result := models.DB.Preload("Items").First(&dbUser, user.GetId()); result.Error != nil {
|
||||
return nil, result.Error
|
||||
}
|
||||
var items []types.IItem
|
||||
for _, item := range dbUser.Items {
|
||||
items = append(items, item)
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
func (prov SqlProvider) GetAllItems() ([]types.IItem, error) {
|
||||
var dbItems []models.Item
|
||||
if err := models.DB.Find(&dbItems).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
var items []types.IItem
|
||||
for _, item := range dbItems {
|
||||
items = append(items, item)
|
||||
}
|
||||
return items, nil
|
||||
}
|
||||
func (prov SqlProvider) Link(user types.IUser, item types.IItem) error {
|
||||
var dbUser models.User
|
||||
if err := models.DB.First(&dbUser, user.GetId()).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
var dbItem models.Item
|
||||
if err := models.DB.First(&dbItem, item.GetId()).Error; err != nil {
|
||||
return err
|
||||
}
|
||||
dbUser.Items = append(dbUser.Items, dbItem)
|
||||
dbItem.UserID = dbUser.ID
|
||||
return models.DB.Save(&dbUser).Save(&dbItem).Error
|
||||
}
|
||||
func (prov SqlProvider) DeleteItem(item types.IItem) error {
|
||||
return models.DB.Where("id = ?", item.GetId()).Delete(&models.Item{}).Error
|
||||
}
|
||||
func (prov SqlProvider) CreateUser(user types.IUser) (int, error) {
|
||||
dbUser := models.User{}
|
||||
err := models.DB.Create(&dbUser).Error
|
||||
return int(dbUser.ID), err
|
||||
}
|
||||
@ -0,0 +1,17 @@
|
||||
package models
|
||||
|
||||
import "gorm.io/gorm"
|
||||
|
||||
type Item struct {
|
||||
gorm.Model
|
||||
Message string
|
||||
UserID uint
|
||||
}
|
||||
|
||||
func (item Item) GetId() int {
|
||||
return int(item.ID)
|
||||
}
|
||||
|
||||
func (item Item) GetMessage() string {
|
||||
return item.Message
|
||||
}
|
||||
@ -0,0 +1,17 @@
|
||||
package models
|
||||
|
||||
import "gorm.io/gorm"
|
||||
|
||||
type User struct {
|
||||
gorm.Model
|
||||
UserName string
|
||||
Items []Item
|
||||
}
|
||||
|
||||
func (user User) GetId() int {
|
||||
return int(user.ID)
|
||||
}
|
||||
|
||||
func (user User) GetUsername() string {
|
||||
return user.UserName
|
||||
}
|
||||
@ -0,0 +1,20 @@
|
||||
package models
|
||||
|
||||
import (
|
||||
"gorm.io/driver/sqlite"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
var DB *gorm.DB
|
||||
|
||||
func ConnectDatabase() error {
|
||||
db, err := gorm.Open(sqlite.Open("dev.db"), &gorm.Config{})
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
db.AutoMigrate(&Item{}, &User{})
|
||||
|
||||
DB = db
|
||||
return nil
|
||||
}
|
||||
Loading…
Reference in New Issue