photoprism/internal/entity/keyword.go
Michael Mayer 24eff21aa4 Search: Default to photo names and keywords #1517 #1560
Default to photo name when search term is too short or on the stop list.
Search full text index otherwise, which now include names of people
(requires reindexing).
2021-09-29 20:09:34 +02:00

69 lines
1.6 KiB
Go

package entity
import (
"strings"
"sync"
"github.com/photoprism/photoprism/pkg/txt"
)
var keywordMutex = sync.Mutex{}
// Keyword used for full text search
type Keyword struct {
ID uint `gorm:"primary_key"`
Keyword string `gorm:"type:VARCHAR(64);index;"`
Skip bool
}
// NewKeyword registers a new keyword in database
func NewKeyword(keyword string) *Keyword {
keyword = strings.ToLower(txt.Clip(keyword, txt.ClipKeyword))
result := &Keyword{
Keyword: keyword,
}
return result
}
// Updates multiple columns in the database.
func (m *Keyword) Updates(values interface{}) error {
return UnscopedDb().Model(m).UpdateColumns(values).Error
}
// Update a column in the database.
func (m *Keyword) Update(attr string, value interface{}) error {
return UnscopedDb().Model(m).UpdateColumn(attr, value).Error
}
// Save updates the existing or inserts a new row.
func (m *Keyword) Save() error {
return Db().Save(m).Error
}
// Create inserts a new row to the database.
func (m *Keyword) Create() error {
keywordMutex.Lock()
defer keywordMutex.Unlock()
return Db().Create(m).Error
}
// FirstOrCreateKeyword returns the existing row, inserts a new row or nil in case of errors.
func FirstOrCreateKeyword(m *Keyword) *Keyword {
result := Keyword{}
if err := Db().Where("keyword = ?", m.Keyword).First(&result).Error; err == nil {
return &result
} else if createErr := m.Create(); createErr == nil {
return m
} else if err := Db().Where("keyword = ?", m.Keyword).First(&result).Error; err == nil {
return &result
} else {
log.Errorf("keyword: %s (find or create %s)", createErr, m.Keyword)
}
return nil
}