summaryrefslogtreecommitdiff
path: root/libodbcpp/column.cpp
blob: d4b6446335322b4c02c63ac815fd3c23fd853a65 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
#include <stdio.h>
#include <stdlib.h>
#include "column.h"
#include "command.h"
#include "error.h"

ODBC::Column::Column(String n, unsigned int i) :
	colNo(i),
	name(n),
	fresh(false)
{
}

ODBC::Column::~Column()
{
}

#define ODBC_DEFAULT_COLUMN_CAST(ctype, rtype) \
	ODBC::Column::operator rtype() const { \
		return (dynamic_cast<const _Column<ctype>& >(*this)).value; \
	}

ODBC_DEFAULT_COLUMN_CAST(SQLINTEGER, unsigned int);
ODBC_DEFAULT_COLUMN_CAST(SQLINTEGER, unsigned long long);
ODBC_DEFAULT_COLUMN_CAST(SQLINTEGER, long long);
ODBC_DEFAULT_COLUMN_CAST(SQLINTEGER, int);
ODBC_DEFAULT_COLUMN_CAST(SQLDOUBLE, double);
ODBC_DEFAULT_COLUMN_CAST(SQLDOUBLE, float);
ODBC_DEFAULT_COLUMN_CAST(SQLCHAR*, const unsigned char *);
ODBC_DEFAULT_COLUMN_CAST(SQLCHAR*, String);
ODBC::Column::operator std::string() const {
	return (const char*)((dynamic_cast<const _Column<SQLCHAR*>& >(*this)).value);
}
ODBC::Column::operator const char * () const {
	return (const char*)((dynamic_cast<const _Column<SQLCHAR*>& >(*this)).value);
}
ODBC::Column::operator struct tm () const {
	const _Column<SQL_TIMESTAMP_STRUCT>& c = dynamic_cast<const _Column<SQL_TIMESTAMP_STRUCT>& >(*this);
	struct tm rtn;
	rtn << c.value;
	return rtn;
}

void
ODBC::Column::bind(SQLHANDLE hStmt, SQLUINTEGER col, SQLSMALLINT ctype, void * buf, size_t size)
{
	bindSize = size;
	RETCODE rc = SQLBindCol(hStmt, col, ctype, buf, bindSize, &bindLen);
	if (rc != SQL_SUCCESS) {
		throw Error(rc, SQL_HANDLE_STMT, hStmt, "%s: Bind column %lu", __FUNCTION__, col);
	}
}

#define REBIND(t, p) \
	template<> void _Column<t>::rebind(Command * cmd, unsigned int col) const \
	{ \
		cmd->p(col, value); \
	}
namespace ODBC {
	REBIND(int, bindParamI)
	REBIND(long, bindParamI)
	REBIND(unsigned int, bindParamI)
	REBIND(long unsigned int, bindParamI)
	REBIND(long long unsigned int, bindParamI)
	REBIND(double, bindParamF)
	REBIND(float, bindParamF)
	REBIND(SQL_TIMESTAMP_STRUCT, bindParamT)
	REBIND(unsigned char *, bindParamS)

	template <>
	int
	_Column<SQLDOUBLE>::writeToBuf(char ** buf, const char * fmt) const
	{
		return asprintf(buf, fmt, value);
	}
	template <>
	int
	_Column<SQLDOUBLE>::writeToBuf(char ** buf) const
	{
		return writeToBuf(buf, "%g");
	}
	template <>
	int
	_Column<SQLINTEGER>::writeToBuf(char ** buf, const char * fmt) const
	{
		return asprintf(buf, fmt, value);
	}
	template <>
	int
	_Column<SQLINTEGER>::writeToBuf(char ** buf) const
	{
		return writeToBuf(buf, "%ld");
	}
	template <>
	int
	_Column<SQLCHAR*>::writeToBuf(char ** buf, const char * fmt) const
	{
		return asprintf(buf, fmt, value);
	}
	template <>
	int
	_Column<SQLCHAR*>::writeToBuf(char ** buf) const
	{
		return writeToBuf(buf, "%s");
	}
	template <>
	int
	_Column<SQL_TIMESTAMP_STRUCT>::writeToBuf(char ** buf, const char * fmt) const
	{
		*buf = (char *)malloc(30);
		struct tm t;
		t << value;
		return strftime(*buf, 30, fmt, &t);
	}
	template <>
	int
	_Column<SQL_TIMESTAMP_STRUCT>::writeToBuf(char ** buf) const
	{
		return writeToBuf(buf, "%F %T");
	}
}

void operator << (SQL_TIMESTAMP_STRUCT & target, const struct tm & src)
{
	target.year = src.tm_year + 1900;
	target.month = src.tm_mon + 1;
	target.day = src.tm_mday;
	target.hour = src.tm_hour;
	target.minute = src.tm_min;
	target.second = src.tm_sec;
	target.fraction = 0;
}
void operator << (struct tm & target, const SQL_TIMESTAMP_STRUCT & src)
{
	target.tm_year = src.year - 1900;
	target.tm_mon = src.month - 1;
	target.tm_mday = src.day;
	target.tm_hour = src.hour;
	target.tm_min = src.minute;
	target.tm_sec = src.second;
}