[Tarantool-patches] [PATCH v4 24/53] sql: introduce mem_set_integer()

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


This patch introduces mem_set_integer(). Function mem_set_integer()
clears MEM and sets it to given integer value.

Part of #5818
---
 src/box/sql/func.c    |  6 +--
 src/box/sql/mem.c     | 88 ++++++++++++++++++-------------------------
 src/box/sql/mem.h     | 21 ++++-------
 src/box/sql/vdbe.c    | 10 ++---
 src/box/sql/vdbeapi.c |  4 +-
 src/box/sql/vdbeaux.c |  6 +--
 6 files changed, 56 insertions(+), 79 deletions(-)

diff --git a/src/box/sql/func.c b/src/box/sql/func.c
index b61de18d8..8f7550f30 100644
--- a/src/box/sql/func.c
+++ b/src/box/sql/func.c
@@ -278,7 +278,7 @@ port_lua_get_vdbemem(struct port *base, uint32_t *size)
 			mem_set_double(&val[i], field.dval);
 			break;
 		case MP_INT:
-			mem_set_i64(&val[i], field.ival);
+			mem_set_integer(&val[i], field.ival, true);
 			break;
 		case MP_UINT:
 			mem_set_u64(&val[i], field.ival);
@@ -349,7 +349,7 @@ port_c_get_vdbemem(struct port *base, uint32_t *size)
 			mem_set_double(&val[i], mp_decode_double(&data));
 			break;
 		case MP_INT:
-			mem_set_i64(&val[i], mp_decode_int(&data));
+			mem_set_integer(&val[i], mp_decode_int(&data), true);
 			break;
 		case MP_UINT:
 			mem_set_u64(&val[i], mp_decode_uint(&data));
@@ -1989,7 +1989,7 @@ sumFinalize(sql_context * context)
 		} else if (p->approx) {
 			sql_result_double(context, p->rSum);
 		} else {
-			mem_set_int(context->pOut, p->iSum, p->is_neg);
+			mem_set_integer(context->pOut, p->iSum, p->is_neg);
 		}
 	}
 }
diff --git a/src/box/sql/mem.c b/src/box/sql/mem.c
index da2aa5c94..13a587aba 100644
--- a/src/box/sql/mem.c
+++ b/src/box/sql/mem.c
@@ -263,6 +263,15 @@ mem_set_null(struct Mem *mem)
 	mem_clear(mem);
 }
 
+void
+mem_set_integer(struct Mem *mem, int64_t value, bool is_neg)
+{
+	mem_clear(mem);
+	mem->u.i = value;
+	mem->flags = is_neg ? MEM_Int : MEM_UInt;
+	mem->field_type = FIELD_TYPE_INTEGER;
+}
+
 int
 mem_copy(struct Mem *to, const struct Mem *from)
 {
@@ -557,8 +566,7 @@ mem_arithmetic(const struct Mem *left, const struct Mem *right,
 	default:
 		unreachable();
 	}
-	result->u.i = ires;
-	result->flags = is_res_neg ? MEM_Int : MEM_UInt;
+	mem_set_integer(result, ires, is_res_neg);
 	return 0;
 }
 
@@ -571,6 +579,7 @@ mem_bitwise(struct Mem *left, struct Mem *right, struct Mem *result, int op)
 		return 0;
 	int64_t l;
 	int64_t r;
+	int64_t res;
 	bool unused;
 	if (sqlVdbeIntValue(left, &l, &unused) != 0) {
 		diag_set(ClientError, ER_SQL_TYPE_MISMATCH,
@@ -583,13 +592,13 @@ mem_bitwise(struct Mem *left, struct Mem *right, struct Mem *result, int op)
 		return -1;
 	}
 	if (op == OP_BitAnd) {
-		result->u.i = l & r;
-		result->flags = result->u.i < 0 ? MEM_Int : MEM_UInt;
+		res = l & r;
+		mem_set_integer(result, res, res < 0);
 		return 0;
 	}
 	if (op == OP_BitOr) {
-		result->u.i = l | r;
-		result->flags = result->u.i < 0 ? MEM_Int : MEM_UInt;
+		res = l | r;
+		mem_set_integer(result, res, res < 0);
 		return 0;
 	}
 	assert(op == OP_ShiftRight || op == OP_ShiftLeft);
@@ -599,17 +608,17 @@ mem_bitwise(struct Mem *left, struct Mem *right, struct Mem *result, int op)
 		r = r < -64 ? 64 : -r;
 	}
 	if (r >= 64) {
-		result->u.i = is_left_shift || l >= 0 ? 0 : -1;
-		result->flags = result->u.i < 0 ? MEM_Int : MEM_UInt;
+		res = is_left_shift || l >= 0 ? 0 : -1;
+		mem_set_integer(result, res, res < 0);
 		return 0;
 	}
 	if (is_left_shift) {
-		result->u.i = l << r;
-		result->flags = result->u.i < 0 ? MEM_Int : MEM_UInt;
+		res = l << r;
+		mem_set_integer(result, res, res < 0);
 		return 0;
 	}
