mfc_Thread_2 - 8BitsCoding/RobotMentor GitHub Wiki

Thread를 여러개 만들어서 협력하며 쓰는 법에 대해 고민

우선 소수를 구하는 함수를 하나 생성해본다.

int IsPrime(int a_num)
{
    int i;
    for(i = 2; i < a_num; i++)
    {
        if(a_num % i == 0) return 0;
    }
    return 1;
}

2~200000 사이의 소수를 모두 합산하는 Thread를 생성해 보자.

DWORD WINAPI InsertItemToList(void * ap_data)
{
    ThreadData * p_data = (ThreadData *)ap_data;
    CString str;
    int index;
    __int64 sum = 0;

    int start_tick = GetTickCount();

    for(unsigned int i = 2; i < 200000; i++) {
        if(WaitForSingleObject(p_data->h_kill_event,0) == WAIT_OBJECT_0) break;

        if(IsPrime(i)) sum += i;

        if(!(i%100000)) {
            str.Format(L"작업 진행중 %d까지 체크함!");
            index = p_data->p_list_box->InsertString(-1, str);
            p_data->p_list_box_SetCurSel(index);
        }
    }

    str.Format(L"2~200000까지 소수의 합은 %lld이다. (%dms)", sum, GetTickCount() - start_tick);
    index = p_data->p_list_box->InsertString(-1, str);
    p_data->p_list_box_SetCurSel(index);

    CloseHandle(p_data->h_thread);
    p_data->h_thread = NULL;

    return 0;
}

---

버튼을 누를때마다 Thread를 만들게 변경해보자.

// Dlg.h
struct ThreadData
{
    HWND h_wnd;         // 대화상자의 Handle
    unsigned int step;  // 
    CListBox * p_list_box;
    HANDLE h_kill_event;
    HANDLE h_thread;
    DWORD thread_id;
};

// Dlg.cpp
private:
    unsigned int m_step = 200000;

// Thread 생성 버튼
ThreadData * p = new ThreadData;
p->h_wnd = m_hWnd;
p->step = m_stop;
p->p_list_box = &m_data_list;
p->h_kill_event = CreateEvent(NULL, 1, 0, NULL);
p->h_thread = CreateThread(NULL, 0, InsertItemToList, p, 0, &p->thread_id);

CString str;
str.Format(L"%08x : %u 까지 합산", p->thread_id, p->step);
int index = m_thread_list.InsertString(-1, str);
m_thread_list.SetItemDataPtr(index, p);

m_step *= m_step;

DWORD WINAPI InsertItemToList(void * ap_data)
{
    ThreadData * p_data = (ThreadData *)ap_data;
    CString str;
    int index;

    str.Format(L"[%08x] 작업을 시작합니다!", p_data->thread_id);
    p_data->p_list_box->SetCurSel(index);
    __int64 sum = 0, kill_flag = 0;
    int start_tick = GetTickCount();
    unsigned int i;

    for(i = 2; i < 200000; i++) {
        if(WaitForSingleObject(p_data->h_kill_event,0) == WAIT_OBJECT_0) {
            str.Format(L"[%08x] 작업 중단 체크함!", p_data->thread_id);
            index = p_data->p_list_box->InsertString(-1, str);
            kill_flag = 1;
            break;
        }

        if(IsPrime(i)) sum += i;

        if(!(i%100000)) {
            str.Format(L"[%08x] 작업 진행중 %d까지 체크함!", p_data->thread_id);
            index = p_data->p_list_box->InsertString(-1, str);
            p_data->p_list_box_SetCurSel(index);
        }
    }

    str.Format(L"[%08x] 2~%u까지 소수의 합은 %lld이다. (%dms)", p_data->thread_id, i, sum, GetTickCount() - start_tick);
    index = p_data->p_list_box->InsertString(-1, str);
    p_data->p_list_box_SetCurSel(index);

    str.Format(L"[%08x] 작업을 중단합니다!", p_data->thread_id);
    p_data->p_list_box->SetCurSel(index);

    CloseHandle(p_data->h_thread);
    
    if(WaitForSingleObject(p_data->h_kill_event, 0) == WAIT_OBJECT_0) kill_flag= 1;
    ::PostMessage(p_data->h_wnd, 27001, kill_flag, (LPARAM)p_data);

    return 0;
}

