@use-tusk/drift-node-sdk 0.1.15 → 0.1.16

package/dist/index.cjs

@@ -11810,6 +11810,7 @@ var PgInstrumentation = class extends TdInstrumentationBase {
 				const inputValue = {
 					text: queryConfig.text,
 					values: queryConfig.values || [],
+					rowMode: queryConfig.rowMode,
 					clientType
 				};
 				if (self.mode === TuskDriftMode.REPLAY) {
@@ -11924,6 +11925,7 @@ var PgInstrumentation = class extends TdInstrumentationBase {
 		if (typeof firstArg === "object" && firstArg.text) return {
 			text: firstArg.text,
 			values: firstArg.values,
+			rowMode: firstArg.rowMode,
 			callback: firstArg.callback || (typeof args[1] === "function" ? args[1] : void 0)
 		};
 		return null;
@@ -12024,7 +12026,7 @@ var PgInstrumentation = class extends TdInstrumentationBase {
 			logger.warn(`[PgInstrumentation] No mock data found for PG query: ${queryText}`);
 			throw new Error(`[PgInstrumentation] No mock data found for PG query: ${queryText}`);
 		}
-		const processedResult = this.convertPostgresTypes(mockData.result);
+		const processedResult = this.convertPostgresTypes(mockData.result, inputValue.rowMode);
 		if (queryConfig.callback) {
 			process.nextTick(() => {
 				queryConfig.callback(null, processedResult);
@@ -12033,35 +12035,56 @@ var PgInstrumentation = class extends TdInstrumentationBase {
 		} else return Promise.resolve(processedResult);
 	}
 	/**
+	* Convert a single value based on its PostgreSQL data type.
+	*
+	* Reference for data type IDs: https://jdbc.postgresql.org/documentation/publicapi/constant-values.html
+	*/
+	convertPostgresValue(value, dataTypeID) {
+		if (value === null || value === void 0) return value;
+		switch (dataTypeID) {
+			case 1184:
+			case 1114:
+			case 1082:
+				if (typeof value === "string") return new Date(value);
+				break;
+			case 1083:
+			case 1266: break;
+			default: break;
+		}
+		return value;
+	}
+	/**
 	* Convert PostgreSQL string values back to appropriate JavaScript types
 	* based on field metadata from the recorded response.
 	*
 	* Reference for data type IDs: https://jdbc.postgresql.org/documentation/publicapi/constant-values.html
 	*/
-	convertPostgresTypes(result) {
+	convertPostgresTypes(result, rowMode) {
 		if (!result || !result.fields || !result.rows) return result;
+		if (rowMode === "array") {
+			const convertedRows$1 = result.rows.map((row) => {
+				if (!Array.isArray(row)) return row;
+				return row.map((value, index) => {
+					const field = result.fields[index];
+					if (!field) return value;
+					return this.convertPostgresValue(value, field.dataTypeID);
+				});
+			});
+			return {
+				...result,
+				rows: convertedRows$1
+			};
+		}
 		const fieldTypeMap = {};
 		result.fields.forEach((field) => {
 			fieldTypeMap[field.name] = field.dataTypeID;
 		});
-		const convertedRows = result.rows.map((row, rowIndex) => {
+		const convertedRows = result.rows.map((row) => {
 			const convertedRow = { ...row };
 			Object.keys(row).forEach((fieldName) => {
 				const dataTypeID = fieldTypeMap[fieldName];
 				const value = row[fieldName];
-				if (value === null || value === void 0) return;
-				switch (dataTypeID) {
-					case 1184:
-					case 1114:
-						if (typeof value === "string") convertedRow[fieldName] = new Date(value);
-						break;
-					case 1082:
-						if (typeof value === "string") convertedRow[fieldName] = new Date(value);
-						break;
-					case 1083:
-					case 1266: break;
-					default: break;
-				}
+				convertedRow[fieldName] = this.convertPostgresValue(value, dataTypeID);
 			});
 			return convertedRow;
 		});
@@ -13289,6 +13312,12 @@ var MysqlInstrumentation = class extends TdInstrumentationBase {
 				logger.debug(`[MysqlInstrumentation] Wrapped Connection.prototype.resume`);
 			}
 		}
+		if (ConnectionClass.prototype && ConnectionClass.prototype.destroy) {
+			if (!isWrapped$1(ConnectionClass.prototype.destroy)) {
+				this._wrap(ConnectionClass.prototype, "destroy", this._getDestroyPatchFn());
+				logger.debug(`[MysqlInstrumentation] Wrapped Connection.prototype.destroy`);
+			}
+		}
 		this.markModuleAsPatched(ConnectionClass);
 		logger.debug(`[MysqlInstrumentation] Connection class patching complete`);
 		return ConnectionClass;
@@ -13703,6 +13732,19 @@ var MysqlInstrumentation = class extends TdInstrumentationBase {
 		};
 	}
 	/**
+	* Get wrapper function for destroy method
+	*/
+	_getDestroyPatchFn() {
+		const self = this;
+		return (originalDestroy) => {
+			return function destroy() {
+				if (self.mode === TuskDriftMode.REPLAY) return;
+				else if (self.mode === TuskDriftMode.RECORD) return originalDestroy.apply(this, arguments);
+				else return originalDestroy.apply(this, arguments);
+			};
+		};
+	}
+	/**
 	* Get wrapper function for Pool.end method
 	*/
 	_getPoolEndPatchFn() {
@@ -32015,7 +32057,7 @@ var require_src = /* @__PURE__ */ __commonJS({ "node_modules/@opentelemetry/sdk-
 //#endregion
 //#region package.json
 var import_src$1 = /* @__PURE__ */ __toESM(require_src(), 1);
-var version = "0.1.15";
+var version = "0.1.16";
 
 //#endregion
 //#region src/version.ts