Implement custom stickiness for flexible rollout

aruizs · Dec 28, 2023 · 187e1e3 · 187e1e3
1 parent 23bf404
commit 187e1e3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/strategies/flexiblerollout.cpp b/src/strategies/flexiblerollout.cpp
@@ -3,6 +3,7 @@
 #include <nlohmann/json.hpp>
 #include <random>
 
+
 namespace unleash {
 FlexibleRollout::FlexibleRollout(std::string_view parameters, std::string_view constraints)
     : Strategy("flexibleRollout", constraints) {
@@ -34,6 +35,10 @@ bool FlexibleRollout::isEnabled(const Context &context) {
         std::mt19937 rng(dev());
         std::uniform_int_distribution<std::mt19937::result_type> dist6(1, 100);
         return dist6(rng) <= m_rollout;
+    } else {
+        auto customFieldIt = context.properties.find(stickinessConfiguration);
+        if (customFieldIt == context.properties.end()) return false;
+        return normalizedMurmur3(m_groupId + ":" + customFieldIt->second) <= m_rollout;
     }
     return false;
 }

diff --git a/test/tests.cpp b/test/tests.cpp
@@ -36,7 +36,7 @@ std::vector<TestParam> readSpecificationTestFromDisk(const std::string &testPath
     // range-based to read each test
     for (auto &element : j) {  // Only features implemented for now
         auto testNumber = std::stoi(element.get<std::string>().substr(0, 2));
-        if (testNumber <= 11) {
+        if (testNumber <= 12) {
             std::cout << testPath + element.get<std::string>() << std::endl;
             std::ifstream testFile(testPath + element.get<std::string>());
             nlohmann::json testJson;