SERVER-33098 Make shards check the client's databaseVersion and throw StaleDbVersion on mismatch

1 files changed, 23 insertions, 0 deletions

diff --git a/src/mongo/s/stale_exception.cpp b/src/mongo/s/stale_exception.cpp
index 6aea3833d52..2fad1b5e142 100644
--- a/src/mongo/s/stale_exception.cpp
+++ b/src/mongo/s/stale_exception.cpp
@@ -35,6 +35,7 @@
 namespace mongo {
 
 MONGO_INIT_REGISTER_ERROR_EXTRA_INFO(StaleConfigInfo);
+MONGO_INIT_REGISTER_ERROR_EXTRA_INFO(StaleDbRoutingVersion);
 
 void StaleConfigInfo::serialize(BSONObjBuilder* bob) const {
     bob->append("ns", _ns);
@@ -51,4 +52,26 @@ std::shared_ptr<const ErrorExtraInfo> StaleConfigInfo::parse(const BSONObj& obj)
     return std::make_shared<StaleConfigInfo>(obj);
 }
 
+void StaleDbRoutingVersion::serialize(BSONObjBuilder* bob) const {
+    bob->append("db", _db);
+    bob->append("vReceived", _received.toBSON());
+    if (_wanted) {
+        bob->append("vWanted", _wanted->toBSON());
+    }
+}
+
+StaleDbRoutingVersion::StaleDbRoutingVersion(const BSONObj& obj)
+    : StaleDbRoutingVersion(
+          obj["db"].String(),
+          DatabaseVersion::parse(IDLParserErrorContext("StaleDbRoutingVersion-vReceived"),
+                                 obj["vReceived"].Obj()),
+          !obj["vWanted"].eoo()
+              ? DatabaseVersion::parse(IDLParserErrorContext("StaleDbRoutingVersion-vWanted"),
+                                       obj["vWanted"].Obj())
+              : boost::optional<DatabaseVersion>{}) {}
+
+std::shared_ptr<const ErrorExtraInfo> StaleDbRoutingVersion::parse(const BSONObj& obj) {
+    return std::make_shared<StaleDbRoutingVersion>(obj);
+}
+
 }  // namespace mongo