Add synchronous SDL readiness waiting API

[ric-plt/sdl.git] / src / databaseconfigurationimpl.cpp

diff --git a/src/databaseconfigurationimpl.cpp b/src/databaseconfigurationimpl.cpp
index 909e4e4..4ee548f 100644 (file)
--- a/src/databaseconfigurationimpl.cpp
+++ b/src/databaseconfigurationimpl.cpp
@@ -14,8 +14,14 @@
    limitations under the License.
 */
 
+/*
+ * This source code is part of the near-RT RIC (RAN Intelligent Controller)
+ * platform project (RICP).
+*/
+
 #include "private/databaseconfigurationimpl.hpp"
 #include <arpa/inet.h>
+#include <boost/crc.hpp>
 
 using namespace shareddatalayer;
 
@@ -48,6 +54,10 @@ void DatabaseConfigurationImpl::checkAndApplyDbType(const std::string& type)
         dbType = DatabaseConfiguration::DbType::REDIS_CLUSTER;
     else if (type == "redis-sentinel")
         dbType = DatabaseConfiguration::DbType::REDIS_SENTINEL;
+    else if (type == "sdl-standalone-cluster")
+        dbType = DatabaseConfiguration::DbType::SDL_STANDALONE_CLUSTER;
+    else if (type == "sdl-sentinel-cluster")
+        dbType = DatabaseConfiguration::DbType::SDL_SENTINEL_CLUSTER;
     else
         throw DatabaseConfiguration::InvalidDbType(type);
 }
@@ -72,6 +82,14 @@ DatabaseConfiguration::Addresses DatabaseConfigurationImpl::getServerAddresses()
     return serverAddresses;
 }
 
+DatabaseConfiguration::Addresses DatabaseConfigurationImpl::getServerAddresses(const boost::optional<std::size_t>& addressIndex) const
+{
+    if (addressIndex)
+        return { HostAndPort(serverAddresses.at(*addressIndex)) };
+
+    return serverAddresses;
+}
+
 DatabaseConfiguration::Addresses DatabaseConfigurationImpl::getDefaultServerAddresses() const
 {
     return { HostAndPort(getDefaultHost(), htons(DEFAULT_PORT)) };
@@ -87,6 +105,14 @@ boost::optional<HostAndPort> DatabaseConfigurationImpl::getSentinelAddress() con
     return sentinelAddress;
 }
 
+boost::optional<HostAndPort> DatabaseConfigurationImpl::getSentinelAddress(const boost::optional<std::size_t>& addressIndex) const
+{
+    if (addressIndex)
+        return { HostAndPort(serverAddresses.at(*addressIndex).getHost(), sentinelAddress->getPort()) };
+
+    return getSentinelAddress();
+}
+
 void DatabaseConfigurationImpl::checkAndApplySentinelMasterName(const std::string& name)
 {
     sentinelMasterName = name;