From 26e04849cec92f9f0f495f857cea6377cedd948c Mon Sep 17 00:00:00 2001
From: matyhtf <rango@swoole.com>
Date: Mon, 28 Oct 2024 12:33:42 +0800
Subject: [PATCH] Added BitMap

---
 core-tests/src/core/util.cpp | 42 ++++++++++++++++++++++++++++++
 include/swoole_util.h        | 50 ++++++++++++++++++++++++++++++++++++
 2 files changed, 92 insertions(+)
 create mode 100644 core-tests/src/core/util.cpp

diff --git a/core-tests/src/core/util.cpp b/core-tests/src/core/util.cpp
new file mode 100644
index 00000000000..d756d542321
--- /dev/null
+++ b/core-tests/src/core/util.cpp
@@ -0,0 +1,42 @@
+/*
+  +----------------------------------------------------------------------+
+  | Swoole                                                               |
+  +----------------------------------------------------------------------+
+  | This source file is subject to version 2.0 of the Apache license,    |
+  | that is bundled with this package in the file LICENSE, and is        |
+  | available through the world-wide-web at the following url:           |
+  | http://www.apache.org/licenses/LICENSE-2.0.html                      |
+  | If you did not receive a copy of the Apache2.0 license and are unable|
+  | to obtain it through the world-wide-web, please send a note to       |
+  | license@swoole.com so we can mail you a copy immediately.            |
+  +----------------------------------------------------------------------+
+  | @link     https://www.swoole.com/                                    |
+  | @contact  team@swoole.com                                            |
+  | @license  https://github.com/swoole/swoole-src/blob/master/LICENSE   |
+  | @Author   Tianfeng Han  <rango@swoole.com>                           |
+  +----------------------------------------------------------------------+
+*/
+
+#include "test_core.h"
+
+#include "swoole_util.h"
+
+TEST(util, bitmap) {
+    swoole::BitMap m(4096);
+
+    m.set(199);
+    m.set(1234);
+    m.set(3048);
+
+    ASSERT_EQ(m.get(199), true);
+    ASSERT_EQ(m.get(1234), true);
+    ASSERT_EQ(m.get(3048), true);
+
+    ASSERT_EQ(m.get(2048), false);
+    ASSERT_EQ(m.get(128), false);
+
+    m.unset(1234);
+    ASSERT_EQ(m.get(1234), false);
+
+    m.clear();
+}
diff --git a/include/swoole_util.h b/include/swoole_util.h
index deb87e73664..8d6d10b11fb 100644
--- a/include/swoole_util.h
+++ b/include/swoole_util.h
@@ -122,6 +122,56 @@ class ScopeGuard {
     bool _active;
 };
 
+class BitMap {
+  private:
+    uint64_t *array_;
+    size_t n_bits_;
+
+    size_t get_array_size(size_t n_bits) {
+        return (((n_bits) + 63) / 64 * 8);
+    }
+
+    size_t get_offset(size_t i) {
+        assert(i < n_bits_);
+        /* (i / 64) */
+        return i >> 6;
+    }
+
+    uint64_t to_int(size_t i, size_t offset) {
+        return ((uint64_t) 1) << (i - (offset << 6));
+    }
+
+  public:
+    BitMap(size_t n_bits) {
+        assert(n_bits > 0);
+        array_ = (uint64_t *) new uint64_t[get_array_size(n_bits)];
+        n_bits_ = n_bits;
+    }
+
+    ~BitMap() {
+        delete[] array_;
+    }
+
+    void clear() {
+        memset(array_, 0, get_array_size(n_bits_));
+    }
+
+    void set(size_t i) {
+        const size_t off = get_offset(i);
+        array_[off] |= to_int(i, off);
+    }
+
+    void unset(size_t i) {
+        const size_t off = get_offset(i);
+        array_[off] &= ~to_int(i, off);
+    }
+
+    bool get(size_t i) {
+        const size_t off = get_offset(i);
+        return array_[off] & to_int(i, off);
+    }
+};
+
 namespace detail {
 enum class ScopeGuardOnExit {};