mirror of
https://github.com/FairRootGroup/FairMQ.git
synced 2025-10-16 01:51:45 +00:00
merge the trunk with the development of ZeroMQ branch
git-svn-id: https://subversion.gsi.de/fairroot/fairbase/trunk@22451 0381ead4-6506-0410-b988-94b70fbc4730
This commit is contained in:
@@ -1,87 +1,115 @@
|
||||
/*
|
||||
/**
|
||||
* runBuffer.cxx
|
||||
*
|
||||
* Created on: Oct 26, 2012
|
||||
* Author: dklein
|
||||
* @since Oct 26, 2012
|
||||
* @authors: D. Klein, A. Rybalchenko
|
||||
*/
|
||||
|
||||
#include "FairMQBuffer.h"
|
||||
#include <sys/types.h>
|
||||
#include <unistd.h>
|
||||
#include "FairMQLogger.h"
|
||||
#include <zmq.hpp>
|
||||
#include <stdio.h>
|
||||
#include <iostream>
|
||||
#include <csignal>
|
||||
|
||||
#include "FairMQLogger.h"
|
||||
#include "FairMQBuffer.h"
|
||||
|
||||
|
||||
FairMQBuffer buffer;
|
||||
|
||||
static void s_signal_handler (int signal)
|
||||
{
|
||||
std::cout << std::endl << "Caught signal " << signal << std::endl;
|
||||
|
||||
buffer.ChangeState(FairMQBuffer::STOP);
|
||||
buffer.ChangeState(FairMQBuffer::END);
|
||||
|
||||
std::cout << "Shutdown complete. Bye!" << std::endl;
|
||||
exit(1);
|
||||
}
|
||||
|
||||
static void s_catch_signals (void)
|
||||
{
|
||||
struct sigaction action;
|
||||
action.sa_handler = s_signal_handler;
|
||||
action.sa_flags = 0;
|
||||
sigemptyset(&action.sa_mask);
|
||||
sigaction(SIGINT, &action, NULL);
|
||||
sigaction(SIGTERM, &action, NULL);
|
||||
}
|
||||
|
||||
int main(int argc, char** argv)
|
||||
{
|
||||
if( argc != 9 ) {
|
||||
std::cout << "Usage: buffer \tID numIoTreads\n" <<
|
||||
"\t\tconnectSocketType connectRcvBufferSize ConnectAddress\n" <<
|
||||
"\t\tbindSocketType bindSndBufferSize BindAddress\n" << std::endl;
|
||||
if ( argc != 11 ) {
|
||||
std::cout << "Usage: buffer \tID numIoTreads\n"
|
||||
<< "\t\tinputSocketType inputRcvBufSize inputMethod inputAddress\n"
|
||||
<< "\t\toutputSocketType outputSndBufSize outputMethod outputAddress\n" << std::endl;
|
||||
return 1;
|
||||
}
|
||||
|
||||
pid_t pid = getpid();
|
||||
s_catch_signals();
|
||||
|
||||
std::stringstream logmsg;
|
||||
logmsg << "PID: " << pid;
|
||||
logmsg << "PID: " << getpid();
|
||||
FairMQLogger::GetInstance()->Log(FairMQLogger::INFO, logmsg.str());
|
||||
|
||||
int i = 1;
|
||||
|
||||
FairMQBuffer* buffer = new FairMQBuffer();
|
||||
buffer->SetProperty(FairMQBuffer::Id, argv[i]);
|
||||
buffer.SetProperty(FairMQBuffer::Id, argv[i]);
|
||||
++i;
|
||||
|
||||
int numIoThreads;
|
||||
std::stringstream(argv[i]) >> numIoThreads;
|
||||
buffer->SetProperty(FairMQBuffer::NumIoThreads, numIoThreads);
|
||||
buffer.SetProperty(FairMQBuffer::NumIoThreads, numIoThreads);
|
||||
++i;
|
||||
buffer.SetProperty(FairMQBuffer::NumInputs, 1);
|
||||
buffer.SetProperty(FairMQBuffer::NumOutputs, 1);
|
||||
|
||||
int numInputs = 1;
|
||||
buffer->SetProperty(FairMQBuffer::NumInputs, numInputs);
|
||||
|
||||
int numOutputs = 1;
|
||||
buffer->SetProperty(FairMQBuffer::NumOutputs, numOutputs);
|
||||
buffer.ChangeState(FairMQBuffer::INIT);
|
||||
|
||||
buffer->Init();
|
||||
|
||||
int connectSocketType = ZMQ_SUB;
|
||||
int inputSocketType = ZMQ_SUB;
|
||||
if (strcmp(argv[i], "pull") == 0) {
|
||||
connectSocketType = ZMQ_PULL;
|
||||
inputSocketType = ZMQ_PULL;
|
||||
}
|
||||
buffer->SetProperty(FairMQBuffer::ConnectSocketType, connectSocketType, 0);
|
||||
buffer.SetProperty(FairMQBuffer::InputSocketType, inputSocketType, 0);
|
||||
++i;
|
||||
int inputRcvBufSize;
|
||||
std::stringstream(argv[i]) >> inputRcvBufSize;
|
||||
buffer.SetProperty(FairMQBuffer::InputRcvBufSize, inputRcvBufSize, 0);
|
||||
++i;
|
||||
buffer.SetProperty(FairMQBuffer::InputMethod, argv[i], 0);
|
||||
++i;
|
||||
buffer.SetProperty(FairMQBuffer::InputAddress, argv[i], 0);
|
||||
++i;
|
||||
|
||||
int connectRcvBufferSize;
|
||||
std::stringstream(argv[i]) >> connectRcvBufferSize;
|
||||
buffer->SetProperty(FairMQBuffer::ConnectRcvBufferSize, connectRcvBufferSize, 0);
|
||||
++i;
|
||||
|
||||
buffer->SetProperty(FairMQBuffer::ConnectAddress, argv[i], 0);
|
||||
++i;
|
||||
|
||||
int bindSocketType = ZMQ_PUB;
|
||||
int outputSocketType = ZMQ_PUB;
|
||||
if (strcmp(argv[i], "push") == 0) {
|
||||
bindSocketType = ZMQ_PUSH;
|
||||
outputSocketType = ZMQ_PUSH;
|
||||
}
|
||||
buffer->SetProperty(FairMQBuffer::BindSocketType, bindSocketType, 0);
|
||||
buffer.SetProperty(FairMQBuffer::OutputSocketType, outputSocketType, 0);
|
||||
++i;
|
||||
|
||||
int bindSndBufferSize;
|
||||
std::stringstream(argv[i]) >> bindSndBufferSize;
|
||||
buffer->SetProperty(FairMQBuffer::BindSndBufferSize, bindSndBufferSize, 0);
|
||||
int outputSndBufSize;
|
||||
std::stringstream(argv[i]) >> outputSndBufSize;
|
||||
buffer.SetProperty(FairMQBuffer::OutputSndBufSize, outputSndBufSize, 0);
|
||||
++i;
|
||||
|
||||
buffer->SetProperty(FairMQBuffer::BindAddress, argv[i], 0);
|
||||
buffer.SetProperty(FairMQBuffer::OutputMethod, argv[i], 0);
|
||||
++i;
|
||||
buffer.SetProperty(FairMQBuffer::OutputAddress, argv[i], 0);
|
||||
++i;
|
||||
|
||||
|
||||
buffer->Bind();
|
||||
buffer->Connect();
|
||||
buffer->Run();
|
||||
buffer.ChangeState(FairMQBuffer::SETOUTPUT);
|
||||
buffer.ChangeState(FairMQBuffer::SETINPUT);
|
||||
buffer.ChangeState(FairMQBuffer::RUN);
|
||||
|
||||
exit(0);
|
||||
|
||||
|
||||
char ch;
|
||||
std::cin.get(ch);
|
||||
|
||||
buffer.ChangeState(FairMQBuffer::STOP);
|
||||
buffer.ChangeState(FairMQBuffer::END);
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
|
Reference in New Issue
Block a user