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

feat: not deleting images to process

Browse Source
This commit is contained in:
John Costa
2025-05-10 15:12:25 +01:00
parent 8e6424aa63
commit 2f3d643278
1 changed files with 25 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

32
backend/models/image.go
View File

@ -130,14 +130,21 @@ func (m ImageModel) FinishProcessing(ctx context.Context, imageId uuid.UUID) (mo
return model.UserImages{}, err
}
removeProcessingStmt := UserImagesToProcess.
DELETE().
WHERE(UserImagesToProcess.ID.EQ(UUID(imageToProcess.ID)))
// TODO:
// We cannot delete the image to process because our events rely on it.
// This indicates our DB structure with the two tables might need some adjusting.
// Or re-doing all together perhaps.
// (switching to a one table (user_images) could work)
// But for now, we can just not delete the images to process and set them to complete
_, err = removeProcessingStmt.ExecContext(ctx, tx)
if err != nil {
return model.UserImages{}, err
}
// removeProcessingStmt := UserImagesToProcess.
// DELETE().
// WHERE(UserImagesToProcess.ID.EQ(UUID(imageToProcess.ID)))
//
// _, err = removeProcessingStmt.ExecContext(ctx, tx)
// if err != nil {
// return model.UserImages{}, err
// }
err = tx.Commit()
return userImage, err
@ -184,6 +191,17 @@ func GetUserId(dbPool *sql.DB, ctx context.Context, imageId uuid.UUID) (uuid.UUI
return userImage.UserID, err
}
func GetUserIdComplete(dbPool *sql.DB, ctx context.Context, imageId uuid.UUID) (uuid.UUID, error) {
getUserIdStmt := UserImages.
SELECT(UserImages.UserID).
WHERE(UserImages.ID.EQ(UUID(imageId)))
userImage := model.UserImagesToProcess{}
err := getUserIdStmt.QueryContext(ctx, dbPool, &userImage)
return userImage.UserID, err
}
func NewImageModel(db *sql.DB) ImageModel {
return ImageModel{dbPool: db}
}