mirror of
https://github.com/acepanel/panel.git
synced 2026-02-04 04:22:33 +08:00
44 lines
1001 B
Go
44 lines
1001 B
Go
package rule
|
||
|
||
import (
|
||
"fmt"
|
||
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// Exists 验证一个值在某个表中的字段中存在，支持同时判断多个字段
|
||
// Exists verify a value exists in a table field, support judging multiple fields at the same time
|
||
// 用法：exists:表名称,字段名称,字段名称,字段名称
|
||
// Usage: exists:table_name,field_name,field_name,field_name
|
||
// 例子：exists:users,phone,email
|
||
// Example: exists:users,phone,email
|
||
type Exists struct {
|
||
DB *gorm.DB
|
||
}
|
||
|
||
func NewExists(db *gorm.DB) *Exists {
|
||
return &Exists{DB: db}
|
||
}
|
||
|
||
func (r *Exists) Passes(val any, options ...any) bool {
|
||
if len(options) < 2 {
|
||
return false
|
||
}
|
||
|
||
tableName := options[0].(string)
|
||
fieldNames := options[1:]
|
||
|
||
query := r.DB.Table(tableName).Where(fmt.Sprintf("%s = ?", fieldNames[0]), val)
|
||
for _, fieldName := range fieldNames[1:] {
|
||
query = query.Or(fmt.Sprintf("%s = ?", fieldName), val)
|
||
}
|
||
|
||
var count int64
|
||
err := query.Count(&count).Error
|
||
if err != nil {
|
||
return false
|
||
}
|
||
|
||
return count != 0
|
||
}
|