From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from localhost (localhost [127.0.0.1]) by turing.freelists.org (Avenir Technologies Mail Multiplex) with ESMTP id 0B01A21AD8 for ; Sat, 22 Dec 2018 06:31:57 -0500 (EST) Received: from turing.freelists.org ([127.0.0.1]) by localhost (turing.freelists.org [127.0.0.1]) (amavisd-new, port 10024) with ESMTP id f2h11WMkOAsB for ; Sat, 22 Dec 2018 06:31:56 -0500 (EST) Received: from smtpng3.m.smailru.net (smtpng3.m.smailru.net [94.100.177.149]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by turing.freelists.org (Avenir Technologies Mail Multiplex) with ESMTPS id B6D441FD27 for ; Sat, 22 Dec 2018 06:31:56 -0500 (EST) From: imeevma@tarantool.org Subject: [tarantool-patches] [PATCH v5 3/5] sql: create method dump_lua for port_sql Date: Sat, 22 Dec 2018 14:31:54 +0300 Message-Id: In-Reply-To: References: Sender: tarantool-patches-bounce@freelists.org Errors-to: tarantool-patches-bounce@freelists.org Reply-To: tarantool-patches@freelists.org List-help: List-unsubscribe: List-software: Ecartis version 1.0.0 List-Id: tarantool-patches List-subscribe: List-owner: List-post: List-archive: To: v.shpilevoy@tarantool.org, tarantool-patches@freelists.org This patch creates the dump_lua method for port_sql. It also defines a new function box.sql.new_execute(), which will be converted to box.sql.execute(). Part of #3505 --- src/box/execute.c | 84 ++++++++++++++++++++++++++++++++++++++++++++++++++++++- src/box/lua/sql.c | 25 +++++++++++++++++ 2 files changed, 108 insertions(+), 1 deletion(-) diff --git a/src/box/execute.c b/src/box/execute.c index b07de28..3cbee4f 100644 --- a/src/box/execute.c +++ b/src/box/execute.c @@ -43,6 +43,7 @@ #include "port.h" #include "tuple.h" #include "sql/vdbe.h" +#include "lua/utils.h" const char *sql_type_strs[] = { NULL, @@ -674,6 +675,87 @@ finish: return rc; } +/** + * Push a metadata of the prepared statement to Lua stack. + * + * @param stmt Prepared statement. + * @param L Lua stack. + * @param column_count Statement's column count. + */ +static inline void +lua_sql_get_description(struct sqlite3_stmt *stmt, struct lua_State *L, + int column_count) +{ + assert(column_count > 0); + lua_createtable(L, column_count, 0); + for (int i = 0; i < column_count; ++i) { + lua_createtable(L, 2, 0); + const char *name = sqlite3_column_name(stmt, i); + const char *type = sqlite3_column_datatype(stmt, i); + /* + * Can not fail, since all column names are + * preallocated during prepare phase and the + * column_name simply returns them. + */ + assert(name != NULL); + lua_pushlstring(L, name, strlen(name)); + lua_setfield(L, -2, "name"); + lua_pushlstring(L, type, strlen(type)); + lua_setfield(L, -2, "type"); + lua_rawseti(L, -2, i + 1); + } + lua_setfield(L, -2, "metadata"); +} + +/** + * Dump a built response into Lua stack. The response is + * destroyed. + * + * @param port port with EXECUTE response. + * @param L Lua stack. + */ +static void +port_sql_dump_lua(struct port *port, struct lua_State *L) +{ + assert(port->vtab == &port_sql_vtab); + sqlite3 *db = sql_get(); + struct sqlite3_stmt *stmt = ((struct port_sql *)port)->stmt; + int column_count = sqlite3_column_count(stmt); + if (column_count > 0) { + int keys = 2; + lua_createtable(L, 0, keys); + lua_sql_get_description(stmt, L, column_count); + port_tuple_vtab.dump_lua(port, L); + lua_setfield(L, -2, "rows"); + } else { + assert(((struct port_tuple *)port)->size == 0); + struct stailq *autoinc_id_list = + vdbe_autoinc_id_list((struct Vdbe *)stmt); + + int dynamic_keys = stailq_empty(autoinc_id_list) ? 0 : 1; + lua_createtable(L, 1, dynamic_keys); + + luaL_pushuint64(L, db->nChange); + lua_setfield(L, -2, "rowcount"); + + if (!stailq_empty(autoinc_id_list)) { + lua_newtable(L); + int i = 1; + struct autoinc_id_entry *id_entry; + stailq_foreach_entry(id_entry, autoinc_id_list, link) { + if (id_entry->id >= 0) + luaL_pushuint64(L, id_entry->id); + else + luaL_pushint64(L, id_entry->id); + lua_rawseti(L, -2, i++); + } + lua_setfield(L, -2, "autoincrement_ids"); + } + } + port_destroy(port); + sqlite3_finalize(stmt); +} + static void port_sql_destroy(struct port *base) { @@ -691,7 +773,7 @@ port_tuple_to_port_sql(struct port *port, struct sqlite3_stmt *stmt) const struct port_vtab port_sql_vtab = { .dump_msgpack = port_sql_dump_msgpack, .dump_msgpack_16 = NULL, - .dump_lua = NULL, + .dump_lua = port_sql_dump_lua, .dump_plain = NULL, .destroy = port_sql_destroy, }; diff --git a/src/box/lua/sql.c b/src/box/lua/sql.c index 6923666..318f331 100644 --- a/src/box/lua/sql.c +++ b/src/box/lua/sql.c @@ -6,6 +6,7 @@ #include #include "lua/info.h" #include "lua/utils.h" +#include "box/execute.h" static void lua_push_column_names(struct lua_State *L, struct sqlite3_stmt *stmt) @@ -111,6 +112,29 @@ sqlerror: } static int +lbox_execute(struct lua_State *L) +{ + struct sql_bind *bind = NULL; + int bind_count = 0; + size_t length; + struct port port; + + struct sqlite3 *db = sql_get(); + if (db == NULL) + return luaL_error(L, "not ready"); + + const char *sql = lua_tolstring(L, 1, &length); + if (sql == NULL) + return luaL_error(L, "usage: box.execute(sqlstring)"); + + if (sql_prepare_and_execute(sql, length, bind, bind_count, &port, + &fiber()->gc) != 0) + return luaT_error(L); + port_dump_lua(&port, L); + return 1; +} + +static int lua_sql_debug(struct lua_State *L) { struct info_handler info; @@ -124,6 +148,7 @@ box_lua_sqlite_init(struct lua_State *L) { static const struct luaL_Reg module_funcs [] = { {"execute", lua_sql_execute}, + {"new_execute", lbox_execute}, {"debug", lua_sql_debug}, {NULL, NULL} }; -- 2.7.4