c++线程同步之信号量

// MutexExDlg.h : 头文件
//

#pragma once

// CMutexExDlg 对话框
class CMutexExDlg : public CDialogEx
{
// 构造
public:
    CMutexExDlg(CWnd* pParent = NULL);    // 标准构造函数

// 对话框数据
    enum { IDD = IDD_MUTEXEX_DIALOG };

    protected:
    virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV 支持
    static HANDLE hThread[];
    static HANDLE m_Semaphore;
    static DWORD WINAPI ThreadProc0(LPVOID lpParameter);
    static DWORD WINAPI ThreadProc1(LPVOID lpParameter);
    static DWORD WINAPI ThreadProc2(LPVOID lpParameter);
    static DWORD WINAPI ThreadProc3(LPVOID lpParameter);

// 实现
protected:
    HICON m_hIcon;

    // 生成的消息映射函数
    virtual BOOL OnInitDialog();
    afx_msg void OnPaint();
    afx_msg HCURSOR OnQueryDragIcon();
    DECLARE_MESSAGE_MAP()
public:
    virtual BOOL PreTranslateMessage(MSG* pMsg);
    afx_msg void OnBnClickedButton1();
    static int m_Edit0;
    static int m_Edit1;
    static int m_Edit2;
    static int m_Edit3;
};

// MutexExDlg.cpp : 实现文件
//

#include "stdafx.h"
#include "MutexEx.h"
#include "MutexExDlg.h"
#include "afxdialogex.h"

#ifdef _DEBUG
#define new DEBUG_NEW
#endif

// CMutexExDlg 对话框

CMutexExDlg::CMutexExDlg(CWnd* pParent /*=NULL*/)
    : CDialogEx(CMutexExDlg::IDD, pParent)
{
    m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
}

void CMutexExDlg::DoDataExchange(CDataExchange* pDX)
{
    CDialogEx::DoDataExchange(pDX);
    DDX_Text(pDX, IDC_EDIT1, m_Edit0);
    DDV_MinMaxInt(pDX, m_Edit0, , );
    DDX_Text(pDX, IDC_EDIT2, m_Edit1);
    DDV_MinMaxInt(pDX, m_Edit1, , );
    DDX_Text(pDX, IDC_EDIT3, m_Edit2);
    DDV_MinMaxInt(pDX, m_Edit2, , );
    DDX_Text(pDX, IDC_EDIT4, m_Edit3);
    DDV_MinMaxInt(pDX, m_Edit3, , );
}

BEGIN_MESSAGE_MAP(CMutexExDlg, CDialogEx)
    ON_WM_PAINT()
    ON_WM_QUERYDRAGICON()
    ON_BN_CLICKED(IDC_BUTTON1, &CMutexExDlg::OnBnClickedButton1)
END_MESSAGE_MAP()

// CMutexExDlg 消息处理程序

BOOL CMutexExDlg::OnInitDialog()
{
    CDialogEx::OnInitDialog();

    // 设置此对话框的图标。  当应用程序主窗口不是对话框时，框架将自动
    //  执行此操作
    SetIcon(m_hIcon, TRUE);            // 设置大图标
    SetIcon(m_hIcon, FALSE);        // 设置小图标

    // TODO:  在此添加额外的初始化代码

    return TRUE;  // 除非将焦点设置到控件，否则返回 TRUE
}

// 如果向对话框添加最小化按钮，则需要下面的代码
//  来绘制该图标。  对于使用文档/视图模型的 MFC 应用程序，
//  这将由框架自动完成。

void CMutexExDlg::OnPaint()
{
    if (IsIconic())
    {
        CPaintDC dc(this); // 用于绘制的设备上下文

        SendMessage(WM_ICONERASEBKGND, reinterpret_cast<WPARAM>(dc.GetSafeHdc()), );

        // 使图标在工作区矩形中居中
        int cxIcon = GetSystemMetrics(SM_CXICON);
        int cyIcon = GetSystemMetrics(SM_CYICON);
        CRect rect;
        GetClientRect(&rect);
        int x = (rect.Width() - cxIcon + ) / ;
        int y = (rect.Height() - cyIcon + ) / ;

        // 绘制图标
        dc.DrawIcon(x, y, m_hIcon);
    }
    else
    {
        CDialogEx::OnPaint();
    }
}

//当用户拖动最小化窗口时系统调用此函数取得光标
//显示。
HCURSOR CMutexExDlg::OnQueryDragIcon()
{
    return static_cast<HCURSOR>(m_hIcon);
}

// 重写虚函数 PreTranslateMessage 屏蔽掉Esc键和Enter键
BOOL CMutexExDlg::PreTranslateMessage(MSG* pMsg)
{
    if (pMsg->message == WM_KEYDOWN)
    {
        int keyCode = (int)pMsg->wParam;
        if (keyCode == VK_ESCAPE || keyCode == VK_RETURN)
        {
            return TRUE;
        }
    }
    return CDialogEx::PreTranslateMessage(pMsg);
}

HANDLE CMutexExDlg::m_Semaphore = NULL;
int CMutexExDlg::m_Edit0 = ;
int CMutexExDlg::m_Edit1 = ;
int CMutexExDlg::m_Edit2 = ;
int CMutexExDlg::m_Edit3 = ;
HANDLE CMutexExDlg::hThread[] = { NULL };

