From a5af0d852cd8bd435b19ecc54a078f3ebfd7db02 Mon Sep 17 00:00:00 2001
From: Brian Cully <bjc@kublai.com>
Date: Sun, 2 Mar 2008 17:53:45 -0500
Subject: Add select_count support.

---
 src/mysqlerl.c | 23 +++++++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

(limited to 'src')

diff --git a/src/mysqlerl.c b/src/mysqlerl.c
index 79731b7..100f26c 100644
--- a/src/mysqlerl.c
+++ b/src/mysqlerl.c
@@ -169,7 +169,6 @@ handle_query(MYSQL *dbh, ETERM *cmd)
   }
   erl_free(q);
 
-  logmsg("DEBUG: prepping buffers and sending.");
   write_msg(resp);
   erl_free_term(resp);
 }
@@ -194,7 +193,7 @@ handle_param_query(MYSQL *dbh, ETERM *msg)
 void
 handle_select_count(MYSQL *dbh, ETERM *msg)
 {
-  ETERM *query;
+  ETERM *query, *resp;
   char *q;
 
   query = erl_element(2, msg);
@@ -202,8 +201,28 @@ handle_select_count(MYSQL *dbh, ETERM *msg)
   erl_free_term(query);
 
   logmsg("DEBUG: got select count msg: %s.", q);
+  if (mysql_query(dbh, q)) {
+    resp = erl_format("{error, {mysql_error, ~i, ~s}}",
+                      mysql_errno(dbh), mysql_error(dbh));
+  } else {
+    MYSQL_RES *result;
 
+    result = mysql_store_result(dbh);
+    if (result) {
+      resp = erl_format("{ok, ~i}", mysql_num_rows(result));
+      mysql_free_result(result);
+    } else {
+      if (mysql_field_count(dbh) == 0)
+        resp = erl_format("{ok, ~i}", mysql_affected_rows(dbh));
+      else
+        resp = erl_format("{error, {mysql_error, ~i, ~s}}",
+                          mysql_errno(dbh), mysql_error(dbh));
+    }
+  }
   erl_free(q);
+
+  write_msg(resp);
+  erl_free_term(resp);
 }
 
 void
-- 
cgit v1.2.3