src/AMQPCommand.cpp
author František Kučera <franta-hg@frantovo.cz>
Sun, 01 May 2022 18:29:58 +0200
branchv_0
changeset 0 08cb319d7c3a
permissions -rw-r--r--
establish project
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     1
/**
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     2
 * Relational pipes
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     3
 * Copyright © 2022 František Kučera (Frantovo.cz, GlobalCode.info)
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     4
 *
08cb319d7c3a 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
08cb319d7c3a 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
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     7
 * the Free Software Foundation, version 3 of the License.
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     8
 *
08cb319d7c3a 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,
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    12
 * GNU General Public License for more details.
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    13
 *
08cb319d7c3a 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
08cb319d7c3a 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/>.
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    16
 */
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    17
#include <cstdlib>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    18
#include <vector>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    19
#include <memory>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    20
#include <locale>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    21
#include <regex>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    22
#include <algorithm>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    23
#include <unistd.h>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    24
#include <sstream>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    25
#include <iomanip>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    26
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    27
#include <relpipe/writer/RelationalWriter.h>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    28
#include <relpipe/writer/RelpipeWriterException.h>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    29
#include <relpipe/writer/AttributeMetadata.h>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    30
#include <relpipe/writer/Factory.h>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    31
#include <relpipe/writer/TypeId.h>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    32
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    33
#include <relpipe/cli/CLI.h>
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    34
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    35
#include "AMQPCommand.h"
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    36
#include "AMQP.h"
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    37
#include "Hex.h"
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    38
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    39
using namespace std;
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    40
using namespace relpipe::cli;
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    41
using namespace relpipe::writer;
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    42
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    43
namespace relpipe {
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    44
namespace in {
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    45
namespace amqp {
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    46
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    47
void AMQPCommand::process(std::shared_ptr<writer::RelationalWriter> writer, Configuration& configuration) {
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    48
	vector<AttributeMetadata> metadata;
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    49
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    50
	std::shared_ptr<AMQP> mq(AMQP::open(convertor.to_bytes(configuration.queue), configuration.unlinkOnClose));
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    51
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    52
	writer->startRelation(configuration.relation,{
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    53
		{L"queue", TypeId::STRING},
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    54
		{L"text", TypeId::STRING},
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    55
		{L"data", TypeId::STRING}
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    56
	}, true);
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    57
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    58
	for (int i = configuration.messageCount; continueProcessing && i > 0; i--) {
08cb319d7c3a 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)
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    60
		std::string message = mq->receive();
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    61
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    62
		writer->writeAttribute(configuration.queue);
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    63
		writer->writeAttribute(Hex::toTxt(message));
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    64
		writer->writeAttribute(Hex::toHex(message));
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    65
	}
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    66
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    67
}
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    68
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    69
AMQPCommand::~AMQPCommand() {
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    70
}
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    71
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    72
}
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    73
}
08cb319d7c3a establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    74
}