demo/demo_cxx_mysql.cpp

   1 #include "../impl/mysql_traits.hpp" // mysql traits
   2 #include "demo_cxx_gen_mysql.hpp" // generated
   3 #include <iostream>
   4 #include <vector>
   5
   6 using namespace std;
   7
   8 typedef sqlgg<mysql_traits> gen_t;
   9 typedef long long int64;
  10
  11 MYSQL* connect()
  12 {
  13   MYSQL* conn = mysql_init (NULL);
  14   if (conn == NULL)
  15   {
  16     fprintf (stderr, "mysql_init() failed (probably out of memory)\n");
  17     return NULL;
  18   }
  19   if (mysql_real_connect (
  20         conn,          /* pointer to connection handler */
  21         NULL, /* host to connect to */
  22         "root", /* user name */
  23         NULL,  /* password */
  24         "test",   /* database to use */
  25         0,             /* port (use default) */
  26         NULL,          /* socket (use default) */
  27         0)             /* flags (none) */
  28       == NULL)
  29   {
  30     fprintf (stderr, "mysql_real_connect() failed:\nError %u (%s)\n",
  31               mysql_errno (conn), mysql_error (conn));
  32     return NULL;
  33   }
  34   return conn;
  35 }
  36
  37 struct output_transfers
  38 {
  39   void operator()(std::string const& fullname, int total)
  40   {
  41      cout << fullname << " = " << total << endl;
  42   }
  43 };
  44
  45 struct output_donors
  46 {
  47   void operator()(std::string const& surname)
  48   {
  49      cout << surname << endl;
  50   }
  51 };
  52
  53 int main()
  54 {
  55   MYSQL* db = connect();
  56   if (!db) return 1;
  57
  58   gen_t gen(db);
  59
  60   mysql_query(db,"DROP TABLE person");
  61   mysql_query(db,"DROP TABLE money");
  62
  63   // create tables
  64   gen.create_person();
  65   gen.create_money();
  66
  67   // add all person records
  68   gen.add_person("John","Black");
  69   int64 john = mysql_insert_id(db);
  70   gen.add_person("Ivan","Petrov");
  71   int64 ivan = mysql_insert_id(db);
  72   gen.add_person("Sancho","Alvares");
  73   int64 sancho = mysql_insert_id(db);
  74
  75   // add money relations
  76   gen.add_money(john,ivan,200);
  77   gen.add_money(john,sancho,100);
  78   gen.add_money(john,sancho,250);
  79   gen.add_money(sancho,ivan,300);
  80
  81   // summarize by person
  82   cout << "Total transfers:" << endl;
  83   gen.calc_total(output_transfers());
  84
  85   // list donors
  86   cout << "Donors:" << endl;
  87   gen.list_donors("petrov",100,output_donors());
  88
  89   // properly close database
  90   mysql_close(db);
  91   db = NULL;
  92
  93   return 0;
  94 }