Add setsockopt callback

fantasy-peak · Jul 24, 2024 · 2cb1c6c · 2cb1c6c
1 parent f8e378f
commit 2cb1c6c
Show file tree

Hide file tree

Showing 10 changed files with 19 additions and 0 deletions.
diff --git a/out/bi_web/include/impl/bi_channel.h b/out/bi_web/include/impl/bi_channel.h
@@ -237,6 +237,8 @@ class BiChannel final {
         if (config.probe) {
             m_socket_ptr->set(zmq::sockopt::probe_router, 1);
         }
+        if (config.setsockopt_callback)
+            config.setsockopt_callback(*m_socket_ptr);
         if (config.bind) {
             if (config.socktype == zmq::socket_type::router) {
                 if (config.mandatory)

diff --git a/out/bi_web/include/impl/uni_channel.h b/out/bi_web/include/impl/uni_channel.h
@@ -160,6 +160,8 @@ class UniChannel final {
         }
         if (config.socktype == zmq::socket_type::sub)
             m_socket_ptr->set(zmq::sockopt::subscribe, "");
+        if (config.setsockopt_callback)
+            config.setsockopt_callback(*m_socket_ptr);
         if (config.bind)
             m_socket_ptr->bind(config.addr);
         else

diff --git a/out/bi_web/include/impl/utils.h b/out/bi_web/include/impl/utils.h
@@ -65,6 +65,7 @@ struct ChannelConfig {
     std::size_t channel_size{50000};
     std::optional<int32_t> heartbeat_timeout;
     std::optional<int32_t> heartbeat_ivl;
+    std::function<void(zmq::socket_t&)> setsockopt_callback;
 };
 
 inline std::string createUuid() {

diff --git a/out/include/impl/bi_channel.h b/out/include/impl/bi_channel.h
@@ -237,6 +237,8 @@ class BiChannel final {
         if (config.probe) {
             m_socket_ptr->set(zmq::sockopt::probe_router, 1);
         }
+        if (config.setsockopt_callback)
+            config.setsockopt_callback(*m_socket_ptr);
         if (config.bind) {
             if (config.socktype == zmq::socket_type::router) {
                 if (config.mandatory)

diff --git a/out/include/impl/uni_channel.h b/out/include/impl/uni_channel.h
@@ -160,6 +160,8 @@ class UniChannel final {
         }
         if (config.socktype == zmq::socket_type::sub)
             m_socket_ptr->set(zmq::sockopt::subscribe, "");
+        if (config.setsockopt_callback)
+            config.setsockopt_callback(*m_socket_ptr);
         if (config.bind)
             m_socket_ptr->bind(config.addr);
         else

diff --git a/out/include/impl/utils.h b/out/include/impl/utils.h
@@ -65,6 +65,7 @@ struct ChannelConfig {
     std::size_t channel_size{50000};
     std::optional<int32_t> heartbeat_timeout;
     std::optional<int32_t> heartbeat_ivl;
+    std::function<void(zmq::socket_t&)> setsockopt_callback;
 };
 
 inline std::string createUuid() {

diff --git a/template/cpp/impl/bi_channel.inja b/template/cpp/impl/bi_channel.inja
@@ -196,6 +196,8 @@ private:
         if (config.probe) {
             m_socket_ptr->set(zmq::sockopt::probe_router, 1);
         }
+        if (config.setsockopt_callback)
+            config.setsockopt_callback(*m_socket_ptr);
         if (config.bind) {
             if (config.socktype == zmq::socket_type::router) {
                 if (config.mandatory)

diff --git a/template/cpp/impl/uni_channel.inja b/template/cpp/impl/uni_channel.inja
@@ -132,6 +132,8 @@ private:
         }
         if (config.socktype == zmq::socket_type::sub)
             m_socket_ptr->set(zmq::sockopt::subscribe, "");
+        if (config.setsockopt_callback)
+            config.setsockopt_callback(*m_socket_ptr);
         if (config.bind)
             m_socket_ptr->bind(config.addr);
         else

diff --git a/template/cpp/impl/utils.inja b/template/cpp/impl/utils.inja
@@ -65,6 +65,7 @@ struct ChannelConfig {
     std::size_t channel_size{50000};
     std::optional<int32_t> heartbeat_timeout;
     std::optional<int32_t> heartbeat_ivl;
+    std::function<void(zmq::socket_t&)> setsockopt_callback;
 };
 
 inline std::string createUuid() {

diff --git a/test/cpp/coro_bi.cpp b/test/cpp/coro_bi.cpp
@@ -75,6 +75,10 @@ asio::awaitable<void> start_client() {
 void start_server() {
     frpc::ChannelConfig bi_config{};
     bi_config.addr = addr;
+    bi_config.setsockopt_callback = [](auto& socket) {
+        spdlog::info("setsockopt_callback");
+        socket.set(zmq::sockopt::heartbeat_ivl, 5000);
+    };
     auto server = fantasy::HelloWorldServer::create(
         bi_config,
         std::make_shared<CoroHandler>(),