src/ZeroMQCommand.cpp
author František Kučera <franta-hg@frantovo.cz>
Sun, 01 May 2022 18:23:45 +0200
branchv_0
changeset 0 e5d547ab0c51
child 1 27c11cea34de
permissions -rw-r--r--
establish project
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     1
/**
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     2
 * Relational pipes
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     3
 * Copyright © 2022 František Kučera (Frantovo.cz, GlobalCode.info)
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     4
 *
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     5
 * This program is free software: you can redistribute it and/or modify
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     6
 * it under the terms of the GNU General Public License as published by
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     7
 * the Free Software Foundation, version 3 of the License.
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     8
 *
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     9
 * This program is distributed in the hope that it will be useful,
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    12
 * GNU General Public License for more details.
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    13
 *
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    14
 * You should have received a copy of the GNU General Public License
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    15
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    16
 */
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    17
#include <cstdlib>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    18
#include <vector>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    19
#include <memory>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    20
#include <locale>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    21
#include <regex>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    22
#include <algorithm>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    23
#include <unistd.h>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    24
#include <sstream>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    25
#include <iomanip>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    26
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    27
#include <relpipe/writer/RelationalWriter.h>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    28
#include <relpipe/writer/RelpipeWriterException.h>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    29
#include <relpipe/writer/AttributeMetadata.h>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    30
#include <relpipe/writer/Factory.h>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    31
#include <relpipe/writer/TypeId.h>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    32
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    33
#include <relpipe/cli/CLI.h>
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    34
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    35
#include "ZeroMQCommand.h"
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    36
#include "ZeroMQ.h"
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    37
#include "Hex.h"
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    38
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    39
using namespace std;
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    40
using namespace relpipe::cli;
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    41
using namespace relpipe::writer;
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    42
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    43
namespace relpipe {
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    44
namespace in {
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    45
namespace zeromq {
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    46
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    47
void ZeroMQCommand::process(std::shared_ptr<writer::RelationalWriter> writer, Configuration& configuration) {
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    48
	vector<AttributeMetadata> metadata;
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    49
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    50
	std::shared_ptr<ZeroMQ> mq(ZeroMQ::open(convertor.to_bytes(configuration.queue), configuration.unlinkOnClose));
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    51
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    52
	writer->startRelation(configuration.relation,{
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    53
		{L"queue", TypeId::STRING},
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    54
		{L"text", TypeId::STRING},
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    55
		{L"data", TypeId::STRING}
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    56
	}, true);
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    57
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    58
	for (int i = configuration.messageCount; continueProcessing && i > 0; i--) {
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    59
		// TODO: maybe rather call mq_timedreceive() inside and check continueProcessing (to be able to stop even when no messages are comming)
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    60
		std::string message = mq->receive();
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    61
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    62
		writer->writeAttribute(configuration.queue);
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    63
		writer->writeAttribute(Hex::toTxt(message));
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    64
		writer->writeAttribute(Hex::toHex(message));
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    65
	}
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    66
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    67
}
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    68
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    69
ZeroMQCommand::~ZeroMQCommand() {
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    70
}
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    71
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    72
}
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    73
}
e5d547ab0c51 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    74
}