1
#include "mysql_server.h"
2
using namespace std;
3

4
// 连接数据库
5
void MySQLServer::Init_sql()
6
{
7
    //初始化数据库连接池
8
    m_connPool = connection_pool::GetInstance();
9
    m_connPool->init("localhost", m_user, m_passWord, m_databaseName, 3306, m_sql_num, 0);
10
    //先从连接池中取一个连接
11
    MYSQL *mysqll = NULL;
12
    connectionRAII mysqlcon(&mysqll, m_connPool);
13
    transaction_sql = mysqll;
14
}
15

16
// 开始事务
17
bool MySQLServer::beginTransaction() {
18
    if (mysql_query(transaction_sql, "START TRANSACTION") != 0) {
19
        std::cerr << "Failed to start transaction: " << mysql_error(transaction_sql) << std::endl;
20
        return false;
21
    }
22
    return true;
23
}
24

25
// 提交事务
26
bool MySQLServer::commitTransaction() {
27
    if (mysql_query(transaction_sql, "COMMIT") != 0) {
28
        std::cerr << "Failed to commit transaction: " << mysql_error(transaction_sql) << std::endl;
29
        return false;
30
    }
31
    return true;
32
}
33

34
// 回滚事务
35
bool MySQLServer::rollbackTransaction() {
36
    if (mysql_query(transaction_sql, "ROLLBACK") != 0) {
37
        std::cerr << "Failed to rollback transaction: " << mysql_error(transaction_sql) << std::endl;
38
        return false;
39
    }
40
    return true;
41
}
42

43
// 禁用自动提交模式
44
bool MySQLServer::disableAutoCommit() {
45
    if (mysql_autocommit(transaction_sql, 0) != 0) {
46
        std::cerr << "Failed to disable autocommit: " << mysql_error(transaction_sql) << std::endl;
47
        return false;
48
    }
49
    return true;
50
}
51

52
// 启用自动提交模式
53
bool MySQLServer::enableAutoCommit() {
54
    if (mysql_autocommit(transaction_sql, 1) != 0) {
55
        std::cerr << "Failed to enable autocommit: " << mysql_error(transaction_sql) << std::endl;
56
        return false;
57
    }
58
        return true;
59
}

1
try{
2
    // 开启事务
3
    mysql_server.beginTransaction();
4

5
    // 多个sql语句执行
6
    // 新增stream_bind_model关联1
7
    if(!mysql_server.insert_stream_bind_model1(stream_id, insert_bind_model_id)){
8
        throw std::runtime_error("setbindAlgorithm: Failed to insert_stream_bind_model1");
9
    }
10
    // 新增stream_bind_model关联2
11
    if(!mysql_server.insert_stream_bind_model2(stream_id, insert_bind_model_id)){
12
        throw std::runtime_error("setbindAlgorithm: Failed to insert_stream_bind_model2");
13
    }
14

15
    mysql_server.commitTransaction();
16
} catch (const std::exception& ex) {
17
    // 如果任意更新操作失败,回滚事务
18
    mysql_server.rollbackTransaction();
19
    cout<<"setbindAlgorithm:"<< std::string(ex.what())<<endl;
20
}