275 lines
11 KiB
C++
275 lines
11 KiB
C++
|
|
/* -*- 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
|
|
* <http://www.openoffice.org/license.html>
|
|
* 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 <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;
|
|
|
|
// 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 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<sal_Int32>(),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<XPropertySet> 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<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_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: */
|