Tony Chen
4 years ago
committed by
GitHub
parent
a500fe2f0d
commit
496edc6fb1
@ -0,0 +1,11 @@ |
||||
// Code generated by protoc-gen-go-errors. DO NOT EDIT.
|
||||
|
||||
package testproto |
||||
|
||||
import ( |
||||
errors "github.com/go-kratos/kratos/v2/errors" |
||||
) |
||||
|
||||
// This is a compile-time assertion to ensure that this generated file
|
||||
// is compatible with the kratos package it is being compiled against.
|
||||
const _ = errors.SupportPackageIsVersion1 |
||||
@ -0,0 +1,91 @@ |
||||
// Code generated by protoc-gen-go. DO NOT EDIT.
|
||||
// versions:
|
||||
// protoc-gen-go v1.25.0
|
||||
// protoc v3.13.0
|
||||
// source: stream.proto
|
||||
|
||||
package testproto |
||||
|
||||
import ( |
||||
proto "github.com/golang/protobuf/proto" |
||||
empty "github.com/golang/protobuf/ptypes/empty" |
||||
_ "google.golang.org/genproto/googleapis/api/annotations" |
||||
httpbody "google.golang.org/genproto/googleapis/api/httpbody" |
||||
protoreflect "google.golang.org/protobuf/reflect/protoreflect" |
||||
protoimpl "google.golang.org/protobuf/runtime/protoimpl" |
||||
reflect "reflect" |
||||
) |
||||
|
||||
const ( |
||||
// Verify that this generated code is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion) |
||||
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
||||
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20) |
||||
) |
||||
|
||||
// This is a compile-time assertion that a sufficiently up-to-date version
|
||||
// of the legacy proto package is being used.
|
||||
const _ = proto.ProtoPackageIsVersion4 |
||||
|
||||
var File_stream_proto protoreflect.FileDescriptor |
||||
|
||||
var file_stream_proto_rawDesc = []byte{ |
||||
0x0a, 0x0c, 0x73, 0x74, 0x72, 0x65, 0x61, 0x6d, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x09, |
||||
0x74, 0x65, 0x73, 0x74, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x1a, 0x1c, 0x67, 0x6f, 0x6f, 0x67, 0x6c, |
||||
0x65, 0x2f, 0x61, 0x70, 0x69, 0x2f, 0x61, 0x6e, 0x6e, 0x6f, 0x74, 0x61, 0x74, 0x69, 0x6f, 0x6e, |
||||
0x73, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x1a, 0x19, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, |
||||
0x61, 0x70, 0x69, 0x2f, 0x68, 0x74, 0x74, 0x70, 0x62, 0x6f, 0x64, 0x79, 0x2e, 0x70, 0x72, 0x6f, |
||||
0x74, 0x6f, 0x1a, 0x1b, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2f, 0x70, 0x72, 0x6f, 0x74, 0x6f, |
||||
0x62, 0x75, 0x66, 0x2f, 0x65, 0x6d, 0x70, 0x74, 0x79, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x32, |
||||
0x69, 0x0a, 0x0d, 0x53, 0x74, 0x72, 0x65, 0x61, 0x6d, 0x53, 0x65, 0x72, 0x76, 0x69, 0x63, 0x65, |
||||
0x12, 0x58, 0x0a, 0x08, 0x44, 0x6f, 0x77, 0x6e, 0x6c, 0x6f, 0x61, 0x64, 0x12, 0x16, 0x2e, 0x67, |
||||
0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x75, 0x66, 0x2e, 0x45, |
||||
0x6d, 0x70, 0x74, 0x79, 0x1a, 0x14, 0x2e, 0x67, 0x6f, 0x6f, 0x67, 0x6c, 0x65, 0x2e, 0x61, 0x70, |
||||
0x69, 0x2e, 0x48, 0x74, 0x74, 0x70, 0x42, 0x6f, 0x64, 0x79, 0x22, 0x1c, 0x82, 0xd3, 0xe4, 0x93, |
||||
0x02, 0x16, 0x12, 0x14, 0x2f, 0x76, 0x31, 0x2f, 0x65, 0x78, 0x61, 0x6d, 0x70, 0x6c, 0x65, 0x2f, |
||||
0x64, 0x6f, 0x77, 0x6e, 0x6c, 0x6f, 0x61, 0x64, 0x30, 0x01, 0x42, 0x51, 0x5a, 0x4f, 0x67, 0x69, |
||||
0x74, 0x68, 0x75, 0x62, 0x2e, 0x63, 0x6f, 0x6d, 0x2f, 0x67, 0x6f, 0x2d, 0x6b, 0x72, 0x61, 0x74, |
||||
0x6f, 0x73, 0x2f, 0x6b, 0x72, 0x61, 0x74, 0x6f, 0x73, 0x2f, 0x63, 0x6d, 0x64, 0x2f, 0x70, 0x72, |
||||
0x6f, 0x74, 0x6f, 0x63, 0x2d, 0x67, 0x65, 0x6e, 0x2d, 0x67, 0x6f, 0x2d, 0x68, 0x74, 0x74, 0x70, |
||||
0x2f, 0x69, 0x6e, 0x74, 0x65, 0x72, 0x6e, 0x61, 0x6c, 0x2f, 0x74, 0x65, 0x73, 0x74, 0x70, 0x72, |
||||
0x6f, 0x74, 0x6f, 0x3b, 0x74, 0x65, 0x73, 0x74, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x62, 0x06, 0x70, |
||||
0x72, 0x6f, 0x74, 0x6f, 0x33, |
||||
} |
||||
|
||||
var file_stream_proto_goTypes = []interface{}{ |
||||
(*empty.Empty)(nil), // 0: google.protobuf.Empty
|
||||
(*httpbody.HttpBody)(nil), // 1: google.api.HttpBody
|
||||
} |
||||
var file_stream_proto_depIdxs = []int32{ |
||||
0, // 0: testproto.StreamService.Download:input_type -> google.protobuf.Empty
|
||||
1, // 1: testproto.StreamService.Download:output_type -> google.api.HttpBody
|
||||
1, // [1:2] is the sub-list for method output_type
|
||||
0, // [0:1] is the sub-list for method input_type
|
||||
0, // [0:0] is the sub-list for extension type_name
|
||||
0, // [0:0] is the sub-list for extension extendee
|
||||
0, // [0:0] is the sub-list for field type_name
|
||||
} |
||||
|
||||
func init() { file_stream_proto_init() } |
||||
func file_stream_proto_init() { |
||||
if File_stream_proto != nil { |
||||
return |
||||
} |
||||
type x struct{} |
||||
out := protoimpl.TypeBuilder{ |
||||
File: protoimpl.DescBuilder{ |
||||
GoPackagePath: reflect.TypeOf(x{}).PkgPath(), |
||||
RawDescriptor: file_stream_proto_rawDesc, |
||||
NumEnums: 0, |
||||
NumMessages: 0, |
||||
NumExtensions: 0, |
||||
NumServices: 1, |
||||
}, |
||||
GoTypes: file_stream_proto_goTypes, |
||||
DependencyIndexes: file_stream_proto_depIdxs, |
||||
}.Build() |
||||
File_stream_proto = out.File |
||||
file_stream_proto_rawDesc = nil |
||||
file_stream_proto_goTypes = nil |
||||
file_stream_proto_depIdxs = nil |
||||
} |
||||
@ -0,0 +1,15 @@ |
||||
syntax = "proto3"; |
||||
option go_package = "github.com/go-kratos/kratos/cmd/protoc-gen-go-http/internal/testproto;testproto"; |
||||
package testproto; |
||||
|
||||
import "google/api/annotations.proto"; |
||||
import "google/api/httpbody.proto"; |
||||
import "google/protobuf/empty.proto"; |
||||
|
||||
service StreamService { |
||||
rpc Download(google.protobuf.Empty) returns (stream google.api.HttpBody) { |
||||
option (google.api.http) = { |
||||
get : "/v1/example/download" |
||||
}; |
||||
} |
||||
} |
||||
@ -0,0 +1,130 @@ |
||||
// Code generated by protoc-gen-go-grpc. DO NOT EDIT.
|
||||
|
||||
package testproto |
||||
|
||||
import ( |
||||
context "context" |
||||
empty "github.com/golang/protobuf/ptypes/empty" |
||||
httpbody "google.golang.org/genproto/googleapis/api/httpbody" |
||||
grpc "google.golang.org/grpc" |
||||
codes "google.golang.org/grpc/codes" |
||||
status "google.golang.org/grpc/status" |
||||
) |
||||
|
||||
// This is a compile-time assertion to ensure that this generated file
|
||||
// is compatible with the grpc package it is being compiled against.
|
||||
// Requires gRPC-Go v1.32.0 or later.
|
||||
const _ = grpc.SupportPackageIsVersion7 |
||||
|
||||
// StreamServiceClient is the client API for StreamService service.
|
||||
//
|
||||
// For semantics around ctx use and closing/ending streaming RPCs, please refer to https://pkg.go.dev/google.golang.org/grpc/?tab=doc#ClientConn.NewStream.
|
||||
type StreamServiceClient interface { |
||||
Download(ctx context.Context, in *empty.Empty, opts ...grpc.CallOption) (StreamService_DownloadClient, error) |
||||
} |
||||
|
||||
type streamServiceClient struct { |
||||
cc grpc.ClientConnInterface |
||||
} |
||||
|
||||
func NewStreamServiceClient(cc grpc.ClientConnInterface) StreamServiceClient { |
||||
return &streamServiceClient{cc} |
||||
} |
||||
|
||||
func (c *streamServiceClient) Download(ctx context.Context, in *empty.Empty, opts ...grpc.CallOption) (StreamService_DownloadClient, error) { |
||||
stream, err := c.cc.NewStream(ctx, &StreamService_ServiceDesc.Streams[0], "/testproto.StreamService/Download", opts...) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
x := &streamServiceDownloadClient{stream} |
||||
if err := x.ClientStream.SendMsg(in); err != nil { |
||||
return nil, err |
||||
} |
||||
if err := x.ClientStream.CloseSend(); err != nil { |
||||
return nil, err |
||||
} |
||||
return x, nil |
||||
} |
||||
|
||||
type StreamService_DownloadClient interface { |
||||
Recv() (*httpbody.HttpBody, error) |
||||
grpc.ClientStream |
||||
} |
||||
|
||||
type streamServiceDownloadClient struct { |
||||
grpc.ClientStream |
||||
} |
||||
|
||||
func (x *streamServiceDownloadClient) Recv() (*httpbody.HttpBody, error) { |
||||
m := new(httpbody.HttpBody) |
||||
if err := x.ClientStream.RecvMsg(m); err != nil { |
||||
return nil, err |
||||
} |
||||
return m, nil |
||||
} |
||||
|
||||
// StreamServiceServer is the server API for StreamService service.
|
||||
// All implementations must embed UnimplementedStreamServiceServer
|
||||
// for forward compatibility
|
||||
type StreamServiceServer interface { |
||||
Download(*empty.Empty, StreamService_DownloadServer) error |
||||
mustEmbedUnimplementedStreamServiceServer() |
||||
} |
||||
|
||||
// UnimplementedStreamServiceServer must be embedded to have forward compatible implementations.
|
||||
type UnimplementedStreamServiceServer struct { |
||||
} |
||||
|
||||
func (UnimplementedStreamServiceServer) Download(*empty.Empty, StreamService_DownloadServer) error { |
||||
return status.Errorf(codes.Unimplemented, "method Download not implemented") |
||||
} |
||||
func (UnimplementedStreamServiceServer) mustEmbedUnimplementedStreamServiceServer() {} |
||||
|
||||
// UnsafeStreamServiceServer may be embedded to opt out of forward compatibility for this service.
|
||||
// Use of this interface is not recommended, as added methods to StreamServiceServer will
|
||||
// result in compilation errors.
|
||||
type UnsafeStreamServiceServer interface { |
||||
mustEmbedUnimplementedStreamServiceServer() |
||||
} |
||||
|
||||
func RegisterStreamServiceServer(s grpc.ServiceRegistrar, srv StreamServiceServer) { |
||||
s.RegisterService(&StreamService_ServiceDesc, srv) |
||||
} |
||||
|
||||
func _StreamService_Download_Handler(srv interface{}, stream grpc.ServerStream) error { |
||||
m := new(empty.Empty) |
||||
if err := stream.RecvMsg(m); err != nil { |
||||
return err |
||||
} |
||||
return srv.(StreamServiceServer).Download(m, &streamServiceDownloadServer{stream}) |
||||
} |
||||
|
||||
type StreamService_DownloadServer interface { |
||||
Send(*httpbody.HttpBody) error |
||||
grpc.ServerStream |
||||
} |
||||
|
||||
type streamServiceDownloadServer struct { |
||||
grpc.ServerStream |
||||
} |
||||
|
||||
func (x *streamServiceDownloadServer) Send(m *httpbody.HttpBody) error { |
||||
return x.ServerStream.SendMsg(m) |
||||
} |
||||
|
||||
// StreamService_ServiceDesc is the grpc.ServiceDesc for StreamService service.
|
||||
// It's only intended for direct use with grpc.RegisterService,
|
||||
// and not to be introspected or modified (even as a copy)
|
||||
var StreamService_ServiceDesc = grpc.ServiceDesc{ |
||||
ServiceName: "testproto.StreamService", |
||||
HandlerType: (*StreamServiceServer)(nil), |
||||
Methods: []grpc.MethodDesc{}, |
||||
Streams: []grpc.StreamDesc{ |
||||
{ |
||||
StreamName: "Download", |
||||
Handler: _StreamService_Download_Handler, |
||||
ServerStreams: true, |
||||
}, |
||||
}, |
||||
Metadata: "stream.proto", |
||||
} |
||||
@ -0,0 +1,33 @@ |
||||
// Code generated by protoc-gen-go-http. DO NOT EDIT.
|
||||
|
||||
package testproto |
||||
|
||||
import ( |
||||
context "context" |
||||
http1 "github.com/go-kratos/kratos/v2/transport/http" |
||||
binding "github.com/go-kratos/kratos/v2/transport/http/binding" |
||||
mux "github.com/gorilla/mux" |
||||
http "net/http" |
||||
) |
||||
|
||||
// This is a compile-time assertion to ensure that this generated file
|
||||
// is compatible with the kratos package it is being compiled against.
|
||||
var _ = new(http.Request) |
||||
var _ = new(context.Context) |
||||
var _ = binding.MapProto |
||||
var _ = mux.NewRouter |
||||
|
||||
const _ = http1.SupportPackageIsVersion1 |
||||
|
||||
type StreamServiceHandler interface { |
||||
} |
||||
|
||||
func NewStreamServiceHandler(srv StreamServiceHandler, opts ...http1.HandleOption) http.Handler { |
||||
h := http1.DefaultHandleOptions() |
||||
for _, o := range opts { |
||||
o(&h) |
||||
} |
||||
r := mux.NewRouter() |
||||
|
||||
return r |
||||
} |
||||
@ -0,0 +1,18 @@ |
||||
package binding |
||||
|
||||
import ( |
||||
"net/http" |
||||
|
||||
"google.golang.org/protobuf/proto" |
||||
) |
||||
|
||||
// BindForm bind form parameters to target.
|
||||
func BindForm(req *http.Request, target interface{}) error { |
||||
if err := req.ParseForm(); err != nil { |
||||
return err |
||||
} |
||||
if msg, ok := target.(proto.Message); ok { |
||||
return mapProto(msg, req.Form) |
||||
} |
||||
return mapForm(target, req.Form) |
||||
} |
||||
@ -0,0 +1,385 @@ |
||||
package binding |
||||
|
||||
// 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.
|
||||
|
||||
import ( |
||||
"encoding/json" |
||||
"errors" |
||||
"fmt" |
||||
"reflect" |
||||
"strconv" |
||||
"strings" |
||||
"time" |
||||
) |
||||
|
||||
var ( |
||||
errUnknownType = errors.New("unknown type") |
||||
emptyField = reflect.StructField{} |
||||
) |
||||
|
||||
func mapForm(ptr interface{}, form map[string][]string) error { |
||||
return mapFormByTag(ptr, form, "json") |
||||
} |
||||
|
||||
func mapFormByTag(ptr interface{}, form map[string][]string, tag string) error { |
||||
ptrVal := reflect.ValueOf(ptr) |
||||
var pointed interface{} |
||||
if ptrVal.Kind() == reflect.Ptr { |
||||
ptrVal = ptrVal.Elem() |
||||
pointed = ptrVal.Interface() |
||||
} |
||||
if ptrVal.Kind() == reflect.Map && |
||||
ptrVal.Type().Key().Kind() == reflect.String { |
||||
if pointed != nil { |
||||
ptr = pointed |
||||
} |
||||
return setFormMap(ptr, form) |
||||
} |
||||
return mappingByPtr(ptr, formSource(form), tag) |
||||
} |
||||
|
||||
// setter tries to set value on a walking by fields of a struct
|
||||
type setter interface { |
||||
TrySet(value reflect.Value, field reflect.StructField, key string, opt setOptions) (isSetted bool, err error) |
||||
} |
||||
|
||||
type formSource map[string][]string |
||||
|
||||
var _ setter = formSource(nil) |
||||
|
||||
// TrySet tries to set a value by request's form source (like map[string][]string)
|
||||
func (form formSource) TrySet(value reflect.Value, field reflect.StructField, tagValue string, opt setOptions) (isSetted bool, err error) { |
||||
return setByForm(value, field, form, tagValue, opt) |
||||
} |
||||
|
||||
func mappingByPtr(ptr interface{}, setter setter, tag string) error { |
||||
_, err := mapping(reflect.ValueOf(ptr), emptyField, setter, tag) |
||||
return err |
||||
} |
||||
|
||||
func mapping(value reflect.Value, field reflect.StructField, setter setter, tag string) (bool, error) { |
||||
if field.Tag.Get(tag) == "-" { // just ignoring this field
|
||||
return false, nil |
||||
} |
||||
|
||||
var vKind = value.Kind() |
||||
|
||||
if vKind == reflect.Ptr { |
||||
var isNew bool |
||||
vPtr := value |
||||
if value.IsNil() { |
||||
isNew = true |
||||
vPtr = reflect.New(value.Type().Elem()) |
||||
} |
||||
isSetted, err := mapping(vPtr.Elem(), field, setter, tag) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
if isNew && isSetted { |
||||
value.Set(vPtr) |
||||
} |
||||
return isSetted, nil |
||||
} |
||||
|
||||
if vKind != reflect.Struct || !field.Anonymous { |
||||
ok, err := tryToSetValue(value, field, setter, tag) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
if ok { |
||||
return true, nil |
||||
} |
||||
} |
||||
|
||||
if vKind == reflect.Struct { |
||||
tValue := value.Type() |
||||
|
||||
var isSetted bool |
||||
for i := 0; i < value.NumField(); i++ { |
||||
sf := tValue.Field(i) |
||||
if sf.PkgPath != "" && !sf.Anonymous { // unexported
|
||||
continue |
||||
} |
||||
ok, err := mapping(value.Field(i), tValue.Field(i), setter, tag) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
isSetted = isSetted || ok |
||||
} |
||||
return isSetted, nil |
||||
} |
||||
return false, nil |
||||
} |
||||
|
||||
type setOptions struct { |
||||
isDefaultExists bool |
||||
defaultValue string |
||||
} |
||||
|
||||
func tryToSetValue(value reflect.Value, field reflect.StructField, setter setter, tag string) (bool, error) { |
||||
var tagValue string |
||||
var setOpt setOptions |
||||
|
||||
tagValue = field.Tag.Get(tag) |
||||
tagValue, opts := head(tagValue, ",") |
||||
|
||||
if tagValue == "" { // default value is FieldName
|
||||
tagValue = field.Name |
||||
} |
||||
if tagValue == "" { // when field is "emptyField" variable
|
||||
return false, nil |
||||
} |
||||
|
||||
var opt string |
||||
for len(opts) > 0 { |
||||
opt, opts = head(opts, ",") |
||||
|
||||
if k, v := head(opt, "="); k == "default" { |
||||
setOpt.isDefaultExists = true |
||||
setOpt.defaultValue = v |
||||
} |
||||
} |
||||
|
||||
return setter.TrySet(value, field, tagValue, setOpt) |
||||
} |
||||
|
||||
func setByForm(value reflect.Value, field reflect.StructField, form map[string][]string, tagValue string, opt setOptions) (isSetted bool, err error) { |
||||
vs, ok := form[tagValue] |
||||
if !ok && !opt.isDefaultExists { |
||||
return false, nil |
||||
} |
||||
|
||||
switch value.Kind() { |
||||
case reflect.Slice: |
||||
if !ok { |
||||
vs = []string{opt.defaultValue} |
||||
} |
||||
return true, setSlice(vs, value, field) |
||||
case reflect.Array: |
||||
if !ok { |
||||
vs = []string{opt.defaultValue} |
||||
} |
||||
if len(vs) != value.Len() { |
||||
return false, fmt.Errorf("%q is not valid value for %s", vs, value.Type().String()) |
||||
} |
||||
return true, setArray(vs, value, field) |
||||
default: |
||||
var val string |
||||
if !ok { |
||||
val = opt.defaultValue |
||||
} |
||||
|
||||
if len(vs) > 0 { |
||||
val = vs[0] |
||||
} |
||||
return true, setWithProperType(val, value, field) |
||||
} |
||||
} |
||||
|
||||
func setWithProperType(val string, value reflect.Value, field reflect.StructField) error { |
||||
switch value.Kind() { |
||||
case reflect.Int: |
||||
return setIntField(val, 0, value) |
||||
case reflect.Int8: |
||||
return setIntField(val, 8, value) |
||||
case reflect.Int16: |
||||
return setIntField(val, 16, value) |
||||
case reflect.Int32: |
||||
return setIntField(val, 32, value) |
||||
case reflect.Int64: |
||||
switch value.Interface().(type) { |
||||
case time.Duration: |
||||
return setTimeDuration(val, value, field) |
||||
} |
||||
return setIntField(val, 64, value) |
||||
case reflect.Uint: |
||||
return setUintField(val, 0, value) |
||||
case reflect.Uint8: |
||||
return setUintField(val, 8, value) |
||||
case reflect.Uint16: |
||||
return setUintField(val, 16, value) |
||||
case reflect.Uint32: |
||||
return setUintField(val, 32, value) |
||||
case reflect.Uint64: |
||||
return setUintField(val, 64, value) |
||||
case reflect.Bool: |
||||
return setBoolField(val, value) |
||||
case reflect.Float32: |
||||
return setFloatField(val, 32, value) |
||||
case reflect.Float64: |
||||
return setFloatField(val, 64, value) |
||||
case reflect.String: |
||||
value.SetString(val) |
||||
case reflect.Struct: |
||||
switch value.Interface().(type) { |
||||
case time.Time: |
||||
return setTimeField(val, field, value) |
||||
} |
||||
return json.Unmarshal([]byte(val), value.Addr().Interface()) |
||||
case reflect.Map: |
||||
return json.Unmarshal([]byte(val), value.Addr().Interface()) |
||||
default: |
||||
return errUnknownType |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func setIntField(val string, bitSize int, field reflect.Value) error { |
||||
if val == "" { |
||||
val = "0" |
||||
} |
||||
intVal, err := strconv.ParseInt(val, 10, bitSize) |
||||
if err == nil { |
||||
field.SetInt(intVal) |
||||
} |
||||
return err |
||||
} |
||||
|
||||
func setUintField(val string, bitSize int, field reflect.Value) error { |
||||
if val == "" { |
||||
val = "0" |
||||
} |
||||
uintVal, err := strconv.ParseUint(val, 10, bitSize) |
||||
if err == nil { |
||||
field.SetUint(uintVal) |
||||
} |
||||
return err |
||||
} |
||||
|
||||
func setBoolField(val string, field reflect.Value) error { |
||||
if val == "" { |
||||
val = "false" |
||||
} |
||||
boolVal, err := strconv.ParseBool(val) |
||||
if err == nil { |
||||
field.SetBool(boolVal) |
||||
} |
||||
return err |
||||
} |
||||
|
||||
func setFloatField(val string, bitSize int, field reflect.Value) error { |
||||
if val == "" { |
||||
val = "0.0" |
||||
} |
||||
floatVal, err := strconv.ParseFloat(val, bitSize) |
||||
if err == nil { |
||||
field.SetFloat(floatVal) |
||||
} |
||||
return err |
||||
} |
||||
|
||||
func setTimeField(val string, structField reflect.StructField, value reflect.Value) error { |
||||
timeFormat := structField.Tag.Get("time_format") |
||||
if timeFormat == "" { |
||||
timeFormat = time.RFC3339 |
||||
} |
||||
|
||||
switch tf := strings.ToLower(timeFormat); tf { |
||||
case "unix", "unixnano": |
||||
tv, err := strconv.ParseInt(val, 10, 64) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
|
||||
d := time.Duration(1) |
||||
if tf == "unixnano" { |
||||
d = time.Second |
||||
} |
||||
|
||||
t := time.Unix(tv/int64(d), tv%int64(d)) |
||||
value.Set(reflect.ValueOf(t)) |
||||
return nil |
||||
|
||||
} |
||||
|
||||
if val == "" { |
||||
value.Set(reflect.ValueOf(time.Time{})) |
||||
return nil |
||||
} |
||||
|
||||
l := time.Local |
||||
if isUTC, _ := strconv.ParseBool(structField.Tag.Get("time_utc")); isUTC { |
||||
l = time.UTC |
||||
} |
||||
|
||||
if locTag := structField.Tag.Get("time_location"); locTag != "" { |
||||
loc, err := time.LoadLocation(locTag) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
l = loc |
||||
} |
||||
|
||||
t, err := time.ParseInLocation(timeFormat, val, l) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
|
||||
value.Set(reflect.ValueOf(t)) |
||||
return nil |
||||
} |
||||
|
||||
func setArray(vals []string, value reflect.Value, field reflect.StructField) error { |
||||
for i, s := range vals { |
||||
err := setWithProperType(s, value.Index(i), field) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
func setSlice(vals []string, value reflect.Value, field reflect.StructField) error { |
||||
slice := reflect.MakeSlice(value.Type(), len(vals), len(vals)) |
||||
err := setArray(vals, slice, field) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
value.Set(slice) |
||||
return nil |
||||
} |
||||
|
||||
func setTimeDuration(val string, value reflect.Value, field reflect.StructField) error { |
||||
d, err := time.ParseDuration(val) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
value.Set(reflect.ValueOf(d)) |
||||
return nil |
||||
} |
||||
|
||||
func head(str, sep string) (head string, tail string) { |
||||
idx := strings.Index(str, sep) |
||||
if idx < 0 { |
||||
return str, "" |
||||
} |
||||
return str[:idx], str[idx+len(sep):] |
||||
} |
||||
|
||||
func setFormMap(ptr interface{}, form map[string][]string) error { |
||||
el := reflect.TypeOf(ptr).Elem() |
||||
|
||||
if el.Kind() == reflect.Slice { |
||||
ptrMap, ok := ptr.(map[string][]string) |
||||
if !ok { |
||||
return errors.New("cannot convert to map slices of strings") |
||||
} |
||||
for k, v := range form { |
||||
ptrMap[k] = v |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
ptrMap, ok := ptr.(map[string]string) |
||||
if !ok { |
||||
return errors.New("cannot convert to map of strings") |
||||
} |
||||
for k, v := range form { |
||||
ptrMap[k] = v[len(v)-1] // pick last
|
||||
} |
||||
|
||||
return nil |
||||
} |
||||
@ -1,81 +0,0 @@ |
||||
package http |
||||
|
||||
import ( |
||||
"fmt" |
||||
"io/ioutil" |
||||
"net/http" |
||||
"strings" |
||||
|
||||
"github.com/go-kratos/kratos/v2/encoding" |
||||
) |
||||
|
||||
const baseContentType = "application" |
||||
|
||||
func contentType(subtype string) string { |
||||
return strings.Join([]string{baseContentType, subtype}, "/") |
||||
} |
||||
|
||||
func contentSubtype(contentType string) string { |
||||
if contentType == baseContentType { |
||||
return "" |
||||
} |
||||
if !strings.HasPrefix(contentType, baseContentType) { |
||||
return "" |
||||
} |
||||
// guaranteed since != baseContentType and has baseContentType prefix
|
||||
switch contentType[len(baseContentType)] { |
||||
case '/', ';': |
||||
if i := strings.Index(contentType, ";"); i != -1 { |
||||
return contentType[len(baseContentType)+1 : i] |
||||
} |
||||
return contentType[len(baseContentType)+1:] |
||||
default: |
||||
return "" |
||||
} |
||||
} |
||||
|
||||
func defaultRequestDecoder(req *http.Request, v interface{}) error { |
||||
data, err := ioutil.ReadAll(req.Body) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
defer req.Body.Close() |
||||
subtype := contentSubtype(req.Header.Get("content-type")) |
||||
codec := encoding.GetCodec(subtype) |
||||
if codec == nil { |
||||
return fmt.Errorf("decoding request failed unknown content-type: %s", subtype) |
||||
} |
||||
return codec.Unmarshal(data, v) |
||||
} |
||||
|
||||
func defaultResponseEncoder(res http.ResponseWriter, req *http.Request, v interface{}) error { |
||||
subtype := contentSubtype(req.Header.Get("accept")) |
||||
codec := encoding.GetCodec(subtype) |
||||
if codec == nil { |
||||
codec = encoding.GetCodec("json") |
||||
} |
||||
data, err := codec.Marshal(v) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
res.Header().Set("content-type", contentType(codec.Name())) |
||||
res.Write(data) |
||||
return nil |
||||
} |
||||
|
||||
func defaultErrorEncoder(res http.ResponseWriter, req *http.Request, err error) { |
||||
se, code := StatusError(err) |
||||
subtype := contentSubtype(req.Header.Get("accept")) |
||||
codec := encoding.GetCodec(subtype) |
||||
if codec == nil { |
||||
codec = encoding.GetCodec("json") |
||||
} |
||||
data, err := codec.Marshal(se) |
||||
if err != nil { |
||||
res.WriteHeader(http.StatusInternalServerError) |
||||
return |
||||
} |
||||
res.Header().Set("content-type", contentType(codec.Name())) |
||||
res.WriteHeader(code) |
||||
res.Write(data) |
||||
} |
||||
@ -1,23 +0,0 @@ |
||||
package http |
||||
|
||||
import "testing" |
||||
|
||||
func TestSubtype(t *testing.T) { |
||||
tests := []struct { |
||||
input string |
||||
expected string |
||||
}{ |
||||
{"application/json", "json"}, |
||||
{"application/json;", "json"}, |
||||
{"application/json; charset=utf-8", "json"}, |
||||
{"application/", ""}, |
||||
{"application", ""}, |
||||
{"foo", ""}, |
||||
{"", ""}, |
||||
} |
||||
for _, test := range tests { |
||||
if contentSubtype(test.input) != test.expected { |
||||
t.Errorf("expected %s got %s", test.expected, test.input) |
||||
} |
||||
} |
||||
} |
||||
@ -1,59 +0,0 @@ |
||||
package http |
||||
|
||||
import ( |
||||
"net/http" |
||||
|
||||
"github.com/go-kratos/kratos/v2/errors" |
||||
) |
||||
|
||||
var ( |
||||
// References: https://github.com/googleapis/googleapis/blob/master/google/rpc/code.proto
|
||||
codesMapping = map[int32]int{ |
||||
0: http.StatusOK, |
||||
1: http.StatusInternalServerError, |
||||
2: http.StatusInternalServerError, |
||||
3: http.StatusBadRequest, |
||||
4: http.StatusRequestTimeout, |
||||
5: http.StatusNotFound, |
||||
6: http.StatusConflict, |
||||
7: http.StatusForbidden, |
||||
8: http.StatusTooManyRequests, |
||||
9: http.StatusPreconditionFailed, |
||||
10: http.StatusConflict, |
||||
11: http.StatusBadRequest, |
||||
12: http.StatusNotImplemented, |
||||
13: http.StatusInternalServerError, |
||||
14: http.StatusServiceUnavailable, |
||||
15: http.StatusInternalServerError, |
||||
16: http.StatusUnauthorized, |
||||
} |
||||
statusMapping = map[int]int32{ |
||||
http.StatusOK: 0, |
||||
http.StatusBadRequest: 3, |
||||
http.StatusRequestTimeout: 4, |
||||
http.StatusNotFound: 5, |
||||
http.StatusConflict: 6, |
||||
http.StatusForbidden: 7, |
||||
http.StatusUnauthorized: 16, |
||||
http.StatusPreconditionFailed: 9, |
||||
http.StatusNotImplemented: 12, |
||||
http.StatusInternalServerError: 13, |
||||
http.StatusServiceUnavailable: 14, |
||||
} |
||||
) |
||||
|
||||
// StatusError converts error to status error.
|
||||
func StatusError(err error) (*errors.StatusError, int) { |
||||
se, ok := errors.FromError(err) |
||||
if !ok { |
||||
se = &errors.StatusError{ |
||||
Code: 2, |
||||
Reason: "Unknown", |
||||
Message: "Unknown: " + err.Error(), |
||||
} |
||||
} |
||||
if status, ok := codesMapping[se.Code]; ok { |
||||
return se, status |
||||
} |
||||
return se, http.StatusInternalServerError |
||||
} |
||||
@ -0,0 +1,147 @@ |
||||
package http |
||||
|
||||
import ( |
||||
"encoding/json" |
||||
"io/ioutil" |
||||
"net/http" |
||||
"strings" |
||||
|
||||
"github.com/go-kratos/kratos/v2/encoding" |
||||
"github.com/go-kratos/kratos/v2/errors" |
||||
"github.com/go-kratos/kratos/v2/middleware" |
||||
"github.com/go-kratos/kratos/v2/transport/http/binding" |
||||
) |
||||
|
||||
const ( |
||||
// SupportPackageIsVersion1 These constants should not be referenced from any other code.
|
||||
SupportPackageIsVersion1 = true |
||||
|
||||
baseContentType = "application" |
||||
) |
||||
|
||||
var ( |
||||
acceptHeader = http.CanonicalHeaderKey("Accept") |
||||
contentTypeHeader = http.CanonicalHeaderKey("Content-Type") |
||||
) |
||||
|
||||
// DecodeRequestFunc is decode request func.
|
||||
type DecodeRequestFunc func(*http.Request, interface{}) error |
||||
|
||||
// EncodeResponseFunc is encode response func.
|
||||
type EncodeResponseFunc func(http.ResponseWriter, *http.Request, interface{}) error |
||||
|
||||
// EncodeErrorFunc is encode error func.
|
||||
type EncodeErrorFunc func(http.ResponseWriter, *http.Request, error) |
||||
|
||||
// HandleOption is handle option.
|
||||
type HandleOption func(*HandleOptions) |
||||
|
||||
// HandleOptions is handle options.
|
||||
type HandleOptions struct { |
||||
Decode DecodeRequestFunc |
||||
Encode EncodeResponseFunc |
||||
Error EncodeErrorFunc |
||||
Middleware middleware.Middleware |
||||
} |
||||
|
||||
// DefaultHandleOptions returns a default handle options.
|
||||
func DefaultHandleOptions() HandleOptions { |
||||
return HandleOptions{ |
||||
Decode: decodeRequest, |
||||
Encode: encodeResponse, |
||||
Error: encodeError, |
||||
} |
||||
} |
||||
|
||||
// RequestDecoder with request decoder.
|
||||
func RequestDecoder(dec DecodeRequestFunc) HandleOption { |
||||
return func(o *HandleOptions) { |
||||
o.Decode = dec |
||||
} |
||||
} |
||||
|
||||
// ResponseEncoder with response encoder.
|
||||
func ResponseEncoder(en EncodeResponseFunc) HandleOption { |
||||
return func(o *HandleOptions) { |
||||
o.Encode = en |
||||
} |
||||
} |
||||
|
||||
// ErrorEncoder with error encoder.
|
||||
func ErrorEncoder(en EncodeErrorFunc) HandleOption { |
||||
return func(o *HandleOptions) { |
||||
o.Error = en |
||||
} |
||||
} |
||||
|
||||
// Middleware with middleware option.
|
||||
func Middleware(m middleware.Middleware) HandleOption { |
||||
return func(o *HandleOptions) { |
||||
o.Middleware = m |
||||
} |
||||
} |
||||
|
||||
// decodeRequest decodes the request body to object.
|
||||
func decodeRequest(req *http.Request, v interface{}) error { |
||||
subtype := contentSubtype(req.Header.Get(contentTypeHeader)) |
||||
if codec := encoding.GetCodec(subtype); codec != nil { |
||||
data, err := ioutil.ReadAll(req.Body) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
return codec.Unmarshal(data, v) |
||||
} |
||||
return binding.BindForm(req, v) |
||||
} |
||||
|
||||
// encodeResponse encodes the object to the HTTP response.
|
||||
func encodeResponse(w http.ResponseWriter, r *http.Request, v interface{}) error { |
||||
for _, accept := range r.Header[acceptHeader] { |
||||
if codec := encoding.GetCodec(contentSubtype(accept)); codec != nil { |
||||
data, err := codec.Marshal(v) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
w.Header().Set(contentTypeHeader, contentType(codec.Name())) |
||||
w.Write(data) |
||||
return nil |
||||
} |
||||
} |
||||
return json.NewEncoder(w).Encode(v) |
||||
} |
||||
|
||||
// encodeError encodes the erorr to the HTTP response.
|
||||
func encodeError(w http.ResponseWriter, r *http.Request, err error) { |
||||
se, ok := errors.FromError(err) |
||||
if !ok { |
||||
se = &errors.StatusError{ |
||||
Code: 2, |
||||
Reason: "", |
||||
Message: err.Error(), |
||||
} |
||||
} |
||||
w.WriteHeader(se.HTTPStatus()) |
||||
encodeResponse(w, r, se) |
||||
} |
||||
|
||||
func contentType(subtype string) string { |
||||
return strings.Join([]string{baseContentType, subtype}, "/") |
||||
} |
||||
|
||||
func contentSubtype(contentType string) string { |
||||
if contentType == baseContentType { |
||||
return "" |
||||
} |
||||
if !strings.HasPrefix(contentType, baseContentType) { |
||||
return "" |
||||
} |
||||
switch contentType[len(baseContentType)] { |
||||
case '/', ';': |
||||
if i := strings.Index(contentType, ";"); i != -1 { |
||||
return contentType[len(baseContentType)+1 : i] |
||||
} |
||||
return contentType[len(baseContentType)+1:] |
||||
default: |
||||
return "" |
||||
} |
||||
} |
||||
@ -0,0 +1,57 @@ |
||||
package http |
||||
|
||||
import ( |
||||
"context" |
||||
"net/http" |
||||
"testing" |
||||
|
||||
"github.com/gorilla/mux" |
||||
) |
||||
|
||||
type HelloRequest struct { |
||||
Name string `json:"name"` |
||||
} |
||||
type HelloReply struct { |
||||
Message string `json:"message"` |
||||
} |
||||
type GreeterService struct { |
||||
} |
||||
|
||||
func (s *GreeterService) SayHello(ctx context.Context, req *HelloRequest) (*HelloReply, error) { |
||||
return &HelloReply{Message: "hello " + req.Name}, nil |
||||
} |
||||
|
||||
func newGreeterHandler(srv *GreeterService, opts ...HandleOption) http.Handler { |
||||
h := DefaultHandleOptions() |
||||
for _, o := range opts { |
||||
o(&h) |
||||
} |
||||
r := mux.NewRouter() |
||||
r.HandleFunc("/helloworld", func(w http.ResponseWriter, r *http.Request) { |
||||
var in HelloRequest |
||||
if err := h.Decode(r, &in); err != nil { |
||||
h.Error(w, r, err) |
||||
return |
||||
} |
||||
next := func(ctx context.Context, req interface{}) (interface{}, error) { |
||||
return srv.SayHello(ctx, &in) |
||||
} |
||||
if h.Middleware != nil { |
||||
next = h.Middleware(next) |
||||
} |
||||
out, err := next(r.Context(), &in) |
||||
if err != nil { |
||||
h.Error(w, r, err) |
||||
return |
||||
} |
||||
if err := h.Encode(w, r, out); err != nil { |
||||
h.Error(w, r, err) |
||||
} |
||||
}).Methods("POST") |
||||
return r |
||||
} |
||||
|
||||
func TestHandler(t *testing.T) { |
||||
s := &GreeterService{} |
||||
_ = newGreeterHandler(s) |
||||
} |
||||
@ -1,49 +0,0 @@ |
||||
package http |
||||
|
||||
import ( |
||||
"context" |
||||
"net/http" |
||||
) |
||||
|
||||
// SupportPackageIsVersion1 These constants should not be referenced from any other code.
|
||||
const SupportPackageIsVersion1 = true |
||||
|
||||
type methodHandler func(srv interface{}, ctx context.Context, req *http.Request, dec func(interface{}) error) (out interface{}, err error) |
||||
|
||||
// MethodDesc represents a Proto service's method specification.
|
||||
type MethodDesc struct { |
||||
Path string |
||||
Method string |
||||
Handler methodHandler |
||||
} |
||||
|
||||
// ServiceDesc represents a Proto service's specification.
|
||||
type ServiceDesc struct { |
||||
ServiceName string |
||||
Methods []MethodDesc |
||||
Metadata interface{} |
||||
} |
||||
|
||||
// ServiceRegistrar wraps a single method that supports service registration.
|
||||
type ServiceRegistrar interface { |
||||
RegisterService(desc *ServiceDesc, impl interface{}) |
||||
} |
||||
|
||||
// RegisterService .
|
||||
func (s *Server) RegisterService(desc *ServiceDesc, impl interface{}) { |
||||
for _, m := range desc.Methods { |
||||
h := m.Handler |
||||
s.router.HandleFunc(m.Path, func(res http.ResponseWriter, req *http.Request) { |
||||
out, err := h(impl, req.Context(), req, func(v interface{}) error { |
||||
return s.requestDecoder(req, v) |
||||
}) |
||||
if err != nil { |
||||
s.errorEncoder(res, req, err) |
||||
return |
||||
} |
||||
if err := s.responseEncoder(res, req, out); err != nil { |
||||
s.errorEncoder(res, req, err) |
||||
} |
||||
}).Methods(m.Method) |
||||
} |
||||
} |
||||
@ -1,93 +0,0 @@ |
||||
package http |
||||
|
||||
import ( |
||||
"bytes" |
||||
"context" |
||||
"encoding/json" |
||||
"fmt" |
||||
"net/http" |
||||
"testing" |
||||
"time" |
||||
|
||||
"github.com/go-kratos/kratos/v2/internal/host" |
||||
) |
||||
|
||||
type testRequest struct { |
||||
Name string `json:"name"` |
||||
} |
||||
type testReply struct { |
||||
Result string `json:"result"` |
||||
} |
||||
type testService struct{} |
||||
|
||||
func (s *testService) SayHello(ctx context.Context, req *testRequest) (*testReply, error) { |
||||
return &testReply{Result: req.Name}, nil |
||||
} |
||||
|
||||
func TestService(t *testing.T) { |
||||
h := func(srv interface{}, ctx context.Context, req *http.Request, dec func(interface{}) error) (interface{}, error) { |
||||
var in testRequest |
||||
if err := dec(&in); err != nil { |
||||
return nil, err |
||||
} |
||||
out, err := srv.(*testService).SayHello(ctx, &in) |
||||
if err != nil { |
||||
return nil, err |
||||
} |
||||
return out, nil |
||||
} |
||||
sd := &ServiceDesc{ |
||||
ServiceName: "helloworld.Greeter", |
||||
Methods: []MethodDesc{ |
||||
{ |
||||
Path: "/helloworld", |
||||
Method: "POST", |
||||
Handler: h, |
||||
}, |
||||
}, |
||||
} |
||||
|
||||
svc := &testService{} |
||||
srv := NewServer() |
||||
srv.RegisterService(sd, svc) |
||||
|
||||
time.AfterFunc(time.Second, func() { |
||||
defer srv.Stop() |
||||
testServiceClient(t, srv) |
||||
}) |
||||
|
||||
if err := srv.Start(); err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
} |
||||
|
||||
func testServiceClient(t *testing.T, srv *Server) { |
||||
client, err := NewClient(context.Background()) |
||||
if err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
port, ok := host.Port(srv.lis) |
||||
if !ok { |
||||
t.Fatalf("extract port error: %v", srv.lis) |
||||
} |
||||
var ( |
||||
in = testRequest{Name: "hello"} |
||||
out = testReply{} |
||||
url = fmt.Sprintf("http://127.0.0.1:%d/helloworld", port) |
||||
) |
||||
data, err := json.Marshal(in) |
||||
if err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
req, err := http.NewRequest("POST", url, bytes.NewReader(data)) |
||||
if err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
req.Header.Set("content-type", "application/json") |
||||
if err := Do(client, req, &out); err != nil { |
||||
t.Fatal(err) |
||||
} |
||||
if out.Result != in.Name { |
||||
t.Fatalf("expected %s got %s", in.Name, out.Result) |
||||
} |
||||
} |
||||
Loading…
Reference in new issue