#ifndef LOCKER_H
#define LOCKER_H

#include <exception>
#include <pthread.h>
#include <semaphore.h>

// 线程同步机制封装类

// 互斥锁类
class locker {
public:
    //构造函数
    locker() {
        if(pthread_mutex_init(&m_mutex, NULL) != 0) {
            throw std::exception();
        }
    }

    //析构函数
    ~locker() {
        pthread_mutex_destroy(&m_mutex);
    }

    //上锁
    bool lock() {
        return pthread_mutex_lock(&m_mutex) == 0;
    }

    //解锁
    bool unlock() {
        return pthread_mutex_unlock(&m_mutex) == 0;
    }

    //获取互斥量
    pthread_mutex_t *get()
    {
        return &m_mutex;
    }

private:
    pthread_mutex_t m_mutex;
};


// 条件变量类：判断队列中是否有数据，条件变量配合互斥锁来使用
class cond {
public:
    //构造函数
    cond(){
        if (pthread_cond_init(&m_cond, NULL) != 0) {
            throw std::exception();
        }
    }

    //析构函数
    ~cond() {
        pthread_cond_destroy(&m_cond);
    }

    //wait
    bool wait(pthread_mutex_t *m_mutex) {
        int ret = 0;
        ret = pthread_cond_wait(&m_cond, m_mutex);
        return ret == 0;
    }

    //带超时时间的wait
    bool timewait(pthread_mutex_t *m_mutex, struct timespec t) {
        int ret = 0;
        ret = pthread_cond_timedwait(&m_cond, m_mutex, &t);
        return ret == 0;
    }

    //让一个或者多个线程唤醒
    bool signal() {
        return pthread_cond_signal(&m_cond) == 0;
    }

    //将所有的线程都唤醒
    bool broadcast() {
        return pthread_cond_broadcast(&m_cond) == 0;
    }

private:
    pthread_cond_t m_cond;
};


// 信号量类
class sem {
public:
    //构造函数
    sem() {
        if( sem_init( &m_sem, 0, 0 ) != 0 ) {
            throw std::exception();
        }
    }

    //带参数的构造函数
    sem(int num) {
        if( sem_init( &m_sem, 0, num ) != 0 ) {
            throw std::exception();
        }
    }

    //析构函数
    ~sem() {
        sem_destroy( &m_sem );
    }

    // 等待信号量
    bool wait() {
        return sem_wait( &m_sem ) == 0;
    }

    // 增加信号量
    bool post() {
        return sem_post( &m_sem ) == 0;
    }
private:
    sem_t m_sem;
};

#endif