photoprism/internal/entity/photo_merge.go

122 lines
4.1 KiB
Go
Raw Normal View History

2020-12-12 22:02:14 +01:00
package entity
import (
"sync"
2020-12-12 22:02:14 +01:00
"github.com/jinzhu/gorm"
"github.com/photoprism/photoprism/pkg/rnd"
)
var photoMergeMutex = sync.Mutex{}
2020-12-12 22:02:14 +01:00
// ResolvePrimary ensures there is only one primary file for a photo.
func (m *Photo) ResolvePrimary() error {
var file File
if err := Db().Where("file_primary = 1 AND photo_id = ?", m.ID).First(&file).Error; err == nil && file.ID > 0 {
return file.ResolvePrimary()
}
return nil
}
// Identical returns identical photos that can be merged.
func (m *Photo) Identical(includeMeta, includeUuid bool) (identical Photos, err error) {
if m.PhotoStack == IsUnstacked || m.PhotoName == "" {
2020-12-12 22:02:14 +01:00
return identical, nil
}
switch {
case includeMeta && includeUuid && m.HasLocation() && m.TakenSrc == SrcMeta && rnd.IsUUID(m.UUID):
2020-12-12 22:02:14 +01:00
if err := Db().
Where("(taken_at = ? AND taken_src = 'meta' AND photo_stack > -1 AND cell_id = ? AND camera_serial = ? AND camera_id = ?) "+
"OR (uuid = ? AND photo_stack > -1)"+
2020-12-12 22:02:14 +01:00
"OR (photo_path = ? AND photo_name = ?)",
m.TakenAt, m.CellID, m.CameraSerial, m.CameraID, m.UUID, m.PhotoPath, m.PhotoName).
Order("photo_quality DESC, id ASC").Find(&identical).Error; err != nil {
2020-12-12 22:02:14 +01:00
return identical, err
}
case includeMeta && m.HasLocation() && m.TakenSrc == SrcMeta:
2020-12-12 22:02:14 +01:00
if err := Db().
Where("(taken_at = ? AND taken_src = 'meta' AND photo_stack > -1 AND cell_id = ? AND camera_serial = ? AND camera_id = ?) "+
2020-12-12 22:02:14 +01:00
"OR (photo_path = ? AND photo_name = ?)",
m.TakenAt, m.CellID, m.CameraSerial, m.CameraID, m.PhotoPath, m.PhotoName).
Order("photo_quality DESC, id ASC").Find(&identical).Error; err != nil {
2020-12-12 22:02:14 +01:00
return identical, err
}
case includeUuid && rnd.IsUUID(m.UUID):
if err := Db().
Where("(uuid = ? AND photo_stack > -1) OR (photo_path = ? AND photo_name = ?)",
2020-12-12 22:02:14 +01:00
m.UUID, m.PhotoPath, m.PhotoName).
Order("photo_quality DESC, id ASC").Find(&identical).Error; err != nil {
2020-12-12 22:02:14 +01:00
return identical, err
}
default:
if err := Db().
Where("photo_path = ? AND photo_name = ?", m.PhotoPath, m.PhotoName).
Order("photo_quality DESC, id ASC").Find(&identical).Error; err != nil {
2020-12-12 22:02:14 +01:00
return identical, err
}
}
return identical, nil
}
// Merge photo with identical ones.
func (m *Photo) Merge(mergeMeta, mergeUuid bool) (original Photo, merged Photos, err error) {
photoMergeMutex.Lock()
defer photoMergeMutex.Unlock()
2020-12-12 22:02:14 +01:00
identical, err := m.Identical(mergeMeta, mergeUuid)
if len(identical) < 2 || err != nil {
return Photo{}, merged, err
}
logResult := func(res *gorm.DB) {
if res.Error != nil {
log.Errorf("merge: %s", res.Error.Error())
err = res.Error
}
}
for i, merge := range identical {
if i == 0 {
original = merge
log.Debugf("photo: merging id %d with %d identical", original.ID, len(identical)-1)
continue
}
deleted := Timestamp()
logResult(UnscopedDb().Exec("UPDATE `files` SET photo_id = ?, photo_uid = ?, file_primary = 0 WHERE photo_id = ?", original.ID, original.PhotoUID, merge.ID))
logResult(UnscopedDb().Exec("UPDATE `photos` SET photo_quality = -1, deleted_at = ? WHERE id = ?", Timestamp(), merge.ID))
switch DbDialect() {
case MySQL:
logResult(UnscopedDb().Exec("UPDATE IGNORE `photos_keywords` SET `photo_id` = ? WHERE photo_id = ?", original.ID, merge.ID))
logResult(UnscopedDb().Exec("UPDATE IGNORE `photos_labels` SET `photo_id` = ? WHERE photo_id = ?", original.ID, merge.ID))
logResult(UnscopedDb().Exec("UPDATE IGNORE `photos_albums` SET `photo_uid` = ? WHERE photo_uid = ?", original.PhotoUID, merge.PhotoUID))
case SQLite:
logResult(UnscopedDb().Exec("UPDATE OR IGNORE `photos_keywords` SET `photo_id` = ? WHERE photo_id = ?", original.ID, merge.ID))
logResult(UnscopedDb().Exec("UPDATE OR IGNORE `photos_labels` SET `photo_id` = ? WHERE photo_id = ?", original.ID, merge.ID))
logResult(UnscopedDb().Exec("UPDATE OR IGNORE `photos_albums` SET `photo_uid` = ? WHERE photo_uid = ?", original.PhotoUID, merge.PhotoUID))
default:
log.Warnf("merge: unknown sql dialect")
}
merge.DeletedAt = &deleted
merge.PhotoQuality = -1
merged = append(merged, merge)
}
if original.ID != m.ID {
deleted := Timestamp()
m.DeletedAt = &deleted
m.PhotoQuality = -1
}
return original, merged, err
}