基于C++文本文件的数据库读取

#include<iostream>
#include <fstream>
#include<string>
#include<vector>
#include<sqlite3.h>

int main() {
    // 打开文本文件
    std::ifstream infile("data.txt");
    if (!infile) {
        std::cerr << "无法打开文件\n";
        return 1;
    }

    // 创建数据库连接
    sqlite3* db;
    int rc = sqlite3_open("data.db", &db);
    if (rc) {
        std::cerr << "无法打开数据库: "<< sqlite3_errmsg(db) << "\n";
        return 1;
    }

    // 创建数据表
    const char* create_table_sql = "CREATE TABLE IF NOT EXISTS data (id INTEGER PRIMARY KEY, name TEXT, age INTEGER);";
    rc = sqlite3_exec(db, create_table_sql, nullptr, nullptr, nullptr);
    if (rc != SQLITE_OK) {
        std::cerr << "无法创建数据表: "<< sqlite3_errmsg(db) << "\n";
        return 1;
    }

    // 从文本文件中读取数据并插入到数据库中
    std::string line;
    while (std::getline(infile, line)) {
        // 解析每行数据
        std::vector<std::string> fields;
        std::istringstream iss(line);
        std::string field;
        while (std::getline(iss, field, ',')) {
            fields.push_back(field);
        }

        // 插入数据到数据库中
        const char* insert_sql = "INSERT INTO data (name, age) VALUES (?, ?);";
        sqlite3_stmt* stmt;
        rc = sqlite3_prepare_v2(db, insert_sql, -1, &stmt, nullptr);
        if (rc != SQLITE_OK) {
            std::cerr << "无法准备SQL语句: "<< sqlite3_errmsg(db) << "\n";
            return 1;
        }
        rc = sqlite3_bind_text(stmt, 1, fields[0].c_str(), -1, SQLITE_STATIC);
        if (rc != SQLITE_OK) {
            std::cerr << "无法绑定参数: "<< sqlite3_errmsg(db) << "\n";
            return 1;
        }
        rc = sqlite3_bind_int(stmt, 2, std::stoi(fields[1]));
        if (rc != SQLITE_OK) {
            std::cerr << "无法绑定参数: "<< sqlite3_errmsg(db) << "\n";
            return 1;
        }
        rc = sqlite3_step(stmt);
        if (rc != SQLITE_DONE) {
            std::cerr << "无法执行SQL语句: "<< sqlite3_errmsg(db) << "\n";
            return 1;
        }
        rc = sqlite3_finalize(stmt);
        if (rc != SQLITE_OK) {
            std::cerr << "无法释放SQL语句: "<< sqlite3_errmsg(db) << "\n";
            return 1;
        }
    }

    // 关闭数据库连接
    rc = sqlite3_close(db);
    if (rc != SQLITE_OK) {
        std::cerr << "无法关闭数据库: "<< sqlite3_errmsg(db) << "\n";
        return 1;
    }

    return 0;
}