/sqlite3cc

To get this branch, use:
bzr branch http://bzr.ed.am/sqlite3cc
2 by edam
- further initial development
1
/*
2
 * command.cc
3
 *
4
 * Copyright (C) 2009 Tim Marston <edam@waxworlds.org>
5
 *
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
 *
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
 *
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
#include <sqlite3cc/command.h>
24
#include <sqlite3cc/database.h>
13 by edam
- made basic_statement::step() protected, for use by query and command only
25
#include <sqlite3cc/manipulator.h>
2 by edam
- further initial development
26
#include <cassert>
27
28
29
sqlite::command::command(
30
	database &database,
31
	const std::string &sql )
32
	:
33
	basic_statement( database, sql ),
34
	_changes( 0 ),
35
	_total_changes( 0 ),
36
	_last_insert_rowid( 0 )
37
{
38
	assert( sqlite3_column_count( _handle ) == 0 );
39
}
40
41
42
sqlite::command::command(
43
	database &database )
44
	:
45
	basic_statement( database ),
46
	_changes( 0 ),
47
	_total_changes( 0 ),
48
	_last_insert_rowid( 0 )
49
{
50
}
51
52
53
int sqlite::command::prepare(
54
	const std::string &sql )
55
{
13 by edam
- made basic_statement::step() protected, for use by query and command only
56
	int code = basic_statement::prepare( sql );
57
	assert( code != SQLITE_OK || sqlite3_column_count( _handle ) == 0 );
58
	return code;
2 by edam
- further initial development
59
}
60
61
62
int sqlite::command::step()
63
{
13 by edam
- made basic_statement::step() protected, for use by query and command only
64
	database::database_mutex_guard lock( _database );
65
66
	int code = basic_statement::step();
67
	if( code == SQLITE_ROW ) throw sqlite_error( "command has results" );
68
	if( code != SQLITE_DONE ) throw sqlite_error( _database, code );
69
70
	_changes = sqlite3_changes( _database._handle );
71
	_total_changes = sqlite3_total_changes( _database._handle );
72
	_last_insert_rowid = sqlite3_last_insert_rowid( _database._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< >
14 by edam
- moved basic_statement::operator <<() back to basic_statement and just create another specialisation in command so that it can use sqlite::exec
85
sqlite::basic_statement &sqlite::basic_statement::operator <<
13 by edam
- made basic_statement::step() protected, for use by query and command only
86
	< sqlite::detail::exec_t >(
87
	const sqlite::detail::exec_t & )
88
{
14 by edam
- moved basic_statement::operator <<() back to basic_statement and just create another specialisation in command so that it can use sqlite::exec
89
	sqlite::command &that = dynamic_cast< sqlite::command & >( *this );
90
91
	int code = that.step();
13 by edam
- made basic_statement::step() protected, for use by query and command only
92
	if( code != SQLITE_DONE ) {
93
		if( code == SQLITE_ROW )
94
			throw sqlite_error( "statement returned results" );
95
		else
96
			throw sqlite_error( _database, code );
2 by edam
- further initial development
97
	}
13 by edam
- made basic_statement::step() protected, for use by query and command only
98
	return *this;
99
}
100