1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
|
#pragma once
#include <yql/essentials/public/issue/yql_issue.h>
#include <yql/essentials/core/yql_type_annotation.h>
#include <library/cpp/yson/consumer.h>
#include <library/cpp/yson/node/node.h>
#include <util/generic/maybe.h>
#include <util/generic/string.h>
#include <util/generic/vector.h>
#include <util/generic/strbuf.h>
#include <functional>
namespace NYql {
class TTypeAnnotationNode;
class TStructExprType;
struct TExprContext;
namespace NCommon {
// empty return value means "remove member"
using TStructMemberMapper = std::function<TMaybe<TStringBuf> (TStringBuf member)>;
void WriteTypeToYson(NYson::TYsonConsumerBase& writer, const TTypeAnnotationNode* type, bool extendedForm = false);
// saves in columns order
void SaveStructTypeToYson(NYson::TYsonConsumerBase& writer, const TStructExprType* type,
const TMaybe<TColumnOrder>& columns = {}, const TStructMemberMapper& mapper = {}, bool extendedForm = false);
NYT::TNode TypeToYsonNode(const TTypeAnnotationNode* type, bool extendedForm = false);
TString WriteTypeToYson(const TTypeAnnotationNode* type, NYT::NYson::EYsonFormat format = NYT::NYson::EYsonFormat::Binary,
bool extendedForm = false);
const TTypeAnnotationNode* ParseTypeFromYson(const TStringBuf yson, TExprContext& ctx, const TPosition& pos = TPosition());
const TTypeAnnotationNode* ParseOrderAwareTypeFromYson(const TStringBuf yson, TColumnOrder& topLevelColumns, TExprContext& ctx, const TPosition& pos = TPosition());
const TTypeAnnotationNode* ParseTypeFromYson(const NYT::TNode& node, TExprContext& ctx, const TPosition& pos = TPosition());
const TTypeAnnotationNode* ParseOrderAwareTypeFromYson(const NYT::TNode& node, TColumnOrder& topLevelColumns, TExprContext& ctx, const TPosition& pos = TPosition());
void WriteResOrPullType(NYson::TYsonConsumerBase& writer, const TTypeAnnotationNode* type,
const TColumnOrder& columns);
} // namespace NCommon
} // namespace NYql
|
