LCOV - code coverage report
Current view: top level - dbaccess/source/ui/misc - RowSetDrop.cxx (source / functions) Hit Total Coverage
Test: commit c8344322a7af75b84dd3ca8f78b05543a976dfd5 Lines: 1 122 0.8 %
Date: 2015-06-13 12:38:46 Functions: 2 7 28.6 %
Legend: Lines: hit not hit

          Line data    Source code
       1             : 
       2             : /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
       3             : /*
       4             :  * This file is part of the LibreOffice project.
       5             :  *
       6             :  * This Source Code Form is subject to the terms of the Mozilla Public
       7             :  * License, v. 2.0. If a copy of the MPL was not distributed with this
       8             :  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
       9             :  *
      10             :  * This file incorporates work covered by the following license notice:
      11             :  *
      12             :  *   Licensed to the Apache Software Foundation (ASF) under one or more
      13             :  *   contributor license agreements. See the NOTICE file distributed
      14             :  *   with this work for additional information regarding copyright
      15             :  *   ownership. The ASF licenses this file to you under the Apache
      16             :  *   License, Version 2.0 (the "License"); you may not use this file
      17             :  *   except in compliance with the License. You may obtain a copy of
      18             :  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
      19             :  */
      20             : 
      21             : #include "TokenWriter.hxx"
      22             : #include <com/sun/star/sdbc/XColumnLocate.hpp>
      23             : #include <com/sun/star/sdbc/XResultSetMetaDataSupplier.hpp>
      24             : #include "dbu_misc.hrc"
      25             : #include "sqlmessage.hxx"
      26             : #include <vcl/msgbox.hxx>
      27             : #include "dbustrings.hrc"
      28             : #include <com/sun/star/sdbc/XRowUpdate.hpp>
      29             : #include <functional>
      30             : 
      31             : using namespace dbaui;
      32             : using namespace ::com::sun::star::uno;
      33             : using namespace ::com::sun::star::beans;
      34             : using namespace ::com::sun::star::container;
      35             : using namespace ::com::sun::star::util;
      36             : using namespace ::com::sun::star::sdbc;
      37             : using namespace ::com::sun::star::sdb;
      38             : using namespace ::com::sun::star::lang;
      39             : 
      40             : // export data
      41           0 : ORowSetImportExport::ORowSetImportExport(   vcl::Window* _pParent,
      42             :                                             const Reference< XResultSetUpdate >& _xResultSetUpdate,
      43             :                                             const svx::ODataAccessDescriptor& _aDataDescriptor,
      44             :                                             const Reference< XComponentContext >& _rM,
      45             :                                             const OUString& rExchange
      46             :                                             )
      47             :                                             : ODatabaseImportExport(_aDataDescriptor,_rM,NULL,rExchange)
      48             :                                             ,m_xTargetResultSetUpdate(_xResultSetUpdate)
      49             :                                             ,m_xTargetRowUpdate(_xResultSetUpdate,UNO_QUERY)
      50             :                                             ,m_pParent(_pParent)
      51           0 :                                             ,m_bAlreadyAsked(false)
      52             : {
      53             :     OSL_ENSURE(_pParent,"Window can't be null!");
      54           0 : }
      55             : 
      56           0 : void ORowSetImportExport::initialize()
      57             : {
      58           0 :     ODatabaseImportExport::initialize();
      59             :     // do namemapping
      60           0 :     Reference<XColumnLocate> xColumnLocate(m_xResultSet,UNO_QUERY);
      61             :     OSL_ENSURE(xColumnLocate.is(),"The rowset normally should support this");
      62             : 
      63           0 :     m_xTargetResultSetMetaData = Reference<XResultSetMetaDataSupplier>(m_xTargetResultSetUpdate,UNO_QUERY)->getMetaData();
      64           0 :     if(!m_xTargetResultSetMetaData.is() || !xColumnLocate.is() || !m_xResultSetMetaData.is() )
      65           0 :         throw SQLException(ModuleRes(STR_UNEXPECTED_ERROR),*this,OUString("S1000") ,0,Any());
      66             : 
      67           0 :     sal_Int32 nCount = m_xTargetResultSetMetaData->getColumnCount();
      68           0 :     m_aColumnMapping.reserve(nCount);
      69           0 :     m_aColumnTypes.reserve(nCount);
      70           0 :     for (sal_Int32 i = 1;i <= nCount; ++i)
      71             :     {
      72           0 :         sal_Int32 nPos = -1; // -1 means column is autoincrement or doesn't exist
      73           0 :         if(!m_xTargetResultSetMetaData->isAutoIncrement(i))
      74             :         {
      75             :             try
      76             :             {
      77           0 :                 OUString sColumnName = m_xTargetResultSetMetaData->getColumnName(i);
      78           0 :                 nPos = xColumnLocate->findColumn(sColumnName);
      79             :             }
      80           0 :             catch(const SQLException&)
      81             :             {
      82           0 :                 if(m_xTargetResultSetMetaData->isNullable(i))
      83           0 :                     nPos = 0; // column doesn't exist but we could set it to null
      84             :             }
      85             :         }
      86             : 
      87           0 :         m_aColumnMapping.push_back(nPos);
      88           0 :         if(nPos > 0)
      89           0 :             m_aColumnTypes.push_back(m_xResultSetMetaData->getColumnType(nPos));
      90             :         else
      91           0 :             m_aColumnTypes.push_back(DataType::OTHER);
      92           0 :     }
      93           0 : }
      94             : 
      95           0 : bool ORowSetImportExport::Write()
      96             : {
      97           0 :     return true;
      98             : }
      99             : 
     100           0 : bool ORowSetImportExport::Read()
     101             : {
     102             :     // check if there is any column to copy
     103           0 :     if(::std::find_if(m_aColumnMapping.begin(),m_aColumnMapping.end(),
     104           0 :                         ::std::bind2nd(::std::greater<sal_Int32>(),0)) == m_aColumnMapping.end())
     105           0 :         return false;
     106           0 :     bool bContinue = true;
     107           0 :     if(m_aSelection.getLength())
     108             :     {
     109           0 :         const Any* pBegin = m_aSelection.getConstArray();
     110           0 :         const Any* pEnd   = pBegin + m_aSelection.getLength();
     111           0 :         for(;pBegin != pEnd && bContinue;++pBegin)
     112             :         {
     113           0 :             sal_Int32 nPos = -1;
     114           0 :             *pBegin >>= nPos;
     115             :             OSL_ENSURE(nPos != -1,"Invalid position!");
     116           0 :             bContinue = (m_xResultSet.is() && m_xResultSet->absolute(nPos) && insertNewRow());
     117             :         }
     118             :     }
     119             :     else
     120             :     {
     121           0 :         Reference<XPropertySet> xProp(m_xResultSet,UNO_QUERY);
     122           0 :         sal_Int32 nRowCount = 0;
     123           0 :         sal_Int32 nCurrentRow = 0;
     124           0 :         sal_Int32 nRowFilterIndex = 0;
     125           0 :         if ( xProp.is() && xProp->getPropertySetInfo()->hasPropertyByName(PROPERTY_ISROWCOUNTFINAL) )
     126             :         {
     127           0 :             bool bFinal = false;
     128           0 :             xProp->getPropertyValue(PROPERTY_ISROWCOUNTFINAL) >>= bFinal;
     129           0 :             if ( !bFinal )
     130           0 :                 m_xResultSet->afterLast();
     131           0 :             xProp->getPropertyValue(PROPERTY_ROWCOUNT) >>= nRowCount;
     132             :         }
     133           0 :         if ( !nRowCount )
     134             :         {
     135           0 :             m_xResultSet->afterLast();
     136           0 :             nRowCount = m_xResultSet->getRow();
     137             :         }
     138             :         OSL_ENSURE(nRowCount,"RowCount is 0!");
     139           0 :         m_xResultSet->beforeFirst();
     140           0 :         while(m_xResultSet.is() && m_xResultSet->next() && bContinue && nRowCount )
     141             :         {
     142           0 :             --nRowCount;
     143           0 :             ++nCurrentRow;
     144           0 :             if(!m_pRowMarker || m_pRowMarker[nRowFilterIndex] == nCurrentRow)
     145             :             {
     146           0 :                 ++nRowFilterIndex;
     147           0 :                 bContinue = insertNewRow();
     148             :             }
     149           0 :         }
     150             :     }
     151           0 :     return true;
     152             : }
     153             : 
     154           0 : bool ORowSetImportExport::insertNewRow()
     155             : {
     156             :     try
     157             :     {
     158           0 :         m_xTargetResultSetUpdate->moveToInsertRow();
     159           0 :         sal_Int32 i = 1;
     160           0 :         ::std::vector<sal_Int32>::iterator aEnd = m_aColumnMapping.end();
     161           0 :         for (::std::vector<sal_Int32>::iterator aIter = m_aColumnMapping.begin(); aIter != aEnd ;++aIter,++i )
     162             :         {
     163           0 :             if(*aIter > 0)
     164             :             {
     165           0 :                 Any aValue;
     166           0 :                 switch(m_aColumnTypes[i-1])
     167             :                 {
     168             :                     case DataType::CHAR:
     169             :                     case DataType::VARCHAR:
     170           0 :                         aValue <<= m_xRow->getString(*aIter);
     171           0 :                         break;
     172             :                     case DataType::DECIMAL:
     173             :                     case DataType::NUMERIC:
     174           0 :                         aValue <<= m_xRow->getDouble(*aIter);
     175           0 :                         break;
     176             :                     case DataType::BIGINT:
     177           0 :                         aValue <<= m_xRow->getLong(*aIter);
     178           0 :                         break;
     179             :                     case DataType::FLOAT:
     180           0 :                         aValue <<= m_xRow->getFloat(*aIter);
     181           0 :                         break;
     182             :                     case DataType::DOUBLE:
     183           0 :                         aValue <<= m_xRow->getDouble(*aIter);
     184           0 :                         break;
     185             :                     case DataType::LONGVARCHAR:
     186           0 :                         aValue <<= m_xRow->getString(*aIter);
     187           0 :                         break;
     188             :                     case DataType::LONGVARBINARY:
     189           0 :                         aValue <<= m_xRow->getBytes(*aIter);
     190           0 :                         break;
     191             :                     case DataType::DATE:
     192           0 :                         aValue <<= m_xRow->getDate(*aIter);
     193           0 :                         break;
     194             :                     case DataType::TIME:
     195           0 :                         aValue <<= m_xRow->getTime(*aIter);
     196           0 :                         break;
     197             :                     case DataType::TIMESTAMP:
     198           0 :                         aValue <<= m_xRow->getTimestamp(*aIter);
     199           0 :                         break;
     200             :                     case DataType::BIT:
     201             :                     case DataType::BOOLEAN:
     202           0 :                         aValue <<= m_xRow->getBoolean(*aIter);
     203           0 :                         break;
     204             :                     case DataType::TINYINT:
     205           0 :                         aValue <<= m_xRow->getByte(*aIter);
     206           0 :                         break;
     207             :                     case DataType::SMALLINT:
     208           0 :                         aValue <<= m_xRow->getShort(*aIter);
     209           0 :                         break;
     210             :                     case DataType::INTEGER:
     211           0 :                         aValue <<= m_xRow->getInt(*aIter);
     212           0 :                         break;
     213             :                     case DataType::REAL:
     214           0 :                         aValue <<= m_xRow->getDouble(*aIter);
     215           0 :                         break;
     216             :                     case DataType::BINARY:
     217             :                     case DataType::VARBINARY:
     218           0 :                         aValue <<= m_xRow->getBytes(*aIter);
     219           0 :                         break;
     220             :                     case DataType::BLOB:
     221           0 :                         aValue <<= m_xRow->getBlob(*aIter);
     222           0 :                         break;
     223             :                     case DataType::CLOB:
     224           0 :                         aValue <<= m_xRow->getClob(*aIter);
     225           0 :                         break;
     226             :                     default:
     227             :                         SAL_WARN("dbaccess.ui", "Unknown type");
     228             :                 }
     229           0 :                 if(m_xRow->wasNull())
     230           0 :                     m_xTargetRowUpdate->updateNull(i);
     231             :                 else
     232           0 :                     m_xTargetRowUpdate->updateObject(i,aValue);
     233             :             }
     234           0 :             else if(*aIter == 0)//now we have know that we to set this column to null
     235           0 :                 m_xTargetRowUpdate->updateNull(i);
     236             :         }
     237           0 :         m_xTargetResultSetUpdate->insertRow();
     238             :     }
     239           0 :     catch(const SQLException&)
     240             :     {
     241           0 :         if(!m_bAlreadyAsked)
     242             :         {
     243           0 :             OUString sAskIfContinue = ModuleRes(STR_ERROR_OCCURRED_WHILE_COPYING);
     244           0 :             ScopedVclPtrInstance< OSQLWarningBox > aDlg( m_pParent, sAskIfContinue, WB_YES_NO | WB_DEF_YES );
     245           0 :             if(aDlg->Execute() == RET_YES)
     246           0 :                 m_bAlreadyAsked = true;
     247             :             else
     248           0 :                 return false;
     249             :         }
     250           0 :     }
     251           0 :     return true;
     252          36 : }
     253             : 
     254             : /* vim:set shiftwidth=4 softtabstop=4 expandtab: */

Generated by: LCOV version 1.11