test2.cc

Go to the documentation of this file.

#include "art/Framework/IO/Root/RootDB/tkeyvfs.h"

extern "C" {
#include <sqlite3.h>
}

#include <cstdio>
#include <cstdlib>
#include <cstring>

#ifndef TKEYVFS_NO_ROOT
#include "TFile.h"
#endif

using namespace std;

static int
callback(void*, int cnt, char** vals, char** col_name)
{
  int i = 0;
  for (i = 0; i < cnt; ++i) {
    if (vals[i]) {
      printf("%s: %s\n", col_name[i], vals[i]);
    } else {
      printf("%s: %s\n", col_name[i], "NULL");
    }
  }
  return 0;
}

int
main(int argc, char** argv)
{
  sqlite3* db = 0;
  char* error_msg = 0;
  int err = 0;
  if (argc < 4 || argc > 5) {
    fprintf(stderr,
            "usage: %s: <op> <db-filename> <db-tkeyname> [<sql-statement>]\n",
            argv[0]);
    exit(1);
  }
  tkeyvfs_init();
#ifndef TKEYVFS_NO_ROOT
  TFile* rootFile = 0;
#endif
  if (!strcmp(argv[1], "r")) {
#ifndef TKEYVFS_NO_ROOT
    rootFile = new TFile(argv[2]);
    err = tkeyvfs_open_v2(argv[3],
#else
    err = tkeyvfs_open_v2(argv[2],
#endif
                          &db,
                          SQLITE_OPEN_READONLY
#ifndef TKEYVFS_NO_ROOT
                          ,
                          rootFile
#endif
    );
  } else if (!strcmp(argv[1], "u")) {
#ifndef TKEYVFS_NO_ROOT
    rootFile = new TFile(argv[2], "UPDATE");
    err = tkeyvfs_open_v2(argv[3],
#else
    err = tkeyvfs_open_v2(argv[2],
#endif
                          &db,
                          SQLITE_OPEN_READWRITE
#ifndef TKEYVFS_NO_ROOT
                          ,
                          rootFile
#endif
    );
  } else if (!strcmp(argv[1], "w")) {
#ifndef TKEYVFS_NO_ROOT
    rootFile = new TFile(argv[2], "RECREATE");
    err = tkeyvfs_open_v2(argv[3],
#else
    err = tkeyvfs_open_v2(argv[2],
#endif
                          &db,
                          SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE
#ifndef TKEYVFS_NO_ROOT
                          ,
                          rootFile
#endif
    );
  } else {
    fprintf(stderr, "Unrecognized file mode designator %s", argv[1]);
    exit(1);
  }
  if (err) {
    fprintf(stderr, "Can't open database: %s\n", sqlite3_errmsg(db));
    sqlite3_close(db);
    exit(1);
  }
  if (argc == 5) {
    err = sqlite3_exec(db, argv[4], callback, 0, &error_msg);
  } else {
    char* buf = 0;
    size_t n = 0;
    getline(&buf, &n, stdin);
    err = sqlite3_exec(db, buf, callback, 0, &error_msg);
    free(buf);
  }
  if (err != SQLITE_OK) {
    fprintf(stderr, "SQL error: %s\n", error_msg);
    sqlite3_free(error_msg);
  }
  sqlite3_close(db);
  return 0;
}