Branch data 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 : :
22 : : #include "TokenWriter.hxx"
23 : : #include <com/sun/star/sdbc/XColumnLocate.hpp>
24 : : #include <com/sun/star/sdbc/XResultSetMetaDataSupplier.hpp>
25 : : #include "dbu_misc.hrc"
26 : : #include "sqlmessage.hxx"
27 : : #include <vcl/msgbox.hxx>
28 : : #include "dbustrings.hrc"
29 : : #include <com/sun/star/sdbc/XRowUpdate.hpp>
30 : : #include <functional>
31 : : #include <rtl/logfile.hxx>
32 : :
33 : : using namespace dbaui;
34 : : using namespace ::com::sun::star::uno;
35 : : using namespace ::com::sun::star::beans;
36 : : using namespace ::com::sun::star::container;
37 : : using namespace ::com::sun::star::util;
38 : : using namespace ::com::sun::star::sdbc;
39 : : using namespace ::com::sun::star::sdb;
40 : : using namespace ::com::sun::star::lang;
41 : :
42 : : // export data
43 : 0 : ORowSetImportExport::ORowSetImportExport( Window* _pParent,
44 : : const Reference< XResultSetUpdate >& _xResultSetUpdate,
45 : : const ::svx::ODataAccessDescriptor& _aDataDescriptor,
46 : : const Reference< XMultiServiceFactory >& _rM,
47 : : const String& rExchange
48 : : )
49 : : : ODatabaseImportExport(_aDataDescriptor,_rM,NULL,rExchange)
50 : : ,m_xTargetResultSetUpdate(_xResultSetUpdate)
51 : : ,m_xTargetRowUpdate(_xResultSetUpdate,UNO_QUERY)
52 : : ,m_pParent(_pParent)
53 [ # # ][ # # ]: 0 : ,m_bAlreadyAsked(sal_False)
[ # # ][ # # ]
[ # # ]
54 : : {
55 : : RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::ORowSetImportExport" );
56 : : OSL_ENSURE(_pParent,"Window can't be null!");
57 : 0 : }
58 : : // -----------------------------------------------------------------------------
59 : 0 : void ORowSetImportExport::initialize()
60 : : {
61 : : RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::initialize" );
62 [ # # ]: 0 : ODatabaseImportExport::initialize();
63 : : // do namemapping
64 [ # # ]: 0 : Reference<XColumnLocate> xColumnLocate(m_xResultSet,UNO_QUERY);
65 : : OSL_ENSURE(xColumnLocate.is(),"The rowset normally should support this");
66 : :
67 [ # # ][ # # ]: 0 : m_xTargetResultSetMetaData = Reference<XResultSetMetaDataSupplier>(m_xTargetResultSetUpdate,UNO_QUERY)->getMetaData();
[ # # ][ # # ]
68 [ # # ][ # # ]: 0 : if(!m_xTargetResultSetMetaData.is() || !xColumnLocate.is() || !m_xResultSetMetaData.is() )
[ # # ][ # # ]
69 [ # # ][ # # ]: 0 : throw SQLException(String(ModuleRes(STR_UNEXPECTED_ERROR)),*this,::rtl::OUString(RTL_CONSTASCII_USTRINGPARAM("S1000")) ,0,Any());
[ # # ][ # # ]
[ # # ][ # # ]
70 : :
71 [ # # ][ # # ]: 0 : sal_Int32 nCount = m_xTargetResultSetMetaData->getColumnCount();
72 [ # # ]: 0 : m_aColumnMapping.reserve(nCount);
73 [ # # ]: 0 : m_aColumnTypes.reserve(nCount);
74 [ # # ]: 0 : for (sal_Int32 i = 1;i <= nCount; ++i)
75 : : {
76 : 0 : sal_Int32 nPos = -1; // -1 means column is autoincrement or doesn't exist
77 [ # # ][ # # ]: 0 : if(!m_xTargetResultSetMetaData->isAutoIncrement(i))
[ # # ]
78 : : {
79 : : try
80 : : {
81 [ # # ][ # # ]: 0 : ::rtl::OUString sColumnName = m_xTargetResultSetMetaData->getColumnName(i);
82 [ # # ][ # # ]: 0 : nPos = xColumnLocate->findColumn(sColumnName);
83 : : }
84 [ # # # # ]: 0 : catch(const SQLException&)
85 : : {
86 [ # # # # : 0 : if(m_xTargetResultSetMetaData->isNullable(i))
# # ]
87 : 0 : nPos = 0; // column doesn't exist but we could set it to null
88 : : }
89 : : }
90 : :
91 [ # # ]: 0 : m_aColumnMapping.push_back(nPos);
92 [ # # ]: 0 : if(nPos > 0)
93 [ # # ][ # # ]: 0 : m_aColumnTypes.push_back(m_xResultSetMetaData->getColumnType(nPos));
[ # # ]
94 : : else
95 [ # # ]: 0 : m_aColumnTypes.push_back(DataType::OTHER);
96 : 0 : }
97 : 0 : }
98 : : // -----------------------------------------------------------------------------
99 : 0 : sal_Bool ORowSetImportExport::Write()
100 : : {
101 : : RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::Write" );
102 : 0 : return sal_True;
103 : : }
104 : : // -----------------------------------------------------------------------------
105 : 0 : sal_Bool ORowSetImportExport::Read()
106 : : {
107 : : RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::Read" );
108 : : // check if there is any column to copy
109 [ # # ][ # # ]: 0 : if(::std::find_if(m_aColumnMapping.begin(),m_aColumnMapping.end(),
110 [ # # ][ # # ]: 0 : ::std::bind2nd(::std::greater<sal_Int32>(),0)) == m_aColumnMapping.end())
111 : 0 : return sal_False;
112 : 0 : sal_Bool bContinue = sal_True;
113 [ # # ]: 0 : if(m_aSelection.getLength())
114 : : {
115 : 0 : const Any* pBegin = m_aSelection.getConstArray();
116 : 0 : const Any* pEnd = pBegin + m_aSelection.getLength();
117 [ # # ][ # # ]: 0 : for(;pBegin != pEnd && bContinue;++pBegin)
[ # # ]
118 : : {
119 : 0 : sal_Int32 nPos = -1;
120 : 0 : *pBegin >>= nPos;
121 : : OSL_ENSURE(nPos != -1,"Invalid position!");
122 [ # # ][ # # ]: 0 : bContinue = (m_xResultSet.is() && m_xResultSet->absolute(nPos) && insertNewRow());
[ # # ][ # # ]
[ # # ][ # # ]
123 : : }
124 : : }
125 : : else
126 : : {
127 [ # # ]: 0 : Reference<XPropertySet> xProp(m_xResultSet,UNO_QUERY);
128 : 0 : sal_Int32 nRowCount = 0;
129 : 0 : sal_Int32 nCurrentRow = 0;
130 : 0 : sal_Int32 nRowFilterIndex = 0;
131 [ # # ][ # # ]: 0 : if ( xProp.is() && xProp->getPropertySetInfo()->hasPropertyByName(PROPERTY_ISROWCOUNTFINAL) )
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ][ # #
# # # # ]
132 : : {
133 : 0 : sal_Bool bFinal = sal_False;
134 [ # # ][ # # ]: 0 : xProp->getPropertyValue(PROPERTY_ISROWCOUNTFINAL) >>= bFinal;
[ # # ]
135 [ # # ]: 0 : if ( !bFinal )
136 [ # # ][ # # ]: 0 : m_xResultSet->afterLast();
137 [ # # ][ # # ]: 0 : xProp->getPropertyValue(PROPERTY_ROWCOUNT) >>= nRowCount;
[ # # ]
138 : : }
139 [ # # ]: 0 : if ( !nRowCount )
140 : : {
141 [ # # ][ # # ]: 0 : m_xResultSet->afterLast();
142 [ # # ][ # # ]: 0 : nRowCount = m_xResultSet->getRow();
143 : : }
144 : : OSL_ENSURE(nRowCount,"RowCount is 0!");
145 [ # # ][ # # ]: 0 : m_xResultSet->beforeFirst();
146 [ # # ][ # # ]: 0 : while(m_xResultSet.is() && m_xResultSet->next() && bContinue && nRowCount )
[ # # ][ # # ]
[ # # ][ # # ]
[ # # ]
147 : : {
148 : 0 : --nRowCount;
149 : 0 : ++nCurrentRow;
150 [ # # ][ # # ]: 0 : if(!m_pRowMarker || m_pRowMarker[nRowFilterIndex] == nCurrentRow)
151 : : {
152 : 0 : ++nRowFilterIndex;
153 [ # # ]: 0 : bContinue = insertNewRow();
154 : : }
155 : 0 : }
156 : : }
157 : 0 : return sal_True;
158 : : }
159 : : // -----------------------------------------------------------------------------
160 : 0 : sal_Bool ORowSetImportExport::insertNewRow()
161 : : {
162 : : RTL_LOGFILE_CONTEXT_AUTHOR( aLogger, "misc", "Ocke.Janssen@sun.com", "ORowSetImportExport::insertNewRow" );
163 : : try
164 : : {
165 [ # # ][ # # ]: 0 : m_xTargetResultSetUpdate->moveToInsertRow();
166 : 0 : sal_Int32 i = 1;
167 : 0 : ::std::vector<sal_Int32>::iterator aEnd = m_aColumnMapping.end();
168 [ # # ][ # # ]: 0 : for (::std::vector<sal_Int32>::iterator aIter = m_aColumnMapping.begin(); aIter != aEnd ;++aIter,++i )
[ # # ]
169 : : {
170 [ # # ][ # # ]: 0 : if(*aIter > 0)
171 : : {
172 : 0 : Any aValue;
173 [ # # # # : 0 : switch(m_aColumnTypes[i-1])
# # # # #
# # # # #
# # # #
# ][ # # ]
174 : : {
175 : : case DataType::CHAR:
176 : : case DataType::VARCHAR:
177 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getString(*aIter);
[ # # ][ # # ]
178 : 0 : break;
179 : : case DataType::DECIMAL:
180 : : case DataType::NUMERIC:
181 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getDouble(*aIter);
[ # # ][ # # ]
182 : 0 : break;
183 : : case DataType::BIGINT:
184 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getLong(*aIter);
[ # # ][ # # ]
185 : 0 : break;
186 : : case DataType::FLOAT:
187 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getFloat(*aIter);
[ # # ][ # # ]
188 : 0 : break;
189 : : case DataType::DOUBLE:
190 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getDouble(*aIter);
[ # # ][ # # ]
191 : 0 : break;
192 : : case DataType::LONGVARCHAR:
193 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getString(*aIter);
[ # # ][ # # ]
194 : 0 : break;
195 : : case DataType::LONGVARBINARY:
196 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getBytes(*aIter);
[ # # ][ # # ]
[ # # ]
197 : 0 : break;
198 : : case DataType::DATE:
199 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getDate(*aIter);
[ # # ][ # # ]
200 : 0 : break;
201 : : case DataType::TIME:
202 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getTime(*aIter);
[ # # ][ # # ]
203 : 0 : break;
204 : : case DataType::TIMESTAMP:
205 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getTimestamp(*aIter);
[ # # ][ # # ]
206 : 0 : break;
207 : : case DataType::BIT:
208 : : case DataType::BOOLEAN:
209 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getBoolean(*aIter);
[ # # ][ # # ]
210 : 0 : break;
211 : : case DataType::TINYINT:
212 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getByte(*aIter);
[ # # ][ # # ]
213 : 0 : break;
214 : : case DataType::SMALLINT:
215 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getShort(*aIter);
[ # # ][ # # ]
216 : 0 : break;
217 : : case DataType::INTEGER:
218 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getInt(*aIter);
[ # # ][ # # ]
219 : 0 : break;
220 : : case DataType::REAL:
221 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getDouble(*aIter);
[ # # ][ # # ]
222 : 0 : break;
223 : : case DataType::BINARY:
224 : : case DataType::VARBINARY:
225 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getBytes(*aIter);
[ # # ][ # # ]
[ # # ]
226 : 0 : break;
227 : : case DataType::BLOB:
228 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getBlob(*aIter);
[ # # ][ # # ]
229 : 0 : break;
230 : : case DataType::CLOB:
231 [ # # ][ # # ]: 0 : aValue <<= m_xRow->getClob(*aIter);
[ # # ][ # # ]
232 : 0 : break;
233 : : default:
234 : : OSL_FAIL("Unknown type");
235 : : }
236 [ # # ][ # # ]: 0 : if(m_xRow->wasNull())
[ # # ]
237 [ # # ][ # # ]: 0 : m_xTargetRowUpdate->updateNull(i);
238 : : else
239 [ # # ][ # # ]: 0 : m_xTargetRowUpdate->updateObject(i,aValue);
240 : : }
241 [ # # ][ # # ]: 0 : else if(*aIter == 0)//now we have know that we to set this column to null
242 [ # # ][ # # ]: 0 : m_xTargetRowUpdate->updateNull(i);
243 : : }
244 [ # # ][ # # ]: 0 : m_xTargetResultSetUpdate->insertRow();
245 : : }
246 [ # # # # ]: 0 : catch(const SQLException&)
247 : : {
248 [ # # ]: 0 : if(!m_bAlreadyAsked)
249 : : {
250 [ # # # # ]: 0 : String sAskIfContinue = String(ModuleRes(STR_ERROR_OCCURRED_WHILE_COPYING));
251 [ # # ]: 0 : OSQLWarningBox aDlg( m_pParent, sAskIfContinue, WB_YES_NO | WB_DEF_YES );
252 [ # # # # ]: 0 : if(aDlg.Execute() == RET_YES)
253 : 0 : m_bAlreadyAsked = sal_True;
254 : : else
255 [ # # # # : 0 : return sal_False;
# # # # ]
256 : : }
257 : : }
258 : 0 : return sal_True;
259 : : }
260 : : // -----------------------------------------------------------------------------
261 : :
262 : :
263 : :
264 : : /* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|