X-Git-Url: https://git.tokkee.org/?a=blobdiff_plain;f=src%2Fcore%2Fdata.c;h=8754f8868a11471fe1e359cee75ac0d52b87721d;hb=419b318ff64a7d9048e9692dba7470eab9532d1f;hp=2994b5f50d3e18348dd183a553c719322e48002c;hpb=b134ee0677f2dd15cb22197e332ea44f47b688f4;p=sysdb.git

diff --git a/src/core/data.c b/src/core/data.c
index 2994b5f..8754f88 100644
--- a/src/core/data.c
+++ b/src/core/data.c
@@ -46,6 +46,77 @@
 
 #include <math.h>
 
+/*
+ * Operator support maxtrix.
+ * <type1> <op> <type2> -> op_matrix[<op>][<type1>][<type2>]
+ */
+
+/* add, sub, mul, div, mod, concat */
+
+/* integer, decimal, string, datetime, binary, regex */
+
+static int op_matrix[6][6][6] = {
+	/* SDB_DATA_ADD */
+	{
+		{ SDB_TYPE_INTEGER, -1, -1, -1, -1, -1 },
+		{ -1, SDB_TYPE_DECIMAL, -1, -1, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ -1, -1, -1, SDB_TYPE_DATETIME, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+	},
+
+	/* SDB_DATA_SUB */
+	{
+		{ SDB_TYPE_INTEGER, -1, -1, -1, -1, -1 },
+		{ -1, SDB_TYPE_DECIMAL, -1, -1, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ -1, -1, -1, SDB_TYPE_DATETIME, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+	},
+
+	/* SDB_DATA_MUL */
+	{
+		{ SDB_TYPE_INTEGER, -1, -1, SDB_TYPE_DATETIME, -1, -1 },
+		{ -1, SDB_TYPE_DECIMAL, -1, SDB_TYPE_DATETIME, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ SDB_TYPE_DATETIME, SDB_TYPE_DATETIME, -1, SDB_TYPE_DATETIME, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+	},
+
+	/* SDB_DATA_DIV */
+	{
+		{ SDB_TYPE_INTEGER, -1, -1, -1, -1, -1 },
+		{ -1, SDB_TYPE_DECIMAL, -1, -1, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ SDB_TYPE_DATETIME, SDB_TYPE_DATETIME, -1, SDB_TYPE_DATETIME, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+	},
+
+	/* SDB_DATA_MOD */
+	{
+		{ SDB_TYPE_INTEGER, -1, -1, -1, -1, -1 },
+		{ -1, SDB_TYPE_DECIMAL, -1, -1, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ SDB_TYPE_DATETIME, SDB_TYPE_DATETIME, -1, SDB_TYPE_DATETIME, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+	},
+
+	/* SDB_DATA_CONCAT */
+	{
+		{ -1, -1, -1, -1, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ -1, -1, SDB_TYPE_STRING, -1, -1, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+		{ -1, -1, -1, -1, SDB_TYPE_BINARY, -1 },
+		{ -1, -1, -1, -1, -1, -1 },
+	},
+};
+
 /*
  * private helper functions
  */
@@ -624,6 +695,32 @@ sdb_data_expr_eval(int op, const sdb_data_t *d1, const sdb_data_t *d2,
 	return -1;
 } /* sdb_data_expr_eval */
 
+int
+sdb_data_expr_type(int op, int type1, int type2)
+{
+	int types_num = (int)SDB_STATIC_ARRAY_LEN(op_matrix[0]);
+
+	assert(SDB_STATIC_ARRAY_LEN(op_matrix[0])
+			== SDB_STATIC_ARRAY_LEN(op_matrix[0][0]));
+
+	if ((op <= 0) || (SDB_STATIC_ARRAY_LEN(op_matrix) < (size_t)op))
+		return -1;
+
+	/* arrays only support concat; type has to match */
+	if ((type1 & SDB_TYPE_ARRAY) || (type2 & SDB_TYPE_ARRAY)) {
+		if ((type1 != type2) || (op != SDB_DATA_CONCAT))
+			return -1;
+		return type1;
+	}
+	if ((type1 < 0) || (types_num < type1)
+			|| (type2 < 0) || (types_num < type2))
+		return -1;
+
+	if ((type1 == SDB_TYPE_NULL) || (type2 == SDB_TYPE_NULL))
+		return SDB_TYPE_NULL;
+	return op_matrix[op - 1][type1 - 1][type2 - 1];
+} /* sdb_data_expr_type */
+
 size_t
 sdb_data_strlen(const sdb_data_t *datum)
 {