31 lines
639 B
Go
31 lines
639 B
Go
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
|
|
}
|