// 抢红包按钮点击事件处理函数
void CMutexExDlg::OnBnClickedButton1()
{
    // 获取编辑框内容到str变量
    CString str;
    GetDlgItem(IDC_EDIT1)->GetWindowText(str);
    // CString 转 int
    m_Edit0 = _ttoi(str);
    m_Edit1 = ;
    m_Edit2 = ;
    m_Edit3 = ;
    // 这里需要创建一个线程 因为 WaitForMultipleObjects 会阻塞住 另外把this指针作为参数传递给线程，用于子线程更新编辑框内容
    HANDLE hThread0 = ::CreateThread(NULL, NULL, ThreadProc0, this, NULL, NULL);
    CloseHandle(hThread0);
}

DWORD WINAPI CMutexExDlg::ThreadProc0(LPVOID lpParameter)
{
    CMutexExDlg* pDlg = (CMutexExDlg*)lpParameter;
    // 创建一个信号量
    m_Semaphore = ::CreateSemaphore(NULL,
        , // 表示初始资源数量，0时不发送信号
        , // 表示最大并发数量 lInitialCount < lMaximumCount
        NULL);
    // 创建三个线程抢红包
    hThread[] = ::CreateThread(NULL, NULL, ThreadProc1, lpParameter, NULL, NULL);
    hThread[] = ::CreateThread(NULL, NULL, ThreadProc2, lpParameter, NULL, NULL);
    hThread[] = ::CreateThread(NULL, NULL, ThreadProc3, lpParameter, NULL, NULL);
    // 设置编辑框内容
    ::SetWindowText(::GetDlgItem(pDlg->m_hWnd, IDC_EDIT1), L"");
    // 将对象设置为已通知状态
    ReleaseSemaphore(m_Semaphore,
        , // 表示增加个数，必须大于0且不超过最大资源数量
        NULL);
    // 使用WaitForMultipleObjects监听所有线程，当线程全部结束后，调用CloseHandle关闭句柄.
    WaitForMultipleObjects(, hThread, TRUE, -);
    ::CloseHandle(hThread[]);
    ::CloseHandle(hThread[]);
    ::CloseHandle(hThread[]);
    ::CloseHandle(m_Semaphore);
    return ;
}

// 线程回调函数1
DWORD WINAPI CMutexExDlg::ThreadProc1(LPVOID lpParameter)
{
    CMutexExDlg* pDlg = (CMutexExDlg*)lpParameter;
    TCHAR szBuffer[];
    DWORD dwTimer = ;
    WaitForSingleObject(m_Semaphore, INFINITE);
    while (dwTimer < )
    {
        Sleep();
        memset(szBuffer, , );
        ::GetWindowText(::GetDlgItem(pDlg->m_hWnd, IDC_EDIT2), szBuffer, );
        swscanf(szBuffer, L"%d", &dwTimer);
        dwTimer++;
        memset(szBuffer, , );
        swprintf(szBuffer, L"%d", dwTimer);
        ::SetWindowText(::GetDlgItem(pDlg->m_hWnd, IDC_EDIT2), szBuffer);
    }
    // 递增信号量的当前资源计数
    ReleaseSemaphore(m_Semaphore, , NULL);
    return ;
}

// 线程回调函数2
DWORD WINAPI CMutexExDlg::ThreadProc2(LPVOID lpParameter)
{
    CMutexExDlg* pDlg = (CMutexExDlg*)lpParameter;
    TCHAR szBuffer[];
    DWORD dwTimer = ;
    WaitForSingleObject(m_Semaphore, INFINITE);
    while (dwTimer < )
    {
        Sleep();
        memset(szBuffer, , );
        ::GetWindowText(::GetDlgItem(pDlg->m_hWnd, IDC_EDIT3), szBuffer, );
        // 将TCHAR* 转换为 int
        swscanf(szBuffer, L"%d", &dwTimer);
        dwTimer++;
        memset(szBuffer, , );
        // int 转换为TCHAR*
        swprintf(szBuffer, L"%d", dwTimer);
        ::SetWindowText(::GetDlgItem(pDlg->m_hWnd, IDC_EDIT3), szBuffer);
    }
    // 递增信号量的当前资源计数
    ReleaseSemaphore(m_Semaphore, , NULL);
    return ;
}

// 线程回调函数3
DWORD WINAPI CMutexExDlg::ThreadProc3(LPVOID lpParameter)
{
    CMutexExDlg* pDlg = (CMutexExDlg*)lpParameter;
    TCHAR szBuffer[];
    DWORD dwTimer = ;
    WaitForSingleObject(m_Semaphore, INFINITE);
    while (dwTimer < )
    {
        Sleep();
        memset(szBuffer, , );
        ::GetWindowText(::GetDlgItem(pDlg->m_hWnd, IDC_EDIT4), szBuffer, );
        swscanf(szBuffer, L"%d", &dwTimer);
        dwTimer++;
        memset(szBuffer, , );
        swprintf(szBuffer, L"%d", dwTimer);
        ::SetWindowText(::GetDlgItem(pDlg->m_hWnd, IDC_EDIT4), szBuffer);
    }
    // 递增信号量的当前资源计数
    ReleaseSemaphore(m_Semaphore, , NULL);
    return ;
}