-	result->u.i = l >> r;
-	result->flags = result->u.i < 0 ? MEM_Int : MEM_UInt;
+	res = l >> r;
+	mem_set_integer(result, res, res < 0);
 	return 0;
 }
 
@@ -628,7 +637,7 @@ mem_bit_not(struct Mem *mem, struct Mem *result)
 		return -1;
 	}
 	i = ~i;
-	mem_set_int(result, i, i < 0);
+	mem_set_integer(result, i, i < 0);
 	return 0;
 }
 
@@ -1328,7 +1337,7 @@ mem_apply_numeric_type(struct Mem *record)
 	int64_t integer_value;
 	bool is_neg;
 	if (sql_atoi64(record->z, &integer_value, &is_neg, record->n) == 0) {
-		mem_set_int(record, integer_value, is_neg);
+		mem_set_integer(record, integer_value, is_neg);
 		return 0;
 	}
 	double float_value;
@@ -1359,15 +1368,14 @@ vdbe_mem_numerify(struct Mem *mem)
 	if ((mem->flags & (MEM_Int | MEM_UInt | MEM_Real | MEM_Null)) != 0)
 		return 0;
 	if ((mem->flags & MEM_Bool) != 0) {
-		mem->u.u = mem->u.b;
-		MemSetTypeFlag(mem, MEM_UInt);
+		mem_set_integer(mem, (int64_t)mem->u.b, false);
 		return 0;
 	}
 	assert((mem->flags & (MEM_Blob | MEM_Str)) != 0);
 	bool is_neg;
 	int64_t i;
 	if (sql_atoi64(mem->z, &i, &is_neg, mem->n) == 0) {
-		mem_set_int(mem, i, is_neg);
+		mem_set_integer(mem, i, is_neg);
 	} else {
 		double d;
 		if (sqlAtoF(mem->z, &d, mem->n) == 0)
@@ -1425,12 +1433,11 @@ sqlVdbeMemCast(Mem * pMem, enum field_type type)
 				return -1;
 			if (type == FIELD_TYPE_UNSIGNED && is_neg)
 				return -1;
-			mem_set_int(pMem, val, is_neg);
+			mem_set_integer(pMem, val, is_neg);
 			return 0;
 		}
 		if ((pMem->flags & MEM_Bool) != 0) {
-			pMem->u.u = pMem->u.b;
-			MemSetTypeFlag(pMem, MEM_UInt);
+			mem_set_integer(pMem, (int64_t)pMem->u.b, false);
 			return 0;
 		}
 		if ((pMem->flags & MEM_Real) != 0) {
@@ -1438,7 +1445,7 @@ sqlVdbeMemCast(Mem * pMem, enum field_type type)
 			if (sqlVdbeRealValue(pMem, &d) != 0)
 				return -1;
 			if (d < (double)INT64_MAX && d >= (double)INT64_MIN) {
-				mem_set_int(pMem, d, d <= -1);
+				mem_set_integer(pMem, d, d <= -1);
 				return 0;
 			}
 			if (d >= (double)INT64_MAX && d < (double)UINT64_MAX) {
@@ -1494,7 +1501,7 @@ mem_apply_integer_type(Mem *pMem)
 	assert(EIGHT_BYTE_ALIGNMENT(pMem));
 
 	if ((rc = doubleToInt64(pMem->u.r, (int64_t *) &ix)) == 0)
-		mem_set_int(pMem, ix, pMem->u.r <= -1);
+		mem_set_integer(pMem, ix, pMem->u.r <= -1);
 	return rc;
 }
 
@@ -1643,8 +1650,10 @@ mem_apply_type(struct Mem *record, enum field_type type)
 			} else {
 				if (double_compare_nint64(d, INT64_MIN, 1) < 0)
 					return 0;
-				if ((double)(int64_t)d == d)
-					mem_set_int(record, (int64_t)d, true);
+				if ((double)(int64_t)d == d) {
+					mem_set_integer(record, (int64_t)d,
+							true);
+				}
 			}
 			return 0;
 		}
