ClickHouse/dbms/src/Formats/ProtobufSchemas.h
2019-06-20 12:12:49 +03:00

45 lines
991 B
C++

#pragma once
#include "config_formats.h"
#if USE_PROTOBUF
#include <memory>
#include <unordered_map>
#include <Core/Types.h>
#include <ext/singleton.h>
namespace google
{
namespace protobuf
{
class Descriptor;
}
}
namespace DB
{
class FormatSchemaInfo;
/** Keeps parsed google protobuf schemas parsed from files.
* This class is used to handle the "Protobuf" input/output formats.
*/
class ProtobufSchemas : public ext::singleton<ProtobufSchemas>
{
public:
ProtobufSchemas();
~ProtobufSchemas();
/// Parses the format schema, then parses the corresponding proto file, and returns the descriptor of the message type.
/// The function never returns nullptr, it throws an exception if it cannot load or parse the file.
const google::protobuf::Descriptor * getMessageTypeForFormatSchema(const FormatSchemaInfo & info);
private:
class ImporterWithSourceTree;
std::unordered_map<String, std::unique_ptr<ImporterWithSourceTree>> importers;
};
}
#endif