diff options
Diffstat (limited to 'rasodmg/partinsert.cc')
| -rw-r--r-- | rasodmg/partinsert.cc | 199 |
1 files changed, 199 insertions, 0 deletions
diff --git a/rasodmg/partinsert.cc b/rasodmg/partinsert.cc new file mode 100644 index 0000000..566f287 --- /dev/null +++ b/rasodmg/partinsert.cc @@ -0,0 +1,199 @@ +/* +* This file is part of rasdaman community. +* +* Rasdaman community is free software: you can redistribute it and/or modify +* it under the terms of the GNU General Public License as published by +* the Free Software Foundation, either version 3 of the License, or +* (at your option) any later version. +* +* Rasdaman community 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 General Public License for more details. +* +* You should have received a copy of the GNU General Public License +* along with rasdaman community. If not, see <http://www.gnu.org/licenses/>. +* +* Copyright 2003, 2004, 2005, 2006, 2007, 2008, 2009 Peter Baumann / +rasdaman GmbH. +* +* For more information please see <http://www.rasdaman.org> +* or contact Peter Baumann via <baumann@rasdaman.com>. +/ +/** + * INCLUDE: partinsert.cc + * + * MODULE: rasodmg + * CLASS: r_Partial_Insert + * + * COMMENTS: + * None +*/ + +#include <string.h> +#include <stdio.h> +#include <vector> + +#include "raslib/rminit.hh" +#include "raslib/rmdebug.hh" +#include "raslib/minterval.hh" +#include "raslib/error.hh" +#include "rasodmg/storagelayout.hh" +#include "rasodmg/alignedtiling.hh" +#include "rasodmg/gmarray.hh" +#include "rasodmg/database.hh" +#include "rasodmg/oqlquery.hh" +#include "rasodmg/partinsert.hh" + + + +// format string for creating collections; parameters: collection name, set type +const char *r_Partial_Insert::format_create = "CREATE COLLECTION %s %s"; +// format string for updating the MDD; parameters: collection name, local oid +const char *r_Partial_Insert::format_update = "UPDATE %s AS x SET x ASSIGN $1 WHERE OID(x) = %f"; + + +r_Partial_Insert::r_Partial_Insert( r_Database &usedb, const char *collname, const char *mddtype, const char *settype, const r_Storage_Layout &stl ) : mydb(usedb) +{ + init_share(collname, mddtype, settype); + mystl = stl.clone(); +} + + +r_Partial_Insert::r_Partial_Insert( r_Database &usedb, const char *collname, const char *mddtype, const char *settype, const r_Minterval &dom, unsigned int tsize ) : mydb(usedb) +{ + init_share(collname, mddtype, settype); + r_Aligned_Tiling *tilingObj = new r_Aligned_Tiling(dom, tsize * dom.cell_count()); + mystl = new r_Storage_Layout(tilingObj); +} + + +r_Partial_Insert::r_Partial_Insert( const r_Partial_Insert &src ) : mydb(src.mydb) +{ + init_share(src.collName, src.mddType, src.setType); + mystl = src.mystl->clone(); +} + + +r_Partial_Insert::~r_Partial_Insert( void ) +{ + if (collName != NULL) + delete [] collName; + if (mddType != NULL) + delete [] mddType; + if (setType != NULL) + delete [] setType; + if (mystl != NULL) + delete mystl; +} + + +void r_Partial_Insert::init_share( const char *collname, const char *mddtype, const char *settype ) +{ + collName = new char[strlen(collname)+1]; + strcpy(collName, collname); + mddType = new char[strlen(mddtype)+1]; + strcpy(mddType, mddtype); + setType = new char[strlen(settype)+1]; + strcpy(setType, settype); + doUpdate = 0; +} + +int r_Partial_Insert::update( r_GMarray *mddPtr, + r_Data_Format transferFormat, + const char* transferFormatParams, + r_Data_Format storageFormat, + const char* storageFormatParams + ) +{ + RMDBGENTER(0, RMDebug::module_rasodmg, "r_Partial_Insert", "update()"); + try { + mddPtr->set_storage_layout(mystl->clone()); + mddPtr->set_type_by_name(mddType); + } + catch (r_Error &err) { + RMInit::logOut << "r_Partial_Insert::update(): unable to set storage_layout for the currend MDD: " + << err.what() << endl; + return -1; + } + + if (doUpdate == 0) + { + char *queryBuffer = new char[strlen(format_create) + strlen(collName) + strlen(setType)]; + sprintf(queryBuffer, format_create, collName, setType); + // first try creating the collection + try + { + // this is basically 3 bytes to long, but that doesn't matter + myta.begin(); + mydb.set_transfer_format(transferFormat, transferFormatParams); + mydb.set_storage_format(storageFormat, storageFormatParams); + r_OQL_Query query(queryBuffer); + r_oql_execute(query); + myta.commit(); + RMDBGMIDDLE(0, RMDebug::module_rasodmg, "r_Partial_Insert", "update(): created new collection " << collName << " with type " << setType ); + } + catch (r_Error &err) + { + RMDBGMIDDLE(0, RMDebug::module_rasodmg, "r_Partial_Insert", "update(): can't create collection: " << err.what() ); + myta.abort(); + // failure to create the collection is not an error + } + delete [] queryBuffer; + + // now create the root object + try + { + myta.begin(); + mydb.set_transfer_format(transferFormat, transferFormatParams); + mydb.set_storage_format(storageFormat, storageFormatParams); + r_Ref<r_GMarray> mddp = new (&mydb, mddType) r_GMarray(*mddPtr); + r_Ref<r_Set<r_Ref<r_GMarray> > > mddCollPtr; + mddCollPtr = (r_Ref<r_Set<r_Ref<r_GMarray> > >)(mydb.lookup_object(collName)); + mddCollPtr->insert_element(mddp); + myOId = mddp->get_oid(); + myta.commit(); + RMDBGMIDDLE(0, RMDebug::module_rasodmg, "r_Partial_Insert", "update(): reated root object OK, oid = " << myOId ); + doUpdate = 1; + } + catch (r_Error &err) + { + RMInit::logOut << "r_Partial_Insert::update(): unable to create root object: " + << err.what() << endl; + myta.abort(); + return -1; + } + } + else + { + char *queryBuffer = new char[strlen(format_update) + strlen(collName) + 32]; + sprintf(queryBuffer, format_update, collName, myOId.get_local_oid()); + + // try the update + try + { + myta.begin(); + mydb.set_transfer_format(transferFormat, transferFormatParams); + mydb.set_storage_format(storageFormat, storageFormatParams); + RMDBGMIDDLE(1, RMDebug::module_rasodmg, "r_Partial_Insert", "update(): QUERY: " << queryBuffer ); + r_OQL_Query query(queryBuffer); + query << (*mddPtr); + r_oql_execute(query); + myta.commit(); + RMDBGMIDDLE(0, RMDebug::module_rasodmg, "r_Partial_Insert", "update(): update object OK" ); + } + catch (r_Error &err) + { + RMInit::logOut << "r_Partial_Insert::update(): failed to update marray: " + << err.what() << endl; + myta.abort(); + delete [] queryBuffer; + RMDBGEXIT(1, RMDebug::module_rasodmg, "r_Partial_Insert", "failed"); + return -1; + } + delete [] queryBuffer; + } + RMDBGEXIT(1, RMDebug::module_rasodmg, "r_Partial_Insert", "update(): exiting OK" ); + + return 0; +} |