Home Explore Help
Sign In
SMusatov
/
ydb
mirror of https://github.com/ydb-platform/ydb.git
1
0
Fork 0
Files
Branch: main
Branches Tags
ydb
/
yql
/
essentials
/
minikql
/
mkql_node_visitor.h

mkql_node_visitor.h 7.5 KB
Permalink History Raw

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195 
  1. #pragma once
    2. 

  2. #include "defs.h"
    3. 

  3. #include "mkql_node.h"
    4. 

  4. #include <util/generic/deque.h>
    5. 

  5. #include <util/generic/hash.h>
    6. 

  6. #include <util/generic/hash_set.h>
    7. 

  7. #include <util/generic/vector.h>
    8. 

  8. #include <library/cpp/containers/stack_vector/stack_vec.h>
    9. 

  9. #include <functional>
    10. 

  10. 

  11. namespace NKikimr {
    12. 

  12. namespace NMiniKQL {
    13. 

  13. 

  14. class INodeVisitor {
    15. 

  15. public:
    16. 

  16.     virtual ~INodeVisitor() {}
    17. 

  17.     virtual void Visit(TTypeType& node) = 0;
    18. 

  18.     virtual void Visit(TVoidType& node) = 0;
    19. 

  19.     virtual void Visit(TNullType& node) = 0;
    20. 

  20.     virtual void Visit(TEmptyListType& node) = 0;
    21. 

  21.     virtual void Visit(TEmptyDictType& node) = 0;
    22. 

  22.     virtual void Visit(TDataType& node) = 0;
    23. 

  23.     virtual void Visit(TPgType& node) = 0;
    24. 

  24.     virtual void Visit(TStructType& node) = 0;
    25. 

  25.     virtual void Visit(TListType& node) = 0;
    26. 

  26.     virtual void Visit(TOptionalType& node) = 0;
    27. 

  27.     virtual void Visit(TDictType& node) = 0;
    28. 

  28.     virtual void Visit(TCallableType& node) = 0;
    29. 

  29.     virtual void Visit(TAnyType& node) = 0;
    30. 

  30.     virtual void Visit(TTupleType& node) = 0;
    31. 

  31.     virtual void Visit(TResourceType& node) = 0;
    32. 

  32.     virtual void Visit(TVariantType& node) = 0;
    33. 

  33.     virtual void Visit(TVoid& node) = 0;
    34. 

  34.     virtual void Visit(TNull& node) = 0;
    35. 

  35.     virtual void Visit(TEmptyList& node) = 0;
    36. 

  36.     virtual void Visit(TEmptyDict& node) = 0;
    37. 

  37.     virtual void Visit(TDataLiteral& node) = 0;
    38. 

  38.     virtual void Visit(TStructLiteral& node) = 0;
    39. 

  39.     virtual void Visit(TListLiteral& node) = 0;
    40. 

  40.     virtual void Visit(TOptionalLiteral& node) = 0;
    41. 

  41.     virtual void Visit(TDictLiteral& node) = 0;
    42. 

  42.     virtual void Visit(TCallable& node) = 0;
    43. 

  43.     virtual void Visit(TAny& node) = 0;
    44. 

  44.     virtual void Visit(TTupleLiteral& node) = 0;
    45. 

  45.     virtual void Visit(TVariantLiteral& node) = 0;
    46. 

  46.     virtual void Visit(TStreamType& node) = 0;
    47. 

  47.     virtual void Visit(TFlowType& node) = 0;
    48. 

  48.     virtual void Visit(TTaggedType& node) = 0;
    49. 

  49.     virtual void Visit(TBlockType& node) = 0;
    50. 

  50.     virtual void Visit(TMultiType& node) = 0;
    51. 

  51. };
    52. 

  52. 

  53. class TThrowingNodeVisitor : public INodeVisitor {
    54. 

  54. public:
    55. 

  55.     void Visit(TTypeType& node) override;
    56. 

  56.     void Visit(TVoidType& node) override;
    57. 

  57.     void Visit(TNullType& node) override;
    58. 

  58.     void Visit(TEmptyListType& node) override;
    59. 

  59.     void Visit(TEmptyDictType& node) override;
    60. 

  60.     void Visit(TDataType& node) override;
    61. 

  61.     void Visit(TPgType& node) override;
    62. 

  62.     void Visit(TStructType& node) override;
    63. 

  63.     void Visit(TListType& node) override;
    64. 

  64.     void Visit(TOptionalType& node) override;
    65. 

  65.     void Visit(TDictType& node) override;
    66. 

  66.     void Visit(TCallableType& node) override;
    67. 

  67.     void Visit(TAnyType& node) override;
    68. 

  68.     void Visit(TTupleType& node) override;
    69. 

  69.     void Visit(TResourceType& node) override;
    70. 

  70.     void Visit(TVariantType& node) override;
    71. 

  71.     void Visit(TVoid& node) override;
    72. 

  72.     void Visit(TNull& node) override;
    73. 

  73.     void Visit(TEmptyList& node) override;
    74. 

  74.     void Visit(TEmptyDict& node) override;
    75. 

  75.     void Visit(TDataLiteral& node) override;
    76. 

  76.     void Visit(TStructLiteral& node) override;
    77. 

  77.     void Visit(TListLiteral& node) override;
    78. 

  78.     void Visit(TOptionalLiteral& node) override;
    79. 

  79.     void Visit(TDictLiteral& node) override;
    80. 

  80.     void Visit(TCallable& node) override;
    81. 

  81.     void Visit(TAny& node) override;
    82. 

  82.     void Visit(TTupleLiteral& node) override;
    83. 

  83.     void Visit(TVariantLiteral& node) override;
    84. 

  84.     void Visit(TStreamType& node) override;
    85. 

  85.     void Visit(TFlowType& node) override;
    86. 

  86.     void Visit(TTaggedType& node) override;
    87. 

  87.     void Visit(TBlockType& node) override;
    88. 

  88.     void Visit(TMultiType& node) override;
    89. 

  89. 

  90. protected:
    91. 

  91.     static void ThrowUnexpectedNodeType();
    92. 

  92. };
    93. 

  93. 

  94. class TEmptyNodeVisitor: public INodeVisitor {
    95. 

  95. public:
    96. 

  96.     void Visit(TTypeType& node) override;
    97. 

  97.     void Visit(TVoidType& node) override;
    98. 

  98.     void Visit(TNullType& node) override;
    99. 

  99.     void Visit(TEmptyListType& node) override;
    100. 

  100.     void Visit(TEmptyDictType& node) override;
    101. 

  101.     void Visit(TDataType& node) override;
    102. 

  102.     void Visit(TPgType& node) override;
    103. 

  103.     void Visit(TStructType& node) override;
    104. 

  104.     void Visit(TListType& node) override;
    105. 

  105.     void Visit(TOptionalType& node) override;
    106. 

  106.     void Visit(TDictType& node) override;
    107. 

  107.     void Visit(TCallableType& node) override;
    108. 

  108.     void Visit(TAnyType& node) override;
    109. 

  109.     void Visit(TTupleType& node) override;
    110. 

  110.     void Visit(TResourceType& node) override;
    111. 

  111.     void Visit(TVariantType& node) override;
    112. 

  112.     void Visit(TVoid& node) override;
    113. 

  113.     void Visit(TNull& node) override;
    114. 

  114.     void Visit(TEmptyList& node) override;
    115. 

  115.     void Visit(TEmptyDict& node) override;
    116. 

  116.     void Visit(TDataLiteral& node) override;
    117. 

  117.     void Visit(TStructLiteral& node) override;
    118. 

  118.     void Visit(TListLiteral& node) override;
    119. 

  119.     void Visit(TOptionalLiteral& node) override;
    120. 

  120.     void Visit(TDictLiteral& node) override;
    121. 

  121.     void Visit(TCallable& node) override;
    122. 

  122.     void Visit(TAny& node) override;
    123. 

  123.     void Visit(TTupleLiteral& node) override;
    124. 

  124.     void Visit(TVariantLiteral& node) override;
    125. 

  125.     void Visit(TStreamType& node) override;
    126. 

  126.     void Visit(TFlowType& node) override;
    127. 

  127.     void Visit(TTaggedType& node) override;
    128. 

  128.     void Visit(TBlockType& node) override;
    129. 

  129.     void Visit(TMultiType& node) override;
    130. 

  130. };
    131. 

  131. 

  132. class TExploringNodeVisitor : public INodeVisitor {
    133. 

  133. public:
    134. 

  134.     using TNodesVec = TStackVec<TNode*, 2>;
    135. 

  135. public:
    136. 

  136.     void Visit(TTypeType& node) override;
    137. 

  137.     void Visit(TVoidType& node) override;
    138. 

  138.     void Visit(TNullType& node) override;
    139. 

  139.     void Visit(TEmptyListType& node) override;
    140. 

  140.     void Visit(TEmptyDictType& node) override;
    141. 

  141.     void Visit(TDataType& node) override;
    142. 

  142.     void Visit(TPgType& node) override;
    143. 

  143.     void Visit(TStructType& node) override;
    144. 

  144.     void Visit(TListType& node) override;
    145. 

  145.     void Visit(TOptionalType& node) override;
    146. 

  146.     void Visit(TDictType& node) override;
    147. 

  147.     void Visit(TCallableType& node) override;
    148. 

  148.     void Visit(TAnyType& node) override;
    149. 

  149.     void Visit(TTupleType& node) override;
    150. 

  150.     void Visit(TResourceType& node) override;
    151. 

  151.     void Visit(TVariantType& node) override;
    152. 

  152.     void Visit(TVoid& node) override;
    153. 

  153.     void Visit(TNull& node) override;
    154. 

  154.     void Visit(TEmptyList& node) override;
    155. 

  155.     void Visit(TEmptyDict& node) override;
    156. 

  156.     void Visit(TDataLiteral& node) override;
    157. 

  157.     void Visit(TStructLiteral& node) override;
    158. 

  158.     void Visit(TListLiteral& node) override;
    159. 

  159.     void Visit(TOptionalLiteral& node) override;
    160. 

  160.     void Visit(TDictLiteral& node) override;
    161. 

  161.     void Visit(TCallable& node) override;
    162. 

  162.     void Visit(TAny& node) override;
    163. 

  163.     void Visit(TTupleLiteral& node) override;
    164. 

  164.     void Visit(TVariantLiteral& node) override;
    165. 

  165.     void Visit(TStreamType& node) override;
    166. 

  166.     void Visit(TFlowType& node) override;
    167. 

  167.     void Visit(TTaggedType& node) override;
    168. 

  168.     void Visit(TBlockType& node) override;
    169. 

  169.     void Visit(TMultiType& node) override;
    170. 

  170. 

  171.     void Walk(TNode* root, const TTypeEnvironment& env, const std::vector<TNode*>& terminalNodes = std::vector<TNode*>(),
    172. 

  172.         bool buildConsumersMap = false, size_t nodesCountHint = 0);
    173. 

  173.     const std::vector<TNode*>& GetNodes();
    174. 

  174.     const TNodesVec& GetConsumerNodes(TNode& node);
    175. 

  175.     void Clear();
    176. 

  176. 

  177. private:
    178. 

  178.     void AddChildNode(TNode* parent, TNode& child);
    179. 

  179. 

  180. private:
    181. 

  181.     std::vector<TNode*> NodeList;
    182. 

  182.     std::vector<TNode*>* Stack = nullptr;
    183. 

  183.     bool BuildConsumersMap = false;
    184. 

  184.     std::unordered_map<TNode*, TNodesVec> ConsumersMap;
    185. 

  185. };
    186. 

  186. 

  187. class TTypeEnvironment;
    188. 

  188. typedef std::function<TRuntimeNode (TCallable& callable, const TTypeEnvironment& env)> TCallableVisitFunc;
    189. 

  189. typedef std::function<TCallableVisitFunc (const TInternName& name)> TCallableVisitFuncProvider;
    190. 

  190. 

  191. TRuntimeNode SinglePassVisitCallables(TRuntimeNode root, TExploringNodeVisitor& explorer,
    192. 

  192.     const TCallableVisitFuncProvider& funcProvider, const TTypeEnvironment& env, bool inPlace, bool& wereChanges);
    193. 

  193. 

  194. }
    195. 

  195. }
    196.