From c6028fe4c36983f7de40540c5df40f0d05f48882 Mon Sep 17 00:00:00 2001
From: Michael Huang <mikhuang@gmail.com>
Date: Thu, 20 Nov 2025 15:04:39 -0800
Subject: [PATCH] Match preparedstatement.js execute behavior to statement.js

---
 lib/preparedstatement.js | 19 +++++++++++++++++--
 1 file changed, 17 insertions(+), 2 deletions(-)

diff --git a/lib/preparedstatement.js b/lib/preparedstatement.js
index 9b5c1be..1e39caa 100644
--- a/lib/preparedstatement.js
+++ b/lib/preparedstatement.js
@@ -28,9 +28,24 @@ PreparedStatement.prototype.clearParameters = function(callback) {
 };
 
 PreparedStatement.prototype.execute = function(callback) {
-  this._ps.execute(function(err, result) {
+  var ps = this._ps
+  ps.execute(function(err, isResultSet) {
     if (err) { winston.error(err); return callback(err); }
-    callback(null, result);
+    if (isResultSet) {
+      ps.getResultSet(function(err, resultset) {
+        if (err) {
+          return callback(err);
+        }
+        return callback(null, new ResultSet(resultset));
+      });
+    } else {
+      ps.getUpdateCount(function(err, count) {
+        if (err) {
+          return callback(err);
+        }
+        return callback(null, count);
+      });
+    }
   });
 };