/sqlite3cc : contents of src/query.cc at revision 13

: (revision 13)

To get this branch, use:

bzr branch
http://bzr.ed.am/sqlite3cc

1 by edam - initial commit	1	/*
2 by edam - further initial development	2	* query.cc
1 by edam - initial commit	3	*
	4	* Copyright (C) 2009 Tim Marston <edam@waxworlds.org>
	5	*
2 by edam - further initial development	6	* This file is part of sqlite3cc (hereafter referred to as "this program").
2 by edam - further initial development	7	* See http://www.waxworlds.org/edam/software/sqlite3cc for more information.
1 by edam - initial commit	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
2 by edam - further initial development	23	#include <sqlite3cc/query.h>
2 by edam - further initial development	24	#include <sqlite3cc/row.h>
13 by edam - made basic_statement::step() protected, for use by query and command only	25	#include <sqlite3cc/manipulator.h>
	26	#include <sqlite3cc/database.h>
2 by edam - further initial development	27	#include <assert.h>
	28
	29
	30	sqlite::query::query(
	31	database &database,
	32	const std::string &sql )
	33	:
9 by edam - added NEWS	34	basic_statement( database, sql ),
9 by edam - added NEWS	35	_next_row( 0 )
2 by edam - further initial development	36	{
	37	assert( sqlite3_column_count( _handle ) > 0 );
	38	}
	39
	40
	41	sqlite::query::query(
	42	database &database )
	43	:
9 by edam - added NEWS	44	basic_statement( database ),
9 by edam - added NEWS	45	_next_row( 0 )
2 by edam - further initial development	46	{
	47	}
	48
	49
	50	int sqlite::query::prepare(
	51	const std::string &sql )
	52	{
9 by edam - added NEWS	53	_next_row = 0;
13 by edam - made basic_statement::step() protected, for use by query and command only	54	int code = basic_statement::prepare( sql );
	55	assert( code != SQLITE_OK \|\| sqlite3_column_count( _handle ) > 0 );
	56	return code;
2 by edam - further initial development	57	}
	58
	59
	60	sqlite::row sqlite::query::step()
	61	{
13 by edam - made basic_statement::step() protected, for use by query and command only	62	database::database_mutex_guard lock( _database );
	63
	64	int code = basic_statement::step();
	65	if( code == SQLITE_DONE ) return row();
	66	if( code == SQLITE_ROW ) return row( _handle, _next_row++ );
	67
	68	throw sqlite_error( _database, code );
2 by edam - further initial development	69	}
	70
	71
	72	unsigned int sqlite::query::column_count()
	73	{
	74	return sqlite3_column_count( _handle );
	75	}
9 by edam - added NEWS	76
	77
	78	const std::string sqlite::query::column_name(
	79	unsigned int index )
	80	{
	81	assert( index <
	82	static_cast< unsigned int >( sqlite3_column_count( _handle ) ) );
	83	return sqlite3_column_name( _handle, index );
	84	}
	85
	86
	87	sqlite::query::iterator::iterator(
	88	sqlite::query &query,
	89	bool valid )
	90	:
	91	_query( query )
	92	{
	93	if( valid )
	94	increment();
	95	else
	96	_valid = false;
	97	}
	98
	99
	100	sqlite::row sqlite::query::iterator::dereference() const
	101	{
	102	return sqlite::row( _query._handle, _query._next_row++ );
	103	}
	104
	105
	106	void sqlite::query::iterator::increment()
	107	{
	108	_valid = _query.step();
	109	}
	110
	111
	112	bool sqlite::query::iterator::equal(
	113	sqlite::query::iterator const &other )
	114	const
	115	{
	116	return _valid == other._valid;
	117	}
	118
	119
	120	sqlite::query::iterator sqlite::query::begin()
	121	{
	122	return sqlite::query::iterator( *this, true );
	123	}
	124
	125
	126	sqlite::query::iterator sqlite::query::end()
	127	{
	128	return sqlite::query::iterator( *this, false );
	129	}
13 by edam - made basic_statement::step() protected, for use by query and command only	130
	131
	132	template< >
	133	sqlite::query &sqlite::query::operator <<
	134	< sqlite::detail::null_t >(
	135	const sqlite::detail::null_t & )
	136	{
	137	int code = bind_null( _bind_index );
	138	if( code != SQLITE_OK ) throw sqlite_error( _database, code );
	139	_bind_index++;
	140	return *this;
	141	}
	142
	143
	144	template< >
	145	sqlite::query &sqlite::query::operator <<
	146	< sqlite::detail::set_index_t >(
	147	const sqlite::detail::set_index_t &t )
	148	{
	149	_bind_index = t._index;
	150	return *this;
	151	}