photoprism/internal/photoprism/places.go

package photoprism

import (
	"fmt"
	"runtime/debug"
	"time"

	"github.com/dustin/go-humanize/english"

	"github.com/photoprism/photoprism/internal/config"
	"github.com/photoprism/photoprism/internal/entity"
	"github.com/photoprism/photoprism/internal/mutex"
	"github.com/photoprism/photoprism/internal/query"
)

// Places represents a geo data worker.
type Places struct {
	conf *config.Config
}

// NewPlaces returns a new Places worker.
func NewPlaces(conf *config.Config) *Places {
	instance := &Places{
		conf: conf,
	}

	return instance
}

// Start runs the Places worker.
func (w *Places) Start() (updated []string, err error) {
	defer func() {
		if r := recover(); r != nil {
			err = fmt.Errorf("index: %s (update locations)\nstack: %s", r, debug.Stack())
			log.Error(err)
		}
	}()

	// Check if a worker is already running.
	if err := mutex.MainWorker.Start(); err != nil {
		log.Warnf("index: %s (update locations)", err.Error())
		return []string{}, err
	}

	defer mutex.MainWorker.Stop()

	// Fetch cell IDs from index.
	cells, err := query.CellIDs()

	// Error?
	if err != nil {
		return []string{}, err
	} else if len(cells) == 0 {
		log.Warnf("index: found no locations to update")
		return []string{}, nil
	}

	// List of updated cells.
	updated = make([]string, 0, len(cells))

	log.Infof("index: retrieving location details")

	// Update known locations.
	for i, cell := range cells {
		if i%10 == 0 {
			log.Infof("index: updated %s, %s remaining",
				english.Plural(i, "location", "locations"),
				english.Plural(len(cells)-i, "location", "locations"))
		}

		if w.Canceled() {
			return updated, nil
		} else if cell.ID == "" || cell.ID == entity.UnknownID {
			// Skip unknown places.
			continue
		}

		// Create cell from location and place ID.
		c := entity.Cell{ID: cell.ID, PlaceID: cell.PlaceID}

		// Fetch updated cell data from backend API.
		if err = c.Refresh(entity.GeoApi); err != nil {
			log.Warnf("index: %s", err)
		} else {
			// Append if successful.
			updated = append(updated, cell.ID)
		}

		// Short break.
		time.Sleep(33 * time.Millisecond)
	}

	// Remove unused entries from the places table.
	if err := query.PurgePlaces(); err != nil {
		log.Errorf("index: %s (purge places)", err)
	}

	// Update location-related photo metadata in the index.
	if _, err := w.UpdatePhotos(); err != nil {
		log.Errorf("index: %s (update photos)", err)
	}

	// Update photo counts in places.
	if err := entity.UpdatePlacesCounts(); err != nil {
		log.Errorf("index: %s (update counts)", err)
	}

	return updated, err
}

// UpdatePhotos updates all location-related photo metadata in the index.
func (w *Places) UpdatePhotos() (affected int, err error) {
	start := time.Now()

	var u []string

	// Find photos without location.
	if err = query.UnscopedDb().
		Raw(`SELECT photo_uid FROM photos WHERE place_id <> 'zz' OR photo_lat <> 0 OR photo_lng <> 0 ORDER BY id`).
		Pluck("photo_uid", &u).Error; err != nil {
		return affected, err
	}

	n := len(u)

	if n == 0 {
		log.Debugf("index: found no photos with location [%s]", time.Since(start))
		return affected, err
	}

	log.Infof("index: updating references, titles, and keywords")

	for i := 0; i < n; i++ {
		if i%10 == 0 {
			log.Infof("index: updated %s, %s remaining",
				english.Plural(i, "photo", "photos"),
				english.Plural(n-i, "photo", "photos"))
		}

		var model entity.Photo

		model, err = query.PhotoByUID(u[i])

		if err != nil {
			log.Errorf("index: %s while loading %s", err, model.String())
			continue
		} else if model.NoLatLng() {
			log.Debugf("index: photo %s has no location", model.String())
			continue
		}

		if err = model.SaveLocation(); err != nil {
			log.Errorf("index: %s while updating %s", err, model.String())
		} else {
			affected++
		}
	}

	return affected, err
}

// Canceled tests if the worker should be stopped.
func (w *Places) Canceled() bool {
	return mutex.MainWorker.Canceled() || mutex.MetaWorker.Canceled()
}

// Cancel stops the current operation.
func (w *Places) Cancel() {
	mutex.MainWorker.Cancel()
}