diff options
Diffstat (limited to 'connectivity/source/cpool/ZPoolCollection.cxx')
| -rw-r--r-- | connectivity/source/cpool/ZPoolCollection.cxx | 468 |
1 files changed, 468 insertions, 0 deletions
diff --git a/connectivity/source/cpool/ZPoolCollection.cxx b/connectivity/source/cpool/ZPoolCollection.cxx new file mode 100644 index 000000000..ba17e6fea --- /dev/null +++ b/connectivity/source/cpool/ZPoolCollection.cxx @@ -0,0 +1,468 @@ +/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */ +/* + * This file is part of the LibreOffice project. + * + * 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/. + * + * This file incorporates work covered by the following license notice: + * + * Licensed to the Apache Software Foundation (ASF) under one or more + * contributor license agreements. See the NOTICE file distributed + * with this work for additional information regarding copyright + * ownership. The ASF licenses this file to you under the Apache + * License, Version 2.0 (the "License"); you may not use this file + * except in compliance with the License. You may obtain a copy of + * the License at http://www.apache.org/licenses/LICENSE-2.0 . + */ + +#include "ZPoolCollection.hxx" +#include "ZDriverWrapper.hxx" +#include "ZConnectionPool.hxx" +#include <com/sun/star/configuration/theDefaultProvider.hpp> +#include <com/sun/star/container/XHierarchicalNameAccess.hpp> +#include <com/sun/star/beans/NamedValue.hpp> +#include <com/sun/star/beans/PropertyValue.hpp> +#include <com/sun/star/frame/Desktop.hpp> +#include <com/sun/star/reflection/ProxyFactory.hpp> +#include <com/sun/star/sdbc/DriverManager.hpp> +#include <cppuhelper/supportsservice.hxx> +#include <com/sun/star/beans/XPropertySet.hpp> +#include <osl/diagnose.h> +#include <sal/log.hxx> +#include <tools/diagnose_ex.h> + +using namespace ::com::sun::star::uno; +using namespace ::com::sun::star::lang; +using namespace ::com::sun::star::sdbc; +using namespace ::com::sun::star::beans; +using namespace ::com::sun::star::container; +using namespace ::com::sun::star::reflection; +using namespace ::osl; +using namespace connectivity; + + +static OUString getConnectionPoolNodeName() +{ + return "org.openoffice.Office.DataAccess/ConnectionPool"; +} + +static OUString getEnablePoolingNodeName() +{ + return "EnablePooling"; +} + +static OUString getDriverNameNodeName() +{ + return "DriverName"; +} + +static OUString getDriverSettingsNodeName() +{ + return "DriverSettings"; +} + +static OUString getEnableNodeName() +{ + return "Enable"; +} + + +OPoolCollection::OPoolCollection(const Reference< XComponentContext >& _rxContext) + :m_xContext(_rxContext) +{ + // bootstrap all objects supporting the .sdb.Driver service + m_xManager = DriverManager::create( m_xContext ); + + m_xProxyFactory = ProxyFactory::create( m_xContext ); + + Reference<XPropertySet> xProp(getConfigPoolRoot(),UNO_QUERY); + if ( xProp.is() ) + xProp->addPropertyChangeListener(getEnablePoolingNodeName(),this); + // attach as desktop listener to know when we have to release our pools + osl_atomic_increment( &m_refCount ); + { + + m_xDesktop = css::frame::Desktop::create( m_xContext ); + m_xDesktop->addTerminateListener(this); + + } + osl_atomic_decrement( &m_refCount ); +} + +OPoolCollection::~OPoolCollection() +{ + clearConnectionPools(false); +} + +Reference< XConnection > SAL_CALL OPoolCollection::getConnection( const OUString& _rURL ) +{ + return getConnectionWithInfo(_rURL,Sequence< PropertyValue >()); +} + +Reference< XConnection > SAL_CALL OPoolCollection::getConnectionWithInfo( const OUString& _rURL, const Sequence< PropertyValue >& _rInfo ) +{ + MutexGuard aGuard(m_aMutex); + Reference< XConnection > xConnection; + Reference< XDriver > xDriver; + Reference< XInterface > xDriverNode; + OUString sImplName; + if(isPoolingEnabledByUrl(_rURL,xDriver,sImplName,xDriverNode) && xDriver.is()) + { + OConnectionPool* pConnectionPool = getConnectionPool(sImplName,xDriver,xDriverNode); + + if(pConnectionPool) + xConnection = pConnectionPool->getConnectionWithInfo(_rURL,_rInfo); + } + else if(xDriver.is()) + xConnection = xDriver->connect(_rURL,_rInfo); + + return xConnection; +} + +void SAL_CALL OPoolCollection::setLoginTimeout( sal_Int32 seconds ) +{ + MutexGuard aGuard(m_aMutex); + m_xManager->setLoginTimeout(seconds); +} + +sal_Int32 SAL_CALL OPoolCollection::getLoginTimeout( ) +{ + MutexGuard aGuard(m_aMutex); + return m_xManager->getLoginTimeout(); +} + +OUString SAL_CALL OPoolCollection::getImplementationName( ) +{ + return "com.sun.star.sdbc.OConnectionPool"; +} + +sal_Bool SAL_CALL OPoolCollection::supportsService( const OUString& _rServiceName ) +{ + return cppu::supportsService(this, _rServiceName); +} + + +Sequence< OUString > SAL_CALL OPoolCollection::getSupportedServiceNames( ) +{ + return { "com.sun.star.sdbc.ConnectionPool" }; +} + +Reference< XDriver > SAL_CALL OPoolCollection::getDriverByURL( const OUString& _rURL ) +{ + // returns the original driver when no connection pooling is enabled else it returns the proxy + MutexGuard aGuard(m_aMutex); + + Reference< XDriver > xDriver; + Reference< XInterface > xDriverNode; + OUString sImplName; + if(isPoolingEnabledByUrl(_rURL,xDriver,sImplName,xDriverNode)) + { + Reference< XDriver > xExistentProxy; + // look if we already have a proxy for this driver + for (const auto& [rxDriver, rxDriverRef] : m_aDriverProxies) + { + // hold the proxy alive as long as we're in this loop round + xExistentProxy = rxDriverRef; + + if (xExistentProxy.is() && (rxDriver.get() == xDriver.get())) + // already created a proxy for this + break; + } + if (xExistentProxy.is()) + { + xDriver = xExistentProxy; + } + else + { // create a new proxy for the driver + // this allows us to control the connections created by it + Reference< XAggregation > xDriverProxy = m_xProxyFactory->createProxy(xDriver); + OSL_ENSURE(xDriverProxy.is(), "OConnectionPool::getDriverByURL: invalid proxy returned by the proxy factory!"); + + OConnectionPool* pConnectionPool = getConnectionPool(sImplName,xDriver,xDriverNode); + xDriver = new ODriverWrapper(xDriverProxy, pConnectionPool); + } + } + + return xDriver; +} + +bool OPoolCollection::isDriverPoolingEnabled(std::u16string_view _sDriverImplName, + Reference< XInterface >& _rxDriverNode) +{ + bool bEnabled = false; + Reference<XInterface> xConnectionPoolRoot = getConfigPoolRoot(); + // then look for which of them settings are stored in the configuration + Reference< XNameAccess > xDirectAccess(openNode(getDriverSettingsNodeName(),xConnectionPoolRoot),UNO_QUERY); + + if(xDirectAccess.is()) + { + Sequence< OUString > aDriverKeys = xDirectAccess->getElementNames(); + const OUString* pDriverKeys = aDriverKeys.getConstArray(); + const OUString* pDriverKeysEnd = pDriverKeys + aDriverKeys.getLength(); + for (;pDriverKeys != pDriverKeysEnd; ++pDriverKeys) + { + // the name of the driver in this round + if(_sDriverImplName == *pDriverKeys) + { + _rxDriverNode = openNode(*pDriverKeys,xDirectAccess); + if(_rxDriverNode.is()) + getNodeValue(getEnableNodeName(),_rxDriverNode) >>= bEnabled; + break; + } + } + } + return bEnabled; +} + +bool OPoolCollection::isPoolingEnabled() +{ + // the config node where all pooling relevant info are stored under + Reference<XInterface> xConnectionPoolRoot = getConfigPoolRoot(); + + // the global "enabled" flag + bool bEnabled = false; + if(xConnectionPoolRoot.is()) + getNodeValue(getEnablePoolingNodeName(),xConnectionPoolRoot) >>= bEnabled; + return bEnabled; +} + +Reference<XInterface> const & OPoolCollection::getConfigPoolRoot() +{ + if(!m_xConfigNode.is()) + m_xConfigNode = createWithProvider( + css::configuration::theDefaultProvider::get(m_xContext), + getConnectionPoolNodeName()); + return m_xConfigNode; +} + +bool OPoolCollection::isPoolingEnabledByUrl(const OUString& _sUrl, + Reference< XDriver >& _rxDriver, + OUString& _rsImplName, + Reference< XInterface >& _rxDriverNode) +{ + bool bEnabled = false; + _rxDriver = m_xManager->getDriverByURL(_sUrl); + if (_rxDriver.is() && isPoolingEnabled()) + { + Reference< XServiceInfo > xServiceInfo(_rxDriver,UNO_QUERY); + OSL_ENSURE(xServiceInfo.is(),"Each driver should have a XServiceInfo interface!"); + + if(xServiceInfo.is()) + { + // look for the implementation name of the driver + _rsImplName = xServiceInfo->getImplementationName(); + bEnabled = isDriverPoolingEnabled(_rsImplName,_rxDriverNode); + } + } + return bEnabled; +} + +void OPoolCollection::clearConnectionPools(bool _bDispose) +{ + for(auto& rEntry : m_aPools) + { + rEntry.second->clear(_bDispose); + } + m_aPools.clear(); +} + +OConnectionPool* OPoolCollection::getConnectionPool(const OUString& _sImplName, + const Reference< XDriver >& _xDriver, + const Reference< XInterface >& _xDriverNode) +{ + OConnectionPool *pRet = nullptr; + OConnectionPools::const_iterator aFind = m_aPools.find(_sImplName); + if (aFind != m_aPools.end()) + pRet = aFind->second.get(); + else if (_xDriver.is() && _xDriverNode.is()) + { + Reference<XPropertySet> xProp(_xDriverNode,UNO_QUERY); + if(xProp.is()) + xProp->addPropertyChangeListener(getEnableNodeName(),this); + rtl::Reference<OConnectionPool> pConnectionPool = new OConnectionPool(_xDriver,_xDriverNode,m_xProxyFactory); + m_aPools.emplace(_sImplName,pConnectionPool); + pRet = pConnectionPool.get(); + } + + OSL_ENSURE(pRet, "Could not query DriverManager from ConnectionPool!"); + + return pRet; +} + +Reference< XInterface > OPoolCollection::createWithProvider(const Reference< XMultiServiceFactory >& _rxConfProvider, + const OUString& _rPath) +{ + OSL_ASSERT(_rxConfProvider.is()); + Sequence< Any > args{ Any(NamedValue( "nodepath", Any(_rPath))) }; + Reference< XInterface > xInterface( + _rxConfProvider->createInstanceWithArguments( + "com.sun.star.configuration.ConfigurationAccess", + args)); + OSL_ENSURE( + xInterface.is(), + "::createWithProvider: could not create the node access!"); + return xInterface; +} + +Reference<XInterface> OPoolCollection::openNode(const OUString& _rPath,const Reference<XInterface>& _xTreeNode) noexcept +{ + Reference< XHierarchicalNameAccess > xHierarchyAccess(_xTreeNode, UNO_QUERY); + Reference< XNameAccess > xDirectAccess(_xTreeNode, UNO_QUERY); + Reference< XInterface > xNode; + + try + { + if (xDirectAccess.is() && xDirectAccess->hasByName(_rPath)) + { + xNode.set(xDirectAccess->getByName(_rPath), css::uno::UNO_QUERY); + SAL_WARN_IF( + !xNode.is(), "connectivity.cpool", + "OConfigurationNode::openNode: could not open the node!"); + } + else if (xHierarchyAccess.is()) + { + xNode.set( + xHierarchyAccess->getByHierarchicalName(_rPath), + css::uno::UNO_QUERY); + SAL_WARN_IF( + !xNode.is(), "connectivity.cpool", + "OConfigurationNode::openNode: could not open the node!"); + } + + } + catch(const NoSuchElementException&) + { + SAL_WARN("connectivity.cpool", "::openNode: there is no element named " << + _rPath << "!"); + } + catch(const Exception&) + { + TOOLS_WARN_EXCEPTION("connectivity.cpool", "OConfigurationNode::openNode: caught an exception while retrieving the node"); + } + return xNode; +} + +Any OPoolCollection::getNodeValue(const OUString& _rPath,const Reference<XInterface>& _xTreeNode) noexcept +{ + Reference< XHierarchicalNameAccess > xHierarchyAccess(_xTreeNode, UNO_QUERY); + Reference< XNameAccess > xDirectAccess(_xTreeNode, UNO_QUERY); + Any aReturn; + try + { + if (xDirectAccess.is() && xDirectAccess->hasByName(_rPath) ) + { + aReturn = xDirectAccess->getByName(_rPath); + } + else if (xHierarchyAccess.is()) + { + aReturn = xHierarchyAccess->getByHierarchicalName(_rPath); + } + } + catch(const NoSuchElementException&) + { + TOOLS_WARN_EXCEPTION("connectivity.cpool", "" ); + } + return aReturn; +} + +void SAL_CALL OPoolCollection::queryTermination( const EventObject& /*Event*/ ) +{ +} + +void SAL_CALL OPoolCollection::notifyTermination( const EventObject& /*Event*/ ) +{ + clearDesktop(); +} + +void SAL_CALL OPoolCollection::disposing( const EventObject& Source ) +{ + MutexGuard aGuard(m_aMutex); + if ( m_xDesktop == Source.Source ) + { + clearDesktop(); + } + else + { + try + { + Reference<XPropertySet> xProp(Source.Source,UNO_QUERY); + if(Source.Source == m_xConfigNode) + { + if ( xProp.is() ) + xProp->removePropertyChangeListener(getEnablePoolingNodeName(),this); + m_xConfigNode.clear(); + } + else if ( xProp.is() ) + xProp->removePropertyChangeListener(getEnableNodeName(),this); + } + catch(const Exception&) + { + TOOLS_WARN_EXCEPTION("connectivity.cpool", ""); + } + } +} + +void SAL_CALL OPoolCollection::propertyChange( const css::beans::PropertyChangeEvent& evt ) +{ + MutexGuard aGuard(m_aMutex); + if(evt.Source == m_xConfigNode) + { + bool bEnabled = true; + evt.NewValue >>= bEnabled; + if(!bEnabled ) + { + m_aDriverProxies.clear(); + m_aDriverProxies = MapDriver2DriverRef(); + clearConnectionPools(false); + } + } + else if(evt.Source.is()) + { + bool bEnabled = true; + evt.NewValue >>= bEnabled; + if(!bEnabled) + { + OUString sThisDriverName; + getNodeValue(getDriverNameNodeName(),evt.Source) >>= sThisDriverName; + // 1st release the driver + // look if we already have a proxy for this driver + MapDriver2DriverRef::iterator aLookup = m_aDriverProxies.begin(); + while( aLookup != m_aDriverProxies.end()) + { + MapDriver2DriverRef::iterator aFind = aLookup; + Reference<XServiceInfo> xInfo(aLookup->first,UNO_QUERY); + ++aLookup; + if(xInfo.is() && xInfo->getImplementationName() == sThisDriverName) + m_aDriverProxies.erase(aFind); + } + + // 2nd clear the connectionpool + OConnectionPools::iterator aFind = m_aPools.find(sThisDriverName); + if(aFind != m_aPools.end()) + { + aFind->second->clear(false); + m_aPools.erase(aFind); + } + } + } +} + +void OPoolCollection::clearDesktop() +{ + clearConnectionPools(true); + if ( m_xDesktop.is() ) + m_xDesktop->removeTerminateListener(this); + m_xDesktop.clear(); +} + +extern "C" SAL_DLLPUBLIC_EXPORT css::uno::XInterface* +connectivity_OPoolCollection_get_implementation( + css::uno::XComponentContext* context , css::uno::Sequence<css::uno::Any> const&) +{ + return cppu::acquire(new OPoolCollection(context)); +} + + +/* vim:set shiftwidth=4 softtabstop=4 expandtab: */ |