/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 09:16:26 UTC
  • Revision ID: edam@waxworlds.org-20100729091626-h8fmg0r74eyfo5ae
- fixed error caused by finialising in-progress queries during rollback that were later finaliased by RAII.

Show diffs side-by-side

added added

removed removed

Lines of Context:
23
23
#include <sqlite3cc/transaction.h>
24
24
#include <sqlite3cc/database.h>
25
25
#include <sqlite3cc/exception.h>
26
 
 
27
 
 
28
 
sqlite::transaction::transaction(
 
26
#include <boost/thread.hpp>
 
27
#include <boost/lexical_cast.hpp>
 
28
 
 
29
 
 
30
sqlite::basic_transaction::basic_transaction(
29
31
        database &database )
30
32
        :
31
33
        _database( database )
32
34
{
33
 
        int error_code = _database.exec( "BEGIN" );
34
 
        if( error_code != SQLITE_OK ) throw sqlite_error( error_code );
35
 
}
36
 
 
37
 
 
38
 
sqlite::transaction::transaction(
39
 
        database &database,
40
 
        const std::string &begin_sql,
41
 
        const std::string &rollback_sql )
 
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_reset( 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" );
 
63
}
 
64
 
 
65
 
 
66
sqlite::immediate_transaction::immediate_transaction(
 
67
        database &database )
42
68
        :
43
 
        _database( database ),
44
 
        _rollback_sql( rollback_sql )
45
 
{
46
 
        int error_code = _database.exec( begin_sql );
47
 
        if( error_code != SQLITE_OK ) throw sqlite_error( error_code );
48
 
}
49
 
 
50
 
 
51
 
sqlite::transaction::~transaction() throw( )
52
 
{
53
 
        rollback();
54
 
}
55
 
 
56
 
 
57
 
void sqlite::transaction::commit()
58
 
{
59
 
        int error_code = _database.exec( "COMMIT" );
60
 
        if( error_code != SQLITE_OK ) throw sqlite_error( error_code );
61
 
}
62
 
 
63
 
 
64
 
void sqlite::transaction::rollback()
65
 
{
66
 
        int error_code = _database.exec(
67
 
                _rollback_sql.empty()? "ROLLBACK" : _rollback_sql );
68
 
        if( error_code != SQLITE_OK ) throw sqlite_error( error_code );
 
69
        basic_transaction( database )
 
70
{
 
71
}
 
72
 
 
73
 
 
74
void sqlite::immediate_transaction::begin()
 
75
{
 
76
        _database.exec( "BEGIN IMMEDIATE" );
69
77
}
70
78
 
71
79
 
72
80
sqlite::exclusive_transaction::exclusive_transaction(
73
81
        database &database )
74
82
        :
75
 
        transaction( database, "BEGIN EXCLUSIVE" )
76
 
{
 
83
        basic_transaction( database )
 
84
{
 
85
}
 
86
 
 
87
 
 
88
void sqlite::exclusive_transaction::begin()
 
89
{
 
90
        _database.exec( "BEGIN EXCLUSIVE" );
 
91
}
 
92
 
 
93
 
 
94
sqlite::recursive_transaction::recursive_transaction(
 
95
        database &database )
 
96
        :
 
97
        basic_transaction( database )
 
98
{
 
99
        static unsigned long long i = 0;
 
100
        static boost::mutex mutex;
 
101
        unsigned long long my_i;
 
102
        {
 
103
                boost::lock_guard< boost::mutex > lock( mutex );
 
104
                my_i = i++;
 
105
        }
 
106
        _sp_name = "_" + boost::lexical_cast< std::string >( my_i++ );
 
107
}
 
108
 
 
109
 
 
110
void sqlite::recursive_transaction::begin()
 
111
{
 
112
        _database.exec( "SAVEPOINT " + _sp_name );
 
113
}
 
114
 
 
115
 
 
116
void sqlite::recursive_transaction::commit()
 
117
{
 
118
        _database.exec( "RELEASE " + _sp_name );
 
119
}
 
120
 
 
121
 
 
122
void sqlite::recursive_transaction::rollback()
 
123
{
 
124
        // we must invalidate any active queries when rolling back
 
125
        invalidate_queries();
 
126
        _database.exec( "ROLLBACK TO " + _sp_name );
 
127
 
 
128
        // we have rolled back this transaction's savepoint, but the savepoint will
 
129
        // remain on the transaction stack unless we also release it
 
130
        _database.exec( "RELEASE " + _sp_name );
77
131
}