1
0
Fork 0
libreoffice/connectivity/source/drivers/postgresql/pq_xindexes.cxx
Daniel Baumann 8e63e14cf6
Adding upstream version 4:25.2.3.
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
2025-06-22 16:20:04 +02:00

303 lines
10 KiB
C++

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/*************************************************************************
*
* Effective License of whole file:
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License version 2.1, as published by the Free Software Foundation.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public
* License along with this library; if not, write to the Free Software
* Foundation, Inc., 59 Temple Place, Suite 330, Boston,
* MA 02111-1307 USA
*
* Parts "Copyright by Sun Microsystems, Inc" prior to August 2011:
*
* The Contents of this file are made available subject to the terms of
* the GNU Lesser General Public License Version 2.1
*
* Copyright: 2000 by Sun Microsystems, Inc.
*
* Contributor(s): Joerg Budischewski
*
* All parts contributed on or after August 2011:
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*
************************************************************************/
#include <sal/log.hxx>
#include <rtl/ref.hxx>
#include <rtl/ustrbuf.hxx>
#include <com/sun/star/lang/IndexOutOfBoundsException.hpp>
#include <com/sun/star/lang/WrappedTargetRuntimeException.hpp>
#include <com/sun/star/sdbc/SQLException.hpp>
#include <com/sun/star/sdbc/XRow.hpp>
#include <com/sun/star/sdbc/XParameters.hpp>
#include <cppuhelper/exc_hlp.hxx>
#include <o3tl/safeint.hxx>
#include <utility>
#include "pq_xindexes.hxx"
#include "pq_xindex.hxx"
#include "pq_statics.hxx"
#include "pq_tools.hxx"
using osl::MutexGuard;
using com::sun::star::beans::XPropertySet;
using com::sun::star::uno::Any;
using com::sun::star::uno::UNO_QUERY;
using com::sun::star::uno::Reference;
using com::sun::star::uno::Sequence;
using com::sun::star::container::XEnumerationAccess;
using com::sun::star::container::XEnumeration;
using com::sun::star::sdbcx::XColumnsSupplier;
using com::sun::star::sdbc::XRow;
using com::sun::star::sdbc::XResultSet;
using com::sun::star::sdbc::XParameters;
using com::sun::star::sdbc::XPreparedStatement;
namespace pq_sdbc_driver
{
Indexes::Indexes(
const ::rtl::Reference< comphelper::RefCountedMutex > & refMutex,
const css::uno::Reference< css::sdbc::XConnection > & origin,
ConnectionSettings *pSettings,
OUString schemaName,
OUString tableName)
: Container( refMutex, origin, pSettings, getStatics().KEY ),
m_schemaName(std::move( schemaName )),
m_tableName(std::move( tableName ))
{
}
Indexes::~Indexes()
{}
void Indexes::refresh()
{
try
{
SAL_INFO("connectivity.postgresql", "sdbcx.Indexes get refreshed for table " << m_schemaName << "." << m_tableName);
osl::MutexGuard guard( m_xMutex->GetMutex() );
Statics & st = getStatics();
Int2StringMap column2NameMap;
fillAttnum2attnameMap( column2NameMap, m_origin, m_schemaName, m_tableName );
// see XDatabaseMetaData::getIndexInfo()
Reference< XPreparedStatement > stmt = m_origin->prepareStatement(
u"SELECT nspname, " // 1
"pg_class.relname, " // 2
"class2.relname, " // 3
"indisclustered, " // 4
"indisunique, " // 5
"indisprimary, " // 6
"indkey " // 7
"FROM pg_index INNER JOIN pg_class ON indrelid = pg_class.oid "
"INNER JOIN pg_namespace ON pg_class.relnamespace = pg_namespace.oid "
"INNER JOIN pg_class as class2 ON pg_index.indexrelid = class2.oid "
"WHERE nspname = ? AND pg_class.relname = ?"_ustr );
Reference< XParameters > params( stmt, UNO_QUERY);
params->setString( 1, m_schemaName );
params->setString( 2, m_tableName );
Reference< XResultSet > rs = stmt->executeQuery();
Reference< XRow > row( rs, UNO_QUERY );
String2IntMap map;
m_values.clear();
sal_Int32 index = 0;
while( rs->next() )
{
// C_SCHEMA = 1
// C_TABLENAME = 2
static const sal_Int32 C_INDEXNAME = 3;
static const sal_Int32 C_IS_CLUSTERED = 4;
static const sal_Int32 C_IS_UNIQUE = 5;
static const sal_Int32 C_IS_PRIMARY = 6;
static const sal_Int32 C_COLUMNS = 7;
OUString currentIndexName = row->getString( C_INDEXNAME );
rtl::Reference<Index> pIndex =
new Index( m_xMutex, m_origin, m_pSettings,
m_schemaName, m_tableName );
bool isUnique = row->getBoolean( C_IS_UNIQUE );
bool isPrimary = row->getBoolean( C_IS_PRIMARY );
bool isClusterd = row->getBoolean( C_IS_CLUSTERED );
Reference< css::beans::XPropertySet > prop = pIndex;
pIndex->setPropertyValue_NoBroadcast_public(
st.IS_UNIQUE, Any( isUnique ) );
pIndex->setPropertyValue_NoBroadcast_public(
st.IS_PRIMARY_KEY_INDEX, Any( isPrimary ) );
pIndex->setPropertyValue_NoBroadcast_public(
st.IS_CLUSTERED, Any( isClusterd ) );
pIndex->setPropertyValue_NoBroadcast_public(
st.NAME, Any( currentIndexName ) );
std::vector< sal_Int32 > seq = parseIntArray( row->getString( C_COLUMNS ) );
Sequence< OUString > columnNames(seq.size());
auto columnNamesRange = asNonConstRange(columnNames);
for( size_t columns = 0 ; columns < seq.size() ; columns ++ )
{
columnNamesRange[columns] = column2NameMap[ seq[columns] ];
}
pIndex->setPropertyValue_NoBroadcast_public(
st.PRIVATE_COLUMN_INDEXES, Any( columnNames ));
{
m_values.emplace_back(prop);
map[ currentIndexName ] = index;
++index;
}
}
m_name2index.swap( map );
}
catch ( css::sdbc::SQLException & e )
{
css::uno::Any anyEx = cppu::getCaughtException();
throw css::lang::WrappedTargetRuntimeException( e.Message,
e.Context, anyEx );
}
fire( RefreshedBroadcaster( *this ) );
}
void Indexes::appendByDescriptor(
const css::uno::Reference< css::beans::XPropertySet >& descriptor )
{
Statics & st = getStatics();
OUString name = extractStringProperty( descriptor, st.NAME );
bool isUnique = extractBoolProperty( descriptor, st.IS_UNIQUE );
OUStringBuffer buf( 128 );
buf.append( "CREATE " );
if( isUnique )
buf.append( "UNIQUE " );
buf.append( "INDEX " );
bufferQuoteIdentifier( buf, name, m_pSettings );
buf.append( " ON " );
bufferQuoteQualifiedIdentifier( buf, m_schemaName, m_tableName, m_pSettings );
buf.append( " ( " );
Reference< XColumnsSupplier > columns( descriptor, UNO_QUERY );
if( columns.is() )
{
Reference< XEnumerationAccess > access( columns->getColumns(), UNO_QUERY );
if( access.is() )
{
Reference< XEnumeration > xEnum( access->createEnumeration() );
bool first = true;
while( xEnum.is() && xEnum->hasMoreElements() )
{
Reference< XPropertySet > column( xEnum->nextElement(), UNO_QUERY );
if( first )
{
first = false;
}
else
{
buf.append( ", " );
}
buf.append( extractStringProperty( column, st.NAME ) );
}
}
}
buf.append( " ) " );
m_origin->createStatement()->executeUpdate( buf.makeStringAndClear() );
refresh();
}
void Indexes::dropByIndex( sal_Int32 index )
{
osl::MutexGuard guard( m_xMutex->GetMutex() );
if( index < 0 || o3tl::make_unsigned(index) >= m_values.size() )
{
throw css::lang::IndexOutOfBoundsException(
"Indexes: Index out of range (allowed 0 to "
+ OUString::number( m_values.size() -1 )
+ ", got " + OUString::number( index )
+ ")",
*this );
}
Reference< XPropertySet > set;
m_values[index] >>= set;
Statics &st = getStatics();
OUStringBuffer buf( 128 );
buf.append( "DROP INDEX " );
bufferQuoteIdentifier( buf, extractStringProperty( set, st.NAME ), m_pSettings );
m_origin->createStatement()->executeUpdate( buf.makeStringAndClear() );
Container::dropByIndex( index );
}
css::uno::Reference< css::beans::XPropertySet > Indexes::createDataDescriptor()
{
return new IndexDescriptor( m_xMutex, m_origin, m_pSettings );
}
Reference< css::container::XNameAccess > Indexes::create(
const ::rtl::Reference< comphelper::RefCountedMutex > & refMutex,
const css::uno::Reference< css::sdbc::XConnection > & origin,
ConnectionSettings *pSettings,
const OUString & schemaName,
const OUString & tableName)
{
rtl::Reference<Indexes> pIndexes
= new Indexes( refMutex, origin, pSettings, schemaName, tableName );
pIndexes->refresh();
return pIndexes;
}
IndexDescriptors::IndexDescriptors(
const ::rtl::Reference< comphelper::RefCountedMutex > & refMutex,
const css::uno::Reference< css::sdbc::XConnection > & origin,
ConnectionSettings *pSettings)
: Container( refMutex, origin, pSettings, getStatics().INDEX )
{}
Reference< css::container::XNameAccess > IndexDescriptors::create(
const ::rtl::Reference< comphelper::RefCountedMutex > & refMutex,
const css::uno::Reference< css::sdbc::XConnection > & origin,
ConnectionSettings *pSettings)
{
return new IndexDescriptors( refMutex, origin, pSettings );
}
css::uno::Reference< css::beans::XPropertySet > IndexDescriptors::createDataDescriptor()
{
return new IndexDescriptor( m_xMutex, m_origin, m_pSettings );
}
};
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */