diff --git a/src/include/utils/env.hpp b/src/include/utils/env.hpp index 74a6cba..f21f24a 100644 --- a/src/include/utils/env.hpp +++ b/src/include/utils/env.hpp @@ -8,6 +8,8 @@ const char *TryGetEnv(const char *name); std::string GetEnvOrDefault(const char *name, const char *default_value); +uint32_t GetEnvOrDefaultInt(const char *name, uint32_t default_value); + bool IsEnvEnabled(const char *name); } // namespace duckdb diff --git a/src/utils/env.cpp b/src/utils/env.cpp index 720cce7..b8ed6e6 100644 --- a/src/utils/env.cpp +++ b/src/utils/env.cpp @@ -21,6 +21,14 @@ std::string GetEnvOrDefault(const char *name, const char *default_value) { return default_value; } +uint32_t GetEnvOrDefaultInt(const char *name, uint32_t default_value) { + const char *res = TryGetEnv(name); + if (res) { + return std::atoi(res); + } + return default_value; +} + bool IsEnvEnabled(const char *name) { const char *res = TryGetEnv(name); if (!res) {