test/drogon
1
0
Fork 0
mirror of https://gitee.com/an-tao/drogon.git synced 2024-11-30 02:37:57 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Add delete methods to Mapper

Browse Source
This commit is contained in:
antao 2018-11-10 22:37:01 +08:00
parent 979c6670c9
commit 11db6673a6
1 changed files with 79 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
orm_lib/inc/drogon/orm/Mapper.h
View File

@ -85,6 +85,12 @@ class Mapper
const ExceptionCallback &ecb) noexcept;
std::future<size_t> updateFuture(const T &obj) noexcept;
size_t deleteOne(const T &obj) noexcept(false);
void deleteOne(const T &obj,
const CountCallback &rcb,
const ExceptionCallback &ecb) noexcept;
std::future<size_t> deleteFutureOne(const T &obj) noexcept;
private:
DbClient &_client;
std::string _limitString;
@ -750,6 +756,79 @@ inline std::future<size_t> Mapper<T>::updateFuture(const T &obj) noexcept
binder.exec();
return prom->get_future();
}
template <typename T>
inline size_t Mapper<T>::deleteOne(const T &obj) noexcept(false)
{
clear();
static_assert(!std::is_same<typename T::PrimaryKeyType, void>::value, "No primary key in the table!");
std::string sql = "delete from ";
sql += T::tableName;
sql += " "; //Replace the last ','
makePrimaryKeyCriteria(sql);
sql = _client.replaceSqlPlaceHolder(sql, "$?");
Result r(nullptr);
{
auto binder = _client << sql;
outputPrimeryKeyToBinder(obj.getPrimaryKey(), binder);
binder << Mode::Blocking;
binder >> [&r](const Result &result) {
r = result;
};
binder.exec(); //Maybe throw exception;
}
return r.affectedRows();
}
template <typename T>
inline void Mapper<T>::deleteOne(const T &obj,
const CountCallback &rcb,
const ExceptionCallback &ecb) noexcept
{
clear();
static_assert(!std::is_same<typename T::PrimaryKeyType, void>::value, "No primary key in the table!");
std::string sql = "delete from ";
sql += T::tableName;
sql += " ";
makePrimaryKeyCriteria(sql);
sql = _client.replaceSqlPlaceHolder(sql, "$?");
auto binder = _client << sql;
outputPrimeryKeyToBinder(obj.getPrimaryKey(), binder);
binder >> [=](const Result &r) {
rcb(r.affectedRows());
};
binder >> ecb;
}
template <typename T>
inline std::future<size_t> Mapper<T>::deleteFutureOne(const T &obj) noexcept
{
clear();
static_assert(!std::is_same<typename T::PrimaryKeyType, void>::value, "No primary key in the table!");
std::string sql = "delete from ";
sql += T::tableName;
sql += " ";
makePrimaryKeyCriteria(sql);
sql = _client.replaceSqlPlaceHolder(sql, "$?");
auto binder = _client << sql;
outputPrimeryKeyToBinder(obj.getPrimaryKey(), binder);
std::shared_ptr<std::promise<size_t>> prom = std::make_shared<std::promise<size_t>>();
binder >> [=](const Result &r) {
prom->set_value(r.affectedRows());
};
binder >> [=](const std::exception_ptr &e) {
prom->set_exception(e);
};
binder.exec();
return prom->get_future();
}
template <typename T>
inline Mapper<T> &Mapper<T>::limit(size_t limit)
{