上一篇寫了一下rpc呼叫過程的實現方式，簡單來說就是服務端把實現了介面的結構體物件進行反射，抽取方法，簽名，儲存，客戶端呼叫的時候go-micro封請求資料，服務端接收到請求時，找到需要呼叫呼叫的物件和對應的方法，利用反射進行呼叫，返回資料。 但是沒有說stream的實現方式，感覺單獨寫一篇帖子來說這個更好一些。上一篇帖子是基礎，理解了上一篇，stream實現原理一點即破。先說一下使用方式，再說原理。 當前go-micro對 rpc 呼叫的方式大概如下： 普通的rpc呼叫 是這樣：
1.連線伺服器或者從快取池得到連線
2.客戶端 ->傳送資料 -> 服務端接收
3.服務端 ->返回資料 -> 客戶端處理資料
4.關閉連線或者把連線返回到快取池
複製程式碼當前 rps stream的實現方式 是這樣子:
1. 連線伺服器
2. 客戶端多次傳送請求-> 服務端接收
3. 服務端多次返回資料-> 客戶端處理資料
4. 關閉連線
複製程式碼當資料量比較大的時候我們可以用stream方式分批次傳輸資料。對於客戶端還是服務端沒有限制，我們可以根據自己的需要使用stream方式，使用方式也非常的簡單，在定義介面的時候在引數或者返回值前面加上stream然後就可以多次進行傳輸了，使用的程式碼還是之前寫的例子,程式碼都在github上： 比如我的例子中定義了兩個使用stream的介面，一個只在返回值使用stream，另一個是在引數和返回值前都加上了stream，最終的使用方式沒有區別
rpc Stream(model.SRequest) returns (stream model.SResponse) {}
rpc BidirectionalStream(stream model.SRequest) returns (stream model.SResponse) {}
複製程式碼看一下go-micro為我們生成的程式碼rpcapi.micro.go裡，不要被嚇到，生成了很多程式碼，但是沒啥理解不了的 Server端
// Server API for Say service
type SayHandler interface {
// .... others
Stream(context.Context, *model.SRequest, Say_StreamStream) error
BidirectionalStream(context.Context, Say_BidirectionalStreamStream) error
}
type Say_StreamStream interface {
SendMsg(interface{}) error
RecvMsg(interface{}) error
Close() error
Send(*model.SResponse) error
}
type Say_BidirectionalStreamStream interface {
SendMsg(interface{}) error
RecvMsg(interface{}) error
Close() error
Send(*model.SResponse) error
Recv() (*model.SRequest, error)
}
// .... others
複製程式碼Client端
// Client API for Say service
type SayService interface {
//... others
Stream(ctx context.Context, in *model.SRequest, opts ...client.CallOption) (Say_StreamService, error)
BidirectionalStream(ctx context.Context, opts ...client.CallOption) (Say_BidirectionalStreamService, error)
}
type Say_StreamService interface {
SendMsg(interface{}) error
RecvMsg(interface{}) error
Close() error
Recv() (*model.SResponse, error)
}
type Say_BidirectionalStreamService interface {
SendMsg(interface{}) error
RecvMsg(interface{}) error
Close() error
Send(*model.SRequest) error
Recv() (*model.SResponse, error)
}
複製程式碼你會發現引數前面加了 Stream後，生成的程式碼會把你的引數變成一個介面，這個介面主要要的方法是
SendMsg(interface{}) error
RecvMsg(interface{}) error
Close() error
複製程式碼剩下的兩個介面方法是根據你是傳送還是接收生成的，如果有傳送就會有Send(你的引數)，如果有接收會生成Rev() (你的引數， error),但這兩個方法只是為了讓你使用時方便，裡面呼叫的還是SendMsg(interface)和RecvMsg(interface)方法，但是他們是怎麼工作的，如何多次傳送和接收傳輸的資料，是不是感覺很神奇。
我就以TsBidirectionalStream 方法為例開始分析，上一篇和再早之前的帖子已經說了服務端啟動的時候都做了哪些操作，這裡就不再贅述，
服務端的實現,很簡單，不斷的獲取客戶端發過來的資料，再給客戶端一次一次的返回一些資料。
/*
模擬資料
*/
func (s *Say) BidirectionalStream(ctx context.Context, stream rpcapi.Say_BidirectionalStreamStream) error {
for {
req, err := stream.Recv()
if err == io.EOF {
break
}
if err != nil {
return err
}
for i := int64(0); i < req.Count; i++ {
if err := stream.Send(&model.SResponse{Value: []string {lib.RandomStr(lib.Random(3, 6))}}); err != nil {
return err
}
}
}
return nil
}
複製程式碼啟動服務，服務開始監聽客戶端傳過來的資料..... 客戶端呼叫服務端方法：
// 呼叫
func TsBidirectionalStream(client rpcapi.SayService) {
rspStream, err := client.BidirectionalStream(context.Background())
if err != nil {
panic(err)
}
// send
go func() {
rspStream.Send(&model.SRequest{Count: 2})
rspStream.Send(&model.SRequest{Count: 5})
// close the stream
if err := rspStream.Close(); err != nil {
fmt.Println("stream close err:", err)
}
}()
// recv
idx := 1
for {
rsp, err := rspStream.Recv()
if err == io.EOF {
break
} else if err != nil {
panic(err)
}
fmt.Printf("test stream get idx %d data %v\n", idx, rsp)
idx++
}
fmt.Println("Read Value End")
}
複製程式碼當客戶端在呼叫rpc的stream方法是要很得到stream
rspStream, err := client.BidirectionalStream(context.Background())
//
func (c *sayService) BidirectionalStream(ctx context.Context, opts ...client.CallOption) (Say_BidirectionalStreamService, error) {
req := c.c.NewRequest(c.name, "Say.BidirectionalStream", &model.SRequest{})
stream, err := c.c.Stream(ctx, req, opts...)
if err != nil {
return nil, err
}
return &sayServiceBidirectionalStream{stream}, nil
}
複製程式碼這個呼叫c.c.Stream(ctx, req, opts...)是關鍵，他的內部實現就是和伺服器進行連線，然後返回一個stream，進行操作。
客戶端：和服務端建立連線，返回Stream，進行接收和傳送資料
服務端：接收客戶端連線請求，利用反射找到相應的方法，組織Strem，傳給方法，進行資料的傳送和接收
複製程式碼建立連線的時候就是一次rpc呼叫，服務端接受連線，然後客戶端傳送一次呼叫，但是傳輸的是空資料，服務端利用反射找到具體的方法，組織stream，呼叫具體方法，利用這個連線，客戶端和服務端進行多次通訊。