src/tests/test-thread.c

   1 #define _GNU_SOURCE
   2 #include <stdio.h>
   3 #include <pthread.h>
   4 #include <time.h>
   5 #include <unistd.h>
   6 #include <sys/syscall.h>
   7
   8 #include <afb/afb-req-itf.h>
   9 #include "../afb-thread.h"
  10 #include "../jobs.h"
  11
  12 struct foo {
  13         int value;
  14         int refcount;
  15 };
  16
  17 void addref(void *closure)
  18 {
  19         struct foo *foo = closure;
  20         foo->refcount++;
  21 }
  22
  23 void unref(void *closure)
  24 {
  25         struct foo *foo = closure;
  26         if(!--foo->refcount) {
  27                 /* printf("%06d FREE\n", foo->value); */
  28                 free(foo);
  29         }
  30 }
  31
  32 void fail(void *closure, const char *status, const char *info)
  33 {
  34         struct foo *foo = closure;
  35         printf("%06d ABORT T%d %s\n", foo->value, (int)syscall(SYS_gettid), status);
  36 }
  37
  38 struct afb_req_itf itf = {
  39         .json = NULL,
  40         .get = NULL,
  41
  42         .success = NULL,
  43         .fail = fail,
  44
  45         .raw = NULL,
  46         .send = NULL,
  47
  48         .context_get = NULL,
  49         .context_set = NULL,
  50
  51         .addref = addref,
  52         .unref = unref,
  53
  54         .session_close = NULL,
  55         .session_set_LOA = NULL,
  56
  57         .subscribe = NULL,
  58         .unsubscribe = NULL,
  59
  60         .subcall = NULL
  61 };
  62
  63 void process(struct afb_req req)
  64 {
  65         struct timespec ts;
  66         struct foo *foo = req.closure;
  67         printf("%06d PROCESS T%d\n", foo->value, (int)syscall(SYS_gettid));
  68         ts.tv_sec = 0;
  69         ts.tv_nsec = foo->value * 1000;
  70 //      nanosleep(&ts, NULL);
  71 }
  72
  73 void terminate(int signum)
  74 {
  75         printf("---------------- TERMINATE T%d (%d)\n", (int)syscall(SYS_gettid), signum);
  76 #if 1
  77         jobs_terminate();
  78 #else
  79         jobs_invoke0(0, jobs_terminate);
  80 #endif
  81         exit(0);
  82 }
  83
  84 void start()
  85 {
  86         int i;
  87         struct foo *foo;
  88         struct afb_req req;
  89         struct timespec ts;
  90
  91         req.itf = &itf;
  92         for (i = 0 ; i  < 10000 ; i++) {
  93                 req.closure = foo = malloc(sizeof *foo);
  94                 foo->value = i;
  95                 foo->refcount = 1;
  96                 afb_thread_req_call(req, process, 5, (&ts) + (i % 7));
  97                 unref(foo);
  98                 if (i == 5000)
  99                         jobs_queue0(NULL, 0, terminate);
 100                 ts.tv_sec = 0;
 101                 ts.tv_nsec = 1000000;
 102 //              nanosleep(&ts, NULL);
 103         }
 104 }
 105
 106
 107
 108 int main()
 109 {
 110         int i;
 111         struct foo *foo;
 112         struct afb_req req;
 113         struct timespec ts;
 114
 115         req.itf = &itf;
 116         jobs_start(4, 0, 20000, start);
 117         return 1;
 118 }
 119
 120
 121