photoprism/internal/entity/entity_count.go

31 lines
639 B
Go
Raw Normal View History

package entity
import (
"github.com/jinzhu/gorm"
)
// Count returns the number of records for a given a model and key values.
func Count(m interface{}, keys []string, values []interface{}) int {
if m == nil || len(keys) != len(values) {
log.Debugf("entity: invalid parameters (count records)")
return -1
}
db, count := UnscopedDb(), 0
stmt := db.Model(m)
// Compose where condition.
for k := range keys {
stmt.Where("? = ?", gorm.Expr(keys[k]), values[k])
}
// Fetch count from database.
if err := stmt.Count(&count).Error; err != nil {
log.Debugf("entity: %s (count records)", err)
return -1
}
return count
}