sql_query.cpp 168 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843
  1. #include "sql_query.h"
  2. #include "sql_expression.h"
  3. #include "sql_select.h"
  4. #include "sql_into_tables.h"
  5. #include "sql_values.h"
  6. #include "node.h"
  7. #include <yql/essentials/parser/proto_ast/gen/v1/SQLv1Lexer.h>
  8. #include <yql/essentials/parser/proto_ast/gen/v1_antlr4/SQLv1Antlr4Lexer.h>
  9. #include <yql/essentials/sql/v1/object_processing.h>
  10. #include <yql/essentials/utils/yql_paths.h>
  11. #include <util/generic/scope.h>
  12. #include <util/string/join.h>
  13. #ifdef GetMessage
  14. #undef GetMessage
  15. #endif
  16. namespace NSQLTranslationV1 {
  17. using NALPDefault::SQLv1LexerTokens;
  18. using NALPDefaultAntlr4::SQLv1Antlr4Lexer;
  19. using namespace NSQLv1Generated;
  20. void FillTargetList(TTranslation& ctx, const TRule_set_target_list& node, TVector<TString>& targetList) {
  21. targetList.push_back(ColumnNameAsSingleStr(ctx, node.GetRule_set_target2().GetRule_column_name1()));
  22. for (auto& block: node.GetBlock3()) {
  23. targetList.push_back(ColumnNameAsSingleStr(ctx, block.GetRule_set_target2().GetRule_column_name1()));
  24. }
  25. }
  26. bool PackageVersionFromString(const TString& s, ui32& version) {
  27. if (s == "release") {
  28. version = 0;
  29. return true;
  30. }
  31. if (s == "draft") {
  32. version = 1;
  33. return true;
  34. }
  35. return TryFromString(s, version);
  36. }
  37. void TSqlQuery::AddStatementToBlocks(TVector<TNodePtr>& blocks, TNodePtr node) {
  38. blocks.emplace_back(node);
  39. }
  40. static bool AsyncReplicationSettingsEntry(std::map<TString, TNodePtr>& out,
  41. const TRule_replication_settings_entry& in, TSqlExpression& ctx, bool create)
  42. {
  43. auto key = IdEx(in.GetRule_an_id1(), ctx);
  44. auto value = ctx.Build(in.GetRule_expr3());
  45. if (!value) {
  46. ctx.Context().Error() << "Invalid replication setting: " << key.Name;
  47. return false;
  48. }
  49. TSet<TString> configSettings = {
  50. "connection_string",
  51. "endpoint",
  52. "database",
  53. "token",
  54. "token_secret_name",
  55. "user",
  56. "password",
  57. "password_secret_name",
  58. };
  59. TSet<TString> modeSettings = {
  60. "consistency_level",
  61. "commit_interval",
  62. };
  63. TSet<TString> stateSettings = {
  64. "state",
  65. "failover_mode",
  66. };
  67. const auto keyName = to_lower(key.Name);
  68. if (!configSettings.count(keyName) && !modeSettings.count(keyName) && !stateSettings.count(keyName)) {
  69. ctx.Context().Error() << "Unknown replication setting: " << key.Name;
  70. return false;
  71. }
  72. if (create && stateSettings.count(keyName)) {
  73. ctx.Context().Error() << key.Name << " is not supported in CREATE";
  74. return false;
  75. }
  76. if (!create && modeSettings.count(keyName)) {
  77. ctx.Context().Error() << key.Name << " is not supported in ALTER";
  78. return false;
  79. }
  80. if (keyName == "commit_interval") {
  81. if (value->GetOpName() != "Interval") {
  82. ctx.Context().Error() << "Literal of Interval type is expected for " << key.Name;
  83. return false;
  84. }
  85. } else {
  86. if (!value->IsLiteral() || value->GetLiteralType() != "String") {
  87. ctx.Context().Error() << "Literal of String type is expected for " << key.Name;
  88. return false;
  89. }
  90. }
  91. if (!out.emplace(keyName, value).second) {
  92. ctx.Context().Error() << "Duplicate replication setting: " << key.Name;
  93. }
  94. return true;
  95. }
  96. static bool AsyncReplicationSettings(std::map<TString, TNodePtr>& out,
  97. const TRule_replication_settings& in, TSqlExpression& ctx, bool create)
  98. {
  99. if (!AsyncReplicationSettingsEntry(out, in.GetRule_replication_settings_entry1(), ctx, create)) {
  100. return false;
  101. }
  102. for (auto& block : in.GetBlock2()) {
  103. if (!AsyncReplicationSettingsEntry(out, block.GetRule_replication_settings_entry2(), ctx, create)) {
  104. return false;
  105. }
  106. }
  107. return true;
  108. }
  109. static bool AsyncReplicationTarget(std::vector<std::pair<TString, TString>>& out, TStringBuf prefixPath,
  110. const TRule_replication_target& in, TTranslation& ctx)
  111. {
  112. const TString remote = Id(in.GetRule_object_ref1().GetRule_id_or_at2(), ctx).second;
  113. const TString local = Id(in.GetRule_object_ref3().GetRule_id_or_at2(), ctx).second;
  114. out.emplace_back(remote, BuildTablePath(prefixPath, local));
  115. return true;
  116. }
  117. static bool AsyncReplicationAlterAction(std::map<TString, TNodePtr>& settings,
  118. const TRule_alter_replication_action& in, TSqlExpression& ctx)
  119. {
  120. // TODO(ilnaz): support other actions
  121. return AsyncReplicationSettings(settings, in.GetRule_alter_replication_set_setting1().GetRule_replication_settings3(), ctx, false);
  122. }
  123. static bool TransferSettingsEntry(std::map<TString, TNodePtr>& out,
  124. const TRule_transfer_settings_entry& in, TSqlExpression& ctx, bool create)
  125. {
  126. auto key = IdEx(in.GetRule_an_id1(), ctx);
  127. auto value = ctx.Build(in.GetRule_expr3());
  128. if (!value) {
  129. ctx.Context().Error() << "Invalid transfer setting: " << key.Name;
  130. return false;
  131. }
  132. TSet<TString> configSettings = {
  133. "connection_string",
  134. "endpoint",
  135. "database",
  136. "token",
  137. "token_secret_name",
  138. "user",
  139. "password",
  140. "password_secret_name",
  141. };
  142. TSet<TString> stateSettings = {
  143. "state",
  144. "failover_mode",
  145. };
  146. const auto keyName = to_lower(key.Name);
  147. if (!configSettings.count(keyName) && !stateSettings.contains(keyName)) {
  148. ctx.Context().Error() << "Unknown transfer setting: " << key.Name;
  149. return false;
  150. }
  151. if (create && stateSettings.count(keyName)) {
  152. ctx.Context().Error() << key.Name << " is not supported in CREATE";
  153. return false;
  154. }
  155. if (!out.emplace(keyName, value).second) {
  156. ctx.Context().Error() << "Duplicate transfer setting: " << key.Name;
  157. }
  158. return true;
  159. }
  160. static bool TransferSettings(std::map<TString, TNodePtr>& out,
  161. const TRule_transfer_settings& in, TSqlExpression& ctx, bool create)
  162. {
  163. if (!TransferSettingsEntry(out, in.GetRule_transfer_settings_entry1(), ctx, create)) {
  164. return false;
  165. }
  166. for (auto& block : in.GetBlock2()) {
  167. if (!TransferSettingsEntry(out, block.GetRule_transfer_settings_entry2(), ctx, create)) {
  168. return false;
  169. }
  170. }
  171. return true;
  172. }
  173. bool TSqlQuery::Statement(TVector<TNodePtr>& blocks, const TRule_sql_stmt_core& core, size_t statementNumber) {
  174. TString internalStatementName;
  175. TString humanStatementName;
  176. ParseStatementName(core, internalStatementName, humanStatementName);
  177. const auto& altCase = core.Alt_case();
  178. if (Mode == NSQLTranslation::ESqlMode::LIMITED_VIEW && (altCase >= TRule_sql_stmt_core::kAltSqlStmtCore4 &&
  179. altCase != TRule_sql_stmt_core::kAltSqlStmtCore13)) {
  180. Error() << humanStatementName << " statement is not supported in limited views";
  181. return false;
  182. }
  183. if (Mode == NSQLTranslation::ESqlMode::SUBQUERY && (altCase >= TRule_sql_stmt_core::kAltSqlStmtCore4 &&
  184. altCase != TRule_sql_stmt_core::kAltSqlStmtCore13 && altCase != TRule_sql_stmt_core::kAltSqlStmtCore6 &&
  185. altCase != TRule_sql_stmt_core::kAltSqlStmtCore18)) {
  186. Error() << humanStatementName << " statement is not supported in subqueries";
  187. return false;
  188. }
  189. if (NeedUseForAllStatements(altCase)) {
  190. Ctx.ForAllStatementsParts.push_back(statementNumber);
  191. }
  192. switch (altCase) {
  193. case TRule_sql_stmt_core::kAltSqlStmtCore1: {
  194. bool success = false;
  195. TNodePtr nodeExpr = PragmaStatement(core.GetAlt_sql_stmt_core1().GetRule_pragma_stmt1(), success);
  196. if (!success) {
  197. return false;
  198. }
  199. if (nodeExpr) {
  200. AddStatementToBlocks(blocks, nodeExpr);
  201. }
  202. break;
  203. }
  204. case TRule_sql_stmt_core::kAltSqlStmtCore2: {
  205. if (Ctx.ParallelModeCount > 0) {
  206. Error() << humanStatementName << " statement is not supported in parallel mode";
  207. return false;
  208. }
  209. Ctx.BodyPart();
  210. TSqlSelect select(Ctx, Mode);
  211. TPosition pos;
  212. auto source = select.Build(core.GetAlt_sql_stmt_core2().GetRule_select_stmt1(), pos);
  213. if (!source) {
  214. return false;
  215. }
  216. blocks.emplace_back(BuildSelectResult(pos, std::move(source),
  217. Mode != NSQLTranslation::ESqlMode::LIMITED_VIEW && Mode != NSQLTranslation::ESqlMode::SUBQUERY, Mode == NSQLTranslation::ESqlMode::SUBQUERY,
  218. Ctx.Scoped));
  219. break;
  220. }
  221. case TRule_sql_stmt_core::kAltSqlStmtCore3: {
  222. Ctx.BodyPart();
  223. TVector<TSymbolNameWithPos> names;
  224. auto nodeExpr = NamedNode(core.GetAlt_sql_stmt_core3().GetRule_named_nodes_stmt1(), names);
  225. if (!nodeExpr) {
  226. return false;
  227. }
  228. TVector<TNodePtr> nodes;
  229. auto subquery = nodeExpr->GetSource();
  230. if (subquery && Mode == NSQLTranslation::ESqlMode::LIBRARY && Ctx.ScopeLevel == 0) {
  231. for (size_t i = 0; i < names.size(); ++i) {
  232. nodes.push_back(BuildInvalidSubqueryRef(subquery->GetPos()));
  233. }
  234. } else if (subquery) {
  235. const auto alias = Ctx.MakeName("subquerynode");
  236. const auto ref = Ctx.MakeName("subquery");
  237. blocks.push_back(BuildSubquery(subquery, alias,
  238. Mode == NSQLTranslation::ESqlMode::SUBQUERY, names.size() == 1 ? -1 : names.size(), Ctx.Scoped));
  239. blocks.back()->SetLabel(ref);
  240. for (size_t i = 0; i < names.size(); ++i) {
  241. nodes.push_back(BuildSubqueryRef(blocks.back(), ref, names.size() == 1 ? -1 : i));
  242. }
  243. } else if (!Ctx.CompactNamedExprs || nodeExpr->GetUdfNode()) {
  244. // Unlike other nodes, TUdfNode is not an independent node, but more like a set of parameters which should be
  245. // applied on UDF call site. For example, TUdfNode can not be Translate()d
  246. // So we can't add it to blocks and use reference, instead we store the TUdfNode itself as named node
  247. // TODO: remove this special case
  248. if (names.size() > 1) {
  249. auto tupleRes = BuildTupleResult(nodeExpr, names.size());
  250. for (size_t i = 0; i < names.size(); ++i) {
  251. nodes.push_back(nodeExpr->Y("Nth", tupleRes, nodeExpr->Q(ToString(i))));
  252. }
  253. } else {
  254. nodes.push_back(std::move(nodeExpr));
  255. }
  256. } else {
  257. const auto ref = Ctx.MakeName("namedexprnode");
  258. blocks.push_back(BuildNamedExpr(names.size() > 1 ? BuildTupleResult(nodeExpr, names.size()) : nodeExpr));
  259. blocks.back()->SetLabel(ref);
  260. for (size_t i = 0; i < names.size(); ++i) {
  261. nodes.push_back(BuildNamedExprReference(blocks.back(), ref, names.size() == 1 ? TMaybe<size_t>() : i));
  262. }
  263. }
  264. for (size_t i = 0; i < names.size(); ++i) {
  265. PushNamedNode(names[i].Pos, names[i].Name, nodes[i]);
  266. }
  267. break;
  268. }
  269. case TRule_sql_stmt_core::kAltSqlStmtCore4: {
  270. Ctx.BodyPart();
  271. const auto& rule = core.GetAlt_sql_stmt_core4().GetRule_create_table_stmt1();
  272. bool replaceIfExists = false;
  273. if (rule.HasBlock2()) { // OR REPLACE
  274. replaceIfExists = true;
  275. Y_DEBUG_ABORT_UNLESS(
  276. (IS_TOKEN(rule.GetBlock2().GetToken1().GetId(), OR) &&
  277. IS_TOKEN(rule.GetBlock2().GetToken2().GetId(), REPLACE))
  278. );
  279. }
  280. const bool isCreateTableAs = rule.HasBlock15();
  281. const auto& block = rule.GetBlock3();
  282. ETableType tableType = ETableType::Table;
  283. bool temporary = false;
  284. if (block.HasAlt2() &&
  285. IS_TOKEN(block.GetAlt2().GetToken1().GetId(), TABLESTORE)
  286. ) {
  287. tableType = ETableType::TableStore;
  288. if (isCreateTableAs) {
  289. Context().Error(GetPos(block.GetAlt2().GetToken1()))
  290. << "CREATE TABLE AS is not supported for TABLESTORE";
  291. return false;
  292. }
  293. } else if (block.HasAlt3() &&
  294. IS_TOKEN(block.GetAlt3().GetToken1().GetId(), EXTERNAL)
  295. ) {
  296. tableType = ETableType::ExternalTable;
  297. if (isCreateTableAs) {
  298. Context().Error(GetPos(block.GetAlt3().GetToken1()))
  299. << "CREATE TABLE AS is not supported for EXTERNAL TABLE";
  300. return false;
  301. }
  302. } else if (block.HasAlt4() && IS_TOKEN(block.GetAlt4().GetToken1().GetId(), TEMP) ||
  303. block.HasAlt5() && IS_TOKEN(block.GetAlt5().GetToken1().GetId(), TEMPORARY)) {
  304. temporary = true;
  305. }
  306. bool existingOk = false;
  307. if (rule.HasBlock4()) { // IF NOT EXISTS
  308. existingOk = true;
  309. Y_DEBUG_ABORT_UNLESS(
  310. IS_TOKEN(rule.GetBlock4().GetToken1().GetId(), IF) &&
  311. IS_TOKEN(rule.GetBlock4().GetToken2().GetId(), NOT) &&
  312. IS_TOKEN(rule.GetBlock4().GetToken3().GetId(), EXISTS)
  313. );
  314. }
  315. if (replaceIfExists && tableType != ETableType::ExternalTable) {
  316. Context().Error(GetPos(rule.GetBlock2().GetToken1()))
  317. << "OR REPLACE feature is supported only for EXTERNAL DATA SOURCE and EXTERNAL TABLE";
  318. return false;
  319. }
  320. TTableRef tr;
  321. if (!SimpleTableRefImpl(rule.GetRule_simple_table_ref5(), tr)) {
  322. return false;
  323. }
  324. TCreateTableParameters params{.TableType=tableType, .Temporary=temporary};
  325. if (!CreateTableEntry(rule.GetRule_create_table_entry7(), params, isCreateTableAs)) {
  326. return false;
  327. }
  328. for (auto& block: rule.GetBlock8()) {
  329. if (!CreateTableEntry(block.GetRule_create_table_entry2(), params, isCreateTableAs)) {
  330. return false;
  331. }
  332. }
  333. if (rule.HasBlock11()) {
  334. Context().Error(GetPos(rule.GetBlock11().GetRule_table_inherits1().GetToken1()))
  335. << "INHERITS clause is not supported yet";
  336. return false;
  337. }
  338. if (rule.HasBlock12()) {
  339. if (tableType == ETableType::TableStore) {
  340. Context().Error(GetPos(rule.GetBlock12().GetRule_table_partition_by1().GetToken1()))
  341. << "PARTITION BY is not supported for TABLESTORE";
  342. return false;
  343. }
  344. const auto list = rule.GetBlock12().GetRule_table_partition_by1().GetRule_pure_column_list4();
  345. params.PartitionByColumns.push_back(IdEx(list.GetRule_an_id2(), *this));
  346. for (auto& node : list.GetBlock3()) {
  347. params.PartitionByColumns.push_back(IdEx(node.GetRule_an_id2(), *this));
  348. }
  349. }
  350. if (rule.HasBlock13()) {
  351. if (!CreateTableSettings(rule.GetBlock13().GetRule_with_table_settings1(), params)) {
  352. return false;
  353. }
  354. }
  355. if (rule.HasBlock14()) {
  356. Context().Error(GetPos(rule.GetBlock14().GetRule_table_tablestore1().GetToken1()))
  357. << "TABLESTORE clause is not supported yet";
  358. return false;
  359. }
  360. TSourcePtr tableSource = nullptr;
  361. if (isCreateTableAs) {
  362. tableSource = TSqlAsValues(Ctx, Mode).Build(rule.GetBlock15().GetRule_table_as_source1().GetRule_values_source2(), "CreateTableAs");
  363. if (!tableSource) {
  364. return false;
  365. }
  366. }
  367. if (!ValidateExternalTable(params)) {
  368. return false;
  369. }
  370. AddStatementToBlocks(blocks, BuildCreateTable(Ctx.Pos(), tr, existingOk, replaceIfExists, params, std::move(tableSource), Ctx.Scoped));
  371. break;
  372. }
  373. case TRule_sql_stmt_core::kAltSqlStmtCore5: {
  374. Ctx.BodyPart();
  375. const auto& rule = core.GetAlt_sql_stmt_core5().GetRule_drop_table_stmt1();
  376. const auto& block = rule.GetBlock2();
  377. ETableType tableType = ETableType::Table;
  378. if (block.HasAlt2()) {
  379. tableType = ETableType::TableStore;
  380. }
  381. if (block.HasAlt3()) {
  382. tableType = ETableType::ExternalTable;
  383. }
  384. bool missingOk = false;
  385. if (rule.HasBlock3()) { // IF EXISTS
  386. missingOk = true;
  387. Y_DEBUG_ABORT_UNLESS(
  388. IS_TOKEN(rule.GetBlock3().GetToken1().GetId(), IF) &&
  389. IS_TOKEN(rule.GetBlock3().GetToken2().GetId(), EXISTS)
  390. );
  391. }
  392. TTableRef tr;
  393. if (!SimpleTableRefImpl(rule.GetRule_simple_table_ref4(), tr)) {
  394. return false;
  395. }
  396. AddStatementToBlocks(blocks, BuildDropTable(Ctx.Pos(), tr, missingOk, tableType, Ctx.Scoped));
  397. break;
  398. }
  399. case TRule_sql_stmt_core::kAltSqlStmtCore6: {
  400. const auto& rule = core.GetAlt_sql_stmt_core6().GetRule_use_stmt1();
  401. Token(rule.GetToken1());
  402. if (!ClusterExpr(rule.GetRule_cluster_expr2(), true, Ctx.Scoped->CurrService, Ctx.Scoped->CurrCluster)) {
  403. return false;
  404. }
  405. break;
  406. }
  407. case TRule_sql_stmt_core::kAltSqlStmtCore7: {
  408. Ctx.BodyPart();
  409. TSqlIntoTable intoTable(Ctx, Mode);
  410. TNodePtr block(intoTable.Build(core.GetAlt_sql_stmt_core7().GetRule_into_table_stmt1()));
  411. if (!block) {
  412. return false;
  413. }
  414. blocks.emplace_back(block);
  415. break;
  416. }
  417. case TRule_sql_stmt_core::kAltSqlStmtCore8: {
  418. if (Ctx.ParallelModeCount > 0) {
  419. Error() << humanStatementName << " statement is not supported in parallel mode";
  420. return false;
  421. }
  422. Ctx.BodyPart();
  423. const auto& rule = core.GetAlt_sql_stmt_core8().GetRule_commit_stmt1();
  424. Token(rule.GetToken1());
  425. blocks.emplace_back(BuildCommitClusters(Ctx.Pos()));
  426. break;
  427. }
  428. case TRule_sql_stmt_core::kAltSqlStmtCore9: {
  429. Ctx.BodyPart();
  430. auto updateNode = Build(core.GetAlt_sql_stmt_core9().GetRule_update_stmt1());
  431. if (!updateNode) {
  432. return false;
  433. }
  434. AddStatementToBlocks(blocks, updateNode);
  435. break;
  436. }
  437. case TRule_sql_stmt_core::kAltSqlStmtCore10: {
  438. Ctx.BodyPart();
  439. auto deleteNode = Build(core.GetAlt_sql_stmt_core10().GetRule_delete_stmt1());
  440. if (!deleteNode) {
  441. return false;
  442. }
  443. blocks.emplace_back(deleteNode);
  444. break;
  445. }
  446. case TRule_sql_stmt_core::kAltSqlStmtCore11: {
  447. if (Ctx.ParallelModeCount > 0) {
  448. Error() << humanStatementName << " statement is not supported in parallel mode";
  449. return false;
  450. }
  451. Ctx.BodyPart();
  452. const auto& rule = core.GetAlt_sql_stmt_core11().GetRule_rollback_stmt1();
  453. Token(rule.GetToken1());
  454. blocks.emplace_back(BuildRollbackClusters(Ctx.Pos()));
  455. break;
  456. }
  457. case TRule_sql_stmt_core::kAltSqlStmtCore12:
  458. if (!DeclareStatement(core.GetAlt_sql_stmt_core12().GetRule_declare_stmt1())) {
  459. return false;
  460. }
  461. break;
  462. case TRule_sql_stmt_core::kAltSqlStmtCore13:
  463. if (!ImportStatement(core.GetAlt_sql_stmt_core13().GetRule_import_stmt1())) {
  464. return false;
  465. }
  466. break;
  467. case TRule_sql_stmt_core::kAltSqlStmtCore14:
  468. if (!ExportStatement(core.GetAlt_sql_stmt_core14().GetRule_export_stmt1())) {
  469. return false;
  470. }
  471. break;
  472. case TRule_sql_stmt_core::kAltSqlStmtCore15: {
  473. Ctx.BodyPart();
  474. const auto& rule = core.GetAlt_sql_stmt_core15().GetRule_alter_table_stmt1();
  475. const bool isTablestore = IS_TOKEN(rule.GetToken2().GetId(), TABLESTORE);
  476. TTableRef tr;
  477. if (!SimpleTableRefImpl(rule.GetRule_simple_table_ref3(), tr)) {
  478. return false;
  479. }
  480. TAlterTableParameters params;
  481. if (isTablestore) {
  482. params.TableType = ETableType::TableStore;
  483. }
  484. if (!AlterTableAction(rule.GetRule_alter_table_action4(), params)) {
  485. return false;
  486. }
  487. for (auto& block : rule.GetBlock5()) {
  488. if (!AlterTableAction(block.GetRule_alter_table_action2(), params)) {
  489. return false;
  490. }
  491. }
  492. AddStatementToBlocks(blocks, BuildAlterTable(Ctx.Pos(), tr, params, Ctx.Scoped));
  493. break;
  494. }
  495. case TRule_sql_stmt_core::kAltSqlStmtCore16: {
  496. // alter_external_table_stmt: ALTER EXTERNAL TABLE simple_table_ref alter_external_table_action (COMMA alter_external_table_action)*
  497. Ctx.BodyPart();
  498. const auto& rule = core.GetAlt_sql_stmt_core16().GetRule_alter_external_table_stmt1();
  499. TTableRef tr;
  500. if (!SimpleTableRefImpl(rule.GetRule_simple_table_ref4(), tr)) {
  501. return false;
  502. }
  503. TAlterTableParameters params;
  504. params.TableType = ETableType::ExternalTable;
  505. if (!AlterExternalTableAction(rule.GetRule_alter_external_table_action5(), params)) {
  506. return false;
  507. }
  508. for (auto& block : rule.GetBlock6()) {
  509. if (!AlterExternalTableAction(block.GetRule_alter_external_table_action2(), params)) {
  510. return false;
  511. }
  512. }
  513. AddStatementToBlocks(blocks, BuildAlterTable(Ctx.Pos(), tr, params, Ctx.Scoped));
  514. break;
  515. }
  516. case TRule_sql_stmt_core::kAltSqlStmtCore17: {
  517. Ctx.BodyPart();
  518. auto node = DoStatement(core.GetAlt_sql_stmt_core17().GetRule_do_stmt1(), false);
  519. if (!node) {
  520. return false;
  521. }
  522. blocks.push_back(node);
  523. break;
  524. }
  525. case TRule_sql_stmt_core::kAltSqlStmtCore18: {
  526. Ctx.BodyPart();
  527. TNodePtr lambda;
  528. TSymbolNameWithPos nameAndPos;
  529. const auto& stmt = core.GetAlt_sql_stmt_core18().GetRule_define_action_or_subquery_stmt1();
  530. const TString kind = to_lower(Ctx.Token(stmt.GetToken2()));
  531. YQL_ENSURE(kind == "action" || kind == "subquery");
  532. if (!DefineActionOrSubqueryStatement(stmt, nameAndPos, lambda)) {
  533. return false;
  534. }
  535. if (Ctx.CompactNamedExprs) {
  536. const auto ref = Ctx.MakeName("named" + kind + "node");
  537. blocks.push_back(BuildNamedExpr(lambda));
  538. blocks.back()->SetLabel(ref);
  539. lambda = BuildNamedExprReference(blocks.back(), ref, {});
  540. }
  541. PushNamedNode(nameAndPos.Pos, nameAndPos.Name, lambda);
  542. break;
  543. }
  544. case TRule_sql_stmt_core::kAltSqlStmtCore19: {
  545. Ctx.BodyPart();
  546. auto node = IfStatement(core.GetAlt_sql_stmt_core19().GetRule_if_stmt1());
  547. if (!node) {
  548. return false;
  549. }
  550. blocks.push_back(node);
  551. break;
  552. }
  553. case TRule_sql_stmt_core::kAltSqlStmtCore20: {
  554. Ctx.BodyPart();
  555. auto node = ForStatement(core.GetAlt_sql_stmt_core20().GetRule_for_stmt1());
  556. if (!node) {
  557. return false;
  558. }
  559. blocks.push_back(node);
  560. break;
  561. }
  562. case TRule_sql_stmt_core::kAltSqlStmtCore21: {
  563. if (Ctx.ParallelModeCount > 0) {
  564. Error() << humanStatementName << " statement is not supported in parallel mode";
  565. return false;
  566. }
  567. Ctx.BodyPart();
  568. TSqlValues values(Ctx, Mode);
  569. TPosition pos;
  570. auto source = values.Build(core.GetAlt_sql_stmt_core21().GetRule_values_stmt1(), pos, {}, TPosition());
  571. if (!source) {
  572. return false;
  573. }
  574. blocks.emplace_back(BuildSelectResult(pos, std::move(source),
  575. Mode != NSQLTranslation::ESqlMode::LIMITED_VIEW && Mode != NSQLTranslation::ESqlMode::SUBQUERY, Mode == NSQLTranslation::ESqlMode::SUBQUERY,
  576. Ctx.Scoped));
  577. break;
  578. }
  579. case TRule_sql_stmt_core::kAltSqlStmtCore22: {
  580. // create_user_stmt: CREATE USER role_name (user_option)*;
  581. Ctx.BodyPart();
  582. auto& node = core.GetAlt_sql_stmt_core22().GetRule_create_user_stmt1();
  583. Ctx.Token(node.GetToken1());
  584. const TPosition pos = Ctx.Pos();
  585. TString service = Ctx.Scoped->CurrService;
  586. TDeferredAtom cluster = Ctx.Scoped->CurrCluster;
  587. if (cluster.Empty()) {
  588. Error() << "USE statement is missing - no default cluster is selected";
  589. return false;
  590. }
  591. TDeferredAtom roleName;
  592. bool allowSystemRoles = false;
  593. if (!RoleNameClause(node.GetRule_role_name3(), roleName, allowSystemRoles)) {
  594. return false;
  595. }
  596. TMaybe<TUserParameters> createUserParams;
  597. const auto& options = node.GetBlock4();
  598. createUserParams.ConstructInPlace();
  599. std::vector<TRule_user_option> opts;
  600. opts.reserve(options.size());
  601. for (const auto& opt : options) {
  602. opts.push_back(opt.GetRule_user_option1());
  603. }
  604. bool isCreateUser = true;
  605. if (!UserParameters(opts, *createUserParams, isCreateUser)) {
  606. return false;
  607. }
  608. AddStatementToBlocks(blocks, BuildControlUser(pos, service, cluster, roleName, createUserParams, Ctx.Scoped, isCreateUser));
  609. break;
  610. }
  611. case TRule_sql_stmt_core::kAltSqlStmtCore23: {
  612. // alter_user_stmt: ALTER USER role_name (WITH? user_option+ | RENAME TO role_name);
  613. Ctx.BodyPart();
  614. auto& node = core.GetAlt_sql_stmt_core23().GetRule_alter_user_stmt1();
  615. Ctx.Token(node.GetToken1());
  616. const TPosition pos = Ctx.Pos();
  617. TString service = Ctx.Scoped->CurrService;
  618. TDeferredAtom cluster = Ctx.Scoped->CurrCluster;
  619. if (cluster.Empty()) {
  620. Error() << "USE statement is missing - no default cluster is selected";
  621. return false;
  622. }
  623. TDeferredAtom roleName;
  624. {
  625. bool allowSystemRoles = true;
  626. if (!RoleNameClause(node.GetRule_role_name3(), roleName, allowSystemRoles)) {
  627. return false;
  628. }
  629. }
  630. TNodePtr stmt;
  631. switch (node.GetBlock4().Alt_case()) {
  632. case TRule_alter_user_stmt_TBlock4::kAlt1: {
  633. TUserParameters alterUserParams;
  634. auto options = node.GetBlock4().GetAlt1().GetBlock2();
  635. std::vector<TRule_user_option> opts;
  636. opts.reserve(options.size());
  637. for (const auto& opt : options) {
  638. opts.push_back(opt.GetRule_user_option1());
  639. }
  640. bool isCreateUser = false;
  641. if (!UserParameters(opts, alterUserParams, isCreateUser)) {
  642. return false;
  643. }
  644. stmt = BuildControlUser(pos, service, cluster, roleName, alterUserParams, Ctx.Scoped, isCreateUser);
  645. break;
  646. }
  647. case TRule_alter_user_stmt_TBlock4::kAlt2: {
  648. TDeferredAtom tgtRoleName;
  649. bool allowSystemRoles = false;
  650. if (!RoleNameClause(node.GetBlock4().GetAlt2().GetRule_role_name3(), tgtRoleName, allowSystemRoles)) {
  651. return false;
  652. }
  653. stmt = BuildRenameUser(pos, service, cluster, roleName, tgtRoleName, Ctx.Scoped);
  654. break;
  655. }
  656. case TRule_alter_user_stmt_TBlock4::ALT_NOT_SET:
  657. Y_ABORT("You should change implementation according to grammar changes");
  658. }
  659. AddStatementToBlocks(blocks, stmt);
  660. break;
  661. }
  662. case TRule_sql_stmt_core::kAltSqlStmtCore24: {
  663. // create_group_stmt: CREATE GROUP role_name (WITH USER role_name (COMMA role_name)* COMMA?)?;
  664. Ctx.BodyPart();
  665. auto& node = core.GetAlt_sql_stmt_core24().GetRule_create_group_stmt1();
  666. Ctx.Token(node.GetToken1());
  667. const TPosition pos = Ctx.Pos();
  668. TString service = Ctx.Scoped->CurrService;
  669. TDeferredAtom cluster = Ctx.Scoped->CurrCluster;
  670. if (cluster.Empty()) {
  671. Error() << "USE statement is missing - no default cluster is selected";
  672. return false;
  673. }
  674. TDeferredAtom roleName;
  675. bool allowSystemRoles = false;
  676. if (!RoleNameClause(node.GetRule_role_name3(), roleName, allowSystemRoles)) {
  677. return false;
  678. }
  679. TCreateGroupParameters createGroupParams;
  680. if (node.HasBlock4()) {
  681. auto& addDropNode = node.GetBlock4();
  682. TVector<TDeferredAtom> roles;
  683. bool allowSystemRoles = false;
  684. createGroupParams.Roles.emplace_back();
  685. if (!RoleNameClause(addDropNode.GetRule_role_name3(), createGroupParams.Roles.back(), allowSystemRoles)) {
  686. return false;
  687. }
  688. for (auto& item : addDropNode.GetBlock4()) {
  689. createGroupParams.Roles.emplace_back();
  690. if (!RoleNameClause(item.GetRule_role_name2(), createGroupParams.Roles.back(), allowSystemRoles)) {
  691. return false;
  692. }
  693. }
  694. }
  695. AddStatementToBlocks(blocks, BuildCreateGroup(pos, service, cluster, roleName, createGroupParams, Ctx.Scoped));
  696. break;
  697. }
  698. case TRule_sql_stmt_core::kAltSqlStmtCore25: {
  699. // alter_group_stmt: ALTER GROUP role_name ((ADD|DROP) USER role_name (COMMA role_name)* COMMA? | RENAME TO role_name);
  700. Ctx.BodyPart();
  701. auto& node = core.GetAlt_sql_stmt_core25().GetRule_alter_group_stmt1();
  702. Ctx.Token(node.GetToken1());
  703. const TPosition pos = Ctx.Pos();
  704. TString service = Ctx.Scoped->CurrService;
  705. TDeferredAtom cluster = Ctx.Scoped->CurrCluster;
  706. if (cluster.Empty()) {
  707. Error() << "USE statement is missing - no default cluster is selected";
  708. return false;
  709. }
  710. TDeferredAtom roleName;
  711. {
  712. bool allowSystemRoles = true;
  713. if (!RoleNameClause(node.GetRule_role_name3(), roleName, allowSystemRoles)) {
  714. return false;
  715. }
  716. }
  717. TNodePtr stmt;
  718. switch (node.GetBlock4().Alt_case()) {
  719. case TRule_alter_group_stmt_TBlock4::kAlt1: {
  720. auto& addDropNode = node.GetBlock4().GetAlt1();
  721. const bool isDrop = IS_TOKEN(addDropNode.GetToken1().GetId(), DROP);
  722. TVector<TDeferredAtom> roles;
  723. bool allowSystemRoles = false;
  724. roles.emplace_back();
  725. if (!RoleNameClause(addDropNode.GetRule_role_name3(), roles.back(), allowSystemRoles)) {
  726. return false;
  727. }
  728. for (auto& item : addDropNode.GetBlock4()) {
  729. roles.emplace_back();
  730. if (!RoleNameClause(item.GetRule_role_name2(), roles.back(), allowSystemRoles)) {
  731. return false;
  732. }
  733. }
  734. stmt = BuildAlterGroup(pos, service, cluster, roleName, roles, isDrop, Ctx.Scoped);
  735. break;
  736. }
  737. case TRule_alter_group_stmt_TBlock4::kAlt2: {
  738. TDeferredAtom tgtRoleName;
  739. bool allowSystemRoles = false;
  740. if (!RoleNameClause(node.GetBlock4().GetAlt2().GetRule_role_name3(), tgtRoleName, allowSystemRoles)) {
  741. return false;
  742. }
  743. stmt = BuildRenameGroup(pos, service, cluster, roleName, tgtRoleName, Ctx.Scoped);
  744. break;
  745. }
  746. case TRule_alter_group_stmt_TBlock4::ALT_NOT_SET:
  747. Y_ABORT("You should change implementation according to grammar changes");
  748. }
  749. AddStatementToBlocks(blocks, stmt);
  750. break;
  751. }
  752. case TRule_sql_stmt_core::kAltSqlStmtCore26: {
  753. // drop_role_stmt: DROP (USER|GROUP) (IF EXISTS)? role_name (COMMA role_name)* COMMA?;
  754. Ctx.BodyPart();
  755. auto& node = core.GetAlt_sql_stmt_core26().GetRule_drop_role_stmt1();
  756. Ctx.Token(node.GetToken1());
  757. const TPosition pos = Ctx.Pos();
  758. TString service = Ctx.Scoped->CurrService;
  759. TDeferredAtom cluster = Ctx.Scoped->CurrCluster;
  760. if (cluster.Empty()) {
  761. Error() << "USE statement is missing - no default cluster is selected";
  762. return false;
  763. }
  764. const bool isUser = IS_TOKEN(node.GetToken2().GetId(), USER);
  765. bool missingOk = false;
  766. if (node.HasBlock3()) { // IF EXISTS
  767. missingOk = true;
  768. Y_DEBUG_ABORT_UNLESS(
  769. IS_TOKEN(node.GetBlock3().GetToken1().GetId(), IF) &&
  770. IS_TOKEN(node.GetBlock3().GetToken2().GetId(), EXISTS)
  771. );
  772. }
  773. TVector<TDeferredAtom> roles;
  774. bool allowSystemRoles = true;
  775. roles.emplace_back();
  776. if (!RoleNameClause(node.GetRule_role_name4(), roles.back(), allowSystemRoles)) {
  777. return false;
  778. }
  779. for (auto& item : node.GetBlock5()) {
  780. roles.emplace_back();
  781. if (!RoleNameClause(item.GetRule_role_name2(), roles.back(), allowSystemRoles)) {
  782. return false;
  783. }
  784. }
  785. AddStatementToBlocks(blocks, BuildDropRoles(pos, service, cluster, roles, isUser, missingOk, Ctx.Scoped));
  786. break;
  787. }
  788. case TRule_sql_stmt_core::kAltSqlStmtCore27: {
  789. // create_object_stmt: CREATE OBJECT (IF NOT EXISTS)? name (TYPE type [WITH k=v,...]);
  790. auto& node = core.GetAlt_sql_stmt_core27().GetRule_create_object_stmt1();
  791. TObjectOperatorContext context(Ctx.Scoped);
  792. if (node.GetRule_object_ref4().HasBlock1()) {
  793. if (!ClusterExpr(node.GetRule_object_ref4().GetBlock1().GetRule_cluster_expr1(),
  794. false, context.ServiceId, context.Cluster)) {
  795. return false;
  796. }
  797. }
  798. bool existingOk = false;
  799. if (node.HasBlock3()) { // IF NOT EXISTS
  800. existingOk = true;
  801. Y_DEBUG_ABORT_UNLESS(
  802. IS_TOKEN(node.GetBlock3().GetToken1().GetId(), IF) &&
  803. IS_TOKEN(node.GetBlock3().GetToken2().GetId(), NOT) &&
  804. IS_TOKEN(node.GetBlock3().GetToken3().GetId(), EXISTS)
  805. );
  806. }
  807. const TString& objectId = Id(node.GetRule_object_ref4().GetRule_id_or_at2(), *this).second;
  808. const TString& typeId = Id(node.GetRule_object_type_ref7().GetRule_an_id_or_type1(), *this);
  809. std::map<TString, TDeferredAtom> kv;
  810. if (node.HasBlock9()) {
  811. if (!ParseObjectFeatures(kv, node.GetBlock9().GetRule_create_object_features1().GetRule_object_features2())) {
  812. return false;
  813. }
  814. }
  815. AddStatementToBlocks(blocks, BuildCreateObjectOperation(Ctx.Pos(), objectId, typeId, existingOk, false, std::move(kv), context));
  816. break;
  817. }
  818. case TRule_sql_stmt_core::kAltSqlStmtCore28: {
  819. // alter_object_stmt: ALTER OBJECT name (TYPE type [SET k=v,...]);
  820. auto& node = core.GetAlt_sql_stmt_core28().GetRule_alter_object_stmt1();
  821. TObjectOperatorContext context(Ctx.Scoped);
  822. if (node.GetRule_object_ref3().HasBlock1()) {
  823. if (!ClusterExpr(node.GetRule_object_ref3().GetBlock1().GetRule_cluster_expr1(),
  824. false, context.ServiceId, context.Cluster)) {
  825. return false;
  826. }
  827. }
  828. const TString& objectId = Id(node.GetRule_object_ref3().GetRule_id_or_at2(), *this).second;
  829. const TString& typeId = Id(node.GetRule_object_type_ref6().GetRule_an_id_or_type1(), *this);
  830. std::map<TString, TDeferredAtom> kv;
  831. if (!ParseObjectFeatures(kv, node.GetRule_alter_object_features8().GetRule_object_features2())) {
  832. return false;
  833. }
  834. AddStatementToBlocks(blocks, BuildAlterObjectOperation(Ctx.Pos(), objectId, typeId, std::move(kv), std::set<TString>(), context));
  835. break;
  836. }
  837. case TRule_sql_stmt_core::kAltSqlStmtCore29: {
  838. // drop_object_stmt: DROP OBJECT (IF EXISTS)? name (TYPE type [WITH k=v,...]);
  839. auto& node = core.GetAlt_sql_stmt_core29().GetRule_drop_object_stmt1();
  840. TObjectOperatorContext context(Ctx.Scoped);
  841. if (node.GetRule_object_ref4().HasBlock1()) {
  842. if (!ClusterExpr(node.GetRule_object_ref4().GetBlock1().GetRule_cluster_expr1(),
  843. false, context.ServiceId, context.Cluster)) {
  844. return false;
  845. }
  846. }
  847. bool missingOk = false;
  848. if (node.HasBlock3()) { // IF EXISTS
  849. missingOk = true;
  850. Y_DEBUG_ABORT_UNLESS(
  851. IS_TOKEN(node.GetBlock3().GetToken1().GetId(), IF) &&
  852. IS_TOKEN(node.GetBlock3().GetToken2().GetId(), EXISTS)
  853. );
  854. }
  855. const TString& objectId = Id(node.GetRule_object_ref4().GetRule_id_or_at2(), *this).second;
  856. const TString& typeId = Id(node.GetRule_object_type_ref7().GetRule_an_id_or_type1(), *this);
  857. std::map<TString, TDeferredAtom> kv;
  858. if (node.HasBlock9()) {
  859. if (!ParseObjectFeatures(kv, node.GetBlock9().GetRule_drop_object_features1().GetRule_object_features2())) {
  860. return false;
  861. }
  862. }
  863. AddStatementToBlocks(blocks, BuildDropObjectOperation(Ctx.Pos(), objectId, typeId, missingOk, std::move(kv), context));
  864. break;
  865. }
  866. case TRule_sql_stmt_core::kAltSqlStmtCore30: {
  867. // create_external_data_source_stmt: CREATE (OR REPLACE)? EXTERNAL DATA SOURCE (IF NOT EXISTS)? name WITH (k=v,...);
  868. auto& node = core.GetAlt_sql_stmt_core30().GetRule_create_external_data_source_stmt1();
  869. TObjectOperatorContext context(Ctx.Scoped);
  870. if (node.GetRule_object_ref7().HasBlock1()) {
  871. if (!ClusterExpr(node.GetRule_object_ref7().GetBlock1().GetRule_cluster_expr1(),
  872. false, context.ServiceId, context.Cluster)) {
  873. return false;
  874. }
  875. }
  876. bool replaceIfExists = false;
  877. if (node.HasBlock2()) { // OR REPLACE
  878. replaceIfExists = true;
  879. Y_DEBUG_ABORT_UNLESS(
  880. IS_TOKEN(node.GetBlock2().GetToken1().GetId(), OR) &&
  881. IS_TOKEN(node.GetBlock2().GetToken2().GetId(), REPLACE)
  882. );
  883. }
  884. bool existingOk = false;
  885. if (node.HasBlock6()) { // IF NOT EXISTS
  886. existingOk = true;
  887. Y_DEBUG_ABORT_UNLESS(
  888. IS_TOKEN(node.GetBlock6().GetToken1().GetId(), IF) &&
  889. IS_TOKEN(node.GetBlock6().GetToken2().GetId(), NOT) &&
  890. IS_TOKEN(node.GetBlock6().GetToken3().GetId(), EXISTS)
  891. );
  892. }
  893. const TString& objectId = Id(node.GetRule_object_ref7().GetRule_id_or_at2(), *this).second;
  894. std::map<TString, TDeferredAtom> kv;
  895. if (!ParseExternalDataSourceSettings(kv, node.GetRule_with_table_settings8())) {
  896. return false;
  897. }
  898. AddStatementToBlocks(blocks, BuildCreateObjectOperation(Ctx.Pos(), BuildTablePath(Ctx.GetPrefixPath(context.ServiceId, context.Cluster), objectId), "EXTERNAL_DATA_SOURCE", existingOk, replaceIfExists, std::move(kv), context));
  899. break;
  900. }
  901. case TRule_sql_stmt_core::kAltSqlStmtCore31: {
  902. // alter_external_data_source_stmt: ALTER EXTERNAL DATA SOURCE object_ref alter_external_data_source_action (COMMA alter_external_data_source_action)*
  903. Ctx.BodyPart();
  904. const auto& node = core.GetAlt_sql_stmt_core31().GetRule_alter_external_data_source_stmt1();
  905. TObjectOperatorContext context(Ctx.Scoped);
  906. if (node.GetRule_object_ref5().HasBlock1()) {
  907. if (!ClusterExpr(node.GetRule_object_ref5().GetBlock1().GetRule_cluster_expr1(),
  908. false, context.ServiceId, context.Cluster)) {
  909. return false;
  910. }
  911. }
  912. const TString& objectId = Id(node.GetRule_object_ref5().GetRule_id_or_at2(), *this).second;
  913. std::map<TString, TDeferredAtom> kv;
  914. std::set<TString> toReset;
  915. if (!ParseExternalDataSourceSettings(kv, toReset, node.GetRule_alter_external_data_source_action6())) {
  916. return false;
  917. }
  918. for (const auto& action : node.GetBlock7()) {
  919. if (!ParseExternalDataSourceSettings(kv, toReset, action.GetRule_alter_external_data_source_action2())) {
  920. return false;
  921. }
  922. }
  923. AddStatementToBlocks(blocks, BuildAlterObjectOperation(Ctx.Pos(), objectId, "EXTERNAL_DATA_SOURCE", std::move(kv), std::move(toReset), context));
  924. break;
  925. }
  926. case TRule_sql_stmt_core::kAltSqlStmtCore32: {
  927. // drop_external_data_source_stmt: DROP EXTERNAL DATA SOURCE (IF EXISTS)? name;
  928. auto& node = core.GetAlt_sql_stmt_core32().GetRule_drop_external_data_source_stmt1();
  929. TObjectOperatorContext context(Ctx.Scoped);
  930. if (node.GetRule_object_ref6().HasBlock1()) {
  931. if (!ClusterExpr(node.GetRule_object_ref6().GetBlock1().GetRule_cluster_expr1(),
  932. false, context.ServiceId, context.Cluster)) {
  933. return false;
  934. }
  935. }
  936. bool missingOk = false;
  937. if (node.HasBlock5()) { // IF EXISTS
  938. missingOk = true;
  939. Y_DEBUG_ABORT_UNLESS(
  940. IS_TOKEN(node.GetBlock5().GetToken1().GetId(), IF) &&
  941. IS_TOKEN(node.GetBlock5().GetToken2().GetId(), EXISTS)
  942. );
  943. }
  944. const TString& objectId = Id(node.GetRule_object_ref6().GetRule_id_or_at2(), *this).second;
  945. AddStatementToBlocks(blocks, BuildDropObjectOperation(Ctx.Pos(), BuildTablePath(Ctx.GetPrefixPath(context.ServiceId, context.Cluster), objectId), "EXTERNAL_DATA_SOURCE", missingOk, {}, context));
  946. break;
  947. }
  948. case TRule_sql_stmt_core::kAltSqlStmtCore33: {
  949. // create_replication_stmt: CREATE ASYNC REPLICATION
  950. auto& node = core.GetAlt_sql_stmt_core33().GetRule_create_replication_stmt1();
  951. TObjectOperatorContext context(Ctx.Scoped);
  952. if (node.GetRule_object_ref4().HasBlock1()) {
  953. const auto& cluster = node.GetRule_object_ref4().GetBlock1().GetRule_cluster_expr1();
  954. if (!ClusterExpr(cluster, false, context.ServiceId, context.Cluster)) {
  955. return false;
  956. }
  957. }
  958. auto prefixPath = Ctx.GetPrefixPath(context.ServiceId, context.Cluster);
  959. std::vector<std::pair<TString, TString>> targets;
  960. if (!AsyncReplicationTarget(targets, prefixPath, node.GetRule_replication_target6(), *this)) {
  961. return false;
  962. }
  963. for (auto& block : node.GetBlock7()) {
  964. if (!AsyncReplicationTarget(targets, prefixPath, block.GetRule_replication_target2(), *this)) {
  965. return false;
  966. }
  967. }
  968. std::map<TString, TNodePtr> settings;
  969. TSqlExpression expr(Ctx, Mode);
  970. if (!AsyncReplicationSettings(settings, node.GetRule_replication_settings10(), expr, true)) {
  971. return false;
  972. }
  973. const TString id = Id(node.GetRule_object_ref4().GetRule_id_or_at2(), *this).second;
  974. AddStatementToBlocks(blocks, BuildCreateAsyncReplication(Ctx.Pos(), BuildTablePath(prefixPath, id),
  975. std::move(targets), std::move(settings), context));
  976. break;
  977. }
  978. case TRule_sql_stmt_core::kAltSqlStmtCore34: {
  979. // drop_replication_stmt: DROP ASYNC REPLICATION
  980. auto& node = core.GetAlt_sql_stmt_core34().GetRule_drop_replication_stmt1();
  981. TObjectOperatorContext context(Ctx.Scoped);
  982. if (node.GetRule_object_ref4().HasBlock1()) {
  983. const auto& cluster = node.GetRule_object_ref4().GetBlock1().GetRule_cluster_expr1();
  984. if (!ClusterExpr(cluster, false, context.ServiceId, context.Cluster)) {
  985. return false;
  986. }
  987. }
  988. const TString id = Id(node.GetRule_object_ref4().GetRule_id_or_at2(), *this).second;
  989. AddStatementToBlocks(blocks, BuildDropAsyncReplication(Ctx.Pos(),
  990. BuildTablePath(Ctx.GetPrefixPath(context.ServiceId, context.Cluster), id),
  991. node.HasBlock5(), context));
  992. break;
  993. }
  994. case TRule_sql_stmt_core::kAltSqlStmtCore35: {
  995. Ctx.BodyPart();
  996. // create_topic_stmt: CREATE TOPIC (IF NOT EXISTS)? topic1 (CONSUMER ...)? [WITH (opt1 = val1, ...]?
  997. auto& rule = core.GetAlt_sql_stmt_core35().GetRule_create_topic_stmt1();
  998. TTopicRef tr;
  999. if (!TopicRefImpl(rule.GetRule_topic_ref4(), tr)) {
  1000. return false;
  1001. }
  1002. bool existingOk = false;
  1003. if (rule.HasBlock3()) { // if not exists
  1004. existingOk = true;
  1005. }
  1006. TCreateTopicParameters params;
  1007. params.ExistingOk = existingOk;
  1008. if (rule.HasBlock5()) { //create_topic_entry (consumers)
  1009. auto& entries = rule.GetBlock5().GetRule_create_topic_entries1();
  1010. auto& firstEntry = entries.GetRule_create_topic_entry2();
  1011. if (!CreateTopicEntry(firstEntry, params)) {
  1012. return false;
  1013. }
  1014. const auto& list = entries.GetBlock3();
  1015. for (auto& node : list) {
  1016. if (!CreateTopicEntry(node.GetRule_create_topic_entry2(), params)) {
  1017. return false;
  1018. }
  1019. }
  1020. }
  1021. if (rule.HasBlock6()) { // with_topic_settings
  1022. auto& topic_settings_node = rule.GetBlock6().GetRule_with_topic_settings1().GetRule_topic_settings3();
  1023. CreateTopicSettings(topic_settings_node, params.TopicSettings);
  1024. }
  1025. AddStatementToBlocks(blocks, BuildCreateTopic(Ctx.Pos(), tr, params, Ctx.Scoped));
  1026. break;
  1027. }
  1028. case TRule_sql_stmt_core::kAltSqlStmtCore36: {
  1029. // alter_topic_stmt: ALTER TOPIC topic_ref alter_topic_action (COMMA alter_topic_action)*;
  1030. // alter_topic_stmt: ALTER TOPIC IF EXISTS topic_ref alter_topic_action (COMMA alter_topic_action)*;
  1031. Ctx.BodyPart();
  1032. auto& rule = core.GetAlt_sql_stmt_core36().GetRule_alter_topic_stmt1();
  1033. TTopicRef tr;
  1034. bool missingOk = false;
  1035. if (rule.HasBlock3()) { // IF EXISTS
  1036. missingOk = true;
  1037. }
  1038. if (!TopicRefImpl(rule.GetRule_topic_ref4(), tr)) {
  1039. return false;
  1040. }
  1041. TAlterTopicParameters params;
  1042. params.MissingOk = missingOk;
  1043. auto& firstEntry = rule.GetRule_alter_topic_action5();
  1044. if (!AlterTopicAction(firstEntry, params)) {
  1045. return false;
  1046. }
  1047. const auto& list = rule.GetBlock6();
  1048. for (auto& node : list) {
  1049. if (!AlterTopicAction(node.GetRule_alter_topic_action2(), params)) {
  1050. return false;
  1051. }
  1052. }
  1053. AddStatementToBlocks(blocks, BuildAlterTopic(Ctx.Pos(), tr, params, Ctx.Scoped));
  1054. break;
  1055. }
  1056. case TRule_sql_stmt_core::kAltSqlStmtCore37: {
  1057. // drop_topic_stmt: DROP TOPIC (IF EXISTS)? topic_ref;
  1058. Ctx.BodyPart();
  1059. const auto& rule = core.GetAlt_sql_stmt_core37().GetRule_drop_topic_stmt1();
  1060. TDropTopicParameters params;
  1061. if (rule.HasBlock3()) { // IF EXISTS
  1062. params.MissingOk = true;
  1063. } else {
  1064. params.MissingOk = false;
  1065. }
  1066. TTopicRef tr;
  1067. if (!TopicRefImpl(rule.GetRule_topic_ref4(), tr)) {
  1068. return false;
  1069. }
  1070. AddStatementToBlocks(blocks, BuildDropTopic(Ctx.Pos(), tr, params, Ctx.Scoped));
  1071. break;
  1072. }
  1073. case TRule_sql_stmt_core::kAltSqlStmtCore38: {
  1074. // GRANT permission_name_target ON an_id_schema (COMMA an_id_schema)* TO role_name (COMMA role_name)* COMMA? (WITH GRANT OPTION)?;
  1075. Ctx.BodyPart();
  1076. auto& node = core.GetAlt_sql_stmt_core38().GetRule_grant_permissions_stmt1();
  1077. Ctx.Token(node.GetToken1());
  1078. const TPosition pos = Ctx.Pos();
  1079. TString service = Ctx.Scoped->CurrService;
  1080. TDeferredAtom cluster = Ctx.Scoped->CurrCluster;
  1081. if (cluster.Empty()) {
  1082. Error() << "USE statement is missing - no default cluster is selected";
  1083. return false;
  1084. }
  1085. TVector<TDeferredAtom> permissions;
  1086. if (!PermissionNameClause(node.GetRule_permission_name_target2(), permissions, node.has_block10())) {
  1087. return false;
  1088. }
  1089. TVector<TDeferredAtom> schemaPaths;
  1090. schemaPaths.emplace_back(Ctx.Pos(), Id(node.GetRule_an_id_schema4(), *this));
  1091. for (const auto& item : node.GetBlock5()) {
  1092. schemaPaths.emplace_back(Ctx.Pos(), Id(item.GetRule_an_id_schema2(), *this));
  1093. }
  1094. TVector<TDeferredAtom> roleNames;
  1095. const bool allowSystemRoles = false;
  1096. roleNames.emplace_back();
  1097. if (!RoleNameClause(node.GetRule_role_name7(), roleNames.back(), allowSystemRoles)) {
  1098. return false;
  1099. }
  1100. for (const auto& item : node.GetBlock8()) {
  1101. roleNames.emplace_back();
  1102. if (!RoleNameClause(item.GetRule_role_name2(), roleNames.back(), allowSystemRoles)) {
  1103. return false;
  1104. }
  1105. }
  1106. AddStatementToBlocks(blocks, BuildGrantPermissions(pos, service, cluster, permissions, schemaPaths, roleNames, Ctx.Scoped));
  1107. break;
  1108. }
  1109. case TRule_sql_stmt_core::kAltSqlStmtCore39:
  1110. {
  1111. // REVOKE (GRANT OPTION FOR)? permission_name_target ON an_id_schema (COMMA an_id_schema)* FROM role_name (COMMA role_name)*;
  1112. Ctx.BodyPart();
  1113. auto& node = core.GetAlt_sql_stmt_core39().GetRule_revoke_permissions_stmt1();
  1114. Ctx.Token(node.GetToken1());
  1115. const TPosition pos = Ctx.Pos();
  1116. TString service = Ctx.Scoped->CurrService;
  1117. TDeferredAtom cluster = Ctx.Scoped->CurrCluster;
  1118. if (cluster.Empty()) {
  1119. Error() << "USE statement is missing - no default cluster is selected";
  1120. return false;
  1121. }
  1122. TVector<TDeferredAtom> permissions;
  1123. if (!PermissionNameClause(node.GetRule_permission_name_target3(), permissions, node.HasBlock2())) {
  1124. return false;
  1125. }
  1126. TVector<TDeferredAtom> schemaPaths;
  1127. schemaPaths.emplace_back(Ctx.Pos(), Id(node.GetRule_an_id_schema5(), *this));
  1128. for (const auto& item : node.GetBlock6()) {
  1129. schemaPaths.emplace_back(Ctx.Pos(), Id(item.GetRule_an_id_schema2(), *this));
  1130. }
  1131. TVector<TDeferredAtom> roleNames;
  1132. const bool allowSystemRoles = false;
  1133. roleNames.emplace_back();
  1134. if (!RoleNameClause(node.GetRule_role_name8(), roleNames.back(), allowSystemRoles)) {
  1135. return false;
  1136. }
  1137. for (const auto& item : node.GetBlock9()) {
  1138. roleNames.emplace_back();
  1139. if (!RoleNameClause(item.GetRule_role_name2(), roleNames.back(), allowSystemRoles)) {
  1140. return false;
  1141. }
  1142. }
  1143. AddStatementToBlocks(blocks, BuildRevokePermissions(pos, service, cluster, permissions, schemaPaths, roleNames, Ctx.Scoped));
  1144. break;
  1145. }
  1146. case TRule_sql_stmt_core::kAltSqlStmtCore40:
  1147. {
  1148. // ALTER TABLESTORE object_ref alter_table_store_action (COMMA alter_table_store_action)*;
  1149. auto& node = core.GetAlt_sql_stmt_core40().GetRule_alter_table_store_stmt1();
  1150. TObjectOperatorContext context(Ctx.Scoped);
  1151. if (node.GetRule_object_ref3().HasBlock1()) {
  1152. if (!ClusterExpr(node.GetRule_object_ref3().GetBlock1().GetRule_cluster_expr1(),
  1153. false, context.ServiceId, context.Cluster)) {
  1154. return false;
  1155. }
  1156. }
  1157. const TString& objectId = Id(node.GetRule_object_ref3().GetRule_id_or_at2(), *this).second;
  1158. const TString& typeId = "TABLESTORE";
  1159. std::map<TString, TDeferredAtom> kv;
  1160. if (!ParseTableStoreFeatures(kv, node.GetRule_alter_table_store_action4())) {
  1161. return false;
  1162. }
  1163. AddStatementToBlocks(blocks, BuildAlterObjectOperation(Ctx.Pos(), objectId, typeId, std::move(kv), std::set<TString>(), context));
  1164. break;
  1165. }
  1166. case TRule_sql_stmt_core::kAltSqlStmtCore41:
  1167. {
  1168. // create_object_stmt: UPSERT OBJECT name (TYPE type [WITH k=v,...]);
  1169. auto& node = core.GetAlt_sql_stmt_core41().GetRule_upsert_object_stmt1();
  1170. TObjectOperatorContext context(Ctx.Scoped);
  1171. if (node.GetRule_object_ref3().HasBlock1()) {
  1172. if (!ClusterExpr(node.GetRule_object_ref3().GetBlock1().GetRule_cluster_expr1(),
  1173. false, context.ServiceId, context.Cluster)) {
  1174. return false;
  1175. }
  1176. }
  1177. const TString& objectId = Id(node.GetRule_object_ref3().GetRule_id_or_at2(), *this).second;
  1178. const TString& typeId = Id(node.GetRule_object_type_ref6().GetRule_an_id_or_type1(), *this);
  1179. std::map<TString, TDeferredAtom> kv;
  1180. if (node.HasBlock8()) {
  1181. if (!ParseObjectFeatures(kv, node.GetBlock8().GetRule_create_object_features1().GetRule_object_features2())) {
  1182. return false;
  1183. }
  1184. }
  1185. AddStatementToBlocks(blocks, BuildUpsertObjectOperation(Ctx.Pos(), objectId, typeId, std::move(kv), context));
  1186. break;
  1187. }
  1188. case TRule_sql_stmt_core::kAltSqlStmtCore42: {
  1189. // create_view_stmt: CREATE VIEW (IF NOT EXISTS)? name (WITH (k = v, ...))? AS select_stmt;
  1190. auto& node = core.GetAlt_sql_stmt_core42().GetRule_create_view_stmt1();
  1191. TObjectOperatorContext context(Ctx.Scoped);
  1192. if (node.GetRule_object_ref4().HasBlock1()) {
  1193. if (!ClusterExpr(node.GetRule_object_ref4().GetBlock1().GetRule_cluster_expr1(),
  1194. false,
  1195. context.ServiceId,
  1196. context.Cluster)) {
  1197. return false;
  1198. }
  1199. }
  1200. const bool existingOk = node.HasBlock3();
  1201. std::map<TString, TDeferredAtom> features;
  1202. if (node.HasBlock5()) {
  1203. if (!ParseObjectFeatures(features, node.GetBlock5().GetRule_create_object_features1().GetRule_object_features2())) {
  1204. return false;
  1205. }
  1206. }
  1207. if (!ParseViewQuery(features, node.GetRule_select_stmt7())) {
  1208. return false;
  1209. }
  1210. const TString objectId = Id(node.GetRule_object_ref4().GetRule_id_or_at2(), *this).second;
  1211. constexpr const char* TypeId = "VIEW";
  1212. AddStatementToBlocks(blocks,
  1213. BuildCreateObjectOperation(Ctx.Pos(),
  1214. BuildTablePath(Ctx.GetPrefixPath(context.ServiceId, context.Cluster), objectId),
  1215. TypeId,
  1216. existingOk,
  1217. false,
  1218. std::move(features),
  1219. context));
  1220. break;
  1221. }
  1222. case TRule_sql_stmt_core::kAltSqlStmtCore43: {
  1223. // drop_view_stmt: DROP VIEW (IF EXISTS)? name;
  1224. auto& node = core.GetAlt_sql_stmt_core43().GetRule_drop_view_stmt1();
  1225. TObjectOperatorContext context(Ctx.Scoped);
  1226. if (node.GetRule_object_ref4().HasBlock1()) {
  1227. if (!ClusterExpr(node.GetRule_object_ref4().GetBlock1().GetRule_cluster_expr1(),
  1228. false,
  1229. context.ServiceId,
  1230. context.Cluster)) {
  1231. return false;
  1232. }
  1233. }
  1234. const bool missingOk = node.HasBlock3();
  1235. const TString objectId = Id(node.GetRule_object_ref4().GetRule_id_or_at2(), *this).second;
  1236. constexpr const char* TypeId = "VIEW";
  1237. AddStatementToBlocks(blocks,
  1238. BuildDropObjectOperation(Ctx.Pos(),
  1239. BuildTablePath(Ctx.GetPrefixPath(context.ServiceId, context.Cluster), objectId),
  1240. TypeId,
  1241. missingOk,
  1242. {},
  1243. context));
  1244. break;
  1245. }
  1246. case TRule_sql_stmt_core::kAltSqlStmtCore44: {
  1247. // alter_replication_stmt: ALTER ASYNC REPLICATION
  1248. auto& node = core.GetAlt_sql_stmt_core44().GetRule_alter_replication_stmt1();
  1249. TObjectOperatorContext context(Ctx.Scoped);
  1250. if (node.GetRule_object_ref4().HasBlock1()) {
  1251. const auto& cluster = node.GetRule_object_ref4().GetBlock1().GetRule_cluster_expr1();
  1252. if (!ClusterExpr(cluster, false, context.ServiceId, context.Cluster)) {
  1253. return false;
  1254. }
  1255. }
  1256. std::map<TString, TNodePtr> settings;
  1257. TSqlExpression expr(Ctx, Mode);
  1258. if (!AsyncReplicationAlterAction(settings, node.GetRule_alter_replication_action5(), expr)) {
  1259. return false;
  1260. }
  1261. for (auto& block : node.GetBlock6()) {
  1262. if (!AsyncReplicationAlterAction(settings, block.GetRule_alter_replication_action2(), expr)) {
  1263. return false;
  1264. }
  1265. }
  1266. const TString id = Id(node.GetRule_object_ref4().GetRule_id_or_at2(), *this).second;
  1267. AddStatementToBlocks(blocks, BuildAlterAsyncReplication(Ctx.Pos(),
  1268. BuildTablePath(Ctx.GetPrefixPath(context.ServiceId, context.Cluster), id),
  1269. std::move(settings), context));
  1270. break;
  1271. }
  1272. case TRule_sql_stmt_core::kAltSqlStmtCore45: {
  1273. // create_resource_pool_stmt: CREATE RESOURCE POOL name WITH (k=v,...);
  1274. auto& node = core.GetAlt_sql_stmt_core45().GetRule_create_resource_pool_stmt1();
  1275. TObjectOperatorContext context(Ctx.Scoped);
  1276. if (node.GetRule_object_ref4().HasBlock1()) {
  1277. if (!ClusterExpr(node.GetRule_object_ref4().GetBlock1().GetRule_cluster_expr1(),
  1278. false, context.ServiceId, context.Cluster)) {
  1279. return false;
  1280. }
  1281. }
  1282. const TString& objectId = Id(node.GetRule_object_ref4().GetRule_id_or_at2(), *this).second;
  1283. std::map<TString, TDeferredAtom> kv;
  1284. if (!ParseResourcePoolSettings(kv, node.GetRule_with_table_settings5())) {
  1285. return false;
  1286. }
  1287. AddStatementToBlocks(blocks, BuildCreateObjectOperation(Ctx.Pos(), objectId, "RESOURCE_POOL", false, false, std::move(kv), context));
  1288. break;
  1289. }
  1290. case TRule_sql_stmt_core::kAltSqlStmtCore46: {
  1291. // alter_resource_pool_stmt: ALTER RESOURCE POOL object_ref alter_resource_pool_action (COMMA alter_external_data_source_action)*
  1292. Ctx.BodyPart();
  1293. const auto& node = core.GetAlt_sql_stmt_core46().GetRule_alter_resource_pool_stmt1();
  1294. TObjectOperatorContext context(Ctx.Scoped);
  1295. if (node.GetRule_object_ref4().HasBlock1()) {
  1296. if (!ClusterExpr(node.GetRule_object_ref4().GetBlock1().GetRule_cluster_expr1(),
  1297. false, context.ServiceId, context.Cluster)) {
  1298. return false;
  1299. }
  1300. }
  1301. const TString& objectId = Id(node.GetRule_object_ref4().GetRule_id_or_at2(), *this).second;
  1302. std::map<TString, TDeferredAtom> kv;
  1303. std::set<TString> toReset;
  1304. if (!ParseResourcePoolSettings(kv, toReset, node.GetRule_alter_resource_pool_action5())) {
  1305. return false;
  1306. }
  1307. for (const auto& action : node.GetBlock6()) {
  1308. if (!ParseResourcePoolSettings(kv, toReset, action.GetRule_alter_resource_pool_action2())) {
  1309. return false;
  1310. }
  1311. }
  1312. AddStatementToBlocks(blocks, BuildAlterObjectOperation(Ctx.Pos(), objectId, "RESOURCE_POOL", std::move(kv), std::move(toReset), context));
  1313. break;
  1314. }
  1315. case TRule_sql_stmt_core::kAltSqlStmtCore47: {
  1316. // drop_resource_pool_stmt: DROP RESOURCE POOL name;
  1317. auto& node = core.GetAlt_sql_stmt_core47().GetRule_drop_resource_pool_stmt1();
  1318. TObjectOperatorContext context(Ctx.Scoped);
  1319. if (node.GetRule_object_ref4().HasBlock1()) {
  1320. if (!ClusterExpr(node.GetRule_object_ref4().GetBlock1().GetRule_cluster_expr1(),
  1321. false, context.ServiceId, context.Cluster)) {
  1322. return false;
  1323. }
  1324. }
  1325. const TString& objectId = Id(node.GetRule_object_ref4().GetRule_id_or_at2(), *this).second;
  1326. AddStatementToBlocks(blocks, BuildDropObjectOperation(Ctx.Pos(), objectId, "RESOURCE_POOL", false, {}, context));
  1327. break;
  1328. }
  1329. case TRule_sql_stmt_core::kAltSqlStmtCore48: {
  1330. // create_backup_collection_stmt: CREATE BACKUP COLLECTION name WITH (k=v,...);
  1331. auto& node = core.GetAlt_sql_stmt_core48().GetRule_create_backup_collection_stmt1();
  1332. TObjectOperatorContext context(Ctx.Scoped);
  1333. if (node.GetRule_backup_collection2().GetRule_object_ref3().HasBlock1()) {
  1334. if (!ClusterExpr(node.GetRule_backup_collection2().GetRule_object_ref3().GetBlock1().GetRule_cluster_expr1(),
  1335. false,
  1336. context.ServiceId,
  1337. context.Cluster)) {
  1338. return false;
  1339. }
  1340. }
  1341. std::map<TString, TDeferredAtom> kv;
  1342. if (!ParseBackupCollectionSettings(kv, node.GetRule_backup_collection_settings6())) {
  1343. return false;
  1344. }
  1345. bool database = false;
  1346. TVector<TDeferredAtom> tables;
  1347. if (node.HasBlock3()) {
  1348. database = node.GetBlock3().GetRule_create_backup_collection_entries1().has_alt_create_backup_collection_entries1();
  1349. if (node.GetBlock3().GetRule_create_backup_collection_entries1().has_alt_create_backup_collection_entries2()) {
  1350. if (!ParseBackupCollectionTables(
  1351. tables,
  1352. node
  1353. .GetBlock3()
  1354. .GetRule_create_backup_collection_entries1()
  1355. .alt_create_backup_collection_entries2()
  1356. .GetRule_create_backup_collection_entries_many1()
  1357. .GetRule_table_list2()))
  1358. {
  1359. return false;
  1360. }
  1361. }
  1362. }
  1363. const TString& objectId = Id(node.GetRule_backup_collection2().GetRule_object_ref3().GetRule_id_or_at2(), *this).second;
  1364. AddStatementToBlocks(blocks,
  1365. BuildCreateBackupCollection(Ctx.Pos(),
  1366. TString(Ctx.GetPrefixPath(context.ServiceId, context.Cluster)),
  1367. objectId,
  1368. TCreateBackupCollectionParameters {
  1369. .Settings = std::move(kv),
  1370. .Database = database,
  1371. .Tables = tables,
  1372. .ExistingOk = false,
  1373. },
  1374. context));
  1375. break;
  1376. }
  1377. case TRule_sql_stmt_core::kAltSqlStmtCore49: {
  1378. // alter_backup_collection_stmt: ALTER BACKUP COLLECTION name alter_backup_collection_action (COMMA alter_backup_collection_action)*;
  1379. auto& node = core.GetAlt_sql_stmt_core49().GetRule_alter_backup_collection_stmt1();
  1380. TObjectOperatorContext context(Ctx.Scoped);
  1381. if (node.GetRule_backup_collection2().GetRule_object_ref3().HasBlock1()) {
  1382. if (!ClusterExpr(node.GetRule_backup_collection2().GetRule_object_ref3().GetBlock1().GetRule_cluster_expr1(),
  1383. false,
  1384. context.ServiceId,
  1385. context.Cluster)) {
  1386. return false;
  1387. }
  1388. }
  1389. std::map<TString, TDeferredAtom> kv;
  1390. std::set<TString> toReset;
  1391. bool addDatabase = false;
  1392. bool dropDatabase = false;
  1393. TVector<TDeferredAtom> addTables;
  1394. TVector<TDeferredAtom> removeTables;
  1395. switch (node.GetBlock3().Alt_case()) {
  1396. case TRule_alter_backup_collection_stmt_TBlock3::kAlt1: {
  1397. if (!ParseBackupCollectionSettings(kv, toReset, node.GetBlock3().GetAlt1().GetRule_alter_backup_collection_actions1())) {
  1398. return false;
  1399. }
  1400. break;
  1401. }
  1402. case TRule_alter_backup_collection_stmt_TBlock3::kAlt2: {
  1403. if (!ParseBackupCollectionEntries(
  1404. addDatabase,
  1405. dropDatabase,
  1406. addTables,
  1407. removeTables,
  1408. node.GetBlock3().GetAlt2().GetRule_alter_backup_collection_entries1()))
  1409. {
  1410. return false;
  1411. }
  1412. break;
  1413. }
  1414. case TRule_alter_backup_collection_stmt_TBlock3::ALT_NOT_SET: {} // do nothing
  1415. }
  1416. auto database = addDatabase ?
  1417. TAlterBackupCollectionParameters::EDatabase::Add :
  1418. dropDatabase ?
  1419. TAlterBackupCollectionParameters::EDatabase::Drop :
  1420. TAlterBackupCollectionParameters::EDatabase::Unchanged;
  1421. const TString& objectId = Id(node.GetRule_backup_collection2().GetRule_object_ref3().GetRule_id_or_at2(), *this).second;
  1422. AddStatementToBlocks(blocks,
  1423. BuildAlterBackupCollection(Ctx.Pos(),
  1424. TString(Ctx.GetPrefixPath(context.ServiceId, context.Cluster)),
  1425. objectId,
  1426. TAlterBackupCollectionParameters {
  1427. .Settings = std::move(kv),
  1428. .SettingsToReset = std::move(toReset),
  1429. .Database = database,
  1430. .TablesToAdd = addTables,
  1431. .TablesToDrop = removeTables,
  1432. .MissingOk = false,
  1433. },
  1434. context));
  1435. break;
  1436. }
  1437. case TRule_sql_stmt_core::kAltSqlStmtCore50: {
  1438. // drop_backup_collection_stmt: DROP BACKUP COLLECTION name;
  1439. auto& node = core.GetAlt_sql_stmt_core50().GetRule_drop_backup_collection_stmt1();
  1440. TObjectOperatorContext context(Ctx.Scoped);
  1441. if (node.GetRule_backup_collection2().GetRule_object_ref3().HasBlock1()) {
  1442. if (!ClusterExpr(node.GetRule_backup_collection2().GetRule_object_ref3().GetBlock1().GetRule_cluster_expr1(),
  1443. false,
  1444. context.ServiceId,
  1445. context.Cluster)) {
  1446. return false;
  1447. }
  1448. }
  1449. const TString& objectId = Id(node.GetRule_backup_collection2().GetRule_object_ref3().GetRule_id_or_at2(), *this).second;
  1450. AddStatementToBlocks(blocks,
  1451. BuildDropBackupCollection(Ctx.Pos(),
  1452. TString(Ctx.GetPrefixPath(context.ServiceId, context.Cluster)),
  1453. objectId,
  1454. TDropBackupCollectionParameters {
  1455. .MissingOk = false,
  1456. },
  1457. context));
  1458. break;
  1459. }
  1460. case TRule_sql_stmt_core::kAltSqlStmtCore51: {
  1461. // analyze_stmt: ANALYZE table_ref
  1462. Ctx.BodyPart();
  1463. const auto& rule = core.GetAlt_sql_stmt_core51().GetRule_analyze_stmt1();
  1464. if (!rule.GetRule_analyze_table_list2().GetBlock2().empty()) {
  1465. Error() << "ANALYZE with multitables hasn't been implemented yet";
  1466. return false;
  1467. }
  1468. auto analyzeTable = rule.GetRule_analyze_table_list2().GetRule_analyze_table1();
  1469. TVector<TString> columns;
  1470. if (analyzeTable.HasBlock2()) {
  1471. auto columnsNode =
  1472. analyzeTable.GetBlock2().GetRule_column_list2();
  1473. if (columnsNode.HasRule_column_name1()) {
  1474. columns.push_back(Id(columnsNode.GetRule_column_name1().GetRule_an_id2(), *this));
  1475. for (const auto& columnNode: columnsNode.GetBlock2()) {
  1476. columns.push_back(Id(columnNode.GetRule_column_name2().GetRule_an_id2(), *this));
  1477. }
  1478. }
  1479. }
  1480. TTableRef tr;
  1481. if (!SimpleTableRefImpl(rule.GetRule_analyze_table_list2().GetRule_analyze_table1().GetRule_simple_table_ref1(), tr)) {
  1482. return false;
  1483. }
  1484. auto params = TAnalyzeParams{.Table = std::make_shared<TTableRef>(tr), .Columns = std::move(columns)};
  1485. AddStatementToBlocks(blocks, BuildAnalyze(Ctx.Pos(), tr.Service, tr.Cluster, params, Ctx.Scoped));
  1486. break;
  1487. }
  1488. case TRule_sql_stmt_core::kAltSqlStmtCore52: {
  1489. // create_resource_pool_classifier_stmt: CREATE RESOURCE POOL CLASSIFIER name WITH (k=v,...);
  1490. auto& node = core.GetAlt_sql_stmt_core52().GetRule_create_resource_pool_classifier_stmt1();
  1491. TObjectOperatorContext context(Ctx.Scoped);
  1492. if (node.GetRule_object_ref5().HasBlock1()) {
  1493. if (!ClusterExpr(node.GetRule_object_ref5().GetBlock1().GetRule_cluster_expr1(),
  1494. false, context.ServiceId, context.Cluster)) {
  1495. return false;
  1496. }
  1497. }
  1498. const TString& objectId = Id(node.GetRule_object_ref5().GetRule_id_or_at2(), *this).second;
  1499. std::map<TString, TDeferredAtom> kv;
  1500. if (!ParseResourcePoolClassifierSettings(kv, node.GetRule_with_table_settings6())) {
  1501. return false;
  1502. }
  1503. AddStatementToBlocks(blocks, BuildCreateObjectOperation(Ctx.Pos(), objectId, "RESOURCE_POOL_CLASSIFIER", false, false, std::move(kv), context));
  1504. break;
  1505. }
  1506. case TRule_sql_stmt_core::kAltSqlStmtCore53: {
  1507. // alter_resource_pool_classifier_stmt: ALTER RESOURCE POOL CLASSIFIER object_ref alter_resource_pool_classifier_action (COMMA alter_resource_pool_classifier_action)*
  1508. Ctx.BodyPart();
  1509. const auto& node = core.GetAlt_sql_stmt_core53().GetRule_alter_resource_pool_classifier_stmt1();
  1510. TObjectOperatorContext context(Ctx.Scoped);
  1511. if (node.GetRule_object_ref5().HasBlock1()) {
  1512. if (!ClusterExpr(node.GetRule_object_ref5().GetBlock1().GetRule_cluster_expr1(),
  1513. false, context.ServiceId, context.Cluster)) {
  1514. return false;
  1515. }
  1516. }
  1517. const TString& objectId = Id(node.GetRule_object_ref5().GetRule_id_or_at2(), *this).second;
  1518. std::map<TString, TDeferredAtom> kv;
  1519. std::set<TString> toReset;
  1520. if (!ParseResourcePoolClassifierSettings(kv, toReset, node.GetRule_alter_resource_pool_classifier_action6())) {
  1521. return false;
  1522. }
  1523. for (const auto& action : node.GetBlock7()) {
  1524. if (!ParseResourcePoolClassifierSettings(kv, toReset, action.GetRule_alter_resource_pool_classifier_action2())) {
  1525. return false;
  1526. }
  1527. }
  1528. AddStatementToBlocks(blocks, BuildAlterObjectOperation(Ctx.Pos(), objectId, "RESOURCE_POOL_CLASSIFIER", std::move(kv), std::move(toReset), context));
  1529. break;
  1530. }
  1531. case TRule_sql_stmt_core::kAltSqlStmtCore54: {
  1532. // drop_resource_pool_classifier_stmt: DROP RESOURCE POOL CLASSIFIER name;
  1533. auto& node = core.GetAlt_sql_stmt_core54().GetRule_drop_resource_pool_classifier_stmt1();
  1534. TObjectOperatorContext context(Ctx.Scoped);
  1535. if (node.GetRule_object_ref5().HasBlock1()) {
  1536. if (!ClusterExpr(node.GetRule_object_ref5().GetBlock1().GetRule_cluster_expr1(),
  1537. false, context.ServiceId, context.Cluster)) {
  1538. return false;
  1539. }
  1540. }
  1541. const TString& objectId = Id(node.GetRule_object_ref5().GetRule_id_or_at2(), *this).second;
  1542. AddStatementToBlocks(blocks, BuildDropObjectOperation(Ctx.Pos(), objectId, "RESOURCE_POOL_CLASSIFIER", false, {}, context));
  1543. break;
  1544. }
  1545. case TRule_sql_stmt_core::kAltSqlStmtCore55: {
  1546. // backup_stmt: BACKUP object_ref (INCREMENTAL)?;
  1547. auto& node = core.GetAlt_sql_stmt_core55().GetRule_backup_stmt1();
  1548. TObjectOperatorContext context(Ctx.Scoped);
  1549. if (node.GetRule_object_ref2().HasBlock1()) {
  1550. if (!ClusterExpr(node.GetRule_object_ref2().GetBlock1().GetRule_cluster_expr1(),
  1551. false, context.ServiceId, context.Cluster)) {
  1552. return false;
  1553. }
  1554. }
  1555. bool incremental = node.HasBlock3();
  1556. const TString& objectId = Id(node.GetRule_object_ref2().GetRule_id_or_at2(), *this).second;
  1557. AddStatementToBlocks(blocks,
  1558. BuildBackup(
  1559. Ctx.Pos(),
  1560. TString(Ctx.GetPrefixPath(context.ServiceId, context.Cluster)),
  1561. objectId,
  1562. TBackupParameters{
  1563. .Incremental = incremental,
  1564. },
  1565. context));
  1566. break;
  1567. }
  1568. case TRule_sql_stmt_core::kAltSqlStmtCore56: {
  1569. // restore_stmt: RESTORE object_ref (AT STRING_VALUE)?;
  1570. auto& node = core.GetAlt_sql_stmt_core56().GetRule_restore_stmt1();
  1571. TObjectOperatorContext context(Ctx.Scoped);
  1572. if (node.GetRule_object_ref2().HasBlock1()) {
  1573. if (!ClusterExpr(node.GetRule_object_ref2().GetBlock1().GetRule_cluster_expr1(),
  1574. false, context.ServiceId, context.Cluster)) {
  1575. return false;
  1576. }
  1577. }
  1578. TString at;
  1579. if (node.HasBlock3()) {
  1580. const TString stringValue = Ctx.Token(node.GetBlock3().GetToken2());
  1581. const auto unescaped = StringContent(Ctx, Ctx.Pos(), stringValue);
  1582. if (!unescaped) {
  1583. return false;
  1584. }
  1585. at = unescaped->Content;
  1586. }
  1587. const TString& objectId = Id(node.GetRule_object_ref2().GetRule_id_or_at2(), *this).second;
  1588. AddStatementToBlocks(blocks,
  1589. BuildRestore(
  1590. Ctx.Pos(),
  1591. TString(Ctx.GetPrefixPath(context.ServiceId, context.Cluster)),
  1592. objectId,
  1593. TRestoreParameters{
  1594. .At = at,
  1595. },
  1596. context));
  1597. break;
  1598. }
  1599. case TRule_sql_stmt_core::kAltSqlStmtCore57: {
  1600. // alter_sequence_stmt: ALTER SEQUENCE (IF EXISTS)? object_ref alter_sequence_action (COMMA alter_sequence_action)*;
  1601. Ctx.BodyPart();
  1602. auto& node = core.GetAlt_sql_stmt_core57().GetRule_alter_sequence_stmt1();
  1603. Ctx.Token(node.GetToken1());
  1604. const TPosition pos = Ctx.Pos();
  1605. TString service = Ctx.Scoped->CurrService;
  1606. TDeferredAtom cluster = Ctx.Scoped->CurrCluster;
  1607. if (cluster.Empty()) {
  1608. Error() << "USE statement is missing - no default cluster is selected";
  1609. return false;
  1610. }
  1611. TObjectOperatorContext context(Ctx.Scoped);
  1612. if (node.GetRule_object_ref4().HasBlock1()) {
  1613. if (!ClusterExpr(node.GetRule_object_ref4().GetBlock1().GetRule_cluster_expr1(),
  1614. false, context.ServiceId, context.Cluster)) {
  1615. return false;
  1616. }
  1617. }
  1618. const TString id = Id(node.GetRule_object_ref4().GetRule_id_or_at2(), *this).second;
  1619. TSequenceParameters params;
  1620. if (node.HasBlock3()) { // IF EXISTS
  1621. params.MissingOk = true;
  1622. Y_DEBUG_ABORT_UNLESS(
  1623. IS_TOKEN(node.GetBlock3().GetToken1().GetId(), IF) &&
  1624. IS_TOKEN(node.GetBlock3().GetToken2().GetId(), EXISTS)
  1625. );
  1626. }
  1627. for (const auto& block : node.GetBlock5()) {
  1628. if (!AlterSequenceAction(block.GetRule_alter_sequence_action1(), params)) {
  1629. return false;
  1630. }
  1631. }
  1632. AddStatementToBlocks(blocks, BuildAlterSequence(pos, service, cluster, id, params, Ctx.Scoped));
  1633. break;
  1634. }
  1635. case TRule_sql_stmt_core::kAltSqlStmtCore58: {
  1636. // create_transfer_stmt: CREATE TRANSFER
  1637. auto& node = core.GetAlt_sql_stmt_core58().GetRule_create_transfer_stmt1();
  1638. TObjectOperatorContext context(Ctx.Scoped);
  1639. if (node.GetRule_object_ref3().HasBlock1()) {
  1640. const auto& cluster = node.GetRule_object_ref3().GetBlock1().GetRule_cluster_expr1();
  1641. if (!ClusterExpr(cluster, false, context.ServiceId, context.Cluster)) {
  1642. return false;
  1643. }
  1644. }
  1645. auto prefixPath = Ctx.GetPrefixPath(context.ServiceId, context.Cluster);
  1646. std::map<TString, TNodePtr> settings;
  1647. TSqlExpression expr(Ctx, Mode);
  1648. if (!TransferSettings(settings, node.GetRule_transfer_settings11(), expr, true)) {
  1649. return false;
  1650. }
  1651. const TString id = Id(node.GetRule_object_ref3().GetRule_id_or_at2(), *this).second;
  1652. const TString source = Id(node.GetRule_object_ref5().GetRule_id_or_at2(), *this).second;
  1653. const TString target = Id(node.GetRule_object_ref7().GetRule_id_or_at2(), *this).second;
  1654. TString transformLambda;
  1655. if (node.GetBlock8().HasRule_lambda_or_parameter2()) {
  1656. if (!ParseTransferLambda(transformLambda, node.GetBlock8().GetRule_lambda_or_parameter2())) {
  1657. return false;
  1658. }
  1659. }
  1660. AddStatementToBlocks(blocks, BuildCreateTransfer(Ctx.Pos(), BuildTablePath(prefixPath, id),
  1661. std::move(source), std::move(target), std::move(transformLambda), std::move(settings), context));
  1662. break;
  1663. }
  1664. case TRule_sql_stmt_core::kAltSqlStmtCore59: {
  1665. // alter_transfer_stmt: ALTER TRANSFER
  1666. auto& node = core.GetAlt_sql_stmt_core59().GetRule_alter_transfer_stmt1();
  1667. TObjectOperatorContext context(Ctx.Scoped);
  1668. if (node.GetRule_object_ref3().HasBlock1()) {
  1669. const auto& cluster = node.GetRule_object_ref3().GetBlock1().GetRule_cluster_expr1();
  1670. if (!ClusterExpr(cluster, false, context.ServiceId, context.Cluster)) {
  1671. return false;
  1672. }
  1673. }
  1674. std::map<TString, TNodePtr> settings;
  1675. std::optional<TString> transformLambda;
  1676. TSqlExpression expr(Ctx, Mode);
  1677. auto transferAlterAction = [&](std::optional<TString>& transformLambda, const TRule_alter_transfer_action& in)
  1678. {
  1679. if (in.HasAlt_alter_transfer_action1()) {
  1680. return TransferSettings(settings, in.GetAlt_alter_transfer_action1().GetRule_alter_transfer_set_setting1().GetRule_transfer_settings3(), expr, false);
  1681. } else if (in.HasAlt_alter_transfer_action2()) {
  1682. TString lb;
  1683. if (!ParseTransferLambda(lb, in.GetAlt_alter_transfer_action2().GetRule_alter_transfer_set_using1().GetRule_lambda_or_parameter3())) {
  1684. return false;
  1685. }
  1686. transformLambda = lb;
  1687. return true;
  1688. }
  1689. return false;
  1690. };
  1691. if (!transferAlterAction(transformLambda, node.GetRule_alter_transfer_action4())) {
  1692. return false;
  1693. }
  1694. for (auto& block : node.GetBlock5()) {
  1695. if (!transferAlterAction(transformLambda, block.GetRule_alter_transfer_action2())) {
  1696. return false;
  1697. }
  1698. }
  1699. const TString id = Id(node.GetRule_object_ref3().GetRule_id_or_at2(), *this).second;
  1700. AddStatementToBlocks(blocks, BuildAlterTransfer(Ctx.Pos(),
  1701. BuildTablePath(Ctx.GetPrefixPath(context.ServiceId, context.Cluster), id),
  1702. std::move(transformLambda), std::move(settings), context));
  1703. break;
  1704. }
  1705. case TRule_sql_stmt_core::kAltSqlStmtCore60: {
  1706. // drop_transfer_stmt: DROP TRANSFER
  1707. auto& node = core.GetAlt_sql_stmt_core60().GetRule_drop_transfer_stmt1();
  1708. TObjectOperatorContext context(Ctx.Scoped);
  1709. if (node.GetRule_object_ref3().HasBlock1()) {
  1710. const auto& cluster = node.GetRule_object_ref3().GetBlock1().GetRule_cluster_expr1();
  1711. if (!ClusterExpr(cluster, false, context.ServiceId, context.Cluster)) {
  1712. return false;
  1713. }
  1714. }
  1715. const TString id = Id(node.GetRule_object_ref3().GetRule_id_or_at2(), *this).second;
  1716. AddStatementToBlocks(blocks, BuildDropTransfer(Ctx.Pos(),
  1717. BuildTablePath(Ctx.GetPrefixPath(context.ServiceId, context.Cluster), id),
  1718. node.HasBlock4(), context));
  1719. break;
  1720. }
  1721. case TRule_sql_stmt_core::kAltSqlStmtCore61: {
  1722. // alter_database_stmt: ALTER DATABASE an_id_schema OWNER TO role_name
  1723. auto& node = core.GetAlt_sql_stmt_core61().GetRule_alter_database_stmt1();
  1724. TDeferredAtom roleName;
  1725. {
  1726. bool allowSystemRoles = true;
  1727. if (!RoleNameClause(node.GetRule_role_name6(), roleName, allowSystemRoles)) {
  1728. return false;
  1729. }
  1730. }
  1731. TAlterDatabaseParameters alterDatabaseParams;
  1732. alterDatabaseParams.Owner = roleName;
  1733. alterDatabaseParams.DbPath = Id(node.GetRule_an_id_schema3(), *this);
  1734. const TPosition pos = Ctx.Pos();
  1735. TString service = Ctx.Scoped->CurrService;
  1736. TDeferredAtom cluster = Ctx.Scoped->CurrCluster;
  1737. auto stmt = BuildAlterDatabase(pos, service, cluster, alterDatabaseParams, Ctx.Scoped);
  1738. AddStatementToBlocks(blocks, stmt);
  1739. break;
  1740. }
  1741. case TRule_sql_stmt_core::ALT_NOT_SET:
  1742. Ctx.IncrementMonCounter("sql_errors", "UnknownStatement" + internalStatementName);
  1743. AltNotImplemented("sql_stmt_core", core);
  1744. return false;
  1745. }
  1746. Ctx.IncrementMonCounter("sql_features", internalStatementName);
  1747. return !Ctx.HasPendingErrors;
  1748. }
  1749. bool TSqlQuery::DeclareStatement(const TRule_declare_stmt& stmt) {
  1750. TNodePtr defaultValue;
  1751. if (stmt.HasBlock5()) {
  1752. TSqlExpression sqlExpr(Ctx, Mode);
  1753. auto exprOrId = sqlExpr.LiteralExpr(stmt.GetBlock5().GetRule_literal_value2());
  1754. if (!exprOrId) {
  1755. return false;
  1756. }
  1757. if (!exprOrId->Expr) {
  1758. Ctx.Error() << "Identifier is not expected here";
  1759. return false;
  1760. }
  1761. defaultValue = exprOrId->Expr;
  1762. }
  1763. if (defaultValue) {
  1764. Error() << "DEFAULT value not supported yet";
  1765. return false;
  1766. }
  1767. if (!Ctx.IsParseHeading()) {
  1768. Error() << "DECLARE statement should be in beginning of query, but it's possible to use PRAGMA or USE before it";
  1769. return false;
  1770. }
  1771. TString varName;
  1772. if (!NamedNodeImpl(stmt.GetRule_bind_parameter2(), varName, *this)) {
  1773. return false;
  1774. }
  1775. const auto varPos = Ctx.Pos();
  1776. const auto typeNode = TypeNode(stmt.GetRule_type_name4());
  1777. if (!typeNode) {
  1778. return false;
  1779. }
  1780. if (IsAnonymousName(varName)) {
  1781. Ctx.Error(varPos) << "Can not use anonymous name '" << varName << "' in DECLARE statement";
  1782. return false;
  1783. }
  1784. if (Ctx.IsAlreadyDeclared(varName)) {
  1785. Ctx.Warning(varPos, TIssuesIds::YQL_DUPLICATE_DECLARE) << "Duplicate declaration of '" << varName << "' will be ignored";
  1786. } else {
  1787. PushNamedAtom(varPos, varName);
  1788. Ctx.DeclareVariable(varName, varPos, typeNode);
  1789. }
  1790. return true;
  1791. }
  1792. bool TSqlQuery::ExportStatement(const TRule_export_stmt& stmt) {
  1793. if (Mode != NSQLTranslation::ESqlMode::LIBRARY || !TopLevel) {
  1794. Error() << "EXPORT statement should be used only in a library on the top level";
  1795. return false;
  1796. }
  1797. TVector<TSymbolNameWithPos> bindNames;
  1798. if (!BindList(stmt.GetRule_bind_parameter_list2(), bindNames)) {
  1799. return false;
  1800. }
  1801. for (auto& bindName : bindNames) {
  1802. if (!Ctx.AddExport(bindName.Pos, bindName.Name)) {
  1803. return false;
  1804. }
  1805. }
  1806. return true;
  1807. }
  1808. bool TSqlQuery::AlterTableAction(const TRule_alter_table_action& node, TAlterTableParameters& params) {
  1809. if (params.RenameTo) {
  1810. // rename action is followed by some other actions
  1811. Error() << "RENAME TO can not be used together with another table action";
  1812. return false;
  1813. }
  1814. switch (node.Alt_case()) {
  1815. case TRule_alter_table_action::kAltAlterTableAction1: {
  1816. // ADD COLUMN
  1817. const auto& addRule = node.GetAlt_alter_table_action1().GetRule_alter_table_add_column1();
  1818. if (!AlterTableAddColumn(addRule, params)) {
  1819. return false;
  1820. }
  1821. break;
  1822. }
  1823. case TRule_alter_table_action::kAltAlterTableAction2: {
  1824. // DROP COLUMN
  1825. const auto& dropRule = node.GetAlt_alter_table_action2().GetRule_alter_table_drop_column1();
  1826. if (!AlterTableDropColumn(dropRule, params)) {
  1827. return false;
  1828. }
  1829. break;
  1830. }
  1831. case TRule_alter_table_action::kAltAlterTableAction3: {
  1832. // ALTER COLUMN
  1833. const auto& alterRule = node.GetAlt_alter_table_action3().GetRule_alter_table_alter_column1();
  1834. if (!AlterTableAlterColumn(alterRule, params)) {
  1835. return false;
  1836. }
  1837. break;
  1838. }
  1839. case TRule_alter_table_action::kAltAlterTableAction4: {
  1840. // ADD FAMILY
  1841. const auto& familyEntry = node.GetAlt_alter_table_action4().GetRule_alter_table_add_column_family1()
  1842. .GetRule_family_entry2();
  1843. if (!AlterTableAddFamily(familyEntry, params)) {
  1844. return false;
  1845. }
  1846. break;
  1847. }
  1848. case TRule_alter_table_action::kAltAlterTableAction5: {
  1849. // ALTER FAMILY
  1850. const auto& alterRule = node.GetAlt_alter_table_action5().GetRule_alter_table_alter_column_family1();
  1851. if (!AlterTableAlterFamily(alterRule, params)) {
  1852. return false;
  1853. }
  1854. break;
  1855. }
  1856. case TRule_alter_table_action::kAltAlterTableAction6: {
  1857. // SET (uncompat)
  1858. const auto& setRule = node.GetAlt_alter_table_action6().GetRule_alter_table_set_table_setting_uncompat1();
  1859. if (!AlterTableSetTableSetting(setRule, params.TableSettings, params.TableType)) {
  1860. return false;
  1861. }
  1862. break;
  1863. }
  1864. case TRule_alter_table_action::kAltAlterTableAction7: {
  1865. // SET (compat)
  1866. const auto& setRule = node.GetAlt_alter_table_action7().GetRule_alter_table_set_table_setting_compat1();
  1867. if (!AlterTableSetTableSetting(setRule, params.TableSettings, params.TableType)) {
  1868. return false;
  1869. }
  1870. break;
  1871. }
  1872. case TRule_alter_table_action::kAltAlterTableAction8: {
  1873. // RESET
  1874. const auto& setRule = node.GetAlt_alter_table_action8().GetRule_alter_table_reset_table_setting1();
  1875. if (!AlterTableResetTableSetting(setRule, params.TableSettings, params.TableType)) {
  1876. return false;
  1877. }
  1878. break;
  1879. }
  1880. case TRule_alter_table_action::kAltAlterTableAction9: {
  1881. // ADD INDEX
  1882. const auto& addIndex = node.GetAlt_alter_table_action9().GetRule_alter_table_add_index1();
  1883. if (!AlterTableAddIndex(addIndex, params)) {
  1884. return false;
  1885. }
  1886. break;
  1887. }
  1888. case TRule_alter_table_action::kAltAlterTableAction10: {
  1889. // DROP INDEX
  1890. const auto& dropIndex = node.GetAlt_alter_table_action10().GetRule_alter_table_drop_index1();
  1891. AlterTableDropIndex(dropIndex, params);
  1892. break;
  1893. }
  1894. case TRule_alter_table_action::kAltAlterTableAction11: {
  1895. // RENAME TO
  1896. if (!params.IsEmpty()) {
  1897. // rename action follows some other actions
  1898. Error() << "RENAME TO can not be used together with another table action";
  1899. return false;
  1900. }
  1901. const auto& renameTo = node.GetAlt_alter_table_action11().GetRule_alter_table_rename_to1();
  1902. AlterTableRenameTo(renameTo, params);
  1903. break;
  1904. }
  1905. case TRule_alter_table_action::kAltAlterTableAction12: {
  1906. // ADD CHANGEFEED
  1907. const auto& rule = node.GetAlt_alter_table_action12().GetRule_alter_table_add_changefeed1();
  1908. if (!AlterTableAddChangefeed(rule, params)) {
  1909. return false;
  1910. }
  1911. break;
  1912. }
  1913. case TRule_alter_table_action::kAltAlterTableAction13: {
  1914. // ALTER CHANGEFEED
  1915. const auto& rule = node.GetAlt_alter_table_action13().GetRule_alter_table_alter_changefeed1();
  1916. if (!AlterTableAlterChangefeed(rule, params)) {
  1917. return false;
  1918. }
  1919. break;
  1920. }
  1921. case TRule_alter_table_action::kAltAlterTableAction14: {
  1922. // DROP CHANGEFEED
  1923. const auto& rule = node.GetAlt_alter_table_action14().GetRule_alter_table_drop_changefeed1();
  1924. AlterTableDropChangefeed(rule, params);
  1925. break;
  1926. }
  1927. case TRule_alter_table_action::kAltAlterTableAction15: {
  1928. // RENAME INDEX TO
  1929. if (!params.IsEmpty()) {
  1930. // rename action follows some other actions
  1931. Error() << "RENAME INDEX TO can not be used together with another table action";
  1932. return false;
  1933. }
  1934. const auto& renameTo = node.GetAlt_alter_table_action15().GetRule_alter_table_rename_index_to1();
  1935. AlterTableRenameIndexTo(renameTo, params);
  1936. break;
  1937. }
  1938. case TRule_alter_table_action::kAltAlterTableAction16: {
  1939. // ALTER INDEX
  1940. const auto& rule = node.GetAlt_alter_table_action16().GetRule_alter_table_alter_index1();
  1941. if (!AlterTableAlterIndex(rule, params)) {
  1942. return false;
  1943. }
  1944. break;
  1945. }
  1946. case TRule_alter_table_action::kAltAlterTableAction17: {
  1947. // ALTER COLUMN id DROP NOT NULL
  1948. const auto& alterRule = node.GetAlt_alter_table_action17().GetRule_alter_table_alter_column_drop_not_null1();
  1949. if (!AlterTableAlterColumnDropNotNull(alterRule, params)) {
  1950. return false;
  1951. }
  1952. break;
  1953. }
  1954. case TRule_alter_table_action::ALT_NOT_SET: {
  1955. AltNotImplemented("alter_table_action", node);
  1956. return false;
  1957. }
  1958. }
  1959. return true;
  1960. }
  1961. bool TSqlQuery::AlterExternalTableAction(const TRule_alter_external_table_action& node, TAlterTableParameters& params) {
  1962. if (params.RenameTo) {
  1963. // rename action is followed by some other actions
  1964. Error() << "RENAME TO can not be used together with another table action";
  1965. return false;
  1966. }
  1967. switch (node.Alt_case()) {
  1968. case TRule_alter_external_table_action::kAltAlterExternalTableAction1: {
  1969. // ADD COLUMN
  1970. const auto& addRule = node.GetAlt_alter_external_table_action1().GetRule_alter_table_add_column1();
  1971. if (!AlterTableAddColumn(addRule, params)) {
  1972. return false;
  1973. }
  1974. break;
  1975. }
  1976. case TRule_alter_external_table_action::kAltAlterExternalTableAction2: {
  1977. // DROP COLUMN
  1978. const auto& dropRule = node.GetAlt_alter_external_table_action2().GetRule_alter_table_drop_column1();
  1979. if (!AlterTableDropColumn(dropRule, params)) {
  1980. return false;
  1981. }
  1982. break;
  1983. }
  1984. case TRule_alter_external_table_action::kAltAlterExternalTableAction3: {
  1985. // SET (uncompat)
  1986. const auto& setRule = node.GetAlt_alter_external_table_action3().GetRule_alter_table_set_table_setting_uncompat1();
  1987. if (!AlterTableSetTableSetting(setRule, params.TableSettings, params.TableType)) {
  1988. return false;
  1989. }
  1990. break;
  1991. }
  1992. case TRule_alter_external_table_action::kAltAlterExternalTableAction4: {
  1993. // SET (compat)
  1994. const auto& setRule = node.GetAlt_alter_external_table_action4().GetRule_alter_table_set_table_setting_compat1();
  1995. if (!AlterTableSetTableSetting(setRule, params.TableSettings, params.TableType)) {
  1996. return false;
  1997. }
  1998. break;
  1999. }
  2000. case TRule_alter_external_table_action::kAltAlterExternalTableAction5: {
  2001. // RESET
  2002. const auto& setRule = node.GetAlt_alter_external_table_action5().GetRule_alter_table_reset_table_setting1();
  2003. if (!AlterTableResetTableSetting(setRule, params.TableSettings, params.TableType)) {
  2004. return false;
  2005. }
  2006. break;
  2007. }
  2008. case TRule_alter_external_table_action::ALT_NOT_SET:
  2009. AltNotImplemented("alter_external_table_action", node);
  2010. return false;
  2011. }
  2012. return true;
  2013. }
  2014. bool TSqlQuery::AlterTableAddColumn(const TRule_alter_table_add_column& node, TAlterTableParameters& params) {
  2015. auto columnSchema = ColumnSchemaImpl(node.GetRule_column_schema3());
  2016. if (!columnSchema) {
  2017. return false;
  2018. }
  2019. if (columnSchema->Families.size() > 1) {
  2020. Ctx.Error() << "Several column families for a single column are not yet supported";
  2021. return false;
  2022. }
  2023. params.AddColumns.push_back(*columnSchema);
  2024. return true;
  2025. }
  2026. bool TSqlQuery::AlterTableDropColumn(const TRule_alter_table_drop_column& node, TAlterTableParameters& params) {
  2027. TString name = Id(node.GetRule_an_id3(), *this);
  2028. params.DropColumns.push_back(name);
  2029. return true;
  2030. }
  2031. bool TSqlQuery::AlterTableAlterColumn(const TRule_alter_table_alter_column& node,
  2032. TAlterTableParameters& params)
  2033. {
  2034. TString name = Id(node.GetRule_an_id3(), *this);
  2035. const TPosition pos(Context().Pos());
  2036. TVector<TIdentifier> families;
  2037. const auto& familyRelation = node.GetRule_family_relation5();
  2038. families.push_back(IdEx(familyRelation.GetRule_an_id2(), *this));
  2039. params.AlterColumns.emplace_back(pos, name, nullptr, false, families, false, nullptr, TColumnSchema::ETypeOfChange::SetFamily);
  2040. return true;
  2041. }
  2042. bool TSqlQuery::AlterTableAddFamily(const TRule_family_entry& node, TAlterTableParameters& params) {
  2043. TFamilyEntry family(IdEx(node.GetRule_an_id2(), *this));
  2044. if (!FillFamilySettings(node.GetRule_family_settings3(), family)) {
  2045. return false;
  2046. }
  2047. params.AddColumnFamilies.push_back(family);
  2048. return true;
  2049. }
  2050. bool TSqlQuery::AlterTableAlterFamily(const TRule_alter_table_alter_column_family& node,
  2051. TAlterTableParameters& params)
  2052. {
  2053. TFamilyEntry* entry = nullptr;
  2054. TIdentifier name = IdEx(node.GetRule_an_id3(), *this);
  2055. for (auto& family : params.AlterColumnFamilies) {
  2056. if (family.Name.Name == name.Name) {
  2057. entry = &family;
  2058. break;
  2059. }
  2060. }
  2061. if (!entry) {
  2062. entry = &params.AlterColumnFamilies.emplace_back(name);
  2063. }
  2064. TIdentifier settingName = IdEx(node.GetRule_an_id5(), *this);
  2065. const TRule_family_setting_value& value = node.GetRule_family_setting_value6();
  2066. if (to_lower(settingName.Name) == "data") {
  2067. if (entry->Data) {
  2068. Ctx.Error() << "Redefinition of 'data' setting for column family '" << name.Name
  2069. << "' in one alter";
  2070. return false;
  2071. }
  2072. if (!StoreString(value, entry->Data, Ctx)) {
  2073. Ctx.Error() << to_upper(settingName.Name) << " value should be a string literal";
  2074. return false;
  2075. }
  2076. } else if (to_lower(settingName.Name) == "compression") {
  2077. if (entry->Compression) {
  2078. Ctx.Error() << "Redefinition of 'compression' setting for column family '" << name.Name
  2079. << "' in one alter";
  2080. return false;
  2081. }
  2082. if (!StoreString(value, entry->Compression, Ctx)) {
  2083. Ctx.Error() << to_upper(settingName.Name) << " value should be a string literal";
  2084. return false;
  2085. }
  2086. } else if (to_lower(settingName.Name) == "compression_level") {
  2087. if (entry->CompressionLevel) {
  2088. Ctx.Error() << "Redefinition of 'compression_level' setting for column family '" << name.Name << "' in one alter";
  2089. return false;
  2090. }
  2091. if (!StoreInt(value, entry->CompressionLevel, Ctx)) {
  2092. Ctx.Error() << to_upper(settingName.Name) << " value should be an integer";
  2093. return false;
  2094. }
  2095. } else {
  2096. Ctx.Error() << "Unknown table setting: " << settingName.Name;
  2097. return false;
  2098. }
  2099. return true;
  2100. }
  2101. bool TSqlQuery::AlterTableSetTableSetting(
  2102. const TRule_alter_table_set_table_setting_uncompat& node, TTableSettings& tableSettings, ETableType tableType
  2103. ) {
  2104. return StoreTableSettingsEntry(
  2105. IdEx(node.GetRule_an_id2(), *this),
  2106. node.GetRule_table_setting_value3(),
  2107. tableSettings,
  2108. tableType,
  2109. true
  2110. );
  2111. }
  2112. bool TSqlQuery::AlterTableSetTableSetting(
  2113. const TRule_alter_table_set_table_setting_compat& node, TTableSettings& tableSettings, ETableType tableType
  2114. ) {
  2115. const auto storeSetting = [&](const TRule_alter_table_setting_entry& entry) {
  2116. return StoreTableSettingsEntry(
  2117. IdEx(entry.GetRule_an_id1(), *this),
  2118. entry.GetRule_table_setting_value3(),
  2119. tableSettings,
  2120. tableType,
  2121. true
  2122. );
  2123. };
  2124. const auto& firstEntry = node.GetRule_alter_table_setting_entry3();
  2125. if (!storeSetting(firstEntry)) {
  2126. return false;
  2127. }
  2128. for (const auto& block : node.GetBlock4()) {
  2129. const auto& entry = block.GetRule_alter_table_setting_entry2();
  2130. if (!storeSetting(entry)) {
  2131. return false;
  2132. }
  2133. }
  2134. return true;
  2135. }
  2136. bool TSqlQuery::AlterTableResetTableSetting(
  2137. const TRule_alter_table_reset_table_setting& node, TTableSettings& tableSettings, ETableType tableType
  2138. ) {
  2139. const auto resetSetting = [&](const TRule_an_id& id) {
  2140. return ResetTableSettingsEntry(IdEx(id, *this), tableSettings, tableType);
  2141. };
  2142. const auto& firstEntry = node.GetRule_an_id3();
  2143. if (!resetSetting(firstEntry)) {
  2144. return false;
  2145. }
  2146. for (const auto& block : node.GetBlock4()) {
  2147. const auto& entry = block.GetRule_an_id2();
  2148. if (!resetSetting(entry)) {
  2149. return false;
  2150. }
  2151. }
  2152. return true;
  2153. }
  2154. bool TSqlQuery::AlterTableAddIndex(const TRule_alter_table_add_index& node, TAlterTableParameters& params) {
  2155. if (!CreateTableIndex(node.GetRule_table_index2(), params.AddIndexes)) {
  2156. return false;
  2157. }
  2158. return true;
  2159. }
  2160. void TSqlQuery::AlterTableDropIndex(const TRule_alter_table_drop_index& node, TAlterTableParameters& params) {
  2161. params.DropIndexes.emplace_back(IdEx(node.GetRule_an_id3(), *this));
  2162. }
  2163. void TSqlQuery::AlterTableRenameTo(const TRule_alter_table_rename_to& node, TAlterTableParameters& params) {
  2164. params.RenameTo = IdEx(node.GetRule_an_id_table3(), *this);
  2165. }
  2166. void TSqlQuery::AlterTableRenameIndexTo(const TRule_alter_table_rename_index_to& node, TAlterTableParameters& params) {
  2167. auto src = IdEx(node.GetRule_an_id3(), *this);
  2168. auto dst = IdEx(node.GetRule_an_id5(), *this);
  2169. params.RenameIndexTo = std::make_pair(src, dst);
  2170. }
  2171. bool TSqlQuery::AlterTableAlterIndex(const TRule_alter_table_alter_index& node, TAlterTableParameters& params) {
  2172. const auto indexName = IdEx(node.GetRule_an_id3(), *this);
  2173. params.AlterIndexes.emplace_back(indexName);
  2174. TTableSettings& indexTableSettings = params.AlterIndexes.back().TableSettings;
  2175. const auto& action = node.GetRule_alter_table_alter_index_action4();
  2176. switch (action.Alt_case()) {
  2177. case TRule_alter_table_alter_index_action::kAltAlterTableAlterIndexAction1: {
  2178. // SET setting value
  2179. const auto& rule = action.GetAlt_alter_table_alter_index_action1().GetRule_alter_table_set_table_setting_uncompat1();
  2180. if (!AlterTableSetTableSetting(rule, indexTableSettings, params.TableType)) {
  2181. return false;
  2182. }
  2183. break;
  2184. }
  2185. case TRule_alter_table_alter_index_action::kAltAlterTableAlterIndexAction2: {
  2186. // SET (setting1 = value1, ...)
  2187. const auto& rule = action.GetAlt_alter_table_alter_index_action2().GetRule_alter_table_set_table_setting_compat1();
  2188. if (!AlterTableSetTableSetting(rule, indexTableSettings, params.TableType)) {
  2189. return false;
  2190. }
  2191. break;
  2192. }
  2193. case TRule_alter_table_alter_index_action::kAltAlterTableAlterIndexAction3: {
  2194. // RESET (setting1, ...)
  2195. const auto& rule = action.GetAlt_alter_table_alter_index_action3().GetRule_alter_table_reset_table_setting1();
  2196. if (!AlterTableResetTableSetting(rule, indexTableSettings, params.TableType)) {
  2197. return false;
  2198. }
  2199. break;
  2200. }
  2201. case TRule_alter_table_alter_index_action::ALT_NOT_SET:
  2202. AltNotImplemented("alter_table_alter_index_action", action);
  2203. return false;
  2204. }
  2205. return true;
  2206. }
  2207. bool TSqlQuery::AlterSequenceAction(const TRule_alter_sequence_action& node, TSequenceParameters& params) {
  2208. switch (node.Alt_case()) {
  2209. case TRule_alter_sequence_action::kAltAlterSequenceAction1: {
  2210. if (params.StartValue) {
  2211. Ctx.Error(Ctx.Pos()) << "Start value defined more than once";
  2212. return false;
  2213. }
  2214. auto literalNumber = LiteralNumber(Ctx, node.GetAlt_alter_sequence_action1().GetRule_integer3());
  2215. if (literalNumber) {
  2216. params.StartValue = TDeferredAtom(literalNumber, Ctx);
  2217. } else {
  2218. return false;
  2219. }
  2220. break;
  2221. }
  2222. case TRule_alter_sequence_action::kAltAlterSequenceAction2: {
  2223. if (params.IsRestart) {
  2224. Ctx.Error(Ctx.Pos()) << "Restart value defined more than once";
  2225. return false;
  2226. }
  2227. auto literalNumber = LiteralNumber(Ctx, node.GetAlt_alter_sequence_action2().GetRule_integer3());
  2228. if (literalNumber) {
  2229. params.IsRestart = true;
  2230. params.RestartValue = TDeferredAtom(literalNumber, Ctx);
  2231. } else {
  2232. return false;
  2233. }
  2234. break;
  2235. }
  2236. case TRule_alter_sequence_action::kAltAlterSequenceAction3: {
  2237. if (params.IsRestart) {
  2238. Ctx.Error(Ctx.Pos()) << "Restart value defined more than once";
  2239. return false;
  2240. }
  2241. params.IsRestart = true;
  2242. break;
  2243. }
  2244. case TRule_alter_sequence_action::kAltAlterSequenceAction4: {
  2245. if (params.Increment) {
  2246. Ctx.Error(Ctx.Pos()) << "Increment defined more than once";
  2247. return false;
  2248. }
  2249. auto literalNumber = LiteralNumber(Ctx, node.GetAlt_alter_sequence_action4().GetRule_integer3());
  2250. if (literalNumber) {
  2251. params.Increment = TDeferredAtom(literalNumber, Ctx);
  2252. } else {
  2253. return false;
  2254. }
  2255. break;
  2256. }
  2257. case TRule_alter_sequence_action::ALT_NOT_SET:
  2258. Y_ABORT("You should change implementation according to grammar changes");
  2259. }
  2260. return true;
  2261. }
  2262. bool TSqlQuery::AlterTableAlterColumnDropNotNull(const TRule_alter_table_alter_column_drop_not_null& node, TAlterTableParameters& params) {
  2263. TString name = Id(node.GetRule_an_id3(), *this);
  2264. const TPosition pos(Context().Pos());
  2265. params.AlterColumns.emplace_back(pos, name, nullptr, false, TVector<TIdentifier>(), false, nullptr, TColumnSchema::ETypeOfChange::DropNotNullConstraint);
  2266. return true;
  2267. }
  2268. bool TSqlQuery::AlterTableAddChangefeed(const TRule_alter_table_add_changefeed& node, TAlterTableParameters& params) {
  2269. TSqlExpression expr(Ctx, Mode);
  2270. return CreateChangefeed(node.GetRule_changefeed2(), expr, params.AddChangefeeds);
  2271. }
  2272. bool TSqlQuery::AlterTableAlterChangefeed(const TRule_alter_table_alter_changefeed& node, TAlterTableParameters& params) {
  2273. params.AlterChangefeeds.emplace_back(IdEx(node.GetRule_an_id3(), *this));
  2274. const auto& alter = node.GetRule_changefeed_alter_settings4();
  2275. switch (alter.Alt_case()) {
  2276. case TRule_changefeed_alter_settings::kAltChangefeedAlterSettings1: {
  2277. // DISABLE
  2278. params.AlterChangefeeds.back().Disable = true;
  2279. break;
  2280. }
  2281. case TRule_changefeed_alter_settings::kAltChangefeedAlterSettings2: {
  2282. // SET
  2283. const auto& rule = alter.GetAlt_changefeed_alter_settings2().GetRule_changefeed_settings3();
  2284. TSqlExpression expr(Ctx, Mode);
  2285. if (!ChangefeedSettings(rule, expr, params.AlterChangefeeds.back().Settings, true)) {
  2286. return false;
  2287. }
  2288. break;
  2289. }
  2290. case TRule_changefeed_alter_settings::ALT_NOT_SET:
  2291. AltNotImplemented("changefeed_alter_settings", alter);
  2292. return false;
  2293. }
  2294. return true;
  2295. }
  2296. void TSqlQuery::AlterTableDropChangefeed(const TRule_alter_table_drop_changefeed& node, TAlterTableParameters& params) {
  2297. params.DropChangefeeds.emplace_back(IdEx(node.GetRule_an_id3(), *this));
  2298. }
  2299. TNodePtr TSqlQuery::PragmaStatement(const TRule_pragma_stmt& stmt, bool& success) {
  2300. success = false;
  2301. const TString& prefix = OptIdPrefixAsStr(stmt.GetRule_opt_id_prefix_or_type2(), *this);
  2302. const TString& lowerPrefix = to_lower(prefix);
  2303. const TString pragma(Id(stmt.GetRule_an_id3(), *this));
  2304. TString normalizedPragma(pragma);
  2305. TMaybe<TIssue> normalizeError = NormalizeName(Ctx.Pos(), normalizedPragma);
  2306. if (!normalizeError.Empty()) {
  2307. Error() << normalizeError->GetMessage();
  2308. Ctx.IncrementMonCounter("sql_errors", "NormalizePragmaError");
  2309. return {};
  2310. }
  2311. TVector<TDeferredAtom> values;
  2312. TVector<const TRule_pragma_value*> pragmaValues;
  2313. bool pragmaValueDefault = false;
  2314. if (stmt.GetBlock4().HasAlt1()) {
  2315. pragmaValues.push_back(&stmt.GetBlock4().GetAlt1().GetRule_pragma_value2());
  2316. }
  2317. else if (stmt.GetBlock4().HasAlt2()) {
  2318. pragmaValues.push_back(&stmt.GetBlock4().GetAlt2().GetRule_pragma_value2());
  2319. for (auto& additionalValue : stmt.GetBlock4().GetAlt2().GetBlock3()) {
  2320. pragmaValues.push_back(&additionalValue.GetRule_pragma_value2());
  2321. }
  2322. }
  2323. const bool withConfigure = prefix || normalizedPragma == "file" || normalizedPragma == "folder" || normalizedPragma == "udf";
  2324. static const THashSet<TStringBuf> lexicalScopePragmas = {
  2325. "classicdivision",
  2326. "strictjoinkeytypes",
  2327. "disablestrictjoinkeytypes",
  2328. "checkedops",
  2329. "unicodeliterals",
  2330. "disableunicodeliterals",
  2331. "warnuntypedstringliterals",
  2332. "disablewarnuntypedstringliterals",
  2333. };
  2334. const bool hasLexicalScope = withConfigure || lexicalScopePragmas.contains(normalizedPragma);
  2335. const bool withFileAlias = normalizedPragma == "file" || normalizedPragma == "folder" || normalizedPragma == "library" || normalizedPragma == "udf";
  2336. for (auto pragmaValue : pragmaValues) {
  2337. if (pragmaValue->HasAlt_pragma_value3()) {
  2338. auto value = Token(pragmaValue->GetAlt_pragma_value3().GetToken1());
  2339. auto parsed = StringContentOrIdContent(Ctx, Ctx.Pos(), value);
  2340. if (!parsed) {
  2341. return {};
  2342. }
  2343. TString prefix;
  2344. if (withFileAlias && (values.size() == 0)) {
  2345. prefix = Ctx.Settings.FileAliasPrefix;
  2346. }
  2347. values.push_back(TDeferredAtom(Ctx.Pos(), prefix + parsed->Content));
  2348. }
  2349. else if (pragmaValue->HasAlt_pragma_value2()
  2350. && pragmaValue->GetAlt_pragma_value2().GetRule_id1().HasAlt_id2()
  2351. && "default" == to_lower(Id(pragmaValue->GetAlt_pragma_value2().GetRule_id1(), *this)))
  2352. {
  2353. pragmaValueDefault = true;
  2354. }
  2355. else if (withConfigure && pragmaValue->HasAlt_pragma_value5()) {
  2356. TString bindName;
  2357. if (!NamedNodeImpl(pragmaValue->GetAlt_pragma_value5().GetRule_bind_parameter1(), bindName, *this)) {
  2358. return {};
  2359. }
  2360. auto namedNode = GetNamedNode(bindName);
  2361. if (!namedNode) {
  2362. return {};
  2363. }
  2364. TString prefix;
  2365. if (withFileAlias && (values.size() == 0)) {
  2366. prefix = Ctx.Settings.FileAliasPrefix;
  2367. }
  2368. TDeferredAtom atom;
  2369. MakeTableFromExpression(Ctx.Pos(), Ctx, namedNode, atom, prefix);
  2370. values.push_back(atom);
  2371. } else {
  2372. Error() << "Expected string" << (withConfigure ? ", named parameter" : "") << " or 'default' keyword as pragma value for pragma: " << pragma;
  2373. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2374. return {};
  2375. }
  2376. }
  2377. if (prefix.empty()) {
  2378. if (!TopLevel && !hasLexicalScope) {
  2379. Error() << "This pragma '" << pragma << "' is not allowed to be used in actions or subqueries";
  2380. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2381. return{};
  2382. }
  2383. if (normalizedPragma == "refselect") {
  2384. Ctx.PragmaRefSelect = true;
  2385. Ctx.IncrementMonCounter("sql_pragma", "RefSelect");
  2386. } else if (normalizedPragma == "sampleselect") {
  2387. Ctx.PragmaSampleSelect = true;
  2388. Ctx.IncrementMonCounter("sql_pragma", "SampleSelect");
  2389. } else if (normalizedPragma == "allowdotinalias") {
  2390. Ctx.PragmaAllowDotInAlias = true;
  2391. Ctx.IncrementMonCounter("sql_pragma", "AllowDotInAlias");
  2392. } else if (normalizedPragma == "udf") {
  2393. if ((values.size() != 1 && values.size() != 2) || pragmaValueDefault) {
  2394. Error() << "Expected file alias as pragma value";
  2395. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2396. return {};
  2397. }
  2398. if (Ctx.Settings.FileAliasPrefix) {
  2399. if (values.size() == 1) {
  2400. values.emplace_back(TDeferredAtom(Ctx.Pos(), ""));
  2401. }
  2402. TString prefix;
  2403. if (!values[1].GetLiteral(prefix, Ctx)) {
  2404. Error() << "Expected literal UDF module prefix in views";
  2405. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2406. return {};
  2407. }
  2408. values[1] = TDeferredAtom(Ctx.Pos(), Ctx.Settings.FileAliasPrefix + prefix);
  2409. }
  2410. Ctx.IncrementMonCounter("sql_pragma", "udf");
  2411. success = true;
  2412. return BuildPragma(Ctx.Pos(), TString(ConfigProviderName), "ImportUdfs", values, false);
  2413. } else if (normalizedPragma == "packageversion") {
  2414. if (values.size() != 2 || pragmaValueDefault) {
  2415. Error() << "Expected package name and version";
  2416. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2417. return {};
  2418. }
  2419. ui32 version = 0;
  2420. TString versionString;
  2421. TString packageName;
  2422. if (!values[0].GetLiteral(packageName, Ctx) || !values[1].GetLiteral(versionString, Ctx)) {
  2423. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2424. return {};
  2425. }
  2426. if (!PackageVersionFromString(versionString, version)) {
  2427. Error() << "Unable to parse package version, possible values 0, 1, draft, release";
  2428. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2429. return {};
  2430. }
  2431. Ctx.SetPackageVersion(packageName, version);
  2432. Ctx.IncrementMonCounter("sql_pragma", "PackageVersion");
  2433. success = true;
  2434. return BuildPragma(Ctx.Pos(), TString(ConfigProviderName), "SetPackageVersion", TVector<TDeferredAtom>{ values[0], TDeferredAtom(values[1].Build()->GetPos(), ToString(version)) }, false);
  2435. } else if (normalizedPragma == "file") {
  2436. if (values.size() < 2U || values.size() > 3U || pragmaValueDefault) {
  2437. Error() << "Expected file alias, url and optional token name as pragma values";
  2438. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2439. return {};
  2440. }
  2441. Ctx.IncrementMonCounter("sql_pragma", "file");
  2442. success = true;
  2443. return BuildPragma(Ctx.Pos(), TString(ConfigProviderName), "AddFileByUrl", values, false);
  2444. } else if (normalizedPragma == "fileoption") {
  2445. if (values.size() < 3U) {
  2446. Error() << "Expected file alias, option key and value";
  2447. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2448. return {};
  2449. }
  2450. Ctx.IncrementMonCounter("sql_pragma", "FileOption");
  2451. success = true;
  2452. return BuildPragma(Ctx.Pos(), TString(ConfigProviderName), "SetFileOption", values, false);
  2453. } else if (normalizedPragma == "folder") {
  2454. if (values.size() < 2U || values.size() > 3U || pragmaValueDefault) {
  2455. Error() << "Expected folder alias, url and optional token name as pragma values";
  2456. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2457. return {};
  2458. }
  2459. Ctx.IncrementMonCounter("sql_pragma", "folder");
  2460. success = true;
  2461. return BuildPragma(Ctx.Pos(), TString(ConfigProviderName), "AddFolderByUrl", values, false);
  2462. } else if (normalizedPragma == "library") {
  2463. if (values.size() < 1) {
  2464. Error() << "Expected non-empty file alias";
  2465. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2466. return{};
  2467. }
  2468. if (values.size() > 3) {
  2469. Error() << "Expected file alias and optional url and token name as pragma values";
  2470. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2471. return{};
  2472. }
  2473. TString alias;
  2474. if (!values.front().GetLiteral(alias, Ctx)) {
  2475. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2476. return{};
  2477. }
  2478. TContext::TLibraryStuff library;
  2479. std::get<TPosition>(library) = values.front().Build()->GetPos();
  2480. if (values.size() > 1) {
  2481. auto& first = std::get<1U>(library);
  2482. first.emplace();
  2483. first->second = values[1].Build()->GetPos();
  2484. if (!values[1].GetLiteral(first->first, Ctx)) {
  2485. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2486. return{};
  2487. }
  2488. TSet<TString> names;
  2489. SubstParameters(first->first, Nothing(), &names);
  2490. for (const auto& name : names) {
  2491. auto namedNode = GetNamedNode(name);
  2492. if (!namedNode) {
  2493. return{};
  2494. }
  2495. }
  2496. if (values.size() > 2) {
  2497. auto& second = std::get<2U>(library);
  2498. second.emplace();
  2499. second->second = values[2].Build()->GetPos();
  2500. if (!values[2].GetLiteral(second->first, Ctx)) {
  2501. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2502. return{};
  2503. }
  2504. }
  2505. }
  2506. Ctx.Libraries[alias] = std::move(library);
  2507. Ctx.IncrementMonCounter("sql_pragma", "library");
  2508. } else if (normalizedPragma == "package") {
  2509. if (values.size() < 2U || values.size() > 3U) {
  2510. Error() << "Expected package name, url and optional token name as pragma values";
  2511. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2512. return {};
  2513. }
  2514. TString packageName;
  2515. if (!values.front().GetLiteral(packageName, Ctx)) {
  2516. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2517. return {};
  2518. }
  2519. TContext::TPackageStuff package;
  2520. std::get<TPosition>(package) = values.front().Build()->GetPos();
  2521. auto fillLiteral = [&](auto& literal, size_t index) {
  2522. if (values.size() <= index) {
  2523. return true;
  2524. }
  2525. constexpr bool optional = std::is_base_of_v<
  2526. std::optional<TContext::TLiteralWithPosition>,
  2527. std::decay_t<decltype(literal)>
  2528. >;
  2529. TContext::TLiteralWithPosition* literalPtr;
  2530. if constexpr (optional) {
  2531. literal.emplace();
  2532. literalPtr = &*literal;
  2533. } else {
  2534. literalPtr = &literal;
  2535. }
  2536. literalPtr->second = values[index].Build()->GetPos();
  2537. if (!values[index].GetLiteral(literalPtr->first, Ctx)) {
  2538. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2539. return false;
  2540. }
  2541. return true;
  2542. };
  2543. // fill url
  2544. auto& urlLiteral = std::get<1U>(package);
  2545. if (!fillLiteral(urlLiteral, 1U)) {
  2546. return {};
  2547. }
  2548. TSet<TString> names;
  2549. SubstParameters(urlLiteral.first, Nothing(), &names);
  2550. for (const auto& name : names) {
  2551. auto namedNode = GetNamedNode(name);
  2552. if (!namedNode) {
  2553. return {};
  2554. }
  2555. }
  2556. // fill token
  2557. if (!fillLiteral(std::get<2U>(package), 2U)) {
  2558. return {};
  2559. }
  2560. Ctx.Packages[packageName] = std::move(package);
  2561. Ctx.IncrementMonCounter("sql_pragma", "package");
  2562. } else if (normalizedPragma == "overridelibrary") {
  2563. if (values.size() != 1U) {
  2564. Error() << "Expected override library alias as pragma value";
  2565. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2566. return {};
  2567. }
  2568. TString alias;
  2569. if (!values.front().GetLiteral(alias, Ctx)) {
  2570. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2571. return {};
  2572. }
  2573. TContext::TOverrideLibraryStuff overrideLibrary;
  2574. std::get<TPosition>(overrideLibrary) = values.front().Build()->GetPos();
  2575. Ctx.OverrideLibraries[alias] = std::move(overrideLibrary);
  2576. Ctx.IncrementMonCounter("sql_pragma", "overridelibrary");
  2577. } else if (normalizedPragma == "directread") {
  2578. Ctx.PragmaDirectRead = true;
  2579. Ctx.IncrementMonCounter("sql_pragma", "DirectRead");
  2580. } else if (normalizedPragma == "equijoin") {
  2581. Ctx.IncrementMonCounter("sql_pragma", "EquiJoin");
  2582. } else if (normalizedPragma == "autocommit") {
  2583. Ctx.PragmaAutoCommit = true;
  2584. Ctx.IncrementMonCounter("sql_pragma", "AutoCommit");
  2585. } else if (normalizedPragma == "usetableprefixforeach") {
  2586. Ctx.PragmaUseTablePrefixForEach = true;
  2587. Ctx.IncrementMonCounter("sql_pragma", "UseTablePrefixForEach");
  2588. } else if (normalizedPragma == "tablepathprefix") {
  2589. TString value;
  2590. TMaybe<TString> arg;
  2591. if (values.size() == 1 || values.size() == 2) {
  2592. if (!values.front().GetLiteral(value, Ctx)) {
  2593. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2594. return {};
  2595. }
  2596. if (values.size() == 2) {
  2597. arg = value;
  2598. if (!values.back().GetLiteral(value, Ctx)) {
  2599. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2600. return {};
  2601. }
  2602. }
  2603. if (!Ctx.SetPathPrefix(value, arg)) {
  2604. return {};
  2605. }
  2606. } else {
  2607. Error() << "Expected path prefix or tuple of (Provider, PathPrefix) or"
  2608. << " (Cluster, PathPrefix) as pragma value";
  2609. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2610. return {};
  2611. }
  2612. Ctx.IncrementMonCounter("sql_pragma", "PathPrefix");
  2613. } else if (normalizedPragma == "groupbylimit") {
  2614. if (values.size() != 1 || !values[0].GetLiteral() || !TryFromString(*values[0].GetLiteral(), Ctx.PragmaGroupByLimit)) {
  2615. Error() << "Expected unsigned integer literal as a single argument for: " << pragma;
  2616. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2617. return {};
  2618. }
  2619. Ctx.IncrementMonCounter("sql_pragma", "GroupByLimit");
  2620. } else if (normalizedPragma == "groupbycubelimit") {
  2621. if (values.size() != 1 || !values[0].GetLiteral() || !TryFromString(*values[0].GetLiteral(), Ctx.PragmaGroupByCubeLimit)) {
  2622. Error() << "Expected unsigned integer literal as a single argument for: " << pragma;
  2623. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2624. return {};
  2625. }
  2626. Ctx.IncrementMonCounter("sql_pragma", "GroupByCubeLimit");
  2627. } else if (normalizedPragma == "simplecolumns") {
  2628. Ctx.SimpleColumns = true;
  2629. Ctx.IncrementMonCounter("sql_pragma", "SimpleColumns");
  2630. } else if (normalizedPragma == "disablesimplecolumns") {
  2631. Ctx.SimpleColumns = false;
  2632. Ctx.IncrementMonCounter("sql_pragma", "DisableSimpleColumns");
  2633. } else if (normalizedPragma == "coalescejoinkeysonqualifiedall") {
  2634. Ctx.CoalesceJoinKeysOnQualifiedAll = true;
  2635. Ctx.IncrementMonCounter("sql_pragma", "CoalesceJoinKeysOnQualifiedAll");
  2636. } else if (normalizedPragma == "disablecoalescejoinkeysonqualifiedall") {
  2637. Ctx.CoalesceJoinKeysOnQualifiedAll = false;
  2638. Ctx.IncrementMonCounter("sql_pragma", "DisableCoalesceJoinKeysOnQualifiedAll");
  2639. } else if (normalizedPragma == "resultrowslimit") {
  2640. if (values.size() != 1 || !values[0].GetLiteral() || !TryFromString(*values[0].GetLiteral(), Ctx.ResultRowsLimit)) {
  2641. Error() << "Expected unsigned integer literal as a single argument for: " << pragma;
  2642. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2643. return {};
  2644. }
  2645. Ctx.IncrementMonCounter("sql_pragma", "ResultRowsLimit");
  2646. } else if (normalizedPragma == "resultsizelimit") {
  2647. if (values.size() != 1 || !values[0].GetLiteral() || !TryFromString(*values[0].GetLiteral(), Ctx.ResultSizeLimit)) {
  2648. Error() << "Expected unsigned integer literal as a single argument for: " << pragma;
  2649. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2650. return {};
  2651. }
  2652. Ctx.IncrementMonCounter("sql_pragma", "ResultSizeLimit");
  2653. } else if (normalizedPragma == "warning") {
  2654. if (values.size() != 2U || values.front().Empty() || values.back().Empty()) {
  2655. Error() << "Expected arguments <action>, <issueId> for: " << pragma;
  2656. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2657. return {};
  2658. }
  2659. TString action;
  2660. TString codePattern;
  2661. if (!values[0].GetLiteral(action, Ctx) || !values[1].GetLiteral(codePattern, Ctx)) {
  2662. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2663. return {};
  2664. }
  2665. TWarningRule rule;
  2666. TString parseError;
  2667. auto parseResult = TWarningRule::ParseFrom(codePattern, action, rule, parseError);
  2668. switch (parseResult) {
  2669. case TWarningRule::EParseResult::PARSE_OK:
  2670. break;
  2671. case TWarningRule::EParseResult::PARSE_PATTERN_FAIL:
  2672. case TWarningRule::EParseResult::PARSE_ACTION_FAIL:
  2673. Ctx.Error() << parseError;
  2674. return {};
  2675. default:
  2676. Y_ENSURE(false, "Unknown parse result");
  2677. }
  2678. Ctx.WarningPolicy.AddRule(rule);
  2679. if (rule.GetPattern() == "*" && rule.GetAction() == EWarningAction::ERROR) {
  2680. // Keep 'unused symbol' warning as warning unless explicitly set to error
  2681. Ctx.SetWarningPolicyFor(TIssuesIds::YQL_UNUSED_SYMBOL, EWarningAction::DEFAULT);
  2682. }
  2683. Ctx.IncrementMonCounter("sql_pragma", "warning");
  2684. } else if (normalizedPragma == "greetings") {
  2685. if (values.size() > 1) {
  2686. Error() << "Multiple arguments are not expected for " << pragma;
  2687. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2688. return {};
  2689. }
  2690. if (values.empty()) {
  2691. values.emplace_back(TDeferredAtom(Ctx.Pos(), "Hello, world! And best wishes from the YQL Team!"));
  2692. }
  2693. TString arg;
  2694. if (!values.front().GetLiteral(arg, Ctx)) {
  2695. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2696. return {};
  2697. }
  2698. Ctx.Info(Ctx.Pos()) << arg;
  2699. } else if (normalizedPragma == "warningmsg") {
  2700. if (values.size() != 1 || !values[0].GetLiteral()) {
  2701. Error() << "Expected string literal as a single argument for: " << pragma;
  2702. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2703. return {};
  2704. }
  2705. Ctx.Warning(Ctx.Pos(), TIssuesIds::YQL_PRAGMA_WARNING_MSG) << *values[0].GetLiteral();
  2706. } else if (normalizedPragma == "errormsg") {
  2707. if (values.size() != 1 || !values[0].GetLiteral()) {
  2708. Error() << "Expected string literal as a single argument for: " << pragma;
  2709. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2710. return {};
  2711. }
  2712. Ctx.Error(Ctx.Pos()) << *values[0].GetLiteral();
  2713. } else if (normalizedPragma == "classicdivision") {
  2714. if (values.size() != 1 || !values[0].GetLiteral() || !TryFromString(*values[0].GetLiteral(), Ctx.Scoped->PragmaClassicDivision)) {
  2715. Error() << "Expected boolean literal as a single argument for: " << pragma;
  2716. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2717. return {};
  2718. }
  2719. Ctx.IncrementMonCounter("sql_pragma", "ClassicDivision");
  2720. } else if (normalizedPragma == "checkedops") {
  2721. if (values.size() != 1 || !values[0].GetLiteral() || !TryFromString(*values[0].GetLiteral(), Ctx.Scoped->PragmaCheckedOps)) {
  2722. Error() << "Expected boolean literal as a single argument for: " << pragma;
  2723. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2724. return {};
  2725. }
  2726. Ctx.IncrementMonCounter("sql_pragma", "CheckedOps");
  2727. } else if (normalizedPragma == "disableunordered") {
  2728. Ctx.Warning(Ctx.Pos(), TIssuesIds::YQL_DEPRECATED_PRAGMA)
  2729. << "Use of deprecated DisableUnordered pragma. It will be dropped soon";
  2730. } else if (normalizedPragma == "pullupflatmapoverjoin") {
  2731. Ctx.PragmaPullUpFlatMapOverJoin = true;
  2732. Ctx.IncrementMonCounter("sql_pragma", "PullUpFlatMapOverJoin");
  2733. } else if (normalizedPragma == "disablepullupflatmapoverjoin") {
  2734. Ctx.PragmaPullUpFlatMapOverJoin = false;
  2735. Ctx.IncrementMonCounter("sql_pragma", "DisablePullUpFlatMapOverJoin");
  2736. } else if (normalizedPragma == "filterpushdownoverjoinoptionalside") {
  2737. Ctx.FilterPushdownOverJoinOptionalSide = true;
  2738. Ctx.IncrementMonCounter("sql_pragma", "FilterPushdownOverJoinOptionalSide");
  2739. } else if (normalizedPragma == "disablefilterpushdownoverjoinoptionalside") {
  2740. Ctx.FilterPushdownOverJoinOptionalSide = false;
  2741. Ctx.IncrementMonCounter("sql_pragma", "DisableFilterPushdownOverJoinOptionalSide");
  2742. } else if (normalizedPragma == "rotatejointree") {
  2743. if (values.size() != 1 || !values[0].GetLiteral() || !TryFromString(*values[0].GetLiteral(), Ctx.RotateJoinTree)) {
  2744. Error() << "Expected boolean literal as a single argument for: " << pragma;
  2745. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2746. return {};
  2747. }
  2748. } else if (normalizedPragma == "allowunnamedcolumns") {
  2749. Ctx.WarnUnnamedColumns = false;
  2750. Ctx.IncrementMonCounter("sql_pragma", "AllowUnnamedColumns");
  2751. } else if (normalizedPragma == "warnunnamedcolumns") {
  2752. Ctx.WarnUnnamedColumns = true;
  2753. Ctx.IncrementMonCounter("sql_pragma", "WarnUnnamedColumns");
  2754. } else if (normalizedPragma == "discoverymode") {
  2755. Ctx.DiscoveryMode = true;
  2756. Ctx.IncrementMonCounter("sql_pragma", "DiscoveryMode");
  2757. } else if (normalizedPragma == "enablesystemcolumns") {
  2758. if (values.size() != 1 || !values[0].GetLiteral() || !TryFromString(*values[0].GetLiteral(), Ctx.EnableSystemColumns)) {
  2759. Error() << "Expected boolean literal as a single argument for: " << pragma;
  2760. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2761. return {};
  2762. }
  2763. Ctx.IncrementMonCounter("sql_pragma", "EnableSystemColumns");
  2764. } else if (normalizedPragma == "ansiinforemptyornullableitemscollections") {
  2765. Ctx.AnsiInForEmptyOrNullableItemsCollections = true;
  2766. Ctx.IncrementMonCounter("sql_pragma", "AnsiInForEmptyOrNullableItemsCollections");
  2767. } else if (normalizedPragma == "disableansiinforemptyornullableitemscollections") {
  2768. Ctx.AnsiInForEmptyOrNullableItemsCollections = false;
  2769. Ctx.IncrementMonCounter("sql_pragma", "DisableAnsiInForEmptyOrNullableItemsCollections");
  2770. } else if (normalizedPragma == "dqengine" || normalizedPragma == "blockengine") {
  2771. Ctx.IncrementMonCounter("sql_pragma", "DqEngine");
  2772. if (values.size() != 1 || !values[0].GetLiteral()
  2773. || ! (*values[0].GetLiteral() == "disable" || *values[0].GetLiteral() == "auto" || *values[0].GetLiteral() == "force"))
  2774. {
  2775. Error() << "Expected `disable|auto|force' argument for: " << pragma;
  2776. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2777. return {};
  2778. }
  2779. const bool isDqEngine = normalizedPragma == "dqengine";
  2780. auto& enable = isDqEngine ? Ctx.DqEngineEnable : Ctx.BlockEngineEnable;
  2781. auto& force = isDqEngine ? Ctx.DqEngineForce : Ctx.BlockEngineForce;
  2782. if (*values[0].GetLiteral() == "disable") {
  2783. enable = false;
  2784. force = false;
  2785. } else if (*values[0].GetLiteral() == "force") {
  2786. enable = true;
  2787. force = true;
  2788. } else if (*values[0].GetLiteral() == "auto") {
  2789. enable = true;
  2790. force = false;
  2791. }
  2792. } else if (normalizedPragma == "ansirankfornullablekeys") {
  2793. Ctx.AnsiRankForNullableKeys = true;
  2794. Ctx.IncrementMonCounter("sql_pragma", "AnsiRankForNullableKeys");
  2795. } else if (normalizedPragma == "disableansirankfornullablekeys") {
  2796. Ctx.AnsiRankForNullableKeys = false;
  2797. Ctx.IncrementMonCounter("sql_pragma", "DisableAnsiRankForNullableKeys");
  2798. } else if (normalizedPragma == "ansiorderbylimitinunionall") {
  2799. Ctx.IncrementMonCounter("sql_pragma", "AnsiOrderByLimitInUnionAll");
  2800. } else if (normalizedPragma == "disableansiorderbylimitinunionall") {
  2801. Error() << "DisableAnsiOrderByLimitInUnionAll pragma is deprecated and no longer supported";
  2802. Ctx.IncrementMonCounter("sql_errors", "DeprecatedPragma");
  2803. return {};
  2804. } else if (normalizedPragma == "ansioptionalas") {
  2805. Ctx.AnsiOptionalAs = true;
  2806. Ctx.IncrementMonCounter("sql_pragma", "AnsiOptionalAs");
  2807. } else if (normalizedPragma == "disableansioptionalas") {
  2808. Ctx.AnsiOptionalAs = false;
  2809. Ctx.IncrementMonCounter("sql_pragma", "DisableAnsiOptionalAs");
  2810. } else if (normalizedPragma == "warnonansialiasshadowing") {
  2811. Ctx.WarnOnAnsiAliasShadowing = true;
  2812. Ctx.IncrementMonCounter("sql_pragma", "WarnOnAnsiAliasShadowing");
  2813. } else if (normalizedPragma == "disablewarnonansialiasshadowing") {
  2814. Ctx.WarnOnAnsiAliasShadowing = false;
  2815. Ctx.IncrementMonCounter("sql_pragma", "DisableWarnOnAnsiAliasShadowing");
  2816. } else if (normalizedPragma == "regexusere2") {
  2817. if (values.size() != 1U || !values.front().GetLiteral() || !TryFromString(*values.front().GetLiteral(), Ctx.PragmaRegexUseRe2)) {
  2818. Error() << "Expected 'true' or 'false' for: " << pragma;
  2819. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2820. return {};
  2821. }
  2822. Ctx.IncrementMonCounter("sql_pragma", "RegexUseRe2");
  2823. } else if (normalizedPragma == "jsonqueryreturnsjsondocument") {
  2824. Ctx.JsonQueryReturnsJsonDocument = true;
  2825. Ctx.IncrementMonCounter("sql_pragma", "JsonQueryReturnsJsonDocument");
  2826. } else if (normalizedPragma == "disablejsonqueryreturnsjsondocument") {
  2827. Ctx.JsonQueryReturnsJsonDocument = false;
  2828. Ctx.IncrementMonCounter("sql_pragma", "DisableJsonQueryReturnsJsonDocument");
  2829. } else if (normalizedPragma == "orderedcolumns") {
  2830. Ctx.OrderedColumns = true;
  2831. Ctx.IncrementMonCounter("sql_pragma", "OrderedColumns");
  2832. } else if (normalizedPragma == "disableorderedcolumns") {
  2833. Ctx.OrderedColumns = false;
  2834. Ctx.IncrementMonCounter("sql_pragma", "DisableOrderedColumns");
  2835. } else if (normalizedPragma == "positionalunionall") {
  2836. Ctx.PositionalUnionAll = true;
  2837. // PositionalUnionAll implies OrderedColumns
  2838. Ctx.OrderedColumns = true;
  2839. Ctx.IncrementMonCounter("sql_pragma", "PositionalUnionAll");
  2840. } else if (normalizedPragma == "pqreadby") {
  2841. if (values.size() != 1 || !values[0].GetLiteral()) {
  2842. Error() << "Expected string literal as a single argument for: " << pragma;
  2843. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2844. return {};
  2845. }
  2846. // special guard to raise error on situation:
  2847. // use cluster1;
  2848. // pragma PqReadPqBy="cluster2";
  2849. const TString* currentClusterLiteral = Ctx.Scoped->CurrCluster.GetLiteral();
  2850. if (currentClusterLiteral && *values[0].GetLiteral() != "dq" && *currentClusterLiteral != *values[0].GetLiteral()) {
  2851. Error() << "Cluster in PqReadPqBy pragma differs from cluster specified in USE statement: " << *values[0].GetLiteral() << " != " << *currentClusterLiteral;
  2852. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2853. return {};
  2854. }
  2855. Ctx.PqReadByRtmrCluster = *values[0].GetLiteral();
  2856. Ctx.IncrementMonCounter("sql_pragma", "PqReadBy");
  2857. } else if (normalizedPragma == "bogousstaringroupbyoverjoin") {
  2858. Ctx.BogousStarInGroupByOverJoin = true;
  2859. Ctx.IncrementMonCounter("sql_pragma", "BogousStarInGroupByOverJoin");
  2860. } else if (normalizedPragma == "strictjoinkeytypes") {
  2861. Ctx.Scoped->StrictJoinKeyTypes = true;
  2862. Ctx.IncrementMonCounter("sql_pragma", "StrictJoinKeyTypes");
  2863. } else if (normalizedPragma == "disablestrictjoinkeytypes") {
  2864. Ctx.Scoped->StrictJoinKeyTypes = false;
  2865. Ctx.IncrementMonCounter("sql_pragma", "DisableStrictJoinKeyTypes");
  2866. } else if (normalizedPragma == "unicodeliterals") {
  2867. Ctx.Scoped->UnicodeLiterals = true;
  2868. Ctx.IncrementMonCounter("sql_pragma", "UnicodeLiterals");
  2869. } else if (normalizedPragma == "disableunicodeliterals") {
  2870. Ctx.Scoped->UnicodeLiterals = false;
  2871. Ctx.IncrementMonCounter("sql_pragma", "DisableUnicodeLiterals");
  2872. } else if (normalizedPragma == "warnuntypedstringliterals") {
  2873. Ctx.Scoped->WarnUntypedStringLiterals = true;
  2874. Ctx.IncrementMonCounter("sql_pragma", "WarnUntypedStringLiterals");
  2875. } else if (normalizedPragma == "disablewarnuntypedstringliterals") {
  2876. Ctx.Scoped->WarnUntypedStringLiterals = false;
  2877. Ctx.IncrementMonCounter("sql_pragma", "DisableWarnUntypedStringLiterals");
  2878. } else if (normalizedPragma == "unorderedsubqueries") {
  2879. Ctx.UnorderedSubqueries = true;
  2880. Ctx.IncrementMonCounter("sql_pragma", "UnorderedSubqueries");
  2881. } else if (normalizedPragma == "disableunorderedsubqueries") {
  2882. Ctx.UnorderedSubqueries = false;
  2883. Ctx.IncrementMonCounter("sql_pragma", "DisableUnorderedSubqueries");
  2884. } else if (normalizedPragma == "datawatermarks") {
  2885. if (values.size() != 1 || !values[0].GetLiteral()
  2886. || ! (*values[0].GetLiteral() == "enable" || *values[0].GetLiteral() == "disable"))
  2887. {
  2888. Error() << "Expected `enable|disable' argument for: " << pragma;
  2889. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2890. return {};
  2891. }
  2892. if (*values[0].GetLiteral() == "enable") {
  2893. Ctx.PragmaDataWatermarks = true;
  2894. } else if (*values[0].GetLiteral() == "disable") {
  2895. Ctx.PragmaDataWatermarks = false;
  2896. }
  2897. Ctx.IncrementMonCounter("sql_pragma", "DataWatermarks");
  2898. } else if (normalizedPragma == "flexibletypes") {
  2899. Ctx.FlexibleTypes = true;
  2900. Ctx.IncrementMonCounter("sql_pragma", "FlexibleTypes");
  2901. } else if (normalizedPragma == "disableflexibletypes") {
  2902. Ctx.Warning(Ctx.Pos(), TIssuesIds::YQL_DEPRECATED_PRAGMA)
  2903. << "Deprecated pragma DisableFlexibleTypes - it will be removed soon. "
  2904. "Consider submitting bug report if FlexibleTypes doesn't work for you";
  2905. Ctx.FlexibleTypes = false;
  2906. Ctx.IncrementMonCounter("sql_pragma", "DisableFlexibleTypes");
  2907. } else if (normalizedPragma == "ansicurrentrow") {
  2908. Ctx.AnsiCurrentRow = true;
  2909. Ctx.IncrementMonCounter("sql_pragma", "AnsiCurrentRow");
  2910. } else if (normalizedPragma == "disableansicurrentrow") {
  2911. Ctx.AnsiCurrentRow = false;
  2912. Ctx.IncrementMonCounter("sql_pragma", "DisableAnsiCurrentRow");
  2913. } else if (normalizedPragma == "emitaggapply") {
  2914. Ctx.EmitAggApply = true;
  2915. Ctx.IncrementMonCounter("sql_pragma", "EmitAggApply");
  2916. } else if (normalizedPragma == "disableemitaggapply") {
  2917. Ctx.EmitAggApply = false;
  2918. Ctx.IncrementMonCounter("sql_pragma", "DisableEmitAggApply");
  2919. } else if (normalizedPragma == "useblocks") {
  2920. Ctx.UseBlocks = true;
  2921. Ctx.IncrementMonCounter("sql_pragma", "UseBlocks");
  2922. } else if (normalizedPragma == "disableuseblocks") {
  2923. Ctx.UseBlocks = false;
  2924. Ctx.IncrementMonCounter("sql_pragma", "DisableUseBlocks");
  2925. } else if (normalizedPragma == "emittablesource") {
  2926. Ctx.EmitTableSource = true;
  2927. Ctx.IncrementMonCounter("sql_pragma", "EmitTableSource");
  2928. } else if (normalizedPragma == "disableemittablesource") {
  2929. Ctx.EmitTableSource = false;
  2930. Ctx.IncrementMonCounter("sql_pragma", "DisableEmitTableSource");
  2931. } else if (normalizedPragma == "ansilike") {
  2932. Ctx.AnsiLike = true;
  2933. Ctx.IncrementMonCounter("sql_pragma", "AnsiLike");
  2934. } else if (normalizedPragma == "disableansilike") {
  2935. Ctx.AnsiLike = false;
  2936. Ctx.IncrementMonCounter("sql_pragma", "DisableAnsiLike");
  2937. } else if (normalizedPragma == "unorderedresult") {
  2938. Ctx.UnorderedResult = true;
  2939. Ctx.IncrementMonCounter("sql_pragma", "UnorderedResult");
  2940. } else if (normalizedPragma == "disableunorderedresult") {
  2941. Ctx.UnorderedResult = false;
  2942. Ctx.IncrementMonCounter("sql_pragma", "DisableUnorderedResult");
  2943. } else if (normalizedPragma == "featurer010") {
  2944. if (values.size() == 1 && values[0].GetLiteral()) {
  2945. const auto& value = *values[0].GetLiteral();
  2946. if ("prototype" == value)
  2947. Ctx.FeatureR010 = true;
  2948. else {
  2949. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2950. return {};
  2951. }
  2952. }
  2953. else {
  2954. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2955. return {};
  2956. }
  2957. Ctx.IncrementMonCounter("sql_pragma", "FeatureR010");
  2958. } else if (normalizedPragma == "compactgroupby") {
  2959. Ctx.CompactGroupBy = true;
  2960. Ctx.IncrementMonCounter("sql_pragma", "CompactGroupBy");
  2961. } else if (normalizedPragma == "disablecompactgroupby") {
  2962. Ctx.CompactGroupBy = false;
  2963. Ctx.IncrementMonCounter("sql_pragma", "DisableCompactGroupBy");
  2964. } else if (normalizedPragma == "costbasedoptimizer") {
  2965. Ctx.IncrementMonCounter("sql_pragma", "CostBasedOptimizer");
  2966. if (values.size() == 1 && values[0].GetLiteral()) {
  2967. Ctx.CostBasedOptimizer = to_lower(*values[0].GetLiteral());
  2968. }
  2969. if (values.size() != 1 || !values[0].GetLiteral()
  2970. || ! (Ctx.CostBasedOptimizer == "disable" || Ctx.CostBasedOptimizer == "pg" || Ctx.CostBasedOptimizer == "native"))
  2971. {
  2972. Error() << "Expected `disable|pg|native' argument for: " << pragma;
  2973. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  2974. return {};
  2975. }
  2976. } else if (normalizedPragma == "compactnamedexprs") {
  2977. Ctx.CompactNamedExprs = true;
  2978. Ctx.IncrementMonCounter("sql_pragma", "CompactNamedExprs");
  2979. } else if (normalizedPragma == "disablecompactnamedexprs") {
  2980. Ctx.Warning(Ctx.Pos(), TIssuesIds::YQL_DEPRECATED_PRAGMA)
  2981. << "Deprecated pragma DisableCompactNamedExprs - it will be removed soon. "
  2982. "Consider submitting bug report if CompactNamedExprs doesn't work for you";
  2983. Ctx.CompactNamedExprs = false;
  2984. Ctx.IncrementMonCounter("sql_pragma", "DisableCompactNamedExprs");
  2985. } else if (normalizedPragma == "validateunusedexprs") {
  2986. Ctx.ValidateUnusedExprs = true;
  2987. Ctx.IncrementMonCounter("sql_pragma", "ValidateUnusedExprs");
  2988. } else if (normalizedPragma == "disablevalidateunusedexprs") {
  2989. Ctx.ValidateUnusedExprs = false;
  2990. Ctx.IncrementMonCounter("sql_pragma", "DisableValidateUnusedExprs");
  2991. } else if (normalizedPragma == "ansiimplicitcrossjoin") {
  2992. Ctx.AnsiImplicitCrossJoin = true;
  2993. Ctx.IncrementMonCounter("sql_pragma", "AnsiImplicitCrossJoin");
  2994. } else if (normalizedPragma == "disableansiimplicitcrossjoin") {
  2995. Ctx.AnsiImplicitCrossJoin = false;
  2996. Ctx.IncrementMonCounter("sql_pragma", "DisableAnsiImplicitCrossJoin");
  2997. } else if (normalizedPragma == "distinctoverwindow") {
  2998. Ctx.DistinctOverWindow = true;
  2999. Ctx.IncrementMonCounter("sql_pragma", "DistinctOverWindow");
  3000. } else if (normalizedPragma == "disabledistinctoverwindow") {
  3001. Ctx.DistinctOverWindow = false;
  3002. Ctx.IncrementMonCounter("sql_pragma", "DisableDistinctOverWindow");
  3003. } else if (normalizedPragma == "seqmode") {
  3004. Ctx.SeqMode = true;
  3005. Ctx.IncrementMonCounter("sql_pragma", "SeqMode");
  3006. } else if (normalizedPragma == "disableseqmode") {
  3007. Ctx.SeqMode = false;
  3008. Ctx.IncrementMonCounter("sql_pragma", "DisableSeqMode");
  3009. } else if (normalizedPragma == "emitunionmerge") {
  3010. Ctx.EmitUnionMerge = true;
  3011. Ctx.IncrementMonCounter("sql_pragma", "EmitUnionMerge");
  3012. } else if (normalizedPragma == "disableemitunionmerge") {
  3013. Ctx.EmitUnionMerge = false;
  3014. Ctx.IncrementMonCounter("sql_pragma", "DisableEmitUnionMerge");
  3015. } else {
  3016. Error() << "Unknown pragma: " << pragma;
  3017. Ctx.IncrementMonCounter("sql_errors", "UnknownPragma");
  3018. return {};
  3019. }
  3020. } else {
  3021. if (lowerPrefix == "yson") {
  3022. if (!TopLevel) {
  3023. Error() << "This pragma '" << pragma << "' is not allowed to be used in actions";
  3024. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  3025. return {};
  3026. }
  3027. if (normalizedPragma == "fast") {
  3028. Ctx.Warning(Ctx.Pos(), TIssuesIds::YQL_DEPRECATED_PRAGMA)
  3029. << "Use of deprecated yson.Fast pragma. It will be dropped soon";
  3030. success = true;
  3031. return {};
  3032. } else if (normalizedPragma == "autoconvert") {
  3033. Ctx.PragmaYsonAutoConvert = true;
  3034. success = true;
  3035. return {};
  3036. } else if (normalizedPragma == "strict") {
  3037. if (values.size() == 0U) {
  3038. Ctx.PragmaYsonStrict = true;
  3039. success = true;
  3040. } else if (values.size() == 1U && values.front().GetLiteral() && TryFromString(*values.front().GetLiteral(), Ctx.PragmaYsonStrict)) {
  3041. success = true;
  3042. } else {
  3043. Error() << "Expected 'true', 'false' or no parameter for: " << pragma;
  3044. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  3045. }
  3046. return {};
  3047. } else if (normalizedPragma == "disablestrict") {
  3048. if (values.size() == 0U) {
  3049. Ctx.PragmaYsonStrict = false;
  3050. success = true;
  3051. return {};
  3052. }
  3053. bool pragmaYsonDisableStrict;
  3054. if (values.size() == 1U && values.front().GetLiteral() && TryFromString(*values.front().GetLiteral(), pragmaYsonDisableStrict)) {
  3055. Ctx.PragmaYsonStrict = !pragmaYsonDisableStrict;
  3056. success = true;
  3057. } else {
  3058. Error() << "Expected 'true', 'false' or no parameter for: " << pragma;
  3059. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  3060. }
  3061. return {};
  3062. } else if (normalizedPragma == "casttostring" || normalizedPragma == "disablecasttostring") {
  3063. const bool allow = normalizedPragma == "casttostring";
  3064. if (values.size() == 0U) {
  3065. Ctx.YsonCastToString = allow;
  3066. success = true;
  3067. return {};
  3068. }
  3069. bool pragmaYsonCastToString;
  3070. if (values.size() == 1U && values.front().GetLiteral() && TryFromString(*values.front().GetLiteral(), pragmaYsonCastToString)) {
  3071. Ctx.PragmaYsonStrict = allow ? pragmaYsonCastToString : !pragmaYsonCastToString;
  3072. success = true;
  3073. } else {
  3074. Error() << "Expected 'true', 'false' or no parameter for: " << pragma;
  3075. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  3076. }
  3077. return {};
  3078. } else {
  3079. Error() << "Unknown pragma: '" << pragma << "'";
  3080. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  3081. return {};
  3082. }
  3083. } else if (std::find(Providers.cbegin(), Providers.cend(), lowerPrefix) == Providers.cend()) {
  3084. if (!Ctx.HasCluster(prefix)) {
  3085. Error() << "Unknown pragma prefix: " << prefix << ", please use cluster name or one of provider " <<
  3086. JoinRange(", ", Providers.cbegin(), Providers.cend());
  3087. Ctx.IncrementMonCounter("sql_errors", "UnknownPragma");
  3088. return {};
  3089. }
  3090. }
  3091. if (normalizedPragma != "flags" && normalizedPragma != "packageversion") {
  3092. if (values.size() > 1) {
  3093. Error() << "Expected at most one value in the pragma";
  3094. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  3095. return {};
  3096. }
  3097. } else {
  3098. if (pragmaValueDefault || values.size() < 1) {
  3099. Error() << "Expected at least one value in the pragma";
  3100. Ctx.IncrementMonCounter("sql_errors", "BadPragmaValue");
  3101. return {};
  3102. }
  3103. }
  3104. success = true;
  3105. Ctx.IncrementMonCounter("sql_pragma", pragma);
  3106. return BuildPragma(Ctx.Pos(), lowerPrefix, normalizedPragma, values, pragmaValueDefault);
  3107. }
  3108. success = true;
  3109. return {};
  3110. }
  3111. TNodePtr TSqlQuery::Build(const TRule_delete_stmt& stmt) {
  3112. TTableRef table;
  3113. if (!SimpleTableRefImpl(stmt.GetRule_simple_table_ref4(), table)) {
  3114. return nullptr;
  3115. }
  3116. const bool isKikimr = table.Service == KikimrProviderName;
  3117. if (!isKikimr) {
  3118. Ctx.Error(GetPos(stmt.GetToken2())) << "DELETE is unsupported for " << table.Service;
  3119. return nullptr;
  3120. }
  3121. TSourcePtr source = BuildTableSource(Ctx.Pos(), table);
  3122. TNodePtr options = nullptr;
  3123. if (stmt.HasBlock6()) {
  3124. options = ReturningList(stmt.GetBlock6().GetRule_returning_columns_list1());
  3125. options = options->Y(options);
  3126. }
  3127. const bool isBatch = stmt.HasBlock1();
  3128. if (stmt.HasBlock5()) {
  3129. switch (stmt.GetBlock5().Alt_case()) {
  3130. case TRule_delete_stmt_TBlock5::kAlt1: {
  3131. const auto& alt = stmt.GetBlock5().GetAlt1();
  3132. TColumnRefScope scope(Ctx, EColumnRefState::Allow);
  3133. TSqlExpression sqlExpr(Ctx, Mode);
  3134. auto whereExpr = sqlExpr.Build(alt.GetRule_expr2());
  3135. if (!whereExpr) {
  3136. return nullptr;
  3137. }
  3138. source->AddFilter(Ctx, whereExpr);
  3139. break;
  3140. }
  3141. case TRule_delete_stmt_TBlock5::kAlt2: {
  3142. const auto& alt = stmt.GetBlock5().GetAlt2();
  3143. auto values = TSqlIntoValues(Ctx, Mode).Build(alt.GetRule_into_values_source2(), "DELETE ON");
  3144. if (!values) {
  3145. return nullptr;
  3146. }
  3147. if (isBatch) {
  3148. Ctx.Error(GetPos(stmt.GetToken2())) << "BATCH DELETE is unsupported with ON";
  3149. return nullptr;
  3150. }
  3151. return BuildWriteColumns(Ctx.Pos(), Ctx.Scoped, table, EWriteColumnMode::DeleteOn, std::move(values), options);
  3152. }
  3153. case TRule_delete_stmt_TBlock5::ALT_NOT_SET:
  3154. return nullptr;
  3155. }
  3156. }
  3157. if (isBatch) {
  3158. return BuildBatchDelete(Ctx.Pos(), Ctx.Scoped, table, std::move(source), options);
  3159. }
  3160. return BuildDelete(Ctx.Pos(), Ctx.Scoped, table, std::move(source), options);
  3161. }
  3162. TNodePtr TSqlQuery::Build(const TRule_update_stmt& stmt) {
  3163. TTableRef table;
  3164. if (!SimpleTableRefImpl(stmt.GetRule_simple_table_ref3(), table)) {
  3165. return nullptr;
  3166. }
  3167. const bool isKikimr = table.Service == KikimrProviderName;
  3168. if (!isKikimr) {
  3169. Ctx.Error(GetPos(stmt.GetToken2())) << "UPDATE is unsupported for " << table.Service;
  3170. return nullptr;
  3171. }
  3172. TNodePtr options = nullptr;
  3173. if (stmt.HasBlock5()) {
  3174. options = ReturningList(stmt.GetBlock5().GetRule_returning_columns_list1());
  3175. options = options->Y(options);
  3176. }
  3177. const bool isBatch = stmt.HasBlock1();
  3178. switch (stmt.GetBlock4().Alt_case()) {
  3179. case TRule_update_stmt_TBlock4::kAlt1: {
  3180. const auto& alt = stmt.GetBlock4().GetAlt1();
  3181. TSourcePtr values = Build(alt.GetRule_set_clause_choice2());
  3182. auto source = BuildTableSource(Ctx.Pos(), table);
  3183. if (alt.HasBlock3()) {
  3184. TColumnRefScope scope(Ctx, EColumnRefState::Allow);
  3185. TSqlExpression sqlExpr(Ctx, Mode);
  3186. auto whereExpr = sqlExpr.Build(alt.GetBlock3().GetRule_expr2());
  3187. if (!whereExpr) {
  3188. return nullptr;
  3189. }
  3190. source->AddFilter(Ctx, whereExpr);
  3191. }
  3192. if (isBatch) {
  3193. return BuildBatchUpdate(Ctx.Pos(), Ctx.Scoped, table, std::move(values), std::move(source), options);
  3194. }
  3195. return BuildUpdateColumns(Ctx.Pos(), Ctx.Scoped, table, std::move(values), std::move(source), options);
  3196. }
  3197. case TRule_update_stmt_TBlock4::kAlt2: {
  3198. const auto& alt = stmt.GetBlock4().GetAlt2();
  3199. auto values = TSqlIntoValues(Ctx, Mode).Build(alt.GetRule_into_values_source2(), "UPDATE ON");
  3200. if (!values) {
  3201. return nullptr;
  3202. }
  3203. if (isBatch) {
  3204. Ctx.Error(GetPos(stmt.GetToken2())) << "BATCH UPDATE is unsupported with ON";
  3205. return nullptr;
  3206. }
  3207. return BuildWriteColumns(Ctx.Pos(), Ctx.Scoped, table, EWriteColumnMode::UpdateOn, std::move(values), options);
  3208. }
  3209. case TRule_update_stmt_TBlock4::ALT_NOT_SET:
  3210. return nullptr;
  3211. }
  3212. }
  3213. TSourcePtr TSqlQuery::Build(const TRule_set_clause_choice& stmt) {
  3214. switch (stmt.Alt_case()) {
  3215. case TRule_set_clause_choice::kAltSetClauseChoice1:
  3216. return Build(stmt.GetAlt_set_clause_choice1().GetRule_set_clause_list1());
  3217. case TRule_set_clause_choice::kAltSetClauseChoice2:
  3218. return Build(stmt.GetAlt_set_clause_choice2().GetRule_multiple_column_assignment1());
  3219. case TRule_set_clause_choice::ALT_NOT_SET:
  3220. AltNotImplemented("set_clause_choice", stmt);
  3221. return nullptr;
  3222. }
  3223. }
  3224. bool TSqlQuery::FillSetClause(const TRule_set_clause& node, TVector<TString>& targetList, TVector<TNodePtr>& values) {
  3225. targetList.push_back(ColumnNameAsSingleStr(*this, node.GetRule_set_target1().GetRule_column_name1()));
  3226. TColumnRefScope scope(Ctx, EColumnRefState::Allow);
  3227. TSqlExpression sqlExpr(Ctx, Mode);
  3228. if (!Expr(sqlExpr, values, node.GetRule_expr3())) {
  3229. return false;
  3230. }
  3231. return true;
  3232. }
  3233. TSourcePtr TSqlQuery::Build(const TRule_set_clause_list& stmt) {
  3234. TVector<TString> targetList;
  3235. TVector<TNodePtr> values;
  3236. const TPosition pos(Ctx.Pos());
  3237. if (!FillSetClause(stmt.GetRule_set_clause1(), targetList, values)) {
  3238. return nullptr;
  3239. }
  3240. for (auto& block: stmt.GetBlock2()) {
  3241. if (!FillSetClause(block.GetRule_set_clause2(), targetList, values)) {
  3242. return nullptr;
  3243. }
  3244. }
  3245. Y_DEBUG_ABORT_UNLESS(targetList.size() == values.size());
  3246. return BuildUpdateValues(pos, targetList, values);
  3247. }
  3248. TSourcePtr TSqlQuery::Build(const TRule_multiple_column_assignment& stmt) {
  3249. TVector<TString> targetList;
  3250. FillTargetList(*this, stmt.GetRule_set_target_list1(), targetList);
  3251. auto simpleValuesNode = stmt.GetRule_simple_values_source4();
  3252. const TPosition pos(Ctx.Pos());
  3253. switch (simpleValuesNode.Alt_case()) {
  3254. case TRule_simple_values_source::kAltSimpleValuesSource1: {
  3255. TVector<TNodePtr> values;
  3256. TSqlExpression sqlExpr(Ctx, Mode);
  3257. if (!ExprList(sqlExpr, values, simpleValuesNode.GetAlt_simple_values_source1().GetRule_expr_list1())) {
  3258. return nullptr;
  3259. }
  3260. return BuildUpdateValues(pos, targetList, values);
  3261. }
  3262. case TRule_simple_values_source::kAltSimpleValuesSource2: {
  3263. TSqlSelect select(Ctx, Mode);
  3264. TPosition selectPos;
  3265. auto source = select.Build(simpleValuesNode.GetAlt_simple_values_source2().GetRule_select_stmt1(), selectPos);
  3266. if (!source) {
  3267. return nullptr;
  3268. }
  3269. return BuildWriteValues(pos, "UPDATE", targetList, std::move(source));
  3270. }
  3271. case TRule_simple_values_source::ALT_NOT_SET:
  3272. Ctx.IncrementMonCounter("sql_errors", "UnknownSimpleValuesSourceAlt");
  3273. AltNotImplemented("simple_values_source", simpleValuesNode);
  3274. return nullptr;
  3275. }
  3276. }
  3277. TNodePtr TSqlQuery::Build(const TSQLv1ParserAST& ast) {
  3278. if (Mode == NSQLTranslation::ESqlMode::QUERY) {
  3279. // inject externally declared named expressions
  3280. for (auto [name, type] : Ctx.Settings.DeclaredNamedExprs) {
  3281. if (name.empty()) {
  3282. Error() << "Empty names for externally declared expressions are not allowed";
  3283. return nullptr;
  3284. }
  3285. TString varName = "$" + name;
  3286. if (IsAnonymousName(varName)) {
  3287. Error() << "Externally declared name '" << name << "' is anonymous";
  3288. return nullptr;
  3289. }
  3290. auto parsed = ParseType(type, *Ctx.Pool, Ctx.Issues, Ctx.Pos());
  3291. if (!parsed) {
  3292. Error() << "Failed to parse type for externally declared name '" << name << "'";
  3293. return nullptr;
  3294. }
  3295. TNodePtr typeNode = BuildBuiltinFunc(Ctx, Ctx.Pos(), "ParseType", { BuildLiteralRawString(Ctx.Pos(), type) });
  3296. PushNamedAtom(Ctx.Pos(), varName);
  3297. // no duplicates are possible at this stage
  3298. bool isWeak = true;
  3299. Ctx.DeclareVariable(varName, {}, typeNode, isWeak);
  3300. // avoid 'Symbol is not used' warning for externally declared expression
  3301. YQL_ENSURE(GetNamedNode(varName));
  3302. }
  3303. }
  3304. const auto& query = ast.GetRule_sql_query();
  3305. TVector<TNodePtr> blocks;
  3306. Ctx.PushCurrentBlocks(&blocks);
  3307. Y_DEFER {
  3308. Ctx.PopCurrentBlocks();
  3309. };
  3310. if (query.Alt_case() == TRule_sql_query::kAltSqlQuery1) {
  3311. size_t statementNumber = 0;
  3312. const auto& statements = query.GetAlt_sql_query1().GetRule_sql_stmt_list1();
  3313. if (!Statement(blocks, statements.GetRule_sql_stmt2().GetRule_sql_stmt_core2(), statementNumber++)) {
  3314. return nullptr;
  3315. }
  3316. for (auto block: statements.GetBlock3()) {
  3317. if (!Statement(blocks, block.GetRule_sql_stmt2().GetRule_sql_stmt_core2(), statementNumber++)) {
  3318. return nullptr;
  3319. }
  3320. }
  3321. }
  3322. ui32 topLevelSelects = 0;
  3323. bool hasTailOps = false;
  3324. for (auto& block : blocks) {
  3325. if (block->SubqueryAlias()) {
  3326. continue;
  3327. }
  3328. if (block->HasSelectResult()) {
  3329. ++topLevelSelects;
  3330. } else if (topLevelSelects) {
  3331. hasTailOps = true;
  3332. }
  3333. }
  3334. if ((Mode == NSQLTranslation::ESqlMode::SUBQUERY || Mode == NSQLTranslation::ESqlMode::LIMITED_VIEW) && (topLevelSelects != 1 || hasTailOps)) {
  3335. Error() << "Strictly one select/process/reduce statement is expected at the end of "
  3336. << (Mode == NSQLTranslation::ESqlMode::LIMITED_VIEW ? "view" : "subquery");
  3337. return nullptr;
  3338. }
  3339. if (!Ctx.PragmaAutoCommit && Ctx.Settings.EndOfQueryCommit && IsQueryMode(Mode)) {
  3340. AddStatementToBlocks(blocks, BuildCommitClusters(Ctx.Pos()));
  3341. }
  3342. auto result = BuildQuery(Ctx.Pos(), blocks, true, Ctx.Scoped, Ctx.SeqMode);
  3343. WarnUnusedNodes();
  3344. return result;
  3345. }
  3346. TNodePtr TSqlQuery::Build(const std::vector<::NSQLv1Generated::TRule_sql_stmt_core>& statements) {
  3347. if (Mode == NSQLTranslation::ESqlMode::QUERY) {
  3348. // inject externally declared named expressions
  3349. for (auto [name, type] : Ctx.Settings.DeclaredNamedExprs) {
  3350. if (name.empty()) {
  3351. Error() << "Empty names for externally declared expressions are not allowed";
  3352. return nullptr;
  3353. }
  3354. TString varName = "$" + name;
  3355. if (IsAnonymousName(varName)) {
  3356. Error() << "Externally declared name '" << name << "' is anonymous";
  3357. return nullptr;
  3358. }
  3359. auto parsed = ParseType(type, *Ctx.Pool, Ctx.Issues, Ctx.Pos());
  3360. if (!parsed) {
  3361. Error() << "Failed to parse type for externally declared name '" << name << "'";
  3362. return nullptr;
  3363. }
  3364. TNodePtr typeNode = BuildBuiltinFunc(Ctx, Ctx.Pos(), "ParseType", { BuildLiteralRawString(Ctx.Pos(), type) });
  3365. PushNamedAtom(Ctx.Pos(), varName);
  3366. // no duplicates are possible at this stage
  3367. bool isWeak = true;
  3368. Ctx.DeclareVariable(varName, {}, typeNode, isWeak);
  3369. // avoid 'Symbol is not used' warning for externally declared expression
  3370. YQL_ENSURE(GetNamedNode(varName));
  3371. }
  3372. }
  3373. TVector<TNodePtr> blocks;
  3374. Ctx.PushCurrentBlocks(&blocks);
  3375. Y_DEFER {
  3376. Ctx.PopCurrentBlocks();
  3377. };
  3378. size_t statementNumber = 0;
  3379. for (const auto& statement : statements) {
  3380. if (!Statement(blocks, statement, statementNumber++)) {
  3381. return nullptr;
  3382. }
  3383. }
  3384. ui32 topLevelSelects = 0;
  3385. bool hasTailOps = false;
  3386. for (auto& block : blocks) {
  3387. if (block->SubqueryAlias()) {
  3388. continue;
  3389. }
  3390. if (block->HasSelectResult()) {
  3391. ++topLevelSelects;
  3392. } else if (topLevelSelects) {
  3393. hasTailOps = true;
  3394. }
  3395. }
  3396. if ((Mode == NSQLTranslation::ESqlMode::SUBQUERY || Mode == NSQLTranslation::ESqlMode::LIMITED_VIEW) && (topLevelSelects != 1 || hasTailOps)) {
  3397. Error() << "Strictly one select/process/reduce statement is expected at the end of "
  3398. << (Mode == NSQLTranslation::ESqlMode::LIMITED_VIEW ? "view" : "subquery");
  3399. return nullptr;
  3400. }
  3401. if (!Ctx.PragmaAutoCommit && Ctx.Settings.EndOfQueryCommit && IsQueryMode(Mode)) {
  3402. AddStatementToBlocks(blocks, BuildCommitClusters(Ctx.Pos()));
  3403. }
  3404. auto result = BuildQuery(Ctx.Pos(), blocks, true, Ctx.Scoped, Ctx.SeqMode);
  3405. return result;
  3406. }
  3407. namespace {
  3408. static bool BuildColumnFeatures(std::map<TString, TDeferredAtom>& result, const TRule_column_schema& columnSchema, const NYql::TPosition& pos, TSqlTranslation& translation) {
  3409. const TString columnName(Id(columnSchema.GetRule_an_id_schema1(), translation));
  3410. TString columnType;
  3411. const auto constraints = ColumnConstraints(columnSchema, translation);
  3412. if (!constraints) {
  3413. return false;
  3414. }
  3415. auto& typeBind = columnSchema.GetRule_type_name_or_bind2();
  3416. switch (typeBind.Alt_case()) {
  3417. case TRule_type_name_or_bind::kAltTypeNameOrBind1:
  3418. {
  3419. auto& typeNameOrBind = typeBind.GetAlt_type_name_or_bind1().GetRule_type_name1();
  3420. if (typeNameOrBind.Alt_case() != TRule_type_name::kAltTypeName2) {
  3421. return false;
  3422. }
  3423. auto& alt = typeNameOrBind.GetAlt_type_name2();
  3424. auto& block = alt.GetBlock1();
  3425. auto& simpleType = block.GetAlt2().GetRule_type_name_simple1();
  3426. columnType = Id(simpleType.GetRule_an_id_pure1(), translation);
  3427. if (columnType.empty()) {
  3428. return false;
  3429. }
  3430. break;
  3431. }
  3432. case TRule_type_name_or_bind::kAltTypeNameOrBind2:
  3433. return false;
  3434. case TRule_type_name_or_bind::ALT_NOT_SET:
  3435. Y_ABORT("You should change implementation according to grammar changes");
  3436. }
  3437. result["NAME"] = TDeferredAtom(pos, columnName);
  3438. YQL_ENSURE(columnType, "Unknown column type");
  3439. result["TYPE"] = TDeferredAtom(pos, columnType);
  3440. if (!constraints->Nullable) {
  3441. result["NOT_NULL"] = TDeferredAtom(pos, "true");
  3442. }
  3443. return true;
  3444. }
  3445. }
  3446. bool TSqlQuery::ParseTableStoreFeatures(std::map<TString, TDeferredAtom> & result, const TRule_alter_table_store_action & actions) {
  3447. switch (actions.Alt_case()) {
  3448. case TRule_alter_table_store_action::kAltAlterTableStoreAction1: {
  3449. // ADD COLUMN
  3450. const auto& addRule = actions.GetAlt_alter_table_store_action1().GetRule_alter_table_add_column1();
  3451. if (!BuildColumnFeatures(result, addRule.GetRule_column_schema3(), Ctx.Pos(), *this)) {
  3452. return false;
  3453. }
  3454. result["ACTION"] = TDeferredAtom(Ctx.Pos(), "NEW_COLUMN");
  3455. break;
  3456. }
  3457. case TRule_alter_table_store_action::kAltAlterTableStoreAction2: {
  3458. // DROP COLUMN
  3459. const auto& dropRule = actions.GetAlt_alter_table_store_action2().GetRule_alter_table_drop_column1();
  3460. TString columnName = Id(dropRule.GetRule_an_id3(), *this);
  3461. if (!columnName) {
  3462. return false;
  3463. }
  3464. result["NAME"] = TDeferredAtom(Ctx.Pos(), columnName);
  3465. result["ACTION"] = TDeferredAtom(Ctx.Pos(), "DROP_COLUMN");
  3466. break;
  3467. }
  3468. case TRule_alter_table_store_action::ALT_NOT_SET:
  3469. Y_ABORT("You should change implementation according to grammar changes");
  3470. }
  3471. return true;
  3472. }
  3473. } // namespace NSQLTranslationV1