implementing prepared statement support in mod-mysql.c

hawk/mod/mod-mysql.c

@@ -55,10 +55,26 @@
 #define __FREE_IMAP_NODE __free_res_node
 #include "../lib/imap-imp.h"
 
+#undef __IMAP_NODE_T_DATA
+#undef __IMAP_LIST_T_DATA
+#undef __IMAP_LIST_T
+#undef __IMAP_NODE_T
+#undef __MAKE_IMAP_NODE
+#undef __FREE_IMAP_NODE
+
+#define __IMAP_NODE_T_DATA  MYSQL_STMT* stmt;
+#define __IMAP_LIST_T_DATA  /* int errnum; */
+#define __IMAP_LIST_T stmt_list_t
+#define __IMAP_NODE_T stmt_node_t
+#define __MAKE_IMAP_NODE __new_stmt_node
+#define __FREE_IMAP_NODE __free_stmt_node
+#include "../lib/imap-imp.h"
+
 struct rtx_data_t
 {
 	sql_list_t sql_list;
 	res_list_t res_list;
+	stmt_list_t stmt_list;
 };
 typedef struct rtx_data_t rtx_data_t;
 
@@ -86,7 +102,6 @@ static void free_sql_node (hawk_rtx_t* rtx, sql_list_t* sql_list, sql_node_t* sq
 	__free_sql_node (rtx, sql_list, sql_node);
 }
 
