2019-12-11 15:55:18 +00:00
|
|
|
package entity
|
2019-12-08 14:05:35 +00:00
|
|
|
|
|
|
|
import (
|
|
|
|
"strings"
|
2020-12-13 14:43:01 +00:00
|
|
|
"sync"
|
2019-12-08 14:05:35 +00:00
|
|
|
|
2020-04-26 12:31:33 +00:00
|
|
|
"github.com/photoprism/photoprism/pkg/txt"
|
2019-12-08 14:05:35 +00:00
|
|
|
)
|
|
|
|
|
2020-12-13 14:43:01 +00:00
|
|
|
var keywordMutex = sync.Mutex{}
|
|
|
|
|
2020-02-21 00:14:45 +00:00
|
|
|
// Keyword used for full text search
|
2019-12-08 14:05:35 +00:00
|
|
|
type Keyword struct {
|
|
|
|
ID uint `gorm:"primary_key"`
|
2020-09-13 15:51:43 +00:00
|
|
|
Keyword string `gorm:"type:VARCHAR(64);index;"`
|
2019-12-08 14:05:35 +00:00
|
|
|
Skip bool
|
|
|
|
}
|
|
|
|
|
2022-09-28 07:01:17 +00:00
|
|
|
// TableName returns the entity table name.
|
2022-09-02 19:30:50 +00:00
|
|
|
func (Keyword) TableName() string {
|
|
|
|
return "keywords"
|
|
|
|
}
|
|
|
|
|
2020-02-21 00:14:45 +00:00
|
|
|
// NewKeyword registers a new keyword in database
|
2019-12-08 14:05:35 +00:00
|
|
|
func NewKeyword(keyword string) *Keyword {
|
2020-04-26 12:31:33 +00:00
|
|
|
keyword = strings.ToLower(txt.Clip(keyword, txt.ClipKeyword))
|
2019-12-08 14:05:35 +00:00
|
|
|
|
|
|
|
result := &Keyword{
|
|
|
|
Keyword: keyword,
|
|
|
|
}
|
|
|
|
|
|
|
|
return result
|
|
|
|
}
|
|
|
|
|
2020-05-26 09:00:39 +00:00
|
|
|
// Updates multiple columns in the database.
|
|
|
|
func (m *Keyword) Updates(values interface{}) error {
|
2022-04-04 06:54:03 +00:00
|
|
|
return UnscopedDb().Model(m).UpdateColumns(values).Error
|
2020-05-26 09:00:39 +00:00
|
|
|
}
|
|
|
|
|
2021-09-29 18:09:34 +00:00
|
|
|
// Update a column in the database.
|
2020-05-26 09:00:39 +00:00
|
|
|
func (m *Keyword) Update(attr string, value interface{}) error {
|
2022-04-04 06:54:03 +00:00
|
|
|
return UnscopedDb().Model(m).UpdateColumn(attr, value).Error
|
2020-05-26 09:00:39 +00:00
|
|
|
}
|
|
|
|
|
2022-10-02 09:38:30 +00:00
|
|
|
// Save updates the record in the database or inserts a new record if it does not already exist.
|
2020-05-26 09:00:39 +00:00
|
|
|
func (m *Keyword) Save() error {
|
|
|
|
return Db().Save(m).Error
|
|
|
|
}
|
|
|
|
|
|
|
|
// Create inserts a new row to the database.
|
|
|
|
func (m *Keyword) Create() error {
|
2020-12-13 14:43:01 +00:00
|
|
|
keywordMutex.Lock()
|
|
|
|
defer keywordMutex.Unlock()
|
|
|
|
|
2020-05-26 09:00:39 +00:00
|
|
|
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
|
2020-07-07 18:44:33 +00:00
|
|
|
} 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 {
|
2020-12-14 12:31:18 +00:00
|
|
|
log.Errorf("keyword: %s (find or create %s)", createErr, m.Keyword)
|
2019-12-19 08:37:10 +00:00
|
|
|
}
|
2019-12-08 14:05:35 +00:00
|
|
|
|
2020-07-07 18:44:33 +00:00
|
|
|
return nil
|
2019-12-08 14:05:35 +00:00
|
|
|
}
|