dcapp/datacenter/bizService/F_InsertRecord.cpp

/*
 * @Description: file content
 * @Author: why
 * @version: 
 * @Date: 2025-02-21 14:03:54
 * @LastEditors: why
 * @LastEditTime: 2025-02-28 17:55:37
 */
#include <iostream>
#include <chrono>
#include <iomanip>
#include <sstream>
#include <string>
#include <ctime>
#include "json.hpp"
#include "F_InsertRecord.h"
#include "bizservice.h"
#include "sqlite3.h"
#include "ToolUtils.h"

using namespace std;
using json = nlohmann::json ;// for convenience
typedef	unsigned char BYTE;

typedef struct topic{

    std::string src;
    std::string des;
    std::string sid;
    std::string fid;
} topic_type;

REGISTER_BIZSERVICE("F-InsertRecord",InsertRecord);

InsertRecord::InsertRecord(void){
    handpara = nullptr;
    initpara = nullptr;
}

InsertRecord::~InsertRecord(void){
}

void InsertRecord::Init(initparam_t *param){
    initpara = param;
}

int  InsertRecord::SQLBatch_exc(std::vector<std::string> &sql_list,std::vector<std::string> &result){
    sqlite3 *db = (sqlite3 *)handpara;
    char *errMsg = nullptr;
    const char *sql;
    int rc;
    int resp = 0;
    // 开启事务
    if (sqlite3_exec(db, "BEGIN TRANSACTION;", nullptr, nullptr, &errMsg) != SQLITE_OK) {
        std::cerr << "事务开启失败: " << errMsg << std::endl;
        sqlite3_free(errMsg);
        return -1;
    }
    // 执行批量插入操作
    sqlite3_stmt *stmt;
    int idx = 0;
    for(const auto &k:sql_list){
        std::cout << "cmd:"<<k.c_str()<<std::endl;
        if (sqlite3_prepare_v2(db, k.c_str(), -1, &stmt, nullptr) != SQLITE_OK) {
                std::cerr << "准备语句失败: " << sqlite3_errmsg(db) << std::endl;
                continue;
            }
        rc = sqlite3_step(stmt);
        if (rc != SQLITE_DONE) {
            std::cerr << "插入失败: " << sqlite3_errmsg(db) << std::endl;
            result[idx] = "-1";
        }else
            result[idx] = "0"; //SUCCESS
        sqlite3_reset(stmt); // 重置语句
    }
    // 提交事务
    if (sqlite3_exec(db, "COMMIT;", nullptr, nullptr, &errMsg) != SQLITE_OK) {
        std::cerr << "事务提交失败: " << errMsg << std::endl;
        sqlite3_free(errMsg);
        resp = -1;
    }
    // 清理
    sqlite3_finalize(stmt);
    return resp;
}
int InsertRecord::HandleMsg(const msg_t msg, msg_t &outMsg,handparam_t *param){
    
    topic.clear();
    topic = msg.subTopicV;
    json value,resp;
    int ret = 0;
    int afn_flag = 0;
    std::vector<std::string> sql_list;
    std::vector<std::string> result;
    sql_list.clear();
    result.clear();
    std::string table_name("dayfrozen");
    try{
        value = json::parse(msg.payload);
    }
    catch(const std::exception  &e){
        std::cout << "topic "<<msg.topic.c_str()<<" parse error:"<<e.what();
        return -1;//format fail
    }
    printf("insert request:%s \n",msg.payload.c_str());
    if(!value.is_object())return -1;
    if(value.contains("token")){
        resp["token"] = value["token"];
    }
    if(value.contains("time")){
       ;// value["time"]
    }
    if(value.contains("data") && value["data"].contains("dataarray")){
        json subvalue = value["data"]["dataarray"];      
         if(subvalue.is_array()){
            int idx = 0;
            for(const auto &k:subvalue){
                std::stringstream ss;
                if(k["afn"].get<std::string>() != "0D"){
                    afn_flag = -1;
                    continue;
                }
                //根据afn/idclass 确定数据库表位置 待添加
                ss << "insert into " <<table_name.c_str() <<" (idx,metid,dataid,frozentime,createdtime,source,crcaux,databuff) values ("\
                << 0 << ","\
                <<std::stoi(k["metid"].get<std::string>())<<","\
                <<k["dataid"].get<std::string>().c_str()<<","\
                <<ToolUtils::string_to_now(k["frozentime"].get<std::string>())<<","\
                <<time(NULL)<<","\
                <<std::stoi(k["crcaux"].get<std::string>())<<","\
                <<k["databuff"].get<std::string>().c_str()<<");";
                printf("insert sql:%s \n",ss.str().c_str());
                sql_list.push_back(ss.str());  
                ss.clear();       
            }
         }
    }
   // ret = SQLBatch_exc(sql_list,result);
    resp["time"]=ToolUtils::now_to_string();
    if(ret == -1 || afn_flag == -1)
        resp["ack"]="false";
    else if(ret == 0)
        resp["ack"]="true";

    resp["data"]["dataidex"]=result;

    outMsg.topic = ToolUtils::makeTpopicStr(topic);
    outMsg.timestamp = time(NULL);
    outMsg.payload = resp.dump();
    std::cout << "resp InsertRecord payload:"<<resp.dump().c_str()<<std::endl;
    return 1;
}
 std::string InsertRecord::GetFID(){
    std::string value("F-InsertRecord");
    return value;
}