Учимся пользоваться odb в нашем меленьком проекте

src/database.hxx

@@ -0,0 +1,177 @@
+// file      : hello/database.hxx
+// copyright : not copyrighted - public domain
+
+//
+// Create concrete database instance based on the DATABASE_* macros.
+//
+
+#ifndef DATABASE_HXX
+#define DATABASE_HXX
+
+#include <cstdlib> // std::exit
+#include <iostream>
+#include <memory> // std::unique_ptr
+#include <string>
+
+#include <odb/database.hxx>
+
+#if defined(DATABASE_MYSQL)
+#include <odb/mysql/database.hxx>
+#elif defined(DATABASE_SQLITE)
+#include <odb/connection.hxx>
+#include <odb/schema-catalog.hxx>
+#include <odb/sqlite/database.hxx>
+#include <odb/transaction.hxx>
+#elif defined(DATABASE_PGSQL)
+#include <odb/pgsql/database.hxx>
+#elif defined(DATABASE_ORACLE)
+#include <odb/oracle/database.hxx>
+#elif defined(DATABASE_MSSQL)
+#include <odb/mssql/database.hxx>
+#else
+#error unknown database; did you forget to define the DATABASE_* macros?
+#endif
+
+#include <filesystem>
+#include <random>
+
+struct testData
+{
+    std::string first;
+    std::string last;
+    int age;
+};
+
+// Функция для генерации случайной строки (имени или фамилии)
+std::string generate_random_string(const std::vector<std::string>& pool)
+{
+    static std::random_device rd;
+    static std::mt19937 gen(rd());
+    std::uniform_int_distribution<> dis(0, pool.size() - 1);
+    return pool[dis(gen)];
+}
+
+// Функция для генерации случайного года
+int generate_random_year(int min_year = 1900, int max_year = 2020)
+{
+    static std::random_device rd;
+    static std::mt19937 gen(rd());
+    std::uniform_int_distribution<> dis(min_year, max_year);
+    return dis(gen);
+}
+
+std::vector<testData> fillDB()
+{
+    std::vector<std::string> first_names = {
+        "John", "Jane", "Alex", "Chris", "Robert", "Emily", "James", "Linda", "David", "Sarah",
+        "Michael", "Elizabeth", "Daniel", "Samantha", "William", "Olivia", "Ethan", "Sophia", "Joshua", "Charlotte",
+        "Daniel", "Grace", "Benjamin", "Isabella", "Matthew", "Victoria", "Henry", "Abigail", "Samuel", "Megan",
+        "Lucas", "Lily", "Andrew", "Madison", "Jackson", "Chloe", "Aiden", "Amelia", "Thomas", "Natalie",
+        "Ryan", "Zoe", "Jack", "Harper", "Elijah", "Ava", "Isaac", "Mia", "Caleb", "Ella"
+    };
+    std::vector<std::string> last_names = {
+        "Doe", "Smith", "Johnson", "Williams", "Jones", "Brown", "Davis", "Miller", "Wilson", "Moore",
+        "Taylor", "Anderson", "Thomas", "Jackson", "White", "Harris", "Martin", "Thompson", "Garcia", "Martinez",
+        "Roberts", "Clark", "Lewis", "Walker", "Young", "Allen", "King", "Wright", "Scott", "Adams",
+        "Baker", "Gonzalez", "Nelson", "Carter", "Mitchell", "Perez", "Robinson", "Hughes", "Flores", "Cook",
+        "Rogers", "Gutierrez", "Ramirez", "Diaz", "Perez", "Ross", "Sanders", "Price", "Howard", "Cooper"
+    };
+
+    std::vector<testData> vecTest;
+
+    for (int i = 0; i < 50; ++i)
+    {
+        std::string first_name = generate_random_string(first_names);
+        std::string last_name = generate_random_string(last_names);
+        int birth_year = generate_random_year(1900, 2000);
+
+        vecTest.push_back({ first_name, last_name, birth_year });
+    }
+
+    return vecTest;
+}
+
+inline std::unique_ptr<odb::database> openDB(std::string path_db)
+{
+    // Открыть БД, если она существует
+    if (std::filesystem::is_regular_file(path_db))
+        return std::unique_ptr<odb::core::database>(new odb::sqlite::database(path_db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE));
+
+    // И создать новую, если не существует
+    std::unique_ptr<odb::core::database> db(new odb::sqlite::database(path_db, SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE));
+
+    odb::core::connection_ptr connect_ptr(db->connection());
+
+    connect_ptr->execute("PRAGMA foreign_keys=OFF");
+
+    odb::core::transaction tans(db->begin());
+    odb::core::schema_catalog::create_schema(*db);
+    tans.commit();
+
+    connect_ptr->execute("PRAGMA foreign_keys=ON");
+
+    return db;
+}
+
+inline std::unique_ptr<odb::database> create_database(int& argc, char* argv[])
+{
+    using namespace std;
+    using namespace odb::core;
+
+    if (argc > 1 && argv[1] == string("--help"))
+    {
+        cout << "Usage: " << argv[0] << " [options]" << endl
+             << "Options:" << endl;
+
+#if defined(DATABASE_MYSQL)
+        odb::mysql::database::print_usage(cout);
+#elif defined(DATABASE_SQLITE)
+        odb::sqlite::database::print_usage(cout);
+#elif defined(DATABASE_PGSQL)
+        odb::pgsql::database::print_usage(cout);
+#elif defined(DATABASE_ORACLE)
+        odb::oracle::database::print_usage(cout);
+#elif defined(DATABASE_MSSQL)
+        odb::mssql::database::print_usage(cout);
+#endif
+
+        exit(0);
+    }
+
+#if defined(DATABASE_MYSQL)
+    unique_ptr<database> db(new odb::mysql::database(argc, argv));
+#elif defined(DATABASE_SQLITE)
+
+    unique_ptr<database> db(new odb::sqlite::database(argc,
+                                                      argv,
+                                                      false,
+                                                      SQLITE_OPEN_READWRITE | SQLITE_OPEN_CREATE));
+
+    // Create the database schema. Due to bugs in SQLite foreign key
+    // support for DDL statements, we need to temporarily disable
+    // foreign keys.
+    //
+    {
+        connection_ptr connect_ptr(db->connection());
+
+        connect_ptr->execute("PRAGMA foreign_keys=OFF");
+
+        transaction tans(connect_ptr->begin());
+        schema_catalog::create_schema(*db);
+        tans.commit();
+
+        connect_ptr->execute("PRAGMA foreign_keys=ON");
+    }
+#elif defined(DATABASE_PGSQL)
+    unique_ptr<database> db(new odb::pgsql::database(argc, argv));
+#elif defined(DATABASE_ORACLE)
+    unique_ptr<database> db(new odb::oracle::database(argc, argv));
+#elif defined(DATABASE_MSSQL)
+    unique_ptr<database> db(
+    new odb::mssql::database(argc, argv, false, "TrustServerCertificate=yes"));
+#endif
+
+    return db;
+}
+
+#endif // DATABASE_HXX