summaryrefslogtreecommitdiff
path: root/src/third_party/wiredtiger/test/cppsuite/test_harness/configuration_settings.h
blob: 9f1d43773c9970e7b0c7d3943b7cefeb214042d7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
/* Include guard. */
#ifndef CONFIGURATION_SETTINGS_H
#define CONFIGURATION_SETTINGS_H

#include "wt_internal.h"
#include <string>
#include <stdexcept>

namespace test_harness {
class configuration {
    private:
    std::string _config;
    WT_CONFIG_PARSER *_config_parser;

    public:
    configuration(const char *test_config_name, const char *config) : _config(config)
    {
        int ret = wiredtiger_config_parser_open(nullptr, config, strlen(config), &_config_parser);
        if (ret != 0)
            throw std::invalid_argument(
              "failed to create configuration parser for provided config");
        if (wiredtiger_test_config_validate(nullptr, nullptr, test_config_name, config) != 0)
            throw std::invalid_argument(
              "failed to validate given config, ensure test config exists");
    }

    configuration(const char *test_config_name, const WT_CONFIG_ITEM &nested)
    {
        if (nested.type != WT_CONFIG_ITEM::WT_CONFIG_ITEM_STRUCT)
            throw std::invalid_argument("provided config item isn't a structure");
        int ret = wiredtiger_config_parser_open(nullptr, nested.str, nested.len, &_config_parser);
        if (ret != 0)
            throw std::invalid_argument(
              "failed to create configuration parser for provided sub config");
    }

    ~configuration()
    {
        if (_config_parser != nullptr) {
            _config_parser->close(_config_parser);
            _config_parser = nullptr;
        }
    }

    std::string
    get_config()
    {
        return _config;
    }

    /*
     * Wrapper functions for retrieving basic configuration values. Ideally the tests can avoid
     * using the config item struct provided by wiredtiger. However if they still wish to use it the
     * get and next functions can be used.
     */
    int
    get_string(const char *key, std::string &value)
    {
        WT_CONFIG_ITEM temp_value;
        WT_RET(_config_parser->get(_config_parser, key, &temp_value));
        if (temp_value.type != WT_CONFIG_ITEM::WT_CONFIG_ITEM_STRING ||
          temp_value.type != WT_CONFIG_ITEM::WT_CONFIG_ITEM_ID)
            return (-1);
        value = std::string(temp_value.str, temp_value.len);
        return (0);
    }

    int
    get_bool(const char *key, bool &value)
    {
        WT_CONFIG_ITEM temp_value;
        WT_RET(_config_parser->get(_config_parser, key, &temp_value));
        if (temp_value.type != WT_CONFIG_ITEM::WT_CONFIG_ITEM_BOOL)
            return (-1);
        value = temp_value.val != 0;
        return (0);
    }

    int
    get_int(const char *key, int64_t &value)
    {
        WT_CONFIG_ITEM temp_value;
        WT_RET(_config_parser->get(_config_parser, key, &temp_value));
        if (temp_value.type != WT_CONFIG_ITEM::WT_CONFIG_ITEM_NUM)
            return (-1);
        value = temp_value.val;
        return (0);
    }

    /*
     * Basic configuration parsing helper functions.
     */
    int
    next(WT_CONFIG_ITEM *key, WT_CONFIG_ITEM *value)
    {
        return _config_parser->next(_config_parser, key, value);
    }

    int
    get(const char *key, WT_CONFIG_ITEM *value)
    {
        return _config_parser->get(_config_parser, key, value);
    }
};
} // namespace test_harness

#endif