vxodbc/vxhelpers.h

   1 #ifndef __VXHELPERS_H
   2 #define __VXHELPERS_H
   3
   4 #include <wvdbusconn.h>
   5 #include "pgtypes.h"
   6
   7
   8 class VxResultSet
   9 {
  10     int maxcol;
  11
  12     int vxtype_to_pgtype(WvStringParm vxtype)
  13     {
  14         if (vxtype == "String")
  15             return PG_TYPE_VARCHAR;
  16         else if (vxtype == "Int64")
  17             return PG_TYPE_INT8;
  18         else if (vxtype == "Int32")
  19             return PG_TYPE_INT4;
  20         else if (vxtype == "Int16")
  21             return PG_TYPE_INT2;
  22         else if (vxtype == "UInt8")
  23             return PG_TYPE_CHAR;
  24         else if (vxtype == "Bool")
  25             return PG_TYPE_BOOL;
  26         else if (vxtype == "Double")
  27             return PG_TYPE_FLOAT8;
  28         else if (vxtype == "Uuid")
  29             return PG_TYPE_VARCHAR;
  30         else if (vxtype == "Binary")
  31             return PG_TYPE_BYTEA;
  32         else if (vxtype == "DateTime")
  33             return VX_TYPE_DATETIME;
  34         else if (vxtype == "Decimal")
  35             return PG_TYPE_NUMERIC;
  36     }
  37 public:
  38     QResultClass *res;
  39
  40     VxResultSet()
  41     {
  42         res = QR_Constructor();
  43         maxcol = -1;
  44         assert(res);
  45         assert(*this == res);
  46     }
  47
  48     void set_field_info(int col, const char *colname, OID type, int typesize)
  49     {
  50         mylog("Col#%d is '%s', type=%d, size=%d\n", col, colname, type, typesize);
  51         if (maxcol < col)
  52         {
  53             // FIXME: This will destroy old data
  54             mylog("!!!!!! Resizing colinfo array, destroying data !!!!!!\n");
  55             maxcol = col;
  56             QR_set_num_fields(res, maxcol+1);
  57         }
  58         QR_set_field_info_v(res, col, colname, type, typesize);
  59         assert(!strcmp(QR_get_fieldname(res, col), colname));
  60     }
  61
  62     operator QResultClass* () const
  63     {
  64         return res;
  65     }
  66
  67     int numcols() const
  68     {
  69         return maxcol+1;
  70     }
  71
  72     void runquery(WvDBusConn &conn, const char *func, const char *query)
  73     {
  74         WvDBusMsg msg("vx.versaplexd", "/db", "vx.db", func);
  75         msg.append(query);
  76         WvDBusMsg reply = conn.send_and_wait(msg, 50000);
  77
  78         if (reply.iserror())
  79             mylog("DBus error: '%s'\n", ((WvString)reply).cstr());
  80         else
  81         {
  82             WvDBusMsg::Iter top(reply);
  83             WvDBusMsg::Iter colinfo(top.getnext().open());
  84             WvDBusMsg::Iter data(top.getnext().open().getnext().open());
  85             WvDBusMsg::Iter flags(top.getnext().open());
  86
  87             int num_cols;
  88             // Sadly there's no better way to get the number of columns
  89             for (num_cols = 0; colinfo.next(); num_cols++) { }
  90             colinfo.rewind();
  91             // Allocate space for the column info data.
  92             QR_set_num_fields(res, num_cols);
  93             maxcol = num_cols - 1;
  94
  95             for (int colnum = 0; colinfo.next(); colnum++)
  96             {
  97                 WvDBusMsg::Iter i(colinfo.open());
  98
  99                 int colsize = i.getnext();
 100                 WvString colname = i.getnext();
 101                 int coltype = vxtype_to_pgtype(i.getnext());
 102
 103                 set_field_info(colnum, colname, coltype, colsize);
 104             }
 105
 106             for (data.rewind(); data.next(); )
 107             {
 108                 TupleField *tuple = QR_AddNew(res);
 109
 110                 WvDBusMsg::Iter cols(data.open());
 111                 for (int colnum = 0;
 112                        cols.next() && colnum < numcols();
 113                        colnum++)
 114                     set_tuplefield_string(&tuple[colnum], *cols);
 115             }
 116         }
 117     }
 118 };
 119
 120
 121 class VxStatement
 122 {
 123     StatementClass *stmt;
 124     RETCODE ret;
 125 public:
 126     VxStatement(StatementClass *_stmt)
 127     {
 128         stmt = _stmt;
 129         ret = SQL_SUCCESS;
 130     }
 131
 132     ~VxStatement()
 133     {
 134         /*
 135          * things need to think that this statement is finished so the
 136          * results can be retrieved.
 137          */
 138         stmt->status = STMT_FINISHED;
 139
 140         /* set up the current tuple pointer for SQLFetch */
 141         stmt->currTuple = -1;
 142         SC_set_rowset_start(stmt, -1, FALSE);
 143         SC_set_current_col(stmt, -1);
 144
 145         if (stmt->internal)
 146             ret = DiscardStatementSvp(stmt, ret, FALSE);
 147     }
 148
 149     void reinit()
 150     {
 151         ret = SC_initialize_and_recycle(stmt);
 152     }
 153
 154     void set_result(VxResultSet &rs)
 155     {
 156         SC_set_Result(stmt, rs);
 157
 158         /* the binding structure for a statement is not set up until
 159          * a statement is actually executed, so we'll have to do this
 160          * ourselves.
 161          */
 162         extend_column_bindings(SC_get_ARDF(stmt), rs.numcols());
 163     }
 164
 165     RETCODE retcode() const
 166     {
 167         return ret;
 168     }
 169
 170     void seterr()
 171     {
 172         ret = SQL_ERROR;
 173     }
 174
 175     void setok()
 176     {
 177         ret = SQL_SUCCESS;
 178     }
 179
 180     bool isok() const
 181     {
 182         return ret == SQL_SUCCESS;
 183     }
 184
 185     WvDBusConn &dbus()
 186     {
 187         WvDBusConn *db = SC_get_conn(stmt)->dbus;
 188         assert(db);
 189         return *db;
 190     }
 191 };
 192
 193 #endif // __VXHELPERS_H