#include "stdafx.h"
|
#include "../include/qtl_postgres.hpp"
|
#include <vector>
|
#include <thread>
|
#include <system_error>
|
#include <time.h>
|
#include <limits.h>
|
#include "../include/qtl_postgres_pool.hpp"
|
#include "../include/qtl_asio.hpp"
|
|
using namespace qtl::postgres;
|
|
void LogError(const error& e)
|
{
|
fprintf(stderr, "PostgreSQL Error: %s\n", e.what());
|
}
|
|
const char postgres_server[] = "localhost";
|
const char postgres_user[] = "postgres";
|
const char postgres_password[] = "111111";
|
const char postgres_database[] = "test";
|
|
qtl::asio::service service;
|
|
void SimpleTest()
|
{
|
async_connection connection;
|
service.reset();
|
std::map<std::string, std::string> params;
|
params["host"] = postgres_server;
|
params["dbname"] = postgres_database;
|
params["user"] = postgres_user;
|
params["password"] = postgres_password;
|
connection.open(service, [&connection](const error& e) {
|
if (e)
|
{
|
LogError(e);
|
service.stop();
|
}
|
else
|
{
|
printf("Connect to PostgreSQL ok.\n");
|
connection.simple_query("select id, name from test", [](const std::string& id, const std::string& name) {
|
printf("%s\t%s\n", id.data(), name.data());
|
return true;
|
}, [&connection](const error& e, size_t row_count) {
|
if (e)
|
LogError(e);
|
else
|
printf("Total %lu rows.\n", row_count);
|
|
connection.close();
|
});
|
}
|
}, params);
|
|
service.run();
|
}
|
|
void insert(async_connection& connection, int next)
|
{
|
qtl::asio::async_execute(connection, [&connection, next](const error& e, uint64_t affected) mutable {
|
if (e)
|
{
|
LogError(e);
|
service.stop();
|
}
|
else
|
{
|
if (next)
|
{
|
--next;
|
asio::post(service.context(), [&connection, next]() {
|
insert(connection, next);
|
});
|
}
|
else
|
{
|
service.stop();
|
}
|
}
|
}, "insert into test(name, createtime) values($1, LOCALTIMESTAMP)", 0, "test name");
|
}
|
|
void ExecuteTest()
|
{
|
async_connection connection;
|
service.reset();
|
std::map<std::string, std::string> params;
|
params["host"] = postgres_server;
|
params["dbname"] = postgres_database;
|
params["user"] = postgres_user;
|
params["password"] = postgres_password;
|
qtl::asio::async_open(service, connection, [&connection](const error& e) {
|
if (e)
|
{
|
LogError(e);
|
service.stop();
|
}
|
else
|
{
|
printf("Connect to PostgreSQL ok.\n");
|
insert(connection, 10);
|
}
|
}, params);
|
|
service.run();
|
};
|
|
void QueryTest()
|
{
|
async_connection connection;
|
service.reset();
|
std::map<std::string, std::string> params;
|
params["host"] = postgres_server;
|
params["dbname"] = postgres_database;
|
params["user"] = postgres_user;
|
params["password"] = postgres_password;
|
connection.open(service, [&connection](const error& e) {
|
if (e)
|
{
|
LogError(e);
|
service.stop();
|
}
|
else
|
{
|
printf("Connect to PostgreSQL ok.\n");
|
connection.query("select id, name, CreateTime from test",
|
[](int32_t id, const std::string& name, const qtl::postgres::timestamp& create_time) {
|
printf("%d\t%s\t%s\n", id, name.data(), create_time.to_string().data());
|
}, [&connection](const error& e) {
|
printf("query has completed.\n");
|
if (e)
|
LogError(e);
|
|
connection.close();
|
});
|
}
|
}, params);
|
|
service.run();
|
}
|
|
int main(int argc, char* argv[])
|
{
|
ExecuteTest();
|
SimpleTest();
|
QueryTest();
|
return 0;
|
}
|
