/**************************************************************
*
* 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;
}
// -----------------------------------------------------------------------------