Code

store: Added sdb_store_get_field().
[sysdb.git] / src / core / store_lookup.c
index 476e0e717d9fb5c788651722d5e2a4948dcd696e..1ec1dbb68e702f6c5e4a9722e885f1042494cf38 100644 (file)
@@ -128,6 +128,42 @@ attr_cmp(sdb_store_obj_t *obj, sdb_store_cond_t *cond,
        return status;
 } /* attr_cmp */
 
+static int
+obj_cmp(sdb_store_obj_t *obj, sdb_store_cond_t *cond,
+               sdb_store_matcher_t __attribute__((unused)) *filter)
+{
+       sdb_data_t obj_value = SDB_DATA_INIT;
+       sdb_data_t value = SDB_DATA_INIT;
+       int status;
+
+       if (sdb_store_get_field(obj, OBJ_C(cond)->field, &obj_value))
+               return INT_MAX;
+       if (sdb_store_expr_eval(OBJ_C(cond)->expr, &value))
+               return INT_MAX;
+
+       if (obj_value.type != value.type) {
+               sdb_data_free_datum(&value);
+               return INT_MAX;
+       }
+       else if (OBJ_C(cond)->field == SDB_FIELD_BACKEND) {
+               /* this implementation is not actually a conditional but rather checks
+                * for equality (or rather, existence) only */
+               size_t i;
+               status = INT_MAX;
+               for (i = 0; i < obj->backends_num; ++i) {
+                       if (! strcasecmp(obj->backends[i], value.data.string)) {
+                               status = 0;
+                               break;
+                       }
+               }
+       }
+       else {
+               status = sdb_data_cmp(&obj_value, &value);
+       }
+       sdb_data_free_datum(&value);
+       return status;
+} /* obj_cmp */
+
 /*
  * matcher implementations
  */
@@ -356,6 +392,32 @@ static sdb_type_t attr_cond_type = {
        /* destroy = */ attr_cond_destroy,
 };
 
+static int
+obj_cond_init(sdb_object_t *obj, va_list ap)
+{
+       int field = va_arg(ap, int);
+       sdb_store_expr_t *expr = va_arg(ap, sdb_store_expr_t *);
+
+       SDB_STORE_COND(obj)->cmp = obj_cmp;
+
+       OBJ_C(obj)->field = field;
+       OBJ_C(obj)->expr = expr;
+       sdb_object_ref(SDB_OBJ(expr));
+       return 0;
+} /* obj_cond_init */
+
+static void
+obj_cond_destroy(sdb_object_t *obj)
+{
+       sdb_object_deref(SDB_OBJ(OBJ_C(obj)->expr));
+} /* obj_cond_destroy */
+
+static sdb_type_t obj_cond_type = {
+       /* size = */ sizeof(obj_cond_t),
+       /* init = */ obj_cond_init,
+       /* destroy = */ obj_cond_destroy,
+};
+
 /*
  * private matcher types
  */
@@ -495,21 +557,34 @@ cond_matcher_destroy(sdb_object_t *obj)
 static char *
 cond_tostring(sdb_store_matcher_t *m, char *buf, size_t buflen)
 {
+       const char *type, *id;
+       sdb_data_t value = SDB_DATA_INIT;
+       char value_str[buflen];
+       sdb_store_expr_t *expr;
+
        if (COND_M(m)->cond->cmp == attr_cmp) {
-               sdb_data_t value = SDB_DATA_INIT;
-               char value_str[buflen];
-               if (sdb_store_expr_eval(ATTR_C(COND_M(m)->cond)->expr, &value))
-                       snprintf(value_str, sizeof(value_str), "ERR");
-               else if (sdb_data_format(&value, value_str, sizeof(value_str),
-                                       SDB_SINGLE_QUOTED) < 0)
-                       snprintf(value_str, sizeof(value_str), "ERR");
-               snprintf(buf, buflen, "ATTR[%s]{ %s %s }",
-                               ATTR_C(COND_M(m)->cond)->name, MATCHER_SYM(m->type),
-                               value_str);
-               sdb_data_free_datum(&value);
+               type = "ATTR";
+               id = ATTR_C(COND_M(m)->cond)->name;
+               expr = ATTR_C(COND_M(m)->cond)->expr;
        }
-       else
+       else if (COND_M(m)->cond->cmp == obj_cmp) {
+               type = "OBJ";
+               id = SDB_FIELD_TO_NAME(OBJ_C(COND_M(m)->cond)->field);
+               expr = OBJ_C(COND_M(m)->cond)->expr;
+       }
+       else {
                snprintf(buf, buflen, "<unknown>");
+               return buf;
+       }
+
+       if (sdb_store_expr_eval(expr, &value))
+               snprintf(value_str, sizeof(value_str), "ERR");
+       else if (sdb_data_format(&value, value_str, sizeof(value_str),
+                               SDB_SINGLE_QUOTED) < 0)
+               snprintf(value_str, sizeof(value_str), "ERR");
+       snprintf(buf, buflen, "%s[%s]{ %s %s }", type, id,
+                       MATCHER_SYM(m->type), value_str);
+       sdb_data_free_datum(&value);
        return buf;
 } /* cond_tostring */
 
@@ -696,6 +771,13 @@ sdb_store_attr_cond(const char *name, sdb_store_expr_t *expr)
                                name, expr));
 } /* sdb_store_attr_cond */
 
+sdb_store_cond_t *
+sdb_store_obj_cond(int field, sdb_store_expr_t *expr)
+{
+       return SDB_STORE_COND(sdb_object_create("obj-cond", obj_cond_type,
+                               field, expr));
+} /* sdb_store_obj_cond */
+
 sdb_store_matcher_t *
 sdb_store_name_matcher(int type, const char *name, _Bool re)
 {
@@ -772,6 +854,44 @@ sdb_store_isnull_matcher(const char *attr_name)
                                MATCHER_ISNULL, attr_name));
 } /* sdb_store_isnull_matcher */
 
