You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

83 lines
2.2 KiB
Go

3 years ago
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
}