mirror of https://github.com/sipwise/sems.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
207 lines
5.9 KiB
207 lines
5.9 KiB
/*
|
|
* $Id: ModMysql.cpp 1764 2010-04-01 14:33:30Z peter_lemenkov $
|
|
*
|
|
* Copyright (C) 2010 TelTech Systems Inc.
|
|
*
|
|
* This file is part of SEMS, a free SIP media server.
|
|
*
|
|
* SEMS 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 Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version. This program is released under
|
|
* the GPL with the additional exemption that compiling, linking,
|
|
* and/or using OpenSSL is allowed.
|
|
*
|
|
* For a license to use the SEMS software under conditions
|
|
* other than those described here, or to purchase support for this
|
|
* software, please contact iptel.org by e-mail at the following addresses:
|
|
* info@iptel.org
|
|
*
|
|
* SEMS is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include "RpcServerThread.h"
|
|
|
|
#include <sys/socket.h>
|
|
#include <unistd.h>
|
|
#include <stdio.h>
|
|
|
|
#include "RpcPeer.h"
|
|
#include "RpcServerLoop.h"
|
|
#include "jsonArg.h"
|
|
#include "JsonRPCServer.h"
|
|
|
|
#include "log.h"
|
|
|
|
RpcServerThread::RpcServerThread()
|
|
: AmEventQueue(this) {
|
|
}
|
|
|
|
RpcServerThread::~RpcServerThread() {
|
|
}
|
|
|
|
void RpcServerThread::run() {
|
|
while (true) {
|
|
waitForEvent();
|
|
processEvents();
|
|
}
|
|
}
|
|
|
|
void RpcServerThread::on_stop() {
|
|
INFO("TODO: stop server thread\n");
|
|
}
|
|
|
|
void RpcServerThread::process(AmEvent* event) {
|
|
JsonServerEvent* server_event = dynamic_cast<JsonServerEvent*>(event);
|
|
if (server_event == NULL) {
|
|
ERROR("invalid event to process\n");
|
|
return;
|
|
}
|
|
JsonrpcNetstringsConnection* connection = server_event->conn;
|
|
|
|
// todo: check event type - for now handle all types equally
|
|
|
|
if (server_event->event_id == JsonServerEvent::SendMessage) {
|
|
JsonServerSendMessageEvent* snd_msg_ev =
|
|
dynamic_cast<JsonServerSendMessageEvent*>(server_event);
|
|
|
|
if (NULL == snd_msg_ev) {
|
|
ERROR("wrong event type received\n");
|
|
return;
|
|
}
|
|
|
|
if (NULL == connection) {
|
|
DBG("getting connection for id %s\n", snd_msg_ev->connection_id.c_str());
|
|
JsonrpcPeerConnection* js_connection = JsonRPCServerLoop::getConnection(snd_msg_ev->connection_id);
|
|
if ((NULL == js_connection) ||
|
|
(NULL ==
|
|
(connection = dynamic_cast<JsonrpcNetstringsConnection*>(js_connection)))) {
|
|
ERROR("getting connection for id %s - message will not be sent\n",
|
|
snd_msg_ev->connection_id.c_str());
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (!snd_msg_ev->is_reply) {
|
|
if (JsonRpcServer::createRequest(snd_msg_ev->reply_link, snd_msg_ev->method,
|
|
snd_msg_ev->params, connection,
|
|
snd_msg_ev->udata,
|
|
snd_msg_ev->id.empty())) {
|
|
ERROR("creating request\n");
|
|
// give back connection into server loop
|
|
JsonRPCServerLoop::returnConnection(connection);
|
|
return;
|
|
}
|
|
} else {
|
|
if (JsonRpcServer::createReply(connection, snd_msg_ev->id, snd_msg_ev->params,
|
|
snd_msg_ev->is_error)) {
|
|
// give back connection into server loop
|
|
JsonRPCServerLoop::returnConnection(connection);
|
|
return;
|
|
}
|
|
}
|
|
connection->msg_recv = false;
|
|
|
|
}
|
|
|
|
bool processed_message = false;
|
|
|
|
int res = 0;
|
|
if (connection->messagePending() && connection->messageIsRecv()) {
|
|
DBG("processing message >%.*s<\n", connection->msg_size, connection->msgbuf);
|
|
res = JsonRpcServer::processMessage(connection->msgbuf, &connection->msg_size,
|
|
connection);
|
|
if (res<0) {
|
|
INFO("error processing message - closing connection\n");
|
|
connection->close();
|
|
connection->notifyDisconnect();
|
|
JsonRPCServerLoop::removeConnection(connection->id);
|
|
delete connection;
|
|
return;
|
|
}
|
|
|
|
connection->msg_recv = false;
|
|
processed_message = true;
|
|
}
|
|
|
|
DBG("connection->messagePending() = %s\n", connection->messagePending()?"true":"false");
|
|
|
|
if (connection->messagePending() && !connection->messageIsRecv()) {
|
|
DBG("calling write\n");
|
|
res = connection->netstringsBlockingWrite();
|
|
if (res == JsonrpcNetstringsConnection::REMOVE) {
|
|
connection->notifyDisconnect();
|
|
JsonRPCServerLoop::removeConnection(connection->id);
|
|
delete connection;
|
|
return;
|
|
}
|
|
}
|
|
|
|
if (processed_message &&
|
|
(connection->flags & JsonrpcPeerConnection::FL_CLOSE_ALWAYS)) {
|
|
DBG("closing connection marked as FL_CLOSE_ALWAYS\n");
|
|
connection->close();
|
|
connection->notifyDisconnect(); // ??
|
|
JsonRPCServerLoop::removeConnection(connection->id);
|
|
delete connection;
|
|
return;
|
|
}
|
|
|
|
// give back connection into server loop
|
|
JsonRPCServerLoop::returnConnection(connection);
|
|
|
|
// ev_io_init(&cli->ev_write,write_cb,cli->fd,EV_WRITE);
|
|
// ev_io_start(loop,&cli->ev_write);
|
|
}
|
|
|
|
|
|
RpcServerThreadpool::RpcServerThreadpool() {
|
|
// one thread is started here so that
|
|
// in app initialization code, there is already
|
|
// a server thread available to receive events
|
|
DBG("starting one server thread for startup requests...\n");
|
|
addThreads(1);
|
|
}
|
|
|
|
RpcServerThreadpool::~RpcServerThreadpool() {
|
|
}
|
|
|
|
/** round-robin dispatch to one thread */
|
|
void RpcServerThreadpool::dispatch(AmEvent* ev) {
|
|
threads_mut.lock();
|
|
if (!threads.size()) {
|
|
ERROR("no threads started for Rpc servers\n");
|
|
delete ev;
|
|
threads_mut.unlock();
|
|
return;
|
|
}
|
|
|
|
(*t_it)->postEvent(ev);
|
|
|
|
t_it++;
|
|
if (t_it == threads.end())
|
|
t_it = threads.begin();
|
|
|
|
threads_mut.unlock();
|
|
}
|
|
|
|
void RpcServerThreadpool::addThreads(unsigned int cnt) {
|
|
DBG("adding %u RPC server threads\n", cnt);
|
|
threads_mut.lock();
|
|
for (unsigned int i=0;i<cnt;i++) {
|
|
RpcServerThread* thr = new RpcServerThread();
|
|
thr->start();
|
|
threads.push_back(thr);
|
|
}
|
|
t_it = threads.begin();
|
|
threads_mut.unlock();
|
|
}
|
|
|