From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from smtpng1.m.smailru.net (smtpng1.m.smailru.net [94.100.181.251]) (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) (No client certificate requested) by dev.tarantool.org (Postfix) with ESMTPS id 3E7364696C8 for ; Thu, 21 Nov 2019 00:28:29 +0300 (MSK) From: Nikita Pettik Date: Thu, 21 Nov 2019 00:28:10 +0300 Message-Id: <0a6806665bbc165dffc23967be57135a46bbad62.1574277369.git.korablev@tarantool.org> In-Reply-To: References: In-Reply-To: References: Subject: [Tarantool-patches] [PATCH v2 11/16] sql: introduce sql_stmt_sizeof() function List-Id: Tarantool development patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , To: tarantool-patches@dev.tarantool.org Cc: v.shpilevoy@tarantool.org To implement memory quota of prepared statement cache, we have to estimate size of prepared statement. This function attempts at that. Part of #2592 --- src/box/execute.h | 8 ++++++++ src/box/sql/vdbeapi.c | 53 +++++++++++++++++++++++++++++++++++++++++++++++++++ 2 files changed, 61 insertions(+) diff --git a/src/box/execute.h b/src/box/execute.h index 6702a18cc..d5b4d8421 100644 --- a/src/box/execute.h +++ b/src/box/execute.h @@ -116,6 +116,14 @@ extern const struct port_vtab port_sql_vtab; int sql_finalize(struct sql_stmt *stmt); +/** + * Calculate estimated size of memory occupied by VM. + * See sqlVdbeMakeReady() for details concerning allocated + * memory. + */ +size_t +sql_stmt_sizeof(const struct sql_stmt *stmt); + /** * Prepare (compile into VDBE byte-code) statement. * diff --git a/src/box/sql/vdbeapi.c b/src/box/sql/vdbeapi.c index da528a4dc..10135bb68 100644 --- a/src/box/sql/vdbeapi.c +++ b/src/box/sql/vdbeapi.c @@ -805,6 +805,59 @@ sql_stmt_schema_version(sql_stmt *stmt) return v->schema_ver; } +size_t +sql_stmt_sizeof(const sql_stmt *stmt) +{ + struct Vdbe *v = (struct Vdbe *) stmt; + size_t size = sizeof(*v); + /* Resulting set */ + size += sizeof(struct Mem) * v->nResColumn * COLNAME_N; + /* Opcodes */ + size += sizeof(struct VdbeOp) * v->nOp; + /* Memory cells */ + size += sizeof(struct Mem) * v->nMem; + /* Bindings */ + size += sizeof(struct Mem) * v->nVar; + /* Bindings included in the result set */ + size += sizeof(uint32_t) * v->res_var_count; + /* Cursors */ + size += sizeof(struct VdbeCursor *) * v->nCursor; + + for (int i = 0; i < v->nOp; ++i) { + /* Estimate size of p4 operand. */ + if (v->aOp[i].p4type != P4_NOTUSED) { + switch (v->aOp[i].p4type) { + case P4_DYNAMIC: + case P4_STATIC: + if (v->aOp[i].opcode == OP_Blob || + v->aOp[i].opcode == OP_String) + size += v->aOp[i].p1; + else if (v->aOp[i].opcode == OP_String8) + size += strlen(v->aOp[i].p4.z); + break; + case P4_BOOL: + size += sizeof(v->aOp[i].p4.b); + break; + case P4_INT32: + size += sizeof(v->aOp[i].p4.i); + break; + case P4_UINT64: + case P4_INT64: + size += sizeof(*v->aOp[i].p4.pI64); + break; + case P4_REAL: + size += sizeof(*v->aOp[i].p4.pReal); + break; + default: + size += sizeof(v->aOp[i].p4.p); + break; + } + } + } + size += strlen(v->zSql); + return size; +} + /******************************* sql_bind_ ************************** * * Routines used to attach values to wildcards in a compiled SQL statement. -- 2.15.1