/sqlite3cc

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

« back to all changes in this revision

Viewing changes to include/sqlite3cc/row.h

  • Committer: Tim Marston
  • Date: 2015-02-26 08:58:24 UTC
  • Revision ID: tim@ed.am-20150226085824-wribs4klulr74n0e
improve blob support: should have been committed with previous commit!

Show diffs side-by-side

added added

removed removed

Lines of Context:
include/sqlite3cc/row.h
 
1
/*
 
2
 * row.h
 
3
 *
 
4
 * Copyright (C) 2009 Tim Marston <tim@ed.am>
 
5
 *
 
6
 * This file is part of sqlite3cc (hereafter referred to as "this program").
 
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
 *
 
19
 * You should have received a copy of the GNU Lesser General Public License
 
20
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 
21
 */
 
22
 
 
23
#ifndef SQLITE3CC_ROW_H_
 
24
#define SQLITE3CC_ROW_H_
 
25
 
 
26
 
 
27
#include <sqlite3.h>
 
28
#include <boost/utility.hpp>
 
29
#include <boost/lexical_cast.hpp>
 
30
#include <boost/utility/value_init.hpp>
 
31
#include <cassert>
 
32
#include <iostream>
 
33
 
 
34
 
 
35
namespace sqlite
 
36
{
 
37
 
 
38
 
 
39
class query;
 
40
namespace detail {
 
41
        struct null_t;
 
42
        struct set_index_t;
 
43
}
 
44
 
 
45
 
 
46
/**
 
47
 * A result row from a query.
 
48
 *
 
49
 * The row is only valid until the next call to step() or reset() on the parent
 
50
 * query object, or until the parent query object is destructed.  This may
 
51
 * change in future versions.
 
52
 */
 
53
class row
 
54
{
 
55
//______________________________________________________________________________
 
56
//                                                                 instantiation
 
57
protected:
 
58
 
 
59
        friend class query;
 
60
 
 
61
        /**
 
62
         * Constructor that produces a valid row.
 
63
         *
 
64
         * @param handle of the statement (query) that created this row
 
65
         * @oaram row_number the index of this row
 
66
         */
 
67
        explicit row(
 
68
                sqlite3_stmt *handle,
 
69
                unsigned long long row_number );
 
70
 
 
71
        /**
 
72
         * Constructor that produces an invalid row.
 
73
         */
 
74
        explicit row();
 
75
 
 
76
//______________________________________________________________________________
 
77
//                                                              public interface
 
78
public:
 
79
 
 
80
        /**
 
81
         * Determine if this row is valid or not.  If it is not valid, there are no
 
82
         * more rows in the results of the query.
 
83
         */
 
84
        operator bool() const;
 
85
 
 
86
        /**
 
87
         * Get the index in to the results that is this row.
 
88
         *
 
89
         * @return index
 
90
         */
 
91
        inline unsigned long long row_number()
 
92
        {
 
93
                return _row_number;
 
94
        }
 
95
 
 
96
        /**
 
97
         * Ascertain a column's type.
 
98
         *
 
99
         * @param index column index
 
100
         * @return sqlite datatype code
 
101
         * @see sqlite3_column_type()
 
102
         */
 
103
        int column_type(
 
104
                unsigned int index );
 
105
 
 
106
        /**
 
107
         * Get the number of bytes in the result for a given column.
 
108
         *
 
109
         * @param index column index
 
110
         * @return number of bytes in result
 
111
         * @see sqlite3_column_bytes()
 
112
         */
 
113
        unsigned int column_bytes(
 
114
                unsigned int index );
 
115
 
 
116
        /**
 
117
         * Get a value from the row.
 
118
         *
 
119
         * @param index column index
 
120
         * @param value reference to object to set with the value
 
121
         * @see sqlite3_column_*()
 
122
         */
 
123
        template< class T >
 
124
        void column(
 
125
                unsigned int index,
 
126
                T &value )
 
127
        {
 
128
                assert( index <
 
129
                        static_cast< unsigned int >( sqlite3_column_count( _handle ) ) );
 
130
 
 
131
                switch( column_type( index ) ) {
 
132
                case SQLITE_NULL:
 
133
                        value = boost::get( boost::value_initialized< T >() );
 
134
                        break;
 
135
                case SQLITE_BLOB: {
 
136
                        int length = sqlite3_column_bytes( _handle, index );
 
137
                        std::string string_value( static_cast< const char * >(
 
138
                                        sqlite3_column_blob( _handle, index ) ), length );
 
139
                        value = boost::lexical_cast< T >( string_value );
 
140
                        break;
 
141
                }
 
142
                default:
 
143
                        const char *text = reinterpret_cast< const char * >(
 
144
                                sqlite3_column_text( _handle, index ) );
 
145
                        value = boost::lexical_cast< T >( text );
 
146
                }
 
147
        }
 
148
 
 
149
        /**
 
150
         * Get a value from the row and return it.
 
151
         *
 
152
         * @param index column index
 
153
         * @return the value
 
154
         * @see sqlite3_column_*()
 
155
         */
 
156
        template< class T >
 
157
        T column(
 
158
                unsigned int index )
 
159
        {
 
160
                T value;
 
161
                column( index, value );
 
162
                return value;
 
163
        }
 
164
 
 
165
        /**
 
166
         * Stream operator is used to obtain values from a result row, fetching from
 
167
         * each column in turn.  In addition, the null and set_index() auto-column-
 
168
         * getting manipulators can be used.
 
169
         *
 
170
         * @param value is a variable to store the retrieved data in
 
171
         */
 
172
        template< class T >
 
173
        row &operator >>(
 
174
                T &value )
 
175
        {
 
176
                column( _column_index, value );
 
177
                _column_index++;
 
178
                return *this;
 
179
        }
 
180
 
 
181
        /**
 
182
         * Stream operator for use with set_index().
 
183
         */
 
184
        row &operator >>(
 
185
                detail::set_index_t t );
 
186
 
 
187
        /**
 
188
         * Test to see if two rows are the same.
 
189
         *
 
190
         * @param other the row to compare this one to
 
191
         * @return true if they are
 
192
         */
 
193
        bool operator ==(
 
194
                const row &other )
 
195
                const;
 
196
 
 
197
//______________________________________________________________________________
 
198
//                                                                implementation
 
199
protected:
 
200
 
 
201
        /** the query's handle, or NULL */
 
202
        sqlite3_stmt *_handle;
 
203
 
 
204
private:
 
205
 
 
206
        /** index used when auto-column-getting */
 
207
        unsigned int _column_index;
 
208
 
 
209
        /** the index of this row */
 
210
        unsigned long long _row_number;
 
211
 
 
212
};
 
213
 
 
214
 
 
215
// template specialisations
 
216
template< >
 
217
row &row::operator >> < detail::null_t >(
 
218
        detail::null_t & );
 
219
 
 
220
 
 
221
} // namespace sqlite
 
222
 
 
223
 
 
224
#endif /* SQLITE3CC_ROW_H_ */