/sqlite3cc

/*

 * command.cc

 *

 * Copyright (C) 2009 Tim Marston <edam@waxworlds.org>

 *

 * This file is part of sqlite3cc (hereafter referred to as "this program").

 * See http://www.waxworlds.org/edam/software/sqlite3cc for more information.

 *

 * This program is free software: you can redistribute it and/or modify

 * it under the terms of the GNU Lesser General Public License as published

 * by the Free Software Foundation, either version 3 of the License, or

 * (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU Lesser General Public License for more details.

 *

 * You should have received a copy of the GNU Lesser General Public License

 * along with this program.  If not, see <http://www.gnu.org/licenses/>.

 */

#include <sqlite3cc/command.h>

#include <sqlite3cc/database.h>

#include <sqlite3cc/manipulator.h>

#include <cassert>

sqlite::command::command(

	database &database,

	const std::string &sql )

	:

	basic_statement( database, sql ),

	_changes( 0 ),

	_total_changes( 0 ),

	_last_insert_rowid( 0 )

{

	assert( sqlite3_column_count( _handle ) == 0 );

}

sqlite::command::command(

	database &database )

	:

	basic_statement( database ),

	_changes( 0 ),

	_total_changes( 0 ),

	_last_insert_rowid( 0 )

{

}

int sqlite::command::prepare(

	const std::string &sql )

{

	int code = basic_statement::prepare( sql );

	assert( code != SQLITE_OK || sqlite3_column_count( _handle ) == 0 );

	return code;

}

int sqlite::command::step()

{

	database::database_mutex_guard lock( _database );

	int code = basic_statement::step();

	if( code == SQLITE_ROW ) throw sqlite_error( "command has results" );

	if( code != SQLITE_DONE ) throw sqlite_error( _database, code );

	_changes = sqlite3_changes( _database._handle );

	_total_changes = sqlite3_total_changes( _database._handle );

	_last_insert_rowid = sqlite3_last_insert_rowid( _database._handle );

	lock.leave();

	// finalise now (which shouldn't matter for commands), so that there are no

	// active statements if we come to begin, commit or rollback a transaction

	finalize();

	return code;

}

template< >

sqlite::basic_statement &sqlite::basic_statement::operator <<

	< sqlite::detail::exec_t >(

	const sqlite::detail::exec_t & )

{

	sqlite::command &that = dynamic_cast< sqlite::command & >( *this );

	int code = that.step();

	if( code != SQLITE_DONE ) {

		if( code == SQLITE_ROW )

			throw sqlite_error( "statement returned results" );

		else

			throw sqlite_error( _database, code );

	}

	return *this;

}