src/YAMLCommand.h
author František Kučera <franta-hg@frantovo.cz>
Thu, 03 Dec 2020 22:31:03 +0100
branchv_0
changeset 2 d68192f0e960
parent 1 8d2d8f4077af
child 3 64f8f8792470
permissions -rw-r--r--
convert YAML to relations Expected structure is: map: key = relation name value = sequence of maps (records) where: key = attribute name value = attribute value Example: -------- fruit: - id: 1 name: cherry - id: 2 name: apple - id: 3 name: plum empty: # this relation has no attributes and will be ignored sparse: - a: aaa b: bbb c: ccc - a: AAA b: BBB - c: C d: D e: E a: A f: F - c: CCC -------- Not yet supported: -------- my_relation_with_metadata: attribute-metadata: - name: a type: string - name: b type: boolean record: - a: a - b: true --------
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
0
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     1
/**
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     2
 * Relational pipes
1
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
     3
 * Copyright © 2020 František Kučera (Frantovo.cz, GlobalCode.info)
0
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     4
 *
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     5
 * This program is free software: you can redistribute it and/or modify
66a454290286 project skeleton
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
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     7
 * the Free Software Foundation, version 3 of the License.
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     8
 *
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     9
 * This program is distributed in the hope that it will be useful,
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    12
 * GNU General Public License for more details.
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    13
 *
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    14
 * You should have received a copy of the GNU General Public License
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    15
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    16
 */
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    17
#pragma once
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    18
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    19
#include <cstdlib>
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    20
#include <iostream>
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    21
#include <memory>
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    22
#include <vector>
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    23
1
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    24
#include <yaml.h>
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    25
0
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    26
#include <relpipe/writer/RelationalWriter.h>
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    27
#include <relpipe/writer/typedefs.h>
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    28
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    29
namespace relpipe {
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    30
namespace in {
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    31
namespace yaml {
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    32
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    33
class YAMLCommand {
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    34
private:
2
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    35
	std::wstring_convert<codecvt_utf8<wchar_t>> convertor; // YAML strings are in UTF-8
0
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    36
1
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    37
	class YAMLEvent {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    38
	private:
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    39
		yaml_event_t event;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    40
	public:
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    41
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    42
		YAMLEvent(yaml_event_t event) : event(event) {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    43
		}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    44
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    45
		virtual ~YAMLEvent() {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    46
			yaml_event_delete(&event);
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    47
		}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    48
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    49
		const yaml_event_type_t getType() const {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    50
			return event.type;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    51
		}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    52
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    53
		const yaml_event_t* getEvent() const {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    54
			return &event;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    55
		}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    56
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    57
		YAMLEvent(const YAMLEvent&) = delete;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    58
		YAMLEvent& operator=(const YAMLEvent&) = delete;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    59
	};
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    60
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    61
	class YAMLParser {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    62
	private:
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    63
		yaml_parser_t yamlParser;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    64
		std::istream* input = nullptr;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    65
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    66
		static int readFromInput(void* instance, unsigned char* buffer, size_t size, size_t* length) {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    67
			std::istream* input = ((YAMLParser*) instance)->input;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    68
			input->read((char*) buffer, size);
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    69
			*length = input->gcount();
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    70
			return (input->good() || input->eof()) ? 1 : 0;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    71
		}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    72
	public:
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    73
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    74
		YAMLParser() {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    75
			yaml_parser_initialize(&yamlParser);
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    76
		}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    77
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    78
		virtual ~YAMLParser() {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    79
			yaml_parser_delete(&yamlParser);
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    80
		}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    81
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    82
		YAMLParser(const YAMLParser&) = delete;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    83
		YAMLParser& operator=(const YAMLParser&) = delete;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    84
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    85
		void setInput(std::istream* input) {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    86
			this->input = input;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    87
			yaml_parser_set_input(&yamlParser, readFromInput, (void*) this);
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    88
		}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    89
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    90
		YAMLEvent* next() {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    91
			yaml_event_t event;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    92
			int result = yaml_parser_parse(&yamlParser, &event);
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    93
			return result == 1 && event.type != yaml_event_type_e::YAML_NO_EVENT ? new YAMLEvent(event) : nullptr; // 1 = OK in yaml.h; YAML_NO_EVENT = end
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    94
		}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    95
	};
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    96
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    97
	YAMLParser parser;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    98
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
    99
	using YAMLEvent_p = std::shared_ptr<YAMLEvent>;
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   100
2
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   101
	enum class State {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   102
		START,
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   103
		RELATIONS,
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   104
		ATTRIBUTE,
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   105
		RECORDS
1
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   106
	};
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   107
2
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   108
	State state;
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   109
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   110
	relpipe::writer::string_t y2s(yaml_char_t* value) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   111
		return value ? convertor.from_bytes((const char*) value) : L"";
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   112
	}
1
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   113
2
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   114
	relpipe::writer::string_t findValue(std::vector<relpipe::writer::string_t> record, relpipe::writer::string_t key) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   115
		if (record.size() % 2) relpipe::writer::RelpipeWriterException(L"Invalid count of values in the record vector");
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   116
		for (size_t i = 0; i < record.size(); i += 2) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   117
			if (record[i] == key) return record[i + 1];
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   118
		}
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   119
		return L"";
1
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   120
	}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   121
0
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   122
public:
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   123
1
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   124
	YAMLCommand() {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   125
	}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   126
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   127
	virtual ~YAMLCommand() {
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   128
	}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   129
0
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   130
	void process(std::istream& input, std::shared_ptr<relpipe::writer::RelationalWriter> writer) {
1
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   131
		parser.setInput(&input);
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   132
2
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   133
		relpipe::writer::string_t relationName;
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   134
		std::vector<relpipe::writer::string_t> record;
0
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   135
		std::vector<relpipe::writer::AttributeMetadata> attributesMetadata;
1
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   136
2
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   137
		state = State::START;
1
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   138
2
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   139
		for (YAMLEvent_p event = YAMLEvent_p(parser.next()); event; event = YAMLEvent_p(parser.next())) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   140
			if (event->getType() == YAML_NO_EVENT) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   141
				throw relpipe::writer::RelpipeWriterException(L"Invalid YAML structure: YAML_NO_EVENT");
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   142
			} else if (event->getType() == YAML_STREAM_START_EVENT) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   143
			} else if (event->getType() == YAML_STREAM_END_EVENT) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   144
			} else if (event->getType() == YAML_DOCUMENT_START_EVENT) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   145
			} else if (event->getType() == YAML_DOCUMENT_END_EVENT) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   146
			} else if (event->getType() == YAML_ALIAS_EVENT) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   147
			} else if (event->getType() == YAML_SCALAR_EVENT) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   148
				relpipe::writer::string_t scalarValue = y2s(event->getEvent()->data.scalar.value);
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   149
				if (state == State::RELATIONS) relationName = scalarValue;
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   150
				else if (state == State::ATTRIBUTE) record.push_back(scalarValue);
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   151
				else throw relpipe::writer::RelpipeWriterException(L"Invalid YAML structure: unexpected YAML_SCALAR_EVENT");
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   152
			} else if (event->getType() == YAML_SEQUENCE_START_EVENT) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   153
				if (state == State::RELATIONS) state = State::RECORDS;
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   154
				else throw relpipe::writer::RelpipeWriterException(L"Invalid YAML structure: unexpected YAML_SEQUENCE_START_EVENT");
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   155
			} else if (event->getType() == YAML_SEQUENCE_END_EVENT) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   156
				if (state == State::RECORDS) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   157
					state = State::RELATIONS;
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   158
					relationName.clear();
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   159
					attributesMetadata.clear();
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   160
				} else {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   161
					throw relpipe::writer::RelpipeWriterException(L"Invalid YAML structure: unexpected YAML_SEQUENCE_END_EVENT");
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   162
				}
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   163
			} else if (event->getType() == YAML_MAPPING_START_EVENT) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   164
				if (state == State::START) state = State::RELATIONS;
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   165
				else if (state == State::RECORDS) state = State::ATTRIBUTE;
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   166
				else if (state == State::RELATIONS) throw relpipe::writer::RelpipeWriterException(L"Not yet implemented"); // TODO: there might be also a map in the relation value (not only a sequence) that would contain metadata (i.e. data types + support for relation containing no records)
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   167
				else throw relpipe::writer::RelpipeWriterException(L"Invalid YAML structure: unexpected YAML_MAPPING_START_EVENT");
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   168
			} else if (event->getType() == YAML_MAPPING_END_EVENT) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   169
				if (state == State::ATTRIBUTE) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   170
					if (attributesMetadata.size() == 0) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   171
						if (record.size() % 2) relpipe::writer::RelpipeWriterException(L"Invalid count of values in the record vector");
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   172
						for (size_t i = 0; i < record.size(); i += 2) attributesMetadata.push_back(relpipe::writer::AttributeMetadata{record[i], relpipe::writer::TypeId::STRING});
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   173
						writer->startRelation(relationName, attributesMetadata, true);
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   174
					}
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   175
					for (auto m : attributesMetadata) writer->writeAttribute(findValue(record, m.attributeName));
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   176
					record.clear();
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   177
					state = State::RECORDS;
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   178
				} else if (state == State::RELATIONS) {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   179
					break; // map of the relations ends (root)
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   180
				} else {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   181
					throw relpipe::writer::RelpipeWriterException(L"Invalid YAML structure: unexpected YAML_MAPPING_END_EVENT");
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   182
				}
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   183
			} else {
d68192f0e960 convert YAML to relations
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   184
				throw relpipe::writer::RelpipeWriterException(L"Invalid YAML structure: unknown event: " + std::to_wstring(event->getType()));
1
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   185
			}
8d2d8f4077af OOP wrapper for yaml.h + list event types
František Kučera <franta-hg@frantovo.cz>
parents: 0
diff changeset
   186
		}
0
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   187
	}
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   188
};
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   189
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   190
}
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   191
}
66a454290286 project skeleton
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   192
}