Fix album downloads when running production

This commit is contained in:
viktorstrate
2021-09-26 15:15:31 +02:00
parent 68b6a49638
commit 7799674b74
2 changed files with 11 additions and 2 deletions

View File

@@ -36,8 +36,15 @@ func RegisterDownloadRoutes(db *gorm.DB, router *mux.Router) {
return
}
var mediaWhereQuery string
if db.Dialector.Name() == "postgres" {
mediaWhereQuery = "\"Media\".album_id = ?"
} else {
mediaWhereQuery = "Media.album_id = ?"
}
var mediaURLs []*models.MediaURL
if err := db.Joins("Media").Where("media.album_id = ?", album.ID).Where("media_urls.purpose IN (?)", mediaPurposeList).Find(&mediaURLs).Error; err != nil {
if err := db.Joins("Media").Where(mediaWhereQuery, album.ID).Where("media_urls.purpose IN (?)", mediaPurposeList).Find(&mediaURLs).Error; err != nil {
w.WriteHeader(http.StatusInternalServerError)
w.Write([]byte("internal server error"))
return