Close socket immediately after server_shutdown error

bozaro · bozaro · commit 6efefcbe174b · 2021-04-19T13:59:45.000+03:00
diff --git a/socket.go b/socket.go
@@ -760,6 +760,16 @@ func (socket *mongoSocket) readLoop() {
 					replyFunc(nil, &reply, i, b)
 				}
 
+				var replyError queryError
+				if err := bson.Unmarshal(b, &replyError); err != nil {
+					debugf("Socket %p to %s: received broken document: %#v", socket, socket.addr, err)
+				}
+				switch replyError.Code {
+				case 11600:
+					socket.kill(errors.New("server_shutdown: "+replyError.ErrMsg), true)
+					return
+				}
+
 				// XXX Do bound checking against totalLen.
 			}
 		}

Original file line number	Diff line number	Diff line change
`@@ -760,6 +760,16 @@ func (socket *mongoSocket) readLoop() {`
`760`	`760`	`replyFunc(nil, &reply, i, b)`
`761`	`761`	`}`
`762`	`762`
	`763`	`+ var replyError queryError`
	`764`	`+ if err := bson.Unmarshal(b, &replyError); err != nil {`
	`765`	`+ debugf("Socket %p to %s: received broken document: %#v", socket, socket.addr, err)`
	`766`	`+ }`
	`767`	`+ switch replyError.Code {`
	`768`	`+ case 11600:`
	`769`	`+ socket.kill(errors.New("server_shutdown: "+replyError.ErrMsg), true)`
	`770`	`+ return`
	`771`	`+ }`
	`772`	`+`
`763`	`773`	`// XXX Do bound checking against totalLen.`
`764`	`774`	`}`
`765`	`775`	`}`