You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
164 lines
3.1 KiB
164 lines
3.1 KiB
package database
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"git.tovijaeschke.xyz/tovi/JumboPetstore/models"
|
|
"git.tovijaeschke.xyz/tovi/JumboPetstore/util"
|
|
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
)
|
|
|
|
func CreatePet(petData *models.Pet) error {
|
|
var (
|
|
photoUrls []string
|
|
photoUrl string
|
|
fileName string
|
|
err error
|
|
)
|
|
|
|
for _, photoUrl = range petData.PhotoUrlJson {
|
|
fileName, err = util.DownloadFile(photoUrl)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
petData.PhotoUrls = append(petData.PhotoUrls, models.PetPhoto{
|
|
PetId: petData.Id,
|
|
FileName: fileName,
|
|
})
|
|
|
|
photoUrls = append(photoUrls, fmt.Sprintf(
|
|
"/images/%s",
|
|
fileName,
|
|
))
|
|
}
|
|
|
|
petData.PhotoUrlJson = photoUrls
|
|
|
|
err = DB.Session(&gorm.Session{FullSaveAssociations: true}).
|
|
Omit("PhotoUrlJson").
|
|
Create(petData).
|
|
Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
func UpdatePet(petData *models.Pet) error {
|
|
var (
|
|
photoUrls []string
|
|
photoUrl string
|
|
fileName string
|
|
err error
|
|
)
|
|
|
|
for _, photoUrl = range petData.PhotoUrlJson {
|
|
fileName, err = util.DownloadFile(photoUrl)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
petData.PhotoUrls = append(petData.PhotoUrls, models.PetPhoto{
|
|
PetId: petData.Id,
|
|
FileName: fileName,
|
|
})
|
|
|
|
photoUrls = append(photoUrls, fmt.Sprintf(
|
|
"/images/%s",
|
|
fileName,
|
|
))
|
|
}
|
|
|
|
petData.PhotoUrlJson = photoUrls
|
|
|
|
// This leaves some orphaned files in ./uploads
|
|
DB.Model(petData).Association("PhotoUrls").Replace(petData.PhotoUrls)
|
|
DB.Model(petData).Association("Tags").Replace(petData.Tags)
|
|
|
|
err = DB.Session(&gorm.Session{FullSaveAssociations: true}).
|
|
Omit("PhotoUrlJson").
|
|
Updates(petData).
|
|
Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
func GetPetById(id int) models.Pet {
|
|
var (
|
|
petData models.Pet
|
|
petPhoto models.PetPhoto
|
|
photoUrls []string
|
|
)
|
|
|
|
DB.Preload(clause.Associations).First(&petData, "id = ?", id)
|
|
|
|
for _, petPhoto = range petData.PhotoUrls {
|
|
photoUrls = append(photoUrls, "/images/"+petPhoto.FileName)
|
|
}
|
|
|
|
petData.PhotoUrlJson = photoUrls
|
|
|
|
return petData
|
|
}
|
|
|
|
func DeletePet(petData models.Pet) {
|
|
DB.Preload(clause.Associations).Delete(&petData)
|
|
}
|
|
|
|
func GetPetsByStatus(status string) ([]models.Pet, error) {
|
|
var (
|
|
petDatas []models.Pet
|
|
petPhoto models.PetPhoto
|
|
photoUrls []string
|
|
i int
|
|
err error
|
|
)
|
|
|
|
err = DB.Preload(clause.Associations).
|
|
Where("status = ?", status).
|
|
Find(&petDatas).
|
|
Error
|
|
if err != nil {
|
|
return petDatas, err
|
|
}
|
|
|
|
for i = range petDatas {
|
|
photoUrls = []string{}
|
|
|
|
for _, petPhoto = range petDatas[i].PhotoUrls {
|
|
photoUrls = append(photoUrls, "/images/"+petPhoto.FileName)
|
|
}
|
|
|
|
petDatas[i].PhotoUrlJson = photoUrls
|
|
}
|
|
|
|
return petDatas, err
|
|
}
|
|
|
|
func AddPhotoToPet(petData *models.Pet, fileName string, additionalMetadata string, fileBytes []byte) (string, error) {
|
|
var (
|
|
err error
|
|
)
|
|
|
|
fileName, err = util.WriteFile(fileName, fileBytes)
|
|
|
|
petData.PhotoUrls = append(petData.PhotoUrls, models.PetPhoto{
|
|
PetId: petData.Id,
|
|
FileName: fileName,
|
|
AdditionalMetadata: additionalMetadata,
|
|
})
|
|
|
|
err = DB.Session(&gorm.Session{FullSaveAssociations: true}).
|
|
Omit("PhotoUrlJson").
|
|
Updates(&petData).
|
|
Error
|
|
|
|
return fileName, err
|
|
}
|