Jelajahi Sumber

Merge pull request #2968 from yichengq/fix-stream-reader-init

rafthttp: always init streamReader before return from newPeer
Yicheng Qin 10 tahun lalu
induk
melakukan
219d304291
1 mengubah file dengan 2 tambahan dan 2 penghapusan
  1. 2 2
      rafthttp/peer.go

+ 2 - 2
rafthttp/peer.go

@@ -143,10 +143,10 @@ func startPeer(tr http.RoundTripper, urls types.URLs, local, to, cid types.ID, r
 		}
 		}
 	}()
 	}()
 
 
+	p.msgAppReader = startStreamReader(tr, picker, streamTypeMsgAppV2, local, to, cid, status, p.recvc, p.propc, errorc)
+	reader := startStreamReader(tr, picker, streamTypeMessage, local, to, cid, status, p.recvc, p.propc, errorc)
 	go func() {
 	go func() {
 		var paused bool
 		var paused bool
-		p.msgAppReader = startStreamReader(tr, picker, streamTypeMsgAppV2, local, to, cid, status, p.recvc, p.propc, errorc)
-		reader := startStreamReader(tr, picker, streamTypeMessage, local, to, cid, status, p.recvc, p.propc, errorc)
 		for {
 		for {
 			select {
 			select {
 			case m := <-p.sendc:
 			case m := <-p.sendc: