亚洲欧美久久久久9999,狠狠色丁香婷婷综合久久来来去,亚洲天堂久久久

gRPC-go 連接管理

gRPC-go 連接管理

(金慶的專(zhuān)欄 2017.12)

把 example greeter 改一下，處理 SayHello() 請(qǐng)求時(shí)，不僅僅返回本次請(qǐng)求者的名字，
還返回上次請(qǐng)求的名字，如：
```
λ go run greeter_client/main.go
2017/12/25 17:59:13 Greeting: Hello 'world' (prev '')
2017/12/25 17:59:15 Greeting: Hello 'world2' (prev 'world')
```

先將客戶端單次請(qǐng)求改為多次請(qǐng)求：

```go
   r, err := c.SayHello(context.Background(), &pb.HelloRequest{Name: name})
   log.Printf("Greeting: %s", r.Message)
   time.Sleep(2 * time.Second)

   r, err = c.SayHello(context.Background(), &pb.HelloRequest{Name: name + "2"})
   log.Printf("Greeting: %s", r.Message)
   ...
```

服務(wù)器需要為每個(gè)連接保存各自的數(shù)據(jù)。連接創(chuàng)建時(shí)初始化數(shù)據(jù)，連接斷開(kāi)時(shí)清理數(shù)據(jù)。
這里利用了連接統(tǒng)計(jì)的接口，不知道是否是最適當(dāng)?shù)膶?shí)現(xiàn)方式?

服務(wù)器創(chuàng)建時(shí)添加 StatsHandler 選項(xiàng)，輸入一個(gè) stats.Handler 的實(shí)現(xiàn)。
```
-   s := grpc.NewServer()
+   s := grpc.NewServer(grpc.StatsHandler(&statshandler{}))
```

statshandler 需實(shí)現(xiàn)4個(gè)方法，只用到2個(gè)連接相關(guān)的方法，TagConn() 和 HandleConn(),
另外2個(gè) TagRPC() 和 HandleRPC() 用于RPC統(tǒng)計(jì), 實(shí)現(xiàn)為空。

```go
type statshandler struct{}

// TagConn 用來(lái)給連接打個(gè)標(biāo)簽，以此來(lái)標(biāo)識(shí)連接(實(shí)在是找不出還有什么辦法來(lái)標(biāo)識(shí)連接).
// 這個(gè)標(biāo)簽是個(gè)指針，可保證每個(gè)連接唯一。
// 將該指針添加到上下文中去，鍵為 connCtxKey{}.
func (h *statshandler) TagConn(ctx context.Context, info *stats.ConnTagInfo) context.Context {
   return context.WithValue(ctx, connCtxKey{}, info)
}

// TagRPC 為空.
func (h *statshandler) TagRPC(ctx context.Context, info *stats.RPCTagInfo) context.Context {
   return ctx
}

// HandleConn 會(huì)在連接開(kāi)始和結(jié)束時(shí)被調(diào)用，分別會(huì)輸入不同的狀態(tài).
func (h *statshandler) HandleConn(ctx context.Context, s stats.ConnStats) {
   tag, ok := getConnTagFromContext(ctx)
   if !ok {
       log.Fatal("can not get conn tag")
   }

   connsMutex.Lock()
   defer connsMutex.Unlock()

   switch s.(type) {
   case *stats.ConnBegin:
       conns[tag] = ""
       log.Printf("begin conn, tag = (%p)%#v, now connections = %d\n", tag, tag, len(conns))
   case *stats.ConnEnd:
       delete(conns, tag)
       log.Printf("end conn, tag = (%p)%#v, now connections = %d\n", tag, tag, len(conns))
   default:
       log.Printf("illegal ConnStats type\n")
   }
}

// HandleRPC 為空.
func (h *statshandler) HandleRPC(ctx context.Context, s stats.RPCStats) {
}
```

用一個(gè)map來(lái)管理所有連接，以連接的標(biāo)簽(是個(gè)指針)為鍵，值為上次請(qǐng)求者的名字。
因?yàn)橛卸嗑€程訪問(wèn)，所有加個(gè) Mutex 來(lái)保護(hù)。
連接結(jié)束時(shí)，將從 conns 中刪除連接相關(guān)的數(shù)據(jù)。

```go
var connsMutex sync.Mutex
var conns map[*stats.ConnTagInfo]string = make(map[*stats.ConnTagInfo]string)
```

