100 lines
2.8 KiB
C++
100 lines
2.8 KiB
C++
|
#define DROGON_TEST_MAIN
|
||
|
|
#include <drogon/drogon_test.h>
|
||
|
|
#include <drogon/HttpAppFramework.h>
|
||
|
|
#include <drogon/config.h>
|
||
|
|
|
||
|
|
using namespace drogon;
|
||
|
|
using namespace trantor;
|
||
|
|
|
||
|
|
#if USE_POSTGRESQL
|
||
|
|
DROGON_TEST(ConnOptionsTest)
|
||
|
|
{
|
||
|
|
auto clientPtr = app().getDbClient();
|
||
|
|
|
||
|
|
// Run a query that will take longer than the statement_timeout
|
||
|
|
clientPtr->execSqlAsync(
|
||
|
|
"select pg_sleep(5);",
|
||
|
|
[TEST_CTX](const drogon::orm::Result &r) {
|
||
|
|
LOG_INFO << "select pg_sleep(5);";
|
||
|
|
FAULT("Statement should be canceled due to timeout.");
|
||
|
|
},
|
||
|
|
[TEST_CTX](const drogon::orm::DrogonDbException &e) {
|
||
|
|
LOG_INFO << "select pg_sleep(5); error(expected):"
|
||
|
|
<< e.base().what();
|
||
|
|
SUCCESS();
|
||
|
|
});
|
||
|
|
|
||
|
|
// Run sql codes to hold a lock for sometime
|
||
|
|
clientPtr->execSqlAsync(
|
||
|
|
R"(
|
||
|
|
DO $$
|
||
|
|
BEGIN
|
||
|
|
perform pg_advisory_xact_lock(12345);
|
||
|
|
perform pg_sleep(2);
|
||
|
|
END $$;)",
|
||
|
|
[TEST_CTX](const drogon::orm::Result &r) {
|
||
|
|
LOG_INFO << "pg_advisory_xact_lock transaction1 finished;";
|
||
|
|
SUCCESS();
|
||
|
|
},
|
||
|
|
[TEST_CTX](const drogon::orm::DrogonDbException &e) {
|
||
|
|
FAULT("query failed: ", e.base().what());
|
||
|
|
});
|
||
|
|
|
||
|
|
// Run sql codes that requires the same lock, and will be canceled due to
|
||
|
|
// lock_timeout
|
||
|
|
clientPtr->execSqlAsync(
|
||
|
|
R"(
|
||
|
|
DO $$
|
||
|
|
BEGIN
|
||
|
|
perform pg_sleep(0.5);
|
||
|
|
perform pg_advisory_xact_lock(12345);
|
||
|
|
END $$;)",
|
||
|
|
[TEST_CTX](const drogon::orm::Result &r) {
|
||
|
|
LOG_INFO << "pg_advisory_xact_lock transaction2 finished;";
|
||
|
|
FAULT("Statement should be canceled due to timeout.");
|
||
|
|
},
|
||
|
|
[TEST_CTX](const drogon::orm::DrogonDbException &e) {
|
||
|
|
LOG_INFO << "pg_advisory_xact_lock() error(expected):"
|
||
|
|
<< e.base().what();
|
||
|
|
SUCCESS();
|
||
|
|
});
|
||
|
|
}
|
||
|
|
|
||
|
|
#endif
|
||
|
|
|
||
|
|
int main(int argc, char **argv)
|
||
|
|
{
|
||
|
|
std::promise<void> p1;
|
||
|
|
std::future<void> f1 = p1.get_future();
|
||
|
|
app().setThreadNum(1);
|
||
|
|
#if USE_POSTGRESQL
|
||
|
|
app().addDbClient(orm::PostgresConfig{
|
||
|
|
"127.0.0.1", // host
|
||
|
|
5432, // port
|
||
|
|
"postgres", // dbname
|
||
|
|
"postgres", // username
|
||
|
|
"12345", // password
|
||
|
|
4, // connectionNum
|
||
|
|
"default", // name
|
||
|
|
false, // isFast
|
||
|
|
"", // charset
|
||
|
|
10, // timeout
|
||
|
|
false, // autobatch
|
||
|
|
{
|
||
|
|
{"statement_timeout", "3s"},
|
||
|
|
{"lock_timeout", "0.5s"},
|
||
|
|
} // connectOptions
|
||
|
|
});
|
||
|
|
#endif
|
||
|
|
std::thread thr([&]() {
|
||
|
|
app().getLoop()->queueInLoop([&]() { p1.set_value(); });
|
||
|
|
app().run();
|
||
|
|
});
|
||
|
|
|
||
|
|
f1.get();
|
||
|
|
int testStatus = test::run(argc, argv);
|
||
|
|
app().getLoop()->queueInLoop([]() { app().quit(); });
|
||
|
|
thr.join();
|
||
|
|
return testStatus;
|
||
|
|
}
|