코드설명

(참고) LPARAM, WPARAM

LPARAM : 정통적으로 4 bytes 주소값을 보통 넘겨준다.

WPARAM : 최근에는 4 bytes 가 되었으나 예전에 2 bytes로 보통 주소값을 넣지 않고 char, byte등을 넣었음.

::PostMessage(p_data->h_wnd, 27001, kill_flag, (LPARAM)p_data); : 부모 다이얼로그(p_data->h_wnd)로 메시지(27001)를 보낸다. 메시지의 WPARAM = kill_flag(운용자가 죽인것인지 자연스럽게 죽은것인지), LPARAM = p_data의 주소값.

if(WaitForSingleObject(p_data->h_kill_event, 0) == WAIT_OBJECT_0) kill_flag= 1; : 마지막에 이 한 줄이 더 들어가는 이유는?? -> 200000번까지 도는데 199999까지 도는 순간에 운용자가 Thread종료를 누르면 kill_flag가 0인채로 끝나버린다. 그 상황을 방지하기 위해서.

// thread 선택 종료
int index = m_thread_list.GetCurSel();
if(LB_ERR != index)
{
    ThreadData* p = (ThreadData*)m_thread_list.GetItemDataPtr(index);
    if(p->h_thread != NULL) {
        SetEvent(p->h_kill_event);
        MSG msg;
        while(p->h_thread != NULL) {
            if(PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) {
                TranslateMessage(&msg);
                DispatchMessage(&msg);
            }
        }
    }
    delete p;
}
}
else {

}

코드설명

PeekMessage(&msg, NULL, 0, 0, PM_REMOVE) : 메시지가 있으면 동작 없으면 그냥 넘어감, RM_REMOVE : 메시지를 복사하지 않고 꺼내옴

기존 GetMessage()와의 차이점은 메시지가 있는지 없는지 확인한다는 점이다.

GetMessage를 쓰면 메시지를 무한정 기다리며 데드락에 걸릴 수 있다.

사용자 지정 메시지(27001) 추가

클래스 마법사(Shift+Ctrl+x) -> 사용자 지정 메시지 추가 -> 27001

// On27001
ThreadData * p = (ThreadData *)lParam;
int count = m_thread_list.GetCount();
for(int i = 0; i < count; i++) {
    if(m_thread_list.GetItemDataPtr(i) == p) {
        m_thread_list.DeleteString(i);
        CloseHandle(p->h_kill_event);

        if(wParam == 0) delete p;
        else p->thread = NULL;

        break;
    }
}

// OnDestroy()
OnBtnAllStop();

// 모든 Thread 종료 버튼 (OnBtnAllStop())
ThreadData * p;
int count = m_thread_list.GetCount();
for(int i = 0; i < count; i++){
    p = (ThreadData*) m_thread_list.GetItemDataPtr(i);
    SetEvent(p->h_kill_event);
}

CString str;
str.Format(L"Thread %d개를 종료합니다.", count);
int index = m_data_list.InsertString(-1, str);
m_data_list.SetCurSel(index);

MSG msg;
while(count) {
    if(PeekMessage(&msg, NULL, 0, 0, PM_REMOVE)) {
        if(msg.message == 27001) {
            count--;
            msg.wParam = 0;     // kill_flag를 1로 바꾸면 여기서 delete를 하지 않아도 된다.
        }
        TranslateMessage(&msg);
        DispatchMessage(&msg);
    }
}

index = m_data_list.InsertString(-1, L"모든 Thread가 종료되었습니다.!");
m_data_list.SetCurSel(index);