getConnTagFromContext() 從上下文中取連接標(biāo)簽：

```go
type connCtxKey struct{}

func getConnTagFromContext(ctx context.Context) (*stats.ConnTagInfo, bool) {
   tag, ok := ctx.Value(connCtxKey{}).(*stats.ConnTagInfo)
   return tag, ok
}
```

最后將 SayHello() 改為記錄請(qǐng)求者名字，并返回上次請(qǐng)求者的名字。

```go
func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) {
   tag, _ := getConnTagFromContext(ctx)
   log.Printf("SayHello(), conn tag = (%p)%#v\n", tag, tag)

   connsMutex.Lock()
   defer connsMutex.Unlock()
   prev := conns[tag]
   conns[tag] = in.Name

   return &pb.HelloReply{Message: fmt.Sprintf("Hello '%s' (prev '%s')", in.Name, prev)}, nil
}
```

測(cè)試多個(gè)客戶端連接，可以看到每個(gè)客戶端有自己的狀態(tài)，互不影響。

```
E:\Git\grpc-go\examples\helloworld (master)
λ go run greeter_server/main.go
2017/12/25 18:39:03 start
2017/12/25 18:39:11 begin conn, tag = (0xc042182040)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0420818f0), LocalAddr:(*net.TCPAddr)(0xc0420818c0)}, now connections = 1
2017/12/25 18:39:11 SayHello(), conn tag = (0xc042182040)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0420818f0), LocalAddr:(*net.TCPAddr)(0xc0420818c0)}
2017/12/25 18:39:13 SayHello(), conn tag = (0xc042182040)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0420818f0), LocalAddr:(*net.TCPAddr)(0xc0420818c0)}
2017/12/25 18:39:13 begin conn, tag = (0xc0421ae200)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0421de060), LocalAddr:(*net.TCPAddr)(0xc0421de030)}, now connections = 2
2017/12/25 18:39:13 SayHello(), conn tag = (0xc0421ae200)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0421de060), LocalAddr:(*net.TCPAddr)(0xc0421de030)}
2017/12/25 18:39:15 SayHello(), conn tag = (0xc042182040)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0420818f0), LocalAddr:(*net.TCPAddr)(0xc0420818c0)}
2017/12/25 18:39:15 SayHello(), conn tag = (0xc0421ae200)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0421de060), LocalAddr:(*net.TCPAddr)(0xc0421de030)}
2017/12/25 18:39:17 SayHello(), conn tag = (0xc042182040)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0420818f0), LocalAddr:(*net.TCPAddr)(0xc0420818c0)}
2017/12/25 18:39:17 SayHello(), conn tag = (0xc0421ae200)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0421de060), LocalAddr:(*net.TCPAddr)(0xc0421de030)}
2017/12/25 18:39:19 end conn, tag = (0xc042182040)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0420818f0), LocalAddr:(*net.TCPAddr)(0xc0420818c0)}, now connections = 1
2017/12/25 18:39:19 SayHello(), conn tag = (0xc0421ae200)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0421de060), LocalAddr:(*net.TCPAddr)(0xc0421de030)}
2017/12/25 18:39:21 end conn, tag = (0xc0421ae200)&stats.ConnTagInfo{RemoteAddr:(*net.TCPAddr)(0xc0421de060), LocalAddr:(*net.TCPAddr)(0xc0421de030)}, now connections = 0
```

posted on 2017-12-25 19:00 金慶閱讀(1282) 評(píng)論(0) 編輯收藏引用所屬分類(lèi): 9. 其它

只有注冊(cè)用戶登錄后才能發(fā)表評(píng)論。
【推薦】100%開(kāi)源！大型工業(yè)跨平臺(tái)軟件C++源碼提供，建模，組態(tài)！

相關(guān)文章: TortoiseGit is OK but GitExtensions fails DeathVoteExpirationTimeout in Orleans How to delete local branches of GitExtension Clustering provider in Orleans Why Orleans' actor is virutal What comes after microservice? Rust Deref coercion example Rust Error Return Check Policy Rust visibility Why does Rust check borrow even in single thread

網(wǎng)站導(dǎo)航: 博客園 IT新聞 BlogJava 博問(wèn) Chat2DB 管理

金慶的專(zhuān)欄

公告

留言簿(12)

隨筆分類(lèi)(502)

隨筆檔案(423)

相冊(cè)

搜索

積分與排名

最新評(píng)論

閱讀排行榜

評(píng)論排行榜