/sqlite3cc

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

« back to all changes in this revision

Viewing changes to src/transaction.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

Lines of Context:
src/transaction.cc
1
1
/*
2
2
 * transaction.cc
3
3
 *
4
 
 * Copyright (C) 2009 Tim Marston <tim@ed.am>
 
4
 * Copyright (C) 2009 Tim Marston <edam@waxworlds.org>
5
5
 *
6
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.
 
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.
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/>.
21
21
 */
22
22
 
23
23
#include <sqlite3cc/transaction.h>
24
 
#include <sqlite3cc/connection.h>
 
24
#include <sqlite3cc/database.h>
25
25
#include <sqlite3cc/exception.h>
26
26
#include <boost/thread.hpp>
27
27
#include <boost/lexical_cast.hpp>
28
28
 
29
29
 
30
 
sqlite::detail::basic_transaction::basic_transaction(
31
 
        connection &connection )
 
30
sqlite::basic_transaction::basic_transaction(
 
31
        database &database )
32
32
        :
33
 
        _connection( connection )
34
 
{
35
 
}
36
 
 
37
 
 
38
 
void sqlite::detail::basic_transaction::begin()
39
 
{
40
 
        _connection.exec( "BEGIN" );
41
 
}
42
 
 
43
 
 
44
 
void sqlite::detail::basic_transaction::commit()
45
 
{
46
 
        // we must reset any active queries when committing
47
 
        reset_active_queries();
48
 
        _connection.exec( "COMMIT" );
49
 
}
50
 
 
51
 
 
52
 
void sqlite::detail::basic_transaction::reset_active_queries()
53
 
{
54
 
        sqlite3_stmt *old_handle = NULL;
55
 
        while( sqlite3_stmt *handle =
56
 
                sqlite3_next_stmt( _connection._handle, old_handle ) )
57
 
        {
58
 
                int code = sqlite3_reset( handle );
59
 
                if( code != SQLITE_OK ) throw sqlite_error( _connection, code );
60
 
                old_handle = handle;
61
 
        }
62
 
}
63
 
 
64
 
 
65
 
void sqlite::detail::basic_transaction::rollback()
66
 
{
67
 
        // we must reset any active queries when rolling back
68
 
        reset_active_queries();
69
 
        _connection.exec( "ROLLBACK" );
 
33
        _database( database )
 
34
{
 
35
}
 
36
 
 
37
 
 
38
void sqlite::basic_transaction::begin()
 
39
{
 
40
        _database.exec( "BEGIN" );
 
41
}
 
42
 
 
43
 
 
44
void sqlite::basic_transaction::commit()
 
45
{
 
46
        _database.exec( "COMMIT" );
 
47
}
 
48
 
 
49
 
 
50
void sqlite::basic_transaction::invalidate_queries()
 
51
{
 
52
        while( sqlite3_stmt *handle = sqlite3_next_stmt( _database._handle, NULL ) )
 
53
                if( int code = sqlite3_finalize( handle ) )
 
54
                        throw sqlite_error( _database, code );
 
55
}
 
56
 
 
57
 
 
58
void sqlite::basic_transaction::rollback()
 
59
{
 
60
        // we must invalidate any active queries when rolling back
 
61
        invalidate_queries();
 
62
        _database.exec( "ROLLBACK" );
70
63
}
71
64
 
72
65
 
73
66
sqlite::immediate_transaction::immediate_transaction(
74
 
        connection &connection )
 
67
        database &database )
75
68
        :
76
 
        basic_transaction( connection )
 
69
        basic_transaction( database )
77
70
{
78
71
}
79
72
 
80
73
 
81
74
void sqlite::immediate_transaction::begin()
82
75
{
83
 
        _connection.exec( "BEGIN IMMEDIATE" );
 
76
        _database.exec( "BEGIN IMMEDIATE" );
84
77
}
85
78
 
86
79
 
87
80
sqlite::exclusive_transaction::exclusive_transaction(
88
 
        connection &connection )
 
81
        database &database )
89
82
        :
90
 
        basic_transaction( connection )
 
83
        basic_transaction( database )
91
84
{
92
85
}
93
86
 
94
87
 
95
88
void sqlite::exclusive_transaction::begin()
96
89
{
97
 
        _connection.exec( "BEGIN EXCLUSIVE" );
 
90
        _database.exec( "BEGIN EXCLUSIVE" );
98
91
}
99
92
 
100
93
 
101
94
sqlite::recursive_transaction::recursive_transaction(
102
 
        connection &connection )
 
95
        database &database )
103
96
        :
104
 
        basic_transaction( connection )
 
97
        basic_transaction( database )
105
98
{
106
99
        static unsigned long long i = 0;
107
100
        static boost::mutex mutex;
116
109
 
117
110
void sqlite::recursive_transaction::begin()
118
111
{
119
 
        _connection.exec( "SAVEPOINT " + _sp_name );
 
112
        _database.exec( "SAVEPOINT " + _sp_name );
120
113
}
121
114
 
122
115
 
123
116
void sqlite::recursive_transaction::commit()
124
117
{
125
 
        // we must reset any active queries when committing
126
 
        reset_active_queries();
127
 
        _connection.exec( "RELEASE " + _sp_name );
 
118
        _database.exec( "RELEASE " + _sp_name );
128
119
}
129
120
 
130
121
 
131
122
void sqlite::recursive_transaction::rollback()
132
123
{
133
 
        // we must reset any active queries when rolling back
134
 
        reset_active_queries();
135
 
        _connection.exec( "ROLLBACK TO " + _sp_name );
 
124
        // we must invalidate any active queries when rolling back
 
125
        invalidate_queries();
 
126
        _database.exec( "ROLLBACK TO " + _sp_name );
136
127
 
137
128
        // we have rolled back this transaction's savepoint, but the savepoint will
138
129
        // remain on the transaction stack unless we also release it
139
 
        _connection.exec( "RELEASE " + _sp_name );
 
130
        _database.exec( "RELEASE " + _sp_name );
140
131
}