Skip to content

Commit

Permalink
fix imports for the interceptor, update send and recieved msg
Browse files Browse the repository at this point in the history
  • Loading branch information
akoserwal committed Aug 7, 2024
1 parent ac1b5c5 commit 932ed31
Showing 1 changed file with 16 additions and 12 deletions.
28 changes: 16 additions & 12 deletions transport/grpc/interceptor.go
Original file line number Diff line number Diff line change
Expand Up @@ -2,12 +2,13 @@ package grpc

import (
"context"
"github.com/go-kratos/kratos/v2/internal/matcher"
"fmt"

"google.golang.org/grpc"
grpcmd "google.golang.org/grpc/metadata"

ic "github.com/go-kratos/kratos/v2/internal/context"
"github.com/go-kratos/kratos/v2/internal/matcher"
"github.com/go-kratos/kratos/v2/middleware"
"github.com/go-kratos/kratos/v2/transport"
)
Expand Down Expand Up @@ -110,34 +111,37 @@ func GetStream(ctx context.Context) grpc.ServerStream {
}

func (w *wrappedStream) SendMsg(m interface{}) error {
err := w.ServerStream.SendMsg(m)
info, _ := transport.FromServerContext(w.ctx)

h := func(ctx context.Context, req interface{}) (interface{}, error) {
return req, nil
return req, w.ServerStream.SendMsg(m)
}

info, ok := transport.FromServerContext(w.ctx)
if !ok {
return fmt.Errorf("transport value stored in ctx returns: %v", ok)
}

if next := w.middleware.Match(info.Operation()); len(next) > 0 {
h = middleware.Chain(next...)(h)
}

_, err = h(w.ctx, m)

_, err := h(w.ctx, m)
return err
}

func (w *wrappedStream) RecvMsg(m interface{}) error {
err := w.ServerStream.RecvMsg(m)
info, _ := transport.FromServerContext(w.ctx)

h := func(ctx context.Context, req interface{}) (interface{}, error) {
return req, nil
return req, w.ServerStream.RecvMsg(m)
}

info, ok := transport.FromServerContext(w.ctx)
if !ok {
return fmt.Errorf("transport value stored in ctx returns: %v", ok)
}

if next := w.middleware.Match(info.Operation()); len(next) > 0 {
h = middleware.Chain(next...)(h)
}

_, err = h(w.ctx, m)
_, err := h(w.ctx, m)
return err
}

0 comments on commit 932ed31

Please sign in to comment.