Linux信号量（简易版）

荷兰的乳牛2024-03-16 16:25

Sem.hpp(用于封装信号量):

cpp 复制代码

#include<iostream>
#include<queue>
#include<unistd.h>
#include <semaphore.h>
using namespace std;
class Sem
{
public:
    Sem(int num)
    {
        sem_init(&_sem,0,num);
    }
    ~Sem()
	{
		sem_destroy(&_sem);
	}
    void V()
    {
        sem_post(&_sem);
    }
    void P()
    {
        sem_wait(&_sem);
    }
private:
    sem_t _sem;
};

Task.hpp(封装任务队列)：

cpp 复制代码

#include "sem.hpp"
#include <iostream>
#include <queue>
#include <pthread.h>
using namespace std;

int num = 10;

class Task
{
public:
    Task(): _space(num), _data(0)
    {
        pthread_mutex_init(&_mtx, nullptr);
    }

    void Push(int x)
    {
        _space.P();  // 生产者等待空间
        pthread_mutex_lock(&_mtx);
        q.push(x);
        cout << "生产了：" << x << endl;
        pthread_mutex_unlock(&_mtx);
        _data.V();   // 通知消费者数据已生产
    }

    void Pop()
    {
        _data.P();   // 消费者等待数据
        pthread_mutex_lock(&_mtx);
        cout << "消费了：" << q.front() << endl;
        q.pop();
        pthread_mutex_unlock(&_mtx);
        _space.V();  // 通知生产者空间已释放
    }

private:
    Sem _space;    // 空闲空间数量
    Sem _data;     // 数据数量
    pthread_mutex_t _mtx;
    queue<int> q;
};

.cc:

cpp 复制代码

#include"Task.hpp"
void* consumer(void* args)
{
    Task* t = (Task*)args;
    while(1)
    {
        sleep(1);
        t->Pop();
    }
    return nullptr;
}
void* producer(void* args)
{
    Task* t = (Task*)args;
    while(1)
    {
        t->Push(rand()%1000);
    }
    return nullptr;
}
int main()
{
    srand(time(nullptr));
    Task* t = new Task;
    pthread_t Consumer[5];
    pthread_t Producer[5];
    for(int i = 0; i < 5; i++)
    {
        pthread_create(&Consumer[i], nullptr, consumer, (void*)t);
        pthread_create(&Producer[i], nullptr, producer, (void*)t);
    }
    for(int i = 0; i < 3; i++)
    {
        pthread_join(Consumer[i], nullptr);
        pthread_join(Producer[i], nullptr);
    }
    delete t;
    return 0;
}