soci-db2.h 7.82 KB
Newer Older
Denis Chapligin's avatar
Denis Chapligin committed
1
//
Mateusz Loskot's avatar
Mateusz Loskot committed
2
// Copyright (C) 2011-2013 Denis Chapligin
Denis Chapligin's avatar
Denis Chapligin committed
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26
// Copyright (C) 2004-2006 Maciej Sobczak, Stephen Hutton
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
//

#ifndef SOCI_DB2_H_INCLUDED
#define SOCI_DB2_H_INCLUDED

#ifdef _WIN32
# ifdef SOCI_DLL
#  ifdef SOCI_DB2_SOURCE
#   define SOCI_DB2_DECL __declspec(dllexport)
#  else
#   define SOCI_DB2_DECL __declspec(dllimport)
#  endif // SOCI_DB2_SOURCE
# endif // SOCI_DLL
#endif // _WIN32
//
// If SOCI_DB2_DECL isn't defined yet define it now
#ifndef SOCI_DB2_DECL
# define SOCI_DB2_DECL
#endif

27
#include <soci/soci-backend.h>
Denis Chapligin's avatar
Denis Chapligin committed
28 29 30

#include <cstddef>
#include <string>
Denis Chapligin's avatar
Denis Chapligin committed
31 32 33
#include <vector>
#include <iostream>
#include <sstream>
34
#include <string>
Denis Chapligin's avatar
Denis Chapligin committed
35
#include <cstring>
Denis Chapligin's avatar
Denis Chapligin committed
36 37 38 39 40

#include <sqlcli1.h>

namespace soci
{
41 42 43 44 45 46 47 48
    namespace details { namespace db2
    {
        enum binding_method
        {
            BOUND_BY_NONE,
            BOUND_BY_NAME,
            BOUND_BY_POSITION
        };
49 50 51 52 53 54 55 56 57 58 59

        inline SQLPOINTER int_as_ptr(int n)
        {
            union
            {
                SQLPOINTER p;
                int n;
            } u;
            u.n = n;
            return u.p;
        }
60 61 62
    }}

    static const std::size_t maxBuffer =  1024 * 1024 * 1024; //CLI limit is about 3 GB, but 1GB should be enough
Denis Chapligin's avatar
Denis Chapligin committed
63 64 65 66

class db2_soci_error : public soci_error {
public:
    db2_soci_error(std::string const & msg, SQLRETURN rc) : soci_error(msg),errorCode(rc) {};
Denis Chapligin's avatar
Denis Chapligin committed
67
    ~db2_soci_error() throw() { };
68 69

    //We have to extract error information before exception throwing, cause CLI handles could be broken at the construction time
Denis Chapligin's avatar
Denis Chapligin committed
70
    static const std::string sqlState(std::string const & msg,const SQLSMALLINT htype,const SQLHANDLE hndl);
71 72

    SQLRETURN errorCode;
Denis Chapligin's avatar
Denis Chapligin committed
73 74
};

75 76 77 78 79 80
// Option allowing to specify the "driver completion" parameter of
// SQLDriverConnect(). Its possible values are the same as the allowed values
// for this parameter in the official DB2 CLI, i.e. one of SQL_DRIVER_XXX
// (in string form as all options are strings currently).
extern SOCI_DB2_DECL char const * db2_option_driver_complete;

Denis Chapligin's avatar
Denis Chapligin committed
81 82 83 84 85
struct db2_statement_backend;

struct SOCI_DB2_DECL db2_standard_into_type_backend : details::standard_into_type_backend
{
    db2_standard_into_type_backend(db2_statement_backend &st)
Denis Chapligin's avatar
Denis Chapligin committed
86
        : statement_(st),buf(NULL)
Denis Chapligin's avatar
Denis Chapligin committed
87 88 89 90 91 92 93 94 95 96
    {}

    void define_by_pos(int& position, void* data, details::exchange_type type);

    void pre_fetch();
    void post_fetch(bool gotData, bool calledFromFetch, indicator* ind);

