/sqlite3cc

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

« back to all changes in this revision

Viewing changes to src/query.cc

  • Committer: edam
  • Date: 2010-07-29 06:28:53 UTC
  • Revision ID: edam@waxworlds.org-20100729062853-4i2fec52m86mh724
- made basic_statement::step() protected, for use by query and command only
- moved basic_statement::operator<<() to command and query instead; one needs to accept sqlite::exec, the other doesn't
- added tests for query::operator<<()
- added code to invlaidate in-progress queries during any transaction rollbacks (currently segfaults in basic_statement::finalize())
- added new sqlite_error constructor that obtains a full error message
- implemented database::database_mutex_guard class
- swapped command's step mutex in favour of the database mutex

Show diffs side-by-side

added added

removed removed

src/query.cc
1
1
/*
2
 
 * query.cpp
 
2
 * query.cc
3
3
 *
4
4
 * Copyright (C) 2009 Tim Marston <edam@waxworlds.org>
5
5
 *
6
 
 * This file is part of sqlitepp (hereafter referred to as "this program").
7
 
 * See http://www.waxworlds.org/edam/software/sqlitepp for more information.
 
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
8
 *
9
9
 * This program is free software: you can redistribute it and/or modify
10
10
 * it under the terms of the GNU Lesser General Public License as published
20
20
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
21
21
 */
22
22
 
23
 
#include <sqlitepp/query.hpp>
24
 
 
25
 
 
 
23
#include <sqlite3cc/query.h>
 
24
#include <sqlite3cc/row.h>
 
25
#include <sqlite3cc/manipulator.h>
 
26
#include <sqlite3cc/database.h>
 
27
#include <assert.h>
 
28
 
 
29
 
 
30
sqlite::query::query(
 
31
        database &database,
 
32
        const std::string &sql )
 
33
        :
 
34
        basic_statement( database, sql ),
 
35
        _next_row( 0 )
 
36
{
 
37
        assert( sqlite3_column_count( _handle ) > 0 );
 
38
}
 
39
 
 
40
 
 
41
sqlite::query::query(
 
42
        database &database )
 
43
        :
 
44
        basic_statement( database ),
 
45
        _next_row( 0 )
 
46
{
 
47
}
 
48
 
 
49
 
 
50
int sqlite::query::prepare(
 
51
        const std::string &sql )
 
52
{
 
53
        _next_row = 0;
 
54
        int code = basic_statement::prepare( sql );
 
55
        assert( code != SQLITE_OK || sqlite3_column_count( _handle ) > 0 );
 
56
        return code;
 
57
}
 
58
 
 
59
 
 
60
sqlite::row sqlite::query::step()
 
61
{
 
62
        database::database_mutex_guard lock( _database );
 
63
 
 
64
        int code = basic_statement::step();
 
65
        if( code == SQLITE_DONE ) return row();
 
66
        if( code == SQLITE_ROW ) return row( _handle, _next_row++ );
 
67
 
 
68
        throw sqlite_error( _database, code );
 
69
}
 
70
 
 
71
 
 
72
unsigned int sqlite::query::column_count()
 
73
{
 
74
        return sqlite3_column_count( _handle );
 
75
}
 
76
 
 
77
 
 
78
const std::string sqlite::query::column_name(
 
79
        unsigned int index )
 
80
{
 
81
        assert( index <
 
82
                static_cast< unsigned int >( sqlite3_column_count( _handle ) ) );
 
83
        return sqlite3_column_name( _handle, index );
 
84
}
 
85
 
 
86
 
 
87
sqlite::query::iterator::iterator(
 
88
        sqlite::query &query,
 
89
        bool valid )
 
90
        :
 
91
        _query( query )
 
92
{
 
93
        if( valid )
 
94
                increment();
 
95
        else
 
96
                _valid = false;
 
97
}
 
98
 
 
99
 
 
100
sqlite::row sqlite::query::iterator::dereference() const
 
101
{
 
102
        return sqlite::row( _query._handle, _query._next_row++ );
 
103
}
 
104
 
 
105
 
 
106
void sqlite::query::iterator::increment()
 
107
{
 
108
        _valid = _query.step();
 
109
}
 
110
 
 
111
 
 
112
bool sqlite::query::iterator::equal(
 
113
        sqlite::query::iterator const &other )
 
114
        const
 
115
{
 
116
        return _valid == other._valid;
 
117
}
 
118
 
 
119
 
 
120
sqlite::query::iterator sqlite::query::begin()
 
121
{
 
122
        return sqlite::query::iterator( *this, true );
 
123
}
 
124
 
 
125
 
 
126
sqlite::query::iterator sqlite::query::end()
 
127
{
 
128
        return sqlite::query::iterator( *this, false );
 
129
}
 
130
 
 
131
 
 
132
template< >
 
133
sqlite::query &sqlite::query::operator <<
 
134
        < sqlite::detail::null_t >(
 
135
        const sqlite::detail::null_t & )
 
136
{
 
137
        int code = bind_null( _bind_index );
 
138
        if( code != SQLITE_OK ) throw sqlite_error( _database, code );
 
139
        _bind_index++;
 
140
        return *this;
 
141
}
 
142
 
 
143
 
 
144
template< >
 
145
sqlite::query &sqlite::query::operator <<
 
146
        < sqlite::detail::set_index_t >(
 
147
        const sqlite::detail::set_index_t &t )
 
148
{
 
149
        _bind_index = t._index;
 
150
        return *this;
 
151
}