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

package/bin.cjs

@@ -28623,41 +28623,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 >= d5) {
-        return Math.round(ms / d5) + "d";
+        return Math.round(ms2 / d5) + "d";
       }
       if (msAbs >= h6) {
-        return Math.round(ms / h6) + "h";
+        return Math.round(ms2 / h6) + "h";
       }
       if (msAbs >= m6) {
-        return Math.round(ms / m6) + "m";
+        return Math.round(ms2 / m6) + "m";
       }
       if (msAbs >= s6) {
-        return Math.round(ms / s6) + "s";
+        return Math.round(ms2 / s6) + "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 >= d5) {
-        return plural2(ms, msAbs, d5, "day");
+        return plural2(ms2, msAbs, d5, "day");
       }
       if (msAbs >= h6) {
-        return plural2(ms, msAbs, h6, "hour");
+        return plural2(ms2, msAbs, h6, "hour");
       }
       if (msAbs >= m6) {
-        return plural2(ms, msAbs, m6, "minute");
+        return plural2(ms2, msAbs, m6, "minute");
       }
       if (msAbs >= s6) {
-        return plural2(ms, msAbs, s6, "second");
+        return plural2(ms2, msAbs, s6, "second");
       }
-      return ms + " ms";
+      return ms2 + " ms";
     }
-    function plural2(ms, msAbs, n5, name) {
+    function plural2(ms2, msAbs, n5, name) {
       var isPlural = msAbs >= n5 * 1.5;
-      return Math.round(ms / n5) + " " + name + (isPlural ? "s" : "");
+      return Math.round(ms2 / n5) + " " + name + (isPlural ? "s" : "");
     }
   }
 });
@@ -28700,8 +28700,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;
@@ -35839,8 +35839,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(
@@ -55589,7 +55589,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: createHash2 } = require("crypto");
     var { Duplex, Readable: Readable2 } = require("stream");
@@ -56320,12 +56320,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);
     }
@@ -58616,7 +58616,7 @@ var require_dist_cjs14 = __commonJS({
       return transformedHeaders;
     };
     var timing = {
-      setTimeout: (cb, ms) => setTimeout(cb, ms),
+      setTimeout: (cb, ms2) => setTimeout(cb, ms2),
       clearTimeout: (timeoutId) => clearTimeout(timeoutId)
     };
     var DEFER_EVENT_LISTENER_TIME$2 = 1e3;
@@ -61675,13 +61675,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 [, sign, offsetH, offsetM] = /([+-])(\d\d):(\d\d)/.exec(offsetStr) || [void 0, "+", 0, 0];
@@ -81572,8 +81572,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);
       }
@@ -81583,12 +81583,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;
@@ -81609,8 +81609,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;
         }
@@ -81624,13 +81624,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);
       }
     };
@@ -81692,13 +81692,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;
@@ -81775,14 +81775,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 * sign, minutes * sign, seconds * sign, ms * sign, us * sign);
+        return new datetime_1.Duration(0, 0, 0, 0, hours * sign, minutes * sign, seconds * sign, ms2 * sign, us * sign);
       }
     };
     exports2.DurationCodec = DurationCodec;
@@ -81828,7 +81828,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);
@@ -81839,7 +81839,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 * sign, minutes * sign, seconds * sign, ms * sign, us * sign);
+        return new datetime_1.RelativeDuration(years, months2, weeks, days, hours * sign, minutes * sign, seconds * sign, ms2 * sign, us * sign);
       }
     };
     exports2.RelativeDurationCodec = RelativeDurationCodec;
