Linux C++多线程与数据库连接池的实现

蜗牛 互联网技术资讯 2024-11-17 4 0

在Linux环境下,使用C++实现多线程和数据库连接池需要考虑以下几个方面:

  1. 多线程的实现

在Linux环境下,可以使用C++11标准中的<thread>库来实现多线程。以下是一个简单的多线程示例:

#include <iostream>
#include <thread>
#include <vector>

void print_hello(int id) {
    std::cout << "Hello from thread " << id << std::endl;
}

int main() {
    const int num_threads = 5;
    std::vector<std::thread> threads;

    for (int i = 0; i < num_threads; ++i) {
        threads.emplace_back(print_hello, i);
    }

    for (auto& t : threads) {
        t.join();
    }

    return 0;
}
  1. 数据库连接池的实现

数据库连接池是一种管理数据库连接的技术,它可以复用已打开的数据库连接,减少创建和关闭连接的开销。以下是一个简单的数据库连接池实现示例:

#include <iostream>
#include <queue>
#include <mutex>
#include <condition_variable>
#include <memory>
#include <stdexcept>
#include <mysql/mysql.h>

class DatabaseConnectionPool {
public:
    DatabaseConnectionPool(const std::string& url, const std::string& user, const std::string& password, unsigned int max_connections)
        : url_(url), user_(user), password_(password), max_connections_(max_connections) {
        for (unsigned int i = 0; i < max_connections_; ++i) {
            connections_.emplace(create_connection());
        }
    }

    std::shared_ptr<MYSQL> acquire() {
        std::unique_lock<std::mutex> lock(mutex_);
        cond_.wait(lock, [this] { return !connections_.empty(); });
        auto connection = connections_.front();
        connections_.pop();
        return connection;
    }

    void release(std::shared_ptr<MYSQL> connection) {
        if (!connection) {
            throw std::invalid_argument("Invalid connection");
        }
        std::unique_lock<std::mutex> lock(mutex_);
        connections_.push(connection);
        cond_.notify_one();
    }

private:
    std::shared_ptr<MYSQL> create_connection() {
        MYSQL* conn = mysql_init(nullptr);
        if (!mysql_real_connect(conn, url_.c_str(), user_.c_str(), password_.c_str(), nullptr, 0, nullptr, 0)) {
            mysql_close(conn);
            throw std::runtime_error(mysql_error(conn));
        }
        return std::shared_ptr<MYSQL>(conn, [](MYSQL* conn) { mysql_close(conn); });
    }

    std::string url_;
    std::string user_;
    std::string password_;
    unsigned int max_connections_;
    std::queue<std::shared_ptr<MYSQL>> connections_;
    std::mutex mutex_;
    std::condition_variable cond_;
};

在这个示例中,我们使用了一个简单的队列来存储数据库连接,并使用互斥锁和条件变量来确保线程安全。当需要获取数据库连接时,我们从队列中取出一个连接;当释放数据库连接时,我们将连接放回队列中。

要使用这个数据库连接池,你可以在多线程程序中调用acquire()方法获取连接,执行数据库操作,然后调用release()方法释放连接。例如:

int main() {
    DatabaseConnectionPool pool("localhost", "user", "password", 5);

    std::vector<std::thread> threads;
    for (int i = 0; i < 10; ++i) {
        threads.emplace_back([&pool, i]() {
            auto conn = pool.acquire();
            // Perform database operations here
            pool.release(conn);
        });
    }

    for (auto& t : threads) {
        t.join();
    }

    return 0;
}

这个示例中,我们创建了一个包含10个线程的程序,每个线程都从数据库连接池中获取一个连接,执行一些数据库操作,然后释放连接。

免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:niceseo6@gmail.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。

评论

有免费节点资源,我们会通知你!加入纸飞机订阅群

×
天气预报查看日历分享网页手机扫码留言评论Telegram