@@ -1653,7 +1662,7 @@ mem_apply_type(struct Mem *record, enum field_type type)
 			int64_t i;
 			if (sql_atoi64(record->z, &i, &is_neg, record->n) != 0)
 				return -1;
-			mem_set_int(record, i, is_neg);
+			mem_set_integer(record, i, is_neg);
 		}
 		if ((record->flags & MEM_Int) == MEM_Int) {
 			if (type == FIELD_TYPE_UNSIGNED)
@@ -1777,9 +1786,9 @@ mem_convert_to_integer(struct Mem *mem)
 	if (d >= (double)UINT64_MAX || d < (double)INT64_MIN)
 		return -1;
 	if (d < (double)INT64_MAX)
-		mem_set_int(mem, (int64_t) d, d < 0);
+		mem_set_integer(mem, (int64_t) d, d < 0);
 	else
-		mem_set_int(mem, (uint64_t) d, false);
+		mem_set_integer(mem, (uint64_t) d, false);
 	return 0;
 }
 
@@ -1898,16 +1907,6 @@ mem_set_ptr(struct Mem *mem, void *ptr)
 	mem->u.p = ptr;
 }
 
-void
-mem_set_i64(struct Mem *mem, int64_t value)
-{
-	mem_clear(mem);
-	mem->u.i = value;
-	int flag = value < 0 ? MEM_Int : MEM_UInt;
-	MemSetTypeFlag(mem, flag);
-	mem->field_type = FIELD_TYPE_INTEGER;
-}
-
 void
 mem_set_u64(struct Mem *mem, uint64_t value)
 {
@@ -1917,21 +1916,6 @@ mem_set_u64(struct Mem *mem, uint64_t value)
 	mem->field_type = FIELD_TYPE_UNSIGNED;
 }
 
