drizzle-kit 1.0.0-beta.2-b705c25 → 1.0.0-beta.2-e689d82

package/api-postgres.mjs

@@ -34431,7 +34431,7 @@ var require_websocket = __commonJS({
     var EventEmitter = __require("events");
     var https2 = __require("https");
     var http3 = __require("http");
-    var net = __require("net");
+    var net2 = __require("net");
     var tls = __require("tls");
     var { randomBytes, createHash: createHash5 } = __require("crypto");
     var { Duplex, Readable: Readable6 } = __require("stream");
@@ -35162,12 +35162,12 @@ var require_websocket = __commonJS({
     }
     function netConnect(options) {
       options.path = options.socketPath;
-      return net.connect(options);
+      return net2.connect(options);
     }
     function tlsConnect(options) {
       options.path = void 0;
       if (!options.servername && options.servername !== "") {
-        options.servername = net.isIP(options.host) ? "" : options.host;
+        options.servername = net2.isIP(options.host) ? "" : options.host;
       }
       return tls.connect(options);
     }
@@ -39485,7 +39485,7 @@ var init_timing = __esm({
   "../node_modules/.pnpm/@smithy+node-http-handler@4.4.5/node_modules/@smithy/node-http-handler/dist-es/timing.js"() {
     "use strict";
     timing = {
-      setTimeout: (cb, ms) => setTimeout(cb, ms),
+      setTimeout: (cb, ms2) => setTimeout(cb, ms2),
       clearTimeout: (timeoutId) => clearTimeout(timeoutId)
     };
   }
@@ -41373,13 +41373,13 @@ var init_schema_date_utils = __esm({
       if (!matches) {
         throw new TypeError(`Invalid RFC3339 timestamp format ${value}`);
       }
-      const [, yearStr, monthStr, dayStr, hours, minutes, seconds, , ms, offsetStr] = matches;
+      const [, yearStr, monthStr, dayStr, hours, minutes, seconds, , ms2, offsetStr] = matches;
       range(monthStr, 1, 12);
       range(dayStr, 1, 31);
       range(hours, 0, 23);
       range(minutes, 0, 59);
       range(seconds, 0, 60);
-      const date2 = new Date(Date.UTC(Number(yearStr), Number(monthStr) - 1, Number(dayStr), Number(hours), Number(minutes), Number(seconds), Number(ms) ? Math.round(parseFloat(`0.${ms}`) * 1e3) : 0));
+      const date2 = new Date(Date.UTC(Number(yearStr), Number(monthStr) - 1, Number(dayStr), Number(hours), Number(minutes), Number(seconds), Number(ms2) ? Math.round(parseFloat(`0.${ms2}`) * 1e3) : 0));
       date2.setUTCFullYear(Number(yearStr));
       if (offsetStr.toUpperCase() != "Z") {
         const [, sign2, offsetH, offsetM] = /([+-])(\d\d):(\d\d)/.exec(offsetStr) || [void 0, "+", 0, 0];
@@ -61392,8 +61392,8 @@ var require_datetime2 = __commonJS({
         if (!(object instanceof Date)) {
           throw new errors_1.InvalidArgumentError(`a Date instance was expected, got "${object}"`);
         }
-        const ms = object.getTime() - TIMESHIFT;
-        const us = ms * 1e3;
+        const ms2 = object.getTime() - TIMESHIFT;
+        const us = ms2 * 1e3;
         buf.writeInt32(8);
         buf.writeInt64(us);
       }
@@ -61403,12 +61403,12 @@ var require_datetime2 = __commonJS({
           return ctx.postDecode(this, us2 + BI_TIMESHIFT_US);
         }
         const us = Number(buf.readBigInt64());
-        let ms = Math.round(us / 1e3);
-        if (Math.abs(us % 1e3) === 500 && Math.abs(ms) % 2 === 1) {
-          ms -= 1;
+        let ms2 = Math.round(us / 1e3);
+        if (Math.abs(us % 1e3) === 500 && Math.abs(ms2) % 2 === 1) {
+          ms2 -= 1;
         }
-        ms += TIMESHIFT;
-        return new Date(ms);
+        ms2 += TIMESHIFT;
+        return new Date(ms2);
       }
     };
     exports2.DateTimeCodec = DateTimeCodec;
@@ -61429,8 +61429,8 @@ var require_datetime2 = __commonJS({
         if (!(object instanceof datetime_1.LocalDateTime)) {
           throw new errors_1.InvalidArgumentError(`a LocalDateTime instance was expected, got "${object}"`);
         }
-        const ms = BigInt(datetime_1.localDateInstances.get(object).getTime() - TIMESHIFT);
-        let us = ms * 1000n + BigInt(object.hour * 36e8 + object.minute * 6e7 + object.second * 1e6 + object.millisecond * 1e3 + object.microsecond);
+        const ms2 = BigInt(datetime_1.localDateInstances.get(object).getTime() - TIMESHIFT);
+        let us = ms2 * 1000n + BigInt(object.hour * 36e8 + object.minute * 6e7 + object.second * 1e6 + object.millisecond * 1e3 + object.microsecond);
         if (object.nanosecond === 500 && Math.abs(object.microsecond) % 2 === 1 || object.nanosecond > 500) {
           us += 1n;
         }
@@ -61444,13 +61444,13 @@ var require_datetime2 = __commonJS({
         }
         const bi_ms = bi_us / 1000n;
         let us = Number(bi_us - bi_ms * 1000n);
-        let ms = Number(bi_ms);
+        let ms2 = Number(bi_ms);
         if (us < 0) {
           us += 1e3;
-          ms -= 1;
+          ms2 -= 1;
         }
-        ms += TIMESHIFT;
-        const date2 = new Date(ms);
+        ms2 += TIMESHIFT;
+        const date2 = new Date(ms2);
         return new datetime_1.LocalDateTime(date2.getUTCFullYear(), date2.getUTCMonth() + 1, date2.getUTCDate(), date2.getUTCHours(), date2.getUTCMinutes(), date2.getUTCSeconds(), date2.getUTCMilliseconds(), us);
       }
     };
@@ -61512,13 +61512,13 @@ var require_datetime2 = __commonJS({
         }
         let us = Number(bius);
         let seconds = Math.floor(us / 1e6);
-        const ms = Math.floor(us % 1e6 / 1e3);
-        us = us % 1e6 - ms * 1e3;
+        const ms2 = Math.floor(us % 1e6 / 1e3);
+        us = us % 1e6 - ms2 * 1e3;
         let minutes = Math.floor(seconds / 60);
         seconds = Math.floor(seconds % 60);
         const hours = Math.floor(minutes / 60);
         minutes = Math.floor(minutes % 60);
-        return new datetime_1.LocalTime(hours, minutes, seconds, ms, us);
+        return new datetime_1.LocalTime(hours, minutes, seconds, ms2, us);
       }
     };
     exports2.LocalTimeCodec = LocalTimeCodec;
@@ -61595,14 +61595,14 @@ var require_datetime2 = __commonJS({
         const biMillion = 1000000n;
         const biSeconds = bius / biMillion;
         let us = Number(bius - biSeconds * biMillion);
-        const ms = Math.floor(us / 1e3);
+        const ms2 = Math.floor(us / 1e3);
         us = us % 1e3;
         let seconds = Number(biSeconds);
         let minutes = Math.floor(seconds / 60);
         seconds = Math.floor(seconds % 60);
         const hours = Math.floor(minutes / 60);
         minutes = Math.floor(minutes % 60);
-        return new datetime_1.Duration(0, 0, 0, 0, hours * sign2, minutes * sign2, seconds * sign2, ms * sign2, us * sign2);
+        return new datetime_1.Duration(0, 0, 0, 0, hours * sign2, minutes * sign2, seconds * sign2, ms2 * sign2, us * sign2);
       }
     };
     exports2.DurationCodec = DurationCodec;
@@ -61648,7 +61648,7 @@ var require_datetime2 = __commonJS({
         const million = BigInt(1e6);
         const biSeconds = bius / million;
         let us = Number(bius - biSeconds * million);
-        const ms = Math.trunc(us / 1e3);
+        const ms2 = Math.trunc(us / 1e3);
         us = us % 1e3;
         let seconds = Number(biSeconds);
         let minutes = Math.trunc(seconds / 60);
@@ -61659,7 +61659,7 @@ var require_datetime2 = __commonJS({
         days = Math.trunc(days % 7);
         const years = Math.trunc(months2 / 12);
         months2 = Math.trunc(months2 % 12);
-        return new datetime_1.RelativeDuration(years, months2, weeks, days, hours * sign2, minutes * sign2, seconds * sign2, ms * sign2, us * sign2);
+        return new datetime_1.RelativeDuration(years, months2, weeks, days, hours * sign2, minutes * sign2, seconds * sign2, ms2 * sign2, us * sign2);
       }
     };
     exports2.RelativeDurationCodec = RelativeDurationCodec;
@@ -71823,41 +71823,41 @@ var require_ms = __commonJS({
           return void 0;
       }
     }
-    function fmtShort(ms) {
-      var msAbs = Math.abs(ms);
+    function fmtShort(ms2) {
+      var msAbs = Math.abs(ms2);
       if (msAbs >= d6) {
-        return Math.round(ms / d6) + "d";
+        return Math.round(ms2 / d6) + "d";
       }
       if (msAbs >= h7) {
-        return Math.round(ms / h7) + "h";
+        return Math.round(ms2 / h7) + "h";
       }
       if (msAbs >= m7) {
-        return Math.round(ms / m7) + "m";
+        return Math.round(ms2 / m7) + "m";
       }
       if (msAbs >= s7) {
-        return Math.round(ms / s7) + "s";
+        return Math.round(ms2 / s7) + "s";
       }
-      return ms + "ms";
+      return ms2 + "ms";
     }
-    function fmtLong(ms) {
-      var msAbs = Math.abs(ms);
+    function fmtLong(ms2) {
+      var msAbs = Math.abs(ms2);
       if (msAbs >= d6) {
-        return plural2(ms, msAbs, d6, "day");
+        return plural2(ms2, msAbs, d6, "day");
       }
       if (msAbs >= h7) {
-        return plural2(ms, msAbs, h7, "hour");
+        return plural2(ms2, msAbs, h7, "hour");
       }
       if (msAbs >= m7) {
-        return plural2(ms, msAbs, m7, "minute");
+        return plural2(ms2, msAbs, m7, "minute");
       }
       if (msAbs >= s7) {
-        return plural2(ms, msAbs, s7, "second");
+        return plural2(ms2, msAbs, s7, "second");
       }
-      return ms + " ms";
+      return ms2 + " ms";
     }
-    function plural2(ms, msAbs, n6, name) {
+    function plural2(ms2, msAbs, n6, name) {
       var isPlural = msAbs >= n6 * 1.5;
-      return Math.round(ms / n6) + " " + name + (isPlural ? "s" : "");
+      return Math.round(ms2 / n6) + " " + name + (isPlural ? "s" : "");
     }
   }
 });
@@ -71901,8 +71901,8 @@ var require_common2 = __commonJS({
           }
           const self2 = debug;
           const curr = Number(/* @__PURE__ */ new Date());
-          const ms = curr - (prevTime || curr);
-          self2.diff = ms;
+          const ms2 = curr - (prevTime || curr);
+          self2.diff = ms2;
           self2.prev = prevTime;
           self2.curr = curr;
           prevTime = curr;
@@ -87405,11 +87405,11 @@ var require_TokenExpiredError = __commonJS({
 var require_timespan = __commonJS({
   "../node_modules/.pnpm/jsonwebtoken@9.0.2/node_modules/jsonwebtoken/lib/timespan.js"(exports2, module) {
     "use strict";
-    var ms = require_ms();
+    var ms2 = require_ms();
     module.exports = function(time2, iat) {
       var timestamp = iat || Math.floor(Date.now() / 1e3);
       if (typeof time2 === "string") {
-        var milliseconds = ms(time2);
+        var milliseconds = ms2(time2);
         if (typeof milliseconds === "undefined") {
           return;
         }
@@ -101947,7 +101947,7 @@ var require_dist = __commonJS({
     };
     Object.defineProperty(exports2, "__esModule", { value: true });
     exports2.Agent = void 0;
-    var net = __importStar2(__require("net"));
+    var net2 = __importStar2(__require("net"));
     var http3 = __importStar2(__require("http"));
     var https_1 = __require("https");
     __exportStar2(require_helpers2(), exports2);
@@ -101987,7 +101987,7 @@ var require_dist = __commonJS({
         if (!this.sockets[name]) {
           this.sockets[name] = [];
         }
-        const fakeSocket = new net.Socket({ writable: false });
+        const fakeSocket = new net2.Socket({ writable: false });
         this.sockets[name].push(fakeSocket);
         this.totalSocketCount++;
         return fakeSocket;
@@ -102199,7 +102199,7 @@ var require_dist2 = __commonJS({
     };
     Object.defineProperty(exports2, "__esModule", { value: true });
     exports2.HttpsProxyAgent = void 0;
-    var net = __importStar2(__require("net"));
+    var net2 = __importStar2(__require("net"));
     var tls = __importStar2(__require("tls"));
     var assert_1 = __importDefault2(__require("assert"));
     var debug_1 = __importDefault2(require_src2());
@@ -102208,7 +102208,7 @@ var require_dist2 = __commonJS({
     var parse_proxy_response_1 = require_parse_proxy_response();
     var debug = (0, debug_1.default)("https-proxy-agent");
     var setServernameFromNonIpHost = (options) => {
-      if (options.servername === void 0 && options.host && !net.isIP(options.host)) {
+      if (options.servername === void 0 && options.host && !net2.isIP(options.host)) {
         return {
           ...options,
           servername: options.host
@@ -102248,10 +102248,10 @@ var require_dist2 = __commonJS({
           socket = tls.connect(setServernameFromNonIpHost(this.connectOpts));
         } else {
           debug("Creating `net.Socket`: %o", this.connectOpts);
-          socket = net.connect(this.connectOpts);
+          socket = net2.connect(this.connectOpts);
         }
         const headers = typeof this.proxyHeaders === "function" ? this.proxyHeaders() : { ...this.proxyHeaders };
-        const host = net.isIPv6(opts.host) ? `[${opts.host}]` : opts.host;
+        const host = net2.isIPv6(opts.host) ? `[${opts.host}]` : opts.host;
         let payload = `CONNECT ${host}:${opts.port} HTTP/1.1\r
 `;
         if (proxy.username || proxy.password) {
@@ -102284,7 +102284,7 @@ var require_dist2 = __commonJS({
           return socket;
         }
         socket.destroy();
-        const fakeSocket = new net.Socket({ writable: false });
+        const fakeSocket = new net2.Socket({ writable: false });
         fakeSocket.readable = true;
         req.once("socket", (s7) => {
           debug("Replaying proxy buffer for failed request");
@@ -102349,7 +102349,7 @@ var require_dist3 = __commonJS({
     };
     Object.defineProperty(exports2, "__esModule", { value: true });
     exports2.HttpProxyAgent = void 0;
-    var net = __importStar2(__require("net"));
+    var net2 = __importStar2(__require("net"));
     var tls = __importStar2(__require("tls"));
     var debug_1 = __importDefault2(require_src2());
     var events_1 = __require("events");
@@ -102422,7 +102422,7 @@ var require_dist3 = __commonJS({
           socket = tls.connect(this.connectOpts);
         } else {
           debug("Creating `net.Socket`: %o", this.connectOpts);
-          socket = net.connect(this.connectOpts);
+          socket = net2.connect(this.connectOpts);
         }
         await (0, events_1.once)(socket, "connect");
         return socket;
@@ -141301,7 +141301,7 @@ var require_connection = __commonJS({
     var _crypto = _interopRequireDefault(__require("crypto"));
     var _os = _interopRequireDefault(__require("os"));
     var tls = _interopRequireWildcard(__require("tls"));
-    var net = _interopRequireWildcard(__require("net"));
+    var net2 = _interopRequireWildcard(__require("net"));
     var _dns = _interopRequireDefault(__require("dns"));
     var _constants = _interopRequireDefault(__require("constants"));
     var _stream = __require("stream");
@@ -142350,7 +142350,7 @@ var require_connection = __commonJS({
       async wrapWithTls(socket, signal) {
         signal.throwIfAborted();
         const secureContext = tls.createSecureContext(this.secureContextOptions);
-        const serverName = !net.isIP(this.config.server) ? this.config.server : "";
+        const serverName = !net2.isIP(this.config.server) ? this.config.server : "";
         const encryptOptions = {
           host: this.config.server,
           socket,
@@ -145259,6 +145259,7 @@ __export(connections_exports, {
   prepareGelDB: () => prepareGelDB,
   preparePostgresDB: () => preparePostgresDB
 });
+import net from "net";
 function parseMssqlUrl(url) {
   return {
     user: url.username,
@@ -145272,7 +145273,7 @@ function parseMssqlUrl(url) {
     }
   };
 }
-var normalisePGliteUrl, preparePostgresDB, prepareCockroach, prepareGelDB, parseSingleStoreCredentials, connectToSingleStore, parseMysqlCredentials, connectToMySQL, parseMssqlCredentials, connectToMsSQL, prepareSqliteParams, preparePGliteParams, connectToSQLite, connectToLibSQL;
+var ms, normalisePGliteUrl, preparePostgresDB, prepareCockroach, prepareGelDB, parseSingleStoreCredentials, connectToSingleStore, parseMysqlCredentials, connectToMySQL, parseMssqlCredentials, connectToMsSQL, prepareSqliteParams, preparePGliteParams, connectToSQLite, connectToLibSQL;
 var init_connections = __esm({
   "src/cli/connections.ts"() {
     "use strict";
@@ -145283,6 +145284,7 @@ var init_connections = __esm({
     init_when_json_met_bigint();
     init_utils3();
     init_outputs();
+    ms = (a6, b6) => Number(b6 - a6) / 1e6;
     normalisePGliteUrl = (it) => {
       if (it.startsWith("file:")) {
         return it.substring(5);
@@ -145437,13 +145439,13 @@ var init_connections = __esm({
             return pg.types.getTypeParser(typeId, format2);
           }
         };
-        const client = "url" in credentials ? new pg.Pool({ connectionString: credentials.url, max: 1 }) : new pg.Pool({ ...credentials, ssl, max: 1 });
-        const db = drizzle({ client });
+        const pool = "url" in credentials ? new pg.Pool({ connectionString: credentials.url, max: 1 }) : new pg.Pool({ ...credentials, ssl, max: 1 });
+        const db = drizzle({ client: pool });
         const migrateFn = async (config) => {
           return migrate(db, config);
         };
         const query = async (sql, params) => {
-          const result2 = await client.query({
+          const result2 = await pool.query({
             text: sql,
             values: params ?? [],
             types: types3
@@ -145453,7 +145455,7 @@ var init_connections = __esm({
           return result2.rows;
         };
         const proxy = async (params) => {
-          const result2 = await client.query({
+          const result2 = await pool.query({
             text: params.sql,
             values: params.params,
             ...params.mode === "array" && { rowMode: "array" },
@@ -145465,7 +145467,7 @@ var init_connections = __esm({
         };
         const transactionProxy = async (queries) => {
           const results = [];
-          const tx = await client.connect();
+          const tx = await pool.connect();
           try {
             await tx.query("BEGIN");
             for (const query2 of queries) {
@@ -145484,7 +145486,72 @@ var init_connections = __esm({
           }
           return results;
         };
-        return { packageName: "pg", query, proxy, transactionProxy, migrate: migrateFn };
+        const benchmarkQuery = async (sql, params) => {
+          const explainResult = await pool.query({
+            text: `EXPLAIN ANALYZE ${sql}`,
+            values: params ?? [],
+            types: types3
+          });
+          const stringifiedResult = JSON.stringify(explainResult.rows);
+          const planningMatch = stringifiedResult.match(/Planning Time:\s*([\d.]+)\s*ms/i);
+          const executionMatch = stringifiedResult.match(/Execution Time:\s*([\d.]+)\s*ms/i);
+          const planningTime = Number(planningMatch[1]);
+          const executionTime = Number(executionMatch[1]);
+          let startAt = 0n;
+          let tcpConnectedAt = 0n;
+          let tlsConnectedAt = null;
+          let dbReadyAt = 0n;
+          let querySentAt = 0n;
+          let firstDataAt = 0n;
+          let lastDataAt = 0n;
+          let bytesReceived = 0;
+          const client = "url" in credentials ? new pg.Client({ connectionString: credentials.url }) : new pg.Client({ ...credentials, ssl });
+          client.connection.once("connect", () => {
+            tcpConnectedAt = process.hrtime.bigint();
+          });
+          client.connection.prependOnceListener("sslconnect", () => {
+            tlsConnectedAt = process.hrtime.bigint();
+          });
+          client.connection.prependOnceListener("readyForQuery", () => {
+            dbReadyAt = process.hrtime.bigint();
+          });
+          client.connection.addListener("rowDescription", (data) => {
+            if (firstDataAt === 0n) {
+              firstDataAt = process.hrtime.bigint();
+            }
+            bytesReceived += data.length;
+          });
+          client.connection.addListener("dataRow", (data) => {
+            bytesReceived += data.length;
+          });
+          client.connection.addListener("commandComplete", () => {
+            lastDataAt = process.hrtime.bigint();
+          });
+          startAt = process.hrtime.bigint();
+          await client.connect();
+          querySentAt = process.hrtime.bigint();
+          await client.query(sql, params);
+          await client.end();
+          return {
+            tcpHandshake: ms(startAt, tcpConnectedAt),
+            tlsHandshake: tlsConnectedAt ? ms(tcpConnectedAt, tlsConnectedAt) : null,
+            dbHandshake: ms(tlsConnectedAt ?? tcpConnectedAt, dbReadyAt),
+            planning: planningTime,
+            execution: executionTime,
+            dataDownload: ms(firstDataAt, lastDataAt) + ms(querySentAt, firstDataAt) - executionTime - planningTime,
+            total: ms(startAt, lastDataAt),
+            dataSize: bytesReceived
+          };
+        };
+        const benchmarkProxy = async ({ sql, params }, repeats) => {
+          const results = [];
+          for (let i7 = 0; i7 < repeats; i7++) {
+            const r7 = await benchmarkQuery(sql, params);
+            results.push(r7);
+          }
+          return results;
+        };
+        return { packageName: "pg", query, proxy, transactionProxy, benchmarkProxy, migrate: migrateFn };
       }
       if (await checkPackage("postgres")) {
         console.log(
@@ -145747,6 +145814,7 @@ var init_connections = __esm({
       console.error(
         "To connect to Postgres database - please install either of 'pg', 'postgres', 'bun', '@neondatabase/serverless' or '@vercel/postgres' drivers"
       );
+      console.warn("For the 'bun' driver, run your script using: bun --bun");
       process.exit(1);
     };
     prepareCockroach = async (credentials) => {
@@ -145978,7 +146046,6 @@ To link your project, please refer https://docs.geldata.com/reference/cli/gel_in
           }
           return next();
         };
-        await connection.connect();
         const query = async (sql, params) => {
           const res = await connection.execute({
             sql,
@@ -146018,11 +146085,111 @@ To link your project, please refer https://docs.geldata.com/reference/cli/gel_in
           }
           return results;
         };
+        const benchmarkQuery = async (sql, params) => {
+          const { createConnection: createConnection2 } = await import("mysql2");
+          const explainResult = await connection.query({
+            sql: `EXPLAIN ANALYZE ${sql}`,
+            values: params ?? [],
+            typeCast
+          });
+          const stringifiedResult = JSON.stringify(explainResult[0]);
+          const timeMatch = stringifiedResult.match(
+            /actual time=([0-9.eE+-]+)\.\.([0-9.eE+-]+)/
+          );
+          const lastRowTime = Number(timeMatch[2]);
+          const executionTime = lastRowTime;
+          let startAt = 0n;
+          let tcpConnectedAt = 0n;
+          let tlsConnectedAt = null;
+          let querySentAt = 0n;
+          let firstDataAt = 0n;
+          let lastDataAt = 0n;
+          let bytesReceived = 0;
+          const createStream = ({ config }) => {
+            let stream;
+            if (config.socketPath) {
+              stream = net.connect(config.socketPath);
+            } else {
+              stream = net.connect(config.port, config.host);
+            }
+            if (config.enableKeepAlive) {
+              stream.on("connect", () => {
+                stream.setKeepAlive(true, config.keepAliveInitialDelay);
+              });
+            }
+            stream.setNoDelay(true);
+            stream.once("connect", () => {
+              tcpConnectedAt = process.hrtime.bigint();
+            });
+            return stream;
+          };
+          startAt = process.hrtime.bigint();
+          const newConnection = result2.url ? createConnection2({
+            // debug: true,
+            uri: result2.url,
+            stream: createStream
+          }) : createConnection2({
+            ...result2.credentials,
+            stream: createStream
+          });
+          await new Promise((resolve2, reject) => {
+            newConnection.connect((err2) => {
+              tlsConnectedAt = process.hrtime.bigint();
+              if (err2) {
+                reject(err2);
+              } else {
+                resolve2();
+              }
+            });
+          });
+          querySentAt = process.hrtime.bigint();
+          await new Promise((resolve2, reject) => {
+            const query2 = newConnection.query({
+              sql,
+              values: params ?? [],
+              typeCast
+              // rowsAsArray: true,
+            });
+            query2.on("error", (err2) => {
+              reject(err2);
+            });
+            query2.on("fields", (fields) => {
+              if (firstDataAt === 0n) {
+                firstDataAt = process.hrtime.bigint();
+              }
+              bytesReceived += fields[0]._buf.length;
+            });
+            query2.on("end", () => {
+              lastDataAt = process.hrtime.bigint();
+              resolve2();
+              newConnection.end();
+            });
+          });
+          return {
+            tcpHandshake: ms(startAt, tcpConnectedAt),
+            tlsHandshake: tlsConnectedAt ? ms(tcpConnectedAt, tlsConnectedAt) : null,
+            dbHandshake: null,
+            planning: null,
+            execution: executionTime,
+            dataDownload: ms(firstDataAt, lastDataAt) + ms(querySentAt, firstDataAt) - executionTime,
+            total: ms(startAt, lastDataAt),
+            dataSize: bytesReceived
+          };
+        };
+        const benchmarkProxy = async ({ sql, params }, repeats) => {
+          const results = [];
+          for (let i7 = 0; i7 < repeats; i7++) {
+            const r7 = await benchmarkQuery(sql, params);
+            results.push(r7);
+          }
+          return results;
+        };
         return {
           db: { query },
           packageName: "mysql2",
           proxy,
           transactionProxy,
+          benchmarkProxy,
           database: result2.database,
           migrate: migrateFn
         };
@@ -146128,6 +146295,7 @@ To link your project, please refer https://docs.geldata.com/reference/cli/gel_in
       console.error(
         "To connect to MySQL database - please install either of 'mysql2', 'bun' or '@planetscale/database' drivers"
       );
+      console.warn("For the 'bun' driver, run your script using: bun --bun");
       process.exit(1);
     };
     parseMssqlCredentials = (credentials) => {
@@ -146576,6 +146744,7 @@ To link your project, please refer https://docs.geldata.com/reference/cli/gel_in
       console.log(
         "Please install either 'better-sqlite3', 'bun', '@libsql/client' or '@tursodatabase/database' for Drizzle Kit to connect to SQLite databases"
       );
+      console.warn("For the 'bun' driver, run your script using: bun --bun");
       process.exit(1);
     };
     connectToLibSQL = async (credentials) => {
@@ -148574,8 +148743,8 @@ ${sql}
   `;
       return content;
     };
-    prepareSnapshotFolderName = (ms) => {
-      const now = ms ? new Date(ms) : /* @__PURE__ */ new Date();
+    prepareSnapshotFolderName = (ms2) => {
+      const now = ms2 ? new Date(ms2) : /* @__PURE__ */ new Date();
       return `${now.getFullYear()}${two(now.getUTCMonth() + 1)}${two(
         now.getUTCDate()
       )}${two(now.getUTCHours())}${two(now.getUTCMinutes())}${two(
@@ -152090,7 +152259,7 @@ import { getTableConfig as singlestoreTableConfig, SingleStoreTable } from "driz
 import { getTableConfig as sqliteTableConfig, SQLiteTable } from "drizzle-orm/sqlite-core";
 import fs10 from "fs";
 import { createServer } from "https";
-var preparePgSchema, prepareMySqlSchema, prepareMsSqlSchema, prepareSQLiteSchema, prepareSingleStoreSchema, getCustomDefaults, drizzleForPostgres, drizzleForMySQL, drizzleForSQLite, drizzleForLibSQL, drizzleForSingleStore, extractRelations, init2, proxySchema, transactionProxySchema, defaultsSchema, schema5, jsonStringify, prepareServer;
+var preparePgSchema, prepareMySqlSchema, prepareMsSqlSchema, prepareSQLiteSchema, prepareSingleStoreSchema, getCustomDefaults, drizzleForPostgres, drizzleForMySQL, drizzleForSQLite, drizzleForLibSQL, drizzleForSingleStore, extractRelations, init2, proxySchema, transactionProxySchema, benchmarkProxySchema, defaultsSchema, schema5, jsonStringify, prepareServer;
 var init_studio = __esm({
   "src/cli/commands/studio.ts"() {
     "use strict";
@@ -152304,6 +152473,7 @@ var init_studio = __esm({
         packageName: db.packageName,
         proxy: db.proxy,
         transactionProxy: db.transactionProxy,
+        benchmarkProxy: db.benchmarkProxy,
         customDefaults,
         schema: pgSchema2,
         relations: relations2,
@@ -152313,7 +152483,7 @@ var init_studio = __esm({
     };
     drizzleForMySQL = async (credentials, mysqlSchema, relations2, schemaFiles, casing2) => {
       const { connectToMySQL: connectToMySQL2 } = await Promise.resolve().then(() => (init_connections(), connections_exports));
-      const { proxy, transactionProxy, database, packageName } = await connectToMySQL2(credentials);
+      const { proxy, transactionProxy, benchmarkProxy, database, packageName } = await connectToMySQL2(credentials);
       const customDefaults = getCustomDefaults(mysqlSchema, casing2);
       let dbUrl;
       if ("url" in credentials) {
@@ -152329,6 +152499,7 @@ var init_studio = __esm({
         databaseName: database,
         proxy,
         transactionProxy,
+        benchmarkProxy,
         customDefaults,
         schema: mysqlSchema,
         relations: relations2,
@@ -152498,6 +152669,23 @@ var init_studio = __esm({
         ]).optional()
       }).array()
     });
+    benchmarkProxySchema = external_exports.object({
+      type: external_exports.literal("bproxy"),
+      data: external_exports.object({
+        query: external_exports.object({
+          sql: external_exports.string(),
+          params: external_exports.array(external_exports.any()).optional(),
+          method: external_exports.union([
+            external_exports.literal("values"),
+            external_exports.literal("get"),
+            external_exports.literal("all"),
+            external_exports.literal("run"),
+            external_exports.literal("execute")
+          ]).optional()
+        }),
+        repeats: external_exports.number().min(1).optional()
+      })
+    });
     defaultsSchema = external_exports.object({
       type: external_exports.literal("defaults"),
       data: external_exports.array(
@@ -152512,6 +152700,7 @@ var init_studio = __esm({
       init2,
       proxySchema,
       transactionProxySchema,
+      benchmarkProxySchema,
       defaultsSchema
     ]);
     jsonStringify = (data) => {
@@ -152534,6 +152723,7 @@ var init_studio = __esm({
       databaseName,
       proxy,
       transactionProxy,
+      benchmarkProxy,
       customDefaults,
       schema: drizzleSchema,
       relations: relations2,
@@ -152597,7 +152787,7 @@ var init_studio = __esm({
             console.warn("Error message:", error3.message);
           }
           return c6.json({
-            version: "6.2",
+            version: "6.3",
             dialect: dialect6,
             driver: driver2,
             packageName,
@@ -152635,6 +152825,21 @@ var init_studio = __esm({
             }
           );
         }
+        if (type === "bproxy") {
+          if (!benchmarkProxy) {
+            throw new Error("Benchmark proxy is not configured for this database.");
+          }
+          const result2 = await benchmarkProxy(body.data.query, body.data.repeats || 1);
+          const res = jsonStringify(result2);
+          return c6.body(
+            res,
+            {
+              headers: {
+                "Content-Type": "application/json"
+              }
+            }
+          );
+        }
         if (type === "defaults") {
           const columns = body.data;
           const result2 = columns.map((column8) => {