    void clean_up();

    db2_statement_backend& statement_;
Denis Chapligin's avatar
Denis Chapligin committed
97 98 99 100 101 102 103

    char* buf;
    void *data;
    details::exchange_type type;
    int position;
    SQLSMALLINT cType;
    SQLLEN valueLen;
Denis Chapligin's avatar
Denis Chapligin committed
104 105 106 107 108
};

struct SOCI_DB2_DECL db2_vector_into_type_backend : details::vector_into_type_backend
{
    db2_vector_into_type_backend(db2_statement_backend &st)
Denis Chapligin's avatar
Denis Chapligin committed
109
        : statement_(st),buf(NULL)
Denis Chapligin's avatar
Denis Chapligin committed
110 111 112 113 114 115 116 117 118 119 120 121 122
    {}

    void define_by_pos(int& position, void* data, details::exchange_type type);

    void pre_fetch();
    void post_fetch(bool gotData, indicator* ind);

    void resize(std::size_t sz);
    std::size_t size();

    void clean_up();

    db2_statement_backend& statement_;
Denis Chapligin's avatar
Denis Chapligin committed
123 124 125 126 127 128 129

    void prepare_indicators(std::size_t size);

    SQLLEN *indptr;
    std::vector<SQLLEN> indVec;
    void *data;
    char *buf;
130
    int position_;
Denis Chapligin's avatar
Denis Chapligin committed
131 132 133
    details::exchange_type type;
    SQLSMALLINT cType;
    std::size_t colSize;
Denis Chapligin's avatar
Denis Chapligin committed
134 135 136 137 138
};

struct SOCI_DB2_DECL db2_standard_use_type_backend : details::standard_use_type_backend
{
    db2_standard_use_type_backend(db2_statement_backend &st)
139
        : statement_(st),buf(NULL),ind(0)
Denis Chapligin's avatar
Denis Chapligin committed
140 141 142 143 144 145 146 147 148 149 150
    {}

    void bind_by_pos(int& position, void* data, details::exchange_type type, bool readOnly);
    void bind_by_name(std::string const& name, void* data, details::exchange_type type, bool readOnly);

    void pre_use(indicator const* ind);
    void post_use(bool gotData, indicator* ind);

    void clean_up();

    db2_statement_backend& statement_;
Denis Chapligin's avatar
Denis Chapligin committed
151

152
    void *prepare_for_bind(void *data, SQLLEN &size, SQLSMALLINT &sqlType, SQLSMALLINT &cType);
Denis Chapligin's avatar
Denis Chapligin committed
153 154 155 156 157 158

    void *data;
    details::exchange_type type;
    int position;
    std::string name;
    char* buf;
159
    SQLLEN ind;
Denis Chapligin's avatar
Denis Chapligin committed
160 161 162 163 164
};

struct SOCI_DB2_DECL db2_vector_use_type_backend : details::vector_use_type_backend
{
    db2_vector_use_type_backend(db2_statement_backend &st)
Denis Chapligin's avatar
Denis Chapligin committed
165
        : statement_(st),buf(NULL) {}
Denis Chapligin's avatar
Denis Chapligin committed
166 167 168 169 170 171 172 173 174 175 176

    void bind_by_pos(int& position, void* data, details::exchange_type type);
    void bind_by_name(std::string const& name, void* data, details::exchange_type type);

    void pre_use(indicator const* ind);

    std::size_t size();

    void clean_up();

    db2_statement_backend& statement_;
Denis Chapligin's avatar
Denis Chapligin committed
177 178 179 180 181 182 183 184 185 186 187

    void prepare_indicators(std::size_t size);
    void prepare_for_bind(void *&data, SQLUINTEGER &size,SQLSMALLINT &sqlType, SQLSMALLINT &cType);
    void bind_helper(int &position, void *data, details::exchange_type type);

