chore: release v0.4.1 (#621)

cloudwego · Aug 29, 2022 · 995948d · 995948d
1 parent 4a7219a
commit 995948d
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 3 deletions.
diff --git a/pkg/remote/codec/default_codec.go b/pkg/remote/codec/default_codec.go
@@ -176,9 +176,14 @@ func (c *defaultCodec) Decode(ctx context.Context, message remote.Message, in re
 	}
 
 	// 2. decode body
+	hasRead := in.ReadLen()
 	if err = c.decodePayload(ctx, message, in); err != nil {
 		return err
 	}
+	if message.PayloadLen() == 0 {
+		// if protocol is PurePayload, should set payload length after decoded
+		message.SetPayloadLen(in.ReadLen() - hasRead)
+	}
 	return nil
 }
 

diff --git a/pkg/remote/codec/protobuf/grpc.go b/pkg/remote/codec/protobuf/grpc.go
@@ -98,6 +98,7 @@ func (c *grpcCodec) Decode(ctx context.Context, message remote.Message, in remot
 	if err != nil {
 		return err
 	}
+	message.SetPayloadLen(dLen)
 	data := message.Data()
 	switch t := data.(type) {
 	case fastpb.Reader:

diff --git a/pkg/remote/trans/detection/server_handler.go b/pkg/remote/trans/detection/server_handler.go
@@ -112,8 +112,7 @@ func (t *svrTransHandler) OnRead(ctx context.Context, conn net.Conn) error {
 
 func (t *svrTransHandler) OnInactive(ctx context.Context, conn net.Conn) {
 	// t.http2 should use the ctx returned by OnActive in r.ctx
-	r := ctx.Value(handlerKey{}).(*handlerWrapper)
-	if r.ctx != nil {
+	if r, ok := ctx.Value(handlerKey{}).(*handlerWrapper); ok && r.ctx != nil {
 		ctx = r.ctx
 	}
 	t.which(ctx).OnInactive(ctx, conn)

diff --git a/version.go b/version.go
@@ -19,5 +19,5 @@ package kitex
 // Name and Version info of this framework, used for statistics and debug
 const (
 	Name    = "Kitex"
-	Version = "v0.4.0"
+	Version = "v0.4.1"
 )