Add drizzleIntegration query header

Added query header for future logging of integration usage
singlestore-labs · Nov 6, 2024 · 6b8676e · 6b8676e
1 parent 465a8a2
commit 6b8676e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/drizzle-orm/src/singlestore/session.ts b/drizzle-orm/src/singlestore/session.ts
@@ -63,8 +63,9 @@ export class SingleStoreDriverPreparedQuery<T extends SingleStorePreparedQueryCo
 		private returningIds?: SelectedFieldsOrdered,
 	) {
 		super();
+		const queryWithComment = `/* drizzleIntegration */ ${queryString}`;
 		this.rawQuery = {
-			sql: queryString,
+			sql: queryWithComment,
 			// rowsAsArray: true,
 			typeCast: function(field: any, next: any) {
 				if (field.type === 'TIMESTAMP' || field.type === 'DATETIME' || field.type === 'DATE') {
@@ -74,7 +75,7 @@ export class SingleStoreDriverPreparedQuery<T extends SingleStorePreparedQueryCo
 			},
 		};
 		this.query = {
-			sql: queryString,
+			sql: queryWithComment,
 			rowsAsArray: true,
 			typeCast: function(field: any, next: any) {
 				if (field.type === 'TIMESTAMP' || field.type === 'DATETIME' || field.type === 'DATE') {