[Tarantool-patches] [PATCH v4 44/53] sql: introduce mem_convert_to_string()

imeevma at tarantool.org imeevma at tarantool.org
Tue Mar 23 12:36:35 MSK 2021


This patch introduces mem_convert_to_string(). This function is used to
convert MEM to MEM contains string value.

Part of #5818
---
 src/box/sql/mem.c | 197 ++++++++++++++++++++++++++++++----------------
 src/box/sql/mem.h |   7 +-
 2 files changed, 133 insertions(+), 71 deletions(-)

diff --git a/src/box/sql/mem.c b/src/box/sql/mem.c
index 8600b5c41..262f48aca 100644
--- a/src/box/sql/mem.c
+++ b/src/box/sql/mem.c
@@ -818,6 +818,131 @@ mem_convert_to_number(struct Mem *mem)
 	return -1;
 }
 
+static inline int
+mem_convert_integer_to_string(struct Mem *mem)
+{
+	const char *str;
+	if ((mem->flags & MEM_UInt) != 0)
+		str = tt_sprintf("%llu", mem->u.u);
+	else
+		str = tt_sprintf("%lld", mem->u.i);
+	return mem_copy_string0(mem, str);
+}
+
+static inline int
+mem_convert_array_to_string(struct Mem *mem)
+{
+	const char *str = mp_str(mem->z);
+	return mem_copy_string0(mem, str);
+}
+
+static inline int
+mem_convert_map_to_string(struct Mem *mem)
+{
+	const char *str = mp_str(mem->z);
+	return mem_copy_string0(mem, str);
+}
+
+static inline int
+mem_convert_boolean_to_string(struct Mem *mem)
+{
+	const char *str = mem->u.b ? "TRUE" : "FALSE";
+	return mem_copy_string0(mem, str);
+}
+
+static inline int
+mem_convert_binary_to_string(struct Mem *mem)
+{
+	if (ExpandBlob(mem) != 0)
+		return -1;
+	mem->flags = (mem->flags & (MEM_Dyn | MEM_Static | MEM_Ephem)) |
+		      MEM_Str;
+	mem->field_type = FIELD_TYPE_STRING;
+	return 0;
+}
+
+static inline int
+mem_convert_binary_to_string0(struct Mem *mem)
+{
+	if (ExpandBlob(mem) != 0)
+		return -1;
+	if (sqlVdbeMemGrow(mem, mem->n + 1, 1) != 0)
+		return -1;
+	mem->z[mem->n] = '\0';
+	mem->flags = MEM_Str | MEM_Term;
+	mem->field_type = FIELD_TYPE_STRING;
+	return 0;
+}
+
+static inline int
+mem_convert_double_to_string(struct Mem *mem)
+{
+	uint32_t size = 32;
+	if (sqlVdbeMemClearAndResize(mem, size) != 0)
+		return -1;
+	sql_snprintf(size, mem->z, "%!.15g", mem->u.r);
+	mem->n = strlen(mem->z);
+	mem->flags = MEM_Str | MEM_Term;
+	mem->field_type = FIELD_TYPE_STRING;
+	return 0;
+}
+
+static inline int
+mem_convert_string_to_string0(struct Mem *mem)
+{
+	assert((mem->flags | MEM_Str) != 0);
+	if (sqlVdbeMemGrow(mem, mem->n + 1, 1) != 0)
+		return -1;
+	mem->z[mem->n] = '\0';
+	mem->flags |= MEM_Term;
+	mem->field_type = FIELD_TYPE_STRING;
+	return 0;
+}
+
+int
+mem_convert_to_string0(struct Mem *mem)
+{
+	if ((mem->flags & MEM_Str) != 0 && (mem->flags & MEM_Term) != 0)
+		return 0;
+	if ((mem->flags & MEM_Str) != 0)
+		return mem_convert_string_to_string0(mem);
+	if ((mem->flags & (MEM_Int | MEM_UInt)) != 0)
+		return mem_convert_integer_to_string(mem);
+	if ((mem->flags & MEM_Real) != 0)
+		return mem_convert_double_to_string(mem);
+	if ((mem->flags & MEM_Bool) != 0)
+		return mem_convert_boolean_to_string(mem);
+	if ((mem->flags & MEM_Blob) != 0) {
+		if ((mem->flags & MEM_Subtype) == 0)
+			return mem_convert_binary_to_string0(mem);
+		if (mp_typeof(*mem->z) == MP_MAP)
+			return mem_convert_map_to_string(mem);
+		return mem_convert_array_to_string(mem);
+	}
+	return -1;
+}
+
+int
+mem_convert_to_string(struct Mem *mem)
+{
+	if ((mem->flags & MEM_Str) != 0)
+		return 0;
+	if ((mem->flags & (MEM_Int | MEM_UInt)) != 0)
+		return mem_convert_integer_to_string(mem);
+	if ((mem->flags & MEM_Real) != 0)
+		return mem_convert_double_to_string(mem);
+	if ((mem->flags & MEM_Bool) != 0)
+		return mem_convert_boolean_to_string(mem);
+	if ((mem->flags & MEM_Blob) != 0) {
+		if ((mem->flags & MEM_Subtype) == 0)
+			return mem_convert_binary_to_string(mem);
+		if (mp_typeof(*mem->z) == MP_MAP)
+			return mem_convert_map_to_string(mem);
+		return mem_convert_array_to_string(mem);
+	}
+	return -1;
+}
+
 int
 mem_copy(struct Mem *to, const struct Mem *from)
 {
@@ -1452,7 +1577,7 @@ valueToText(sql_value * pVal)
 		pVal->flags |= MEM_Str;
 		sqlVdbeMemNulTerminate(pVal);	/* IMP: R-31275-44060 */
 	} else {
-		sqlVdbeMemStringify(pVal);
+		mem_convert_to_string(pVal);
 		assert(0 == (1 & SQL_PTR_TO_INT(pVal->z)));
 	}
 	return pVal->z;
