gin/binding/protobuf.go

37 lines
864 B
Go
Raw Normal View History

2015-07-12 09:42:39 +00:00
// 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 (
"io/ioutil"
"net/http"
2017-06-12 02:40:15 +00:00
"github.com/golang/protobuf/proto"
2015-07-12 09:42:39 +00:00
)
type protobufBinding struct{}
2016-04-14 23:16:46 +00:00
func (protobufBinding) Name() string {
2015-07-12 09:42:39 +00:00
return "protobuf"
}
func (b protobufBinding) Bind(req *http.Request, obj interface{}) error {
2015-07-12 09:42:39 +00:00
buf, err := ioutil.ReadAll(req.Body)
if err != nil {
return err
}
return b.BindBody(buf, obj)
}
2015-07-12 09:42:39 +00:00
func (protobufBinding) BindBody(body []byte, obj interface{}) error {
if err := proto.Unmarshal(body, obj.(proto.Message)); err != nil {
2015-07-12 09:42:39 +00:00
return err
}
2018-11-05 06:17:04 +00:00
// Here it's same to return validate(obj), but util now we can't add
// `binding:""` to the struct which automatically generate by gen-proto
2015-07-12 09:42:39 +00:00
return nil
// return validate(obj)
2015-07-12 09:42:39 +00:00
}