Написание кастомного правила для go-playground/validator

DrType — Thu, 04 Jan 2024 11:14:54 GMT

Юзаю пакет github.com/go-playground/validator/v10 для валидации запросов в Fiber (в пет-прожекте).
Захотелось иметь правило валидации наподобие ларавелевского exists. Реализовал следующим образом:

Go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
package validators
 
import (
    "my-project/db"
    "fmt"
    "reflect"
    "strings"
 
    "github.com/go-playground/validator/v10"
)
 
func Exists(fl validator.FieldLevel) bool {
    params := strings.Split(fl.Param(), ".")
    table := params[0]
    column := "id"
    if len(params) > 1 {
        column = params[1]
    }
 
    var value any
    switch valueType := fl.Field().Kind(); valueType {
    case reflect.String:
        value = fl.Field().String()
    case reflect.Int:
        value = fl.Field().Int()
    }
 
    var result int
    db.Database.Raw(
        fmt.Sprintf("SELECT COUNT(*) FROM %s WHERE %s = ?", table, column),
        value,
    ).Scan(&result)
 
    return result > 0
}

Регистрирую новое правило:

Go
1
2
validatorInstance := validator.New()
validatorInstance.RegisterValidation("exists", validators.Exists)

Возможные примеры использования:

Go
1
2
3
4
5
6
7
type UpdateEntityRequest struct {
    Id  int  `json:"id" validate:"omitempty,exists=entities"` //в таблице entities должна найтись запись с таким же id
}
 
type SomeRequest struct {
    Name  string  `json:"name" validate:"exists=users.name"` //в таблице users должна найтись запись с таким же name
}

По поводу реализации. Для транзакций с БД использую Gorm, и db.Database в 29 строчке имеет тип *gorm.DB. Возможно есть более канонический способ динамического построения запроса на основе имени таблицы и имени столбца; но в любом случае об инъекциях речи все равно не идет, так как эти два параметра задаются разработчиком (который при этом легко может поломать запрос, что я как-то раз и сделал, добавив лишний символ ; в struct tag)... Также можно было бы добавить обработку ошибки запроса, но вроде не критично, так как в случае неудачи паники все равно не возникает, а функция просто возвращает false.

Это я к тому, что напиание кастомных правил валидации - плодотворная затея, может упростить жизнь. И при написании таких функций имхо полезно ориентироваться на реализации встроенных (baked in) правил в исходном коде пакета.

Форум программистов и сисадминов Киберфорум - Блоги - DrType

Написание кастомного правила для go-playground/validator