-
 static res_node_t* new_res_node (hawk_rtx_t* rtx, res_list_t* res_list, MYSQL_RES* res)
 {
 	res_node_t* res_node;
@@ -107,6 +122,24 @@ static void free_res_node (hawk_rtx_t* rtx, res_list_t* res_list, res_node_t* re
 	__free_res_node (rtx, res_list, res_node);
 }
 
+static stmt_node_t* new_stmt_node (hawk_rtx_t* rtx, stmt_list_t* stmt_list, MYSQL_STMT* stmt)
+{
+	stmt_node_t* stmt_node;
+
+	stmt_node = __new_stmt_node(rtx, stmt_list);
+	if (!stmt_node) return HAWK_NULL;
+
+	stmt_node->stmt = stmt;
+
+	return stmt_node;
+}
+
+static void free_stmt_node (hawk_rtx_t* rtx, stmt_list_t* stmt_list, stmt_node_t* stmt_node)
+{
+	mysql_stmt_close (stmt_node->stmt);
+	stmt_node->stmt = HAWK_NULL;
+	__free_stmt_node (rtx, stmt_list, stmt_node);
+}
 /* ------------------------------------------------------------------------ */
 
 static HAWK_INLINE sql_list_t* rtx_to_sql_list (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
@@ -141,6 +174,22 @@ static HAWK_INLINE res_node_t* get_res_list_node (res_list_t* res_list, hawk_int
 	return res_list->map.tab[id];
 }
 
+
+static HAWK_INLINE stmt_list_t* rtx_to_stmt_list (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
+{
+	hawk_rbt_pair_t* pair;
+	rtx_data_t* data;
+	pair = hawk_rbt_search((hawk_rbt_t*)fi->mod->ctx, &rtx, HAWK_SIZEOF(rtx));
+	HAWK_ASSERT (pair != HAWK_NULL);
+	data = (rtx_data_t*)HAWK_RBT_VPTR(pair);
+	return &data->stmt_list;
+}
+
+static HAWK_INLINE stmt_node_t* get_stmt_list_node (stmt_list_t* stmt_list, hawk_int_t id)
+{
+	if (id < 0 || id >= stmt_list->map.high || !stmt_list->map.tab[id]) return HAWK_NULL;
+	return stmt_list->map.tab[id];
+}
 /* ------------------------------------------------------------------------ */
 
 static void set_error_on_sql_list (hawk_rtx_t* rtx, sql_list_t* sql_list, const hawk_ooch_t* errfmt, ...)
@@ -211,7 +260,58 @@ static res_node_t* get_res_list_node_with_arg (hawk_rtx_t* rtx, sql_list_t* sql_
 	return res_node;
 }
 
+static stmt_node_t* get_stmt_list_node_with_arg (hawk_rtx_t* rtx, sql_list_t* sql_list, stmt_list_t* stmt_list, hawk_val_t* arg)
+{
+	hawk_int_t id;
+	stmt_node_t* stmt_node;
+
+	if (hawk_rtx_valtoint(rtx, arg, &id) <= -1)
+	{
+		set_error_on_sql_list (rtx, sql_list, HAWK_T("illegal statement id"));
+		return HAWK_NULL;
+	}
+	else if (!(stmt_node = get_stmt_list_node(stmt_list, id)))
+	{
+		set_error_on_sql_list (rtx, sql_list, HAWK_T("invalid statement id - %zd"), (hawk_oow_t)id);
+		return HAWK_NULL;
+	}
+
+	return stmt_node;
+}
+
 /* ------------------------------------------------------------------------ */
+/*
+	mysql = mysql::open();
+
+	if (mysql::connect(mysql, "localhost", "username", "password", "database") <= -1)
+	{
+		print "connect error -", mysql::errmsg();
+	}
+
+
+	if (mysql::query(mysql, "select * from mytable") <= -1)
+	{
+		print "query error -", mysql::errmsg();
+	}
+
+	result = mysql::store_result(mysql);
+	if (result <= -1)
+	{
+		print "store result error - ", mysql::errmsg();
+	}
+
+	while (mysql::fetch_row(result, row) > 0)
+	{
+		ncols = length(row);
+		for (i = 0; i < ncols; i++) print row[i];
+		print "----";
+	}
+
+	mysql::free_result(result);
+
+	mysql::close(mysql);
+*/
+
 static int fnc_open (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
 {
 	sql_list_t* sql_list;
@@ -613,10 +713,6 @@ done:
 	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
 	hawk_rtx_setretval (rtx, hawk_rtx_makeintval(rtx, ret));
 	return 0;
-
-oops:
-	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
-	return -1;
 }
 
 static int fnc_commit (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
@@ -629,10 +725,6 @@ static int fnc_commit (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
 	sql_node = get_sql_list_node_with_arg(rtx, sql_list, hawk_rtx_getarg(rtx, 0));
 	if (sql_node)
 	{
-		int v;
-
-		v = hawk_rtx_valtobool(rtx, hawk_rtx_getarg(rtx, 1));
-
 		ENSURE_CONNECT_EVER_ATTEMPTED(rtx, sql_list, sql_node);
 
 		if (mysql_commit(sql_node->mysql) != 0)
@@ -648,10 +740,6 @@ done:
 	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
 	hawk_rtx_setretval (rtx, hawk_rtx_makeintval(rtx, ret));
 	return 0;
-
-oops:
-	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
-	return -1;
 }
 
 static int fnc_rollback (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
@@ -664,10 +752,6 @@ static int fnc_rollback (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
 	sql_node = get_sql_list_node_with_arg(rtx, sql_list, hawk_rtx_getarg(rtx, 0));
 	if (sql_node)
 	{
-		int v;
-
-		v = hawk_rtx_valtobool(rtx, hawk_rtx_getarg(rtx, 1));
-
 		ENSURE_CONNECT_EVER_ATTEMPTED(rtx, sql_list, sql_node);
 
 		if (mysql_rollback(sql_node->mysql) != 0)
@@ -683,10 +767,6 @@ done:
 	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
 	hawk_rtx_setretval (rtx, hawk_rtx_makeintval(rtx, ret));
 	return 0;
-
-oops:
-	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
-	return -1;
 }
 
 static int fnc_affected_rows (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
@@ -773,7 +853,6 @@ static int fnc_escape_string (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
 			goto oops;
 		}
 
-
 		if (hawk_rtx_setrefval(rtx, (hawk_val_ref_t*)hawk_rtx_getarg(rtx, 2), retv) <= -1)
 		{
 			hawk_rtx_refupval (rtx, retv);
@@ -786,7 +865,6 @@ static int fnc_escape_string (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
 	}
 
 done:
-	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
 	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
 	if (ebuf) hawk_rtx_freemem (rtx, ebuf);
 	if (qstr) hawk_rtx_freevalbcstr (rtx, a1, qstr);
@@ -794,6 +872,7 @@ done:
 	return 0;
 
 oops:
+	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
 	if (ebuf) hawk_rtx_freemem (rtx, ebuf);
 	if (qstr) hawk_rtx_freevalbcstr (rtx, a1, qstr);
 	return -1;
@@ -987,6 +1066,154 @@ oops:
 	return -1;
 }
 
+/* -------------------------------------------------------------------------- */
+static int fnc_stmt_init (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
+{
+	sql_list_t* sql_list;
+	sql_node_t* sql_node;
+
+	int ret = -1;
+	int take_rtx_err = 0;
+
+	sql_list = rtx_to_sql_list(rtx, fi);
+	sql_node = get_sql_list_node_with_arg(rtx, sql_list, hawk_rtx_getarg(rtx, 0));
+
+	if (sql_list)
+	{
+		stmt_list_t* stmt_list;
+		stmt_node_t* stmt_node;
+		MYSQL_STMT* stmt;
+
+		stmt_list = rtx_to_stmt_list(rtx, fi);
+
+		ENSURE_CONNECT_EVER_ATTEMPTED(rtx, sql_list, sql_node);
+
+		stmt = mysql_stmt_init(sql_node->mysql);
+		if (!stmt)
+		{
+			set_error_on_sql_list (rtx, sql_list, HAWK_T("%hs"), mysql_error(sql_node->mysql));
+			goto done;
+		}
+
+		stmt_node = new_stmt_node(rtx, stmt_list, stmt);
+		if (!stmt_node)
+		{
+			mysql_stmt_close (stmt);
+			take_rtx_err = 1;
+			goto done;
+		}
+
+		ret = stmt_node->id;
+	}
+
+done:
+	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
+	hawk_rtx_setretval (rtx, hawk_rtx_makeintval(rtx, ret));
+	return 0;
+}
+
+static int fnc_stmt_close (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
+{
+	sql_list_t* sql_list;
+	stmt_list_t* stmt_list;
+	stmt_node_t* stmt_node;
+	int ret = -1;
+
+	sql_list = rtx_to_sql_list(rtx, fi);
+	stmt_list = rtx_to_stmt_list(rtx, fi);
+	stmt_node = get_stmt_list_node_with_arg(rtx, sql_list, stmt_list, hawk_rtx_getarg(rtx, 0));
+	if (stmt_node)
+	{
+		free_stmt_node (rtx, stmt_list, stmt_node);
+		ret = 0;
+	}
+
+	hawk_rtx_setretval (rtx, hawk_rtx_makeintval(rtx, ret));
+	return 0;
+}
+
+static int fnc_stmt_prepare (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
+{
+	sql_list_t* sql_list;
+	sql_node_t* sql_node;
+	stmt_list_t* stmt_list;
+	stmt_node_t* stmt_node;
+	int ret = -1;
+	int take_rtx_err = 0;
+	hawk_val_t* a1;
+	hawk_bch_t* qstr = HAWK_NULL;
+
+	sql_list = rtx_to_sql_list(rtx, fi);
+	stmt_list = rtx_to_stmt_list(rtx, fi);
+	sql_node = get_sql_list_node_with_arg(rtx, sql_list, hawk_rtx_getarg(rtx, 0));
+	stmt_node = get_stmt_list_node_with_arg(rtx, sql_list, stmt_list, hawk_rtx_getarg(rtx, 0));
+	if (stmt_node)
+	{
+		hawk_oow_t qlen;
+
+		a1 = hawk_rtx_getarg(rtx, 1);
+
+		qstr = hawk_rtx_getvalbcstr(rtx, a1, &qlen);
+		if (!qstr) { take_rtx_err = 1; goto oops; }
+
+		ENSURE_CONNECT_EVER_ATTEMPTED(rtx, sql_list, sql_node);
+
+		if (mysql_stmt_prepare(stmt_node->stmt, qstr, qlen) != 0)
+		{
+			set_error_on_sql_list (rtx, sql_list, HAWK_T("%hs"), mysql_stmt_error(stmt_node->stmt));
+			goto done;
+		}
+
+		ret = 0;
+	}
+
+done:
+	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
+	if (qstr) hawk_rtx_freevalbcstr (rtx, a1, qstr);
+	hawk_rtx_setretval (rtx, hawk_rtx_makeintval(rtx, ret));
+	return 0;
+
+oops:
+	if (take_rtx_err) set_error_on_sql_list (rtx, sql_list, HAWK_NULL);
+	if (qstr) hawk_rtx_freevalbcstr (rtx, a1, qstr);
+	return -1;
+}
+
+static int fnc_stmt_bind_param (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
+{
+	/* TODO: */
+	return -1;
+}
+
+static int fnc_stmt_execute (hawk_rtx_t* rtx, const hawk_fnc_info_t* fi)
+{
+	sql_list_t* sql_list;
+	sql_node_t* sql_node;
+	stmt_list_t* stmt_list;
+	stmt_node_t* stmt_node;
+	int ret = -1;
+
+	sql_list = rtx_to_sql_list(rtx, fi);
+	stmt_list = rtx_to_stmt_list(rtx, fi);
+	sql_node = get_sql_list_node_with_arg(rtx, sql_list, hawk_rtx_getarg(rtx, 0));
+	stmt_node = get_stmt_list_node_with_arg(rtx, sql_list, stmt_list, hawk_rtx_getarg(rtx, 0));
+	if (stmt_node)
+	{
+		ENSURE_CONNECT_EVER_ATTEMPTED(rtx, sql_list, sql_node);
+
+		if (mysql_stmt_execute(stmt_node->stmt) != 0)
+		{
+			set_error_on_sql_list (rtx, sql_list, HAWK_T("%hs"), mysql_stmt_error(stmt_node->stmt));
+			goto done;
+		}
+		ret = 0;
+	}
+
+done:
+	hawk_rtx_setretval (rtx, hawk_rtx_makeintval(rtx, ret));
+	return 0;
+}
+
 typedef struct fnctab_t fnctab_t;
 struct fnctab_t
 {
@@ -1023,6 +1250,13 @@ static fnctab_t fnctab[] =
 	{ HAWK_T("rollback"),          { { 1, 1, HAWK_NULL },     fnc_rollback,        0 } },
 	{ HAWK_T("select_db"),         { { 2, 2, HAWK_NULL },     fnc_select_db,       0 } },
 	{ HAWK_T("set_option"),        { { 3, 3, HAWK_NULL },     fnc_set_option,      0 } },
+
+	{ HAWK_T("stmt_bind_param"),   { { 1, 1, HAWK_NULL },     fnc_stmt_bind_param, 0 } },
+	{ HAWK_T("stmt_close"),        { { 1, 1, HAWK_NULL },     fnc_stmt_close,      0 } },
+	{ HAWK_T("stmt_execute"),      { { 1, 1, HAWK_NULL },     fnc_stmt_execute,    0 } },
+	{ HAWK_T("stmt_init"),         { { 1, 1, HAWK_NULL },     fnc_stmt_init,       0 } },
+	{ HAWK_T("stmt_prepare"),      { { 2, 2, HAWK_NULL },     fnc_stmt_prepare,    0 } },
+
 	{ HAWK_T("store_result"),      { { 1, 1, HAWK_NULL },     fnc_store_result,    0 } }
 };