mirror/sqlgen
1
0
Fork 0
mirror of https://github.com/getml/sqlgen.git synced 2026-05-05 14:43:08 -05:00
Code Issues Packages Projects Releases Wiki Activity

Added operations on time stamps (#24)

Browse Source
This commit is contained in:
Dr. Patrick Urbanke (劉自成)
2025-07-02 20:04:50 +02:00
committed by GitHub
parent aa1a96f1c0
commit f95156df77
30 changed files with 1410 additions and 436 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/postgres/test_select_from_with_timestamps.cpp
+89
View File
@@ -0,0 +1,89 @@
#ifndef SQLGEN_BUILD_DRY_TESTS_ONLY
#include <gtest/gtest.h>
#include <chrono>
#include <ranges>
#include <rfl.hpp>
#include <rfl/json.hpp>
#include <sqlgen.hpp>
#include <sqlgen/postgres.hpp>
#include <vector>
namespace test_range_select_from_with_timestamps {
struct Person {
sqlgen::PrimaryKey<uint32_t, sqlgen::auto_incr> id;
std::string first_name;
std::string last_name;
sqlgen::Timestamp<"%Y-%m-%d"> birthday;
};
TEST(postgres, test_range_select_from_with_timestamps) {
const auto people1 = std::vector<Person>(
{Person{.first_name = "Homer",
.last_name = "Simpson",
.birthday = sqlgen::Timestamp<"%Y-%m-%d">("1970-01-01")},
Person{.first_name = "Bart",
.last_name = "Simpson",
.birthday = sqlgen::Timestamp<"%Y-%m-%d">("2000-01-01")},
Person{.first_name = "Lisa",
.last_name = "Simpson",
.birthday = sqlgen::Timestamp<"%Y-%m-%d">("2002-01-01")},
Person{.first_name = "Maggie",
.last_name = "Simpson",
.birthday = sqlgen::Timestamp<"%Y-%m-%d">("2010-01-01")}});
const auto credentials = sqlgen::postgres::Credentials{.user = "postgres",
.password = "password",
.host = "localhost",
.dbname = "postgres"};
using namespace sqlgen;
struct Birthday {
Date birthday;
Date birthday_recreated;
time_t birthday_unixepoch;
double age_in_days;
int hour;
int minute;
int second;
int weekday;
};
const auto get_birthdays =
select_from<Person>(
("birthday"_c + std::chrono::days(10)) | as<"birthday">,
((cast<Date>(concat(cast<std::string>(year("birthday"_c)), "-",
cast<std::string>(month("birthday"_c)), "-",
cast<std::string>(day("birthday"_c))))) +
std::chrono::days(10)) |
as<"birthday_recreated">,
days_between("birthday"_c, Date("2011-01-01")) | as<"age_in_days">,
unixepoch("birthday"_c + std::chrono::days(10)) |
as<"birthday_unixepoch">,
hour("birthday"_c) | as<"hour">, minute("birthday"_c) | as<"minute">,
second("birthday"_c) | as<"second">,
weekday("birthday"_c) | as<"weekday">) |
order_by("id"_c) | to<std::vector<Birthday>>;
const auto birthdays = postgres::connect(credentials)
.and_then(drop<Person> | if_exists)
.and_then(write(std::ref(people1)))
.and_then(get_birthdays)
.value();
const std::string expected_query =
R"(SELECT "birthday" + INTERVAL '10 days' AS "birthday", cast((cast(extract(YEAR from "birthday") as TEXT) || '-' || cast(extract(MONTH from "birthday") as TEXT) || '-' || cast(extract(DAY from "birthday") as TEXT)) as TIMESTAMP) + INTERVAL '10 days' AS "birthday_recreated", cast('2011-01-01' as DATE) - cast("birthday" as DATE) AS "age_in_days", extract(EPOCH FROM "birthday" + INTERVAL '10 days') AS "birthday_unixepoch", extract(HOUR from "birthday") AS "hour", extract(MINUTE from "birthday") AS "minute", extract(SECOND from "birthday") AS "second", extract(DOW from "birthday") AS "weekday" FROM "Person" ORDER BY "id";)";
const std::string expected =
R"([{"birthday":"1970-01-11","birthday_recreated":"1970-01-11","birthday_unixepoch":864000,"age_in_days":14975.0,"hour":0,"minute":0,"second":0,"weekday":4},{"birthday":"2000-01-11","birthday_recreated":"2000-01-11","birthday_unixepoch":947548800,"age_in_days":4018.0,"hour":0,"minute":0,"second":0,"weekday":6},{"birthday":"2002-01-11","birthday_recreated":"2002-01-11","birthday_unixepoch":1010707200,"age_in_days":3287.0,"hour":0,"minute":0,"second":0,"weekday":2},{"birthday":"2010-01-11","birthday_recreated":"2010-01-11","birthday_unixepoch":1263168000,"age_in_days":365.0,"hour":0,"minute":0,"second":0,"weekday":5}])";
EXPECT_EQ(postgres::to_sql(get_birthdays), expected_query);
EXPECT_EQ(rfl::json::write(birthdays), expected);
}
} // namespace test_range_select_from_with_timestamps
#endif
tests/postgres/test_where_with_timestamps.cpp
+66
View File
@@ -0,0 +1,66 @@
#ifndef SQLGEN_BUILD_DRY_TESTS_ONLY
#include <gtest/gtest.h>
#include <chrono>
#include <rfl.hpp>
#include <rfl/json.hpp>
#include <sqlgen.hpp>
#include <sqlgen/postgres.hpp>
#include <vector>
namespace test_where_with_timestamps {
struct Person {
sqlgen::PrimaryKey<uint32_t, sqlgen::auto_incr> id;
std::string first_name;
std::string last_name;
sqlgen::Date birthday;
};
TEST(postgres, test_where_with_timestamps) {
const auto people1 =
std::vector<Person>({Person{.first_name = "Homer",
.last_name = "Simpson",
.birthday = sqlgen::Date("1970-01-01")},
Person{.first_name = "Bart",
.last_name = "Simpson",
.birthday = sqlgen::Date("2000-01-01")},
Person{.first_name = "Lisa",
.last_name = "Simpson",
.birthday = sqlgen::Date("2002-01-01")},
Person{.first_name = "Maggie",
.last_name = "Simpson",
.birthday = sqlgen::Date("2010-01-01")}});
const auto credentials = sqlgen::postgres::Credentials{.user = "postgres",
.password = "password",
.host = "localhost",
.dbname = "postgres"};
using namespace sqlgen;
using namespace std::literals::chrono_literals;
const auto conn =
sqlgen::postgres::connect(credentials).and_then(drop<Person> | if_exists);
sqlgen::write(conn, people1).value();
const auto query =
sqlgen::read<std::vector<Person>> |
where("birthday"_c + std::chrono::years(11) - std::chrono::weeks(10) +
std::chrono::milliseconds(4000000) >
Date("2010-01-01")) |
order_by("id"_c);
const auto people2 = query(conn).value();
const std::string expected =
R"([{"id":2,"first_name":"Bart","last_name":"Simpson","birthday":"2000-01-01"},{"id":3,"first_name":"Lisa","last_name":"Simpson","birthday":"2002-01-01"},{"id":4,"first_name":"Maggie","last_name":"Simpson","birthday":"2010-01-01"}])";
EXPECT_EQ(rfl::json::write(people2), expected);
}
} // namespace test_where_with_timestamps
#endif