set values of missing in the config file statement types to 0

duckdb · Sep 26, 2024 · f8aea67 · f8aea67
1 parent 2b9ae4d
commit f8aea67
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/src/include/random_nums_config.hpp b/src/include/random_nums_config.hpp
@@ -41,7 +41,9 @@ enum class RandomPercentagesEnum : idx_t {
     LIMIT_PERCENT_MODIFIER_LIMIT = 20,
     LIMIT_PERCENT_MODIFIER_OFFSET = 21,
     LIMIT_MODIFIER_LIMIT = 22,
-    LIMIT_MODIFIER_OFFSET = 23
+    LIMIT_MODIFIER_OFFSET = 23,
+
+    COUNT
 
 };
 

diff --git a/src/random_nums_config.cpp b/src/random_nums_config.cpp
@@ -67,6 +67,8 @@ void ParseJsonObj(yyjson_val *obj, unordered_map<RandomPercentagesEnum, idx_t> &
     }
 }
 
+
+
 unordered_map<RandomPercentagesEnum, idx_t> GetConfigFromFile(const char *json_string) {
 
     unordered_map<RandomPercentagesEnum, idx_t> config_from_file;
@@ -82,6 +84,13 @@ unordered_map<RandomPercentagesEnum, idx_t> GetConfigFromFile(const char *json_s
         yyjson_doc_free(doc);
         return GetDefaultConfig();
     }
+    // set values of missing statement types to 0
+    for (idx_t i = 0; i < static_cast<int>(RandomPercentagesEnum::COUNT); ++i) {
+        RandomPercentagesEnum statement_type = static_cast<RandomPercentagesEnum>(i);
+        if (config_from_file.find(statement_type) == config_from_file.end()) {
+            config_from_file[statement_type] = 0;
+        }
+    }
     return config_from_file;
 }
 } // namespace duckdb