业务场景中使用 SQL 实现快速数据更新与插入

news2025/5/26 1:52:04

一、业务背景

在气象数据处理系统中，我们经常需要对分钟级的降水数据进行更新和插入操作。具体场景如下：

• 数据源会定期发送分钟级的降水数据，包括降水值（PRECA）和质控码（PRECA_QC2）。

• 如果系统中已存在对应时间点的数据，则需要更新降水值和质控码。

• 如果系统中不存在对应时间点的数据，则需要插入新的记录。

二、SQL 实现方法

为了高效地实现上述需求，我们采用INSERT INTO ... ON DUPLICATE KEY UPDATE语法。这种方法可以在尝试插入数据时，自动检测是否存在冲突（如主键或唯一键冲突），并在冲突发生时执行更新操作。

1.数据表结构

假设数据表名为pre_YPREC00_N01，其结构如下：

• pre_time：时间戳，主键或唯一键，格式为YYYY-MM-DD HH:MM:SS。

• PRECA：降水值，类型为浮点数。

• PRECA_QC2：质控码，类型为整数。

2.SQL 语句构造

插入并更新的 SQL 语句

INSERT INTO pre_YPREC00_N01 (pre_time, PRECA, PRECA_QC2)
VALUES
('2025-05-23 13:21:00', 0.1, 4),
('2025-05-23 13:22:00', 0.1, 4),
('2025-05-23 13:23:00', 0.1, 4),
('2025-05-23 13:24:00', 0.1, 4),
('2025-05-23 13:25:00', 0.1, 4),
('2025-05-23 13:26:00', 0.1, 4),
('2025-05-23 13:31:00', 0.1, 4),
('2025-05-23 13:32:00', 0.1, 4),
('2025-05-23 13:33:00', 0.1, 4),
('2025-05-23 13:34:00', 0.1, 4),
('2025-05-23 13:35:00', 0.1, 4),
('2025-05-23 13:36:00', 0.1, 4)
ON DUPLICATE KEY UPDATE
PRECA = VALUES(PRECA),
PRECA_QC2 = VALUES(PRECA_QC2);

3.代码实现

以下是使用 C++和 MySQL 的示例代码，展示如何动态生成上述 SQL 语句并执行：

#include <iostream>
#include <string>
#include <vector>
#include <mysql/mysql.h>

// 假设 request 是一个包含分钟级数据的对象
struct MinuteData {
    std::string minute; // 时间戳，格式为 YYYY-MM-DD HH:MM:SS
    std::string v;      // 降水值
    std::string qc;     // 质控码
};

struct Request {
    std::vector<MinuteData> minutelist;
};

void SetRainDataInDB(const Request& request, const std::string& strTableName) {
    int totalUpdates = request.minutelist.size();

    std::string strSqlBase = "INSERT INTO pre_" + strTableName + " (pre_time, PRECA, PRECA_QC2) VALUES ";
    std::string strSqlValues;

    for (size_t i = 0; i < totalUpdates; i++) {
        if (request.minutelist[i].v.empty()) {
            continue;
        }
        std::cout << "打印 订正的 [" << request.minutelist[i].minute << "] 分钟的 降水数据 [" << request.minutelist[i].v << "] 质控值 是[" << request.minutelist[i].qc << "]" << std::endl;

        strSqlValues += "('" + request.minutelist[i].minute + "', " + request.minutelist[i].v + ", 4), ";
    }

    // 去掉最后一个逗号
    if (!strSqlValues.empty()) {
        strSqlValues.pop_back();
        strSqlValues.pop_back(); // 去掉多余的逗号和空格
    }

    std::string strSqlOnDuplicateKeyUpdate = " ON DUPLICATE KEY UPDATE PRECA = VALUES(PRECA), PRECA_QC2 = VALUES(PRECA_QC2)";
    std::string strFullSQL = strSqlBase + strSqlValues + strSqlOnDuplicateKeyUpdate;

    std::cout << "SetRainDataInDB SQL: [" << strFullSQL << "]" << std::endl;

    // 连接数据库并执行 SQL 语句
    MYSQL* conn = mysql_init(NULL);
    if (mysql_real_connect(conn, "localhost", "username", "password", "database_name", 0, NULL, 0)) {
        if (mysql_query(conn, strFullSQL.c_str())) {
            std::cerr << "SQL 执行失败: " << mysql_error(conn) << std::endl;
        } else {
            std::cout << "SQL 执行成功" << std::endl;
        }
        mysql_close(conn);
    } else {
        std::cerr << "数据库连接失败: " << mysql_error(conn) << std::endl;
    }
}

int main() {
    Request request;
    request.minutelist.push_back({"2025-05-23 13:21:00", "0.1", "4"});
    request.minutelist.push_back({"2025-05-23 13:22:00", "0.1", "4"});
    request.minutelist.push_back({"2025-05-23 13:23:00", "0.1", "4"});
    // 添加更多分钟级数据...

    SetRainDataInDB(request, "YPREC00_N01");

    return 0;
}