connectivity/source/commontools/TKey.cxx

   1 /*************************************************************************
   2  *
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * Copyright 2008 by Sun Microsystems, Inc.
   6  *
   7  * OpenOffice.org - a multi-platform office productivity suite
   8  *
   9  * $RCSfile: TKey.cxx,v $
  10  * $Revision: 1.5 $
  11  *
  12  * This file is part of OpenOffice.org.
  13  *
  14  * OpenOffice.org is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU Lesser General Public License version 3
  16  * only, as published by the Free Software Foundation.
  17  *
  18  * OpenOffice.org is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU Lesser General Public License version 3 for more details
  22  * (a copy is included in the LICENSE file that accompanied this code).
  23  *
  24  * You should have received a copy of the GNU Lesser General Public License
  25  * version 3 along with OpenOffice.org.  If not, see
  26  * <http://www.openoffice.org/license.html>
  27  * for a copy of the LGPLv3 License.
  28  *
  29  ************************************************************************/
  30
  31 // MARKER(update_precomp.py): autogen include statement, do not remove
  32 #include "precompiled_connectivity.hxx"
  33
  34
  35 #include "connectivity/TKey.hxx"
  36 #include "connectivity/TKeyColumns.hxx"
  37 #include <com/sun/star/sdbc/XRow.hpp>
  38 #include <com/sun/star/sdbc/XResultSet.hpp>
  39 #include "TConnection.hxx"
  40 #include "connectivity/TTableHelper.hxx"
  41
  42 using namespace connectivity;
  43 using namespace ::com::sun::star::uno;
  44 using namespace ::com::sun::star::beans;
  45 //      using namespace ::com::sun::star::sdbcx;
  46 using namespace ::com::sun::star::sdbc;
  47 using namespace ::com::sun::star::container;
  48 using namespace ::com::sun::star::lang;
  49 // -------------------------------------------------------------------------
  50 OTableKeyHelper::OTableKeyHelper(OTableHelper* _pTable) : connectivity::sdbcx::OKey(sal_True)
  51     ,m_pTable(_pTable)
  52 {
  53     construct();
  54 }
  55 // -------------------------------------------------------------------------
  56 OTableKeyHelper::OTableKeyHelper(       OTableHelper* _pTable
  57             ,const ::rtl::OUString& _Name
  58             ,const sdbcx::TKeyProperties& _rProps
  59             ) : connectivity::sdbcx::OKey(_Name,_rProps,sal_True)
  60                 ,m_pTable(_pTable)
  61 {
  62     construct();
  63     refreshColumns();
  64 }
  65 // -------------------------------------------------------------------------
  66 void OTableKeyHelper::refreshColumns()
  67 {
  68     if ( !m_pTable )
  69         return;
  70
  71     ::std::vector< ::rtl::OUString> aVector;
  72     if ( !isNew() )
  73     {
  74         aVector = m_aProps->m_aKeyColumnNames;
  75         if ( aVector.empty() )
  76         {
  77             ::dbtools::OPropertyMap& rPropMap = OMetaConnection::getPropMap();
  78             ::rtl::OUString aSchema,aTable;
  79             m_pTable->getPropertyValue(rPropMap.getNameByIndex(PROPERTY_ID_SCHEMANAME)) >>= aSchema;
  80             m_pTable->getPropertyValue(rPropMap.getNameByIndex(PROPERTY_ID_NAME))               >>= aTable;
  81
  82             if ( m_Name.getLength() ) // foreign key
  83             {
  84
  85                 Reference< XResultSet > xResult = m_pTable->getMetaData()->getImportedKeys(m_pTable->getPropertyValue(rPropMap.getNameByIndex(PROPERTY_ID_CATALOGNAME)),
  86                     aSchema,aTable);
  87
  88                 if ( xResult.is() )
  89                 {
  90                     Reference< XRow > xRow(xResult,UNO_QUERY);
  91                     while( xResult->next() )
  92                     {
  93                         ::rtl::OUString aForeignKeyColumn = xRow->getString(8);
  94                         if(xRow->getString(12) == m_Name)
  95                             aVector.push_back(aForeignKeyColumn);
  96                     }
  97                 }
  98             }
  99
 100             if ( aVector.empty() )
 101             {
 102                 const Reference< XResultSet > xResult = m_pTable->getMetaData()->getPrimaryKeys(m_pTable->getPropertyValue(rPropMap.getNameByIndex(PROPERTY_ID_CATALOGNAME)),
 103                     aSchema,aTable);
 104
 105                 if ( xResult.is() )
 106                 {
 107                     const Reference< XRow > xRow(xResult,UNO_QUERY);
 108                     while( xResult->next() )
 109                         aVector.push_back(xRow->getString(4));
 110                 } // if ( xResult.is() )
 111             }
 112         }
 113     }
 114
 115
 116     if ( m_pColumns )
 117         m_pColumns ->reFill(aVector);
 118     else
 119         m_pColumns      = new OKeyColumnsHelper(this,m_aMutex,aVector);
 120 }
 121 // -----------------------------------------------------------------------------
 122