linux/dbstore.c

   1 #include <stdio.h>
   2 #include <stdlib.h>
   3 #include <stdint.h>
   4 #include <time.h>
   5 #include <ctype.h>
   6 #include <string.h>
   7
   8 #include <glib.h>
   9
  10 #include <mysql/mysql.h>
  11
  12 #include "dbstore.h"
  13
  14 static char *host = NULL;
  15 static char *user = NULL;
  16 static char *pass = NULL;
  17 static char *dbnm = "watermeter";
  18
  19 int dbconfig(char *conffile)
  20 {
  21         FILE *fp = fopen(conffile, "r");
  22         int lc = 0;
  23         int rc = 0;
  24         char buf[128];
  25
  26         if (!fp) return 1;
  27         while (fgets(buf, sizeof(buf), fp)) {
  28                 char *k, *v, *e;
  29
  30                 lc++;
  31                 e = buf + strlen(buf) - 1;
  32                 if (*e == '\n') *e = '\0';
  33                 else {
  34                         g_warning("%s:%d line too long", conffile, lc);
  35                         rc = 1;
  36                         break;
  37                 }
  38                 if ((k = strchr(buf, '#'))) {
  39                         e=k;
  40                         *e = '\0';
  41                 }
  42                 for (k = buf; k < e && isspace(*k); k++) /*nothing*/ ;
  43                 for (v = k; v < e && !isspace(*v)
  44                             && *v != ':' && *v != '='; v++) /*nothing*/ ;
  45                 *v++ = '\0';
  46                 if (*k == '\0') continue; /* empty or comment-only line */
  47                 for (; v < e && (isspace(*v) || *v == ':' || *v == '=')
  48                                                         ; v++) /*nothing*/ ;
  49                 if (v >= e) {
  50                         g_warning("%s:%d no value for key \"%s\"",
  51                                         conffile, lc, k);
  52                         rc = 1;
  53                         break;
  54                 }
  55 #ifdef TEST_CONFIG
  56                 printf("k=%s v=%s\n", k, v);
  57 #endif
  58                 if      (!strcmp(k, "host"))     host = strdup(v);
  59                 else if (!strcmp(k, "user"))     user = strdup(v);
  60                 else if (!strcmp(k, "password")) pass = strdup(v);
  61                 else if (!strcmp(k, "database")) dbnm = strdup(v);
  62                 else {
  63                         g_warning("%s:%d unrecognized key \"%s\"",
  64                                         conffile, lc, k);
  65                         rc = 1;
  66                         break;
  67                 }
  68         }
  69         return rc;
  70 }
  71
  72 int dbstore(uint8_t which, uint32_t val)
  73 {
  74         time_t t;
  75         int i;
  76         MYSQL mysql;
  77         int rc = 0;
  78         struct tm tm;
  79         char buf[64];
  80         char tstr[32];
  81         char *table = (which == 1) ? "cold" : "hot";
  82         char statement[64];
  83         MYSQL_RES *result;
  84         uint32_t prev_val = 0;
  85
  86         t = time(NULL);
  87         (void)gmtime_r(&t, &tm);
  88         (void)strftime(tstr, sizeof(tstr), "%Y-%m-%d %H:%M:%S", &tm);
  89         mysql_init(&mysql);
  90         if(!mysql_real_connect(&mysql, host, user, pass, dbnm, 0, NULL, 0)) {
  91                 g_warning("mysql connect error: %s\n", mysql_error(&mysql));
  92                 return 1;
  93         }
  94         mysql_autocommit(&mysql, FALSE);
  95         /* ======== */
  96         snprintf(statement, sizeof(statement),
  97                  "select value from %scnt order by timestamp desc limit 1;\n",
  98                  table);
  99         if ((rc = mysql_query(&mysql, statement)))
 100                 g_warning("mysql \"%s\" error: %s\n",
 101                                 statement, mysql_error(&mysql));
 102         else if ((result = mysql_store_result(&mysql))){
 103                 MYSQL_ROW row = mysql_fetch_row(result);
 104                 if (row && *row) prev_val = atoi(*row);
 105                 mysql_free_result(result);
 106         }
 107         if (val < prev_val) {
 108                 snprintf(statement, sizeof(statement),
 109                          "insert into %sadj values (\"%s\",%u);\n",
 110                          table, tstr, prev_val);
 111                 g_info("%s %u <= %u, %s", table, val, prev_val, statement);
 112                 if ((rc = mysql_query(&mysql, statement)))
 113                         g_warning("mysql \"%s\" error: %s\n",
 114                                         statement, mysql_error(&mysql));
 115         }
 116         snprintf(statement, sizeof(statement),
 117                  "insert into %scnt values (\"%s\",%u);\n",
 118                  table, tstr, val);
 119         if ((rc = mysql_query(&mysql, statement)))
 120                 g_warning("mysql \"%s\" error: %s\n",
 121                                 statement, mysql_error(&mysql));
 122         /* ======== */
 123         if (!rc) {
 124                 if ((rc = mysql_commit(&mysql)))
 125                         g_warning("mysql commit error: %s\n",
 126                                         mysql_error(&mysql));
 127         }
 128         mysql_close(&mysql);
 129         return rc;
 130 }
 131
 132 #ifdef TEST_CONFIG
 133 int main(int const argc, char *argv[])
 134 {
 135         if (dbconfig(argv[1])) {
 136                 printf("could not parse config file\n");
 137                 return 1;
 138         }
 139         printf("host: %s\nuser: %s\npass: %s\ndbnm: %s\n",
 140                 host, user, pass, dbnm);
 141         return 0;
 142 }
 143 #endif