/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ /************************************************************************* * * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER. * * Copyright 2000, 2010 Oracle and/or its affiliates. * * OpenOffice.org - a multi-platform office productivity suite * * This file is part of OpenOffice.org. * * OpenOffice.org is free software: you can redistribute it and/or modify * it under the terms of the GNU Lesser General Public License version 3 * only, as published by the Free Software Foundation. * * OpenOffice.org 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 version 3 for more details * (a copy is included in the LICENSE file that accompanied this code). * * You should have received a copy of the GNU Lesser General Public License * version 3 along with OpenOffice.org. If not, see * * for a copy of the LGPLv3 License. * ************************************************************************/ // MARKER(update_precomp.py): autogen include statement, do not remove #include "precompiled_dbaccess.hxx" #include "TokenWriter.hxx" #include #include #include "dbu_misc.hrc" #include "sqlmessage.hxx" #include #include "dbustrings.hrc" #include #include #include using namespace dbaui; using namespace ::com::sun::star::uno; using namespace ::com::sun::star::beans; using namespace ::com::sun::star::container; using namespace ::com::sun::star::util; using namespace ::com::sun::star::sdbc; using namespace ::com::sun::star::sdb; using namespace ::com::sun::star::lang; // export data ORowSetImportExport::ORowSetImportExport( Window* _pParent, const Reference< XResultSetUpdate >& _xResultSetUpdate, const ::svx::ODataAccessDescriptor& _aDataDescriptor, const Reference< XMultiServiceFactory >& _rM, const String& rExchange ) : ODatabaseImportExport(_aDataDescriptor,_rM,NULL,rExchange) ,m_xTargetResultSetUpdate(_xResultSetUpdate) ,m_xTargetRowUpdate(_xResultSetUpdate,UNO_QUERY) ,m_pParent(_pParent) ,m_bAlreadyAsked(sal_False) { RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::ORowSetImportExport" ); OSL_ENSURE(_pParent,"Window can't be null!"); } // ----------------------------------------------------------------------------- void ORowSetImportExport::initialize() { RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::initialize" ); ODatabaseImportExport::initialize(); // do namemapping Reference xColumnLocate(m_xResultSet,UNO_QUERY); OSL_ENSURE(xColumnLocate.is(),"The rowset normally should support this"); m_xTargetResultSetMetaData = Reference(m_xTargetResultSetUpdate,UNO_QUERY)->getMetaData(); if(!m_xTargetResultSetMetaData.is() || !xColumnLocate.is() || !m_xResultSetMetaData.is() ) throw SQLException(String(ModuleRes(STR_UNEXPECTED_ERROR)),*this,::rtl::OUString(RTL_CONSTASCII_USTRINGPARAM("S1000")) ,0,Any()); sal_Int32 nCount = m_xTargetResultSetMetaData->getColumnCount(); m_aColumnMapping.reserve(nCount); m_aColumnTypes.reserve(nCount); for (sal_Int32 i = 1;i <= nCount; ++i) { sal_Int32 nPos = -1; // -1 means column is autoincrement or doesn't exist if(!m_xTargetResultSetMetaData->isAutoIncrement(i)) { try { ::rtl::OUString sColumnName = m_xTargetResultSetMetaData->getColumnName(i); nPos = xColumnLocate->findColumn(sColumnName); } catch(const SQLException&) { if(m_xTargetResultSetMetaData->isNullable(i)) nPos = 0; // column doesn't exist but we could set it to null } } m_aColumnMapping.push_back(nPos); if(nPos > 0) m_aColumnTypes.push_back(m_xResultSetMetaData->getColumnType(nPos)); else m_aColumnTypes.push_back(DataType::OTHER); } } // ----------------------------------------------------------------------------- BOOL ORowSetImportExport::Write() { RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::Write" ); return TRUE; } // ----------------------------------------------------------------------------- BOOL ORowSetImportExport::Read() { RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::Read" ); // check if there is any column to copy if(::std::find_if(m_aColumnMapping.begin(),m_aColumnMapping.end(), ::std::bind2nd(::std::greater(),0)) == m_aColumnMapping.end()) return FALSE; sal_Bool bContinue = sal_True; if(m_aSelection.getLength()) { const Any* pBegin = m_aSelection.getConstArray(); const Any* pEnd = pBegin + m_aSelection.getLength(); for(;pBegin != pEnd && bContinue;++pBegin) { sal_Int32 nPos = -1; *pBegin >>= nPos; OSL_ENSURE(nPos != -1,"Invalid position!"); bContinue = (m_xResultSet.is() && m_xResultSet->absolute(nPos) && insertNewRow()); } } else { Reference xProp(m_xResultSet,UNO_QUERY); sal_Int32 nRowCount = 0; sal_Int32 nCurrentRow = 0; sal_Int32 nRowFilterIndex = 0; if ( xProp.is() && xProp->getPropertySetInfo()->hasPropertyByName(PROPERTY_ISROWCOUNTFINAL) ) { sal_Bool bFinal = sal_False; xProp->getPropertyValue(PROPERTY_ISROWCOUNTFINAL) >>= bFinal; if ( !bFinal ) m_xResultSet->afterLast(); xProp->getPropertyValue(PROPERTY_ROWCOUNT) >>= nRowCount; } if ( !nRowCount ) { m_xResultSet->afterLast(); nRowCount = m_xResultSet->getRow(); } OSL_ENSURE(nRowCount,"RowCount is 0!"); m_xResultSet->beforeFirst(); while(m_xResultSet.is() && m_xResultSet->next() && bContinue && nRowCount ) { --nRowCount; ++nCurrentRow; if(!m_pRowMarker || m_pRowMarker[nRowFilterIndex] == nCurrentRow) { ++nRowFilterIndex; bContinue = insertNewRow(); } } } return TRUE; } // ----------------------------------------------------------------------------- sal_Bool ORowSetImportExport::insertNewRow() { RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::insertNewRow" ); try { m_xTargetResultSetUpdate->moveToInsertRow(); sal_Int32 i = 1; ::std::vector::iterator aEnd = m_aColumnMapping.end(); for (::std::vector::iterator aIter = m_aColumnMapping.begin(); aIter != aEnd ;++aIter,++i ) { if(*aIter > 0) { Any aValue; switch(m_aColumnTypes[i-1]) { case DataType::CHAR: case DataType::VARCHAR: aValue <<= m_xRow->getString(*aIter); break; case DataType::DECIMAL: case DataType::NUMERIC: aValue <<= m_xRow->getDouble(*aIter); break; case DataType::BIGINT: aValue <<= m_xRow->getLong(*aIter); break; case DataType::FLOAT: aValue <<= m_xRow->getFloat(*aIter); break; case DataType::DOUBLE: aValue <<= m_xRow->getDouble(*aIter); break; case DataType::LONGVARCHAR: aValue <<= m_xRow->getString(*aIter); break; case DataType::LONGVARBINARY: aValue <<= m_xRow->getBytes(*aIter); break; case DataType::DATE: aValue <<= m_xRow->getDate(*aIter); break; case DataType::TIME: aValue <<= m_xRow->getTime(*aIter); break; case DataType::TIMESTAMP: aValue <<= m_xRow->getTimestamp(*aIter); break; case DataType::BIT: case DataType::BOOLEAN: aValue <<= m_xRow->getBoolean(*aIter); break; case DataType::TINYINT: aValue <<= m_xRow->getByte(*aIter); break; case DataType::SMALLINT: aValue <<= m_xRow->getShort(*aIter); break; case DataType::INTEGER: aValue <<= m_xRow->getInt(*aIter); break; case DataType::REAL: aValue <<= m_xRow->getDouble(*aIter); break; case DataType::BINARY: case DataType::VARBINARY: aValue <<= m_xRow->getBytes(*aIter); break; case DataType::BLOB: aValue <<= m_xRow->getBlob(*aIter); break; case DataType::CLOB: aValue <<= m_xRow->getClob(*aIter); break; default: OSL_ENSURE(0,"Unknown type"); } if(m_xRow->wasNull()) m_xTargetRowUpdate->updateNull(i); else m_xTargetRowUpdate->updateObject(i,aValue); } else if(*aIter == 0)//now we have know that we to set this column to null m_xTargetRowUpdate->updateNull(i); } m_xTargetResultSetUpdate->insertRow(); } catch(const SQLException&) { if(!m_bAlreadyAsked) { String sAskIfContinue = String(ModuleRes(STR_ERROR_OCCURRED_WHILE_COPYING)); OSQLWarningBox aDlg( m_pParent, sAskIfContinue, WB_YES_NO | WB_DEF_YES ); if(aDlg.Execute() == RET_YES) m_bAlreadyAsked = sal_True; else return sal_False; } } return sal_True; } // ----------------------------------------------------------------------------- /* vim:set shiftwidth=4 softtabstop=4 expandtab: */