zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

add more tests (#1114)

Browse Source
This commit is contained in:
Kevin Wan
2021-10-04 20:02:25 +08:00
committed by GitHub
parent 9f5bfa0088
commit 3cdfcb05f1
2 changed files with 119 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

36
zrpc/internal/clientinterceptors/tracinginterceptor.go
View File

@@ -100,6 +100,24 @@ type (
}
)
func (w *clientStream) CloseSend() error {
err := w.ClientStream.CloseSend()
if err != nil {
w.sendStreamEvent(errorEvent, err)
}
return err
}
func (w *clientStream) Header() (metadata.MD, error) {
md, err := w.ClientStream.Header()
if err != nil {
w.sendStreamEvent(errorEvent, err)
}
return md, err
}
func (w *clientStream) RecvMsg(m interface{}) error {
err := w.ClientStream.RecvMsg(m)
if err == nil && !w.desc.ServerStreams {
@@ -127,24 +145,6 @@ func (w *clientStream) SendMsg(m interface{}) error {
return err
}
func (w *clientStream) Header() (metadata.MD, error) {
md, err := w.ClientStream.Header()
if err != nil {
w.sendStreamEvent(errorEvent, err)
}
return md, err
}
func (w *clientStream) CloseSend() error {
err := w.ClientStream.CloseSend()
if err != nil {
w.sendStreamEvent(errorEvent, err)
}
return err
}
func (w *clientStream) sendStreamEvent(eventType streamEventType, err error) {
select {
case <-w.eventsDone: