X-Git-Url: https://git.tokkee.org/?p=sysdb.git;a=blobdiff_plain;f=src%2Fcore%2Fstore_expr.c;h=6e39305a89ecd18718ad7984a4e7cc85f407728d;hp=a0047a1f12e1fa8498ef0ed5255373478f98a070;hb=da52b4da948c741d20c7a3995ce5d133af377def;hpb=9935ca9e91f79b4f5ee6e9a7f94912b7992e8858

diff --git a/src/core/store_expr.c b/src/core/store_expr.c
index a0047a1..6e39305 100644
--- a/src/core/store_expr.c
+++ b/src/core/store_expr.c
@@ -46,6 +46,23 @@
  * private data types
  */
 
+/* iterate through either a list of child nodes or arrays */
+struct sdb_store_expr_iter {
+	sdb_store_obj_t *obj;
+	sdb_store_expr_t *expr;
+
+	sdb_avltree_iter_t *tree;
+
+	sdb_data_t array;
+	size_t array_idx;
+
+	sdb_store_matcher_t *filter;
+};
+
+/*
+ * private types
+ */
+
 static int
 expr_init(sdb_object_t *obj, va_list ap)
 {
@@ -150,7 +167,7 @@ sdb_store_expr_fieldvalue(int field)
 	sdb_data_t value = { SDB_TYPE_INTEGER, { .integer = field } };
 	sdb_store_expr_t *e;
 
-	if ((field < SDB_FIELD_NAME) || (SDB_FIELD_BACKEND < field))
+	if ((field < SDB_FIELD_NAME) || (SDB_FIELD_VALUE < field))
 		return NULL;
 	e = SDB_STORE_EXPR(sdb_object_create("store-fieldvalue", expr_type,
 				FIELD_VALUE, NULL, NULL, &value));
@@ -244,5 +261,206 @@ sdb_store_expr_eval(sdb_store_expr_t *expr, sdb_store_obj_t *obj,
 	return status;
 } /* sdb_store_expr_eval */
 
+bool
+sdb_store_expr_iterable(sdb_store_expr_t *expr, int context)
+{
+	if (expr->type == TYPED_EXPR) {
+		if ((context != SDB_HOST) && (context != SDB_SERVICE)
+				&& (context != SDB_METRIC))
+			return 0;
+		if (context == expr->data.data.integer)
+			return 0;
+		if ((expr->data.data.integer != SDB_SERVICE)
+				&& (expr->data.data.integer != SDB_METRIC)
+				&& (expr->data.data.integer != SDB_ATTRIBUTE))
+			return 0;
+		if ((context == SDB_SERVICE)
+				&& (expr->data.data.integer == SDB_METRIC))
+			return 0;
+		else if ((context == SDB_METRIC)
+				&& (expr->data.data.integer == SDB_SERVICE))
+			return 0;
+		return 1;
+	}
+	else if (expr->type == FIELD_VALUE) {
+		if ((context != SDB_HOST) && (context != SDB_SERVICE)
+				&& (context != SDB_METRIC) && (context != SDB_ATTRIBUTE))
+			return 0;
+		if (expr->data.data.integer == SDB_FIELD_BACKEND)
+			return 1;
+		else if (expr->data.data.integer == SDB_FIELD_VALUE)
+			/* we don't current support this just like when using
+			 * the attribute[<name>] syntax */
+			return 0;
+	}
+	else if (! expr->type) {
+		return !!(expr->data.type & SDB_TYPE_ARRAY);
+	}
+	return 0;
+} /* sdb_store_expr_iterable */
+
+sdb_store_expr_iter_t *
+sdb_store_expr_iter(sdb_store_expr_t *expr, sdb_store_obj_t *obj,
+		sdb_store_matcher_t *filter)
+{
+	sdb_store_expr_iter_t *iter;
+	sdb_avltree_iter_t *tree = NULL;
+	sdb_data_t array = SDB_DATA_INIT;
+
+	if (! expr)
+		return NULL;
+
+	if (expr->type == TYPED_EXPR) {
+		if (! obj)
+			return NULL;
+		if (obj->type == SDB_HOST) {
+			if (expr->data.data.integer == SDB_SERVICE)
+				tree = sdb_avltree_get_iter(HOST(obj)->services);
+			else if (expr->data.data.integer == SDB_METRIC)
+				tree = sdb_avltree_get_iter(HOST(obj)->metrics);
+			else if (expr->data.data.integer == SDB_ATTRIBUTE)
+				tree = sdb_avltree_get_iter(HOST(obj)->attributes);
+		}
+		else if (obj->type == SDB_SERVICE) {
+			if (expr->data.data.integer == SDB_ATTRIBUTE)
+				tree = sdb_avltree_get_iter(SVC(obj)->attributes);
+		}
+		else if (obj->type == SDB_METRIC) {
+			if (expr->data.data.integer == SDB_ATTRIBUTE)
+				tree = sdb_avltree_get_iter(METRIC(obj)->attributes);
+		}
+	}
+	else if (expr->type == FIELD_VALUE) {
+		if (! obj)
+			return NULL;
+		if (expr->data.data.integer == SDB_FIELD_BACKEND) {
+			/* while scanning the store, we hold a read lock, so it's safe to
+			 * access the data without copying */
+			array.type = SDB_TYPE_ARRAY | SDB_TYPE_STRING;
+			array.data.array.length = obj->backends_num;
+			array.data.array.values = obj->backends;
+		}
+	}
+	else if (! expr->type) {
+		if (expr->data.type & SDB_TYPE_ARRAY)
+			array = expr->data;
+	}
+	else
+		return NULL;
+
+	if ((! tree) && (array.type == SDB_TYPE_NULL))
+		return NULL;
+
+	iter = calloc(1, sizeof(*iter));
+	if (! iter)
+		return NULL;
+
+	sdb_object_ref(SDB_OBJ(obj));
+	sdb_object_ref(SDB_OBJ(expr));
+	sdb_object_ref(SDB_OBJ(filter));
+
+	iter->obj = obj;
+	iter->expr = expr;
+	iter->tree = tree;
+	iter->array = array;
+	iter->filter = filter;
+	return iter;
+} /* sdb_store_expr_iter */
+
+void
+sdb_store_expr_iter_destroy(sdb_store_expr_iter_t *iter)
+{
+	sdb_data_t null = SDB_DATA_INIT;
+
+	if (! iter)
+		return;
+
+	if (iter->tree)
+		sdb_avltree_iter_destroy(iter->tree);
+	iter->tree = NULL;
+
+	iter->array = null;
+	iter->array_idx = 0;
+
+	sdb_object_deref(SDB_OBJ(iter->obj));
+	sdb_object_deref(SDB_OBJ(iter->expr));
+	sdb_object_deref(SDB_OBJ(iter->filter));
+	free(iter);
+} /* sdb_store_expr_iter_destroy */
+
+bool
+sdb_store_expr_iter_has_next(sdb_store_expr_iter_t *iter)
+{
+	if (! iter)
+		return 0;
+
+	if (iter->tree) {
+		/* this function may be called before get_next,
+		 * so we'll have to apply filters here as well */
+		if (iter->filter) {
+			sdb_store_obj_t *child;
+			while ((child = STORE_OBJ(sdb_avltree_iter_peek_next(iter->tree)))) {
+				if (sdb_store_matcher_matches(iter->filter, child, NULL))
+					break;
+				(void)sdb_avltree_iter_get_next(iter->tree);
+			}
+		}
+
+		return sdb_avltree_iter_has_next(iter->tree);
+	}
+
+	return iter->array_idx < iter->array.data.array.length;
+} /* sdb_store_expr_iter_has_next */
+
+sdb_data_t
+sdb_store_expr_iter_get_next(sdb_store_expr_iter_t *iter)
+{
+	sdb_data_t null = SDB_DATA_INIT;
+	sdb_data_t ret = SDB_DATA_INIT;
+	sdb_data_t tmp = SDB_DATA_INIT;
+
+	if (! iter)
+		return null;
+
+	if (iter->tree) {
+		sdb_store_obj_t *child;
+
+		while (42) {
+			child = STORE_OBJ(sdb_avltree_iter_get_next(iter->tree));
+			if (! child)
+				break;
+			if (iter->filter
+					&& (! sdb_store_matcher_matches(iter->filter, child, NULL)))
+				continue;
+
+			if (sdb_store_expr_eval(iter->expr, child, &ret, iter->filter))
+				return null;
+			break;
+		}
+
+		/* Skip over any filtered objects */
+		if (iter->filter) {
+			while ((child = STORE_OBJ(sdb_avltree_iter_peek_next(iter->tree)))) {
+				if (sdb_store_matcher_matches(iter->filter, child, NULL))
+					break;
+				(void)sdb_avltree_iter_get_next(iter->tree);
+			}
+		}
+
+		return ret;
+	}
+
+	if (iter->array_idx >= iter->array.data.array.length)
+		return null;
+
+	++iter->array_idx;
+	if (sdb_data_array_get(&iter->array, iter->array_idx - 1, &ret))
+		return null;
+	if (sdb_data_copy(&tmp, &ret))
+		return null;
+	ret = tmp;
+	return ret;
+} /* sdb_store_expr_iter_get_next */
+
 /* vim: set tw=78 sw=4 ts=4 noexpandtab : */