mirror of
https://github.com/rbock/sqlpp11.git
synced 2024-11-16 12:51:13 +08:00
496 lines
12 KiB
C++
496 lines
12 KiB
C++
#pragma once
|
|
|
|
/*
|
|
* Copyright (c) 2013-2015, 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 <iostream>
|
|
#include <sqlpp11/connection.h>
|
|
#include <sqlpp11/detail/float_safe_ostringstream.h>
|
|
#include <sqlpp11/transaction.h>
|
|
#include <sqlpp11/schema.h>
|
|
#include <sqlpp11/serialize.h>
|
|
#include <sqlpp11/serializer_context.h>
|
|
#include <sstream>
|
|
|
|
// an object to store internal Mock flags and values to validate in tests
|
|
struct InternalMockData
|
|
{
|
|
sqlpp::isolation_level _last_isolation_level;
|
|
sqlpp::isolation_level _default_isolation_level;
|
|
};
|
|
|
|
struct MockDb : public sqlpp::connection
|
|
{
|
|
using _traits =
|
|
::sqlpp::make_traits<::sqlpp::no_value_t>;
|
|
|
|
struct _serializer_context_t
|
|
{
|
|
::sqlpp::detail::float_safe_ostringstream _os;
|
|
|
|
_serializer_context_t() = default;
|
|
_serializer_context_t(const _serializer_context_t& rhs)
|
|
{
|
|
_os << rhs._os.str();
|
|
}
|
|
|
|
std::string str() const
|
|
{
|
|
return _os.str();
|
|
}
|
|
|
|
void reset()
|
|
{
|
|
_os.str("");
|
|
}
|
|
|
|
template <typename T>
|
|
::sqlpp::detail::float_safe_ostringstream& operator<<(T t)
|
|
{
|
|
return _os << t;
|
|
}
|
|
|
|
static std::string escape(std::string arg)
|
|
{
|
|
return sqlpp::serializer_context_t::escape(arg);
|
|
}
|
|
};
|
|
|
|
using _interpreter_context_t = _serializer_context_t;
|
|
|
|
_serializer_context_t get_serializer_context()
|
|
{
|
|
return {};
|
|
}
|
|
|
|
template <typename T>
|
|
static _serializer_context_t& _serialize_interpretable(const T& x, _serializer_context_t& context)
|
|
{
|
|
serialize(context, x);
|
|
return context;
|
|
}
|
|
|
|
template <typename T>
|
|
static _serializer_context_t& _interpret_interpretable(const T& x, _interpreter_context_t& context)
|
|
{
|
|
serialize(context, x);
|
|
return context;
|
|
}
|
|
|
|
class result_t
|
|
{
|
|
public:
|
|
constexpr bool operator==(const result_t&) const
|
|
{
|
|
return true;
|
|
}
|
|
|
|
template <typename ResultRow>
|
|
void next(ResultRow& result_row)
|
|
{
|
|
result_row._invalidate();
|
|
}
|
|
};
|
|
|
|
// Directly executed statements start here
|
|
template <typename T>
|
|
auto _run(const T& t, ::sqlpp::consistent_t) -> decltype(t._run(*this))
|
|
{
|
|
return t._run(*this);
|
|
}
|
|
|
|
template <typename Check, typename T>
|
|
auto _run(const T& t, Check) -> Check;
|
|
|
|
template <typename T>
|
|
auto operator()(const T& t) -> decltype(this->_run(t, sqlpp::run_check_t<_serializer_context_t, T>{}))
|
|
{
|
|
return _run(t, sqlpp::run_check_t<_serializer_context_t, T>{});
|
|
}
|
|
|
|
size_t execute(const std::string&)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
template <
|
|
typename Statement,
|
|
typename Enable = typename std::enable_if<not std::is_convertible<Statement, std::string>::value, void>::type>
|
|
size_t execute(const Statement& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running execute call with\n" << context.str() << std::endl;
|
|
return execute(context.str());
|
|
}
|
|
|
|
template <typename Insert>
|
|
size_t insert(const Insert& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running insert call with\n" << context.str() << std::endl;
|
|
return 0;
|
|
}
|
|
|
|
template <typename Update>
|
|
size_t update(const Update& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running update call with\n" << context.str() << std::endl;
|
|
return 0;
|
|
}
|
|
|
|
template <typename Remove>
|
|
size_t remove(const Remove& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running remove call with\n" << context.str() << std::endl;
|
|
return 0;
|
|
}
|
|
|
|
template <typename Select>
|
|
result_t select(const Select& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running select call with\n" << context.str() << std::endl;
|
|
return {};
|
|
}
|
|
|
|
// Prepared statements start here
|
|
using _prepared_statement_t = std::nullptr_t;
|
|
|
|
template <typename T>
|
|
auto _prepare(const T& t, ::sqlpp::consistent_t) -> decltype(t._prepare(*this))
|
|
{
|
|
return t._prepare(*this);
|
|
}
|
|
|
|
template <typename Check, typename T>
|
|
auto _prepare(const T& t, Check) -> Check;
|
|
|
|
template <typename T>
|
|
auto prepare(const T& t) -> decltype(this->_prepare(t, sqlpp::prepare_check_t<_serializer_context_t, T>{}))
|
|
{
|
|
return _prepare(t, sqlpp::prepare_check_t<_serializer_context_t, T>{});
|
|
}
|
|
|
|
template <typename Statement>
|
|
_prepared_statement_t prepare_execute(Statement& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running prepare execute call with\n" << context.str() << std::endl;
|
|
return nullptr;
|
|
}
|
|
|
|
template <typename Insert>
|
|
_prepared_statement_t prepare_insert(Insert& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running prepare insert call with\n" << context.str() << std::endl;
|
|
return nullptr;
|
|
}
|
|
|
|
template <typename PreparedExecute>
|
|
size_t run_prepared_execute(const PreparedExecute&)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
template <typename PreparedInsert>
|
|
size_t run_prepared_insert(const PreparedInsert&)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
template <typename Select>
|
|
_prepared_statement_t prepare_select(Select& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running prepare select call with\n" << context.str() << std::endl;
|
|
return nullptr;
|
|
}
|
|
|
|
template <typename PreparedSelect>
|
|
result_t run_prepared_select(PreparedSelect&)
|
|
{
|
|
return {};
|
|
}
|
|
|
|
auto attach(std::string name) -> ::sqlpp::schema_t
|
|
{
|
|
return {name};
|
|
}
|
|
|
|
void start_transaction()
|
|
{
|
|
_mock_data._last_isolation_level = _mock_data._default_isolation_level;
|
|
}
|
|
|
|
void start_transaction(sqlpp::isolation_level level)
|
|
{
|
|
_mock_data._last_isolation_level = level;
|
|
}
|
|
|
|
void set_default_isolation_level(sqlpp::isolation_level level)
|
|
{
|
|
_mock_data._default_isolation_level = level;
|
|
}
|
|
|
|
sqlpp::isolation_level get_default_isolation_level()
|
|
{
|
|
return _mock_data._default_isolation_level;
|
|
}
|
|
|
|
void rollback_transaction(bool)
|
|
{
|
|
}
|
|
|
|
void commit_transaction()
|
|
{
|
|
}
|
|
|
|
void report_rollback_failure(std::string)
|
|
{
|
|
}
|
|
|
|
// temporary data store to verify the expected results were produced
|
|
InternalMockData _mock_data;
|
|
};
|
|
|
|
struct MockSizeDb : public sqlpp::connection
|
|
{
|
|
using _traits = MockDb::_traits;
|
|
|
|
using _serializer_context_t = MockDb::_serializer_context_t;
|
|
|
|
using _interpreter_context_t = _serializer_context_t;
|
|
|
|
_serializer_context_t get_serializer_context()
|
|
{
|
|
return {};
|
|
}
|
|
|
|
template <typename T>
|
|
static _serializer_context_t& _serialize_interpretable(const T& x, _serializer_context_t& context)
|
|
{
|
|
serialize(context, x);
|
|
return context;
|
|
}
|
|
|
|
template <typename T>
|
|
static _serializer_context_t& _interpret_interpretable(const T& x, _interpreter_context_t& context)
|
|
{
|
|
serialize(context, x);
|
|
return context;
|
|
}
|
|
|
|
class result_t : public MockDb::result_t
|
|
{
|
|
public:
|
|
size_t size() const
|
|
{
|
|
return 0;
|
|
}
|
|
};
|
|
|
|
// Directly executed statements start here
|
|
template <typename T>
|
|
auto _run(const T& t, ::sqlpp::consistent_t) -> decltype(t._run(*this))
|
|
{
|
|
return t._run(*this);
|
|
}
|
|
|
|
template <typename Check, typename T>
|
|
auto _run(const T& t, Check) -> Check;
|
|
|
|
template <typename T>
|
|
auto operator()(const T& t) -> decltype(this->_run(t, sqlpp::run_check_t<_serializer_context_t, T>{}))
|
|
{
|
|
return _run(t, sqlpp::run_check_t<_serializer_context_t, T>{});
|
|
}
|
|
|
|
size_t execute(const std::string&)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
template <
|
|
typename Statement,
|
|
typename Enable = typename std::enable_if<not std::is_convertible<Statement, std::string>::value, void>::type>
|
|
size_t execute(const Statement& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running execute call with\n" << context.str() << std::endl;
|
|
return execute(context.str());
|
|
}
|
|
|
|
template <typename Insert>
|
|
size_t insert(const Insert& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running insert call with\n" << context.str() << std::endl;
|
|
return 0;
|
|
}
|
|
|
|
template <typename Update>
|
|
size_t update(const Update& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running update call with\n" << context.str() << std::endl;
|
|
return 0;
|
|
}
|
|
|
|
template <typename Remove>
|
|
size_t remove(const Remove& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running remove call with\n" << context.str() << std::endl;
|
|
return 0;
|
|
}
|
|
|
|
template <typename Select>
|
|
result_t select(const Select& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running select call with\n" << context.str() << std::endl;
|
|
return {};
|
|
}
|
|
|
|
// Prepared statements start here
|
|
using _prepared_statement_t = std::nullptr_t;
|
|
|
|
template <typename T>
|
|
auto _prepare(const T& t, ::sqlpp::consistent_t) -> decltype(t._prepare(*this))
|
|
{
|
|
return t._prepare(*this);
|
|
}
|
|
|
|
template <typename Check, typename T>
|
|
auto _prepare(const T& t, Check) -> Check;
|
|
|
|
template <typename T>
|
|
auto prepare(const T& t) -> decltype(this->_prepare(t, sqlpp::prepare_check_t<_serializer_context_t, T>{}))
|
|
{
|
|
return _prepare(t, sqlpp::prepare_check_t<_serializer_context_t, T>{});
|
|
}
|
|
|
|
template <typename Statement>
|
|
_prepared_statement_t prepare_execute(Statement& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running prepare execute call with\n" << context.str() << std::endl;
|
|
return nullptr;
|
|
}
|
|
|
|
template <typename Insert>
|
|
_prepared_statement_t prepare_insert(Insert& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running prepare insert call with\n" << context.str() << std::endl;
|
|
return nullptr;
|
|
}
|
|
|
|
template <typename PreparedExecute>
|
|
size_t run_prepared_execute(const PreparedExecute&)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
template <typename PreparedInsert>
|
|
size_t run_prepared_insert(const PreparedInsert&)
|
|
{
|
|
return 0;
|
|
}
|
|
|
|
template <typename Select>
|
|
_prepared_statement_t prepare_select(Select& x)
|
|
{
|
|
_serializer_context_t context;
|
|
serialize(context, x);
|
|
std::cout << "Running prepare select call with\n" << context.str() << std::endl;
|
|
return nullptr;
|
|
}
|
|
|
|
template <typename PreparedSelect>
|
|
result_t run_prepared_select(PreparedSelect&)
|
|
{
|
|
return {};
|
|
}
|
|
|
|
auto attach(std::string name) -> ::sqlpp::schema_t
|
|
{
|
|
return {name};
|
|
}
|
|
|
|
void start_transaction()
|
|
{
|
|
_mock_data._last_isolation_level = _mock_data._default_isolation_level;
|
|
}
|
|
|
|
void start_transaction(sqlpp::isolation_level level)
|
|
{
|
|
_mock_data._last_isolation_level = level;
|
|
}
|
|
|
|
void set_default_isolation_level(sqlpp::isolation_level level)
|
|
{
|
|
_mock_data._default_isolation_level = level;
|
|
}
|
|
|
|
sqlpp::isolation_level get_default_isolation_level()
|
|
{
|
|
return _mock_data._default_isolation_level;
|
|
}
|
|
|
|
void rollback_transaction(bool)
|
|
{
|
|
}
|
|
|
|
void commit_transaction()
|
|
{
|
|
}
|
|
|
|
void report_rollback_failure(std::string)
|
|
{
|
|
}
|
|
|
|
// temporary data store to verify the expected results were produced
|
|
InternalMockData _mock_data;
|
|
};
|