+static sdb_store_matcher_t *
+maybe_inv_matcher(sdb_store_matcher_t *m, _Bool inv)
+{
+       sdb_store_matcher_t *tmp;
+
+       if ((! m) || (! inv))
+               return m;
+
+       tmp = sdb_store_inv_matcher(m);
+       /* pass ownership to the inverse matcher */
+       sdb_object_deref(SDB_OBJ(m));
+       return tmp;
+} /* maybe_inv_matcher */
+
+static int
+parse_cond_op(const char *op,
+               sdb_store_matcher_t *(**matcher)(sdb_store_cond_t *), _Bool *inv)
+{
+       *inv = 0;
+       if (! strcasecmp(op, "<"))
+               *matcher = sdb_store_lt_matcher;
+       else if (! strcasecmp(op, "<="))
+               *matcher = sdb_store_le_matcher;
+       else if (! strcasecmp(op, "="))
+               *matcher = sdb_store_eq_matcher;
+       else if (! strcasecmp(op, ">="))
+               *matcher = sdb_store_ge_matcher;
+       else if (! strcasecmp(op, ">"))
+               *matcher = sdb_store_gt_matcher;
+       else if (! strcasecmp(op, "!=")) {
+               *matcher = sdb_store_eq_matcher;
+               *inv = 1;
+       }
+       else
+               return -1;
+       return 0;
+} /* parse_cond_op */
+
 static sdb_store_matcher_t *
 parse_attr_cmp(const char *attr, const char *op, sdb_store_expr_t *expr)
 {
@@ -791,21 +911,7 @@ parse_attr_cmp(const char *attr, const char *op, sdb_store_expr_t *expr)
        }
        else if (! expr)
                return NULL;
-       else if (! strcasecmp(op, "<"))
-               matcher = sdb_store_lt_matcher;
-       else if (! strcasecmp(op, "<="))
-               matcher = sdb_store_le_matcher;
-       else if (! strcasecmp(op, "="))
-               matcher = sdb_store_eq_matcher;
-       else if (! strcasecmp(op, ">="))
-               matcher = sdb_store_ge_matcher;
-       else if (! strcasecmp(op, ">"))
-               matcher = sdb_store_gt_matcher;
-       else if (! strcasecmp(op, "!=")) {
-               matcher = sdb_store_eq_matcher;
-               inv = 1;
-       }
-       else
+       else if (parse_cond_op(op, &matcher, &inv))
                return NULL;
 
        cond = sdb_store_attr_cond(attr, expr);
@@ -815,17 +921,7 @@ parse_attr_cmp(const char *attr, const char *op, sdb_store_expr_t *expr)
        m = matcher(cond);
        /* pass ownership to 'm' or destroy in case of an error */
        sdb_object_deref(SDB_OBJ(cond));
-       if (! m)
-               return NULL;
-
-       if (inv) {
-               sdb_store_matcher_t *tmp;
-               tmp = sdb_store_inv_matcher(m);
-               /* pass ownership to the inverse matcher */
-               sdb_object_deref(SDB_OBJ(m));
-               m = tmp;
-       }
-       return m;
+       return maybe_inv_matcher(m, inv);
 } /* parse_attr_cmp */
 
 sdb_store_matcher_t *
@@ -883,19 +979,47 @@ sdb_store_matcher_parse_cmp(const char *obj_type, const char *attr,
                m = sdb_store_attr_matcher(attr, value.data.string, re);
 
        sdb_data_free_datum(&value);
+       return maybe_inv_matcher(m, inv);
+} /* sdb_store_matcher_parse_cmp */
 
-       if (! m)
+sdb_store_matcher_t *
+sdb_store_matcher_parse_field_cmp(const char *name, const char *op,
+               sdb_store_expr_t *expr)
+{
+       sdb_store_matcher_t *(*matcher)(sdb_store_cond_t *) = NULL;
+       sdb_store_matcher_t *m;
+       sdb_store_cond_t *cond;
+       _Bool inv = 0;
+
+       int field;
+
+       if (! expr)
                return NULL;
 
-       if (inv) {
-               sdb_store_matcher_t *tmp;
-               tmp = sdb_store_inv_matcher(m);
-               /* pass ownership to the inverse matcher */
-               sdb_object_deref(SDB_OBJ(m));
-               m = tmp;
-       }
-       return m;
-} /* sdb_store_matcher_parse_cmp */
+       if (! strcasecmp(name, "last_update"))
+               field = SDB_FIELD_LAST_UPDATE;
+       else if (! strcasecmp(name, "age"))
+               field = SDB_FIELD_AGE;
+       else if (! strcasecmp(name, "interval"))
+               field = SDB_FIELD_INTERVAL;
+       else if (! strcasecmp(name, "backend"))
+               field = SDB_FIELD_BACKEND;
+       else
+               return NULL;
+
+       if (parse_cond_op(op, &matcher, &inv))
+               return NULL;
+
+       cond = sdb_store_obj_cond(field, expr);
+       if (! cond)
+               return NULL;
+
+       assert(matcher);
+       m = matcher(cond);
+       /* pass ownership to 'm' or destroy in case of an error */
+       sdb_object_deref(SDB_OBJ(cond));
+       return maybe_inv_matcher(m, inv);
+} /* sdb_store_matcher_parse_field_cmp */
 
 sdb_store_matcher_t *
 sdb_store_dis_matcher(sdb_store_matcher_t *left, sdb_store_matcher_t *right)