diff --git a/lib/inc/Channel.h b/lib/inc/Channel.h index 26fc2f5eca1b..266039ccf2f2 100644 --- a/lib/inc/Channel.h +++ b/lib/inc/Channel.h @@ -28,6 +28,13 @@ namespace sairedis virtual ~Channel(); + public: + + void setResponseTimeout( + _In_ uint64_t responseTimeout); + + uint64_t getResponseTimeout() const; + public: virtual void setBuffered( @@ -56,6 +63,8 @@ namespace sairedis Callback m_callback; + uint64_t m_responseTimeoutMs; + protected: // notification /** diff --git a/lib/inc/RedisRemoteSaiInterface.h b/lib/inc/RedisRemoteSaiInterface.h index eb0ea9e78f8a..4927e2321a99 100644 --- a/lib/inc/RedisRemoteSaiInterface.h +++ b/lib/inc/RedisRemoteSaiInterface.h @@ -465,6 +465,8 @@ namespace sairedis std::shared_ptr m_communicationChannel; + uint64_t m_responseTimeoutMs; + std::function)> m_notificationCallback; std::map m_tableDump; diff --git a/lib/inc/sairedis.h b/lib/inc/sairedis.h index 49993b5f3a2b..24029fa7e4ae 100644 --- a/lib/inc/sairedis.h +++ b/lib/inc/sairedis.h @@ -13,6 +13,11 @@ extern "C" { */ #define SAI_REDIS_KEY_CONTEXT_CONFIG "SAI_REDIS_CONTEXT_CONFIG" +/** + * @brief Default synchronous operation response timeout in milliseconds. + */ +#define SAI_REDIS_DEFAULT_SYNC_OPERATION_RESPONSE_TIMEOUT (60*1000) + typedef enum _sai_redis_notify_syncd_t { SAI_REDIS_NOTIFY_SYNCD_INIT_VIEW, @@ -205,4 +210,16 @@ typedef enum _sai_redis_switch_attr_t */ SAI_REDIS_SWITCH_ATTR_RECORDING_FILENAME, + /** + * @brief Synchronous operation response timeout in milliseconds. + * + * Used for every synchronous API call. In asynchronous mode used for GET + * operation. + * + * @type sai_uint64_t + * @flags CREATE_AND_SET + * @default 60000 + */ + SAI_REDIS_SWITCH_ATTR_SYNC_OPERATION_RESPONSE_TIMEOUT, + } sai_redis_switch_attr_t; diff --git a/lib/src/Channel.cpp b/lib/src/Channel.cpp index 50b1e870f4fa..9df5919e023d 100644 --- a/lib/src/Channel.cpp +++ b/lib/src/Channel.cpp @@ -1,12 +1,15 @@ #include "Channel.h" +#include "sairedis.h" + #include "swss/logger.h" using namespace sairedis; Channel::Channel( _In_ Callback callback): - m_callback(callback) + m_callback(callback), + m_responseTimeoutMs(SAI_REDIS_DEFAULT_SYNC_OPERATION_RESPONSE_TIMEOUT) { SWSS_LOG_ENTER(); @@ -19,3 +22,18 @@ Channel::~Channel() // empty } + +void Channel::setResponseTimeout( + _In_ uint64_t responseTimeout) +{ + SWSS_LOG_ENTER(); + + m_responseTimeoutMs = responseTimeout; +} + +uint64_t Channel::getResponseTimeout() const +{ + SWSS_LOG_ENTER(); + + return m_responseTimeoutMs; +} diff --git a/lib/src/RedisChannel.cpp b/lib/src/RedisChannel.cpp index a6b5d4a39566..8d6cda7c5f1b 100644 --- a/lib/src/RedisChannel.cpp +++ b/lib/src/RedisChannel.cpp @@ -9,11 +9,6 @@ using namespace sairedis; -/** - * @brief Get response timeout in milliseconds. - */ -#define REDIS_ASIC_STATE_COMMAND_GETRESPONSE_TIMEOUT_MS (60*1000) - RedisChannel::RedisChannel( _In_ const std::string& dbAsic, _In_ Channel::Callback callback): @@ -179,7 +174,7 @@ sai_status_t RedisChannel::wait( swss::Selectable *sel; - int result = s.select(&sel, REDIS_ASIC_STATE_COMMAND_GETRESPONSE_TIMEOUT_MS); + int result = s.select(&sel, (int)m_responseTimeoutMs); if (result == swss::Select::OBJECT) { diff --git a/lib/src/RedisRemoteSaiInterface.cpp b/lib/src/RedisRemoteSaiInterface.cpp index 6b3e02d25b77..7507f7866a1c 100644 --- a/lib/src/RedisRemoteSaiInterface.cpp +++ b/lib/src/RedisRemoteSaiInterface.cpp @@ -92,6 +92,8 @@ sai_status_t RedisRemoteSaiInterface::initialize( std::bind(&RedisRemoteSaiInterface::handleNotification, this, _1, _2, _3)); } + m_responseTimeoutMs = m_communicationChannel->getResponseTimeout(); + m_db = std::make_shared(m_contextConfig->m_dbAsic, 0); m_redisVidIndexGenerator = std::make_shared(m_db, REDIS_KEY_VIDCOUNTER); @@ -355,6 +357,16 @@ sai_status_t RedisRemoteSaiInterface::setRedisExtensionAttribute( return SAI_STATUS_SUCCESS; + case SAI_REDIS_SWITCH_ATTR_SYNC_OPERATION_RESPONSE_TIMEOUT: + + m_responseTimeoutMs = attr->value.u64; + + m_communicationChannel->setResponseTimeout(m_responseTimeoutMs); + + SWSS_LOG_NOTICE("set response timeout to %lu ms", m_responseTimeoutMs); + + return SAI_STATUS_SUCCESS; + case SAI_REDIS_SWITCH_ATTR_SYNC_MODE: SWSS_LOG_WARN("sync mode is depreacated, use communication mode"); @@ -402,6 +414,8 @@ sai_status_t RedisRemoteSaiInterface::setRedisExtensionAttribute( m_contextConfig->m_dbAsic, std::bind(&RedisRemoteSaiInterface::handleNotification, this, _1, _2, _3)); + m_communicationChannel->setResponseTimeout(m_responseTimeoutMs); + m_communicationChannel->setBuffered(true); return SAI_STATUS_SUCCESS; @@ -416,6 +430,8 @@ sai_status_t RedisRemoteSaiInterface::setRedisExtensionAttribute( m_contextConfig->m_dbAsic, std::bind(&RedisRemoteSaiInterface::handleNotification, this, _1, _2, _3)); + m_communicationChannel->setResponseTimeout(m_responseTimeoutMs); + m_communicationChannel->setBuffered(false); return SAI_STATUS_SUCCESS; @@ -432,6 +448,8 @@ sai_status_t RedisRemoteSaiInterface::setRedisExtensionAttribute( m_contextConfig->m_zmqNtfEndpoint, std::bind(&RedisRemoteSaiInterface::handleNotification, this, _1, _2, _3)); + m_communicationChannel->setResponseTimeout(m_responseTimeoutMs); + SWSS_LOG_NOTICE("zmq enabled, forcing sync mode"); m_syncMode = true; diff --git a/lib/src/ZeroMQChannel.cpp b/lib/src/ZeroMQChannel.cpp index 6b543e083725..9d05ca228a63 100644 --- a/lib/src/ZeroMQChannel.cpp +++ b/lib/src/ZeroMQChannel.cpp @@ -12,11 +12,6 @@ using namespace sairedis; -/** - * @brief Get response timeout in milliseconds. - */ -#define ZMQ_GETRESPONSE_TIMEOUT_MS (60*1000) - #define ZMQ_RESPONSE_BUFFER_SIZE (4*1024*1024) ZeroMQChannel::ZeroMQChannel( @@ -273,7 +268,7 @@ sai_status_t ZeroMQChannel::wait( items[0].socket = m_socket; items[0].events = ZMQ_POLLIN; - int rc = zmq_poll(items, 1, ZMQ_GETRESPONSE_TIMEOUT_MS); + int rc = zmq_poll(items, 1, (int)m_responseTimeoutMs); if (rc == 0) {