You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
467 lines
9.7 KiB
467 lines
9.7 KiB
/* This file is part of TDevelop
|
|
Copyright (C) 2003 Roberto Raggi <roberto@kdevelop.org>
|
|
|
|
This library is free software; you can redistribute it and/or
|
|
modify it under the terms of the GNU Library General Public
|
|
License as published by the Free Software Foundation; either
|
|
version 2 of the License, or (at your option) any later version.
|
|
|
|
This library 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
|
|
Library General Public License for more details.
|
|
|
|
You should have received a copy of the GNU Library General Public License
|
|
along with this library; see the file COPYING.LIB. If not, write to
|
|
the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
|
|
Boston, MA 02110-1301, USA.
|
|
*/
|
|
#include "catalog.h"
|
|
#include <tqdir.h>
|
|
#include <tqfile.h>
|
|
#include <tqfileinfo.h>
|
|
#include <tqdatastream.h>
|
|
|
|
#include <krandomsequence.h>
|
|
#include <kdebug.h>
|
|
|
|
#include <cstring>
|
|
#include <cstdlib>
|
|
|
|
#include <config.h>
|
|
|
|
#ifdef USE_DB_H_PATH
|
|
#include USE_DB_H_PATH
|
|
#else
|
|
#include <db.h>
|
|
#endif
|
|
|
|
struct _Catalog_Private
|
|
{
|
|
TQString dbName;
|
|
|
|
DB* dbp;
|
|
TQMap<TQCString, DB*> indexList;
|
|
KRandomSequence rnd;
|
|
bool enabled;
|
|
|
|
_Catalog_Private()
|
|
: dbp( 0 ), enabled( true )
|
|
{
|
|
}
|
|
|
|
bool hasIndex( const TQCString& name ) const
|
|
{
|
|
return indexList.contains( name );
|
|
}
|
|
|
|
DB* index( const TQCString& name )
|
|
{
|
|
return indexList[ name ];
|
|
}
|
|
|
|
bool addItem( DB* dbp, const TQCString& id, const Tag& tag )
|
|
{
|
|
Q_ASSERT( dbp != 0 );
|
|
|
|
DBT key, data;
|
|
int ret;
|
|
|
|
std::memset( &key, 0, sizeof(key) );
|
|
std::memset( &data, 0, sizeof(data) );
|
|
|
|
TQByteArray a1;
|
|
{
|
|
TQDataStream stream( a1, IO_WriteOnly );
|
|
stream << id;
|
|
key.data = a1.data();
|
|
key.size = a1.size();
|
|
}
|
|
|
|
TQByteArray a2;
|
|
{
|
|
TQDataStream stream( a2, IO_WriteOnly );
|
|
tag.store( stream );
|
|
data.data = a2.data();
|
|
data.size = a2.size();
|
|
}
|
|
|
|
ret = dbp->put( dbp, 0, &key, &data, 0 );
|
|
|
|
return ret == 0;
|
|
}
|
|
|
|
bool addItem( DB* dbp, const TQVariant& id, const TQCString& v )
|
|
{
|
|
Q_ASSERT( dbp != 0 );
|
|
|
|
DBT key, data;
|
|
int ret;
|
|
|
|
std::memset( &key, 0, sizeof(key) );
|
|
std::memset( &data, 0, sizeof(data) );
|
|
|
|
TQByteArray a1;
|
|
{
|
|
TQDataStream stream( a1, IO_WriteOnly );
|
|
stream << id;
|
|
key.data = a1.data();
|
|
key.size = a1.size();
|
|
}
|
|
|
|
TQByteArray a2;
|
|
{
|
|
TQDataStream stream( a2, IO_WriteOnly );
|
|
stream << v;
|
|
data.data = a2.data();
|
|
data.size = a2.size();
|
|
}
|
|
|
|
ret = dbp->put( dbp, 0, &key, &data, 0 );
|
|
|
|
return ret == 0;
|
|
}
|
|
|
|
};
|
|
|
|
|
|
/*!
|
|
\fn Catalog::Catalog
|
|
*/
|
|
Catalog::Catalog()
|
|
: d( new _Catalog_Private() )
|
|
{
|
|
}
|
|
|
|
/*!
|
|
\fn Catalog::~Catalog
|
|
*/
|
|
Catalog::~Catalog()
|
|
{
|
|
close();
|
|
delete( d );
|
|
d = 0;
|
|
}
|
|
|
|
/*!
|
|
\fn Catalog::indexList() const
|
|
*/
|
|
TQValueList<TQCString> Catalog::indexList() const
|
|
{
|
|
TQValueList<TQCString> l;
|
|
TQMap<TQCString, DB*>::Iterator it = d->indexList.begin();
|
|
while( it != d->indexList.end() ){
|
|
l << it.key();
|
|
++it;
|
|
}
|
|
|
|
return l;
|
|
}
|
|
|
|
bool Catalog::enabled() const
|
|
{
|
|
return d->enabled;
|
|
}
|
|
|
|
void Catalog::setEnabled( bool isEnabled )
|
|
{
|
|
d->enabled = isEnabled;
|
|
}
|
|
|
|
/*!
|
|
\fn Catalog::addIndex( const TQString& name )
|
|
@todo document these functions
|
|
*/
|
|
void Catalog::addIndex( const TQCString& name )
|
|
{
|
|
Q_ASSERT( d->dbp != 0 );
|
|
|
|
TQMap<TQCString, DB*>::Iterator it = d->indexList.find( name );
|
|
if( it == d->indexList.end() ){
|
|
DB* dbp = 0;
|
|
|
|
int ret;
|
|
|
|
if ((ret = db_create(&dbp, 0, 0)) != 0) {
|
|
kdDebug() << "db_create: " << db_strerror(ret) << endl;
|
|
return /*false*/;
|
|
}
|
|
|
|
if ((ret = dbp->set_flags(dbp, DB_DUP | DB_DUPSORT)) != 0) {
|
|
dbp->err(dbp, ret, "set_flags: DB_DUP | DB_DUPSORT");
|
|
dbp->close( dbp, 0 );
|
|
return;
|
|
}
|
|
|
|
TQFileInfo fileInfo( d->dbName );
|
|
TQString indexName = fileInfo.dirPath(true) + "/" + fileInfo.baseName(true) + "." + TQString(name) + ".idx";
|
|
|
|
if( (ret = dbp->set_cachesize( dbp, 0, 2 * 1024 * 1024, 0 )) != 0 ){
|
|
kdDebug() << "set_cachesize: " << db_strerror(ret) << endl;
|
|
}
|
|
|
|
if ((ret = dbp->open(
|
|
dbp, NULL, TQFile::encodeName( indexName ).data(), 0, DB_BTREE, DB_CREATE, 0664)) != 0) {
|
|
kdDebug() << "db_open: " << db_strerror(ret) << endl;
|
|
dbp->close( dbp, 0 );
|
|
return;
|
|
}
|
|
|
|
d->indexList[ name ] = dbp;
|
|
}
|
|
}
|
|
|
|
/*!
|
|
\fn Catalog::close()
|
|
*/
|
|
|
|
void Catalog::close()
|
|
{
|
|
d->dbName = TQString();
|
|
|
|
TQMap<TQCString, DB*>::Iterator it = d->indexList.begin();
|
|
while( it != d->indexList.end() ){
|
|
if( it.data() ){
|
|
it.data()->close( it.data(), 0 );
|
|
}
|
|
++it;
|
|
}
|
|
d->indexList.clear();
|
|
|
|
if( d->dbp != 0 ){
|
|
d->dbp->close( d->dbp, 0 );
|
|
d->dbp = 0;
|
|
}
|
|
}
|
|
|
|
/*!
|
|
\fn Catalog::open( const TQString& dbName )
|
|
*/
|
|
|
|
void Catalog::open( const TQString& dbName )
|
|
{
|
|
Q_ASSERT( d->dbp == 0 );
|
|
|
|
d->dbName = dbName;
|
|
|
|
int ret;
|
|
|
|
if ((ret = db_create(&d->dbp, 0, 0)) != 0) {
|
|
kdDebug() << "db_create: " << db_strerror(ret) << endl;
|
|
return /*false*/;
|
|
}
|
|
|
|
if ((ret = d->dbp->set_flags(d->dbp, DB_RECNUM)) != 0) {
|
|
d->dbp->err(d->dbp, ret, "set_flags: DB_RECNUM");
|
|
close();
|
|
return;
|
|
}
|
|
|
|
if( (ret = d->dbp->set_cachesize( d->dbp, 0, 2 * 1024 * 1024, 0 )) != 0 ){
|
|
kdDebug() << "set_cachesize: " << db_strerror(ret) << endl;
|
|
}
|
|
|
|
if ((ret = d->dbp->open(
|
|
d->dbp, NULL, d->dbName.local8Bit(), 0, DB_BTREE, DB_CREATE, 0664)) != 0) {
|
|
kdDebug() << "db_open: " << db_strerror(ret) << endl;
|
|
close();
|
|
return;
|
|
}
|
|
}
|
|
|
|
/*!
|
|
\fn Catalog::dbName() const
|
|
*/
|
|
|
|
TQString Catalog::dbName() const
|
|
{
|
|
return d->dbName;
|
|
}
|
|
|
|
/*!
|
|
\fn Catalog::isValid() const
|
|
*/
|
|
|
|
bool Catalog::isValid() const
|
|
{
|
|
return d->dbp != 0;
|
|
}
|
|
|
|
/*!
|
|
\fn Catalog::addItem( Tag& tag )
|
|
*/
|
|
|
|
void Catalog::addItem( Tag& tag )
|
|
{
|
|
if( tag.name().isEmpty() )
|
|
return;
|
|
|
|
TQCString id = generateId();
|
|
|
|
tag.setId( id );
|
|
if( d->addItem(d->dbp, id, tag) ){
|
|
TQMap<TQCString, DB*>::Iterator it = d->indexList.begin();
|
|
while( it != d->indexList.end() ){
|
|
if( tag.hasAttribute(it.key()) )
|
|
d->addItem( it.data(), tag.attribute(it.key()), id );
|
|
++it;
|
|
}
|
|
}
|
|
}
|
|
|
|
/*!
|
|
\fn Catalog::getItemById( const TQString& id )
|
|
*/
|
|
|
|
Tag Catalog::getItemById( const TQCString& id )
|
|
{
|
|
Q_ASSERT( d->dbp != 0 );
|
|
|
|
DBT key, data;
|
|
std::memset( &key, 0, sizeof(key) );
|
|
std::memset( &data, 0, sizeof(data) );
|
|
|
|
TQByteArray a1;
|
|
{
|
|
TQDataStream stream( a1, IO_WriteOnly );
|
|
stream << id;
|
|
key.data = a1.data();
|
|
key.size = a1.size();
|
|
}
|
|
|
|
int ret = d->dbp->get( d->dbp, 0, &key, &data, 0 );
|
|
Q_ASSERT( ret == 0 );
|
|
|
|
Tag tag;
|
|
|
|
if( ret == 0 ){
|
|
TQByteArray a;
|
|
a.setRawData( (const char*) data.data, data.size );
|
|
TQDataStream stream( a, IO_ReadOnly );
|
|
tag.load( stream );
|
|
a.resetRawData( (const char*) data.data, data.size );
|
|
}
|
|
|
|
return tag;
|
|
}
|
|
|
|
/*!
|
|
\fn Catalog::sync()
|
|
*/
|
|
|
|
void Catalog::sync()
|
|
{
|
|
Q_ASSERT( d->dbp != 0 );
|
|
d->dbp->sync( d->dbp, 0 );
|
|
|
|
TQMap<TQCString, DB*>::Iterator it = d->indexList.begin();
|
|
while( it != d->indexList.end() ){
|
|
it.data()->sync( it.data(), 0 );
|
|
++it;
|
|
}
|
|
}
|
|
|
|
/*!
|
|
\fn Catalog::query( const TQValueList<QueryArgument>& args )
|
|
*/
|
|
|
|
TQValueList<Tag> Catalog::query( const TQValueList<QueryArgument>& args )
|
|
{
|
|
TQValueList<Tag> tags;
|
|
|
|
DBT key, data;
|
|
|
|
DBC** cursors = new DBC* [ args.size() + 1 ];
|
|
|
|
TQValueList< TQPair<TQCString,TQVariant> >::ConstIterator it = args.begin();
|
|
int current = 0;
|
|
while( it != args.end() ){
|
|
TQCString indexName = (*it).first;
|
|
TQVariant value = (*it).second;
|
|
|
|
if( d->hasIndex(indexName) ) {
|
|
DB* dbp = d->index( indexName );
|
|
Q_ASSERT( dbp != 0 );
|
|
|
|
std::memset( &key, 0, sizeof(key) );
|
|
std::memset( &data, 0, sizeof(data) );
|
|
|
|
TQByteArray a1;
|
|
{
|
|
TQDataStream stream( a1, IO_WriteOnly );
|
|
stream << value;
|
|
key.data = a1.data();
|
|
key.size = a1.size();
|
|
}
|
|
|
|
DBC* cursor = 0;
|
|
int rtn = dbp->cursor( dbp, 0, &cursor, 0 );
|
|
|
|
if ( rtn == 0 ) {
|
|
|
|
rtn = cursor->c_get( cursor, &key, &data, DB_SET );
|
|
|
|
if ( rtn == 0 ) {
|
|
cursors[ current++ ] = cursor;
|
|
}
|
|
else if ( rtn != DB_NOTFOUND) {
|
|
kdDebug() << "fetching cursor failed: " << db_strerror(rtn) << endl;
|
|
cursor->c_close(cursor);
|
|
}
|
|
}
|
|
else {
|
|
kdDebug() << "creating cursor failed: " << db_strerror(rtn) << endl;
|
|
}
|
|
}
|
|
++it;
|
|
}
|
|
|
|
cursors[ current ] = 0;
|
|
|
|
if( current == args.size() ) {
|
|
|
|
DBC* join_curs = 0;
|
|
int rtn = d->dbp->join( d->dbp, cursors, &join_curs, 0 );
|
|
|
|
if ( rtn == 0 ) {
|
|
|
|
std::memset( &key, 0, sizeof(key) );
|
|
std::memset( &data, 0, sizeof(data) );
|
|
|
|
while( join_curs->c_get(join_curs, &key, &data, 0) == 0 ) {
|
|
|
|
TQByteArray a2;
|
|
a2.setRawData( (const char*) data.data, data.size );
|
|
TQDataStream s( a2, IO_ReadOnly );
|
|
Tag tag;
|
|
tag.load( s );
|
|
a2.resetRawData( (const char*) data.data, data.size );
|
|
tags << tag;
|
|
}
|
|
|
|
join_curs->c_close( join_curs );
|
|
}
|
|
else {
|
|
kdDebug() << "joining results failed: " << db_strerror(rtn) << endl;
|
|
}
|
|
}
|
|
|
|
DBC** c = cursors;
|
|
while( *c != 0 ){
|
|
(*c)->c_close( *c );
|
|
++c;
|
|
}
|
|
delete[] cursors;
|
|
|
|
return tags;
|
|
}
|
|
|
|
TQCString Catalog::generateId()
|
|
{
|
|
static int n = 1;
|
|
TQCString asStr;
|
|
asStr.sprintf( "%05d", n++ );
|
|
return asStr;
|
|
}
|
|
|