Commit b6f4e985 authored by libangzhu's avatar libangzhu Committed by vipwzw

用google.status.Code提换google.grpc.Code

parent 4c0de86d
...@@ -15,6 +15,7 @@ import ( ...@@ -15,6 +15,7 @@ import (
"golang.org/x/net/context" "golang.org/x/net/context"
"google.golang.org/grpc" "google.golang.org/grpc"
"google.golang.org/grpc/codes" "google.golang.org/grpc/codes"
"google.golang.org/grpc/status"
) )
// Start peer start // Start peer start
...@@ -265,7 +266,7 @@ func (p *Peer) sendStream() { ...@@ -265,7 +266,7 @@ func (p *Peer) sendStream() {
P2pComm.CollectPeerStat(err, p) P2pComm.CollectPeerStat(err, p)
if err != nil { if err != nil {
log.Error("sendStream", "send", err) log.Error("sendStream", "send", err)
if grpc.Code(err) == codes.Unimplemented { //maybe order peers delete peer to BlackList if status.Code(err) == codes.Unimplemented { //maybe order peers delete peer to BlackList
p.node.nodeInfo.blacklist.Add(p.Addr(), 3600) p.node.nodeInfo.blacklist.Add(p.Addr(), 3600)
} }
time.Sleep(time.Second) //have a rest time.Sleep(time.Second) //have a rest
...@@ -335,13 +336,13 @@ func (p *Peer) readStream() { ...@@ -335,13 +336,13 @@ func (p *Peer) readStream() {
log.Error("CloseSend", "err", errs) log.Error("CloseSend", "err", errs)
} }
if grpc.Code(err) == codes.Unavailable { if status.Code(err) == codes.Unavailable {
break //重新创建新的流 break //重新创建新的流
} }
log.Error("readStream", "recv,err:", err.Error(), "peerIp", p.Addr()) log.Error("readStream", "recv,err:", err.Error(), "peerIp", p.Addr())
if grpc.Code(err) == codes.Unimplemented { //maybe order peers delete peer to BlackList if status.Code(err) == codes.Unimplemented { //maybe order peers delete peer to BlackList
p.node.nodeInfo.blacklist.Add(p.Addr(), 3600) p.node.nodeInfo.blacklist.Add(p.Addr(), 3600)
return return
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment