]> git.sur5r.net Git - bacula/bacula/blobdiff - bacula/src/lib/workq.c
- Convert more atoi to str_to_int64() for DB.
[bacula/bacula] / bacula / src / lib / workq.c
index dbe0001ffff78551386943c16b80cbad422a56da..a00adbfb3782db9dc0ee883ba70278aad911f573 100755 (executable)
@@ -28,7 +28,7 @@
  *
  */
 /*
-   Copyright (C) 2000-2003 Kern Sibbald and John Walker
+   Copyright (C) 2000-2004 Kern Sibbald and John Walker
 
    This program is free software; you can redistribute it and/or
    modify it under the terms of the GNU General Public License as
 #include "bacula.h"
 
 /* Forward referenced functions */
-static void *workq_server(void *arg);
+extern "C" void *workq_server(void *arg);
 
-/*   
+/*
  * Initialize a work queue
  *
  *  Returns: 0 on success
- *          errno on failure
+ *           errno on failure
  */
 int workq_init(workq_t *wq, int threads, void *(*engine)(void *arg))
 {
    int stat;
-                       
+
    if ((stat = pthread_attr_init(&wq->attr)) != 0) {
       return stat;
    }
@@ -80,11 +80,11 @@ int workq_init(workq_t *wq, int threads, void *(*engine)(void *arg))
    }
    wq->quit = 0;
    wq->first = wq->last = NULL;
-   wq->max_workers = threads;        /* max threads to create */
-   wq->num_workers = 0;              /* no threads yet */
-   wq->idle_workers = 0;             /* no idle threads */
-   wq->engine = engine;              /* routine to run */
-   wq->valid = WORKQ_VALID; 
+   wq->max_workers = threads;         /* max threads to create */
+   wq->num_workers = 0;               /* no threads yet */
+   wq->idle_workers = 0;              /* no idle threads */
+   wq->engine = engine;               /* routine to run */
+   wq->valid = WORKQ_VALID;
    return 0;
 }
 
@@ -92,7 +92,7 @@ int workq_init(workq_t *wq, int threads, void *(*engine)(void *arg))
  * Destroy a work queue
  *
  * Returns: 0 on success
