impl/pqxx_traits.hpp

   1 /*
   2   Postgresql/pqxx C++ traits for sqlgg
   3   by ygrek
   4   2014-06-08
   5
   6   This is free and unencumbered software released into the public domain.
   7
   8   Anyone is free to copy, modify, publish, use, compile, sell, or
   9   distribute this software, either in source code form or as a compiled
  10   binary, for any purpose, commercial or non-commercial, and by any
  11   means.
  12
  13   For more information, please refer to <http://unlicense.org/>
  14 */
  15
  16 #include <pqxx/pqxx>
  17 #define SQLGG_STR(x) x
  18
  19 #include <assert.h>
  20 #include <string.h>
  21 #include <stdlib.h>
  22
  23 #include <string>
  24 #include <vector>
  25
  26 #if defined(SQLGG_DEBUG)
  27 #include <iostream>
  28 using namespace std;
  29 #endif
  30
  31 struct pqxx_traits
  32 {
  33   typedef int Int;
  34   typedef std::string Text;
  35   typedef Text Any;
  36
  37   typedef pqxx::result::const_iterator row;
  38   typedef pqxx::work& connection;
  39
  40   static void get_column(row r, int index, Int& data)
  41   {
  42     // nothing
  43   }
  44
  45   static void get_column(row r, int index, Text& data)
  46   {
  47   }
  48
  49   typedef pqxx::prepare::declaration stmt_decl;
  50
  51   static void set_param(stmt_decl const& stmt, const Text& val, int index)
  52   {
  53     stmt("varchar",pqxx::prepare::treat_string);
  54   }
  55
  56   static void set_param(stmt_decl const& stmt, const Int& val, int index)
  57   {
  58     stmt("INTEGER",pqxx::prepare::treat_direct);
  59   }
  60
  61   template<typename T>
  62   static void set_param(pqxx::prepare::invocation& stmt, const T& val, int index)
  63   {
  64     stmt(val);
  65   }
  66
  67   template<class Container, class Binder, class Params>
  68   static bool do_select(connection db, Container& result, const char* sql, Binder binder, Params params)
  69   {
  70     const char* name = "sqlgg_stmt";
  71     cout << "start prepare" << endl;
  72     pqxx::prepare::declaration decl = db.conn().prepare(name,sql);
  73     params.set_params(decl);
  74
  75     cout << "start invoke" << endl;
  76     pqxx::prepare::invocation call = db.prepared(name);
  77     cout << "set params" << endl;
  78     params.set_params(call);
  79     cout << "exec" << endl;
  80     result = call.exec();
  81     cout << "execed" << endl;
  82
  83     db.conn().unprepare(name);
  84
  85     return true;
  86   }
  87
  88   struct no_params
  89   {
  90     void set_params(pqxx::prepare::declaration const&) {}
  91     void set_params(pqxx::prepare::invocation&) {}
  92     enum { count = 0 };
  93   };
  94
  95   template<class T>
  96   struct no_binder
  97   {
  98     void get(row,T&) {}
  99     void bind(row,T&) {}
 100     enum { count = 0 };
 101   };
 102
 103   template<class Params>
 104   static bool do_execute(connection db, const char* sql, Params params)
 105   {
 106     pqxx::result R;
 107     return do_select(db,R,sql,no_binder<int>(),params);
 108   }
 109
 110 }; // mysql_traits
 111