-/* $Id: msg-thread.cpp,v 1.2 2005-06-02 06:40:46 adam Exp $
- Copyright (c) 1998-2005, Index Data.
-
-This file is part of the yaz-proxy.
+/* This file is part of YAZ proxy
+ Copyright (C) 1998-2011 Index Data
YAZ proxy is free software; you can redistribute it and/or modify it under
the terms of the GNU General Public License as published by the Free
for more details.
You should have received a copy of the GNU General Public License
-along with YAZ proxy; see the file LICENSE. If not, write to the
-Free Software Foundation, 59 Temple Place - Suite 330, Boston, MA
-02111-1307, USA.
- */
+along with this program; if not, write to the Free Software
+Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
+*/
+
+#if YAZ_POSIX_THREADS
#include <pthread.h>
+#endif
+
+#if HAVE_UNISTD_H
#include <unistd.h>
+#endif
+
#include <ctype.h>
+#include <stdio.h>
-#include <yaz++/socket-observer.h>
+#include <yazpp/socket-observer.h>
#include <yaz/log.h>
#include "msg-thread.h"
using namespace yazpp_1;
+class Msg_Thread::Private {
+public:
+ int m_no_threads;
+ Msg_Thread_Queue m_input;
+ Msg_Thread_Queue m_output;
+#if YAZ_POSIX_THREADS
+ int m_fd[2];
+ yazpp_1::ISocketObservable *m_SocketObservable;
+ pthread_t *m_thread_id;
+ pthread_mutex_t m_mutex_input_data;
+ pthread_cond_t m_cond_input_data;
+ pthread_mutex_t m_mutex_output_data;
+ bool m_stop_flag;
+#endif
+};
+
+IMsg_Thread::~IMsg_Thread()
+{
+
+}
+
Msg_Thread_Queue::Msg_Thread_Queue()
{
m_list = 0;
if (!*l)
return 0;
while ((*l)->m_next)
- l = &(*l)->m_next;
+ l = &(*l)->m_next;
IMsg_Thread *m = (*l)->m_item;
delete *l;
*l = 0;
return m;
}
+#if YAZ_POSIX_THREADS
static void *tfunc(void *p)
{
Msg_Thread *pt = (Msg_Thread *) p;
pt->run(0);
return 0;
}
+#endif
-
-Msg_Thread::Msg_Thread(IYazSocketObservable *obs)
- : m_SocketObservable(obs)
+Msg_Thread::Msg_Thread(ISocketObservable *obs, int no_threads)
{
- pipe(m_fd);
- obs->addObserver(m_fd[0], this);
- obs->maskObserver(this, YAZ_SOCKET_OBSERVE_READ);
-
- m_stop_flag = false;
- pthread_mutex_init(&m_mutex_input_data, 0);
- pthread_cond_init(&m_cond_input_data, 0);
- pthread_mutex_init(&m_mutex_output_data, 0);
- pthread_cond_init(&m_cond_output_data, 0);
- pthread_create(&m_thread_id, 0, tfunc, this);
+ m_p = new Private;
+
+#if YAZ_POSIX_THREADS
+ m_p->m_SocketObservable = obs;
+
+ pipe(m_p->m_fd);
+ obs->addObserver(m_p->m_fd[0], this);
+ obs->maskObserver(this, SOCKET_OBSERVE_READ);
+
+ m_p->m_stop_flag = false;
+ pthread_mutex_init(&m_p->m_mutex_input_data, 0);
+ pthread_cond_init(&m_p->m_cond_input_data, 0);
+ pthread_mutex_init(&m_p->m_mutex_output_data, 0);
+
+ m_p->m_no_threads = no_threads;
+ m_p->m_thread_id = new pthread_t[no_threads];
+ int i;
+ for (i = 0; i<m_p->m_no_threads; i++)
+ pthread_create(&m_p->m_thread_id[i], 0, tfunc, this);
+#endif
}
Msg_Thread::~Msg_Thread()
{
- pthread_mutex_lock(&m_mutex_input_data);
- m_stop_flag = true;
- pthread_cond_signal(&m_cond_input_data);
- pthread_mutex_unlock(&m_mutex_input_data);
-
- pthread_join(m_thread_id, 0);
-
- m_SocketObservable->deleteObserver(this);
-
- pthread_cond_destroy(&m_cond_input_data);
- pthread_mutex_destroy(&m_mutex_input_data);
- pthread_cond_destroy(&m_cond_output_data);
- pthread_mutex_destroy(&m_mutex_output_data);
- close(m_fd[0]);
- close(m_fd[1]);
+#if YAZ_POSIX_THREADS
+ pthread_mutex_lock(&m_p->m_mutex_input_data);
+ m_p->m_stop_flag = true;
+ pthread_cond_broadcast(&m_p->m_cond_input_data);
+ pthread_mutex_unlock(&m_p->m_mutex_input_data);
+
+ int i;
+ for (i = 0; i<m_p->m_no_threads; i++)
+ pthread_join(m_p->m_thread_id[i], 0);
+ delete [] m_p->m_thread_id;
+
+ m_p->m_SocketObservable->deleteObserver(this);
+
+ pthread_cond_destroy(&m_p->m_cond_input_data);
+ pthread_mutex_destroy(&m_p->m_mutex_input_data);
+ pthread_mutex_destroy(&m_p->m_mutex_output_data);
+ close(m_p->m_fd[0]);
+ close(m_p->m_fd[1]);
+#endif
+
+ delete m_p;
}
void Msg_Thread::socketNotify(int event)
{
- if (event & YAZ_SOCKET_OBSERVE_READ)
+#if YAZ_POSIX_THREADS
+ if (event & SOCKET_OBSERVE_READ)
{
- char buf[2];
- read(m_fd[0], buf, 1);
- pthread_mutex_lock(&m_mutex_output_data);
- IMsg_Thread *out = m_output.dequeue();
- pthread_mutex_unlock(&m_mutex_output_data);
- if (out)
- out->result();
+ char buf[2];
+ read(m_p->m_fd[0], buf, 1);
+ pthread_mutex_lock(&m_p->m_mutex_output_data);
+ IMsg_Thread *out = m_p->m_output.dequeue();
+ pthread_mutex_unlock(&m_p->m_mutex_output_data);
+ if (out)
+ out->result();
}
+#endif
}
+#if YAZ_POSIX_THREADS
void Msg_Thread::run(void *p)
{
while(1)
{
- pthread_mutex_lock(&m_mutex_input_data);
- pthread_cond_wait(&m_cond_input_data, &m_mutex_input_data);
- while (1)
- {
- if (m_stop_flag)
- {
- pthread_mutex_unlock(&m_mutex_input_data);
- return;
- }
- IMsg_Thread *in = m_input.dequeue();
- pthread_mutex_unlock(&m_mutex_input_data);
- if (!in)
- break;
- IMsg_Thread *out = in->handle();
- pthread_mutex_lock(&m_mutex_output_data);
- m_output.enqueue(out);
- pthread_cond_signal(&m_cond_output_data);
- pthread_mutex_unlock(&m_mutex_output_data);
-
- write(m_fd[1], "", 1);
-
- pthread_mutex_lock(&m_mutex_input_data);
- }
+ pthread_mutex_lock(&m_p->m_mutex_input_data);
+ while (!m_p->m_stop_flag && m_p->m_input.size() == 0)
+ pthread_cond_wait(&m_p->m_cond_input_data, &m_p->m_mutex_input_data);
+ if (m_p->m_stop_flag)
+ {
+ pthread_mutex_unlock(&m_p->m_mutex_input_data);
+ break;
+ }
+ IMsg_Thread *in = m_p->m_input.dequeue();
+ pthread_mutex_unlock(&m_p->m_mutex_input_data);
+
+ IMsg_Thread *out = in->handle();
+ pthread_mutex_lock(&m_p->m_mutex_output_data);
+ m_p->m_output.enqueue(out);
+
+ write(m_p->m_fd[1], "", 1);
+ pthread_mutex_unlock(&m_p->m_mutex_output_data);
}
}
+#endif
void Msg_Thread::put(IMsg_Thread *m)
{
- pthread_mutex_lock(&m_mutex_input_data);
- m_input.enqueue(m);
- pthread_cond_signal(&m_cond_input_data);
- pthread_mutex_unlock(&m_mutex_input_data);
+#if YAZ_POSIX_THREADS
+ pthread_mutex_lock(&m_p->m_mutex_input_data);
+ m_p->m_input.enqueue(m);
+ pthread_cond_signal(&m_p->m_cond_input_data);
+ pthread_mutex_unlock(&m_p->m_mutex_input_data);
+#else
+ IMsg_Thread *out = m->handle();
+ if (out)
+ out->result();
+#endif
}
+/*
+ * Local variables:
+ * c-basic-offset: 4
+ * c-file-style: "Stroustrup"
+ * indent-tabs-mode: nil
+ * End:
+ * vim: shiftwidth=4 tabstop=8 expandtab
+ */
+