From c4265fb5a792db45b84ed3e2f93b871ac51af3d5 Mon Sep 17 00:00:00 2001
From: Florian Forster <octo@leeloo.lan.home.verplant.org>
Date: Tue, 3 Feb 2009 21:34:36 +0100
Subject: [PATCH] src/utils_db_query.[ch]: Add support for min-/max-version.

These settings can be used to check if a particular query is suited to the
database server version we're currently connected to.
---
 src/utils_db_query.c | 41 +++++++++++++++++++++++++++++++++++++++++
 src/utils_db_query.h |  2 ++
 2 files changed, 43 insertions(+)

diff --git a/src/utils_db_query.c b/src/utils_db_query.c
index 27e8a55f..1bd5b216 100644
--- a/src/utils_db_query.c
+++ b/src/utils_db_query.c
@@ -55,6 +55,9 @@ struct udb_query_s /* {{{ */
   char *statement;
   void *user_data;
 
+  unsigned int min_version;
+  unsigned int max_version;
+
   /* Preparation area */
   size_t column_num;
   char *host;
@@ -144,6 +147,27 @@ static int udb_config_add_string (char ***ret_array, /* {{{ */
   return (0);
 } /* }}} int udb_config_add_string */
 
+static int udb_config_set_uint (unsigned int *ret_value, /* {{{ */
+    oconfig_item_t *ci)
+{
+  double tmp;
+
+  if ((ci->values_num != 1)
+      || (ci->values[0].type != OCONFIG_TYPE_NUMBER))
+  {
+    WARNING ("db query utils: The `%s' config option "
+        "needs exactly one numeric argument.", ci->key);
+    return (-1);
+  }
+
+  tmp = ci->values[0].value.number;
+  if ((tmp < 0.0) || (tmp > ((double) UINT_MAX)))
+    return (-ERANGE);
+
+  *ret_value = (unsigned int) (tmp + .5);
+  return (0);
+} /* }}} int udb_config_set_uint */
+
 /*
  * Result private functions
  */
@@ -545,6 +569,8 @@ int udb_query_create (udb_query_t ***ret_query_list, /* {{{ */
     return (-1);
   }
   memset (q, 0, sizeof (*q));
+  q->min_version = 0;
+  q->max_version = UINT_MAX;
 
   status = udb_config_set_string (&q->name, ci);
   if (status != 0)
@@ -562,6 +588,10 @@ int udb_query_create (udb_query_t ***ret_query_list, /* {{{ */
       status = udb_config_set_string (&q->statement, child);
     else if (strcasecmp ("Result", child->key) == 0)
       status = udb_result_create (q->name, &q->results, child);
+    else if (strcasecmp ("MinVersion", child->key) == 0)
+      status = udb_config_set_uint (&q->min_version, child);
+    else if (strcasecmp ("MaxVersion", child->key) == 0)
+      status = udb_config_set_uint (&q->max_version, child);
     else
     {
       WARNING ("db query utils: Query `%s': Option `%s' not allowed here.",
@@ -726,6 +756,17 @@ void *udb_query_get_user_data (udb_query_t *q) /* {{{ */
   return (q->user_data);
 } /* }}} void *udb_query_get_user_data */
 
+int udb_query_check_version (udb_query_t *q, unsigned int version) /* {{{ */
+{
+  if (q == NULL)
+    return (-EINVAL);
+
+  if ((version < q->min_version) || (version > q->max_version))
+    return (0);
+
+  return (1);
+} /* }}} int udb_query_check_version */
+
 void udb_query_finish_result (udb_query_t *q) /* {{{ */
 {
   udb_result_t *r;
diff --git a/src/utils_db_query.h b/src/utils_db_query.h
index 87c3e4e4..4e12cedd 100644
--- a/src/utils_db_query.h
+++ b/src/utils_db_query.h
@@ -47,6 +47,8 @@ const char *udb_query_get_statement (udb_query_t *q);
 void  udb_query_set_user_data (udb_query_t *q, void *user_data);
 void *udb_query_get_user_data (udb_query_t *q);
 
+int udb_query_check_version (udb_query_t *q, unsigned int version);
+
 int udb_query_prepare_result (udb_query_t *q,
     const char *host, const char *plugin, const char *db_name,
     char **column_names, size_t column_num);
-- 
2.11.0