src/msg-thread.cpp

   1 /* $Id: msg-thread.cpp,v 1.5 2005-06-25 15:58:33 adam Exp $
   2    Copyright (c) 1998-2005, Index Data.
   3
   4 This file is part of the yaz-proxy.
   5
   6 YAZ proxy is free software; you can redistribute it and/or modify it under
   7 the terms of the GNU General Public License as published by the Free
   8 Software Foundation; either version 2, or (at your option) any later
   9 version.
  10
  11 YAZ proxy is distributed in the hope that it will be useful, but WITHOUT ANY
  12 WARRANTY; without even the implied warranty of MERCHANTABILITY or
  13 FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  14 for more details.
  15
  16 You should have received a copy of the GNU General Public License
  17 along with YAZ proxy; see the file LICENSE.  If not, write to the
  18 Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
  19 02111-1307, USA.
  20  */
  21 #include <pthread.h>
  22 #include <unistd.h>
  23 #include <ctype.h>
  24 #include <stdio.h>
  25
  26 #include <yaz++/socket-observer.h>
  27 #include <yaz/log.h>
  28
  29 #include "msg-thread.h"
  30
  31 using namespace yazpp_1;
  32
  33 Msg_Thread_Queue::Msg_Thread_Queue()
  34 {
  35     m_list = 0;
  36 }
  37
  38 int Msg_Thread_Queue::size()
  39 {
  40     int no = 0;
  41     Msg_Thread_Queue_List *l;
  42     for (l = m_list; l; l = l->m_next)
  43         no++;
  44     return no;
  45 }
  46
  47 void Msg_Thread_Queue::enqueue(IMsg_Thread *m)
  48 {
  49     Msg_Thread_Queue_List *l = new Msg_Thread_Queue_List;
  50     l->m_next = m_list;
  51     l->m_item = m;
  52     m_list = l;
  53 }
  54
  55 IMsg_Thread *Msg_Thread_Queue::dequeue()
  56 {
  57     Msg_Thread_Queue_List **l = &m_list;
  58     if (!*l)
  59         return 0;
  60     while ((*l)->m_next)
  61         l = &(*l)->m_next;
  62     IMsg_Thread *m = (*l)->m_item;
  63     delete *l;
  64     *l = 0;
  65     return m;
  66 }
  67
  68 static void *tfunc(void *p)
  69 {
  70     Msg_Thread *pt = (Msg_Thread *) p;
  71     pt->run(0);
  72     return 0;
  73 }
  74
  75
  76 Msg_Thread::Msg_Thread(ISocketObservable *obs)
  77     : m_SocketObservable(obs)
  78 {
  79     pipe(m_fd);
  80     obs->addObserver(m_fd[0], this);
  81     obs->maskObserver(this, SOCKET_OBSERVE_READ);
  82
  83     m_stop_flag = false;
  84     pthread_mutex_init(&m_mutex_input_data, 0);
  85     pthread_cond_init(&m_cond_input_data, 0);
  86     pthread_mutex_init(&m_mutex_output_data, 0);
  87     pthread_cond_init(&m_cond_output_data, 0);
  88     pthread_create(&m_thread_id, 0, tfunc, this);
  89 }
  90
  91 Msg_Thread::~Msg_Thread()
  92 {
  93     pthread_mutex_lock(&m_mutex_input_data);
  94     m_stop_flag = true;
  95     pthread_cond_signal(&m_cond_input_data);
  96     pthread_mutex_unlock(&m_mutex_input_data);
  97
  98     pthread_join(m_thread_id, 0);
  99
 100     m_SocketObservable->deleteObserver(this);
 101
 102     pthread_cond_destroy(&m_cond_input_data);
 103     pthread_mutex_destroy(&m_mutex_input_data);
 104     pthread_cond_destroy(&m_cond_output_data);
 105     pthread_mutex_destroy(&m_mutex_output_data);
 106     close(m_fd[0]);
 107     close(m_fd[1]);
 108 }
 109
 110 void Msg_Thread::socketNotify(int event)
 111 {
 112     if (event & SOCKET_OBSERVE_READ)
 113     {
 114         char buf[2];
 115         read(m_fd[0], buf, 1);
 116         pthread_mutex_lock(&m_mutex_output_data);
 117         IMsg_Thread *out = m_output.dequeue();
 118         pthread_mutex_unlock(&m_mutex_output_data);
 119         if (out)
 120             out->result();
 121     }
 122 }
 123
 124 void Msg_Thread::run(void *p)
 125 {
 126     while(1)
 127     {
 128         pthread_mutex_lock(&m_mutex_input_data);
 129         pthread_cond_wait(&m_cond_input_data, &m_mutex_input_data);
 130         while (1)
 131         {
 132             if (m_stop_flag)
 133             {
 134                 pthread_mutex_unlock(&m_mutex_input_data);
 135                 return;
 136             }
 137             IMsg_Thread *in = m_input.dequeue();
 138             pthread_mutex_unlock(&m_mutex_input_data);
 139             if (!in)
 140                 break;
 141             IMsg_Thread *out = in->handle();
 142             pthread_mutex_lock(&m_mutex_output_data);
 143             m_output.enqueue(out);
 144             pthread_cond_signal(&m_cond_output_data);
 145             pthread_mutex_unlock(&m_mutex_output_data);
 146
 147             write(m_fd[1], "", 1);
 148
 149             pthread_mutex_lock(&m_mutex_input_data);
 150         }
 151     }
 152 }
 153
 154 void Msg_Thread::put(IMsg_Thread *m)
 155 {
 156     pthread_mutex_lock(&m_mutex_input_data);
 157     m_input.enqueue(m);
 158     pthread_cond_signal(&m_cond_input_data);
 159     pthread_mutex_unlock(&m_mutex_input_data);
 160 }
 161 /*
 162  * Local variables:
 163  * c-basic-offset: 4
 164  * indent-tabs-mode: nil
 165  * End:
 166  * vim: shiftwidth=4 tabstop=8 expandtab
 167  */
 168