Update ipc.cpp

Optimizes the number of handles to the large message cache.

Author: mutouyun

src/ipc.cpp

@@ -21,6 +21,7 @@
 
 #include "libipc/utility/log.h"
 #include "libipc/utility/id_pool.h"
+#include "libipc/utility/utility.h"
 
 #include "libipc/memory/resource.h"
 
@@ -90,77 +91,68 @@ auto cc_acc() {
     return static_cast<acc_t*>(acc_h.get());
 }
 
-auto& cls_storages() {
-    struct cls_t {
-        ipc::shm::handle id_info_;
-        std::array<ipc::shm::handle, ipc::id_pool<>::max_count> mems_;
-    };
-    static ipc::unordered_map<std::size_t, cls_t> cls_s;
-    return cls_s;
-}
-
-auto& cls_lock() {
-    static ipc::spin_lock cls_l;
-    return cls_l;
-}
-
-struct cls_info_t {
+struct chunk_info_t {
     ipc::id_pool<> pool_;
     ipc::spin_lock lock_;
+
+    IPC_CONSTEXPR_ static std::size_t chunks_elem_size(std::size_t chunk_size) noexcept {
+        return ipc::make_align(alignof(std::max_align_t), chunk_size + sizeof(acc_t));
+    }
+
+    IPC_CONSTEXPR_ static std::size_t chunks_mem_size(std::size_t chunk_size) noexcept {
+        return ipc::id_pool<>::max_count * chunks_elem_size(chunk_size);
+    }
+
+    ipc::byte_t *at(std::size_t chunk_size, std::size_t id) noexcept {
+        return reinterpret_cast<ipc::byte_t *>(this + 1) + (chunks_elem_size(chunk_size) * id);
+    }
 };
 
-constexpr std::size_t calc_cls_size(std::size_t size) noexcept {
-    return (((size - 1) / ipc::large_msg_limit) + 1) * ipc::large_msg_limit;
+auto& chunk_storages() {
+    class chunk_t {
+        ipc::shm::handle handle_;
+
+    public:
+        chunk_info_t *get_info(std::size_t chunk_size) {
+            if (!handle_.valid() &&
+                !handle_.acquire( ("__CHUNK_INFO__" + ipc::to_string(chunk_size)).c_str(), 
+                                  sizeof(chunk_info_t) + chunk_info_t::chunks_mem_size(chunk_size) )) {
+                ipc::error("[chunk_storages] chunk_shm.id_info_.acquire failed: chunk_size = %zd\n", chunk_size);
+                return nullptr;
+            }
+            auto info = static_cast<chunk_info_t*>(handle_.get());
+            if (info == nullptr) {
+                ipc::error("[chunk_storages] chunk_shm.id_info_.get failed: chunk_size = %zd\n", chunk_size);
+                return nullptr;
+            }
+            return info;
+        }
+    };
+    static ipc::unordered_map<std::size_t, chunk_t> chunk_s;
+    return chunk_s;
 }
 
-auto& cls_storage(std::size_t cls_size) {
-    IPC_UNUSED_ auto guard = ipc::detail::unique_lock(cls_lock());
-    return cls_storages()[cls_size];
+auto& chunk_lock() {
+    static ipc::spin_lock chunk_l;
+    return chunk_l;
 }
 
-template <typename T>
-cls_info_t* cls_storage_info(const char* func, T& cls_shm, std::size_t cls_size) {
-    if (!cls_shm.id_info_.valid() &&
-        !cls_shm.id_info_.acquire(("__CLS_INFO__" + ipc::to_string(cls_size)).c_str(), sizeof(cls_info_t))) {
-        ipc::error("[%s] cls_shm.id_info_.acquire failed: cls_size = %zd\n", func, cls_size);
-        return nullptr;
-    }
-    auto info = static_cast<cls_info_t*>(cls_shm.id_info_.get());
-    if (info == nullptr) {
-        ipc::error("[%s] cls_shm.id_info_.get failed: cls_size = %zd\n", func, cls_size);
-        return nullptr;
-    }
-    return info;
+constexpr std::size_t calc_chunk_size(std::size_t size) noexcept {
+    return ( ((size - 1) / ipc::large_msg_align) + 1 ) * ipc::large_msg_align;
 }
 
-template <typename T>
-ipc::byte_t* cls_storage_mem(const char* func, T& cls_shm, std::size_t cls_size, std::size_t id) {
-    if (id == ipc::invalid_value) {
-        return nullptr;
-    }
-    if (!cls_shm.mems_[id].valid() &&
-        !cls_shm.mems_[id].acquire(("__CLS_MEM_BLOCK__" + ipc::to_string(cls_size) +
-                                                   "__" + ipc::to_string(id)).c_str(), 
-                                   cls_size + sizeof(acc_t))) {
-        ipc::error("[%s] cls_shm.mems_[id].acquire failed: id = %zd, cls_size = %zd\n", func, id, cls_size);
-        return nullptr;
-    }
-
-    auto ptr = static_cast<ipc::byte_t *>(cls_shm.mems_[id].get());
-    if (ptr == nullptr) {
-        ipc::error("[%s] cls_shm.mems_[id].get failed: id = %zd, cls_size = %zd\n", func, id, cls_size);
-        return nullptr;
-    }
-    return ptr;
+auto& chunk_storage(std::size_t chunk_size) {
+    IPC_UNUSED_ auto guard = ipc::detail::unique_lock(chunk_lock());
+    return chunk_storages()[chunk_size];
 }
 
 std::pair<std::size_t, void*> apply_storage(std::size_t conn_count, std::size_t size) {
     if (conn_count == 0) return {};
 
-    std::size_t cls_size = calc_cls_size(size);
-    auto &      cls_shm  = cls_storage(cls_size);
+    std::size_t chunk_size = calc_chunk_size(size);
+    auto &      chunk_shm  = chunk_storage(chunk_size);
 
-    auto info = cls_storage_info("apply_storage", cls_shm, cls_size);
+    auto info = chunk_shm.get_info(chunk_size);
     if (info == nullptr) return {};
 
     info->lock_.lock();
@@ -169,20 +161,23 @@ std::pair<std::size_t, void*> apply_storage(std::size_t conn_count, std::size_t
     auto id = info->pool_.acquire();
     info->lock_.unlock();
 
-    auto ptr = cls_storage_mem("apply_storage", cls_shm, cls_size, id);
+    auto ptr = info->at(chunk_size, id);
     if (ptr == nullptr) return {};
-    reinterpret_cast<acc_t*>(ptr + cls_size)->store(static_cast<msg_id_t>(conn_count), std::memory_order_release);
+    reinterpret_cast<acc_t*>(ptr + chunk_size)->store(static_cast<msg_id_t>(conn_count), std::memory_order_release);
     return { id, ptr };
 }
 
-void* find_storage(std::size_t id, std::size_t size) {
-    std::size_t cls_size = calc_cls_size(size);
-    auto &      cls_shm  = cls_storage(cls_size);
+void *find_storage(std::size_t id, std::size_t size) {
+    std::size_t chunk_size = calc_chunk_size(size);
+    auto &      chunk_shm  = chunk_storage(chunk_size);
+
+    auto info = chunk_shm.get_info(chunk_size);
+    if (info == nullptr) return nullptr;
 
-    auto ptr = cls_storage_mem("find_storage", cls_shm, cls_size, id);
+    auto ptr = info->at(chunk_size, id);
     if (ptr == nullptr) return nullptr;
-    if (reinterpret_cast<acc_t*>(ptr + cls_size)->load(std::memory_order_acquire) == 0) {
-        ipc::error("[find_storage] cc test failed: id = %zd, cls_size = %zd\n", id, cls_size);
+    if (reinterpret_cast<acc_t*>(ptr + chunk_size)->load(std::memory_order_acquire) == 0) {
+        ipc::error("[find_storage] cc test failed: id = %zd, chunk_size = %zd\n", id, chunk_size);
         return nullptr;
     }
     return ptr;
@@ -194,26 +189,22 @@ void recycle_storage(std::size_t id, std::size_t size) {
         return;
     }
 
-    std::size_t cls_size = calc_cls_size(size);
-    auto &      cls_shm  = cls_storage(cls_size);
+    std::size_t chunk_size = calc_chunk_size(size);
+    auto &      chunk_shm  = chunk_storage(chunk_size);
 
-    if (!cls_shm.mems_[id].valid()) {
-        ipc::error("[recycle_storage] should find storage first: id = %zd, cls_size = %zd\n", id, cls_size);
-        return;
-    }
-    auto ptr = static_cast<ipc::byte_t*>(cls_shm.mems_[id].get());
+    auto info = chunk_shm.get_info(chunk_size);
+    if (info == nullptr) return;
+
+    auto ptr = info->at(chunk_size, id);
     if (ptr == nullptr) {
-        ipc::error("[recycle_storage] cls_shm.mems_[id].get failed: id = %zd, cls_size = %zd\n", id, cls_size);
+        ipc::error("[recycle_storage] chunk_shm.mems[%zd] failed: chunk_size = %zd\n", id, chunk_size);
         return;
     }
-    if (reinterpret_cast<acc_t*>(ptr + cls_size)->fetch_sub(1, std::memory_order_acq_rel) > 1) {
+    if (reinterpret_cast<acc_t*>(ptr + chunk_size)->fetch_sub(1, std::memory_order_acq_rel) > 1) {
         // not the last receiver, just return
         return;
     }
 
-    auto info = cls_storage_info("recycle_storage", cls_shm, cls_size);
-    if (info == nullptr) return;
-
     info->lock_.lock();
     info->pool_.release(id);
     info->lock_.unlock();
@@ -225,13 +216,16 @@ void clear_storage(std::size_t id, std::size_t size) {
         return;
     }
 
-    std::size_t cls_size = calc_cls_size(size);
-    auto &      cls_shm  = cls_storage(cls_size);
+    std::size_t chunk_size = calc_chunk_size(size);
+    auto &      chunk_shm  = chunk_storage(chunk_size);
+
+    auto info = chunk_shm.get_info(chunk_size);
+    if (info == nullptr) return;
 
-    auto ptr = cls_storage_mem("clear_storage", cls_shm, cls_size, id);
+    auto ptr = info->at(chunk_size, id);
     if (ptr == nullptr) return;
 
-    auto cc_flag = reinterpret_cast<acc_t*>(ptr + cls_size);
+    auto cc_flag = reinterpret_cast<acc_t*>(ptr + chunk_size);
     for (unsigned k = 0;;) {
         auto cc_curr = cc_flag->load(std::memory_order_acquire);
         if (cc_curr == 0) return; // means this id has been cleared
@@ -241,9 +235,6 @@ void clear_storage(std::size_t id, std::size_t size) {
         ipc::yield(k);
     }
 
-    auto info = cls_storage_info("clear_storage", cls_shm, cls_size);
-    if (info == nullptr) return;
-
     info->lock_.lock();
     info->pool_.release(id);
     info->lock_.unlock();