/sqlite3cc

To get this branch, use:
bzr branch http://bzr.ed.am/sqlite3cc

« back to all changes in this revision

Viewing changes to src/basic_statement.cc

  • Committer: Tim Marston
  • Date: 2015-02-26 08:59:16 UTC
  • Revision ID: tim@ed.am-20150226085916-1z7761knxbutpr7n
updated tests for blob support

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
1
/*
2
2
 * basic_statement.cc
3
3
 *
4
 
 * Copyright (C) 2009 Tim Marston <edam@waxworlds.org>
 
4
 * Copyright (C) 2009 Tim Marston <tim@ed.am>
5
5
 *
6
6
 * This file is part of sqlite3cc (hereafter referred to as "this program").
7
 
 * See http://www.waxworlds.org/edam/software/sqlite3cc for more information.
8
 
 *
9
 
 * This program is free software: you can redistribute it and/or modify
10
 
 * it under the terms of the GNU Lesser General Public License as published
11
 
 * by the Free Software Foundation, either version 3 of the License, or
12
 
 * (at your option) any later version.
13
 
 *
14
 
 * This program is distributed in the hope that it will be useful,
15
 
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16
 
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17
 
 * GNU Lesser General Public License for more details.
 
7
 * See http://ed.am/dev/sqlite3cc for more information.
 
8
 *
 
9
 * This program is free software: you can redistribute it and/or modify it under
 
10
 * the terms of the GNU Lesser General Public License as published by the Free
 
11
 * Software Foundation, either version 3 of the License, or (at your option) any
 
12
 * later version.
 
13
 *
 
14
 * This program is distributed in the hope that it will be useful, but WITHOUT
 
15
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
 
16
 * FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
 
17
 * details.
18
18
 *
19
19
 * You should have received a copy of the GNU Lesser General Public License
20
20
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
22
22
 
23
23
#include <sqlite3cc/basic_statement.h>
24
24
#include <sqlite3cc/exception.h>
25
 
#include <sqlite3cc/database.h>
 
25
#include <sqlite3cc/connection.h>
26
26
#include <sqlite3cc/manipulator.h>
27
 
#include <string.h>
28
 
#include <iomanip>
29
 
 
30
 
 
31
 
sqlite::basic_statement::basic_statement(
32
 
        database &database,
33
 
        const std::string &sql )
34
 
        :
35
 
        _database( database ),
36
 
        _handle( NULL ),
37
 
        _bind_index( 1 )
38
 
{
39
 
        int error_code = prepare( sql );
40
 
        if( error_code != SQLITE_OK ) throw sqlite_error( error_code );
41
 
}
42
 
 
43
 
 
44
 
sqlite::basic_statement::basic_statement(
45
 
        database &database )
46
 
        :
47
 
        _database( database ),
48
 
        _handle( NULL ),
49
 
        _bind_index( 1 )
50
 
{
51
 
}
52
 
 
53
 
 
54
 
sqlite::basic_statement::~basic_statement()
55
 
{
56
 
        finalize();
57
 
}
58
 
 
59
 
 
60
 
int sqlite::basic_statement::prepare(
61
 
        const std::string &sql )
62
 
{
63
 
        finalize();
64
 
        return sqlite3_prepare_v2( _database._handle, sql.c_str(),
 
27
 
 
28
 
 
29
sqlite::detail::basic_statement::basic_statement(
 
30
        connection &connection,
 
31
        const std::string &sql )
 
32
        :
 
33
        _connection( connection ),
 
34
        _handle( NULL ),
 
35
        _bind_index( 1 )
 
36
{
 
37
        int code = prepare( sql );
 
38
        if( code != SQLITE_OK ) throw sqlite_error( connection, code );
 
39
}
 
40
 
 
41
 
 
42
sqlite::detail::basic_statement::basic_statement(
 
43
        connection &connection )
 
44
        :
 
45
        _connection( connection ),
 
46
        _handle( NULL ),
 
47
        _bind_index( 1 )
 
48
{
 
49
}
 
50
 
 
51
 
 
52
sqlite::detail::basic_statement::~basic_statement()
 
53
{
 
54
        finalize();
 
55
}
 
56
 
 
57
 
 
58
int sqlite::detail::basic_statement::prepare(
 
59
        const std::string &sql )
 
60
{
 
61
        finalize();
 
62
        return sqlite3_prepare_v2( _connection._handle, sql.c_str(),
65
63
                sql.length() + 1, &_handle, NULL );
66
64
}
67
65
 
68
66
 
69
 
int sqlite::basic_statement::reset()
 
67
int sqlite::detail::basic_statement::reset()
70
68
{
71
69
        return sqlite3_reset( _handle );
72
70
}
73
71
 
74
72
 
75
 
int sqlite::basic_statement::clear_bindings()
 
73
int sqlite::detail::basic_statement::clear_bindings()
76
74
{
77
75
        _bind_index = 1;
78
76
        return sqlite3_clear_bindings( _handle );
79
77
}
80
78
 
81
79
 
82
 
int sqlite::basic_statement::bind_static(
 
80
int sqlite::detail::basic_statement::bind_static(
83
81
        unsigned int index,
84
82
        const char *value,
85
83
        unsigned int value_length )
86
84
{
 
85
                // bind as text (applying the type affinity of the underlying column)
87
86
        return sqlite3_bind_text( _handle, index, value, value_length,
88
87
                SQLITE_STATIC );
89
88
}
90
89
 
91
90
 
92
 
int sqlite::basic_statement::bind_static(
 
91
int sqlite::detail::basic_statement::bind_static(
93
92
        unsigned int index,
94
93
        const char *value )
95
94
{
97
96
}
98
97
 
99
98
 
100
 
int sqlite::basic_statement::bind_static(
 
99
int sqlite::detail::basic_statement::bind_static(
101
100
        unsigned int index,
102
101
        const std::string &value )
103
102
{
105
104
}
106
105
 
107
106
 
108
 
int sqlite::basic_statement::bind_null(
 
107
int sqlite::detail::basic_statement::bind_null(
109
108
        unsigned int index )
110
109
{
111
110
        return sqlite3_bind_null( _handle, index );
112
111
}
113
112
 
114
113
 
115
 
int sqlite::basic_statement::bind_static(
 
114
int sqlite::detail::basic_statement::bind_static(
116
115
        const std::string &name,
117
116
        const char *value,
118
117
        unsigned int value_length )
121
120
}
122
121
 
123
122
 
124
 
int sqlite::basic_statement::bind_static(
 
123
int sqlite::detail::basic_statement::bind_static(
125
124
        const std::string &name,
126
125
        const char *value )
127
126
{
129
128
}
130
129
 
131
130
 
132
 
int sqlite::basic_statement::bind_static(
 
131
int sqlite::detail::basic_statement::bind_static(
133
132
        const std::string &name,
134
133
        const std::string &value )
135
134
{
137
136
}
138
137
 
139
138
 
140
 
int sqlite::basic_statement::bind_null(
 
139
int sqlite::detail::basic_statement::bind_null(
141
140
        const std::string &name )
142
141
{
143
142
        return bind_null( bind_parameter_index( name ) );
144
143
}
145
144
 
146
145
 
147
 
int sqlite::basic_statement::finalize()
148
 
{
149
 
        int error_code = SQLITE_OK;
 
146
int sqlite::detail::basic_statement::bind_blob(
 
147
        unsigned int index,
 
148
        const char *value,
 
149
        unsigned int value_length )
 
150
{
 
151
        return sqlite3_bind_blob( _handle, index, value, value_length,
 
152
                SQLITE_TRANSIENT );
 
153
}
 
154
 
 
155
 
 
156
int sqlite::detail::basic_statement::bind_blob(
 
157
        unsigned int index,
 
158
        const std::string &value )
 
159
{
 
160
        return bind_blob( index, value.c_str(), value.length() );
 
161
}
 
162
 
 
163
 
 
164
int sqlite::detail::basic_statement::bind_blob(
 
165
        const std::string &name,
 
166
        const char *value,
 
167
        unsigned int value_length )
 
168
{
 
169
        return bind_blob( bind_parameter_index( name ), value, value_length );
 
170
}
 
171
 
 
172
 
 
173
int sqlite::detail::basic_statement::bind_blob(
 
174
        const std::string &name,
 
175
        const std::string &value )
 
176
{
 
177
        return bind_blob( bind_parameter_index( name ), value );
 
178
}
 
179
 
 
180
 
 
181
int sqlite::detail::basic_statement::finalize()
 
182
{
 
183
        int code = SQLITE_OK;
150
184
 
151
185
        if( _handle ) {
152
 
                error_code = sqlite3_finalize( _handle );
 
186
                code = sqlite3_finalize( _handle );
153
187
                _handle = NULL;
154
188
        }
155
189
 
156
 
        return error_code;
157
 
}
158
 
 
159
 
 
160
 
int sqlite::basic_statement::step()
161
 
{
162
 
        return sqlite3_step( _handle );
163
 
}
164
 
 
165
 
 
166
 
int sqlite::basic_statement::bind_parameter_index(
 
190
        return code;
 
191
}
 
192
 
 
193
 
 
194
int sqlite::detail::basic_statement::bind_parameter_index(
167
195
        const std::string &name )
168
196
{
169
197
        unsigned int index = sqlite3_bind_parameter_index( _handle, name.c_str() );
172
200
}
173
201
 
174
202
 
 
203
int sqlite::detail::basic_statement::step()
 
204
{
 
205
        return sqlite3_step( _handle );
 
206
}
 
207
 
 
208
 
175
209
template< >
176
 
sqlite::basic_statement &sqlite::basic_statement::operator <<
177
 
        < sqlite::_null_t >(
178
 
        const sqlite::_null_t & )
 
210
sqlite::detail::basic_statement &sqlite::detail::basic_statement::operator <<
 
211
        < sqlite::detail::null_t >(
 
212
        const sqlite::detail::null_t & )
179
213
{
180
 
        int error_code = bind_null( _bind_index );
181
 
        if( error_code != SQLITE_OK ) throw sqlite_error( error_code );
 
214
        int code = bind_null( _bind_index );
 
215
        if( code != SQLITE_OK ) throw sqlite_error( _connection, code );
182
216
        _bind_index++;
183
217
        return *this;
184
218
}
185
219
 
186
220
 
187
221
template< >
188
 
sqlite::basic_statement &sqlite::basic_statement::operator <<
189
 
        < sqlite::_exec_t >(
190
 
        const sqlite::_exec_t & )
191
 
{
192
 
        int error_code = step();
193
 
        if( error_code != SQLITE_DONE ) {
194
 
                if( error_code == SQLITE_ROW )
195
 
                        throw sqlite_error( "statement returned results" );
196
 
                else
197
 
                        throw sqlite_error( error_code );
198
 
        }
199
 
        return *this;
200
 
}
201
 
 
202
 
 
203
 
template< >
204
 
sqlite::basic_statement &sqlite::basic_statement::operator <<
205
 
        < sqlite::_set_index_t >(
206
 
        const sqlite::_set_index_t &t )
 
222
sqlite::detail::basic_statement &sqlite::detail::basic_statement::operator <<
 
223
        < sqlite::detail::set_index_t >(
 
224
        const sqlite::detail::set_index_t &t )
207
225
{
208
226
        _bind_index = t._index;
209
227
        return *this;
210
228
}
 
229
 
 
230
 
 
231
template< >
 
232
sqlite::detail::basic_statement &sqlite::detail::basic_statement::operator <<
 
233
        < sqlite::detail::blob_t >(
 
234
        const sqlite::detail::blob_t &t )
 
235
{
 
236
        int code = bind_blob( _bind_index, t.value, t.value_length );
 
237
        if( code != SQLITE_OK ) throw sqlite_error( _connection, code );
 
238
        _bind_index++;
 
239
        return *this;
 
240
}
 
241
 
 
242
 
 
243
template< >
 
244
int sqlite::detail::basic_statement::bind< int >(
 
245
        unsigned int index,
 
246
        const int &value )
 
247
{
 
248
        return sqlite3_bind_int( _handle, index, value );
 
249
}
 
250
 
 
251
 
 
252
template< >
 
253
int sqlite::detail::basic_statement::bind< double >(
 
254
        unsigned int index,
 
255
        const double &value )
 
256
{
 
257
        return sqlite3_bind_double( _handle, index, value );
 
258
}