db-cpp/data_types.hpp at main · a1eaiactaest/db-cpp · GitHub

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
//
// Created by Piotrek Rybiec on 29/04/2025.
//

#ifndef DATA_TYPES_H
#define DATA_TYPES_H

#include <string>
#include <unordered_map>
#include <chrono>
#include <ctime>

enum class ConstraintType {
    PRIMARY_KEY,
    FOREIGN_KEY,
    UNIQUE,
    NOT_NULL,
    DEFAULT,
};

enum class DataType {
    INTEGER,
    FLOAT,
    BOOLEAN,
    STRING,
    DATE,
    DATETIME,
    NULL_VALUE,
};

class Constraint;
class ForeignKeyConstraint;
class Value;

using Date = std::chrono::year_month_day;
using DateTime = std::chrono::sys_time<std::chrono::milliseconds>;


// type safety, corresponding types are self-explanatory
using VariantType = std::variant<
    int,
    double,
    bool,
    std::string,
    Date,
    DateTime,
    std::monostate // for NULL
>;

inline auto dataTypeToString(const DataType& t) -> std::string {
    const std::unordered_map<DataType, std::string> tmap = {
        {DataType::INTEGER, "INTEGER"},
        {DataType::FLOAT, "FLOAT"},
        {DataType::BOOLEAN, "BOOLEAN"},
        {DataType::STRING, "STRING"},
        {DataType::DATE, "DATE"},
        {DataType::DATETIME, "DATETIME"},
    };
    return tmap.at(t);
}

inline auto stringToDataType(const std::string& s) -> DataType {
    const std::unordered_map<std::string, DataType> smap = {
        {"INTEGER", DataType::INTEGER},
        {"FLOAT", DataType::FLOAT},
        {"BOOLEAN", DataType::BOOLEAN},
        {"STRING", DataType::STRING},
        {"DATE", DataType::DATE},
        {"DATETIME", DataType::DATETIME},
    };
    return smap.at(s);
}


#endif //DATA_TYPES_H