ambrop7 15 лет назад
Родитель
Сommit
3de2af71eb
2 измененных файлов с 90 добавлено и 0 удалено
  1. 3 0
      tests/CMakeLists.txt
  2. 87 0
      tests/threadwork_test.c

+ 3 - 0
tests/CMakeLists.txt

@@ -3,3 +3,6 @@ add_executable(chunkbuffer2_test chunkbuffer2_test.c)
 add_executable(bproto_test bproto_test.c)
 
 add_executable(bstruct_test bstruct_test.c)
+
+add_executable(threadwork_test threadwork_test.c)
+target_link_libraries(threadwork_test threadwork)

+ 87 - 0
tests/threadwork_test.c

@@ -0,0 +1,87 @@
+#include <stddef.h>
+#include <stdio.h>
+
+#include <misc/debug.h>
+#include <system/BLog.h>
+#include <system/DebugObject.h>
+#include <threadwork/BThreadWork.h>
+
+BReactor reactor;
+BThreadWorkDispatcher twd;
+BThreadWork tw1;
+BThreadWork tw2;
+BThreadWork tw3;
+int num_left;
+
+static void handler_done (void *user)
+{
+    printf("work done\n");
+    
+    num_left--;
+    
+    if (num_left == 0) {
+        printf("all works done, quitting\n");
+        BReactor_Quit(&reactor, 0);
+    }
+}
+
+static void work_func (void *user)
+{
+    unsigned int x = 0;
+    
+    for (int i = 0; i < 10000; i++) {
+        for (int j = 0; j < 10000; j++) {
+            x++;
+        }
+    }
+}
+
+static void dummy_works (int n)
+{
+    for (int i = 0; i < n; i++) {
+        BThreadWork tw_tmp;
+        BThreadWork_Init(&tw_tmp, &twd, handler_done, NULL, work_func, NULL);
+        BThreadWork_Free(&tw_tmp);
+    }
+}
+
+int main ()
+{
+    BLog_InitStdout();
+    BLog_SetChannelLoglevel(BLOG_CHANNEL_BThreadWork, BLOG_DEBUG);
+    
+    if (!BReactor_Init(&reactor)) {
+        DEBUG("BReactor_Init failed");
+        goto fail1;
+    }
+    
+    if (!BThreadWorkDispatcher_Init(&twd, &reactor, 1)) {
+        DEBUG("BThreadWorkDispatcher_Init failed");
+        goto fail2;
+    }
+    
+    dummy_works(200);
+    
+    BThreadWork_Init(&tw1, &twd, handler_done, NULL, work_func, NULL);
+    
+    BThreadWork_Init(&tw2, &twd, handler_done, NULL, work_func, NULL);
+    
+    BThreadWork_Init(&tw3, &twd, handler_done, NULL, work_func, NULL);
+    
+    dummy_works(200);
+    
+    num_left = 3;
+    
+    BReactor_Exec(&reactor);
+    
+    BThreadWork_Free(&tw3);
+    BThreadWork_Free(&tw2);
+    BThreadWork_Free(&tw1);
+    BThreadWorkDispatcher_Free(&twd);
+fail2:
+    BReactor_Free(&reactor);
+fail1:
+    BLog_Free();
+    DebugObjectGlobal_Finish();
+    return 0;
+}