26c3f42095
* feat(binding): Add support for custom validation tags * docs: Add example for custom validation tag * test(binding): Add test for registering custom validation
83 lines
2.4 KiB
Go
83 lines
2.4 KiB
Go
// Copyright 2014 Manu Martinez-Almeida. All rights reserved.
|
|
// Use of this source code is governed by a MIT style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package binding
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
validator "gopkg.in/go-playground/validator.v8"
|
|
)
|
|
|
|
const (
|
|
MIMEJSON = "application/json"
|
|
MIMEHTML = "text/html"
|
|
MIMEXML = "application/xml"
|
|
MIMEXML2 = "text/xml"
|
|
MIMEPlain = "text/plain"
|
|
MIMEPOSTForm = "application/x-www-form-urlencoded"
|
|
MIMEMultipartPOSTForm = "multipart/form-data"
|
|
MIMEPROTOBUF = "application/x-protobuf"
|
|
MIMEMSGPACK = "application/x-msgpack"
|
|
MIMEMSGPACK2 = "application/msgpack"
|
|
)
|
|
|
|
type Binding interface {
|
|
Name() string
|
|
Bind(*http.Request, interface{}) error
|
|
}
|
|
|
|
type StructValidator interface {
|
|
// ValidateStruct can receive any kind of type and it should never panic, even if the configuration is not right.
|
|
// If the received type is not a struct, any validation should be skipped and nil must be returned.
|
|
// If the received type is a struct or pointer to a struct, the validation should be performed.
|
|
// If the struct is not valid or the validation itself fails, a descriptive error should be returned.
|
|
// Otherwise nil must be returned.
|
|
ValidateStruct(interface{}) error
|
|
|
|
// RegisterValidation adds a validation Func to a Validate's map of validators denoted by the key
|
|
// NOTE: if the key already exists, the previous validation function will be replaced.
|
|
// NOTE: this method is not thread-safe it is intended that these all be registered prior to any validation
|
|
RegisterValidation(string, validator.Func) error
|
|
}
|
|
|
|
var Validator StructValidator = &defaultValidator{}
|
|
|
|
var (
|
|
JSON = jsonBinding{}
|
|
XML = xmlBinding{}
|
|
Form = formBinding{}
|
|
Query = queryBinding{}
|
|
FormPost = formPostBinding{}
|
|
FormMultipart = formMultipartBinding{}
|
|
ProtoBuf = protobufBinding{}
|
|
MsgPack = msgpackBinding{}
|
|
)
|
|
|
|
func Default(method, contentType string) Binding {
|
|
if method == "GET" {
|
|
return Form
|
|
}
|
|
|
|
switch contentType {
|
|
case MIMEJSON:
|
|
return JSON
|
|
case MIMEXML, MIMEXML2:
|
|
return XML
|
|
case MIMEPROTOBUF:
|
|
return ProtoBuf
|
|
case MIMEMSGPACK, MIMEMSGPACK2:
|
|
return MsgPack
|
|
default: //case MIMEPOSTForm, MIMEMultipartPOSTForm:
|
|
return Form
|
|
}
|
|
}
|
|
|
|
func validate(obj interface{}) error {
|
|
if Validator == nil {
|
|
return nil
|
|
}
|
|
return Validator.ValidateStruct(obj)
|
|
}
|