    SQLLEN *indptr;
    std::vector<SQLLEN> indVec;
    void *data;
    char *buf;
    details::exchange_type type;
    std::size_t colSize;
Denis Chapligin's avatar
Denis Chapligin committed
188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203
};

struct db2_session_backend;
struct SOCI_DB2_DECL db2_statement_backend : details::statement_backend
{
    db2_statement_backend(db2_session_backend &session);

    void alloc();
    void clean_up();
    void prepare(std::string const& query, details::statement_type eType);

    exec_fetch_result execute(int number);
    exec_fetch_result fetch(int number);

    long long get_affected_rows();
    int get_number_of_rows();
204
    std::string get_parameter_name(int index) const;
Denis Chapligin's avatar
Denis Chapligin committed
205 206 207 208 209

    std::string rewrite_for_procedure_call(std::string const& query);

    int prepare_for_describe();
    void describe_column(int colNum, data_type& dtype, std::string& columnName);
210
    size_t column_size(int col);
Denis Chapligin's avatar
Denis Chapligin committed
211 212 213 214 215 216 217

    db2_standard_into_type_backend* make_into_type_backend();
    db2_standard_use_type_backend* make_use_type_backend();
    db2_vector_into_type_backend* make_vector_into_type_backend();
    db2_vector_use_type_backend* make_vector_use_type_backend();

    db2_session_backend& session_;
Denis Chapligin's avatar
Denis Chapligin committed
218 219 220

    SQLHANDLE hStmt;
    std::string query_;
221
    std::vector<std::string> names_;
Denis Chapligin's avatar
Denis Chapligin committed
222 223
    bool hasVectorUseElements;
    SQLUINTEGER numRowsFetched;
224
    details::db2::binding_method use_binding_method_;
Denis Chapligin's avatar
Denis Chapligin committed
225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250
};

struct db2_rowid_backend : details::rowid_backend
{
    db2_rowid_backend(db2_session_backend &session);

    ~db2_rowid_backend();
};

struct db2_blob_backend : details::blob_backend
{
    db2_blob_backend(db2_session_backend& session);

    ~db2_blob_backend();

    std::size_t get_len();
    std::size_t read(std::size_t offset, char* buf, std::size_t toRead);
    std::size_t write(std::size_t offset, char const* buf, std::size_t toWrite);
    std::size_t append(char const* buf, std::size_t toWrite);
    void trim(std::size_t newLen);

    db2_session_backend& session_;
};

struct db2_session_backend : details::session_backend
{
251
    db2_session_backend(connection_parameters const& parameters);
Denis Chapligin's avatar
Denis Chapligin committed
252 253 254 255 256 257 258 259 260 261 262 263 264 265 266

    ~db2_session_backend();

    void begin();
    void commit();
    void rollback();

    std::string get_backend_name() const { return "DB2"; }

    void clean_up();

    db2_statement_backend* make_statement_backend();
    db2_rowid_backend* make_rowid_backend();
    db2_blob_backend* make_blob_backend();

Denis Chapligin's avatar
Denis Chapligin committed
267 268 269
    void parseConnectString(std::string const &);
    void parseKeyVal(std::string const &);

270
    std::string connection_string_;
Denis Chapligin's avatar
Denis Chapligin committed
271
    bool autocommit;
272
    bool in_transaction;
Denis Chapligin's avatar
Denis Chapligin committed
273 274 275 276 277 278 279

    SQLHANDLE hEnv; /* Environment handle */
    SQLHANDLE hDbc; /* Connection handle */
};

struct SOCI_DB2_DECL db2_backend_factory : backend_factory
{
280
    db2_backend_factory() {}
281 282
    db2_session_backend* make_session(
        connection_parameters const & parameters) const;
Denis Chapligin's avatar
Denis Chapligin committed
283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298
};

extern SOCI_DB2_DECL db2_backend_factory const db2;

extern "C"
{

// for dynamic backend loading
SOCI_DB2_DECL backend_factory const* factory_db2();
SOCI_DB2_DECL void register_factory_db2();

} // extern "C"

} // namespace soci

#endif // SOCI_DB2_H_INCLUDED