/sqlite3cc

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

« back to all changes in this revision

Viewing changes to src/command.cc

  • Committer: edam
  • Date: 2010-07-23 12:57:07 UTC
  • Revision ID: edam@waxworlds.org-20100723125707-qu9jk9vvg2uewx7t
- cleaned up test-main
- fixed comment type
- made sqlite::exec() throw instead of returning sqlite error code

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
1
/*
2
2
 * command.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/command.h>
24
 
#include <sqlite3cc/connection.h>
25
 
#include <sqlite3cc/manipulator.h>
 
24
#include <sqlite3cc/database.h>
 
25
#include <boost/thread/locks.hpp>
26
26
#include <cassert>
27
27
 
28
28
 
 
29
boost::recursive_mutex sqlite::command::_command_mutex;
 
30
 
 
31
 
29
32
sqlite::command::command(
30
 
        connection &connection,
 
33
        database &database,
31
34
        const std::string &sql )
32
35
        :
33
 
        detail::basic_statement( connection, sql ),
 
36
        basic_statement( database, sql ),
34
37
        _changes( 0 ),
35
38
        _total_changes( 0 ),
36
39
        _last_insert_rowid( 0 )
40
43
 
41
44
 
42
45
sqlite::command::command(
43
 
        connection &connection )
 
46
        database &database )
44
47
        :
45
 
        detail::basic_statement( connection ),
 
48
        basic_statement( database ),
46
49
        _changes( 0 ),
47
50
        _total_changes( 0 ),
48
51
        _last_insert_rowid( 0 )
53
56
int sqlite::command::prepare(
54
57
        const std::string &sql )
55
58
{
56
 
        int code = detail::basic_statement::prepare( sql );
57
 
        assert( code != SQLITE_OK || sqlite3_column_count( _handle ) == 0 );
58
 
        return code;
 
59
        int error_code = basic_statement::prepare( sql );
 
60
        assert( error_code != SQLITE_OK || sqlite3_column_count( _handle ) == 0 );
 
61
        return error_code;
59
62
}
60
63
 
61
64
 
62
65
int sqlite::command::step()
63
66
{
64
 
        connection::mutex_guard lock( _connection );
65
 
 
66
 
        int code = detail::basic_statement::step();
67
 
        if( code == SQLITE_ROW ) throw sqlite_error( "command has results" );
68
 
        if( code != SQLITE_DONE ) throw sqlite_error( _connection, code );
69
 
 
70
 
        _changes = sqlite3_changes( _connection._handle );
71
 
        _total_changes = sqlite3_total_changes( _connection._handle );
72
 
        _last_insert_rowid = sqlite3_last_insert_rowid( _connection._handle );
73
 
 
74
 
        lock.leave();
75
 
 
76
 
        // finalise now (which shouldn't matter for commands), so that there are no
77
 
        // active statements if we come to begin, commit or rollback a transaction
78
 
        finalize();
79
 
 
80
 
        return code;
81
 
}
82
 
 
83
 
 
84
 
template< >
85
 
sqlite::detail::basic_statement &sqlite::detail::basic_statement::operator <<
86
 
        < sqlite::detail::exec_t >(
87
 
        const sqlite::detail::exec_t & )
88
 
{
89
 
        sqlite::command &that = dynamic_cast< sqlite::command & >( *this );
90
 
 
91
 
        int code = that.step();
92
 
        if( code != SQLITE_DONE ) {
93
 
                if( code == SQLITE_ROW )
94
 
                        throw sqlite_error( "statement returned results" );
95
 
                else
96
 
                        throw sqlite_error( _connection, code );
 
67
        boost::lock_guard< boost::recursive_mutex > lock( _command_mutex );
 
68
        int error_code = basic_statement::step();
 
69
        if( error_code == SQLITE_OK ) {
 
70
                _changes = sqlite3_changes( _database._handle );
 
71
                _total_changes = sqlite3_total_changes( _database._handle );
 
72
                _last_insert_rowid = sqlite3_last_insert_rowid( _database._handle );
97
73
        }
98
 
        return *this;
 
74
        return error_code;
99
75
}
100