2015-12-17 22:37:35 +02:00
|
|
|
package server
|
|
|
|
|
|
|
|
import (
|
2018-03-03 13:53:52 +02:00
|
|
|
"context"
|
2019-08-15 16:22:53 +02:00
|
|
|
"errors"
|
|
|
|
"io"
|
2015-12-17 22:37:35 +02:00
|
|
|
"sync"
|
2019-01-08 17:38:25 +02:00
|
|
|
|
2021-10-12 13:55:53 +02:00
|
|
|
"go-micro.dev/v4/codec"
|
2015-12-17 22:37:35 +02:00
|
|
|
)
|
|
|
|
|
2022-09-30 16:27:07 +02:00
|
|
|
// Implements the Streamer interface.
|
2015-12-17 22:37:35 +02:00
|
|
|
type rpcStream struct {
|
|
|
|
sync.RWMutex
|
2019-01-08 17:38:25 +02:00
|
|
|
id string
|
2015-12-17 22:37:35 +02:00
|
|
|
closed bool
|
|
|
|
err error
|
|
|
|
request Request
|
2019-01-08 17:38:25 +02:00
|
|
|
codec codec.Codec
|
2015-12-17 22:37:35 +02:00
|
|
|
context context.Context
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *rpcStream) Context() context.Context {
|
|
|
|
return r.context
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *rpcStream) Request() Request {
|
|
|
|
return r.request
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *rpcStream) Send(msg interface{}) error {
|
|
|
|
r.Lock()
|
|
|
|
defer r.Unlock()
|
|
|
|
|
2019-01-08 17:38:25 +02:00
|
|
|
resp := codec.Message{
|
2019-01-18 12:12:57 +02:00
|
|
|
Target: r.request.Service(),
|
|
|
|
Method: r.request.Method(),
|
2019-01-10 23:25:31 +02:00
|
|
|
Endpoint: r.request.Endpoint(),
|
|
|
|
Id: r.id,
|
|
|
|
Type: codec.Response,
|
2015-12-17 22:37:35 +02:00
|
|
|
}
|
|
|
|
|
2019-06-03 16:55:47 +02:00
|
|
|
if err := r.codec.Write(&resp, msg); err != nil {
|
|
|
|
r.err = err
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
2015-12-17 22:37:35 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func (r *rpcStream) Recv(msg interface{}) error {
|
2019-01-08 17:38:25 +02:00
|
|
|
req := new(codec.Message)
|
|
|
|
req.Type = codec.Request
|
2015-12-17 22:37:35 +02:00
|
|
|
|
2020-01-12 11:13:14 +02:00
|
|
|
err := r.codec.ReadHeader(req, req.Type)
|
|
|
|
r.Lock()
|
|
|
|
defer r.Unlock()
|
|
|
|
if err != nil {
|
2015-12-17 22:37:35 +02:00
|
|
|
// discard body
|
2019-01-07 20:20:47 +02:00
|
|
|
r.codec.ReadBody(nil)
|
2019-06-03 16:55:47 +02:00
|
|
|
r.err = err
|
2015-12-17 22:37:35 +02:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2019-08-15 16:22:53 +02:00
|
|
|
// check the error
|
|
|
|
if len(req.Error) > 0 {
|
|
|
|
// Check the client closed the stream
|
|
|
|
switch req.Error {
|
2022-09-29 16:44:53 +02:00
|
|
|
case errLastStreamResponse.Error():
|
2019-08-15 16:22:53 +02:00
|
|
|
// discard body
|
2020-01-12 11:13:14 +02:00
|
|
|
r.Unlock()
|
2019-08-15 16:22:53 +02:00
|
|
|
r.codec.ReadBody(nil)
|
2020-01-12 11:13:14 +02:00
|
|
|
r.Lock()
|
2019-08-15 16:22:53 +02:00
|
|
|
r.err = io.EOF
|
|
|
|
return io.EOF
|
|
|
|
default:
|
|
|
|
return errors.New(req.Error)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2016-04-06 18:53:16 +02:00
|
|
|
// we need to stay up to date with sequence numbers
|
2019-01-08 17:38:25 +02:00
|
|
|
r.id = req.Id
|
2020-01-12 11:13:14 +02:00
|
|
|
r.Unlock()
|
|
|
|
err = r.codec.ReadBody(msg)
|
|
|
|
r.Lock()
|
|
|
|
if err != nil {
|
2019-06-03 16:55:47 +02:00
|
|
|
r.err = err
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
2015-12-17 22:37:35 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
func (r *rpcStream) Error() error {
|
|
|
|
r.RLock()
|
|
|
|
defer r.RUnlock()
|
|
|
|
return r.err
|
|
|
|
}
|
|
|
|
|
|
|
|
func (r *rpcStream) Close() error {
|
|
|
|
r.Lock()
|
|
|
|
defer r.Unlock()
|
|
|
|
r.closed = true
|
|
|
|
return r.codec.Close()
|
|
|
|
}
|