| /************************************************************** |
| * |
| * 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 |
| * |
| * Unless required by applicable law or agreed to in writing, |
| * software distributed under the License is distributed on an |
| * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY |
| * KIND, either express or implied. See the License for the |
| * specific language governing permissions and limitations |
| * under the License. |
| * |
| *************************************************************/ |
| |
| |
| |
| // MARKER(update_precomp.py): autogen include statement, do not remove |
| #include "precompiled_dbaccess.hxx" |
| |
| #include "TokenWriter.hxx" |
| #include <com/sun/star/sdbc/XColumnLocate.hpp> |
| #include <com/sun/star/sdbc/XResultSetMetaDataSupplier.hpp> |
| #include "dbu_misc.hrc" |
| #include "sqlmessage.hxx" |
| #include <vcl/msgbox.hxx> |
| #include "dbustrings.hrc" |
| #include <com/sun/star/sdbc/XRowUpdate.hpp> |
| #include <functional> |
| #include <rtl/logfile.hxx> |
| |
| 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; |
| // using namespace ::com::sun::star::sdbcx; |
| |
| // 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> xColumnLocate(m_xResultSet,UNO_QUERY); |
| OSL_ENSURE(xColumnLocate.is(),"The rowset normally should support this"); |
| |
| m_xTargetResultSetMetaData = Reference<XResultSetMetaDataSupplier>(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 exists |
| 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 exists 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<sal_Int32>(),0)) == m_aColumnMapping.end()) |
| return sal_False; |
| sal_Int32 nCurrentRow = 0; |
| sal_Int32 nRowFilterIndex = 0; |
| 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 posiotion!"); |
| bContinue = (m_xResultSet.is() && m_xResultSet->absolute(nPos) && insertNewRow()); |
| } |
| } |
| else |
| { |
| Reference<XPropertySet> xProp(m_xResultSet,UNO_QUERY); |
| sal_Int32 nRowCount = 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<sal_Int32>::iterator aEnd = m_aColumnMapping.end(); |
| for (::std::vector<sal_Int32>::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_OCCURED_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; |
| } |
| // ----------------------------------------------------------------------------- |
| |
| |
| |