@@ -106326,11 +106326,11 @@ var require_TokenExpiredError = __commonJS({
 // ../node_modules/.pnpm/jsonwebtoken@9.0.2/node_modules/jsonwebtoken/lib/timespan.js
 var require_timespan = __commonJS({
   "../node_modules/.pnpm/jsonwebtoken@9.0.2/node_modules/jsonwebtoken/lib/timespan.js"(exports2, module2) {
-    var ms = require_ms();
+    var ms2 = require_ms();
     module2.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;
         }
@@ -120917,7 +120917,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);
@@ -120957,7 +120957,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;
@@ -121169,7 +121169,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());
@@ -121178,7 +121178,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
@@ -121218,10 +121218,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 payload2 = `CONNECT ${host}:${opts.port} HTTP/1.1\r
 `;
         if (proxy.username || proxy.password) {
@@ -121254,7 +121254,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", (s6) => {
           debug("Replaying proxy buffer for failed request");
@@ -121319,7 +121319,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");
@@ -121392,7 +121392,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;
@@ -160306,7 +160306,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");
@@ -161355,7 +161355,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,
@@ -164282,10 +164282,11 @@ function parseMssqlUrl(url) {
     }
   };
 }
-var normalisePGliteUrl, preparePostgresDB, prepareCockroach, prepareGelDB, parseSingleStoreCredentials, connectToSingleStore, parseMysqlCredentials, connectToMySQL, parseMssqlCredentials, connectToMsSQL, prepareSqliteParams, preparePGliteParams, connectToSQLite, connectToLibSQL;
+var import_net, 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";
+    import_net = __toESM(require("net"));
     init_src();
     init_wrapper();
     init_utils2();
@@ -164293,6 +164294,7 @@ var init_connections = __esm({
     init_when_json_met_bigint();
     init_utils4();
     init_outputs();
+    ms = (a5, b5) => Number(b5 - a5) / 1e6;
     normalisePGliteUrl = (it2) => {
       if (it2.startsWith("file:")) {
         return it2.substring(5);
@@ -164447,13 +164449,13 @@ var init_connections = __esm({
             return pg.types.getTypeParser(typeId, format2);
           }
         };
-        const client = "url" in credentials2 ? new pg.Pool({ connectionString: credentials2.url, max: 1 }) : new pg.Pool({ ...credentials2, ssl, max: 1 });
-        const db = drizzle({ client });
+        const pool = "url" in credentials2 ? new pg.Pool({ connectionString: credentials2.url, max: 1 }) : new pg.Pool({ ...credentials2, ssl, max: 1 });
+        const db = drizzle({ client: pool });
         const migrateFn = async (config) => {
           return migrate2(db, config);
         };
         const query = async (sql, params) => {
-          const result2 = await client.query({
+          const result2 = await pool.query({
             text: sql,
             values: params ?? [],
             types: types3
@@ -164463,7 +164465,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" },
@@ -164475,7 +164477,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) {
@@ -164494,7 +164496,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 credentials2 ? new pg.Client({ connectionString: credentials2.url }) : new pg.Client({ ...credentials2, 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", (data2) => {
+            if (firstDataAt === 0n) {
+              firstDataAt = process.hrtime.bigint();
+            }
+            bytesReceived += data2.length;
+          });
+          client.connection.addListener("dataRow", (data2) => {
+            bytesReceived += data2.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 i6 = 0; i6 < repeats; i6++) {
+            const r6 = await benchmarkQuery(sql, params);
+            results.push(r6);
+          }
+          return results;
+        };
+        return { packageName: "pg", query, proxy, transactionProxy, benchmarkProxy, migrate: migrateFn };
       }
       if (await checkPackage("postgres")) {
         console.log(
@@ -164757,6 +164824,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 (credentials2) => {
@@ -164988,7 +165056,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,
@@ -165028,11 +165095,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 = import_net.default.connect(config.socketPath);
+            } else {
+              stream = import_net.default.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((resolve3, reject) => {
+            newConnection.connect((err2) => {
+              tlsConnectedAt = process.hrtime.bigint();
+              if (err2) {
+                reject(err2);
+              } else {
+                resolve3();
+              }
+            });
+          });
+          querySentAt = process.hrtime.bigint();
+          await new Promise((resolve3, 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();
+              resolve3();
+              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 i6 = 0; i6 < repeats; i6++) {
+            const r6 = await benchmarkQuery(sql, params);
+            results.push(r6);
+          }
+          return results;
+        };
         return {
           db: { query },
           packageName: "mysql2",
           proxy,
           transactionProxy,
+          benchmarkProxy,
           database: result2.database,
           migrate: migrateFn
         };
@@ -165138,6 +165305,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 = (credentials2) => {
@@ -165586,6 +165754,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 (credentials2) => {
@@ -176685,7 +176854,7 @@ __export(studio_exports, {
   prepareServer: () => prepareServer,
   prepareSingleStoreSchema: () => prepareSingleStoreSchema
 });
-var import_crypto11, import_drizzle_orm8, import_relations8, import_mssql_core3, import_mysql_core3, import_pg_core3, import_singlestore_core2, import_sqlite_core3, import_fs16, import_node_https2, preparePgSchema, prepareMySqlSchema, prepareMsSqlSchema, prepareSQLiteSchema, prepareSingleStoreSchema, getCustomDefaults, drizzleForPostgres, drizzleForMySQL, drizzleForSQLite, drizzleForLibSQL, drizzleForSingleStore, extractRelations, init, proxySchema, transactionProxySchema, defaultsSchema, schema5, jsonStringify, prepareServer;
+var import_crypto11, import_drizzle_orm8, import_relations8, import_mssql_core3, import_mysql_core3, import_pg_core3, import_singlestore_core2, import_sqlite_core3, import_fs16, import_node_https2, preparePgSchema, prepareMySqlSchema, prepareMsSqlSchema, prepareSQLiteSchema, prepareSingleStoreSchema, getCustomDefaults, drizzleForPostgres, drizzleForMySQL, drizzleForSQLite, drizzleForLibSQL, drizzleForSingleStore, extractRelations, init, proxySchema, transactionProxySchema, benchmarkProxySchema, defaultsSchema, schema5, jsonStringify, prepareServer;
 var init_studio = __esm({
   "src/cli/commands/studio.ts"() {
     "use strict";
@@ -176909,6 +177078,7 @@ var init_studio = __esm({
         packageName: db.packageName,
         proxy: db.proxy,
         transactionProxy: db.transactionProxy,
+        benchmarkProxy: db.benchmarkProxy,
         customDefaults,
         schema: pgSchema2,
         relations: relations5,
@@ -176918,7 +177088,7 @@ var init_studio = __esm({
     };
     drizzleForMySQL = async (credentials2, mysqlSchema, relations5, schemaFiles, casing2) => {
       const { connectToMySQL: connectToMySQL2 } = await Promise.resolve().then(() => (init_connections(), connections_exports));
-      const { proxy, transactionProxy, database, packageName } = await connectToMySQL2(credentials2);
+      const { proxy, transactionProxy, benchmarkProxy, database, packageName } = await connectToMySQL2(credentials2);
       const customDefaults = getCustomDefaults(mysqlSchema, casing2);
       let dbUrl;
       if ("url" in credentials2) {
@@ -176934,6 +177104,7 @@ var init_studio = __esm({
         databaseName: database,
         proxy,
         transactionProxy,
+        benchmarkProxy,
         customDefaults,
         schema: mysqlSchema,
         relations: relations5,
@@ -177103,6 +177274,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(
@@ -177117,6 +177305,7 @@ var init_studio = __esm({
       init,
       proxySchema,
       transactionProxySchema,
+      benchmarkProxySchema,
       defaultsSchema
     ]);
     jsonStringify = (data2) => {
@@ -177139,6 +177328,7 @@ var init_studio = __esm({
       databaseName,
       proxy,
       transactionProxy,
+      benchmarkProxy,
       customDefaults,
       schema: drizzleSchema,
       relations: relations5,
@@ -177202,7 +177392,7 @@ var init_studio = __esm({
             console.warn("Error message:", error4.message);
           }
           return c5.json({
-            version: "6.2",
+            version: "6.3",
             dialect: dialect6,
             driver: driver2,
             packageName,
@@ -177240,6 +177430,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 c5.body(
+            res,
+            {
+              headers: {
+                "Content-Type": "application/json"
+              }
+            }
+          );
+        }
         if (type === "defaults") {
           const columns = body.data;
           const result2 = columns.map((column12) => {