-void
-mem_set_int(struct Mem *mem, int64_t value, bool is_neg)
-{
-	mem_clear(mem);
-	if (is_neg) {
-		assert(value < 0);
-		mem->u.i = value;
-		MemSetTypeFlag(mem, MEM_Int);
-	} else {
-		mem->u.u = value;
-		MemSetTypeFlag(mem, MEM_UInt);
-	}
-	mem->field_type = FIELD_TYPE_INTEGER;
-}
-
 void
 mem_set_double(struct Mem *mem, double value)
 {
diff --git a/src/box/sql/mem.h b/src/box/sql/mem.h
index dc0d4a284..0fd902274 100644
--- a/src/box/sql/mem.h
+++ b/src/box/sql/mem.h
@@ -166,6 +166,13 @@ mem_destroy(struct Mem *mem);
 void
 mem_set_null(struct Mem *mem);
 
+/**
+ * Set integer value. According to is_neg flag value is considered
+ * to be signed or unsigned.
+ */
+void
+mem_set_integer(struct Mem *mem, int64_t value, bool is_neg);
+
 /**
  * Copy content of MEM from one MEM to another. In case source MEM contains
  * string or binary and allocation type is not STATIC, this value is copied to
@@ -409,24 +416,10 @@ mem_set_bool(struct Mem *mem, bool value);
 void
 mem_set_ptr(struct Mem *mem, void *ptr);
 
-/**
- * Set integer value. Depending on its sign MEM_Int (in case
- * of negative value) or MEM_UInt flag is set.
- */
-void
-mem_set_i64(struct Mem *mem, int64_t value);
-
 /** Set unsigned value and MEM_UInt flag. */
 void
 mem_set_u64(struct Mem *mem, uint64_t value);
 
-/**
- * Set integer value. According to is_neg flag value is considered
- * to be signed or unsigned.
- */
-void
-mem_set_int(struct Mem *mem, int64_t value, bool is_neg);
-
 /** Set double value and MEM_Real flag. */
 void
 mem_set_double(struct Mem *mem, double value);
diff --git a/src/box/sql/vdbe.c b/src/box/sql/vdbe.c
index f0b56033a..92845d66d 100644
--- a/src/box/sql/vdbe.c
+++ b/src/box/sql/vdbe.c
@@ -731,7 +731,7 @@ case OP_Halt: {
  */
 case OP_Integer: {         /* out2 */
 	pOut = vdbe_prepare_null_out(p, pOp->p2);
-	mem_set_i64(pOut, pOp->p1);
+	mem_set_integer(pOut, pOp->p1, pOp->p1 < 0);
 	break;
 }
 
@@ -756,7 +756,7 @@ case OP_Bool: {         /* out2 */
 case OP_Int64: {           /* out2 */
 	pOut = vdbe_prepare_null_out(p, pOp->p2);
 	assert(pOp->p4.pI64!=0);
-	mem_set_int(pOut, *pOp->p4.pI64, pOp->p4type == P4_INT64);
+	mem_set_integer(pOut, *pOp->p4.pI64, pOp->p4type == P4_INT64);
 	break;
 }
 
@@ -2708,7 +2708,7 @@ skip_truncate:
 	r.nField = (u16)nField;
 
 	if (int_field > 0)
-		mem_set_int(&aMem[int_field], iKey, is_neg);
+		mem_set_integer(&aMem[int_field], iKey, is_neg);
 
 	r.default_rc = ((1 & (oc - OP_SeekLT)) ? -1 : +1);
 	assert(oc!=OP_SeekGT || r.default_rc==-1);
@@ -2999,8 +2999,8 @@ case OP_FCopy: {     /* out2 */
 		assert(mem_is_integer(pIn1));
 
 		pOut = vdbe_prepare_null_out(p, pOp->p2);
-		mem_set_int(pOut, pIn1->u.i, pIn1->flags == MEM_Int);
-		pOut->field_type = pIn1->field_type;
+		if (mem_copy(pOut, pIn1) != 0)
+			goto abort_due_to_error;
 	}
 	break;
 }
diff --git a/src/box/sql/vdbeapi.c b/src/box/sql/vdbeapi.c
index 76af333e8..b0f1f202f 100644
--- a/src/box/sql/vdbeapi.c
+++ b/src/box/sql/vdbeapi.c
@@ -189,7 +189,7 @@ sql_result_uint(sql_context *ctx, uint64_t u_val)
 void
 sql_result_int(sql_context *ctx, int64_t val)
 {
-	mem_set_i64(ctx->pOut, val);
+	mem_set_integer(ctx->pOut, val, val < 0);
 }
 
 void
@@ -840,7 +840,7 @@ sql_bind_int64(sql_stmt * pStmt, int i, sql_int64 iValue)
 		return -1;
 	int rc = sql_bind_type(p, i, "integer");
 	assert(iValue < 0);
-	mem_set_int(&p->aVar[i - 1], iValue, true);
+	mem_set_integer(&p->aVar[i - 1], iValue, true);
 	return rc;
 }
 
diff --git a/src/box/sql/vdbeaux.c b/src/box/sql/vdbeaux.c
index 9d07358e6..3d1f094b8 100644
--- a/src/box/sql/vdbeaux.c
+++ b/src/box/sql/vdbeaux.c
@@ -1324,13 +1324,13 @@ sqlVdbeList(Vdbe * p)
 			}
 		}
 
-		mem_set_i64(pMem, pOp->p1);
+		mem_set_integer(pMem, pOp->p1, pOp->p1 < 0);
 		pMem++;
 
-		mem_set_i64(pMem, pOp->p2);
+		mem_set_integer(pMem, pOp->p2, pOp->p2 < 0);
 		pMem++;
 
-		mem_set_i64(pMem, pOp->p3);
+		mem_set_integer(pMem, pOp->p3, pOp->p3 < 0);
 		pMem++;
 
 		if (sqlVdbeMemClearAndResize(pMem, 256)) {
-- 
2.25.1



More information about the Tarantool-patches mailing list