/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ /* * This file is part of the LibreOffice project. * * This Source Code Form is subject to the terms of the Mozilla Public * License, v. 2.0. If a copy of the MPL was not distributed with this * file, You can obtain one at http://mozilla.org/MPL/2.0/. * * This file incorporates work covered by the following license notice: * * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed * with this work for additional information regarding copyright * ownership. The ASF licenses this file to you under the Apache * License, Version 2.0 (the "License"); you may not use this file * except in compliance with the License. You may obtain a copy of * the License at http://www.apache.org/licenses/LICENSE-2.0 . */ #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< XComponentContext >& _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,OUString("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 { 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); } } // ----------------------------------------------------------------------------- sal_Bool ORowSetImportExport::Write() { RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::Write" ); return sal_True; } // ----------------------------------------------------------------------------- sal_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 sal_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 sal_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_FAIL("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: */