sqlite3-statement.hpp

/* -*- Mode:C++; c-file-style:"gnu"; indent-tabs-mode:nil -*- */
#ifndef NDN_SQLITE3_STATEMENT_HPP
#define NDN_SQLITE3_STATEMENT_HPP

// Only compile if ndn-cpp-config.h defines NDN_CPP_HAVE_SQLITE3.
#include <ndn-cpp/ndn-cpp-config.h>
#ifdef NDN_CPP_HAVE_SQLITE3

#include <string>
#include <sqlite3.h>
#include <ndn-cpp/util/blob.hpp>

namespace ndn {

/*
 * Sqlite3Statement is a utility class to wrap an SQLite3 prepared statement,
 * provide access methods, and finalize the statement in the destructor.
 */
class Sqlite3Statement
{
public:
  Sqlite3Statement(sqlite3* database, const std::string& statement);

  ~Sqlite3Statement();

  int
  bind(int index, const char* value, size_t size)
  {
    return sqlite3_bind_text(statement_, index, value, size, SQLITE_TRANSIENT);
  }

  int
  bind(int index, const std::string& value)
  {
    return bind(index, value.c_str(), value.size());
  }

  int
  bind(int index, const uint8_t* buf, size_t size, bool bufIsStatic = false)
  {
    return sqlite3_bind_blob
      (statement_, index, buf, size,
       bufIsStatic ? SQLITE_STATIC : SQLITE_TRANSIENT);
  }

  int
  bind(int index, const Blob& blob, bool bufIsStatic = false)
  {
    return bind(index, blob.buf(), blob.size(), bufIsStatic);
  }

  int
  bind(int index, int integer)
  {
    return sqlite3_bind_int(statement_, index, integer);
  }

  std::string
  getString(int column)
  {
    return std::string
      (reinterpret_cast<const char*>(sqlite3_column_text(statement_, column)),
       sqlite3_column_bytes(statement_, column));
  }

  int
  getInt(int column) { return sqlite3_column_int(statement_, column); }

  const uint8_t*
  getBuf(int column)
  {
    return static_cast<const uint8_t*>(sqlite3_column_blob(statement_, column));
  }

  int
  getSize(int column) { return sqlite3_column_bytes(statement_, column); }

  Blob
  getBlob(int column) { return Blob(getBuf(column), getSize(column)); }

  int
  step() { return sqlite3_step(statement_); }

private:
  // Disable the copy constructor and assignment operator.
  Sqlite3Statement(const Sqlite3Statement& other);
  Sqlite3Statement& operator=(const Sqlite3Statement& other);

  sqlite3_stmt* statement_;
};

}

#endif // NDN_CPP_HAVE_SQLITE3

#endif