hqlsource.cpp 181 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881
  1. /*##############################################################################
  2. HPCC SYSTEMS software Copyright (C) 2012 HPCC Systems®.
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. ############################################################################## */
  13. #include "jliball.hpp"
  14. #include "hql.hpp"
  15. #include "platform.h"
  16. #include "jlib.hpp"
  17. #include "jmisc.hpp"
  18. #include "jstream.ipp"
  19. #include "jdebug.hpp"
  20. #include "eclrtl_imp.hpp"
  21. #include "rtlkey.hpp"
  22. #include "hql.hpp"
  23. #include "hqlattr.hpp"
  24. #include "hqlmeta.hpp"
  25. #include "hqlthql.hpp"
  26. #include "hqlhtcpp.ipp"
  27. #include "hqlttcpp.ipp"
  28. #include "hqlutil.hpp"
  29. #include "hqlthql.hpp"
  30. #include "hqlwcpp.hpp"
  31. #include "hqlcpputil.hpp"
  32. #include "hqltcppc.ipp"
  33. #include "hqlopt.hpp"
  34. #include "hqlfold.hpp"
  35. #include "hqlcerrors.hpp"
  36. #include "hqlcatom.hpp"
  37. #include "hqlccommon.hpp"
  38. #include "hqltrans.ipp"
  39. #include "hqlpmap.hpp"
  40. #include "hqlttcpp.ipp"
  41. #include "hqlsource.ipp"
  42. #include "hqlcse.ipp"
  43. #include "hqliter.ipp"
  44. #include "thorcommon.hpp"
  45. #include "hqlinline.hpp"
  46. #include "hqliproj.hpp"
  47. //#define FLATTEN_DATASETS
  48. //#define HACK_TO_IGNORE_TABLE
  49. //#define TraceExprPrintLog(x, expr) PrintLog(x ": %s", expr->toString(StringBuffer()).str());
  50. #define TraceExprPrintLog(x, expr)
  51. //#define TraceTableFields
  52. inline bool needToSerializeRecord(node_operator mode)
  53. {
  54. return (mode == no_thor || mode == no_flat);
  55. }
  56. inline bool needToSerializeRecord(IHqlExpression * mode)
  57. {
  58. return needToSerializeRecord(mode->getOperator());
  59. }
  60. //---------------------------------------------------------------------------
  61. void HqlCppTranslator::addGlobalOnWarning(IHqlExpression * setMetaExpr)
  62. {
  63. globalOnWarnings->addOnWarning(setMetaExpr);
  64. }
  65. unsigned HqlCppTranslator::getSourceAggregateOptimizeFlags() const
  66. {
  67. const bool insideChildQuery = false; // value does not currently matter
  68. return getOptimizeFlags(insideChildQuery)|HOOfold|HOOinsidecompound;
  69. }
  70. void HqlCppTranslator::doBuildExprFilepos(BuildCtx & ctx, IHqlExpression * expr, CHqlBoundExpr & tgt)
  71. {
  72. if (buildExprInCorrectContext(ctx, expr, tgt, false))
  73. return;
  74. throwError(HQLERR_CouldNotResolveFileposition); // internal error: fileposition should have been available.
  75. }
  76. void HqlCppTranslator::doBuildExprFileLogicalName(BuildCtx & ctx, IHqlExpression * expr, CHqlBoundExpr & tgt)
  77. {
  78. if (buildExprInCorrectContext(ctx, expr, tgt, false))
  79. return;
  80. throwError(HQLERR_CouldNotResolveFileposition); // internal error: fileposition should have been available.
  81. }
  82. //---------------------------------------------------------------------------
  83. node_operator getDatasetKind(IHqlExpression * dataset)
  84. {
  85. dataset = queryPhysicalRootTable(dataset);
  86. if (dataset)
  87. {
  88. IHqlExpression * mode = dataset->queryChild(2);
  89. if (mode)
  90. return mode->getOperator();
  91. }
  92. return no_none;
  93. }
  94. unsigned getProjectCount(IHqlExpression * expr)
  95. {
  96. unsigned projectCount = 0;
  97. while (expr->getOperator() != no_table)
  98. {
  99. switch (expr->getOperator())
  100. {
  101. case no_hqlproject:
  102. case no_usertable:
  103. case no_newusertable:
  104. case no_selectfields:
  105. projectCount++;
  106. break;
  107. }
  108. expr = expr->queryChild(0);
  109. }
  110. return projectCount;
  111. }
  112. IHqlExpression * queryFetch(IHqlExpression * expr)
  113. {
  114. for (;;)
  115. {
  116. switch (expr->getOperator())
  117. {
  118. case no_fetch:
  119. return expr;
  120. case no_filter:
  121. case no_compound_fetch:
  122. case no_limit:
  123. case no_keyedlimit:
  124. case no_choosen:
  125. break;
  126. default:
  127. UNIMPLEMENTED;
  128. }
  129. expr = expr->queryChild(0);
  130. }
  131. }
  132. bool isSimpleSource(IHqlExpression * expr)
  133. {
  134. for (;;)
  135. {
  136. switch (expr->getOperator())
  137. {
  138. case no_keyindex:
  139. case no_newkeyindex:
  140. case no_table:
  141. case no_temptable:
  142. case no_inlinetable:
  143. case no_workunit_dataset:
  144. case no_xmlproject:
  145. case no_null:
  146. case no_datasetfromrow:
  147. case no_datasetfromdictionary:
  148. case no_getgraphresult:
  149. case no_getgraphloopresult:
  150. case no_rows:
  151. return true;
  152. case no_choosen:
  153. case no_limit:
  154. case no_keyedlimit:
  155. case no_sorted:
  156. case no_stepped:
  157. case no_distributed:
  158. case no_preservemeta:
  159. case no_unordered:
  160. case no_grouped:
  161. case no_compound_diskread:
  162. case no_compound_disknormalize:
  163. case no_compound_diskaggregate:
  164. case no_compound_diskcount:
  165. case no_compound_diskgroupaggregate:
  166. case no_compound_indexread:
  167. case no_compound_indexnormalize:
  168. case no_compound_indexaggregate:
  169. case no_compound_indexcount:
  170. case no_compound_indexgroupaggregate:
  171. case no_compound_childread:
  172. case no_compound_childnormalize:
  173. case no_compound_childaggregate:
  174. case no_compound_childcount:
  175. case no_compound_childgroupaggregate:
  176. case no_compound_selectnew:
  177. case no_compound_inline:
  178. case no_section:
  179. case no_sectioninput:
  180. case no_nofold:
  181. case no_nohoist:
  182. case no_nocombine:
  183. case no_dataset_alias:
  184. break;
  185. default:
  186. return false;
  187. }
  188. expr = expr->queryChild(0);
  189. }
  190. }
  191. static bool isSimpleProjectingDiskRead(IHqlExpression * expr)
  192. {
  193. bool projected = false;
  194. for (;;)
  195. {
  196. switch (expr->getOperator())
  197. {
  198. case no_table:
  199. return true;
  200. case no_hqlproject:
  201. case no_newusertable:
  202. if (projected)
  203. return false;
  204. //MORE: HPCC-18469 Check if the transform only assigns fields with the same name from source to the target
  205. if (!isSimpleProject(expr))
  206. return false;
  207. projected = true;
  208. break;
  209. default:
  210. return false;
  211. }
  212. expr = expr->queryChild(0);
  213. }
  214. }
  215. IHqlExpression * getVirtualSelector(IHqlExpression * dataset)
  216. {
  217. IHqlExpression * table = queryPhysicalRootTable(dataset);
  218. if (!table)
  219. table = dataset;
  220. return LINK(table->queryNormalizedSelector());
  221. }
  222. IHqlExpression * getFilepos(IHqlExpression * dataset, bool isLocal)
  223. {
  224. IHqlExpression * attr = isLocal ? createLocalAttribute() : NULL;
  225. return createValue(no_filepos, LINK(fposType), getVirtualSelector(dataset), attr);
  226. }
  227. IHqlExpression * getFileLogicalName(IHqlExpression * dataset)
  228. {
  229. return createValue(no_file_logicalname, makeVarStringType(UNKNOWN_LENGTH, NULL, NULL), getVirtualSelector(dataset));
  230. }
  231. IHqlExpression * getVirtualReplacement(IHqlExpression * field, IHqlExpression * virtualDef, IHqlExpression * dataset)
  232. {
  233. IAtom * virtualKind = virtualDef->queryName();
  234. if (virtualKind == filepositionAtom)
  235. return getFilepos(dataset, false);
  236. else if (virtualKind == localFilePositionAtom)
  237. return getFilepos(dataset, true);
  238. else if (virtualKind == sizeofAtom)
  239. return createValue(no_sizeof, LINK(sizetType), getVirtualSelector(dataset));
  240. else if (virtualKind == logicalFilenameAtom)
  241. return createValue(no_implicitcast, field->getType(), getFileLogicalName(dataset));
  242. throwError1(HQLERR_UnknownVirtualAttr, str(virtualKind));
  243. return NULL;
  244. }
  245. static IHqlExpression * createFileposCall(HqlCppTranslator & translator, IIdAtom * name, const char * provider, const char * rowname)
  246. {
  247. HqlExprArray args;
  248. args.append(*createVariable(provider, makeBoolType()));
  249. args.append(*createVariable(rowname, makeBoolType())); // really a row
  250. return translator.bindFunctionCall(name, args);
  251. }
  252. //---------------------------------------------------------------------------
  253. void VirtualFieldsInfo::gatherVirtualFields(IHqlExpression * _record, bool ignoreVirtuals, bool ensureSerialized)
  254. {
  255. OwnedHqlExpr record = ensureSerialized ? getSerializedForm(_record, diskAtom) : LINK(_record);
  256. if (record != _record)
  257. requiresDeserialize = true;
  258. //MORE: This should really recurse through records to check for nested virtual fields.
  259. // e.g., inside ifblocks, or even records....
  260. ForEachChild(idx, record)
  261. {
  262. IHqlExpression * cur = record->queryChild(idx);
  263. IHqlExpression * virtualAttr = NULL;
  264. if (!ignoreVirtuals)
  265. virtualAttr = cur->queryAttribute(virtualAtom);
  266. if (virtualAttr)
  267. {
  268. selects.append(*LINK(cur));
  269. if (isUnknownSize(cur->queryType()))
  270. simpleVirtualsAtEnd = false;
  271. if (virtuals.find(*virtualAttr) == NotFound)
  272. virtuals.append(*LINK(virtualAttr));
  273. }
  274. else
  275. {
  276. //Also adds attributes...
  277. physicalFields.append(*LINK(cur));
  278. if (virtuals.ordinality())
  279. simpleVirtualsAtEnd = false;
  280. }
  281. }
  282. }
  283. IHqlExpression * VirtualFieldsInfo::createPhysicalRecord()
  284. {
  285. if (physicalFields.ordinality() == 1)
  286. if (physicalFields.item(0).getOperator() == no_record)
  287. return LINK(&physicalFields.item(0));
  288. return createRecord(physicalFields);
  289. }
  290. //---------------------------------------------------------------------------
  291. class VirtualRecordTransformCreator : public RecordTransformCreator
  292. {
  293. public:
  294. VirtualRecordTransformCreator(IHqlExpression * _dataset) { dataset = _dataset; }
  295. virtual IHqlExpression * getMissingAssignValue(IHqlExpression * expr)
  296. {
  297. IHqlExpression * virtualAttr = expr->queryAttribute(virtualAtom);
  298. assertex(virtualAttr);
  299. return getVirtualReplacement(expr, virtualAttr->queryChild(0), dataset);
  300. }
  301. protected:
  302. IHqlExpression * dataset;
  303. };
  304. IHqlExpression * createTableWithoutVirtuals(VirtualFieldsInfo & info, IHqlExpression * tableExpr)
  305. {
  306. IHqlExpression * record = tableExpr->queryChild(1);
  307. OwnedHqlExpr diskRecord = info.createPhysicalRecord();
  308. //Clone the annotations to improve the regenerated text in the graph
  309. OwnedHqlExpr diskRecordWithMeta = record->cloneAllAnnotations(diskRecord);
  310. HqlExprArray args;
  311. unwindChildren(args, tableExpr);
  312. args.replace(*LINK(diskRecordWithMeta), 1);
  313. IHqlExpression * newDataset = tableExpr->queryBody()->clone(args);
  314. VirtualRecordTransformCreator mapper(newDataset);
  315. IHqlExpression * newTransform = mapper.createMappingTransform(no_newtransform, record, newDataset);
  316. OwnedHqlExpr projected = createDatasetF(no_newusertable, newDataset, LINK(record), newTransform, createAttribute(_internal_Atom), NULL);
  317. return tableExpr->cloneAllAnnotations(projected);
  318. }
  319. IHqlExpression * buildTableWithoutVirtuals(VirtualFieldsInfo & info, IHqlExpression * expr)
  320. {
  321. IHqlExpression * tableExpr = queryPhysicalRootTable(expr);
  322. OwnedHqlExpr projected = createTableWithoutVirtuals(info, tableExpr);
  323. return replaceExpression(expr, tableExpr, projected);
  324. }
  325. static IHqlExpression * createTableFromSerialized(IHqlExpression * tableExpr)
  326. {
  327. IHqlExpression * record = tableExpr->queryChild(1);
  328. OwnedHqlExpr diskRecord = getSerializedForm(record, diskAtom);
  329. OwnedHqlExpr diskRecordWithMeta = record->cloneAllAnnotations(diskRecord);
  330. OwnedHqlExpr newTable = replaceChild(tableExpr->queryBody(), 1, diskRecordWithMeta);
  331. OwnedHqlExpr transform = createRecordMappingTransform(no_newtransform, record, newTable->queryNormalizedSelector());
  332. OwnedHqlExpr projected = createDatasetF(no_newusertable, LINK(newTable), LINK(record), LINK(transform), createAttribute(_internal_Atom), NULL);
  333. return tableExpr->cloneAllAnnotations(projected);
  334. }
  335. static IHqlExpression * buildTableFromSerialized(IHqlExpression * expr)
  336. {
  337. IHqlExpression * tableExpr = queryPhysicalRootTable(expr);
  338. OwnedHqlExpr projected = createTableFromSerialized(tableExpr);
  339. return replaceExpression(expr, tableExpr, projected);
  340. }
  341. static IHqlExpression * nextDiskField(IHqlExpression * diskRecord, unsigned & diskIndex)
  342. {
  343. for (;;)
  344. {
  345. IHqlExpression * cur = diskRecord->queryChild(diskIndex++);
  346. if (!cur || !cur->isAttribute())
  347. return cur;
  348. }
  349. }
  350. static void createPhysicalLogicalAssigns(HqlExprArray & assigns, IHqlExpression * self, IHqlExpression * diskRecord, IHqlExpression * record, IHqlExpression * diskDataset, bool allowTranslate, unsigned fileposIndex)
  351. {
  352. unsigned numFields = record->numChildren();
  353. unsigned diskIndex = 0;
  354. for (unsigned idx2=0; idx2 < numFields; idx2++)
  355. {
  356. IHqlExpression * cur = record->queryChild(idx2);
  357. switch (cur->getOperator())
  358. {
  359. case no_ifblock:
  360. {
  361. IHqlExpression * ifblock = nextDiskField(diskRecord, diskIndex);
  362. assertex(ifblock && ifblock->getOperator() == no_ifblock);
  363. createPhysicalLogicalAssigns(assigns, self, ifblock->queryChild(1), cur->queryChild(1), diskDataset, false, NotFound);
  364. break;
  365. }
  366. case no_record:
  367. {
  368. IHqlExpression * srcRecord = nextDiskField(diskRecord, diskIndex);
  369. assertex(srcRecord && srcRecord->getOperator() == no_record);
  370. createPhysicalLogicalAssigns(assigns, self, srcRecord, cur, diskDataset, allowTranslate, NotFound);
  371. break;
  372. }
  373. case no_field:
  374. {
  375. OwnedHqlExpr target = createSelectExpr(LINK(self), LINK(cur));
  376. OwnedHqlExpr newValue;
  377. IHqlExpression * curPhysical = nextDiskField(diskRecord, diskIndex);
  378. OwnedHqlExpr physicalSelect = createSelectExpr(LINK(diskDataset), LINK(curPhysical));
  379. if (cur->isDatarow() && !cur->hasAttribute(blobAtom) && (!isInPayload() || (physicalSelect->queryType() != target->queryType())))
  380. {
  381. HqlExprArray subassigns;
  382. OwnedHqlExpr childSelf = createSelector(no_self, cur, NULL);
  383. createPhysicalLogicalAssigns(subassigns, childSelf, curPhysical->queryRecord(), cur->queryRecord(), physicalSelect, false, NotFound);
  384. OwnedHqlExpr transform = createValue(no_transform, makeTransformType(cur->queryRecord()->getType()), subassigns);
  385. newValue.setown(createRow(no_createrow, transform.getClear()));
  386. }
  387. else
  388. newValue.setown(convertIndexPhysical2LogicalValue(cur, physicalSelect, allowTranslate && (idx2 != fileposIndex)));
  389. if (newValue)
  390. assigns.append(*createAssign(target.getClear(), newValue.getClear()));
  391. break;
  392. }
  393. }
  394. }
  395. }
  396. static void createPhysicalLogicalAssigns(HqlExprArray & assigns, IHqlExpression * diskDataset, IHqlExpression * tableExpr, bool hasFilePosition)
  397. {
  398. IHqlExpression * record = tableExpr->queryRecord();
  399. unsigned fileposIndex = (hasFilePosition ? record->numChildren() - 1 : NotFound);
  400. OwnedHqlExpr self = getSelf(record);
  401. createPhysicalLogicalAssigns(assigns, self, diskDataset->queryRecord(), record, diskDataset, true, fileposIndex);
  402. }
  403. IHqlExpression * HqlCppTranslator::convertToPhysicalIndex(IHqlExpression * tableExpr)
  404. {
  405. LinkedHqlExpr * match = physicalIndexCache.getValue(tableExpr);
  406. if (match)
  407. return LINK(*match);
  408. if (tableExpr->hasAttribute(_original_Atom))
  409. return LINK(tableExpr);
  410. assertex(tableExpr->getOperator() == no_newkeyindex);
  411. IHqlExpression * record = tableExpr->queryRecord();
  412. HqlMapTransformer mapper;
  413. bool hasFileposition = getBoolAttribute(tableExpr, filepositionAtom, true);
  414. IHqlExpression * diskRecord = createPhysicalIndexRecord(mapper, record, hasFileposition, false);
  415. unsigned payload = numPayloadFields(tableExpr);
  416. assertex(payload || !hasFileposition);
  417. HqlExprArray args;
  418. unwindChildren(args, tableExpr);
  419. args.replace(*diskRecord, 1);
  420. removeAttribute(args, _payload_Atom);
  421. args.append(*createAttribute(_payload_Atom, getSizetConstant(payload)));
  422. args.append(*createAttribute(_original_Atom, LINK(tableExpr)));
  423. //remove the preload attribute and replace with correct value
  424. IHqlExpression * newDataset = createDataset(tableExpr->getOperator(), args);
  425. HqlExprArray assigns;
  426. createPhysicalLogicalAssigns(assigns, newDataset, tableExpr, hasFileposition);
  427. OwnedHqlExpr projectedTable = createDataset(no_newusertable, newDataset, createComma(LINK(record), createValue(no_newtransform, makeTransformType(record->getType()), assigns)));
  428. physicalIndexCache.setValue(tableExpr, projectedTable);
  429. return projectedTable.getClear();
  430. }
  431. IHqlExpression * convertToPhysicalTable(IHqlExpression * tableExpr, bool ensureSerialized)
  432. {
  433. VirtualFieldsInfo fieldInfo;
  434. fieldInfo.gatherVirtualFields(tableExpr->queryRecord(), tableExpr->hasAttribute(_noVirtual_Atom), ensureSerialized);
  435. if (fieldInfo.hasVirtualsOrDeserialize())
  436. return createTableWithoutVirtuals(fieldInfo, tableExpr);
  437. return LINK(tableExpr);
  438. }
  439. IHqlExpression * HqlCppTranslator::buildIndexFromPhysical(IHqlExpression * expr)
  440. {
  441. IHqlExpression * tableExpr = queryPhysicalRootTable(expr);
  442. OwnedHqlExpr newProject = convertToPhysicalIndex(tableExpr);
  443. return replaceExpression(expr, tableExpr, newProject);
  444. }
  445. //---------------------------------------------------------------------------
  446. class SourceSteppingInfo
  447. {
  448. public:
  449. inline bool exists() { return rawStepping.exists(); }
  450. IHqlExpression * firstStepped()
  451. {
  452. if (rawStepping.exists())
  453. return rawStepping.fields->queryChild(0);
  454. return NULL;
  455. }
  456. void extractRaw()
  457. {
  458. rawSteppingProject.extractFields(rawStepping);
  459. }
  460. void checkKeyable(CppFilterExtractor & monitors)
  461. {
  462. if (!rawStepping.exists())
  463. return;
  464. unsigned prev = NotFound;
  465. ForEachChild(i, rawStepping.fields)
  466. {
  467. IHqlExpression * cur = rawStepping.fields->queryChild(i);
  468. unsigned thisMatch = monitors.queryKeySelectIndex(cur);
  469. if (thisMatch == NotFound)
  470. throwError1(HQLERR_StepFieldNotKeyed, str(cur->queryChild(1)->queryName()));
  471. if ((prev != NotFound) && (thisMatch != prev+1))
  472. throwError1(HQLERR_StepFieldNotContiguous, str(cur->queryChild(1)->queryName()));
  473. prev = thisMatch;
  474. }
  475. }
  476. void generateMembers(HqlCppTranslator & translator, BuildCtx & ctx)
  477. {
  478. rawStepping.generateSteppingMetaMember(translator, ctx, "RawSteppingMeta");
  479. if (outputStepping.exists())
  480. {
  481. if (outputStepping.exists() && outputStepping.ds != rawStepping.ds)
  482. {
  483. outputStepping.generateSteppingMetaMember(translator, ctx, "ProjectedSteppingMeta");
  484. MemberFunction func(translator, ctx, "virtual void mapOutputToInput(ARowBuilder & crSelf, const void * _projected, unsigned numFields) override");
  485. translator.ensureRowAllocated(func.ctx, "crSelf");
  486. func.ctx.addQuotedLiteral("const byte * pr = (const byte *)_projected;");
  487. translator.bindTableCursor(func.ctx, rawStepping.ds, "crSelf.row()");
  488. translator.bindTableCursor(func.ctx, outputStepping.ds, "pr");
  489. StringBuffer s;
  490. ForEachChild(i, outputStepping.fields)
  491. {
  492. IHqlExpression * curOutput = outputStepping.fields->queryChild(i);
  493. IHqlExpression * curRawExpr = rawSteppingProject.fields->queryChild(i);
  494. IHqlExpression * curRawSelect = rawStepping.fields->queryChild(i);
  495. OwnedHqlExpr original = outputStepping.invertTransform(curRawExpr, curOutput);
  496. func.ctx.addQuoted(s.clear().append("if (numFields < ").append(i+1).append(") return;"));
  497. translator.buildAssign(func.ctx, curRawSelect, original);
  498. }
  499. }
  500. }
  501. else
  502. {
  503. OwnedHqlExpr fail = createValue(no_fail, makeVoidType(), createConstant("Cannot step output of index read"));
  504. MemberFunction func(translator, ctx, "virtual void mapOutputToInput(ARowBuilder & crSelf, const void * _projected, unsigned numFields) override");
  505. translator.buildStmt(func.ctx, fail);
  506. }
  507. }
  508. public:
  509. SteppingFieldSelection outputStepping;
  510. SteppingFieldSelection rawSteppingProject;
  511. SteppingFieldSelection rawStepping;
  512. };
  513. //---------------------------------------------------------------------------
  514. class SourceBuilder
  515. {
  516. public:
  517. SourceBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  518. : tableExpr(_tableExpr), translator(_translator)
  519. {
  520. nameExpr.setown(foldHqlExpression(_nameExpr));
  521. needDefaultTransform = true;
  522. needToCallTransform = false;
  523. isPreloaded = false;
  524. isCompoundCount = false;
  525. transformCanFilter = false;
  526. IHqlExpression * preload = tableExpr ? tableExpr->queryAttribute(preloadAtom) : NULL;
  527. if (preload)
  528. {
  529. isPreloaded = true;
  530. preloadSize.set(preload->queryChild(0));
  531. }
  532. failedFilterValue.set(queryZero());
  533. isNormalize = false;
  534. aggregation = false;
  535. instance = NULL;
  536. returnIfFilterFails = true;
  537. useFilterMappings = true;
  538. generateUnfilteredTransform = false;
  539. allowDynamicFormatChange = tableExpr && !tableExpr->hasAttribute(fixedAtom);
  540. onlyExistsAggreate = false;
  541. monitorsForGrouping = false;
  542. useImplementationClass = false;
  543. isUnfilteredCount = false;
  544. isVirtualLogicalFilenameUsed = false;
  545. requiresOrderedMerge = false;
  546. rootSelfRow = NULL;
  547. activityKind = TAKnone;
  548. if (tableExpr)
  549. {
  550. IHqlExpression * virtualAttr = tableExpr->queryAttribute(virtualAtom);
  551. newDiskReadMapping = translator.queryOptions().newDiskReadMapping;
  552. if (virtualAttr)
  553. newDiskReadMapping = !getBoolAttribute(virtualAttr, legacyAtom);
  554. switch (tableExpr->getOperator())
  555. {
  556. case no_fetch:
  557. case no_compound_fetch:
  558. newDiskReadMapping = false;
  559. break;
  560. }
  561. }
  562. else
  563. newDiskReadMapping = false;
  564. }
  565. virtual ~SourceBuilder() {}
  566. virtual void buildMembers(IHqlExpression * expr) = 0;
  567. virtual void buildTransformFpos(BuildCtx & transformCtx) = 0;
  568. virtual void extractMonitors(IHqlExpression * ds, SharedHqlExpr & unkeyedFilter, HqlExprArray & conds);
  569. virtual void buildTransformElements(BuildCtx & ctx, IHqlExpression * expr, bool ignoreFilters);
  570. virtual void buildTransform(IHqlExpression * expr) = 0;
  571. virtual void analyse(IHqlExpression * expr);
  572. void buildCanMatch(IHqlExpression * expr);
  573. void buildMatchFilter(BuildCtx & ctx, IHqlExpression * expr);
  574. void buildFilenameMember();
  575. void appendFilter(SharedHqlExpr & unkeyedFilter, IHqlExpression * expr);
  576. void buildKeyedLimitHelper(IHqlExpression * expr);
  577. void buildLimits(BuildCtx & classctx, IHqlExpression * expr, unique_id_t id);
  578. void buildReadMembers( IHqlExpression * expr);
  579. void buildSteppingMeta(IHqlExpression * expr, CppFilterExtractor * monitors);
  580. void buildTransformBody(BuildCtx & transformCtx, IHqlExpression * expr, bool returnSize, bool ignoreFilters, bool bindInputRow);
  581. void checkDependencies(BuildCtx & ctx, IHqlExpression * expr);
  582. bool containsStepping(IHqlExpression * expr);
  583. ABoundActivity * buildActivity(BuildCtx & ctx, IHqlExpression * expr, ThorActivityKind activityKind, const char *kind, ABoundActivity *input);
  584. void gatherVirtualFields(bool ignoreVirtuals, bool ensureSerialized);
  585. bool recordHasVirtuals() { return fieldInfo.hasVirtuals(); }
  586. bool recordHasVirtualsOrDeserialize() { return fieldInfo.hasVirtualsOrDeserialize(); }
  587. bool isSourceInvariant(IHqlExpression * dataset, IHqlExpression * expr);
  588. bool hasExistChoosenLimit() { return (choosenValue && getIntValue(choosenValue) == 1); }
  589. bool isRootSelector(IHqlExpression * expr) const;
  590. protected:
  591. void assignLocalExtract(BuildCtx & ctx, ParentExtract * extractBuilder, IHqlExpression * dataset, const char * argName);
  592. void associateFilePositions(BuildCtx & ctx, const char * provider, const char * rowname);
  593. void buildSteppedHelpers();
  594. void doBuildAggregateSelectIterator(BuildCtx & ctx, IHqlExpression * expr);
  595. void doBuildNormalizeIterators(BuildCtx & ctx, IHqlExpression * expr, bool isChildIterator);
  596. void buildAggregateHelpers(IHqlExpression * expr);
  597. void buildCountHelpers(IHqlExpression * expr, bool allowMultiple);
  598. virtual void buildFlagsMember(IHqlExpression * expr) {}
  599. void buildGlobalGroupAggregateHelpers(IHqlExpression * expr);
  600. void buildGroupAggregateHelpers(ParentExtract * extractBuilder, IHqlExpression * aggregate);
  601. void buildGroupAggregateCompareHelper(ParentExtract * extractBuilder, IHqlExpression * aggregate, HqlExprArray & recordFields, HqlExprArray & agrgegateFields);
  602. void buildGroupAggregateHashHelper(ParentExtract * extractBuilder, IHqlExpression * dataset, IHqlExpression * fields);
  603. void buildGroupAggregateProcessHelper(ParentExtract * extractBuilder, IHqlExpression * aggregate, const char * name, bool doneAny);
  604. void buildGroupingMonitors(IHqlExpression * expr, CppFilterExtractor & monitors);
  605. void buildGroupAggregateTransformBody(BuildCtx & transformctx, IHqlExpression * expr, bool useExtract, bool bindInputRow);
  606. void buildNormalizeHelpers(IHqlExpression * expr);
  607. void buildTargetCursor(Shared<BoundRow> & tempRow, Shared<BoundRow> & rowBuilder, BuildCtx & ctx, IHqlExpression * expr);
  608. void associateTargetCursor(BuildCtx & subctx, BuildCtx & ctx, BoundRow * tempRow, BoundRow * rowBuilder, IHqlExpression * expr);
  609. IHqlExpression * ensureAggregateGroupingAliased(IHqlExpression * aggregate);
  610. void gatherSteppingMeta(IHqlExpression * expr, SourceSteppingInfo & info);
  611. void gatherSteppingMeta(IHqlExpression * expr, SteppingFieldSelection & outputStepping, SteppingFieldSelection & rawStepping);
  612. void gatherFieldUsage(SourceFieldUsage * fieldUsage, IHqlExpression * expr);
  613. void rebindFilepositons(BuildCtx & ctx, IHqlExpression * dataset, node_operator side, IHqlExpression * selSeq, bool isLocal);
  614. void rebindFilepositons(BuildCtx & ctx, IHqlExpression * dataset, node_operator side, IHqlExpression * selSeq);
  615. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr);
  616. virtual void analyseGraph(IHqlExpression * expr);
  617. virtual bool isExplicitExists() { return false; }
  618. public:
  619. VirtualFieldsInfo fieldInfo;
  620. ActivityInstance * instance;
  621. BoundRow * rootSelfRow;
  622. HqlExprAttr tableExpr;
  623. HqlExprAttr nameExpr;
  624. HqlExprAttr fpos;
  625. HqlExprAttr lfpos;
  626. HqlExprAttr limitExpr;
  627. HqlExprAttr keyedLimitExpr;
  628. HqlExprAttr choosenValue;
  629. HqlExprAttr preloadSize;
  630. HqlExprAttr firstTransformer;
  631. HqlExprAttr lastTransformer;
  632. HqlExprAttr alreadyDoneFlag;
  633. HqlExprArray originalFilters;
  634. HqlExprArray mappedFilters;
  635. HqlExprArray removedFilters;
  636. HqlExprAttr failedFilterValue;
  637. HqlExprAttr compoundCountVar;
  638. HqlExprAttr physicalRecord;
  639. LinkedHqlExpr expectedRecord;
  640. LinkedHqlExpr projectedRecord;
  641. LinkedHqlExpr tableSelector;
  642. LinkedHqlExpr projectedSelector;
  643. HqlExprAttr steppedExpr;
  644. Linked<BuildCtx> globaliterctx;
  645. HqlExprCopyArray parentCursors;
  646. HqlExprAttr logicalFilenameMarker;
  647. ThorActivityKind activityKind;
  648. bool useFilterMappings;
  649. bool needDefaultTransform;
  650. bool needToCallTransform;
  651. bool isPreloaded;
  652. bool transformCanFilter;
  653. bool isCompoundCount;
  654. bool isNormalize;
  655. bool aggregation;
  656. bool returnIfFilterFails;
  657. bool allowDynamicFormatChange;
  658. bool onlyExistsAggreate;
  659. bool monitorsForGrouping;
  660. bool generateUnfilteredTransform;
  661. bool useImplementationClass;
  662. bool isUnfilteredCount;
  663. bool isVirtualLogicalFilenameUsed;
  664. bool requiresOrderedMerge;
  665. bool newDiskReadMapping;
  666. bool extractCanMatch = false;
  667. protected:
  668. HqlCppTranslator & translator;
  669. };
  670. struct HQLCPP_API HqlFilePositionDefinedValue : public HqlSimpleDefinedValue
  671. {
  672. public:
  673. HqlFilePositionDefinedValue(SourceBuilder & _builder, IHqlExpression * _original, IHqlExpression * _expr)
  674. : HqlSimpleDefinedValue(_original, _expr), builder(_builder)
  675. { }
  676. virtual IHqlExpression * queryExpr() const
  677. {
  678. builder.isVirtualLogicalFilenameUsed = true;
  679. return HqlSimpleDefinedValue::queryExpr();
  680. }
  681. public:
  682. SourceBuilder & builder;
  683. };
  684. bool SourceBuilder::isSourceInvariant(IHqlExpression * dataset, IHqlExpression * expr)
  685. {
  686. if (containsAssertKeyed(expr))
  687. return false;
  688. if (!containsActiveDataset(expr))
  689. return true;
  690. HqlExprCopyArray inScope;
  691. expr->gatherTablesUsed(NULL, &inScope);
  692. if (!canEvaluateInScope(parentCursors, inScope))
  693. return false;
  694. //Carefull.... It looks ok, but it is possible that the same dataset occurs at multiple levels (sqfilt.hql)
  695. //So need to be careful that the datasets being referenced aren't newly in scope for this activity...
  696. for (;;)
  697. {
  698. if (inScope.contains(*dataset->queryNormalizedSelector()))
  699. return false;
  700. //To be strictly correct we need to walk no_select chain of root datasets!
  701. dataset = queryNextMultiLevelDataset(dataset, true);
  702. if (!dataset)
  703. return true;
  704. }
  705. }
  706. bool SourceBuilder::isRootSelector(IHqlExpression * expr) const
  707. {
  708. if (!tableExpr)
  709. return false;
  710. return (expr->queryNormalizedSelector() == tableExpr->queryNormalizedSelector());
  711. }
  712. void SourceBuilder::analyse(IHqlExpression * expr)
  713. {
  714. IHqlExpression * body = expr->queryBody(true);
  715. if (expr != body)
  716. {
  717. switch (expr->getAnnotationKind())
  718. {
  719. case annotate_meta:
  720. //the onwarning state will be restored by the scope held in HqlCppTranslator::buildActivity
  721. translator.localOnWarnings->processMetaAnnotation(expr);
  722. break;
  723. case annotate_location:
  724. instance->addLocationAttribute(expr);
  725. break;
  726. case annotate_symbol:
  727. //don't clear onWarnings when we hit a symbol because the warnings within a compound activity aren't generated at the correct point
  728. instance->addNameAttribute(expr);
  729. break;
  730. }
  731. analyse(body);
  732. return;
  733. }
  734. node_operator op = expr->getOperator();
  735. switch (op)
  736. {
  737. case no_cachealias:
  738. analyse(expr->queryChild(1));
  739. return;
  740. case no_newkeyindex:
  741. case no_table:
  742. case no_fetch:
  743. case no_select: // handled below
  744. case no_null:
  745. case no_anon:
  746. case no_pseudods:
  747. case no_workunit_dataset:
  748. case no_call:
  749. case no_externalcall:
  750. case no_rows:
  751. case no_libraryinput:
  752. break;
  753. default:
  754. analyse(expr->queryChild(0));
  755. break;
  756. }
  757. switch (op)
  758. {
  759. case no_table:
  760. case no_newkeyindex:
  761. if (!newDiskReadMapping)
  762. {
  763. assertex(!fieldInfo.hasVirtuals());
  764. }
  765. if (newDiskReadMapping)
  766. {
  767. if (!tableExpr->hasAttribute(_noVirtual_Atom) && (tableExpr->queryChild(2)->getOperator() != no_pipe))
  768. {
  769. if (containsVirtualField(tableExpr->queryRecord(), logicalFilenameAtom))
  770. isVirtualLogicalFilenameUsed = true;
  771. }
  772. }
  773. break;
  774. case no_null:
  775. case no_anon:
  776. case no_pseudods:
  777. case no_workunit_dataset:
  778. case no_getgraphresult:
  779. case no_call:
  780. case no_externalcall:
  781. case no_rows:
  782. case no_libraryinput:
  783. break;
  784. case no_filter:
  785. {
  786. OwnedHqlExpr unkeyedFilter;
  787. HqlExprArray conds;
  788. unwindFilterConditions(conds, expr);
  789. extractMonitors(expr, unkeyedFilter, conds);
  790. if (unkeyedFilter)
  791. {
  792. if (!extractCanMatch || !isRootSelector(expr))
  793. transformCanFilter = true;
  794. originalFilters.append(*LINK(expr));
  795. mappedFilters.append(*unkeyedFilter.getClear());
  796. }
  797. else
  798. removedFilters.append(*LINK(expr));
  799. break;
  800. }
  801. case no_select:
  802. {
  803. bool isNew;
  804. IHqlExpression * ds = querySelectorDataset(expr, isNew);
  805. if (isNew && isMultiLevelDatasetSelector(expr, false))
  806. {
  807. if (!translator.resolveSelectorDataset(instance->startctx, ds))
  808. {
  809. analyse(ds);
  810. isNormalize = true;
  811. }
  812. }
  813. break;
  814. }
  815. case no_stepped:
  816. if (steppedExpr)
  817. throwError(HQLERR_MultipleStepped);
  818. steppedExpr.set(expr);
  819. break;
  820. case no_sorted:
  821. case no_distributed:
  822. case no_preservemeta:
  823. case no_unordered:
  824. case no_grouped:
  825. case no_alias_scope:
  826. case no_section:
  827. case no_sectioninput:
  828. case no_nofold:
  829. case no_nohoist:
  830. case no_nocombine:
  831. case no_dataset_alias:
  832. break;
  833. case no_preload:
  834. isPreloaded = true;
  835. preloadSize.set(queryRealChild(expr, 1));
  836. break;
  837. case no_limit:
  838. limitExpr.set(expr);
  839. break;
  840. case no_keyedlimit:
  841. keyedLimitExpr.set(expr);
  842. break;
  843. case no_choosen:
  844. {
  845. choosenValue.set(expr->queryChild(1));
  846. IHqlExpression * first = queryRealChild(expr, 2);
  847. if (first)
  848. {
  849. Owned<ITypeInfo> type = makeIntType(8, true);
  850. choosenValue.setown(createValue(no_sub, LINK(type), createValue(no_add, LINK(type), ensureExprType(choosenValue, type), ensureExprType(first, type)), createConstant(I64C(1))));
  851. }
  852. choosenValue.setown(foldHqlExpression(choosenValue));
  853. }
  854. break;
  855. case no_hqlproject:
  856. needToCallTransform = true;
  857. needDefaultTransform = false;
  858. if (!firstTransformer)
  859. firstTransformer.set(expr);
  860. lastTransformer.set(expr);
  861. break;
  862. case no_newusertable:
  863. needToCallTransform = true;
  864. needDefaultTransform = false;
  865. if (!firstTransformer)
  866. firstTransformer.set(expr);
  867. lastTransformer.set(expr);
  868. break;
  869. case no_aggregate:
  870. {
  871. needToCallTransform = true;
  872. needDefaultTransform = false;
  873. aggregation = true;
  874. if (!firstTransformer)
  875. firstTransformer.set(expr);
  876. lastTransformer.set(expr);
  877. break;
  878. }
  879. case no_newaggregate:
  880. {
  881. needToCallTransform = true;
  882. needDefaultTransform = false;
  883. aggregation = true;
  884. if (!firstTransformer)
  885. firstTransformer.set(expr);
  886. lastTransformer.set(expr);
  887. IHqlExpression * transform = expr->queryChild(2);
  888. node_operator aggOp = queryTransformSingleAggregate(transform);
  889. onlyExistsAggreate = ((aggOp == no_existsgroup) || (aggOp == no_none)); // The implicit project code can remove all the aggregate() operators....
  890. if (isCompoundCount)
  891. {
  892. IHqlExpression * rhs = transform->queryChild(0)->queryChild(1);
  893. IHqlExpression * filter = queryRealChild(rhs, 0);
  894. if (filter)
  895. transformCanFilter = true;
  896. }
  897. break;
  898. }
  899. case no_usertable:
  900. case no_selectfields:
  901. UNIMPLEMENTED;
  902. break;
  903. case no_fetch:
  904. needToCallTransform = true;
  905. needDefaultTransform = false;
  906. if (!firstTransformer)
  907. firstTransformer.set(expr);
  908. lastTransformer.set(expr);
  909. break;
  910. case no_compound_diskread:
  911. case no_compound_disknormalize:
  912. case no_compound_diskaggregate:
  913. case no_compound_diskcount:
  914. case no_compound_diskgroupaggregate:
  915. case no_compound_indexread:
  916. case no_compound_indexnormalize:
  917. case no_compound_indexaggregate:
  918. case no_compound_indexcount:
  919. case no_compound_indexgroupaggregate:
  920. case no_compound_childread:
  921. case no_compound_childnormalize:
  922. case no_compound_childaggregate:
  923. case no_compound_childcount:
  924. case no_compound_childgroupaggregate:
  925. case no_compound_fetch:
  926. case no_compound_selectnew:
  927. break;
  928. default:
  929. throwUnexpectedOp(op);
  930. }
  931. }
  932. void SourceBuilder::appendFilter(SharedHqlExpr & unkeyedFilter, IHqlExpression * expr)
  933. {
  934. if (expr)
  935. {
  936. if (expr->queryValue())
  937. {
  938. if (!expr->queryValue()->getBoolValue())
  939. unkeyedFilter.set(expr);
  940. }
  941. else
  942. {
  943. if (unkeyedFilter)
  944. unkeyedFilter.setown(createValue(no_and, unkeyedFilter.getClear(), LINK(expr)));
  945. else
  946. unkeyedFilter.set(expr);
  947. }
  948. }
  949. }
  950. void SourceBuilder::associateFilePositions(BuildCtx & ctx, const char * provider, const char * rowname)
  951. {
  952. if (fpos)
  953. {
  954. Owned<IHqlExpression> fposExpr = createFileposCall(translator, getFilePositionId, provider, rowname);
  955. ctx.associateExpr(fpos, fposExpr);
  956. }
  957. if (lfpos)
  958. {
  959. Owned<IHqlExpression> fposExpr = createFileposCall(translator, getLocalFilePositionId, provider, rowname);
  960. ctx.associateExpr(lfpos, fposExpr);
  961. }
  962. if (logicalFilenameMarker)
  963. {
  964. Owned<IHqlExpression> nameExpr = createFileposCall(translator, queryLogicalFilenameId, provider, rowname);
  965. ctx.associateOwn(*new HqlFilePositionDefinedValue(*this, logicalFilenameMarker, nameExpr));
  966. }
  967. }
  968. void SourceBuilder::rebindFilepositons(BuildCtx & ctx, IHqlExpression * dataset, node_operator side, IHqlExpression * selSeq, bool isLocal)
  969. {
  970. if (!tableExpr)
  971. return;
  972. OwnedHqlExpr searchPos = getFilepos(tableExpr, isLocal);
  973. HqlExprAssociation * match = ctx.queryMatchExpr(searchPos);
  974. if (match)
  975. {
  976. OwnedHqlExpr selector = createSelector(side, dataset, selSeq);
  977. OwnedHqlExpr selectorFpos = getFilepos(selector, isLocal);
  978. ctx.associateExpr(selectorFpos, match->queryExpr());
  979. }
  980. }
  981. void SourceBuilder::rebindFilepositons(BuildCtx & ctx, IHqlExpression * dataset, node_operator side, IHqlExpression * selSeq)
  982. {
  983. bool savedIsVirtualLogicalFilenameUsed = isVirtualLogicalFilenameUsed; // don't allow the rebinding to set the flag.
  984. rebindFilepositons(ctx, dataset, side, selSeq, true);
  985. rebindFilepositons(ctx, dataset, side, selSeq, false);
  986. OwnedHqlExpr searchLogicalFilename = getFileLogicalName(dataset);
  987. HqlExprAssociation * match = ctx.queryMatchExpr(searchLogicalFilename);
  988. if (match)
  989. {
  990. OwnedHqlExpr selector = createSelector(side, dataset, selSeq);
  991. OwnedHqlExpr selectorLogicalFilename = getFileLogicalName(dataset);
  992. ctx.associateOwn(*new HqlFilePositionDefinedValue(*this, selectorLogicalFilename, match->queryExpr()));
  993. }
  994. isVirtualLogicalFilenameUsed = savedIsVirtualLogicalFilenameUsed;
  995. }
  996. void SourceBuilder::buildFilenameMember()
  997. {
  998. //---- virtual const char * getFileName() { return "x.d00"; } ----
  999. translator.buildFilenameFunction(*instance, instance->startctx, WaFilename, "getFileName", nameExpr, translator.hasDynamicFilename(tableExpr));
  1000. }
  1001. void SourceBuilder::buildReadMembers(IHqlExpression * expr)
  1002. {
  1003. buildFilenameMember();
  1004. //---- virtual bool needTransform() { return <bool>; } ----
  1005. if (needToCallTransform || transformCanFilter)
  1006. translator.doBuildBoolFunction(instance->classctx, "needTransform", true);
  1007. //---- virtual bool transformMayFilter() { return <bool>; } ----
  1008. if (transformCanFilter)
  1009. translator.doBuildBoolFunction(instance->classctx, "transformMayFilter", true);
  1010. }
  1011. void SourceBuilder::buildLimits(BuildCtx & classctx, IHqlExpression * expr, unique_id_t id)
  1012. {
  1013. if (limitExpr)
  1014. translator.buildLimitHelpers(classctx, limitExpr, nameExpr, id);
  1015. if (choosenValue)
  1016. {
  1017. MemberFunction func(translator, classctx, "virtual unsigned __int64 getChooseNLimit() override");
  1018. OwnedHqlExpr newLimit = ensurePositiveOrZeroInt64(choosenValue);
  1019. translator.buildReturn(func.ctx, newLimit);
  1020. }
  1021. }
  1022. void SourceBuilder::buildTransformBody(BuildCtx & transformCtx, IHqlExpression * expr, bool returnSize, bool ignoreFilters, bool bindInputRow)
  1023. {
  1024. if (tableExpr && bindInputRow)
  1025. {
  1026. IHqlExpression * mode = (tableExpr->getOperator() == no_table) ? tableExpr->queryChild(2) : NULL;
  1027. if (mode && mode->getOperator() == no_csv)
  1028. {
  1029. translator.bindCsvTableCursor(transformCtx, tableExpr, "Src", no_none, NULL, true, queryCsvEncoding(mode));
  1030. }
  1031. else
  1032. {
  1033. translator.bindTableCursor(transformCtx, projectedSelector, "left");
  1034. }
  1035. transformCtx.addGroup();
  1036. }
  1037. rootSelfRow = translator.bindSelf(transformCtx, expr, "crSelf");
  1038. buildTransformFpos(transformCtx);
  1039. buildTransformElements(transformCtx, expr, ignoreFilters);
  1040. if (returnSize)
  1041. {
  1042. CHqlBoundExpr boundTargetSize;
  1043. if (needDefaultTransform)
  1044. {
  1045. IHqlExpression * left = expr->queryNormalizedSelector();
  1046. OwnedHqlExpr source = ensureActiveRow(left);
  1047. translator.buildAssign(transformCtx, rootSelfRow->querySelector(), source);
  1048. }
  1049. translator.getRecordSize(transformCtx, rootSelfRow->querySelector(), boundTargetSize);
  1050. transformCtx.setNextDestructor();
  1051. transformCtx.addReturn(boundTargetSize.expr);
  1052. }
  1053. rootSelfRow = NULL;
  1054. }
  1055. void SourceBuilder::buildTargetCursor(Shared<BoundRow> & tempRow, Shared<BoundRow> & rowBuilder, BuildCtx & ctx, IHqlExpression * expr)
  1056. {
  1057. assertex(lastTransformer != NULL);
  1058. if (expr == lastTransformer)
  1059. {
  1060. rowBuilder.set(rootSelfRow);
  1061. }
  1062. else
  1063. {
  1064. tempRow.setown(translator.declareTempAnonRow(ctx, ctx, expr));
  1065. ctx.addGroup();
  1066. // group is important, otherwise sizeof(self.x) gets cached incorrectly
  1067. // not so sure, but references to LEFT(x) may be misresolved
  1068. rowBuilder.setown(translator.createRowBuilder(ctx, tempRow));
  1069. }
  1070. }
  1071. void SourceBuilder::associateTargetCursor(BuildCtx & subctx, BuildCtx & ctx, BoundRow * tempRow, BoundRow * rowBuilder, IHqlExpression * expr)
  1072. {
  1073. //First remove the old active dataset
  1074. //This is not strictly necessary, but it avoids the redundant row being serialized to any child queries
  1075. BoundRow * oldCursor = translator.resolveSelectorDataset(ctx, expr->queryChild(0));
  1076. ctx.removeAssociation(oldCursor);
  1077. //And add an association for expr
  1078. if (tempRow)
  1079. {
  1080. translator.finalizeTempRow(subctx, tempRow, rowBuilder);
  1081. translator.bindTableCursor(ctx, expr, tempRow->queryBound());
  1082. }
  1083. else
  1084. translator.bindTableCursor(ctx, expr, rowBuilder->queryBound());
  1085. }
  1086. void SourceBuilder::buildTransformElements(BuildCtx & ctx, IHqlExpression * expr, bool ignoreFilters)
  1087. {
  1088. //This function can be called again for the unfiltered tranform. Don't process annotations again.
  1089. if ((expr != instance->activityExpr) && !ignoreFilters)
  1090. instance->processAnnotations(expr);
  1091. expr = expr->queryBody();
  1092. node_operator op = expr->getOperator();
  1093. switch (op)
  1094. {
  1095. case no_cachealias:
  1096. buildTransformElements(ctx, expr->queryChild(1), ignoreFilters);
  1097. return;
  1098. case no_newkeyindex:
  1099. case no_table:
  1100. case no_fetch:
  1101. case no_select: // handled below
  1102. case no_null:
  1103. case no_anon:
  1104. case no_pseudods:
  1105. case no_workunit_dataset:
  1106. case no_getgraphresult:
  1107. case no_call:
  1108. case no_externalcall:
  1109. case no_rows:
  1110. case no_libraryinput:
  1111. break;
  1112. default:
  1113. buildTransformElements(ctx, expr->queryChild(0), ignoreFilters);
  1114. break;
  1115. }
  1116. switch (op)
  1117. {
  1118. case no_newkeyindex:
  1119. case no_table:
  1120. if (newDiskReadMapping)
  1121. {
  1122. }
  1123. else
  1124. {
  1125. assertex(!fieldInfo.hasVirtuals());
  1126. }
  1127. break;
  1128. case no_null:
  1129. case no_anon:
  1130. case no_pseudods:
  1131. break;
  1132. case no_workunit_dataset:
  1133. case no_getgraphresult:
  1134. case no_call:
  1135. case no_externalcall:
  1136. case no_rows:
  1137. case no_libraryinput:
  1138. throwUnexpectedOp(op);
  1139. break;
  1140. case no_select:
  1141. processTransformSelect(ctx, expr);
  1142. break;
  1143. case no_sorted:
  1144. case no_stepped:
  1145. case no_distributed:
  1146. case no_preservemeta:
  1147. case no_unordered:
  1148. case no_grouped:
  1149. case no_preload:
  1150. case no_limit:
  1151. case no_keyedlimit:
  1152. case no_choosen:
  1153. case no_alias_scope:
  1154. case no_section:
  1155. case no_sectioninput:
  1156. case no_nofold:
  1157. case no_nohoist:
  1158. case no_nocombine:
  1159. break;
  1160. case no_filter:
  1161. {
  1162. if (!ignoreFilters && (!extractCanMatch || !isRootSelector(expr)))
  1163. {
  1164. LinkedHqlExpr cond;
  1165. if (useFilterMappings)
  1166. {
  1167. unsigned match = originalFilters.find(*expr);
  1168. if (match != NotFound)
  1169. cond.set(&mappedFilters.item(match));
  1170. }
  1171. else
  1172. {
  1173. HqlExprArray args;
  1174. unwindRealChildren(args, expr, 1);
  1175. cond.setown(createBalanced(no_and, queryBoolType(), args));
  1176. }
  1177. if (cond)
  1178. {
  1179. IHqlExpression * ds = expr->queryChild(0);
  1180. OwnedHqlExpr test = returnIfFilterFails ? getInverse(cond) : LINK(cond);
  1181. if (translator.queryOptions().foldFilter)
  1182. test.setown(foldScopedHqlExpression(translator.queryErrorProcessor(), ds->queryNormalizedSelector(), test));
  1183. if (translator.options.spotCSE)
  1184. test.setown(spotScalarCSE(test, ds, translator.queryOptions().spotCseInIfDatasetConditions));
  1185. if (!returnIfFilterFails)
  1186. translator.buildFilter(ctx, test);
  1187. else
  1188. {
  1189. LinkedHqlExpr mismatchReturnValue = failedFilterValue;
  1190. //If the output row has already been generated, then returning at this point will leak any
  1191. //child datasets. To avoid that we explicitly call the destructor on the output row.
  1192. if (recordRequiresDestructor(expr->queryRecord()))
  1193. {
  1194. if (lastTransformer && lastTransformer->queryNormalizedSelector() == expr->queryNormalizedSelector())
  1195. {
  1196. StringBuffer s;
  1197. translator.buildMetaForRecord(s, expr->queryRecord());
  1198. s.append(".destruct(crSelf.row())");
  1199. OwnedHqlExpr cleanupAction = createQuoted(s.str(), makeVoidType());
  1200. //Create a compound expression (destroy-old, return-value)
  1201. mismatchReturnValue.setown(createCompound(LINK(cleanupAction), LINK(mismatchReturnValue)));
  1202. }
  1203. }
  1204. translator.buildFilteredReturn(ctx, test, mismatchReturnValue);
  1205. }
  1206. }
  1207. }
  1208. }
  1209. break;
  1210. case no_hqlproject:
  1211. {
  1212. IHqlExpression * dataset = expr->queryChild(0);
  1213. IHqlExpression * datasetSelector = dataset->queryNormalizedSelector();
  1214. IHqlExpression * selSeq = querySelSeq(expr);
  1215. OwnedHqlExpr leftSelect = createSelector(no_left, dataset, querySelSeq(expr));
  1216. if ((projectedSelector != tableSelector) && (expr == firstTransformer))
  1217. datasetSelector = projectedSelector;
  1218. //Following is a bit nasty....
  1219. //Converting the no_hqlproject to a no_newusertable means that some of the expressions
  1220. //are commoned up with expressions calculated by a previous filter, reducing the code.
  1221. //However, it isn't valid if transform contains an instance of newSelect -
  1222. //e.g. project(i(x), transform(exists(i....))) - see jholt25.xhql
  1223. //And unfortunately it fails silently.
  1224. //So we use queryReplaceSelector which fails if an ambiguity is introduced by the replacement
  1225. OwnedHqlExpr newSelect = ensureActiveRow(datasetSelector);
  1226. OwnedHqlExpr transform = queryNewReplaceSelector(expr->queryChild(1), leftSelect, newSelect);
  1227. BuildCtx subctx(ctx); // buildTargetCursor adds group if necessary
  1228. Linked<BoundRow> tempRow;
  1229. Linked<BoundRow> rowBuilder;
  1230. buildTargetCursor(tempRow, rowBuilder, subctx, expr);
  1231. if (!transform)
  1232. {
  1233. //The replace introduced an ambiguity => need to use the unmapped expression.
  1234. BoundRow * prevCursor = translator.resolveSelectorDataset(ctx, datasetSelector);
  1235. transform.set(expr->queryChild(1));
  1236. translator.rebindTableCursor(subctx, dataset, prevCursor, no_left, selSeq);
  1237. rebindFilepositons(subctx, dataset, no_left, selSeq);
  1238. }
  1239. if (returnIfFilterFails)
  1240. {
  1241. translator.associateSkipReturnMarker(subctx, failedFilterValue, NULL); // failedFilterValue already handles clearing the result
  1242. }
  1243. else
  1244. {
  1245. //MORE: Probably not implementable, should try and prevent this happening!
  1246. //translator.associateSkipReturnMarker(subctx, failedFilterValue, NULL);
  1247. }
  1248. translator.doTransform(subctx, transform, rowBuilder);
  1249. ctx.addGroup();
  1250. associateTargetCursor(subctx, ctx, tempRow, rowBuilder, expr);
  1251. }
  1252. break;
  1253. case no_newusertable:
  1254. {
  1255. BuildCtx subctx(ctx);
  1256. Linked<BoundRow> tempRow;
  1257. Linked<BoundRow> rowBuilder;
  1258. buildTargetCursor(tempRow, rowBuilder, subctx, expr);
  1259. LinkedHqlExpr transform = expr->queryChild(2);
  1260. if (tableExpr && (expr == firstTransformer))
  1261. {
  1262. if (tableSelector != projectedSelector)
  1263. transform.setown(newReplaceSelector(transform, tableSelector, projectedSelector));
  1264. }
  1265. if (returnIfFilterFails)
  1266. {
  1267. translator.associateSkipReturnMarker(subctx, failedFilterValue, NULL); // failedFilterValue already handles clearing the result
  1268. }
  1269. else
  1270. {
  1271. //MORE: Probably not implementable
  1272. //translator.associateSkipReturnMarker(subctx, failedFilterValue, NULL);
  1273. }
  1274. translator.doTransform(subctx, transform, rowBuilder);
  1275. ctx.addGroup();
  1276. associateTargetCursor(subctx, ctx, tempRow, rowBuilder, expr);
  1277. }
  1278. break;
  1279. case no_aggregate:
  1280. {
  1281. if (alreadyDoneFlag)
  1282. {
  1283. IHqlExpression * dataset = expr->queryChild(0);
  1284. IHqlExpression * selSeq = querySelSeq(expr);
  1285. IHqlExpression * transform = expr->queryChild(2);
  1286. BuildCtx subctx(ctx);
  1287. //Similar code to no_hqlproject. Should possibly try and do a similar mapping to the transforms
  1288. //(but complicated by having merge/first)
  1289. BoundRow * prevCursor = translator.resolveSelectorDataset(ctx, dataset);
  1290. translator.rebindTableCursor(subctx, dataset, prevCursor, no_left, selSeq);
  1291. rebindFilepositons(subctx, dataset, no_left, selSeq);
  1292. Linked<BoundRow> tempRow;
  1293. Linked<BoundRow> rowBuilder;
  1294. buildTargetCursor(tempRow, rowBuilder, subctx, expr);
  1295. translator.doBuildUserAggregateProcessTransform(subctx, rowBuilder, expr, transform, alreadyDoneFlag);
  1296. subctx.addAssign(alreadyDoneFlag, queryBoolExpr(true));
  1297. associateTargetCursor(subctx, ctx, tempRow, rowBuilder, expr);
  1298. }
  1299. }
  1300. break;
  1301. case no_newaggregate:
  1302. {
  1303. IHqlExpression * transform = expr->queryChild(2);
  1304. if (isCompoundCount)
  1305. {
  1306. //This should really be special cased in a count() baseclass, but can't be bothered at the moment.
  1307. IHqlExpression * rhs = transform->queryChild(0)->queryChild(1);
  1308. IHqlExpression * filter = queryRealChild(rhs, 0);
  1309. if (filter)
  1310. {
  1311. if (!returnIfFilterFails)
  1312. translator.buildFilter(ctx, filter);
  1313. else
  1314. {
  1315. OwnedHqlExpr test = getInverse(filter);
  1316. translator.buildFilteredReturn(ctx, test, failedFilterValue);
  1317. }
  1318. }
  1319. if (compoundCountVar)
  1320. {
  1321. OwnedHqlExpr inc = adjustValue(compoundCountVar, 1);
  1322. ctx.addAssign(compoundCountVar, inc);
  1323. }
  1324. }
  1325. else if (alreadyDoneFlag)
  1326. {
  1327. BuildCtx subctx(ctx);
  1328. Linked<BoundRow> tempRow;
  1329. Linked<BoundRow> rowBuilder;
  1330. buildTargetCursor(tempRow, rowBuilder, subctx, expr);
  1331. translator.doBuildAggregateProcessTransform(subctx, rowBuilder, expr, alreadyDoneFlag);
  1332. subctx.addAssign(alreadyDoneFlag, queryBoolExpr(true));
  1333. associateTargetCursor(subctx, ctx, tempRow, rowBuilder, expr);
  1334. }
  1335. }
  1336. break;
  1337. case no_usertable:
  1338. case no_selectfields:
  1339. UNIMPLEMENTED;
  1340. break;
  1341. case no_fetch:
  1342. {
  1343. BuildCtx subctx(ctx);
  1344. Linked<BoundRow> tempRow;
  1345. Linked<BoundRow> rowBuilder;
  1346. buildTargetCursor(tempRow, rowBuilder, subctx, expr);
  1347. // MORE - don't understand why this is required here but not in hqlproject above
  1348. IHqlExpression * dataset = expr->queryChild(0);
  1349. BoundRow * leftCursor;
  1350. switch (getDatasetKind(tableExpr))
  1351. {
  1352. case no_csv:
  1353. leftCursor = translator.bindCsvTableCursor(subctx, dataset, "Left", no_left, querySelSeq(expr), true, queryCsvTableEncoding(tableExpr));
  1354. break;
  1355. case no_xml:
  1356. case no_json:
  1357. leftCursor = translator.bindXmlTableCursor(subctx, dataset, "xmlLeft", no_left, querySelSeq(expr), true);
  1358. break;
  1359. default:
  1360. leftCursor = translator.bindTableCursor(subctx, dataset, "left", no_left, querySelSeq(expr));
  1361. break;
  1362. }
  1363. BoundRow * rightCursor = NULL;
  1364. LinkedHqlExpr transform = expr->queryChild(3);
  1365. if (!containsOnlyLeft(transform, true))
  1366. {
  1367. IHqlExpression * rhs = expr->queryChild(1);
  1368. IHqlExpression * memoryRhsRecord = rhs->queryRecord();
  1369. //RIGHT is the input data row.
  1370. //a) In hthor the remote file is read locally, so RIGHT is a deserialized unmodified row.
  1371. //b) in roxie it is serialized, and accessed in its serialized form.
  1372. //c) in thor it is serialized, but also (inefficiently) deserialized, so use the deserialized form.
  1373. //NOTE: Currently extractJoinFields either does nothing, or sends the entire row. Needless to say that could be improved -
  1374. //and would mean the roxie/thor code required changing
  1375. if (translator.targetRoxie())
  1376. {
  1377. OwnedHqlExpr serializedRhsRecord = getSerializedForm(memoryRhsRecord, diskAtom);
  1378. OwnedHqlExpr serializedRhs = createDataset(no_null, LINK(serializedRhsRecord));
  1379. rightCursor = translator.bindTableCursor(subctx, serializedRhs, "right", no_right, querySelSeq(expr));
  1380. transform.setown(replaceMemorySelectorWithSerializedSelector(transform, memoryRhsRecord, no_right, querySelSeq(expr), diskAtom));
  1381. }
  1382. else
  1383. {
  1384. rightCursor = translator.bindTableCursor(subctx, rhs, "right", no_right, querySelSeq(expr));
  1385. }
  1386. }
  1387. buildTransformFpos(subctx); // unusual, but this must occur after the row cursor is bound
  1388. translator.associateSkipReturnMarker(subctx, failedFilterValue, NULL); // failedFilterValue already handles clearing the result
  1389. translator.doTransform(subctx, transform, rowBuilder);
  1390. subctx.removeAssociation(leftCursor);
  1391. subctx.removeAssociation(rightCursor);
  1392. associateTargetCursor(subctx, ctx, tempRow, rowBuilder, expr);
  1393. }
  1394. break;
  1395. case no_compound_diskread:
  1396. case no_compound_disknormalize:
  1397. case no_compound_diskaggregate:
  1398. case no_compound_diskcount:
  1399. case no_compound_diskgroupaggregate:
  1400. case no_compound_indexread:
  1401. case no_compound_indexnormalize:
  1402. case no_compound_indexaggregate:
  1403. case no_compound_indexcount:
  1404. case no_compound_indexgroupaggregate:
  1405. case no_compound_childread:
  1406. case no_compound_childnormalize:
  1407. case no_compound_childaggregate:
  1408. case no_compound_childcount:
  1409. case no_compound_childgroupaggregate:
  1410. case no_compound_fetch:
  1411. case no_compound_selectnew:
  1412. break;
  1413. default:
  1414. throwUnexpectedOp(op);
  1415. }
  1416. }
  1417. void SourceBuilder::buildMatchFilter(BuildCtx & ctx, IHqlExpression * expr)
  1418. {
  1419. expr = expr->queryBody();
  1420. node_operator op = expr->getOperator();
  1421. switch (op)
  1422. {
  1423. case no_cachealias:
  1424. buildMatchFilter(ctx, expr->queryChild(1));
  1425. return;
  1426. case no_newkeyindex:
  1427. case no_table:
  1428. case no_fetch:
  1429. case no_select: // handled below
  1430. case no_null:
  1431. case no_anon:
  1432. case no_pseudods:
  1433. case no_workunit_dataset:
  1434. case no_getgraphresult:
  1435. case no_call:
  1436. case no_externalcall:
  1437. case no_rows:
  1438. case no_libraryinput:
  1439. return;
  1440. default:
  1441. buildMatchFilter(ctx, expr->queryChild(0));
  1442. break;
  1443. }
  1444. switch (op)
  1445. {
  1446. case no_filter:
  1447. {
  1448. LinkedHqlExpr cond;
  1449. if (useFilterMappings)
  1450. {
  1451. unsigned match = originalFilters.find(*expr);
  1452. if (match != NotFound)
  1453. cond.set(&mappedFilters.item(match));
  1454. }
  1455. else
  1456. {
  1457. HqlExprArray args;
  1458. unwindRealChildren(args, expr, 1);
  1459. cond.setown(createBalanced(no_and, queryBoolType(), args));
  1460. }
  1461. if (cond)
  1462. {
  1463. IHqlExpression * ds = expr->queryChild(0);
  1464. OwnedHqlExpr test = getInverse(cond);
  1465. if (translator.queryOptions().foldFilter)
  1466. test.setown(foldScopedHqlExpression(translator.queryErrorProcessor(), ds->queryNormalizedSelector(), test));
  1467. if (translator.options.spotCSE)
  1468. test.setown(spotScalarCSE(test, ds, translator.queryOptions().spotCseInIfDatasetConditions));
  1469. if (tableSelector != projectedSelector)
  1470. test.setown(newReplaceSelector(test, tableSelector, projectedSelector));
  1471. translator.buildFilteredReturn(ctx, test, queryBoolExpr(false));
  1472. }
  1473. }
  1474. break;
  1475. }
  1476. }
  1477. void SourceBuilder::doBuildAggregateSelectIterator(BuildCtx & ctx, IHqlExpression * expr)
  1478. {
  1479. IHqlExpression * ds = expr->queryChild(0);
  1480. if (isNewSelector(expr))
  1481. buildTransformElements(ctx, ds, false);
  1482. Owned<IHqlCppDatasetCursor> cursor = translator.createDatasetSelector(ctx, expr->queryNormalizedSelector());
  1483. cursor->buildIterateLoop(ctx, false);
  1484. }
  1485. void SourceBuilder::processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  1486. {
  1487. throwUnexpected();
  1488. }
  1489. void SourceBuilder::doBuildNormalizeIterators(BuildCtx & ctx, IHqlExpression * expr, bool isChildIterator)
  1490. {
  1491. HqlExprArray iterators;
  1492. IHqlExpression * root = gatherSelectorLevels(iterators, expr);
  1493. //MORE: transform also needs to be inside this iterctx
  1494. BuildCtx iterctx(*globaliterctx);
  1495. MemberFunction firstFunc(translator, instance->startctx);
  1496. CursorArray cursors;
  1497. if (isChildIterator)
  1498. {
  1499. assertex(!root);
  1500. firstFunc.start("virtual bool first() override");
  1501. }
  1502. else
  1503. {
  1504. assertex(root);
  1505. firstFunc.start("virtual bool first(const void * _src) override");
  1506. bool isProjected = (root->queryNormalizedSelector() != tableExpr->queryNormalizedSelector());
  1507. if (!isProjected)
  1508. {
  1509. iterctx.addQuotedLiteral("byte * src;");
  1510. associateFilePositions(iterctx, "activity->fpp", "activity->src"); // in case no projection in first()
  1511. firstFunc.ctx.addQuotedLiteral("src = (byte *)_src;");
  1512. }
  1513. else
  1514. {
  1515. firstFunc.ctx.addQuotedLiteral("byte * src = (byte *)_src;");
  1516. }
  1517. translator.associateBlobHelper(firstFunc.ctx, tableExpr, "fpp");
  1518. BoundRow * tableCursor = translator.bindTableCursor(firstFunc.ctx, tableExpr, "src");
  1519. associateFilePositions(firstFunc.ctx, "fpp", "src");
  1520. TransformSequenceBuilder builder(translator, queryBoolExpr(false));
  1521. builder.buildSequence(firstFunc.ctx, &iterctx, root);
  1522. if (!isProjected)
  1523. cursors.append(*LINK(tableCursor));
  1524. else
  1525. {
  1526. BoundRow * match = translator.resolveSelectorDataset(firstFunc.ctx, root);
  1527. assertex(match);
  1528. cursors.append(*LINK(match));
  1529. }
  1530. }
  1531. CompoundIteratorBuilder iterBuilder(translator, instance->nestedctx, iterctx);
  1532. if (iterators.ordinality() == 1)
  1533. {
  1534. StringBuffer s, iterName, cursorName;
  1535. iterBuilder.createSingleLevelIterator(iterName, cursorName, &iterators.item(0), cursors);
  1536. firstFunc.ctx.addQuoted(s.clear().append("return (").append(cursorName).append(" = (byte *)").append(iterName).append(".first()) != 0;"));
  1537. {
  1538. BuildCtx nextctx(instance->startctx);
  1539. nextctx.addQuotedFunction("virtual bool next() override");
  1540. nextctx.addQuoted(s.clear().append("return (").append(cursorName).append(" = (byte *)").append(iterName).append(".next()) != 0;"));
  1541. }
  1542. }
  1543. else
  1544. {
  1545. iterBuilder.buildCompoundIterator(instance->onlyEvalOnceContext(), iterators, cursors);
  1546. firstFunc.ctx.addQuotedLiteral("return iter.first();");
  1547. {
  1548. BuildCtx nextctx(instance->startctx);
  1549. nextctx.addQuotedFunction("virtual bool next() override");
  1550. nextctx.addQuotedLiteral("return iter.next();");
  1551. }
  1552. }
  1553. ForEachItemIn(i, cursors)
  1554. ctx.associate(cursors.item(i));
  1555. }
  1556. void SourceBuilder::checkDependencies(BuildCtx & ctx, IHqlExpression * expr)
  1557. {
  1558. //Add dependency information
  1559. Owned<ABoundActivity> bound = instance->getBoundActivity();
  1560. translator.addFileDependency(nameExpr, bound);
  1561. }
  1562. void SourceBuilder::extractMonitors(IHqlExpression * ds, SharedHqlExpr & unkeyedFilter, HqlExprArray & conds)
  1563. {
  1564. ForEachItemIn(i, conds)
  1565. {
  1566. IHqlExpression * filter = &conds.item(i);
  1567. #ifdef ALLOW_CHILD_ITERATORS_TO_CALL_CANMATCHANY
  1568. if (isSourceInvariant(ds, filter)) // more actually isSourceInvariant.
  1569. extendAndCondition(globalGuard, filter);
  1570. else
  1571. #endif
  1572. appendFilter(unkeyedFilter, filter);
  1573. }
  1574. }
  1575. void SourceBuilder::analyseGraph(IHqlExpression * expr)
  1576. {
  1577. analyse(expr);
  1578. SourceFieldUsage * fieldUsage = translator.querySourceFieldUsage(tableExpr);
  1579. if (fieldUsage && !fieldUsage->seenAll())
  1580. {
  1581. if (expr->queryNormalizedSelector() == tableExpr->queryNormalizedSelector())
  1582. fieldUsage->noteAll();
  1583. else
  1584. gatherFieldUsage(fieldUsage, expr);
  1585. }
  1586. }
  1587. void SourceBuilder::gatherFieldUsage(SourceFieldUsage * fieldUsage, IHqlExpression * expr)
  1588. {
  1589. for (;;)
  1590. {
  1591. if (expr->queryBody() == tableExpr->queryBody())
  1592. return;
  1593. if (fieldUsage->seenAll())
  1594. return;
  1595. IHqlExpression * ds = expr->queryChild(0);
  1596. switch (expr->getOperator())
  1597. {
  1598. case no_fetch:
  1599. {
  1600. assertex(ds->queryBody() == tableExpr->queryBody());
  1601. IHqlExpression * selSeq = querySelSeq(expr);
  1602. OwnedHqlExpr left = createSelector(no_left, ds, selSeq);
  1603. ::gatherFieldUsage(fieldUsage, expr, left);
  1604. return;
  1605. }
  1606. }
  1607. assertex(getNumChildTables(expr) == 1);
  1608. if (ds->queryNormalizedSelector() == tableExpr->queryNormalizedSelector())
  1609. gatherParentFieldUsage(fieldUsage, expr);
  1610. expr = ds;
  1611. }
  1612. }
  1613. inline bool useDescriptiveGraphLabel(ThorActivityKind kind)
  1614. {
  1615. switch (kind)
  1616. {
  1617. case TAKcsvfetch:
  1618. case TAKxmlfetch:
  1619. case TAKjsonfetch:
  1620. case TAKfetch:
  1621. return false;
  1622. }
  1623. return true;
  1624. }
  1625. static bool expandGraphLabel(ThorActivityKind kind)
  1626. {
  1627. switch (kind)
  1628. {
  1629. case TAKdiskread:
  1630. case TAKcsvread:
  1631. case TAKxmlread:
  1632. case TAKjsonread:
  1633. case TAKdiskcount:
  1634. case TAKdiskexists:
  1635. case TAKspillread:
  1636. return true;
  1637. default:
  1638. return false;
  1639. }
  1640. }
  1641. ABoundActivity * SourceBuilder::buildActivity(BuildCtx & ctx, IHqlExpression * expr, ThorActivityKind _activityKind, const char *kind, ABoundActivity *input)
  1642. {
  1643. activityKind = _activityKind;
  1644. translator.gatherActiveCursors(ctx, parentCursors);
  1645. bool isSpill = tableExpr && tableExpr->hasAttribute(_spill_Atom);
  1646. if (isSpill && (activityKind == TAKdiskread))
  1647. activityKind = TAKspillread;
  1648. useImplementationClass = translator.queryOptions().minimizeActivityClasses && translator.targetRoxie() && isSpill;
  1649. Owned<ActivityInstance> localInstance = new ActivityInstance(translator, ctx, activityKind, expr, kind);
  1650. if (useImplementationClass)
  1651. localInstance->setImplementationClass(newMemorySpillReadArgId);
  1652. if (((activityKind >= TAKdiskread) && (activityKind <= TAKdiskgroupaggregate)) || (activityKind == TAKspillread))
  1653. {
  1654. IHqlExpression * seq = querySequence(tableExpr);
  1655. translator.noteResultAccessed(ctx, seq, nameExpr);
  1656. }
  1657. instance = localInstance;
  1658. StringBuffer graphLabel;
  1659. graphLabel.append(getActivityText(activityKind));
  1660. bool isFiltered = false;
  1661. double filterLikelihood = 1.0;
  1662. if (expandGraphLabel(activityKind))
  1663. {
  1664. graphLabel.clear();
  1665. if (expr != tableExpr)
  1666. {
  1667. IHqlExpression * cur = expr;
  1668. bool isProjected = false;
  1669. for (;;)
  1670. {
  1671. switch (cur->getOperator())
  1672. {
  1673. case no_filter:
  1674. if (isKnownLikelihood(filterLikelihood))
  1675. {
  1676. double likelihood = queryActivityLikelihood(cur);
  1677. if (isKnownLikelihood(likelihood))
  1678. // Combine the likelihood of the 2 filter conditions
  1679. // N.B. this only works if the filter probability are independent
  1680. filterLikelihood *= likelihood;
  1681. else
  1682. // One of the filter probability is unknown, so the overall probability is unknown
  1683. setUnknownLikelihood(filterLikelihood);
  1684. }
  1685. isFiltered = true;
  1686. break;
  1687. case no_hqlproject:
  1688. case no_newusertable:
  1689. case no_transformascii:
  1690. case no_transformebcdic:
  1691. if (!cur->hasAttribute(_internal_Atom))
  1692. isProjected = true;
  1693. break;
  1694. case no_table:
  1695. cur = NULL;
  1696. break;
  1697. }
  1698. if (!cur)
  1699. break;
  1700. cur = cur->queryChild(0);
  1701. }
  1702. if (isFiltered)
  1703. {
  1704. graphLabel.append("Filtered\n");
  1705. }
  1706. if (isProjected)
  1707. graphLabel.append("Projected\n");
  1708. }
  1709. if ((translator.getTargetClusterType() == RoxieCluster) && isSpill)
  1710. graphLabel.append("Read");
  1711. else if (isExplicitExists() && (activityKind == TAKdiskcount))
  1712. graphLabel.append("Disk Exists");
  1713. else
  1714. graphLabel.append(getActivityText(activityKind));
  1715. }
  1716. if (isExplicitExists())
  1717. {
  1718. if (activityKind == TAKindexcount)
  1719. graphLabel.clear().append("Index Exists");
  1720. }
  1721. if (useDescriptiveGraphLabel(activityKind))
  1722. {
  1723. StringBuffer eclChunk;
  1724. bool isLimited = false;
  1725. bool isStepped = false;
  1726. IHqlExpression * cur = expr;
  1727. instance->graphEclText.clear();
  1728. for (;;)
  1729. {
  1730. eclChunk.clear();
  1731. switch (cur->getOperator())
  1732. {
  1733. case no_hqlproject:
  1734. case no_newusertable:
  1735. case no_transformascii:
  1736. case no_transformebcdic:
  1737. break;
  1738. case no_filter:
  1739. toECL(cur->queryBody(), eclChunk, false, true);
  1740. break;
  1741. case no_table:
  1742. case no_newkeyindex:
  1743. case no_select:
  1744. toECL(cur->queryBody(), eclChunk, false, true);
  1745. cur = NULL;
  1746. break;
  1747. case no_stepped:
  1748. isStepped = true;
  1749. break;
  1750. case no_limit:
  1751. case no_keyedlimit:
  1752. isLimited = true;
  1753. break;
  1754. default:
  1755. if (getNumChildTables(cur) == 0)
  1756. {
  1757. toECL(cur->queryBody(), eclChunk, false, true);
  1758. cur = NULL;
  1759. }
  1760. break;
  1761. }
  1762. instance->graphEclText.insert(0, eclChunk);
  1763. if (!cur)
  1764. break;
  1765. cur = cur->queryChild(0);
  1766. }
  1767. if (isLimited)
  1768. graphLabel.insert(0, "Limited\n");
  1769. if (isStepped)
  1770. graphLabel.insert(0, "Stepped\n");
  1771. if (localInstance->isLocal && !isSpill)
  1772. graphLabel.insert(0, "Local ");
  1773. }
  1774. if (nameExpr && nameExpr->queryValue())
  1775. {
  1776. if (isSpill)
  1777. {
  1778. if (activityKind != TAKspillread)
  1779. graphLabel.append("\nSpill");
  1780. }
  1781. else
  1782. {
  1783. graphLabel.newline();
  1784. if (tableExpr->hasAttribute(_workflowPersist_Atom))
  1785. graphLabel.append("Persist ");
  1786. StringBuffer filename;
  1787. //Call getStringValue() rather than generateECL() to avoid 't quote \ etc. in the string
  1788. getStringValue(filename.append("'"), nameExpr).append("'");
  1789. const char * coloncolon = strstr(filename, "::");
  1790. if (coloncolon)
  1791. {
  1792. for (;;)
  1793. {
  1794. const char * next = strstr(coloncolon+2, "::");
  1795. if (!next)
  1796. break;
  1797. coloncolon = next;
  1798. }
  1799. graphLabel.append("'...").append(coloncolon);
  1800. }
  1801. else
  1802. graphLabel.append(filename);
  1803. }
  1804. }
  1805. instance->graphLabel.set(graphLabel.str());
  1806. translator.buildActivityFramework(localInstance);
  1807. translator.buildInstancePrefix(localInstance);
  1808. analyseGraph(expr);
  1809. if (!useImplementationClass)
  1810. {
  1811. buildMembers(expr);
  1812. buildTransform(expr);
  1813. buildCanMatch(expr);
  1814. buildFlagsMember(expr);
  1815. if (tableExpr && (activityKind < TAKchildread || activityKind > TAKchildthroughnormalize))
  1816. {
  1817. switch (activityKind)
  1818. {
  1819. case TAKindexread:
  1820. case TAKindexnormalize:
  1821. case TAKindexaggregate:
  1822. case TAKindexcount:
  1823. case TAKindexgroupaggregate:
  1824. case TAKindexexists:
  1825. {
  1826. LinkedHqlExpr indexExpr = queryAttributeChild(tableExpr, _original_Atom, 0);
  1827. OwnedHqlExpr serializedRecord;
  1828. unsigned numPayload = numPayloadFields(indexExpr);
  1829. if (numPayload)
  1830. serializedRecord.setown(notePayloadFields(indexExpr->queryRecord(), numPayload));
  1831. else
  1832. serializedRecord.set(indexExpr->queryRecord());
  1833. serializedRecord.setown(getSerializedForm(serializedRecord, diskAtom));
  1834. bool hasFilePosition = getBoolAttribute(indexExpr, filepositionAtom, true);
  1835. serializedRecord.setown(createMetadataIndexRecord(serializedRecord, hasFilePosition));
  1836. translator.buildMetaMember(instance->classctx, serializedRecord, false, "queryDiskRecordSize");
  1837. translator.buildMetaMember(instance->classctx, serializedRecord, false, "queryProjectedDiskRecordSize");
  1838. break;
  1839. }
  1840. }
  1841. }
  1842. }
  1843. else
  1844. {
  1845. assertex(!hasDynamic(tableExpr));
  1846. bool matched = translator.registerGlobalUsage(nameExpr);
  1847. if (!matched)
  1848. {
  1849. StringBuffer spillName;
  1850. getExprECL(nameExpr, spillName);
  1851. if (translator.queryOptions().allowThroughSpill)
  1852. throwError1(HQLERR_ReadSpillBeforeWriteFix, spillName.str());
  1853. else
  1854. throwError1(HQLERR_ReadSpillBeforeWrite, spillName.str());
  1855. }
  1856. translator.addFilenameConstructorParameter(*instance, WaFilename, nameExpr);
  1857. }
  1858. if (steppedExpr)
  1859. buildSteppedHelpers();
  1860. if (translator.targetRoxie())
  1861. instance->addAttributeBool(WaIsSpill, isSpill);
  1862. else if (needToCallTransform || transformCanFilter)
  1863. instance->addAttributeBool(WaIsTransformSpill, isSpill);
  1864. else
  1865. instance->addAttributeBool(WaIsSpill, isSpill);
  1866. if (isFiltered)
  1867. {
  1868. if (isKnownLikelihood(filterLikelihood))
  1869. {
  1870. StringBuffer text;
  1871. filterLikelihood *= 100;
  1872. text.setf("%3.2f%%", filterLikelihood);
  1873. instance->addAttribute(WaMatchLikelihood, text);
  1874. }
  1875. }
  1876. IHqlExpression * spillReason = tableExpr ? queryAttributeChild(tableExpr, _spillReason_Atom, 0) : NULL;
  1877. if (spillReason && !translator.queryOptions().obfuscateOutput)
  1878. {
  1879. StringBuffer text;
  1880. getStringValue(text, spillReason);
  1881. instance->addAttribute(WaSpillReason, text.str());
  1882. }
  1883. if (tableExpr)
  1884. instance->addSignedAttribute(tableExpr->queryAttribute(_signed_Atom));
  1885. checkDependencies(ctx, expr);
  1886. translator.buildInstanceSuffix(localInstance);
  1887. if (input)
  1888. translator.buildConnectInputOutput(ctx, localInstance, input, 0, 0);
  1889. instance = NULL;
  1890. return localInstance->getBoundActivity();
  1891. }
  1892. void SourceBuilder::buildKeyedLimitHelper(IHqlExpression * self)
  1893. {
  1894. if (keyedLimitExpr)
  1895. {
  1896. IHqlExpression * limitValue = keyedLimitExpr->queryChild(1);
  1897. {
  1898. MemberFunction func(translator, instance->startctx, "virtual unsigned __int64 getKeyedLimit() override");
  1899. translator.buildReturn(func.ctx, limitValue);
  1900. if (isZero(limitValue))
  1901. translator.WARNING(CategoryUnusual, HQLWRN_KeyedLimitIsZero);
  1902. }
  1903. LinkedHqlExpr fail = keyedLimitExpr->queryChild(2);
  1904. if (!fail || fail->isAttribute())
  1905. fail.setown(translator.createFailAction("Keyed limit exceeded", limitValue, NULL, instance->activityId));
  1906. {
  1907. MemberFunction func(translator, instance->startctx, "virtual void onKeyedLimitExceeded() override");
  1908. translator.buildStmt(func.ctx, fail);
  1909. }
  1910. IHqlExpression * transform = queryAttributeChild(keyedLimitExpr, onFailAtom, 0);
  1911. if (transform)
  1912. {
  1913. MemberFunction func(translator, instance->startctx, "virtual size32_t transformOnKeyedLimitExceeded(ARowBuilder & crSelf) override");
  1914. translator.ensureRowAllocated(func.ctx, "crSelf");
  1915. translator.buildTransformBody(func.ctx, transform, NULL, NULL, self, NULL);
  1916. }
  1917. }
  1918. }
  1919. void SourceBuilder::buildSteppedHelpers()
  1920. {
  1921. StringBuffer steppedFlags;
  1922. IHqlExpression * priority = steppedExpr->queryAttribute(priorityAtom);
  1923. if (priority)
  1924. {
  1925. steppedFlags.append("|SSFhaspriority");
  1926. translator.doBuildFunction(instance->startctx, doubleType, "getPriority", priority->queryChild(0));
  1927. }
  1928. IHqlExpression * prefetch = steppedExpr->queryAttribute(prefetchAtom);
  1929. if (prefetch)
  1930. {
  1931. steppedFlags.append("|SSFhasprefetch");
  1932. translator.doBuildUnsignedFunction(instance->startctx, "getPrefetchSize", prefetch->queryChild(0));
  1933. }
  1934. IHqlExpression * filtered = steppedExpr->queryAttribute(filteredAtom);
  1935. if (filtered)
  1936. steppedFlags.append("|SSFalwaysfilter");
  1937. if (steppedFlags.length())
  1938. translator.doBuildUnsignedFunction(instance->classctx, "getSteppedFlags", steppedFlags.str()+1);
  1939. }
  1940. void SourceBuilder::assignLocalExtract(BuildCtx & ctx, ParentExtract * extractBuilder, IHqlExpression * dataset, const char * argName)
  1941. {
  1942. if (extractBuilder)
  1943. {
  1944. StringBuffer s;
  1945. s.append("byte * ");
  1946. translator.generateExprCpp(s, extractBuilder->queryExtractName());
  1947. s.append(" = (byte *) ").append(argName).append(";");
  1948. ctx.addQuoted(s);
  1949. }
  1950. else
  1951. {
  1952. StringBuffer s;
  1953. ctx.addQuoted(s.append("unsigned char * left = (unsigned char *) ").append(argName).append(";"));
  1954. translator.bindTableCursor(ctx, dataset, "left");
  1955. }
  1956. }
  1957. void SourceBuilder::buildGroupAggregateHashHelper(ParentExtract * extractBuilder, IHqlExpression * dataset, IHqlExpression * fields)
  1958. {
  1959. instance->classctx.addQuotedLiteral("virtual IHash * queryHash() override { return &hash; }");
  1960. BuildCtx classctx(instance->nestedctx);
  1961. IHqlStmt * classStmt = translator.beginNestedClass(classctx, "hash", "IHash", NULL, extractBuilder);
  1962. {
  1963. MemberFunction func(translator, classctx, "virtual unsigned hash(const void * _self) override");
  1964. assignLocalExtract(func.ctx, extractBuilder, dataset, "_self");
  1965. OwnedHqlExpr hash = createValue(no_hash32, LINK(unsignedType), LINK(fields), createAttribute(internalAtom));
  1966. translator.buildReturn(func.ctx, hash);
  1967. }
  1968. translator.endNestedClass(classStmt);
  1969. }
  1970. void SourceBuilder::buildGroupAggregateCompareHelper(ParentExtract * extractBuilder, IHqlExpression * aggregate, HqlExprArray & recordFields, HqlExprArray & aggregateFields)
  1971. {
  1972. //Special case comparison of IF(<global-bool>, value, some-constant), so that the order expression only compares if <global-bool> is true
  1973. //specifically to improve david's BIRPs code.
  1974. HqlExprArray optimizedLeft, optimizedRight;
  1975. ForEachItemIn(i, recordFields)
  1976. {
  1977. IHqlExpression & curLeft = recordFields.item(i);
  1978. IHqlExpression & curRight = aggregateFields.item(i);
  1979. IHqlExpression * newLeft = NULL;
  1980. IHqlExpression * newRight = NULL;
  1981. if (curLeft.getOperator() == no_if)
  1982. {
  1983. IHqlExpression * cond = curLeft.queryChild(0);
  1984. if (isIndependentOfScope(cond))
  1985. {
  1986. IHqlExpression * elseValue = curLeft.queryChild(2);
  1987. if (elseValue->getOperator() == no_constant)
  1988. {
  1989. newLeft = LINK(&curLeft);
  1990. newRight = createValue(no_if, curRight.getType(), LINK(cond), LINK(&curRight), LINK(elseValue));
  1991. }
  1992. }
  1993. }
  1994. if (newLeft)
  1995. {
  1996. optimizedLeft.append(*newLeft);
  1997. optimizedRight.append(*newRight);
  1998. }
  1999. else
  2000. {
  2001. optimizedLeft.append(OLINK(curLeft));
  2002. optimizedRight.append(OLINK(curRight));
  2003. }
  2004. }
  2005. OwnedHqlExpr leftList = createSortList(optimizedLeft);
  2006. DatasetReference datasetRight(aggregate, no_activetable, NULL);
  2007. OwnedHqlExpr selSeq = createDummySelectorSequence();
  2008. OwnedHqlExpr rightList = createSortList(optimizedRight);
  2009. OwnedHqlExpr rightSelect = datasetRight.getSelector(no_right, selSeq);
  2010. OwnedHqlExpr rightResolved = datasetRight.mapCompound(rightList, rightSelect);
  2011. OwnedHqlExpr order = createValue(no_order, makeIntType(sizeof(signed), true), LINK(leftList), LINK(rightResolved));
  2012. //Now generate the nested class
  2013. instance->classctx.addQuotedLiteral("virtual ICompare * queryCompareRowElement() override { return &compareRowElement; }");
  2014. BuildCtx classctx(instance->nestedctx);
  2015. IHqlStmt * classStmt = translator.beginNestedClass(classctx, "compareRowElement", "ICompare", NULL, extractBuilder);
  2016. {
  2017. MemberFunction func(translator, classctx, "virtual int docompare(const void * _left, const void * _right) const override");
  2018. assignLocalExtract(func.ctx, extractBuilder, aggregate->queryChild(0), "_left");
  2019. func.ctx.addQuotedLiteral("const unsigned char * right = (const unsigned char *) _right;");
  2020. func.ctx.associateExpr(constantMemberMarkerExpr, constantMemberMarkerExpr);
  2021. translator.bindTableCursor(func.ctx, aggregate, "right", no_right, selSeq);
  2022. translator.doBuildReturnCompare(func.ctx, order, no_eq, false, false);
  2023. }
  2024. translator.endNestedClass(classStmt);
  2025. }
  2026. void SourceBuilder::buildGroupAggregateProcessHelper(ParentExtract * extractBuilder, IHqlExpression * aggregate, const char * name, bool doneAny)
  2027. {
  2028. StringBuffer proto;
  2029. IHqlExpression * dataset = aggregate->queryChild(0);
  2030. IHqlExpression * tgtRecord = aggregate->queryChild(1);
  2031. OwnedHqlExpr resultDataset = createDataset(no_anon, LINK(tgtRecord));
  2032. proto.append("virtual size32_t ").append(name).append("(ARowBuilder & crSelf, const void * _src) override");
  2033. MemberFunction validateFunc(translator, instance->nestedctx, proto, MFdynamicproto);
  2034. translator.ensureRowAllocated(validateFunc.ctx, "crSelf");
  2035. assignLocalExtract(validateFunc.ctx, extractBuilder, dataset, "_src");
  2036. BoundRow * selfRow = translator.bindSelf(validateFunc.ctx, resultDataset, "crSelf");
  2037. if (extractBuilder)
  2038. {
  2039. MemberEvalContext * evalContext = new MemberEvalContext(translator, extractBuilder, translator.queryEvalContext(validateFunc.ctx), validateFunc.ctx);
  2040. validateFunc.ctx.associateOwn(*evalContext);
  2041. evalContext->initContext();
  2042. validateFunc.ctx.associateExpr(codeContextMarkerExpr, codeContextMarkerExpr);
  2043. }
  2044. if (aggregate->getOperator() == no_aggregate)
  2045. {
  2046. //It is inefficient to call processUserAggregateTransform() twice, but this is an unusual construct, so leave as it
  2047. //is for the moment.
  2048. OwnedHqlExpr firstTransform;
  2049. OwnedHqlExpr nextTransform;
  2050. translator.processUserAggregateTransform(aggregate, aggregate->queryChild(2), firstTransform, nextTransform);
  2051. IHqlExpression * transform = doneAny ? nextTransform : firstTransform;
  2052. OwnedHqlExpr left = createSelector(no_left, dataset, querySelSeq(aggregate));
  2053. OwnedHqlExpr mappedTransform = replaceSelector(transform, left, dataset);
  2054. translator.doBuildUserAggregateProcessTransform(validateFunc.ctx, selfRow, aggregate, mappedTransform, queryBoolExpr(doneAny));
  2055. }
  2056. else
  2057. translator.doBuildAggregateProcessTransform(validateFunc.ctx, selfRow, aggregate, queryBoolExpr(doneAny));
  2058. translator.buildReturnRecordSize(validateFunc.ctx, selfRow);
  2059. }
  2060. void SourceBuilder::buildGroupAggregateHelpers(ParentExtract * extractBuilder, IHqlExpression * aggregate)
  2061. {
  2062. IHqlExpression * dataset = aggregate->queryChild(0);
  2063. LinkedHqlExpr transform = aggregate->queryChild(2);
  2064. LinkedHqlExpr grouping = aggregate->queryChild(3);
  2065. if (aggregate->getOperator() == no_aggregate)
  2066. {
  2067. OwnedHqlExpr left = createSelector(no_left, dataset, querySelSeq(aggregate));
  2068. grouping.setown(replaceSelector(grouping, left, dataset));
  2069. transform.setown(replaceSelector(transform, left, dataset));
  2070. }
  2071. HqlExprArray recordFields, aggregateFields;
  2072. grouping->unwindList(recordFields, no_sortlist);
  2073. getMappedFields(aggregateFields, transform, recordFields, queryActiveTableSelector());
  2074. OwnedHqlExpr allRecordFields = createValueSafe(no_sortlist, makeSortListType(NULL), recordFields);
  2075. OwnedHqlExpr allAggregateFields = createValueSafe(no_sortlist, makeSortListType(NULL), aggregateFields);
  2076. //virtual size32_t processFirst(void * target, const void * src)
  2077. buildGroupAggregateProcessHelper(extractBuilder, aggregate, "processFirst", false);
  2078. //virtual size32_t processNext(void * target, const void * src)
  2079. buildGroupAggregateProcessHelper(extractBuilder, aggregate, "processNext", true);
  2080. //virtual IHash * queryHash()
  2081. buildGroupAggregateHashHelper(extractBuilder, dataset, allRecordFields);
  2082. //virtual ICompare * queryCompareElements()
  2083. DatasetReference outRef(aggregate, no_activetable, NULL);
  2084. translator.buildCompareMember(instance->nestedctx, "CompareElements", allAggregateFields, outRef); // compare transformed elements
  2085. //virtual ICompare * queryCompareRowElement()
  2086. buildGroupAggregateCompareHelper(extractBuilder, aggregate, recordFields, aggregateFields);
  2087. //virtual IHash * queryHashElement()
  2088. translator.buildHashOfExprsClass(instance->nestedctx, "HashElement", allAggregateFields, outRef, true);
  2089. }
  2090. IHqlExpression * SourceBuilder::ensureAggregateGroupingAliased(IHqlExpression * aggregate)
  2091. {
  2092. IHqlExpression * dataset = aggregate->queryChild(0);
  2093. IHqlExpression * grouping = aggregate->queryChild(3);
  2094. //Force complex grouping fields into aliases to reduce processing...
  2095. HqlMapTransformer transformer;
  2096. transformer.setMapping(dataset, dataset);
  2097. ForEachChild(i, grouping)
  2098. {
  2099. IHqlExpression * cur = grouping->queryChild(i);
  2100. if (translator.requiresTemp(instance->nestedctx, cur, true) && (cur->getOperator() != no_alias))
  2101. {
  2102. OwnedHqlExpr alias = createAliasOwn(LINK(cur), createAttribute(internalAtom));
  2103. transformer.setMapping(cur, alias);
  2104. }
  2105. }
  2106. return transformer.transformRoot(aggregate);
  2107. }
  2108. void SourceBuilder::buildGlobalGroupAggregateHelpers(IHqlExpression * expr)
  2109. {
  2110. IHqlExpression * aggregate = expr->queryChild(0);
  2111. node_operator op = aggregate->getOperator();
  2112. assertex(op == no_newaggregate || op == no_aggregate);
  2113. StringBuffer s;
  2114. //virtual size32_t clearAggregate(void * self) = 0;
  2115. translator.doBuildAggregateClearFunc(instance->startctx, aggregate);
  2116. //virtual size32_t mergeAggregate(ARowBuilder & crSelf, const void * src) = 0; //only call if transform called at least once on src.
  2117. translator.doBuildAggregateMergeFunc(instance->startctx, aggregate, requiresOrderedMerge);
  2118. //virtual void processRow(void * self, const void * src) = 0;
  2119. {
  2120. BuildCtx rowctx(instance->startctx);
  2121. rowctx.addQuotedFunction("virtual void processRow(const void * src, IHThorGroupAggregateCallback * callback) override");
  2122. rowctx.addQuotedLiteral("doProcessRow((byte *)src, callback);");
  2123. }
  2124. //virtual void processRows(void * self, size32_t srcLen, const void * src) = 0;
  2125. //Only meaningful for a dataset, and even then I'm not sure it is ever used.
  2126. if (!isKey(tableExpr))
  2127. {
  2128. OwnedHqlExpr newTableExpr = LINK(tableExpr);
  2129. MemberFunction func(translator, instance->startctx, "virtual void processRows(size32_t srcLen, const void * _left, IHThorGroupAggregateCallback * callback) override");
  2130. func.ctx.addQuotedLiteral("unsigned char * left = (unsigned char *)_left;");
  2131. OwnedHqlExpr ds = createVariable("left", makeReferenceModifier(newTableExpr->getType()));
  2132. OwnedHqlExpr len = createVariable("srcLen", LINK(sizetType));
  2133. OwnedHqlExpr fullDs = createTranslated(ds, len);
  2134. BoundRow * curRow = translator.buildDatasetIterate(func.ctx, fullDs, false);
  2135. s.clear().append("doProcessRow(");
  2136. translator.generateExprCpp(s, curRow->queryBound());
  2137. s.append(", callback);");
  2138. func.ctx.addQuoted(s);
  2139. }
  2140. }
  2141. void SourceBuilder::buildGroupingMonitors(IHqlExpression * expr, CppFilterExtractor & monitors)
  2142. {
  2143. IHqlExpression * aggregate = expr->queryChild(0);
  2144. node_operator op = aggregate->getOperator();
  2145. assertex(op == no_newaggregate || op == no_aggregate);
  2146. IHqlExpression * dataset = aggregate->queryChild(0);
  2147. IHqlExpression * grouping = aggregate->queryChild(3);
  2148. //virtual void createGroupSegmentMonitors(IIndexReadContext *ctx) = 0;
  2149. MemberFunction func(translator, instance->startctx, "virtual bool createGroupSegmentMonitors(IIndexReadContext * irc) override");
  2150. monitorsForGrouping = true;
  2151. if (op == no_newaggregate)
  2152. translator.bindTableCursor(func.ctx, dataset, "_dummy");
  2153. else
  2154. translator.bindTableCursor(func.ctx, dataset, "_dummy", no_left, querySelSeq(aggregate));
  2155. unsigned maxField = 0;
  2156. ForEachChild(i, grouping)
  2157. {
  2158. unsigned nextField = 0;
  2159. if (!monitors.createGroupingMonitor(func.ctx, "irc", grouping->queryChild(i), nextField))
  2160. {
  2161. monitorsForGrouping = false;
  2162. func.setIncluded(false);
  2163. break;
  2164. }
  2165. if (maxField < nextField)
  2166. maxField = nextField;
  2167. }
  2168. if (monitorsForGrouping)
  2169. func.ctx.addReturn(queryBoolExpr(true));
  2170. if (monitorsForGrouping)
  2171. {
  2172. translator.doBuildUnsignedFunction(instance->classctx, "getGroupingMaxField", maxField);
  2173. }
  2174. }
  2175. void SourceBuilder::buildAggregateHelpers(IHqlExpression * expr)
  2176. {
  2177. IHqlExpression * aggregate = expr->queryChild(0);
  2178. node_operator op = aggregate->getOperator();
  2179. assertex(op == no_newaggregate || op == no_aggregate);
  2180. StringBuffer s;
  2181. alreadyDoneFlag.setown(instance->startctx.getTempDeclare(queryBoolType(), NULL));
  2182. instance->onstartctx.addAssign(alreadyDoneFlag, queryBoolExpr(false));
  2183. //virtual bool processedAnyRows() = 0;
  2184. translator.doBuildBoolFunction(instance->startctx, "processedAnyRows", alreadyDoneFlag);
  2185. //virtual size32_t clearAggregate(ARowBuilder & crSelf) = 0;
  2186. translator.doBuildAggregateClearFunc(instance->startctx, aggregate);
  2187. //virtual size32_t mergeAggregate(ARowBuilder & crSelf, const void * src) = 0; //only call if transform called at least once on src.
  2188. translator.doBuildAggregateMergeFunc(instance->startctx, aggregate, requiresOrderedMerge);
  2189. }
  2190. void SourceBuilder::buildCountHelpers(IHqlExpression * expr, bool allowMultiple)
  2191. {
  2192. StringBuffer s;
  2193. //---- virtual bool hasFilter() { return <bool>; } ----
  2194. if (transformCanFilter||isNormalize)
  2195. translator.doBuildBoolFunction(instance->classctx, "hasFilter", true);
  2196. if (allowMultiple)
  2197. {
  2198. bool isExists = hasExistChoosenLimit();
  2199. OwnedHqlExpr one = getSizetConstant(1);
  2200. if (transformCanFilter||isNormalize)
  2201. {
  2202. //virtual size32_t numValid(const void * src) = 0;
  2203. {
  2204. BuildCtx rowctx(instance->startctx);
  2205. rowctx.addQuotedFunction("virtual size32_t numValid(const void * src) override");
  2206. rowctx.addQuotedLiteral("return valid((byte *)src);");
  2207. }
  2208. //virtual size32_t numValid(size32_t srcLen, const void * src);
  2209. {
  2210. MemberFunction func(translator, instance->startctx, "virtual size32_t numValid(size32_t srcLen, const void * _src) override");
  2211. func.ctx.addQuotedLiteral("unsigned char * src = (unsigned char *)_src;");
  2212. OwnedHqlExpr ds = createVariable("src", makeReferenceModifier(tableExpr->getType()));
  2213. OwnedHqlExpr len = createVariable("srcLen", LINK(sizetType));
  2214. OwnedHqlExpr fullDs = createTranslated(ds, len);
  2215. if (isExists)
  2216. {
  2217. BuildCtx iterctx(func.ctx);
  2218. BoundRow * curRow = translator.buildDatasetIterate(iterctx, fullDs, false);
  2219. s.clear().append("if (valid(");
  2220. translator.generateExprCpp(s, curRow->queryBound());
  2221. s.append("))");
  2222. iterctx.addQuotedCompound(s, nullptr);
  2223. iterctx.addReturn(one);
  2224. func.ctx.addQuotedLiteral("return 0;");
  2225. }
  2226. else
  2227. {
  2228. func.ctx.addQuotedLiteral("size32_t cnt = 0;");
  2229. BuildCtx iterctx(func.ctx);
  2230. BoundRow * curRow = translator.buildDatasetIterate(iterctx, fullDs, false);
  2231. s.clear().append("cnt += valid(");
  2232. translator.generateExprCpp(s, curRow->queryBound());
  2233. s.append(");");
  2234. iterctx.addQuoted(s);
  2235. func.ctx.addQuotedLiteral("return cnt;");
  2236. }
  2237. }
  2238. }
  2239. else
  2240. {
  2241. //virtual size32_t numValid(size32_t srcLen, const void * src);
  2242. MemberFunction func(translator, instance->startctx, "virtual size32_t numValid(size32_t srcLen, const void * _src) override");
  2243. if (isExists)
  2244. func.ctx.addReturn(one);
  2245. else
  2246. {
  2247. func.ctx.addQuotedLiteral("unsigned char * src = (unsigned char *)_src;");
  2248. CHqlBoundExpr bound;
  2249. bound.length.setown(createVariable("srcLen", LINK(sizetType)));
  2250. bound.expr.setown(createVariable("src", makeReferenceModifier(tableExpr->getType())));
  2251. OwnedHqlExpr count = translator.getBoundCount(bound);
  2252. func.ctx.addReturn(count);
  2253. }
  2254. }
  2255. }
  2256. }
  2257. void SourceBuilder::buildNormalizeHelpers(IHqlExpression * expr)
  2258. {
  2259. }
  2260. void SourceBuilder::buildCanMatch(IHqlExpression * expr)
  2261. {
  2262. if (extractCanMatch)
  2263. {
  2264. MemberFunction func(translator, instance->startctx);
  2265. func.start("virtual bool canMatch(const void * _left) override");
  2266. func.ctx.addQuotedLiteral("unsigned char * left = (unsigned char *)_left;");
  2267. if (newDiskReadMapping)
  2268. translator.bindTableCursor(func.ctx, projectedSelector, "left");
  2269. else
  2270. translator.bindTableCursor(func.ctx, tableExpr, "left");
  2271. //This will have no ill effect for disk read, and is used for blob lookup
  2272. translator.associateBlobHelper(func.ctx, tableExpr, "fpp");
  2273. buildTransformFpos(func.ctx);
  2274. unsigned mark = func.numStmts();
  2275. buildMatchFilter(func.ctx, firstTransformer ? firstTransformer->queryChild(0) : expr);
  2276. if (func.numStmts() != mark)
  2277. {
  2278. func.ctx.addReturn(queryBoolExpr(true));
  2279. translator.doBuildBoolFunction(instance->classctx, "hasMatchFilter", true);
  2280. }
  2281. else
  2282. func.setIncluded(false);
  2283. }
  2284. }
  2285. void SourceBuilder::buildGroupAggregateTransformBody(BuildCtx & transformCtx, IHqlExpression * expr, bool useExtract, bool bindInputRow)
  2286. {
  2287. buildTransformBody(transformCtx, expr, false, false, bindInputRow);
  2288. IHqlExpression * aggregate = expr->queryChild(0);
  2289. OwnedHqlExpr mappedAggregate = ensureAggregateGroupingAliased(aggregate);
  2290. Owned<ParentExtract> extractBuilder;
  2291. if (useExtract || (aggregate != mappedAggregate))
  2292. {
  2293. extractBuilder.setown(translator.createExtractBuilder(transformCtx, PETcallback, NULL, GraphCoLocal, true));
  2294. if (!translator.queryOptions().serializeRowsetInExtract)
  2295. extractBuilder->setAllowDestructor();
  2296. translator.beginExtract(transformCtx, extractBuilder);
  2297. buildGroupAggregateHelpers(extractBuilder, mappedAggregate);
  2298. translator.endExtract(transformCtx, extractBuilder);
  2299. }
  2300. else
  2301. buildGroupAggregateHelpers(NULL, mappedAggregate);
  2302. HqlExprArray args;
  2303. args.append(*createVariable("callback", makeBoolType()));
  2304. if (extractBuilder)
  2305. {
  2306. CHqlBoundExpr boundExtract;
  2307. extractBuilder->endCreateExtract(boundExtract);
  2308. args.append(*boundExtract.getTranslatedExpr());
  2309. }
  2310. else
  2311. {
  2312. BoundRow * match = translator.resolveDatasetRequired(transformCtx, aggregate->queryChild(0));
  2313. Owned<ITypeInfo> rowType = makeReferenceModifier(makeRowType(queryNullRecord()->getType()));
  2314. OwnedHqlExpr rowAddr = getPointer(match->queryBound());
  2315. OwnedHqlExpr castBound = createValue(no_typetransfer, LINK(rowType), LINK(rowAddr));
  2316. args.append(*createTranslated(castBound));
  2317. }
  2318. OwnedHqlExpr call = translator.bindFunctionCall(addAggregateRowId, args);
  2319. translator.buildStmt(transformCtx, call);
  2320. }
  2321. void SourceBuilder::gatherVirtualFields(bool ignoreVirtuals, bool ensureSerialized)
  2322. {
  2323. IHqlExpression * record = tableExpr->queryRecord();
  2324. fieldInfo.gatherVirtualFields(record, ignoreVirtuals, ensureSerialized);
  2325. if (fieldInfo.hasVirtuals())
  2326. physicalRecord.setown(fieldInfo.createPhysicalRecord());
  2327. else
  2328. physicalRecord.set(record);
  2329. if (newDiskReadMapping)
  2330. {
  2331. projectedRecord.set(tableExpr->queryRecord());
  2332. expectedRecord.setown(getSerializedForm(physicalRecord, diskAtom));
  2333. //MORE: HPCC-18469 Reduce projected to the fields that are actually required by the dataset, and will need to remap field references.
  2334. if (fieldInfo.hasVirtuals())
  2335. {
  2336. StringBuffer typeName;
  2337. unsigned recordTypeFlags = translator.buildRtlType(typeName, projectedRecord->queryType());
  2338. if (recordTypeFlags & RFTMcannotinterpret)
  2339. throwError(HQLERR_CannotInterpretRecord);
  2340. }
  2341. }
  2342. else
  2343. {
  2344. expectedRecord.set(physicalRecord);
  2345. projectedRecord.set(physicalRecord);
  2346. }
  2347. tableSelector.set(tableExpr->queryNormalizedSelector());
  2348. projectedSelector.set(tableSelector);
  2349. }
  2350. /*
  2351. interface ICompoundSourceSteppingMeta : extends ISteppingMeta
  2352. {
  2353. virtual ISteppingMeta * queryRawSteppingMeta() = 0;
  2354. virtual ISteppingMeta * queryProjectedSteppingMeta() = 0; // if null no projection takes place
  2355. virtual void mapOutputToIInput(void * originalRow, const void * projectedRow, unsigned firstField, unsigned numFields) = 0;
  2356. };
  2357. */
  2358. bool SourceBuilder::containsStepping(IHqlExpression * expr)
  2359. {
  2360. for (;;)
  2361. {
  2362. switch (expr->getOperator())
  2363. {
  2364. case no_stepped:
  2365. return true;
  2366. default:
  2367. {
  2368. if (expr->queryBody() == tableExpr)
  2369. return false;
  2370. unsigned numChildren = getNumChildTables(expr);
  2371. if (numChildren == 0)
  2372. return false;
  2373. assertex(numChildren == 1);
  2374. }
  2375. }
  2376. expr = expr->queryChild(0);
  2377. }
  2378. }
  2379. void SourceBuilder::gatherSteppingMeta(IHqlExpression * expr, SteppingFieldSelection & outputStepping, SteppingFieldSelection & rawStepping)
  2380. {
  2381. for (;;)
  2382. {
  2383. switch (expr->getOperator())
  2384. {
  2385. case no_newusertable:
  2386. case no_hqlproject:
  2387. if (rawStepping.exists())
  2388. {
  2389. rawStepping.expandTransform(expr);
  2390. }
  2391. else
  2392. {
  2393. gatherSteppingMeta(expr->queryChild(0), outputStepping, rawStepping);
  2394. outputStepping.clear();
  2395. return;
  2396. //throwError(HQLERR_CantProjectStepping);
  2397. //gatherSteppingMeta(expr->queryChild(0), outputStepping, rawStepping);
  2398. //apply projection to the output fields and inverse to the raw fields.
  2399. }
  2400. break;
  2401. case no_stepped:
  2402. {
  2403. outputStepping.setStepping(expr);
  2404. rawStepping.setStepping(expr);
  2405. break;
  2406. }
  2407. default:
  2408. {
  2409. if (expr->queryBody() == tableExpr)
  2410. return;
  2411. unsigned numChildren = getNumChildTables(expr);
  2412. if (numChildren == 0)
  2413. return;
  2414. assertex(numChildren == 1);
  2415. }
  2416. }
  2417. expr = expr->queryChild(0);
  2418. }
  2419. }
  2420. void SourceBuilder::gatherSteppingMeta(IHqlExpression * expr, SourceSteppingInfo & info)
  2421. {
  2422. if (!steppedExpr)
  2423. return;
  2424. gatherSteppingMeta(expr, info.outputStepping, info.rawSteppingProject);
  2425. if (info.rawSteppingProject.exists())
  2426. info.extractRaw();
  2427. }
  2428. //-----------------------------------------------------------------------------------------------
  2429. //-- Disk file processing
  2430. //-----------------------------------------------------------------------------------------------
  2431. class DiskReadBuilderBase : public SourceBuilder
  2432. {
  2433. public:
  2434. DiskReadBuilderBase(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  2435. : SourceBuilder(_translator, _tableExpr, _nameExpr), monitors(_tableExpr, _translator, 0, true, false)
  2436. {
  2437. fpos.setown(getFilepos(tableExpr, false));
  2438. lfpos.setown(getFilepos(tableExpr, true));
  2439. logicalFilenameMarker.setown(getFileLogicalName(tableExpr));
  2440. mode = tableExpr->queryChild(2);
  2441. modeOp = mode->getOperator();
  2442. includeFormatCrc = (modeOp != no_csv && modeOp != no_pipe);
  2443. }
  2444. virtual void buildMembers(IHqlExpression * expr);
  2445. virtual void buildTransformFpos(BuildCtx & transformCtx);
  2446. virtual void extractMonitors(IHqlExpression * ds, SharedHqlExpr & unkeyedFilter, HqlExprArray & conds);
  2447. protected:
  2448. virtual void buildFlagsMember(IHqlExpression * expr);
  2449. protected:
  2450. CppFilterExtractor monitors;
  2451. OwnedHqlExpr globalGuard;
  2452. IHqlExpression * mode;
  2453. node_operator modeOp;
  2454. bool includeFormatCrc;
  2455. };
  2456. void DiskReadBuilderBase::buildMembers(IHqlExpression * expr)
  2457. {
  2458. StringBuffer s;
  2459. //Process any KEYED() information
  2460. if (monitors.isKeyed())
  2461. {
  2462. MemberFunction func(translator, instance->startctx, "virtual void createSegmentMonitors(IIndexReadContext *irc) override");
  2463. monitors.buildSegments(func.ctx, "irc", true);
  2464. }
  2465. instance->addAttributeBool(WaIsKeyed, monitors.isKeyed());
  2466. //---- virtual unsigned getFlags()
  2467. instance->addAttributeBool(WaIsPreload, isPreloaded);
  2468. bool matched = translator.registerGlobalUsage(tableExpr->queryChild(0));
  2469. if (translator.getTargetClusterType() == RoxieCluster)
  2470. {
  2471. instance->addAttributeBool(WaIsFileOpt, tableExpr->hasAttribute(optAtom));
  2472. instance->addAttributeBool(WaIsGlobalSpill, tableExpr->hasAttribute(jobTempAtom));
  2473. if (tableExpr->hasAttribute(jobTempAtom) && !matched)
  2474. throwUnexpected();
  2475. }
  2476. if (!matched && expr->hasAttribute(_spill_Atom))
  2477. {
  2478. StringBuffer spillName;
  2479. getExprECL(tableExpr->queryChild(0), spillName);
  2480. if (translator.queryOptions().allowThroughSpill)
  2481. throwError1(HQLERR_ReadSpillBeforeWriteFix, spillName.str());
  2482. else
  2483. throwError1(HQLERR_ReadSpillBeforeWrite, spillName.str());
  2484. }
  2485. //---- virtual bool canMatchAny() { return <value>; } ----
  2486. LinkedHqlExpr guard = globalGuard.get();
  2487. extendConditionOwn(guard, no_and, LINK(monitors.queryGlobalGuard()));
  2488. if (guard)
  2489. translator.doBuildBoolFunction(instance->startctx, "canMatchAny", guard);
  2490. translator.buildEncryptHelper(instance->startctx, tableExpr->queryAttribute(encryptAtom));
  2491. //---- preloadSize is passed via the xgmml, not via a member
  2492. if (preloadSize)
  2493. {
  2494. instance->addAttributeInt(WaSizePreload, preloadSize->queryValue()->getIntValue());
  2495. }
  2496. if (includeFormatCrc)
  2497. {
  2498. //Spill files can still have virtual attributes in their physical records => remove them.
  2499. OwnedHqlExpr noVirtualRecord = removeVirtualAttributes(expectedRecord);
  2500. translator.buildFormatCrcFunction(instance->classctx, "getDiskFormatCrc", noVirtualRecord);
  2501. if (newDiskReadMapping)
  2502. translator.buildFormatCrcFunction(instance->classctx, "getProjectedFormatCrc", projectedRecord);
  2503. else
  2504. translator.buildFormatCrcFunction(instance->classctx, "getProjectedFormatCrc", noVirtualRecord);
  2505. }
  2506. translator.buildMetaMember(instance->classctx, expectedRecord, isGrouped(tableExpr), "queryDiskRecordSize");
  2507. if (activityKind != TAKpiperead)
  2508. translator.buildMetaMember(instance->classctx, projectedRecord, isGrouped(tableExpr), "queryProjectedDiskRecordSize");
  2509. buildLimits(instance->startctx, expr, instance->activityId);
  2510. buildKeyedLimitHelper(expr);
  2511. //Note the helper base class contains code like the following
  2512. //IThorDiskCallback * fpp;");
  2513. //virtual void setCallback(IThorDiskCallback * _tc) { fpp = _tc; }");
  2514. }
  2515. void DiskReadBuilderBase::buildFlagsMember(IHqlExpression * expr)
  2516. {
  2517. StringBuffer flags;
  2518. if (tableExpr->hasAttribute(_spill_Atom)) flags.append("|TDXtemporary");
  2519. if (tableExpr->hasAttribute(jobTempAtom)) flags.append("|TDXjobtemp");
  2520. if (tableExpr->hasAttribute(groupedAtom)) flags.append("|TDXgrouped");
  2521. if (tableExpr->hasAttribute(__compressed__Atom)) flags.append("|TDXcompress");
  2522. if (tableExpr->hasAttribute(unsortedAtom)) flags.append("|TDRunsorted");
  2523. if (tableExpr->hasAttribute(optAtom)) flags.append("|TDRoptional");
  2524. if (tableExpr->hasAttribute(_workflowPersist_Atom)) flags.append("|TDXupdateaccessed");
  2525. if (isPreloaded) flags.append("|TDRpreload");
  2526. if (monitors.isKeyed()) flags.append("|TDRkeyed");
  2527. if (limitExpr)
  2528. {
  2529. if (limitExpr->hasAttribute(onFailAtom))
  2530. flags.append("|TDRlimitcreates");
  2531. else if (limitExpr->hasAttribute(skipAtom))
  2532. flags.append("|TDRlimitskips");
  2533. }
  2534. if (keyedLimitExpr)
  2535. {
  2536. if (keyedLimitExpr->hasAttribute(onFailAtom))
  2537. flags.append("|TDRkeyedlimitcreates|TDRcountkeyedlimit"); // is count correct?
  2538. else if (keyedLimitExpr->hasAttribute(skipAtom))
  2539. flags.append("|TDRkeyedlimitskips|TDRcountkeyedlimit");
  2540. else if (keyedLimitExpr->hasAttribute(countAtom))
  2541. flags.append("|TDRcountkeyedlimit");
  2542. }
  2543. if (onlyExistsAggreate) flags.append("|TDRaggregateexists");
  2544. if (monitorsForGrouping) flags.append("|TDRgroupmonitors");
  2545. if (!nameExpr->isConstant()) flags.append("|TDXvarfilename");
  2546. if (translator.hasDynamicFilename(tableExpr)) flags.append("|TDXdynamicfilename");
  2547. if (isUnfilteredCount) flags.append("|TDRunfilteredcount");
  2548. if (isVirtualLogicalFilenameUsed) flags.append("|TDRfilenamecallback");
  2549. if (requiresOrderedMerge) flags.append("|TDRorderedmerge");
  2550. if (flags.length())
  2551. translator.doBuildUnsignedFunction(instance->classctx, "getFlags", flags.str()+1);
  2552. }
  2553. void DiskReadBuilderBase::buildTransformFpos(BuildCtx & transformCtx)
  2554. {
  2555. if (modeOp == no_csv)
  2556. associateFilePositions(transformCtx, "fpp", "dataSrc[0]");
  2557. else
  2558. associateFilePositions(transformCtx, "fpp", "left");
  2559. }
  2560. void DiskReadBuilderBase::extractMonitors(IHqlExpression * ds, SharedHqlExpr & unkeyedFilter, HqlExprArray & conds)
  2561. {
  2562. HqlExprAttr mode = tableExpr->queryChild(2);
  2563. //KEYED filters can only currently be implemented for binary files - not csv, xml or pipe....
  2564. if (queryTableMode(tableExpr) == no_flat)
  2565. {
  2566. OwnedHqlExpr implicitFilter;
  2567. ForEachItemIn(i, conds)
  2568. {
  2569. IHqlExpression * filter = &conds.item(i);
  2570. if (isSourceInvariant(ds, filter)) // more actually isSourceInvariant.
  2571. extendConditionOwn(globalGuard, no_and, LINK(filter));
  2572. else
  2573. {
  2574. node_operator op = filter->getOperator();
  2575. switch (op)
  2576. {
  2577. case no_assertkeyed:
  2578. case no_assertwild:
  2579. {
  2580. //MORE: This needs to test that the fields are at fixed offsets, fixed length, and collatable.
  2581. OwnedHqlExpr extraFilter;
  2582. monitors.extractFilters(filter, extraFilter);
  2583. //NB: Even if it is keyed then (part of) the test condition might be duplicated.
  2584. appendFilter(unkeyedFilter, extraFilter);
  2585. break;
  2586. }
  2587. default:
  2588. // Add this condition to the catchall filter
  2589. appendFilter(implicitFilter, filter);
  2590. break;
  2591. }
  2592. }
  2593. }
  2594. if (implicitFilter)
  2595. {
  2596. if (translator.queryOptions().implicitKeyedDiskFilter && !monitors.isKeyed())
  2597. {
  2598. OwnedHqlExpr extraFilter;
  2599. monitors.extractFilters(implicitFilter.get(), extraFilter);
  2600. appendFilter(unkeyedFilter, extraFilter);
  2601. }
  2602. else
  2603. appendFilter(unkeyedFilter, implicitFilter);
  2604. }
  2605. }
  2606. else
  2607. SourceBuilder::extractMonitors(ds, unkeyedFilter, conds);
  2608. }
  2609. //---------------------------------------------------------------------------
  2610. class DiskReadBuilder : public DiskReadBuilderBase
  2611. {
  2612. public:
  2613. DiskReadBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  2614. : DiskReadBuilderBase(_translator, _tableExpr, _nameExpr)
  2615. {
  2616. extractCanMatch = (modeOp == no_thor) || (modeOp == no_flat);
  2617. }
  2618. protected:
  2619. virtual void buildTransform(IHqlExpression * expr) override;
  2620. virtual void buildMembers(IHqlExpression * expr) override;
  2621. virtual void analyseGraph(IHqlExpression * expr) override;
  2622. };
  2623. void DiskReadBuilder::analyseGraph(IHqlExpression * expr)
  2624. {
  2625. DiskReadBuilderBase::analyseGraph(expr);
  2626. if (newDiskReadMapping && extractCanMatch && firstTransformer)
  2627. {
  2628. //Calculate the minimum set of fields required by any post-filters and projects.
  2629. projectedRecord.setown(getMinimumInputRecord(translator, firstTransformer));
  2630. if (projectedRecord != firstTransformer->queryChild(0)->queryRecord())
  2631. {
  2632. OwnedHqlExpr selSeq = createUniqueSelectorSequence();
  2633. projectedSelector.setown(createSelector(no_left, projectedRecord, selSeq));
  2634. //Check if projecting the input fields to the minimum now means the transform can be removed.
  2635. if ((firstTransformer == lastTransformer) && (projectedRecord == firstTransformer->queryRecord()))
  2636. {
  2637. if (isSimpleProject(firstTransformer))
  2638. needToCallTransform = false;
  2639. }
  2640. }
  2641. }
  2642. }
  2643. void DiskReadBuilder::buildMembers(IHqlExpression * expr)
  2644. {
  2645. if (modeOp == no_csv)
  2646. buildFilenameMember();
  2647. else if (modeOp != no_pipe)
  2648. buildReadMembers(expr);
  2649. DiskReadBuilderBase::buildMembers(expr);
  2650. //---- virtual const char * getPipeProgram() { return "grep"; } ----
  2651. if (modeOp==no_pipe)
  2652. {
  2653. if (expr->hasAttribute(_disallowed_Atom))
  2654. throwError(HQLERR_PipeNotAllowed);
  2655. {
  2656. MemberFunction func(translator, instance->startctx, "virtual const char * getPipeProgram() override");
  2657. translator.buildReturn(func.ctx, mode->queryChild(0), unknownVarStringType);
  2658. }
  2659. IHqlExpression * csvFromPipe = tableExpr->queryAttribute(csvAtom);
  2660. IHqlExpression * xmlFromPipe = tableExpr->queryAttribute(xmlAtom);
  2661. bool usesContents = false;
  2662. if (csvFromPipe)
  2663. {
  2664. if (isValidCsvRecord(tableExpr->queryRecord()))
  2665. {
  2666. StringBuffer csvInstanceName;
  2667. translator.buildCsvReadTransformer(tableExpr, csvInstanceName, csvFromPipe);
  2668. StringBuffer s;
  2669. s.append("virtual ICsvToRowTransformer * queryCsvTransformer() override { return &").append(csvInstanceName).append("; }");
  2670. instance->classctx.addQuoted(s);
  2671. }
  2672. else
  2673. {
  2674. throwUnexpected(); // should be caught earlier
  2675. }
  2676. }
  2677. else if (xmlFromPipe)
  2678. {
  2679. translator.doBuildXmlReadMember(*instance, expr, "queryXmlTransformer", usesContents);
  2680. translator.doBuildVarStringFunction(instance->classctx, "getXmlIteratorPath", queryAttributeChild(xmlFromPipe, rowAtom, 0));
  2681. }
  2682. StringBuffer flags;
  2683. if (tableExpr->hasAttribute(groupAtom)) // not supported in parser?
  2684. flags.append("|TPFgroupeachrow");
  2685. if (tableExpr->hasAttribute(optAtom)) // not supported in parser?
  2686. flags.append("|TPFnofail");
  2687. if (csvFromPipe)
  2688. flags.append("|TPFreadcsvfrompipe");
  2689. if (xmlFromPipe)
  2690. flags.append("|TPFreadxmlfrompipe");
  2691. if (usesContents)
  2692. flags.append("|TPFreadusexmlcontents");
  2693. if (flags.length())
  2694. translator.doBuildUnsignedFunction(instance->classctx, "getPipeFlags", flags.str()+1);
  2695. }
  2696. }
  2697. void DiskReadBuilder::buildTransform(IHqlExpression * expr)
  2698. {
  2699. if (modeOp == no_pipe)
  2700. {
  2701. assertex(!(needToCallTransform || transformCanFilter));
  2702. return;
  2703. }
  2704. if (recordRequiresSerialization(tableExpr->queryRecord(), diskAtom))
  2705. {
  2706. //Sanity check to ensure that the projected row is only in the in memory format if no transform needs to be called.
  2707. if (needToCallTransform || transformCanFilter)
  2708. throwUnexpectedX("Projected dataset should have been serialized");
  2709. //Base implementation for a disk read throws an exception if it is called.
  2710. return;
  2711. }
  2712. if (modeOp == no_csv)
  2713. {
  2714. translator.buildCsvParameters(instance->nestedctx, mode, NULL, true);
  2715. {
  2716. MemberFunction func(translator, instance->startctx, "virtual size32_t transform(ARowBuilder & crSelf, unsigned * lenSrc, const char * * dataSrc) override");
  2717. translator.ensureRowAllocated(func.ctx, "crSelf");
  2718. //associateVirtualCallbacks(*this, func.ctx, tableExpr);
  2719. buildTransformBody(func.ctx, expr, true, false, true);
  2720. }
  2721. rootSelfRow = NULL;
  2722. unsigned maxColumns = countTotalFields(tableExpr->queryRecord(), false);
  2723. translator.doBuildUnsignedFunction(instance->classctx, "getMaxColumns", maxColumns);
  2724. return;
  2725. }
  2726. MemberFunction func(translator, instance->startctx);
  2727. if ((instance->kind == TAKdiskread) || (instance->kind == TAKspillread))
  2728. func.start("virtual size32_t transform(ARowBuilder & crSelf, const void * _left) override");
  2729. else
  2730. func.start("virtual size32_t transform(ARowBuilder & crSelf, const void * _left, IFilePositionProvider * fpp) override");
  2731. translator.ensureRowAllocated(func.ctx, "crSelf");
  2732. func.ctx.addQuotedLiteral("unsigned char * left = (unsigned char *)_left;");
  2733. buildTransformBody(func.ctx, expr, true, false, true);
  2734. }
  2735. //---------------------------------------------------------------------------
  2736. ABoundActivity * HqlCppTranslator::doBuildActivityDiskRead(BuildCtx & ctx, IHqlExpression * expr)
  2737. {
  2738. // assertex(!isGroupedActivity(expr));
  2739. IHqlExpression *tableExpr = queryPhysicalRootTable(expr);
  2740. if (!tableExpr)
  2741. return buildCachedActivity(ctx, expr->queryChild(0)); // Somehow a null appeared.
  2742. HqlExprAttr mode = tableExpr->queryChild(2);
  2743. node_operator modeOp = mode->getOperator();
  2744. bool isPiped = modeOp==no_pipe;
  2745. DiskReadBuilder info(*this, tableExpr, tableExpr->queryChild(0));
  2746. info.gatherVirtualFields(tableExpr->hasAttribute(_noVirtual_Atom) || isPiped, needToSerializeRecord(modeOp));
  2747. unsigned optFlags = (options.foldOptimized ? HOOfold : 0);
  2748. if (info.newDiskReadMapping && (modeOp != no_csv) && (modeOp != no_xml) && (modeOp != no_pipe))
  2749. {
  2750. //The projected disk information (which is passed to the transform) uses the in memory format IFF
  2751. // - The disk read is a trivial slimming transform (so no transform needs calling on the projected disk format.
  2752. // - It is used for all disk reads since directly transforming is always at least as efficient as going via
  2753. // the serialized form.
  2754. // Otherwise the table is converted to the serialized format.
  2755. const bool forceAllProjectedSerialized = options.forceAllProjectedDiskSerialized;
  2756. //Reading from a spill file uses the in-memory format to optimize on-demand spilling.
  2757. bool optimizeInMemorySpill = targetThor();
  2758. bool useInMemoryFormat = optimizeInMemorySpill && isSimpleProjectingDiskRead(expr);
  2759. if (forceAllProjectedSerialized || !useInMemoryFormat)
  2760. {
  2761. //else if the the table isn't serialized, then map to a serialized table, and then project to the real format
  2762. if (recordRequiresSerialization(tableExpr->queryRecord(), diskAtom))
  2763. {
  2764. OwnedHqlExpr transformed = buildTableFromSerialized(expr);
  2765. //Need to wrap a possible no_usertable, otherwise the localisation can go wrong.
  2766. if (expr->getOperator() == no_table)
  2767. transformed.setown(createDataset(no_compound_diskread, LINK(transformed)));
  2768. OwnedHqlExpr optimized = optimizeHqlExpression(queryErrorProcessor(), transformed, optFlags);
  2769. traceExpression("after disk optimize", optimized);
  2770. return doBuildActivityDiskRead(ctx, optimized);
  2771. }
  2772. }
  2773. //Otherwise the dataset is in the correct format
  2774. }
  2775. else
  2776. {
  2777. if (info.recordHasVirtuals() || info.fieldInfo.requiresDeserialize)
  2778. {
  2779. OwnedHqlExpr transformed = buildTableWithoutVirtuals(info.fieldInfo, expr);
  2780. //Need to wrap a possible no_usertable, otherwise the localisation can go wrong.
  2781. if (expr->getOperator() == no_table)
  2782. transformed.setown(createDataset(no_compound_diskread, LINK(transformed)));
  2783. OwnedHqlExpr optimized = optimizeHqlExpression(queryErrorProcessor(), transformed, optFlags);
  2784. traceExpression("after disk optimize", optimized);
  2785. return doBuildActivityDiskRead(ctx, optimized);
  2786. }
  2787. }
  2788. OwnedHqlExpr optimized;
  2789. if (expr->getOperator() == no_table)
  2790. optimized.set(expr);
  2791. else
  2792. optimized.setown(optimizeHqlExpression(queryErrorProcessor(), expr, optFlags));
  2793. if (optimized != expr)
  2794. return buildActivity(ctx, optimized, false);
  2795. if (isPiped)
  2796. return info.buildActivity(ctx, expr, TAKpiperead, "PipeRead", NULL);
  2797. ensureDiskAccessAllowed(tableExpr);
  2798. if (modeOp == no_csv)
  2799. return info.buildActivity(ctx, expr, TAKcsvread, "CsvRead", NULL);
  2800. return info.buildActivity(ctx, expr, TAKdiskread, "DiskRead", NULL);
  2801. }
  2802. //---------------------------------------------------------------------------
  2803. class DiskNormalizeBuilder : public DiskReadBuilderBase
  2804. {
  2805. public:
  2806. DiskNormalizeBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  2807. : DiskReadBuilderBase(_translator, _tableExpr, _nameExpr)
  2808. {
  2809. }
  2810. virtual void buildTransform(IHqlExpression * expr);
  2811. virtual void buildMembers(IHqlExpression * expr);
  2812. protected:
  2813. virtual void analyseGraph(IHqlExpression * expr);
  2814. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  2815. {
  2816. doBuildNormalizeIterators(ctx, expr, false);
  2817. }
  2818. };
  2819. void DiskNormalizeBuilder::analyseGraph(IHqlExpression * expr)
  2820. {
  2821. DiskReadBuilderBase::analyseGraph(expr);
  2822. needDefaultTransform = (expr->queryNormalizedSelector()->getOperator() == no_select);
  2823. }
  2824. void DiskNormalizeBuilder::buildMembers(IHqlExpression * expr)
  2825. {
  2826. buildFilenameMember();
  2827. DiskReadBuilderBase::buildMembers(expr);
  2828. buildNormalizeHelpers(expr);
  2829. }
  2830. void DiskNormalizeBuilder::buildTransform(IHqlExpression * expr)
  2831. {
  2832. globaliterctx.setown(new BuildCtx(instance->startctx));
  2833. globaliterctx->addGroup();
  2834. MemberFunction func(translator, instance->startctx, "virtual size32_t transform(ARowBuilder & crSelf) override");
  2835. translator.ensureRowAllocated(func.ctx, "crSelf");
  2836. buildTransformBody(func.ctx, expr, true, false, false);
  2837. }
  2838. //---------------------------------------------------------------------------
  2839. ABoundActivity * HqlCppTranslator::doBuildActivityDiskNormalize(BuildCtx & ctx, IHqlExpression * expr)
  2840. {
  2841. assertex(!isGroupedActivity(expr));
  2842. IHqlExpression *tableExpr = queryPhysicalRootTable(expr);
  2843. ensureDiskAccessAllowed(tableExpr);
  2844. HqlExprAttr mode = tableExpr->queryChild(2);
  2845. assertex(mode->getOperator()!=no_pipe);
  2846. DiskNormalizeBuilder info(*this, tableExpr, tableExpr->queryChild(0));
  2847. info.gatherVirtualFields(tableExpr->hasAttribute(_noVirtual_Atom), needToSerializeRecord(mode));
  2848. LinkedHqlExpr transformed = expr;
  2849. if (info.recordHasVirtualsOrDeserialize())
  2850. transformed.setown(buildTableWithoutVirtuals(info.fieldInfo, expr));
  2851. unsigned optFlags = (options.foldOptimized ? HOOfold : 0);
  2852. OwnedHqlExpr optimized = optimizeHqlExpression(queryErrorProcessor(), transformed, optFlags);
  2853. if (optimized != expr)
  2854. return buildActivity(ctx, optimized, false);
  2855. return info.buildActivity(ctx, expr, TAKdisknormalize, "DiskNormalize", NULL);
  2856. }
  2857. //---------------------------------------------------------------------------
  2858. class DiskAggregateBuilder : public DiskReadBuilderBase
  2859. {
  2860. public:
  2861. DiskAggregateBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  2862. : DiskReadBuilderBase(_translator, _tableExpr, _nameExpr)
  2863. {
  2864. failedFilterValue.clear();
  2865. }
  2866. virtual void buildTransform(IHqlExpression * expr);
  2867. virtual void buildMembers(IHqlExpression * expr);
  2868. protected:
  2869. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  2870. {
  2871. doBuildAggregateSelectIterator(ctx, expr);
  2872. }
  2873. virtual void analyseGraph(IHqlExpression * expr)
  2874. {
  2875. DiskReadBuilderBase::analyseGraph(expr);
  2876. returnIfFilterFails = !isNormalize;
  2877. }
  2878. };
  2879. void DiskAggregateBuilder::buildMembers(IHqlExpression * expr)
  2880. {
  2881. StringBuffer s;
  2882. buildFilenameMember();
  2883. DiskReadBuilderBase::buildMembers(expr);
  2884. buildAggregateHelpers(expr);
  2885. //virtual void processRow(void * self, const void * src) = 0;
  2886. {
  2887. BuildCtx rowctx(instance->startctx);
  2888. rowctx.addQuotedFunction("virtual void processRow(ARowBuilder & crSelf, const void * src) override");
  2889. rowctx.addQuotedLiteral("doProcessRow(crSelf, (const byte *)src);");
  2890. }
  2891. //virtual void processRows(void * self, size32_t srcLen, const void * src) = 0;
  2892. {
  2893. MemberFunction func(translator, instance->startctx, "virtual void processRows(ARowBuilder & crSelf, size32_t srcLen, const void * _left) override");
  2894. func.ctx.addQuotedLiteral("unsigned char * left = (unsigned char *)_left;");
  2895. OwnedHqlExpr ds = createVariable("left", makeReferenceModifier(tableExpr->getType()));
  2896. OwnedHqlExpr len = createVariable("srcLen", LINK(sizetType));
  2897. OwnedHqlExpr fullDs = createTranslated(ds, len);
  2898. Owned<IHqlCppDatasetCursor> iter = translator.createDatasetSelector(func.ctx, fullDs);
  2899. BoundRow * curRow = iter->buildIterateLoop(func.ctx, false);
  2900. s.clear().append("doProcessRow(crSelf, ");
  2901. translator.generateExprCpp(s, curRow->queryBound());
  2902. s.append(");");
  2903. func.ctx.addQuoted(s);
  2904. }
  2905. }
  2906. void DiskAggregateBuilder::buildTransform(IHqlExpression * expr)
  2907. {
  2908. MemberFunction func(translator, instance->startctx, "void doProcessRow(ARowBuilder & crSelf, const byte * left)");
  2909. translator.ensureRowAllocated(func.ctx, "crSelf");
  2910. buildTransformBody(func.ctx, expr, false, false, true);
  2911. }
  2912. //---------------------------------------------------------------------------
  2913. class DiskCountBuilder : public DiskReadBuilderBase
  2914. {
  2915. public:
  2916. DiskCountBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr, node_operator _aggOp)
  2917. : DiskReadBuilderBase(_translator, _tableExpr, _nameExpr)
  2918. {
  2919. aggOp = _aggOp;
  2920. isCompoundCount = true;
  2921. failedFilterValue.set(queryZero());
  2922. }
  2923. virtual void buildTransform(IHqlExpression * expr);
  2924. virtual void buildMembers(IHqlExpression * expr);
  2925. virtual bool isExplicitExists() { return (aggOp == no_existsgroup); }
  2926. protected:
  2927. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  2928. {
  2929. doBuildAggregateSelectIterator(ctx, expr);
  2930. }
  2931. virtual void analyseGraph(IHqlExpression * expr)
  2932. {
  2933. DiskReadBuilderBase::analyseGraph(expr);
  2934. returnIfFilterFails = !isNormalize;
  2935. if (aggOp == no_existsgroup)
  2936. choosenValue.setown(getSizetConstant(1));
  2937. }
  2938. protected:
  2939. node_operator aggOp;
  2940. };
  2941. void DiskCountBuilder::buildMembers(IHqlExpression * expr)
  2942. {
  2943. isUnfilteredCount = !(transformCanFilter||isNormalize);
  2944. buildFilenameMember();
  2945. DiskReadBuilderBase::buildMembers(expr);
  2946. buildCountHelpers(expr, true);
  2947. }
  2948. void DiskCountBuilder::buildTransform(IHqlExpression * expr)
  2949. {
  2950. if (transformCanFilter||isNormalize)
  2951. {
  2952. MemberFunction func(translator, instance->startctx, "size32_t valid(const byte * left)");
  2953. OwnedHqlExpr cnt;
  2954. if (isNormalize)
  2955. {
  2956. compoundCountVar.setown(func.ctx.getTempDeclare(sizetType, queryZero()));
  2957. cnt.set(compoundCountVar);
  2958. }
  2959. else
  2960. cnt.setown(getSizetConstant(1));
  2961. BuildCtx subctx(func.ctx);
  2962. buildTransformBody(subctx, expr, false, false, true);
  2963. func.ctx.addReturn(cnt);
  2964. }
  2965. }
  2966. //---------------------------------------------------------------------------
  2967. ABoundActivity * HqlCppTranslator::doBuildActivityDiskAggregate(BuildCtx & ctx, IHqlExpression * expr)
  2968. {
  2969. assertex(!isGroupedActivity(expr));
  2970. IHqlExpression *tableExpr = queryPhysicalRootTable(expr);
  2971. ensureDiskAccessAllowed(tableExpr);
  2972. HqlExprAttr mode = tableExpr->queryChild(2);
  2973. assertex(mode->getOperator()!=no_pipe);
  2974. DiskAggregateBuilder info(*this, tableExpr, tableExpr->queryChild(0));
  2975. info.gatherVirtualFields(tableExpr->hasAttribute(_noVirtual_Atom), needToSerializeRecord(mode));
  2976. LinkedHqlExpr transformed = expr;
  2977. if (info.recordHasVirtualsOrDeserialize())
  2978. transformed.setown(buildTableWithoutVirtuals(info.fieldInfo, expr));
  2979. transformed.setown(optimizeHqlExpression(queryErrorProcessor(), transformed, getSourceAggregateOptimizeFlags()));
  2980. if (transformed != expr)
  2981. return buildActivity(ctx, transformed, false);
  2982. node_operator aggOp = querySimpleAggregate(expr, true, false);
  2983. if (aggOp == no_countgroup || aggOp == no_existsgroup)
  2984. {
  2985. DiskCountBuilder info(*this, tableExpr, tableExpr->queryChild(0), aggOp);
  2986. info.gatherVirtualFields(tableExpr->hasAttribute(_noVirtual_Atom), needToSerializeRecord(mode));
  2987. return info.buildActivity(ctx, expr, TAKdiskcount, "DiskCount", NULL);
  2988. }
  2989. else
  2990. return info.buildActivity(ctx, expr, TAKdiskaggregate, "DiskAggregate", NULL);
  2991. }
  2992. //---------------------------------------------------------------------------
  2993. class DiskGroupAggregateBuilder : public DiskReadBuilderBase
  2994. {
  2995. public:
  2996. DiskGroupAggregateBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  2997. : DiskReadBuilderBase(_translator, _tableExpr, _nameExpr)
  2998. {
  2999. failedFilterValue.clear();
  3000. }
  3001. virtual void buildTransform(IHqlExpression * expr);
  3002. virtual void buildMembers(IHqlExpression * expr);
  3003. protected:
  3004. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  3005. {
  3006. doBuildAggregateSelectIterator(ctx, expr);
  3007. }
  3008. virtual void analyseGraph(IHqlExpression * expr)
  3009. {
  3010. DiskReadBuilderBase::analyseGraph(expr);
  3011. returnIfFilterFails = !isNormalize;
  3012. }
  3013. };
  3014. void DiskGroupAggregateBuilder::buildMembers(IHqlExpression * expr)
  3015. {
  3016. buildFilenameMember();
  3017. buildGroupingMonitors(expr, monitors);
  3018. DiskReadBuilderBase::buildMembers(expr);
  3019. buildGlobalGroupAggregateHelpers(expr);
  3020. }
  3021. void DiskGroupAggregateBuilder::buildTransform(IHqlExpression * expr)
  3022. {
  3023. MemberFunction func(translator, instance->startctx, "void doProcessRow(const byte * left, IHThorGroupAggregateCallback * callback)");
  3024. bool accessesCallback = containsOperator(expr, no_filepos) || containsOperator(expr, no_file_logicalname);
  3025. buildGroupAggregateTransformBody(func.ctx, expr, isNormalize || accessesCallback, true);
  3026. }
  3027. //---------------------------------------------------------------------------
  3028. ABoundActivity * HqlCppTranslator::doBuildActivityDiskGroupAggregate(BuildCtx & ctx, IHqlExpression * expr)
  3029. {
  3030. IHqlExpression *tableExpr = queryPhysicalRootTable(expr);
  3031. ensureDiskAccessAllowed(tableExpr);
  3032. HqlExprAttr mode = tableExpr->queryChild(2);
  3033. assertex(mode->getOperator()!=no_pipe);
  3034. DiskGroupAggregateBuilder info(*this, tableExpr, tableExpr->queryChild(0));
  3035. info.gatherVirtualFields(tableExpr->hasAttribute(_noVirtual_Atom), needToSerializeRecord(mode));
  3036. LinkedHqlExpr transformed = expr;
  3037. if (info.recordHasVirtualsOrDeserialize())
  3038. transformed.setown(buildTableWithoutVirtuals(info.fieldInfo, expr));
  3039. transformed.setown(optimizeHqlExpression(queryErrorProcessor(), transformed, getSourceAggregateOptimizeFlags()));
  3040. if (transformed != expr)
  3041. return buildActivity(ctx, transformed, false);
  3042. return info.buildActivity(ctx, expr, TAKdiskgroupaggregate, "DiskGroupAggregate", NULL);
  3043. }
  3044. //-----------------------------------------------------------------------------------------------
  3045. //-- Child dataset processing
  3046. //-----------------------------------------------------------------------------------------------
  3047. class ChildBuilderBase : public SourceBuilder
  3048. {
  3049. public:
  3050. ChildBuilderBase(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  3051. : SourceBuilder(_translator, _tableExpr, _nameExpr)
  3052. {
  3053. }
  3054. virtual void buildMembers(IHqlExpression * expr) {}
  3055. virtual void buildTransformFpos(BuildCtx & transformCtx) {}
  3056. };
  3057. class ChildNormalizeBuilder : public ChildBuilderBase
  3058. {
  3059. public:
  3060. ChildNormalizeBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  3061. : ChildBuilderBase(_translator, _tableExpr, _nameExpr)
  3062. {
  3063. }
  3064. virtual void buildTransform(IHqlExpression * expr);
  3065. virtual void buildMembers(IHqlExpression * expr);
  3066. protected:
  3067. virtual void analyseGraph(IHqlExpression * expr);
  3068. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  3069. {
  3070. doBuildNormalizeIterators(ctx, expr, true);
  3071. }
  3072. };
  3073. void ChildNormalizeBuilder::analyseGraph(IHqlExpression * expr)
  3074. {
  3075. ChildBuilderBase::analyseGraph(expr);
  3076. needDefaultTransform = (expr->queryNormalizedSelector()->getOperator() == no_select);
  3077. }
  3078. void ChildNormalizeBuilder::buildMembers(IHqlExpression * expr)
  3079. {
  3080. ChildBuilderBase::buildMembers(expr);
  3081. buildNormalizeHelpers(expr);
  3082. }
  3083. void ChildNormalizeBuilder::buildTransform(IHqlExpression * expr)
  3084. {
  3085. globaliterctx.setown(new BuildCtx(instance->startctx));
  3086. globaliterctx->addGroup();
  3087. MemberFunction func(translator, instance->startctx, "virtual size32_t transform(ARowBuilder & crSelf) override");
  3088. translator.ensureRowAllocated(func.ctx, "crSelf");
  3089. buildTransformBody(func.ctx, expr, true, false, false);
  3090. }
  3091. //---------------------------------------------------------------------------
  3092. ABoundActivity * HqlCppTranslator::doBuildActivityChildNormalize(BuildCtx & ctx, IHqlExpression * expr)
  3093. {
  3094. ChildNormalizeBuilder info(*this, NULL, NULL);
  3095. OwnedHqlExpr optimized = optimizeHqlExpression(queryErrorProcessor(), expr, HOOfold);
  3096. if (optimized != expr)
  3097. return buildActivity(ctx, optimized, false);
  3098. return info.buildActivity(ctx, expr, TAKchildnormalize, "ChildNormalize", NULL);
  3099. }
  3100. //---------------------------------------------------------------------------
  3101. class ChildAggregateBuilder : public ChildBuilderBase
  3102. {
  3103. public:
  3104. ChildAggregateBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  3105. : ChildBuilderBase(_translator, _tableExpr, _nameExpr)
  3106. {
  3107. failedFilterValue.clear();
  3108. }
  3109. virtual void buildTransform(IHqlExpression * expr);
  3110. virtual void buildMembers(IHqlExpression * expr);
  3111. protected:
  3112. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  3113. {
  3114. doBuildAggregateSelectIterator(ctx, expr);
  3115. }
  3116. virtual void analyseGraph(IHqlExpression * expr)
  3117. {
  3118. ChildBuilderBase::analyseGraph(expr);
  3119. returnIfFilterFails = false;
  3120. }
  3121. };
  3122. void ChildAggregateBuilder::buildMembers(IHqlExpression * expr)
  3123. {
  3124. ChildBuilderBase::buildMembers(expr);
  3125. buildAggregateHelpers(expr);
  3126. }
  3127. void ChildAggregateBuilder::buildTransform(IHqlExpression * expr)
  3128. {
  3129. MemberFunction func(translator, instance->startctx, "virtual void processRows(ARowBuilder & crSelf) override");
  3130. translator.ensureRowAllocated(func.ctx, "crSelf");
  3131. buildTransformBody(func.ctx, expr, false, false, false);
  3132. }
  3133. //---------------------------------------------------------------------------
  3134. ABoundActivity * HqlCppTranslator::doBuildActivityChildAggregate(BuildCtx & ctx, IHqlExpression * expr)
  3135. {
  3136. ChildAggregateBuilder info(*this, NULL, NULL);
  3137. OwnedHqlExpr transformed = optimizeHqlExpression(queryErrorProcessor(), expr, getSourceAggregateOptimizeFlags());
  3138. if (transformed != expr)
  3139. return buildActivity(ctx, transformed, false);
  3140. return info.buildActivity(ctx, expr, TAKchildaggregate, "ChildAggregate", NULL);
  3141. }
  3142. //---------------------------------------------------------------------------
  3143. class ChildGroupAggregateBuilder : public ChildBuilderBase
  3144. {
  3145. public:
  3146. ChildGroupAggregateBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  3147. : ChildBuilderBase(_translator, _tableExpr, _nameExpr)
  3148. {
  3149. failedFilterValue.clear();
  3150. }
  3151. virtual void buildTransform(IHqlExpression * expr);
  3152. virtual void buildMembers(IHqlExpression * expr);
  3153. protected:
  3154. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  3155. {
  3156. doBuildAggregateSelectIterator(ctx, expr);
  3157. }
  3158. virtual void analyseGraph(IHqlExpression * expr)
  3159. {
  3160. ChildBuilderBase::analyseGraph(expr);
  3161. returnIfFilterFails = false;
  3162. }
  3163. };
  3164. void ChildGroupAggregateBuilder::buildMembers(IHqlExpression * expr)
  3165. {
  3166. ChildBuilderBase::buildMembers(expr);
  3167. IHqlExpression * aggregate = expr->queryChild(0);
  3168. assertex(aggregate->getOperator() == no_newaggregate);
  3169. StringBuffer s;
  3170. //virtual size32_t clearAggregate(void * self) = 0;
  3171. translator.doBuildAggregateClearFunc(instance->startctx, aggregate);
  3172. //virtual size32_t mergeAggregate(ARowBuilder & crSelf, const void * src) - never actually called.
  3173. instance->startctx.addQuotedLiteral("virtual size32_t mergeAggregate(ARowBuilder & crSelf, const void * src) override { return 0; }");
  3174. }
  3175. void ChildGroupAggregateBuilder::buildTransform(IHqlExpression * expr)
  3176. {
  3177. MemberFunction func(translator, instance->startctx, "void processRows(IHThorGroupAggregateCallback * callback)");
  3178. buildGroupAggregateTransformBody(func.ctx, expr, true, false);
  3179. }
  3180. //---------------------------------------------------------------------------
  3181. ABoundActivity * HqlCppTranslator::doBuildActivityChildGroupAggregate(BuildCtx & ctx, IHqlExpression * expr)
  3182. {
  3183. ChildGroupAggregateBuilder info(*this, NULL, NULL);
  3184. OwnedHqlExpr transformed = optimizeHqlExpression(queryErrorProcessor(), expr, getSourceAggregateOptimizeFlags());
  3185. if (transformed != expr)
  3186. return buildActivity(ctx, transformed, false);
  3187. return info.buildActivity(ctx, expr, TAKchildgroupaggregate, "ChildGroupAggregate", NULL);
  3188. }
  3189. //---------------------------------------------------------------------------
  3190. class ChildThroughNormalizeBuilder : public ChildBuilderBase
  3191. {
  3192. public:
  3193. ChildThroughNormalizeBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  3194. : ChildBuilderBase(_translator, _tableExpr, _nameExpr)
  3195. {
  3196. }
  3197. virtual void buildTransform(IHqlExpression * expr);
  3198. virtual void buildMembers(IHqlExpression * expr);
  3199. protected:
  3200. virtual void analyseGraph(IHqlExpression * expr);
  3201. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  3202. {
  3203. translator.bindTableCursor(ctx, expr, "left");
  3204. doBuildNormalizeIterators(ctx, expr, false);
  3205. }
  3206. };
  3207. void ChildThroughNormalizeBuilder::analyseGraph(IHqlExpression * expr)
  3208. {
  3209. ChildBuilderBase::analyseGraph(expr);
  3210. needDefaultTransform = (expr->queryNormalizedSelector()->getOperator() == no_select);
  3211. }
  3212. void ChildThroughNormalizeBuilder::buildMembers(IHqlExpression * expr)
  3213. {
  3214. ChildBuilderBase::buildMembers(expr);
  3215. buildNormalizeHelpers(expr);
  3216. }
  3217. void ChildThroughNormalizeBuilder::buildTransform(IHqlExpression * expr)
  3218. {
  3219. globaliterctx.setown(new BuildCtx(instance->startctx));
  3220. globaliterctx->addGroup();
  3221. MemberFunction func(translator, instance->startctx, "virtual size32_t transform(ARowBuilder & crSelf) override");
  3222. translator.ensureRowAllocated(func.ctx, "crSelf");
  3223. buildTransformBody(func.ctx, expr, true, false, false);
  3224. }
  3225. ABoundActivity * HqlCppTranslator::doBuildActivityCompoundSelectNew(BuildCtx & ctx, IHqlExpression * expr)
  3226. {
  3227. OwnedHqlExpr optimized = optimizeHqlExpression(queryErrorProcessor(), expr, HOOfold);
  3228. if (optimized->getOperator() == no_null)
  3229. return buildCachedActivity(ctx, optimized);
  3230. IHqlExpression * select = queryRoot(optimized);
  3231. if (!(select && select->getOperator() == no_select && select->hasAttribute(newAtom)))
  3232. {
  3233. if (optimized->getOperator() == no_compound_selectnew)
  3234. return buildCachedActivity(ctx, optimized->queryChild(0));
  3235. return buildCachedActivity(ctx, optimized);
  3236. }
  3237. IHqlExpression * ds = select->queryChild(0);
  3238. Owned<ABoundActivity> childActivity = buildCachedActivity(ctx, ds);
  3239. OwnedHqlExpr fakeDataset = createDataset(no_anon, LINK(ds->queryRecord()));
  3240. OwnedHqlExpr fakeSelect = createNewSelectExpr(LINK(fakeDataset), LINK(select->queryChild(1)));
  3241. OwnedHqlExpr activityExpr = replaceExpression(optimized, select, fakeSelect);
  3242. ChildThroughNormalizeBuilder info(*this, fakeDataset, NULL);
  3243. info.gatherVirtualFields(true, true); // ,false?
  3244. return info.buildActivity(ctx, activityExpr, TAKchildthroughnormalize, "ChildThroughNormalize", childActivity);
  3245. }
  3246. //---------------------------------------------------------------------------
  3247. static ABoundActivity * buildNullIndexActivity(HqlCppTranslator & translator, BuildCtx & ctx, IHqlExpression * expr)
  3248. {
  3249. while (isCompoundSource(expr))
  3250. expr = expr->queryChild(0);
  3251. return translator.buildCachedActivity(ctx, expr);
  3252. }
  3253. class IndexReadBuilderBase : public SourceBuilder
  3254. {
  3255. friend class MonitorRemovalTransformer;
  3256. public:
  3257. IndexReadBuilderBase(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  3258. : SourceBuilder(_translator, _tableExpr, _nameExpr), monitors(_tableExpr, _translator, -(int)numPayloadFields(_tableExpr), false, false)
  3259. {
  3260. }
  3261. virtual void buildMembers(IHqlExpression * expr);
  3262. virtual void extractMonitors(IHqlExpression * ds, SharedHqlExpr & unkeyedFilter, HqlExprArray & conds);
  3263. protected:
  3264. virtual void buildFlagsMember(IHqlExpression * expr);
  3265. virtual void buildTransformFpos(BuildCtx & transformCtx)
  3266. {
  3267. associateFilePositions(transformCtx, "fpp", "left");
  3268. }
  3269. IHqlExpression * removeMonitors(IHqlExpression * expr);
  3270. protected:
  3271. CppFilterExtractor monitors;
  3272. SourceSteppingInfo steppingInfo;
  3273. };
  3274. void IndexReadBuilderBase::buildMembers(IHqlExpression * expr)
  3275. {
  3276. //---- virtual void createSegmentMonitors(struct IIndexReadContext *) { ... } ----
  3277. {
  3278. MemberFunction func(translator, instance->startctx, "virtual void createSegmentMonitors(IIndexReadContext *irc) override");
  3279. monitors.buildSegments(func.ctx, "irc", false);
  3280. }
  3281. buildLimits(instance->startctx, expr, instance->activityId);
  3282. if (!limitExpr && !keyedLimitExpr && !choosenValue && (instance->kind == TAKindexread || instance->kind == TAKindexnormalize) && !steppedExpr)
  3283. {
  3284. unsigned implicitLimit = translator.getDefaultImplicitIndexReadLimit();
  3285. if (translator.checkIndexReadLimit())
  3286. {
  3287. StringBuffer keyname;
  3288. if (implicitLimit)
  3289. translator.WARNINGAT2(CategoryLimit, queryLocation(expr), HQLWRN_ImplicitReadAddLimit, implicitLimit, monitors.queryKeyName(keyname));
  3290. else
  3291. translator.WARNINGAT1(CategoryLimit, queryLocation(expr), HQLWRN_ImplicitReadLimit, monitors.queryKeyName(keyname));
  3292. }
  3293. if (implicitLimit)
  3294. {
  3295. OwnedHqlExpr limit = getSizetConstant(implicitLimit);
  3296. translator.buildLimitHelpers(instance->startctx, limit, NULL, false, nameExpr, instance->activityId);
  3297. }
  3298. }
  3299. instance->addAttributeBool(WaIsPreload, isPreloaded);
  3300. if (translator.getTargetClusterType() == RoxieCluster)
  3301. instance->addAttributeBool(WaIsIndexOpt, tableExpr->hasAttribute(optAtom));
  3302. if (monitors.queryGlobalGuard())
  3303. translator.doBuildBoolFunction(instance->startctx, "canMatchAny", monitors.queryGlobalGuard());
  3304. buildKeyedLimitHelper(expr);
  3305. translator.buildFormatCrcFunction(instance->classctx, "getDiskFormatCrc", true, tableExpr, tableExpr, 1);
  3306. translator.buildFormatCrcFunction(instance->classctx, "getProjectedFormatCrc", true, tableExpr, tableExpr, 1);
  3307. IHqlExpression * originalKey = queryAttributeChild(tableExpr, _original_Atom, 0);
  3308. translator.buildSerializedLayoutMember(instance->classctx, originalKey->queryRecord(), "getIndexLayout", numKeyedFields(originalKey));
  3309. //Note the helper base class contains code like the following
  3310. //IThorIndexCallback * fpp;");
  3311. //virtual void setCallback(IThorIndexCallback * _tc) { fpp = _tc; }");
  3312. }
  3313. void IndexReadBuilderBase::buildFlagsMember(IHqlExpression * expr)
  3314. {
  3315. StringBuffer flags;
  3316. if (tableExpr->hasAttribute(sortedAtom))
  3317. flags.append("|TIRsorted");
  3318. else if (!isOrdered(tableExpr))
  3319. flags.append("|TIRunordered");
  3320. if (!monitors.isFiltered())
  3321. flags.append("|TIRnofilter");
  3322. if (isPreloaded)
  3323. flags.append("|TIRpreload");
  3324. if (tableExpr->hasAttribute(optAtom))
  3325. flags.append("|TIRoptional");
  3326. if (limitExpr && limitExpr->hasAttribute(skipAtom))
  3327. flags.append("|TIRlimitskips");
  3328. if (limitExpr && limitExpr->hasAttribute(onFailAtom))
  3329. flags.append("|TIRlimitcreates");
  3330. if (generateUnfilteredTransform)
  3331. flags.append("|TIRunfilteredtransform");
  3332. if (keyedLimitExpr)
  3333. {
  3334. if (keyedLimitExpr->hasAttribute(onFailAtom))
  3335. flags.append("|TIRkeyedlimitcreates|TIRcountkeyedlimit");
  3336. else if (keyedLimitExpr->hasAttribute(skipAtom))
  3337. flags.append("|TIRkeyedlimitskips|TIRcountkeyedlimit");
  3338. else if (keyedLimitExpr->hasAttribute(countAtom))
  3339. flags.append("|TIRcountkeyedlimit");
  3340. }
  3341. IHqlExpression * firstStepped = steppingInfo.firstStepped();
  3342. if (firstStepped && monitors.isEqualityFilterBefore(firstStepped))
  3343. flags.append("|TIRstepleadequality");
  3344. if (onlyExistsAggreate) flags.append("|TIRaggregateexists");
  3345. if (monitorsForGrouping) flags.append("|TIRgroupmonitors");
  3346. if (!nameExpr->isConstant()) flags.append("|TIRvarfilename");
  3347. if (translator.hasDynamicFilename(tableExpr)) flags.append("|TIRdynamicfilename");
  3348. if (requiresOrderedMerge) flags.append("|TIRorderedmerge");
  3349. if (translator.queryOptions().createValueSets)
  3350. flags.append("|TIRnewfilters");
  3351. if (containsOperator(expr, no_id2blob))
  3352. flags.append("|TIRusesblob");
  3353. if (flags.length())
  3354. translator.doBuildUnsignedFunction(instance->classctx, "getFlags", flags.str()+1);
  3355. }
  3356. void IndexReadBuilderBase::extractMonitors(IHqlExpression * ds, SharedHqlExpr & unkeyedFilter, HqlExprArray & conds)
  3357. {
  3358. OwnedHqlExpr extraFilter;
  3359. monitors.extractFilters(conds, extraFilter);
  3360. appendFilter(unkeyedFilter, extraFilter);
  3361. }
  3362. class MonitorRemovalTransformer : public HqlMapTransformer
  3363. {
  3364. public:
  3365. MonitorRemovalTransformer(IndexReadBuilderBase & _builder) : builder(_builder) {}
  3366. virtual IHqlExpression * createTransformed(IHqlExpression * expr);
  3367. protected:
  3368. IndexReadBuilderBase & builder;
  3369. };
  3370. IHqlExpression * MonitorRemovalTransformer::createTransformed(IHqlExpression * expr)
  3371. {
  3372. switch (expr->getOperator())
  3373. {
  3374. case no_filter:
  3375. {
  3376. IHqlExpression * ds = expr->queryChild(0);
  3377. IHqlExpression * body = expr->queryBody();
  3378. unsigned match = builder.originalFilters.find(*body);
  3379. if (match == NotFound)
  3380. {
  3381. if (builder.removedFilters.find(*body))
  3382. return transform(ds);
  3383. return NewHqlTransformer::createTransformed(expr);
  3384. }
  3385. IHqlExpression & newFilter = builder.mappedFilters.item(match);
  3386. HqlExprArray args;
  3387. args.append(*transform(ds));
  3388. args.append(*transform(&newFilter));
  3389. return expr->clone(args);
  3390. }
  3391. }
  3392. return NewHqlTransformer::createTransformed(expr);
  3393. }
  3394. IHqlExpression * IndexReadBuilderBase::removeMonitors(IHqlExpression * expr)
  3395. {
  3396. MonitorRemovalTransformer mapper(*this);
  3397. mapper.setMapping(tableExpr, tableExpr);
  3398. return mapper.transformRoot(expr);
  3399. }
  3400. //---------------------------------------------------------------------------
  3401. class NewIndexReadBuilder : public IndexReadBuilderBase
  3402. {
  3403. public:
  3404. NewIndexReadBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  3405. : IndexReadBuilderBase(_translator, _tableExpr, _nameExpr)
  3406. {
  3407. extractCanMatch = true;
  3408. }
  3409. virtual void analyseGraph(IHqlExpression * expr)
  3410. {
  3411. IndexReadBuilderBase::analyseGraph(expr);
  3412. gatherSteppingMeta(expr, steppingInfo);
  3413. if (steppedExpr && transformCanFilter && translator.queryOptions().optimizeSteppingPostfilter)
  3414. {
  3415. //If sorted, but can't step output no point in duplicating the transform
  3416. if (steppingInfo.outputStepping.exists())
  3417. generateUnfilteredTransform = true;
  3418. }
  3419. returnIfFilterFails = !isNormalize;
  3420. }
  3421. virtual void buildTransform(IHqlExpression * expr);
  3422. virtual void buildMembers(IHqlExpression * expr);
  3423. };
  3424. void NewIndexReadBuilder::buildMembers(IHqlExpression * expr)
  3425. {
  3426. if (steppingInfo.exists())
  3427. {
  3428. steppingInfo.checkKeyable(monitors);
  3429. monitors.preventMerge(steppingInfo.firstStepped());
  3430. }
  3431. buildReadMembers(expr);
  3432. if (steppingInfo.exists())
  3433. steppingInfo.generateMembers(translator, instance->classctx);
  3434. IndexReadBuilderBase::buildMembers(expr);
  3435. }
  3436. void NewIndexReadBuilder::buildTransform(IHqlExpression * expr)
  3437. {
  3438. if (true)
  3439. {
  3440. MemberFunction func(translator, instance->startctx, "virtual size32_t transform(ARowBuilder & crSelf, const void * _left) override");
  3441. translator.ensureRowAllocated(func.ctx, "crSelf");
  3442. func.ctx.addQuotedLiteral("unsigned char * left = (unsigned char *)_left;");
  3443. translator.associateBlobHelper(func.ctx, tableExpr, "fpp");
  3444. buildTransformBody(func.ctx, expr, true, false, true);
  3445. }
  3446. if (generateUnfilteredTransform)
  3447. {
  3448. MemberFunction func(translator, instance->startctx, "virtual size32_t unfilteredTransform(ARowBuilder & crSelf, const void * _left) override");
  3449. translator.ensureRowAllocated(func.ctx, "crSelf");
  3450. func.ctx.addQuotedLiteral("unsigned char * left = (unsigned char *)_left;");
  3451. translator.associateBlobHelper(func.ctx, tableExpr, "fpp");
  3452. buildTransformBody(func.ctx, expr, true, true, true);
  3453. }
  3454. }
  3455. //---------------------------------------------------------------------------
  3456. ABoundActivity * HqlCppTranslator::doBuildActivityIndexRead(BuildCtx & ctx, IHqlExpression * expr)
  3457. {
  3458. OwnedHqlExpr transformed = buildIndexFromPhysical(expr);
  3459. OwnedHqlExpr optimized = optimizeHqlExpression(queryErrorProcessor(), transformed, HOOfold);
  3460. IHqlExpression *tableExpr = queryPhysicalRootTable(optimized);
  3461. //If the filter is false, then it may get reduced to a NULL operation!
  3462. if (!tableExpr)
  3463. return buildNullIndexActivity(*this, ctx, optimized);
  3464. ensureDiskAccessAllowed(tableExpr);
  3465. if (optimized->getOperator() != no_compound_indexread)
  3466. optimized.setown(createDataset(no_compound_indexread, LINK(optimized)));
  3467. traceExpression("before index read", optimized);
  3468. assertex(tableExpr->getOperator() == no_newkeyindex);
  3469. NewIndexReadBuilder info(*this, tableExpr, tableExpr->queryChild(3));
  3470. info.gatherVirtualFields(true, true);
  3471. if (info.containsStepping(optimized))
  3472. return info.buildActivity(ctx, optimized, TAKindexread, "SteppedIndexRead", NULL);
  3473. return info.buildActivity(ctx, optimized, TAKindexread, "IndexRead", NULL);
  3474. }
  3475. //---------------------------------------------------------------------------
  3476. class IndexNormalizeBuilder : public IndexReadBuilderBase
  3477. {
  3478. public:
  3479. IndexNormalizeBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  3480. : IndexReadBuilderBase(_translator, _tableExpr, _nameExpr)
  3481. {
  3482. }
  3483. virtual void buildTransform(IHqlExpression * expr);
  3484. virtual void buildMembers(IHqlExpression * expr);
  3485. protected:
  3486. virtual void analyseGraph(IHqlExpression * expr);
  3487. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  3488. {
  3489. doBuildNormalizeIterators(ctx, expr, false);
  3490. }
  3491. };
  3492. void IndexNormalizeBuilder::analyseGraph(IHqlExpression * expr)
  3493. {
  3494. IndexReadBuilderBase::analyseGraph(expr);
  3495. needDefaultTransform = (expr->queryNormalizedSelector()->getOperator() == no_select);
  3496. }
  3497. void IndexNormalizeBuilder::buildMembers(IHqlExpression * expr)
  3498. {
  3499. buildFilenameMember();
  3500. IndexReadBuilderBase::buildMembers(expr);
  3501. buildNormalizeHelpers(expr);
  3502. }
  3503. void IndexNormalizeBuilder::buildTransform(IHqlExpression * expr)
  3504. {
  3505. globaliterctx.setown(new BuildCtx(instance->startctx));
  3506. globaliterctx->addGroup();
  3507. MemberFunction func(translator, instance->startctx, "virtual size32_t transform(ARowBuilder & crSelf) override");
  3508. translator.ensureRowAllocated(func.ctx, "crSelf");
  3509. //Because this transform creates iterator classes for the child iterators the expression tree needs to be modified
  3510. //instead of using an inline tests. We could switch to using this all the time for indexes once I trust it!
  3511. OwnedHqlExpr simplified = removeMonitors(expr);
  3512. lastTransformer.set(queryExpression(simplified->queryDataset()->queryTable()));
  3513. useFilterMappings=false;
  3514. buildTransformBody(func.ctx, simplified, true, false, false);
  3515. }
  3516. //---------------------------------------------------------------------------
  3517. ABoundActivity * HqlCppTranslator::doBuildActivityIndexNormalize(BuildCtx & ctx, IHqlExpression * expr)
  3518. {
  3519. OwnedHqlExpr transformed = buildIndexFromPhysical(expr);
  3520. OwnedHqlExpr optimized = optimizeHqlExpression(queryErrorProcessor(), transformed, HOOfold);
  3521. traceExpression("after optimize", optimized);
  3522. IHqlExpression *tableExpr = queryPhysicalRootTable(optimized);
  3523. if (!tableExpr)
  3524. return buildNullIndexActivity(*this, ctx, optimized);
  3525. ensureDiskAccessAllowed(tableExpr);
  3526. assertex(tableExpr->getOperator() == no_newkeyindex);
  3527. IndexNormalizeBuilder info(*this, tableExpr, tableExpr->queryChild(3));
  3528. info.gatherVirtualFields(true, true);
  3529. return info.buildActivity(ctx, optimized, TAKindexnormalize, "IndexNormalize", NULL);
  3530. }
  3531. //---------------------------------------------------------------------------
  3532. class IndexAggregateBuilder : public IndexReadBuilderBase
  3533. {
  3534. public:
  3535. IndexAggregateBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  3536. : IndexReadBuilderBase(_translator, _tableExpr, _nameExpr)
  3537. {
  3538. failedFilterValue.clear();
  3539. }
  3540. virtual void buildTransform(IHqlExpression * expr);
  3541. virtual void buildMembers(IHqlExpression * expr);
  3542. protected:
  3543. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  3544. {
  3545. doBuildAggregateSelectIterator(ctx, expr);
  3546. }
  3547. virtual void analyseGraph(IHqlExpression * expr)
  3548. {
  3549. IndexReadBuilderBase::analyseGraph(expr);
  3550. returnIfFilterFails = !isNormalize;
  3551. }
  3552. };
  3553. void IndexAggregateBuilder::buildMembers(IHqlExpression * expr)
  3554. {
  3555. StringBuffer s;
  3556. buildFilenameMember();
  3557. IndexReadBuilderBase::buildMembers(expr);
  3558. buildAggregateHelpers(expr);
  3559. //virtual void processRow(void * self, const void * src) = 0;
  3560. {
  3561. BuildCtx rowctx(instance->startctx);
  3562. rowctx.addQuotedFunction("virtual void processRow(ARowBuilder & crSelf, const void * src) override");
  3563. rowctx.addQuotedLiteral("doProcessRow(crSelf, (const byte *)src);");
  3564. }
  3565. //virtual void processRows(ARowBuilder & crSelf, size32_t srcLen, const void * _left)
  3566. //is meaningless for an index - uses the default error implementation in the base class
  3567. }
  3568. void IndexAggregateBuilder::buildTransform(IHqlExpression * expr)
  3569. {
  3570. MemberFunction func(translator, instance->startctx, "void doProcessRow(ARowBuilder & crSelf, const byte * left)");
  3571. translator.ensureRowAllocated(func.ctx, "crSelf");
  3572. translator.associateBlobHelper(func.ctx, tableExpr, "fpp");
  3573. buildTransformBody(func.ctx, expr, false, false, true);
  3574. }
  3575. //---------------------------------------------------------------------------
  3576. class IndexCountBuilder : public IndexReadBuilderBase
  3577. {
  3578. public:
  3579. IndexCountBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr, node_operator _aggOp)
  3580. : IndexReadBuilderBase(_translator, _tableExpr, _nameExpr)
  3581. {
  3582. aggOp = _aggOp;
  3583. isCompoundCount = true;
  3584. failedFilterValue.set(queryZero());
  3585. }
  3586. virtual void buildTransform(IHqlExpression * expr);
  3587. virtual void buildMembers(IHqlExpression * expr);
  3588. virtual bool isExplicitExists() { return (aggOp == no_existsgroup); }
  3589. protected:
  3590. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  3591. {
  3592. doBuildAggregateSelectIterator(ctx, expr);
  3593. }
  3594. virtual void analyseGraph(IHqlExpression * expr)
  3595. {
  3596. IndexReadBuilderBase::analyseGraph(expr);
  3597. returnIfFilterFails = !isNormalize;
  3598. IHqlExpression * aggregate = expr->queryChild(0);
  3599. if (isKeyedCountAggregate(aggregate))
  3600. {
  3601. if (isNormalize)
  3602. translator.throwError(HQLERR_KeyedCountCantNormalize);
  3603. if (!monitors.isKeyedExplicitly())
  3604. translator.throwError(HQLERR_KeyedCountNotKeyed);
  3605. if (transformCanFilter)
  3606. {
  3607. ForEachItemIn(i, originalFilters)
  3608. removedFilters.append(OLINK(originalFilters.item(i)));
  3609. originalFilters.kill();
  3610. mappedFilters.kill();
  3611. transformCanFilter = false;
  3612. }
  3613. }
  3614. if (aggOp == no_existsgroup)
  3615. choosenValue.setown(getSizetConstant(1));
  3616. }
  3617. protected:
  3618. node_operator aggOp;
  3619. };
  3620. void IndexCountBuilder::buildMembers(IHqlExpression * expr)
  3621. {
  3622. buildFilenameMember();
  3623. IndexReadBuilderBase::buildMembers(expr);
  3624. buildCountHelpers(expr, false);
  3625. }
  3626. void IndexCountBuilder::buildTransform(IHqlExpression * expr)
  3627. {
  3628. if (transformCanFilter||isNormalize)
  3629. {
  3630. MemberFunction func(translator, instance->startctx, "virtual size32_t numValid(const void * _left) override");
  3631. func.ctx.addQuotedLiteral("unsigned char * left = (unsigned char *)_left;");
  3632. translator.associateBlobHelper(func.ctx, tableExpr, "fpp");
  3633. OwnedHqlExpr cnt;
  3634. if (isNormalize)
  3635. {
  3636. compoundCountVar.setown(func.ctx.getTempDeclare(sizetType, queryZero()));
  3637. cnt.set(compoundCountVar);
  3638. }
  3639. else
  3640. cnt.setown(getSizetConstant(1));
  3641. BuildCtx subctx(func.ctx);
  3642. buildTransformBody(subctx, expr, false, false, true);
  3643. func.ctx.addReturn(cnt);
  3644. }
  3645. }
  3646. //---------------------------------------------------------------------------
  3647. ABoundActivity * HqlCppTranslator::doBuildActivityIndexAggregate(BuildCtx & ctx, IHqlExpression * expr)
  3648. {
  3649. OwnedHqlExpr transformed = buildIndexFromPhysical(expr);
  3650. OwnedHqlExpr optimized = optimizeHqlExpression(queryErrorProcessor(), transformed, getSourceAggregateOptimizeFlags());
  3651. IHqlExpression *tableExpr = queryPhysicalRootTable(optimized);
  3652. if (!tableExpr)
  3653. return buildNullIndexActivity(*this, ctx, optimized);
  3654. ensureDiskAccessAllowed(tableExpr);
  3655. assertex(tableExpr->getOperator() == no_newkeyindex);
  3656. node_operator aggOp = querySimpleAggregate(expr, true, false);
  3657. if (aggOp == no_countgroup || aggOp == no_existsgroup)
  3658. {
  3659. IndexCountBuilder info(*this, tableExpr, tableExpr->queryChild(3), aggOp);
  3660. info.gatherVirtualFields(true, true);
  3661. return info.buildActivity(ctx, optimized, TAKindexcount, "IndexCount", NULL);
  3662. }
  3663. else
  3664. {
  3665. IndexAggregateBuilder info(*this, tableExpr, tableExpr->queryChild(3));
  3666. info.gatherVirtualFields(true, true);
  3667. return info.buildActivity(ctx, optimized, TAKindexaggregate, "IndexAggregate", NULL);
  3668. }
  3669. }
  3670. //---------------------------------------------------------------------------
  3671. class IndexGroupAggregateBuilder : public IndexReadBuilderBase
  3672. {
  3673. public:
  3674. IndexGroupAggregateBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr)
  3675. : IndexReadBuilderBase(_translator, _tableExpr, _nameExpr)
  3676. {
  3677. failedFilterValue.clear();
  3678. transformAccessesCallback = false;
  3679. }
  3680. virtual void buildTransform(IHqlExpression * expr);
  3681. virtual void buildMembers(IHqlExpression * expr);
  3682. protected:
  3683. void doBuildProcessCountMembers(BuildCtx & ctx, IHqlExpression * aggregate);
  3684. virtual void processTransformSelect(BuildCtx & ctx, IHqlExpression * expr)
  3685. {
  3686. doBuildAggregateSelectIterator(ctx, expr);
  3687. }
  3688. virtual void analyseGraph(IHqlExpression * expr)
  3689. {
  3690. IndexReadBuilderBase::analyseGraph(expr);
  3691. returnIfFilterFails = !isNormalize;
  3692. }
  3693. protected:
  3694. bool transformAccessesCallback;
  3695. };
  3696. void IndexGroupAggregateBuilder::buildMembers(IHqlExpression * expr)
  3697. {
  3698. transformAccessesCallback = containsOperator(expr, no_filepos) || containsOperator(expr, no_id2blob);
  3699. buildFilenameMember();
  3700. buildGroupingMonitors(expr, monitors);
  3701. IndexReadBuilderBase::buildMembers(expr);
  3702. buildGlobalGroupAggregateHelpers(expr);
  3703. if (!isNormalize && !transformCanFilter && monitorsForGrouping && !transformAccessesCallback)
  3704. {
  3705. IHqlExpression * aggregate = expr->queryChild(0);
  3706. ThorActivityKind newKind = TAKnone;
  3707. switch (querySingleAggregate(aggregate, false, true, true))
  3708. {
  3709. case no_countgroup:
  3710. newKind = TAKindexgroupcount;
  3711. break;
  3712. case no_existsgroup:
  3713. newKind = TAKindexgroupexists;
  3714. break;
  3715. }
  3716. if (newKind)
  3717. {
  3718. instance->changeActivityKind(newKind);
  3719. doBuildProcessCountMembers(instance->startctx, aggregate);
  3720. }
  3721. }
  3722. }
  3723. void IndexGroupAggregateBuilder::doBuildProcessCountMembers(BuildCtx & ctx, IHqlExpression * aggregate)
  3724. {
  3725. IHqlExpression * dataset = aggregate->queryChild(0);
  3726. IHqlExpression * tgtRecord = aggregate->queryChild(1);
  3727. IHqlExpression * transform = aggregate->queryChild(2);
  3728. OwnedHqlExpr resultDataset = createDataset(no_anon, LINK(tgtRecord));
  3729. {
  3730. MemberFunction func(translator, ctx, "virtual size32_t initialiseCountGrouping(ARowBuilder & crSelf, const void * _src) override");
  3731. translator.ensureRowAllocated(func.ctx, "crSelf");
  3732. func.ctx.addQuotedLiteral("unsigned char * src = (unsigned char *) _src;");
  3733. translator.associateBlobHelper(func.ctx, tableExpr, "fpp");
  3734. BoundRow * selfCursor = translator.bindSelf(func.ctx, resultDataset, "crSelf");
  3735. translator.bindTableCursor(func.ctx, dataset, "src");
  3736. //Replace count() with 0, exists() with true and call as a transform - which will error if the replacement fails.
  3737. OwnedHqlExpr count = createValue(no_countgroup, LINK(defaultIntegralType));
  3738. OwnedHqlExpr exists = createValue(no_existsgroup, makeBoolType());
  3739. OwnedHqlExpr newCount = createNullExpr(count);
  3740. OwnedHqlExpr newTransform = replaceExpression(transform, count, newCount);
  3741. newTransform.setown(replaceExpression(newTransform, exists, queryBoolExpr(true)));
  3742. translator.doTransform(func.ctx, newTransform, selfCursor);
  3743. translator.buildReturnRecordSize(func.ctx, selfCursor);
  3744. }
  3745. {
  3746. MemberFunction func(translator, ctx, "virtual size32_t processCountGrouping(ARowBuilder & crSelf, unsigned __int64 count) override");
  3747. translator.ensureRowAllocated(func.ctx, "crSelf");
  3748. BoundRow * selfCursor = translator.bindSelf(func.ctx, resultDataset, "crSelf");
  3749. OwnedHqlExpr newCount = createTranslatedOwned(createVariable("count", LINK(defaultIntegralType)));
  3750. OwnedHqlExpr self = getSelf(aggregate);
  3751. ForEachChild(idx, transform)
  3752. {
  3753. IHqlExpression * cur = transform->queryChild(idx);
  3754. if (cur->isAttribute())
  3755. continue;
  3756. OwnedHqlExpr target = selfCursor->bindToRow(cur->queryChild(0), self);
  3757. IHqlExpression * src = cur->queryChild(1);
  3758. IHqlExpression * arg = queryRealChild(src, 0);
  3759. BuildCtx condctx(func.ctx);
  3760. node_operator srcOp = src->getOperator();
  3761. switch (srcOp)
  3762. {
  3763. case no_countgroup:
  3764. {
  3765. if (arg)
  3766. translator.buildFilter(condctx, arg);
  3767. OwnedHqlExpr newValue = createValue(no_add, target->getType(), LINK(target), ensureExprType(newCount, target->queryType()));
  3768. translator.buildAssign(condctx, target, newValue);
  3769. }
  3770. break;
  3771. }
  3772. }
  3773. translator.buildReturnRecordSize(func.ctx, selfCursor);
  3774. }
  3775. }
  3776. void IndexGroupAggregateBuilder::buildTransform(IHqlExpression * expr)
  3777. {
  3778. MemberFunction func(translator, instance->startctx, "void doProcessRow(const byte * left, IHThorGroupAggregateCallback * callback)");
  3779. translator.associateBlobHelper(func.ctx, tableExpr, "fpp");
  3780. buildGroupAggregateTransformBody(func.ctx, expr, isNormalize || transformAccessesCallback, true);
  3781. }
  3782. //---------------------------------------------------------------------------
  3783. ABoundActivity * HqlCppTranslator::doBuildActivityIndexGroupAggregate(BuildCtx & ctx, IHqlExpression * expr)
  3784. {
  3785. OwnedHqlExpr transformed = buildIndexFromPhysical(expr);
  3786. OwnedHqlExpr optimized = optimizeHqlExpression(queryErrorProcessor(), transformed, getSourceAggregateOptimizeFlags());
  3787. IHqlExpression *tableExpr = queryPhysicalRootTable(optimized);
  3788. if (!tableExpr)
  3789. return buildNullIndexActivity(*this, ctx, optimized);
  3790. ensureDiskAccessAllowed(tableExpr);
  3791. IHqlExpression * aggregate = expr->queryChild(0);
  3792. assertex(aggregate->getOperator() == no_newaggregate || aggregate->getOperator() == no_aggregate);
  3793. ThorActivityKind tak = TAKindexgroupaggregate;
  3794. assertex(tableExpr->getOperator() == no_newkeyindex);
  3795. IndexGroupAggregateBuilder info(*this, tableExpr, tableExpr->queryChild(3));
  3796. info.gatherVirtualFields(true, true);
  3797. return info.buildActivity(ctx, optimized, tak, "IndexGroupAggregate", NULL);
  3798. }
  3799. //---------------------------------------------------------------------------
  3800. void associateVirtualCallbacks(HqlCppTranslator & translator, BuildCtx & ctx, IHqlExpression * dataset)
  3801. {
  3802. OwnedHqlExpr fpos = getFilepos(dataset, false);
  3803. OwnedHqlExpr lfpos = getFilepos(dataset, true);
  3804. Owned<IHqlExpression> fposExpr = createFileposCall(translator, getFilePositionId, "fpp", "crSelf.row()");
  3805. Owned<IHqlExpression> lfposExpr = createFileposCall(translator, getLocalFilePositionId, "fpp", "crSelf.row()");
  3806. ctx.associateExpr(fpos, fposExpr);
  3807. ctx.associateExpr(lfpos, lfposExpr);
  3808. }
  3809. void HqlCppTranslator::buildXmlReadTransform(IHqlExpression * dataset, StringBuffer & factoryName, bool & usesContents)
  3810. {
  3811. OwnedHqlExpr xmlMarker = createAttribute(xmlReadMarkerAtom, LINK(dataset->queryRecord()));
  3812. BuildCtx declarectx(*code, declareAtom);
  3813. HqlExprAssociation * match = declarectx.queryMatchExpr(xmlMarker);
  3814. if (match)
  3815. {
  3816. IHqlExpression * matchExpr = match->queryExpr();
  3817. matchExpr->queryChild(0)->queryValue()->getStringValue(factoryName);
  3818. usesContents = matchExpr->queryChild(1)->queryValue()->getBoolValue();
  3819. return;
  3820. }
  3821. StringBuffer s, id, className;
  3822. getUniqueId(id);
  3823. className.append("cx2r").append(id);
  3824. const char * interfaceName = "IXmlToRowTransformer";
  3825. StringBuffer prolog, epilog;
  3826. prolog.append("struct ").append(className).append(" : public RtlCInterface, implements ").append(interfaceName);
  3827. epilog.append(";");
  3828. GlobalClassBuilder builder(*this, declarectx, className, "CXmlToRowTransformer", interfaceName, true);
  3829. builder.buildClass(XmlTransformerPrio);
  3830. builder.setIncomplete(true);
  3831. BuildCtx & classctx = builder.classctx;
  3832. s.clear().append("inline ").append(className).append("(unsigned _activityId) : CXmlToRowTransformer(_activityId) {}");
  3833. classctx.addQuoted(s);
  3834. {
  3835. MemberFunction func(*this, classctx, "virtual size32_t transform(ARowBuilder & crSelf, IColumnProvider * row, IThorDiskCallback * fpp) override");
  3836. ensureRowAllocated(func.ctx, "crSelf");
  3837. xmlUsesContents = false;
  3838. //MORE: If this becomes a compound activity
  3839. BoundRow * rootSelfRow = bindSelf(func.ctx, dataset, "crSelf");
  3840. bindXmlTableCursor(func.ctx, dataset, "row", no_none, NULL, true);
  3841. OwnedHqlExpr activityId = createVariable("activityId", LINK(sizetType));
  3842. func.ctx.associateExpr(queryActivityIdMarker(), activityId);
  3843. associateVirtualCallbacks(*this, func.ctx, dataset);
  3844. OwnedHqlExpr active = ensureActiveRow(dataset);
  3845. buildAssign(func.ctx, rootSelfRow->querySelector(), active);
  3846. buildReturnRecordSize(func.ctx, rootSelfRow);
  3847. usesContents = xmlUsesContents;
  3848. rootSelfRow = NULL;
  3849. }
  3850. buildMetaMember(classctx, dataset, false, "queryRecordSize");
  3851. builder.setIncomplete(false);
  3852. builder.completeClass(XmlTransformerPrio);
  3853. factoryName.append(builder.accessorName);
  3854. OwnedHqlExpr matchedValue = createAttribute(internalAtom, createConstant(factoryName.str()), createConstant(usesContents));
  3855. declarectx.associateExpr(xmlMarker, matchedValue);
  3856. }
  3857. //---------------------------------------------------------------------------
  3858. unsigned HqlCppTranslator::buildCsvReadTransform(BuildCtx & subctx, IHqlExpression * dataset, bool newInterface, IHqlExpression * csvAttr)
  3859. {
  3860. MemberFunction func(*this, subctx);
  3861. if (newInterface)
  3862. func.start("virtual size32_t transform(ARowBuilder & crSelf, unsigned * lenSrc, const char * * dataSrc) override");
  3863. else
  3864. func.start("virtual size32_t transform(ARowBuilder & crSelf, unsigned * lenSrc, const char * * dataSrc, unsigned __int64 _fpos) override");
  3865. //MORE: If this becomes a compound activity
  3866. BoundRow * rootSelfRow = bindSelf(func.ctx, dataset, "crSelf");
  3867. bindCsvTableCursor(func.ctx, dataset, "Src", no_none, NULL, true, queryCsvEncoding(csvAttr));
  3868. ensureRowAllocated(func.ctx, rootSelfRow);
  3869. if (newInterface)
  3870. {
  3871. associateVirtualCallbacks(*this, func.ctx, dataset);
  3872. }
  3873. else
  3874. {
  3875. OwnedHqlExpr fpos = getFilepos(dataset, false);
  3876. OwnedHqlExpr fposVar = createVariable("_fpos", fpos->getType());
  3877. func.ctx.associateExpr(fpos, fposVar);
  3878. }
  3879. OwnedHqlExpr active = ensureActiveRow(dataset);
  3880. buildAssign(func.ctx, rootSelfRow->querySelector(), active);
  3881. buildReturnRecordSize(func.ctx, rootSelfRow);
  3882. rootSelfRow = NULL;
  3883. return countTotalFields(dataset->queryRecord(), false);
  3884. }
  3885. void HqlCppTranslator::buildCsvReadTransformer(IHqlExpression * dataset, StringBuffer & instanceName, IHqlExpression * optCsvAttr)
  3886. {
  3887. OwnedHqlExpr csvMarker = createAttribute(csvReadMarkerAtom, LINK(dataset->queryRecord()), LINK(optCsvAttr));
  3888. BuildCtx declarectx(*code, declareAtom);
  3889. HqlExprAssociation * match = declarectx.queryMatchExpr(csvMarker);
  3890. if (match)
  3891. {
  3892. IHqlExpression * matchExpr = match->queryExpr();
  3893. matchExpr->queryChild(0)->queryValue()->getStringValue(instanceName);
  3894. return;
  3895. }
  3896. StringBuffer id, className;
  3897. getUniqueId(id);
  3898. instanceName.append("c2r").append(id);
  3899. className.append("cc2r").append(id);
  3900. StringBuffer prolog, epilog;
  3901. prolog.append("struct ").append(className).append(" : public RtlCInterface, implements ICsvToRowTransformer");
  3902. epilog.append(" ").append(instanceName).append(";");
  3903. BuildCtx classctx(declarectx);
  3904. classctx.setNextPriority(XmlTransformerPrio);
  3905. IHqlStmt * transformClass = classctx.addQuotedCompound(prolog, epilog);
  3906. transformClass->setIncomplete(true);
  3907. transformClass->setIncluded(false); // if can't generate csv for this record, then don't generate an invalid class.
  3908. classctx.addQuotedLiteral("virtual void Link() const override { RtlCInterface::Link(); }");
  3909. classctx.addQuotedLiteral("virtual bool Release() const override { return RtlCInterface::Release(); }");
  3910. unsigned maxColumns = buildCsvReadTransform(classctx, dataset, false, optCsvAttr);
  3911. doBuildUnsignedFunction(classctx, "getMaxColumns", maxColumns);
  3912. buildMetaMember(classctx, dataset, false, "queryRecordSize");
  3913. buildCsvParameters(classctx, optCsvAttr, NULL, true);
  3914. transformClass->setIncomplete(false);
  3915. transformClass->setIncluded(true);
  3916. if (options.spanMultipleCpp)
  3917. {
  3918. StringBuffer helperFunc;
  3919. createAccessFunctions(helperFunc, declarectx, XmlTransformerPrio, "ICsvToRowTransformer", instanceName.str());
  3920. instanceName.clear().append(helperFunc).append("()");
  3921. }
  3922. OwnedHqlExpr matchedValue = createAttribute(internalAtom, createConstant(instanceName.str()));
  3923. declarectx.associateExpr(csvMarker, matchedValue);
  3924. }
  3925. ABoundActivity * HqlCppTranslator::doBuildActivityXmlRead(BuildCtx & ctx, IHqlExpression * expr)
  3926. {
  3927. IHqlExpression * tableExpr = expr;
  3928. ensureDiskAccessAllowed(tableExpr);
  3929. IHqlExpression * filename = tableExpr->queryChild(0);
  3930. IHqlExpression * mode = tableExpr->queryChild(2);
  3931. node_operator modeType = mode->getOperator();
  3932. StringBuffer s;
  3933. ThorActivityKind kind = (modeType == no_json) ? TAKjsonread : TAKxmlread;
  3934. Owned<ActivityInstance> instance = new ActivityInstance(*this, ctx, kind, expr, "XmlRead");
  3935. buildActivityFramework(instance);
  3936. buildInstancePrefix(instance);
  3937. //MORE: Improve when we support projecting xml instead of reading all
  3938. SourceFieldUsage * fieldUsage = querySourceFieldUsage(tableExpr);
  3939. if (fieldUsage && !fieldUsage->seenAll())
  3940. fieldUsage->noteAll();
  3941. //---- virtual const char * getFileName() { return "x.d00"; } ----
  3942. buildFilenameFunction(*instance, instance->startctx, WaFilename, "getFileName", filename, hasDynamicFilename(tableExpr));
  3943. buildEncryptHelper(instance->startctx, tableExpr->queryAttribute(encryptAtom));
  3944. bool usesContents = false;
  3945. doBuildXmlReadMember(*instance, tableExpr, "queryTransformer", usesContents);
  3946. doBuildVarStringFunction(instance->classctx, "getXmlIteratorPath", queryRealChild(mode, 0));
  3947. buildMetaMember(instance->classctx, tableExpr, false, "queryDiskRecordSize"); // A lie, but I don't care....
  3948. buildMetaMember(instance->classctx, tableExpr, false, "queryProjectedDiskRecordSize"); // A lie, but I don't care....
  3949. //virtual unsigned getFlags() = 0;
  3950. StringBuffer flags;
  3951. if (expr->hasAttribute(_spill_Atom)) flags.append("|TDXtemporary");
  3952. if (expr->hasAttribute(unsortedAtom)) flags.append("|TDRunsorted");
  3953. if (expr->hasAttribute(optAtom)) flags.append("|TDRoptional");
  3954. if (usesContents) flags.append("|TDRusexmlcontents");
  3955. if (mode->hasAttribute(noRootAtom)) flags.append("|TDRxmlnoroot");
  3956. if (!filename->isConstant()) flags.append("|TDXvarfilename");
  3957. if (hasDynamicFilename(expr)) flags.append("|TDXdynamicfilename");
  3958. if (flags.length())
  3959. doBuildUnsignedFunction(instance->classctx, "getFlags", flags.str()+1);
  3960. //Note the helper base class contains code like the following
  3961. //IThorDiskCallback * fpp;");
  3962. //virtual void setCallback(IThorDiskCallback * _tc) { fpp = _tc; }");
  3963. buildInstanceSuffix(instance);
  3964. addFileDependency(filename, instance->queryBoundActivity());
  3965. return instance->getBoundActivity();
  3966. }
  3967. //---------------------------------------------------------------------------
  3968. ABoundActivity * HqlCppTranslator::doBuildActivityTable(BuildCtx & ctx, IHqlExpression * expr)
  3969. {
  3970. node_operator mode = expr->queryChild(2)->getOperator();
  3971. switch (mode)
  3972. {
  3973. case no_thor:
  3974. case no_flat:
  3975. case no_pipe:
  3976. case no_csv:
  3977. return doBuildActivityDiskRead(ctx, expr);
  3978. case no_xml:
  3979. case no_json:
  3980. return doBuildActivityXmlRead(ctx, expr);
  3981. default:
  3982. UNIMPLEMENTED;
  3983. }
  3984. }
  3985. //---------------------------------------------------------------------------
  3986. class FetchBuilder : public SourceBuilder
  3987. {
  3988. public:
  3989. FetchBuilder(HqlCppTranslator & _translator, IHqlExpression *_tableExpr, IHqlExpression *_nameExpr, IHqlExpression * _fetchExpr)
  3990. : SourceBuilder(_translator, _tableExpr, _nameExpr)
  3991. {
  3992. compoundExpr.set(_fetchExpr);
  3993. fetchExpr.set(queryFetch(_fetchExpr));
  3994. selSeq.set(querySelSeq(fetchExpr));
  3995. fetchRhs = fetchExpr->queryChild(1);
  3996. memoryRhsRecord = fetchRhs->queryRecord();
  3997. serializedRhsRecord.setown(getSerializedForm(memoryRhsRecord, diskAtom));
  3998. }
  3999. virtual void buildMembers(IHqlExpression * expr);
  4000. virtual void buildTransform(IHqlExpression * expr);
  4001. virtual void buildTransformFpos(BuildCtx & transformCtx);
  4002. protected:
  4003. HqlExprAttr compoundExpr;
  4004. HqlExprAttr fetchExpr;
  4005. HqlExprAttr selSeq;
  4006. HqlExprAttr serializedRhsRecord;
  4007. IHqlExpression * fetchRhs;
  4008. IHqlExpression * memoryRhsRecord;
  4009. };
  4010. void FetchBuilder::buildMembers(IHqlExpression * expr)
  4011. {
  4012. buildFilenameMember();
  4013. IHqlExpression * fetch = queryFetch(expr);
  4014. {
  4015. MemberFunction func(translator, instance->startctx, "virtual unsigned __int64 extractPosition(const void * _right) override");
  4016. func.ctx.addQuotedLiteral("const unsigned char * right = (const unsigned char *) _right;");
  4017. translator.bindTableCursor(func.ctx, fetch->queryChild(1), "right", no_right, selSeq);
  4018. translator.buildReturn(func.ctx, fetch->queryChild(2));
  4019. }
  4020. translator.buildEncryptHelper(instance->startctx, tableExpr->queryAttribute(encryptAtom), "getFileEncryptKey");
  4021. //Fetch flags
  4022. StringBuffer flags;
  4023. if (tableExpr->hasAttribute(optAtom))
  4024. flags.append("|FFdatafileoptional");
  4025. if (!nameExpr->isConstant())
  4026. flags.append("|FFvarfilename");
  4027. if (translator.hasDynamicFilename(tableExpr))
  4028. flags.append("|FFdynamicfilename");
  4029. if (flags.length())
  4030. translator.doBuildUnsignedFunction(instance->classctx, "getFetchFlags", flags.str()+1);
  4031. if (tableExpr->hasAttribute(optAtom) && translator.targetRoxie())
  4032. instance->addAttributeBool(WaIsFileOpt, true);
  4033. buildLimits(instance->startctx, expr, instance->activityId);
  4034. switch (getDatasetKind(tableExpr))
  4035. {
  4036. case no_csv:
  4037. {
  4038. translator.buildCsvParameters(instance->nestedctx, tableExpr->queryChild(2), NULL, true);
  4039. unsigned maxColumns = getFieldCount(tableExpr->queryRecord());
  4040. StringBuffer s;
  4041. s.clear().append("virtual unsigned getMaxColumns() override { return ").append(maxColumns).append("; }");
  4042. instance->classctx.addQuoted(s);
  4043. break;
  4044. }
  4045. case no_xml:
  4046. case no_json:
  4047. break;
  4048. default:
  4049. translator.buildFormatCrcFunction(instance->classctx, "getDiskFormatCrc", physicalRecord);
  4050. break;
  4051. }
  4052. if (!containsOnlyLeft(fetch->queryChild(3), true))
  4053. {
  4054. //MORE: Need to change following if we optimize it to only extract the relevant fields.
  4055. instance->classctx.addQuotedLiteral("virtual bool extractAllJoinFields() override { return true; }");
  4056. {
  4057. MemberFunction func(translator, instance->startctx, "virtual size32_t extractJoinFields(ARowBuilder & crSelf, const void * _left) override");
  4058. translator.ensureRowAllocated(func.ctx, "crSelf");
  4059. translator.buildRecordSerializeExtract(func.ctx, memoryRhsRecord);
  4060. }
  4061. StringBuffer s;
  4062. MetaInstance meta(translator, serializedRhsRecord, false);
  4063. translator.buildMetaInfo(meta);
  4064. instance->classctx.addQuoted(s.clear().append("virtual IOutputMetaData * queryExtractedSize() override { return &").append(meta.queryInstanceObject()).append("; }"));
  4065. }
  4066. translator.buildMetaMember(instance->classctx, expectedRecord, isGrouped(tableExpr), "queryDiskRecordSize");
  4067. translator.buildMetaMember(instance->classctx, projectedRecord, isGrouped(tableExpr), "queryProjectedDiskRecordSize");
  4068. }
  4069. void FetchBuilder::buildTransform(IHqlExpression * expr)
  4070. {
  4071. translator.xmlUsesContents = false;
  4072. MemberFunction func(translator, instance->startctx);
  4073. switch (getDatasetKind(tableExpr))
  4074. {
  4075. case no_csv:
  4076. func.start("virtual size32_t transform(ARowBuilder & crSelf, unsigned * lenLeft, const char * * dataLeft, const void * _right, unsigned __int64 _fpos) override");
  4077. func.ctx.addQuotedLiteral("unsigned char * right = (unsigned char *)_right;");
  4078. break;
  4079. case no_xml:
  4080. case no_json:
  4081. func.start("virtual size32_t transform(ARowBuilder & crSelf, IColumnProvider * xmlLeft, const void * _right, unsigned __int64 _fpos) override");
  4082. func.ctx.addQuotedLiteral("unsigned char * right = (unsigned char *)_right;");
  4083. break;
  4084. default:
  4085. func.start("virtual size32_t transform(ARowBuilder & crSelf, const void * _left, const void * _right, unsigned __int64 _fpos) override");
  4086. func.ctx.addQuotedLiteral("unsigned char * left = (unsigned char *)_left;");
  4087. func.ctx.addQuotedLiteral("unsigned char * right = (unsigned char *)_right;");
  4088. break;
  4089. }
  4090. translator.ensureRowAllocated(func.ctx, "crSelf");
  4091. buildTransformBody(func.ctx, expr, true, false, true);
  4092. if (translator.xmlUsesContents)
  4093. instance->classctx.addQuotedLiteral("virtual bool requiresContents() override { return true; }");
  4094. }
  4095. void FetchBuilder::buildTransformFpos(BuildCtx & transformCtx)
  4096. {
  4097. fpos.setown(createVariable("_fpos", LINK(fposType)));
  4098. //NB: Because the fetch gets merged with the usertable used to project the dataset, the
  4099. //transform contains filepos(LEFT) not filepos(tableExpr)
  4100. OwnedHqlExpr leftSelect = createSelector(no_left, fetchExpr->queryChild(0), selSeq);
  4101. OwnedHqlExpr fposField = getFilepos(leftSelect, false);
  4102. transformCtx.associateExpr(fposField, fpos);
  4103. //MORE: Could possibly support virtual(filename) here
  4104. }
  4105. static HqlTransformerInfo fetchInputReplacerInfo("FetchInputReplacer");
  4106. class FetchInputReplacer : public NewHqlTransformer
  4107. {
  4108. public:
  4109. FetchInputReplacer(IHqlExpression * _newDataset, node_operator side)
  4110. : NewHqlTransformer(fetchInputReplacerInfo)
  4111. {
  4112. newDataset = _newDataset;
  4113. child = (side == no_left) ? 0 : 1;
  4114. }
  4115. virtual IHqlExpression * createTransformed(IHqlExpression * expr)
  4116. {
  4117. if (expr->getOperator() == no_fetch)
  4118. return replaceChild(expr, child, newDataset);
  4119. return NewHqlTransformer::createTransformed(expr);
  4120. }
  4121. protected:
  4122. IHqlExpression * newDataset;
  4123. unsigned child;
  4124. };
  4125. IHqlExpression * replaceFetchInput(IHqlExpression * expr, IHqlExpression * newDataset, node_operator side)
  4126. {
  4127. FetchInputReplacer replacer(newDataset, side);
  4128. return replacer.transformRoot(expr);
  4129. }
  4130. ABoundActivity * HqlCppTranslator::doBuildActivityFetch(BuildCtx & ctx, IHqlExpression * expr)
  4131. {
  4132. IHqlExpression *fetch = queryFetch(expr);
  4133. IHqlExpression *tableExpr = queryPhysicalRootTable(fetch->queryChild(0));
  4134. if (!tableExpr)
  4135. throwError(HQLERR_FetchNonDiskfile);
  4136. FetchBuilder info(*this, tableExpr, tableExpr->queryChild(0), expr);
  4137. info.gatherVirtualFields(false, true);//?needToSerializeRecord(mode)
  4138. unsigned optFlags = (options.foldOptimized ? HOOfold : 0);
  4139. if (info.recordHasVirtualsOrDeserialize())
  4140. {
  4141. OwnedHqlExpr projected = createTableWithoutVirtuals(info.fieldInfo, tableExpr);
  4142. //Nasty: We don't want to optimize the rhs, otherwise references get changed!
  4143. //so optimize everything except the rhs, and then add the rhs back in again.
  4144. IHqlExpression * fetchRhs = fetch->queryChild(1);
  4145. OwnedHqlExpr null = createDataset(no_anon, LINK(fetchRhs->queryRecord()));
  4146. OwnedHqlExpr simple = replaceFetchInput(expr, null, no_right);
  4147. OwnedHqlExpr transformed = replaceExpression(simple, tableExpr, projected);
  4148. OwnedHqlExpr optSimple = optimizeHqlExpression(queryErrorProcessor(), transformed, optFlags);
  4149. IHqlExpression * newFetch = queryFetch(optSimple);
  4150. assertex(newFetch);
  4151. IHqlExpression * lhs = newFetch->queryChild(0);
  4152. if (lhs->getOperator() != no_table)
  4153. throwError1(HQLERR_ExpectedFileLhsFetch, getOpString(lhs->getOperator()));
  4154. OwnedHqlExpr optimized = replaceFetchInput(optSimple, fetchRhs, no_right);
  4155. return doBuildActivityFetch(ctx, optimized);
  4156. }
  4157. if (getProjectCount(expr) > 1)
  4158. {
  4159. OwnedHqlExpr optimized = optimizeHqlExpression(queryErrorProcessor(), expr, optFlags);
  4160. return doBuildActivityFetch(ctx, optimized);
  4161. }
  4162. Owned<ABoundActivity> childActivity = buildCachedActivity(ctx, fetch->queryChild(1));
  4163. node_operator kind = getDatasetKind(tableExpr);
  4164. switch (kind)
  4165. {
  4166. case no_csv:
  4167. return info.buildActivity(ctx, expr, TAKcsvfetch, "CsvFetch", childActivity);
  4168. case no_xml:
  4169. return info.buildActivity(ctx, expr, TAKxmlfetch, "XmlFetch", childActivity);
  4170. case no_json:
  4171. //Note use of "XmlFetch" because we want the code generator to leverage existing xml classes
  4172. return info.buildActivity(ctx, expr, TAKjsonfetch, "XmlFetch", childActivity);
  4173. case no_flat:
  4174. case no_thor:
  4175. return info.buildActivity(ctx, expr, TAKfetch, "Fetch", childActivity);
  4176. }
  4177. throwError1(HQLERR_FetchNotSupportMode, getOpString(kind));
  4178. return NULL;
  4179. }