@@ -1978,74 +2103,6 @@ sqlVdbeMemCast(Mem * pMem, enum field_type type)
 	}
 }
 
-/*
- * Add MEM_Str to the set of representations for the given Mem.  Numbers
- * are converted using sql_snprintf().  Converting a BLOB to a string
- * is a no-op.
- *
- * Existing representations MEM_Int and MEM_Real are invalidated if
- * bForce is true but are retained if bForce is false.
- *
- * A MEM_Null value will never be passed to this function. This function is
- * used for converting values to text for returning to the user (i.e. via
- * sql_value_text()), or for ensuring that values to be used as btree
- * keys are strings. In the former case a NULL pointer is returned the
- * user and the latter is an internal programming error.
- */
-int
-sqlVdbeMemStringify(Mem * pMem)
-{
-	int fg = pMem->flags;
-	int nByte = 32;
-
-	if ((fg & (MEM_Null | MEM_Str | MEM_Blob)) != 0 &&
-	    !mem_has_msgpack_subtype(pMem))
-		return 0;
-
-	assert(!(fg & MEM_Zero));
-	assert((fg & (MEM_Int | MEM_UInt | MEM_Real | MEM_Bool |
-		      MEM_Blob)) != 0);
-	assert(EIGHT_BYTE_ALIGNMENT(pMem));
-
-	/*
-	 * In case we have ARRAY/MAP we should save decoded value
-	 * before clearing pMem->z.
-	 */
-	char *value = NULL;
-	if (mem_has_msgpack_subtype(pMem)) {
-		const char *value_str = mp_str(pMem->z);
-		nByte = strlen(value_str) + 1;
-		value = region_alloc(&fiber()->gc, nByte);
-		memcpy(value, value_str, nByte);
-	}
-
-	if (sqlVdbeMemClearAndResize(pMem, nByte)) {
-		return -1;
-	}
-	if (fg & MEM_Int) {
-		sql_snprintf(nByte, pMem->z, "%lld", pMem->u.i);
-		pMem->flags &= ~MEM_Int;
-	} else if ((fg & MEM_UInt) != 0) {
-		sql_snprintf(nByte, pMem->z, "%llu", pMem->u.u);
-		pMem->flags &= ~MEM_UInt;
-	} else if ((fg & MEM_Bool) != 0) {
-		sql_snprintf(nByte, pMem->z, "%s",
-			     SQL_TOKEN_BOOLEAN(pMem->u.b));
-		pMem->flags &= ~MEM_Bool;
-	} else if (mem_has_msgpack_subtype(pMem)) {
-		sql_snprintf(nByte, pMem->z, "%s", value);
-		pMem->flags &= ~MEM_Subtype;
-		pMem->subtype = SQL_SUBTYPE_NO;
-	} else {
-		assert(fg & MEM_Real);
-		sql_snprintf(nByte, pMem->z, "%!.15g", pMem->u.r);
-		pMem->flags &= ~MEM_Real;
-	}
-	pMem->n = sqlStrlen30(pMem->z);
-	pMem->flags |= MEM_Str | MEM_Term;
-	return 0;
-}
-
 /*
  * Make sure the given Mem is \u0000 terminated.
  */
@@ -2163,7 +2220,7 @@ mem_apply_type(struct Mem *record, enum field_type type)
 		 */
 		if ((record->flags & MEM_Str) == 0 &&
 		    (record->flags & (MEM_Real | MEM_Int | MEM_UInt)) != 0)
-			sqlVdbeMemStringify(record);
+			mem_convert_to_string(record);
 		record->flags &= ~(MEM_Real | MEM_Int | MEM_UInt);
 		return 0;
 	case FIELD_TYPE_VARBINARY:
diff --git a/src/box/sql/mem.h b/src/box/sql/mem.h
index d5f0d7891..d370b448b 100644
--- a/src/box/sql/mem.h
+++ b/src/box/sql/mem.h
@@ -330,6 +330,12 @@ mem_convert_to_double(struct Mem *mem);
 int
 mem_convert_to_number(struct Mem *mem);
 
+int
+mem_convert_to_string(struct Mem *mem);
+
+int
+mem_convert_to_string0(struct Mem *mem);
+
 /**
  * Simple type to str convertor. It is used to simplify
  * error reporting.
@@ -381,7 +387,6 @@ registerTrace(int iReg, Mem *p);
 #endif
 
 int sqlVdbeMemCast(struct Mem *, enum field_type type);
-int sqlVdbeMemStringify(struct Mem *);
 int sqlVdbeMemNulTerminate(struct Mem *);
 int sqlVdbeMemExpandBlob(struct Mem *);
 #define ExpandBlob(P) (mem_is_zeroblob(P)? sqlVdbeMemExpandBlob(P) : 0)
-- 
2.25.1



More information about the Tarantool-patches mailing list