mirror of
https://github.com/rbock/sqlpp11.git
synced 2024-11-15 20:31:16 +08:00
179 lines
6.7 KiB
C++
179 lines
6.7 KiB
C++
/*
|
|
* Copyright (c) 2013 - 2016, Roland Bock
|
|
* All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without modification,
|
|
* are permitted provided that the following conditions are met:
|
|
*
|
|
* * Redistributions of source code must retain the above copyright notice,
|
|
* this list of conditions and the following disclaimer.
|
|
* * Redistributions in binary form must reproduce the above copyright notice,
|
|
* this list of conditions and the following disclaimer in the documentation
|
|
* and/or other materials provided with the distribution.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
|
* IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT,
|
|
* INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
|
|
* BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
* DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
|
|
* LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE
|
|
* OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
|
|
* OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
#include "TabSample.h"
|
|
#include <cassert>
|
|
#include <sqlpp11/alias_provider.h>
|
|
#include <sqlpp11/functions.h>
|
|
#include <sqlpp11/insert.h>
|
|
#include <sqlpp11/mysql/connection.h>
|
|
#include <sqlpp11/remove.h>
|
|
#include <sqlpp11/select.h>
|
|
#include <sqlpp11/transaction.h>
|
|
#include <sqlpp11/update.h>
|
|
|
|
#include <iostream>
|
|
#include <vector>
|
|
|
|
const auto library_raii = sqlpp::mysql::scoped_library_initializer_t{0, nullptr, nullptr};
|
|
|
|
namespace sql = sqlpp::mysql;
|
|
const auto tab = TabSample{};
|
|
|
|
void testSelectAll(sql::connection& db, int expectedRowCount)
|
|
{
|
|
std::cerr << "--------------------------------------" << std::endl;
|
|
int i = 0;
|
|
for (const auto& row : db(sqlpp::select(all_of(tab)).from(tab).unconditionally()))
|
|
{
|
|
++i;
|
|
std::cerr << ">>> row.alpha: " << row.alpha << ", row.beta: " << row.beta << ", row.gamma: " << row.gamma
|
|
<< std::endl;
|
|
assert(i == row.alpha);
|
|
};
|
|
assert(i == expectedRowCount);
|
|
|
|
auto preparedSelectAll = db.prepare(sqlpp::select(all_of(tab)).from(tab).unconditionally());
|
|
std::cerr << "--------------------------------------" << std::endl;
|
|
i = 0;
|
|
for (const auto& row : db(preparedSelectAll))
|
|
{
|
|
++i;
|
|
std::cerr << ">>> row.alpha: " << row.alpha << ", row.beta: " << row.beta << ", row.gamma: " << row.gamma
|
|
<< std::endl;
|
|
assert(i == row.alpha);
|
|
};
|
|
assert(i == expectedRowCount);
|
|
|
|
// Try running the same prepared statement again
|
|
std::cerr << "--------------------------------------" << std::endl;
|
|
i = 0;
|
|
for (const auto& row : db(preparedSelectAll))
|
|
{
|
|
++i;
|
|
std::cerr << ">>> row.alpha: " << row.alpha << ", row.beta: " << row.beta << ", row.gamma: " << row.gamma
|
|
<< std::endl;
|
|
assert(i == row.alpha);
|
|
};
|
|
assert(i == expectedRowCount);
|
|
std::cerr << "--------------------------------------" << std::endl;
|
|
}
|
|
|
|
int Select(int, char*[])
|
|
{
|
|
auto config = std::make_shared<sql::connection_config>();
|
|
config->user = "root";
|
|
config->database = "sqlpp_mysql";
|
|
config->debug = true;
|
|
try
|
|
{
|
|
sql::connection db(config);
|
|
}
|
|
catch (const sqlpp::exception& e)
|
|
{
|
|
std::cerr << "For testing, you'll need to create a database sqlpp_mysql for user root (no password)" << std::endl;
|
|
std::cerr << e.what() << std::endl;
|
|
return 1;
|
|
}
|
|
try
|
|
{
|
|
sql::connection db(config);
|
|
db.execute(R"(DROP TABLE IF EXISTS tab_sample)");
|
|
db.execute(R"(CREATE TABLE tab_sample (
|
|
alpha bigint(20) AUTO_INCREMENT,
|
|
beta varchar(255) DEFAULT NULL,
|
|
gamma bool DEFAULT NULL,
|
|
PRIMARY KEY (alpha)
|
|
))");
|
|
db.execute(R"(DROP TABLE IF EXISTS tab_foo)");
|
|
db.execute(R"(CREATE TABLE tab_foo (
|
|
omega bigint(20) DEFAULT NULL
|
|
))");
|
|
|
|
testSelectAll(db, 0);
|
|
db(insert_into(tab).default_values());
|
|
testSelectAll(db, 1);
|
|
db(insert_into(tab).set(tab.gamma = true, tab.beta = "cheesecake"));
|
|
testSelectAll(db, 2);
|
|
db(insert_into(tab).set(tab.gamma = true, tab.beta = "cheesecake"));
|
|
testSelectAll(db, 3);
|
|
|
|
// test functions and operators
|
|
db(select(all_of(tab)).from(tab).where(tab.alpha.is_null()));
|
|
db(select(all_of(tab)).from(tab).where(tab.alpha.is_not_null()));
|
|
db(select(all_of(tab)).from(tab).where(tab.alpha.in(1, 2, 3)));
|
|
db(select(all_of(tab)).from(tab).where(tab.alpha.in(sqlpp::value_list(std::vector<int>{1, 2, 3, 4}))));
|
|
db(select(all_of(tab)).from(tab).where(tab.alpha.not_in(1, 2, 3)));
|
|
db(select(all_of(tab)).from(tab).where(tab.alpha.not_in(sqlpp::value_list(std::vector<int>{1, 2, 3, 4}))));
|
|
db(select(count(tab.alpha)).from(tab).unconditionally());
|
|
db(select(avg(tab.alpha)).from(tab).unconditionally());
|
|
db(select(max(tab.alpha)).from(tab).unconditionally());
|
|
db(select(min(tab.alpha)).from(tab).unconditionally());
|
|
db(select(exists(select(tab.alpha).from(tab).where(tab.alpha > 7))).from(tab).unconditionally());
|
|
db(select(all_of(tab)).from(tab).where(tab.alpha == any(select(tab.alpha).from(tab).where(tab.alpha < 3))));
|
|
db(select(all_of(tab)).from(tab).where(tab.alpha == some(select(tab.alpha).from(tab).where(tab.alpha < 3))));
|
|
|
|
db(select(all_of(tab)).from(tab).where(tab.alpha + tab.alpha > 3));
|
|
db(select(all_of(tab)).from(tab).where((tab.beta + tab.beta) == ""));
|
|
db(select(all_of(tab)).from(tab).where((tab.beta + tab.beta).like("%'\"%")));
|
|
|
|
// insert
|
|
db(insert_into(tab).set(tab.gamma = true));
|
|
|
|
// update
|
|
db(update(tab).set(tab.gamma = false).where(tab.alpha.in(1)));
|
|
db(update(tab).set(tab.gamma = false).where(tab.alpha.in(sqlpp::value_list(std::vector<int>{1, 2, 3, 4}))));
|
|
|
|
// remove
|
|
{
|
|
db(remove_from(tab).where(tab.alpha == tab.alpha + 3));
|
|
|
|
auto result = db(select(all_of(tab)).from(tab).unconditionally());
|
|
std::cerr << "Accessing a field directly from the result (using the current row): " << result.begin()->alpha
|
|
<< std::endl;
|
|
std::cerr << "Can do that again, no problem: " << result.begin()->alpha << std::endl;
|
|
}
|
|
|
|
// transaction
|
|
{
|
|
auto tx = start_transaction(db);
|
|
auto result = db(select(all_of(tab), select(max(tab.alpha)).from(tab)).from(tab).unconditionally());
|
|
if (const auto& row = *result.begin())
|
|
{
|
|
long a = row.alpha;
|
|
long m = row.max;
|
|
std::cerr << "-----------------------------" << a << ", " << m << std::endl;
|
|
}
|
|
tx.commit();
|
|
}
|
|
}
|
|
catch (const std::exception& e)
|
|
{
|
|
std::cerr << "Exception: " << e.what() << std::endl;
|
|
return 1;
|
|
}
|
|
return 0;
|
|
}
|