diff options
| author | Brian Cully <bjc@kublai.com> | 2008-03-02 17:53:45 -0500 |
|---|---|---|
| committer | Brian Cully <github.20.shmit@spamgourmet.com> | 2008-03-02 17:53:45 -0500 |
| commit | a5af0d852cd8bd435b19ecc54a078f3ebfd7db02 (patch) | |
| tree | 2b5835f74ee3905407ac006311b4fa1730f8c721 | |
| parent | 52a4f21f1aa608b327af5d2e83cf0c0a29ff7c7e (diff) | |
| download | mysqlerl-a5af0d852cd8bd435b19ecc54a078f3ebfd7db02.tar.gz mysqlerl-a5af0d852cd8bd435b19ecc54a078f3ebfd7db02.zip | |
Add select_count support.
| -rw-r--r-- | src/mysqlerl.c | 23 |
1 files changed, 21 insertions, 2 deletions
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 |