196de5490SAndrew Rist /**************************************************************
2cdf0e10cSrcweir *
396de5490SAndrew Rist * Licensed to the Apache Software Foundation (ASF) under one
496de5490SAndrew Rist * or more contributor license agreements. See the NOTICE file
596de5490SAndrew Rist * distributed with this work for additional information
696de5490SAndrew Rist * regarding copyright ownership. The ASF licenses this file
796de5490SAndrew Rist * to you under the Apache License, Version 2.0 (the
896de5490SAndrew Rist * "License"); you may not use this file except in compliance
996de5490SAndrew Rist * with the License. You may obtain a copy of the License at
1096de5490SAndrew Rist *
1196de5490SAndrew Rist * http://www.apache.org/licenses/LICENSE-2.0
1296de5490SAndrew Rist *
1396de5490SAndrew Rist * Unless required by applicable law or agreed to in writing,
1496de5490SAndrew Rist * software distributed under the License is distributed on an
1596de5490SAndrew Rist * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
1696de5490SAndrew Rist * KIND, either express or implied. See the License for the
1796de5490SAndrew Rist * specific language governing permissions and limitations
1896de5490SAndrew Rist * under the License.
1996de5490SAndrew Rist *
2096de5490SAndrew Rist *************************************************************/
2196de5490SAndrew Rist
2296de5490SAndrew Rist
23cdf0e10cSrcweir
24cdf0e10cSrcweir // MARKER(update_precomp.py): autogen include statement, do not remove
25*b63233d8Sdamjan #include "precompiled_dbui.hxx"
26cdf0e10cSrcweir
27cdf0e10cSrcweir #include "TokenWriter.hxx"
28cdf0e10cSrcweir #include <com/sun/star/sdbc/XColumnLocate.hpp>
29cdf0e10cSrcweir #include <com/sun/star/sdbc/XResultSetMetaDataSupplier.hpp>
30cdf0e10cSrcweir #include "dbu_misc.hrc"
31cdf0e10cSrcweir #include "sqlmessage.hxx"
32cdf0e10cSrcweir #include <vcl/msgbox.hxx>
33cdf0e10cSrcweir #include "dbustrings.hrc"
34cdf0e10cSrcweir #include <com/sun/star/sdbc/XRowUpdate.hpp>
35cdf0e10cSrcweir #include <functional>
36cdf0e10cSrcweir #include <rtl/logfile.hxx>
37cdf0e10cSrcweir
38cdf0e10cSrcweir using namespace dbaui;
39cdf0e10cSrcweir using namespace ::com::sun::star::uno;
40cdf0e10cSrcweir using namespace ::com::sun::star::beans;
41cdf0e10cSrcweir using namespace ::com::sun::star::container;
42cdf0e10cSrcweir using namespace ::com::sun::star::util;
43cdf0e10cSrcweir using namespace ::com::sun::star::sdbc;
44cdf0e10cSrcweir using namespace ::com::sun::star::sdb;
45cdf0e10cSrcweir using namespace ::com::sun::star::lang;
46cdf0e10cSrcweir // using namespace ::com::sun::star::sdbcx;
47cdf0e10cSrcweir
48cdf0e10cSrcweir // export data
ORowSetImportExport(Window * _pParent,const Reference<XResultSetUpdate> & _xResultSetUpdate,const::svx::ODataAccessDescriptor & _aDataDescriptor,const Reference<XMultiServiceFactory> & _rM,const String & rExchange)49cdf0e10cSrcweir ORowSetImportExport::ORowSetImportExport( Window* _pParent,
50cdf0e10cSrcweir const Reference< XResultSetUpdate >& _xResultSetUpdate,
51cdf0e10cSrcweir const ::svx::ODataAccessDescriptor& _aDataDescriptor,
52cdf0e10cSrcweir const Reference< XMultiServiceFactory >& _rM,
53cdf0e10cSrcweir const String& rExchange
54cdf0e10cSrcweir )
55cdf0e10cSrcweir : ODatabaseImportExport(_aDataDescriptor,_rM,NULL,rExchange)
56cdf0e10cSrcweir ,m_xTargetResultSetUpdate(_xResultSetUpdate)
57cdf0e10cSrcweir ,m_xTargetRowUpdate(_xResultSetUpdate,UNO_QUERY)
58cdf0e10cSrcweir ,m_pParent(_pParent)
59cdf0e10cSrcweir ,m_bAlreadyAsked(sal_False)
60cdf0e10cSrcweir {
61cdf0e10cSrcweir RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::ORowSetImportExport" );
62cdf0e10cSrcweir OSL_ENSURE(_pParent,"Window can't be null!");
63cdf0e10cSrcweir }
64cdf0e10cSrcweir // -----------------------------------------------------------------------------
initialize()65cdf0e10cSrcweir void ORowSetImportExport::initialize()
66cdf0e10cSrcweir {
67cdf0e10cSrcweir RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::initialize" );
68cdf0e10cSrcweir ODatabaseImportExport::initialize();
69cdf0e10cSrcweir // do namemapping
70cdf0e10cSrcweir Reference<XColumnLocate> xColumnLocate(m_xResultSet,UNO_QUERY);
71cdf0e10cSrcweir OSL_ENSURE(xColumnLocate.is(),"The rowset normally should support this");
72cdf0e10cSrcweir
73cdf0e10cSrcweir m_xTargetResultSetMetaData = Reference<XResultSetMetaDataSupplier>(m_xTargetResultSetUpdate,UNO_QUERY)->getMetaData();
74cdf0e10cSrcweir if(!m_xTargetResultSetMetaData.is() || !xColumnLocate.is() || !m_xResultSetMetaData.is() )
75cdf0e10cSrcweir throw SQLException(String(ModuleRes(STR_UNEXPECTED_ERROR)),*this,::rtl::OUString(RTL_CONSTASCII_USTRINGPARAM("S1000")) ,0,Any());
76cdf0e10cSrcweir
77cdf0e10cSrcweir sal_Int32 nCount = m_xTargetResultSetMetaData->getColumnCount();
78cdf0e10cSrcweir m_aColumnMapping.reserve(nCount);
79cdf0e10cSrcweir m_aColumnTypes.reserve(nCount);
80cdf0e10cSrcweir for (sal_Int32 i = 1;i <= nCount; ++i)
81cdf0e10cSrcweir {
82cdf0e10cSrcweir sal_Int32 nPos = -1; // -1 means column is autoincrement or doesn't exists
83cdf0e10cSrcweir if(!m_xTargetResultSetMetaData->isAutoIncrement(i))
84cdf0e10cSrcweir {
85cdf0e10cSrcweir try
86cdf0e10cSrcweir {
87cdf0e10cSrcweir ::rtl::OUString sColumnName = m_xTargetResultSetMetaData->getColumnName(i);
88cdf0e10cSrcweir nPos = xColumnLocate->findColumn(sColumnName);
89cdf0e10cSrcweir }
90cdf0e10cSrcweir catch(const SQLException&)
91cdf0e10cSrcweir {
92cdf0e10cSrcweir if(m_xTargetResultSetMetaData->isNullable(i))
93cdf0e10cSrcweir nPos = 0; // column doesn't exists but we could set it to null
94cdf0e10cSrcweir }
95cdf0e10cSrcweir }
96cdf0e10cSrcweir
97cdf0e10cSrcweir m_aColumnMapping.push_back(nPos);
98cdf0e10cSrcweir if(nPos > 0)
99cdf0e10cSrcweir m_aColumnTypes.push_back(m_xResultSetMetaData->getColumnType(nPos));
100cdf0e10cSrcweir else
101cdf0e10cSrcweir m_aColumnTypes.push_back(DataType::OTHER);
102cdf0e10cSrcweir }
103cdf0e10cSrcweir }
104cdf0e10cSrcweir // -----------------------------------------------------------------------------
Write()105cdf0e10cSrcweir sal_Bool ORowSetImportExport::Write()
106cdf0e10cSrcweir {
107cdf0e10cSrcweir RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::Write" );
108cdf0e10cSrcweir return sal_True;
109cdf0e10cSrcweir }
110cdf0e10cSrcweir // -----------------------------------------------------------------------------
Read()111cdf0e10cSrcweir sal_Bool ORowSetImportExport::Read()
112cdf0e10cSrcweir {
113cdf0e10cSrcweir RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::Read" );
114cdf0e10cSrcweir // check if there is any column to copy
115cdf0e10cSrcweir if(::std::find_if(m_aColumnMapping.begin(),m_aColumnMapping.end(),
116cdf0e10cSrcweir ::std::bind2nd(::std::greater<sal_Int32>(),0)) == m_aColumnMapping.end())
117cdf0e10cSrcweir return sal_False;
118cdf0e10cSrcweir sal_Int32 nCurrentRow = 0;
119cdf0e10cSrcweir sal_Int32 nRowFilterIndex = 0;
120cdf0e10cSrcweir sal_Bool bContinue = sal_True;
121cdf0e10cSrcweir if(m_aSelection.getLength())
122cdf0e10cSrcweir {
123cdf0e10cSrcweir const Any* pBegin = m_aSelection.getConstArray();
124cdf0e10cSrcweir const Any* pEnd = pBegin + m_aSelection.getLength();
125cdf0e10cSrcweir for(;pBegin != pEnd && bContinue;++pBegin)
126cdf0e10cSrcweir {
127cdf0e10cSrcweir sal_Int32 nPos = -1;
128cdf0e10cSrcweir *pBegin >>= nPos;
129cdf0e10cSrcweir OSL_ENSURE(nPos != -1,"Invalid posiotion!");
130cdf0e10cSrcweir bContinue = (m_xResultSet.is() && m_xResultSet->absolute(nPos) && insertNewRow());
131cdf0e10cSrcweir }
132cdf0e10cSrcweir }
133cdf0e10cSrcweir else
134cdf0e10cSrcweir {
135cdf0e10cSrcweir Reference<XPropertySet> xProp(m_xResultSet,UNO_QUERY);
136cdf0e10cSrcweir sal_Int32 nRowCount = 0;
137cdf0e10cSrcweir if ( xProp.is() && xProp->getPropertySetInfo()->hasPropertyByName(PROPERTY_ISROWCOUNTFINAL) )
138cdf0e10cSrcweir {
139cdf0e10cSrcweir sal_Bool bFinal = sal_False;
140cdf0e10cSrcweir xProp->getPropertyValue(PROPERTY_ISROWCOUNTFINAL) >>= bFinal;
141cdf0e10cSrcweir if ( !bFinal )
142cdf0e10cSrcweir m_xResultSet->afterLast();
143cdf0e10cSrcweir xProp->getPropertyValue(PROPERTY_ROWCOUNT) >>= nRowCount;
144cdf0e10cSrcweir }
145cdf0e10cSrcweir if ( !nRowCount )
146cdf0e10cSrcweir {
147cdf0e10cSrcweir m_xResultSet->afterLast();
148cdf0e10cSrcweir nRowCount = m_xResultSet->getRow();
149cdf0e10cSrcweir }
150cdf0e10cSrcweir OSL_ENSURE(nRowCount,"RowCount is 0!");
151cdf0e10cSrcweir m_xResultSet->beforeFirst();
152cdf0e10cSrcweir while(m_xResultSet.is() && m_xResultSet->next() && bContinue && nRowCount )
153cdf0e10cSrcweir {
154cdf0e10cSrcweir --nRowCount;
155cdf0e10cSrcweir ++nCurrentRow;
156cdf0e10cSrcweir if(!m_pRowMarker || m_pRowMarker[nRowFilterIndex] == nCurrentRow)
157cdf0e10cSrcweir {
158cdf0e10cSrcweir ++nRowFilterIndex;
159cdf0e10cSrcweir bContinue = insertNewRow();
160cdf0e10cSrcweir }
161cdf0e10cSrcweir }
162cdf0e10cSrcweir }
163cdf0e10cSrcweir return sal_True;
164cdf0e10cSrcweir }
165cdf0e10cSrcweir // -----------------------------------------------------------------------------
insertNewRow()166cdf0e10cSrcweir sal_Bool ORowSetImportExport::insertNewRow()
167cdf0e10cSrcweir {
168cdf0e10cSrcweir RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::insertNewRow" );
169cdf0e10cSrcweir try
170cdf0e10cSrcweir {
171cdf0e10cSrcweir m_xTargetResultSetUpdate->moveToInsertRow();
172cdf0e10cSrcweir sal_Int32 i = 1;
173cdf0e10cSrcweir ::std::vector<sal_Int32>::iterator aEnd = m_aColumnMapping.end();
174cdf0e10cSrcweir for (::std::vector<sal_Int32>::iterator aIter = m_aColumnMapping.begin(); aIter != aEnd ;++aIter,++i )
175cdf0e10cSrcweir {
176cdf0e10cSrcweir if(*aIter > 0)
177cdf0e10cSrcweir {
178cdf0e10cSrcweir Any aValue;
179cdf0e10cSrcweir switch(m_aColumnTypes[i-1])
180cdf0e10cSrcweir {
181cdf0e10cSrcweir case DataType::CHAR:
182cdf0e10cSrcweir case DataType::VARCHAR:
183cdf0e10cSrcweir aValue <<= m_xRow->getString(*aIter);
184cdf0e10cSrcweir break;
185cdf0e10cSrcweir case DataType::DECIMAL:
186cdf0e10cSrcweir case DataType::NUMERIC:
187cdf0e10cSrcweir aValue <<= m_xRow->getDouble(*aIter);
188cdf0e10cSrcweir break;
189cdf0e10cSrcweir case DataType::BIGINT:
190cdf0e10cSrcweir aValue <<= m_xRow->getLong(*aIter);
191cdf0e10cSrcweir break;
192cdf0e10cSrcweir case DataType::FLOAT:
193cdf0e10cSrcweir aValue <<= m_xRow->getFloat(*aIter);
194cdf0e10cSrcweir break;
195cdf0e10cSrcweir case DataType::DOUBLE:
196cdf0e10cSrcweir aValue <<= m_xRow->getDouble(*aIter);
197cdf0e10cSrcweir break;
198cdf0e10cSrcweir case DataType::LONGVARCHAR:
199cdf0e10cSrcweir aValue <<= m_xRow->getString(*aIter);
200cdf0e10cSrcweir break;
201cdf0e10cSrcweir case DataType::LONGVARBINARY:
202cdf0e10cSrcweir aValue <<= m_xRow->getBytes(*aIter);
203cdf0e10cSrcweir break;
204cdf0e10cSrcweir case DataType::DATE:
205cdf0e10cSrcweir aValue <<= m_xRow->getDate(*aIter);
206cdf0e10cSrcweir break;
207cdf0e10cSrcweir case DataType::TIME:
208cdf0e10cSrcweir aValue <<= m_xRow->getTime(*aIter);
209cdf0e10cSrcweir break;
210cdf0e10cSrcweir case DataType::TIMESTAMP:
211cdf0e10cSrcweir aValue <<= m_xRow->getTimestamp(*aIter);
212cdf0e10cSrcweir break;
213cdf0e10cSrcweir case DataType::BIT:
214cdf0e10cSrcweir case DataType::BOOLEAN:
215cdf0e10cSrcweir aValue <<= m_xRow->getBoolean(*aIter);
216cdf0e10cSrcweir break;
217cdf0e10cSrcweir case DataType::TINYINT:
218cdf0e10cSrcweir aValue <<= m_xRow->getByte(*aIter);
219cdf0e10cSrcweir break;
220cdf0e10cSrcweir case DataType::SMALLINT:
221cdf0e10cSrcweir aValue <<= m_xRow->getShort(*aIter);
222cdf0e10cSrcweir break;
223cdf0e10cSrcweir case DataType::INTEGER:
224cdf0e10cSrcweir aValue <<= m_xRow->getInt(*aIter);
225cdf0e10cSrcweir break;
226cdf0e10cSrcweir case DataType::REAL:
227cdf0e10cSrcweir aValue <<= m_xRow->getDouble(*aIter);
228cdf0e10cSrcweir break;
229cdf0e10cSrcweir case DataType::BINARY:
230cdf0e10cSrcweir case DataType::VARBINARY:
231cdf0e10cSrcweir aValue <<= m_xRow->getBytes(*aIter);
232cdf0e10cSrcweir break;
233cdf0e10cSrcweir case DataType::BLOB:
234cdf0e10cSrcweir aValue <<= m_xRow->getBlob(*aIter);
235cdf0e10cSrcweir break;
236cdf0e10cSrcweir case DataType::CLOB:
237cdf0e10cSrcweir aValue <<= m_xRow->getClob(*aIter);
238cdf0e10cSrcweir break;
239cdf0e10cSrcweir default:
240cdf0e10cSrcweir OSL_ENSURE(0,"Unknown type");
241cdf0e10cSrcweir }
242cdf0e10cSrcweir if(m_xRow->wasNull())
243cdf0e10cSrcweir m_xTargetRowUpdate->updateNull(i);
244cdf0e10cSrcweir else
245cdf0e10cSrcweir m_xTargetRowUpdate->updateObject(i,aValue);
246cdf0e10cSrcweir }
247cdf0e10cSrcweir else if(*aIter == 0)//now we have know that we to set this column to null
248cdf0e10cSrcweir m_xTargetRowUpdate->updateNull(i);
249cdf0e10cSrcweir }
250cdf0e10cSrcweir m_xTargetResultSetUpdate->insertRow();
251cdf0e10cSrcweir }
252cdf0e10cSrcweir catch(const SQLException&)
253cdf0e10cSrcweir {
254cdf0e10cSrcweir if(!m_bAlreadyAsked)
255cdf0e10cSrcweir {
256cdf0e10cSrcweir String sAskIfContinue = String(ModuleRes(STR_ERROR_OCCURED_WHILE_COPYING));
257cdf0e10cSrcweir OSQLWarningBox aDlg( m_pParent, sAskIfContinue, WB_YES_NO | WB_DEF_YES );
258cdf0e10cSrcweir if(aDlg.Execute() == RET_YES)
259cdf0e10cSrcweir m_bAlreadyAsked = sal_True;
260cdf0e10cSrcweir else
261cdf0e10cSrcweir return sal_False;
262cdf0e10cSrcweir }
263cdf0e10cSrcweir }
264cdf0e10cSrcweir return sal_True;
265cdf0e10cSrcweir }
266cdf0e10cSrcweir // -----------------------------------------------------------------------------
267cdf0e10cSrcweir
268cdf0e10cSrcweir
269cdf0e10cSrcweir
270