src/ZeroMQHandler.h
author František Kučera <franta-hg@frantovo.cz>
Sun, 01 May 2022 22:27:42 +0200
branchv_0
changeset 1 ccaed729f8d3
parent 0 452d06d24ac2
child 2 e6294ba5017f
permissions -rw-r--r--
first version
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     1
/**
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     2
 * Relational pipes
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     3
 * Copyright © 2022 František Kučera (Frantovo.cz, GlobalCode.info)
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     4
 *
452d06d24ac2 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
452d06d24ac2 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
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     7
 * the Free Software Foundation, version 3 of the License.
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     8
 *
452d06d24ac2 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,
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    12
 * GNU General Public License for more details.
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    13
 *
452d06d24ac2 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
452d06d24ac2 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/>.
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    16
 */
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    17
#pragma once
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    18
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    19
#include <memory>
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    20
#include <string>
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    21
#include <vector>
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    22
#include <iostream>
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    23
#include <sstream>
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    24
#include <locale>
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    25
#include <codecvt>
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    26
1
ccaed729f8d3 first version
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    27
#include <zmq.hpp>
ccaed729f8d3 first version
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    28
0
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    29
#include <relpipe/common/type/typedefs.h>
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    30
#include <relpipe/reader/TypeId.h>
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    31
#include <relpipe/reader/handlers/RelationalReaderStringHandler.h>
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    32
#include <relpipe/reader/handlers/AttributeMetadata.h>
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    33
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    34
#include "Configuration.h"
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    35
#include "Hex.h"
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    36
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    37
namespace relpipe {
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    38
namespace out {
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    39
namespace zeromq {
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    40
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    41
class ZeroMQHandler : public relpipe::reader::handlers::RelationalReaderStringHandler {
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    42
private:
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    43
	std::wstring_convert<codecvt_utf8<wchar_t>> convertor; // TODO: support also other encodings.
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    44
	Configuration configuration;
1
ccaed729f8d3 first version
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    45
	zmq::context_t zmqContext;
ccaed729f8d3 first version
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    46
	zmq::socket_t zmqSocket;
0
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    47
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    48
	struct CurrentRelation {
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    49
		relpipe::common::type::StringX name;
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    50
		std::vector<relpipe::reader::handlers::AttributeMetadata> attributes;
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    51
		relpipe::common::type::Integer attributeIndex = 0;
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    52
		std::string currentValue;
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    53
	} currentRelation;
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    54
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    55
public:
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    56
1
ccaed729f8d3 first version
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    57
	ZeroMQHandler(Configuration configuration) : configuration(configuration), zmqSocket(zmqContext, zmq::socket_type::push) {
ccaed729f8d3 first version
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    58
		zmqSocket.connect(convertor.to_bytes(configuration.endpointUrl));
0
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    59
	}
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    60
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    61
	void startRelation(relpipe::common::type::StringX name, std::vector<relpipe::reader::handlers::AttributeMetadata> attributes) override {
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    62
		currentRelation = CurrentRelation{name, attributes};
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    63
	}
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    64
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    65
	void attribute(const relpipe::common::type::StringX& value) override {
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    66
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    67
		auto attributeName = currentRelation.attributes[currentRelation.attributeIndex].getAttributeName();
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    68
		if (attributeName == L"text" && value.size()) currentRelation.currentValue = convertor.to_bytes(value);
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    69
		else if (attributeName == L"data" && value.size()) currentRelation.currentValue = Hex::fromHex(value).str();
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    70
		else if (attributeName == L"text"); // keep empty or value from 'data'
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    71
		else if (attributeName == L"data"); // keep empty or value from 'text'
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    72
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    73
		currentRelation.attributeIndex++;
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    74
		if (currentRelation.attributeIndex == currentRelation.attributes.size()) {
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    75
			currentRelation.attributeIndex = 0;
1
ccaed729f8d3 first version
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    76
			zmqSocket.send(currentRelation.currentValue.c_str(), currentRelation.currentValue.size(), 0); // FIXME: check return value
0
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    77
		}
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    78
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    79
	}
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    80
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    81
	void endOfPipe() {
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    82
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    83
	}
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    84
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    85
};
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    86
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    87
}
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    88
}
452d06d24ac2 establish project
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    89
}