90 lines
2.4 KiB
Go
90 lines
2.4 KiB
Go
package entity
|
|
|
|
import (
|
|
"crypto/sha1"
|
|
"fmt"
|
|
"time"
|
|
)
|
|
|
|
type PeopleFaces []PersonFace
|
|
|
|
// PersonFace represents the face of a Person.
|
|
type PersonFace struct {
|
|
ID string `gorm:"type:VARBINARY(42);primary_key;auto_increment:false;" json:"ID" yaml:"ID"`
|
|
PersonUID string `gorm:"type:VARBINARY(42);index;" json:"PersonUID" yaml:"PersonUID"`
|
|
FaceSrc string `gorm:"type:VARBINARY(8);" json:"Src" yaml:"Src"`
|
|
Embedding string `gorm:"type:LONGTEXT;" json:"Embedding" yaml:"Embedding,omitempty"`
|
|
PhotoCount int `gorm:"default:0" json:"PhotoCount" yaml:"-"`
|
|
CreatedAt time.Time `json:"CreatedAt" yaml:"-"`
|
|
UpdatedAt time.Time `json:"UpdatedAt" yaml:"-"`
|
|
DeletedAt *time.Time `sql:"index" json:"DeletedAt,omitempty" yaml:"-"`
|
|
}
|
|
|
|
// TableName returns the entity database table name.
|
|
func (PersonFace) TableName() string {
|
|
return "people_faces_dev"
|
|
}
|
|
|
|
/*
|
|
// BeforeCreate creates a random UID if needed before inserting a new row to the database.
|
|
func (m *PersonFace) BeforeCreate(scope *gorm.Scope) error {
|
|
return scope.SetColumn("ID")
|
|
}*/
|
|
|
|
// NewPersonFace returns a new face.
|
|
func NewPersonFace(personUID, faceSrc, embedding string, photoCount int) *PersonFace {
|
|
result := &PersonFace{
|
|
ID: fmt.Sprintf("%x", sha1.Sum([]byte(embedding))),
|
|
PersonUID: personUID,
|
|
FaceSrc: faceSrc,
|
|
Embedding: embedding,
|
|
PhotoCount: photoCount,
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
// UnmarshalEmbedding parses the face embedding JSON string.
|
|
func (m *PersonFace) UnmarshalEmbedding() (result Embedding) {
|
|
return UnmarshalEmbedding(m.Embedding)
|
|
}
|
|
|
|
// Save updates the existing or inserts a new face.
|
|
func (m *PersonFace) Save() error {
|
|
peopleMutex.Lock()
|
|
defer peopleMutex.Unlock()
|
|
|
|
return Db().Save(m).Error
|
|
}
|
|
|
|
// Create inserts the face to the database.
|
|
func (m *PersonFace) Create() error {
|
|
peopleMutex.Lock()
|
|
defer peopleMutex.Unlock()
|
|
|
|
return Db().Create(m).Error
|
|
}
|
|
|
|
// Delete removes the face from the database.
|
|
func (m *PersonFace) Delete() error {
|
|
return Db().Delete(m).Error
|
|
}
|
|
|
|
// Deleted returns true if the face is deleted.
|
|
func (m *PersonFace) Deleted() bool {
|
|
return m.DeletedAt != nil
|
|
}
|
|
|
|
// Restore restores the face in the database.
|
|
func (m *PersonFace) Restore() error {
|
|
if m.Deleted() {
|
|
return UnscopedDb().Model(m).Update("DeletedAt", nil).Error
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
// Update a face property in the database.
|
|
func (m *PersonFace) Update(attr string, value interface{}) error {
|
|
return UnscopedDb().Model(m).UpdateColumn(attr, value).Error
|
|
}
|