- *         errno on failure
+ *          errno on failure
  */
 int workq_destroy(workq_t *wq)
 {
@@ -104,30 +104,30 @@ int workq_destroy(workq_t *wq)
   if ((stat = pthread_mutex_lock(&wq->mutex)) != 0) {
      return stat;
   }
-  wq->valid = 0;                     /* prevent any more operations */
+  wq->valid = 0;                      /* prevent any more operations */
 
-  /* 
-   * If any threads are active, wake them 
+  /*
+   * If any threads are active, wake them
    */
   if (wq->num_workers > 0) {
      wq->quit = 1;
      if (wq->idle_workers) {
-       if ((stat = pthread_cond_broadcast(&wq->work)) != 0) {
-          pthread_mutex_unlock(&wq->mutex);
-          return stat;
-       }
+        if ((stat = pthread_cond_broadcast(&wq->work)) != 0) {
+           pthread_mutex_unlock(&wq->mutex);
+           return stat;
+        }
      }
      while (wq->num_workers > 0) {
-       if ((stat = pthread_cond_wait(&wq->work, &wq->mutex)) != 0) {
-          pthread_mutex_unlock(&wq->mutex);
-          return stat;
-       }
+        if ((stat = pthread_cond_wait(&wq->work, &wq->mutex)) != 0) {
+           pthread_mutex_unlock(&wq->mutex);
+           return stat;
+        }
      }
   }
   if ((stat = pthread_mutex_unlock(&wq->mutex)) != 0) {
      return stat;
   }
-  stat = pthread_mutex_destroy(&wq->mutex);
+  stat  = pthread_mutex_destroy(&wq->mutex);
   stat1 = pthread_cond_destroy(&wq->work);
   stat2 = pthread_attr_destroy(&wq->attr);
   return (stat != 0 ? stat : (stat1 != 0 ? stat1 : stat2));
@@ -137,19 +137,19 @@ int workq_destroy(workq_t *wq)
 /*
  *  Add work to a queue
  *    wq is a queue that was created with workq_init
- *    element is a user unique item that will be passed to the 
- *       processing routine
+ *    element is a user unique item that will be passed to the
+ *        processing routine
  *    work_item will get internal work queue item -- if it is not NULL
  *    priority if non-zero will cause the item to be placed on the
- *       head of the list instead of the tail.
+ *        head of the list instead of the tail.
  */
 int workq_add(workq_t *wq, void *element, workq_ele_t **work_item, int priority)
 {
    int stat;
    workq_ele_t *item;
    pthread_t id;
-    
-   Dmsg0(200, "workq_add\n");
+
+   Dmsg0(1400, "workq_add\n");
    if (wq->valid != WORKQ_VALID) {
       return EINVAL;
    }
@@ -164,45 +164,45 @@ int workq_add(workq_t *wq, void *element, workq_ele_t **work_item, int priority)
       return stat;
    }
 
-   Dmsg0(200, "add item to queue\n");
+   Dmsg0(1400, "add item to queue\n");
    if (priority) {
       /* Add to head of queue */
       if (wq->first == NULL) {
-        wq->first = item;
-        wq->last = item;
+         wq->first = item;
+         wq->last = item;
       } else {
-        item->next = wq->first;
-        wq->first = item;
+         item->next = wq->first;
+         wq->first = item;
       }
    } else {
       /* Add to end of queue */
       if (wq->first == NULL) {
-        wq->first = item;
+         wq->first = item;
       } else {
-        wq->last->next = item;
+         wq->last->next = item;
       }
       wq->last = item;
    }
 
    /* if any threads are idle, wake one */
    if (wq->idle_workers > 0) {
-      Dmsg0(200, "Signal worker\n");
-      if ((stat = pthread_cond_signal(&wq->work)) != 0) {
-        pthread_mutex_unlock(&wq->mutex);
-        return stat;
+      Dmsg0(1400, "Signal worker\n");
+      if ((stat = pthread_cond_broadcast(&wq->work)) != 0) {
+         pthread_mutex_unlock(&wq->mutex);
+         return stat;
       }
    } else if (wq->num_workers < wq->max_workers) {
-      Dmsg0(200, "Create worker thread\n");
+      Dmsg0(1400, "Create worker thread\n");
       /* No idle threads so create a new one */
       set_thread_concurrency(wq->max_workers + 1);
       if ((stat = pthread_create(&id, &wq->attr, workq_server, (void *)wq)) != 0) {
-        pthread_mutex_unlock(&wq->mutex);
-        return stat;
+         pthread_mutex_unlock(&wq->mutex);
+         return stat;
       }
       wq->num_workers++;
    }
    pthread_mutex_unlock(&wq->mutex);
-   Dmsg0(200, "Return workq_add\n");
+   Dmsg0(1400, "Return workq_add\n");
    /* Return work_item if requested */
    if (work_item) {
       *work_item = item;
@@ -224,8 +224,8 @@ int workq_remove(workq_t *wq, workq_ele_t *work_item)
    int stat, found = 0;
    pthread_t id;
    workq_ele_t *item, *prev;
-    
-   Dmsg0(200, "workq_remove\n");
+
+   Dmsg0(1400, "workq_remove\n");
    if (wq->valid != WORKQ_VALID) {
       return EINVAL;
    }
@@ -236,20 +236,20 @@ int workq_remove(workq_t *wq, workq_ele_t *work_item)
 
    for (prev=item=wq->first; item; item=item->next) {
       if (item == work_item) {
-        found = 1;
-        break;
+         found = 1;
+         break;
       }
       prev = item;
    }
    if (!found) {
       return EINVAL;
    }
-   
+
    /* Move item to be first on list */
    if (wq->first != work_item) {
-      prev->next = work_item->next;   
+      prev->next = work_item->next;
       if (wq->last == work_item) {
-        wq->last = prev;
+         wq->last = prev;
       }
       work_item->next = wq->first;
       wq->first = work_item;
@@ -257,39 +257,40 @@ int workq_remove(workq_t *wq, workq_ele_t *work_item)
 
    /* if any threads are idle, wake one */
    if (wq->idle_workers > 0) {
-      Dmsg0(200, "Signal worker\n");
-      if ((stat = pthread_cond_signal(&wq->work)) != 0) {
-        pthread_mutex_unlock(&wq->mutex);
-        return stat;
+      Dmsg0(1400, "Signal worker\n");
+      if ((stat = pthread_cond_broadcast(&wq->work)) != 0) {
+         pthread_mutex_unlock(&wq->mutex);
+         return stat;
       }
    } else {
-      Dmsg0(200, "Create worker thread\n");
+      Dmsg0(1400, "Create worker thread\n");
       /* No idle threads so create a new one */
       set_thread_concurrency(wq->max_workers + 1);
       if ((stat = pthread_create(&id, &wq->attr, workq_server, (void *)wq)) != 0) {
-        pthread_mutex_unlock(&wq->mutex);
-        return stat;
+         pthread_mutex_unlock(&wq->mutex);
+         return stat;
       }
       wq->num_workers++;
    }
    pthread_mutex_unlock(&wq->mutex);
-   Dmsg0(200, "Return workq_remove\n");
+   Dmsg0(1400, "Return workq_remove\n");
    return stat;
 }
 
 
-/* 
+/*
  * This is the worker thread that serves the work queue.
  * In due course, it will call the user's engine.
  */
-static void *workq_server(void *arg)
+extern "C"
+void *workq_server(void *arg)
 {
    struct timespec timeout;
    workq_t *wq = (workq_t *)arg;
    workq_ele_t *we;
    int stat, timedout;
 
-   Dmsg0(200, "Start workq_server\n");
+   Dmsg0(1400, "Start workq_server\n");
    if ((stat = pthread_mutex_lock(&wq->mutex)) != 0) {
       return NULL;
    }
@@ -298,91 +299,91 @@ static void *workq_server(void *arg)
       struct timeval tv;
       struct timezone tz;
 
-      Dmsg0(200, "Top of for loop\n");
+      Dmsg0(1400, "Top of for loop\n");
       timedout = 0;
-      Dmsg0(200, "gettimeofday()\n");
+      Dmsg0(1400, "gettimeofday()\n");
       gettimeofday(&tv, &tz);
       timeout.tv_nsec = 0;
       timeout.tv_sec = tv.tv_sec + 2;
 
       while (wq->first == NULL && !wq->quit) {
-        /*
-         * Wait 2 seconds, then if no more work, exit
-         */
-         Dmsg0(200, "pthread_cond_timedwait()\n");
+         /*
+          * Wait 2 seconds, then if no more work, exit
+          */
+         Dmsg0(1400, "pthread_cond_timedwait()\n");
 #ifdef xxxxxxxxxxxxxxxx_was_HAVE_CYGWIN
-        /* CYGWIN dies with a page fault the second
-         * time that pthread_cond_timedwait() is called
-         * so fake it out.
-         */
-        pthread_mutex_lock(&wq->mutex);
-        stat = ETIMEDOUT;
+         /* CYGWIN dies with a page fault the second
+          * time that pthread_cond_timedwait() is called
+          * so fake it out.
+          */
+         pthread_mutex_lock(&wq->mutex);
+         stat = ETIMEDOUT;
 #else
-        stat = pthread_cond_timedwait(&wq->work, &wq->mutex, &timeout);
+         stat = pthread_cond_timedwait(&wq->work, &wq->mutex, &timeout);
 #endif
-         Dmsg1(200, "timedwait=%d\n", stat);
-        if (stat == ETIMEDOUT) {
-           timedout = 1;
-           break;
-        } else if (stat != 0) {
+         Dmsg1(1400, "timedwait=%d\n", stat);
+         if (stat == ETIMEDOUT) {
+            timedout = 1;
+            break;
+         } else if (stat != 0) {
             /* This shouldn't happen */
-            Dmsg0(200, "This shouldn't happen\n");
-           wq->num_workers--;
-           pthread_mutex_unlock(&wq->mutex);
-           return NULL;
-        }
-      } 
+            Dmsg0(1400, "This shouldn't happen\n");
+            wq->num_workers--;
+            pthread_mutex_unlock(&wq->mutex);
+            return NULL;
+         }
+      }
       we = wq->first;
       if (we != NULL) {
-        wq->first = we->next;
-        if (wq->last == we) {
-           wq->last = NULL;
-        }
-        if ((stat = pthread_mutex_unlock(&wq->mutex)) != 0) {
-           return NULL;
-        }
+         wq->first = we->next;
+         if (wq->last == we) {
+            wq->last = NULL;
+         }
+         if ((stat = pthread_mutex_unlock(&wq->mutex)) != 0) {
+            return NULL;
+         }
          /* Call user's routine here */
-         Dmsg0(200, "Calling user engine.\n");
-        wq->engine(we->data);
-         Dmsg0(200, "Back from user engine.\n");
-        free(we);                    /* release work entry */
-         Dmsg0(200, "relock mutex\n"); 
-        if ((stat = pthread_mutex_lock(&wq->mutex)) != 0) {
-           return NULL;
-        }
-         Dmsg0(200, "Done lock mutex\n");
+         Dmsg0(1400, "Calling user engine.\n");
+         wq->engine(we->data);
+         Dmsg0(1400, "Back from user engine.\n");
+         free(we);                    /* release work entry */
+         Dmsg0(1400, "relock mutex\n");
+         if ((stat = pthread_mutex_lock(&wq->mutex)) != 0) {
+            return NULL;
+         }
+         Dmsg0(1400, "Done lock mutex\n");
       }
       /*
        * If no more work request, and we are asked to quit, then do it
        */
       if (wq->first == NULL && wq->quit) {
-        wq->num_workers--;
-        if (wq->num_workers == 0) {
-            Dmsg0(200, "Wake up destroy routine\n");
-           /* Wake up destroy routine if he is waiting */
-           pthread_cond_broadcast(&wq->work);
-        }
-         Dmsg0(200, "Unlock mutex\n");
-        pthread_mutex_unlock(&wq->mutex);
-         Dmsg0(200, "Return from workq_server\n");
-        return NULL;
+         wq->num_workers--;
+         if (wq->num_workers == 0) {
+            Dmsg0(1400, "Wake up destroy routine\n");
+            /* Wake up destroy routine if he is waiting */
+            pthread_cond_broadcast(&wq->work);
+         }
+         Dmsg0(1400, "Unlock mutex\n");
+         pthread_mutex_unlock(&wq->mutex);
+         Dmsg0(1400, "Return from workq_server\n");
+         return NULL;
       }
-      Dmsg0(200, "Check for work request\n");
-      /* 
+      Dmsg0(1400, "Check for work request\n");
+      /*
        * If no more work requests, and we waited long enough, quit
        */
-      Dmsg1(200, "wq->first==NULL = %d\n", wq->first==NULL);
-      Dmsg1(200, "timedout=%d\n", timedout);
+      Dmsg1(1400, "wq->first==NULL = %d\n", wq->first==NULL);
+      Dmsg1(1400, "timedout=%d\n", timedout);
       if (wq->first == NULL && timedout) {
-         Dmsg0(200, "break big loop\n");
-        wq->num_workers--;
-        break;
+         Dmsg0(1400, "break big loop\n");
+         wq->num_workers--;
+         break;
       }
-      Dmsg0(200, "Loop again\n");
+      Dmsg0(1400, "Loop again\n");
    } /* end of big for loop */
 
-   Dmsg0(200, "unlock mutex\n");
+   Dmsg0(1400, "unlock mutex\n");
    pthread_mutex_unlock(&wq->mutex);
-   Dmsg0(200, "End workq_server\n");
+   Dmsg0(1400, "End workq_server\n");
    return NULL;
 }