Работа с двумя потоками
11.05.2015, 00:55. Показов 1769. Ответов 1
Задаются две маски файлов. Первая маска - это маска директорий, вторая - это маска файлов. Также задается начальный каталог работы. После запуска происходит работа двумя потоками. Первый поток производит поиск с начального каталога работы директорий, удовлетворяющих первой маске. Если она находит такие директории, то их имена она передает второму потоку, который ищет в них файлы, удовлетворяющие второй маске.
Весь день сижу не могу понять как это сделать. Вот мои наработки, можно использовать для решения этой задачи.
| C++ | 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
| #include <Windows.h>
#include <process.h>
#include "resource.h"
#include <iostream>
#include <string>
#include <fstream>
#define IDM_CIRCLES 1
#define IDM_BARS 2
#define IDM_STOP 3
#define IDM_BEGIN 4
using namespace std;
//12. Задаются две маски файлов.Первая маска - это маска директорий, вторая - это маска файлов.
//Также задается начальный каталог работы.После запуска происходит работа двумя потоками.
//Первый поток производит поиск с начального каталога работы директорий, удовлетворяющих первой маске.
//Если она находит такие директории, то их имена она передает второму потоку, который ищет в них файлы, удовлетворяющие второй маске.
HWND hList;
WIN32_FIND_DATAA fd;
string mask;
string folders;
HANDLE hThread[2];
DWORD dwThread[2];
struct PARAM
{
BOOL type; // тип выполняемых потоком действий
HWND wnd; // окно, в оконной процедуре которого создается поток
BOOL stop; // признак завершения работы потока
HANDLE event; // объект-событие, поток работает, если оно установлено
};
// --- Описание функции главного окна
LRESULT CALLBACK WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam);
BOOL CALLBACK DlgProc(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam);
DWORD WINAPI FolderSearch(LPVOID param);
// --- Глобальные переменные
HINSTANCE hInst; // Дескриптор экземпляра приложения
char ClassName[] = "Window"; // Название класса окна
char AppTitle[] = "Application Win32"; // Заголовок главного окна
// --- Функция WinMain
int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
{
HWND hWndMain;
MSG msg;
hInst = hInstance;
//Создаем диалоговое окно
DialogBox(hInst, MAKEINTRESOURCE(IDD_DIALOG1), NULL, DlgProc);
while (GetMessage(&msg, NULL, 0, 0))
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
return msg.wParam;
}
void ThreadFun(void *lpvoid)
{
struct PARAM *p = (struct PARAM *)lpvoid;
while (!p->stop)
{
WaitForSingleObject(p->event, INFINITE);
}
_endthread(); // полная остановка потока
}
DWORD WINAPI FolderSearch(LPVOID param)
{
int dir = 0;
if (fd.dwFileAttributes != FILE_ATTRIBUTE_REPARSE_POINT)
{
if (fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)
{
printf("%s: %s\n", "Folder", fd.cFileName);
dir++;
}
}
return dir;
}
// --- Функция окна
LRESULT CALLBACK WndProc(HWND hWnd, UINT msg, WPARAM wParam, LPARAM lParam)
{
static struct PARAM p; // используется как общая память потоков
switch (msg)
{
case WM_COMMAND: // обработка сообщений от меню
{
switch (LOWORD(wParam))
{
case IDC_BUTTON_SEARCH: // выполнение потока
{ // сделать события установленным
hThread[0] = CreateThread(NULL, 0, (LPTHREAD_START_ROUTINE)FolderSearch,
(LPVOID)folders.c_str(), 0, &dwThread[0]);
if (hThread[0] == NULL)
return GetLastError();
hThread[1] = CreateThread(NULL, 1, (LPTHREAD_START_ROUTINE)ThreadFun,
NULL, 0, &dwThread[0]);
if (hThread[1] == NULL)
return GetLastError();
// запускаем второй поток
}; break;
}
}; return 0;
case WM_DESTROY:
{ // записать признак завершения потока
p.stop = 1;
PostQuitMessage(0);
}; return 0l;
// обработка других сообщений
default: return DefWindowProc(hWnd, msg, wParam, lParam);
}
return 0l;
}
BOOL CALLBACK DlgProc(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam)
{
//Собственно сама папка
switch (msg)
{
case WM_CLOSE:
{
PostQuitMessage(0);
}; break;
}
return FALSE;
} |
|
Вот второй кусок кода
| C++ | 1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
| #include <Windows.h>
#include "resource.h"
#include <string>
#include <CommCtrl.h>
BOOL CALLBACK DlgProc(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam);
DWORD WINAPI FindFolder(PVOID p); // функция - поток для поиска по маске папки
DWORD WINAPI FindFile(PVOID p); // функция - поток для поиска по маске файлов
struct Property
{
WCHAR path[255]; // Путь к файлу
UINT result; // Результат в байтах
UINT countFile; // Количество файлов в каталоге
HWND hDlg; // Дискриптор диалогового окна
HANDLE event; // Событие
WIN32_FIND_DATA *wfd; // Информации о файлах
HANDLE hThread1, hThread2; // дискрипторы потока, как бы не логично иметь их тут в структуре проперти каталога,
//но меня бесит когда много глобальных переменных
DWORD dwThread1, dwThread2;
};
HINSTANCE hInst;
int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)
{
HWND hWndMain;
MSG msg;
hInst = hInstance;
//Создаем диалоговое окно
DialogBox(hInst, MAKEINTRESOURCE(IDD_DIALOG1), NULL, DlgProc);
while (GetMessage(&msg, NULL, 0, 0))
{
TranslateMessage(&msg);
DispatchMessage(&msg);
}
return msg.wParam;
}
BOOL CALLBACK DlgProc(HWND hDlg, UINT msg, WPARAM wParam, LPARAM lParam){
//Собственно сама папка
static Property folder;
switch (msg){
case WM_INITDIALOG:{
folder.hDlg = hDlg; // Запоминаем дискриптор диалогового окна, чтобы посылать в него сообщения из потока
folder.result = 0;
}; break;
case WM_COMMAND:{
switch (LOWORD(wParam)){
case IDC_BUTTON_INPUT: GetDlgItemText(hDlg, IDC_EDIT_PATH, (LPSTR)folder.path, 255); break; //Считываем путь к каталогу из едита
case IDC_BUTTON_START: folder.hThread1 = CreateThread(NULL, 0, FindFolder, &folder, 0, &folder.dwThread1); break;
case IDC_BUTTON_STOP: {
// Фунция приостановки потока
SuspendThread(folder.hThread1);
SuspendThread(folder.hThread2);
}; break;
case IDC_BUTTON_REMOVE: {
// Функция возобновление потока
ResumeThread(folder.hThread1);
ResumeThread(folder.hThread2);
}; break;
case IDC_BUTTON_EXIT: {
// Функция завершения потока
TerminateThread(folder.hThread1, folder.dwThread1);
TerminateThread(folder.hThread2, folder.dwThread2);
}; break;
}
}; break;
case WM_CLOSE:{
PostQuitMessage(0);
}; break;
}
return FALSE;
}
// в поток всегда передает ОДИН АРГУМЕНТ, он делается указателем на воид, чтобы можно было все что угодно передать
DWORD WINAPI FindFolder(PVOID p){
//Преобразуем войдовский указатель в проперти
Property *tmp = (Property*)p;
tmp->result = 0;
WIN32_FIND_DATA tmpWfd;
// Про FindFirstFile и FindNextFile почитай в методички, там легко, просто писать в комменты много
HANDLE hFind = FindFirstFile(LPSTR(tmp->path), &tmpWfd);
//Если файл найден
if (hFind != INVALID_HANDLE_VALUE){
UINT i = 0;
// Находим количество файлов в каталоге
do{
i++;
} while (FindNextFile(hFind, &tmpWfd) != NULL);
FindClose(hFind);
//Выделяем память под количество файлов, тут будет хранить инфу о найденных файлах
tmp->wfd = new WIN32_FIND_DATA[i];
// Инициализируем прогресс бар, устанавливаем его длину, т.е количество итераций для полного его заполнения (Количество файлов умножить на 2)
// умножить на 2, так как, сначала мы типо их ищим, это половина прогресс бара, а потом еще по ним бежим и считаем размер, это еще половина прогрес бара
SendMessage(GetDlgItem(tmp->hDlg, IDC_PROGRESS1), PBM_SETRANGE, 0, MAKELPARAM(0, i * 2));
// Шаг заполнения прогресс бара
SendMessage(GetDlgItem(tmp->hDlg, IDC_PROGRESS1), PBM_SETSTEP, 1, 0);
SendMessage(GetDlgItem(tmp->hDlg, IDC_PROGRESS1), PBM_SETPOS, 0, 0);
//Поехали заполнять структуру
i = 0;
hFind = FindFirstFile(LPSTR(tmp->path), &tmp->wfd[i]);
do{
i++;
// увеличиваем заполненность прогресс бара
SendMessage(GetDlgItem(tmp->hDlg, IDC_PROGRESS1), PBM_STEPIT, 0, 0);
// Слип, просто чтобы медленнее выполнялось все
Sleep(100);
} while (FindNextFile(hFind, &tmp->wfd[i]) != NULL);
FindClose(hFind);
// Запоминаем количество файлов в структуре
tmp->countFile = i;
// Создаем событие установленное в сброшенное
tmp->event = CreateEvent(NULL, 0, 0, NULL);
// Создаем второй поток, который будет считать нам размер файлов
tmp->hThread2 = CreateThread(NULL, 0, GetFolderSize, tmp, 0, &tmp->dwThread2);
// А теперь начинаем ЖДАТЬ, пока событие не перейден в установленное состояние, в результате работы второго потока
DWORD rc = WaitForSingleObject(tmp->event, INFINITE);
// Если мы дождались установление события, то
if (rc == WAIT_OBJECT_0)
{
//Формируем красивую строку для вывода
std::string output;
output += "Размер ";
output += (LPSTR)tmp->path;
output += " : \n";
output += std::to_string(tmp->result);
output += " (";
output += std::to_string(tmp->result / 1024 / 1024);
output += " МБ)";
SendMessage(GetDlgItem(tmp->hDlg, IDC_LIST_SIZE), LB_ADDSTRING, 0, (LPARAM)output.c_str());
}
p = tmp;
}
return 0;
}
DWORD WINAPI GetFolderSize(PVOID p){
Property *tmp = (Property*)p;
for (int i = 0; i < tmp->countFile; i++){
// поле nFileSizeLow структуры wfd - хранит нижние биты размера файла( то что нам и нужно)
tmp->result += tmp->wfd[i].nFileSizeLow;
// тут тоже увеличиваем заполненность прогресс бара
SendMessage(GetDlgItem(tmp->hDlg, IDC_PROGRESS1), PBM_STEPIT, 0, 0);
Sleep(100);
}
//После нахождения размера файлов, устанавливаем событие
SetEvent(tmp->event);
return 0;
} |
|
Может есть у кого уже готовое? Было бы хорошо. Или хотя бы подскажите.
0
|
(
