sharded <4.4 support

Author: aditi-khare-mongoDB

src/cmap/connect.ts

@@ -31,6 +31,7 @@ import {
   MIN_SUPPORTED_SERVER_VERSION,
   MIN_SUPPORTED_WIRE_VERSION
 } from './wire_protocol/constants';
+import { isSharded } from './wire_protocol/shared';
 
 /** @public */
 export type Stream = Socket | TLSSocket;
@@ -164,7 +165,7 @@ export async function performInitialHandshake(
     } catch (error) {
       if (error instanceof MongoError) {
         error.addErrorLabel(MongoErrorLabel.HandshakeError);
-        if (needsRetryableWriteLabel(error, response.maxWireVersion)) {
+        if (needsRetryableWriteLabel(error, response.maxWireVersion, isSharded(conn))) {
           error.addErrorLabel(MongoErrorLabel.RetryableWriteError);
         }
       }

src/error.ts

@@ -1215,11 +1215,15 @@ const RETRYABLE_READ_ERROR_CODES = new Set<number>([
 // see: https://github.com/mongodb/specifications/blob/master/source/retryable-writes/retryable-writes.rst#terms
 const RETRYABLE_WRITE_ERROR_CODES = RETRYABLE_READ_ERROR_CODES;
 
-export function needsRetryableWriteLabel(error: Error, maxWireVersion: number): boolean {
+export function needsRetryableWriteLabel(
+  error: Error,
+  maxWireVersion: number,
+  isSharded: boolean
+): boolean {
   // pre-4.4 server, then the driver adds an error label for every valid case
   // execute operation will only inspect the label, code/message logic is handled here
   if (error instanceof MongoNetworkError) {
-    return true;
+    return !isSharded;
   }
 
   if (error instanceof MongoError) {
@@ -1235,7 +1239,9 @@ export function needsRetryableWriteLabel(error: Error, maxWireVersion: number):
   }
 
   if (error instanceof MongoWriteConcernError) {
-    return RETRYABLE_WRITE_ERROR_CODES.has(error.result?.code ?? error.code ?? 0);
+    return isSharded && maxWireVersion < 9
+      ? false
+      : RETRYABLE_WRITE_ERROR_CODES.has(error.result?.code ?? error.code ?? 0);
   }
 
   if (error instanceof MongoError && typeof error.code === 'number') {

src/sdam/server.ts

@@ -8,6 +8,7 @@ import {
 } from '../cmap/connection_pool';
 import { PoolClearedError } from '../cmap/errors';
 import { type MongoDBResponseConstructor } from '../cmap/wire_protocol/responses';
+import { isSharded } from '../cmap/wire_protocol/shared';
 import {
   APM_EVENTS,
   CLOSED,
@@ -453,7 +454,7 @@ export class Server extends TypedEventEmitter<ServerEvents> {
     } else {
       if (
         (isRetryableWritesEnabled(this.topology) || isTransactionCommand(cmd)) &&
-        needsRetryableWriteLabel(error, maxWireVersion(this)) &&
+        needsRetryableWriteLabel(error, maxWireVersion(this), isSharded(this)) &&
         !inActiveTransaction(session, cmd)
       ) {
         error.addErrorLabel(MongoErrorLabel.RetryableWriteError);

src/sessions.ts

@@ -391,7 +391,7 @@ export class ClientSession extends TypedEventEmitter<ClientSessionEvents> {
 
     const topologyMaxWireVersion = maxWireVersion(this.client.topology);
     if (
-      isSharded(this.client.topology) &&
+      is(this.client.topology) &&
       topologyMaxWireVersion != null &&
       topologyMaxWireVersion < minWireVersionForShardedTransactions
     ) {