首页
/ SQLPP11 中实现 MySQL 的 INSERT ON DUPLICATE KEY UPDATE 功能

SQLPP11 中实现 MySQL 的 INSERT ON DUPLICATE KEY UPDATE 功能

2025-06-30 12:53:22作者:董宙帆

在 SQLPP11 这个 C++ SQL 查询构建库中,实现 MySQL 特有的 INSERT ON DUPLICATE KEY UPDATE 功能需要一些特殊处理。这个功能允许我们在插入数据时,如果遇到主键或唯一键冲突,则执行更新操作而不是抛出错误。

基本实现方式

SQLPP11 提供了 custom_query 功能来实现这种特殊语法。最直接的方式是构建原生 SQL 字符串:

auto custom_query = sqlpp::custom_query(
    "INSERT INTO articles (id, title, content) VALUES ("
    + std::to_string(item.id) + ", "
    + db.escape(item.title) + ", "
    + db.escape(item.content) +
    ") ON DUPLICATE KEY UPDATE title = VALUES(title), content = VALUES(content)");

这里需要注意使用 db.escape() 方法来转义字符串值,而不是直接拼接,以防止 SQL 注入攻击。

更优雅的类型安全实现

SQLPP11 提供了更类型安全的方式来实现这一功能。我们可以定义一个辅助结构体:

namespace {
    struct on_duplicate_key_update {
        std::string _serialized;

        template <typename Db, typename Assignment>
        on_duplicate_key_update(Db& db, Assignment assignment) {
            typename Db::_serializer_context_t context(db);
            _serialized = " ON DUPLICATE KEY UPDATE " + serialize(assignment, context).str();
        }

        template <typename Db, typename Assignment>
        auto operator()(Db& db, Assignment assignment) -> on_duplicate_key_update& {
            typename Db::_serializer_context_t context(db);
            _serialized += ", " + serialize(assignment, context).str();
            return *this;
        }

        auto get() const -> sqlpp::verbatim_t<::sqlpp::no_value_t> {
            return ::sqlpp::verbatim(_serialized);
        }
    };
}

然后可以这样使用:

auto custom_query = custom_query(
    sqlpp::insert_into(tbl).set(tbl.title = "sample", tbl.content = "content", tbl.id = 100),
    on_duplicate_key_update(db, tbl.title = "sample")(db, tbl.content = "content").get());

这种方式利用了 SQLPP11 的类型系统,提供了编译时检查,比直接拼接 SQL 字符串更安全可靠。

注意事项

  1. 上下文对象构造时需要传入数据库连接对象,不能使用默认构造函数
  2. 对于 MySQL 连接,需要使用 db.escape() 而不是 sqlpp::mysql::quote()
  3. 这种方式保持了 SQLPP11 的类型安全特性,同时支持 MySQL 特有的语法

通过这种方式,我们可以在保持 SQLPP11 类型安全特性的同时,充分利用 MySQL 特有的功能,实现高效的数据插入和更新操作。

登录后查看全文
热门项目推荐
相关项目推荐