format.h

Go to the documentation of this file.

#pragma once

 
#include "node.h"
 
#include <google/protobuf/descriptor.h>
 
namespace NYT {


struct TYamredDsvAttributes
{
    TVector<TString> KeyColumnNames;

    TVector<TString> SubkeyColumnNames;
};


struct TFormat
{
public:
    TNode Config;
 
public:
    explicit TFormat(const TNode& config = TNode());

    static TFormat YsonText();

    static TFormat YsonBinary();

    static TFormat YaMRLenval();

    static TFormat Protobuf(
        const TVector<const ::google::protobuf::Descriptor*>& descriptors,
        bool withDescriptors = false);

    static TFormat Json();

    static TFormat Dsv();

    template<typename T>
    static inline TFormat Protobuf(bool withDescriptors = false);

    bool IsTextYson() const;

    bool IsProtobuf() const;

    bool IsYamredDsv() const;

    TYamredDsvAttributes GetYamredDsvAttributes() const;
};

 
template<typename T>
TFormat TFormat::Protobuf(bool withDescriptors) {
    return TFormat::Protobuf({T::descriptor()}, withDescriptors);
}

TTableSchema CreateTableSchema(
    const ::google::protobuf::Descriptor& messageDescriptor,
    bool keepFieldsWithoutExtension);

 
} // namespace NYT