connectivity/source/commontools/TKey.cxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  *
   9  * This file incorporates work covered by the following license notice:
  10  *
  11  *   Licensed to the Apache Software Foundation (ASF) under one or more
  12  *   contributor license agreements. See the NOTICE file distributed
  13  *   with this work for additional information regarding copyright
  14  *   ownership. The ASF licenses this file to you under the Apache
  15  *   License, Version 2.0 (the "License"); you may not use this file
  16  *   except in compliance with the License. You may obtain a copy of
  17  *   the License at http://www.apache.org/licenses/LICENSE-2.0 .
  18  */
  19
  20 #include <TKey.hxx>
  21 #include <TKeyColumns.hxx>
  22 #include <com/sun/star/sdbc/XRow.hpp>
  23 #include <com/sun/star/sdbc/XResultSet.hpp>
  24 #include <TConnection.hxx>
  25 #include <connectivity/TTableHelper.hxx>
  26
  27 using namespace connectivity;
  28 using namespace ::com::sun::star::uno;
  29 using namespace ::com::sun::star::sdbc;
  30
  31 OTableKeyHelper::OTableKeyHelper(OTableHelper* _pTable) : connectivity::sdbcx::OKey(true)
  32     ,m_pTable(_pTable)
  33 {
  34     construct();
  35 }
  36
  37 OTableKeyHelper::OTableKeyHelper(   OTableHelper* _pTable
  38             ,const OUString& Name
  39             ,const std::shared_ptr<sdbcx::KeyProperties>& _rProps
  40             ) : connectivity::sdbcx::OKey(Name,_rProps,true)
  41                 ,m_pTable(_pTable)
  42 {
  43     construct();
  44     refreshColumns();
  45 }
  46
  47 void OTableKeyHelper::refreshColumns()
  48 {
  49     if ( !m_pTable )
  50         return;
  51
  52     std::vector< OUString> aVector;
  53     if ( !isNew() )
  54     {
  55         aVector = m_aProps->m_aKeyColumnNames;
  56         if ( aVector.empty() )
  57         {
  58             ::dbtools::OPropertyMap& rPropMap = OMetaConnection::getPropMap();
  59             OUString aSchema,aTable;
  60             m_pTable->getPropertyValue(rPropMap.getNameByIndex(PROPERTY_ID_SCHEMANAME)) >>= aSchema;
  61             m_pTable->getPropertyValue(rPropMap.getNameByIndex(PROPERTY_ID_NAME))       >>= aTable;
  62
  63             if ( !m_Name.isEmpty() ) // foreign key
  64             {
  65
  66                 Reference< XResultSet > xResult = m_pTable->getMetaData()->getImportedKeys(m_pTable->getPropertyValue(rPropMap.getNameByIndex(PROPERTY_ID_CATALOGNAME)),
  67                     aSchema,aTable);
  68
  69                 if ( xResult.is() )
  70                 {
  71                     Reference< XRow > xRow(xResult,UNO_QUERY);
  72                     while( xResult->next() )
  73                     {
  74                         OUString aForeignKeyColumn = xRow->getString(8);
  75                         if(xRow->getString(12) == m_Name)
  76                             aVector.push_back(aForeignKeyColumn);
  77                     }
  78                 }
  79             }
  80
  81             if ( aVector.empty() )
  82             {
  83                 const Reference< XResultSet > xResult = m_pTable->getMetaData()->getPrimaryKeys(m_pTable->getPropertyValue(rPropMap.getNameByIndex(PROPERTY_ID_CATALOGNAME)),
  84                     aSchema,aTable);
  85
  86                 if ( xResult.is() )
  87                 {
  88                     const Reference< XRow > xRow(xResult,UNO_QUERY);
  89                     while( xResult->next() )
  90                         aVector.push_back(xRow->getString(4));
  91                 } // if ( xResult.is() )
  92             }
  93         }
  94     }
  95
  96
  97     if ( m_pColumns )
  98         m_pColumns->reFill(aVector);
  99     else
 100         m_pColumns.reset(new OKeyColumnsHelper(this,m_aMutex,aVector));
 101 }
 102
 103
 104 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */