JohnCosta27/Haystack
1
0
Fork 0
Code Issues 7 Pull Requests Actions Packages Projects 1 Releases Wiki Activity

feat: methods to get image

Browse Source
This commit is contained in:
John Costa
2025-02-24 20:02:58 +00:00
parent c0fe7e1853
commit e69d7b5c08
2 changed files with 43 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
backend/models/image.go
View File

@@ -3,8 +3,6 @@ package models
import (
"errors"
"fmt"
"log"
"screenmark/screenmark/.gen/haystack/haystack/model"
. "screenmark/screenmark/.gen/haystack/haystack/table"
@@ -13,18 +11,18 @@ import (
"github.com/google/uuid"
)
func SaveImage(userId string, imageName string, imageData []byte) error {
stmt := UserImages.INSERT(UserImages.UserID, UserImages.ImageName, UserImages.Image).VALUES(userId, imageName, imageData)
_, err := stmt.Exec(db)
func SaveImage(userId string, imageName string, imageData []byte) (model.UserImages, error) {
stmt := UserImages.INSERT(UserImages.UserID, UserImages.ImageName, UserImages.Image).VALUES(userId, imageName, imageData).RETURNING(UserImages.ID, UserImages.UserID, UserImages.ImageName)
return err
userImage := model.UserImages{}
err := stmt.Query(db, &userImage)
return userImage, err
}
func GetImage(imageId string) (model.UserImages, error) {
id := uuid.MustParse(imageId)
stmt := UserImages.SELECT(UserImages.ImageName, UserImages.Image).WHERE(UserImages.ID.EQ(UUID(id)))
log.Println(stmt.DebugSql())
stmt := UserImages.SELECT(UserImages.AllColumns).WHERE(UserImages.ID.EQ(UUID(id)))
images := []model.UserImages{}
err := stmt.Query(db, &images)