12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733173417351736173717381739174017411742174317441745174617471748174917501751175217531754175517561757175817591760176117621763176417651766176717681769177017711772177317741775177617771778177917801781178217831784178517861787178817891790179117921793179417951796179717981799180018011802180318041805180618071808180918101811181218131814181518161817181818191820182118221823182418251826182718281829183018311832183318341835183618371838183918401841184218431844184518461847184818491850185118521853185418551856185718581859186018611862186318641865186618671868186918701871187218731874187518761877187818791880188118821883188418851886188718881889189018911892189318941895189618971898189919001901190219031904190519061907190819091910191119121913191419151916191719181919192019211922192319241925192619271928192919301931193219331934193519361937193819391940194119421943194419451946194719481949195019511952195319541955195619571958195919601961196219631964196519661967196819691970197119721973197419751976197719781979198019811982198319841985198619871988198919901991199219931994199519961997199819992000200120022003200420052006200720082009201020112012201320142015201620172018201920202021202220232024202520262027202820292030203120322033203420352036203720382039204020412042204320442045204620472048204920502051205220532054205520562057205820592060206120622063206420652066206720682069207020712072207320742075207620772078207920802081208220832084208520862087208820892090209120922093209420952096209720982099210021012102210321042105210621072108210921102111211221132114211521162117211821192120212121222123212421252126212721282129213021312132213321342135213621372138213921402141214221432144214521462147214821492150215121522153215421552156215721582159216021612162216321642165216621672168216921702171217221732174217521762177217821792180218121822183218421852186218721882189219021912192219321942195219621972198219922002201220222032204220522062207220822092210221122122213221422152216221722182219222022212222222322242225222622272228222922302231223222332234223522362237223822392240224122422243224422452246224722482249225022512252225322542255225622572258225922602261226222632264226522662267226822692270227122722273227422752276227722782279228022812282228322842285228622872288228922902291229222932294229522962297229822992300230123022303230423052306230723082309231023112312231323142315231623172318231923202321232223232324232523262327232823292330233123322333233423352336233723382339234023412342234323442345234623472348234923502351235223532354235523562357235823592360236123622363236423652366236723682369237023712372237323742375237623772378237923802381238223832384238523862387238823892390239123922393239423952396239723982399240024012402240324042405240624072408240924102411241224132414241524162417241824192420242124222423242424252426242724282429243024312432243324342435243624372438243924402441244224432444244524462447244824492450245124522453245424552456245724582459246024612462246324642465246624672468246924702471247224732474247524762477247824792480248124822483248424852486248724882489249024912492249324942495249624972498249925002501250225032504250525062507250825092510251125122513251425152516251725182519252025212522252325242525252625272528252925302531253225332534253525362537253825392540254125422543254425452546254725482549255025512552255325542555255625572558255925602561256225632564256525662567256825692570257125722573257425752576257725782579258025812582258325842585258625872588258925902591259225932594259525962597259825992600260126022603260426052606260726082609261026112612261326142615261626172618261926202621262226232624262526262627262826292630263126322633263426352636263726382639264026412642264326442645264626472648264926502651265226532654265526562657265826592660266126622663266426652666266726682669267026712672267326742675267626772678267926802681268226832684268526862687268826892690269126922693269426952696269726982699270027012702270327042705270627072708270927102711271227132714271527162717271827192720272127222723272427252726272727282729273027312732273327342735273627372738273927402741274227432744274527462747274827492750275127522753275427552756275727582759276027612762276327642765276627672768276927702771277227732774277527762777277827792780278127822783278427852786278727882789279027912792279327942795279627972798279928002801280228032804280528062807280828092810281128122813281428152816281728182819282028212822282328242825282628272828282928302831283228332834283528362837283828392840284128422843284428452846284728482849285028512852285328542855285628572858285928602861286228632864286528662867286828692870287128722873287428752876287728782879288028812882288328842885288628872888288928902891289228932894289528962897289828992900290129022903290429052906290729082909291029112912291329142915291629172918291929202921292229232924292529262927292829292930293129322933293429352936293729382939294029412942294329442945294629472948294929502951295229532954295529562957295829592960296129622963296429652966296729682969297029712972297329742975297629772978297929802981298229832984298529862987298829892990299129922993299429952996299729982999300030013002300330043005300630073008300930103011301230133014301530163017301830193020302130223023302430253026302730283029303030313032303330343035303630373038303930403041304230433044304530463047304830493050305130523053305430553056305730583059306030613062306330643065306630673068306930703071307230733074307530763077307830793080308130823083308430853086308730883089309030913092309330943095309630973098309931003101310231033104310531063107310831093110311131123113311431153116311731183119312031213122312331243125312631273128312931303131313231333134313531363137313831393140314131423143314431453146314731483149315031513152315331543155315631573158315931603161316231633164316531663167316831693170317131723173317431753176317731783179318031813182318331843185318631873188318931903191319231933194319531963197319831993200320132023203320432053206320732083209321032113212321332143215321632173218321932203221322232233224322532263227322832293230323132323233323432353236323732383239324032413242324332443245324632473248324932503251325232533254325532563257325832593260326132623263326432653266326732683269327032713272327332743275327632773278327932803281328232833284328532863287328832893290329132923293329432953296329732983299330033013302330333043305330633073308330933103311331233133314331533163317331833193320332133223323332433253326332733283329333033313332333333343335333633373338333933403341334233433344334533463347334833493350335133523353335433553356335733583359336033613362336333643365336633673368336933703371337233733374337533763377337833793380338133823383338433853386338733883389339033913392339333943395339633973398339934003401340234033404340534063407340834093410341134123413341434153416341734183419342034213422342334243425342634273428342934303431343234333434343534363437343834393440344134423443344434453446344734483449345034513452345334543455345634573458345934603461346234633464346534663467346834693470347134723473347434753476347734783479348034813482348334843485348634873488348934903491349234933494349534963497349834993500350135023503350435053506350735083509351035113512351335143515351635173518351935203521352235233524352535263527352835293530353135323533353435353536353735383539354035413542354335443545354635473548354935503551355235533554355535563557355835593560356135623563356435653566356735683569357035713572357335743575357635773578357935803581358235833584358535863587358835893590359135923593359435953596359735983599360036013602360336043605360636073608360936103611361236133614361536163617361836193620362136223623362436253626362736283629363036313632363336343635363636373638363936403641364236433644364536463647364836493650365136523653365436553656365736583659366036613662366336643665366636673668366936703671367236733674367536763677367836793680368136823683368436853686368736883689369036913692369336943695369636973698369937003701370237033704370537063707370837093710371137123713371437153716371737183719372037213722372337243725372637273728372937303731373237333734373537363737373837393740374137423743374437453746374737483749375037513752375337543755375637573758375937603761376237633764376537663767376837693770377137723773377437753776377737783779378037813782378337843785378637873788378937903791379237933794379537963797379837993800380138023803380438053806380738083809381038113812381338143815381638173818381938203821382238233824382538263827382838293830383138323833383438353836383738383839384038413842384338443845384638473848384938503851385238533854385538563857385838593860386138623863386438653866386738683869387038713872387338743875387638773878387938803881388238833884388538863887388838893890389138923893389438953896389738983899390039013902390339043905390639073908390939103911391239133914391539163917391839193920392139223923392439253926392739283929393039313932393339343935393639373938393939403941394239433944394539463947394839493950395139523953395439553956395739583959396039613962396339643965396639673968396939703971397239733974397539763977397839793980398139823983398439853986398739883989399039913992399339943995399639973998399940004001400240034004400540064007400840094010401140124013401440154016401740184019402040214022402340244025402640274028402940304031403240334034403540364037403840394040404140424043404440454046404740484049405040514052405340544055405640574058405940604061406240634064406540664067406840694070407140724073407440754076407740784079408040814082408340844085408640874088408940904091409240934094409540964097409840994100410141024103410441054106410741084109411041114112411341144115411641174118411941204121412241234124412541264127412841294130413141324133413441354136413741384139414041414142414341444145414641474148414941504151415241534154415541564157415841594160416141624163416441654166416741684169417041714172417341744175417641774178417941804181418241834184418541864187418841894190419141924193419441954196419741984199420042014202420342044205420642074208420942104211421242134214421542164217421842194220422142224223422442254226422742284229423042314232423342344235423642374238423942404241424242434244424542464247424842494250425142524253425442554256425742584259426042614262426342644265426642674268426942704271427242734274427542764277427842794280428142824283428442854286428742884289429042914292429342944295429642974298429943004301430243034304430543064307430843094310431143124313431443154316431743184319432043214322432343244325432643274328432943304331433243334334433543364337433843394340434143424343434443454346434743484349435043514352435343544355435643574358435943604361436243634364436543664367436843694370437143724373437443754376437743784379438043814382438343844385438643874388438943904391439243934394439543964397439843994400440144024403440444054406440744084409441044114412441344144415441644174418441944204421442244234424442544264427442844294430443144324433443444354436443744384439444044414442444344444445444644474448444944504451445244534454445544564457445844594460446144624463446444654466446744684469447044714472447344744475447644774478447944804481448244834484448544864487448844894490449144924493449444954496449744984499450045014502450345044505450645074508450945104511451245134514451545164517451845194520452145224523452445254526452745284529453045314532453345344535453645374538453945404541454245434544454545464547454845494550455145524553455445554556455745584559456045614562456345644565456645674568456945704571457245734574457545764577457845794580458145824583458445854586458745884589459045914592459345944595459645974598459946004601460246034604460546064607460846094610461146124613461446154616461746184619462046214622462346244625462646274628462946304631463246334634463546364637463846394640464146424643464446454646464746484649465046514652465346544655465646574658465946604661466246634664466546664667466846694670467146724673467446754676467746784679468046814682468346844685468646874688468946904691469246934694469546964697469846994700470147024703470447054706470747084709471047114712471347144715471647174718471947204721472247234724472547264727472847294730473147324733473447354736473747384739474047414742474347444745474647474748474947504751475247534754475547564757475847594760476147624763476447654766476747684769477047714772477347744775477647774778477947804781478247834784478547864787478847894790479147924793479447954796479747984799480048014802480348044805480648074808480948104811481248134814481548164817481848194820482148224823482448254826482748284829483048314832483348344835483648374838483948404841484248434844484548464847484848494850485148524853485448554856485748584859486048614862486348644865486648674868486948704871487248734874487548764877487848794880488148824883488448854886488748884889489048914892489348944895489648974898489949004901490249034904490549064907490849094910491149124913491449154916491749184919492049214922492349244925492649274928492949304931493249334934493549364937493849394940494149424943494449454946494749484949495049514952495349544955495649574958495949604961496249634964496549664967496849694970497149724973497449754976497749784979498049814982498349844985498649874988498949904991499249934994499549964997499849995000500150025003500450055006500750085009501050115012501350145015501650175018501950205021502250235024502550265027502850295030503150325033503450355036503750385039504050415042504350445045504650475048504950505051505250535054505550565057505850595060506150625063506450655066506750685069507050715072507350745075507650775078507950805081508250835084508550865087508850895090509150925093509450955096509750985099510051015102510351045105510651075108510951105111511251135114511551165117511851195120512151225123512451255126512751285129513051315132513351345135513651375138513951405141514251435144514551465147514851495150515151525153515451555156515751585159516051615162516351645165516651675168516951705171517251735174517551765177517851795180518151825183518451855186518751885189519051915192519351945195519651975198519952005201520252035204520552065207520852095210521152125213521452155216521752185219522052215222522352245225522652275228522952305231523252335234523552365237523852395240524152425243524452455246524752485249525052515252525352545255525652575258525952605261526252635264526552665267526852695270527152725273527452755276527752785279528052815282528352845285528652875288528952905291529252935294529552965297529852995300530153025303530453055306530753085309531053115312531353145315531653175318531953205321532253235324532553265327532853295330533153325333533453355336533753385339534053415342534353445345534653475348534953505351535253535354535553565357535853595360536153625363536453655366536753685369537053715372537353745375537653775378537953805381538253835384538553865387538853895390539153925393539453955396539753985399540054015402540354045405540654075408540954105411541254135414541554165417541854195420542154225423542454255426542754285429543054315432543354345435543654375438543954405441544254435444544554465447544854495450545154525453545454555456545754585459546054615462546354645465546654675468546954705471547254735474547554765477547854795480548154825483548454855486548754885489549054915492549354945495549654975498549955005501550255035504550555065507550855095510551155125513551455155516551755185519552055215522552355245525552655275528552955305531553255335534553555365537553855395540554155425543554455455546554755485549555055515552555355545555555655575558555955605561556255635564556555665567556855695570557155725573557455755576557755785579558055815582558355845585558655875588558955905591559255935594559555965597559855995600560156025603560456055606560756085609561056115612561356145615561656175618561956205621562256235624562556265627562856295630563156325633563456355636563756385639564056415642564356445645564656475648564956505651565256535654565556565657565856595660566156625663566456655666566756685669567056715672567356745675567656775678567956805681568256835684568556865687568856895690569156925693569456955696569756985699570057015702570357045705570657075708570957105711571257135714571557165717571857195720572157225723572457255726572757285729573057315732573357345735573657375738573957405741574257435744574557465747574857495750575157525753575457555756575757585759576057615762576357645765576657675768576957705771577257735774577557765777577857795780578157825783578457855786578757885789579057915792579357945795579657975798579958005801580258035804580558065807580858095810581158125813581458155816581758185819582058215822582358245825582658275828582958305831583258335834583558365837583858395840584158425843584458455846584758485849585058515852585358545855585658575858585958605861586258635864586558665867586858695870587158725873587458755876587758785879588058815882588358845885588658875888588958905891589258935894589558965897589858995900590159025903590459055906590759085909591059115912591359145915591659175918591959205921592259235924592559265927592859295930593159325933593459355936593759385939594059415942594359445945594659475948594959505951595259535954595559565957595859595960596159625963596459655966596759685969597059715972597359745975597659775978597959805981598259835984598559865987598859895990599159925993599459955996599759985999600060016002600360046005600660076008600960106011601260136014601560166017601860196020602160226023602460256026602760286029603060316032603360346035603660376038603960406041604260436044604560466047604860496050605160526053605460556056605760586059606060616062606360646065606660676068606960706071607260736074607560766077607860796080608160826083608460856086608760886089609060916092609360946095609660976098609961006101610261036104610561066107610861096110611161126113611461156116611761186119612061216122612361246125612661276128612961306131613261336134613561366137613861396140614161426143614461456146614761486149615061516152615361546155615661576158615961606161616261636164616561666167616861696170617161726173617461756176617761786179618061816182618361846185618661876188618961906191619261936194619561966197619861996200620162026203620462056206620762086209621062116212621362146215621662176218621962206221622262236224622562266227622862296230623162326233623462356236623762386239624062416242624362446245624662476248624962506251625262536254625562566257625862596260626162626263626462656266626762686269627062716272627362746275627662776278627962806281628262836284628562866287628862896290629162926293629462956296629762986299630063016302630363046305630663076308630963106311631263136314631563166317631863196320632163226323632463256326632763286329633063316332633363346335633663376338633963406341634263436344634563466347634863496350635163526353635463556356635763586359636063616362636363646365636663676368636963706371637263736374637563766377637863796380638163826383638463856386638763886389639063916392639363946395639663976398639964006401640264036404640564066407640864096410641164126413641464156416641764186419642064216422642364246425642664276428642964306431643264336434643564366437643864396440644164426443644464456446644764486449645064516452645364546455645664576458645964606461646264636464646564666467646864696470647164726473647464756476647764786479648064816482648364846485648664876488648964906491649264936494649564966497649864996500650165026503650465056506650765086509651065116512651365146515651665176518651965206521652265236524652565266527652865296530653165326533653465356536653765386539654065416542654365446545654665476548654965506551655265536554655565566557655865596560656165626563656465656566656765686569657065716572657365746575657665776578657965806581658265836584658565866587658865896590659165926593659465956596659765986599660066016602660366046605660666076608660966106611661266136614661566166617661866196620662166226623662466256626662766286629663066316632663366346635663666376638663966406641664266436644664566466647664866496650665166526653665466556656665766586659666066616662666366646665666666676668666966706671667266736674667566766677667866796680668166826683668466856686668766886689669066916692669366946695669666976698669967006701670267036704670567066707670867096710671167126713671467156716671767186719672067216722672367246725672667276728672967306731673267336734673567366737673867396740674167426743674467456746674767486749675067516752675367546755675667576758675967606761676267636764676567666767676867696770677167726773677467756776677767786779678067816782678367846785678667876788678967906791679267936794679567966797679867996800680168026803680468056806680768086809681068116812681368146815681668176818681968206821682268236824682568266827682868296830683168326833683468356836683768386839684068416842684368446845684668476848684968506851685268536854685568566857685868596860686168626863686468656866686768686869687068716872687368746875687668776878687968806881688268836884688568866887688868896890689168926893689468956896689768986899690069016902690369046905690669076908690969106911691269136914691569166917691869196920692169226923692469256926692769286929693069316932693369346935693669376938693969406941694269436944694569466947694869496950695169526953695469556956695769586959696069616962696369646965696669676968696969706971697269736974697569766977697869796980698169826983698469856986698769886989699069916992699369946995699669976998699970007001700270037004700570067007700870097010701170127013701470157016701770187019702070217022702370247025702670277028702970307031703270337034703570367037703870397040704170427043704470457046704770487049705070517052705370547055705670577058705970607061706270637064706570667067706870697070707170727073707470757076707770787079708070817082708370847085708670877088708970907091709270937094709570967097709870997100710171027103710471057106710771087109711071117112711371147115711671177118711971207121712271237124712571267127712871297130713171327133713471357136713771387139714071417142714371447145714671477148714971507151715271537154715571567157715871597160716171627163716471657166716771687169717071717172717371747175717671777178717971807181718271837184718571867187718871897190719171927193719471957196719771987199720072017202720372047205720672077208720972107211721272137214721572167217721872197220722172227223722472257226722772287229723072317232723372347235723672377238723972407241724272437244724572467247724872497250725172527253725472557256725772587259726072617262726372647265726672677268726972707271727272737274727572767277727872797280728172827283728472857286728772887289729072917292729372947295729672977298729973007301730273037304730573067307730873097310731173127313731473157316731773187319732073217322732373247325732673277328732973307331733273337334733573367337733873397340734173427343734473457346734773487349735073517352735373547355735673577358735973607361736273637364736573667367736873697370737173727373737473757376737773787379738073817382738373847385738673877388738973907391739273937394739573967397739873997400740174027403740474057406740774087409741074117412741374147415741674177418741974207421742274237424742574267427742874297430743174327433743474357436743774387439744074417442744374447445744674477448744974507451745274537454745574567457745874597460746174627463746474657466746774687469747074717472747374747475747674777478747974807481748274837484748574867487748874897490749174927493749474957496749774987499750075017502750375047505750675077508750975107511751275137514751575167517751875197520752175227523752475257526752775287529753075317532753375347535753675377538753975407541754275437544754575467547754875497550755175527553755475557556755775587559756075617562756375647565756675677568756975707571757275737574757575767577757875797580758175827583758475857586758775887589759075917592759375947595759675977598759976007601760276037604760576067607760876097610761176127613761476157616761776187619762076217622762376247625762676277628762976307631763276337634763576367637763876397640764176427643764476457646764776487649765076517652765376547655765676577658765976607661766276637664766576667667766876697670767176727673767476757676767776787679768076817682768376847685768676877688768976907691769276937694769576967697769876997700770177027703770477057706770777087709771077117712771377147715771677177718771977207721772277237724772577267727772877297730773177327733773477357736773777387739774077417742774377447745774677477748774977507751775277537754775577567757775877597760776177627763776477657766776777687769777077717772777377747775777677777778777977807781778277837784778577867787778877897790779177927793779477957796779777987799780078017802780378047805780678077808780978107811781278137814781578167817781878197820782178227823782478257826782778287829783078317832783378347835783678377838783978407841784278437844784578467847784878497850785178527853785478557856785778587859786078617862786378647865786678677868786978707871787278737874787578767877787878797880788178827883788478857886788778887889789078917892789378947895789678977898789979007901790279037904790579067907790879097910791179127913791479157916791779187919792079217922792379247925792679277928792979307931793279337934793579367937793879397940794179427943794479457946794779487949795079517952795379547955795679577958795979607961796279637964796579667967796879697970797179727973797479757976797779787979798079817982798379847985798679877988798979907991799279937994799579967997799879998000800180028003800480058006800780088009801080118012801380148015801680178018801980208021802280238024802580268027802880298030803180328033803480358036803780388039804080418042804380448045804680478048804980508051805280538054805580568057805880598060806180628063806480658066806780688069807080718072807380748075807680778078807980808081808280838084808580868087808880898090809180928093809480958096809780988099810081018102810381048105810681078108810981108111811281138114811581168117811881198120812181228123812481258126812781288129813081318132813381348135813681378138813981408141814281438144814581468147814881498150815181528153815481558156815781588159816081618162816381648165816681678168816981708171817281738174817581768177817881798180818181828183818481858186818781888189819081918192819381948195819681978198819982008201820282038204820582068207820882098210821182128213821482158216821782188219822082218222822382248225822682278228822982308231823282338234823582368237823882398240824182428243824482458246824782488249825082518252825382548255825682578258825982608261826282638264826582668267826882698270827182728273827482758276827782788279828082818282828382848285828682878288828982908291829282938294829582968297829882998300830183028303830483058306830783088309831083118312831383148315831683178318831983208321832283238324832583268327832883298330833183328333833483358336833783388339834083418342834383448345834683478348834983508351835283538354835583568357835883598360836183628363836483658366836783688369837083718372837383748375837683778378837983808381838283838384838583868387838883898390839183928393839483958396839783988399840084018402840384048405840684078408840984108411841284138414841584168417841884198420842184228423842484258426842784288429843084318432843384348435843684378438843984408441844284438444844584468447844884498450845184528453845484558456845784588459846084618462846384648465846684678468846984708471847284738474847584768477847884798480848184828483848484858486848784888489849084918492849384948495849684978498849985008501850285038504850585068507850885098510851185128513851485158516851785188519852085218522852385248525852685278528852985308531853285338534853585368537853885398540854185428543854485458546854785488549855085518552855385548555855685578558855985608561856285638564856585668567856885698570857185728573857485758576857785788579858085818582858385848585858685878588858985908591859285938594859585968597859885998600860186028603860486058606860786088609861086118612861386148615861686178618861986208621862286238624862586268627862886298630863186328633863486358636863786388639864086418642864386448645864686478648864986508651865286538654865586568657865886598660866186628663866486658666866786688669867086718672867386748675867686778678867986808681868286838684868586868687868886898690869186928693869486958696869786988699870087018702870387048705870687078708870987108711871287138714871587168717871887198720872187228723872487258726872787288729873087318732873387348735873687378738873987408741874287438744874587468747874887498750875187528753875487558756875787588759876087618762876387648765876687678768876987708771877287738774877587768777877887798780878187828783878487858786878787888789879087918792879387948795879687978798879988008801880288038804880588068807880888098810881188128813881488158816881788188819882088218822882388248825882688278828882988308831883288338834883588368837883888398840884188428843884488458846884788488849885088518852885388548855885688578858885988608861886288638864886588668867886888698870887188728873887488758876887788788879888088818882888388848885888688878888888988908891889288938894889588968897889888998900890189028903890489058906890789088909891089118912891389148915891689178918891989208921892289238924892589268927892889298930893189328933893489358936893789388939894089418942894389448945894689478948894989508951895289538954895589568957895889598960896189628963896489658966896789688969897089718972897389748975897689778978897989808981898289838984898589868987898889898990899189928993899489958996899789988999900090019002900390049005900690079008900990109011901290139014901590169017901890199020902190229023902490259026902790289029903090319032903390349035903690379038903990409041904290439044904590469047904890499050905190529053905490559056905790589059906090619062906390649065906690679068906990709071907290739074907590769077907890799080908190829083908490859086908790889089909090919092909390949095909690979098909991009101910291039104910591069107910891099110911191129113911491159116911791189119912091219122912391249125912691279128912991309131913291339134913591369137913891399140914191429143914491459146914791489149915091519152915391549155915691579158915991609161916291639164916591669167916891699170917191729173917491759176917791789179918091819182918391849185918691879188918991909191919291939194919591969197919891999200920192029203920492059206920792089209921092119212921392149215921692179218921992209221922292239224922592269227922892299230923192329233923492359236923792389239924092419242924392449245924692479248924992509251925292539254925592569257925892599260926192629263926492659266926792689269927092719272927392749275927692779278927992809281928292839284928592869287928892899290929192929293929492959296929792989299930093019302930393049305930693079308930993109311931293139314931593169317931893199320932193229323932493259326932793289329933093319332933393349335933693379338933993409341934293439344934593469347934893499350935193529353935493559356935793589359936093619362936393649365936693679368936993709371937293739374937593769377937893799380938193829383938493859386938793889389939093919392939393949395939693979398939994009401940294039404940594069407940894099410941194129413941494159416941794189419942094219422942394249425942694279428942994309431943294339434943594369437943894399440944194429443944494459446944794489449945094519452945394549455945694579458945994609461946294639464946594669467946894699470947194729473947494759476947794789479948094819482948394849485948694879488948994909491949294939494949594969497949894999500950195029503950495059506950795089509951095119512951395149515951695179518951995209521952295239524952595269527952895299530953195329533953495359536953795389539954095419542954395449545954695479548954995509551955295539554955595569557955895599560956195629563956495659566956795689569957095719572957395749575957695779578957995809581958295839584958595869587958895899590959195929593959495959596959795989599960096019602960396049605960696079608960996109611961296139614961596169617961896199620962196229623962496259626962796289629963096319632963396349635963696379638963996409641964296439644964596469647964896499650965196529653965496559656965796589659966096619662966396649665966696679668966996709671967296739674967596769677967896799680968196829683968496859686968796889689969096919692969396949695969696979698969997009701970297039704970597069707970897099710971197129713971497159716971797189719972097219722972397249725972697279728972997309731973297339734973597369737973897399740974197429743974497459746974797489749975097519752975397549755975697579758975997609761976297639764976597669767976897699770977197729773977497759776977797789779978097819782978397849785978697879788978997909791979297939794979597969797979897999800980198029803980498059806980798089809981098119812981398149815981698179818981998209821982298239824982598269827982898299830983198329833983498359836983798389839984098419842984398449845984698479848984998509851985298539854985598569857985898599860986198629863986498659866986798689869987098719872987398749875987698779878987998809881988298839884988598869887988898899890989198929893989498959896989798989899990099019902990399049905990699079908990999109911991299139914991599169917991899199920992199229923992499259926992799289929993099319932993399349935993699379938993999409941994299439944994599469947994899499950995199529953995499559956995799589959996099619962996399649965996699679968996999709971997299739974997599769977997899799980998199829983998499859986998799889989999099919992999399949995999699979998999910000100011000210003100041000510006100071000810009100101001110012100131001410015100161001710018100191002010021100221002310024100251002610027100281002910030100311003210033100341003510036100371003810039100401004110042100431004410045100461004710048100491005010051100521005310054100551005610057100581005910060100611006210063100641006510066100671006810069100701007110072100731007410075100761007710078100791008010081100821008310084100851008610087100881008910090100911009210093100941009510096100971009810099101001010110102101031010410105101061010710108101091011010111101121011310114101151011610117101181011910120101211012210123101241012510126101271012810129101301013110132101331013410135101361013710138101391014010141101421014310144101451014610147101481014910150101511015210153101541015510156101571015810159101601016110162101631016410165101661016710168101691017010171101721017310174101751017610177101781017910180101811018210183101841018510186101871018810189101901019110192101931019410195101961019710198101991020010201102021020310204102051020610207102081020910210102111021210213102141021510216102171021810219102201022110222102231022410225102261022710228102291023010231102321023310234102351023610237102381023910240102411024210243102441024510246102471024810249102501025110252102531025410255102561025710258102591026010261102621026310264102651026610267102681026910270102711027210273102741027510276102771027810279102801028110282102831028410285102861028710288102891029010291102921029310294102951029610297102981029910300103011030210303103041030510306103071030810309103101031110312103131031410315103161031710318103191032010321103221032310324103251032610327103281032910330103311033210333103341033510336103371033810339103401034110342103431034410345103461034710348103491035010351103521035310354103551035610357103581035910360103611036210363103641036510366103671036810369103701037110372103731037410375103761037710378103791038010381103821038310384103851038610387103881038910390103911039210393103941039510396103971039810399104001040110402104031040410405104061040710408104091041010411104121041310414104151041610417104181041910420104211042210423104241042510426104271042810429104301043110432104331043410435104361043710438104391044010441104421044310444104451044610447104481044910450104511045210453104541045510456104571045810459104601046110462104631046410465104661046710468104691047010471104721047310474104751047610477104781047910480104811048210483104841048510486104871048810489104901049110492104931049410495104961049710498104991050010501105021050310504105051050610507105081050910510105111051210513105141051510516105171051810519105201052110522105231052410525105261052710528105291053010531105321053310534105351053610537105381053910540105411054210543105441054510546105471054810549105501055110552105531055410555105561055710558105591056010561105621056310564105651056610567105681056910570105711057210573105741057510576105771057810579105801058110582105831058410585105861058710588105891059010591105921059310594105951059610597105981059910600106011060210603106041060510606106071060810609106101061110612106131061410615106161061710618106191062010621106221062310624106251062610627106281062910630106311063210633106341063510636106371063810639106401064110642106431064410645106461064710648106491065010651106521065310654106551065610657106581065910660106611066210663106641066510666106671066810669106701067110672106731067410675106761067710678106791068010681106821068310684106851068610687106881068910690106911069210693106941069510696106971069810699107001070110702107031070410705107061070710708107091071010711107121071310714107151071610717107181071910720107211072210723107241072510726107271072810729107301073110732107331073410735107361073710738107391074010741107421074310744107451074610747107481074910750107511075210753107541075510756107571075810759107601076110762107631076410765107661076710768107691077010771107721077310774107751077610777107781077910780107811078210783107841078510786107871078810789107901079110792107931079410795107961079710798107991080010801108021080310804108051080610807108081080910810108111081210813108141081510816108171081810819108201082110822108231082410825108261082710828108291083010831108321083310834108351083610837108381083910840108411084210843108441084510846108471084810849108501085110852108531085410855108561085710858108591086010861108621086310864108651086610867108681086910870108711087210873108741087510876108771087810879108801088110882108831088410885108861088710888108891089010891108921089310894108951089610897108981089910900109011090210903109041090510906109071090810909109101091110912109131091410915109161091710918109191092010921109221092310924109251092610927109281092910930109311093210933109341093510936109371093810939109401094110942109431094410945109461094710948109491095010951109521095310954109551095610957109581095910960109611096210963109641096510966109671096810969109701097110972109731097410975109761097710978109791098010981109821098310984109851098610987109881098910990109911099210993109941099510996109971099810999110001100111002110031100411005110061100711008110091101011011110121101311014110151101611017110181101911020110211102211023110241102511026110271102811029110301103111032110331103411035110361103711038110391104011041110421104311044110451104611047110481104911050110511105211053110541105511056110571105811059110601106111062110631106411065110661106711068110691107011071110721107311074110751107611077110781107911080110811108211083110841108511086110871108811089110901109111092110931109411095110961109711098110991110011101111021110311104111051110611107111081110911110111111111211113111141111511116111171111811119111201112111122111231112411125111261112711128111291113011131111321113311134111351113611137111381113911140111411114211143111441114511146111471114811149111501115111152111531115411155111561115711158111591116011161111621116311164111651116611167111681116911170111711117211173111741117511176111771117811179111801118111182111831118411185111861118711188111891119011191111921119311194111951119611197111981119911200112011120211203112041120511206112071120811209112101121111212112131121411215112161121711218112191122011221112221122311224112251122611227112281122911230112311123211233112341123511236112371123811239112401124111242112431124411245112461124711248112491125011251112521125311254112551125611257112581125911260112611126211263112641126511266112671126811269112701127111272112731127411275112761127711278112791128011281112821128311284112851128611287112881128911290112911129211293112941129511296112971129811299113001130111302113031130411305113061130711308113091131011311113121131311314113151131611317113181131911320113211132211323113241132511326113271132811329113301133111332113331133411335113361133711338113391134011341113421134311344113451134611347113481134911350113511135211353113541135511356113571135811359113601136111362113631136411365113661136711368113691137011371113721137311374113751137611377113781137911380113811138211383113841138511386113871138811389113901139111392113931139411395113961139711398113991140011401114021140311404114051140611407114081140911410114111141211413114141141511416114171141811419114201142111422114231142411425114261142711428114291143011431114321143311434114351143611437114381143911440114411144211443114441144511446114471144811449114501145111452114531145411455114561145711458114591146011461114621146311464114651146611467114681146911470114711147211473114741147511476114771147811479114801148111482114831148411485114861148711488114891149011491114921149311494114951149611497114981149911500115011150211503115041150511506115071150811509115101151111512115131151411515115161151711518115191152011521115221152311524115251152611527115281152911530115311153211533115341153511536115371153811539115401154111542115431154411545115461154711548115491155011551115521155311554115551155611557115581155911560115611156211563115641156511566115671156811569115701157111572115731157411575115761157711578115791158011581115821158311584115851158611587115881158911590115911159211593115941159511596115971159811599116001160111602116031160411605116061160711608116091161011611116121161311614116151161611617116181161911620116211162211623116241162511626116271162811629116301163111632116331163411635116361163711638116391164011641116421164311644116451164611647116481164911650116511165211653116541165511656116571165811659116601166111662116631166411665116661166711668116691167011671116721167311674116751167611677116781167911680116811168211683116841168511686116871168811689116901169111692116931169411695116961169711698116991170011701117021170311704117051170611707117081170911710117111171211713117141171511716117171171811719117201172111722117231172411725117261172711728117291173011731117321173311734117351173611737117381173911740117411174211743117441174511746117471174811749117501175111752117531175411755117561175711758117591176011761117621176311764117651176611767117681176911770117711177211773117741177511776117771177811779117801178111782117831178411785117861178711788117891179011791117921179311794117951179611797117981179911800118011180211803118041180511806118071180811809118101181111812118131181411815118161181711818118191182011821118221182311824118251182611827118281182911830118311183211833118341183511836118371183811839118401184111842118431184411845118461184711848118491185011851118521185311854118551185611857118581185911860118611186211863118641186511866118671186811869118701187111872118731187411875118761187711878118791188011881118821188311884118851188611887118881188911890118911189211893118941189511896118971189811899119001190111902119031190411905119061190711908119091191011911119121191311914119151191611917119181191911920119211192211923119241192511926119271192811929119301193111932119331193411935119361193711938119391194011941119421194311944119451194611947119481194911950119511195211953119541195511956119571195811959119601196111962119631196411965119661196711968119691197011971119721197311974119751197611977119781197911980119811198211983119841198511986119871198811989119901199111992119931199411995119961199711998119991200012001120021200312004120051200612007120081200912010120111201212013120141201512016120171201812019120201202112022120231202412025120261202712028120291203012031120321203312034120351203612037120381203912040120411204212043120441204512046120471204812049120501205112052120531205412055120561205712058120591206012061120621206312064120651206612067120681206912070120711207212073120741207512076120771207812079120801208112082120831208412085120861208712088120891209012091120921209312094120951209612097120981209912100121011210212103121041210512106121071210812109121101211112112121131211412115121161211712118121191212012121121221212312124121251212612127121281212912130121311213212133121341213512136121371213812139121401214112142121431214412145121461214712148121491215012151121521215312154121551215612157121581215912160121611216212163121641216512166121671216812169121701217112172121731217412175121761217712178121791218012181121821218312184121851218612187121881218912190121911219212193121941219512196121971219812199122001220112202122031220412205122061220712208122091221012211122121221312214122151221612217122181221912220122211222212223122241222512226122271222812229122301223112232122331223412235122361223712238122391224012241122421224312244122451224612247122481224912250122511225212253122541225512256122571225812259122601226112262122631226412265122661226712268122691227012271122721227312274122751227612277122781227912280122811228212283122841228512286122871228812289122901229112292122931229412295122961229712298122991230012301123021230312304123051230612307123081230912310123111231212313123141231512316123171231812319123201232112322123231232412325123261232712328123291233012331123321233312334123351233612337123381233912340123411234212343123441234512346123471234812349123501235112352123531235412355123561235712358123591236012361123621236312364123651236612367123681236912370123711237212373123741237512376123771237812379123801238112382123831238412385123861238712388123891239012391123921239312394123951239612397123981239912400124011240212403124041240512406124071240812409124101241112412124131241412415124161241712418124191242012421124221242312424124251242612427124281242912430124311243212433124341243512436124371243812439124401244112442124431244412445124461244712448124491245012451124521245312454124551245612457124581245912460124611246212463124641246512466124671246812469124701247112472124731247412475124761247712478124791248012481124821248312484124851248612487124881248912490124911249212493124941249512496124971249812499125001250112502125031250412505125061250712508125091251012511125121251312514125151251612517125181251912520125211252212523125241252512526125271252812529125301253112532125331253412535125361253712538125391254012541125421254312544125451254612547125481254912550125511255212553125541255512556125571255812559125601256112562125631256412565125661256712568125691257012571125721257312574125751257612577125781257912580125811258212583125841258512586125871258812589125901259112592125931259412595125961259712598125991260012601126021260312604126051260612607126081260912610126111261212613126141261512616126171261812619126201262112622126231262412625126261262712628126291263012631126321263312634126351263612637126381263912640126411264212643126441264512646126471264812649126501265112652126531265412655126561265712658126591266012661126621266312664126651266612667126681266912670126711267212673126741267512676126771267812679126801268112682126831268412685126861268712688126891269012691126921269312694126951269612697126981269912700127011270212703127041270512706127071270812709127101271112712127131271412715127161271712718127191272012721127221272312724127251272612727127281272912730127311273212733127341273512736127371273812739127401274112742127431274412745127461274712748127491275012751127521275312754127551275612757127581275912760127611276212763127641276512766127671276812769127701277112772127731277412775127761277712778127791278012781127821278312784127851278612787127881278912790127911279212793127941279512796127971279812799128001280112802128031280412805128061280712808128091281012811128121281312814128151281612817128181281912820128211282212823128241282512826128271282812829128301283112832128331283412835128361283712838128391284012841128421284312844128451284612847128481284912850128511285212853128541285512856128571285812859128601286112862128631286412865128661286712868128691287012871128721287312874128751287612877128781287912880128811288212883128841288512886128871288812889128901289112892128931289412895128961289712898128991290012901129021290312904129051290612907129081290912910129111291212913129141291512916129171291812919129201292112922129231292412925129261292712928129291293012931129321293312934129351293612937129381293912940129411294212943129441294512946129471294812949129501295112952129531295412955129561295712958129591296012961129621296312964129651296612967129681296912970129711297212973129741297512976129771297812979129801298112982129831298412985129861298712988129891299012991129921299312994129951299612997129981299913000130011300213003130041300513006130071300813009130101301113012130131301413015130161301713018130191302013021130221302313024130251302613027130281302913030130311303213033130341303513036130371303813039130401304113042130431304413045130461304713048130491305013051130521305313054130551305613057130581305913060130611306213063130641306513066130671306813069130701307113072130731307413075130761307713078130791308013081130821308313084130851308613087130881308913090130911309213093130941309513096130971309813099131001310113102131031310413105131061310713108131091311013111131121311313114131151311613117131181311913120131211312213123131241312513126131271312813129131301313113132131331313413135131361313713138131391314013141131421314313144131451314613147131481314913150131511315213153131541315513156131571315813159131601316113162131631316413165131661316713168131691317013171131721317313174131751317613177131781317913180131811318213183131841318513186131871318813189131901319113192131931319413195131961319713198131991320013201132021320313204132051320613207132081320913210132111321213213132141321513216132171321813219132201322113222132231322413225132261322713228132291323013231132321323313234132351323613237132381323913240132411324213243132441324513246132471324813249132501325113252132531325413255132561325713258132591326013261132621326313264132651326613267132681326913270132711327213273132741327513276132771327813279132801328113282132831328413285132861328713288132891329013291132921329313294132951329613297132981329913300133011330213303133041330513306133071330813309133101331113312133131331413315133161331713318133191332013321133221332313324133251332613327133281332913330133311333213333133341333513336133371333813339133401334113342133431334413345133461334713348133491335013351133521335313354133551335613357133581335913360133611336213363133641336513366133671336813369133701337113372133731337413375133761337713378133791338013381133821338313384133851338613387133881338913390133911339213393133941339513396133971339813399134001340113402134031340413405134061340713408134091341013411134121341313414134151341613417134181341913420134211342213423134241342513426134271342813429134301343113432134331343413435134361343713438134391344013441134421344313444134451344613447134481344913450134511345213453134541345513456134571345813459134601346113462134631346413465134661346713468134691347013471134721347313474134751347613477134781347913480134811348213483134841348513486134871348813489134901349113492134931349413495134961349713498134991350013501135021350313504135051350613507135081350913510135111351213513135141351513516135171351813519135201352113522135231352413525135261352713528135291353013531135321353313534135351353613537135381353913540135411354213543135441354513546135471354813549135501355113552135531355413555135561355713558135591356013561135621356313564135651356613567135681356913570135711357213573135741357513576135771357813579135801358113582135831358413585135861358713588135891359013591135921359313594135951359613597135981359913600136011360213603136041360513606136071360813609136101361113612136131361413615136161361713618136191362013621136221362313624136251362613627136281362913630136311363213633136341363513636136371363813639136401364113642136431364413645136461364713648136491365013651136521365313654136551365613657136581365913660136611366213663136641366513666136671366813669136701367113672136731367413675136761367713678136791368013681136821368313684136851368613687136881368913690136911369213693136941369513696136971369813699137001370113702137031370413705137061370713708137091371013711137121371313714137151371613717137181371913720137211372213723137241372513726137271372813729137301373113732137331373413735137361373713738137391374013741137421374313744137451374613747137481374913750137511375213753137541375513756137571375813759137601376113762137631376413765137661376713768137691377013771137721377313774137751377613777137781377913780137811378213783137841378513786137871378813789137901379113792137931379413795137961379713798137991380013801138021380313804138051380613807138081380913810138111381213813138141381513816138171381813819138201382113822138231382413825138261382713828138291383013831138321383313834138351383613837138381383913840138411384213843138441384513846138471384813849138501385113852138531385413855138561385713858138591386013861138621386313864138651386613867138681386913870138711387213873138741387513876138771387813879138801388113882138831388413885138861388713888138891389013891138921389313894138951389613897138981389913900139011390213903139041390513906139071390813909139101391113912139131391413915139161391713918139191392013921139221392313924139251392613927139281392913930139311393213933139341393513936139371393813939139401394113942139431394413945139461394713948139491395013951139521395313954139551395613957139581395913960139611396213963139641396513966139671396813969139701397113972139731397413975139761397713978139791398013981139821398313984139851398613987139881398913990139911399213993139941399513996139971399813999140001400114002140031400414005140061400714008140091401014011140121401314014140151401614017140181401914020140211402214023140241402514026140271402814029140301403114032140331403414035140361403714038140391404014041140421404314044140451404614047140481404914050140511405214053140541405514056140571405814059140601406114062140631406414065140661406714068140691407014071140721407314074140751407614077140781407914080140811408214083140841408514086140871408814089140901409114092140931409414095140961409714098140991410014101141021410314104141051410614107141081410914110141111411214113141141411514116141171411814119141201412114122141231412414125141261412714128141291413014131141321413314134141351413614137141381413914140141411414214143141441414514146141471414814149141501415114152141531415414155141561415714158141591416014161141621416314164141651416614167141681416914170141711417214173141741417514176141771417814179141801418114182141831418414185141861418714188141891419014191141921419314194141951419614197141981419914200142011420214203142041420514206142071420814209142101421114212142131421414215142161421714218142191422014221142221422314224142251422614227142281422914230142311423214233142341423514236142371423814239142401424114242142431424414245142461424714248142491425014251142521425314254142551425614257142581425914260142611426214263142641426514266142671426814269142701427114272142731427414275142761427714278142791428014281142821428314284142851428614287142881428914290142911429214293142941429514296142971429814299143001430114302143031430414305143061430714308143091431014311143121431314314143151431614317143181431914320143211432214323143241432514326143271432814329143301433114332143331433414335143361433714338143391434014341143421434314344143451434614347143481434914350143511435214353143541435514356143571435814359143601436114362143631436414365143661436714368143691437014371143721437314374143751437614377143781437914380143811438214383143841438514386143871438814389143901439114392143931439414395143961439714398143991440014401144021440314404144051440614407144081440914410144111441214413144141441514416144171441814419144201442114422144231442414425144261442714428144291443014431144321443314434144351443614437144381443914440144411444214443144441444514446144471444814449144501445114452144531445414455144561445714458144591446014461144621446314464144651446614467144681446914470144711447214473144741447514476144771447814479144801448114482144831448414485144861448714488144891449014491144921449314494144951449614497144981449914500145011450214503145041450514506145071450814509145101451114512145131451414515145161451714518145191452014521145221452314524145251452614527145281452914530145311453214533145341453514536145371453814539145401454114542145431454414545145461454714548145491455014551145521455314554145551455614557145581455914560145611456214563145641456514566145671456814569145701457114572145731457414575145761457714578145791458014581145821458314584145851458614587145881458914590145911459214593145941459514596145971459814599146001460114602146031460414605146061460714608146091461014611146121461314614146151461614617146181461914620146211462214623146241462514626146271462814629146301463114632146331463414635146361463714638146391464014641146421464314644146451464614647146481464914650146511465214653146541465514656146571465814659146601466114662146631466414665146661466714668146691467014671146721467314674146751467614677146781467914680146811468214683146841468514686146871468814689146901469114692146931469414695146961469714698146991470014701147021470314704147051470614707147081470914710147111471214713147141471514716147171471814719147201472114722147231472414725147261472714728147291473014731147321473314734147351473614737147381473914740147411474214743147441474514746147471474814749147501475114752147531475414755147561475714758147591476014761147621476314764147651476614767147681476914770147711477214773147741477514776147771477814779147801478114782147831478414785147861478714788147891479014791147921479314794147951479614797147981479914800148011480214803148041480514806148071480814809148101481114812148131481414815148161481714818148191482014821148221482314824148251482614827148281482914830148311483214833148341483514836148371483814839148401484114842148431484414845148461484714848148491485014851148521485314854148551485614857148581485914860148611486214863148641486514866148671486814869148701487114872148731487414875148761487714878148791488014881148821488314884148851488614887148881488914890148911489214893148941489514896148971489814899149001490114902149031490414905149061490714908149091491014911149121491314914149151491614917149181491914920149211492214923149241492514926149271492814929149301493114932149331493414935149361493714938149391494014941149421494314944149451494614947149481494914950149511495214953149541495514956149571495814959149601496114962149631496414965149661496714968149691497014971149721497314974149751497614977149781497914980149811498214983149841498514986149871498814989149901499114992149931499414995149961499714998149991500015001150021500315004150051500615007150081500915010150111501215013150141501515016150171501815019150201502115022150231502415025150261502715028150291503015031150321503315034150351503615037150381503915040150411504215043150441504515046150471504815049150501505115052150531505415055150561505715058150591506015061150621506315064150651506615067150681506915070150711507215073150741507515076150771507815079150801508115082150831508415085150861508715088150891509015091150921509315094150951509615097150981509915100151011510215103151041510515106151071510815109151101511115112151131511415115151161511715118151191512015121151221512315124151251512615127151281512915130151311513215133151341513515136151371513815139151401514115142151431514415145151461514715148151491515015151151521515315154151551515615157151581515915160151611516215163151641516515166151671516815169151701517115172151731517415175151761517715178151791518015181151821518315184151851518615187151881518915190151911519215193151941519515196151971519815199152001520115202152031520415205152061520715208152091521015211152121521315214152151521615217152181521915220152211522215223152241522515226152271522815229152301523115232152331523415235152361523715238152391524015241152421524315244152451524615247152481524915250152511525215253152541525515256152571525815259152601526115262152631526415265152661526715268152691527015271152721527315274152751527615277152781527915280152811528215283152841528515286152871528815289152901529115292152931529415295152961529715298152991530015301153021530315304153051530615307153081530915310153111531215313153141531515316153171531815319153201532115322153231532415325153261532715328153291533015331153321533315334153351533615337153381533915340153411534215343153441534515346153471534815349153501535115352153531535415355153561535715358153591536015361153621536315364153651536615367153681536915370153711537215373153741537515376153771537815379153801538115382153831538415385153861538715388153891539015391153921539315394153951539615397153981539915400154011540215403154041540515406154071540815409154101541115412154131541415415154161541715418154191542015421154221542315424154251542615427154281542915430154311543215433154341543515436154371543815439154401544115442154431544415445154461544715448154491545015451154521545315454154551545615457154581545915460154611546215463154641546515466154671546815469154701547115472154731547415475154761547715478154791548015481154821548315484154851548615487154881548915490154911549215493154941549515496154971549815499155001550115502155031550415505155061550715508155091551015511155121551315514155151551615517155181551915520155211552215523155241552515526155271552815529155301553115532155331553415535155361553715538155391554015541155421554315544155451554615547155481554915550155511555215553155541555515556155571555815559155601556115562155631556415565155661556715568155691557015571155721557315574155751557615577155781557915580155811558215583155841558515586155871558815589155901559115592155931559415595155961559715598155991560015601156021560315604156051560615607156081560915610156111561215613156141561515616156171561815619156201562115622156231562415625156261562715628156291563015631156321563315634156351563615637156381563915640156411564215643156441564515646156471564815649156501565115652156531565415655156561565715658156591566015661156621566315664156651566615667156681566915670156711567215673156741567515676156771567815679156801568115682156831568415685156861568715688156891569015691156921569315694156951569615697156981569915700157011570215703157041570515706157071570815709157101571115712157131571415715157161571715718157191572015721157221572315724157251572615727157281572915730157311573215733157341573515736157371573815739157401574115742157431574415745157461574715748157491575015751157521575315754157551575615757157581575915760157611576215763157641576515766157671576815769157701577115772157731577415775157761577715778157791578015781157821578315784157851578615787157881578915790157911579215793157941579515796157971579815799158001580115802158031580415805158061580715808158091581015811158121581315814158151581615817158181581915820158211582215823158241582515826158271582815829158301583115832158331583415835158361583715838158391584015841158421584315844158451584615847158481584915850158511585215853158541585515856158571585815859158601586115862158631586415865158661586715868158691587015871158721587315874158751587615877158781587915880158811588215883158841588515886158871588815889158901589115892158931589415895158961589715898158991590015901159021590315904159051590615907159081590915910159111591215913159141591515916159171591815919159201592115922159231592415925159261592715928159291593015931159321593315934159351593615937159381593915940159411594215943159441594515946159471594815949159501595115952159531595415955159561595715958159591596015961159621596315964159651596615967159681596915970159711597215973159741597515976159771597815979159801598115982159831598415985159861598715988159891599015991159921599315994159951599615997159981599916000160011600216003160041600516006160071600816009160101601116012160131601416015160161601716018160191602016021160221602316024160251602616027160281602916030160311603216033160341603516036160371603816039160401604116042160431604416045160461604716048160491605016051160521605316054160551605616057160581605916060160611606216063160641606516066160671606816069160701607116072160731607416075160761607716078160791608016081160821608316084160851608616087160881608916090160911609216093160941609516096160971609816099161001610116102161031610416105161061610716108161091611016111161121611316114161151611616117161181611916120161211612216123161241612516126161271612816129161301613116132161331613416135161361613716138161391614016141161421614316144161451614616147161481614916150161511615216153161541615516156161571615816159161601616116162161631616416165161661616716168161691617016171161721617316174161751617616177161781617916180161811618216183161841618516186161871618816189161901619116192161931619416195161961619716198161991620016201162021620316204162051620616207162081620916210162111621216213162141621516216162171621816219162201622116222162231622416225162261622716228162291623016231162321623316234162351623616237162381623916240162411624216243162441624516246162471624816249162501625116252162531625416255162561625716258162591626016261162621626316264162651626616267162681626916270162711627216273162741627516276162771627816279162801628116282162831628416285162861628716288162891629016291162921629316294162951629616297162981629916300163011630216303163041630516306163071630816309163101631116312163131631416315163161631716318163191632016321163221632316324163251632616327163281632916330163311633216333163341633516336163371633816339163401634116342163431634416345163461634716348163491635016351163521635316354163551635616357163581635916360163611636216363163641636516366163671636816369163701637116372163731637416375163761637716378163791638016381163821638316384163851638616387163881638916390163911639216393163941639516396163971639816399164001640116402164031640416405164061640716408164091641016411164121641316414164151641616417164181641916420164211642216423164241642516426164271642816429164301643116432164331643416435164361643716438164391644016441164421644316444164451644616447164481644916450164511645216453164541645516456164571645816459164601646116462164631646416465164661646716468164691647016471164721647316474164751647616477164781647916480164811648216483164841648516486164871648816489164901649116492164931649416495164961649716498164991650016501165021650316504165051650616507165081650916510165111651216513165141651516516165171651816519165201652116522165231652416525165261652716528165291653016531165321653316534165351653616537165381653916540165411654216543165441654516546165471654816549165501655116552165531655416555165561655716558165591656016561165621656316564165651656616567165681656916570165711657216573165741657516576165771657816579165801658116582165831658416585165861658716588165891659016591165921659316594165951659616597165981659916600166011660216603166041660516606166071660816609166101661116612166131661416615166161661716618166191662016621166221662316624166251662616627166281662916630166311663216633166341663516636166371663816639166401664116642166431664416645166461664716648166491665016651166521665316654166551665616657166581665916660166611666216663166641666516666166671666816669166701667116672166731667416675166761667716678166791668016681166821668316684166851668616687166881668916690166911669216693166941669516696166971669816699167001670116702167031670416705167061670716708167091671016711167121671316714167151671616717167181671916720167211672216723167241672516726167271672816729167301673116732167331673416735167361673716738167391674016741167421674316744167451674616747167481674916750167511675216753167541675516756167571675816759167601676116762167631676416765167661676716768167691677016771167721677316774167751677616777167781677916780167811678216783167841678516786167871678816789167901679116792167931679416795167961679716798167991680016801168021680316804168051680616807168081680916810168111681216813168141681516816168171681816819168201682116822168231682416825168261682716828168291683016831168321683316834168351683616837168381683916840168411684216843168441684516846168471684816849168501685116852168531685416855168561685716858168591686016861168621686316864168651686616867168681686916870168711687216873168741687516876168771687816879168801688116882168831688416885168861688716888168891689016891168921689316894168951689616897168981689916900169011690216903169041690516906169071690816909169101691116912169131691416915169161691716918169191692016921169221692316924169251692616927169281692916930169311693216933169341693516936169371693816939169401694116942169431694416945169461694716948169491695016951169521695316954169551695616957169581695916960169611696216963169641696516966169671696816969169701697116972169731697416975169761697716978169791698016981169821698316984169851698616987169881698916990169911699216993169941699516996169971699816999170001700117002170031700417005170061700717008170091701017011170121701317014170151701617017170181701917020170211702217023170241702517026170271702817029170301703117032170331703417035170361703717038170391704017041170421704317044170451704617047170481704917050170511705217053170541705517056170571705817059170601706117062170631706417065170661706717068170691707017071170721707317074170751707617077170781707917080170811708217083170841708517086170871708817089170901709117092170931709417095170961709717098170991710017101171021710317104171051710617107171081710917110171111711217113171141711517116171171711817119171201712117122171231712417125171261712717128171291713017131171321713317134171351713617137171381713917140171411714217143171441714517146171471714817149171501715117152171531715417155171561715717158171591716017161171621716317164171651716617167171681716917170171711717217173171741717517176171771717817179171801718117182171831718417185171861718717188171891719017191171921719317194171951719617197171981719917200172011720217203172041720517206172071720817209172101721117212172131721417215172161721717218172191722017221172221722317224172251722617227172281722917230172311723217233172341723517236172371723817239172401724117242172431724417245172461724717248172491725017251172521725317254172551725617257172581725917260172611726217263172641726517266172671726817269172701727117272172731727417275172761727717278172791728017281172821728317284172851728617287172881728917290172911729217293172941729517296172971729817299173001730117302173031730417305173061730717308173091731017311173121731317314173151731617317173181731917320173211732217323173241732517326173271732817329173301733117332173331733417335173361733717338173391734017341173421734317344173451734617347173481734917350173511735217353173541735517356173571735817359173601736117362173631736417365173661736717368173691737017371173721737317374173751737617377173781737917380173811738217383173841738517386173871738817389173901739117392173931739417395173961739717398173991740017401174021740317404174051740617407174081740917410174111741217413174141741517416174171741817419174201742117422174231742417425174261742717428174291743017431174321743317434174351743617437174381743917440174411744217443174441744517446174471744817449174501745117452174531745417455174561745717458174591746017461174621746317464174651746617467174681746917470174711747217473174741747517476174771747817479174801748117482174831748417485174861748717488174891749017491174921749317494174951749617497174981749917500175011750217503175041750517506175071750817509175101751117512175131751417515175161751717518175191752017521175221752317524175251752617527175281752917530175311753217533175341753517536175371753817539175401754117542175431754417545175461754717548175491755017551175521755317554175551755617557175581755917560175611756217563175641756517566175671756817569175701757117572175731757417575175761757717578175791758017581175821758317584175851758617587175881758917590175911759217593175941759517596175971759817599176001760117602176031760417605176061760717608176091761017611176121761317614176151761617617176181761917620176211762217623176241762517626176271762817629176301763117632176331763417635176361763717638176391764017641176421764317644176451764617647176481764917650176511765217653176541765517656176571765817659176601766117662176631766417665176661766717668176691767017671176721767317674176751767617677176781767917680176811768217683176841768517686176871768817689176901769117692176931769417695176961769717698176991770017701177021770317704177051770617707177081770917710177111771217713177141771517716177171771817719177201772117722177231772417725177261772717728177291773017731177321773317734177351773617737177381773917740177411774217743177441774517746177471774817749177501775117752177531775417755177561775717758177591776017761177621776317764177651776617767177681776917770177711777217773177741777517776177771777817779177801778117782177831778417785177861778717788177891779017791177921779317794177951779617797177981779917800178011780217803178041780517806178071780817809178101781117812178131781417815178161781717818178191782017821178221782317824178251782617827178281782917830178311783217833178341783517836178371783817839178401784117842178431784417845178461784717848178491785017851178521785317854178551785617857178581785917860178611786217863178641786517866178671786817869178701787117872178731787417875178761787717878178791788017881178821788317884178851788617887178881788917890178911789217893178941789517896178971789817899179001790117902179031790417905179061790717908179091791017911179121791317914179151791617917179181791917920179211792217923179241792517926179271792817929179301793117932179331793417935179361793717938179391794017941179421794317944179451794617947179481794917950179511795217953179541795517956179571795817959179601796117962179631796417965179661796717968179691797017971179721797317974179751797617977179781797917980179811798217983179841798517986179871798817989179901799117992179931799417995179961799717998179991800018001180021800318004180051800618007180081800918010180111801218013180141801518016180171801818019180201802118022180231802418025180261802718028180291803018031180321803318034180351803618037180381803918040180411804218043180441804518046180471804818049180501805118052180531805418055180561805718058180591806018061180621806318064180651806618067180681806918070180711807218073180741807518076180771807818079180801808118082180831808418085180861808718088180891809018091180921809318094180951809618097180981809918100181011810218103181041810518106181071810818109181101811118112181131811418115181161811718118181191812018121181221812318124181251812618127181281812918130181311813218133181341813518136181371813818139181401814118142181431814418145181461814718148181491815018151181521815318154181551815618157181581815918160181611816218163181641816518166181671816818169181701817118172181731817418175181761817718178181791818018181181821818318184181851818618187181881818918190181911819218193181941819518196181971819818199182001820118202182031820418205182061820718208182091821018211182121821318214182151821618217182181821918220182211822218223182241822518226182271822818229182301823118232182331823418235182361823718238182391824018241182421824318244182451824618247182481824918250182511825218253182541825518256182571825818259182601826118262182631826418265182661826718268182691827018271182721827318274182751827618277182781827918280182811828218283182841828518286182871828818289182901829118292182931829418295182961829718298182991830018301183021830318304183051830618307183081830918310183111831218313183141831518316183171831818319183201832118322183231832418325183261832718328183291833018331183321833318334183351833618337183381833918340183411834218343183441834518346183471834818349183501835118352183531835418355183561835718358183591836018361183621836318364183651836618367183681836918370183711837218373183741837518376183771837818379183801838118382183831838418385183861838718388183891839018391183921839318394183951839618397183981839918400184011840218403184041840518406184071840818409184101841118412184131841418415184161841718418184191842018421184221842318424184251842618427184281842918430184311843218433184341843518436184371843818439184401844118442184431844418445184461844718448184491845018451184521845318454184551845618457184581845918460184611846218463184641846518466184671846818469184701847118472184731847418475184761847718478184791848018481184821848318484184851848618487184881848918490184911849218493184941849518496184971849818499185001850118502185031850418505185061850718508185091851018511185121851318514185151851618517185181851918520185211852218523185241852518526185271852818529185301853118532185331853418535185361853718538185391854018541185421854318544185451854618547185481854918550185511855218553185541855518556185571855818559185601856118562185631856418565185661856718568185691857018571185721857318574185751857618577185781857918580185811858218583185841858518586185871858818589185901859118592185931859418595185961859718598185991860018601186021860318604186051860618607186081860918610186111861218613186141861518616186171861818619186201862118622186231862418625186261862718628186291863018631186321863318634186351863618637186381863918640186411864218643186441864518646186471864818649186501865118652186531865418655186561865718658186591866018661186621866318664186651866618667186681866918670186711867218673186741867518676186771867818679186801868118682186831868418685186861868718688186891869018691186921869318694186951869618697186981869918700187011870218703187041870518706187071870818709187101871118712187131871418715187161871718718187191872018721187221872318724187251872618727187281872918730187311873218733187341873518736187371873818739187401874118742187431874418745187461874718748187491875018751187521875318754187551875618757187581875918760187611876218763187641876518766187671876818769187701877118772187731877418775187761877718778187791878018781187821878318784187851878618787187881878918790187911879218793187941879518796187971879818799188001880118802188031880418805188061880718808188091881018811188121881318814188151881618817188181881918820188211882218823188241882518826188271882818829188301883118832188331883418835188361883718838188391884018841188421884318844188451884618847188481884918850188511885218853188541885518856188571885818859188601886118862188631886418865188661886718868188691887018871188721887318874188751887618877188781887918880188811888218883188841888518886188871888818889188901889118892188931889418895188961889718898188991890018901189021890318904189051890618907189081890918910189111891218913189141891518916189171891818919189201892118922189231892418925189261892718928189291893018931189321893318934189351893618937189381893918940189411894218943189441894518946189471894818949189501895118952189531895418955189561895718958189591896018961189621896318964189651896618967189681896918970189711897218973189741897518976189771897818979189801898118982189831898418985189861898718988189891899018991189921899318994189951899618997189981899919000190011900219003190041900519006190071900819009190101901119012190131901419015190161901719018190191902019021190221902319024190251902619027190281902919030190311903219033190341903519036190371903819039190401904119042190431904419045190461904719048190491905019051190521905319054190551905619057190581905919060190611906219063190641906519066190671906819069190701907119072190731907419075190761907719078190791908019081190821908319084190851908619087190881908919090190911909219093190941909519096190971909819099191001910119102191031910419105191061910719108191091911019111191121911319114191151911619117191181911919120191211912219123191241912519126191271912819129191301913119132191331913419135191361913719138191391914019141191421914319144191451914619147191481914919150191511915219153191541915519156191571915819159191601916119162191631916419165191661916719168191691917019171191721917319174191751917619177191781917919180191811918219183191841918519186191871918819189191901919119192191931919419195191961919719198191991920019201192021920319204192051920619207192081920919210192111921219213192141921519216192171921819219192201922119222192231922419225192261922719228192291923019231192321923319234192351923619237192381923919240192411924219243192441924519246192471924819249192501925119252192531925419255192561925719258192591926019261192621926319264192651926619267192681926919270192711927219273192741927519276192771927819279192801928119282192831928419285192861928719288192891929019291192921929319294192951929619297192981929919300193011930219303193041930519306193071930819309193101931119312193131931419315193161931719318193191932019321193221932319324193251932619327193281932919330193311933219333193341933519336193371933819339193401934119342193431934419345193461934719348193491935019351193521935319354193551935619357193581935919360193611936219363193641936519366193671936819369193701937119372193731937419375193761937719378193791938019381193821938319384193851938619387193881938919390193911939219393193941939519396193971939819399194001940119402194031940419405194061940719408194091941019411194121941319414194151941619417194181941919420194211942219423194241942519426194271942819429194301943119432194331943419435194361943719438194391944019441194421944319444194451944619447194481944919450194511945219453194541945519456194571945819459194601946119462194631946419465194661946719468194691947019471194721947319474194751947619477194781947919480194811948219483194841948519486194871948819489194901949119492194931949419495194961949719498194991950019501195021950319504195051950619507195081950919510195111951219513195141951519516195171951819519195201952119522195231952419525195261952719528195291953019531195321953319534195351953619537195381953919540195411954219543195441954519546195471954819549195501955119552195531955419555195561955719558195591956019561195621956319564195651956619567195681956919570195711957219573195741957519576195771957819579195801958119582195831958419585195861958719588195891959019591195921959319594195951959619597195981959919600196011960219603196041960519606196071960819609196101961119612196131961419615196161961719618196191962019621196221962319624196251962619627196281962919630196311963219633196341963519636196371963819639196401964119642196431964419645196461964719648196491965019651196521965319654196551965619657196581965919660196611966219663196641966519666196671966819669196701967119672196731967419675196761967719678196791968019681196821968319684196851968619687196881968919690196911969219693196941969519696196971969819699197001970119702197031970419705197061970719708197091971019711197121971319714197151971619717197181971919720197211972219723197241972519726197271972819729197301973119732197331973419735197361973719738197391974019741197421974319744197451974619747197481974919750197511975219753197541975519756197571975819759197601976119762197631976419765197661976719768197691977019771197721977319774197751977619777197781977919780197811978219783197841978519786197871978819789197901979119792197931979419795197961979719798197991980019801198021980319804198051980619807198081980919810198111981219813198141981519816198171981819819198201982119822198231982419825198261982719828198291983019831198321983319834198351983619837198381983919840198411984219843198441984519846198471984819849198501985119852198531985419855198561985719858198591986019861198621986319864198651986619867198681986919870198711987219873198741987519876198771987819879198801988119882198831988419885198861988719888198891989019891198921989319894198951989619897198981989919900199011990219903199041990519906199071990819909199101991119912199131991419915199161991719918199191992019921199221992319924199251992619927199281992919930199311993219933199341993519936199371993819939199401994119942199431994419945199461994719948199491995019951199521995319954199551995619957199581995919960199611996219963199641996519966199671996819969199701997119972199731997419975199761997719978199791998019981199821998319984199851998619987199881998919990199911999219993199941999519996199971999819999200002000120002200032000420005200062000720008200092001020011200122001320014200152001620017200182001920020200212002220023200242002520026200272002820029200302003120032200332003420035200362003720038200392004020041200422004320044200452004620047200482004920050200512005220053200542005520056200572005820059200602006120062200632006420065200662006720068200692007020071200722007320074200752007620077200782007920080200812008220083200842008520086200872008820089200902009120092200932009420095200962009720098200992010020101201022010320104201052010620107201082010920110201112011220113201142011520116201172011820119201202012120122201232012420125201262012720128201292013020131201322013320134201352013620137201382013920140201412014220143201442014520146201472014820149201502015120152201532015420155201562015720158201592016020161201622016320164201652016620167201682016920170201712017220173201742017520176201772017820179201802018120182201832018420185201862018720188201892019020191201922019320194201952019620197201982019920200202012020220203202042020520206202072020820209202102021120212202132021420215202162021720218202192022020221202222022320224202252022620227202282022920230202312023220233202342023520236202372023820239202402024120242202432024420245202462024720248202492025020251202522025320254202552025620257202582025920260202612026220263202642026520266202672026820269202702027120272202732027420275202762027720278202792028020281202822028320284202852028620287202882028920290202912029220293202942029520296202972029820299203002030120302203032030420305203062030720308203092031020311203122031320314203152031620317203182031920320203212032220323203242032520326203272032820329203302033120332203332033420335203362033720338203392034020341203422034320344203452034620347203482034920350203512035220353203542035520356203572035820359203602036120362203632036420365203662036720368203692037020371203722037320374203752037620377203782037920380203812038220383203842038520386203872038820389203902039120392203932039420395203962039720398203992040020401204022040320404204052040620407204082040920410204112041220413204142041520416204172041820419204202042120422204232042420425204262042720428204292043020431204322043320434204352043620437204382043920440204412044220443204442044520446204472044820449204502045120452204532045420455204562045720458204592046020461204622046320464204652046620467204682046920470204712047220473204742047520476204772047820479204802048120482204832048420485204862048720488204892049020491204922049320494204952049620497204982049920500205012050220503205042050520506205072050820509205102051120512205132051420515205162051720518205192052020521205222052320524205252052620527205282052920530205312053220533205342053520536205372053820539205402054120542205432054420545205462054720548205492055020551205522055320554205552055620557205582055920560205612056220563205642056520566205672056820569205702057120572205732057420575205762057720578205792058020581205822058320584205852058620587205882058920590205912059220593205942059520596205972059820599206002060120602206032060420605206062060720608206092061020611206122061320614206152061620617206182061920620206212062220623206242062520626206272062820629206302063120632206332063420635206362063720638206392064020641206422064320644206452064620647206482064920650206512065220653206542065520656206572065820659206602066120662206632066420665206662066720668206692067020671206722067320674206752067620677206782067920680206812068220683206842068520686206872068820689206902069120692206932069420695206962069720698206992070020701207022070320704207052070620707207082070920710207112071220713207142071520716207172071820719207202072120722207232072420725207262072720728207292073020731207322073320734207352073620737207382073920740207412074220743207442074520746207472074820749207502075120752207532075420755207562075720758207592076020761207622076320764207652076620767207682076920770207712077220773207742077520776207772077820779207802078120782207832078420785207862078720788207892079020791207922079320794207952079620797207982079920800208012080220803208042080520806208072080820809208102081120812208132081420815208162081720818208192082020821208222082320824208252082620827208282082920830208312083220833208342083520836208372083820839208402084120842208432084420845208462084720848208492085020851208522085320854208552085620857208582085920860208612086220863208642086520866208672086820869208702087120872208732087420875208762087720878208792088020881208822088320884208852088620887208882088920890208912089220893208942089520896208972089820899209002090120902209032090420905209062090720908209092091020911209122091320914209152091620917209182091920920209212092220923209242092520926209272092820929209302093120932209332093420935209362093720938209392094020941209422094320944209452094620947209482094920950209512095220953209542095520956209572095820959209602096120962209632096420965209662096720968209692097020971209722097320974209752097620977209782097920980209812098220983209842098520986209872098820989209902099120992209932099420995209962099720998209992100021001210022100321004210052100621007210082100921010210112101221013210142101521016210172101821019210202102121022210232102421025210262102721028210292103021031210322103321034210352103621037210382103921040210412104221043210442104521046210472104821049210502105121052210532105421055210562105721058210592106021061210622106321064210652106621067210682106921070210712107221073210742107521076210772107821079210802108121082210832108421085210862108721088210892109021091210922109321094210952109621097210982109921100211012110221103211042110521106211072110821109211102111121112211132111421115211162111721118211192112021121211222112321124211252112621127211282112921130211312113221133211342113521136211372113821139211402114121142211432114421145211462114721148211492115021151211522115321154211552115621157211582115921160211612116221163211642116521166211672116821169211702117121172211732117421175211762117721178211792118021181211822118321184211852118621187211882118921190211912119221193211942119521196211972119821199212002120121202212032120421205212062120721208212092121021211212122121321214212152121621217212182121921220212212122221223212242122521226212272122821229212302123121232212332123421235212362123721238212392124021241212422124321244212452124621247212482124921250212512125221253212542125521256212572125821259212602126121262212632126421265212662126721268212692127021271212722127321274212752127621277212782127921280212812128221283212842128521286212872128821289212902129121292212932129421295212962129721298212992130021301213022130321304213052130621307213082130921310213112131221313213142131521316213172131821319213202132121322213232132421325213262132721328213292133021331213322133321334213352133621337213382133921340213412134221343213442134521346213472134821349213502135121352213532135421355213562135721358213592136021361213622136321364213652136621367213682136921370213712137221373213742137521376213772137821379213802138121382213832138421385213862138721388213892139021391213922139321394213952139621397213982139921400214012140221403214042140521406214072140821409214102141121412214132141421415214162141721418214192142021421214222142321424214252142621427214282142921430214312143221433214342143521436214372143821439214402144121442214432144421445214462144721448214492145021451214522145321454214552145621457214582145921460214612146221463214642146521466214672146821469214702147121472214732147421475214762147721478214792148021481214822148321484214852148621487214882148921490214912149221493214942149521496214972149821499215002150121502215032150421505215062150721508215092151021511215122151321514215152151621517215182151921520215212152221523215242152521526215272152821529215302153121532215332153421535215362153721538215392154021541215422154321544215452154621547215482154921550215512155221553215542155521556215572155821559215602156121562215632156421565215662156721568215692157021571215722157321574215752157621577215782157921580215812158221583215842158521586215872158821589215902159121592215932159421595215962159721598215992160021601216022160321604216052160621607216082160921610216112161221613216142161521616216172161821619216202162121622216232162421625216262162721628216292163021631216322163321634216352163621637216382163921640216412164221643216442164521646216472164821649216502165121652216532165421655216562165721658216592166021661216622166321664216652166621667216682166921670216712167221673216742167521676216772167821679216802168121682216832168421685216862168721688216892169021691216922169321694216952169621697216982169921700217012170221703217042170521706217072170821709217102171121712217132171421715217162171721718217192172021721217222172321724217252172621727217282172921730217312173221733217342173521736217372173821739217402174121742217432174421745217462174721748217492175021751217522175321754217552175621757217582175921760217612176221763217642176521766217672176821769217702177121772217732177421775217762177721778217792178021781217822178321784217852178621787217882178921790217912179221793217942179521796217972179821799218002180121802218032180421805218062180721808218092181021811218122181321814218152181621817218182181921820218212182221823218242182521826218272182821829218302183121832218332183421835218362183721838218392184021841218422184321844218452184621847218482184921850218512185221853218542185521856218572185821859218602186121862218632186421865218662186721868218692187021871218722187321874218752187621877218782187921880218812188221883218842188521886218872188821889218902189121892218932189421895218962189721898218992190021901219022190321904219052190621907219082190921910219112191221913219142191521916219172191821919219202192121922219232192421925219262192721928219292193021931219322193321934219352193621937219382193921940219412194221943219442194521946219472194821949219502195121952219532195421955219562195721958219592196021961219622196321964219652196621967219682196921970219712197221973219742197521976219772197821979219802198121982219832198421985219862198721988219892199021991219922199321994219952199621997219982199922000220012200222003220042200522006220072200822009220102201122012220132201422015220162201722018220192202022021220222202322024220252202622027220282202922030220312203222033220342203522036220372203822039220402204122042220432204422045220462204722048220492205022051220522205322054220552205622057220582205922060220612206222063220642206522066220672206822069220702207122072220732207422075220762207722078220792208022081220822208322084220852208622087220882208922090220912209222093220942209522096220972209822099221002210122102221032210422105221062210722108221092211022111221122211322114221152211622117221182211922120221212212222123221242212522126221272212822129221302213122132221332213422135221362213722138221392214022141221422214322144221452214622147221482214922150221512215222153221542215522156221572215822159221602216122162221632216422165221662216722168221692217022171221722217322174221752217622177221782217922180221812218222183221842218522186221872218822189221902219122192221932219422195221962219722198221992220022201222022220322204222052220622207222082220922210222112221222213222142221522216222172221822219222202222122222222232222422225222262222722228222292223022231222322223322234222352223622237222382223922240222412224222243222442224522246222472224822249222502225122252222532225422255222562225722258222592226022261222622226322264222652226622267222682226922270222712227222273222742227522276222772227822279222802228122282222832228422285222862228722288222892229022291222922229322294222952229622297222982229922300223012230222303223042230522306223072230822309223102231122312223132231422315223162231722318223192232022321223222232322324223252232622327223282232922330223312233222333223342233522336223372233822339223402234122342223432234422345223462234722348223492235022351223522235322354223552235622357223582235922360223612236222363223642236522366223672236822369223702237122372223732237422375223762237722378223792238022381223822238322384223852238622387223882238922390223912239222393223942239522396223972239822399224002240122402224032240422405224062240722408224092241022411224122241322414224152241622417224182241922420224212242222423224242242522426224272242822429224302243122432224332243422435224362243722438224392244022441224422244322444224452244622447224482244922450224512245222453224542245522456224572245822459224602246122462224632246422465224662246722468224692247022471224722247322474224752247622477224782247922480224812248222483224842248522486224872248822489224902249122492224932249422495224962249722498224992250022501225022250322504225052250622507225082250922510225112251222513225142251522516225172251822519225202252122522225232252422525225262252722528225292253022531225322253322534225352253622537225382253922540225412254222543225442254522546225472254822549225502255122552225532255422555225562255722558225592256022561225622256322564225652256622567225682256922570225712257222573225742257522576225772257822579225802258122582225832258422585225862258722588225892259022591225922259322594225952259622597225982259922600226012260222603226042260522606226072260822609226102261122612226132261422615226162261722618226192262022621226222262322624226252262622627226282262922630226312263222633226342263522636226372263822639226402264122642226432264422645226462264722648226492265022651226522265322654226552265622657226582265922660226612266222663226642266522666226672266822669226702267122672226732267422675226762267722678226792268022681226822268322684226852268622687226882268922690226912269222693226942269522696226972269822699227002270122702227032270422705227062270722708227092271022711227122271322714227152271622717227182271922720227212272222723227242272522726227272272822729227302273122732227332273422735227362273722738227392274022741227422274322744227452274622747227482274922750227512275222753227542275522756227572275822759227602276122762227632276422765227662276722768227692277022771227722277322774227752277622777227782277922780227812278222783227842278522786227872278822789227902279122792227932279422795227962279722798227992280022801228022280322804228052280622807228082280922810228112281222813228142281522816228172281822819228202282122822228232282422825228262282722828228292283022831228322283322834228352283622837228382283922840228412284222843228442284522846228472284822849228502285122852228532285422855228562285722858228592286022861228622286322864228652286622867228682286922870228712287222873228742287522876228772287822879228802288122882228832288422885228862288722888228892289022891228922289322894228952289622897228982289922900229012290222903229042290522906229072290822909229102291122912229132291422915229162291722918229192292022921229222292322924229252292622927229282292922930229312293222933229342293522936229372293822939229402294122942229432294422945229462294722948229492295022951229522295322954229552295622957229582295922960229612296222963229642296522966229672296822969229702297122972229732297422975229762297722978229792298022981229822298322984229852298622987229882298922990229912299222993229942299522996229972299822999230002300123002230032300423005230062300723008230092301023011230122301323014230152301623017230182301923020230212302223023230242302523026230272302823029230302303123032230332303423035230362303723038230392304023041230422304323044230452304623047230482304923050230512305223053230542305523056230572305823059230602306123062230632306423065230662306723068230692307023071230722307323074230752307623077230782307923080230812308223083230842308523086230872308823089230902309123092230932309423095230962309723098230992310023101231022310323104231052310623107231082310923110231112311223113231142311523116231172311823119231202312123122231232312423125231262312723128231292313023131231322313323134231352313623137231382313923140231412314223143231442314523146231472314823149231502315123152231532315423155231562315723158231592316023161231622316323164231652316623167231682316923170231712317223173231742317523176231772317823179231802318123182231832318423185231862318723188231892319023191231922319323194231952319623197231982319923200232012320223203232042320523206232072320823209232102321123212232132321423215232162321723218232192322023221232222322323224232252322623227232282322923230232312323223233232342323523236232372323823239232402324123242232432324423245232462324723248232492325023251232522325323254232552325623257232582325923260232612326223263232642326523266232672326823269232702327123272232732327423275232762327723278232792328023281232822328323284232852328623287232882328923290232912329223293232942329523296232972329823299233002330123302233032330423305233062330723308233092331023311233122331323314233152331623317233182331923320233212332223323233242332523326233272332823329233302333123332233332333423335233362333723338233392334023341233422334323344233452334623347233482334923350233512335223353233542335523356233572335823359233602336123362233632336423365233662336723368233692337023371233722337323374233752337623377233782337923380233812338223383233842338523386233872338823389233902339123392233932339423395233962339723398233992340023401234022340323404234052340623407234082340923410234112341223413234142341523416234172341823419234202342123422234232342423425234262342723428234292343023431234322343323434234352343623437234382343923440234412344223443234442344523446234472344823449234502345123452234532345423455234562345723458234592346023461234622346323464234652346623467234682346923470234712347223473234742347523476234772347823479234802348123482234832348423485234862348723488234892349023491234922349323494234952349623497234982349923500235012350223503235042350523506235072350823509235102351123512235132351423515235162351723518235192352023521235222352323524235252352623527235282352923530235312353223533235342353523536235372353823539235402354123542235432354423545235462354723548235492355023551235522355323554235552355623557235582355923560235612356223563235642356523566235672356823569235702357123572235732357423575235762357723578235792358023581235822358323584235852358623587235882358923590235912359223593235942359523596235972359823599236002360123602236032360423605236062360723608236092361023611236122361323614236152361623617236182361923620236212362223623236242362523626236272362823629236302363123632236332363423635236362363723638236392364023641236422364323644236452364623647236482364923650236512365223653236542365523656236572365823659236602366123662236632366423665236662366723668236692367023671236722367323674236752367623677236782367923680236812368223683236842368523686236872368823689236902369123692236932369423695236962369723698236992370023701237022370323704237052370623707237082370923710237112371223713237142371523716237172371823719237202372123722237232372423725237262372723728237292373023731237322373323734237352373623737237382373923740237412374223743237442374523746237472374823749237502375123752237532375423755237562375723758237592376023761237622376323764237652376623767237682376923770237712377223773237742377523776237772377823779237802378123782237832378423785237862378723788237892379023791237922379323794237952379623797237982379923800238012380223803238042380523806238072380823809238102381123812238132381423815238162381723818238192382023821238222382323824238252382623827238282382923830238312383223833238342383523836238372383823839238402384123842238432384423845238462384723848238492385023851238522385323854238552385623857238582385923860238612386223863238642386523866238672386823869238702387123872238732387423875238762387723878238792388023881238822388323884238852388623887238882388923890238912389223893238942389523896238972389823899239002390123902239032390423905239062390723908239092391023911239122391323914239152391623917239182391923920239212392223923239242392523926239272392823929239302393123932239332393423935239362393723938239392394023941239422394323944239452394623947239482394923950239512395223953239542395523956239572395823959239602396123962239632396423965239662396723968239692397023971239722397323974239752397623977239782397923980239812398223983239842398523986239872398823989239902399123992239932399423995239962399723998239992400024001240022400324004240052400624007240082400924010240112401224013240142401524016240172401824019240202402124022240232402424025240262402724028240292403024031240322403324034240352403624037240382403924040240412404224043240442404524046240472404824049240502405124052240532405424055240562405724058240592406024061240622406324064240652406624067240682406924070240712407224073240742407524076240772407824079240802408124082240832408424085240862408724088240892409024091240922409324094240952409624097240982409924100241012410224103241042410524106241072410824109241102411124112241132411424115241162411724118241192412024121241222412324124241252412624127241282412924130241312413224133241342413524136241372413824139241402414124142241432414424145241462414724148241492415024151241522415324154241552415624157241582415924160241612416224163241642416524166241672416824169241702417124172241732417424175241762417724178241792418024181241822418324184241852418624187241882418924190241912419224193241942419524196241972419824199242002420124202242032420424205242062420724208242092421024211242122421324214242152421624217242182421924220242212422224223242242422524226242272422824229242302423124232242332423424235242362423724238242392424024241242422424324244242452424624247242482424924250242512425224253242542425524256242572425824259242602426124262242632426424265242662426724268242692427024271242722427324274242752427624277242782427924280242812428224283242842428524286242872428824289242902429124292242932429424295242962429724298242992430024301243022430324304243052430624307243082430924310243112431224313243142431524316243172431824319243202432124322243232432424325243262432724328243292433024331243322433324334243352433624337243382433924340243412434224343243442434524346243472434824349243502435124352243532435424355243562435724358243592436024361243622436324364243652436624367243682436924370243712437224373243742437524376243772437824379243802438124382243832438424385243862438724388243892439024391243922439324394243952439624397243982439924400244012440224403244042440524406244072440824409244102441124412244132441424415244162441724418244192442024421244222442324424244252442624427244282442924430244312443224433244342443524436244372443824439244402444124442244432444424445244462444724448244492445024451244522445324454244552445624457244582445924460244612446224463244642446524466244672446824469244702447124472244732447424475244762447724478244792448024481244822448324484244852448624487244882448924490244912449224493244942449524496244972449824499245002450124502245032450424505245062450724508245092451024511245122451324514245152451624517245182451924520245212452224523245242452524526245272452824529245302453124532245332453424535245362453724538245392454024541245422454324544245452454624547245482454924550245512455224553245542455524556245572455824559245602456124562245632456424565245662456724568245692457024571245722457324574245752457624577245782457924580245812458224583245842458524586245872458824589245902459124592245932459424595245962459724598245992460024601246022460324604246052460624607246082460924610246112461224613246142461524616246172461824619246202462124622246232462424625246262462724628246292463024631246322463324634246352463624637246382463924640246412464224643246442464524646246472464824649246502465124652246532465424655246562465724658246592466024661246622466324664246652466624667246682466924670246712467224673246742467524676246772467824679246802468124682246832468424685246862468724688246892469024691246922469324694246952469624697246982469924700247012470224703247042470524706247072470824709247102471124712247132471424715247162471724718247192472024721247222472324724247252472624727247282472924730247312473224733247342473524736247372473824739247402474124742247432474424745247462474724748247492475024751247522475324754247552475624757247582475924760247612476224763247642476524766247672476824769247702477124772247732477424775247762477724778247792478024781247822478324784247852478624787247882478924790247912479224793247942479524796247972479824799248002480124802248032480424805248062480724808248092481024811248122481324814248152481624817248182481924820248212482224823248242482524826248272482824829248302483124832248332483424835248362483724838248392484024841248422484324844248452484624847248482484924850248512485224853248542485524856248572485824859248602486124862248632486424865248662486724868248692487024871248722487324874248752487624877248782487924880248812488224883248842488524886248872488824889248902489124892248932489424895248962489724898248992490024901249022490324904249052490624907249082490924910249112491224913249142491524916249172491824919249202492124922249232492424925249262492724928249292493024931249322493324934249352493624937249382493924940249412494224943249442494524946249472494824949249502495124952249532495424955249562495724958249592496024961249622496324964249652496624967249682496924970249712497224973249742497524976249772497824979249802498124982249832498424985249862498724988249892499024991249922499324994249952499624997249982499925000250012500225003250042500525006250072500825009250102501125012250132501425015250162501725018250192502025021250222502325024250252502625027250282502925030250312503225033250342503525036250372503825039250402504125042250432504425045250462504725048250492505025051250522505325054250552505625057250582505925060250612506225063250642506525066250672506825069250702507125072250732507425075250762507725078250792508025081250822508325084250852508625087250882508925090250912509225093250942509525096250972509825099251002510125102251032510425105251062510725108251092511025111251122511325114251152511625117251182511925120251212512225123251242512525126251272512825129251302513125132251332513425135251362513725138251392514025141251422514325144251452514625147251482514925150251512515225153251542515525156251572515825159251602516125162251632516425165251662516725168251692517025171251722517325174251752517625177251782517925180251812518225183251842518525186251872518825189251902519125192251932519425195251962519725198251992520025201252022520325204252052520625207252082520925210252112521225213252142521525216252172521825219252202522125222252232522425225252262522725228252292523025231252322523325234252352523625237252382523925240252412524225243252442524525246252472524825249252502525125252252532525425255252562525725258252592526025261252622526325264252652526625267252682526925270252712527225273252742527525276252772527825279252802528125282252832528425285252862528725288252892529025291252922529325294252952529625297252982529925300253012530225303253042530525306253072530825309253102531125312253132531425315253162531725318253192532025321253222532325324253252532625327253282532925330253312533225333253342533525336253372533825339253402534125342253432534425345253462534725348253492535025351253522535325354253552535625357253582535925360253612536225363253642536525366253672536825369253702537125372253732537425375253762537725378253792538025381253822538325384253852538625387253882538925390253912539225393253942539525396253972539825399254002540125402254032540425405254062540725408254092541025411254122541325414254152541625417254182541925420254212542225423254242542525426254272542825429254302543125432254332543425435254362543725438254392544025441254422544325444254452544625447254482544925450254512545225453254542545525456254572545825459254602546125462254632546425465254662546725468254692547025471254722547325474254752547625477254782547925480254812548225483254842548525486254872548825489254902549125492254932549425495254962549725498254992550025501255022550325504255052550625507255082550925510255112551225513255142551525516255172551825519255202552125522255232552425525255262552725528255292553025531255322553325534255352553625537255382553925540255412554225543255442554525546255472554825549255502555125552255532555425555255562555725558255592556025561255622556325564255652556625567255682556925570255712557225573255742557525576255772557825579255802558125582255832558425585255862558725588255892559025591255922559325594255952559625597255982559925600256012560225603256042560525606256072560825609256102561125612256132561425615256162561725618256192562025621256222562325624256252562625627256282562925630256312563225633256342563525636256372563825639256402564125642256432564425645256462564725648256492565025651256522565325654256552565625657256582565925660256612566225663256642566525666256672566825669256702567125672256732567425675256762567725678256792568025681256822568325684256852568625687256882568925690256912569225693256942569525696256972569825699257002570125702257032570425705257062570725708257092571025711257122571325714257152571625717257182571925720257212572225723257242572525726257272572825729257302573125732257332573425735257362573725738257392574025741257422574325744257452574625747257482574925750257512575225753257542575525756257572575825759257602576125762257632576425765257662576725768257692577025771257722577325774257752577625777257782577925780257812578225783257842578525786257872578825789257902579125792257932579425795257962579725798257992580025801258022580325804258052580625807258082580925810258112581225813258142581525816258172581825819258202582125822258232582425825258262582725828258292583025831258322583325834258352583625837258382583925840258412584225843258442584525846258472584825849258502585125852258532585425855258562585725858258592586025861258622586325864258652586625867258682586925870258712587225873258742587525876258772587825879258802588125882258832588425885258862588725888258892589025891258922589325894258952589625897258982589925900259012590225903259042590525906259072590825909259102591125912259132591425915259162591725918259192592025921259222592325924259252592625927259282592925930259312593225933259342593525936259372593825939259402594125942259432594425945259462594725948259492595025951259522595325954259552595625957259582595925960259612596225963259642596525966259672596825969259702597125972259732597425975259762597725978259792598025981259822598325984259852598625987259882598925990259912599225993259942599525996259972599825999260002600126002260032600426005260062600726008260092601026011260122601326014260152601626017260182601926020260212602226023260242602526026260272602826029260302603126032260332603426035260362603726038260392604026041260422604326044260452604626047260482604926050260512605226053260542605526056260572605826059260602606126062260632606426065260662606726068260692607026071260722607326074260752607626077260782607926080260812608226083260842608526086260872608826089260902609126092260932609426095260962609726098260992610026101261022610326104261052610626107261082610926110261112611226113261142611526116261172611826119261202612126122261232612426125261262612726128261292613026131261322613326134261352613626137261382613926140261412614226143261442614526146261472614826149261502615126152261532615426155261562615726158261592616026161261622616326164261652616626167261682616926170261712617226173261742617526176261772617826179261802618126182261832618426185261862618726188261892619026191261922619326194261952619626197261982619926200262012620226203262042620526206262072620826209262102621126212262132621426215262162621726218262192622026221262222622326224262252622626227262282622926230262312623226233262342623526236262372623826239262402624126242262432624426245262462624726248262492625026251262522625326254262552625626257262582625926260262612626226263262642626526266262672626826269262702627126272262732627426275262762627726278262792628026281262822628326284262852628626287262882628926290262912629226293262942629526296262972629826299263002630126302263032630426305263062630726308263092631026311263122631326314263152631626317263182631926320263212632226323263242632526326263272632826329263302633126332263332633426335263362633726338263392634026341263422634326344263452634626347263482634926350263512635226353263542635526356263572635826359263602636126362263632636426365263662636726368263692637026371263722637326374263752637626377263782637926380263812638226383263842638526386263872638826389263902639126392263932639426395263962639726398263992640026401264022640326404264052640626407264082640926410264112641226413264142641526416264172641826419264202642126422264232642426425264262642726428264292643026431264322643326434264352643626437264382643926440264412644226443264442644526446264472644826449264502645126452264532645426455264562645726458264592646026461264622646326464264652646626467264682646926470264712647226473264742647526476264772647826479264802648126482264832648426485264862648726488264892649026491264922649326494264952649626497264982649926500265012650226503265042650526506265072650826509265102651126512265132651426515265162651726518265192652026521265222652326524265252652626527265282652926530265312653226533265342653526536265372653826539265402654126542265432654426545265462654726548265492655026551265522655326554265552655626557265582655926560265612656226563265642656526566265672656826569265702657126572265732657426575265762657726578265792658026581265822658326584265852658626587265882658926590265912659226593265942659526596265972659826599266002660126602266032660426605266062660726608266092661026611266122661326614266152661626617266182661926620266212662226623266242662526626266272662826629266302663126632266332663426635266362663726638266392664026641266422664326644266452664626647266482664926650266512665226653266542665526656266572665826659266602666126662266632666426665266662666726668266692667026671266722667326674266752667626677266782667926680266812668226683266842668526686266872668826689266902669126692266932669426695266962669726698266992670026701267022670326704267052670626707267082670926710267112671226713267142671526716267172671826719267202672126722267232672426725267262672726728267292673026731267322673326734267352673626737267382673926740267412674226743267442674526746267472674826749267502675126752267532675426755267562675726758267592676026761267622676326764267652676626767267682676926770267712677226773267742677526776267772677826779267802678126782267832678426785267862678726788267892679026791267922679326794267952679626797267982679926800268012680226803268042680526806268072680826809268102681126812268132681426815268162681726818268192682026821268222682326824268252682626827268282682926830268312683226833268342683526836268372683826839268402684126842268432684426845268462684726848268492685026851268522685326854268552685626857268582685926860268612686226863268642686526866268672686826869268702687126872268732687426875268762687726878268792688026881268822688326884268852688626887268882688926890268912689226893268942689526896268972689826899269002690126902269032690426905269062690726908269092691026911269122691326914269152691626917269182691926920269212692226923269242692526926269272692826929269302693126932269332693426935269362693726938269392694026941269422694326944269452694626947269482694926950269512695226953269542695526956269572695826959269602696126962269632696426965269662696726968269692697026971269722697326974269752697626977269782697926980269812698226983269842698526986269872698826989269902699126992269932699426995269962699726998269992700027001270022700327004270052700627007270082700927010270112701227013270142701527016270172701827019270202702127022270232702427025270262702727028270292703027031270322703327034270352703627037270382703927040270412704227043270442704527046270472704827049270502705127052270532705427055270562705727058270592706027061270622706327064270652706627067270682706927070270712707227073270742707527076270772707827079270802708127082270832708427085270862708727088270892709027091270922709327094270952709627097270982709927100271012710227103271042710527106271072710827109271102711127112271132711427115271162711727118271192712027121271222712327124271252712627127271282712927130271312713227133271342713527136271372713827139271402714127142271432714427145271462714727148271492715027151271522715327154271552715627157271582715927160271612716227163271642716527166271672716827169271702717127172271732717427175271762717727178271792718027181271822718327184271852718627187271882718927190271912719227193271942719527196271972719827199272002720127202272032720427205272062720727208272092721027211272122721327214272152721627217272182721927220272212722227223272242722527226272272722827229272302723127232272332723427235272362723727238272392724027241272422724327244272452724627247272482724927250272512725227253272542725527256272572725827259272602726127262272632726427265272662726727268272692727027271272722727327274272752727627277272782727927280272812728227283272842728527286272872728827289272902729127292272932729427295272962729727298272992730027301273022730327304273052730627307273082730927310273112731227313273142731527316273172731827319273202732127322273232732427325273262732727328273292733027331273322733327334273352733627337273382733927340273412734227343273442734527346273472734827349273502735127352273532735427355273562735727358273592736027361273622736327364273652736627367273682736927370273712737227373273742737527376273772737827379273802738127382273832738427385273862738727388273892739027391273922739327394273952739627397273982739927400274012740227403274042740527406274072740827409274102741127412274132741427415274162741727418274192742027421274222742327424274252742627427274282742927430274312743227433274342743527436274372743827439274402744127442274432744427445274462744727448274492745027451274522745327454274552745627457274582745927460274612746227463274642746527466274672746827469274702747127472274732747427475274762747727478274792748027481274822748327484274852748627487274882748927490274912749227493274942749527496274972749827499275002750127502275032750427505275062750727508275092751027511275122751327514275152751627517275182751927520275212752227523275242752527526275272752827529275302753127532275332753427535275362753727538275392754027541275422754327544275452754627547275482754927550275512755227553275542755527556275572755827559275602756127562275632756427565275662756727568275692757027571275722757327574275752757627577275782757927580275812758227583275842758527586275872758827589275902759127592275932759427595275962759727598275992760027601276022760327604276052760627607276082760927610276112761227613276142761527616276172761827619276202762127622276232762427625276262762727628276292763027631276322763327634276352763627637276382763927640276412764227643276442764527646276472764827649276502765127652276532765427655276562765727658276592766027661276622766327664276652766627667276682766927670276712767227673276742767527676276772767827679276802768127682276832768427685276862768727688276892769027691276922769327694276952769627697276982769927700277012770227703277042770527706277072770827709277102771127712277132771427715277162771727718277192772027721277222772327724277252772627727277282772927730277312773227733277342773527736277372773827739277402774127742277432774427745277462774727748277492775027751277522775327754277552775627757277582775927760277612776227763277642776527766277672776827769277702777127772277732777427775277762777727778277792778027781277822778327784277852778627787277882778927790277912779227793277942779527796277972779827799278002780127802278032780427805278062780727808278092781027811278122781327814278152781627817278182781927820278212782227823278242782527826278272782827829278302783127832278332783427835278362783727838278392784027841278422784327844278452784627847278482784927850278512785227853278542785527856278572785827859278602786127862278632786427865278662786727868278692787027871278722787327874278752787627877278782787927880278812788227883278842788527886278872788827889278902789127892278932789427895278962789727898278992790027901279022790327904279052790627907279082790927910279112791227913279142791527916279172791827919279202792127922279232792427925279262792727928279292793027931279322793327934279352793627937279382793927940279412794227943279442794527946279472794827949279502795127952279532795427955279562795727958279592796027961279622796327964279652796627967279682796927970279712797227973279742797527976279772797827979279802798127982279832798427985279862798727988279892799027991279922799327994279952799627997279982799928000280012800228003280042800528006280072800828009280102801128012280132801428015280162801728018280192802028021280222802328024280252802628027280282802928030280312803228033280342803528036280372803828039280402804128042280432804428045280462804728048280492805028051280522805328054280552805628057280582805928060280612806228063280642806528066280672806828069280702807128072280732807428075280762807728078280792808028081280822808328084280852808628087280882808928090280912809228093280942809528096280972809828099281002810128102281032810428105281062810728108281092811028111281122811328114281152811628117281182811928120281212812228123281242812528126281272812828129281302813128132281332813428135281362813728138281392814028141 |
- /*##############################################################################
- HPCC SYSTEMS software Copyright (C) 2012 HPCC Systems®.
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
- http://www.apache.org/licenses/LICENSE-2.0
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- ############################################################################## */
- #include "jmisc.hpp"
- #include "jdebug.hpp"
- #include "jptree.hpp"
- #include "rtlkey.hpp"
- #include "jsort.hpp"
- #include "jhtree.hpp"
- #include "jqueue.tpp"
- #include "jisem.hpp"
- #include "thorxmlread.hpp"
- #include "thorrparse.ipp"
- #include "thorxmlwrite.hpp"
- #include "thorsoapcall.hpp"
- #include "thorcommon.ipp"
- #include "thorsort.hpp"
- #include "thorstats.hpp"
- #include "jlzw.hpp"
- #include "javahash.hpp"
- #include "javahash.tpp"
- #include "thorstep.ipp"
- #include "thorpipe.hpp"
- #include "thorfile.hpp"
- #include "eclhelper.hpp"
- #include "eclrtl_imp.hpp"
- #include "rtlfield_imp.hpp"
- #include "rtlds_imp.hpp"
- #include "rtlread_imp.hpp"
- #include <algorithm>
- #include "dafdesc.hpp"
- #include "dautils.hpp"
- #include "ftbase.ipp"
- namespace ccdserver_hqlhelper
- {
- #include "eclhelper_base.hpp"
- }
- #include "ccd.hpp"
- #include "ccdserver.hpp"
- #include "ccdcontext.hpp"
- #include "ccdactivities.hpp"
- #include "ccdquery.hpp"
- #include "ccdstate.hpp"
- #include "ccdqueue.ipp"
- #include "ccdsnmp.hpp"
- #include "ccddali.hpp"
- #include "jsmartsock.hpp"
- #include "dllserver.hpp"
- #include "workflow.hpp"
- #include "roxiemem.hpp"
- #include "roxierowbuff.hpp"
- #include "roxiehelper.hpp"
- #include "roxielmj.hpp"
- #include "roxierow.hpp"
- #include "thorplugin.hpp"
- #include "keybuild.hpp"
- #include "thorstrand.hpp"
- #define MAX_HTTP_HEADERSIZE 8000
- #define MIN_PAYLOAD_SIZE 800
- #pragma warning(disable : 4355)
- #define DEFAULT_PARALLEL_LOOP_THREADS 1
- #define PROBE
- #ifdef _DEBUG
- //#define FAKE_EXCEPTIONS
- //#define TRACE_JOINGROUPS
- //#define TRACE_SPLIT
- //#define _CHECK_HEAPSORT
- //#undef PARALLEL_EXECUTE
- //#define TRACE_SEEK_REQUESTS
- #endif
- using roxiemem::OwnedRoxieRow;
- using roxiemem::OwnedRoxieString;
- using roxiemem::OwnedConstRoxieRow;
- using roxiemem::IRowManager;
- using roxiemem::ReleaseRoxieRows;
- // There is a bug in VC6 implemetation of protected which prevents nested classes from accessing owner's data. It can be tricky to work around - hence...
- #if _MSC_VER==1200
- #define protected public
- #endif
- #define TRACE_STARTSTOP // This determines if it is available - it is enabled/disabled by a configuration option
- #define TRACE_STRANDS
- #define MAX_SENSIBLE_STRANDS 1024 // Architecture dependent...
- static const SmartStepExtra dummySmartStepExtra(SSEFreadAhead, NULL);
- //=================================================================================
- class RestartableThread : public CInterface
- {
- class MyThread : public Thread
- {
- Linked<RestartableThread> owner;
- public:
- MyThread(RestartableThread *_owner, const char *name) : Thread(name), owner(_owner)
- {
- }
- virtual int run()
- {
- owner->started.signal();
- return owner->run();
- }
- };
- friend class MyThread;
- Semaphore started;
- Owned<MyThread> thread;
- CriticalSection crit;
- StringAttr name;
- public:
- RestartableThread(const char *_name) : name(_name)
- {
- }
- virtual void start(const char *namePrefix)
- {
- StringBuffer s(namePrefix);
- s.append(name);
- {
- CriticalBlock b(crit);
- assertex(!thread);
- thread.setown(new MyThread(this, s));
- thread->start();
- }
- started.wait();
- }
- virtual void join()
- {
- {
- Owned<Thread> tthread;
- {
- CriticalBlock b(crit);
- tthread.setown(thread.getClear());
- }
- if (tthread)
- tthread->join();
- }
- }
- virtual int run() = 0;
- };
- //================================================================================
- //The following don't link their arguments because that creates a circular reference
- //But I wish there was a better way
- class IndirectSlaveContext : implements IRoxieSlaveContext, public CInterface
- {
- public:
- IndirectSlaveContext(IRoxieSlaveContext * _ctx) : ctx(_ctx) {}
- IMPLEMENT_IINTERFACE
- // void set(IRoxieSlaveContext * _ctx) { ctx = _ctx; }
- virtual ICodeContext *queryCodeContext()
- {
- return ctx->queryCodeContext();
- }
- virtual void checkAbort()
- {
- ctx->checkAbort();
- }
- virtual void notifyAbort(IException *E)
- {
- ctx->notifyAbort(E);
- }
- virtual IActivityGraph * queryChildGraph(unsigned id)
- {
- return ctx->queryChildGraph(id);
- }
- virtual void noteChildGraph(unsigned id, IActivityGraph *childGraph)
- {
- ctx->noteChildGraph(id, childGraph) ;
- }
- virtual IRowManager &queryRowManager()
- {
- return ctx->queryRowManager();
- }
- virtual void noteStatistic(StatisticKind kind, unsigned __int64 value) const
- {
- ctx->noteStatistic(kind, value);
- }
- virtual void mergeStats(const CRuntimeStatisticCollection &from) const
- {
- ctx->mergeStats(from);
- }
- virtual const CRuntimeStatisticCollection &queryStats() const
- {
- return ctx->queryStats();
- }
- virtual void CTXLOGva(const char *format, va_list args) const __attribute__((format(printf,2,0)))
- {
- ctx->CTXLOGva(format, args);
- }
- virtual void CTXLOGa(TracingCategory category, const char *prefix, const char *text) const
- {
- ctx->CTXLOGa(category, prefix, text);
- }
- virtual void logOperatorExceptionVA(IException *E, const char *file, unsigned line, const char *format, va_list args) const __attribute__((format(printf,5,0)))
- {
- ctx->logOperatorExceptionVA(E, file, line, format, args);
- }
- virtual void CTXLOGaeva(IException *E, const char *file, unsigned line, const char *prefix, const char *format, va_list args) const __attribute__((format(printf,6,0)))
- {
- ctx->CTXLOGaeva(E, file, line, prefix, format, args);
- }
- virtual void CTXLOGl(LogItem *log) const
- {
- ctx->CTXLOGl(log);
- }
- virtual StringBuffer &getLogPrefix(StringBuffer &ret) const
- {
- return ctx->getLogPrefix(ret);
- }
- virtual unsigned queryTraceLevel() const
- {
- return ctx->queryTraceLevel();
- }
- virtual bool isIntercepted() const
- {
- return ctx->isIntercepted();
- }
- virtual bool isBlind() const
- {
- return ctx->isBlind();
- }
- virtual const QueryOptions &queryOptions() const
- {
- return ctx->queryOptions();
- }
- virtual void addSlavesReplyLen(unsigned len)
- {
- ctx->addSlavesReplyLen(len);
- }
- virtual const char *queryAuthToken()
- {
- return ctx->queryAuthToken();
- }
- virtual const IResolvedFile *resolveLFN(const char *filename, bool isOpt)
- {
- return ctx->resolveLFN(filename, isOpt);
- }
- virtual IRoxieWriteHandler *createLFN(const char *filename, bool overwrite, bool extend, const StringArray &clusters)
- {
- return ctx->createLFN(filename, overwrite, extend, clusters);
- }
- virtual void onFileCallback(const RoxiePacketHeader &header, const char *lfn, bool isOpt, bool isLocal)
- {
- ctx->onFileCallback(header, lfn, isOpt, isLocal);
- }
- virtual IActivityGraph *getLibraryGraph(const LibraryCallFactoryExtra &extra, IRoxieServerActivity *parentActivity)
- {
- return ctx->getLibraryGraph(extra, parentActivity);
- }
- virtual void noteProcessed(unsigned subgraphId, unsigned activityId, unsigned _idx, unsigned _processed, unsigned _strands) const
- {
- ctx->noteProcessed(subgraphId, activityId, _idx, _processed, _strands);
- }
- virtual void mergeActivityStats(const CRuntimeStatisticCollection &fromStats, unsigned subgraphId, unsigned activityId, const ActivityTimeAccumulator &_totalCycles, cycle_t _localCycles) const
- {
- ctx->mergeActivityStats(fromStats, subgraphId, activityId, _totalCycles, _localCycles);
- }
- virtual IProbeManager *queryProbeManager() const
- {
- return ctx->queryProbeManager();
- }
- virtual IDebuggableContext *queryDebugContext() const
- {
- return ctx->queryDebugContext();
- }
- virtual void printResults(IXmlWriter *output, const char *name, unsigned sequence)
- {
- ctx->printResults(output, name, sequence);
- }
- virtual void setWUState(WUState state)
- {
- ctx->setWUState(state);
- }
- virtual bool checkWuAborted()
- {
- return ctx->checkWuAborted();
- }
- virtual IWorkUnit *updateWorkUnit() const
- {
- return ctx->updateWorkUnit();
- }
- virtual IConstWorkUnit *queryWorkUnit() const
- {
- return ctx->queryWorkUnit();
- }
- virtual IRoxieServerContext *queryServerContext()
- {
- return ctx->queryServerContext();
- }
- virtual IWorkUnitRowReader *getWorkunitRowReader(const char *wuid, const char * name, unsigned sequence, IXmlToRowTransformer * xmlTransformer, IEngineRowAllocator *rowAllocator, bool isGrouped)
- {
- return ctx->getWorkunitRowReader(wuid, name, sequence, xmlTransformer, rowAllocator, isGrouped);
- }
- virtual IEngineRowAllocator *getRowAllocatorEx(IOutputMetaData * meta, unsigned activityId, roxiemem::RoxieHeapFlags flags) const
- {
- return ctx->getRowAllocatorEx(meta, activityId, flags);
- }
- protected:
- IRoxieSlaveContext * ctx;
- };
- //=================================================================================
- #define RESULT_FLUSH_THRESHOLD 10000u
- #ifdef _DEBUG
- #define SOAP_SPLIT_THRESHOLD 100u
- #define SOAP_SPLIT_RESERVE 200u
- #else
- #define SOAP_SPLIT_THRESHOLD 64000u
- #define SOAP_SPLIT_RESERVE 65535u
- #endif
- //=================================================================================
- // General activity statistics
- static const StatisticsMapping actStatistics(StWhenFirstRow, StTimeElapsed, StTimeLocalExecute, StTimeTotalExecute, StSizeMaxRowSize,
- StNumRowsProcessed, StNumSlaves, StNumStarted, StNumStopped, StNumStrands,
- StNumScansPerRow, StNumAllocations, StNumAllocationScans, StKindNone);
- static const StatisticsMapping joinStatistics(&actStatistics, StNumAtmostTriggered, StKindNone);
- static const StatisticsMapping keyedJoinStatistics(&joinStatistics, StNumServerCacheHits, StNumIndexSeeks, StNumIndexScans, StNumIndexWildSeeks,
- StNumIndexSkips, StNumIndexNullSkips, StNumIndexMerges, StNumIndexMergeCompares,
- StNumPreFiltered, StNumPostFiltered, StNumIndexAccepted, StNumIndexRejected,
- StNumIndexRowsRead, StNumDiskRowsRead, StNumDiskSeeks, StNumDiskAccepted,
- StNumBlobCacheHits, StNumLeafCacheHits, StNumNodeCacheHits,
- StNumBlobCacheAdds, StNumLeafCacheAdds, StNumNodeCacheAdds,
- StNumDiskRejected, StKindNone);
- static const StatisticsMapping indexStatistics(&actStatistics, StNumServerCacheHits, StNumIndexSeeks, StNumIndexScans, StNumIndexWildSeeks,
- StNumIndexSkips, StNumIndexNullSkips, StNumIndexMerges, StNumIndexMergeCompares,
- StNumPreFiltered, StNumPostFiltered, StNumIndexAccepted, StNumIndexRejected,
- StNumBlobCacheHits, StNumLeafCacheHits, StNumNodeCacheHits,
- StNumBlobCacheAdds, StNumLeafCacheAdds, StNumNodeCacheAdds,
- StNumIndexRowsRead, StKindNone);
- static const StatisticsMapping diskStatistics(&actStatistics, StNumServerCacheHits, StNumDiskRowsRead, StNumDiskSeeks, StNumDiskAccepted,
- StNumDiskRejected, StKindNone);
- static const StatisticsMapping soapStatistics(&actStatistics, StTimeSoapcall, StKindNone);
- static const StatisticsMapping groupStatistics(&actStatistics, StNumGroups, StNumGroupMax, StKindNone);
- static const StatisticsMapping sortStatistics(&actStatistics, StTimeSortElapsed, StKindNone);
- //=================================================================================
- const static unsigned minus1U = (0U-1U);
- class CRoxieServerActivityFactoryBase : public CActivityFactory, implements IRoxieServerActivityFactory
- {
- protected:
- IntArray dependencies; // things I am dependent on
- IntArray dependencyIndexes; // things I am dependent on
- IntArray dependencyControlIds; // things I am dependent on
- StringArray dependencyEdgeIds; // How to describe them to the debugger
- unsigned dependentCount; // things dependent on me
- unsigned optParallel = 0; // -1=enable,0=not specified,1=single-threaded,n=<num threads>
- bool optStableInput = true; // is the input forced to ordered?
- bool optUnstableInput = false; // is the input forced to unordered?
- bool optUnordered = false; // is the output specified as unordered?
- unsigned heapFlags;
- mutable CriticalSection statsCrit;
- mutable __int64 processed;
- mutable __int64 started;
- mutable __int64 totalCycles;
- mutable __int64 localCycles;
- public:
- IMPLEMENT_IINTERFACE;
- CRoxieServerActivityFactoryBase(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind)
- {
- processed = 0;
- started = 0;
- totalCycles = 0;
- localCycles = 0;
- dependentCount = 0;
- optParallel = _graphNode.getPropInt("att[@name='parallel']/@value", 0);
- optUnordered = !_graphNode.getPropBool("att[@name='ordered']/@value", true);
- heapFlags = _graphNode.getPropInt("hint[@name='heapflags']/@value", _queryFactory.queryOptions().heapFlags);
- }
-
- ~CRoxieServerActivityFactoryBase()
- {
- }
- StringBuffer &toString(StringBuffer &ret) const
- {
- return ret.appendf("%p", this);
- }
- virtual void addDependency(unsigned _source, ThorActivityKind _kind, unsigned _sourceIdx, int controlId, const char *edgeId)
- {
- dependencies.append(_source);
- dependencyIndexes.append(_sourceIdx);
- dependencyControlIds.append(controlId);
- dependencyEdgeIds.append(edgeId);
- }
- virtual bool isInputOrdered(bool consumerOrdered, unsigned idx) const
- {
- if (optStableInput)
- return true;
- if (optUnstableInput)
- return false;
- if (optUnordered)
- return false;
- return consumerOrdered;
- }
- virtual void noteDependent(unsigned target)
- {
- dependentCount++;
- }
- virtual IntArray &queryDependencies() { return dependencies; }
- virtual IntArray &queryDependencyIndexes() { return dependencyIndexes; }
- virtual IntArray &queryDependencyControlIds() { return dependencyControlIds; }
- virtual StringArray &queryDependencyEdgeIds() { return dependencyEdgeIds; }
- virtual unsigned queryId() const { return id; }
- virtual unsigned querySubgraphId() const { return subgraphId; }
- virtual ThorActivityKind getKind() const { return kind; }
- virtual IOutputMetaData * queryOutputMeta() const
- {
- return meta;
- }
- virtual bool isSink() const
- {
- return false;
- }
- virtual bool isFunction() const
- {
- return false;
- }
- virtual bool isGraphInvariant() const
- {
- return false;
- }
- virtual IHThorArg &getHelper() const
- {
- return *helperFactory();
- }
- virtual IRoxieServerActivity *createFunction(IHThorArg &arg, IProbeManager *_probeManager) const
- {
- arg.Release();
- throwUnexpected();
- }
- virtual void mergeStats(const CRuntimeStatisticCollection &from) const
- {
- CActivityFactory::mergeStats(from);
- }
- virtual void noteProcessed(unsigned _idx, unsigned _processed) const
- {
- dbgassertex(!_idx);
- if (_processed)
- {
- CriticalBlock b(statsCrit);
- processed += _processed;
- }
- }
- virtual void mergeActivityStats(const CRuntimeStatisticCollection &fromStats, const ActivityTimeAccumulator &_totalCycles, cycle_t _localCycles) const
- {
- if (_totalCycles.totalCycles || _localCycles)
- {
- CriticalBlock b(statsCrit);
- dbgassertex(_totalCycles.totalCycles >= _localCycles);
- totalCycles += _totalCycles.totalCycles;
- localCycles += _localCycles;
- }
- CActivityFactory::mergeStats(fromStats);
- }
- virtual void noteStarted() const
- {
- CriticalBlock b(statsCrit);
- started ++;
- }
- virtual void noteStarted(unsigned idx) const
- {
- throwUnexpected(); // should be implemented/required by multiOutput cases only
- }
- virtual void getEdgeProgressInfo(unsigned output, IPropertyTree &edge) const
- {
- CriticalBlock b(statsCrit);
- if (output == 0)
- {
- putStatsValue(&edge, "count", "sum", processed);
- if (started)
- putStatsValue(&edge, "started", "sum", started);
- }
- else
- ERRLOG("unexpected call to getEdgeProcessInfo for output %d in activity %d", output, queryId());
- }
- virtual void getNodeProgressInfo(IPropertyTree &node) const
- {
- CActivityFactory::getNodeProgressInfo(node);
- CriticalBlock b(statsCrit);
- if (started)
- putStatsValue(&node, "_roxieStarted", "sum", started);
- if (totalCycles)
- putStatsValue(&node, "totalTime", "sum", (unsigned) (cycle_to_nanosec(totalCycles)/1000));
- if (localCycles)
- putStatsValue(&node, "localTime", "sum", (unsigned) (cycle_to_nanosec(localCycles)/1000));
- }
- virtual void resetNodeProgressInfo()
- {
- CActivityFactory::resetNodeProgressInfo();
- CriticalBlock b(statsCrit);
- processed = 0;
- started = 0;
- totalCycles = 0;
- localCycles = 0;
- }
- virtual void getActivityMetrics(StringBuffer &reply) const
- {
- CActivityFactory::getActivityMetrics(reply);
- CriticalBlock b(statsCrit);
- putStatsValue(reply, "_roxieStarted", "sum", started);
- putStatsValue(reply, "totalTime", "sum", (unsigned) (cycle_to_nanosec(totalCycles)/1000));
- putStatsValue(reply, "localTime", "sum", (unsigned) (cycle_to_nanosec(localCycles)/1000));
- }
- virtual unsigned __int64 queryLocalCycles() const
- {
- return localCycles;
- }
- virtual IQueryFactory &queryQueryFactory() const
- {
- return CActivityFactory::queryQueryFactory();
- }
- virtual ActivityArray *queryChildQuery(unsigned idx, unsigned &id)
- {
- return CActivityFactory::queryChildQuery(idx, id);
- }
- virtual void addChildQuery(unsigned id, ActivityArray *childQuery)
- {
- CActivityFactory::addChildQuery(id, childQuery);
- }
- virtual void onCreateChildQueries(IRoxieSlaveContext *ctx, IHThorArg *colocalArg, IArrayOf<IActivityGraph> &childGraphs, IRoxieServerActivity *parentActivity, IProbeManager *_probeManager, const IRoxieContextLogger &_logctx, unsigned numParallel) const
- {
- ForEachItemIn(idx, childQueries)
- {
- childGraphs.append(*createActivityGraph(ctx, NULL, childQueryIndexes.item(idx), childQueries.item(idx), parentActivity, _probeManager, _logctx, numParallel));
- ctx->noteChildGraph(childQueryIndexes.item(idx), &childGraphs.item(idx));
- childGraphs.item(idx).onCreate(colocalArg);
- }
- }
- IActivityGraph * createChildGraph(IRoxieSlaveContext * ctx, IHThorArg *colocalArg, unsigned childId, IRoxieServerActivity *parentActivity, IProbeManager * _probeManager, const IRoxieContextLogger &_logctx) const
- {
- unsigned match = childQueryIndexes.find(childId);
- assertex(match != NotFound);
- Owned<IActivityGraph> graph = createActivityGraph(ctx, NULL, childQueryIndexes.item(match), childQueries.item(match), parentActivity, _probeManager, _logctx, 1);
- graph->onCreate(colocalArg);
- return graph.getClear();
- }
- virtual IRoxieServerSideCache *queryServerSideCache() const
- {
- return NULL; // Activities that wish to support server-side caching will need to do better....
- }
- virtual bool getEnableFieldTranslation() const
- {
- throwUnexpected(); // only implemented by index-related subclasses
- }
- virtual IDefRecordMeta *queryActivityMeta() const
- {
- throwUnexpected(); // only implemented by index-related subclasses
- }
- virtual void getXrefInfo(IPropertyTree &reply, const IRoxieContextLogger &logctx) const
- {
- // Most activities have nothing to say...
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return actStatistics; // Overridden by anyone that needs more
- }
- virtual roxiemem::RoxieHeapFlags getHeapFlags() const
- {
- return (roxiemem::RoxieHeapFlags)heapFlags;
- }
- };
- class CRoxieServerMultiInputInfo
- {
- private:
- UnsignedArray inputs;
- UnsignedArray inputIndexes;
- public:
- void set(unsigned idx, unsigned source, unsigned sourceidx)
- {
- if (idx==inputs.length())
- {
- inputs.append(source);
- inputIndexes.append(sourceidx);
- }
- else
- {
- while (!inputs.isItem(idx))
- {
- inputs.append(0);
- inputIndexes.append(0);
- }
- inputs.replace(source, idx);
- inputIndexes.replace(sourceidx, idx);
- }
- }
- unsigned get(unsigned idx, unsigned &sourceidx) const
- {
- if (inputs.isItem(idx))
- {
- sourceidx = inputIndexes.item(idx);
- return inputs.item(idx);
- }
- else
- return (unsigned) -1;
- }
- inline unsigned ordinality() const { return inputs.ordinality(); }
- };
- class CRoxieServerMultiInputFactory : public CRoxieServerActivityFactoryBase
- {
- private:
- CRoxieServerMultiInputInfo inputs;
- public:
- CRoxieServerMultiInputFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactoryBase(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- inputs.set(idx, source, sourceidx);
- }
- virtual unsigned getInput(unsigned idx, unsigned &sourceidx) const
- {
- return inputs.get(idx, sourceidx);
- }
- virtual unsigned numInputs() const { return inputs.ordinality(); }
- };
- class CWrappedException : implements IException, public CInterface
- {
- Owned<IException> wrapped;
- ThorActivityKind kind;
- unsigned queryId;
- public:
- IMPLEMENT_IINTERFACE;
- CWrappedException(IException *_wrapped, ThorActivityKind _kind, unsigned _queryId)
- : wrapped(_wrapped), kind(_kind), queryId(_queryId)
- {
- }
- virtual int errorCode() const { return wrapped->errorCode(); }
- virtual StringBuffer & errorMessage(StringBuffer &msg) const { return wrapped->errorMessage(msg).appendf(" (in %s %d)", getActivityText(kind), queryId); }
- virtual MessageAudience errorAudience() const { return wrapped->errorAudience(); }
- };
- class CRoxieServerActivityFactory : public CRoxieServerActivityFactoryBase
- {
- protected:
- unsigned input;
- unsigned inputidx;
- public:
- CRoxieServerActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactoryBase(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- input = (unsigned) -1;
- inputidx = 0;
- }
- inline void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- if (idx != 0)
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: id = %d : setInput() parameter out of bounds idx = %d at %s(%d)", id, idx, __FILE__, __LINE__);
- if (input != -1)
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: id = %d : setInput() called twice for input = %d source = %d inputidx = %d sourceidx = %d at %s(%d)", id, input, source, inputidx, sourceidx, __FILE__, __LINE__);
- input = source;
- inputidx = sourceidx;
- }
- virtual unsigned getInput(unsigned idx, unsigned &sourceidx) const
- {
- if (!idx)
- {
- sourceidx = inputidx;
- return input;
- }
- return (unsigned) -1;
- }
- virtual unsigned numInputs() const { return (input == (unsigned)-1) ? 0 : 1; }
- };
- class CRoxieServerMultiOutputFactory : public CRoxieServerActivityFactory
- {
- protected:
- unsigned numOutputs;
- unsigned __int64 *processedArray;
- bool *startedArray;
- CRoxieServerMultiOutputFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- numOutputs = 0;
- processedArray = NULL;
- startedArray = NULL;
- }
- ~CRoxieServerMultiOutputFactory()
- {
- delete [] processedArray;
- delete [] startedArray;
- }
- void setNumOutputs(unsigned num)
- {
- numOutputs = num;
- if (!num)
- num = 1; // Even sink activities like to track how many records they process
- processedArray = new unsigned __int64[num];
- startedArray = new bool[num];
- for (unsigned i = 0; i < num; i++)
- {
- processedArray[i] = 0;
- startedArray[i] = 0;
- }
- }
- virtual void getEdgeProgressInfo(unsigned idx, IPropertyTree &edge) const
- {
- assertex(numOutputs ? idx < numOutputs : idx==0);
- CriticalBlock b(statsCrit);
- putStatsValue(&edge, "count", "sum", processedArray[idx]);
- putStatsValue(&edge, "started", "sum", startedArray[idx]);
- }
- virtual void resetNodeProgressInfo()
- {
- CRoxieServerActivityFactory::resetNodeProgressInfo();
- CriticalBlock b(statsCrit);
- for (unsigned i = 0; i < numOutputs; i++)
- {
- processedArray[i] = 0;
- startedArray[i] = 0;
- }
- }
- virtual void noteProcessed(unsigned idx, unsigned _processed) const
- {
- assertex(numOutputs ? idx < numOutputs : idx==0);
- CriticalBlock b(statsCrit);
- processedArray[idx] += _processed;
- }
- virtual void noteStarted(unsigned idx) const
- {
- assertex(numOutputs ? idx < numOutputs : idx==0);
- CriticalBlock b(statsCrit);
- startedArray[idx] = true;
- }
- };
- class CRoxieServerInternalSinkFactory : public CRoxieServerActivityFactory
- {
- protected:
- bool isInternal;
- bool isRoot;
- unsigned usageCount;
- public:
- CRoxieServerInternalSinkFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- usageCount = _usageCount;
- isRoot = _isRoot;
- isInternal = false; // filled in by derived class constructor
- }
- virtual bool isSink() const
- {
- //only a sink if a root activity
- return isRoot && !(isInternal && dependentCount && dependentCount==usageCount); // MORE - it's possible for this to get the answer wrong still, since usageCount does not include references from main procedure. Gavin?
- }
- virtual void getEdgeProgressInfo(unsigned idx, IPropertyTree &edge) const
- {
- // There is no meaningful info to return along the dependency edge - we don't detect how many times the value has been read from the context
- // Just leave it blank is safest.
- }
- };
- typedef enum { STATEreset, STATEstarted, STATEstopped, STATEstarting } activityState;
- const char *queryStateText(activityState state)
- {
- switch (state)
- {
- case STATEreset: return "reset";
- case STATEstarted: return "started";
- case STATEstopped: return "stopped";
- case STATEstarting: return "starting";
- default: return "unknown";
- }
- }
- typedef ICopyArrayOf<IRoxieServerActivity> IRoxieServerActivityCopyArray;
- class CParallelActivityExecutor : public CAsyncFor
- {
- public:
- unsigned parentExtractSize;
- const byte * parentExtract;
- CParallelActivityExecutor(IRoxieServerActivityCopyArray & _activities, unsigned _parentExtractSize, const byte * _parentExtract) :
- activities(_activities), parentExtractSize(_parentExtractSize), parentExtract(_parentExtract) { }
- void Do(unsigned i)
- {
- activities.item(i).execute(parentExtractSize, parentExtract);
- }
- private:
- IRoxieServerActivityCopyArray & activities;
- };
- extern IEngineRowStream * ensureSingleStream(IRoxieSlaveContext *ctx, PointerArrayOf<IEngineRowStream> & instreams, Owned<IStrandJunction> &junction)
- {
- if (instreams.length() != 1)
- {
- assertex(instreams.length());
- if (!junction)
- junction.setown(createStrandJunction(ctx->queryRowManager(), instreams.length(), 1, ctx->queryOptions().strandBlockSize, false));
- ForEachItemIn(stream, instreams)
- {
- junction->setInput(stream, instreams.item(stream));
- }
- return junction->queryOutput(0);
- }
- else
- return instreams.item(0);
- }
- extern IEngineRowStream *connectSingleStream(IRoxieSlaveContext *ctx, IFinalRoxieInput *input, unsigned idx, Owned<IStrandJunction> &junction, bool consumerOrdered)
- {
- if (input)
- {
- PointerArrayOf<IEngineRowStream> instreams;
- junction.setown(input->getOutputStreams(ctx, idx, instreams, NULL, consumerOrdered, nullptr));
- return ensureSingleStream(ctx, instreams, junction);
- }
- else
- return NULL;
- }
- extern IEngineRowStream *connectSingleStream(IRoxieSlaveContext *ctx, IFinalRoxieInput *input, unsigned idx, bool consumerOrdered)
- {
- Owned<IStrandJunction> junction;
- IEngineRowStream * result = connectSingleStream(ctx, input, idx, junction, consumerOrdered);
- assertex(!junction);
- return result;
- }
- class CRoxieServerActivity : implements CInterfaceOf<IRoxieServerActivity>, implements IFinalRoxieInput, implements IEngineRowStream, implements IRoxieContextLogger
- {
- friend class StrandProcessor;
- protected:
- IFinalRoxieInput *input;
- unsigned sourceIdx;
- IEngineRowStream *inputStream;
- Owned<IStrandJunction> junction;
- IHThorArg &basehelper;
- IRoxieSlaveContext *ctx;
- const IRoxieServerActivityFactory *factory;
- IRoxieServerActivityCopyArray dependencies;
- IntArray dependencyIndexes;
- IntArray dependencyControlIds;
- IArrayOf<IActivityGraph> childGraphs;
- CachedOutputMetaData meta;
- IHThorArg *colocalParent;
- IEngineRowAllocator *rowAllocator;
- CriticalSection statecrit;
- CriticalSection statscrit;
- mutable CRuntimeStatisticCollection stats;
- MapStringToMyClass<ThorSectionTimer> functionTimers;
- unsigned processed;
- ActivityTimeAccumulator totalCycles;
- cycle_t localCycles;
- unsigned activityId;
- activityState state;
- bool createPending;
- bool debugging;
- bool timeActivities;
- bool aborted;
- bool connected = false;
- IProbeManager *probeManager = NULL;
- public:
- IMPLEMENT_IINTERFACE_USING(CInterfaceOf<IRoxieServerActivity>)
- CRoxieServerActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : ctx(_ctx),
- factory(_factory),
- basehelper(_factory->getHelper()),
- activityId(_factory->queryId()),
- stats(_factory->queryStatsMapping()),
- probeManager(_probeManager)
- {
- input = NULL;
- sourceIdx = 0;
- inputStream = NULL;
- meta.set(basehelper.queryOutputMeta());
- processed = 0;
- localCycles = 0;
- state=STATEreset;
- rowAllocator = NULL;
- debugging = _probeManager != NULL; // Don't want to collect timing stats from debug sessions
- colocalParent = NULL;
- createPending = true;
- timeActivities = defaultTimeActivities;
- aborted = false;
- }
-
- CRoxieServerActivity(IRoxieSlaveContext *_ctx, IHThorArg & _helper) : ctx(_ctx), factory(NULL), basehelper(_helper), stats(allStatistics)
- {
- activityId = 0;
- input = NULL;
- sourceIdx = 0;
- inputStream = NULL;
- ctx = NULL;
- meta.set(basehelper.queryOutputMeta());
- processed = 0;
- localCycles = 0;
- state=STATEreset;
- rowAllocator = NULL;
- debugging = false;
- colocalParent = NULL;
- createPending = true;
- timeActivities = defaultTimeActivities;
- aborted = false;
- }
- ~CRoxieServerActivity()
- {
- if (traceStartStop)
- {
- // There was an old comment here stating // Note- CTXLOG may not be safe
- // There were problems in the destruction order of graphs that might mean an IndirectSlaveContext
- // was released while activities still referenced it.
- // However these should now all be fixed
- DBGLOG("%p destroy %d state=%s", this, activityId, queryStateText(state));
- if (watchActivityId && watchActivityId==activityId)
- {
- DBGLOG("WATCH: %p destroy %d state=%s", this, activityId, queryStateText(state));
- }
- }
- if (state!=STATEreset)
- {
- DBGLOG("STATE: Activity %d destroyed but not reset", activityId);
- state = STATEreset; // bit pointless but there you go...
- }
- if (factory && !debugging)
- {
- if (processed)
- factory->noteProcessed(0, processed);
- factory->mergeActivityStats(stats, totalCycles, localCycles);
- }
- if (ctx && factory)
- {
- if (processed)
- ctx->noteProcessed(factory->querySubgraphId(), activityId, 0, processed, 0);
- ctx->mergeActivityStats(stats, factory->querySubgraphId(), activityId, totalCycles, localCycles);
- }
- basehelper.Release();
- ::Release(rowAllocator);
- }
- virtual const IRoxieContextLogger &queryLogCtx()const
- {
- return *this;
- }
- inline const StatisticsMapping & queryStatsMapping() const
- {
- return factory ? factory->queryStatsMapping() : actStatistics;
- }
- virtual void mergeStats(MemoryBuffer &buf)
- {
- stats.deserializeMerge(buf);
- }
- virtual ISectionTimer *registerTimer(unsigned activityId, const char * name)
- {
- CriticalBlock b(statscrit); // reuse statscrit to protect functionTimers - it will not be held concurrently
- ISectionTimer *timer = functionTimers.getValue(name);
- if (!timer)
- {
- timer = ThorSectionTimer::createTimer(stats, name);
- functionTimers.setValue(name, timer);
- timer->Release(); // Value returned is not linked
- }
- return timer;
- }
- virtual IRoxieServerActivity * queryChildActivity(unsigned activityId)
- {
- ForEachItemIn(i, childGraphs)
- {
- IRoxieServerActivity * activity = childGraphs.item(i).queryActivity(activityId);
- if (activity)
- return activity;
- }
- #ifdef _DEBUG
- throwUnexpectedX("Unable to map child activity id to an activity");
- #endif
- return nullptr;
- }
- void mergeStrandStats(unsigned strandProcessed, const ActivityTimeAccumulator & strandCycles, const CRuntimeStatisticCollection & strandStats)
- {
- CriticalBlock cb(statscrit);
- processed += strandProcessed;
- totalCycles.merge(strandCycles);
- stats.merge(strandStats);
- }
- inline void ensureRowAllocator()
- {
- if (!rowAllocator)
- rowAllocator = createRowAllocator(meta.queryOriginal());
- }
- virtual IEngineRowAllocator * createRowAllocator(IOutputMetaData * metadata)
- {
- return ctx->getRowAllocatorEx(metadata, activityId, factory->getHeapFlags());
- }
- virtual IEngineRowAllocator * createRowAllocatorEx(IOutputMetaData * metadata, roxiemem::RoxieHeapFlags extraFlags)
- {
- return ctx->getRowAllocatorEx(metadata, activityId, (roxiemem::RoxieHeapFlags)(factory->getHeapFlags()|extraFlags));
- }
- inline ICodeContext *queryCodeContext()
- {
- return ctx->queryCodeContext();
- }
- // MORE - most of this is copied from ccd.hpp - can't we refactor?
- virtual void CTXLOGa(TracingCategory category, const char *prefix, const char *text) const
- {
- if (ctx)
- ctx->CTXLOGa(category, prefix, text);
- else
- DBGLOG("[%s] %s", prefix, text);
- }
- virtual void CTXLOGaeva(IException *E, const char *file, unsigned line, const char *prefix, const char *format, va_list args) const __attribute__((format(printf,6,0)))
- {
- if (ctx)
- ctx->CTXLOGaeva(E, file, line, prefix, format, args);
- else
- {
- StringBuffer ss;
- ss.appendf("[%s] ERROR", prefix);
- if (E)
- ss.append(": ").append(E->errorCode());
- if (file)
- ss.appendf(": %s(%d) ", file, line);
- if (E)
- E->errorMessage(ss.append(": "));
- if (format)
- {
- ss.append(": ").valist_appendf(format, args);
- }
- LOG(MCoperatorProgress, unknownJob, "%s", ss.str());
- }
- }
- virtual void CTXLOGl(LogItem *log) const
- {
- if (ctx)
- ctx->CTXLOGl(log);
- else
- {
- assert(ctx);
- log->Release(); // Should never happen
- }
- }
- virtual void noteStatistic(StatisticKind kind, unsigned __int64 value) const
- {
- stats.addStatistic(kind, value);
- }
- virtual void mergeStats(const CRuntimeStatisticCollection &from) const
- {
- stats.merge(from);
- }
- virtual const CRuntimeStatisticCollection &queryStats() const
- {
- return stats;
- }
- virtual StringBuffer &getLogPrefix(StringBuffer &ret) const
- {
- if (ctx)
- ctx->getLogPrefix(ret);
- return ret.append('@').append(activityId);
- }
- virtual bool isIntercepted() const
- {
- return ctx ? ctx->isIntercepted() : false;
- }
- virtual bool isBlind() const
- {
- return ctx ? ctx->isBlind() : blindLogging;
- }
- virtual unsigned queryTraceLevel() const
- {
- if (ctx)
- return ctx->queryTraceLevel();
- else
- return traceLevel;
- }
- virtual bool isPassThrough()
- {
- return false;
- }
- virtual const IResolvedFile *resolveLFN(const char *filename, bool isOpt)
- {
- return ctx->resolveLFN(filename, isOpt);
- }
- virtual const IResolvedFile *queryVarFileInfo() const
- {
- throwUnexpected(); // should be implemented in more derived class by anyone that has a remote adaptor
- return NULL;
- }
- virtual void serializeSkipInfo(MemoryBuffer &out, unsigned seekLen, const void *rawSeek, unsigned numFields, const void * seek, const SmartStepExtra &stepExtra) const
- {
- throwUnexpected(); // should be implemented in more derived class wherever needed
- }
- virtual IRoxieSlaveContext *queryContext()
- {
- return ctx;
- }
- virtual IEngineRowStream *queryConcreteOutputStream(unsigned whichInput) { assertex(whichInput==0); return this; }
- virtual IStrandJunction *queryConcreteOutputJunction(unsigned idx) const { assertex(idx==0); return junction; }
- virtual IRoxieServerActivity *queryActivity() { return this; }
- virtual IIndexReadActivityInfo *queryIndexReadActivity() { return NULL; }
- virtual bool needsAllocator() const { return false; }
- void _onCreate(IHThorArg *_colocalParent, unsigned _numParallel)
- {
- colocalParent = _colocalParent;
- createPending = true;
- if (needsAllocator())
- ensureRowAllocator();
- processed = 0;
- if (factory)
- factory->onCreateChildQueries(ctx, &basehelper, childGraphs, this, probeManager, *this, _numParallel);
- if (ctx)
- timeActivities = ctx->queryOptions().timeActivities;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- _onCreate(_colocalParent, 1);
- }
- virtual void serializeCreateStartContext(MemoryBuffer &out)
- {
- //This should only be called after onStart has been called on the helper
- assertex(!createPending);
- assertex(state==STATEstarted);
- unsigned startlen = out.length();
- basehelper.serializeCreateContext(out);
- basehelper.serializeStartContext(out);
- if (queryTraceLevel() > 10)
- CTXLOG("serializeCreateStartContext for %d added %d bytes", activityId, out.length()-startlen);
- }
- virtual void serializeExtra(MemoryBuffer &out) {}
- inline void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CriticalBlock cb(statecrit);
- if (state != STATEreset && state != STATEstarting)
- {
- CTXLOG("STATE: Expected state to be reset, but was %s, in activity %d", queryStateText(state), activityId);
- }
- state=STATEstarted;
- #ifdef TRACE_STARTSTOP
- if (traceStartStop)
- {
- CTXLOG("start %p %d", this, activityId);
- if (watchActivityId && watchActivityId==activityId)
- {
- CTXLOG("WATCH: start %p %d", this, activityId);
- }
- }
- #endif
- executeDependencies(parentExtractSize, parentExtract, 0);
- if (input)
- input->start(parentExtractSize, parentExtract, paused);
- ensureCreated();
- basehelper.onStart(parentExtract, NULL);
- if (factory)
- factory->noteStarted();
- startJunction(junction);
- }
- void executeDependencies(unsigned parentExtractSize, const byte *parentExtract, unsigned controlId)
- {
- //MORE: Create a filtered list and then use asyncfor
- ForEachItemIn(idx, dependencies)
- {
- if (dependencyControlIds.item(idx) == controlId)
- dependencies.item(idx).execute(parentExtractSize, parentExtract);
- }
- }
- void stopDependencies(unsigned parentExtractSize, const byte *parentExtract, unsigned controlId)
- {
- ForEachItemIn(idx, dependencies)
- {
- if (dependencyControlIds.item(idx) == controlId)
- dependencies.item(idx).stop();
- }
- }
- virtual unsigned __int64 queryTotalCycles() const
- {
- return totalCycles.totalCycles;
- }
- virtual unsigned __int64 queryLocalCycles() const
- {
- __int64 ret = totalCycles.totalCycles;
- if (input) ret -= input->queryTotalCycles();
- if (ret < 0)
- ret = 0;
- return ret;
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (idx==0)
- return input;
- else
- return NULL;
- }
- inline void ensureCreated()
- {
- if (createPending)
- {
- createPending = false;
- basehelper.onCreate(ctx->queryCodeContext(), colocalParent, NULL);
- }
- }
- virtual void stop()
- {
- // NOTE - don't be tempted to skip the stop for activities that are reset - splitters need to see the stops
- if (state != STATEstopped)
- {
- CriticalBlock cb(statecrit);
- if (state != STATEstopped)
- {
- #ifdef TRACE_STARTSTOP
- if (traceStartStop)
- {
- CTXLOG("stop %p %d (state currently %s)", this, activityId, queryStateText(state));
- if (watchActivityId && watchActivityId==activityId)
- {
- CTXLOG("WATCH: stop %p %d", this, activityId);
- }
- }
- #endif
- state=STATEstopped;
- // NOTE - this is needed to ensure that dependencies which were not used are properly stopped
- ForEachItemIn(idx, dependencies)
- {
- if (dependencyControlIds.item(idx) == 0)
- dependencies.item(idx).stopSink(dependencyIndexes.item(idx));
- }
- if (inputStream)
- inputStream->stop();
- if (rowAllocator)
- {
- stats.reset(heapStatistics);
- rowAllocator->gatherStats(stats);
- }
- }
- }
- }
- virtual void abort()
- {
- aborted = true;
- // MORE - should abort dependencies here?
- stop();
- }
- inline void reset()
- {
- if (state != STATEreset)
- {
- CriticalBlock cb(statecrit);
- if (state != STATEreset)
- {
- if (state==STATEstarted || state==STATEstarting)
- {
- if (traceStartStop || traceLevel > 2)
- CTXLOG("STATE: activity %d reset without stop", activityId);
- stop();
- }
- state = STATEreset;
- #ifdef TRACE_STARTSTOP
- if (traceStartStop)
- {
- CTXLOG("reset %p %d", this, activityId);
- if (watchActivityId && watchActivityId==activityId)
- {
- CTXLOG("WATCH: reset %p %d", this, activityId);
- }
- }
- #endif
- resetJunction(junction);
- ForEachItemIn(idx, dependencies)
- dependencies.item(idx).reset();
- localCycles = queryLocalCycles(); // We can't call queryLocalCycles() in the destructor, so save the information here when we can.
- if (input)
- input->reset();
- }
- }
- aborted = false;
- }
- virtual void addDependency(IRoxieServerActivity &source, unsigned sourceIdx, int controlId)
- {
- dependencies.append(source);
- dependencyIndexes.append(sourceIdx);
- dependencyControlIds.append(controlId);
- }
- virtual void resetEOF()
- {
- //would make more sense if the default implementation (and eof member) were in the base class
- }
- // Sink activities should override this....
- virtual void execute(unsigned parentExtractSize, const byte * parentExtract)
- {
- throw MakeStringException(ROXIE_SINK, "Internal error: execute() requires a sink");
- }
- virtual void executeChild(size32_t & retSize, void * & ret, unsigned parentExtractSize, const byte * parentExtract)
- {
- throw MakeStringException(ROXIE_SINK, "Internal error: executeChild() requires a suitable sink");
- }
- virtual void stopSink(unsigned idx)
- {
- throw MakeStringException(ROXIE_SINK, "Internal error: stopSink() requires a suitable sink");
- }
- virtual void connectInputStreams(bool consumerOrdered)
- {
- if (input && !inputStream)
- inputStream = connectSingleStream(ctx, input, sourceIdx, junction, isInputOrdered(consumerOrdered, 0));
- if (!connected)
- {
- connectDependencies();
- connected = true;
- }
- }
- void connectDependencies()
- {
- ForEachItemIn(i, dependencies)
- dependencies.item(i).connectInputStreams(true);
- }
- virtual __int64 evaluate()
- {
- throw MakeStringException(ROXIE_SINK, "Internal error: evaluate() requires a function");
- }
- virtual IFinalRoxieInput * querySelectOutput(unsigned id)
- {
- return NULL;
- }
- virtual bool querySetStreamInput(unsigned id, unsigned _sourceIdx, IFinalRoxieInput * _input)
- {
- return false;
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- assertex(!idx);
- input = _in;
- sourceIdx = _sourceIdx;
- inputStream = NULL;
- }
- bool isInputOrdered(bool consumerOrdered, unsigned idx) const
- {
- if (!factory)
- return true;
- //Grouping implies the stream processing must be ordered - I think this always applies
- IFinalRoxieInput * curInput = queryInput(idx);
- if (curInput && curInput->queryOutputMeta()->isGrouped())
- return true;
- return factory->isInputOrdered(consumerOrdered, idx);
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- assertex(!idx);
- // By default, activities are assumed NOT to support streams
- bool inputOrdered = isInputOrdered(consumerOrdered, 0);
- connectInputStreams(inputOrdered);
- // Return a single stream
- streams.append(this);
- return NULL;
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- if (idx == (unsigned) -1)
- idx = 0;
- return idx ? NULL : this;
- }
- virtual IOutputMetaData *queryOutputMeta() const
- {
- return meta.queryOriginal();
- }
- virtual unsigned queryId() const
- {
- return activityId;
- }
- virtual unsigned querySubgraphId() const
- {
- return factory->querySubgraphId();
- }
- virtual void checkAbort()
- {
- ctx->checkAbort();
- }
- IException *makeWrappedException(IException *e)
- {
- StringBuffer msg;
- ThorActivityKind activityKind = factory ? factory->getKind() : TAKnone;
- CTXLOG("makeWrappedException - %s (in %s %d)", e->errorMessage(msg).str(), getActivityText(activityKind), activityId);
- if (QUERYINTERFACE(e, CWrappedException) || QUERYINTERFACE(e, IUserException))
- return e;
- else
- return new CWrappedException(e, activityKind, activityId);
- }
- virtual void gatherIterationUsage(IRoxieServerLoopResultProcessor & processor, unsigned parentExtractSize, const byte * parentExtract)
- {
- }
- virtual void associateIterationOutputs(IRoxieServerLoopResultProcessor & processor, unsigned parentExtractSize, const byte * parentExtract, IProbeManager *probeManager, IArrayOf<IRoxieProbe> &probes)
- {
- }
- virtual void resetOutputsUsed()
- {
- }
- virtual void noteOutputUsed()
- {
- }
- virtual IRoxieServerSideCache *queryServerSideCache() const
- {
- return factory->queryServerSideCache();
- }
- virtual const IRoxieServerActivityFactory *queryFactory() const
- {
- return factory;
- }
- inline ThorActivityKind getKind() const
- {
- return factory->getKind();
- }
- inline bool isSink() const
- {
- return (factory != NULL) && factory->isSink();
- }
- };
- //=====================================================================================================
- class CRoxieServerLateStartActivity : public CRoxieServerActivity
- {
- protected:
- bool prefiltered;
- bool eof;
- void lateStart(unsigned parentExtractSize, const byte *parentExtract, bool any)
- {
- prefiltered = !any;
- eof = prefiltered;
- if (!prefiltered)
- {
- input->start(parentExtractSize, parentExtract, false);
- startJunction(junction);
- }
- else
- {
- if (traceStartStop)
- CTXLOG("lateStart activity stopping input early as prefiltered");
- inputStream->stop();
- }
- }
- public:
- CRoxieServerLateStartActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager)
- {
- prefiltered = false;
- eof = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- IFinalRoxieInput *saveInput = input;
- Owned<IStrandJunction> saveJunction = junction.getClear();
- input = NULL; // Make sure parent does not start the chain yet
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- input = saveInput;
- junction.setown(saveJunction.getClear());
- }
- virtual void stop()
- {
- if (!prefiltered)
- {
- inputStream->stop();
- }
- else if (traceStartStop)
- CTXLOG("lateStart activity NOT stopping input late as prefiltered");
- IEngineRowStream *save = inputStream;
- inputStream = NULL; // We already stopped inputStream - prevent parent from doing so again
- CRoxieServerActivity::stop();
- inputStream = save;
- }
- virtual void reset()
- {
- CRoxieServerActivity::reset();
- prefiltered = false;
- }
- };
- //=================================================================================
- class StrandOptions
- {
- // Typically set from hints, common to many stranded activities
- public:
- explicit StrandOptions(IPropertyTree &_graphNode)
- {
- //PARALLEL(1) can be used to explicitly disable parallel processing.
- numStrands = _graphNode.getPropInt("att[@name='parallel']/@value", 0);
- if ((numStrands == minus1U) || (numStrands > MAX_SENSIBLE_STRANDS))
- numStrands = getAffinityCpus();
- blockSize = _graphNode.getPropInt("hint[@name='strandblocksize']/@value", 0);
- }
- StrandOptions(const StrandOptions &from, IRoxieSlaveContext *ctx)
- {
- numStrands = from.numStrands;
- blockSize = from.blockSize;
- if (!blockSize)
- blockSize = ctx->queryOptions().strandBlockSize;
- if (numStrands == 0)
- numStrands = ctx->queryOptions().forceNumStrands;
- }
- public:
- unsigned numStrands = 0; // if 1 it forces single-stranded operations. (Useful for testing.)
- unsigned blockSize = 0;
- };
- class StrandProcessor : public CInterfaceOf<IEngineRowStream>
- {
- protected:
- CRoxieServerActivity &parent;
- IEngineRowAllocator *rowAllocator;
- IEngineRowStream *inputStream;
- ActivityTimeAccumulator totalCycles;
- mutable CRuntimeStatisticCollection stats;
- unsigned processed = 0;
- unsigned numProcessedLastGroup = 0;
- const bool timeActivities;
- bool stopped = false;
- std::atomic<bool> abortRequested;
- public:
- explicit StrandProcessor(CRoxieServerActivity &_parent, IEngineRowStream *_inputStream, bool needsAllocator)
- : parent(_parent), inputStream(_inputStream), stats(parent.queryStatsMapping()), timeActivities(_parent.timeActivities), abortRequested(false)
- {
- if (needsAllocator)
- {
- rowAllocator = parent.createRowAllocatorEx(parent.queryOutputMeta(), roxiemem::RHFunique);
- }
- else
- rowAllocator = NULL;
- }
- ~StrandProcessor()
- {
- ::Release(rowAllocator);
- }
- virtual void start()
- {
- processed = 0;
- numProcessedLastGroup = 0;
- totalCycles.reset();
- stats.reset();
- }
- virtual void stop()
- {
- if (!stopped)
- {
- if (inputStream)
- inputStream->stop();
- stats.reset(heapStatistics); // Heap stats are always gathered from scratch each time
- if (rowAllocator)
- rowAllocator->gatherStats(stats);
- parent.stop();
- parent.mergeStrandStats(processed, totalCycles, stats);
- }
- stopped = true;
- }
- virtual void reset()
- {
- stopped = false;
- abortRequested.store(false, std::memory_order_relaxed);
- }
- virtual void resetEOF()
- {
- inputStream->resetEOF();
- }
- inline void requestAbort() { abortRequested.store(true, std::memory_order_relaxed); }
- inline bool isAborting() { return abortRequested.load(std::memory_order_relaxed); }
- };
- class CRoxieServerStrandedActivity : public CRoxieServerActivity
- {
- protected:
- StrandOptions strandOptions;
- IArrayOf<StrandProcessor> strands;
- Owned<IStrandBranch> branch;
- Owned<IStrandJunction> splitter;
- Owned<IStrandJunction> sourceJunction; // A junction applied to the output of a source activity
- std::atomic<unsigned> active;
- public:
- CRoxieServerStrandedActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const StrandOptions &_strandOptions)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- strandOptions(_strandOptions, ctx)
- {
- active = 0;
- }
- ~CRoxieServerStrandedActivity()
- {
- if (strands.ordinality() > 1)
- {
- if (factory && !debugging)
- factory->noteProcessed(0, processed);
- if (ctx && factory)
- ctx->noteProcessed(factory->querySubgraphId(), activityId, 0, processed, strands.ordinality());
- processed = 0; // To avoid reprocessing in base destructor
- }
- }
- virtual void onCreate(IHThorArg *_colocalArg)
- {
- CRoxieServerActivity::_onCreate(_colocalArg, strands.ordinality());
- }
- //This function is pure (But also implemented out of line) to force the derived classes to implement it.
- //After calling the base class start method, and initialising any values from the helper they must call onStartStrands(),
- //this must also happen before any rows are read from the strands (e.g., by a source junction)
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused) = 0;
- virtual void reset()
- {
- assertex(active==0);
- ForEachItemIn(idx, strands)
- strands.item(idx).reset();
- resetJunction(splitter);
- CRoxieServerActivity::reset();
- resetJunction(sourceJunction);
- }
- virtual void stop()
- {
- // Called from the strands... which should ensure that stop is not called more than once per strand
- //The first strand to call
- if (active)
- --active;
- if (!active)
- CRoxieServerActivity::stop();
- }
- void requestAbort()
- {
- ForEachItemIn(idx, strands)
- strands.item(idx).requestAbort();
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks) override
- {
- assertex(idx == 0);
- assertex(strands.empty());
- CRoxieServerActivity::connectDependencies();
- bool inputOrdered = isInputOrdered(consumerOrdered, idx);
- //Note, numStrands == 1 is an explicit request to disable threading
- if (consumerOptions && (consumerOptions->numStrands != 1) && (strandOptions.numStrands != 1))
- {
- //Check to see if the consumer's settings should override
- if (strandOptions.numStrands == 0)
- {
- strandOptions.numStrands = consumerOptions->numStrands;
- strandOptions.blockSize = consumerOptions->blockSize;
- }
- else if (consumerOptions->numStrands > strandOptions.numStrands)
- {
- strandOptions.numStrands = consumerOptions->numStrands;
- }
- }
- Owned <IStrandJunction> recombiner;
- if (input)
- {
- if (strandOptions.numStrands == 1)
- {
- // 1 means explicitly requested single-strand.
- IEngineRowStream *instream = connectSingleStream(ctx, input, sourceIdx, junction, inputOrdered);
- strands.append(*createStrandProcessor(instream));
- }
- else
- {
- PointerArrayOf<IEngineRowStream> instreams;
- recombiner.setown(input->getOutputStreams(ctx, sourceIdx, instreams, &strandOptions, inputOrdered, orderedCallbacks));
- if ((instreams.length() == 1) && (strandOptions.numStrands != 0)) // 0 means did not specify - we should use the strands that our upstream provides
- {
- assertex(recombiner == NULL);
- // Create a splitter to split the input into n... and a recombiner if need to preserve sorting
- if (inputOrdered)
- {
- branch.setown(createStrandBranch(ctx->queryRowManager(), strandOptions.numStrands, strandOptions.blockSize, true, input->queryOutputMeta()->isGrouped(), false, orderedCallbacks));
- splitter.set(branch->queryInputJunction());
- recombiner.set(branch->queryOutputJunction());
- }
- else
- {
- splitter.setown(createStrandJunction(ctx->queryRowManager(), 1, strandOptions.numStrands, strandOptions.blockSize, false));
- }
- splitter->setInput(0, instreams.item(0));
- for (unsigned strandNo = 0; strandNo < strandOptions.numStrands; strandNo++)
- strands.append(*createStrandProcessor(splitter->queryOutput(strandNo)));
- }
- else
- {
- // Ignore my hint and just use the width already split into...
- ForEachItemIn(strandNo, instreams)
- strands.append(*createStrandProcessor(instreams.item(strandNo)));
- }
- }
- }
- else
- {
- unsigned numStrands = strandOptions.numStrands ? strandOptions.numStrands : 1;
- for (unsigned i=0; i < numStrands; i++)
- strands.append(*createStrandSourceProcessor(inputOrdered));
- if (inputOrdered && (numStrands > 1))
- {
- if (consumerOptions)
- {
- //If the output activities are also stranded then need to create a version of the branch
- bool isGrouped = queryOutputMeta()->isGrouped();
- branch.setown(createStrandBranch(ctx->queryRowManager(), strandOptions.numStrands, strandOptions.blockSize, true, isGrouped, true, orderedCallbacks));
- sourceJunction.set(branch->queryInputJunction());
- recombiner.set(branch->queryOutputJunction());
- assertex((orderedCallbacks && !recombiner) || (!orderedCallbacks && recombiner));
- //This is different from the branch above. The first "junction" has the source activity as the input, and the outputs as the result of the activity
- for (unsigned strandNo = 0; strandNo < strandOptions.numStrands; strandNo++)
- {
- sourceJunction->setInput(strandNo, &strands.item(strandNo));
- streams.append(sourceJunction->queryOutput(strandNo));
- }
- #ifdef TRACE_STRANDS
- if (traceLevel > 2)
- DBGLOG("Executing activity %u with %u strands", activityId, strands.ordinality());
- #endif
- return recombiner.getClear();
- }
- else
- {
- //Feeding into a non threaded activity, so create a M:1 junction to combine the source strands
- recombiner.setown(createStrandJunction(ctx->queryRowManager(), numStrands, 1, strandOptions.blockSize, inputOrdered));
- }
- }
- }
- ForEachItemIn(i, strands)
- streams.append(&strands.item(i));
- #ifdef TRACE_STRANDS
- if (traceLevel > 2)
- DBGLOG("Executing activity %u with %u strands", activityId, strands.ordinality());
- #endif
- return recombiner.getClear();
- }
- virtual StrandProcessor *createStrandProcessor(IEngineRowStream *instream) = 0;
- //MORE: Possibly this class should be split into two for sinks and non sinks...
- virtual StrandProcessor *createStrandSourceProcessor(bool inputOrdered) = 0;
- virtual const void * nextRow()
- {
- throwUnexpected();
- }
- inline unsigned numStrands() const { return strands.ordinality(); }
- protected:
- void onStartStrands()
- {
- ForEachItemIn(idx, strands)
- {
- strands.item(idx).start();
- active++;
- }
- }
- };
- //For some reason gcc doesn't let you specify a function as pure virtual and define it at the same time.
- void CRoxieServerStrandedActivity::start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- startJunction(splitter);
- }
- class CRoxieServerStrandedLateStartActivity : public CRoxieServerStrandedActivity
- {
- protected:
- bool prefiltered;
- bool eof;
- void lateStart(unsigned parentExtractSize, const byte *parentExtract, bool any)
- {
- prefiltered = !any;
- eof = prefiltered;
- if (!prefiltered)
- {
- //At this point
- input->start(parentExtractSize, parentExtract, false);
- startJunction(splitter);
- onStartStrands(); // Initialise the strands - the helper will have been correctly initialized by this point.
- startJunction(sourceJunction);
- }
- else
- {
- if (traceStartStop)
- CTXLOG("strandedLateStart activity stopping input early as prefiltered");
- ForEachItemIn(idx, strands)
- strands.item(idx).stop();
- }
- }
- public:
- CRoxieServerStrandedLateStartActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const StrandOptions &_strandOptions)
- : CRoxieServerStrandedActivity(_ctx, _factory, _probeManager, _strandOptions)
- {
- prefiltered = false;
- eof = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- IFinalRoxieInput *save = input;
- input = NULL; // Make sure parent does not start the chain yet - but we do want to do the dependencies (because the decision about whether to start may depend on them)
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- input = save;
- }
- virtual void reset()
- {
- CRoxieServerStrandedActivity::reset();
- prefiltered = false;
- }
- };
- //=====================================================================================================
- atomic_t nextInstanceId;
- extern unsigned getNextInstanceId()
- {
- return atomic_add_exchange(&nextInstanceId, 1)+1;
- }
- atomic_t nextRuid;
- ruid_t getNextRuid()
- {
- ruid_t ret = atomic_add_exchange(&nextRuid, 1)+1;
- while (ret < RUID_FIRST)
- ret = atomic_add_exchange(&nextRuid, 1)+1; // ruids 0 and 1 are reserved for pings/unwanted discarder.
- return ret;
- }
- void setStartRuid(unsigned restarts)
- {
- atomic_set(&nextRuid, restarts * 0x10000);
- atomic_set(&nextInstanceId, restarts * 10000);
- }
- enum { LimitSkipErrorCode = 0, KeyedLimitSkipErrorCode = 1 };
- class LimitSkipException : public IException, public CInterface
- {
- int code;
- public:
- LimitSkipException(int _code) { code = _code; }
- IMPLEMENT_IINTERFACE;
- virtual int errorCode() const { return code; }
- virtual StringBuffer & errorMessage(StringBuffer &msg) const { return msg.append("LimitSkipException"); }
- virtual MessageAudience errorAudience() const { return MSGAUD_internal; }
- };
- IException *makeLimitSkipException(bool isKeyed)
- {
- // We need to make sure what we throw is IException not something derived from it....
- return new LimitSkipException(isKeyed ? KeyedLimitSkipErrorCode : LimitSkipErrorCode);
- }
- //=================================================================================
- interface IRecordPullerCallback : extends IExceptionHandler
- {
- virtual void processRow(const void *row) = 0;
- virtual void processEOG() = 0;
- virtual void processGroup(const ConstPointerArray &rows) = 0;
- virtual void processDone() = 0;
- };
- class RecordPullerThread : public RestartableThread
- {
- protected:
- IFinalRoxieInput *input;
- Owned<IStrandJunction> junction;
- unsigned sourceIdx = 0;
- IEngineRowStream *inputStream;
- IRecordPullerCallback *helper;
- Semaphore started; // MORE: GH->RKC I'm pretty sure this can be deleted, since handled by RestartableThread
- bool groupAtOnce, eof, eog;
- CriticalSection crit;
- public:
- RecordPullerThread(bool _groupAtOnce)
- : RestartableThread("RecordPullerThread"), groupAtOnce(_groupAtOnce)
- {
- input = NULL;
- inputStream = NULL;
- helper = NULL;
- eof = eog = FALSE;
- }
- inline unsigned __int64 queryTotalCycles() const
- {
- return input->queryTotalCycles();
- }
- void setInput(IRecordPullerCallback *_helper, unsigned _sourceIdx, IFinalRoxieInput *_input)
- {
- helper = _helper;
- input = _input;
- sourceIdx = _sourceIdx;
- }
- IFinalRoxieInput *queryInput() const
- {
- return input;
- }
- virtual void connectInputStreams(IRoxieSlaveContext *ctx, bool consumerOrdered)
- {
- inputStream = connectSingleStream(ctx, input, sourceIdx, junction, consumerOrdered);
- }
- IEngineRowStream *queryStream() const
- {
- return inputStream;
- }
- void start(unsigned parentExtractSize, const byte *parentExtract, bool paused, unsigned preload, bool noThread, IRoxieSlaveContext *ctx)
- {
- eof = false;
- eog = false;
- input->start(parentExtractSize, parentExtract, paused);
- startJunction(junction);
- try
- {
- if (preload && !paused)
- {
- if (traceLevel > 4)
- DBGLOG("Preload fetching first %d records", preload);
- if (groupAtOnce)
- pullGroups(preload);
- else
- pullRecords(preload);
- }
- if (eof)
- {
- if (traceLevel > 4)
- DBGLOG("No need to start puller after preload");
- helper->processDone();
- }
- else
- {
- if (!noThread)
- {
- StringBuffer logPrefix("[");
- if (ctx) ctx->getLogPrefix(logPrefix);
- logPrefix.append("] ");
- RestartableThread::start(logPrefix);
- started.wait();
- }
- }
- }
- catch (IException *e)
- {
- helper->fireException(e);
- }
- catch (...)
- {
- helper->fireException(MakeStringException(ROXIE_INTERNAL_ERROR, "Unexpected exception caught in RecordPullerThread::start"));
- }
- }
- void stop()
- {
- if (traceStartStop)
- DBGLOG("RecordPullerThread::stop");
- {
- CriticalBlock c(crit); // stop is called on our consumer's thread. We need to take care calling stop for our input to make sure it is not in mid-nextRow etc etc.
- if (inputStream)
- inputStream->stop();
- }
- RestartableThread::join();
- }
- void reset()
- {
- input->reset();
- resetJunction(junction);
- }
- virtual int run()
- {
- started.signal();
- try
- {
- if (groupAtOnce)
- pullGroups((unsigned) -1);
- else
- pullRecords((unsigned) -1);
- helper->processDone();
- }
- catch (IException *e)
- {
- helper->fireException(e);
- }
- catch (...)
- {
- helper->fireException(MakeStringException(ROXIE_INTERNAL_ERROR, "Unexpected exception caught in RecordPullerThread::run"));
- }
- return 0;
- }
- void done()
- {
- helper->processDone();
- }
- bool pullRecords(unsigned preload)
- {
- if (eof)
- return false;
- while (preload)
- {
- const void * row;
- {
- CriticalBlock c(crit); // See comments in stop for why this is needed
- row = inputStream->nextRow();
- }
- if (row)
- {
- eog = false;
- helper->processRow(row);
- }
- else if (!eog)
- {
- helper->processEOG();
- eog = true;
- }
- else
- {
- eof = true;
- return false;
- }
- if (preload != (unsigned) -1)
- preload--;
- }
- return true;
- }
- void pullGroups(unsigned preload)
- {
- ConstPointerArray thisGroup;
- unsigned rowsDone = 0;
- while (preload && !eof)
- {
- const void *row;
- {
- CriticalBlock c(crit);
- row = inputStream->nextRow();
- }
- if (row)
- {
- thisGroup.append(row);
- rowsDone++;
- }
- else if (thisGroup.length())
- {
- helper->processGroup(thisGroup);
- thisGroup.kill();
- if (preload != (unsigned) -1)
- {
- if (preload > rowsDone)
- preload -= rowsDone;
- else
- break;
- }
- rowsDone = 0;
- }
- else
- {
- eof = true;
- break;
- }
- }
- }
- };
- //=================================================================================
- #define READAHEAD_SIZE 1000
- // MORE - this code copied from ThreadedConcat code - may be able to common up some.
- class CRoxieServerReadAheadInput : implements IEngineRowStream, implements IFinalRoxieInput, implements IRecordPullerCallback, public CInterface
- {
- QueueOf<const void, true> buffer;
- InterruptableSemaphore ready;
- InterruptableSemaphore space;
- CriticalSection crit;
- bool eof;
- bool disabled;
- IRoxieSlaveContext *ctx;
- RecordPullerThread puller;
- unsigned preload;
- cycle_t totalCycles;
- bool timeActivities;
- public:
- IMPLEMENT_IINTERFACE;
- CRoxieServerReadAheadInput(IRoxieSlaveContext *_ctx, unsigned _preload) : ctx(_ctx), puller(true), preload(_preload)
- {
- eof = false;
- disabled = (ctx->queryDebugContext() != NULL);
- totalCycles = 0;
- timeActivities = ctx->queryOptions().timeActivities;
- }
- virtual IRoxieServerActivity *queryActivity()
- {
- return puller.queryInput()->queryActivity();
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks) override
- {
- assertex(!idx);
- puller.connectInputStreams(ctx, consumerOrdered);
- streams.append(this);
- return NULL;
- }
- virtual IEngineRowStream *queryConcreteOutputStream(unsigned whichInput) { return this; }
- virtual IStrandJunction *queryConcreteOutputJunction(unsigned idx) const { return nullptr; }
- virtual IIndexReadActivityInfo *queryIndexReadActivity()
- {
- return puller.queryInput()->queryIndexReadActivity();
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- if (disabled)
- puller.queryInput()->start(parentExtractSize, parentExtract, paused);
- else
- {
- space.reinit(READAHEAD_SIZE);
- ready.reinit();
- puller.start(parentExtractSize, parentExtract, paused, preload, false, ctx);
- }
- }
- virtual void stop()
- {
- if (disabled)
- puller.queryStream()->stop();
- else
- {
- space.interrupt();
- ready.interrupt();
- puller.stop();
- }
- }
- virtual void reset()
- {
- if (disabled)
- puller.queryInput()->reset();
- else
- {
- puller.reset();
- ForEachItemIn(idx1, buffer)
- ReleaseRoxieRow(buffer.item(idx1));
- buffer.clear();
- }
- }
- virtual void resetEOF()
- {
- throwUnexpected();
- }
- virtual IOutputMetaData * queryOutputMeta() const
- {
- return puller.queryInput()->queryOutputMeta();
- }
- void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- assertex(!idx);
- puller.setInput(this, _sourceIdx, _in);
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (idx==0)
- return puller.queryInput();
- else
- return NULL;
- }
- virtual unsigned __int64 queryTotalCycles() const
- {
- return totalCycles;
- }
- virtual const void * nextRow()
- {
- SimpleActivityTimer t(totalCycles, timeActivities);
- if (disabled)
- return puller.queryStream()->nextRow();
- else
- {
- loop
- {
- {
- CriticalBlock b(crit);
- if (eof && !buffer.ordinality())
- return NULL; // eof
- }
- ready.wait();
- const void *ret;
- {
- CriticalBlock b(crit);
- ret = buffer.dequeue();
- }
- space.signal();
- return ret;
- }
- }
- }
- virtual bool fireException(IException *e)
- {
- // called from puller thread on failure
- ready.interrupt(LINK(e));
- space.interrupt(e);
- return true;
- }
- virtual void processRow(const void *row)
- {
- {
- CriticalBlock b(crit);
- buffer.enqueue(row);
- }
- ready.signal();
- space.wait();
- }
- virtual void processGroup(const ConstPointerArray &rows)
- {
- // NOTE - a bit bizarre in that it waits for the space AFTER using it.
- // But the space semaphore is only there to stop infinite readahead. And otherwise it would deadlock
- // if group was larger than max(space)
- {
- CriticalBlock b(crit);
- ForEachItemIn(idx, rows)
- buffer.enqueue(rows.item(idx));
- buffer.enqueue(NULL);
- }
- for (unsigned i2 = 0; i2 <= rows.length(); i2++) // note - does 1 extra for the null
- {
- ready.signal();
- space.wait();
- }
- }
- virtual void processEOG()
- {
- // Used when output is not grouped - just ignore
- }
- virtual void processDone()
- {
- CriticalBlock b(crit);
- eof = true;
- ready.signal();
- }
- };
- //=================================================================================
- class CRoxieServerTwoInputActivity : public CRoxieServerActivity
- {
- protected:
- IFinalRoxieInput *input1;
- IEngineRowStream *inputStream1;
- Owned<IStrandJunction> junction1;
- unsigned sourceIdx1 = 0;
- Owned<CRoxieServerReadAheadInput> puller;
- public:
- CRoxieServerTwoInputActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager)
- {
- input1 = NULL;
- inputStream1 = NULL;
- }
- ~CRoxieServerTwoInputActivity()
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- input1->start(parentExtractSize, parentExtract, paused);
- startJunction(junction1);
- }
- virtual void stop()
- {
- inputStream1->stop();
- CRoxieServerActivity::stop();
- }
- virtual unsigned __int64 queryLocalCycles() const
- {
- __int64 ret;
- __int64 inputCycles = input->queryTotalCycles();
- __int64 input1Cycles = input1->queryTotalCycles();
- if (puller)
- ret = totalCycles.totalCycles - (inputCycles > input1Cycles ? inputCycles : input1Cycles);
- else
- ret = totalCycles.totalCycles - (inputCycles + input1Cycles);
- if (ret < 0)
- ret = 0;
- return ret;
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- switch (idx)
- {
- case 0:
- return input;
- case 1:
- return input1;
- default:
- return NULL;
- }
- }
- virtual void reset()
- {
- CRoxieServerActivity::reset();
- if (input1)
- input1->reset();
- resetJunction(junction1);
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- switch(idx)
- {
- case 0:
- CRoxieServerActivity::setInput(idx, _sourceIdx, _in);
- break;
- case 1:
- input1 = _in;
- sourceIdx1 = _sourceIdx;
- break;
- default:
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() parameter out of bounds at %s(%d)", __FILE__, __LINE__);
- }
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks) override
- {
- inputStream1 = connectSingleStream(ctx, input1, sourceIdx1, junction1, isInputOrdered(consumerOrdered, 1));
- //Assumes input1 will be stranded if any.
- return CRoxieServerActivity::getOutputStreams(ctx, idx, streams, consumerOptions, consumerOrdered, orderedCallbacks);
- }
- };
- //=================================================================================
- class CRoxieServerMultiInputBaseActivity : public CRoxieServerActivity
- {
- protected:
- unsigned numInputs;
- unsigned numStreams;
- IFinalRoxieInput **inputArray;
- unsigned *sourceIdxArray;
- IEngineRowStream **streamArray;
- Owned<IStrandJunction> *junctionArray;
- public:
- CRoxieServerMultiInputBaseActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), numInputs(_numInputs)
- {
- numStreams = numInputs;
- inputArray = new IFinalRoxieInput*[numInputs];
- sourceIdxArray = new unsigned[numInputs];
- streamArray = new IEngineRowStream*[numStreams];
- junctionArray = new Owned<IStrandJunction>[numStreams];
- for (unsigned i1 = 0; i1 < numInputs; i1++)
- {
- inputArray[i1] = NULL;
- sourceIdxArray[i1] = 0;
- }
- for (unsigned i2 = 0; i2 < numStreams; i2++)
- streamArray[i2] = NULL;
- }
- ~CRoxieServerMultiInputBaseActivity()
- {
- delete [] inputArray;
- delete [] sourceIdxArray;
- delete [] streamArray;
- delete [] junctionArray;
- }
- virtual unsigned __int64 queryLocalCycles() const
- {
- __int64 localCycles = totalCycles.totalCycles;
- for (unsigned i = 0; i < numInputs; i++)
- localCycles -= inputArray[i]->queryTotalCycles();
- if (localCycles < 0)
- localCycles = 0;
- return localCycles;
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (idx < numInputs)
- return inputArray[idx];
- else
- return NULL;
- }
- virtual IFinalRoxieInput * queryConcreteInput(unsigned idx)
- {
- return queryInput(idx);
- }
- virtual IEngineRowStream *queryConcreteOutputStream(unsigned whichInput)
- {
- if (whichInput < numInputs)
- return streamArray[whichInput];
- else
- return NULL;
- }
- virtual IStrandJunction *queryConcreteOutputJunction(unsigned idx) const
- {
- if (idx < numInputs)
- return junctionArray[idx];
- else
- return NULL;
- }
- virtual void reset()
- {
- for (unsigned i = 0; i < numInputs; i++)
- inputArray[i]->reset();
- for (unsigned iS = 0; iS < numStreams; iS++)
- {
- resetJunction(junctionArray[iS]);
- }
- CRoxieServerActivity::reset();
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- assertex(idx < numInputs && idx < numStreams);
- inputArray[idx] = _in;
- sourceIdxArray[idx] = _sourceIdx;
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks) override
- {
- //There could be situations (e.g., NONEMPTY), where you might want to strand the activity.
- for (unsigned i = 0; i < numInputs; i++)
- streamArray[i] = connectSingleStream(ctx, inputArray[i], sourceIdxArray[i], junctionArray[i], consumerOrdered);
- return CRoxieServerActivity::getOutputStreams(ctx, idx, streams, NULL, consumerOrdered, nullptr); // The input basesclass does not have an input
- }
- };
- //=================================================================================
- class CRoxieServerMultiInputActivity : public CRoxieServerMultiInputBaseActivity
- {
- public:
- CRoxieServerMultiInputActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerMultiInputBaseActivity(_ctx, _factory, _probeManager, _numInputs)
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerMultiInputBaseActivity::start(parentExtractSize, parentExtract, paused);
- for (unsigned i = 0; i < numInputs; i++)
- {
- inputArray[i]->start(parentExtractSize, parentExtract, paused);
- }
- for (unsigned iS = 0; iS < numStreams; iS++)
- {
- startJunction(junctionArray[iS]);
- }
- }
- virtual void stop()
- {
- for (unsigned i = 0; i < numStreams; i++)
- {
- streamArray[i]->stop();
- }
- CRoxieServerMultiInputBaseActivity::stop();
- }
- };
- //=====================================================================================================
- class CRoxieServerInternalSinkActivity : public CRoxieServerActivity
- {
- protected:
- unsigned numOutputs;
- bool executed;
- bool *stopped;
- CriticalSection ecrit;
- Owned<IException> exception;
- public:
- CRoxieServerInternalSinkActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numOutputs)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), numOutputs(_numOutputs)
- {
- executed = false;
- stopped = new bool[numOutputs];
- for (unsigned s = 0; s < numOutputs; s++)
- stopped[s] = false;
- }
- ~CRoxieServerInternalSinkActivity()
- {
- delete [] stopped;
- }
- virtual void reset()
- {
- for (unsigned s = 0; s < numOutputs; s++)
- stopped[s] = false;
- executed = false;
- exception.clear();
- CRoxieServerActivity::reset();
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- return NULL;
- }
- virtual void stopSink(unsigned outputIdx)
- {
- if (!stopped[outputIdx])
- {
- stopped[outputIdx] = true;
- for (unsigned s = 0; s < numOutputs; s++)
- if (!stopped[s])
- return;
- stop(); // all outputs stopped - stop parent.
- }
- }
- virtual const void *nextRow()
- {
- throwUnexpected(); // I am nobody's input
- }
- virtual void onExecute() = 0;
- virtual void execute(unsigned parentExtractSize, const byte * parentExtract)
- {
- CriticalBlock b(ecrit);
- if (exception)
- throw exception.getLink();
- if (!executed)
- {
- try
- {
- start(parentExtractSize, parentExtract, false);
- {
- ActivityTimer t(totalCycles, timeActivities); // unfortunately this is not really best place for seeing in debugger.
- onExecute();
- }
- stop();
- executed = true;
- }
- catch (IException *E)
- {
- exception.set(E); // (or maybe makeWrappedException?)
- abort();
- throw;
- }
- catch (...)
- {
- exception.set(MakeStringException(ROXIE_INTERNAL_ERROR, "Unknown exception caught at %s:%d", __FILE__, __LINE__));
- abort();
- throw;
- }
- }
- }
- };
- //=================================================================================
- class CRoxieServerQueryPacket : implements IRoxieServerQueryPacket, public CInterface
- {
- protected:
- Owned<IMessageResult> result;
- Owned<IRoxieQueryPacket> packet;
- Linked<IRoxieServerQueryPacket> continuation;
- unsigned hash;
- unsigned seq;
- unsigned lastDebugSequence;
- Owned<IRoxieQueryPacket> lastDebugResponse;
- ILRUChain *prev;
- ILRUChain *next;
- bool delayed;
- public:
- IMPLEMENT_IINTERFACE;
- CRoxieServerQueryPacket(IRoxieQueryPacket *p) : packet(p)
- {
- hash = 0;
- seq = 0;
- prev = NULL;
- next = NULL;
- delayed = false;
- lastDebugSequence = 0;
- }
- virtual IRoxieQueryPacket *queryPacket() const
- {
- return packet;
- }
- virtual bool isContinuation() const
- {
- return packet && (packet->queryHeader().continueSequence & ~CONTINUE_SEQUENCE_SKIPTO) != 0;
- }
- virtual bool isDelayed() const
- {
- return delayed;
- }
- virtual bool isEnd() const
- {
- return false;
- }
- virtual bool isLimit(unsigned __int64 &_rowLimit, unsigned __int64 &_keyedLimit, unsigned __int64 &_stopAfter) const
- {
- return false;
- }
- virtual bool hasResult() const
- {
- return result != NULL;
- }
- virtual bool hasContinuation() const
- {
- return continuation != NULL;
- }
- virtual void setDelayed(bool _delayed)
- {
- delayed = _delayed;
- }
- virtual void setPacket(IRoxieQueryPacket *_packet)
- {
- packet.setown(_packet);
- }
- virtual void setSequence(unsigned _seq)
- {
- assertex(!IsShared());
- seq = _seq;
- }
- virtual unsigned getSequence() const
- {
- return seq;
- }
- IMessageResult *getResult()
- {
- return result.getLink();
- }
- IMessageResult *queryResult()
- {
- return result;
- }
- void setResult(IMessageResult *r)
- {
- result.setown(r);
- }
- IRoxieServerQueryPacket *queryContinuation()
- {
- return continuation;
- }
- void setContinuation(IRoxieServerQueryPacket *c)
- {
- continuation.setown(c);
- }
- virtual unsigned queryHash() const
- {
- return hash;
- }
- virtual void setHash(unsigned _hash)
- {
- hash = _hash;
- }
- virtual ILRUChain *queryPrev() const { return prev; }
- virtual ILRUChain *queryNext() const { return next; }
- virtual void setPrev(ILRUChain *p) { prev = p; }
- virtual void setNext(ILRUChain *n) { next = n; }
- virtual void unchain()
- {
- if (prev && next)
- {
- prev->setNext(next);
- next->setPrev(prev);
- }
- next = NULL;
- prev = NULL;
- }
- virtual IRoxieQueryPacket *getDebugResponse(unsigned sequence)
- {
- if (sequence == lastDebugSequence)
- return lastDebugResponse.getLink();
- else if (sequence > lastDebugSequence)
- {
- lastDebugResponse.clear();
- return NULL;
- }
- else
- throwUnexpected();
- }
- virtual void setDebugResponse(unsigned sequence, IRoxieQueryPacket *response)
- {
- lastDebugSequence = sequence;
- lastDebugResponse.set(response);
- }
- };
- class CRoxieServerQueryPacketEndMarker : public CRoxieServerQueryPacket
- {
- public:
- CRoxieServerQueryPacketEndMarker() : CRoxieServerQueryPacket(NULL)
- {
- }
- virtual bool isEnd() const
- {
- return true;
- }
- };
- class CRoxieServerQueryPacketLimitMarker : public CRoxieServerQueryPacket
- {
- unsigned __int64 rowLimit;
- unsigned __int64 keyedLimit;
- unsigned __int64 stopAfter;
- public:
- CRoxieServerQueryPacketLimitMarker(unsigned __int64 _rowLimit, unsigned __int64 _keyedLimit, unsigned __int64 _stopAfter) : CRoxieServerQueryPacket(NULL)
- {
- rowLimit = _rowLimit;
- keyedLimit = _keyedLimit;
- stopAfter = _stopAfter;
- }
- virtual bool isLimit(unsigned __int64 &_rowLimit, unsigned __int64 &_keyedLimit, unsigned __int64 &_stopAfter) const
- {
- _rowLimit = rowLimit;
- _keyedLimit = keyedLimit;
- _stopAfter = stopAfter;
- return true;
- }
- };
- class CRoxieServerSideCache : implements IRoxieServerSideCache, implements ILRUChain
- {
- protected:
- unsigned cacheTableSize;
- unsigned cacheTableSpace;
- IRoxieServerQueryPacket **cacheTable;
- mutable ILRUChain *prev;
- mutable ILRUChain *next;
- mutable CriticalSection crit;
- virtual ILRUChain *queryPrev() const { return prev; }
- virtual ILRUChain *queryNext() const { return next; }
- virtual void setPrev(ILRUChain *p) { prev = p; }
- virtual void setNext(ILRUChain *n) { next = n; }
- virtual void unchain()
- {
- prev->setNext(next);
- next->setPrev(prev);
- next = NULL;
- prev = NULL;
- }
- void moveToHead(IRoxieServerQueryPacket *mru)
- {
- mru->unchain();
- mru->setNext(next);
- next->setPrev(mru);
- mru->setPrev(this);
- next = mru;
- }
- IRoxieServerQueryPacket *removeLRU()
- {
- if (next==this)
- assertex(next != this);
- IRoxieServerQueryPacket *goer = (IRoxieServerQueryPacket *) next;
- goer->unchain(); // NOTE - this will modify the value of next
- return goer;
- }
- void removeEntry(IRoxieServerQueryPacket *goer)
- {
- unsigned v = goer->queryHash() % cacheTableSize;
- loop
- {
- IRoxieServerQueryPacket *found = cacheTable[v];
- assertex(found);
- if (found == goer)
- {
- cacheTable[v] = NULL;
- unsigned vn = v;
- loop
- {
- vn++;
- if (vn==cacheTableSize) vn = 0;
- IRoxieServerQueryPacket *found2 = cacheTable[vn];
- if (!found2)
- break;
- unsigned vm = found2->queryHash() % cacheTableSize;
- if (((vn+cacheTableSize-vm) % cacheTableSize)>=((vn+cacheTableSize-v) % cacheTableSize)) // diff(vn,vm)>=diff(vn,v)
- {
- cacheTable[v] = found2;
- v = vn;
- cacheTable[v] = NULL;
- }
- }
- cacheTableSpace++;
- break;
- }
- v++;
- if (v==cacheTableSize)
- v = 0;
- }
- goer->Release();
- }
- public:
- CRoxieServerSideCache(unsigned _cacheSize)
- {
- cacheTableSize = (_cacheSize*4)/3;
- cacheTable = new IRoxieServerQueryPacket *[cacheTableSize];
- memset(cacheTable, 0, cacheTableSize * sizeof(IRoxieServerQueryPacket *));
- cacheTableSpace = _cacheSize;
- prev = this;
- next = this;
- }
- ~CRoxieServerSideCache()
- {
- for (unsigned i = 0; i < cacheTableSize; i++)
- {
- ::Release(cacheTable[i]);
- }
- delete [] cacheTable;
- }
- virtual IRoxieServerQueryPacket *findCachedResult(const IRoxieContextLogger &logctx, IRoxieQueryPacket *p) const
- {
- unsigned hash = p->hash();
- unsigned et = hash % cacheTableSize;
- if (traceServerSideCache)
- {
- StringBuffer s;
- logctx.CTXLOG("CRoxieServerSideCache::findCachedResult hash %x slot %d %s", hash, et, p->queryHeader().toString(s).str());
- }
- CriticalBlock b(crit);
- loop
- {
- IRoxieServerQueryPacket *found = cacheTable[et];
- if (!found)
- return NULL;
- if (found->queryHash() == hash && found->queryPacket()->cacheMatch(p))
- {
- const_cast<CRoxieServerSideCache *>(this)->moveToHead(found);
- if (traceServerSideCache)
- logctx.CTXLOG("CRoxieServerSideCache::findCachedResult cache hit");
- logctx.noteStatistic(StNumServerCacheHits, 1);
- return NULL;
- // Because IMessageResult cannot be replayed, this scheme is flawed. I'm leaving the code here just as a stats gatherer to see how useful it would have been....
- //IRoxieServerQueryPacket *ret = new CRoxieServerQueryPacket(p);
- //ret->setResult(found->getResult());
- //return ret;
- }
- et++;
- if (et == cacheTableSize)
- et = 0;
- }
- }
- virtual void noteCachedResult(IRoxieServerQueryPacket *out, IMessageResult *in)
- {
- if (true) //!in->getLength()) // MORE - separate caches for hits and nohits
- {
- unsigned hash = out->queryPacket()->hash();
- out->setHash(hash);
- unsigned et = hash % cacheTableSize;
- if (traceServerSideCache)
- {
- StringBuffer s;
- DBGLOG("CRoxieServerSideCache::noteCachedResult hash %x slot %d %s", hash, et, out->queryPacket()->queryHeader().toString(s).str());
- }
- CriticalBlock b(crit);
- loop
- {
- IRoxieServerQueryPacket *found = cacheTable[et];
- if (!found)
- {
- if (cacheTableSpace)
- {
- out->setResult(LINK(in));
- cacheTable[et] = LINK(out);
- cacheTableSpace--;
- moveToHead(out);
- break;
- }
- else
- {
- IRoxieServerQueryPacket *goer = removeLRU();
- removeEntry(goer);
- et = hash % cacheTableSize;
- continue;
- }
- }
- else if (found->queryHash()==hash && found->queryPacket()->cacheMatch(out->queryPacket()))
- {
- moveToHead(found);
- return; // already in the cache. Because we don't cache until we have result, this can happen where
- // multiple copies of a slave query are in-flight at once.
- }
- et++;
- if (et == cacheTableSize)
- et = 0;
- }
- }
- // MORE - do we need to worry about the attachment between the MessageUnpacker and the current row manager. May all fall out ok...
- // Can I easily spot a null result? Do I want to cache null results separately? only?
- }
- // Note that this caching mechanism (unlike the old keyed-join specific one) does not common up cases where multiple
- // identical queries are in-flight at the same time. But if we can make it persistent between queries that will
- // more than make up for it
- };
- class CRowArrayMessageUnpackCursor : implements IMessageUnpackCursor, public CInterface
- {
- ConstPointerArray &data;
- Linked<IMessageResult> result;
- public:
- IMPLEMENT_IINTERFACE;
- CRowArrayMessageUnpackCursor(ConstPointerArray &_data, IMessageResult *_result)
- : data(_data), result(_result)
- {
- }
- virtual bool atEOF() const
- {
- return data.length()==0;
- }
- virtual bool isSerialized() const
- {
- return false;
- }
- virtual const void * getNext(int length)
- {
- if (!data.length())
- return NULL;
- const void *ret = data.item(0);
- data.remove(0);
- return ret;
- }
- };
- // MORE - should possibly move more over to the lazy version used in indexread?
- class CRowArrayMessageResult : implements IMessageResult, public CInterface
- {
- ConstPointerArray data;
- IRowManager &rowManager;
- bool variableSize;
- public:
- IMPLEMENT_IINTERFACE;
- CRowArrayMessageResult(IRowManager &_rowManager, bool _variableSize) : rowManager(_rowManager), variableSize(_variableSize)
- {
- }
- ~CRowArrayMessageResult()
- {
- ReleaseRoxieRows(data);
- }
- virtual IMessageUnpackCursor *getCursor(IRowManager *rowMgr) const
- {
- CRowArrayMessageResult *_this = (CRowArrayMessageResult *) this;
- return new CRowArrayMessageUnpackCursor(_this->data, _this);
- }
- virtual const void *getMessageHeader(unsigned &length) const
- {
- throwUnexpected(); // should never get called - I don't have a header available
- }
- virtual const void *getMessageMetadata(unsigned &length) const
- {
- length = 0;
- return NULL;
- }
- virtual void discard() const
- {
- throwUnexpected();
- }
- void append(const void *row)
- {
- data.append(row);
- }
- };
- void throwRemoteException(IMessageUnpackCursor *extra)
- {
- RecordLengthType *rowlen = (RecordLengthType *) extra->getNext(sizeof(RecordLengthType));
- if (rowlen)
- {
- char *xml = (char *) extra->getNext(*rowlen);
- ReleaseRoxieRow(rowlen);
- Owned<IPropertyTree> p = createPTreeFromXMLString(xml);
- ReleaseRoxieRow(xml);
- unsigned code = p->getPropInt("Code", 0);
- const char *msg = p->queryProp("Message");
- if (!msg)
- msg = xml;
- throw MakeStringException(code, "%s", msg);
- }
- throwUnexpected();
- }
- class CRemoteResultAdaptor : implements IEngineRowStream, implements IFinalRoxieInput, implements IExceptionHandler, public CInterface
- {
- friend class CRemoteResultMerger;
- class CRemoteResultMerger
- {
- class HeapEntry : public CInterface
- {
- private:
- CRemoteResultAdaptor &adaptor;
- IMessageUnpackCursor *cursor;
- public:
- const void *current;
- bool isLast;
- bool lastIsComplete;
- IRoxieServerQueryPacket *packet;
- unsigned seq;
- public:
- inline const void *noteResult(IMessageUnpackCursor *_cursor, bool _lastIsComplete)
- {
- cursor = _cursor;
- lastIsComplete = _lastIsComplete;
- return next();
- }
- public:
- HeapEntry(CRemoteResultAdaptor &_adaptor, IRoxieServerQueryPacket *_packet, unsigned _seq) : adaptor(_adaptor), packet(_packet), seq(_seq)
- {
- cursor = NULL;
- current = NULL;
- isLast = false;
- lastIsComplete = true;
- }
- ~HeapEntry()
- {
- ::Release(packet);
- ::Release(cursor);
- ReleaseRoxieRow(current);
- }
- bool isCompleteMatch() const
- {
- if (!isLast || lastIsComplete)
- return true;
- else
- return false;
- }
- const void *next()
- {
- if (cursor)
- {
- ReleaseClearRoxieRow(current);
- current = adaptor.getRow(cursor);
- isLast = cursor->atEOF();
- if (!current)
- {
- cursor->Release();
- cursor = NULL;
- }
- }
- return current;
- }
- unsigned skipTo(IRangeCompare *compare, const void *seek, unsigned numFields, bool requireExactMatch)
- {
- // MORE - This loop should possibly be a binchop... though it's not absolutely clear that is true (depends on term frequencies)
- unsigned skipped = 0;
- loop
- {
- int c = compare->docompare(current, seek, numFields);
- //If larger than the seek values, then we may be allowed to return an inexact match,
- //if equal then it is required to be an exact match,
- if (c > 0)
- {
- if (!requireExactMatch || isCompleteMatch())
- break;
- }
- else if ((c == 0) && isCompleteMatch())
- break;
- skipped++;
- if (!next())
- break;
- }
- return skipped;
- }
- };
- CRemoteResultAdaptor &adaptor;
- CIArrayOf<HeapEntry> heapEntries;
- UnsignedArray heap;
- IRowManager *rowManager;
- unsigned numPending;
- unsigned numFields;
- bool endSeen;
- bool remakePending;
- IRangeCompare *compare;
- bool deferredContinuation;
- inline int doCompare(unsigned l, unsigned r)
- {
- int ret = compare->docompare(heapEntries.item(l).current, heapEntries.item(r).current, numFields);
- if (!ret) ret = heapEntries.item(l).seq - heapEntries.item(r).seq;
- return ret;
- }
- void makeHeap()
- {
- /* Permute blocks to establish the heap property
- For each element p, the children are p*2+1 and p*2+2 (provided these are in range)
- The children of p must both be greater than or equal to p
- The parent of a child c is given by p = (c-1)/2
- */
- unsigned i;
- unsigned n = heap.length();
- unsigned *s = heap.getArray();
- for (i=1; i<n; i++)
- {
- unsigned r = s[i];
- int c = i; /* child */
- while (c > 0)
- {
- int p = (c-1)/2; /* parent */
- if ( doCompare( s[c], s[p] ) >= 0 )
- break;
- s[c] = s[p];
- s[p] = r;
- c = p;
- }
- }
- remakePending = false;
- }
- void remakeHeap()
- {
- /* The row associated with block[0] will have changed
- This code restores the heap property
- */
- unsigned p = 0; /* parent */
- unsigned n = heap.length();
- unsigned *s = heap.getArray();
- while (1)
- {
- unsigned c = p*2 + 1; /* child */
- if ( c >= n )
- break;
- /* Select smaller child */
- if ( c+1 < n && doCompare( s[c+1], s[c] ) < 0 ) c += 1;
- /* If child is greater or equal than parent then we are done */
- if ( doCompare( s[c], s[p] ) >= 0 )
- break;
- /* Swap parent and child */
- unsigned r = s[c];
- s[c] = s[p];
- s[p] = r;
- /* child becomes parent */
- p = c;
- }
- remakePending = false;
- }
- void append(IRoxieServerQueryPacket *p, unsigned seq)
- {
- HeapEntry &h = *new HeapEntry(adaptor, LINK(p), seq);
- IMessageResult *result = p->queryResult();
- assertex(result);
- if (h.noteResult(result->getCursor(rowManager), isCompleteMatchFlag(result)))
- {
- heapEntries.append(h);
- heap.append(heap.ordinality());
- }
- else
- h.Release();
- }
- void removeHeap(unsigned idx)
- {
- heapEntries.remove(idx);
- ForEachItemIn(i, heap)
- {
- unsigned v = heap.item(i);
- assertex(v != idx);
- if (v > idx)
- heap.replace(v-1, i);
- }
- }
- bool isCompleteMatchFlag(IMessageResult *result)
- {
- unsigned metaLen;
- const byte *metaInfo = (const byte *) result->getMessageMetadata(metaLen);
- if (metaLen)
- {
- unsigned short continuationLen = *(unsigned short *) metaInfo;
- if (continuationLen >= sizeof(bool))
- {
- metaInfo += sizeof(unsigned short);
- return *(bool *) metaInfo;
- }
- }
- return true; // if no continuation info, last row was complete.
- }
- public:
- CRemoteResultMerger(CRemoteResultAdaptor &_adaptor) : adaptor(_adaptor)
- {
- init(NULL, NULL);
- }
- void init(ISteppingMeta *meta, IRowManager *_rowManager)
- {
- if (meta)
- {
- numFields = meta->getNumFields();
- compare = meta->queryCompare();
- }
- else
- {
- numFields = 0;
- compare = NULL;
- }
- rowManager = _rowManager;
- numPending = 0;
- endSeen = false;
- remakePending = false;
- deferredContinuation = false;
- }
- void reset()
- {
- heapEntries.kill();
- heap.kill();
- numPending = 0;
- endSeen = false;
- remakePending = false;
- deferredContinuation = false;
- }
- inline bool noteEndSeen()
- {
- bool hadSeen = endSeen;
- if (!endSeen)
- makeHeap();
- endSeen = true;
- return !hadSeen;
- }
- void noteResult(IRoxieServerQueryPacket *p, unsigned seq)
- {
- if (!p->isContinuation())
- append(p, seq);
- else
- {
- ForEachItemIn(idx, heapEntries)
- {
- HeapEntry &h = heapEntries.item(idx);
- if (h.packet == p)
- {
- IMessageResult *result = p->queryResult();
- if (!h.noteResult(result->getCursor(rowManager), isCompleteMatchFlag(result)))
- {
- heap.zap(idx);
- removeHeap(idx);
- }
- numPending--;
- if (!numPending)
- makeHeap();
- return;
- }
- }
- }
- // If we get here it must be a continuation for one that I have not yet consumed... we don't need to do anything.
- return;
- }
- unsigned skipRows(unsigned &idx, const void *seek, const void *rawSeek, unsigned numFields, unsigned seekLen, const SmartStepExtra & stepExtra)
- {
- HeapEntry &entry = heapEntries.item(idx);
- unsigned skipped = entry.current ? entry.skipTo(compare, seek, numFields, !stepExtra.returnMismatches()) : 0;
- if (!entry.current)
- {
- IRoxieServerQueryPacket *continuation = entry.packet->queryContinuation();
- if (continuation)
- {
- continuation->Link();
- entry.packet->Release();
- entry.packet = continuation;
- if (continuation->hasResult())
- {
- IMessageResult *result = continuation->queryResult();
- bool lastIsCompleteMatch = isCompleteMatchFlag(result);
- entry.noteResult(result->getCursor(rowManager), lastIsCompleteMatch);
- }
- else
- {
- if (continuation->isDelayed())
- {
- continuation->setDelayed(false);
- MemoryBuffer serializedSkip;
- adaptor.activity.serializeSkipInfo(serializedSkip, seekLen, rawSeek, numFields, seek, stepExtra);
- continuation->setPacket(continuation->queryPacket()->insertSkipData(serializedSkip.length(), serializedSkip.toByteArray()));
- ROQ->sendPacket(continuation->queryPacket(), adaptor.activity.queryLogCtx());
- adaptor.sentsome.signal();
- }
- numPending++;
- }
- }
- else
- {
- heap.zap(idx);
- removeHeap(idx);
- idx--;
- }
- }
- return skipped;
- }
- const void * nextRowGE(const void * seek, const void *rawSeek, unsigned numFields, unsigned seeklen, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- // We discard all rows < seekval from all entries in heap
- // If this results in additional slave requests, we return NULL so that we can wait for them
- // If not, we rebuild the heap (if any were skipped) and return the first row
- deferredContinuation = false;
- if (heap.length())
- {
- unsigned skipped = 0;
- unsigned idx = 0;
- while(heapEntries.isItem(idx))
- {
- skipped += skipRows(idx, seek, rawSeek, numFields, seeklen, stepExtra);
- idx++;
- }
- if (numPending)
- return NULL; // can't answer yet, need more results from slaves
- else
- {
- if (skipped)
- makeHeap();
- return next(wasCompleteMatch, stepExtra);
- }
- }
- else
- return NULL;
- }
- bool doContinuation(HeapEntry &topEntry, bool canDefer)
- {
- IRoxieServerQueryPacket *continuation = topEntry.packet->queryContinuation();
- if (continuation)
- {
- if (continuation->isDelayed() && canDefer)
- {
- if (adaptor.activity.queryLogCtx().queryTraceLevel() > 10)
- adaptor.activity.queryLogCtx().CTXLOG("Deferring continuation");
- deferredContinuation = true;
- }
- else
- {
- deferredContinuation = false;
- continuation->Link();
- topEntry.packet->Release();
- topEntry.packet = continuation;
- if (continuation->hasResult())
- {
- IMessageResult *result = continuation->queryResult();
- bool lastIsCompleteMatch = isCompleteMatchFlag(result);
- topEntry.noteResult(result->getCursor(rowManager), lastIsCompleteMatch);
- }
- else
- {
- if (continuation->isDelayed()) // has the continuation been requested yet?
- {
- continuation->Link();
- topEntry.packet->Release();
- topEntry.packet = continuation;
- continuation->setDelayed(false);
- if (adaptor.activity.queryLogCtx().queryTraceLevel() > 10)
- adaptor.activity.queryLogCtx().CTXLOG("About to send continuation, from doContinuation");
- ROQ->sendPacket(continuation->queryPacket(), adaptor.activity.queryLogCtx());
- adaptor.sentsome.signal();
- }
- numPending++; // we are waiting for one that is already in flight
- }
- }
- return true; // next not known yet
- }
- else
- return false;
- }
- const void *next(bool & wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- OwnedConstRoxieRow ret;
- if (heap.length())
- {
- if (deferredContinuation)
- {
- unsigned top = heap.item(0);
- HeapEntry &topEntry = heapEntries.item(top);
- doContinuation(topEntry, false);
- return NULL;
- }
- if (remakePending)
- remakeHeap();
- unsigned top = heap.item(0);
- HeapEntry &topEntry = heapEntries.item(top);
- ret.set(topEntry.current);
- wasCompleteMatch = topEntry.isCompleteMatch();
- const void *next = topEntry.next();
- if (!next)
- {
- if (!doContinuation(topEntry, stepExtra.returnMismatches()))
- {
- unsigned last = heap.popGet();
- if (heap.length())
- heap.replace(last, 0);
- removeHeap(top);
- }
- }
- remakePending = true;
- }
- return ret.getClear();
- }
- bool ready()
- {
- return endSeen && numPending == 0;
- }
- };
- IRoxieServerQueryPacket *createRoxieServerQueryPacket(IRoxieQueryPacket *p, bool &cached)
- {
- if (serverSideCache && !debugContext)
- {
- IRoxieServerQueryPacket *ret = serverSideCache->findCachedResult(activity.queryLogCtx(), p);
- if (ret)
- {
- p->Release();
- cached = true;
- return ret;
- }
- }
- cached = false;
- return new CRoxieServerQueryPacket(p);
- }
- #ifdef _DEBUG
- void dumpPending()
- {
- CriticalBlock b(pendingCrit);
- ForEachItemIn(idx, pending)
- {
- IRoxieServerQueryPacket &p = pending.item(idx);
- StringBuffer s;
- unsigned __int64 dummy;
- if (p.isEnd())
- s.append("END");
- else if (p.isLimit(dummy, dummy, dummy))
- s.append("LIMIT");
- else
- {
- IRoxieQueryPacket *i = p.queryPacket();
- s.appendf("%s", p.hasResult() ? "COMPLETE " : "PENDING ");
- if (i)
- {
- RoxiePacketHeader &header = i->queryHeader();
- header.toString(s);
- }
- }
- DBGLOG("Pending %d %s", idx, s.str());
- }
- }
- #endif
- void abortPending()
- {
- CriticalBlock b(pendingCrit);
- ForEachItemIn(idx, pending)
- {
- IRoxieServerQueryPacket &p = pending.item(idx);
- if (!p.hasResult())
- {
- IRoxieQueryPacket *i = p.queryPacket();
- if (i)
- {
- RoxiePacketHeader &header = i->queryHeader();
- ROQ->sendAbort(header, activity.queryLogCtx());
- }
- }
- }
- pending.kill();
- }
- void checkDelayed()
- {
- if (ctx->queryDebugContext() && ctx->queryDebugContext()->getExecuteSequentially())
- {
- bool allDelayed = true;
- CriticalBlock b(pendingCrit);
- unsigned sendIdx = (unsigned) -1;
- ForEachItemIn(idx, pending)
- {
- IRoxieServerQueryPacket &p = pending.item(idx);
- if (p.queryPacket())
- {
- if (p.isDelayed())
- {
- if (sendIdx == (unsigned) -1)
- sendIdx = idx;
- }
- else if (!p.hasResult())
- {
- allDelayed = false;
- break;
- }
- }
- }
- if (allDelayed && sendIdx != (unsigned) -1)
- {
- if (activity.queryLogCtx().queryTraceLevel() > 10)
- activity.queryLogCtx().CTXLOG("About to send debug-deferred from next");
- pending.item(sendIdx).setDelayed(false);
- ROQ->sendPacket(pending.item(sendIdx).queryPacket(), activity.queryLogCtx());
- sentsome.signal();
- }
- }
- else if (deferredStart)
- {
- CriticalBlock b(pendingCrit);
- ForEachItemIn(idx, pending)
- {
- IRoxieServerQueryPacket &p = pending.item(idx);
- if (p.isDelayed())
- {
- if (activity.queryLogCtx().queryTraceLevel() > 10)
- activity.queryLogCtx().CTXLOG("About to send deferred start from next");
- p.setDelayed(false);
- ROQ->sendPacket(p.queryPacket(), activity.queryLogCtx());
- sentsome.signal();
- }
- }
- deferredStart = false;
- }
- }
- void retryPending()
- {
- CriticalBlock b(pendingCrit);
- checkDelayed();
- ForEachItemIn(idx, pending)
- {
- IRoxieServerQueryPacket &p = pending.item(idx);
- if (!p.hasResult() && !p.isDelayed())
- {
- IRoxieQueryPacket *i = p.queryPacket();
- if (i)
- {
- if (!i->queryHeader().retry())
- {
- StringBuffer s;
- IException *E = MakeStringException(ROXIE_MULTICAST_ERROR, "Failed to get response from slave(s) for %s in activity %d", i->queryHeader().toString(s).str(), activity.queryId());
- activity.queryLogCtx().logOperatorException(E, __FILE__, __LINE__, "CRemoteResultAdaptor::retry");
- throw E;
- }
- if (!localSlave)
- {
- ROQ->sendPacket(i, activity.queryLogCtx());
- atomic_inc(&retriesSent);
- }
- }
- }
- }
- }
- class ChannelBuffer
- {
- protected:
- unsigned bufferLeft;
- MemoryBuffer buffer;
- char *nextBuf;
- unsigned overflowSequence;
- unsigned channel; // == bonded channel
- bool needsFlush;
- InterruptableSemaphore flowController;
- const CRemoteResultAdaptor &owner;
- CriticalSection crit;
- public:
- ChannelBuffer(const CRemoteResultAdaptor &_owner, unsigned _channel) : owner(_owner), channel(_channel), flowController(perChannelFlowLimit)
- {
- overflowSequence = 0;
- needsFlush = false;
- bufferLeft = 0;
- nextBuf = NULL;
- }
- void init(unsigned minSize)
- {
- assertex(!buffer.length());
- if (minSize < MIN_PAYLOAD_SIZE)
- minSize = MIN_PAYLOAD_SIZE;
- unsigned headerSize = sizeof(RoxiePacketHeader)+owner.headerLength();
- unsigned bufferSize = headerSize+minSize;
- if (bufferSize < mtu_size)
- bufferSize = mtu_size;
- buffer.reserveTruncate(bufferSize);
- bufferLeft = bufferSize - headerSize;
- assertex(buffer.toByteArray());
- nextBuf = (char *) buffer.toByteArray() + headerSize;
- needsFlush = false;
- }
- inline IRoxieQueryPacket *flush()
- {
- CriticalBlock cb(crit);
- Owned<IRoxieQueryPacket> ret;
- if (needsFlush)
- {
- buffer.setLength(nextBuf - buffer.toByteArray());
- RoxiePacketHeader *h = (RoxiePacketHeader *) buffer.toByteArray();
- h->init(owner.remoteId, owner.ruid, channel, overflowSequence);
- //patch logPrefix, cachedContext and parent extract into the place reserved in the message buffer
- byte * tgt = (byte*)(h+1);
- owner.copyHeader(tgt, channel);
- ret.setown(createRoxiePacket(buffer));
- if (overflowSequence == OVERFLOWSEQUENCE_MAX)
- overflowSequence = 1; // don't wrap to 0 - that is a bit special
- else
- overflowSequence++;
- needsFlush = false;
- bufferLeft = 0;
- if (owner.flowControlled)
- {
- CriticalUnblock cub(crit);
- while (!flowController.wait(1000))
- {
- StringBuffer s;
- owner.activity.queryLogCtx().CTXLOG("Channel %d blocked by flow control: %s", channel, h->toString(s).str());
- }
- }
- }
- return ret.getClear();
- }
- inline void signal()
- {
- if (owner.flowControlled)
- flowController.signal();
- }
- inline void interrupt(IException *e)
- {
- flowController.interrupt(e);
- }
- inline void *getBuffer(unsigned size)
- {
- CriticalBlock cb(crit);
- if (bufferLeft >= size)
- {
- needsFlush = true;
- void * ret = nextBuf;
- nextBuf += size;
- bufferLeft -= size;
- return ret;
- }
- else if (!needsFlush)
- {
- init(size);
- return getBuffer(size);
- }
- else if (owner.mergeOrder)
- {
- return buffer.reserve(size); // whole query needs to go as single packet if we are to merge
- }
- else
- return NULL; // will force it to flush and start a new packet
- }
- };
- private:
- friend class CRemoteResultMerger;
- bool allread;
- bool contextCached;
- bool preserveOrder;
- InterruptableSemaphore sentsome;
- Owned <IMessageCollator> mc;
- Owned<IMessageUnpackCursor> mu;
- Owned<IMessageResult> mr;
- ChannelBuffer **buffers;
- IHThorArg &helper;
- unsigned __int64 stopAfter;
- unsigned resendSequence;
- IHThorArg *colocalArg;
- IArrayOf<IRoxieServerQueryPacket> pending;
- CriticalSection pendingCrit;
- IRoxieServerSideCache *serverSideCache;
- unsigned sentSequence;
- Owned<IOutputRowDeserializer> deserializer;
- Owned<IEngineRowAllocator> rowAllocator;
- CRemoteResultMerger merger;
- // this is only used to avoid recreating a bufferStream for each row. A better solution may be needed
- MemoryBuffer tempRowBuffer;
- Owned<ISerialStream> bufferStream;
- CThorStreamDeserializerSource rowSource;
- protected:
- IRowManager *rowManager;
- unsigned __int64 rowLimit;
- unsigned __int64 keyedLimit;
- IRoxieServerErrorHandler *errorHandler;
- CachedOutputMetaData meta;
- public:
- ISteppingMeta *mergeOrder;
- IRoxieSlaveContext *ctx;
- IDebuggableContext *debugContext;
- IRoxieServerActivity &activity;
- unsigned parentExtractSize;
- const byte * parentExtract;
- bool flowControlled;
- bool deferredStart;
- MemoryBuffer logInfo;
- MemoryBuffer cachedContext;
- const RemoteActivityId &remoteId;
- ruid_t ruid;
- mutable CriticalSection buffersCrit;
- unsigned processed;
- cycle_t totalCycles;
- bool timeActivities;
- //private: //vc6 doesn't like this being private yet accessed by nested class...
- const void *getRow(IMessageUnpackCursor *mu)
- {
- if (!mu->isSerialized() || (meta.isFixedSize() && !deserializer))
- return mu->getNext(meta.getFixedSize());
- else
- {
- RecordLengthType *rowlen = (RecordLengthType *) mu->getNext(sizeof(RecordLengthType));
- if (rowlen)
- {
- RecordLengthType len = *rowlen;
- ReleaseRoxieRow(rowlen);
- const void *slaveRec = mu->getNext(len);
- if (deserializer && mu->isSerialized())
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- tempRowBuffer.setBuffer(len, const_cast<void *>(slaveRec), false);
- size_t outsize = deserializer->deserialize(rowBuilder, rowSource);
- ReleaseRoxieRow(slaveRec);
- return rowBuilder.finalizeRowClear(outsize);
- }
- else
- return slaveRec;
- }
- else
- return NULL;
- }
- }
- private:
- ChannelBuffer *queryChannelBuffer(unsigned channel, bool force=false)
- {
- CriticalBlock cb(buffersCrit);
- ChannelBuffer *b = buffers[channel];
- if (!b && force)
- {
- if (!contextCached)
- {
- logInfo.clear();
- unsigned char loggingFlags = LOGGING_FLAGSPRESENT | LOGGING_TRACELEVELSET;
- unsigned char ctxTraceLevel = activity.queryLogCtx().queryTraceLevel() + 1; // Avoid passing a 0
- const QueryOptions &options = ctx->queryOptions();
- if (activity.queryLogCtx().isIntercepted())
- loggingFlags |= LOGGING_INTERCEPTED;
- if (options.timeActivities)
- loggingFlags |= LOGGING_TIMEACTIVITIES;
- if (activity.queryLogCtx().isBlind())
- loggingFlags |= LOGGING_BLIND;
- if (options.checkingHeap)
- loggingFlags |= LOGGING_CHECKINGHEAP;
- if (ctx->queryWorkUnit())
- loggingFlags |= LOGGING_WUID;
- if (debugContext)
- {
- loggingFlags |= LOGGING_DEBUGGERACTIVE;
- logInfo.append(loggingFlags).append(ctxTraceLevel);
- MemoryBuffer bpInfo;
- debugContext->serialize(bpInfo);
- bpInfo.append((__uint64)(memsize_t) &activity);
- logInfo.append((unsigned short) bpInfo.length());
- logInfo.append(bpInfo.length(), bpInfo.toByteArray());
- }
- else
- logInfo.append(loggingFlags).append(ctxTraceLevel);
- StringBuffer logPrefix;
- activity.queryLogCtx().getLogPrefix(logPrefix);
- logInfo.append(logPrefix);
- activity.serializeCreateStartContext(cachedContext.clear());
- activity.serializeExtra(cachedContext);
- if (activity.queryVarFileInfo())
- {
- activity.queryVarFileInfo()->queryTimeStamp().serialize(cachedContext);
- cachedContext.append(activity.queryVarFileInfo()->queryCheckSum());
- }
- contextCached = true;
- }
- b = buffers[channel] = new ChannelBuffer(*this, channel);
- }
- return b;
- }
- void processRow(const void *got)
- {
- processed++;
- if (processed > rowLimit)
- {
- ReleaseRoxieRow(got);
- errorHandler->onLimitExceeded(false); // NOTE - should throw exception
- throwUnexpected();
- }
- else if (processed > keyedLimit)
- {
- ReleaseRoxieRow(got);
- errorHandler->onLimitExceeded(true); // NOTE - should throw exception
- throwUnexpected();
- }
- }
- public:
- IMPLEMENT_IINTERFACE;
- CRemoteResultAdaptor(IRoxieSlaveContext *_ctx, IRoxieServerErrorHandler *_errorHandler, const RemoteActivityId &_remoteId, IOutputMetaData *_meta, IHThorArg &_helper, IRoxieServerActivity &_activity, bool _preserveOrder, bool _flowControlled)
- : remoteId(_remoteId), meta(_meta), activity(_activity), helper(_helper), preserveOrder(_preserveOrder), flowControlled(_flowControlled), merger(*this)
- {
- ctx = _ctx;
- errorHandler = _errorHandler;
- colocalArg = NULL;
- allread = false;
- ruid = 0;
- rowLimit = (unsigned __int64) -1;
- keyedLimit = (unsigned __int64) -1;
- contextCached = false;
- stopAfter = I64C(0x7FFFFFFFFFFFFFFF);
- buffers = new ChannelBuffer*[numChannels+1];
- memset(buffers, 0, (numChannels+1)*sizeof(ChannelBuffer *));
- parentExtractSize = 0;
- rowManager = NULL;
- parentExtract = NULL;
- debugContext = NULL;
- mergeOrder = NULL;
- deferredStart = false;
- processed = 0;
- sentSequence = 0;
- resendSequence = 0;
- totalCycles = 0;
- serverSideCache = activity.queryServerSideCache();
- bufferStream.setown(createMemoryBufferSerialStream(tempRowBuffer));
- rowSource.setStream(bufferStream);
- timeActivities = defaultTimeActivities;
- }
- ~CRemoteResultAdaptor()
- {
- if (mc)
- {
- ROQ->queryReceiveManager()->detachCollator(mc);
- mc.clear();
- }
- for (unsigned channel = 0; channel <= numChannels; channel++)
- {
- delete(buffers[channel]);
- }
- delete [] buffers;
- }
- void setMeta(IOutputMetaData *newmeta)
- {
- meta.set(newmeta);
- }
- virtual IEngineRowStream *queryConcreteOutputStream(unsigned whichInput) { assertex(whichInput==0); return this; }
- virtual IStrandJunction *queryConcreteOutputJunction(unsigned idx) const { assertex(idx==0); return nullptr; }
- virtual IRoxieServerActivity *queryActivity()
- {
- return &activity;
- }
- virtual IIndexReadActivityInfo *queryIndexReadActivity()
- {
- return NULL;
- }
- void setMergeInfo(ISteppingMeta *_mergeOrder)
- {
- mergeOrder = _mergeOrder;
- deferredStart = true;
- }
- void send(IRoxieQueryPacket *p)
- {
- if (p)
- {
- Linked<IRoxieQueryPacket> saver(p); // avoids a race with abortPending, without keeping pendingCrit locked over the send which we might prefer not to
- assertex(p->queryHeader().uid==ruid);
- // MORE: Maybe we should base the fastlane flag on some other
- // criteria !! (i.e A Roxie server prediction based on the
- // activity type/activity behaviour/expected reply size .. etc).
- //
- // Currently (code below) based on high priority, seq=0, and none-child activity.
- // But this could still cause too many reply packets on the fatlane
- // (higher priority output Q), which may cause the activities on the
- // low priority output Q to not get service on time.
- if ((colocalArg == 0) && // not a child query activity??
- (p->queryHeader().activityId & (ROXIE_SLA_PRIORITY | ROXIE_HIGH_PRIORITY)) &&
- (p->queryHeader().overflowSequence == 0) &&
- (p->queryHeader().continueSequence & ~CONTINUE_SEQUENCE_SKIPTO)==0)
- p->queryHeader().retries |= ROXIE_FASTLANE;
- if (p->queryHeader().channel)
- {
- bool cached = false;
- IRoxieServerQueryPacket *rsqp = createRoxieServerQueryPacket(p, cached);
- if (deferredStart)
- rsqp->setDelayed(true);
- rsqp->setSequence(sentSequence++);
- {
- CriticalBlock b(pendingCrit);
- pending.append(*rsqp);
- }
- if (!deferredStart)
- {
- if (!cached)
- ROQ->sendPacket(p, activity.queryLogCtx());
- sentsome.signal();
- }
- }
- else
- {
- // Care is needed here. If I send the packet before I add to the pending there is a danger that I'll get results that I discard
- // Need to add first, then send
- unsigned i;
- bool allCached = true;
- for (i = 1; i <= numChannels; i++)
- {
- IRoxieQueryPacket *q = p->clonePacket(i);
- bool thisChannelCached;
- IRoxieServerQueryPacket *rsqp = createRoxieServerQueryPacket(q, thisChannelCached);
- if (!thisChannelCached)
- allCached = false;
- rsqp->setSequence(sentSequence++);
- if (deferredStart)
- {
- rsqp->setDelayed(true);
- }
- {
- CriticalBlock b(pendingCrit);
- pending.append(*rsqp);
- }
- if (!deferredStart)
- sentsome.signal();
- }
- if (!allCached && !deferredStart)
- ROQ->sendPacket(p, activity.queryLogCtx());
- buffers[0]->signal(); // since replies won't come back on that channel...
- p->Release();
- }
- }
- }
- void *getMem(unsigned partNo, unsigned fileNo, unsigned size)
- {
- unsigned channel = partNo ? getBondedChannel(partNo) : 0;
- size += sizeof(PartNoType);
- ChannelBuffer *b = queryChannelBuffer(channel, true);
- char *buffer = (char *) b->getBuffer(size);
- if (!buffer)
- {
- send(b->flush());
- buffer = (char *) b->getBuffer(size);
- }
- PartNoType sp;
- sp.partNo = partNo;
- sp.fileNo = fileNo;
- memcpy(buffer, &sp, sizeof(sp));
- buffer += sizeof(sp);
- return buffer;
- }
- void injectResult(IMessageResult *result)
- {
- IRoxieServerQueryPacket *f = new CRoxieServerQueryPacket(NULL);
- f->setSequence(sentSequence++);
- f->setResult(result);
- CriticalBlock b(pendingCrit);
- pending.append(*f);
- sentsome.signal(); // MORE - arguably should only send if there is any point waking up the listener thread, to save context swicth
- }
- void flush()
- {
- for (unsigned channel = 0; channel <= numChannels; channel++)
- {
- ChannelBuffer *b = queryChannelBuffer(channel, false);
- if (b)
- send(b->flush());
- }
- }
- void interruptBuffers(IException *e)
- {
- for (unsigned channel = 0; channel <= numChannels; channel++)
- {
- ChannelBuffer *b = queryChannelBuffer(channel, false);
- if (b)
- b->interrupt(LINK(e));
- }
- }
- void senddone()
- {
- CriticalBlock b(pendingCrit);
- pending.append(*new CRoxieServerQueryPacketEndMarker);
- sentsome.signal();
- }
- bool fireException(IException *e)
- {
- {
- CriticalBlock b(pendingCrit);
- pending.append(*new CRoxieServerQueryPacketEndMarker);
- }
- interruptBuffers(e);
- if (mc)
- mc->interrupt(LINK(e));
- sentsome.interrupt(e);
- return true;
- }
- virtual void onCreate(IHThorArg *_colocalArg)
- {
- debugContext = ctx->queryDebugContext();
- colocalArg = _colocalArg;
- if (meta.needsSerializeDisk())
- {
- deserializer.setown(meta.createDiskDeserializer(ctx->queryCodeContext(), activity.queryId()));
- rowAllocator.setown(activity.createRowAllocator(meta.queryOriginal()));
- }
- if (ctx->queryDebugContext() && ctx->queryDebugContext()->getExecuteSequentially())
- deferredStart = true;
- timeActivities = ctx->queryOptions().timeActivities;
- }
- virtual void onStart(unsigned _parentExtractSize, const byte * _parentExtract)
- {
- #ifdef TRACE_STARTSTOP
- if (traceStartStop)
- activity.queryLogCtx().CTXLOG("RRAonstart");
- #endif
- sentsome.reinit();
- ruid = getNextRuid();
- rowManager = &ctx->queryRowManager();
- if (mergeOrder)
- merger.init(mergeOrder, rowManager);
- if (mc)
- {
- ROQ->queryReceiveManager()->detachCollator(mc); // Should never happen - implies someone forgot to call onReset!
- }
- mc.setown(ROQ->queryReceiveManager()->createMessageCollator(rowManager, ruid));
- allread = false;
- mu.clear();
- contextCached = false;
- processed = 0;
- resendSequence = 0;
- sentSequence = 0;
- for (unsigned channel = 0; channel <= numChannels; channel++)
- {
- delete(buffers[channel]);
- buffers[channel] = NULL;
- }
- flush();
- parentExtractSize = _parentExtractSize;
- parentExtract = _parentExtract;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- #ifdef TRACE_STARTSTOP
- if (traceStartStop)
- activity.queryLogCtx().CTXLOG("RRAstart");
- #endif
- activity.start(parentExtractSize, parentExtract, paused);
- }
- void setLimits(unsigned __int64 _rowLimit, unsigned __int64 _keyedLimit, unsigned __int64 _stopAfter)
- {
- if (ctx->queryProbeManager())
- {
- if (_rowLimit != (unsigned __int64) -1) ctx->queryProbeManager()->setNodePropertyInt(&activity, "rowLimit", _rowLimit);
- if (_keyedLimit != (unsigned __int64) -1) ctx->queryProbeManager()->setNodePropertyInt(&activity, "keyedLimit", _keyedLimit);
- if (_stopAfter != I64C(0x7FFFFFFFFFFFFFFF)) ctx->queryProbeManager()->setNodePropertyInt(&activity, "choosenLimit", _stopAfter);
- }
- {
- CriticalBlock b(pendingCrit);
- if (pending.length())
- {
- #ifdef _DEBUG
- dumpPending(); // MORE - only defined in debug build - could have put the ifdef inside the dumpPending method
- #endif
- assertex(pending.length()==0);
- }
- pending.append(*new CRoxieServerQueryPacketLimitMarker(_rowLimit, _keyedLimit, _stopAfter));
- }
- sentsome.signal();
- rowLimit = _rowLimit;
- keyedLimit = _keyedLimit;
- stopAfter = _stopAfter;
- }
- virtual void stop()
- {
- #ifdef TRACE_STARTSTOP
- if (traceStartStop)
- activity.queryLogCtx().CTXLOG("RRAstop");
- #endif
- onStop();
- activity.stop();
- }
- void onStop()
- {
- #ifdef TRACE_STARTSTOP
- if (traceStartStop)
- activity.queryLogCtx().CTXLOG("RRAonstop");
- #endif
- abortPending();
- interruptBuffers(NULL);
- sentsome.interrupt();
- if (mc) // May not be set if start() chain threw exception
- mc->interrupt();
- }
- virtual void reset()
- {
- #ifdef TRACE_STARTSTOP
- if (traceStartStop)
- activity.queryLogCtx().CTXLOG("RRAreset");
- #endif
- activity.reset();
- onReset();
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- // Make sure parent activity calls up the tree
- PointerArrayOf<IEngineRowStream> instreams;
- IStrandJunction *junction = activity.getOutputStreams(ctx, idx, instreams, consumerOptions, consumerOrdered, orderedCallbacks);
- assertex (junction == NULL);
- // I return a single strand (we could change that sometime I guess...)
- streams.append(this);
- return NULL;
- }
- virtual void resetEOF()
- {
- throwUnexpected();
- }
- virtual void onReset()
- {
- #ifdef TRACE_STARTSTOP
- if (traceStartStop)
- activity.queryLogCtx().CTXLOG("RRAonreset");
- #endif
- if (mc)
- ROQ->queryReceiveManager()->detachCollator(mc);
- merger.reset();
- pending.kill();
- if (mc && ctx)
- ctx->addSlavesReplyLen(mc->queryBytesReceived());
- mc.clear(); // Or we won't free memory for graphs that get recreated
- mu.clear(); //ditto
- deferredStart = false;
- // NOTE: do NOT clear mergeOrder - this is set at create time not per child query
- }
- virtual IOutputMetaData * queryOutputMeta() const
- {
- return helper.queryOutputMeta();
- }
- virtual unsigned __int64 queryTotalCycles() const
- {
- return totalCycles;
- }
- const void * nextRowGE(const void *seek, const void *rawSeek, unsigned numFields, unsigned seekLen, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- if (activity.queryLogCtx().queryTraceLevel() > 20)
- {
- StringBuffer recstr;
- unsigned i;
- for (i = 0; i < seekLen; i++)
- {
- recstr.appendf("%02x ", ((unsigned char *) rawSeek)[i]);
- }
- activity.queryLogCtx().CTXLOG("CRemoteResultAdaptor::nextRowGE(rawSeek=%s numFields=%d, seeklen=%d, returnMismatches=%d)", recstr.str(), numFields, seekLen, stepExtra.returnMismatches());
- }
- assertex(mergeOrder);
- if (deferredStart)
- {
- CriticalBlock b(pendingCrit);
- ForEachItemIn(idx, pending)
- {
- IRoxieServerQueryPacket &p = pending.item(idx);
- if (p.isDelayed())
- {
- p.setDelayed(false);
- if (activity.queryLogCtx().queryTraceLevel() > 10)
- activity.queryLogCtx().CTXLOG("About to send deferred start from nextRowGE, setting requireExact to %d", !stepExtra.returnMismatches());
- MemoryBuffer serializedSkip;
- activity.serializeSkipInfo(serializedSkip, seekLen, rawSeek, numFields, seek, stepExtra);
- p.setPacket(p.queryPacket()->insertSkipData(serializedSkip.length(), serializedSkip.toByteArray()));
- ROQ->sendPacket(p.queryPacket(), activity.queryLogCtx());
- sentsome.signal();
- }
- }
- deferredStart = false;
- }
- SimpleActivityTimer t(totalCycles, timeActivities);
- if (processed==stopAfter)
- return NULL;
- if (allread)
- return NULL;
- loop
- {
- if (merger.ready())
- {
- const void *got = merger.nextRowGE(seek, rawSeek, numFields, seekLen, wasCompleteMatch, stepExtra);
- if (got)
- {
- processRow(got);
- return got;
- }
- }
- if (!reload()) // MORE - should pass the seek info here...
- return NULL;
- }
- }
- virtual const void *nextRow()
- {
- // If we are merging then we need to do a heapsort on all
- SimpleActivityTimer t(totalCycles, timeActivities);
- if (activity.queryLogCtx().queryTraceLevel() > 10)
- {
- activity.queryLogCtx().CTXLOG("CRemoteResultAdaptor::nextRow()");
- }
- loop
- {
- checkDelayed();
- if (processed==stopAfter)
- return NULL;
- if (allread)
- return NULL;
- // If we can still consume from the merger or the most recently retrieved mu, do so.
- const void *got = NULL;
- if (mergeOrder && merger.ready())
- {
- bool matched = true;
- got = merger.next(matched, dummySmartStepExtra);
- }
- else if (mu)
- got = getRow(mu);
- if (got)
- {
- processRow(got);
- return got;
- }
- if (!reload())
- return NULL;
- }
- }
- bool reload()
- {
- // Wait for something to be returned from a slave....
- mu.clear();
- sentsome.wait();
- // must be at least an endMarker on the queue since sentsome was signalled
- {
- CriticalBlock b(pendingCrit);
- IRoxieServerQueryPacket &top = pending.item(0);
- if (top.isLimit(rowLimit, keyedLimit, stopAfter)) // This is really a start marker...
- {
- pending.remove(0);
- return true;
- }
- else if (top.isEnd())
- {
- pending.remove(0);
- allread = true;
- if (activity.queryLogCtx().queryTraceLevel() > 5)
- activity.queryLogCtx().CTXLOG("All read on ruid %x", ruid);
- return false;
- }
- else if (mergeOrder)
- {
- unsigned idx = 0;
- bool added = false;
- while (pending.isItem(idx))
- {
- IRoxieServerQueryPacket &item = pending.item(idx);
- if (item.isEnd())
- {
- if (merger.noteEndSeen())
- {
- sentsome.signal(); // Because we waited, yet didn't actually consume anything
- added = true;
- }
- break;
- }
- else if (item.hasResult())
- {
- merger.noteResult(&item, item.getSequence());
- pending.remove(idx);
- added = true;
- }
- else if (item.isContinuation())
- idx++;
- else
- break;
- }
- if (added)
- return true;
- }
- else if (top.hasResult())
- {
- mr.setown(pending.item(0).getResult());
- mu.setown(mr->getCursor(rowManager));
- pending.remove(0);
- return true;
- }
- }
- getNextUnpacker();
- return true;
- }
- void getNextUnpacker()
- {
- mu.clear();
- unsigned ctxTraceLevel = activity.queryLogCtx().queryTraceLevel();
- loop
- {
- checkDelayed();
- unsigned timeout = remoteId.isSLAPriority() ? slaTimeout : (remoteId.isHighPriority() ? highTimeout : lowTimeout);
- activity.queryContext()->checkAbort();
- bool anyActivity;
- if (ctxTraceLevel > 5)
- activity.queryLogCtx().CTXLOG("Calling getNextUnpacker(%d)", timeout);
- mr.setown(mc->getNextResult(timeout, anyActivity));
- if (ctxTraceLevel > 6)
- activity.queryLogCtx().CTXLOG("Called getNextUnpacker(%d), activity=%d", timeout, anyActivity);
- activity.queryContext()->checkAbort();
- if (mr)
- {
- unsigned roxieHeaderLen;
- const RoxiePacketHeader &header = *(const RoxiePacketHeader *) mr->getMessageHeader(roxieHeaderLen);
- #ifdef _DEBUG
- assertex(roxieHeaderLen == sizeof(RoxiePacketHeader));
- #endif
- if (ctxTraceLevel > 5)
- {
- StringBuffer s;
- activity.queryLogCtx().CTXLOG("getNextUnpacker got packet %s", header.toString(s).str());
- }
- CriticalBlock b(pendingCrit);
- unsigned idx = 0;
- IRoxieServerQueryPacket *original = NULL;
- IRoxieQueryPacket *op;
- while (pending.isItem(idx))
- {
- original = &pending.item(idx);
- op = original->queryPacket();
- if (op && header.matchPacket(op->queryHeader()))
- break;
- original = NULL;
- idx++;
- }
- if (!original || original->hasResult())
- {
- switch (header.activityId)
- {
- case ROXIE_FILECALLBACK:
- {
- // tell slave to abort
- //if (ctxTraceLevel > 5)
- {
- StringBuffer s;
- activity.queryLogCtx().CTXLOG("Redundant callback on query %s", header.toString(s).str());
- }
- Owned<IMessageUnpackCursor> callbackData = mr->getCursor(rowManager);
- OwnedConstRoxieRow len = callbackData->getNext(sizeof(RecordLengthType));
- if (len)
- {
- RecordLengthType *rowlen = (RecordLengthType *) len.get();
- OwnedConstRoxieRow row = callbackData->getNext(*rowlen);
- const char *rowdata = (const char *) row.get();
- // bool isOpt = * (bool *) rowdata;
- // bool isLocal = * (bool *) (rowdata+1);
- ROQ->sendAbortCallback(header, rowdata+2, activity.queryLogCtx());
- }
- else
- throwUnexpected();
- break;
- }
- // MORE - ROXIE_ALIVE perhaps should go here too
- default:
- if (ctxTraceLevel > 3)
- activity.queryLogCtx().CTXLOG("Discarding packet %p - original %p is NULL or has result already", mr.get(), original);
- mr->discard();
- break;
- }
- mr.clear();
- }
- else
- {
- atomic_inc(&resultsReceived);
- switch (header.activityId)
- {
- case ROXIE_DEBUGCALLBACK:
- {
- Owned<IMessageUnpackCursor> callbackData = mr->getCursor(rowManager);
- OwnedConstRoxieRow len = callbackData->getNext(sizeof(RecordLengthType));
- if (len)
- {
- RecordLengthType *rowlen = (RecordLengthType *) len.get();
- OwnedConstRoxieRow row = callbackData->getNext(*rowlen);
- char *rowdata = (char *) row.get();
- if (ctxTraceLevel > 5)
- {
- StringBuffer s;
- activity.queryLogCtx().CTXLOG("Callback on query %s for debug", header.toString(s).str());
- }
- MemoryBuffer slaveInfo;
- slaveInfo.setBuffer(*rowlen, rowdata, false);
- unsigned debugSequence;
- slaveInfo.read(debugSequence);
- Owned<IRoxieQueryPacket> reply = original->getDebugResponse(debugSequence);
- if (!reply)
- reply.setown(activity.queryContext()->queryDebugContext()->onDebugCallback(header, *rowlen, rowdata));
- if (reply)
- {
- original->setDebugResponse(debugSequence, reply);
- ROQ->sendPacket(reply, activity.queryLogCtx());
- }
- }
- else
- throwUnexpected();
- // MORE - somehow we need to make sure slave gets a reply even if I'm not waiting (in udp layer)
- // Leave original message on pending queue in original location - this is not a reply to it.
- break;
- }
- case ROXIE_FILECALLBACK:
- {
- // we need to send back to the slave a message containing the file info requested.
- Owned<IMessageUnpackCursor> callbackData = mr->getCursor(rowManager);
- OwnedConstRoxieRow len = callbackData->getNext(sizeof(RecordLengthType));
- if (len)
- {
- RecordLengthType *rowlen = (RecordLengthType *) len.get();
- OwnedConstRoxieRow row = callbackData->getNext(*rowlen);
- const char *rowdata = (const char *) row.get();
- bool isOpt = * (bool *) rowdata;
- bool isLocal = * (bool *) (rowdata+1);
- const char *lfn = rowdata+2;
- if (ctxTraceLevel > 5)
- {
- StringBuffer s;
- activity.queryLogCtx().CTXLOG("Callback on query %s file %s", header.toString(s).str(),(const char *) lfn);
- }
- activity.queryContext()->onFileCallback(header, lfn, isOpt, isLocal);
- }
- else
- throwUnexpected();
- // MORE - somehow we need to make sure slave gets a reply even if I'm not waiting (in udp layer)
- // Leave original message on pending queue in original location - this is not a reply to it.
- break;
- }
- case ROXIE_KEYEDLIMIT_EXCEEDED:
- activity.queryLogCtx().CTXLOG("ROXIE_KEYEDLIMIT_EXCEEDED");
- errorHandler->onLimitExceeded(true); // NOTE - should throw exception!
- throwUnexpected();
- case ROXIE_LIMIT_EXCEEDED:
- activity.queryLogCtx().CTXLOG("ROXIE_LIMIT_EXCEEDED");
- errorHandler->onLimitExceeded(false); // NOTE - should throw exception!
- throwUnexpected();
- case ROXIE_TRACEINFO:
- {
- Owned<IMessageUnpackCursor> extra = mr->getCursor(rowManager);
- loop
- {
- RecordLengthType *rowlen = (RecordLengthType *) extra->getNext(sizeof(RecordLengthType));
- if (rowlen)
- {
- char *logInfo = (char *) extra->getNext(*rowlen);
- MemoryBuffer buf;
- buf.setBuffer(*rowlen, logInfo, false);
- switch ((TracingCategory) *logInfo)
- {
- case LOG_TRACING:
- case LOG_ERROR:
- activity.queryLogCtx().CTXLOGl(new LogItem(buf));
- break;
- case LOG_STATVALUES:
- buf.skip(1);
- activity.mergeStats(buf);
- break;
- case LOG_CHILDCOUNT:
- case LOG_CHILDSTATS:
- unsigned graphId, childId;
- buf.skip(1);
- buf.read(graphId);
- buf.read(childId);
- if (*logInfo == LOG_CHILDCOUNT)
- {
- unsigned idx;
- unsigned childProcessed;
- unsigned childStrands;
- buf.read(idx);
- buf.read(childProcessed);
- buf.read(childStrands);
- if (traceLevel > 5)
- activity.queryLogCtx().CTXLOG("Processing ChildCount %d idx %d strands %d for child %d subgraph %d", childProcessed, idx, childStrands, childId, graphId);
- activity.queryContext()->noteProcessed(graphId, childId, idx, childProcessed, childStrands);
- }
- else
- {
- ActivityTimeAccumulator dummy; // We could serialize from slave? Would get confusing though
- CRuntimeStatisticCollection childStats(allStatistics);
- childStats.deserialize(buf);
- if (traceLevel > 5)
- {
- StringBuffer s;
- activity.queryLogCtx().CTXLOG("Processing ChildStats for child %d subgraph %d: %s", childId, graphId, childStats.toStr(s).str());
- }
- activity.queryContext()->mergeActivityStats(childStats, graphId, childId, dummy, 0);
- }
- }
- ReleaseRoxieRow(rowlen);
- ReleaseRoxieRow(logInfo);
- }
- else
- break;
- }
- break;
- }
- case ROXIE_EXCEPTION:
- if (ctxTraceLevel > 1)
- {
- StringBuffer s;
- activity.queryLogCtx().CTXLOG("Exception on query %s", header.toString(s).str());
- }
- op->queryHeader().noteException(header.retries);
- if (op->queryHeader().allChannelsFailed())
- {
- activity.queryLogCtx().CTXLOG("Multiple exceptions on query - aborting");
- Owned<IMessageUnpackCursor> exceptionData = mr->getCursor(rowManager);
- throwRemoteException(exceptionData);
- }
- // Leave it on pending queue in original location
- break;
- case ROXIE_ALIVE:
- if (ctxTraceLevel > 4)
- {
- StringBuffer s;
- activity.queryLogCtx().CTXLOG("ROXIE_ALIVE: %s", header.toString(s).str());
- }
- op->queryHeader().noteAlive(header.retries & ROXIE_RETRIES_MASK);
- // Leave it on pending queue in original location
- break;
- default:
- if (header.retries & ROXIE_RETRIES_MASK)
- atomic_inc(&retriesNeeded);
- unsigned metaLen;
- const void *metaData = mr->getMessageMetadata(metaLen);
- if (metaLen)
- {
- // We got back first chunk but there is more.
- // resend the packet, with the cursor info provided.
- // MORE - if smart-stepping, we don't want to send the continuation immediately. Other cases it's not clear that we do.
- if (ctxTraceLevel > 1)
- {
- StringBuffer s;
- activity.queryLogCtx().CTXLOG("Additional data size %d on query %s mergeOrder %p", metaLen, header.toString(s).str(), mergeOrder);
- }
- if (*((unsigned short *) metaData) + sizeof(unsigned short) != metaLen)
- {
- StringBuffer s;
- activity.queryLogCtx().CTXLOG("Additional data size %d on query %s mergeOrder %p", metaLen, header.toString(s).str(), mergeOrder);
- activity.queryLogCtx().CTXLOG("Additional data is corrupt");
- throwUnexpected();
- }
- MemoryBuffer nextQuery;
- nextQuery.append(sizeof(RoxiePacketHeader), &header);
- nextQuery.append(metaLen, metaData);
- nextQuery.append(op->getTraceLength(), op->queryTraceInfo());
- nextQuery.append(op->getContextLength(), op->queryContextData());
- if (resendSequence == CONTINUESEQUENCE_MAX)
- {
- activity.queryLogCtx().CTXLOG("ERROR: Continuation sequence wrapped"); // shouldn't actually matter.... but suggests a very iffy query!
- resendSequence = 1;
- }
- else
- resendSequence++;
- RoxiePacketHeader *newHeader = (RoxiePacketHeader *) nextQuery.toByteArray();
- newHeader->continueSequence = resendSequence; // NOTE - we clear the skipTo flag since continuation of a skip is NOT a skip...
- newHeader->retries &= ~ROXIE_RETRIES_MASK;
- IRoxieQueryPacket *resend = createRoxiePacket(nextQuery);
- CRoxieServerQueryPacket *fqp = new CRoxieServerQueryPacket(resend);
- fqp->setSequence(original->getSequence());
- pending.add(*fqp, idx+1); // note that pending takes ownership. sendPacket does not release.
- original->setContinuation(LINK(fqp));
- if (mergeOrder)
- fqp->setDelayed(true);
- else
- {
- ROQ->sendPacket(resend, activity.queryLogCtx());
- sentsome.signal();
- }
- // Note that we don't attempt to cache results that have continuation records - too tricky !
- }
- else
- {
- if (serverSideCache)
- serverSideCache->noteCachedResult(original, mr);
- }
- unsigned channel = header.channel;
- {
- ChannelBuffer *b = queryChannelBuffer(channel); // If not something is wrong, or we sent out on channel 0?
- if (b)
- b->signal();
- }
- original->setResult(mr.getClear());
- sentsome.signal();
- return;
- }
- }
- }
- else
- {
- if (!anyActivity && !localSlave)
- {
- activity.queryLogCtx().CTXLOG("Input has stalled - retry required?");
- retryPending();
- }
- }
- }
- }
- inline unsigned headerLength() const
- {
- return logInfo.length() + cachedContext.length() + sizeof(unsigned) + parentExtractSize;
- }
- void copyHeader(byte *tgt, unsigned channel) const
- {
- unsigned len = logInfo.length();
- memcpy(tgt, logInfo.toByteArray(), len);
- tgt += len;
- *(unsigned *) tgt = parentExtractSize;
- tgt += sizeof(unsigned);
- memcpy(tgt, parentExtract, parentExtractSize);
- tgt += parentExtractSize;
- memcpy(tgt, cachedContext.toByteArray(), cachedContext.length());
- tgt += cachedContext.length();
- }
- };
- class CSkippableRemoteResultAdaptor : public CRemoteResultAdaptor
- {
- Owned <IException> exception;
- bool skipping;
- ConstPointerArray buff;
- unsigned index;
- bool pulled;
- void pullInput()
- {
- try
- {
- if (exception)
- throw exception.getClear();
- unsigned __int64 count = 0;
- loop
- {
- const void * next = CRemoteResultAdaptor::nextRow();
- if (next == NULL)
- {
- next = CRemoteResultAdaptor::nextRow();
- if(next == NULL)
- break;
- buff.append(NULL);
- }
- count++;
- if (count > rowLimit)
- {
- ReleaseRoxieRow(next);
- ReleaseRoxieRows(buff);
- errorHandler->onLimitExceeded(false); // throws an exception - user or LimitSkipException
- throwUnexpected();
- }
- else if (count > keyedLimit)
- {
- ReleaseRoxieRow(next);
- ReleaseRoxieRows(buff);
- errorHandler->onLimitExceeded(true); // throws an exception - user or LimitSkipException
- throwUnexpected();
- }
- buff.append(next);
- }
- }
- catch (IException *E)
- {
- if (QUERYINTERFACE(E, LimitSkipException))
- {
- Owned<IException> cleanup = E;
- ReleaseRoxieRows(buff);
- const void *onfail = errorHandler->createLimitFailRow(E->errorCode() == KeyedLimitSkipErrorCode);
- if (onfail)
- buff.append(onfail);
- }
- else
- throw;
- }
- pulled = true;
- }
- public:
- CSkippableRemoteResultAdaptor(IRoxieSlaveContext *_ctx, IRoxieServerErrorHandler *_errorHandler, const RemoteActivityId &_remoteId, IOutputMetaData *_meta, IHThorArg &_helper, IRoxieServerActivity &_activity, bool _preserveOrder, bool _flowControlled, bool _skipping) :
- CRemoteResultAdaptor(_ctx, _errorHandler, _remoteId, _meta, _helper, _activity, _preserveOrder, _flowControlled)
- {
- skipping = _skipping;
- index = 0;
- pulled = false;
- }
- void setException(IException *E)
- {
- exception.setown(E);
- }
- virtual void onReset()
- {
- roxiemem::ReleaseRoxieRowRange(buff.getArray(), index, buff.ordinality());
- buff.kill();
- index = 0;
- pulled = false;
- exception.clear();
- CRemoteResultAdaptor::onReset();
- }
- void onStart(unsigned _parentExtractSize, const byte * _parentExtract)
- {
- index = 0;
- pulled = false;
- CRemoteResultAdaptor::onStart(_parentExtractSize, _parentExtract);
- }
- virtual const void * nextRowGE(const void *seek, const void *rawSeek, unsigned numFields, unsigned seeklen, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- // MORE - not sure what we need to do about the skip case... but we need at least this to prevent issues with exception getting lost
- if (exception)
- throw exception.getClear();
- return CRemoteResultAdaptor::nextRowGE(seek, rawSeek, numFields, seeklen, wasCompleteMatch, stepExtra);
- }
- virtual const void *nextRow()
- {
- if (skipping)
- {
- if(!pulled)
- pullInput();
- if(buff.isItem(index))
- {
- const void * next = buff.item(index++);
- if(next)
- processed++;
- return next;
- }
- return NULL;
- }
- else
- {
- if (exception)
- throw exception.getClear();
- return CRemoteResultAdaptor::nextRow();
- }
- }
- };
- //=================================================================================
- class CRoxieServerApplyActivity : public CRoxieServerInternalSinkActivity
- {
- IHThorApplyArg &helper;
- public:
- CRoxieServerApplyActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, 0), helper((IHThorApplyArg &) basehelper)
- {
- }
- virtual void onExecute()
- {
- helper.start();
- loop
- {
- const void * next = inputStream->ungroupedNextRow();
- if (!next)
- break;
- helper.apply(next);
- ReleaseRoxieRow(next);
- }
- helper.end();
- }
- };
- class CRoxieServerApplyActivityFactory : public CRoxieServerActivityFactory
- {
- bool isRoot;
- public:
- CRoxieServerApplyActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), isRoot(_isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerApplyActivity(_ctx, this, _probeManager);
- }
- virtual bool isSink() const
- {
- return isRoot;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerApplyActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerApplyActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //=================================================================================
- class CRoxieServerNullActivity : public CRoxieServerActivity
- {
- public:
- CRoxieServerNullActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager)
- {
- }
- virtual const void *nextRow()
- {
- return NULL;
- }
- };
- IRoxieServerActivity * createRoxieServerNullActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- {
- return new CRoxieServerNullActivity(_ctx, _factory, _probeManager);
- }
- class CRoxieServerNullActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerNullActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerNullActivity(_ctx, this, _probeManager);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for null activity");
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNullActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerNullActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerPassThroughActivity : public CRoxieServerActivity
- {
- public:
- CRoxieServerPassThroughActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager)
- {
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- inputStream->resetEOF();
- }
- virtual const void *nextRow()
- {
- const void * next = inputStream->nextRow();
- if (next)
- processed++;
- return next;
- }
- virtual bool isPassThrough()
- {
- return true;
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * next = inputStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- if (next)
- processed++;
- return next;
- }
- IInputSteppingMeta * querySteppingMeta()
- {
- return input->querySteppingMeta();
- }
- };
- IRoxieServerActivity * createRoxieServerPassThroughActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- {
- return new CRoxieServerPassThroughActivity(_ctx, _factory, _probeManager);
- }
- //=================================================================================
- class CRoxieServerChildBaseActivity : public CRoxieServerActivity
- {
- protected:
- bool eof;
- bool first;
- public:
- CRoxieServerChildBaseActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager)
- {
- eof = false;
- first = true;
- }
- ~CRoxieServerChildBaseActivity()
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- first = true;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- };
- class CRoxieServerChildBaseActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerChildBaseActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for %s activity", getActivityText(kind));
- }
- };
- //=================================================================================
- class CRoxieServerChildIteratorActivity : public CRoxieServerChildBaseActivity
- {
- IHThorChildIteratorArg &helper;
- public:
- CRoxieServerChildIteratorActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerChildBaseActivity(_ctx, _factory, _probeManager), helper((IHThorChildIteratorArg &) basehelper)
- {
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- bool ok;
- if (first)
- {
- ok = helper.first();
- first = false;
- }
- else
- ok = helper.next();
- try
- {
- while (ok)
- {
- processed++;
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned outSize = helper.transform(rowBuilder);
- if (outSize)
- return rowBuilder.finalizeRowClear(outSize);
- ok = helper.next();
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- eof = true;
- return NULL;
- }
- };
- class CRoxieServerChildIteratorActivityFactory : public CRoxieServerChildBaseActivityFactory
- {
- public:
- CRoxieServerChildIteratorActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerChildBaseActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerChildIteratorActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerChildIteratorActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerChildIteratorActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerChildNormalizeActivity : public CRoxieServerChildBaseActivity
- {
- IHThorChildNormalizeArg &helper;
- public:
- CRoxieServerChildNormalizeActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerChildBaseActivity(_ctx, _factory, _probeManager), helper((IHThorChildNormalizeArg &) basehelper)
- {
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- bool ok;
- if (first)
- {
- ok = helper.first();
- first = false;
- }
- else
- ok = helper.next();
- if (ok)
- {
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- do {
- unsigned outSize = helper.transform(rowBuilder);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- ok = helper.next();
- }
- while (ok);
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- eof = true;
- return NULL;
- }
- };
- class CRoxieServerChildNormalizeActivityFactory : public CRoxieServerChildBaseActivityFactory
- {
- public:
- CRoxieServerChildNormalizeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerChildBaseActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerChildNormalizeActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNewChildNormalizeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerChildNormalizeActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerChildAggregateActivity : public CRoxieServerChildBaseActivity
- {
- IHThorChildAggregateArg &helper;
- public:
- CRoxieServerChildAggregateActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerChildBaseActivity(_ctx, _factory, _probeManager), helper((IHThorChildAggregateArg &) basehelper)
- {
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- eof = true;
- processed++;
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- helper.clearAggregate(rowBuilder);
- helper.processRows(rowBuilder);
- size32_t finalSize = meta.getRecordSize(rowBuilder.getSelf());
- return rowBuilder.finalizeRowClear(finalSize);
- }
- };
- class CRoxieServerChildAggregateActivityFactory : public CRoxieServerChildBaseActivityFactory
- {
- public:
- CRoxieServerChildAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerChildBaseActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerChildAggregateActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNewChildAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerChildAggregateActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerChildGroupAggregateActivity : public CRoxieServerChildBaseActivity, public IHThorGroupAggregateCallback
- {
- IHThorChildGroupAggregateArg &helper;
- RowAggregator aggregated;
- public:
- IMPLEMENT_IINTERFACE_USING(CRoxieServerChildBaseActivity)
- CRoxieServerChildGroupAggregateActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerChildBaseActivity(_ctx, _factory, _probeManager), helper((IHThorChildGroupAggregateArg &) basehelper),
- aggregated(helper, helper)
- {
- }
- void processRow(const void * next)
- {
- aggregated.addRow(next);
- }
-
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerChildBaseActivity::start(parentExtractSize, parentExtract, paused);
- aggregated.start(rowAllocator);
- }
- virtual void reset()
- {
- aggregated.reset();
- CRoxieServerChildBaseActivity::reset();
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- if (first)
- {
- helper.processRows(this);
- first = false;
- }
- Owned<AggregateRowBuilder> next = aggregated.nextResult();
- if (next)
- {
- processed++;
- return next->finalizeRowClear();
- }
- eof = true;
- return NULL;
- }
- };
- class CRoxieServerChildGroupAggregateActivityFactory : public CRoxieServerChildBaseActivityFactory
- {
- public:
- CRoxieServerChildGroupAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerChildBaseActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerChildGroupAggregateActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNewChildGroupAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerChildGroupAggregateActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerChildThroughNormalizeActivity : public CRoxieServerChildBaseActivity
- {
- IHThorChildThroughNormalizeArg &helper;
- const void * lastInput;
- unsigned numProcessedLastGroup;
- bool ok;
- public:
- CRoxieServerChildThroughNormalizeActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerChildBaseActivity(_ctx, _factory, _probeManager), helper((IHThorChildThroughNormalizeArg &) basehelper)
- {
- lastInput = NULL;
- numProcessedLastGroup = 0;
- ok = false;
- }
- virtual void stop()
- {
- CRoxieServerChildBaseActivity::stop();
- ReleaseRoxieRow(lastInput);
- lastInput = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerChildBaseActivity::start(parentExtractSize, parentExtract, paused);
- numProcessedLastGroup = processed;
- ok = false;
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- loop
- {
- if (ok)
- ok = helper.next();
- while (!ok)
- {
- ReleaseRoxieRow(lastInput);
- lastInput = inputStream->nextRow();
- if (!lastInput)
- {
- if (numProcessedLastGroup != processed)
- {
- numProcessedLastGroup = processed;
- return NULL;
- }
- lastInput = inputStream->nextRow();
- if (!lastInput)
- return NULL;
- }
- ok = helper.first(lastInput);
- }
- try
- {
- do
- {
- unsigned outSize = helper.transform(rowBuilder);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- ok = helper.next();
- } while (ok);
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerChildThroughNormalizeActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerChildThroughNormalizeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerChildThroughNormalizeActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNewChildThroughNormalizeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerChildThroughNormalizeActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerDistributionActivity : public CRoxieServerInternalSinkActivity
- {
- IHThorDistributionArg &helper;
- public:
- CRoxieServerDistributionActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, 0), helper((IHThorDistributionArg &)basehelper)
- {
- }
- virtual void onExecute()
- {
- MemoryAttr ma;
- IDistributionTable * * accumulator = (IDistributionTable * *)ma.allocate(helper.queryInternalRecordSize()->getMinRecordSize());
- helper.clearAggregate(accumulator);
- OwnedConstRoxieRow nextrec(inputStream->nextRow());
- loop
- {
- if (!nextrec)
- {
- nextrec.setown(inputStream->nextRow());
- if (!nextrec)
- break;
- }
- helper.process(accumulator, nextrec);
- nextrec.setown(inputStream->nextRow());
- }
- StringBuffer result;
- result.append("<XML>");
- helper.gatherResult(accumulator, result);
- result.append("</XML>");
- helper.sendResult(result.length(), result.str());
- helper.destruct(accumulator);
- }
- };
- class CRoxieServerDistributionActivityFactory : public CRoxieServerActivityFactory
- {
- bool isRoot;
- public:
- CRoxieServerDistributionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), isRoot(_isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerDistributionActivity(_ctx, this, _probeManager);
- }
- virtual bool isSink() const
- {
- return isRoot;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerDistributionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerDistributionActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //=================================================================================
- class CRoxieServerLinkedRawIteratorActivity : public CRoxieServerActivity
- {
- IHThorLinkedRawIteratorArg &helper;
- public:
- CRoxieServerLinkedRawIteratorActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorLinkedRawIteratorArg &) basehelper)
- {
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void *ret =helper.next();
- if (ret)
- {
- LinkRoxieRow(ret);
- processed++;
- }
- return ret;
- }
- };
- class CRoxieServerLinkedRawIteratorActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerLinkedRawIteratorActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerLinkedRawIteratorActivity(_ctx, this, _probeManager);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for %s activity", getActivityText(kind));
- }
- };
- IRoxieServerActivityFactory *createRoxieServerLinkedRawIteratorActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerLinkedRawIteratorActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerDatasetResultActivity : public CRoxieServerActivity
- {
- public:
- CRoxieServerDatasetResultActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager)
- {
- }
- virtual const void *nextRow()
- {
- throwUnexpected();
- }
- virtual void executeChild(size32_t & retSize, void * & ret, unsigned parentExtractSize, const byte * parentExtract)
- {
- try
- {
- start(parentExtractSize, parentExtract, false);
- {
- ActivityTimer t(totalCycles, timeActivities);
- MemoryBuffer result;
- IRecordSize * inputMeta = input->queryOutputMeta();
- loop
- {
- const void *nextrec = inputStream->ungroupedNextRow();
- if (!nextrec)
- break;
- result.append(inputMeta->getRecordSize(nextrec), nextrec);
- ReleaseRoxieRow(nextrec);
- }
- retSize = result.length();
- ret = result.detach();
- }
- stop();
- reset();
- }
- catch(IException *E)
- {
- ctx->notifyAbort(E);
- abort();
- reset();
- throw;
- }
- catch(...)
- {
- Owned<IException> E = MakeStringException(ROXIE_INTERNAL_ERROR, "Unknown exception caught at %s:%d", __FILE__, __LINE__);
- ctx->notifyAbort(E);
- abort();
- reset();
- throw;
- }
- }
- };
- class CRoxieServerDatasetResultActivityFactory : public CRoxieServerActivityFactory
- {
- bool isRoot;
- public:
- CRoxieServerDatasetResultActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), isRoot(_isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerDatasetResultActivity(_ctx, this, _probeManager);
- }
- virtual bool isSink() const
- {
- return isRoot;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerDatasetResultActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerDatasetResultActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //=================================================================================
- class CRoxieServerInlineTableActivity : public CRoxieServerActivity
- {
- IHThorInlineTableArg &helper;
- __uint64 curRow;
- __uint64 numRows;
- public:
- CRoxieServerInlineTableActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorInlineTableArg &) basehelper)
- {
- curRow = 0;
- numRows = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- curRow = 0;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- numRows = helper.numRows();
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- // Filtering empty rows, returns the next valid row
- while (curRow < numRows)
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned outSize = helper.getRow(rowBuilder, curRow++);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- return NULL;
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() called for source activity");
- }
- };
- class CRoxieServerStrandedInlineTableActivity : public CRoxieServerStrandedActivity
- {
- IHThorInlineTableArg &helper;
- __uint64 numRows;
- //An inline table that isn't stranded
- class InlineTableSimpleProcessor : public StrandProcessor
- {
- protected:
- IHThorInlineTableArg &helper;
- __uint64 curRow;
- __uint64 maxRows;
- public:
- InlineTableSimpleProcessor(CRoxieServerActivity &_parent, IHThorInlineTableArg &_helper)
- : StrandProcessor(_parent, NULL, true), helper(_helper), curRow(0), maxRows(0)
- {
- }
- virtual void start() override
- {
- StrandProcessor::start();
- curRow = 0;
- maxRows = queryParent().numRows;
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- //Return rows while the current section is active
- while (curRow < maxRows)
- {
- RtlDynamicRowBuilder rowBuilder(*rowAllocator);
- unsigned outSize = helper.getRow(rowBuilder, curRow++);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- return NULL;
- }
- CRoxieServerStrandedInlineTableActivity & queryParent() { return static_cast<CRoxieServerStrandedInlineTableActivity&>(parent); }
- };
- class InlineTableStrandProcessor : public StrandProcessor
- {
- protected:
- IHThorInlineTableArg &helper;
- unsigned whichStrand;
- unsigned numStrands = 0;
- __uint64 sectionSize;
- __uint64 curRow = 0;
- __uint64 sectionMaxRows = 0;
- __uint64 maxRows = 0;
- bool isOrdered;
- bool eosPending;
- public:
- InlineTableStrandProcessor(CRoxieServerActivity &_parent, IHThorInlineTableArg &_helper, unsigned _whichStrand, bool _isOrdered)
- : StrandProcessor(_parent, NULL, true), helper(_helper), whichStrand(_whichStrand), isOrdered(_isOrdered)
- {
- sectionSize = queryParent().strandOptions.blockSize;
- eosPending = false;
- }
- virtual void start() override
- {
- StrandProcessor::start();
- numStrands = queryParent().numStrands();
- maxRows = queryParent().numRows;
- curRow = std::min(whichStrand * sectionSize, maxRows);
- sectionMaxRows = std::min(curRow + sectionSize, maxRows);
- eosPending = isOrdered && (curRow != maxRows);
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- //Return rows while the current section is active
- while (curRow < sectionMaxRows)
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned outSize = helper.getRow(rowBuilder, curRow++);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- if (curRow == maxRows)
- return NULL;
- //Update current position to the next section
- curRow += (numStrands - 1) * sectionSize;
- if (curRow > maxRows)
- curRow = maxRows;
- sectionMaxRows = std::min(curRow + sectionSize, maxRows);
- //If we have output a block of rows, then send the block as a unit - results now come from the next strand
- if (eosPending)
- {
- //If there is another block, then second an end of section marker after it, otherwise make sure NULL is returned next.
- eosPending = isOrdered && (curRow != maxRows);
- return queryEndOfSectionMarker();
- }
- }
- }
- CRoxieServerStrandedInlineTableActivity & queryParent() { return static_cast<CRoxieServerStrandedInlineTableActivity&>(parent); }
- };
- public:
- CRoxieServerStrandedInlineTableActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const StrandOptions &_strandOptions)
- : CRoxieServerStrandedActivity(_ctx, _factory, _probeManager, _strandOptions), helper((IHThorInlineTableArg &) basehelper)
- {
- numRows = 0;
- }
- virtual StrandProcessor *createStrandProcessor(IEngineRowStream *instream) { throwUnexpected(); }
- virtual StrandProcessor *createStrandSourceProcessor(bool inputOrdered)
- {
- if (strandOptions.numStrands <= 1)
- return new InlineTableSimpleProcessor(*this, helper);
- return new InlineTableStrandProcessor(*this, helper, strands.ordinality(), inputOrdered);
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerStrandedActivity::start(parentExtractSize, parentExtract, paused);
- numRows = helper.numRows();
- onStartStrands();
- startJunction(sourceJunction); // This must be started *after* all the strands have been initialised
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() called for source activity");
- }
- };
- class CRoxieServerInlineTableActivityFactory : public CRoxieServerActivityFactory
- {
- StrandOptions strandOptions;
- public:
- CRoxieServerInlineTableActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), strandOptions(_graphNode)
- {
- optStableInput = false; // do not force the output to be ordered
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerStrandedInlineTableActivity(_ctx, this, _probeManager, strandOptions);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for InlineTable activity");
- }
- };
- IRoxieServerActivityFactory *createRoxieServerInlineTableActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerInlineTableActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerWorkUnitReadActivity : public CRoxieServerActivity
- {
- IHThorWorkunitReadArg &helper;
- CriticalSection readerCrit;
- Owned<IWorkUnitRowReader> wuReader;
- public:
- CRoxieServerWorkUnitReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorWorkunitReadArg &)basehelper)
- {
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- if (!ctx->queryServerContext())
- {
- throw MakeStringException(ROXIE_INTERNAL_ERROR, "Workunit read activity cannot be executed in slave context");
- }
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- IXmlToRowTransformer * xmlTransformer = helper.queryXmlTransformer();
- OwnedRoxieString fromWuid(helper.getWUID());
- wuReader.setown(ctx->getWorkunitRowReader(fromWuid, helper.queryName(), helper.querySequence(), xmlTransformer, rowAllocator, meta.isGrouped()));
- // MORE _ should that be in onCreate?
- }
- virtual void reset()
- {
- {
- CriticalBlock b(readerCrit);
- wuReader.clear();
- }
- CRoxieServerActivity::reset();
- };
- virtual bool needsAllocator() const { return true; }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- Linked<IWorkUnitRowReader> useReader;
- {
- CriticalBlock b(readerCrit);
- if (!wuReader)
- return NULL;
- useReader.set(wuReader);
- }
- const void *ret = useReader->nextRow();
- if (ret)
- processed++;
- return ret;
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() called for source activity");
- }
- };
- class CRoxieServerWorkUnitReadActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerWorkUnitReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerWorkUnitReadActivity(_ctx, this, _probeManager);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for WorkUnitRead activity");
- }
- };
- IRoxieServerActivityFactory *createRoxieServerWorkUnitReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerWorkUnitReadActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- interface ILocalGraphEx : public IEclGraphResults
- {
- public:
- virtual void setResult(unsigned id, IGraphResult * result) = 0;
- virtual IEngineRowStream * createResultIterator(unsigned id) = 0;
- virtual void setGraphLoopResult(IGraphResult * result) = 0;
- virtual IEngineRowStream * createGraphLoopResultIterator(unsigned id) = 0;
- };
- class CSafeRoxieInput : implements IEngineRowStream, implements IFinalRoxieInput, public CInterface
- {
- public:
- CSafeRoxieInput(unsigned _sourceIdx, IFinalRoxieInput * _input) : input(_input), inputStream(NULL), sourceIdx(_sourceIdx) {}
- IMPLEMENT_IINTERFACE
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks) override
- {
- connectInputStreams(ctx, consumerOrdered);
- streams.append(this);
- return NULL;
- }
- void connectInputStreams(IRoxieSlaveContext *ctx, bool consumerOrdered)
- {
- inputStream.set(connectSingleStream(ctx, input, sourceIdx, junction, consumerOrdered));
- }
- virtual IOutputMetaData * queryOutputMeta() const
- {
- return input->queryOutputMeta();
- }
- virtual unsigned __int64 queryTotalCycles() const
- {
- return input->queryTotalCycles();
- }
- virtual IEngineRowStream *queryConcreteOutputStream(unsigned whichInput) { assertex(whichInput==0); return this; }
- virtual IStrandJunction *queryConcreteOutputJunction(unsigned idx) const { assertex(idx==0); return nullptr; }
- virtual IRoxieServerActivity *queryActivity()
- {
- return input->queryActivity();
- }
- virtual IIndexReadActivityInfo *queryIndexReadActivity()
- {
- return input->queryIndexReadActivity();
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CriticalBlock procedure(cs);
- input->start(parentExtractSize, parentExtract, paused);
- startJunction(junction);
- }
- virtual void stop()
- {
- CriticalBlock procedure(cs);
- inputStream->stop();
- }
- virtual void reset()
- {
- CriticalBlock procedure(cs);
- input->reset();
- resetJunction(junction);
- }
- virtual void resetEOF()
- {
- CriticalBlock procedure(cs);
- inputStream->resetEOF();
- }
- virtual const void *nextRow()
- {
- CriticalBlock procedure(cs);
- return inputStream->nextRow();
- }
- virtual bool nextGroup(ConstPointerArray & group)
- {
- CriticalBlock procedure(cs);
- return inputStream->nextGroup(group);
- }
- private:
- CriticalSection cs;
- Linked<IFinalRoxieInput> input;
- Linked<IEngineRowStream> inputStream;
- Owned<IStrandJunction> junction;
- unsigned sourceIdx;
- };
- //=================================================================================
- class CPseudoRoxieInput : implements IEngineRowStream, implements IFinalRoxieInput, public CInterface
- {
- public:
- IMPLEMENT_IINTERFACE;
-
- CPseudoRoxieInput()
- {
- }
- virtual unsigned __int64 queryTotalCycles() const
- {
- return 0;
- }
- virtual IEngineRowStream *queryConcreteOutputStream(unsigned whichInput) { assertex(whichInput==0); return this; }
- virtual IStrandJunction *queryConcreteOutputJunction(unsigned idx) const { assertex(idx==0); return nullptr; }
- virtual IRoxieServerActivity *queryActivity()
- {
- throwUnexpected();
- }
- virtual IIndexReadActivityInfo *queryIndexReadActivity()
- {
- throwUnexpected();
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks) override
- {
- assertex(!idx);
- streams.append(this);
- return NULL;
- }
- virtual IOutputMetaData * queryOutputMeta() const { throwUnexpected(); }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused) { }
- virtual void stop() { }
- virtual void reset() { }
- virtual void checkAbort() { }
- virtual void resetEOF() { }
- };
- class CIndirectRoxieInput : public CPseudoRoxieInput
- {
- public:
- CIndirectRoxieInput() : input(NULL)
- {
- stream = NULL;
- sourceIdx = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- input->start(parentExtractSize, parentExtract, paused);
- startJunction(junction);
- }
- virtual void stop()
- {
- stream->stop();
- }
- virtual void reset()
- {
- input->reset();
- resetJunction(junction);
- }
- virtual const void * nextRow()
- {
- return stream->nextRow();
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- return stream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- }
- virtual unsigned __int64 queryTotalCycles() const
- {
- return input->queryTotalCycles();
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- stream->resetEOF();
- }
- virtual unsigned numConcreteOutputs() const
- {
- return input->numConcreteOutputs();
- }
-
- virtual IFinalRoxieInput * queryConcreteInput(unsigned idx)
- {
- return input->queryConcreteInput(idx);
- }
- virtual IEngineRowStream *queryConcreteOutputStream(unsigned whichInput)
- {
- return input->queryConcreteOutputStream(whichInput);
- }
- virtual IStrandJunction *queryConcreteOutputJunction(unsigned idx) const
- {
- return input->queryConcreteOutputJunction(idx);
- }
- virtual IOutputMetaData * queryOutputMeta() const
- {
- return input->queryOutputMeta();
- }
- virtual IRoxieServerActivity *queryActivity()
- {
- return input->queryActivity();
- }
- inline void setInput(unsigned _sourceIdx, IFinalRoxieInput * _input)
- {
- input = _input;
- sourceIdx = _sourceIdx;
- }
- inline void connectInputStreams(IRoxieSlaveContext *ctx, bool consumerOrdered)
- {
- stream = connectSingleStream(ctx, input, sourceIdx, junction, consumerOrdered);
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks) override
- {
- assertex(idx == 0);
- connectInputStreams(ctx, consumerOrdered);
- streams.append(this);
- return NULL;
- }
- inline IFinalRoxieInput *queryInput() const
- {
- return input;
- }
- protected:
- unsigned sourceIdx;
- IFinalRoxieInput *input;
- IEngineRowStream *stream;
- Owned<IStrandJunction> junction;
- };
- class CExtractMapperInput : public CIndirectRoxieInput
- {
- unsigned savedParentExtractSize;
- const byte * savedParentExtract;
- public:
- CExtractMapperInput()
- {
- savedParentExtractSize = 0;
- savedParentExtract = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CIndirectRoxieInput::start(savedParentExtractSize, savedParentExtract, paused);
- }
- void setParentExtract(unsigned _savedParentExtractSize, const byte * _savedParentExtract)
- {
- savedParentExtractSize = _savedParentExtractSize;
- savedParentExtract = _savedParentExtract;
- }
- };
- class CGraphResult : implements IGraphResult, public CInterface
- {
- CriticalSection cs;
- byte **rowset;
- size32_t count;
- bool complete;
- void clear()
- {
- CriticalBlock func(cs);
- rtlReleaseRowset(count, rowset);
- rowset = NULL;
- count = 0;
- complete = false;
- }
- public:
- IMPLEMENT_IINTERFACE
- CGraphResult()
- {
- complete = false; // dummy result is not supposed to be used...
- rowset = NULL;
- count = 0;
- }
- CGraphResult(size32_t _count, byte **_rowset)
- : count(_count), rowset(_rowset)
- {
- complete = true;
- }
- ~CGraphResult()
- {
- clear();
- }
- // interface IGraphResult
- virtual IEngineRowStream * createIterator()
- {
- if (!complete)
- throw MakeStringException(ROXIE_GRAPH_PROCESSING_ERROR, "Internal Error: Reading uninitialised graph result");
- return new CGraphResultIterator(this);
- }
- virtual void getLinkedResult(unsigned & countResult, byte * * & result)
- {
- if (!complete)
- throw MakeStringException(ROXIE_GRAPH_PROCESSING_ERROR, "Internal Error: Reading uninitialised graph result");
- result = rtlLinkRowset(rowset);
- countResult = count;
- }
- virtual const void * getLinkedRowResult()
- {
- if (!complete)
- throw MakeStringException(ROXIE_GRAPH_PROCESSING_ERROR, "Internal Error: Reading uninitialised graph result");
- if (count != 1)
- throw MakeStringException(ROXIE_GRAPH_PROCESSING_ERROR, "Internal Error: Expected a single row result");
- const void * ret = rowset[0];
- LinkRoxieRow(ret);
- return ret;
- }
- //other
- const void * getRow(unsigned i)
- {
- CriticalBlock func(cs);
- if (i >= count)
- return NULL;
- const void * ret = rowset[i];
- if (ret) LinkRoxieRow(ret);
- return ret;
- }
- protected:
- class CGraphResultIterator : public CInterfaceOf<IEngineRowStream>
- {
- unsigned i;
- Linked<CGraphResult> result;
- public:
- CGraphResultIterator(CGraphResult * _result) : result(_result) { i = 0; }
- public:
- virtual const void * nextRow()
- {
- return result->getRow(i++);
- }
- virtual void stop()
- {
- }
- virtual void resetEOF()
- {
- throwUnexpected();
- }
- };
- };
- //=================================================================================
- class CRoxieServerLocalResultReadActivity : public CRoxieServerActivity
- {
- IHThorLocalResultReadArg &helper;
- CriticalSection iterCrit;
- Owned<IEngineRowStream> iter;
- ILocalGraphEx * graph;
- unsigned graphId;
- unsigned sequence;
- public:
- CRoxieServerLocalResultReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _graphId)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorLocalResultReadArg &)basehelper), graphId(_graphId)
- {
- graph = NULL;
- sequence = 0;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- graph = static_cast<ILocalGraphEx *>(ctx->queryCodeContext()->resolveLocalQuery(graphId));
- sequence = helper.querySequence();
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- iter.setown(graph->createResultIterator(sequence));
- }
- virtual void reset()
- {
- {
- CriticalBlock b(iterCrit);
- iter.clear();
- }
- CRoxieServerActivity::reset();
- };
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- Linked<IEngineRowStream> useIter;
- {
- CriticalBlock b(iterCrit);
- if (!iter)
- return NULL;
- useIter.set(iter);
- }
- const void * next = useIter->nextRow();
- if (next)
- {
- processed++;
- atomic_inc(&rowsIn);
- }
- return next;
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() called for source activity");
- }
- };
- class CRoxieServerLocalResultReadActivityFactory : public CRoxieServerActivityFactory
- {
- unsigned graphId;
- public:
- CRoxieServerLocalResultReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _graphId)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), graphId(_graphId)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerLocalResultReadActivity(_ctx, this, _probeManager, graphId);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for LocalResultRead activity");
- }
- };
- IRoxieServerActivityFactory *createRoxieServerLocalResultReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned graphId)
- {
- return new CRoxieServerLocalResultReadActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, graphId);
- }
- //=================================================================================
- class CRoxieServerLocalResultStreamReadActivity : public CRoxieServerActivity
- {
- IHThorLocalResultReadArg &helper;
- unsigned sequence;
- public:
- CRoxieServerLocalResultStreamReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorLocalResultReadArg &)basehelper)
- {
- sequence = 0;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- sequence = helper.querySequence();
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * next = inputStream->nextRow();
- if (next)
- {
- processed++;
- atomic_inc(&rowsIn);
- }
- return next;
- }
- virtual bool querySetStreamInput(unsigned id, unsigned _sourceIdx, IFinalRoxieInput * _input)
- {
- if (id == sequence)
- {
- CRoxieServerActivity::setInput(0, _sourceIdx, _input);
- connectInputStreams(true);
- return true;
- }
- return false;
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() called for source activity");
- }
- };
- class CRoxieServerLocalResultStreamReadActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerLocalResultStreamReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerLocalResultStreamReadActivity(_ctx, this, _probeManager);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for LocalResultRead activity");
- }
- };
- IRoxieServerActivityFactory *createRoxieServerLocalResultStreamReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerLocalResultStreamReadActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- class CRoxieServerLocalResultWriteActivity : public CRoxieServerInternalSinkActivity
- {
- IHThorLocalResultWriteArg &helper;
- ILocalGraphEx * graph;
- unsigned graphId;
- public:
- CRoxieServerLocalResultWriteActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _graphId, unsigned _numOutputs)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, _numOutputs), helper((IHThorLocalResultWriteArg &)basehelper), graphId(_graphId)
- {
- graph = NULL;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerInternalSinkActivity::onCreate(_colocalParent);
- graph = static_cast<ILocalGraphEx *>(ctx->queryCodeContext()->resolveLocalQuery(graphId));
- }
- virtual void onExecute()
- {
- RtlLinkedDatasetBuilder builder(rowAllocator);
- inputStream->readAll(builder);
- Owned<CGraphResult> result = new CGraphResult(builder.getcount(), builder.linkrows());
- graph->setResult(helper.querySequence(), result);
- }
- virtual const void *nextRow()
- {
- return inputStream->nextRow(); // I can act as a passthrough input
- }
- IFinalRoxieInput * querySelectOutput(unsigned id)
- {
- if (id == helper.querySequence())
- {
- executed = true; // Ensure that we don't try to pull as a sink as well as via the passthrough
- return LINK(this);
- }
- return NULL;
- }
- };
- class CRoxieServerLocalResultWriteActivityFactory : public CRoxieServerInternalSinkFactory
- {
- unsigned graphId;
- public:
- CRoxieServerLocalResultWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, unsigned _graphId, bool _isRoot)
- : CRoxieServerInternalSinkFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot), graphId(_graphId)
- {
- isInternal = true;
- Owned<IHThorLocalResultWriteArg> helper = (IHThorLocalResultWriteArg *) helperFactory();
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerLocalResultWriteActivity(_ctx, this, _probeManager, graphId, usageCount);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerLocalResultWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, unsigned _graphId, bool _isRoot)
- {
- return new CRoxieServerLocalResultWriteActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _graphId, _isRoot);
- }
- //=====================================================================================================
- class CRoxieServerDictionaryResultWriteActivity : public CRoxieServerInternalSinkActivity
- {
- IHThorDictionaryResultWriteArg &helper;
- ILocalGraphEx * graph;
- unsigned graphId;
- public:
- CRoxieServerDictionaryResultWriteActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _usageCount, unsigned _graphId)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, _usageCount), helper((IHThorDictionaryResultWriteArg &)basehelper), graphId(_graphId)
- {
- graph = NULL;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerInternalSinkActivity::onCreate(_colocalParent);
- graph = static_cast<ILocalGraphEx *>(ctx->queryCodeContext()->resolveLocalQuery(graphId));
- }
- virtual void onExecute()
- {
- unsigned sequence = helper.querySequence();
- RtlLinkedDictionaryBuilder builder(rowAllocator, helper.queryHashLookupInfo());
- loop
- {
- const void *row = inputStream->ungroupedNextRow();
- if (!row)
- break;
- builder.appendOwn(row);
- processed++;
- }
- Owned<CGraphResult> result = new CGraphResult(builder.getcount(), builder.linkrows());
- graph->setResult(helper.querySequence(), result);
- }
- virtual const void *nextRow()
- {
- return inputStream->nextRow(); // I can act as a passthrough input
- }
- IFinalRoxieInput * querySelectOutput(unsigned id)
- {
- if (id == helper.querySequence())
- {
- executed = true; // Ensure that we don't try to pull as a sink as well as via the passthrough
- return LINK(this);
- }
- return NULL;
- }
- };
- class CRoxieServerDictionaryResultWriteActivityFactory : public CRoxieServerInternalSinkFactory
- {
- unsigned graphId;
- public:
- CRoxieServerDictionaryResultWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, unsigned _graphId, bool _isRoot)
- : CRoxieServerInternalSinkFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot), graphId(_graphId)
- {
- isInternal = true;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerDictionaryResultWriteActivity(_ctx, this, _probeManager, usageCount, graphId);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerDictionaryResultWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, unsigned _graphId, bool _isRoot)
- {
- return new CRoxieServerDictionaryResultWriteActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _graphId, _isRoot);
- }
- //=================================================================================
- class CRoxieServerGraphLoopResultReadActivity : public CRoxieServerActivity
- {
- protected:
- IHThorGraphLoopResultReadArg &helper;
- CriticalSection iterCrit;
- Owned<IFinalRoxieInput> iterInput;
- Owned<IEngineRowStream> iterStream;
- Owned<IStrandJunction> iterJunction;
- ILocalGraphEx * graph;
- unsigned graphId;
- unsigned sequence;
- public:
- CRoxieServerGraphLoopResultReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _graphId)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorGraphLoopResultReadArg &)basehelper), graphId(_graphId)
- {
- graph = NULL;
- sequence = 0;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- graph = static_cast<ILocalGraphEx *>(ctx->queryCodeContext()->resolveLocalQuery(graphId));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if (iterInput)
- iterInput->start(parentExtractSize, parentExtract, paused);
- else
- {
- sequence = helper.querySequence();
- if ((int)sequence >= 0)
- {
- try
- {
- iterStream.setown(graph->createGraphLoopResultIterator(sequence));
- iterInput.clear();
- }
- catch (IException * E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- startJunction(iterJunction);
- }
- virtual void stop()
- {
- if (iterStream)
- iterStream->stop();
- CRoxieServerActivity::stop();
- }
- virtual void reset()
- {
- {
- CriticalBlock b(iterCrit);
- if (iterInput)
- iterInput->reset();
- resetJunction(iterJunction);
- iterInput.clear();
- iterStream.clear();
- iterJunction.clear();
- }
- CRoxieServerActivity::reset();
- };
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- Linked<IEngineRowStream> useStream;
- {
- CriticalBlock b(iterCrit);
- if (!iterStream)
- return NULL;
- useStream.set(iterStream);
- }
- const void * next = useStream->nextRow();
- if (next)
- {
- processed++;
- atomic_inc(&rowsIn);
- }
- return next;
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() called for source activity");
- }
- virtual void gatherIterationUsage(IRoxieServerLoopResultProcessor & processor, unsigned parentExtractSize, const byte * parentExtract)
- {
- ensureCreated();
- basehelper.onStart(parentExtract, NULL);
- processor.noteUseIteration(helper.querySequence());
- }
- virtual void associateIterationOutputs(IRoxieServerLoopResultProcessor & processor, unsigned parentExtractSize, const byte * parentExtract, IProbeManager *probeManager, IArrayOf<IRoxieProbe> &probes)
- {
- //helper already initialised from the gatherIterationUsage() call.
- CRoxieServerActivity::connectInputStreams(true);
- iterInput.set(processor.connectIterationOutput(helper.querySequence(), probeManager, probes, this, 0));
- iterStream.set(connectSingleStream(ctx, iterInput, 0, iterJunction, true));
- }
- virtual IInputSteppingMeta * querySteppingMeta()
- {
- assertex(iterInput);
- return iterInput->querySteppingMeta();
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- assertex(iterStream);
- return iterStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- }
- };
- //variety of CRoxieServerGraphLoopResultReadActivity created internally with a predefined sequence number
- class CRoxieServerInternalGraphLoopResultReadActivity : public CRoxieServerGraphLoopResultReadActivity
- {
- public:
- CRoxieServerInternalGraphLoopResultReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _graphId, unsigned _sequence)
- : CRoxieServerGraphLoopResultReadActivity(_ctx, _factory, _probeManager, _graphId)
- {
- sequence = _sequence;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if ((int)sequence >= 0)
- {
- try
- {
- iterStream.setown(graph->createGraphLoopResultIterator(sequence));
- iterInput.clear();
- }
- catch (IException * E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerGraphLoopResultReadActivityFactory : public CRoxieServerActivityFactory
- {
- unsigned graphId;
- public:
- CRoxieServerGraphLoopResultReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _graphId)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), graphId(_graphId)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerGraphLoopResultReadActivity(_ctx, this, _probeManager, graphId);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for GraphLoopResultRead activity");
- }
- };
- IRoxieServerActivityFactory *createRoxieServerGraphLoopResultReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned graphId)
- {
- return new CRoxieServerGraphLoopResultReadActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, graphId);
- }
- //=====================================================================================================
- class CRoxieServerGraphLoopResultWriteActivity : public CRoxieServerInternalSinkActivity
- {
- IHThorGraphLoopResultWriteArg &helper;
- ILocalGraphEx * graph;
- unsigned graphId;
- public:
- CRoxieServerGraphLoopResultWriteActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _graphId, unsigned _numOutputs)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, _numOutputs), helper((IHThorGraphLoopResultWriteArg &)basehelper), graphId(_graphId)
- {
- graph = NULL;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerInternalSinkActivity::onCreate(_colocalParent);
- graph = static_cast<ILocalGraphEx *>(ctx->queryCodeContext()->resolveLocalQuery(graphId));
- }
- virtual void onExecute()
- {
- RtlLinkedDatasetBuilder builder(rowAllocator);
- inputStream->readAll(builder);
- Owned<CGraphResult> result = new CGraphResult(builder.getcount(), builder.linkrows());
- graph->setGraphLoopResult(result);
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- if (idx==0)
- return this;
- else
- return NULL;
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- inputStream->resetEOF();
- }
- virtual const void *nextRow()
- {
- const void * next = inputStream->nextRow();
- if (next)
- processed++;
- return next;
- }
- virtual bool isPassThrough()
- {
- return true;
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * next = inputStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- if (next)
- processed++;
- return next;
- }
- IInputSteppingMeta * querySteppingMeta()
- {
- return input->querySteppingMeta();
- }
- virtual IOutputMetaData * queryOutputMeta() const
- {
- return input->queryOutputMeta();
- }
- };
- class CRoxieServerGraphLoopResultWriteActivityFactory : public CRoxieServerInternalSinkFactory
- {
- unsigned graphId;
- public:
- CRoxieServerGraphLoopResultWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, unsigned _graphId)
- : CRoxieServerInternalSinkFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, true), graphId(_graphId)
- {
- isInternal = true;
- Owned<IHThorGraphLoopResultWriteArg> helper = (IHThorGraphLoopResultWriteArg *) helperFactory();
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerGraphLoopResultWriteActivity(_ctx, this, _probeManager, graphId, usageCount);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerGraphLoopResultWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, unsigned _graphId)
- {
- return new CRoxieServerGraphLoopResultWriteActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _graphId);
- }
- //=================================================================================
- class CRoxieServerDedupActivity : public CRoxieServerActivity
- {
- protected:
- IHThorDedupArg &helper;
- unsigned numKept;
- unsigned numToKeep;
- public:
- CRoxieServerDedupActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorDedupArg &)basehelper)
- {
- numKept = 0;
- numToKeep = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- numKept = 0;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- numToKeep = helper.numToKeep();
- }
- };
- class CRoxieServerDedupKeepLeftActivity : public CRoxieServerDedupActivity
- {
- IRangeCompare * stepCompare;
- const void *prev;
- public:
- CRoxieServerDedupKeepLeftActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerDedupActivity(_ctx, _factory, _probeManager)
- {
- prev = NULL;
- stepCompare = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- prev = NULL;
- CRoxieServerDedupActivity::start(parentExtractSize, parentExtract, paused);
- IInputSteppingMeta * stepMeta = input->querySteppingMeta();
- stepCompare = NULL;
- if (stepMeta)
- stepCompare = stepMeta->queryCompare();
- }
- virtual void reset()
- {
- ReleaseClearRoxieRow(prev);
- CRoxieServerDedupActivity::reset();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * next;
- loop
- {
- next = inputStream->nextRow();
- if (!prev || !next || !helper.matches(prev,next))
- {
- numKept = 0;
- break;
- }
- if (numKept < numToKeep-1)
- {
- numKept++;
- break;
- }
- ReleaseRoxieRow(next);
- }
- ReleaseRoxieRow(prev);
- prev = next;
- if (next)
- {
- LinkRoxieRow(next);
- processed++;
- }
- return next;
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * next;
- loop
- {
- next = inputStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- //If the record was an in-exact match from the index then return it immediately
- //and don't cause it to dedup following legal records.
- if (!wasCompleteMatch)
- {
- assertex(stepExtra.returnMismatches());
- return next;
- }
- if (!prev || !next || !helper.matches(prev,next))
- {
- numKept = 0;
- break;
- }
- if (numKept < numToKeep-1)
- {
- numKept++;
- break;
- }
- //Unusual - deduping by x,y stepped on x,y,z - still want any record back as soon as possible.
- if (stepExtra.returnMismatches())
- {
- //If asked to return mismatches we are only interested in mismatches that will force the stepped
- //condition to advance
- if (stepCompare->docompare(next, seek, numFields) != 0)
- {
- wasCompleteMatch = false;
- break;
- }
- }
- ReleaseRoxieRow(next);
- }
- ReleaseRoxieRow(prev);
- prev = next;
- if (next)
- {
- LinkRoxieRow(next);
- processed++;
- }
- return next;
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- inputStream->resetEOF();
- }
- IInputSteppingMeta * querySteppingMeta()
- {
- return input->querySteppingMeta();
- }
- };
- //=================================================================================
- class CRoxieServerDedupKeepRightActivity : public CRoxieServerDedupActivity
- {
- const void *kept;
- bool first;
- public:
- CRoxieServerDedupKeepRightActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerDedupActivity(_ctx, _factory, _probeManager)
- {
- kept = NULL;
- first = true;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- first = true;
- kept = NULL;
- CRoxieServerDedupActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- ReleaseClearRoxieRow(kept);
- CRoxieServerActivity::reset();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (first)
- {
- kept = inputStream->nextRow();
- first = false;
- }
- const void * next;
- loop
- {
- next = inputStream->nextRow();
- if (!kept || !next || !helper.matches(kept,next))
- {
- numKept = 0;
- break;
- }
- if (numKept < numToKeep-1)
- {
- numKept++;
- break;
- }
- ReleaseRoxieRow(kept);
- kept = next;
- }
- const void * ret = kept;
- kept = next;
- // CTXLOG("dedup returns %p", ret);
- if (ret) processed++;
- return ret;
- }
- };
- class CRoxieServerDedupAllActivity : public CRoxieServerActivity
- {
- IHThorDedupArg &helper;
- unsigned survivorIndex;
- ConstPointerArray survivors;
- bool keepLeft;
- bool eof;
- bool first;
- ICompare *primaryCompare;
- public:
- CRoxieServerDedupAllActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, bool _keepLeft)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorDedupArg &)basehelper)
- {
- keepLeft = _keepLeft;
- primaryCompare = helper.queryComparePrimary();
- eof = false;
- first = true;
- survivorIndex = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- first = true;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- #ifdef _DEBUG
- while (survivors.isItem(survivorIndex))
- {
- ReleaseRoxieRow(survivors.item(survivorIndex++));
- }
- #endif
- survivors.kill();
- eof = false;
- first = true;
- CRoxieServerActivity::reset();
- }
- void dedupRange(unsigned first, unsigned last, ConstPointerArray & group)
- {
- for (unsigned idxL = first; idxL < last; idxL++)
- {
- const void * left = group.item(idxL);
- if (left)
- {
- for (unsigned idxR = first; idxR < last; idxR++)
- {
- const void * right = group.item(idxR);
- if ((idxL != idxR) && right)
- {
- if (helper.matches(left, right))
- {
- if (keepLeft)
- {
- group.replace(NULL, idxR);
- ReleaseRoxieRow(right);
- }
- else
- {
- group.replace(NULL, idxL);
- ReleaseRoxieRow(left);
- break;
- }
- }
- }
- }
- }
- }
- }
- void dedupRange(unsigned first, unsigned last, void ** rows)
- {
- for (unsigned idxL = first; idxL < last; idxL++)
- {
- void * left = rows[idxL];
- if (left)
- {
- for (unsigned idxR = first; idxR < last; idxR++)
- {
- void * right = rows[idxR];
- if ((idxL != idxR) && right)
- {
- if (helper.matches(left, right))
- {
- if (keepLeft)
- {
- rows[idxR] = NULL;
- ReleaseRoxieRow(right);
- }
- else
- {
- rows[idxL] = NULL;
- ReleaseRoxieRow(left);
- break;
- }
- }
- }
- }
- }
- }
- }
- bool calcNextDedupAll()
- {
- survivors.kill();
- survivorIndex = 0;
- ConstPointerArray group;
- if (eof || !inputStream->nextGroup(group))
- {
- eof = true;
- return false;
- }
- unsigned max = group.ordinality();
- if (primaryCompare)
- {
- //hard, if not impossible, to hit this code once optimisations in place
- MemoryAttr indexbuff(max*sizeof(void *));
- void ** temp = (void **)indexbuff.bufferBase();
- void * *rows = const_cast<void * *>(group.getArray());
- msortvecstableinplace(rows, max, *primaryCompare, temp);
- unsigned first = 0;
- for (unsigned idx = 1; idx < max; idx++)
- {
- if (primaryCompare->docompare(rows[first], rows[idx]) != 0)
- {
- dedupRange(first, idx, rows);
- first = idx;
- }
- }
- dedupRange(first, max, rows);
- for(unsigned idx2=0; idx2<max; ++idx2)
- {
- void * cur = rows[idx2];
- if(cur)
- survivors.append(cur);
- }
- }
- else
- {
- dedupRange(0, max, group);
- for(unsigned idx=0; idx<max; ++idx)
- {
- const void * cur = group.item(idx);
- if(cur)
- survivors.append(cur);
- }
- }
- return true;
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (first)
- {
- calcNextDedupAll();
- first = false;
- }
- while (survivors.isItem(survivorIndex))
- {
- const void *ret = survivors.item(survivorIndex++);
- if (ret)
- {
- processed++;
- return ret;
- }
- }
- calcNextDedupAll();
- return NULL;
- }
- };
- class CRoxieServerDedupActivityFactory : public CRoxieServerActivityFactory
- {
- bool compareAll;
- bool keepLeft;
- public:
- CRoxieServerDedupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- Owned<IHThorDedupArg> helper = (IHThorDedupArg *) helperFactory();
- compareAll = helper->compareAll();
- keepLeft = helper->keepLeft();
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- if (compareAll)
- return new CRoxieServerDedupAllActivity(_ctx, this, _probeManager, keepLeft);
- else if (keepLeft)
- return new CRoxieServerDedupKeepLeftActivity(_ctx, this, _probeManager);
- else
- return new CRoxieServerDedupKeepRightActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerDedupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerDedupActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerHashDedupActivity : public CRoxieServerActivity
- {
- bool eof;
- IHThorHashDedupArg &helper;
- class HashDedupElement
- {
- public:
- HashDedupElement(unsigned _hash, const void *_keyRow)
- : hash(_hash), keyRow(_keyRow)
- {
- }
- ~HashDedupElement()
- {
- ReleaseRoxieRow(keyRow);
- }
- inline unsigned queryHash() const
- {
- return hash;
- }
- inline const void *queryRow() const
- {
- return keyRow;
- }
- private:
- unsigned hash;
- const void *keyRow;
- };
- class HashDedupTable : public SuperHashTable
- {
- public:
- HashDedupTable(IHThorHashDedupArg & _helper, CRoxieServerHashDedupActivity & _activity)
- : helper(_helper),
- activity(_activity),
- keySize(helper.queryKeySize())
- {
- }
- virtual ~HashDedupTable() { _releaseAll(); }
- virtual unsigned getHashFromElement(const void *et) const
- {
- const HashDedupElement *element = reinterpret_cast<const HashDedupElement *>(et);
- return element->queryHash();
- }
- virtual unsigned getHashFromFindParam(const void *fp) const { throwUnexpected(); }
- virtual const void * getFindParam(const void *et) const { throwUnexpected(); }
- virtual bool matchesElement(const void *et, const void *searchET) const { throwUnexpected(); }
- virtual bool matchesFindParam(const void *et, const void *key, unsigned fphash) const
- {
- const HashDedupElement *element = reinterpret_cast<const HashDedupElement *>(et);
- if (fphash != element->queryHash())
- return false;
- return (helper.queryKeyCompare()->docompare(element->queryRow(), key) == 0);
- }
- virtual void onAdd(void *et) {}
- virtual void onRemove(void *et)
- {
- const HashDedupElement *element = reinterpret_cast<const HashDedupElement *>(et);
- delete element;
- }
- void onCreate(IRoxieSlaveContext *ctx)
- {
- keyRowAllocator.setown(activity.createRowAllocator(keySize.queryOriginal()));
- }
- void reset()
- {
- kill();
- }
- bool insert(const void * row)
- {
- unsigned hash = helper.queryHash()->hash(row);
- RtlDynamicRowBuilder keyRowBuilder(keyRowAllocator, true);
- size32_t thisKeySize = helper.recordToKey(keyRowBuilder, row);
- OwnedConstRoxieRow keyRow = keyRowBuilder.finalizeRowClear(thisKeySize);
- if (find(hash, keyRow.get()))
- return false;
- addNew(new HashDedupElement(hash, keyRow.getClear()), hash);
- return true;
- }
- private:
- IHThorHashDedupArg & helper;
- CachedOutputMetaData keySize;
- Owned<IEngineRowAllocator> keyRowAllocator;
- CRoxieServerHashDedupActivity & activity;
- } table;
- public:
- CRoxieServerHashDedupActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorHashDedupArg &)basehelper), table(helper, *this)
- {
- eof = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(colocalParent);
- table.onCreate(ctx);
- }
- virtual void reset()
- {
- table.reset();
- eof = false;
- CRoxieServerActivity::reset();
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- while(!eof)
- {
- const void * next = inputStream->nextRow();
- if(!next)
- {
- if (table.count() == 0)
- eof = true;
- table.reset();
- return NULL;
- }
- if(table.insert(next))
- return next;
- else
- ReleaseRoxieRow(next);
- }
- return NULL;
- }
- };
- class CRoxieServerHashDedupActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerHashDedupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerHashDedupActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerHashDedupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerHashDedupActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerRollupActivity : public CRoxieServerActivity
- {
- IHThorRollupArg &helper;
- OwnedConstRoxieRow left;
- OwnedConstRoxieRow prev;
- OwnedConstRoxieRow right;
- bool readFirstRow;
- public:
- CRoxieServerRollupActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorRollupArg &)basehelper)
- {
- readFirstRow = false;
- }
- ~CRoxieServerRollupActivity()
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- readFirstRow = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- left.clear();
- prev.clear();
- right.clear();
- CRoxieServerActivity::reset();
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (!readFirstRow)
- {
- left.setown(inputStream->nextRow());
- prev.set(left);
- readFirstRow = true;
- }
- loop
- {
- right.setown(inputStream->nextRow());
- if(!prev || !right || !helper.matches(prev,right))
- {
- const void * ret = left.getClear();
- if(ret)
- processed++;
- left.setown(right.getClear());
- prev.set(left);
- return ret;
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned outSize = helper.transform(rowBuilder, left, right);
- if (outSize)
- left.setown(rowBuilder.finalizeRowClear(outSize));
- if (helper.getFlags() & RFrolledismatchleft)
- prev.set(left);
- else
- prev.set(right);
- }
- catch(IException * E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerRollupActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerRollupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerRollupActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerRollupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerRollupActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerNormalizeActivity : public CRoxieServerActivity
- {
- IHThorNormalizeArg &helper;
- unsigned numThisRow;
- unsigned curRow;
- const void *buffer;
- unsigned numProcessedLastGroup;
- public:
- CRoxieServerNormalizeActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorNormalizeArg &)basehelper)
- {
- buffer = NULL;
- numThisRow = 0;
- curRow = 0;
- numProcessedLastGroup = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- numThisRow = 0;
- curRow = 0;
- numProcessedLastGroup = 0;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- ReleaseClearRoxieRow(buffer);
- CRoxieServerActivity::reset();
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- while (curRow == numThisRow)
- {
- if (buffer)
- ReleaseClearRoxieRow(buffer);
- buffer = inputStream->nextRow();
- if (!buffer && (processed == numProcessedLastGroup))
- buffer = inputStream->nextRow();
- if (!buffer)
- {
- numProcessedLastGroup = processed;
- return NULL;
- }
- curRow = 0;
- numThisRow = helper.numExpandedRows(buffer);
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned actualSize = helper.transform(rowBuilder, buffer, ++curRow);
- if (actualSize != 0)
- {
- processed++;
- return rowBuilder.finalizeRowClear(actualSize);
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerNormalizeActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerNormalizeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerNormalizeActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNormalizeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerNormalizeActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerNormalizeChildActivity : public CRoxieServerActivity
- {
- IHThorNormalizeChildArg &helper;
- unsigned numThisRow;
- unsigned curRow;
- const void *buffer;
- unsigned numProcessedLastGroup;
- INormalizeChildIterator * cursor;
- const void * curChildRow;
- bool advanceInput()
- {
- loop
- {
- ReleaseClearRoxieRow(buffer);
- buffer = inputStream->nextRow();
- if (!buffer && (processed == numProcessedLastGroup))
- buffer = inputStream->nextRow();
- if (!buffer)
- {
- numProcessedLastGroup = processed;
- return false;
- }
- curChildRow = cursor->first(buffer);
- if (curChildRow)
- {
- curRow = 0;
- return true;
- }
- }
- }
- public:
- CRoxieServerNormalizeChildActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorNormalizeChildArg &)basehelper)
- {
- buffer = NULL;
- cursor = NULL;
- numThisRow = 0;
- curRow = 0;
- numProcessedLastGroup = 0;
- curChildRow = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- numThisRow = 0;
- curRow = 0;
- numProcessedLastGroup = 0;
- curChildRow = NULL;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- cursor = helper.queryIterator();
- }
- virtual void stop()
- {
- CRoxieServerActivity::stop();
- }
- virtual void reset()
- {
- cursor = NULL;
- ReleaseClearRoxieRow(buffer);
- CRoxieServerActivity::reset();
- }
- virtual bool needsAllocator() const { return true; }
- const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- if (!buffer)
- {
- if (!advanceInput())
- return NULL;
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize = helper.transform(rowBuilder, buffer, curChildRow, ++curRow);
- curChildRow = cursor->next();
- if (!curChildRow)
- ReleaseClearRoxieRow(buffer);
- if (outSize != 0)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerNormalizeChildActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerNormalizeChildActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerNormalizeChildActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNormalizeChildActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerNormalizeChildActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerNormalizeLinkedChildActivity : public CRoxieServerActivity
- {
- IHThorNormalizeLinkedChildArg &helper;
- OwnedConstRoxieRow curParent;
- OwnedConstRoxieRow curChild;
- unsigned numProcessedLastGroup;
- bool advanceInput()
- {
- loop
- {
- curParent.setown(inputStream->nextRow());
- if (!curParent && (processed == numProcessedLastGroup))
- curParent.setown(inputStream->nextRow());
- if (!curParent)
- {
- numProcessedLastGroup = processed;
- return false;
- }
- curChild.set(helper.first(curParent));
- if (curChild)
- return true;
- }
- }
- public:
- CRoxieServerNormalizeLinkedChildActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorNormalizeLinkedChildArg &)basehelper)
- {
- numProcessedLastGroup = 0;
- }
- ~CRoxieServerNormalizeLinkedChildActivity()
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- numProcessedLastGroup = 0;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- curParent.clear();
- curChild.clear();
- CRoxieServerActivity::reset();
- }
- const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- if (!curParent)
- {
- if (!advanceInput())
- return NULL;
- }
- try
- {
- const void *ret = curChild.getClear();
- curChild.set(helper.next());
- if (!curChild)
- curParent.clear();
- if (ret)
- {
- processed++;
- return ret;
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerNormalizeLinkedChildActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerNormalizeLinkedChildActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerNormalizeLinkedChildActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNormalizeLinkedChildActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerNormalizeLinkedChildActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerSortActivity : public CRoxieServerActivity
- {
- protected:
- IHThorSortArg &helper;
- ICompare *compare;
- Owned<ISortAlgorithm> sorter;
- bool readInput;
- RoxieSortAlgorithm sortAlgorithm;
- unsigned sortFlags;
- public:
- CRoxieServerSortActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, RoxieSortAlgorithm _sortAlgorithm, unsigned _sortFlags)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorSortArg &)basehelper), sortAlgorithm(_sortAlgorithm), sortFlags(_sortFlags)
- {
- compare = helper.queryCompare();
- readInput = false;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- if (sortAlgorithm==unknownSortAlgorithm)
- sorter.clear();
- else
- sorter.setown(createSortAlgorithm(sortAlgorithm, compare, ctx->queryRowManager(), meta, ctx->queryCodeContext(), tempDirectory, activityId));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- assertex(!readInput);
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- if (sorter)
- sorter->reset();
- readInput = false;
- CRoxieServerActivity::reset();
- }
- static RoxieSortAlgorithm useAlgorithm(const char *algorithmName, unsigned sortFlags)
- {
- RoxieSortAlgorithm sortAlgorithm = unknownSortAlgorithm;
- if (!algorithmName)
- algorithmName = "";
- if (stricmp(algorithmName, "quicksort")==0)
- {
- switch (sortFlags & (TAFstable|TAFspill))
- {
- case 0: sortAlgorithm = quickSortAlgorithm; break;
- case TAFstable: sortAlgorithm = stableQuickSortAlgorithm; break;
- case TAFspill: sortAlgorithm = spillingQuickSortAlgorithm; break;
- case TAFstable|TAFspill: sortAlgorithm = stableSpillingQuickSortAlgorithm; break;
- }
- }
- else if (stricmp(algorithmName, "parquicksort")==0)
- {
- switch (sortFlags & TAFstable)
- {
- case 0: sortAlgorithm = parallelQuickSortAlgorithm; break;
- case TAFstable: sortAlgorithm = parallelStableQuickSortAlgorithm; break;
- default: throwUnexpected();
- }
- }
- else if (stricmp(algorithmName, "heapsort")==0)
- sortAlgorithm = heapSortAlgorithm; // NOTE - we do allow UNSTABLE('heapsort') in order to facilitate runtime selection. Also explicit selection of heapsort overrides request to spill
- else if (stricmp(algorithmName, "mergesort")==0)
- sortAlgorithm = (sortFlags & TAFspill) ? spillingMergeSortAlgorithm : mergeSortAlgorithm;
- else if (stricmp(algorithmName, "parmergesort")==0)
- sortAlgorithm = (sortFlags & TAFspill) ? spillingParallelMergeSortAlgorithm : parallelMergeSortAlgorithm;
- else if (stricmp(algorithmName, "tbbqsort")==0)
- sortAlgorithm = tbbQuickSortAlgorithm;
- else if (stricmp(algorithmName, "tbbstableqsort")==0)
- sortAlgorithm = tbbStableQuickSortAlgorithm;
- else
- {
- if (*algorithmName)
- WARNLOG(ROXIE_UNKNOWN_ALGORITHM, "Ignoring unsupported sort order algorithm '%s', using default", algorithmName);
- if (sortFlags & TAFspill)
- sortAlgorithm = ((sortFlags & TAFunstable) != 0) ? spillingQuickSortAlgorithm : stableSpillingQuickSortAlgorithm;
- else if (sortFlags & TAFunstable)
- sortAlgorithm = quickSortAlgorithm;
- else
- sortAlgorithm = heapSortAlgorithm;
- }
- return sortAlgorithm;
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (!readInput)
- {
- if (sortAlgorithm == unknownSortAlgorithm)
- {
- sorter.clear();
- IHThorAlgorithm *sortMethod = static_cast<IHThorAlgorithm *>(helper.selectInterface(TAIalgorithm_1));
- OwnedRoxieString algorithmName(sortMethod->getAlgorithm());
- sortAlgorithm = useAlgorithm(algorithmName, sortFlags);
- sorter.setown(createSortAlgorithm(sortAlgorithm, compare, ctx->queryRowManager(), meta, ctx->queryCodeContext(), tempDirectory, activityId));
- }
- sorter->prepare(inputStream);
- noteStatistic(StTimeSortElapsed, cycle_to_nanosec(sorter->getElapsedCycles(true)));
- readInput = true;
- }
- const void *ret = sorter->next();
- if (ret)
- processed++;
- else
- {
- sorter->reset();
- readInput = false; // ready for next group
- }
- return ret;
- }
- };
- class CRoxieServerSortActivityFactory : public CRoxieServerActivityFactory
- {
- RoxieSortAlgorithm sortAlgorithm;
- unsigned sortFlags;
- public:
- CRoxieServerSortActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- Owned<IHThorSortArg> sortHelper = (IHThorSortArg *) helperFactory();
- const char *algorithmName = NULL;
- IHThorAlgorithm *sortMethod = static_cast<IHThorAlgorithm *>(sortHelper->selectInterface(TAIalgorithm_1));
- if (sortMethod)
- {
- sortFlags = sortMethod->getAlgorithmFlags();
- if (sortFlags & TAFconstant)
- algorithmName = sortMethod->getAlgorithm();
- }
- else
- sortFlags = TAFstable|TAFconstant;
- bool forceSpill = _queryFactory.queryOptions().allSortsMaySpill || _graphNode.getPropBool("hint[@name='spill']/@value", false);
- if (forceSpill)
- sortFlags |= TAFspill;
- if (!(sortFlags & TAFunstable))
- sortFlags |= TAFstable; // Assume stable unless specifically requested otherwise
- if (sortFlags & TAFconstant)
- sortAlgorithm = CRoxieServerSortActivity::useAlgorithm(algorithmName, sortFlags);
- else
- sortAlgorithm = unknownSortAlgorithm;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSortActivity(_ctx, this, _probeManager, sortAlgorithm, sortFlags);
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return sortStatistics;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSortActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerSortActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- static int compareUint64(const void * _left, const void * _right)
- {
- const unsigned __int64 left = *static_cast<const unsigned __int64 *>(_left);
- const unsigned __int64 right = *static_cast<const unsigned __int64 *>(_right);
- if (left < right)
- return -1;
- if (left > right)
- return -1;
- return 0;
- }
- class CRoxieServerQuantileActivity : public CRoxieServerActivity
- {
- protected:
- Owned<ISortAlgorithm> sorter;
- IHThorQuantileArg &helper;
- ConstPointerArray sorted;
- ICompare *compare;
- unsigned flags;
- double skew;
- unsigned __int64 numDivisions;
- bool rangeIsAll;
- size32_t rangeSize;
- rtlDataAttr rangeValues;
- bool calculated;
- bool processedAny;
- bool anyThisGroup;
- bool eof;
- unsigned curQuantile;
- unsigned curIndex;
- unsigned curIndexExtra;
- unsigned skipSize;
- unsigned skipExtra;
- unsigned prevIndex;
- public:
- CRoxieServerQuantileActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _flags)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorQuantileArg &)basehelper), flags(_flags)
- {
- compare = helper.queryCompare();
- skew = 0.0;
- numDivisions = 0;
- calculated = false;
- processedAny = false;
- anyThisGroup = false;
- eof = false;
- rangeIsAll = true;
- rangeSize = 0;
- curQuantile = 0;
- prevIndex = 0;
- curIndex = 0;
- curIndexExtra = 0;
- skipSize = 0;
- skipExtra = 0;
- if (flags & TQFunstable)
- sorter.setown(createQuickSortAlgorithm(compare));
- else
- sorter.setown(createMergeSortAlgorithm(compare));
- }
- virtual void reset()
- {
- sorter->reset();
- calculated = false;
- processedAny = false;
- anyThisGroup = false;
- CRoxieServerActivity::reset();
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- skew = helper.getSkew();
- numDivisions = helper.getNumDivisions();
- //Check for -ve integer values and treat as out of range
- if ((__int64)numDivisions < 1)
- numDivisions = 1;
- if (flags & TQFhasrange)
- helper.getRange(rangeIsAll, rangeSize, rangeValues.refdata());
- else
- {
- rangeIsAll = true;
- rangeSize = 0;
- }
- if (rangeSize)
- {
- //Sort the range items into order to allow quick comparison
- qsort(rangeValues.getdata(), rangeSize / sizeof(unsigned __int64), sizeof(unsigned __int64), compareUint64);
- }
- calculated = false;
- processedAny = false;
- anyThisGroup = false;
- eof = false;
- curQuantile = 0;
- prevIndex = 0;
- curIndex = 0;
- curIndexExtra = 0;
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- const void * ret = NULL;
- for(;;)
- {
- if (!calculated)
- {
- if (flags & TQFlocalsorted)
- {
- for (;;)
- {
- const void * next = inputStream->nextRow();
- if (!next)
- break;
- sorted.append(next);
- }
- }
- else
- {
- sorter->prepare(inputStream);
- sorter->getSortedGroup(sorted);
- }
- if (sorted.ordinality() == 0)
- {
- if (processedAny)
- {
- eof = true;
- return NULL;
- }
- //Unusual case 0 rows - add a default row instead
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t thisSize = helper.createDefault(rowBuilder);
- sorted.append(rowBuilder.finalizeRowClear(thisSize));
- }
- calculated = true;
- processedAny = true;
- anyThisGroup = false;
- curQuantile = 0;
- curIndex = 0;
- curIndexExtra = (numDivisions-1) / 2; // to ensure correctly rounded up
- prevIndex = curIndex-1; // Ensure it doesn't match
- skipSize = (sorted.ordinality() / numDivisions);
- skipExtra = (sorted.ordinality() % numDivisions);
- }
- if (isQuantileIncluded(curQuantile))
- {
- if (!(flags & TQFdedup) || (prevIndex != curIndex))
- {
- const void * lhs = sorted.item(curIndex);
- if (flags & TQFneedtransform)
- {
- unsigned outSize;
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- try
- {
- outSize = helper.transform(rowBuilder, lhs, curQuantile);
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- if (outSize)
- ret = rowBuilder.finalizeRowClear(outSize);
- }
- else
- {
- LinkRoxieRow(lhs);
- ret = lhs;
- }
- prevIndex = curIndex; // even if output was skipped?
- }
- }
- curIndex += skipSize;
- curIndexExtra += skipExtra;
- if (curIndexExtra >= numDivisions)
- {
- curIndex++;
- curIndexExtra -= numDivisions;
- }
- //Ensure the current index always stays valid.
- if (curIndex >= sorted.ordinality())
- curIndex = sorted.ordinality()-1;
- curQuantile++;
- if (curQuantile > numDivisions)
- {
- sorted.kill();
- sorter->reset();
- calculated = false; // ready for next group
- }
- if (ret)
- {
- anyThisGroup = true;
- processed++;
- return ret;
- }
- if (curQuantile > numDivisions)
- {
- if (anyThisGroup)
- return NULL;
- }
- }
- }
- bool isQuantileIncluded(unsigned quantile) const
- {
- if (quantile == 0)
- return (flags & TQFfirst) != 0;
- if (quantile == numDivisions)
- return (flags & TQFlast) != 0;
- if (rangeIsAll)
- return true;
- //Compare against the list of ranges provided
- //MORE: Since the list is sorted should only need to compare the next (and allow for dups)
- unsigned rangeNum = rangeSize / sizeof(unsigned __int64);
- const unsigned __int64 * ranges = static_cast<const unsigned __int64 *>(rangeValues.getdata());
- for (unsigned i = 0; i < rangeNum; i++)
- {
- if (ranges[i] == quantile)
- return true;
- }
- return false;
- }
- };
- class CRoxieServerQuantileActivityFactory : public CRoxieServerActivityFactory
- {
- unsigned flags;
- public:
- CRoxieServerQuantileActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- Owned<IHThorQuantileArg> quantileHelper = (IHThorQuantileArg *) helperFactory();
- flags = quantileHelper->getFlags();
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerQuantileActivity(_ctx, this, _probeManager, flags);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerQuantileActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerQuantileActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- class CRoxieServerSortedActivity : public CRoxieServerActivity
- {
- IHThorSortedArg &helper;
- ICompare * compare;
- const void *prev;
- IRangeCompare * stepCompare;
- public:
- CRoxieServerSortedActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorSortedArg &)basehelper)
- {
- prev = NULL;
- compare = helper.queryCompare();
- stepCompare = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- IInputSteppingMeta * stepMeta = input->querySteppingMeta();
- if (stepMeta)
- stepCompare = stepMeta->queryCompare();
- prev = NULL;
- }
- virtual void reset()
- {
- ReleaseClearRoxieRow(prev);
- CRoxieServerActivity::reset();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void *ret = inputStream->nextRow();
- if (ret && prev && compare->docompare(prev, ret) > 0)
- {
- // MORE - better to give mismatching rows that indexes?
- throw MakeStringException(ROXIE_NOT_SORTED, "SORTED(%u) detected incorrectly sorted rows (row %d, %d))", activityId, processed, processed+1);
- }
- ReleaseRoxieRow(prev);
- prev = ret;
- if (ret)
- {
- LinkRoxieRow(prev);
- processed++;
- }
- return ret;
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void *ret = inputStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- if (ret && prev && compare->docompare(prev, ret) > 0)
- {
- // MORE - better to give mismatching rows that indexes?
- throw MakeStringException(ROXIE_NOT_SORTED, "SORTED(%u) detected incorrectly sorted rows (row %d, %d))", activityId, processed, processed+1);
- }
- ReleaseRoxieRow(prev);
- prev = ret;
- if (ret)
- {
- LinkRoxieRow(prev);
- processed++;
- }
- return ret;
- }
- IInputSteppingMeta * querySteppingMeta()
- {
- return input->querySteppingMeta();
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- inputStream->resetEOF();
- }
- };
- class CRoxieServerSortedActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerSortedActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSortedActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSortedActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerSortedActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- class CRoxieServerThroughSpillActivity : public CRoxieServerActivity
- {
- /*
- BE VERY CAREFUL - this code is tricky.
- Note that starts and stops (and resets) can occur in strange orders
- The FIRST start OR stop must initialize the activity but only the first START should call the upstream start.
- The last stop should call the upstream stop.
- The first reset should call the upstream reset.
- The calculation of whether a row is needed for other (yet to come) outputs needs to work correctly even if the output in question has
- not yet had start or stop called - for this to happen init() is called on all outputs on the first start or stop.
- Some outputs may be completely pruned away when used in a GRAPH - these outputs should not receive any start/stop/reset and should be
- ignored in the minIndex calculation
- */
- public:
- IHThorArg &helper;
- unsigned activeOutputs;
- unsigned numOutputs;
- unsigned numOriginalOutputs;
- QueueOf<const void, true> buffer;
- CriticalSection crit;
- CriticalSection crit2;
- unsigned tailIdx;
- unsigned headIdx;
- Owned<IException> error;
- class OutputAdaptor : implements IEngineRowStream, implements IFinalRoxieInput, public CInterface
- {
- bool eof, eofpending, stopped;
- public:
- CRoxieServerThroughSpillActivity *parent;
- unsigned idx;
- unsigned oid;
- unsigned processed;
- cycle_t totalCycles; // We track this per output so that the pullers get a meaningful value to use when calculating their localtime
- public:
- IMPLEMENT_IINTERFACE;
- OutputAdaptor()
- {
- parent = NULL;
- oid = 0;
- idx = 0;
- processed = 0;
- totalCycles = 0;
- eofpending = false;
- eof = false;
- stopped = false;
- }
- ~OutputAdaptor()
- {
- if (traceStartStop)
- DBGLOG("%p ~OutputAdaptor %d", this, oid);
- if (processed && parent && parent->factory)
- {
- parent->factory->noteProcessed(oid, processed);
- if (parent->ctx)
- parent->ctx->noteProcessed(parent->querySubgraphId(), parent->activityId, oid, processed, 0);
- }
- }
- void init()
- {
- if (traceStartStop)
- DBGLOG("%p init Input adaptor %d", this, oid);
- idx = 0;
- processed = 0;
- eofpending = false;
- eof = false;
- stopped = false;
- }
- virtual IEngineRowStream *queryConcreteOutputStream(unsigned whichInput) { assertex(idx==0); return this; }
- virtual IStrandJunction *queryConcreteOutputJunction(unsigned idx) const { assertex(idx==0); return nullptr; }
- virtual IRoxieServerActivity *queryActivity()
- {
- return parent;
- }
- virtual IIndexReadActivityInfo *queryIndexReadActivity()
- {
- return parent->queryIndexReadActivity();
- }
- virtual unsigned __int64 queryTotalCycles() const
- {
- return totalCycles;
- }
- virtual const void * nextRow()
- {
- SimpleActivityTimer t(totalCycles, parent->timeActivities);
- if (eof)
- return NULL;
- const void *ret = parent->readBuffered(idx, oid);
- #ifdef TRACE_SPLIT
- parent->CTXLOG("Adaptor %d got back %p for record %d", oid, ret, idx);
- #endif
- if (ret)
- {
- processed++;
- eofpending = false;
- }
- else if (eofpending)
- eof = true;
- else
- eofpending = true;
- return ret;
- }
- virtual IOutputMetaData * queryOutputMeta() const
- {
- return parent->queryOutputMeta();
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- // NOTE: it is tempting to move the init() of all output adaptors here. However that is not a good idea,
- // since adaptors that have not yet started or stopped (but are going to) still need to have been init()'ed
- // for minIndex to give the correct answers
- // therefore, we call init() on all adaptors on receipt of the first start() or stop()
- if (traceStartStop)
- parent->CTXLOG("%p start Input adaptor %d stopped = %d", this, oid, stopped);
- parent->start(oid, parentExtractSize, parentExtract, paused);
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks) override
- {
- parent->connectInputStreams(consumerOrdered);
- streams.append(this);
- return NULL;
- }
- virtual void stop()
- {
- if (traceStartStop)
- parent->CTXLOG("%p stop Input adaptor %d stopped = %d", this, oid, stopped);
- if (!stopped)
- {
- parent->stop(oid, idx); // NOTE - may call init()
- stopped = true; // parent code relies on stop being called exactly once per adaptor, so make sure it is!
- }
- };
- virtual void reset()
- {
- if (traceStartStop)
- parent->CTXLOG("%p reset Input adaptor %d stopped = %d", this, oid, stopped);
- parent->reset(oid);
- idx = 0; // value should not be relevant really but this is the safest...
- stopped = false;
- };
- virtual void resetEOF()
- {
- parent->resetEOF();
- }
- virtual void checkAbort()
- {
- parent->checkAbort();
- }
- } *adaptors;
- bool *used;
- unsigned nextFreeOutput()
- {
- unsigned i = numOutputs;
- while (i)
- {
- i--;
- if (!used[i])
- return i;
- }
- throwUnexpected();
- }
- unsigned minIndex(unsigned exceptOid)
- {
- // MORE - yukky code (and slow). Could keep them heapsorted by idx or something
- // this is trying to determine whether any of the adaptors will in the future read a given record
- unsigned minIdx = (unsigned) -1;
- for (unsigned i = 0; i < numOutputs; i++)
- {
- if (i != exceptOid && used[i] && adaptors[i].idx < minIdx)
- minIdx = adaptors[i].idx;
- }
- return minIdx;
- }
- inline bool isLastTailReader(unsigned exceptOid)
- {
- for (unsigned i = 0; i < numOutputs; i++)
- {
- if (i != exceptOid && adaptors[i].idx == tailIdx && used[i])
- return false;
- }
- return true;
- }
- void initOutputs()
- {
- activeOutputs = numOutputs;
- for (unsigned i = 0; i < numOriginalOutputs; i++)
- if (used[i])
- adaptors[i].init();
- state = STATEstarting;
- }
- public:
- CRoxieServerThroughSpillActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numOutputs)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper(basehelper), numOutputs(_numOutputs)
- {
- numOriginalOutputs = numOutputs;
- adaptors = new OutputAdaptor[numOutputs];
- used = new bool[numOutputs];
- for (unsigned i = 0; i < numOutputs; i++)
- {
- adaptors[i].parent = this;
- adaptors[i].oid = i;
- used[i] = false;
- }
- tailIdx = 0;
- headIdx = 0;
- activeOutputs = numOutputs;
- }
- ~CRoxieServerThroughSpillActivity()
- {
- delete [] adaptors;
- delete [] used;
- }
- const void *readBuffered(unsigned &idx, unsigned oid)
- {
- //False positives are fine, false negatives are not.. so headIdx must only be updated when a row will be available.
- const unsigned curIdx = idx;
- if (curIdx == headIdx) // test once without getting the crit2 sec
- {
- CriticalBlock b2(crit2); // but only one puller gets to read the head
- if (curIdx == headIdx) // test again now that we have it
- {
- ActivityTimer t(totalCycles, timeActivities); // NOTE - time spent waiting for crit not included here. But it will have been included on the totalTime of the person holding the crit, so that is right
- if (error)
- throw error.getLink();
- try
- {
- const void *row = inputStream->nextRow();
- if (activeOutputs==1)
- {
- #ifdef TRACE_SPLIT
- CTXLOG("spill %d optimised return of %p", activityId, row);
- #endif
- headIdx++;
- idx++;
- return row; // optimization for the case where only one output still active.
- }
- CriticalBlock b(crit);
- headIdx++;
- idx++;
- if (activeOutputs==1)
- return row; // optimization for the case where only one output still active.
- buffer.enqueue(row);
- if (row) LinkRoxieRow(row);
- return row;
- }
- catch (IException *E)
- {
- #ifdef TRACE_SPLIT
- CTXLOG("spill %d caught exception", activityId);
- #endif
- error.set(E);
- throw;
- }
- catch (...)
- {
- IException *E = MakeStringException(ROXIE_INTERNAL_ERROR, "Unknown exception caught in CRoxieServerThroughSpillActivity::readBuffered");
- error.set(E);
- throw E;
- }
- }
- }
- CriticalBlock b(crit);
- ActivityTimer t(totalCycles, timeActivities); // NOTE - time spent waiting for crit not included here. But it will have been included on the totalTime of the person holding the crit, so that is right
- unsigned lidx = curIdx - tailIdx;
- idx++;
- if (!lidx)
- {
- // The numOutputs <= 2 check optimizes slightly the common case of 2-way splitters.
- // The rationale is that I MUST be the last reader if there are only 2 - the other guy must have put the record there in the first place
- if (numOutputs <= 2 || isLastTailReader(oid)) // NOTE - this includes the case where minIndex returns (unsigned) -1, meaning no other active pullers
- {
- tailIdx++;
- const void *ret = buffer.dequeue(); // no need to link - last puller
- #ifdef TRACE_SPLIT
- CTXLOG("last puller return of %p", ret);
- #endif
- return ret;
- }
- }
- const void *ret = buffer.item(lidx);
- if (ret) LinkRoxieRow(ret);
- #ifdef TRACE_SPLIT
- CTXLOG("standard return of %p", ret);
- #endif
- return ret;
- }
- virtual void start(unsigned oid, unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CriticalBlock b(crit);
- if (error)
- throw error.getLink();
- if (factory)
- factory->noteStarted(oid);
- if (traceStartStop)
- CTXLOG("SPLIT %p: start %d child %d activeOutputs %d numOutputs %d numOriginalOutputs %d state %s", this, activityId, oid, activeOutputs, numOutputs, numOriginalOutputs, queryStateText(state));
- if (state != STATEstarted)
- {
- if (state != STATEstarting)
- initOutputs();
- tailIdx = 0;
- headIdx = 0;
- error.clear();
- try
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- catch (IException *E)
- {
- #ifdef TRACE_SPLIT
- CTXLOG("spill %d caught exception in start", activityId);
- #endif
- error.set(E);
- throw;
- }
- catch (...)
- {
- IException *E = MakeStringException(ROXIE_INTERNAL_ERROR, "Unknown exception caught in CRoxieServerThroughSpillActivity::start");
- error.set(E);
- throw E;
- }
- }
- }
- void stop(unsigned oid, unsigned & idx)
- {
- // Note that OutputAdaptor code ensures that stop is not called more than once per adaptor
- CriticalBlock b(crit);
- #ifdef TRACE_STARTSTOP
- if (traceStartStop)
- {
- CTXLOG("SPLIT %p: stop %d child %d activeOutputs %d numOutputs %d numOriginalOutputs %d state %s", this, activityId, oid, activeOutputs, numOutputs, numOriginalOutputs, queryStateText(state));
- if (watchActivityId && watchActivityId==activityId)
- {
- CTXLOG("WATCH: stop %d", activityId);
- }
- }
- #endif
- if (state != STATEstarting && state != STATEstarted)
- initOutputs();
- if (activeOutputs > 1)
- {
- if (tailIdx==idx)
- {
- // Discard all buffered rows that are there purely for this adaptor to read them
- unsigned min = minIndex(oid);
- if (min != (unsigned) -1)
- // what does -1 signify?? No-one wants anything? In which case can't we kill all rows??
- // Should never happen though if there are still some active.
- // there may be a small window where adaptors are blocked on the semaphore...
- {
- #ifdef TRACE_SPLIT
- CTXLOG("%p: Discarding buffered rows from %d to %d for oid %x (%d outputs active)", this, idx, min, oid, activeOutputs);
- #endif
- while (tailIdx < min)
- {
- ReleaseRoxieRow(buffer.dequeue());
- tailIdx++;
- }
- }
- }
- activeOutputs--;
- idx = (unsigned) -1; // causes minIndex not to save rows for me...
- return;
- }
- #ifdef TRACE_SPLIT
- CTXLOG("%p: All outputs done", this);
- #endif
- activeOutputs = numOutputs;
- CRoxieServerActivity::stop();
- };
- void reset(unsigned oid)
- {
- if (traceStartStop)
- CTXLOG("SPLIT %p: reset %d child %d activeOutputs %d numOutputs %d numOriginalOutputs %d state %s", this, activityId, oid, activeOutputs, numOutputs, numOriginalOutputs, queryStateText(state));
- activeOutputs = numOutputs;
- while (buffer.ordinality())
- ReleaseRoxieRow(buffer.dequeue());
- error.clear();
- if (state != STATEreset) // make sure input is only reset once
- CRoxieServerActivity::reset();
- };
- void connectInputStreams(bool consumerOrdered)
- {
- if (!inputStream)
- CRoxieServerActivity::connectInputStreams(consumerOrdered);
- }
- virtual const void *nextRow()
- {
- throwUnexpected(); // Internal logic error - we are not anybody's input
- }
- virtual IOutputMetaData * queryOutputMeta() const
- {
- // if (outputMeta)
- // return outputMeta;
- // else
- return input->queryOutputMeta(); // not always known (e.g. disk write - though Gavin _could_ fill it in)
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- if (idx==(unsigned)-1)
- idx = nextFreeOutput(); // MORE - what is this used for?
- assertex(idx < numOriginalOutputs);
- assertex(!used[idx]);
- used[idx] = true;
- return &adaptors[idx];
- }
- virtual void resetOutputsUsed()
- {
- numOutputs = 1;
- activeOutputs = 1;
- // MORE RKC->GH should we be clearing the used array here? anywhere?
- }
- virtual void noteOutputUsed()
- {
- assertex(numOutputs < numOriginalOutputs);
- numOutputs++;
- activeOutputs = numOutputs;
- }
- virtual bool isPassThrough()
- {
- return numOutputs==1;
- }
- };
- class CRoxieServerThroughSpillActivityFactory : public CRoxieServerMultiOutputFactory
- {
- public:
- CRoxieServerThroughSpillActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerMultiOutputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- Owned<IHThorSpillArg> helper = (IHThorSpillArg *) helperFactory();
- setNumOutputs(helper->getTempUsageCount() + 1);
- }
- CRoxieServerThroughSpillActivityFactory(IQueryFactory &_queryFactory, HelperFactory *_helperFactory, unsigned _numOutputs, IPropertyTree &_graphNode)
- : CRoxieServerMultiOutputFactory(0, 0, _queryFactory, _helperFactory, TAKsplit, _graphNode)
- {
- setNumOutputs(_numOutputs);
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerThroughSpillActivity(_ctx, this, _probeManager, numOutputs);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerThroughSpillActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerThroughSpillActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- IRoxieServerActivityFactory *createRoxieServerThroughSpillActivityFactory(IQueryFactory &_queryFactory, HelperFactory *_factory, unsigned _numOutputs, IPropertyTree &_graphNode)
- {
- return new CRoxieServerThroughSpillActivityFactory(_queryFactory, _factory, _numOutputs, _graphNode);
- }
- //----------------------------------------------------------------------------------------------
- class CRoxieServerSplitActivityFactory : public CRoxieServerMultiOutputFactory
- {
- public:
- CRoxieServerSplitActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerMultiOutputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- Owned<IHThorSplitArg> helper = (IHThorSplitArg *) helperFactory();
- setNumOutputs(helper->numBranches());
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerThroughSpillActivity(_ctx, this, _probeManager, numOutputs);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSplitActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerSplitActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- #define PIPE_BUFSIZE 0x8000
- static IException *createPipeFailureException(const char *cmd, unsigned retcode, IPipeProcess *pipe)
- {
- StringBuffer msg;
- if(pipe->hasError())
- {
- try
- {
- char error[512];
- size32_t sz = pipe->readError(sizeof(error), error);
- if(sz && sz!=(size32_t)-1)
- msg.append(", stderr: '").append(sz, error).append("'");
- }
- catch (IException *e)
- {
- EXCLOG(e, "Error reading pipe stderr");
- e->Release();
- }
- }
- return MakeStringException(ROXIE_PIPE_ERROR, "Pipe process %s returned error %u%s", cmd, retcode, msg.str());
- }
- class CRoxieServerPipeReadActivity : public CRoxieServerActivity
- {
- IHThorPipeReadArg &helper;
- Owned<IPipeProcess> pipe;
- StringAttr pipeCommand;
- Owned<IOutputRowDeserializer> rowDeserializer;
- Owned<IReadRowStream> readTransformer;
- bool groupSignalled;
- public:
- CRoxieServerPipeReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorPipeReadArg &)basehelper)
- {
- groupSignalled = true;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- rowDeserializer.setown(rowAllocator->createDiskDeserializer(ctx->queryCodeContext()));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- groupSignalled = true; // i.e. don't start with a NULL row
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if (!readTransformer)
- {
- OwnedRoxieString xmlIteratorPath(helper.getXmlIteratorPath());
- readTransformer.setown(createReadRowStream(rowAllocator, rowDeserializer, helper.queryXmlTransformer(), helper.queryCsvTransformer(), xmlIteratorPath, helper.getPipeFlags()));
- }
- OwnedRoxieString pipeProgram(helper.getPipeProgram());
- openPipe(pipeProgram);
- }
- virtual void stop()
- {
- CRoxieServerActivity::stop();
- pipe.clear();
- readTransformer->setStream(NULL);
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- while (!waitForPipe())
- {
- if (!pipe)
- return NULL;
- if (helper.getPipeFlags() & TPFgroupeachrow)
- {
- if (!groupSignalled)
- {
- groupSignalled = true;
- return NULL;
- }
- }
- }
- const void *ret = readTransformer->next();
- assertex(ret != NULL); // if ret can ever be NULL then we need to recode this logic
- processed++;
- groupSignalled = false;
- return ret;
- }
- protected:
- bool waitForPipe()
- {
- if (!pipe)
- return false; // done
- if (!readTransformer->eos())
- return true;
- verifyPipe();
- return false;
- }
- void openPipe(char const * cmd)
- {
- pipeCommand.setown(cmd);
- pipe.setown(createPipeProcess());
- if(!pipe->run(NULL, cmd, ".", false, true, true, 0x10000))
- throw MakeStringException(ROXIE_PIPE_ERROR, "Could not run pipe process %s", cmd);
- Owned<ISimpleReadStream> pipeReader = pipe->getOutputStream();
- readTransformer->setStream(pipeReader.get());
- }
- void verifyPipe()
- {
- if (pipe)
- {
- unsigned err = pipe->wait();
- if(err && !(helper.getPipeFlags() & TPFnofail))
- {
- throw createPipeFailureException(pipeCommand.get(), err, pipe);
- }
- pipe.clear();
- }
- }
- };
- class CRoxieServerPipeThroughActivity : public CRoxieServerActivity, implements IRecordPullerCallback
- {
- IHThorPipeThroughArg &helper;
- RecordPullerThread puller;
- Owned<IPipeProcess> pipe;
- StringAttr pipeCommand;
- InterruptableSemaphore pipeVerified;
- InterruptableSemaphore pipeOpened;
- CachedOutputMetaData inputMeta;
- Owned<IOutputRowSerializer> rowSerializer;
- Owned<IOutputRowDeserializer> rowDeserializer;
- Owned<IPipeWriteXformHelper> writeTransformer;
- Owned<IReadRowStream> readTransformer;
- bool firstRead;
- bool recreate;
- bool inputExhausted;
- bool groupSignalled;
- public:
- CRoxieServerPipeThroughActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorPipeThroughArg &)basehelper), puller(false)
- {
- recreate = helper.recreateEachRow();
- groupSignalled = true;
- firstRead = false;
- inputExhausted = false;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- rowSerializer.setown(inputMeta.createDiskSerializer(ctx->queryCodeContext(), activityId));
- rowDeserializer.setown(rowAllocator->createDiskDeserializer(ctx->queryCodeContext()));
- writeTransformer.setown(createPipeWriteXformHelper(helper.getPipeFlags(), helper.queryXmlOutput(), helper.queryCsvOutput(), rowSerializer));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- firstRead = true;
- inputExhausted = false;
- groupSignalled = true; // i.e. don't start with a NULL row
- pipeVerified.reinit();
- pipeOpened.reinit();
- writeTransformer->ready();
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if (!readTransformer)
- {
- OwnedRoxieString xmlIterator(helper.getXmlIteratorPath());
- readTransformer.setown(createReadRowStream(rowAllocator, rowDeserializer, helper.queryXmlTransformer(), helper.queryCsvTransformer(), xmlIterator, helper.getPipeFlags()));
- }
- if(!recreate)
- {
- OwnedRoxieString pipeProgram(helper.getPipeProgram());
- openPipe(pipeProgram);
- }
- puller.start(parentExtractSize, parentExtract, paused, 0, false, ctx); // Pipe does not support preload presently - locks up
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- if (idx)
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() parameter out of bounds at %s(%d)", __FILE__, __LINE__);
- puller.setInput(this, _sourceIdx, _in);
- inputMeta.set(_in->queryOutputMeta());
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- puller.connectInputStreams(ctx, consumerOrdered);
- return CRoxieServerActivity::getOutputStreams(ctx, idx, streams, consumerOptions, consumerOrdered, orderedCallbacks);
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (idx==0)
- return puller.queryInput();
- else
- return NULL;
- }
- virtual void stop()
- {
- pipeVerified.interrupt(NULL);
- pipeOpened.interrupt(NULL);
- puller.stop();
- CRoxieServerActivity::stop();
- pipe.clear();
- if (readTransformer)
- readTransformer->setStream(NULL);
- }
- virtual void reset()
- {
- puller.reset();
- CRoxieServerActivity::reset();
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- while (!waitForPipe())
- {
- if (!pipe)
- return NULL;
- if (helper.getPipeFlags() & TPFgroupeachrow)
- {
- if (!groupSignalled)
- {
- groupSignalled = true;
- return NULL;
- }
- }
- }
- const void *ret = readTransformer->next();
- assertex(ret != NULL); // if ret can ever be NULL then we need to recode this logic
- processed++;
- groupSignalled = false;
- return ret;
- }
- virtual void processRow(const void *row)
- {
- // called from puller thread
- if(recreate)
- openPipe(helper.getNameFromRow(row));
- writeTransformer->writeTranslatedText(row, pipe);
- ReleaseRoxieRow(row);
- if(recreate)
- {
- closePipe();
- pipeVerified.wait();
- }
- }
- virtual void processDone()
- {
- // called from puller thread
- if(recreate)
- {
- inputExhausted = true;
- pipeOpened.signal();
- }
- else
- {
- closePipe();
- pipeVerified.wait();
- }
- }
- virtual void processEOG()
- {
- }
- void processGroup(const ConstPointerArray &)
- {
- throwUnexpected();
- }
- virtual bool fireException(IException *e)
- {
- pipeOpened.interrupt(LINK(e));
- pipeVerified.interrupt(e);
- return true;
- }
- private:
- bool waitForPipe()
- {
- if (firstRead)
- {
- pipeOpened.wait();
- firstRead = false;
- }
- if (!pipe)
- return false; // done
- if (!readTransformer->eos())
- return true;
- verifyPipe();
- if (recreate && !inputExhausted)
- pipeOpened.wait();
- return false;
- }
- void openPipe(char const * cmd)
- {
- pipeCommand.setown(cmd);
- pipe.setown(createPipeProcess());
- if(!pipe->run(NULL, cmd, ".", true, true, true, 0x10000))
- throw MakeStringException(ROXIE_PIPE_ERROR, "Could not run pipe process %s", cmd);
- writeTransformer->writeHeader(pipe);
- Owned<ISimpleReadStream> pipeReader = pipe->getOutputStream();
- readTransformer->setStream(pipeReader.get());
- pipeOpened.signal();
- }
- void closePipe()
- {
- writeTransformer->writeFooter(pipe);
- pipe->closeInput();
- }
- void verifyPipe()
- {
- if (pipe)
- {
- unsigned err = pipe->wait();
- if(err && !(helper.getPipeFlags() & TPFnofail))
- {
- throw createPipeFailureException(pipeCommand.get(), err, pipe);
- }
- pipe.clear();
- pipeVerified.signal();
- }
- }
- };
- class CRoxieServerPipeWriteActivity : public CRoxieServerInternalSinkActivity
- {
- IHThorPipeWriteArg &helper;
- Owned<IPipeProcess> pipe;
- StringAttr pipeCommand;
- CachedOutputMetaData inputMeta;
- Owned<IOutputRowSerializer> rowSerializer;
- Owned<IPipeWriteXformHelper> writeTransformer;
- bool firstRead;
- bool recreate;
- bool inputExhausted;
- public:
- CRoxieServerPipeWriteActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numOutputs)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, _numOutputs), helper((IHThorPipeWriteArg &)basehelper)
- {
- recreate = helper.recreateEachRow();
- firstRead = false;
- inputExhausted = false;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- inputMeta.set(input->queryOutputMeta());
- rowSerializer.setown(inputMeta.createDiskSerializer(ctx->queryCodeContext(), activityId));
- writeTransformer.setown(createPipeWriteXformHelper(helper.getPipeFlags(), helper.queryXmlOutput(), helper.queryCsvOutput(), rowSerializer));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- firstRead = true;
- inputExhausted = false;
- writeTransformer->ready();
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if(!recreate)
- {
- OwnedRoxieString pipeProgram(helper.getPipeProgram());
- openPipe(pipeProgram);
- }
- }
- virtual void stop()
- {
- if (!aborted && helper.getSequence() >= 0)
- {
- WorkunitUpdate wu = ctx->updateWorkUnit();
- if (wu)
- {
- Owned<IWUResult> result = wu->updateResultBySequence(helper.getSequence());
- if (result)
- {
- result->setResultTotalRowCount(processed);
- result->setResultStatus(ResultStatusCalculated);
- }
- }
- }
- CRoxieServerActivity::stop();
- pipe.clear();
- }
- virtual void onExecute()
- {
- loop
- {
- const void *row = inputStream->ungroupedNextRow();
- if (!row)
- break;
- processed++;
- if(recreate)
- openPipe(helper.getNameFromRow(row));
- writeTransformer->writeTranslatedText(row, pipe);
- ReleaseRoxieRow(row);
- if(recreate)
- closePipe();
- }
- closePipe();
- }
- private:
- void openPipe(char const * cmd)
- {
- pipeCommand.setown(cmd);
- pipe.setown(createPipeProcess());
- if(!pipe->run(NULL, cmd, ".", true, false, true, 0x10000))
- throw MakeStringException(ROXIE_PIPE_ERROR, "Could not run pipe process %s", cmd);
- writeTransformer->writeHeader(pipe);
- }
- void closePipe()
- {
- writeTransformer->writeFooter(pipe);
- pipe->closeInput();
- unsigned err = pipe->wait();
- if(err && !(helper.getPipeFlags() & TPFnofail))
- {
- throw createPipeFailureException(pipeCommand.get(), err, pipe);
- }
- pipe.clear();
- }
- };
- class CRoxieServerPipeReadActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerPipeReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerPipeReadActivity(_ctx, this, _probeManager);
- }
- };
- class CRoxieServerPipeThroughActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerPipeThroughActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerPipeThroughActivity(_ctx, this, _probeManager);
- }
- };
- class CRoxieServerPipeWriteActivityFactory : public CRoxieServerInternalSinkFactory
- {
- public:
- CRoxieServerPipeWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, bool _isRoot)
- : CRoxieServerInternalSinkFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerPipeWriteActivity(_ctx, this, _probeManager, usageCount);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerPipeReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerPipeReadActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- IRoxieServerActivityFactory *createRoxieServerPipeThroughActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerPipeThroughActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- IRoxieServerActivityFactory *createRoxieServerPipeWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, bool _isRoot)
- {
- return new CRoxieServerPipeWriteActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot);
- }
- //=====================================================================================================
- class CRoxieServerStreamedIteratorActivity : public CRoxieServerActivity
- {
- IHThorStreamedIteratorArg &helper;
- Owned<IRowStream> rows;
- public:
- CRoxieServerStreamedIteratorActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorStreamedIteratorArg &)basehelper)
- {
- }
- ~CRoxieServerStreamedIteratorActivity()
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- rows.setown(helper.createInput());
- }
- virtual void stop()
- {
- if (rows)
- {
- rows->stop();
- rows.clear();
- }
- CRoxieServerActivity::stop();
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- assertex(rows != NULL);
- const void * next = rows->nextRow();
- if (next)
- processed++;
- return next;
- }
- };
- class CRoxieServerStreamedIteratorActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerStreamedIteratorActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerStreamedIteratorActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerStreamedIteratorActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerStreamedIteratorActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- class CRoxieServerFilterActivity : public CRoxieServerLateStartActivity
- {
- IHThorFilterArg &helper;
- bool anyThisGroup;
- IRangeCompare * stepCompare;
- public:
- CRoxieServerFilterActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerLateStartActivity(_ctx, _factory, _probeManager), helper((IHThorFilterArg &)basehelper)
- {
- anyThisGroup = false;
- stepCompare = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- anyThisGroup = false;
- CRoxieServerLateStartActivity::start(parentExtractSize, parentExtract, paused);
- lateStart(parentExtractSize, parentExtract, helper.canMatchAny());
- stepCompare = NULL;
- if (!eof)
- {
- IInputSteppingMeta * stepMeta = input->querySteppingMeta();
- if (stepMeta)
- stepCompare = stepMeta->queryCompare();
- }
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- loop
- {
- const void * ret = inputStream->nextRow();
- if (!ret)
- {
- //stop returning two NULLs in a row.
- if (anyThisGroup)
- {
- anyThisGroup = false;
- return NULL;
- }
- ret = inputStream->nextRow();
- if (!ret)
- {
- eof = true;
- return NULL; // eof...
- }
- }
- if (helper.isValid(ret))
- {
- anyThisGroup = true;
- processed++;
- return ret;
- }
- ReleaseRoxieRow(ret);
- }
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- //Could assert that this isn't grouped
- // MORE - will need rethinking once we rethink the nextRowGE interface for global smart-stepping.
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- loop
- {
- const void * ret = inputStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- if (!ret)
- {
- eof = true;
- return NULL;
- }
- if (!wasCompleteMatch)
- {
- anyThisGroup = false; // RKC->GH - is this right??
- return ret;
- }
- if (helper.isValid(ret))
- {
- anyThisGroup = true;
- processed++;
- return ret;
- }
- if (!stepExtra.returnMismatches())
- {
- ReleaseRoxieRow(ret);
- return nextRow();
- }
- //If asked to return mismatches we are only interested in mismatches that will force the stepped
- //condition to advance
- if (stepCompare->docompare(ret, seek, numFields) != 0)
- {
- wasCompleteMatch = false;
- anyThisGroup = false; // WHY?
- return ret;
- }
- ReleaseRoxieRow(ret);
- }
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- eof = prefiltered;
- anyThisGroup = false;
- inputStream->resetEOF();
- }
- IInputSteppingMeta * querySteppingMeta()
- {
- return input->querySteppingMeta();
- }
- };
- class CRoxieServerFilterActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerFilterActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- optStableInput = false;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerFilterActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerFilterActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerFilterActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- class CRoxieServerFilterGroupActivity : public CRoxieServerLateStartActivity
- {
- IHThorFilterGroupArg &helper;
- unsigned curIndex;
- ConstPointerArray gathered;
- IRangeCompare * stepCompare;
- public:
- CRoxieServerFilterGroupActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerLateStartActivity(_ctx, _factory, _probeManager), helper((IHThorFilterGroupArg &)basehelper)
- {
- curIndex = 0;
- stepCompare = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerLateStartActivity::start(parentExtractSize, parentExtract, paused);
- lateStart(parentExtractSize, parentExtract, helper.canMatchAny());//sets eof
- assertex(eof == !helper.canMatchAny());
- curIndex = 0;
- stepCompare = NULL;
- if (!eof)
- {
- IInputSteppingMeta * inputStepping = input->querySteppingMeta();
- if (inputStepping)
- stepCompare = inputStepping->queryCompare();
- }
- }
- virtual void reset()
- {
- releaseGathered();
- CRoxieServerLateStartActivity::reset();
- }
- inline void releaseGathered()
- {
- roxiemem::ReleaseRoxieRowRange(gathered.getArray(), curIndex, gathered.ordinality());
- gathered.kill();
- curIndex = 0;
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- if (eof)
- return NULL;
- if (gathered.ordinality())
- {
- if (gathered.isItem(curIndex))
- {
- const void * ret = gathered.item(curIndex++);
- processed++;
- return ret;
- }
- curIndex = 0;
- gathered.kill();
- return NULL;
- }
- const void * ret = inputStream->nextRow();
- while (ret)
- {
- gathered.append(ret);
- ret = inputStream->nextRow();
- }
- unsigned num = gathered.ordinality();
- if (num != 0)
- {
- if (!helper.isValid(num, (const void * *)gathered.getArray()))
- ReleaseRoxieRows(gathered); // read next group
- }
- else
- eof = true;
- }
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- if (gathered.ordinality())
- {
- while (gathered.isItem(curIndex))
- {
- const void * ret = gathered.item(curIndex++);
- if (stepCompare->docompare(ret, seek, numFields) >= 0)
- {
- processed++;
- return ret;
- }
- ReleaseRoxieRow(ret);
- }
- curIndex = 0;
- gathered.kill();
- //nextRowGE never returns an end of group marker.
- }
- //Not completely sure about this - it could lead the the start of a group being skipped,
- //so the group filter could potentially work on a different group. If so, we'd need to check the
- //next fields were a subset of the grouping fields - more an issue for the group activity.
-
- //MORE: What do we do with wasCompleteMatch? something like the following????
- #if 0
- loop
- {
- const void * ret;
- if (stepExtra.returnMismatches())
- {
- bool matchedCompletely = true;
- ret = input->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- if (!wasCompleteMatch)
- return ret;
- }
- else
- ret = input->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- #endif
- const void * ret = inputStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- while (ret)
- {
- gathered.append(ret);
- ret = inputStream->nextRow();
- }
- unsigned num = gathered.ordinality();
- if (num != 0)
- {
- if (!helper.isValid(num, (const void * *)gathered.getArray()))
- ReleaseRoxieRows(gathered); // read next group
- }
- else
- eof = true;
- return IEngineRowStream::ungroupedNextRow();
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- eof = false;
- releaseGathered();
- inputStream->resetEOF();
- }
- IInputSteppingMeta * querySteppingMeta()
- {
- return input->querySteppingMeta();
- }
- };
- class CRoxieServerFilterGroupActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerFilterGroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerFilterGroupActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerFilterGroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerFilterGroupActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerSideEffectActivity : public CRoxieServerActivity
- {
- IHThorSideEffectArg &helper;
- CriticalSection ecrit;
- Owned<IException> exception;
- bool executed;
- public:
- CRoxieServerSideEffectActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorSideEffectArg &)basehelper)
- {
- executed = false;
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- CriticalBlock b(ecrit);
- if (exception)
- throw(exception.getLink());
- if (!executed)
- {
- try
- {
- executed = true;
- helper.action();
- }
- catch(IException *E)
- {
- exception.set(E);
- throw;
- }
- }
- return NULL;
- }
- virtual void execute(unsigned parentExtractSize, const byte * parentExtract)
- {
- CriticalBlock b(ecrit);
- if (exception)
- throw(exception.getLink());
- if (!executed)
- {
- try
- {
- ActivityTimer t(totalCycles, timeActivities);
- executed = true;
- start(parentExtractSize, parentExtract, false);
- helper.action();
- stop();
- }
- catch(IException *E)
- {
- ctx->notifyAbort(E);
- abort();
- exception.set(E);
- throw;
- }
- }
- }
- virtual void reset()
- {
- executed = false;
- exception.clear();
- CRoxieServerActivity::reset();
- }
- };
- class CRoxieServerSideEffectActivityFactory : public CRoxieServerActivityFactory
- {
- bool isRoot;
- public:
- CRoxieServerSideEffectActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), isRoot(_isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSideEffectActivity(_ctx, this, _probeManager);
- }
- virtual bool isSink() const
- {
- return isRoot && !meta.queryOriginal();
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSideEffectActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerSideEffectActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //=================================================================================
- class CRoxieServerActionActivity : public CRoxieServerInternalSinkActivity
- {
- IHThorActionArg &helper;
- public:
- CRoxieServerActionActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numOutputs)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, _numOutputs), helper((IHThorActionArg &)basehelper)
- {
- }
- virtual void onExecute()
- {
- helper.action();
- }
- };
- class CRoxieServerActionActivityFactory : public CRoxieServerInternalSinkFactory
- {
- public:
- CRoxieServerActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, bool _isRoot)
- : CRoxieServerInternalSinkFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerActionActivity(_ctx, this, _probeManager, usageCount);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, bool _isRoot)
- {
- return new CRoxieServerActionActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot);
- }
- //=================================================================================
- class CRoxieServerSampleActivity : public CRoxieServerActivity
- {
- IHThorSampleArg &helper;
- unsigned numSamples;
- unsigned numToSkip;
- unsigned whichSample;
- bool anyThisGroup;
- bool eof;
- public:
- CRoxieServerSampleActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorSampleArg &)basehelper)
- {
- numSamples = 0;
- numToSkip = 0;
- whichSample = 0;
- anyThisGroup = false;
- eof = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- anyThisGroup = false;
- eof = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- numSamples = helper.getProportion();
- whichSample = helper.getSampleNumber();
- numToSkip = (whichSample ? whichSample-1 : 0);
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- loop
- {
- const void * ret = inputStream->nextRow();
- if (!ret)
- {
- //this does work with groups - may or may not be useful...
- //reset the sample for each group.... probably best.
- numToSkip = (whichSample ? whichSample-1 : 0);
- if (anyThisGroup)
- {
- anyThisGroup = false;
- return NULL;
- }
- ret = inputStream->nextRow();
- if (!ret)
- {
- eof = true;
- return NULL; // eof...
- }
- }
- if (numToSkip == 0)
- {
- anyThisGroup = true;
- numToSkip = numSamples-1;
- processed++;
- return ret;
- }
- numToSkip--;
- ReleaseRoxieRow(ret);
- }
- }
- };
- class CRoxieServerSampleActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerSampleActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSampleActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSampleActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerSampleActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerChooseSetsActivity : public CRoxieServerActivity
- {
- IHThorChooseSetsArg &helper;
- unsigned numSets;
- unsigned * setCounts;
- bool done;
- public:
- CRoxieServerChooseSetsActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorChooseSetsArg &)basehelper)
- {
- setCounts = NULL;
- numSets = 0;
- done = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- done = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- numSets = helper.getNumSets();
- setCounts = new unsigned[numSets];
- memset(setCounts, 0, sizeof(unsigned)*numSets);
- helper.setCounts(setCounts);
- }
- virtual void reset()
- {
- delete [] setCounts;
- setCounts = NULL;
- CRoxieServerActivity::reset();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (done)
- return NULL;
- loop
- {
- const void * ret = inputStream->ungroupedNextRow();
- if (!ret)
- {
- done = true;
- return NULL;
- }
- processed++;
- switch (helper.getRecordAction(ret))
- {
- case 2:
- done = true;
- return ret;
- case 1:
- return ret;
- }
- ReleaseRoxieRow(ret);
- }
- }
- };
- class CRoxieServerChooseSetsActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerChooseSetsActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerChooseSetsActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerChooseSetsActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerChooseSetsActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerChooseSetsExActivity : public CRoxieServerActivity
- {
- protected:
- IHThorChooseSetsExArg &helper;
- unsigned numSets;
- unsigned curIndex;
- unsigned * setCounts;
- count_t * limits;
- bool done;
- ConstPointerArray gathered;
- virtual bool includeRow(const void * row) = 0;
- virtual void calculateSelection() = 0;
- public:
- CRoxieServerChooseSetsExActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorChooseSetsExArg &)basehelper)
- {
- setCounts = NULL;
- limits = NULL;
- done = false;
- curIndex = 0;
- numSets = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- done = false;
- curIndex = 0;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- numSets = helper.getNumSets();
- setCounts = new unsigned[numSets];
- memset(setCounts, 0, sizeof(unsigned)*numSets);
- limits = (count_t *)calloc(sizeof(count_t), numSets);
- helper.getLimits(limits);
- }
- virtual void reset()
- {
- delete [] setCounts;
- setCounts = NULL;
- free(limits);
- limits = NULL;
- roxiemem::ReleaseRoxieRowRange(gathered.getArray(), curIndex, gathered.ordinality());
- gathered.kill();
- curIndex = 0;
- CRoxieServerActivity::reset();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (gathered.ordinality() == 0)
- {
- curIndex = 0;
- if (!inputStream->nextGroup(gathered))
- {
- done = true;
- return NULL;
- }
- ForEachItemIn(idx1, gathered)
- {
- unsigned category = helper.getCategory(gathered.item(idx1));
- if (category)
- setCounts[category-1]++;
- }
- calculateSelection();
- }
- while (gathered.isItem(curIndex))
- {
- const void * row = gathered.item(curIndex);
- gathered.replace(NULL, curIndex);
- curIndex++;
- if (includeRow(row))
- {
- processed++;
- return row;
- }
- ReleaseRoxieRow(row);
- }
- gathered.kill();
- return NULL;
- }
- };
- class CRoxieServerChooseSetsLastActivity : public CRoxieServerChooseSetsExActivity
- {
- unsigned * numToSkip;
- public:
- CRoxieServerChooseSetsLastActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager) : CRoxieServerChooseSetsExActivity(_ctx, _factory, _probeManager)
- {
- numToSkip = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerChooseSetsExActivity::start(parentExtractSize, parentExtract, paused);
- numToSkip = (unsigned *)calloc(sizeof(unsigned), numSets);
- }
- virtual void reset()
- {
- free(numToSkip);
- numToSkip = NULL;
- CRoxieServerChooseSetsExActivity::reset();
- }
- protected:
- virtual void calculateSelection()
- {
- for (unsigned idx=0; idx < numSets; idx++)
- {
- if (setCounts[idx] < limits[idx])
- numToSkip[idx] = 0;
- else
- numToSkip[idx] = (unsigned)(setCounts[idx] - limits[idx]);
- }
- }
- virtual bool includeRow(const void * row)
- {
- unsigned category = helper.getCategory(row);
- if (category)
- {
- if (numToSkip[category-1] == 0)
- return true;
- numToSkip[category-1]--;
- }
- return false;
- }
- };
- class CRoxieServerChooseSetsEnthActivity : public CRoxieServerChooseSetsExActivity
- {
- count_t * counter;
- public:
- CRoxieServerChooseSetsEnthActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager) : CRoxieServerChooseSetsExActivity(_ctx, _factory, _probeManager)
- {
- counter = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerChooseSetsExActivity::start(parentExtractSize, parentExtract, paused);
- counter = (count_t *)calloc(sizeof(count_t), numSets);
- }
- virtual void reset()
- {
- free(counter);
- counter = NULL;
- CRoxieServerChooseSetsExActivity::reset();
- }
- protected:
- virtual void calculateSelection()
- {
- }
- virtual bool includeRow(const void * row)
- {
- unsigned category = helper.getCategory(row);
- if (category)
- {
- assertex(category <= numSets);
- counter[category-1] += limits[category-1];
- if(counter[category-1] >= setCounts[category-1])
- {
- counter[category-1] -= setCounts[category-1];
- return true;
- }
- }
- return false;
- }
- };
- class CRoxieServerChooseSetsEnthActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerChooseSetsEnthActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerChooseSetsEnthActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerChooseSetsEnthActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerChooseSetsEnthActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- class CRoxieServerChooseSetsLastActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerChooseSetsLastActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerChooseSetsLastActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerChooseSetsLastActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerChooseSetsLastActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerEnthActivity : public CRoxieServerActivity
- {
- IHThorEnthArg &helper;
- unsigned __int64 numerator;
- unsigned __int64 denominator;
- unsigned __int64 counter;
- bool eof;
- public:
- CRoxieServerEnthActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorEnthArg &)basehelper)
- {
- eof = false;
- numerator = denominator = counter = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- numerator = helper.getProportionNumerator();
- denominator = helper.getProportionDenominator();
- if(denominator == 0) denominator = 1; //MORE: simplest way to avoid disaster in this case
- counter = (helper.getSampleNumber()-1) * greatestCommonDivisor(numerator, denominator);
- if (counter >= denominator)
- counter %= denominator;
- }
- inline bool wanted()
- {
- counter += numerator;
- if(counter >= denominator)
- {
- counter -= denominator;
- return true;
- }
- return false;
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- const void * ret;
- loop
- {
- ret = inputStream->ungroupedNextRow();
- if (!ret) //eof
- {
- eof = true;
- return ret;
- }
- if (wanted())
- return ret;
- ReleaseRoxieRow(ret);
- }
- }
- };
- class CRoxieServerEnthActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerEnthActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerEnthActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerEnthActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerEnthActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerAggregateActivity : public CRoxieServerActivity
- {
- IHThorAggregateArg &helper;
- bool eof;
- bool isInputGrouped;
- bool abortEarly;
- public:
- CRoxieServerAggregateActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorAggregateArg &)basehelper)
- {
- eof = false;
- isInputGrouped = false;
- abortEarly = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- isInputGrouped = input->queryOutputMeta()->isGrouped(); // could be done earlier, in setInput?
- abortEarly = !isInputGrouped && (factory->getKind() == TAKexistsaggregate); // ditto
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- const void * next = inputStream->nextRow();
- if (!next && isInputGrouped)
- {
- eof = true;
- return NULL;
- }
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t finalSize = helper.clearAggregate(rowBuilder);
- if (next)
- {
- finalSize = helper.processFirst(rowBuilder, next);
- ReleaseRoxieRow(next);
- if (!abortEarly)
- {
- loop
- {
- next = inputStream->nextRow();
- if (!next)
- break;
- finalSize = helper.processNext(rowBuilder, next);
- ReleaseRoxieRow(next);
- }
- }
- }
- if (!isInputGrouped) // either read all, or aborted early
- eof = true;
- processed++;
- return rowBuilder.finalizeRowClear(finalSize);
- }
- };
- class CRoxieServerStrandedAggregateActivity : public CRoxieServerStrandedActivity, implements IOrderedCallbackCollection
- {
- class AggregateProcessor : public StrandProcessor, implements IStrandThreaded, implements IOrderedOutputCallback
- {
- protected:
- IHThorAggregateArg &helper;
- OwnedConstRoxieRow result;
- bool eof = false;
- bool isInputGrouped = false;
- bool abortEarly = false;
- public:
- AggregateProcessor(CRoxieServerActivity &_parent, IEngineRowStream *_inputStream, IHThorAggregateArg &_helper, bool _isInputGrouped, bool _abortEarly)
- : StrandProcessor(_parent, _inputStream, true), helper(_helper), isInputGrouped(_isInputGrouped), abortEarly(_abortEarly)
- {
- }
- virtual const void * nextRow() override
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof || isAborting())
- return NULL;
- const void * next = inputStream->nextRow();
- if (!next && isInputGrouped)
- {
- eof = true;
- return NULL;
- }
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t finalSize = helper.clearAggregate(rowBuilder);
- if (next)
- {
- finalSize = helper.processFirst(rowBuilder, next);
- ReleaseRoxieRow(next);
- if (!abortEarly)
- {
- loop
- {
- next = inputStream->nextRow();
- if (!next)
- break;
- finalSize = helper.processNext(rowBuilder, next);
- ReleaseRoxieRow(next);
- if (isAborting())
- break;
- }
- }
- }
- if (!isInputGrouped) // either read all, or aborted early
- eof = true;
- processed++;
- return rowBuilder.finalizeRowClear(finalSize);
- }
- virtual void main() override
- {
- isInputGrouped = true; // gather no row if there are no input rows.
- result.setown(nextRow());
- queryActivity().noteStrandFinished(inputStream);
- }
- virtual void reset() override
- {
- StrandProcessor::reset();
- result.clear();
- eof = false;
- }
- //interface IOutputOutputCallback
- virtual bool noteEndOfInputChunk() { return true; }
- virtual void noteEndOfInput() {}
- virtual void stopStream()
- {
- inputStream->stop();
- }
- const void * queryResult() const { return result; }
- CRoxieServerStrandedAggregateActivity & queryActivity() { return static_cast<CRoxieServerStrandedAggregateActivity &>(parent); }
- };
- //This class is only used when a single aggregate is being calculated.
- class AggregateCombiner : public CInterfaceOf<IEngineRowStream>
- {
- public:
- explicit AggregateCombiner(CRoxieServerStrandedAggregateActivity & _parent) : parent(_parent)
- {
- }
- virtual const void *nextRow()
- {
- return parent.getCombinedAggregate();
- }
- virtual void stop()
- {
- parent.outputStopped();
- }
- virtual void resetEOF()
- {
- throwUnexpected();
- }
- protected:
- CRoxieServerStrandedAggregateActivity & parent;
- } combiner;
- class NullCallback : implements IOrderedOutputCallback
- {
- public:
- virtual bool noteEndOfInputChunk() { return true; }
- virtual void noteEndOfInput() {}
- } nullCallback;
- public:
- CRoxieServerStrandedAggregateActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const StrandOptions &_strandOptions)
- : CRoxieServerStrandedActivity(_ctx, _factory, _probeManager, _strandOptions), helper((IHThorAggregateArg &)basehelper), combiner(*this)
- {
- isInputGrouped = false;
- abortEarly = false;
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- CRoxieServerStrandedActivity::setInput(idx, _sourceIdx, _in);
- isInputGrouped = input->queryOutputMeta()->isGrouped(); // could be done earlier, in setInput?
- abortEarly = !isInputGrouped && (factory->getKind() == TAKexistsaggregate); // ditto
- //For EXISTS(), and aggregates that rely on the input order, force to execute single threaded
- //MORE: Ordered should be able to process a block at a time and then merge them, similar to disk write processing.
- if (abortEarly || (!isInputGrouped && queryFactory()->isInputOrdered(false, 0)))
- strandOptions.numStrands = 1;
- else if (!isInputGrouped)
- combineStreams = true;
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks) override
- {
- PointerArrayOf<IEngineRowStream> localStreams;
- //If streams are being combined, this class should be notified of end of sections, otherwise grouped aggregates can inform a downstream activity.
- IOrderedCallbackCollection * callbacks = combineStreams ? static_cast<IOrderedCallbackCollection *>(this) : orderedCallbacks;
- Owned<IStrandJunction> outJunction = CRoxieServerStrandedActivity::getOutputStreams(ctx, idx, localStreams, consumerOptions, consumerOrdered, callbacks);
- if (localStreams.ordinality() == 1)
- combineStreams = false; // Select a more efficient/simpler code path
- if (combineStreams)
- {
- assertex(!outJunction); // this needs more thought!
- streams.append(&combiner);
- }
- else
- {
- streams.swapWith(localStreams);
- }
- return outJunction.getClear();
- }
- virtual StrandProcessor *createStrandProcessor(IEngineRowStream *instream)
- {
- DBGLOG("Create aggregate strand processor %u", strandOptions.numStrands);
- return new AggregateProcessor(*this, instream, (IHThorAggregateArg &) basehelper, isInputGrouped && !combineStreams, abortEarly);
- }
- virtual StrandProcessor *createStrandSourceProcessor(bool inputOrdered) { throwUnexpected(); }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused) override
- {
- CRoxieServerStrandedActivity::start(parentExtractSize, parentExtract, paused);
- if (combineStreams)
- {
- barrier.setown(createStrandBarrier());
- ForEachItemIn(i, strands)
- barrier->startStrand(queryAggregate(i));
- active.store(1);
- }
- else
- active.store(0);
- }
- virtual void reset() override
- {
- CRoxieServerStrandedActivity::reset();
- barrier.clear();
- calculated = false;
- }
- //Helper functions for the single aggregate case
- const void * getCombinedAggregate()
- {
- if (calculated)
- return NULL;
- calculated= true;
- barrier->waitForStrands();
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t finalSize = helper.clearAggregate(rowBuilder);
- bool isFirst = true;
- for (unsigned i=0; i < numStrands(); i++)
- {
- const void * next = queryAggregate(i).queryResult();
- if (next)
- {
- if (isFirst)
- {
- finalSize = cloneRow(rowBuilder, next, helper.queryOutputMeta());
- isFirst = false;
- }
- else
- finalSize = helper.mergeAggregate(rowBuilder, next);
- }
- }
- return rowBuilder.finalizeRowClear(finalSize);
- }
- void outputStopped()
- {
- //only called if generating a single result.
- assertex(combineStreams);
- if (barrier)
- {
- requestAbort(); // if strands are still running, abort them
- barrier->waitForStrands();
- }
- else
- {
- ForEachItemIn(i, strands)
- queryAggregate(i).stopStream();
- }
- CRoxieServerStrandedActivity::stop();
- }
- void noteStrandFinished(IRowStream * stream)
- {
- barrier->noteStrandFinished(stream);
- }
- virtual IOrderedOutputCallback * queryCallback(unsigned i)
- {
- //The non-ordered aggregate does not need to process the ordered call back information.
- //If the ordered single aggregate was implemented then it would need to be processed instead of ignored,
- //however it needs some care since the strands are not yet created. So cannot return strand(i).
- return &nullCallback;
- }
- inline AggregateProcessor & queryAggregate(unsigned i)
- {
- return static_cast<AggregateProcessor &>(strands.item(i));
- }
- protected:
- IHThorAggregateArg &helper;
- Owned<IStrandBarrier> barrier;
- IEngineRowStream * combinedStream = nullptr;
- bool isInputGrouped;
- bool abortEarly;
- bool combineStreams = false;
- bool calculated = false;
- };
- class CRoxieServerAggregateActivityFactory : public CRoxieServerActivityFactory
- {
- StrandOptions strandOptions;
- public:
- CRoxieServerAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), strandOptions(_graphNode)
- {
- Owned<IHThorAggregateArg> helper = (IHThorAggregateArg *) helperFactory();
- if (!(helper->getAggregateFlags() & TAForderedmerge))
- optStableInput = false;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerStrandedAggregateActivity(_ctx, this, _probeManager, strandOptions);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerAggregateActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- typedef unsigned t_hashPrefix;
- class CRoxieServerHashAggregateActivity : public CRoxieServerActivity
- {
- IHThorHashAggregateArg &helper;
- RowAggregator aggregated;
- bool eof;
- bool gathered;
- bool isGroupedAggregate;
- public:
- CRoxieServerHashAggregateActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, bool _isGroupedAggregate, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorHashAggregateArg &)basehelper),
- isGroupedAggregate(_isGroupedAggregate),
- aggregated(helper, helper)
- {
- eof = false;
- gathered = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- gathered = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- aggregated.reset();
- CRoxieServerActivity::reset();
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- if (!gathered)
- {
- aggregated.start(rowAllocator);
- bool eog = true;
- loop
- {
- const void * next = inputStream->nextRow();
- if (!next)
- {
- if (isGroupedAggregate)
- {
- if (eog)
- eof = true;
- break;
- }
- next = inputStream->nextRow();
- if (!next)
- break;
- }
- eog = false;
- aggregated.addRow(next);
- ReleaseRoxieRow(next);
- }
- gathered = true;
- }
- Owned<AggregateRowBuilder> next = aggregated.nextResult();
- if (next)
- {
- processed++;
- return next->finalizeRowClear();
- }
- if (!isGroupedAggregate)
- eof = true;
- aggregated.reset();
- gathered = false;
- return NULL;
- }
- };
- class CRoxieServerHashAggregateActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerHashAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- isGroupedAggregate = _graphNode.getPropBool("att[@name='grouped']/@value");
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerHashAggregateActivity(_ctx, this, isGroupedAggregate, _probeManager);
- }
- protected:
- bool isGroupedAggregate;
- };
- IRoxieServerActivityFactory *createRoxieServerHashAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerHashAggregateActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerDegroupActivity : public CRoxieServerActivity
- {
- IHThorDegroupArg &helper;
- bool eof;
- public:
- CRoxieServerDegroupActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorDegroupArg &)basehelper)
- {
- eof = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- const void * ret = inputStream->ungroupedNextRow();
- if (ret)
- processed++;
- else
- eof = true;
- return ret;
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- const void * ret = inputStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- if (ret)
- processed++;
- else
- eof = true;
- return ret;
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- eof = false;
- inputStream->resetEOF();
- }
- IInputSteppingMeta * querySteppingMeta()
- {
- return input->querySteppingMeta();
- }
- };
- class CRoxieServerDegroupActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerDegroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerDegroupActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerDegroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerDegroupActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerSpillReadActivity : public CRoxieServerActivity
- {
- IHThorDiskReadArg &helper;
- bool needTransform;
- bool eof;
- bool anyThisGroup;
- unsigned __int64 rowLimit;
- unsigned __int64 choosenLimit;
- public:
- CRoxieServerSpillReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorDiskReadArg &)basehelper)
- {
- needTransform = helper.needTransform();
- rowLimit = (unsigned __int64) -1;
- choosenLimit = 0;
- eof = false;
- anyThisGroup = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- anyThisGroup = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if (helper.canMatchAny())
- eof = false;
- else
- eof = true;
- choosenLimit = helper.getChooseNLimit();
- rowLimit = helper.getRowLimit();
- helper.setCallback(NULL); // members should not be called - change if they are
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- if (processed==choosenLimit)
- {
- eof = true;
- return NULL;
- }
- if (needTransform)
- {
- loop
- {
- const void *in = inputStream->nextRow();
- if (!in)
- {
- if (anyThisGroup)
- {
- anyThisGroup = false;
- return NULL;
- }
- in = inputStream->nextRow();
- if (!in)
- {
- eof = true;
- return NULL; // eof...
- }
- }
- unsigned outSize;
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- try
- {
- outSize = helper.transform(rowBuilder, in);
- ReleaseRoxieRow(in);
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- if (outSize)
- {
- anyThisGroup = true;
- processed++;
- if (processed==rowLimit)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d line = %d", activityId, __LINE__);
- helper.onLimitExceeded();
- }
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- }
- else
- {
- const void *ret = inputStream->nextRow();
- if (ret)
- {
- processed++;
- if (processed==rowLimit)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d line = %d", activityId, __LINE__);
- ReleaseClearRoxieRow(ret);
- helper.onLimitExceeded(); // should not return
- throwUnexpected();
- }
- }
- return ret;
- }
- }
- };
- class CRoxieServerSpillReadActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerSpillReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSpillReadActivity(_ctx, this, _probeManager);
- }
- virtual void addDependency(unsigned source, ThorActivityKind sourceKind, unsigned sourceIdx, int controlId, const char *edgeId)
- {
- if (sourceKind==TAKspill || sourceKind==TAKdiskwrite) // Bit of a hack - codegen probably should differentiate
- setInput(0, source, sourceIdx);
- else
- CRoxieServerActivityFactory::addDependency(source, kind, sourceIdx, controlId, edgeId);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSpillReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerSpillReadActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerSpillWriteActivity : public CRoxieServerActivity
- {
- public:
- CRoxieServerSpillWriteActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager)
- {
- }
- ~CRoxieServerSpillWriteActivity()
- {
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- return inputStream->nextRow();
- }
- };
- //==================================================================================
- class CRoxieServerDiskWriteActivity : public CRoxieServerInternalSinkActivity, implements IRoxiePublishCallback
- {
- protected:
- Owned<IExtRowWriter> outSeq;
- Owned<IOutputRowSerializer> rowSerializer;
- Linked<IFileIOStream> diskout;
- bool blockcompressed;
- bool extend;
- bool overwrite;
- bool encrypted;
- bool grouped;
- IHThorDiskWriteArg &helper;
- CachedOutputMetaData diskmeta;
- Owned<IRoxieWriteHandler> writer;
- bool tallycrc;
- unsigned __int64 uncompressedBytesWritten;
- CRC32 crc;
- void updateWorkUnitResult(unsigned __int64 reccount)
- {
- assertex(writer);
- // MORE - a lot of this is common with hthor
- WorkunitUpdate wu = ctx->updateWorkUnit();
- if (wu)
- {
- OwnedRoxieString rawLogicalName = helper.getFileName();
- StringBuffer lfn; // logical filename
- expandLogicalFilename(lfn, rawLogicalName, wu, false, false);
- if (lfn.length())
- {
- unsigned flags = helper.getFlags();
- WUFileKind fileKind;
- if (TDXtemporary & flags)
- fileKind = WUFileTemporary;
- else if(TDXjobtemp & flags)
- fileKind = WUFileJobOwned;
- else if(TDWowned & flags)
- fileKind = WUFileOwned;
- else
- fileKind = WUFileStandard;
- StringArray clusters;
- writer->getClusters(clusters);
- wu->addFile(lfn.str(), &clusters, helper.getTempUsageCount(), fileKind, NULL);
- if (!(flags & TDXtemporary) && helper.getSequence() >= 0)
- {
- Owned<IWUResult> result = wu->updateResultBySequence(helper.getSequence());
- if (result)
- {
- result->setResultTotalRowCount(reccount);
- result->setResultStatus(ResultStatusCalculated);
- if (helper.getFlags() & TDWresult)
- result->setResultFilename(lfn.str());
- else
- result->setResultLogicalName(lfn.str());
- }
- }
- }
- }
- }
- void resolve()
- {
- OwnedRoxieString rawLogicalName = helper.getFileName();
- assertex(rawLogicalName);
- assertex((helper.getFlags() & TDXtemporary) == 0);
- StringArray clusters;
- unsigned clusterIdx = 0;
- while(true)
- {
- OwnedRoxieString cluster(helper.getCluster(clusterIdx));
- if(!cluster)
- break;
- clusters.append(cluster);
- clusterIdx++;
- }
- if (clusters.length())
- {
- if (extend)
- throw MakeStringException(0, "Cannot combine EXTEND and CLUSTER flags on disk write of file %s", rawLogicalName.get());
- }
- else
- {
- if (roxieName.length())
- clusters.append(roxieName.str());
- else
- clusters.append(".");
- }
- writer.setown(ctx->createLFN(rawLogicalName, overwrite, extend, clusters));
- // MORE - need to check somewhere that single part if it's an existing file or an external one...
- }
- public:
- CRoxieServerDiskWriteActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, 0), helper((IHThorDiskWriteArg &)basehelper)
- {
- extend = ((helper.getFlags() & TDWextend) != 0);
- overwrite = ((helper.getFlags() & TDWoverwrite) != 0);
- grouped = false; // don't think we need to support it...
- diskmeta.set(helper.queryDiskRecordSize());
- blockcompressed = (((helper.getFlags() & TDWnewcompress) != 0) || (((helper.getFlags() & TDXcompress) != 0) && (diskmeta.getFixedSize() >= MIN_ROWCOMPRESS_RECSIZE))); //always use new compression
- encrypted = false; // set later
- tallycrc = true;
- uncompressedBytesWritten = 0;
- }
- ~CRoxieServerDiskWriteActivity()
- {
- }
- virtual bool needsAllocator() const
- {
- return true;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
-
- resolve();
- Owned<IFileIO> io;
- void *ekey;
- size32_t ekeylen;
- helper.getEncryptKey(ekeylen, ekey);
- Owned<ICompressor> ecomp;
- if (ekeylen!=0)
- {
- ecomp.setown(createAESCompressor256(ekeylen,ekey));
- memset(ekey,0,ekeylen);
- rtlFree(ekey);
- encrypted = true;
- blockcompressed = true;
- }
- if (blockcompressed)
- io.setown(createCompressedFileWriter(writer->queryFile(), (diskmeta.isFixedSize() ? diskmeta.getFixedSize() : 0), extend, true, ecomp));
- else
- io.setown(writer->queryFile()->open(extend ? IFOwrite : IFOcreate));
- if (!io)
- throw MakeStringException(errno, "Failed to create%s file %s for writing", (encrypted ? " encrypted" : (blockcompressed ? " compressed" : "")), writer->queryFile()->queryFilename());
- diskout.setown(createBufferedIOStream(io));
- if (extend)
- diskout->seek(0, IFSend);
- tallycrc = !factory->queryQueryFactory().queryOptions().skipFileFormatCrcCheck && !(helper.getFlags() & TDRnocrccheck) && !blockcompressed;
- Owned<IRowInterfaces> rowIf = createRowInterfaces(input->queryOutputMeta(), activityId, factory->getHeapFlags(), ctx->queryCodeContext());
- rowSerializer.set(rowIf->queryRowSerializer());
- unsigned rwFlags = rw_autoflush;
- if(grouped)
- rwFlags |= rw_grouped;
- if(tallycrc)
- rwFlags |= rw_crc;
- outSeq.setown(createRowWriter(diskout, rowIf, rwFlags));
- }
- virtual void stop()
- {
- if (aborted)
- {
- if (writer)
- writer->finish(false, this);
- }
- else
- {
- if (outSeq)
- outSeq->flush(&crc);
- if (outSeq)
- uncompressedBytesWritten = outSeq->getPosition();
- outSeq.clear();
- diskout.clear(); // Make sure file is properly closed or date may not match published info
- if (writer)
- {
- updateWorkUnitResult(processed);
- writer->finish(true, this);
- }
- }
- writer.clear();
- CRoxieServerActivity::stop();
- }
- virtual void reset()
- {
- CRoxieServerActivity::reset();
- diskout.clear();
- outSeq.clear();
- writer.clear();
- uncompressedBytesWritten = 0;
- crc.reset();
- }
- virtual void onExecute()
- {
- loop
- {
- const void *nextrec = inputStream->ungroupedNextRow();
- if (!nextrec)
- break;
- processed++;
- outSeq->putRow(nextrec);
- }
- }
- virtual void setFileProperties(IFileDescriptor *desc) const
- {
- IPropertyTree &partProps = desc->queryPart(0)->queryProperties(); //properties of the first file part.
- IPropertyTree &fileProps = desc->queryProperties(); // properties of the logical file
- if (blockcompressed)
- {
- // caller has already set @size from file size...
- fileProps.setPropBool("@blockCompressed", true);
- fileProps.setPropInt64("@compressedSize", partProps.getPropInt64("@size", 0));
- partProps.setPropInt64("@compressedSize", partProps.getPropInt64("@size", 0));
- fileProps.setPropInt64("@size", uncompressedBytesWritten);
- partProps.setPropInt64("@size", uncompressedBytesWritten);
- }
- else if (tallycrc && crc.get())
- partProps.setPropInt64("@fileCrc", crc.get());
- if (encrypted)
- fileProps.setPropBool("@encrypted", true);
- fileProps.setPropInt64("@recordCount", processed);
- unsigned flags = helper.getFlags();
- if (flags & TDWpersist)
- fileProps.setPropBool("@persistent", true);
- if (grouped)
- fileProps.setPropBool("@grouped", true);
- if (flags & (TDWowned|TDXjobtemp|TDXtemporary))
- fileProps.setPropBool("@owned", true);
- if (flags & TDWresult)
- fileProps.setPropBool("@result", true);
- IConstWorkUnit *workUnit = ctx->queryWorkUnit();
- if (workUnit)
- {
- fileProps.setProp("@owner", workUnit->queryUser());
- fileProps.setProp("@workunit", workUnit->queryWuid());
- fileProps.setProp("@job", workUnit->queryJobName());
- }
- if (flags & TDWexpires)
- setExpiryTime(fileProps, helper.getExpiryDays());
- if (flags & TDWupdate)
- {
- unsigned eclCRC;
- unsigned __int64 totalCRC;
- helper.getUpdateCRCs(eclCRC, totalCRC);
- fileProps.setPropInt("@eclCRC", eclCRC);
- fileProps.setPropInt64("@totalCRC", totalCRC);
- }
- fileProps.setPropInt("@formatCrc", helper.getFormatCrc());
- IRecordSize * inputMeta = input->queryOutputMeta();
- if ((inputMeta->isFixedSize()) && !isOutputTransformed())
- fileProps.setPropInt("@recordSize", inputMeta->getFixedSize() + (grouped ? 1 : 0));
- const char *recordECL = helper.queryRecordECL();
- if (recordECL && *recordECL)
- fileProps.setProp("ECL", recordECL);
- fileProps.setProp("@kind", "flat"); // default, derivitives may override
- }
- virtual IUserDescriptor *queryUserDescriptor() const
- {
- IConstWorkUnit *workUnit = ctx->queryWorkUnit();
- if (workUnit)
- return workUnit->queryUserDescriptor();
- else
- return NULL;
- }
- virtual bool isOutputTransformed() const { return false; }
- };
- //=================================================================================
- class CRoxieServerCsvWriteActivity : public CRoxieServerDiskWriteActivity
- {
- IHThorCsvWriteArg &csvHelper;
- CSVOutputStream csvOutput;
- public:
- CRoxieServerCsvWriteActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerDiskWriteActivity(_ctx, _factory, _probeManager), csvHelper(static_cast<IHThorCsvWriteArg &>(helper))
- {
- csvOutput.init(csvHelper.queryCsvParameters(), false);
- }
- virtual void onExecute()
- {
- OwnedRoxieString header(csvHelper.queryCsvParameters()->getHeader());
- if (header)
- {
- csvOutput.beginLine();
- csvOutput.writeHeaderLn(strlen(header), header);
- diskout->write(csvOutput.length(), csvOutput.str());
- }
- loop
- {
- const void *nextrec = inputStream->ungroupedNextRow();
- if (!nextrec)
- break;
- processed++;
- csvOutput.beginLine();
- csvHelper.writeRow((const byte *)nextrec, &csvOutput);
- csvOutput.endLine();
- diskout->write(csvOutput.length(), csvOutput.str());
- ReleaseRoxieRow(nextrec);
- }
- OwnedRoxieString footer(csvHelper.queryCsvParameters()->getFooter());
- if (footer)
- {
- csvOutput.beginLine();
- csvOutput.writeHeaderLn(strlen(footer), footer);
- diskout->write(csvOutput.length(), csvOutput.str());
- }
- }
- virtual void setFileProperties(IFileDescriptor *desc) const
- {
- CRoxieServerDiskWriteActivity::setFileProperties(desc);
- IPropertyTree &props = desc->queryProperties();
- props.setProp("@format","utf8n");
- ICsvParameters *csvParameters = csvHelper.queryCsvParameters();
- StringBuffer separator;
- OwnedRoxieString rs(csvParameters->getSeparator(0));
- const char *s = rs;
- while (s && *s)
- {
- if (',' == *s)
- separator.append("\\,");
- else
- separator.append(*s);
- ++s;
- }
- props.setProp("@csvSeparate", separator.str());
- props.setProp("@csvQuote", rs.setown(csvParameters->getQuote(0)));
- props.setProp("@csvTerminate", rs.setown(csvParameters->getTerminator(0)));
- props.setProp("@csvEscape", rs.setown(csvParameters->getEscape(0)));
- props.setProp("@kind", "csv");
- }
- virtual bool isOutputTransformed() const { return true; }
- };
- class CRoxieServerXmlWriteActivity : public CRoxieServerDiskWriteActivity
- {
- IHThorXmlWriteArg &xmlHelper;
- StringAttr rowTag;
- ThorActivityKind kind;
- unsigned headerLength;
- unsigned footerLength;
- public:
- CRoxieServerXmlWriteActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, ThorActivityKind _kind)
- : CRoxieServerDiskWriteActivity(_ctx, _factory, _probeManager), xmlHelper(static_cast<IHThorXmlWriteArg &>(helper)), kind(_kind), headerLength(0), footerLength(0)
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerDiskWriteActivity::start(parentExtractSize, parentExtract, paused);
- OwnedRoxieString xmlpath(xmlHelper.getXmlIteratorPath());
- if (!xmlpath)
- rowTag.set(DEFAULTXMLROWTAG);
- else
- {
- const char *path = xmlpath;
- if (*path == '/') path++;
- if (strchr(path, '/')) UNIMPLEMENTED; // more what do we do with /mydata/row
- rowTag.set(path);
- }
- }
- virtual void onExecute()
- {
- StringBuffer header;
- OwnedRoxieString suppliedHeader(xmlHelper.getHeader());
- if (kind==TAKjsonwrite)
- buildJsonHeader(header, suppliedHeader, rowTag);
- else if (suppliedHeader)
- header.set(suppliedHeader);
- else
- header.append(DEFAULTXMLHEADER).newline();
- headerLength = header.length();
- diskout->write(header.length(), header.str());
- Owned<IXmlWriterExt> writer = createIXmlWriterExt(xmlHelper.getXmlFlags(), 0, NULL, (kind==TAKjsonwrite) ? WTJSON : WTStandard);
- writer->outputBeginArray(rowTag); //need to set this
- writer->clear(); //but not output it
- loop
- {
- OwnedConstRoxieRow nextrec = inputStream->ungroupedNextRow();
- if (!nextrec)
- break;
- processed++;
- writer->clear().outputBeginNested(rowTag, false);
- xmlHelper.toXML((const byte *)nextrec.get(), *writer);
- writer->outputEndNested(rowTag);
- diskout->write(writer->length(), writer->str());
- }
- OwnedRoxieString suppliedFooter(xmlHelper.getFooter());
- StringBuffer footer;
- if (kind==TAKjsonwrite)
- buildJsonFooter(footer.newline(), suppliedFooter, rowTag);
- else if (suppliedFooter)
- footer.append(suppliedFooter);
- else
- footer.append(DEFAULTXMLFOOTER);
- footerLength=footer.length();
- diskout->write(footer.length(), footer);
- }
- virtual void reset()
- {
- CRoxieServerDiskWriteActivity::reset();
- rowTag.clear();
- }
- virtual void setFileProperties(IFileDescriptor *desc) const
- {
- CRoxieServerDiskWriteActivity::setFileProperties(desc);
- desc->queryProperties().setProp("@format","utf8n");
- desc->queryProperties().setProp("@rowTag",rowTag.get());
- desc->queryProperties().setProp("@kind", (kind==TAKjsonwrite) ? "json" : "xml");
- desc->queryProperties().setPropInt(FPheaderLength, headerLength);
- desc->queryProperties().setPropInt(FPfooterLength, footerLength);
- }
- virtual bool isOutputTransformed() const { return true; }
- };
- class CRoxieServerDiskWriteActivityFactory : public CRoxieServerMultiOutputFactory
- {
- bool isRoot;
- bool isTemp;
- public:
- CRoxieServerDiskWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerMultiOutputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), isRoot(_isRoot)
- {
- Owned<IHThorDiskWriteArg> helper = (IHThorDiskWriteArg *) helperFactory();
- isTemp = (helper->getFlags() & TDXtemporary) != 0;
- setNumOutputs(helper->getTempUsageCount());
- if (_kind!=TAKdiskwrite)
- assertex(numOutputs == 0);
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- switch (numOutputs)
- {
- case 0:
- switch (kind)
- {
- case TAKdiskwrite: return new CRoxieServerDiskWriteActivity(_ctx, this, _probeManager);
- case TAKcsvwrite: return new CRoxieServerCsvWriteActivity(_ctx, this, _probeManager);
- case TAKxmlwrite:
- case TAKjsonwrite:
- return new CRoxieServerXmlWriteActivity(_ctx, this, _probeManager, kind);
- };
- throwUnexpected();
- case 1:
- return new CRoxieServerSpillWriteActivity(_ctx, this, _probeManager);
- default:
- return new CRoxieServerThroughSpillActivity(_ctx, this, _probeManager, numOutputs);
- }
- }
- virtual bool isSink() const
- {
- return numOutputs == 0 && !isTemp; // MORE - check with Gavin if this is right if not a temp but reread in same job...
- }
- };
- IRoxieServerActivityFactory *createRoxieServerDiskWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerDiskWriteActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //=================================================================================
- class CRoxieServerIndexWriteActivity : public CRoxieServerInternalSinkActivity, implements IRoxiePublishCallback
- {
- IHThorIndexWriteArg &helper;
- bool overwrite;
- Owned<ClusterWriteHandler> clusterHandler;
- Owned<IRoxieWriteHandler> writer;
- unsigned __int64 reccount;
- unsigned int fileCrc;
- StringBuffer filename;
- void updateWorkUnitResult()
- {
- // MORE - a lot of this is common with hthor
- WorkunitUpdate wu = ctx->updateWorkUnit();
- if (wu)
- {
- OwnedRoxieString rawLogicalName = helper.getFileName();
- StringBuffer lfn; // logical filename
- expandLogicalFilename(lfn, rawLogicalName, wu, false, false);
- if (lfn.length())
- {
- if (helper.getSequence() >= 0)
- {
- Owned<IWUResult> result = wu->updateResultBySequence(helper.getSequence());
- if (result)
- {
- result->setResultTotalRowCount(reccount);
- result->setResultStatus(ResultStatusCalculated);
- result->setResultLogicalName(lfn.str());
- }
- }
- if(clusterHandler)
- clusterHandler->finish(writer->queryFile());
- CTXLOG("Created roxie index file %s", lfn.str());
- }
- }
- }
- virtual void resolve()
- {
- StringArray clusters;
- unsigned clusterIdx = 0;
- while(true)
- {
- OwnedRoxieString cluster(helper.getCluster(clusterIdx));
- if(!cluster)
- break;
- clusters.append(cluster);
- clusterIdx++;
- }
- if (roxieName.length())
- clusters.append(roxieName.str());
- else
- clusters.append(".");
- OwnedRoxieString fname(helper.getFileName());
- writer.setown(ctx->createLFN(fname, overwrite, false, clusters)); // MORE - if there's a workunit, use if for scope.
- filename.set(writer->queryFile()->queryFilename());
- if (writer->queryFile()->exists())
- {
- if (overwrite)
- {
- CTXLOG("Removing existing %s from DFS",filename.str());
- writer->queryFile()->remove();
- }
- else
- throw MakeStringException(99, "Cannot write index file %s, file already exists (missing OVERWRITE attribute?)", filename.str());
- }
- }
- void buildUserMetadata(Owned<IPropertyTree> & metadata)
- {
- size32_t nameLen;
- char * nameBuff;
- size32_t valueLen;
- char * valueBuff;
- unsigned idx = 0;
- while(helper.getIndexMeta(nameLen, nameBuff, valueLen, valueBuff, idx++))
- {
- StringBuffer name(nameLen, nameBuff);
- StringBuffer value(valueLen, valueBuff);
- if(*nameBuff == '_' && strcmp(name, "_nodeSize") != 0)
- {
- OwnedRoxieString fname(helper.getFileName());
- throw MakeStringException(0, "Invalid name %s in user metadata for index %s (names beginning with underscore are reserved)", name.str(), fname.get());
- }
- if(!validateXMLTag(name.str()))
- {
- OwnedRoxieString fname(helper.getFileName());
- throw MakeStringException(0, "Invalid name %s in user metadata for index %s (not legal XML element name)", name.str(), fname.get());
- }
- if(!metadata)
- metadata.setown(createPTree("metadata"));
- metadata->setProp(name.str(), value.str());
- }
- }
- void buildLayoutMetadata(Owned<IPropertyTree> & metadata)
- {
- if(!metadata)
- metadata.setown(createPTree("metadata"));
- metadata->setProp("_record_ECL", helper.queryRecordECL());
- void * layoutMetaBuff;
- size32_t layoutMetaSize;
- if(helper.getIndexLayout(layoutMetaSize, layoutMetaBuff))
- {
- metadata->setPropBin("_record_layout", layoutMetaSize, layoutMetaBuff);
- rtlFree(layoutMetaBuff);
- }
- }
- public:
- CRoxieServerIndexWriteActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, 0), helper(static_cast<IHThorIndexWriteArg &>(basehelper))
- {
- overwrite = ((helper.getFlags() & TIWoverwrite) != 0);
- reccount = 0;
- fileCrc = 0;
- }
- ~CRoxieServerIndexWriteActivity()
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- resolve();
- }
- virtual void onExecute()
- {
- bool isVariable = helper.queryDiskRecordSize()->isVariableSize();
- size32_t maxDiskRecordSize;
- if (isVariable)
- {
- if (helper.getFlags() & TIWmaxlength)
- maxDiskRecordSize = helper.getMaxKeySize();
- else
- maxDiskRecordSize = KEYBUILD_MAXLENGTH; // Current default behaviour, could be improved in the future
- }
- else
- maxDiskRecordSize = helper.queryDiskRecordSize()->getFixedSize();
- if (maxDiskRecordSize > KEYBUILD_MAXLENGTH)
- throw MakeStringException(99, "Index maximum record length (%d) exceeds 32k internal limit", maxDiskRecordSize);
- OwnedMalloc<char> rowBuffer(maxDiskRecordSize, true);
- fileCrc = -1;
- {
- Owned<IFileIO> io;
- try
- {
- io.setown(writer->queryFile()->open(IFOcreate));
- }
- catch(IException * e)
- {
- e->Release();
- clearKeyStoreCache(false);
- io.setown(writer->queryFile()->open(IFOcreate));
- }
- if(!io)
- throw MakeStringException(errno, "Failed to create file %s for writing", filename.str());
- Owned<IFileIOStream> out = createIOStream(io);
- unsigned flags = COL_PREFIX | HTREE_FULLSORT_KEY;
- if (helper.getFlags() & TIWrowcompress)
- flags |= HTREE_COMPRESSED_KEY|HTREE_QUICK_COMPRESSED_KEY;
- else if (!(helper.getFlags() & TIWnolzwcompress))
- flags |= HTREE_COMPRESSED_KEY;
- if (isVariable)
- flags |= HTREE_VARSIZE;
- Owned<IPropertyTree> metadata;
- buildUserMetadata(metadata);
- buildLayoutMetadata(metadata);
- unsigned nodeSize = metadata ? metadata->getPropInt("_nodeSize", NODESIZE) : NODESIZE;
- Owned<IKeyBuilder> builder = createKeyBuilder(out, flags, maxDiskRecordSize, nodeSize, helper.getKeyedSize(), 0);
- class BcWrapper : implements IBlobCreator
- {
- IKeyBuilder *builder;
- public:
- BcWrapper(IKeyBuilder *_builder) : builder(_builder) {}
- virtual unsigned __int64 createBlob(size32_t size, const void * ptr)
- {
- return builder->createBlob(size, (const char *) ptr);
- }
- } bc(builder);
- // Loop thru the results
- loop
- {
- OwnedConstRoxieRow nextrec(inputStream->ungroupedNextRow());
- if (!nextrec)
- break;
- try
- {
- unsigned __int64 fpos=0;
- RtlStaticRowBuilder rowBuilder(rowBuffer, maxDiskRecordSize);
- size32_t thisSize = helper.transform(rowBuilder, nextrec, &bc, fpos);
- builder->processKeyData(rowBuffer, fpos, thisSize);
- }
- catch(IException * e)
- {
- throw makeWrappedException(e);
- }
- reccount++;
- }
- if(metadata)
- builder->finish(metadata,&fileCrc);
- else
- builder->finish(&fileCrc);
- }
- }
- virtual void stop()
- {
- if (writer)
- {
- if (!aborted)
- updateWorkUnitResult();
- writer->finish(!aborted, this);
- writer.clear();
- }
- CRoxieServerActivity::stop();
- }
- virtual void reset()
- {
- CRoxieServerActivity::reset();
- writer.clear();
- }
- //interface IRoxiePublishCallback
- virtual void setFileProperties(IFileDescriptor *desc) const
- {
- IPropertyTree &partProps = desc->queryPart(0)->queryProperties(); //properties of the first file part.
- IPropertyTree &fileProps = desc->queryProperties(); // properties of the logical file
- // Now publish to name services
- StringBuffer dir,base;
- offset_t indexFileSize = writer->queryFile()->size();
- if(clusterHandler)
- clusterHandler->splitPhysicalFilename(dir, base);
- else
- splitFilename(filename.str(), &dir, &dir, &base, &base);
- desc->setDefaultDir(dir.str());
- //properties of the first file part.
- Owned<IPropertyTree> attrs;
- if(clusterHandler)
- attrs.setown(createPTree("Part")); // clusterHandler is going to set attributes
- else
- {
- // add cluster
- StringBuffer mygroupname;
- desc->setNumParts(1);
- desc->setPartMask(base.str());
- attrs.set(&desc->queryPart(0)->queryProperties());
- }
- attrs->setPropInt64("@size", indexFileSize);
- CDateTime createTime, modifiedTime, accessedTime;
- writer->queryFile()->getTime(&createTime, &modifiedTime, &accessedTime);
- // round file time down to nearest sec. Nanosec accurancy is not preserved elsewhere and can lead to mismatch later.
- unsigned hour, min, sec, nanosec;
- modifiedTime.getTime(hour, min, sec, nanosec);
- modifiedTime.setTime(hour, min, sec, 0);
- StringBuffer timestr;
- modifiedTime.getString(timestr);
- if(timestr.length())
- attrs->setProp("@modified", timestr.str());
- if(clusterHandler)
- clusterHandler->setDescriptorParts(desc, base.str(), attrs);
- // properties of the logical file
- IPropertyTree & properties = desc->queryProperties();
- properties.setProp("@kind", "key");
- properties.setPropInt64("@size", indexFileSize);
- properties.setPropInt64("@recordCount", reccount);
- WorkunitUpdate workUnit = ctx->updateWorkUnit();
- if (workUnit)
- {
- properties.setProp("@owner", workUnit->queryUser());
- properties.setProp("@workunit", workUnit->queryWuid());
- properties.setProp("@job", workUnit->queryJobName());
- }
- char const * rececl = helper.queryRecordECL();
- if(rececl && *rececl)
- properties.setProp("ECL", rececl);
- if (helper.getFlags() & TIWexpires)
- setExpiryTime(properties, helper.getExpiryDays());
- if (helper.getFlags() & TIWupdate)
- {
- unsigned eclCRC;
- unsigned __int64 totalCRC;
- helper.getUpdateCRCs(eclCRC, totalCRC);
- properties.setPropInt("@eclCRC", eclCRC);
- properties.setPropInt64("@totalCRC", totalCRC);
- }
- properties.setPropInt("@fileCrc", fileCrc);
- properties.setPropInt("@formatCrc", helper.getFormatCrc());
- void * layoutMetaBuff;
- size32_t layoutMetaSize;
- if(helper.getIndexLayout(layoutMetaSize, layoutMetaBuff))
- {
- properties.setPropBin("_record_layout", layoutMetaSize, layoutMetaBuff);
- rtlFree(layoutMetaBuff);
- }
- }
- IUserDescriptor *queryUserDescriptor() const
- {
- IConstWorkUnit *workUnit = ctx->queryWorkUnit();
- if (workUnit)
- return workUnit->queryUserDescriptor();
- else
- return NULL;
- }
- };
- //=================================================================================
- class CRoxieServerIndexWriteActivityFactory : public CRoxieServerMultiOutputFactory
- {
- public:
- CRoxieServerIndexWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerMultiOutputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- setNumOutputs(0);
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerIndexWriteActivity(_ctx, this, _probeManager);
- }
- virtual bool isSink() const
- {
- return true;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerIndexWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerIndexWriteActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //=================================================================================
- static inline void getLimitType(unsigned flags, bool & limitFail, bool & limitOnFail)
- {
- if((flags & JFmatchAbortLimitSkips) != 0)
- {
- limitFail = false;
- limitOnFail = false;
- }
- else
- {
- limitOnFail = ((flags & JFonfail) != 0);
- limitFail = !limitOnFail;
- }
- }
- class CRoxieServerJoinActivity : public CRoxieServerTwoInputActivity
- {
- enum { JSfill, JSfillleft, JSfillright, JScollate, JScompare, JSleftonly, JSrightonly } state;
- IHThorJoinArg &helper;
- ICompare * collate;
- ICompare * collateupper;
- ThorActivityKind activityKind;
- bool leftOuterJoin;
- bool rightOuterJoin;
- bool exclude;
- bool limitFail;
- bool limitOnFail;
- bool forceSpill;
- unsigned keepLimit;
- unsigned joinLimit;
- unsigned atmostLimit;
- unsigned abortLimit;
- unsigned atmostsTriggered;
- bool betweenjoin;
- OwnedRowArray right;
- const void * left;
- const void * pendingRight;
- unsigned rightIndex;
- unsigned joinCounter;
- BoolArray matchedRight;
- bool matchedLeft;
- Owned<IException> failingLimit;
- ConstPointerArray filteredRight;
- Owned<IRHLimitedCompareHelper> limitedhelper;
- OwnedConstRoxieRow defaultLeft;
- OwnedConstRoxieRow defaultRight;
- Owned<IEngineRowAllocator> defaultLeftAllocator;
- Owned<IEngineRowAllocator> defaultRightAllocator;
- Owned<IGroupedInput> sortedLeft;
- Owned<IGroupedInput> groupedSortedRight;
- bool cloneLeft;
- void createDefaultLeft()
- {
- if (!defaultLeft)
- {
- if (!defaultLeftAllocator)
- defaultLeftAllocator.setown(createRowAllocator(input->queryOutputMeta()));
- RtlDynamicRowBuilder rowBuilder(defaultLeftAllocator);
- size32_t thisSize = helper.createDefaultLeft(rowBuilder);
- defaultLeft.setown(rowBuilder.finalizeRowClear(thisSize));
- }
- }
- void createDefaultRight()
- {
- if (!defaultRight)
- {
- if (!defaultRightAllocator)
- defaultRightAllocator.setown(createRowAllocator(input1->queryOutputMeta()));
- RtlDynamicRowBuilder rowBuilder(defaultRightAllocator);
- size32_t thisSize = helper.createDefaultRight(rowBuilder);
- defaultRight.setown(rowBuilder.finalizeRowClear(thisSize));
- }
- }
- public:
- CRoxieServerJoinActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, bool _forceSpill)
- : CRoxieServerTwoInputActivity(_ctx, _factory, _probeManager), helper((IHThorJoinArg &)basehelper), forceSpill(_forceSpill)
- {
- // MORE - some of this should be done in factory
- unsigned joinFlags = helper.getJoinFlags();
- leftOuterJoin = (joinFlags & JFleftouter) != 0;
- rightOuterJoin = (joinFlags & JFrightouter) != 0;
- exclude = (joinFlags & JFexclude) != 0;
- cloneLeft = (joinFlags & JFtransformmatchesleft) != 0;
- getLimitType(joinFlags, limitFail, limitOnFail);
- if (joinFlags & JFslidingmatch)
- {
- betweenjoin = true;
- collate = helper.queryCompareLeftRightLower();
- collateupper = helper.queryCompareLeftRightUpper();
- }
- else
- {
- betweenjoin = false;
- collate = collateupper = helper.queryCompareLeftRight();
- }
- rightIndex = 0;
- joinCounter = 0;
- state = JSfill;
- matchedLeft = false;
- joinLimit = 0;
- keepLimit = 0; // wait until ctx available
- atmostLimit = 0; // wait until ctx available
- abortLimit = 0; // wait until ctx available
- atmostsTriggered = 0;
- assertex((joinFlags & (JFfirst | JFfirstleft | JFfirstright)) == 0);
- left = NULL;
- pendingRight = NULL;
- activityKind = _factory->getKind();
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- left = NULL;
- rightIndex = 0;
- joinCounter = 0;
- state = JSfill;
- matchedLeft = false;
- CRoxieServerTwoInputActivity::start(parentExtractSize, parentExtract, paused);
- keepLimit = helper.getKeepLimit();
- if (keepLimit == 0)
- keepLimit = (unsigned)-1;
- atmostsTriggered = 0;
- atmostLimit = helper.getJoinLimit();
- if(atmostLimit == 0)
- atmostLimit = (unsigned)-1;
- else
- assertex(!rightOuterJoin && !betweenjoin);
- abortLimit = helper.getMatchAbortLimit();
- if (abortLimit == 0)
- abortLimit = (unsigned)-1;
- if (rightOuterJoin)
- createDefaultLeft();
- if ((leftOuterJoin && (activityKind==TAKjoin || activityKind==TAKjoinlight || activityKind==TAKdenormalizegroup)) || limitOnFail)
- createDefaultRight();
- bool isStable = (helper.getJoinFlags() & JFunstable) == 0;
- RoxieSortAlgorithm sortAlgorithm;
- if (forceSpill)
- sortAlgorithm = isStable ? stableSpillingQuickSortAlgorithm : spillingQuickSortAlgorithm;
- else
- sortAlgorithm = isStable ? stableQuickSortAlgorithm : quickSortAlgorithm;
- if (helper.isLeftAlreadySorted())
- sortedLeft.setown(createDegroupedInputReader(inputStream));
- else
- sortedLeft.setown(createSortedInputReader(inputStream, createSortAlgorithm(sortAlgorithm, helper.queryCompareLeft(), ctx->queryRowManager(), input->queryOutputMeta(), ctx->queryCodeContext(), tempDirectory, activityId)));
- ICompare *compareRight = helper.queryCompareRight();
- if (helper.isRightAlreadySorted())
- groupedSortedRight.setown(createGroupedInputReader(inputStream1, compareRight));
- else
- groupedSortedRight.setown(createSortedGroupedInputReader(inputStream1, compareRight, createSortAlgorithm(sortAlgorithm, compareRight, ctx->queryRowManager(), input1->queryOutputMeta(), ctx->queryCodeContext(), tempDirectory, activityId)));
- if ((helper.getJoinFlags() & JFlimitedprefixjoin) && helper.getJoinLimit())
- { //limited match join (s[1..n])
- limitedhelper.setown(createRHLimitedCompareHelper());
- limitedhelper->init( helper.getJoinLimit(), groupedSortedRight, collate, helper.queryPrefixCompare() );
- }
- }
- virtual void reset()
- {
- if (atmostsTriggered)
- noteStatistic(StNumAtmostTriggered, atmostsTriggered);
- right.clear();
- ReleaseClearRoxieRow(left);
- ReleaseClearRoxieRow(pendingRight);
- defaultRight.clear();
- defaultLeft.clear();
- sortedLeft.clear();
- groupedSortedRight.clear();
- CRoxieServerTwoInputActivity::reset();
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- if (!idx && (helper.getJoinFlags() & JFparallel) != 0)
- {
- puller.setown(new CRoxieServerReadAheadInput(ctx, ctx->queryOptions().parallelJoinPreload));
- puller->setInput(0, _sourceIdx, _in);
- _in = puller;
- _sourceIdx = 0;
- }
- CRoxieServerTwoInputActivity::setInput(idx, _sourceIdx, _in);
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- if (idx==(unsigned)-1)
- idx = 0;
- return idx ? NULL : this;
- }
- void fillLeft()
- {
- matchedLeft = false;
- left = sortedLeft->nextRow(); // NOTE: already degrouped by the IGroupedInput we attached
- if (betweenjoin && left && pendingRight && (collate->docompare(left, pendingRight) >= 0))
- fillRight();
- if (limitedhelper && 0==rightIndex)
- {
- rightIndex = 0;
- joinCounter = 0;
- right.clear();
- matchedRight.kill();
- if (left)
- {
- limitedhelper->getGroup(right,left);
- ForEachItemIn(idx, right)
- matchedRight.append(false);
- }
- }
- }
- void fillRight()
- {
- if (limitedhelper)
- return;
- failingLimit.clear();
- if(betweenjoin && left)
- {
- aindex_t start = 0;
- while(right.isItem(start) && (collateupper->docompare(left, right.item(start)) > 0))
- start++;
- if(start>0)
- right.clearPart(0, start);
- }
- else
- right.clear();
- rightIndex = 0;
- joinCounter = 0;
- unsigned groupCount = 0;
- const void * next;
- while(true)
- {
- if(pendingRight)
- {
- next = pendingRight;
- pendingRight = NULL;
- }
- else
- {
- next = groupedSortedRight->nextRow();
- }
- if(!rightOuterJoin && next && (!left || (collateupper->docompare(left, next) > 0))) // if right is less than left, and not right outer, can skip group
- {
- while(next)
- {
- ReleaseClearRoxieRow(next);
- next = groupedSortedRight->nextRow();
- }
- continue;
- }
- while(next)
- {
- if(groupCount==abortLimit)
- {
- if(limitFail)
- failLimit();
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- if(limitOnFail)
- {
- assertex(!failingLimit);
- try
- {
- failLimit();
- }
- catch(IException * except)
- {
- failingLimit.setown(except);
- }
- assertex(failingLimit != NULL);
- }
- right.append(next);
- do
- {
- next = groupedSortedRight->nextRow();
- ReleaseRoxieRow(next);
- } while(next);
- break;
- }
- else if(groupCount==atmostLimit)
- {
- atmostsTriggered++;
- right.clear();
- groupCount = 0;
- while(next)
- {
- ReleaseRoxieRow(next);
- next = groupedSortedRight->nextRow();
- }
- }
- else
- {
- right.append(next);
- groupCount++;
- }
- next = groupedSortedRight->nextRow();
- }
- // normally only want to read one right group, but if is between join and next right group is in window for left, need to continue
- if(betweenjoin && left)
- {
- pendingRight = groupedSortedRight->nextRow();
- if(!pendingRight || (collate->docompare(left, pendingRight) < 0))
- break;
- }
- else
- break;
- }
- matchedRight.kill();
- ForEachItemIn(idx, right)
- matchedRight.append(false);
- }
- const void * joinRecords(const void * curLeft, const void * curRight, unsigned counter)
- {
- if (cloneLeft)
- {
- LinkRoxieRow(curLeft);
- return curLeft;
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t thisSize = helper.transform(rowBuilder, curLeft, curRight, counter);
- if (thisSize)
- return rowBuilder.finalizeRowClear(thisSize);
- else
- return NULL;
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- const void * denormalizeRecords(const void * curLeft, ConstPointerArray & rows)
- {
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned numRows = rows.ordinality();
- const void * right = numRows ? rows.item(0) : defaultRight.get();
- size32_t thisSize = helper.transform(rowBuilder, curLeft, right, numRows, (const void * *)rows.getArray());
- if (thisSize)
- return rowBuilder.finalizeRowClear(thisSize);
- else
- return NULL;
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- const void * joinException(const void * curLeft, IException * except)
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t thisSize = helper.onFailTransform(rowBuilder, curLeft, defaultRight, except);
- return rowBuilder.finalizeRowClear(thisSize);
- }
- void failLimit()
- {
- helper.onMatchAbortLimitExceeded();
- CommonXmlWriter xmlwrite(0);
- if (input->queryOutputMeta() && input->queryOutputMeta()->hasXML())
- {
- input->queryOutputMeta()->toXML((byte *) left, xmlwrite);
- }
- throw MakeStringException(ROXIE_TOO_MANY_RESULTS, "More than %d match candidates in join %d for row %s", abortLimit, queryId(), xmlwrite.str());
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- switch (state)
- {
- case JSfill:
- fillLeft();
- state = JSfillright;
- break;
- case JSfillright:
- fillRight();
- state = JScollate;
- break;
- case JSfillleft:
- fillLeft();
- state = JScollate;
- break;
- case JScollate:
- if (right.ordinality() == 0)
- {
- if (left == NULL)
- return NULL;
- state = JSleftonly;
- }
- else
- {
- if (!left)
- state = JSrightonly;
- else
- {
- int diff;
- if(betweenjoin)
- diff = ((collate->docompare(left, right.item(0)) < 0) ? -1 : ((collateupper->docompare(left, right.item(right.ordinality()-1)) > 0) ? +1 : 0));
- else
- diff = collate->docompare(left, right.item(0));
- bool limitExceeded = right.ordinality()>abortLimit;
- if (diff == 0)
- {
- if (limitExceeded)
- {
- const void * ret = NULL;
- if(failingLimit)
- ret = joinException(left, failingLimit);
- ReleaseRoxieRow(left);
- left = NULL;
- state = JSfillleft;
- ForEachItemIn(idx, right)
- matchedRight.replace(true, idx);
- if(ret)
- {
- processed++;
- return ret;
- }
- }
- else
- {
- state = JScompare;
- joinLimit = keepLimit;
- }
- }
- else if (diff < 0)
- state = JSleftonly;
- else if (limitExceeded)
- {
- // MORE - Roxie code seems to think there should be a destroyRowset(right) here....
- state = JSfillright;
- }
- else
- state = JSrightonly;
- }
- }
- break;
- case JSrightonly:
- if (rightOuterJoin)
- {
- switch (activityKind)
- {
- case TAKjoin:
- {
- while (right.isItem(rightIndex))
- {
- if (!matchedRight.item(rightIndex))
- {
- const void * rhs = right.item(rightIndex++);
- const void *ret = joinRecords(defaultLeft, rhs, 0);
- if (ret)
- {
- processed++;
- return ret;
- }
- }
- else
- rightIndex++;
- }
- break;
- }
- //Probably excessive to implement the following, but possibly useful
- case TAKdenormalize:
- {
- OwnedConstRoxieRow newLeft;
- newLeft.set(defaultLeft);
- unsigned rowSize = 0;
- unsigned leftCount = 0;
- while (right.isItem(rightIndex))
- {
- if (!matchedRight.item(rightIndex))
- {
- const void * rhs = right.item(rightIndex);
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned thisSize = helper.transform(rowBuilder, newLeft, rhs, ++leftCount);
- if (thisSize)
- {
- rowSize = thisSize;
- newLeft.setown(rowBuilder.finalizeRowClear(rowSize));
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- rightIndex++;
- }
- state = JSfillright;
- if (rowSize)
- {
- processed++;
- return newLeft.getClear();
- }
- break;
- }
- case TAKdenormalizegroup:
- {
- filteredRight.kill();
- while (right.isItem(rightIndex))
- {
- if (!matchedRight.item(rightIndex))
- filteredRight.append(right.item(rightIndex));
- rightIndex++;
- }
- state = JSfillright;
- if (filteredRight.ordinality())
- {
- const void * ret = denormalizeRecords(defaultLeft, filteredRight);
- filteredRight.kill();
- if (ret)
- {
- processed++;
- return ret;
- }
- }
- break;
- }
- }
- }
- state = JSfillright;
- break;
-
- case JSleftonly:
- {
- const void * ret = NULL;
- if (!matchedLeft && leftOuterJoin)
- {
- switch (activityKind)
- {
- case TAKjoin:
- ret = joinRecords(left, defaultRight, 0);
- break;
- case TAKdenormalize:
- ret = left;
- left = NULL;
- break;
- case TAKdenormalizegroup:
- filteredRight.kill();
- ret = denormalizeRecords(left, filteredRight);
- break;
- }
- }
- ReleaseRoxieRow(left);
- left = NULL;
- state = JSfillleft;
- if (ret)
- {
- processed++;
- return ret;
- }
- break;
- }
- case JScompare:
- if (joinLimit != 0)
- {
- switch (activityKind)
- {
- case TAKjoin:
- {
- while (right.isItem(rightIndex))
- {
- const void * rhs = right.item(rightIndex++);
- if (helper.match(left, rhs))
- {
- matchedRight.replace(true, rightIndex-1);
- matchedLeft = true;
- if (!exclude)
- {
- const void *ret = joinRecords(left, rhs, ++joinCounter);
- if (ret)
- {
- processed++;
- joinLimit--;
- return ret;
- }
- }
- }
- }
- break;
- }
- case TAKdenormalize:
- {
- OwnedConstRoxieRow newLeft;
- newLeft.set(left);
- unsigned rowSize = 0;
- unsigned leftCount = 0;
- while (right.isItem(rightIndex) && joinLimit)
- {
- try
- {
- const void * rhs = right.item(rightIndex++);
- if (helper.match(left, rhs))
- {
- matchedRight.replace(true, rightIndex-1);
- matchedLeft = true;
- if (!exclude)
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned thisSize = helper.transform(rowBuilder, newLeft, rhs, ++leftCount);
- if (thisSize)
- {
- rowSize = thisSize;
- newLeft.setown(rowBuilder.finalizeRowClear(rowSize));
- joinLimit--;
- }
- }
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- state = JSleftonly;
- rightIndex = 0;
- if (rowSize)
- {
- processed++;
- return newLeft.getClear();
- }
- break;
- }
- case TAKdenormalizegroup:
- {
- filteredRight.kill();
- while (right.isItem(rightIndex))
- {
- const void * rhs = right.item(rightIndex++);
- if (helper.match(left, rhs))
- {
- matchedRight.replace(true, rightIndex-1);
- filteredRight.append(rhs);
- matchedLeft = true;
- if (filteredRight.ordinality()==joinLimit)
- break;
- }
- }
- state = JSleftonly;
- rightIndex = 0;
- if (!exclude && filteredRight.ordinality())
- {
- const void * ret = denormalizeRecords(left, filteredRight);
- filteredRight.kill();
- if (ret)
- {
- processed++;
- return ret;
- }
- }
- break;
- }
- }
- }
- state = JSleftonly;
- rightIndex = 0;
- joinCounter = 0;
- break;
- }
- }
- }
- };
- class CRoxieServerJoinActivityFactory : public CRoxieServerActivityFactory
- {
- unsigned input2;
- unsigned input2idx;
- bool forceSpill;
- public:
- CRoxieServerJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- forceSpill = _graphNode.getPropBool("hint[@name='spill']/@value", _queryFactory.queryOptions().allSortsMaySpill);
- input2 = 0;
- input2idx = 0;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerJoinActivity(_ctx, this, _probeManager, forceSpill);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- if (idx==1)
- {
- input2 = source;
- input2idx = sourceidx;
- }
- else
- CRoxieServerActivityFactory::setInput(idx, source, sourceidx);
- }
- virtual unsigned getInput(unsigned idx, unsigned &sourceidx) const
- {
- switch (idx)
- {
- case 1:
- sourceidx = input2idx;
- return input2;
- case 0:
- return CRoxieServerActivityFactory::getInput(idx, sourceidx);
- default:
- return (unsigned) -1;
- }
- }
- virtual unsigned numInputs() const { return 2; }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return joinStatistics;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerJoinActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- #define CONCAT_READAHEAD 1000
- class CRoxieThreadedConcatReader : implements IRecordPullerCallback, public CInterface
- {
- public:
- IMPLEMENT_IINTERFACE;
- CRoxieThreadedConcatReader(InterruptableSemaphore &_ready, bool _grouped)
- : puller(false), grouped(_grouped), atEog(true), ready(_ready), eof(false)
- {
- }
- void start(unsigned parentExtractSize, const byte *parentExtract, bool paused, IRoxieSlaveContext *ctx)
- {
- space.reinit(CONCAT_READAHEAD);
- puller.start(parentExtractSize, parentExtract, paused, ctx->queryOptions().concatPreload, false, ctx);
- }
- void stop()
- {
- space.interrupt();
- puller.stop();
- }
- IFinalRoxieInput *queryInput() const
- {
- return puller.queryInput();
- }
- void reset()
- {
- puller.reset();
- ForEachItemIn(idx, buffer)
- ReleaseRoxieRow(buffer.item(idx));
- buffer.clear();
- eof = false;
- atEog = true;
- }
- void setInput(unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- puller.setInput(this, _sourceIdx, _in);
- }
- void connectInputStreams(IRoxieSlaveContext *ctx, bool consumerOrdered)
- {
- puller.connectInputStreams(ctx, consumerOrdered);
- }
- virtual void processRow(const void *row)
- {
- buffer.enqueue(row);
- ready.signal();
- space.wait();
- }
- virtual void processGroup(const ConstPointerArray &rows)
- {
- // We use record-by-record input mode of the puller thread even in grouped mode.
- throwUnexpected();
- }
- virtual void processEOG()
- {
- if (grouped)
- processRow(NULL);
- }
- virtual void processDone()
- {
- processRow(NULL);
- }
- virtual bool fireException(IException *e)
- {
- // called from puller thread on failure
- ready.interrupt(LINK(e));
- space.interrupt(e);
- return true;
- }
- bool peek(const void * &row, bool &anyActive)
- {
- if (!eof)
- {
- if (buffer.ordinality())
- {
- space.signal();
- row = buffer.dequeue();
- if (row==NULL)
- {
- if (atEog)
- {
- eof = true;
- return false;
- }
- else
- atEog = true;
- }
- else if (grouped)
- atEog = false;
- return true;
- }
- anyActive = true;
- }
- return false;
- }
- protected:
- RecordPullerThread puller;
- InterruptableSemaphore space;
- InterruptableSemaphore &ready;
- SafeQueueOf<const void, true> buffer;
- bool atEog;
- bool eof;
- bool grouped;
- };
- typedef CopyReferenceArrayOf<CRoxieThreadedConcatReader> ReaderArray;
- class CRoxieServerThreadedConcatActivity : public CRoxieServerActivity
- {
- InterruptableSemaphore ready;
- ReaderArray pullers;
- unsigned numInputs;
- unsigned nextPuller; // for round robin
- unsigned readyPending;
- bool eof;
- bool inGroup;
- bool grouped;
- public:
- CRoxieServerThreadedConcatActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, bool _grouped, unsigned _numInputs)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), grouped(_grouped)
- {
- numInputs = _numInputs;
- eof = (numInputs==0);
- inGroup = false;
- nextPuller = 0;
- readyPending = 0;
- for (unsigned i = 0; i < numInputs; i++)
- pullers.append(*new CRoxieThreadedConcatReader(ready, _grouped));
- }
- ~CRoxieServerThreadedConcatActivity()
- {
- ForEachItemIn(idx, pullers)
- delete &pullers.item(idx);
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = (numInputs==0);
- inGroup = false;
- nextPuller = 0;
- readyPending = 0;
- ready.reinit();
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- ForEachItemIn(idx, pullers)
- {
- pullers.item(idx).start(parentExtractSize, parentExtract, paused, ctx);
- // NOTE - it is ok to start the thread running while parts of the subgraph are still being started, since everything
- // in the part of the subgraph that the thread uses has been started.
- // Note that splitters are supposed to cope with being used when only some outputs have been started.
- }
- }
- virtual void stop()
- {
- ready.interrupt();
- ForEachItemIn(idx, pullers)
- pullers.item(idx).stop();
- CRoxieServerActivity::stop();
- }
- virtual unsigned __int64 queryLocalCycles() const
- {
- return 0;
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (pullers.isItem(idx))
- return pullers.item(idx).queryInput();
- else
- return NULL;
- }
- virtual void reset()
- {
- CRoxieServerActivity::reset();
- ForEachItemIn(idx, pullers)
- pullers.item(idx).reset();
- eof = false;
- inGroup = false;
- nextPuller = 0;
- readyPending = 0;
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- if (pullers.isItem(idx))
- pullers.item(idx).setInput(_sourceIdx, _in);
- else
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() parameter out of bounds at %s(%d)", __FILE__, __LINE__);
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- ForEachItemIn(i, pullers)
- pullers.item(i).connectInputStreams(ctx, consumerOrdered);
- return CRoxieServerActivity::getOutputStreams(ctx, idx, streams, nullptr, consumerOrdered, nullptr); // No input actually connected
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- loop
- {
- if (readyPending && !inGroup)
- {
- if (readyPending > 1)
- ready.signal(readyPending-1);
- readyPending = 0;
- }
- else
- ready.wait();
- bool anyActive = false;
- ForEachItemIn(unused_index, pullers)
- {
- // NOTE - we round robin not just because it's more efficient, but because it ensures the preservation of grouping information
- const void *ret;
- bool fetched = pullers.item(nextPuller).peek(ret, anyActive);
- if (fetched)
- {
- inGroup = (ret != NULL);
- return ret;
- }
- if (inGroup && grouped)
- {
- // Some other puller has data, but we can't consume it until the group we are reading is complete.
- readyPending++;
- anyActive = true;
- break;
- }
- nextPuller++;
- if (nextPuller==pullers.ordinality())
- nextPuller = 0;
- }
- if (!anyActive)
- break;
- // A ready signal without anything being ready means someone reached end-of-file.
- }
- eof = true;
- return NULL;
- }
- };
- class CRoxieServerOrderedConcatActivity : public CRoxieServerMultiInputActivity
- {
- IEngineRowStream *curStream;
- bool eogSeen;
- bool anyThisGroup;
- bool grouped;
- unsigned streamIdx;
- public:
- CRoxieServerOrderedConcatActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, bool _grouped, unsigned _numInputs)
- : CRoxieServerMultiInputActivity(_ctx, _factory, _probeManager, _numInputs)
- {
- eogSeen = false;
- anyThisGroup = false;
- grouped = _grouped;
- streamIdx = 0;
- curStream = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerMultiInputActivity::start(parentExtractSize, parentExtract, paused);
- streamIdx = 0;
- curStream = streamArray[streamIdx];
- eogSeen = false;
- anyThisGroup = false;
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (!curStream)
- return NULL; // eof
- const void * next = curStream->nextRow();
- if (next)
- {
- anyThisGroup = true;
- eogSeen = false;
- processed++;
- return next;
- }
- else if (!eogSeen)
- {
- eogSeen = true;
- if (grouped)
- {
- if (anyThisGroup)
- {
- anyThisGroup = false;
- return NULL;
- }
- else
- return nextRow();
- }
- else
- return nextRow();
- }
- else if (streamIdx < numStreams-1)
- {
- streamIdx++;
- curStream = streamArray[streamIdx];
- eogSeen = false;
- return nextRow();
- }
- else
- {
- curStream = NULL;
- return NULL;
- }
- }
- };
- class CRoxieServerConcatActivityFactory : public CRoxieServerMultiInputFactory
- {
- bool ordered;
- bool grouped;
- public:
- CRoxieServerConcatActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerMultiInputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- Owned <IHThorFunnelArg> helper = (IHThorFunnelArg *) helperFactory();
- ordered = helper->isOrdered();
- grouped = helper->queryOutputMeta()->isGrouped();
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- if (ordered || (_probeManager && _probeManager->queryDebugManager()))
- return new CRoxieServerOrderedConcatActivity(_ctx, this, _probeManager, grouped, numInputs());
- else
- return new CRoxieServerThreadedConcatActivity(_ctx, this, _probeManager, grouped, numInputs());
- }
- };
- IRoxieServerActivityFactory *createRoxieServerConcatActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerConcatActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerNonEmptyActivity : public CRoxieServerMultiInputBaseActivity
- {
- IEngineRowStream* selectedStream;
- unsigned savedParentExtractSize;
- const byte * savedParentExtract;
- bool foundInput;
- public:
- CRoxieServerNonEmptyActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerMultiInputBaseActivity(_ctx, _factory, _probeManager, _numInputs)
- {
- foundInput = false;
- selectedStream = NULL;
- savedParentExtractSize = 0;;
- savedParentExtract = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- //Don't start the inputs yet so we can short-circuit...
- CRoxieServerMultiInputBaseActivity::start(parentExtractSize, parentExtract, paused);
- savedParentExtractSize = parentExtractSize;
- savedParentExtract = parentExtract;
- }
- virtual void stop()
- {
- if (foundInput)
- {
- if (selectedStream)
- selectedStream->stop();
- }
- else
- {
- for (unsigned i = 0; i < numStreams; i++)
- streamArray[i]->stop();
- }
- CRoxieServerMultiInputBaseActivity::stop();
- }
- virtual void reset()
- {
- CRoxieServerMultiInputBaseActivity::reset();
- foundInput = false;
- selectedStream = NULL;
- }
- virtual unsigned __int64 queryLocalCycles() const
- {
- return 0; // Can't easily calculate anything reliable but local processing is negligible
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (!foundInput)
- {
- foundInput = true;
- assertex(numInputs == numStreams); // If this ceases to be true we will need to rethink some of this
- //If we get an exception in this loop then stop() will stop any started inputs
- for (unsigned i=0; i < numStreams; i++)
- {
- selectedStream = streamArray[i];
- inputArray[i]->start(savedParentExtractSize, savedParentExtract, false); // Assumes 1:1 mapping streams to inputs
- startJunction(junctionArray[i]);
- const void * next = selectedStream->nextRow();
- if (next)
- {
- //Found a row so stop remaining
- for (unsigned j=i+1; j < numInputs; j++)
- streamArray[j]->stop();
- processed++;
- return next;
- }
- selectedStream->stop();
- }
- selectedStream = NULL;
- return NULL;
- }
- if (!selectedStream)
- return NULL;
- const void * next = selectedStream->nextRow();
- if (next)
- processed++;
- return next;
- }
- };
- class CRoxieServerNonEmptyActivityFactory : public CRoxieServerMultiInputFactory
- {
- public:
- CRoxieServerNonEmptyActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerMultiInputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerNonEmptyActivity(_ctx, this, _probeManager, numInputs());
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNonEmptyActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerNonEmptyActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerMergeActivity : public CRoxieServerMultiInputActivity
- {
- IHThorMergeArg &helper;
- unsigned *mergeheap;
- unsigned activeInputs;
- const void **pending;
- bool first;
- ICompare *compare;
- bool dedup;
- void permute()
- {
- assertex(activeInputs == 0);
- for(unsigned i = 0; i < numStreams; i++)
- if(pullInput(i))
- mergeheap[activeInputs++] = i;
- // the tree structure: element p has children p*2+1 and p*2+2, or element c has parent (unsigned)(c-1)/2
- // the heap property: no element should be smaller than its parent
- // the dedup variant: if(dedup), the top of the heap should also not be equal to either child
- // the method: establish this by starting with the parent of the bottom element and working up to the top element, sifting each down to its correct place
- if (activeInputs >= 2)
- for(unsigned p = (activeInputs-2)/2; p > 0; --p)
- siftDown(p);
- if(dedup)
- siftDownDedupTop();
- else
- siftDown(0);
- }
- void readNext()
- {
- if(!pullInput(mergeheap[0]))
- if(!promote(0))
- return;
- // we have changed the element at the top of the heap, so need to sift it down to maintain the heap property
- if(dedup)
- siftDownDedupTop();
- else
- siftDown(0);
- }
- bool pullInput(unsigned i)
- {
- const void *next = streamArray[i]->ungroupedNextRow();
- pending[i] = next;
- return (next != NULL);
- }
- bool promote(unsigned p)
- {
- activeInputs--;
- if(activeInputs == p)
- return false;
- mergeheap[p] = mergeheap[activeInputs];
- return true;
- }
- bool siftDown(unsigned p)
- {
- // assuming that all descendants of p form a heap, sift p down to its correct position, and so include it in the heap
- bool nochange = true;
- while(1)
- {
- unsigned c = p*2 + 1;
- if(c >= activeInputs)
- return nochange;
- if(c+1 < activeInputs)
- {
- int childcmp = BuffCompare(c+1, c);
- if((childcmp < 0) || ((childcmp == 0) && (mergeheap[c+1] < mergeheap[c])))
- ++c;
- }
- int cmp = BuffCompare(c, p);
- if((cmp > 0) || ((cmp == 0) && (mergeheap[c] > mergeheap[p])))
- return nochange;
- nochange = false;
- unsigned r = mergeheap[c];
- mergeheap[c] = mergeheap[p];
- mergeheap[p] = r;
- p = c;
- }
- }
- void siftDownDedupTop()
- {
- // same as siftDown(0), except that it also ensures that the top of the heap is not equal to either of its children
- if(activeInputs < 2)
- return;
- unsigned c = 1;
- int childcmp = 1;
- if(activeInputs >= 3)
- {
- childcmp = BuffCompare(2, 1);
- if(childcmp < 0)
- c = 2;
- }
- int cmp = BuffCompare(c, 0);
- if(cmp > 0)
- return;
- // the following loop ensures the correct property holds on the smaller branch, and that childcmp==0 iff the top matches the other branch
- while(cmp <= 0)
- {
- if(cmp == 0)
- {
- if(mergeheap[c] < mergeheap[0])
- {
- unsigned r = mergeheap[c];
- mergeheap[c] = mergeheap[0];
- mergeheap[0] = r;
- }
- ReleaseClearRoxieRow(pending[mergeheap[c]]);
- if(!pullInput(mergeheap[c]))
- if(!promote(c))
- break;
- siftDown(c);
- }
- else
- {
- unsigned r = mergeheap[c];
- mergeheap[c] = mergeheap[0];
- mergeheap[0] = r;
- if(siftDown(c))
- break;
- }
- cmp = BuffCompare(c, 0);
- }
- // the following loop ensures the uniqueness property holds on the other branch too
- c = 3-c;
- if(activeInputs <= c)
- return;
- while(childcmp == 0)
- {
- if(mergeheap[c] < mergeheap[0])
- {
- unsigned r = mergeheap[c];
- mergeheap[c] = mergeheap[0];
- mergeheap[0] = r;
- }
- ReleaseClearRoxieRow(pending[mergeheap[c]]);
- if(!pullInput(mergeheap[c]))
- if(!promote(c))
- break;
- siftDown(c);
- childcmp = BuffCompare(c, 0);
- }
- }
- inline int BuffCompare(unsigned a, unsigned b)
- {
- return compare->docompare(pending[mergeheap[a]], pending[mergeheap[b]]);
- }
- public:
- CRoxieServerMergeActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerMultiInputActivity(_ctx, _factory, _probeManager, _numInputs), helper((IHThorMergeArg &)basehelper)
- {
- activeInputs = 0;
- first = true;
- mergeheap = new unsigned[numInputs];
- pending = new const void *[numStreams];
- compare = helper.queryCompare();
- dedup = helper.dedup();
- for (unsigned i = 0; i < numStreams; i++)
- pending[i] = NULL;
- }
- ~CRoxieServerMergeActivity()
- {
- delete [] mergeheap;
- delete [] pending;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- activeInputs = 0;
- first = true;
- CRoxieServerMultiInputActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- for (unsigned i = 0; i < numStreams; i++)
- ReleaseClearRoxieRow(pending[i]);
- CRoxieServerMultiInputActivity::reset();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (first)
- {
- permute();
- first = false;
- }
- if (activeInputs)
- {
- const void *next = pending[mergeheap[0]];
- readNext();
- if (next)
- processed++;
- return next;
- }
- else
- return NULL;
- }
- };
- class CRoxieServerMergeActivityFactory : public CRoxieServerMultiInputFactory
- {
- public:
- CRoxieServerMergeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerMultiInputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerMergeActivity(_ctx, this, _probeManager, numInputs());
- }
- };
- IRoxieServerActivityFactory *createRoxieServerMergeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerMergeActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerRegroupActivity : public CRoxieServerMultiInputActivity
- {
- IHThorRegroupArg &helper;
- unsigned streamIndex;
- bool eof;
- unsigned __int64 numProcessedLastGroup;
- public:
- CRoxieServerRegroupActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerMultiInputActivity(_ctx, _factory, _probeManager, _numInputs), helper((IHThorRegroupArg &)basehelper)
- {
- streamIndex = 0;
- eof = false;
- numProcessedLastGroup = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- streamIndex = 0;
- eof = false;
- numProcessedLastGroup = processed;
- CRoxieServerMultiInputActivity::start(parentExtractSize, parentExtract, paused);
- }
- const void * nextFromInputs()
- {
- unsigned initialStream = streamIndex;
- while (streamIndex < numStreams)
- {
- const void * next = streamArray[streamIndex]->nextRow();
- if (next)
- {
- if ((streamIndex != initialStream) && (streamIndex != initialStream+1))
- {
- ReleaseRoxieRow(next);
- throw MakeStringException(ROXIE_MISMATCH_GROUP_ERROR, "Mismatched groups supplied to Regroup (%d)", factory->queryId());
- }
- return next;
- }
- streamIndex++;
- }
- if ((initialStream != 0) && (initialStream+1 != numStreams))
- throw MakeStringException(ROXIE_MISMATCH_GROUP_ERROR, "Mismatched groups supplied to Regroup (%d)", factory->queryId());
- streamIndex = 0;
- return NULL;
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- const void * ret = nextFromInputs();
- if (ret)
- {
- processed++;
- return ret;
- }
- if (numProcessedLastGroup != processed)
- {
- numProcessedLastGroup = processed;
- return NULL;
- }
- eof = true;
- return NULL;
- }
- };
- class CRoxieServerRegroupActivityFactory : public CRoxieServerMultiInputFactory
- {
- public:
- CRoxieServerRegroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerMultiInputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerRegroupActivity(_ctx, this, _probeManager, numInputs());
- }
- };
- IRoxieServerActivityFactory *createRoxieServerRegroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerRegroupActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerCombineActivity : public CRoxieServerMultiInputActivity
- {
- IHThorCombineArg &helper;
- unsigned __int64 numProcessedLastGroup;
- public:
- CRoxieServerCombineActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerMultiInputActivity(_ctx, _factory, _probeManager, _numInputs), helper((IHThorCombineArg &)basehelper)
- {
- numProcessedLastGroup = 0;
- }
- ~CRoxieServerCombineActivity()
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- numProcessedLastGroup = processed;
- CRoxieServerMultiInputActivity::start(parentExtractSize, parentExtract, paused);
- }
- void nextInputs(ConstPointerArray & out)
- {
- for (unsigned i=0; i < numStreams; i++)
- {
- const void * next = streamArray[i]->nextRow();
- if (next)
- out.append(next);
- }
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- ConstPointerArray group;
- nextInputs(group);
- if ((group.ordinality() == 0) && (numProcessedLastGroup == processed))
- nextInputs(group);
- if (group.ordinality() == 0)
- {
- numProcessedLastGroup = processed;
- return NULL;
- }
- else if (group.ordinality() != numInputs)
- {
- ReleaseRoxieRows(group);
- throw MakeStringException(ROXIE_MISMATCH_GROUP_ERROR, "Mismatched group input for Combine Activity(%d)", factory->queryId());
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize = helper.transform(rowBuilder, group.ordinality(), group.getArray());
- ReleaseRoxieRows(group);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- catch (IException *E)
- {
- ReleaseRoxieRows(group);
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerCombineActivityFactory : public CRoxieServerMultiInputFactory
- {
- public:
- CRoxieServerCombineActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerMultiInputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerCombineActivity(_ctx, this, _probeManager, numInputs());
- }
- };
- IRoxieServerActivityFactory *createRoxieServerCombineActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerCombineActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerCombineGroupActivity : public CRoxieServerTwoInputActivity
- {
- IHThorCombineGroupArg &helper;
- unsigned __int64 numProcessedLastGroup;
- public:
- CRoxieServerCombineGroupActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerTwoInputActivity(_ctx, _factory, _probeManager), helper((IHThorCombineGroupArg &)basehelper)
- {
- numProcessedLastGroup = 0;
- }
- ~CRoxieServerCombineGroupActivity()
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- numProcessedLastGroup = processed;
- CRoxieServerTwoInputActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- if (idx==(unsigned)-1)
- idx = 0;
- return idx ? NULL : this;
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- const void * left = inputStream->nextRow();
- if (!left && (numProcessedLastGroup == processed))
- left = inputStream->nextRow();
- if (!left)
- {
- if (numProcessedLastGroup == processed)
- {
- const void * nextRight = inputStream1->nextRow();
- if (nextRight)
- {
- ReleaseRoxieRow(nextRight);
- throw MakeStringException(ROXIE_MISSING_GROUP_ERROR, "Missing LEFT record for Combine Group (%d)", factory->queryId());
- }
- }
- else
- numProcessedLastGroup = processed;
- return NULL;
- }
- ConstPointerArray group;
- loop
- {
- const void * in = inputStream1->nextRow();
- if (!in)
- break;
- group.append(in);
- }
- if (group.ordinality() == 0)
- {
- ReleaseRoxieRow(left);
- ReleaseRoxieRows(group);
- throw MakeStringException(ROXIE_MISSING_GROUP_ERROR, "Missing RIGHT group for Combine Group (%d)", factory->queryId());
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize = helper.transform(rowBuilder, left, group.ordinality(), (const void * *)group.getArray());
- ReleaseRoxieRow(left);
- ReleaseRoxieRows(group);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- catch (IException *E)
- {
- ReleaseRoxieRow(left);
- ReleaseRoxieRows(group);
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerCombineGroupActivityFactory : public CRoxieServerActivityFactory
- {
- unsigned input2;
- unsigned input2idx;
- public:
- CRoxieServerCombineGroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- input2 = 0;
- input2idx = 0;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerCombineGroupActivity(_ctx, this, _probeManager);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- if (idx==1)
- {
- input2 = source;
- input2idx = sourceidx;
- }
- else
- CRoxieServerActivityFactory::setInput(idx, source, sourceidx);
- }
- virtual unsigned getInput(unsigned idx, unsigned &sourceidx) const
- {
- switch (idx)
- {
- case 1:
- sourceidx = input2idx;
- return input2;
- case 0:
- return CRoxieServerActivityFactory::getInput(idx, sourceidx);
- default:
- return (unsigned) -1;
- }
- }
- virtual unsigned numInputs() const { return 2; }
- };
- IRoxieServerActivityFactory *createRoxieServerCombineGroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerCombineGroupActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerRollupGroupActivity : public CRoxieServerActivity
- {
- IHThorRollupGroupArg &helper;
- bool eof;
- public:
- CRoxieServerRollupGroupActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- helper((IHThorRollupGroupArg &)basehelper)
- {
- eof = false;
- }
- ~CRoxieServerRollupGroupActivity()
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- loop
- {
- ConstPointerArray group;
- loop
- {
- const void * in = inputStream->nextRow();
- if (!in)
- break;
- group.append(in);
- }
- if (group.ordinality() == 0)
- {
- eof = true;
- return NULL;
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize = helper.transform(rowBuilder, group.ordinality(), (const void * *)group.getArray());
- ReleaseRoxieRows(group);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- catch (IException * E)
- {
- ReleaseRoxieRows(group);
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerRollupGroupActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerRollupGroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerRollupGroupActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerRollupGroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerRollupGroupActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerFilterProjectActivity : public CRoxieServerLateStartActivity
- {
- IHThorFilterProjectArg &helper;
- unsigned numProcessedLastGroup;
- unsigned __int64 recordCount;
- public:
- CRoxieServerFilterProjectActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerLateStartActivity(_ctx, _factory, _probeManager), helper((IHThorFilterProjectArg &)basehelper)
- {
- numProcessedLastGroup = 0;
- recordCount = 0;
- }
- ~CRoxieServerFilterProjectActivity()
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- numProcessedLastGroup = 0;
- recordCount = 0;
- CRoxieServerLateStartActivity::start(parentExtractSize, parentExtract, paused);
- lateStart(parentExtractSize, parentExtract, helper.canMatchAny()); //sets eof
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- loop
- {
- const void * in = inputStream->nextRow();
- if (!in)
- {
- recordCount = 0;
- if (numProcessedLastGroup == processed)
- in = inputStream->nextRow();
- if (!in)
- {
- numProcessedLastGroup = processed;
- return NULL;
- }
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize = helper.transform(rowBuilder, in, ++recordCount);
- ReleaseRoxieRow(in);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerFilterProjectActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerFilterProjectActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- //MORE: Code generator needs to indicate if the count is used.
- //optStableInput = false;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerFilterProjectActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerFilterProjectActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerFilterProjectActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================
- class CRoxieServerStrandedProjectActivity : public CRoxieServerStrandedActivity
- {
- class ProjectProcessor : public StrandProcessor
- {
- protected:
- IHThorProjectArg &helper;
- public:
- ProjectProcessor(CRoxieServerActivity &_parent, IEngineRowStream *_inputStream, IHThorProjectArg &_helper)
- : StrandProcessor(_parent, _inputStream, true), helper(_helper)
- {
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- OwnedConstRoxieRow in = inputStream->nextRow();
- if (!in)
- {
- if (numProcessedLastGroup == processed)
- in.setown(inputStream->nextRow());
- if (!in)
- {
- numProcessedLastGroup = processed;
- return NULL;
- }
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize;
- outSize = helper.transform(rowBuilder, in);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- catch (IException *E)
- {
- throw parent.makeWrappedException(E);
- }
- }
- }
- };
- public:
- CRoxieServerStrandedProjectActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const StrandOptions &_strandOptions)
- : CRoxieServerStrandedActivity(_ctx, _factory, _probeManager, _strandOptions)
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerStrandedActivity::start(parentExtractSize, parentExtract, paused);
- onStartStrands();
- }
- virtual StrandProcessor *createStrandProcessor(IEngineRowStream *instream)
- {
- return new ProjectProcessor(*this, instream, (IHThorProjectArg &) basehelper);
- }
- virtual StrandProcessor *createStrandSourceProcessor(bool inputOrdered) { throwUnexpected(); }
- };
- class CRoxieServerProjectActivity : public CRoxieServerActivity
- {
- unsigned numProcessedLastGroup;
- unsigned __int64 recordCount;
- bool count;
- public:
- CRoxieServerProjectActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, bool _count)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- count(_count)
- {
- numProcessedLastGroup = 0;
- recordCount = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- numProcessedLastGroup = 0;
- recordCount = 0;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- OwnedConstRoxieRow in = inputStream->nextRow();
- if (!in)
- {
- recordCount = 0;
- if (numProcessedLastGroup == processed)
- in.setown(inputStream->nextRow());
- if (!in)
- {
- numProcessedLastGroup = processed;
- return NULL;
- }
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize;
- if (count)
- outSize = ((IHThorCountProjectArg &) basehelper).transform(rowBuilder, in, ++recordCount);
- else
- outSize = ((IHThorProjectArg &) basehelper).transform(rowBuilder, in);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerProjectActivityFactory : public CRoxieServerActivityFactory
- {
- protected:
- StrandOptions strandOptions;
- bool count;
- public:
- CRoxieServerProjectActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), strandOptions(_graphNode)
- {
- count = (kind==TAKcountproject || kind==TAKprefetchcountproject);
- optStableInput = count;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- // NOTE - if they explicitly requested numStrands = 1 via hint, we could use either of the below. For now use the stranded version to ensure it is tested.
- if (kind == TAKproject) // Not supported on prefetch or count projects
- return new CRoxieServerStrandedProjectActivity(_ctx, this, _probeManager, strandOptions);
- else
- return new CRoxieServerProjectActivity(_ctx, this, _probeManager, count);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerProjectActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerProjectActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerPrefetchProjectActivity : public CRoxieServerActivity, implements IRecordPullerCallback
- {
- unsigned numProcessedLastGroup;
- bool count;
- bool eof;
- bool allPulled;
- bool isThreaded;
- unsigned preload;
- unsigned __int64 recordCount;
- IHThorPrefetchProjectArg &helper;
- RecordPullerThread puller;
- InterruptableSemaphore ready;
- InterruptableSemaphore space;
- class PrefetchInfo : public CInterface
- {
- public:
- inline PrefetchInfo(IHThorPrefetchProjectArg &helper, const void *_in, unsigned __int64 _recordCount)
- {
- if (helper.preTransform(extract, _in, _recordCount))
- {
- in.setown(_in);
- recordCount = _recordCount;
- }
- else
- {
- ReleaseRoxieRow(_in);
- recordCount = 0;
- }
- }
- OwnedConstRoxieRow in;
- unsigned __int64 recordCount;
- rtlRowBuilder extract;
- };
- QueueOf<PrefetchInfo, true> pulled;
- CriticalSection pulledCrit;
- public:
- CRoxieServerPrefetchProjectActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, bool _count)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- helper((IHThorPrefetchProjectArg &) basehelper),
- puller(false),
- count(_count)
- {
- numProcessedLastGroup = 0;
- recordCount = 0;
- eof = false;
- allPulled = false;
- isThreaded = (helper.getFlags() & PPFparallel) != 0;
- preload = 0;
- }
- ~CRoxieServerPrefetchProjectActivity()
- {
- while (pulled.ordinality())
- ::Release(pulled.dequeue());
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- if (idx)
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() parameter out of bounds at %s(%d)", __FILE__, __LINE__);
- puller.setInput(this, _sourceIdx, _in);
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- puller.connectInputStreams(ctx, consumerOrdered);
- return CRoxieServerActivity::getOutputStreams(ctx, idx, streams, consumerOptions, consumerOrdered, orderedCallbacks);
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (idx==0)
- return puller.queryInput();
- else
- return NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- numProcessedLastGroup = 0;
- recordCount = 0;
- eof = false;
- allPulled = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- preload = helper.getLookahead();
- if (!preload)
- preload = ctx->queryOptions().prefetchProjectPreload;
- space.reinit(preload);
- ready.reinit();
- puller.start(parentExtractSize, parentExtract, paused, preload, !isThreaded, ctx);
- }
- virtual void stop()
- {
- space.interrupt();
- ready.interrupt();
- CRoxieServerActivity::stop();
- puller.stop();
- }
- virtual void reset()
- {
- CRoxieServerActivity::reset();
- puller.reset();
- allPulled = false;
- while (pulled.ordinality())
- ::Release(pulled.dequeue());
- }
- virtual PrefetchInfo *readNextRecord()
- {
- if (!isThreaded)
- {
- if (!allPulled) // This looks like it's thread unsafe but we are inside the if(!isThreaded) so should be ok
- puller.pullRecords(1);
- }
- else
- ready.wait();
- CriticalBlock b(pulledCrit);
- PrefetchInfo *ret = pulled.ordinality() ? pulled.dequeue() : NULL;
- space.signal();
- return ret;
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- loop
- {
- Owned<PrefetchInfo> in = readNextRecord();
- if (!in)
- {
- recordCount = 0;
- if (numProcessedLastGroup == processed)
- in.setown(readNextRecord());
- if (!in)
- {
- numProcessedLastGroup = processed;
- eof = true;
- return NULL;
- }
- }
- try
- {
- if (in->in)
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize;
- IThorChildGraph *child = helper.queryChild();
- Owned<IEclGraphResults> results;
- if (child)
- results.setown(child->evaluate(in->extract.size(), in->extract.getbytes()));
- outSize = helper.transform(rowBuilder, in->in, results, in->recordCount);
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- // interface IExceptionHandler
- virtual bool fireException(IException *e)
- {
- // called from puller thread on failure
- ready.interrupt(LINK(e));
- space.interrupt(e);
- return true;
- }
- // interface IRecordPullerCallback
- virtual void processRow(const void *row)
- {
- {
- CriticalBlock b(pulledCrit);
- pulled.enqueue(new PrefetchInfo(helper, row, ++recordCount));
- }
- if (isThreaded)
- {
- ready.signal();
- space.wait();
- }
- }
- virtual void processEOG()
- {
- {
- CriticalBlock b(pulledCrit);
- pulled.enqueue(NULL);
- }
- if (isThreaded)
- {
- ready.signal();
- space.wait();
- }
- }
- virtual void processGroup(const ConstPointerArray &rows)
- {
- throwUnexpected();
- }
- virtual void processDone()
- {
- CriticalBlock b(pulledCrit);
- allPulled = true;
- if (isThreaded)
- ready.signal();
- }
- };
- class CRoxieServerPrefetchProjectActivityFactory : public CRoxieServerProjectActivityFactory
- {
- public:
- CRoxieServerPrefetchProjectActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerProjectActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerPrefetchProjectActivity(_ctx, this, _probeManager, count);
- }
- };
- extern IRoxieServerActivityFactory *createRoxieServerPrefetchProjectActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerPrefetchProjectActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CPointerArrayRoxieInput : public CPseudoRoxieInput
- {
- public:
- CPointerArrayRoxieInput()
- {
- rowset = NULL;
- rowcount = 0;
- curRow = 0;
- }
- void init(size32_t _rowcount, byte **_rowset)
- {
- rowset = _rowset;
- rowcount = _rowcount;
- curRow = 0;
- }
- virtual const void * nextRow()
- {
- if (curRow < rowcount)
- {
- const void * ret = rowset[curRow];
- if (ret)
- LinkRoxieRow(ret);
- curRow++;
- return ret;
- }
- return NULL;
- }
- protected:
- byte **rowset;
- size32_t rowcount;
- size32_t curRow;
- };
- class CRoxieServerLoopActivity : public CRoxieServerActivity
- {
- protected:
- IHThorLoopArg &helper;
- ThorActivityKind activityKind;
- unsigned maxIterations;
- bool finishedLooping;
- unsigned flags;
- bool eof;
- rtlRowBuilder loopExtractBuilder;
- unsigned loopGraphId;
- Linked<IOutputMetaData> counterMeta;
- public:
- CRoxieServerLoopActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _loopGraphId, IOutputMetaData * _counterMeta)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- helper((IHThorLoopArg &)basehelper), loopGraphId(_loopGraphId), counterMeta(_counterMeta)
- {
- eof = false;
- finishedLooping = false;
- activityKind = factory->getKind();
- flags = helper.getFlags();
- maxIterations = 0;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- int iterations = (int) helper.numIterations();
- maxIterations = (iterations >= 0) ? iterations : 0;
- finishedLooping = ((activityKind == TAKloopcount) && (maxIterations == 0));
- if ((flags & IHThorLoopArg::LFnewloopagain) && !helper.loopFirstTime())
- finishedLooping = true;
- loopExtractBuilder.clear();
- helper.createParentExtract(loopExtractBuilder); // could possibly delay this until execution actually happens
- }
- virtual void stop()
- {
- CRoxieServerActivity::stop();
- loopExtractBuilder.clear();
- }
- void createCounterResult(IRoxieServerChildGraph * graph, unsigned counter)
- {
- if (flags & IHThorLoopArg::LFcounter)
- {
- void * counterRow = ctx->queryRowManager().allocate(sizeof(thor_loop_counter_t), activityId);
- *((thor_loop_counter_t *)counterRow) = counter;
- RtlLinkedDatasetBuilder builder(rowAllocator);
- builder.appendOwn(counterRow);
- Owned<CGraphResult> counterResult = new CGraphResult(builder.getcount(), builder.linkrows());
- graph->setInputResult(2, counterResult);
- }
- }
- };
- //=================================================================================
- class CRoxieServerSequentialLoopActivity : public CRoxieServerLoopActivity
- {
- Owned<IActivityGraph> loopQuery;
- Owned<IRoxieServerChildGraph> loopGraph;
- IEngineRowStream * curStream;
- RtlLinkedDatasetBuilder *loopInputBuilder;
- CPointerArrayRoxieInput arrayInput;
- Linked<IEngineRowStream> resultStream;
- unsigned loopCounter;
- public:
- CRoxieServerSequentialLoopActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _loopGraphId, IOutputMetaData * _counterMeta)
- : CRoxieServerLoopActivity(_ctx, _factory, _probeManager, _loopGraphId, _counterMeta)
- {
- curStream = NULL;
- loopCounter = 0;
- loopInputBuilder = NULL;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerLoopActivity::onCreate(_colocalParent);
- loopQuery.set(ctx->queryChildGraph(loopGraphId));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- curStream = inputStream;
- loopCounter = 1;
- CRoxieServerLoopActivity::start(parentExtractSize, parentExtract, paused);
- //MORE: Not sure about this, should IRoxieServerChildGraph be combined with IActivityGraph?
- loopGraph.set(loopQuery->queryLoopGraph());
- loopInputBuilder = new RtlLinkedDatasetBuilder(rowAllocator);
- }
- virtual void stop()
- {
- delete loopInputBuilder;
- loopInputBuilder = NULL;
- CRoxieServerLoopActivity::stop();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- unsigned emptyIterations = 0;
- loop
- {
- loop
- {
- const void * ret = curStream->nextRow();
- if (!ret)
- {
- ret = curStream->nextRow(); // more cope with groups somehow....
- if (!ret)
- {
- if (finishedLooping)
- {
- eof = true;
- return NULL;
- }
- break;
- }
- }
- if (finishedLooping ||
- ((flags & IHThorLoopArg::LFfiltered) && !helper.sendToLoop(loopCounter, ret)))
- {
- processed++;
- return ret;
- }
- loopInputBuilder->appendOwn(ret);
- }
- switch (activityKind)
- {
- case TAKloopdataset:
- {
- if (!(flags & IHThorLoopArg::LFnewloopagain))
- {
- if (!helper.loopAgain(loopCounter, loopInputBuilder->getcount(), (const void**) loopInputBuilder->queryrows()))
- {
- if (loopInputBuilder->getcount() == 0)
- {
- eof = true;
- return NULL;
- }
- arrayInput.init(loopInputBuilder->getcount(), loopInputBuilder->linkrows());
- // MORE - should builder be cleared here?
- curStream = &arrayInput;
- finishedLooping = true;
- continue; // back to the input loop again
- }
- }
- break;
- }
- case TAKlooprow:
- if (!loopInputBuilder->getcount())
- {
- finishedLooping = true;
- eof = true;
- return NULL;
- }
- break;
- }
- if (loopInputBuilder->getcount())
- emptyIterations = 0;
- else
- {
- //note: any outputs which didn't go around the loop again, would return the record, reinitializing emptyIterations
- emptyIterations++;
- if (emptyIterations > maxEmptyLoopIterations)
- throw MakeStringException(ROXIE_TOO_MANY_EMPTY_LOOP, "Executed LOOP with empty input and output %u times", emptyIterations);
- if (emptyIterations % 32 == 0)
- CTXLOG("Executing LOOP with empty input and output %u times", emptyIterations);
- }
- checkAbort();
- try
- {
- Owned<IRoxieGraphResults> results = executeIteration(loopExtractBuilder.size(), loopExtractBuilder.getbytes(), loopCounter);
- resultStream.setown(results->createIterator(0));
- if (flags & IHThorLoopArg::LFnewloopagain)
- {
- Owned<IEngineRowStream> againResult = results->createIterator(helper.loopAgainResult());
- OwnedConstRoxieRow row = againResult->nextRow();
- assertex(row);
- //Result is a row which contains a single boolean field.
- if (!((const bool *)row.get())[0])
- finishedLooping = true;
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- curStream = resultStream.get();
- loopCounter++;
- if ((activityKind == TAKloopcount) && (loopCounter > maxIterations))
- finishedLooping = true;
- }
- }
- IRoxieGraphResults * executeIteration(unsigned parentExtractSize, const byte *parentExtract, unsigned counter)
- {
- try
- {
- loopGraph->beforeExecute();
- Owned<IGraphResult> inputRowsResult = new CGraphResult(loopInputBuilder->getcount(), loopInputBuilder->linkrows());
- loopInputBuilder->clear();
- loopGraph->setInputResult(1, inputRowsResult);
- createCounterResult(loopGraph, counter);
- Owned<IRoxieGraphResults> ret = loopGraph->execute(parentExtractSize, parentExtract);
- loopGraph->afterExecute();
- return ret.getClear();
- }
- catch (...)
- {
- CTXLOG("Exception thrown in loop body - cleaning up");
- loopGraph->afterExecute();
- throw;
- }
- }
- };
- //=================================================================================
- typedef SafeQueueOf<const void, true> SafeRowQueue;
- class CRowQueuePseudoInput : public CPseudoRoxieInput
- {
- public:
- CRowQueuePseudoInput(SafeRowQueue & _input, IOutputMetaData * _meta)
- : meta(_meta), input(_input)
- {
- eof = false;
- }
- virtual const void * nextRow()
- {
- if (eof)
- return NULL;
- const void * ret = input.dequeue();
- if (!ret)
- eof = true;
- return ret;
- }
- virtual IOutputMetaData * queryOutputMeta() const { return meta; }
- protected:
- SafeRowQueue & input;
- IOutputMetaData * meta;
- bool eof;
- };
- class CRoxieServerParallelLoopActivity;
- class LoopFilterPseudoInput : public CIndirectRoxieInput
- {
- public:
- LoopFilterPseudoInput(CRoxieServerParallelLoopActivity * _activity, unsigned _counter) :
- activity(_activity), counter(_counter)
- {
- }
- virtual const void * nextRow();
- protected:
- CRoxieServerParallelLoopActivity * activity;
- unsigned counter;
- };
- class LoopExecutorThread : public RestartableThread
- {
- protected:
- Owned<CSafeRoxieInput> safeInput;
- CRoxieServerParallelLoopActivity * activity;
- bool eof;
- CriticalSection crit;
- unsigned flags;
- SafeRowQueue tempResults[2];
- unsigned savedParentExtractSize;
- const byte * savedParentExtract;
- IArrayOf<IActivityGraph> cachedGraphs;
- IRoxieSlaveContext *ctx;
- public:
- LoopExecutorThread()
- : RestartableThread("LoopExecutorThread")
- {
- activity = NULL;
- eof = false;
- flags = 0;
- ctx = NULL;
- savedParentExtract = NULL;
- savedParentExtractSize = 0;
- }
- void setInput(CRoxieServerParallelLoopActivity * _activity, unsigned _sourceIdx, IFinalRoxieInput *_input, unsigned _flags)
- {
- activity = _activity;
- flags = _flags;
- // stop is called on our consumer's thread. We need to take care calling stop for our input to make sure it is not in mid-nextRow etc etc.
- safeInput.setown(new CSafeRoxieInput(_sourceIdx, _input));
- }
- void connectInputStreams(bool consumerOrdered)
- {
- safeInput->connectInputStreams(ctx, consumerOrdered);
- }
- IFinalRoxieInput *queryInput() const
- {
- return safeInput;
- }
- void onCreate(IRoxieSlaveContext * _ctx);
- void start(unsigned parentExtractSize, const byte *parentExtract, bool paused);
- void stop();
- void reset();
- virtual int run();
- protected:
- void executeLoop();
- void executeLoopInstance(unsigned counter, unsigned numIterations, unsigned inputIdx, IFinalRoxieInput * input, SafeRowQueue * spillOutput);
- IFinalRoxieInput * createLoopIterationGraph(unsigned i, unsigned inputIdx, IFinalRoxieInput * input, unsigned counter);
- };
- class CRoxieServerParallelLoopActivity : public CRoxieServerLoopActivity
- {
- friend class LoopFilterPseudoInput;
- friend class LoopExecutorThread;
- QueueOf<const void, true> ready;
- CriticalSection helperCS;
- CriticalSection cs;
- unsigned defaultNumParallel;
- LoopExecutorThread executor;
- IProbeManager* probeManager;
- CriticalSection canAccess;
- CriticalSection scrit;
- InterruptableSemaphore readySpace;
- InterruptableSemaphore recordsReady;
- protected:
- bool includeInLoop(unsigned counter, const void * row)
- {
- CriticalBlock b(helperCS);
- return helper.sendToLoop(counter, row);
- }
- public:
- CRoxieServerParallelLoopActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _loopGraphId, IOutputMetaData * _counterMeta)
- : CRoxieServerLoopActivity(_ctx, _factory, _probeManager, _loopGraphId, _counterMeta),
- readySpace(parallelLoopFlowLimit)
- {
- probeManager = _probeManager;
- defaultNumParallel = 0;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- executor.onCreate(ctx);
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CriticalBlock b(scrit); // can stop while still starting, if unlucky...
- readySpace.reinit(parallelLoopFlowLimit);
- recordsReady.reinit();
- CRoxieServerLoopActivity::start(parentExtractSize, parentExtract, paused);
- defaultNumParallel = helper.defaultParallelIterations();
- if (!defaultNumParallel)
- defaultNumParallel = DEFAULT_PARALLEL_LOOP_THREADS;
- //MORE: If numIterations <= number of parallel iterations[1],
- //then we don't need to create a separate thread to do the processing, and the results will also avoid
- //being transferred via a queue
- executor.start(parentExtractSize, parentExtract, paused);
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- if (idx)
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() parameter out of bounds at %s(%d)", __FILE__, __LINE__);
- executor.setInput(this, _sourceIdx, _in, flags);
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- executor.connectInputStreams(consumerOrdered);
- return CRoxieServerActivity::getOutputStreams(ctx, idx, streams, nullptr, consumerOrdered, nullptr);
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (idx==0)
- return executor.queryInput();
- else
- return NULL;
- }
- virtual void stop()
- {
- CriticalBlock b(scrit); // can stop while still starting, if unlucky...
- readySpace.interrupt();
- recordsReady.interrupt();
- executor.join(); // MORE - may not be needed given stop/reset split
- CRoxieServerLoopActivity::stop();
- }
- virtual void reset()
- {
- while (ready.ordinality())
- ReleaseRoxieRow(ready.dequeue());
- executor.reset();
- CRoxieServerActivity::reset();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- if (eof)
- return NULL;
- recordsReady.wait();
- CriticalBlock procedure(canAccess);
- if (ready.ordinality())
- {
- const void *result = ready.dequeue();
- readySpace.signal();
- if (result)
- processed++;
- return result;
- }
- else
- eof = true;
- }
- }
- unsigned getNumParallel(unsigned iter)
- {
- return defaultNumParallel;
- }
- inline void enqueueResult(const void * row)
- {
- try
- {
- while(!readySpace.wait(1000))
- {
- CTXLOG("Blocked waiting for space in loop %p activity id: %d output queue: %d records in queue", this, queryId(), ready.ordinality());
- }
- }
- catch (...)
- {
- ReleaseRoxieRow(row);
- throw;
- }
- CriticalBlock b2(canAccess);
- ready.enqueue(row);
- recordsReady.signal();
- }
- inline void finishResults()
- {
- recordsReady.signal();
- }
- virtual bool fireException(IException *e)
- {
- readySpace.interrupt(LINK(e));
- recordsReady.interrupt(e);
- return true;
- }
- IActivityGraph * createChildGraphInstance()
- {
- return factory->createChildGraph(ctx, &helper, loopGraphId, this, probeManager, *this);
- }
- IActivityGraph * queryChildGraph()
- {
- return ctx->queryChildGraph(loopGraphId);
- }
- };
- //=================================================================================
- const void * LoopFilterPseudoInput::nextRow()
- {
- loop
- {
- const void * next = stream->nextRow();
- if (!next || activity->includeInLoop(counter, next))
- return next;
- activity->enqueueResult(next);
- }
- }
- void LoopExecutorThread::onCreate(IRoxieSlaveContext * _ctx)
- {
- //Initialise the cached graph list with the child instance that will always be created. Other iterations will be created on demand.
- ctx = _ctx;
- cachedGraphs.append(*LINK(activity->queryChildGraph()));
- }
- void LoopExecutorThread::start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- savedParentExtractSize = parentExtractSize;
- savedParentExtract = parentExtract;
- eof = false;
- StringBuffer logPrefix("[");
- ctx->getLogPrefix(logPrefix).append("] ");
- RestartableThread::start(logPrefix);
- }
- int LoopExecutorThread::run()
- {
- try
- {
- executeLoop();
- }
- catch (IException *e)
- {
- activity->fireException(e);
- }
- catch (...)
- {
- activity->fireException(MakeStringException(ROXIE_INTERNAL_ERROR, "Unexpected exception caught in LoopExecutorThread::run"));
- }
- return 0;
- }
- void LoopExecutorThread::stop()
- {
- safeInput->stop();
- RestartableThread::join();
- }
- void LoopExecutorThread::reset()
- {
- safeInput->reset();
- }
- void LoopExecutorThread::executeLoop()
- {
- unsigned iterations = 0;
- unsigned counter = 0;
- unsigned outputIndex = 0;
- //Note, activities don't link inputs, so need to be careful that special inputs remain linked while the activity is executing.
- loop
- {
- if (activity->activityKind == TAKloopcount)
- {
- if (counter == activity->maxIterations)
- break;
- }
- else
- {
- //This condition isn't quite right because it needs to be whether the filtered
- //input is empty. May be ok if we include that in the semantics,
- if (tempResults[1-outputIndex].ordinality() == 0)
- break;
- }
- unsigned numParallel = activity->getNumParallel(iterations);
- Linked<IFinalRoxieInput> curInput;
- if (iterations == 0)
- curInput.set(safeInput);
- else
- {
- SafeRowQueue & inputQueue = tempResults[1-outputIndex];
- inputQueue.enqueue(NULL);
- curInput.setown(new CRowQueuePseudoInput(inputQueue, activity->queryOutputMeta()));
- }
- SafeRowQueue * curOutput = NULL;
- if (counter+numParallel > activity->maxIterations)
- numParallel = activity->maxIterations - counter;
- else if (counter+numParallel < activity->maxIterations)
- curOutput = &tempResults[outputIndex];
- executeLoopInstance(counter, numParallel, 0, curInput, curOutput); // CurInput is always SOME sort of proxy so idx always 0
- outputIndex = 1-outputIndex;
- counter += numParallel;
- iterations++;
- }
- //Check for TAKlooprow, where end of loop couldn't be determined ahead of time
- SafeRowQueue & inputQueue = tempResults[1-outputIndex];
- while (inputQueue.ordinality())
- {
- const void * next = inputQueue.dequeue();
- activity->enqueueResult(next);
- }
- activity->finishResults();
- }
- void LoopExecutorThread::executeLoopInstance(unsigned counter, unsigned numIterations, unsigned inputIdx, IFinalRoxieInput * input, SafeRowQueue * spillOutput)
- {
- IArrayOf<IFinalRoxieInput> savedInputs; // activities don't link their inputs, so this list keeps filters alive.
- Linked<IFinalRoxieInput> curInput = input;
- unsigned i;
- for (i= 0; i != numIterations; i++)
- {
- unsigned thisCounter = counter+i+1;
- IFinalRoxieInput * filtered = curInput;
- if (flags & IHThorLoopArg::LFfiltered)
- {
- LoopFilterPseudoInput *newFiltered = new LoopFilterPseudoInput(activity, thisCounter);
- newFiltered->setInput(inputIdx, curInput);
- filtered = newFiltered;
- savedInputs.append(*filtered);
- inputIdx = 0;
- }
- //graph is kept, so new curInput will be guaranteed to exist
- curInput.setown(createLoopIterationGraph(i, inputIdx, filtered, thisCounter));
- inputIdx = 0;
- }
- Owned<IStrandJunction> curJunction;
- IEngineRowStream *curStream = connectSingleStream(ctx, curInput, inputIdx, curJunction, true);
- try
- {
- curInput->start(savedParentExtractSize, savedParentExtract, false);
- if (spillOutput)
- {
- loop
- {
- const void * next = curStream->nextRow();
- if (!next)
- break;
- spillOutput->enqueue(next);
- }
- }
- else
- {
- loop
- {
- const void * next = curStream->nextRow();
- if (!next)
- break;
- activity->enqueueResult(next);
- }
- }
- }
- catch (IException *E)
- {
- ctx->notifyAbort(E);
- for (i= 0; i != numIterations; i++)
- {
- cachedGraphs.item(i).queryLoopGraph()->afterExecute();
- }
- curStream->stop();
- curInput->reset();
- throw;
- }
- for (i= 0; i != numIterations; i++)
- {
- cachedGraphs.item(i).queryLoopGraph()->afterExecute();
- }
- curStream->stop();
- curInput->reset();
- }
- IFinalRoxieInput * LoopExecutorThread::createLoopIterationGraph(unsigned i, unsigned inputIdx, IFinalRoxieInput * input, unsigned counter)
- {
- if (!cachedGraphs.isItem(i))
- cachedGraphs.append(*activity->createChildGraphInstance());
- Linked<IRoxieServerChildGraph> loopGraph = cachedGraphs.item(i).queryLoopGraph();
- loopGraph->beforeExecute();
- if (!loopGraph->querySetInputResult(1, inputIdx, input))
- throwUnexpected(); // a loop which doesn't use the value from the previous iteration. Should probably handle even if daft.
- activity->createCounterResult(loopGraph, counter);
- return loopGraph->selectOutput(0);
- }
- //=================================================================================
- class CCounterRowMetaData : implements IOutputMetaData, public CInterface
- {
- public:
- IMPLEMENT_IINTERFACE
- virtual size32_t getRecordSize(const void *) { return sizeof(thor_loop_counter_t); }
- virtual size32_t getMinRecordSize() const { return sizeof(thor_loop_counter_t); }
- virtual size32_t getFixedSize() const { return sizeof(thor_loop_counter_t); }
- virtual void toXML(const byte * self, IXmlWriter & out) { }
- virtual unsigned getVersion() const { return OUTPUTMETADATA_VERSION; }
- virtual unsigned getMetaFlags() { return 0; }
- virtual void destruct(byte * self) {}
- virtual IOutputRowSerializer * createDiskSerializer(ICodeContext * ctx, unsigned activityId) { return NULL; }
- virtual IOutputRowDeserializer * createDiskDeserializer(ICodeContext * ctx, unsigned activityId) { return NULL; }
- virtual ISourceRowPrefetcher * createDiskPrefetcher(ICodeContext * ctx, unsigned activityId) { return NULL; }
- virtual IOutputMetaData * querySerializedDiskMeta() { return this; }
- virtual IOutputRowSerializer * createInternalSerializer(ICodeContext * ctx, unsigned activityId) { return NULL; }
- virtual IOutputRowDeserializer * createInternalDeserializer(ICodeContext * ctx, unsigned activityId) { return NULL; }
- virtual void walkIndirectMembers(const byte * self, IIndirectMemberVisitor & visitor) {}
- virtual IOutputMetaData * queryChildMeta(unsigned i) { return NULL; }
- };
- class CRoxieServerLoopActivityFactory : public CRoxieServerActivityFactory
- {
- unsigned loopGraphId;
- unsigned flags;
- Linked<IOutputMetaData> counterMeta;
- public:
- CRoxieServerLoopActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _loopGraphId)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), loopGraphId(_loopGraphId)
- {
- Owned<IHThorLoopArg> helper = (IHThorLoopArg *) helperFactory();
- flags = helper->getFlags();
- counterMeta.setown(new CCounterRowMetaData);
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- if (flags & IHThorLoopArg::LFparallel)
- return new CRoxieServerParallelLoopActivity(_ctx, this, _probeManager, loopGraphId, counterMeta);
- else
- return new CRoxieServerSequentialLoopActivity(_ctx, this, _probeManager, loopGraphId, counterMeta);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerLoopActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _loopGraphId)
- {
- return new CRoxieServerLoopActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _loopGraphId);
- }
- //=================================================================================
- class CRoxieServerGraphLoopActivity : public CRoxieServerActivity
- {
- protected:
- IHThorGraphLoopArg &helper;
- unsigned maxIterations;
- unsigned flags;
- rtlRowBuilder GraphExtractBuilder;
- unsigned loopGraphId;
- Linked<IOutputMetaData> counterMeta;
- public:
- CRoxieServerGraphLoopActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _GraphGraphId, IOutputMetaData * _counterMeta)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- helper((IHThorGraphLoopArg &)basehelper), loopGraphId(_GraphGraphId), counterMeta(_counterMeta)
- {
- flags = helper.getFlags();
- maxIterations = 0;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- int iterations = (int) helper.numIterations();
- maxIterations = (iterations >= 0) ? iterations : 0;
- if (maxIterations > maxGraphLoopIterations)
- throw MakeStringException(ROXIE_TOO_MANY_GRAPH_LOOP, "Attempt to execute graph %u times", maxIterations);
- if (maxIterations != 0)
- {
- GraphExtractBuilder.clear();
- helper.createParentExtract(GraphExtractBuilder);
- }
- }
- virtual void stop()
- {
- CRoxieServerActivity::stop();
- GraphExtractBuilder.clear();
- }
- void createCounterResult(IRoxieServerChildGraph * graph, unsigned counter)
- {
- if (flags & IHThorGraphLoopArg::GLFcounter)
- {
- void * counterRow = ctx->queryRowManager().allocate(sizeof(thor_loop_counter_t), activityId);
- *((thor_loop_counter_t *)counterRow) = counter;
- RtlLinkedDatasetBuilder builder(rowAllocator);
- builder.appendOwn(counterRow);
- Owned<CGraphResult> counterResult = new CGraphResult(builder.getcount(), builder.linkrows());
- graph->setInputResult(0, counterResult);
- }
- }
- };
- //=================================================================================
- class CRoxieServerSequentialGraphLoopActivity : public CRoxieServerGraphLoopActivity
- {
- Owned<IActivityGraph> GraphQuery;
- Owned<IRoxieServerChildGraph> loopGraph;
- Linked<IEngineRowStream> resultStream;
- bool evaluated;
- public:
- CRoxieServerSequentialGraphLoopActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _GraphGraphId, IOutputMetaData * _counterMeta)
- : CRoxieServerGraphLoopActivity(_ctx, _factory, _probeManager, _GraphGraphId, _counterMeta)
- {
- evaluated = false;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerGraphLoopActivity::onCreate(_colocalParent);
- GraphQuery.set(ctx->queryChildGraph(loopGraphId));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerGraphLoopActivity::start(parentExtractSize, parentExtract, paused);
- //MORE: Not sure about this, should IRoxieServerChildGraph be combined with IActivityGraph?
- loopGraph.set(GraphQuery->queryLoopGraph());
- evaluated = false;
- }
- virtual void stop()
- {
- if (loopGraph)
- loopGraph->clearGraphLoopResults();
- CRoxieServerGraphLoopActivity::stop();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (!evaluated)
- {
- executeEntireGraph();
- evaluated = true;
- }
- const void * ret = resultStream->nextRow();
- if (ret)
- processed++;
- return ret;
- }
- void executeIteration(unsigned parentExtractSize, const byte *parentExtract, unsigned counter)
- {
- try
- {
- loopGraph->beforeExecute();
- createCounterResult(loopGraph, counter);
- loopGraph->executeGraphLoop(parentExtractSize, parentExtract);
- loopGraph->afterExecute();
- }
- catch (...)
- {
- CTXLOG("Exception thrown in graph body - cleaning up");
- loopGraph->afterExecute();
- throw;
- }
- }
- void createInitialGraphInput()
- {
- loopGraph->clearGraphLoopResults();
- RtlLinkedDatasetBuilder builder(rowAllocator);
- inputStream->readAll(builder);
- Owned<CGraphResult> result = new CGraphResult(builder.getcount(), builder.linkrows());
- loopGraph->setGraphLoopResult(0, result);
- }
- void executeEntireGraph()
- {
- createInitialGraphInput();
- for (unsigned loopCounter=1; loopCounter <= maxIterations; loopCounter++)
- {
- executeIteration(GraphExtractBuilder.size(), GraphExtractBuilder.getbytes(), loopCounter);
- }
- resultStream.setown(loopGraph->getGraphLoopResult(maxIterations));
- }
- };
- //=================================================================================
- struct GraphOutputSplitterArg : public ccdserver_hqlhelper::CThorSplitArg
- {
- public:
- virtual unsigned numBranches()
- {
- return 0;
- }
- virtual IOutputMetaData * queryOutputMeta()
- {
- return NULL;// get it from the parent..
- }
- };
- extern "C" IHThorArg * createGraphOutputSplitter() { return new GraphOutputSplitterArg; }
- class CGraphIterationInfo : public CInterface
- {
- private:
- Owned<IRoxieServerActivityFactory> factory; // Note - before sourceAct, so destroyed last
- unsigned sourceIdx;
- Linked<IRoxieServerActivity> sourceAct;
- Linked<IFinalRoxieInput> sourceInput;
- unsigned numUses;
- unsigned iteration;
- public:
- CGraphIterationInfo(IRoxieServerActivity * _sourceAct, IFinalRoxieInput *_input, unsigned _sourceIdx, unsigned _iteration)
- : sourceAct(_sourceAct), sourceInput(_input), sourceIdx(_sourceIdx), iteration(_iteration)
- {
- numUses = 0;
- }
- inline void noteUsed()
- {
- numUses++;
- }
- void createSplitter(IRoxieSlaveContext *ctx, IProbeManager *probeManager, IArrayOf<IRoxieProbe> &probes)
- {
- if (numUses > 1)
- {
- Owned<IPropertyTree> splitterNode = createPTree();
- factory.setown(createRoxieServerThroughSpillActivityFactory(sourceAct->queryFactory()->queryQueryFactory(), createGraphOutputSplitter, numUses, *splitterNode));
- IRoxieServerActivity *splitter = factory->createActivity(ctx, NULL);
- splitter->onCreate(NULL);
- IFinalRoxieInput *input = sourceAct->queryOutput(sourceIdx);
- if (probeManager)
- {
- IRoxieProbe * inputProbe = probeManager->createProbe(static_cast<IInputBase*>(input), sourceAct, splitter, sourceIdx, 0, iteration);
- probes.append(*LINK(inputProbe));
- input = &dynamic_cast<IFinalRoxieInput &>(inputProbe->queryInput());
- sourceIdx = 0;
- }
- sourceAct.setown(splitter);
- sourceAct->setInput(0, sourceIdx, input);
- sourceIdx = 0;
- sourceInput.clear();
- }
- }
- IFinalRoxieInput *connectOutput(IProbeManager *probeManager, IArrayOf<IRoxieProbe> &probes, IRoxieServerActivity *targetAct, unsigned targetIdx)
- {
- // MORE - not really necessary to create splitters in separate pass, is it?
- if (factory) // we created a splitter....
- sourceInput.set(sourceAct->queryOutput(sourceIdx));
- IFinalRoxieInput *ret = sourceInput;
- if (probeManager)
- {
- IRoxieProbe * inputProbe = probeManager->createProbe(sourceInput, sourceAct, targetAct, sourceIdx, targetIdx, iteration);
- probes.append(*LINK(inputProbe));
- ret = &dynamic_cast<IFinalRoxieInput &>(inputProbe->queryInput());
- // MORE - do we need to connect the stream in the probe?
- }
- if (factory) // we created a splitter....
- sourceIdx++;
- return ret;
- }
- };
- class CRoxieServerParallelGraphLoopActivity : public CRoxieServerGraphLoopActivity, implements IRoxieServerLoopResultProcessor
- {
- Owned<IActivityGraph> childGraph;
- IFinalRoxieInput * resultInput;
- IEngineRowStream * resultStream;
- Owned<IStrandJunction> resultJunction;
- CIArrayOf<CGraphIterationInfo> outputs;
- IArrayOf<IRoxieServerChildGraph> iterationGraphs;
- Owned<CExtractMapperInput> inputExtractMapper;
- IProbeManager *probeManager;
- unsigned createLoopCounter;
- IArrayOf<IRoxieProbe> probes;
- public:
- CRoxieServerParallelGraphLoopActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _GraphGraphId, IOutputMetaData * _counterMeta)
- : CRoxieServerGraphLoopActivity(_ctx, _factory, _probeManager, _GraphGraphId, _counterMeta), probeManager(_probeManager)
- {
- inputExtractMapper.setown(new CExtractMapperInput);
- resultInput = NULL;
- resultStream = NULL;
- createLoopCounter = 0;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerGraphLoopActivity::onCreate(_colocalParent);
- childGraph.set(ctx->queryChildGraph(loopGraphId));
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- //Input needs to be handled very carefully.....
- //We don't want to call onStart on the input unless it is actually used, so don't use the base CRoxieServerActivity implementation.
- //This activity's input needs to be started with (parentExtractSize, parentExtract), but the elements in the graph need to be started with the
- //GraphExtractBuilder parent extract. So we need to wrap the input in a pseudo-input (inputExtractMapper) that passes through a different
- //parentExtract. Something very similar will be needed for query library calls with streaming inputs when they are implemented.
- assertex(idx == 0);
- inputExtractMapper->setInput(_sourceIdx, _in);
- }
- virtual void connectInputStreams(bool consumerOrdered)
- {
- //NB: inputExtractMapper is not connected at this point - only if/when it is used from within the graph
- CRoxieServerActivity::connectInputStreams(consumerOrdered);
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (idx==0)
- return inputExtractMapper->queryInput();
- else
- return NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerGraphLoopActivity::start(parentExtractSize, parentExtract, paused); // initialises GraphExtractBuilder
- inputExtractMapper->setParentExtract(parentExtractSize, parentExtract);
- createExpandedGraph(GraphExtractBuilder.size(), GraphExtractBuilder.getbytes(), probeManager);
- resultInput->start(GraphExtractBuilder.size(), GraphExtractBuilder.getbytes(), paused);
- startJunction(resultJunction);
- }
- virtual void stop()
- {
- if (resultStream)
- resultStream->stop();
- CRoxieServerGraphLoopActivity::stop();
- }
- virtual void reset()
- {
- if (resultInput)
- resultInput->reset();
- resetJunction(resultJunction);
- resultInput = NULL;
- resultStream = NULL;
- resultJunction.clear();
- outputs.kill();
- iterationGraphs.kill(); // must be done after all activities killed
- if (probeManager)
- {
- probeManager->deleteGraph(NULL, (IArrayOf<IInputBase>*)&probes);
- probes.kill();
- }
- CRoxieServerGraphLoopActivity::reset();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * ret = resultStream->nextRow();
- if (ret)
- processed++;
- return ret;
- }
- void createExpandedGraph(unsigned parentExtractSize, const byte *parentExtract, IProbeManager *probeManager)
- {
- //result(0) is the input to the graph.
- resultInput = inputExtractMapper;
- outputs.append(* new CGraphIterationInfo(resultInput->queryActivity(), resultInput, 0, 1));
- for (createLoopCounter=1; createLoopCounter <= maxIterations; createLoopCounter++)
- {
- IRoxieServerChildGraph * graph = childGraph->createGraphLoopInstance(ctx, createLoopCounter, parentExtractSize, parentExtract, *this);
- graph->beforeExecute();
- iterationGraphs.append(*graph);
- graph->gatherIterationUsage(*this);
- CGraphIterationInfo *iteration = graph->selectGraphLoopOutput();
- outputs.append(*iteration);
- }
- createLoopCounter = 0;
- createSplitters(probeManager, probes);
- ForEachItemIn(i2, iterationGraphs)
- iterationGraphs.item(i2).associateIterationOutputs(*this);
- resultInput = outputs.tos().connectOutput(probeManager, probes, this, 0);
- resultStream = connectSingleStream(ctx, resultInput, 0, resultJunction, true);
- }
- void createSplitters(IProbeManager *probeManager, IArrayOf<IRoxieProbe> &probes)
- {
- ForEachItemIn(i, outputs)
- {
- CGraphIterationInfo & next = outputs.item(i);
- next.createSplitter(ctx, probeManager, probes);
- }
- }
- //IRoxieServerLoopResultProcessor
- virtual void noteUseIteration(unsigned _whichIteration)
- {
- int whichIteration = (int) _whichIteration; // May go negative - API is unsigned for historical reasons
- if (whichIteration >= 0)
- {
- if (!outputs.isItem(whichIteration))
- throw MakeStringException(ROXIE_GRAPH_PROCESSING_ERROR, "Error reading graph result %d from iteration %d", whichIteration, createLoopCounter);
- outputs.item(whichIteration).noteUsed();
- }
- }
- virtual IFinalRoxieInput* connectIterationOutput(unsigned whichIteration, IProbeManager *probeManager, IArrayOf<IRoxieProbe> &probes, IRoxieServerActivity *targetAct, unsigned targetIdx)
- {
- if (outputs.isItem(whichIteration))
- {
- CGraphIterationInfo & next = outputs.item(whichIteration);
- return next.connectOutput(probeManager, probes, targetAct, targetIdx);
- }
- return NULL;
- }
- };
- //=================================================================================
- class CRoxieServerGraphLoopActivityFactory : public CRoxieServerActivityFactory
- {
- unsigned loopGraphId;
- unsigned flags;
- Linked<IOutputMetaData> counterMeta;
- public:
- CRoxieServerGraphLoopActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _loopGraphId)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), loopGraphId(_loopGraphId)
- {
- Owned<IHThorGraphLoopArg> helper = (IHThorGraphLoopArg *) helperFactory();
- flags = helper->getFlags();
- counterMeta.setown(new CCounterRowMetaData);
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- if (kind == TAKparallelgraphloop)
- return new CRoxieServerParallelGraphLoopActivity(_ctx, this, _probeManager, loopGraphId, counterMeta);
- else
- return new CRoxieServerSequentialGraphLoopActivity(_ctx, this, _probeManager, loopGraphId, counterMeta);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerGraphLoopActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _loopGraphId)
- {
- return new CRoxieServerGraphLoopActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _loopGraphId);
- }
- //=====================================================================================================
- class CRoxieServerLibraryCallActivity : public CRoxieServerActivity
- {
- class OutputAdaptor : public CExtractMapperInput
- {
- bool stopped;
- public:
- CRoxieServerLibraryCallActivity *parent;
- IOutputMetaData * meta = nullptr;
- unsigned oid;
- unsigned processed;
- public:
- OutputAdaptor()
- {
- parent = NULL;
- oid = 0;
- init();
- }
- ~OutputAdaptor()
- {
- if (traceStartStop)
- DBGLOG("%p ~OutputAdaptor %d", this, oid);
- if (processed && parent && parent->factory)
- {
- parent->factory->noteProcessed(oid, processed);
- if (parent->ctx)
- parent->ctx->noteProcessed(parent->querySubgraphId(), parent->activityId, oid, processed, 0);
- }
- }
- void setParent(CRoxieServerLibraryCallActivity * _parent, IOutputMetaData * _meta, unsigned _oid)
- {
- parent = _parent;
- meta = _meta;
- oid = _oid;
- }
- void init()
- {
- processed = 0;
- stopped = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- parent->start(oid, parentExtractSize, parentExtract, paused);
- CExtractMapperInput::start(parentExtractSize, parentExtract, paused);
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- //MORE value of consumerOrdered depends on whether this activity is marked as ordered.
- if (!oid) // Only need to set up once
- {
- parent->connectInputStreams(true); // To handle dependencies etc
- }
- return CExtractMapperInput::getOutputStreams(ctx, 0, streams, nullptr, true, nullptr);
- }
- //This override should not really be needed, but input is set up too late for the call to connect strands
- virtual IOutputMetaData * queryOutputMeta() const
- {
- return meta;
- }
- virtual void stop()
- {
- if (!stopped)
- {
- stopped = true;
- parent->stop(oid); // parent code relies on stop being called exactly once per adaptor, so make sure it is!
- CExtractMapperInput::stop();
- }
- };
- virtual void reset()
- {
- parent->reset(oid);
- CExtractMapperInput::reset();
- init();
- };
- virtual void checkAbort()
- {
- parent->checkAbort();
- }
- };
-
- IHThorLibraryCallArg &helper;
- unsigned numInputs;
- unsigned numOutputs;
- unsigned numActiveOutputs;
- bool started;
- OutputAdaptor* outputAdaptors;
- CExtractMapperInput * * inputAdaptors;
- bool * inputUsed;
- bool * outputUsed;
- Owned<IException> error;
- CriticalSection crit;
- rtlRowBuilder libraryExtractBuilder;
- Owned<IActivityGraph> libraryGraph;
- const LibraryCallFactoryExtra & extra;
- public:
- CRoxieServerLibraryCallActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs, unsigned _numOutputs, const LibraryCallFactoryExtra & _extra)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- helper((IHThorLibraryCallArg &)basehelper), extra(_extra)
- {
- numInputs = _numInputs;
- numOutputs = _numOutputs;
- numActiveOutputs = numOutputs;
- inputAdaptors = new CExtractMapperInput*[numInputs];
- inputUsed = new bool[numInputs];
- for (unsigned i1 = 0; i1 < numInputs; i1++)
- {
- inputAdaptors[i1] = new CExtractMapperInput;
- inputUsed[i1] = false;
- }
- outputAdaptors = new OutputAdaptor[numOutputs];
- outputUsed = new bool[numOutputs];
- for (unsigned i2 = 0; i2 < numOutputs; i2++)
- {
- outputAdaptors[i2].setParent(this, helper.queryOutputMeta(i2), i2);
- outputUsed[i2] = false;
- }
- started = false;
- }
- ~CRoxieServerLibraryCallActivity()
- {
- for (unsigned i1 = 0; i1 < numInputs; i1++)
- ::Release(inputAdaptors[i1]);
- delete [] inputAdaptors;
- delete [] inputUsed;
- delete [] outputAdaptors;
- delete [] outputUsed;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- libraryGraph.setown(ctx->getLibraryGraph(extra, this));
- libraryGraph->onCreate(_colocalParent);
- //Now map the inputs and outputs to the adapters
- IRoxieServerChildGraph * graph = libraryGraph->queryLoopGraph();
- for (unsigned i1=0; i1<numInputs; i1++)
- inputUsed[i1] = graph->querySetInputResult(i1, 0, inputAdaptors[i1]);
- for (unsigned i2=0; i2<numOutputs; i2++)
- {
- unsigned outputIndex = extra.outputs.item(i2);
- Owned<IFinalRoxieInput> output = graph->selectOutput(numInputs+outputIndex);
- outputAdaptors[i2].setInput(0, output); // MORE - no idea if this 0 is right??
- outputAdaptors[i2].connectInputStreams(ctx, true);
- }
- }
- virtual void start(unsigned oid, unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CriticalBlock b(crit);
- if (error)
- throw error.getLink();
- if (factory)
- factory->noteStarted(oid);
- if (!started)
- {
- // even though it is not complete, we don't want to run this again if it fails.
- started = true;
- //see notes on splitter above
- try
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- catch (IException *E)
- {
- #ifdef TRACE_SPLIT
- CTXLOG("spill %d caught exception in start", activityId);
- #endif
- error.set(E);
- throw;
- }
- catch (...)
- {
- IException *E = MakeStringException(ROXIE_INTERNAL_ERROR, "Unknown exception caught in CRoxieServerLibraryCallActivity::start");
- error.set(E);
- throw E;
- }
- //recreate the parent extract, and use it to reinitialize the graphs...
- libraryExtractBuilder.clear();
- helper.createParentExtract(libraryExtractBuilder);
- // NOTE - do NOT set activeOutputs = numOutputs here - we must rely on the value set in reset and constructor. This is because we can see stop on
- // some inputs before we see start on others.
- for (unsigned i1 = 0; i1 < numInputs; i1++)
- {
- if (inputUsed[i1])
- inputAdaptors[i1]->setParentExtract(parentExtractSize, parentExtract);
- else
- inputAdaptors[i1]->stop();
- }
- for (unsigned i2 = 0; i2 < numOutputs; i2++)
- outputAdaptors[i2].setParentExtract(libraryExtractBuilder.size(), libraryExtractBuilder.getbytes());
- //call stop on all the unused inputs.
- IRoxieServerChildGraph * graph = libraryGraph->queryLoopGraph();
- graph->beforeExecute();
- ForEachItemIn(i3, extra.unusedOutputs)
- {
- Owned<IFinalRoxieInput> output = graph->selectOutput(numInputs+extra.unusedOutputs.item(i3));
- // output->queryStream().stop(); Is this needed??
- }
- }
- }
- virtual void stop(unsigned oid)
- {
- CriticalBlock b(crit);
- if (--numActiveOutputs == 0)
- {
- //call stop on all the unused inputs.
- IRoxieServerChildGraph * graph = libraryGraph->queryLoopGraph();
- graph->beforeExecute();
- ForEachItemIn(i3, extra.unusedOutputs)
- {
- Owned<IFinalRoxieInput> output = graph->selectOutput(numInputs+extra.unusedOutputs.item(i3));
- // Is this needed ?? output->queryStream().stop();
- }
- CRoxieServerActivity::stop();
- }
- }
- void reset(unsigned oid)
- {
- started = false;
- error.clear();
- numActiveOutputs = numOutputs;
- if (state != STATEreset) // make sure input is only reset once
- {
- CRoxieServerActivity::reset();
- libraryGraph->reset();
- //Call reset on all unused inputs/outputs from the graph - no one else will.
- for (unsigned i1 = 0; i1 < numInputs; i1++)
- {
- if (!inputUsed[i1])
- inputAdaptors[i1]->reset();
- }
- IRoxieServerChildGraph * graph = libraryGraph->queryLoopGraph();
- ForEachItemIn(i3, extra.unusedOutputs)
- {
- Owned<IFinalRoxieInput> output = graph->selectOutput(numInputs+extra.unusedOutputs.item(i3));
- output->reset();
- }
- }
- };
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- inputAdaptors[idx]->setInput(_sourceIdx, _in);
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (idx < numInputs && inputAdaptors[idx])
- return inputAdaptors[idx]->queryInput();
- else
- return NULL;
- }
- public:
- virtual const void *nextRow()
- {
- throwUnexpected(); // Internal logic error - we are not anybody's input
- }
- virtual IOutputMetaData * queryOutputMeta() const
- {
- throwUnexpected(); // should be called on outputs instead
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- assertex(idx!=(unsigned)-1);
- assertex(!outputUsed[idx]);
- outputUsed[idx] = true;
- return &outputAdaptors[idx];
- }
- };
- void LibraryCallFactoryExtra::set(const LibraryCallFactoryExtra & _other)
- {
- ForEachItemIn(i1, _other.outputs)
- outputs.append(_other.outputs.item(i1));
- ForEachItemIn(i2, _other.unusedOutputs)
- unusedOutputs.append(_other.unusedOutputs.item(i2));
- maxOutputs = _other.maxOutputs;
- graphid = _other.graphid;
- libraryName.set(_other.libraryName);
- embeddedGraphName.set(_other.embeddedGraphName);
- interfaceHash = _other.interfaceHash;
- embedded = _other.embedded;
- }
- void LibraryCallFactoryExtra::calcUnused()
- {
- for (unsigned i=0; i < maxOutputs; i++)
- if (!outputs.contains(i))
- unusedOutputs.append(i);
- }
- class CRoxieServerLibraryCallActivityFactory : public CRoxieServerMultiOutputFactory
- {
- private:
- CRoxieServerMultiInputInfo inputs;
- LibraryCallFactoryExtra extra;
- public:
- CRoxieServerLibraryCallActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, LibraryCallFactoryExtra & _extra)
- : CRoxieServerMultiOutputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- extra.set(_extra);
- extra.calcUnused();
- setNumOutputs(extra.outputs.ordinality());
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerLibraryCallActivity(_ctx, this, _probeManager, numInputs(), numOutputs, extra);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- inputs.set(idx, source, sourceidx);
- }
- virtual unsigned getInput(unsigned idx, unsigned &sourceidx) const
- {
- return inputs.get(idx, sourceidx);
- }
- virtual unsigned numInputs() const { return inputs.ordinality(); }
- virtual void getXrefInfo(IPropertyTree &reply, const IRoxieContextLogger &logctx) const
- {
- addXrefLibraryInfo(reply, extra.libraryName);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerLibraryCallActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, LibraryCallFactoryExtra & _extra)
- {
- return new CRoxieServerLibraryCallActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _extra);
- }
- //=====================================================================================================
- // CRoxieServerNWayInputActivity is a multi-input, multi-output activity, where the outputs are a subset of the inputs
- // Used to implement RANGE(<rowset-expression>, [set-of-indices])
- class CRoxieServerNWayInputBaseActivity : public CRoxieServerMultiInputBaseActivity
- {
- protected:
- PointerArrayOf<IFinalRoxieInput> selectedInputs;
- PointerArrayOf<IEngineRowStream> selectedStreams;
- PointerArrayOf<IStrandJunction> selectedJunctions;
- public:
- CRoxieServerNWayInputBaseActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerMultiInputBaseActivity(_ctx, _factory, _probeManager, _numInputs)
- {
- }
- virtual void stop()
- {
- ForEachItemIn(i2, selectedStreams)
- selectedStreams.item(i2)->stop();
- CRoxieServerMultiInputBaseActivity::stop();
- }
- virtual unsigned __int64 queryLocalCycles() const
- {
- __int64 localCycles = totalCycles.totalCycles;
- ForEachItemIn(i, selectedInputs)
- {
- localCycles -= selectedInputs.item(i)->queryTotalCycles();
- }
- if (localCycles < 0)
- localCycles = 0;
- return localCycles;
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (selectedInputs.isItem(idx))
- return selectedInputs.item(idx);
- else
- return NULL;
- }
- virtual void reset()
- {
- ForEachItemIn(i, selectedInputs)
- selectedInputs.item(i)->reset();
- selectedInputs.kill();
- selectedStreams.kill();
- selectedJunctions.kill();
- CRoxieServerMultiInputBaseActivity::reset();
- }
- virtual const void * nextRow()
- {
- throwUnexpected();
- }
- virtual unsigned numConcreteOutputs() const
- {
- return selectedInputs.ordinality();
- }
- virtual IFinalRoxieInput * queryConcreteInput(unsigned idx)
- {
- return queryInput(idx);
- }
- virtual IEngineRowStream *queryConcreteOutputStream(unsigned whichInput)
- {
- if (selectedStreams.isItem(whichInput))
- return selectedStreams.item(whichInput);
- return NULL;
- }
- virtual IStrandJunction *queryConcreteOutputJunction(unsigned idx) const
- {
- if (selectedJunctions.isItem(idx))
- return selectedJunctions.item(idx);
- return NULL;
- }
- };
- class CRoxieServerNWayInputActivity : public CRoxieServerNWayInputBaseActivity
- {
- IHThorNWayInputArg & helper;
- public:
- CRoxieServerNWayInputActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerNWayInputBaseActivity(_ctx, _factory, _probeManager, _numInputs), helper((IHThorNWayInputArg &)basehelper)
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerNWayInputBaseActivity::start(parentExtractSize, parentExtract, paused);
- bool selectionIsAll;
- size32_t selectionLen;
- rtlDataAttr selection;
- helper.getInputSelection(selectionIsAll, selectionLen, selection.refdata());
- selectedInputs.kill();
- selectedStreams.kill();
- selectedJunctions.kill();
- assertex(numInputs==numStreams); // Will need refactoring when that ceases to be true
- if (selectionIsAll)
- {
- for (unsigned i=0; i < numInputs; i++)
- {
- selectedInputs.append(inputArray[i]);
- selectedStreams.append(streamArray[i]); // Assumes 1:1 relationship - is that good?
- selectedJunctions.append(junctionArray[i]);
- }
- }
- else
- {
- const size32_t * selections = (const size32_t *)selection.getdata();
- unsigned max = selectionLen/sizeof(size32_t);
- for (unsigned i = 0; i < max; i++)
- {
- unsigned nextIndex = selections[i];
- //Check there are no duplicates..... Assumes there are a fairly small number of inputs, so n^2 search is ok.
- for (unsigned j=i+1; j < max; j++)
- {
- if (nextIndex == selections[j])
- throw MakeStringException(ROXIE_NWAY_INPUT_ERROR, "Selection list for nway input can not contain duplicates");
- }
- if (nextIndex > numInputs || nextIndex > numStreams)
- throw MakeStringException(ROXIE_NWAY_INPUT_ERROR, "Index %d in RANGE selection list is out of range", nextIndex);
- selectedInputs.append(inputArray[nextIndex-1]);
- selectedStreams.append(streamArray[nextIndex-1]); // Assumes 1:1 relationship - is that good?
- selectedJunctions.append(junctionArray[nextIndex-1]);
- }
- }
- // NB: Whatever pulls this nwayinput activity, starts and stops the selectedInputs and selectedJunctions
- }
- virtual void stop()
- {
- // NB: Whatever pulls this nwayinput activity, starts and stops the selectedInputs
- }
- };
- class CRoxieServerNWayInputActivityFactory : public CRoxieServerMultiInputFactory
- {
- public:
- CRoxieServerNWayInputActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerMultiInputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerNWayInputActivity(_ctx, this, _probeManager, numInputs());
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNWayInputActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerNWayInputActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- class CRoxieServerNWayGraphLoopResultReadActivity : public CRoxieServerNWayInputBaseActivity
- {
- IHThorNWayGraphLoopResultReadArg & helper;
- IArrayOf<IRoxieServerActivity> resultReaders;
- Owned<IStrandJunction> *resultJunctions;
- unsigned graphId;
- bool grouped;
- bool selectionIsAll;
- size32_t selectionLen;
- rtlDataAttr selection;
- public:
- CRoxieServerNWayGraphLoopResultReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _graphId)
- : CRoxieServerNWayInputBaseActivity(_ctx, _factory, _probeManager, 0), helper((IHThorNWayGraphLoopResultReadArg &)basehelper)
- {
- grouped = helper.isGrouped();
- graphId = _graphId;
- selectionIsAll = false;
- selectionLen = 0;
- resultJunctions = NULL;
- }
- ~CRoxieServerNWayGraphLoopResultReadActivity()
- {
- delete [] resultJunctions;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if (selectedInputs.ordinality() == 0)
- {
- initInputSelection();
- unsigned max = selectionLen / sizeof(size32_t);
- const size32_t * selections = (const size32_t *)selection.getdata();
- IProbeManager * probeManager = NULL; // MORE!!
- for (unsigned i = 0; i < max; i++)
- {
- CRoxieServerActivity * resultInput = new CRoxieServerInternalGraphLoopResultReadActivity(ctx, factory, probeManager, graphId, selections[i]);
- resultReaders.append(*resultInput);
- selectedInputs.append(resultInput);
- resultInput->onCreate(colocalParent);
- resultInput->start(parentExtractSize, parentExtract, paused);
- selectedStreams.append(connectSingleStream(ctx, resultInput, 0, resultJunctions[i], true));
- selectedJunctions.append(resultJunctions[i]);
- }
- }
- }
- virtual void reset()
- {
- resultReaders.kill();
- CRoxieServerNWayInputBaseActivity::reset();
- ForEachItemIn(i, selectedStreams)
- resetJunction(resultJunctions[i]);
- }
- virtual void gatherIterationUsage(IRoxieServerLoopResultProcessor & processor, unsigned parentExtractSize, const byte * parentExtract)
- {
- ensureCreated();
- basehelper.onStart(parentExtract, NULL);
- initInputSelection();
- unsigned max = selectionLen / sizeof(size32_t);
- const size32_t * selections = (const size32_t *)selection.getdata();
- for (unsigned i = 0; i < max; i++)
- processor.noteUseIteration(selections[i]);
- }
- virtual void associateIterationOutputs(IRoxieServerLoopResultProcessor & processor, unsigned parentExtractSize, const byte * parentExtract, IProbeManager *probeManager, IArrayOf<IRoxieProbe> &probes)
- {
- //selection etc. already initialised from the gatherIterationUsage() call.
- unsigned max = selectionLen / sizeof(size32_t);
- const size32_t * selections = (const size32_t *)selection.getdata();
- resultJunctions = new Owned<IStrandJunction> [max];
- for (unsigned i = 0; i < max; i++)
- {
- IFinalRoxieInput *in = processor.connectIterationOutput(selections[i], probeManager, probes, this, i);
- selectedInputs.append(in);
- selectedStreams.append(connectSingleStream(ctx, in, 0, resultJunctions[i], true));
- selectedJunctions.append(resultJunctions[i]);
- }
- }
- protected:
- void initInputSelection()
- {
- helper.getInputSelection(selectionIsAll, selectionLen, selection.refdata());
- if (selectionIsAll)
- throw MakeStringException(ROXIE_NWAY_INPUT_ERROR, "ALL not yet supported for NWay graph inputs");
- }
- };
- class CRoxieServerNWayGraphLoopResultReadActivityFactory : public CRoxieServerActivityFactory
- {
- unsigned graphId;
- public:
- CRoxieServerNWayGraphLoopResultReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _graphId)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), graphId(_graphId)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerNWayGraphLoopResultReadActivity(_ctx, this, _probeManager, graphId);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for NWay GraphLoopResultRead activity");
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNWayGraphLoopResultReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned graphId)
- {
- return new CRoxieServerNWayGraphLoopResultReadActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, graphId);
- }
- //=================================================================================
- class RoxieSteppedInput : implements ISteppedInput, public CInterface
- {
- public:
- RoxieSteppedInput(IFinalRoxieInput * _input, IEngineRowStream *_stream) { input = _input; inputStream = _stream; }
- IMPLEMENT_IINTERFACE
- protected:
- virtual const void * nextInputRow()
- {
- #ifdef TRACE_SEEK_REQUESTS
- IRoxieContextLogger * logger = input->queryActivity();
- const void * ret = doNextInputRow();
- {
- CommonXmlWriter xmlwrite(XWFtrim|XWFopt|XWFnoindent);
- if (!ret)
- xmlwrite.outputBool(true,"eof");
- else if (input->queryOutputMeta()->hasXML())
- input->queryOutputMeta()->toXML((byte *) ret, xmlwrite);
- logger->CTXLOG("next() returns (%s)", xmlwrite.str());
- }
- return ret;
- #else
- return inputStream->ungroupedNextRow();
- #endif
- }
- virtual const void * nextInputRowGE(const void * seek, unsigned numFields, bool & wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- #ifdef TRACE_SEEK_REQUESTS
- IRoxieContextLogger * logger = input->queryActivity();
- {
- CommonXmlWriter xmlwrite(XWFtrim|XWFopt|XWFnoindent);
- if (input->queryOutputMeta()->hasXML())
- input->queryOutputMeta()->toXML((byte *) seek, xmlwrite);
- logger->CTXLOG("nextInputRowGE(%d, %s%s%s, %s) seek(%s)",
- numFields,
- stepExtra.readAheadManyResults() ? "readahead " : "",
- stepExtra.returnMismatches() ? "mismatch" : "exact",
- stepExtra.onlyReturnFirstSeekMatch() ? " single-match" : "",
- stepExtra.queryExtraSeeks() ? "multi-seek":"",
- xmlwrite.str());
- }
- const void * ret = doNextInputRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- {
- CommonXmlWriter xmlwrite(XWFtrim|XWFopt|XWFnoindent);
- if (!ret)
- xmlwrite.outputBool(true,"eof");
- else if (input->queryOutputMeta()->hasXML())
- input->queryOutputMeta()->toXML((byte *) ret, xmlwrite);
- logger->CTXLOG("nextInputRowGE(%d, %s%s%s, %s) result(%s)",
- numFields,
- stepExtra.readAheadManyResults() ? "readahead " : "",
- stepExtra.returnMismatches() ? "mismatch" : "exact",
- stepExtra.onlyReturnFirstSeekMatch() ? " single-match" : "",
- stepExtra.queryExtraSeeks() ? "multi-seek":"",
- xmlwrite.str());
- }
- return ret;
- #else
- return doNextInputRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- #endif
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- inputStream->resetEOF();
- }
- virtual IInputSteppingMeta * queryInputSteppingMeta()
- {
- return input->querySteppingMeta();
- }
- inline const void * doNextInputRowGE(const void * seek, unsigned numFields, bool & wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- assertex(wasCompleteMatch);
- return inputStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- }
- protected:
- IFinalRoxieInput * input;
- IEngineRowStream * inputStream;
- };
- //=================================================================================
- class CRoxieServerNWayBaseActivity : public CRoxieServerMultiInputBaseActivity
- {
- public:
- CRoxieServerNWayBaseActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerMultiInputBaseActivity(_ctx, _factory, _probeManager, _numInputs)
- {
- }
- void startExpandedInputs(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- ForEachItemIn(ei, expandedInputs)
- expandedInputs.item(ei)->start(parentExtractSize, parentExtract, paused);
- ForEachItemIn(idx, expandedInputs)
- startJunction(expandedJunctions.item(idx));
- }
- virtual void stop()
- {
- ForEachItemIn(i, expandedStreams)
- expandedStreams.item(i)->stop();
- CRoxieServerMultiInputBaseActivity::stop();
- }
- virtual void reset()
- {
- ForEachItemIn(idx, expandedInputs)
- resetJunction(expandedJunctions.item(idx));
- expandedInputs.kill();
- expandedStreams.kill();
- expandedJunctions.kill();
- CRoxieServerMultiInputBaseActivity::reset();
- }
- protected:
- PointerArrayOf<IFinalRoxieInput> expandedInputs;
- PointerArrayOf<IEngineRowStream> expandedStreams;
- PointerArrayOf<IStrandJunction> expandedJunctions;
- };
- //=================================================================================
- class CRoxieServerNaryActivity : public CRoxieServerNWayBaseActivity
- {
- public:
- CRoxieServerNaryActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerNWayBaseActivity(_ctx, _factory, _probeManager, _numInputs)
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerNWayBaseActivity::start(parentExtractSize, parentExtract, paused);
- for (unsigned i=0; i < numInputs; i++)
- {
- IFinalRoxieInput * cur = inputArray[i];
- CRoxieServerNWayInputBaseActivity *nWayInput = dynamic_cast<CRoxieServerNWayInputBaseActivity *>(cur);
- if (nWayInput)
- {
- nWayInput->start(parentExtractSize, parentExtract, paused);
- unsigned numRealInputs = cur->numConcreteOutputs();
- for (unsigned j = 0; j < numRealInputs; j++)
- {
- expandedInputs.append(cur->queryConcreteInput(j));
- expandedStreams.append(cur->queryConcreteOutputStream(j));
- expandedJunctions.append(cur->queryConcreteOutputJunction(j));
- }
- }
- else
- {
- expandedInputs.append(cur);
- // NB: this activities input streams + junction have been setup and held in CRoxieServerMultiInputActivity base
- expandedStreams.append(queryConcreteOutputStream(i));
- expandedJunctions.append(queryConcreteOutputJunction(i));
- }
- }
- startExpandedInputs(parentExtractSize, parentExtract, paused);
- }
- };
- //=================================================================================
- class CRoxieStreamMerger : public CStreamMerger
- {
- public:
- CRoxieStreamMerger() : CStreamMerger(true)
- {
- streamArray = NULL;
- }
- void initInputs(unsigned _numInputs, IEngineRowStream ** _streamArray)
- {
- CStreamMerger::initInputs(_numInputs);
- streamArray = _streamArray;
- }
- virtual bool pullInput(unsigned i, const void * seek, unsigned numFields, const SmartStepExtra * stepExtra)
- {
- const void * next;
- bool matches = true;
- if (seek)
- next = streamArray[i]->nextRowGE(seek, numFields, matches, *stepExtra);
- else
- next = streamArray[i]->ungroupedNextRow();
- pending[i] = next;
- pendingMatches[i] = matches;
- return (next != NULL);
- }
- virtual void releaseRow(const void * row)
- {
- ReleaseRoxieRow(row);
- }
- protected:
- IEngineRowStream **streamArray;
- };
- class CRoxieServerNWayMergeActivity : public CRoxieServerNaryActivity
- {
- public:
- CRoxieServerNWayMergeActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerNaryActivity(_ctx, _factory, _probeManager, _numInputs),
- helper((IHThorNWayMergeArg &)basehelper)
- {
- initializedMeta = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerNaryActivity::start(parentExtractSize, parentExtract, paused);
- merger.init(helper.queryCompare(), helper.dedup(), helper.querySteppingMeta()->queryCompare());
- merger.initInputs(expandedStreams.length(), expandedStreams.getArray());
- }
- virtual void stop()
- {
- merger.done();
- CRoxieServerNaryActivity::stop();
- }
- virtual void reset()
- {
- merger.cleanup();
- CRoxieServerNaryActivity::reset();
- initializedMeta = false;
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * next = merger.nextRow();
- if (next)
- processed++;
- return next;
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool & wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * next = merger.nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- if (next)
- processed++;
- return next;
- }
- virtual IInputSteppingMeta * querySteppingMeta()
- {
- if (expandedStreams.ordinality() == 0)
- return NULL;
- if (!initializedMeta)
- {
- meta.init(helper.querySteppingMeta(), false);
- ForEachItemIn(i, expandedStreams)
- {
- if (meta.getNumFields() == 0)
- break;
- IInputSteppingMeta * inputMeta = expandedInputs.item(i)->querySteppingMeta();
- meta.intersect(inputMeta);
- }
- initializedMeta = true;
- }
- if (meta.getNumFields() == 0)
- return NULL;
- return &meta;
- }
- protected:
- IHThorNWayMergeArg &helper;
- CRoxieStreamMerger merger;
- CSteppingMeta meta;
- bool initializedMeta;
- };
- class CRoxieServerNWayMergeActivityFactory : public CRoxieServerMultiInputFactory
- {
- public:
- CRoxieServerNWayMergeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerMultiInputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerNWayMergeActivity(_ctx, this, _probeManager, numInputs());
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNWayMergeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerNWayMergeActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerNWayMergeJoinActivity : public CRoxieServerNaryActivity
- {
- public:
- CRoxieServerNWayMergeJoinActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs, CMergeJoinProcessor & _processor)
- : CRoxieServerNaryActivity(_ctx, _factory, _probeManager, _numInputs),processor(_processor),
- helper((IHThorNWayMergeJoinArg &)basehelper)
- {
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerNaryActivity::onCreate(_colocalParent);
- ICodeContext * codectx = ctx->queryCodeContext();
- inputAllocator.setown(createRowAllocator(helper.queryInputMeta()));
- outputAllocator.setown(createRowAllocator(helper.queryOutputMeta()));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerNaryActivity::start(parentExtractSize, parentExtract, paused);
- ForEachItemIn(i1, expandedInputs)
- {
- Owned<RoxieSteppedInput> stepInput = new RoxieSteppedInput(expandedInputs.item(i1), expandedStreams.item(i1));
- processor.addInput(stepInput);
- }
- processor.beforeProcessing(inputAllocator, outputAllocator);
- }
- virtual void stop()
- {
- processor.afterProcessing();
- CRoxieServerNaryActivity::stop();
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return processor.gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- processor.queryResetEOF();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * next = processor.nextRow();
- if (next)
- processed++;
- return next;
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * next = processor.nextGE(seek, numFields, wasCompleteMatch, stepExtra);
- if (next)
- processed++;
- return next;
- }
- virtual IInputSteppingMeta * querySteppingMeta()
- {
- return processor.queryInputSteppingMeta();
- }
- protected:
- IHThorNWayMergeJoinArg & helper;
- CMergeJoinProcessor & processor;
- Owned<IEngineRowAllocator> inputAllocator;
- Owned<IEngineRowAllocator> outputAllocator;
- };
- class CRoxieServerAndMergeJoinActivity : public CRoxieServerNWayMergeJoinActivity
- {
- public:
- CRoxieServerAndMergeJoinActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerNWayMergeJoinActivity(_ctx, _factory, _probeManager, _numInputs, andProcessor), andProcessor(helper)
- {
- }
- protected:
- CAndMergeJoinProcessor andProcessor;
- };
- class CRoxieServerAndLeftMergeJoinActivity : public CRoxieServerNWayMergeJoinActivity
- {
- public:
- CRoxieServerAndLeftMergeJoinActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerNWayMergeJoinActivity(_ctx, _factory, _probeManager, _numInputs, andLeftProcessor), andLeftProcessor(helper)
- {
- }
- protected:
- CAndLeftMergeJoinProcessor andLeftProcessor;
- };
- class CRoxieServerMofNMergeJoinActivity : public CRoxieServerNWayMergeJoinActivity
- {
- public:
- CRoxieServerMofNMergeJoinActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerNWayMergeJoinActivity(_ctx, _factory, _probeManager, _numInputs, mofnProcessor), mofnProcessor(helper)
- {
- }
- protected:
- CMofNMergeJoinProcessor mofnProcessor;
- };
- class CRoxieServerProximityJoinActivity : public CRoxieServerNWayMergeJoinActivity
- {
- public:
- CRoxieServerProximityJoinActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerNWayMergeJoinActivity(_ctx, _factory, _probeManager, _numInputs, proximityProcessor), proximityProcessor(helper)
- {
- }
- protected:
- CProximityJoinProcessor proximityProcessor;
- };
- class CRoxieServerNWayMergeJoinActivityFactory : public CRoxieServerMultiInputFactory
- {
- unsigned flags;
- public:
- CRoxieServerNWayMergeJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerMultiInputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- Owned<IHThorNWayMergeJoinArg> helper = (IHThorNWayMergeJoinArg *) helperFactory();
- flags = helper->getJoinFlags();
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- if (flags & IHThorNWayMergeJoinArg::MJFhasrange)
- return new CRoxieServerProximityJoinActivity(_ctx, this, _probeManager, numInputs());
- switch (flags & IHThorNWayMergeJoinArg::MJFkindmask)
- {
- case IHThorNWayMergeJoinArg::MJFinner:
- return new CRoxieServerAndMergeJoinActivity(_ctx, this, _probeManager, numInputs());
- case IHThorNWayMergeJoinArg::MJFleftonly:
- case IHThorNWayMergeJoinArg::MJFleftouter:
- return new CRoxieServerAndLeftMergeJoinActivity(_ctx, this, _probeManager, numInputs());
- case IHThorNWayMergeJoinArg::MJFmofn:
- return new CRoxieServerMofNMergeJoinActivity(_ctx, this, _probeManager, numInputs());
- default:
- throwUnexpected();
- }
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNWayMergeJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerNWayMergeJoinActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerNWaySelectActivity : public CRoxieServerNWayBaseActivity
- {
- IHThorNWaySelectArg &helper;
- public:
- CRoxieServerNWaySelectActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerNWayBaseActivity(_ctx, _factory, _probeManager, _numInputs),
- helper((IHThorNWaySelectArg &)basehelper)
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerNWayBaseActivity::start(parentExtractSize, parentExtract, paused);
- unsigned whichInput = helper.getInputIndex();
- selectedInput = nullptr;
- selectedStream = nullptr;
- if (whichInput--)
- {
- for (unsigned i=0; i < numInputs; i++)
- {
- IFinalRoxieInput * cur = inputArray[i];
- CRoxieServerNWayInputBaseActivity *nWayInput = dynamic_cast<CRoxieServerNWayInputBaseActivity *>(cur);
- if (nWayInput)
- {
- nWayInput->start(parentExtractSize, parentExtract, paused);
- unsigned numRealInputs = cur->numConcreteOutputs();
- if (whichInput < numRealInputs)
- {
- expandedInputs.append(cur->queryConcreteInput(whichInput));
- expandedStreams.append(cur->queryConcreteOutputStream(whichInput));
- expandedJunctions.append(cur->queryConcreteOutputJunction(whichInput));
- break;
- }
- whichInput -= numRealInputs;
- }
- else
- {
- if (whichInput == 0)
- {
- expandedInputs.append(cur);
- // NB: this activities input streams + junction have been setup and held in CRoxieServerMultiInputActivity base
- expandedStreams.append(queryConcreteOutputStream(i));
- expandedJunctions.append(queryConcreteOutputJunction(i));
- break;
- }
- whichInput -= 1;
- }
- }
- }
- if (expandedInputs.ordinality())
- {
- startExpandedInputs(parentExtractSize, parentExtract, paused);
- selectedInput = expandedInputs.item(0);
- selectedStream = expandedStreams.item(0);
- }
- }
- const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (!selectedStream)
- return NULL;
- return selectedStream->nextRow();
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- if (!selectedInput)
- return false;
- return selectedInput->gatherConjunctions(collector);
- }
-
- virtual void resetEOF()
- {
- if (selectedStream)
- selectedStream->resetEOF();
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (!selectedStream)
- return NULL;
- return selectedStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- }
- IInputSteppingMeta * querySteppingMeta()
- {
- if (selectedInput)
- return selectedInput->querySteppingMeta();
- return NULL;
- }
- protected:
- IFinalRoxieInput * selectedInput = nullptr;
- IEngineRowStream * selectedStream = nullptr;
- Owned<IStrandJunction> selectedJunction;
- };
- class CRoxieServerNWaySelectActivityFactory : public CRoxieServerMultiInputFactory
- {
- public:
- CRoxieServerNWaySelectActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerMultiInputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerNWaySelectActivity(_ctx, this, _probeManager, numInputs());
- }
- };
- IRoxieServerActivityFactory *createRoxieServerNWaySelectActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerNWaySelectActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerRemoteActivity : public CRoxieServerActivity, implements IRoxieServerErrorHandler
- {
- protected:
- IHThorRemoteArg &helper;
- CRemoteResultAdaptor remote;
- public:
- CRoxieServerRemoteActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteID)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- helper((IHThorRemoteArg &)basehelper),
- remote(_ctx, this, _remoteID, meta.queryOriginal(), helper, *this, false, false) // MORE - if they need it stable we'll have to think!
- {
- }
- virtual const IResolvedFile *queryVarFileInfo() const
- {
- return NULL;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- remote.onCreate(_colocalParent);
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- remote.onStart(parentExtractSize, parentExtract);
- remote.setLimits(helper.getRowLimit(), (unsigned __int64) -1, I64C(0x7FFFFFFFFFFFFFFF));
- unsigned fileNo = 0; // MORE - superfiles require us to do this per file part... maybe (needs thought)
- remote.getMem(0, fileNo, 0); // the cached context is all we need to send
- remote.flush();
- remote.senddone();
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() called for source activity");
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- if (idx==(unsigned)-1)
- idx = 0;
- return idx ? NULL: &remote;
- }
- virtual void reset()
- {
- processed = remote.processed;
- remote.processed = 0;
- CRoxieServerActivity::reset();
- }
- virtual void onLimitExceeded(bool isKeyed)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- helper.onLimitExceeded();
- }
- virtual const void *createLimitFailRow(bool isKeyed)
- {
- UNIMPLEMENTED; // MORE - is there an ONFAIL for a limit folded into a remote?
- }
- virtual const void *nextRow()
- {
- throwUnexpected(); // I am nobody's input
- }
- };
- class CRoxieServerRemoteActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- RemoteActivityId remoteId;
- bool isRoot;
- CRoxieServerRemoteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), remoteId(_remoteId), isRoot(_isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerRemoteActivity(_ctx, this, _probeManager, remoteId);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for %s activity", getActivityText(kind));
- }
- virtual bool isSink() const
- {
- //I don't think the action version of this is implemented - but this would be the code
- return isRoot && !meta.queryOriginal();
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return allStatistics; // Child queries...
- }
- };
- IRoxieServerActivityFactory *createRoxieServerRemoteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId, bool _isRoot)
- {
- return new CRoxieServerRemoteActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId, _isRoot);
- }
- //=================================================================================
- class CRoxieServerIterateActivity : public CRoxieServerActivity
- {
- IHThorIterateArg &helper;
- OwnedConstRoxieRow defaultRecord;
- OwnedConstRoxieRow left;
- OwnedConstRoxieRow right;
- unsigned counter;
- public:
- CRoxieServerIterateActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- helper((IHThorIterateArg &)basehelper)
- {
- counter = 0;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- counter = 0;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t thisSize = helper.createDefault(rowBuilder);
- defaultRecord.setown(rowBuilder.finalizeRowClear(thisSize));
- }
- virtual void reset()
- {
- defaultRecord.clear();
- right.clear();
- left.clear();
- CRoxieServerActivity::reset();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- right.setown(inputStream->nextRow());
- if (!right)
- {
- bool skippedGroup = (left == NULL) && (counter > 0); //we have just skipped entire group, but shouldn't output a double null
- left.clear();
- counter = 0;
- if (skippedGroup) continue;
- return NULL;
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned outSize = helper.transform(rowBuilder, left ? left : defaultRecord, right, ++counter);
- if (outSize)
- {
- left.setown(rowBuilder.finalizeRowClear(outSize));
- processed++;
- return left.getLink();
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- };
- class CRoxieServerIterateActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerIterateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerIterateActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerIterateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerIterateActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerProcessActivity : public CRoxieServerActivity
- {
- IHThorProcessArg &helper;
- OwnedConstRoxieRow curRight;
- OwnedConstRoxieRow initialRight;
- unsigned counter;
- Owned<IEngineRowAllocator> rightRowAllocator;
- public:
- CRoxieServerProcessActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorProcessArg &)basehelper)
- {
- counter = 0;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- rightRowAllocator.setown(createRowAllocator(helper.queryRightRecordSize()));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- counter = 0;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- RtlDynamicRowBuilder rowBuilder(rightRowAllocator);
- size32_t thisSize = helper.createInitialRight(rowBuilder);
- initialRight.setown(rowBuilder.finalizeRowClear(thisSize));
- curRight.set(initialRight);
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- try
- {
- loop
- {
- const void * in = inputStream->nextRow();
- if (!in)
- {
- bool eog = (curRight != initialRight); // processed any records?
- counter = 0;
- curRight.set(initialRight);
- if (eog)
- return NULL;
- in = inputStream->nextRow();
- if (!in)
- return NULL;
- }
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- RtlDynamicRowBuilder rightRowBuilder(rightRowAllocator);
- size32_t outSize = helper.transform(rowBuilder, rightRowBuilder, in, curRight, ++counter);
- ReleaseRoxieRow(in);
- if (outSize)
- {
- //MORE: This should be returned...
- size32_t rightSize = rightRowAllocator->queryOutputMeta()->getRecordSize(rightRowBuilder.getSelf());
- curRight.setown(rightRowBuilder.finalizeRowClear(rightSize));
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- };
- class CRoxieServerProcessActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerProcessActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerProcessActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerProcessActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerProcessActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerGroupActivity : public CRoxieServerActivity
- {
- IHThorGroupArg &helper;
- bool endPending;
- bool eof;
- bool first;
- unsigned numGroups;
- unsigned numGroupMax;
- unsigned numProcessedLastGroup;
- const void *next;
- public:
- CRoxieServerGroupActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorGroupArg &)basehelper)
- {
- next = NULL;
- endPending = false;
- eof = false;
- first = true;
- numGroups = 0;
- numGroupMax = 0;
- numProcessedLastGroup = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- endPending = false;
- eof = false;
- first = true;
- numGroups = 0;
- numGroupMax = 0;
- numProcessedLastGroup = processed;
- assertex(next == NULL);
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- noteStatistic(StNumGroups, numGroups);
- noteStatistic(StNumGroupMax, numGroupMax);
- ReleaseClearRoxieRow(next);
- CRoxieServerActivity::reset();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (first)
- {
- next = inputStream->nextRow();
- first = false;
- }
- if (eof || endPending)
- {
- endPending = false;
- return NULL;
- }
- const void * prev = next;
- next = inputStream->ungroupedNextRow();
- if (next)
- {
- assertex(prev);
- if (!helper.isSameGroup(prev, next))
- {
- noteEndOfGroup();
- endPending = true;
- }
- }
- else
- {
- noteEndOfGroup();
- eof = true;
- }
- if (prev)
- processed++;
- return prev;
- }
- inline void noteEndOfGroup()
- {
- unsigned numThisGroup = processed - numProcessedLastGroup;
- if (numThisGroup == 0)
- return;
- numProcessedLastGroup = processed;
- if (numThisGroup > numGroupMax)
- numGroupMax = numThisGroup;
- numGroups++;
- }
- };
- class CRoxieServerGroupActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerGroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerGroupActivity(_ctx, this, _probeManager);
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return groupStatistics;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerGroupActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerGroupActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerFirstNActivity : public CRoxieServerLateStartActivity
- {
- unsigned __int64 limit;
- unsigned __int64 skip;
- unsigned doneThisGroup;
- IHThorFirstNArg &helper;
- public:
- CRoxieServerFirstNActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerLateStartActivity(_ctx, _factory, _probeManager), helper((IHThorFirstNArg &)basehelper)
- {
- doneThisGroup = 0;
- limit = 0;
- skip = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- doneThisGroup = 0;
- CRoxieServerLateStartActivity::start(parentExtractSize, parentExtract, paused);
- limit = helper.getLimit();
- skip = helper.numToSkip();
- lateStart(parentExtractSize, parentExtract, limit > 0);
- if (limit + skip >= limit)
- limit += skip;
- }
- const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- const void *ret;
- loop
- {
- ret = inputStream->nextRow();
- if (!ret)
- {
- if (meta.isGrouped())
- {
- if (doneThisGroup > skip)
- {
- doneThisGroup = 0;
- return NULL;
- }
- doneThisGroup = 0;
- }
- ret = inputStream->nextRow();
- if (!ret)
- {
- eof = true;
- return NULL;
- }
- }
- doneThisGroup++;
- if (doneThisGroup > skip)
- break;
- ReleaseRoxieRow(ret);
- }
- if (doneThisGroup <= limit)
- {
- processed++;
- return ret;
- }
- ReleaseRoxieRow(ret);
- if (meta.isGrouped())
- {
- while ((ret = inputStream->nextRow()) != NULL)
- ReleaseRoxieRow(ret);
- doneThisGroup = 0;
- }
- else
- eof = true;
- return NULL;
- }
- };
- class CRoxieServerFirstNActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerFirstNActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerFirstNActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerFirstNActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerFirstNActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerSelectNActivity : public CRoxieServerActivity
- {
- bool done;
- IHThorSelectNArg &helper;
- public:
- CRoxieServerSelectNActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorSelectNArg &)basehelper)
- {
- done = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- done = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- const void *defaultRow()
- {
- if (!rowAllocator)
- ensureRowAllocator(); // We delay as often not needed...
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t thisSize = helper.createDefault(rowBuilder);
- return rowBuilder.finalizeRowClear(thisSize);
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (done)
- return NULL;
- done = true;
- processed++; // always going to return a row!
- unsigned __int64 index = helper.getRowToSelect();
- while (--index)
- {
- const void * next = inputStream->ungroupedNextRow();
- if (!next)
- return defaultRow();
- ReleaseRoxieRow(next);
- }
- const void * next = inputStream->ungroupedNextRow();
- if (!next)
- next = defaultRow();
- return next;
- }
- };
- class CRoxieServerSelectNActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerSelectNActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSelectNActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSelectNActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerSelectNActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerSelfJoinActivity : public CRoxieServerActivity
- {
- IHThorJoinArg &helper;
- ICompare *collate;
- OwnedRowArray group;
- bool matchedLeft;
- BoolArray matchedRight;
- bool eof;
- bool first;
- unsigned leftIndex;
- unsigned rightIndex;
- unsigned rightOuterIndex;
- unsigned joinLimit;
- unsigned atmostLimit;
- unsigned atmostsTriggered;
- unsigned abortLimit;
- unsigned keepLimit;
- unsigned joinCounter;
- bool leftOuterJoin;
- bool rightOuterJoin;
- bool exclude;
- bool limitFail;
- bool limitOnFail;
- bool cloneLeft;
- bool forceSpill;
- OwnedConstRoxieRow defaultLeft;
- OwnedConstRoxieRow defaultRight;
- OwnedConstRoxieRow lhs;
- Owned<IException> failingLimit;
- bool failingOuterAtmost;
- Owned<IEngineRowAllocator> defaultAllocator;
- Owned<IRHLimitedCompareHelper> limitedhelper;
- Owned<CRHDualCache> dualcache;
- Owned<IGroupedInput> groupedInput;
- IRowStream *dualCacheInput;
- bool fillGroup()
- {
- group.clear();
- matchedLeft = false;
- matchedRight.kill();
- failingOuterAtmost = false;
- const void * next;
- unsigned groupCount = 0;
- while((next = groupedInput->nextRow()) != NULL)
- {
- if(groupCount==abortLimit)
- {
- if(limitFail)
- failLimit(next);
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- if(limitOnFail)
- {
- assertex(!failingLimit);
- try
- {
- failLimit(next);
- }
- catch(IException * except)
- {
- failingLimit.setown(except);
- }
- assertex(failingLimit != NULL);
- group.append(next);
- groupCount++;
- break;
- }
- group.clear();
- groupCount = 0;
- while(next)
- {
- ReleaseRoxieRow(next);
- next = groupedInput->nextRow();
- }
- }
- else if(groupCount==atmostLimit)
- {
- atmostsTriggered++;
- if(leftOuterJoin)
- {
- group.append(next);
- groupCount++;
- failingOuterAtmost = true;
- break;
- }
- else
- {
- group.clear();
- groupCount = 0;
- while (next)
- {
- ReleaseRoxieRow(next);
- next = groupedInput->nextRow();
- }
- }
- }
- else
- {
- group.append(next);
- groupCount++;
- }
- }
- if(group.ordinality()==0)
- {
- eof = true;
- return false;
- }
- leftIndex = 0;
- rightIndex = 0;
- rightOuterIndex = 0;
- joinCounter = 0;
- joinLimit = keepLimit;
- ForEachItemIn(idx, group)
- matchedRight.append(false);
- return true;
- }
- void failLimit(const void * next)
- {
- helper.onMatchAbortLimitExceeded();
- CommonXmlWriter xmlwrite(XWFtrim|XWFopt );
- if (!ctx->isBlind() && input->queryOutputMeta() && input->queryOutputMeta()->hasXML())
- {
- input->queryOutputMeta()->toXML((byte *) next, xmlwrite);
- }
- throw MakeStringException(ROXIE_TOO_MANY_RESULTS, "More than %d match candidates in self-join %d for row %s", abortLimit, queryId(), xmlwrite.str());
- }
- virtual bool needsAllocator() const { return true; }
- const void *joinRecords(const void * curLeft, const void * curRight, unsigned counter, IException * except)
- {
- try
- {
- if (cloneLeft && !except)
- {
- LinkRoxieRow(curLeft);
- return curLeft;
- }
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outsize = except ? helper.onFailTransform(rowBuilder, curLeft, curRight, except) : helper.transform(rowBuilder, curLeft, curRight, counter);
- if (outsize)
- return rowBuilder.finalizeRowClear(outsize);
- else
- return NULL;
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- void createDefaultLeft()
- {
- if (!defaultLeft)
- {
- if (!defaultAllocator)
- defaultAllocator.setown(createRowAllocator(input->queryOutputMeta()));
- RtlDynamicRowBuilder rowBuilder(defaultAllocator);
- size32_t thisSize = helper.createDefaultLeft(rowBuilder);
- defaultLeft.setown(rowBuilder.finalizeRowClear(thisSize));
- }
- }
- void createDefaultRight()
- {
- if (!defaultRight)
- {
- if (!defaultAllocator)
- defaultAllocator.setown(createRowAllocator(input->queryOutputMeta()));
- RtlDynamicRowBuilder rowBuilder(defaultAllocator);
- size32_t thisSize = helper.createDefaultRight(rowBuilder);
- defaultRight.setown(rowBuilder.finalizeRowClear(thisSize));
- }
- }
- public:
- CRoxieServerSelfJoinActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, bool _forceSpill)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorJoinArg &)basehelper), forceSpill(_forceSpill)
- {
- collate = helper.queryCompareLeftRight();
- eof = false;
- first = true;
- keepLimit = 0;
- atmostLimit = 0;
- atmostsTriggered = 0;
- unsigned joinFlags = helper.getJoinFlags();
- leftOuterJoin = (joinFlags & JFleftouter) != 0;
- rightOuterJoin = (joinFlags & JFrightouter) != 0;
- cloneLeft = (joinFlags & JFtransformmatchesleft) != 0;
- exclude = (joinFlags & JFexclude) != 0;
- abortLimit = 0;
- joinLimit = 0;
- assertex((joinFlags & (JFfirst | JFfirstleft | JFfirstright)) == 0); // no longer supported
- getLimitType(joinFlags, limitFail, limitOnFail);
- if((joinFlags & JFslidingmatch) != 0)
- throw MakeStringException(ROXIE_UNIMPLEMENTED_ERROR, "Internal Error: Sliding self join not supported");
- failingOuterAtmost = false;
- matchedLeft = false;
- leftIndex = 0;
- rightIndex = 0;
- rightOuterIndex = 0;
- joinCounter = 0;
- dualCacheInput = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- first = true;
- failingLimit.clear();
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- keepLimit = helper.getKeepLimit();
- if(keepLimit == 0)
- keepLimit = (unsigned)-1;
- atmostsTriggered = 0;
- atmostLimit = helper.getJoinLimit();
- if(atmostLimit == 0)
- atmostLimit = (unsigned)-1;
- else
- assertex(!rightOuterJoin);
- abortLimit = helper.getMatchAbortLimit();
- if (abortLimit == 0)
- abortLimit = (unsigned)-1;
- if (rightOuterJoin)
- createDefaultLeft();
- if (leftOuterJoin || limitOnFail)
- createDefaultRight();
- ICompare *compareLeft = helper.queryCompareLeft();
- if (helper.isLeftAlreadySorted())
- groupedInput.setown(createGroupedInputReader(inputStream, compareLeft));
- else
- {
- bool isStable = (helper.getJoinFlags() & JFunstable) == 0;
- RoxieSortAlgorithm sortAlgorithm;
- if (forceSpill)
- sortAlgorithm = isStable ? stableSpillingQuickSortAlgorithm : spillingQuickSortAlgorithm;
- else
- sortAlgorithm = isStable ? stableQuickSortAlgorithm : quickSortAlgorithm;
- groupedInput.setown(createSortedGroupedInputReader(inputStream, compareLeft, createSortAlgorithm(sortAlgorithm, compareLeft, ctx->queryRowManager(), input->queryOutputMeta(), ctx->queryCodeContext(), tempDirectory, activityId)));
- }
- if ((helper.getJoinFlags() & JFlimitedprefixjoin) && helper.getJoinLimit())
- { //limited match join (s[1..n])
- dualcache.setown(new CRHDualCache());
- dualcache->init(groupedInput);
- dualCacheInput = dualcache->queryOut1();
- failingOuterAtmost = false;
- matchedLeft = false;
- leftIndex = 0;
- rightOuterIndex = 0;
- joinCounter = 0;
- limitedhelper.setown(createRHLimitedCompareHelper());
- limitedhelper->init( helper.getJoinLimit(), dualcache->queryOut2(), collate, helper.queryPrefixCompare() );
- }
- }
- virtual void reset()
- {
- if (atmostsTriggered)
- noteStatistic(StNumAtmostTriggered, atmostsTriggered);
- group.clear();
- groupedInput.clear();
- CRoxieServerActivity::reset();
- defaultLeft.clear();
- defaultRight.clear();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (limitedhelper)
- {
- while(!eof) //limited match join
- {
- if (!group.isItem(rightIndex))
- {
- lhs.setown(dualCacheInput->nextRow());
- if (lhs)
- {
- rightIndex = 0;
- joinCounter = 0;
- group.clear();
- limitedhelper->getGroup(group,lhs);
- }
- else
- {
- eof = true;
- }
- }
- if (group.isItem(rightIndex))
- {
- const void * rhs = group.item(rightIndex++);
- if(helper.match(lhs, rhs))
- {
- const void * ret = joinRecords(lhs, rhs, ++joinCounter, NULL);
- if(ret)
- {
- processed++;
- return ret;
- }
- }
- }
- }
- return NULL;
- }
- else
- {
- if (first)
- {
- first = false;
- fillGroup();
- }
- while(!eof)
- {
- if(failingOuterAtmost)
- while(group.isItem(leftIndex))
- {
- const void * ret = joinRecords(group.item(leftIndex++), defaultRight, 0, NULL);
- if(ret)
- {
- processed++;
- return ret;
- }
- }
- if((joinLimit == 0) || !group.isItem(rightIndex))
- {
- if(leftOuterJoin && !matchedLeft && !failingLimit)
- {
- const void * ret = joinRecords(group.item(leftIndex), defaultRight, 0, NULL);
- if(ret)
- {
- matchedLeft = true;
- processed++;
- return ret;
- }
- }
- leftIndex++;
- matchedLeft = false;
- rightIndex = 0;
- joinCounter = 0;
- joinLimit = keepLimit;
- }
- if(!group.isItem(leftIndex))
- {
- if(failingLimit || failingOuterAtmost)
- {
- const void * lhs;
- while((lhs = groupedInput->nextRow()) != NULL) // dualCache never active here
- {
- const void * ret = joinRecords(lhs, defaultRight, 0, failingLimit);
- ReleaseRoxieRow(lhs);
- if(ret)
- {
- processed++;
- return ret;
- }
- }
- failingLimit.clear();
- }
- if(rightOuterJoin && !failingLimit)
- while(group.isItem(rightOuterIndex))
- if(!matchedRight.item(rightOuterIndex++))
- {
- const void * ret = joinRecords(defaultLeft, group.item(rightOuterIndex-1), 0, NULL);
- if(ret)
- {
- processed++;
- return ret;
- }
- }
- if(!fillGroup())
- return NULL;
- continue;
- }
- const void * lhs = group.item(leftIndex);
- if(failingLimit)
- {
- leftIndex++;
- const void * ret = joinRecords(lhs, defaultRight, 0, failingLimit);
- if(ret)
- {
- processed++;
- return ret;
- }
- }
- else
- {
- const void * rhs = group.item(rightIndex++);
- if(helper.match(lhs, rhs))
- {
- matchedLeft = true;
- matchedRight.replace(true, rightIndex-1);
- if(!exclude)
- {
- const void * ret = joinRecords(lhs, rhs, ++joinCounter, NULL);
- if(ret)
- {
- processed++;
- joinLimit--;
- return ret;
- }
- }
- }
- }
- }
- return NULL;
- }
- }
- };
- class CRoxieServerSelfJoinActivityFactory : public CRoxieServerActivityFactory
- {
- bool forceSpill;
- public:
- CRoxieServerSelfJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- forceSpill = queryFactory.queryOptions().allSortsMaySpill || _graphNode.getPropBool("hint[@name='spill']/@value", false);;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSelfJoinActivity(_ctx, this, _probeManager, forceSpill);
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return joinStatistics;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSelfJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerSelfJoinActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- class CRoxieServerLookupJoinActivity : public CRoxieServerTwoInputActivity
- {
- private:
- class LookupTable : public CInterface
- {
- public:
- LookupTable(IHThorHashJoinArg &helper)
- : leftRightCompare(helper.queryCompareLeftRight()), rightCompare(helper.queryCompareRight()),
- leftHash(helper.queryHashLeft()), rightHash(helper.queryHashRight())
- {
- size = 0;
- }
- virtual const void *find(const void * left) const = 0;
- virtual const void *findNext(const void * left) const = 0;
- protected:
- ICompare * leftRightCompare;
- ICompare * rightCompare;
- IHash * leftHash;
- IHash * rightHash;
- unsigned size;
- };
- class DedupLookupTable : public LookupTable
- {
- public:
- DedupLookupTable(ConstPointerArray &rightRows, IHThorHashJoinArg &helper)
- : LookupTable(helper)
- {
- size = (4*rightRows.length())/3 + 1;
- table = (const void * *)calloc(size, sizeof(void *)); // This should probably be allocated from roxiemem (and size rounded up to actual available size)
- ForEachItemIn(idx, rightRows)
- add(rightRows.item(idx));
- }
- ~DedupLookupTable()
- {
- roxiemem::ReleaseRoxieRowArray(size, table);
- free(table);
- }
- virtual const void *find(const void * left) const
- {
- unsigned index = leftHash->hash(left) % size;
- unsigned start = index;
- while (table[index])
- {
- if(leftRightCompare->docompare(left, table[index]) == 0)
- return table[index];
- index++;
- if (index==size)
- index = 0;
- if (index==start)
- throw MakeStringException(ROXIE_JOIN_ERROR, "Internal error in lookup join activity (hash table full on lookup)");
- }
- return NULL;
- }
- virtual const void *findNext(const void * left) const
- {
- return NULL;
- }
- protected:
- void add(const void * right)
- {
- unsigned index = rightHash->hash(right) % size;
- unsigned start = index;
- while (table[index])
- {
- if (rightCompare->docompare(table[index], right) == 0)
- {
- ReleaseRoxieRow(right);
- return;
- }
- index++;
- if (index==size)
- index = 0;
- if (index==start)
- throw MakeStringException(ROXIE_JOIN_ERROR, "Internal error in lookup join activity (hash table full on add)");
- }
- table[index] = right;
- }
- const void * * table;
- };
- class FewLookupTable : public LookupTable
- {
- public:
- FewLookupTable(ConstPointerArray &rightRows, IHThorHashJoinArg &helper)
- : LookupTable(helper)
- {
- size = (4*rightRows.length())/3 + 1;
- table = (const void * *)calloc(size, sizeof(void *)); // This should probably be allocated from roxiemem
- findex = fstart = BadIndex;
- ForEachItemIn(idx, rightRows)
- add(rightRows.item(idx));
- }
- ~FewLookupTable()
- {
- roxiemem::ReleaseRoxieRowArray(size, table);
- free(table);
- }
- virtual const void *find(const void * left) const
- {
- fstart = leftHash->hash(left) % size;
- findex = fstart;
- return doFind(left);
- }
- virtual const void *findNext(const void * left) const
- {
- if (findex == BadIndex)
- return NULL;
- advance();
- return doFind(left);
- }
- protected:
- void add(const void * right)
- {
- unsigned start = rightHash->hash(right) % size;
- unsigned index = start;
- while (table[index])
- {
- index++;
- if (index==size)
- index = 0;
- if (index==start)
- throwUnexpected(); //table is full, should never happen
- }
- table[index] = right;
- }
- void advance() const
- {
- findex++;
- if(findex==size)
- findex = 0;
- if(findex==fstart)
- throw MakeStringException(ROXIE_JOIN_ERROR, "Internal error in lookup join activity (hash table full on lookup)");
- }
- const void *doFind(const void * left) const
- {
- while(table[findex])
- {
- if (leftRightCompare->docompare(left, table[findex]) == 0)
- return table[findex];
- advance();
- }
- findex = BadIndex;
- return NULL;
- }
- const void * * table;
- unsigned mutable fstart;
- unsigned mutable findex;
- static unsigned const BadIndex;
- };
- class ManyLookupTable : public LookupTable
- {
- public:
- ManyLookupTable(ConstPointerArray &rightRows, IHThorHashJoinArg &helper)
- : LookupTable(helper)
- {
- rightRows.swapWith(rowtable);
- UInt64Array groups;
- unsigned numRows = rowtable.length();
- if (numRows)
- {
- unsigned groupStart = 0;
- const void *groupStartRow = rowtable.item(0);
- for (unsigned i=1; i < numRows; i++)
- {
- const void *thisRow = rowtable.item(i);
- if (rightCompare->docompare(groupStartRow, thisRow))
- {
- groups.append(makeint64(groupStart, i-groupStart));
- groupStart = i;
- groupStartRow = thisRow;
- }
- }
- groups.append(makeint64(groupStart, numRows-groupStart));
- }
- size = (4*groups.length())/3 + 1;
- table = (__uint64 *) calloc(size, sizeof(__uint64)); // This should probably be allocated from roxiemem
- ForEachItemIn(idx, groups)
- {
- unsigned __int64 group = groups.item(idx);
- unsigned groupstart = high(group);
- const void *row = rowtable.item(groupstart);
- add(row, group);
- }
- }
- ~ManyLookupTable()
- {
- roxiemem::ReleaseRoxieRows(rowtable);
- free(table);
- }
- void add(const void *row, unsigned __int64 group)
- {
- unsigned start = rightHash->hash(row) % size;
- unsigned index = start;
- while (table[index])
- {
- index++;
- if (index==size)
- index = 0;
- if (index==start)
- throwUnexpected(); //table is full, should never happen
- }
- table[index] = group;
- }
- virtual const void *find(const void * left) const
- {
- unsigned index = leftHash->hash(left) % size;
- unsigned start = index;
- while (table[index])
- {
- __uint64 group = table[index];
- currentMatch = high(group);
- const void *right = rowtable.item(currentMatch);
- if (leftRightCompare->docompare(left, right) == 0)
- {
- currentMatch++;
- matchCount = low(group) - 1;
- return right;
- }
- index++;
- if (index==size)
- index = 0;
- if (index==start)
- throw MakeStringException(ROXIE_JOIN_ERROR, "Internal error in lookup join activity (hash table full on lookup)");
- }
- matchCount = 0;
- return NULL;
- }
- virtual const void *findNext(const void * left) const
- {
- if (!matchCount)
- return NULL;
- matchCount--;
- return rowtable.item(currentMatch++);
- }
- protected:
- __uint64 *table;
- ConstPointerArray rowtable;
- mutable unsigned currentMatch = 0;
- mutable unsigned matchCount = 0;
- };
- IHThorHashJoinArg &helper;
- bool leftOuterJoin;
- bool exclude;
- bool eog;
- bool many;
- bool dedupRHS;
- bool useFewTable;
- bool matchedGroup;
- const void *left;
- OwnedConstRoxieRow defaultRight;
- Owned<LookupTable> table;
- unsigned keepLimit;
- unsigned atmostLimit;
- unsigned atmostsTriggered;
- unsigned limitLimit;
- unsigned joinCounter;
- bool limitFail;
- bool limitOnFail;
- bool hasGroupLimit;
- bool isSmartJoin;
- unsigned keepCount;
- bool gotMatch;
- bool cloneLeft;
- ConstPointerArray rightGroup;
- aindex_t rightGroupIndex;
- Owned<IException> failingLimit;
- ConstPointerArray filteredRight;
- ThorActivityKind activityKind;
- Owned<IEngineRowAllocator> defaultRightAllocator;
-
- void createDefaultRight()
- {
- if (!defaultRight)
- {
- if (!defaultRightAllocator)
- defaultRightAllocator.setown(createRowAllocator(input1->queryOutputMeta()));
- RtlDynamicRowBuilder rowBuilder(defaultRightAllocator);
- size32_t thisSize = helper.createDefaultRight(rowBuilder);
- defaultRight.setown(rowBuilder.finalizeRowClear(thisSize));
- }
- }
- public:
- CRoxieServerLookupJoinActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, bool _useFewTable)
- : CRoxieServerTwoInputActivity(_ctx, _factory, _probeManager), helper((IHThorHashJoinArg &)basehelper), useFewTable(_useFewTable)
- {
- unsigned joinFlags = helper.getJoinFlags();
- leftOuterJoin = (joinFlags & JFleftouter) != 0;
- assertex((joinFlags & JFrightouter) == 0);
- exclude = (joinFlags & JFexclude) != 0;
- many = (joinFlags & JFmanylookup) != 0;
- cloneLeft = (joinFlags & JFtransformmatchesleft) != 0;
- dedupRHS = (joinFlags & (JFmanylookup | JFmatchrequired | JFtransformMaySkip)) == 0; // optimisation: can implicitly dedup RHS unless is many lookup, or match required, or transform may skip
- left = NULL;
- activityKind = factory->getKind();
- eog = false;
- matchedGroup = false;
- gotMatch = false;
- keepLimit = 0;
- keepCount = 0;
- joinCounter = 0;
- atmostLimit = 0;
- atmostsTriggered = 0;
- limitLimit = 0;
- rightGroupIndex = 0;
- hasGroupLimit = false;
- isSmartJoin = (joinFlags & JFsmart) != 0;
- getLimitType(joinFlags, limitFail, limitOnFail);
- }
- void loadRight()
- {
- ConstPointerArray rightset;
- try
- {
- const void * next;
- while(true)
- {
- next = inputStream1->ungroupedNextRow();
- if(!next)
- break;
- rightset.append(next);
- }
- if (!dedupRHS)
- {
- if (useFewTable)
- {
- table.setown(new FewLookupTable(rightset, helper)); // NOTE - takes ownership of rightset
- }
- else
- {
- if (!helper.isRightAlreadySorted())
- {
- if (helper.getJoinFlags() & JFunstable)
- {
- qsortvec(const_cast<void * *>(rightset.getArray()), rightset.ordinality(), *helper.queryCompareRight());
- }
- else
- {
- unsigned rightord = rightset.ordinality();
- MemoryAttr tempAttr(rightord*sizeof(void **)); // Temp storage for stable sort. This should probably be allocated from roxiemem
- void **temp = (void **) tempAttr.bufferBase();
- void **_rows = const_cast<void * *>(rightset.getArray());
- msortvecstableinplace(_rows, rightord, *helper.queryCompareRight(), temp);
- }
- }
- table.setown(new ManyLookupTable(rightset, helper)); // NOTE - takes ownership of rightset
- }
- }
- else
- {
- table.setown(new DedupLookupTable(rightset, helper)); // NOTE - takes ownership of rightset
- }
- }
- catch (...)
- {
- roxiemem::ReleaseRoxieRows(rightset);
- throw;
- }
- };
- virtual void reset()
- {
- if (atmostsTriggered)
- noteStatistic(StNumAtmostTriggered, atmostsTriggered);
- CRoxieServerTwoInputActivity::reset();
- ReleaseClearRoxieRow(left);
- defaultRight.clear();
- table.clear();
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eog = false;
- matchedGroup = false;
- left = NULL;
- CRoxieServerTwoInputActivity::start(parentExtractSize, parentExtract, paused);
- keepLimit = helper.getKeepLimit();
- if(keepLimit==0) keepLimit = static_cast<unsigned>(-1);
- atmostsTriggered = 0;
- atmostLimit = helper.getJoinLimit();
- limitLimit = helper.getMatchAbortLimit();
- hasGroupLimit = ((atmostLimit > 0) || (limitLimit > 0));
- if(atmostLimit==0) atmostLimit = static_cast<unsigned>(-1);
- if(limitLimit==0) limitLimit = static_cast<unsigned>(-1);
- getLimitType(helper.getJoinFlags(), limitFail, limitOnFail);
- switch (activityKind)
- {
- case TAKlookupjoin:
- case TAKlookupdenormalizegroup:
- case TAKsmartjoin:
- case TAKsmartdenormalizegroup:
- if (leftOuterJoin)
- createDefaultRight();
- break;
- }
- if (limitOnFail)
- createDefaultRight();
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- if (!idx && (helper.getJoinFlags() & JFparallel) != 0)
- {
- puller.setown(new CRoxieServerReadAheadInput(ctx, ctx->queryOptions().parallelJoinPreload));
- puller->setInput(0, _sourceIdx, _in);
- _in = puller;
- _sourceIdx = 0;
- }
- CRoxieServerTwoInputActivity::setInput(idx, _sourceIdx, _in);
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if(!table)
- loadRight();
- switch (activityKind)
- {
- case TAKlookupjoin:
- case TAKsmartjoin:
- return nextRowJoin();
- case TAKlookupdenormalize:
- case TAKlookupdenormalizegroup:
- case TAKsmartdenormalize:
- case TAKsmartdenormalizegroup:
- return nextRowDenormalize();
- }
- throwUnexpected();
- }
- private:
- const void * nextRowJoin()
- {
- if(!table)
- loadRight();
- while(true)
- {
- const void * right = NULL;
- if(!left)
- {
- left = inputStream->nextRow();
- keepCount = keepLimit;
- joinCounter = 0;
- if(!left)
- {
- if (isSmartJoin)
- left = inputStream->nextRow();
- if (!left)
- {
- if(matchedGroup || eog)
- {
- matchedGroup = false;
- eog = true;
- return NULL;
- }
- eog = true;
- continue;
- }
- }
- eog = false;
- gotMatch = false;
- right = getRightFirst();
- }
- else
- right = getRightNext();
- const void * ret = NULL;
- if(failingLimit)
- {
- ret = joinException(left, failingLimit);
- }
- else
- {
- while(right)
- {
- if(helper.match(left, right))
- {
- gotMatch = true;
- if(exclude)
- break;
- ret = joinRecords(left, right, ++joinCounter);
- if(ret)
- {
- processed++;
- break;
- }
- }
- right = getRightNext();
- ret = NULL;
- }
- if(leftOuterJoin && !gotMatch)
- {
- ret = joinRecords(left, defaultRight, 0);
- gotMatch = true;
- }
- }
- if(ret)
- {
- matchedGroup = true;
- processed++;
- if(!many || (--keepCount == 0) || failingLimit)
- {
- ReleaseClearRoxieRow(left);
- failingLimit.clear();
- }
- return ret;
- }
- ReleaseClearRoxieRow(left);
- }
- }
- const void * nextRowDenormalize()
- {
- while(true)
- {
- left = inputStream->nextRow();
- if(!left)
- {
- if (!matchedGroup || isSmartJoin)
- left = inputStream->nextRow();
- if (!left)
- {
- matchedGroup = false;
- return NULL;
- }
- }
- gotMatch = false;
- const void * right = getRightFirst();
- const void * ret = NULL;
- if (failingLimit)
- ret = joinException(left, failingLimit);
- else if (activityKind == TAKlookupdenormalize || activityKind == TAKsmartdenormalize)
- {
- OwnedConstRoxieRow newLeft;
- newLeft.set(left);
- unsigned rowSize = 0;
- unsigned leftCount = 0;
- keepCount = keepLimit;
- while (right)
- {
- try
- {
- if (helper.match(left, right))
- {
- gotMatch = true;
- if (exclude)
- break;
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned thisSize = helper.transform(rowBuilder, newLeft, right, ++leftCount);
- if (thisSize)
- {
- rowSize = thisSize;
- newLeft.setown(rowBuilder.finalizeRowClear(rowSize));
- }
- if(!many || (--keepCount == 0))
- break;
- }
- right = getRightNext();
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- if (rowSize)
- ret = newLeft.getClear();
- else if (leftOuterJoin && !gotMatch)
- {
- ret = left;
- left = NULL;
- }
- }
- else
- {
- filteredRight.kill();
- keepCount = keepLimit;
- while (right)
- {
- if (helper.match(left, right))
- {
- gotMatch = true;
- if(exclude)
- break;
- filteredRight.append(right);
- if(!many || (--keepCount == 0))
- break;
- }
- right = getRightNext();
- }
- if((filteredRight.ordinality() > 0) || (leftOuterJoin && !gotMatch))
- ret = denormalizeRecords(left, filteredRight);
- filteredRight.kill();
- }
- ReleaseRoxieRow(left);
- left = NULL;
- failingLimit.clear();
- if(ret)
- {
- matchedGroup = true;
- processed++;
- return ret;
- }
- }
- }
- const void * joinRecords(const void * left, const void * right, unsigned counter)
- {
- if (cloneLeft)
- {
- LinkRoxieRow(left);
- return left;
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned outSize = helper.transform(rowBuilder, left, right, counter);
- if (outSize)
- return rowBuilder.finalizeRowClear(outSize);
- else
- return NULL;
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- const void * joinException(const void * left, IException * except)
- {
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned outSize = helper.onFailTransform(rowBuilder, left, defaultRight, except);
- if (outSize)
- return rowBuilder.finalizeRowClear(outSize);
- else
- return NULL;
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- const void * denormalizeRecords(const void * left, ConstPointerArray & rows)
- {
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned numRows = rows.ordinality();
- const void * right = numRows ? rows.item(0) : defaultRight.get();
- unsigned outSize = helper.transform(rowBuilder, left, right, numRows, (const void * *)rows.getArray());
- if (outSize)
- return rowBuilder.finalizeRowClear(outSize);
- else
- return NULL;
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- const void * getRightFirst() { if(hasGroupLimit) return fillRightGroup(); else return table->find(left); }
- const void * getRightNext() { if(hasGroupLimit) return readRightGroup(); else return table->findNext(left); }
- const void * readRightGroup() { if(rightGroup.isItem(rightGroupIndex)) return rightGroup.item(rightGroupIndex++); else return NULL; }
- const void *fillRightGroup()
- {
- rightGroup.kill();
- for(const void * right = table->find(left); right; right = table->findNext(left))
- {
- rightGroup.append(right);
- if(rightGroup.ordinality() > limitLimit)
- {
- if(limitFail)
- failLimit();
- gotMatch = true;
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- if(limitOnFail)
- {
- assertex(!failingLimit);
- try
- {
- failLimit();
- }
- catch(IException * e)
- {
- failingLimit.setown(e);
- }
- assertex(failingLimit != NULL);
- }
- else
- {
- rightGroup.kill();
- }
- break;
- }
- if(rightGroup.ordinality() > atmostLimit)
- {
- atmostsTriggered++;
- rightGroup.kill();
- break;
- }
- }
- rightGroupIndex = 0;
- return readRightGroup();
- }
- void failLimit()
- {
- helper.onMatchAbortLimitExceeded();
- CommonXmlWriter xmlwrite(XWFtrim|XWFopt );
- if(!ctx->isBlind() && input->queryOutputMeta() && input->queryOutputMeta()->hasXML())
- {
- input->queryOutputMeta()->toXML(static_cast<const unsigned char *>(left), xmlwrite);
- }
- throw MakeStringException(ROXIE_TOO_MANY_RESULTS, "More than %u match candidates in join %d for row %s", limitLimit, queryId(), xmlwrite.str());
- }
-
- };
- unsigned const CRoxieServerLookupJoinActivity::FewLookupTable::BadIndex(static_cast<unsigned>(-1));
- class CRoxieServerLookupJoinActivityFactory : public CRoxieServerJoinActivityFactory
- {
- public:
- CRoxieServerLookupJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerJoinActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- Owned<IHThorHashJoinArg> helper = (IHThorHashJoinArg *) helperFactory();
- useFewTable = _graphNode.getPropBool("hint[@name='usefewtable']/@value", false);
- if((helper->getJoinFlags() & (JFfirst | JFfirstleft | JFfirstright | JFslidingmatch)) != 0)
- throw MakeStringException(ROXIE_INVALID_FLAGS, "Invalid flags for lookup join activity"); // code generator should never create such an activity
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerLookupJoinActivity(_ctx, this, _probeManager, useFewTable);
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return joinStatistics;
- }
- protected:
- bool useFewTable;
- };
- IRoxieServerActivityFactory *createRoxieServerLookupJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerLookupJoinActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- class CRoxieServerAllJoinActivity : public CRoxieServerTwoInputActivity
- {
- private:
- IHThorAllJoinArg &helper;
- bool leftOuterJoin;
- bool rightOuterJoin;
- bool exclude;
- OwnedConstRoxieRow defaultRight;
- OwnedConstRoxieRow defaultLeft;
- Owned<IEngineRowAllocator> defaultRightAllocator;
- Owned<IEngineRowAllocator> defaultLeftAllocator;
- const void *left;
- unsigned countForLeft;
- ConstPointerArray rightset;
- BoolArray matchedRight; // MORE - could use a bitset...
- unsigned keepLimit;
- bool started;
- bool eog;
- bool eos;
- bool matchedLeft;
- bool matchedGroup;
- bool leftIsGrouped;
- bool cloneLeft;
- unsigned rightIndex;
- unsigned joinCounter;
- unsigned rightOrdinality;
- ThorActivityKind activityKind;
- ConstPointerArray filteredRight;
- void createDefaultLeft()
- {
- if (!defaultLeft)
- {
- if (!defaultLeftAllocator)
- defaultLeftAllocator.setown(createRowAllocator(input->queryOutputMeta()));
- RtlDynamicRowBuilder rowBuilder(defaultLeftAllocator);
- size32_t thisSize = helper.createDefaultLeft(rowBuilder);
- defaultLeft.setown(rowBuilder.finalizeRowClear(thisSize));
- }
- }
- void createDefaultRight()
- {
- if (!defaultRight)
- {
- if (!defaultRightAllocator)
- defaultRightAllocator.setown(createRowAllocator(input1->queryOutputMeta()));
- RtlDynamicRowBuilder rowBuilder(defaultRightAllocator);
- size32_t thisSize = helper.createDefaultRight(rowBuilder);
- defaultRight.setown(rowBuilder.finalizeRowClear(thisSize));
- }
- }
- public:
- CRoxieServerAllJoinActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerTwoInputActivity(_ctx, _factory, _probeManager), helper((IHThorAllJoinArg &)basehelper)
- {
- unsigned joinFlags = helper.getJoinFlags();
- leftOuterJoin = (joinFlags & JFleftouter) != 0;
- rightOuterJoin = (joinFlags & JFrightouter) != 0;
- cloneLeft = (joinFlags & JFtransformmatchesleft) != 0;
- keepLimit = (unsigned) -1;
- exclude = (joinFlags & JFexclude) != 0;
- left = NULL;
- started = true;
- eog = false;
- eos = false;
- matchedLeft = false;
- matchedGroup = false;
- activityKind = factory->getKind();
- rightIndex = 0;
- rightOrdinality = 0;
- leftIsGrouped = false;
- countForLeft = 0;
- joinCounter = 0;
- }
- virtual void reset()
- {
- defaultRight.clear();
- defaultLeft.clear();
- ReleaseRoxieRows(rightset);
- matchedRight.kill();
- CRoxieServerTwoInputActivity::reset();
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eog = false;
- eos = false;
- matchedLeft = false;
- matchedGroup = false;
- started = false;
- left = NULL;
- CRoxieServerTwoInputActivity::start(parentExtractSize, parentExtract, paused);
- keepLimit = helper.getKeepLimit();
- if(keepLimit==0)
- keepLimit = (unsigned) -1;
- countForLeft = keepLimit;
- joinCounter = 0;
- leftIsGrouped = input->queryOutputMeta()->isGrouped();
- if((activityKind==TAKalljoin || activityKind==TAKalldenormalizegroup) && leftOuterJoin)
- createDefaultRight();
- if(rightOuterJoin)
- createDefaultLeft();
- }
- void loadRight()
- {
- const void * next;
- while(true)
- {
- next = inputStream1->ungroupedNextRow();
- if(!next)
- break;
- rightset.append(next);
- matchedRight.append(false);
- }
- rightIndex = 0;
- joinCounter = 0;
- rightOrdinality = rightset.ordinality();
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- if (!idx && (helper.getJoinFlags() & JFparallel) != 0)
- {
- puller.setown(new CRoxieServerReadAheadInput(ctx, ctx->queryOptions().parallelJoinPreload));
- puller->setInput(0, _sourceIdx, _in);
- _in = puller;
- _sourceIdx = 0;
- }
- CRoxieServerTwoInputActivity::setInput(idx, _sourceIdx, _in);
- }
- const void * joinRecords(const void * left, const void * right, unsigned counter)
- {
- // MORE - could share some code with lookup join
- if (cloneLeft)
- {
- LinkRoxieRow(left);
- return left;
- }
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned outSize = helper.transform(rowBuilder, left, right, counter);
- if (outSize)
- return rowBuilder.finalizeRowClear(outSize);
- else
- return NULL;
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- const void * denormalizeRecords(const void * curLeft, ConstPointerArray & rows)
- {
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned numRows = rows.ordinality();
- const void * right = numRows ? rows.item(0) : defaultRight.get();
- unsigned outSize = helper.transform(rowBuilder, curLeft, right, numRows, rows.getArray());
- if (outSize)
- return rowBuilder.finalizeRowClear(outSize);
- else
- return NULL;
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if(!started)
- {
- started = true;
- left = inputStream->nextRow();
- matchedLeft = false;
- countForLeft = keepLimit;
- joinCounter = 0;
- if(left == NULL)
- {
- eos = true;
- return NULL;
- }
- loadRight();
- }
- const void * ret;
- const void * right;
- if(eos)
- return NULL;
- while(true)
- {
- ret = NULL;
- if((rightIndex == rightOrdinality) || (countForLeft==0))
- {
- if(leftOuterJoin && left && !matchedLeft)
- {
- switch(activityKind)
- {
- case TAKalljoin:
- ret = joinRecords(left, defaultRight, 0);
- break;
- case TAKalldenormalize:
- ret = left;
- left = NULL;
- break;
- case TAKalldenormalizegroup:
- filteredRight.kill();
- ret = denormalizeRecords(left, filteredRight);
- break;
- default:
- throwUnexpected();
- }
- }
- rightIndex = 0;
- joinCounter = 0;
- ReleaseRoxieRow(left);
- left = NULL;
- if(ret)
- {
- matchedGroup = true;
- processed++;
- return ret;
- }
- }
- if(!left)
- {
- left = inputStream->nextRow();
- matchedLeft = false;
- countForLeft = keepLimit;
- joinCounter = 0;
- }
- if(!left)
- {
- if(eog)
- {
- eos = true;
- matchedGroup = false;
- return NULL;
- }
- eog = true;
- if (matchedGroup && leftIsGrouped)
- {
- matchedGroup = false;
- return NULL;
- }
- matchedGroup = false;
- continue;
- }
- eog = false;
- switch(activityKind)
- {
- case TAKalljoin:
- while(rightIndex < rightOrdinality)
- {
- right = rightset.item(rightIndex);
- if(helper.match(left, right))
- {
- matchedLeft = true;
- matchedRight.replace(true, rightIndex);
- if(!exclude)
- ret = joinRecords(left, right, ++joinCounter);
- }
- rightIndex++;
- if(ret)
- {
- countForLeft--;
- matchedGroup = true;
- processed++;
- return ret;
- }
- }
- break;
- case TAKalldenormalize:
- {
- OwnedConstRoxieRow newLeft;
- newLeft.set(left);
- unsigned rowSize = 0;
- unsigned leftCount = 0;
- while((rightIndex < rightOrdinality) && countForLeft)
- {
- right = rightset.item(rightIndex);
- if(helper.match(left, right))
- {
- matchedLeft = true;
- matchedRight.replace(true, rightIndex);
- if(!exclude)
- {
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned thisSize = helper.transform(rowBuilder, newLeft, right, ++leftCount);
- if(thisSize)
- {
- rowSize = thisSize;
- newLeft.setown(rowBuilder.finalizeRowClear(rowSize));
- --countForLeft;
- }
- }
- catch (IException *e)
- {
- throw makeWrappedException(e);
- }
- }
- }
- rightIndex++;
- }
- if(rowSize)
- {
- processed++;
- return newLeft.getClear();
- }
- }
- break;
- case TAKalldenormalizegroup:
- filteredRight.kill();
- while((rightIndex < rightOrdinality) && countForLeft)
- {
- right = rightset.item(rightIndex);
- if(helper.match(left, right))
- {
- matchedLeft = true;
- matchedRight.replace(true, rightIndex);
- filteredRight.append(right);
- --countForLeft;
- }
- ++rightIndex;
- }
- if(!exclude && filteredRight.ordinality())
- {
- ret = denormalizeRecords(left, filteredRight);
- filteredRight.kill();
- if(ret)
- {
- processed++;
- return ret;
- }
- }
- break;
- default:
- throwUnexpected();
- }
- }
- }
- };
- class CRoxieServerAllJoinActivityFactory : public CRoxieServerJoinActivityFactory
- {
- public:
- CRoxieServerAllJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerJoinActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- Owned<IHThorAllJoinArg> helper = (IHThorAllJoinArg *) helperFactory();
- if((helper->getJoinFlags() & (JFfirst | JFfirstleft | JFfirstright)) != 0)
- throw MakeStringException(ROXIE_INVALID_FLAGS, "Invalid flags for join all activity"); // code generator should never create such an activity
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerAllJoinActivity(_ctx, this, _probeManager);
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return joinStatistics;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerAllJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerAllJoinActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- class CRoxieServerTopNActivity : public CRoxieServerLateStartActivity
- {
- unsigned limit;
- bool hasBest;
- bool eoi;
- const void **sorted;
- unsigned sortedCount;
- unsigned curIndex;
- IHThorTopNArg &helper;
- ICompare &compare;
- public:
- CRoxieServerTopNActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerLateStartActivity(_ctx, _factory, _probeManager), helper((IHThorTopNArg &)basehelper), compare(*helper.queryCompare())
- {
- sorted = NULL;
- sortedCount = 0;
- curIndex = 0;
- limit = 0;
- eoi = false;
- hasBest = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- assertex(sorted == NULL);
- sortedCount = 0;
- curIndex = 0;
- eoi = false;
-
- CRoxieServerLateStartActivity::start(parentExtractSize, parentExtract, paused);
- limit = (unsigned) helper.getLimit();
- hasBest = helper.hasBest();
- lateStart(parentExtractSize, parentExtract, limit > 0);
- // MORE - should we use an expanding array instead?
- if (limit > 0)
- sorted = (const void **) ctx->queryRowManager().allocate((limit+1) * sizeof(const void *), activityId);
- }
- virtual void reset()
- {
- if (sorted)
- {
- roxiemem::ReleaseRoxieRowRange(sorted, curIndex, sortedCount);
- curIndex = 0;
- sortedCount = 0;
- ReleaseRoxieRow(sorted);
- sorted = NULL;
- }
- CRoxieServerLateStartActivity::reset();
- }
- bool abortEarly()
- {
- if (hasBest && (sortedCount == limit))
- {
- int compare = helper.compareBest(sorted[sortedCount-1]);
- if (compare == 0)
- {
- if (meta.isGrouped())
- {
- //MORE: This would be more efficient if we had a away of skipping to the end of the incoming group.
- const void * next;
- while ((next = inputStream->nextRow()) != NULL)
- ReleaseRoxieRow(next);
- }
- else
- eoi = true;
- return true;
- }
- //This only checks the lowest element - we could check all elements inserted, but it would increase the number of compares
- if (compare < 0)
- throw MakeStringException(ROXIE_TOPN_ROW_ERROR, "TOPN: row found that exceeds the best value");
- }
- return false;
- }
- void getSorted()
- {
- curIndex = 0;
- sortedCount = 0;
- if(eoi)
- return;
- const void * next;
- while ((next = inputStream->nextRow()) != NULL)
- {
- if (sortedCount < limit)
- {
- binary_vec_insert_stable(next, sorted, sortedCount, compare);
- sortedCount++;
- if(abortEarly())
- return;
- }
- else
- {
- if(limit && compare.docompare(sorted[sortedCount-1], next) > 0) // MORE - if stability is an issue, need to consider whether this should be > or >=
- {
- binary_vec_insert_stable(next, sorted, sortedCount, compare); // MORE - not sure this is stable!
- ReleaseRoxieRow(sorted[sortedCount]);
- if(abortEarly())
- return;
- }
- else
- {
- ReleaseRoxieRow(next); // do not bother with insertion sort if we know next will fall off the end
- }
- }
- }
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- if (curIndex >= sortedCount)
- {
- bool eog = sortedCount != 0;
- getSorted();
- if(sortedCount == 0)
- {
- eof = true;
- return NULL;
- }
- if (eog)
- return NULL;
- }
- processed++;
- return sorted[curIndex++];
- }
- };
- class CRoxieServerTopNActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerTopNActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerTopNActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerTopNActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerTopNActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerLimitActivity : public CRoxieServerActivity
- {
- protected:
- unsigned __int64 rowLimit;
- IHThorLimitArg &helper;
- public:
- CRoxieServerLimitActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorLimitArg &)basehelper)
- {
- rowLimit = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- rowLimit = helper.getRowLimit(); // could conceivably depend on context so should not compute any earlier than this
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * ret = inputStream->nextRow();
- if (ret)
- {
- processed++;
- if (processed > rowLimit)
- {
- ReleaseRoxieRow(ret);
- if (traceLevel > 4)
- DBGLOG("activityid = %d line = %d", activityId, __LINE__);
- helper.onLimitExceeded();
- }
- }
- return ret;
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * ret = inputStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- if (ret)
- {
- if (wasCompleteMatch)
- processed++;
- if (processed > rowLimit)
- {
- ReleaseRoxieRow(ret);
- if (traceLevel > 4)
- DBGLOG("activityid = %d line = %d", activityId, __LINE__);
- helper.onLimitExceeded();
- }
- }
- return ret;
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- //Do not reset the rowLimit
- inputStream->resetEOF();
- }
- IInputSteppingMeta * querySteppingMeta()
- {
- return input->querySteppingMeta();
- }
- };
- class CRoxieServerLimitActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerLimitActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerLimitActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerLimitActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerLimitActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- class CRoxieServerSkipLimitActivity : public CRoxieServerLimitActivity
- {
- ConstPointerArray buff;
- bool started;
- unsigned index;
- IHThorLimitTransformExtra * transformExtra;
- public:
- CRoxieServerSkipLimitActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, bool _onFail)
- : CRoxieServerLimitActivity(_ctx, _factory, _probeManager)
- {
- transformExtra = NULL;
- started = false;
- index = 0;
- if (_onFail)
- transformExtra = static_cast<IHThorLimitTransformExtra *>(helper.selectInterface(TAIlimittransformextra_1));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- started = false;
- index = 0;
- CRoxieServerLimitActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- roxiemem::ReleaseRoxieRowRange(buff.getArray(), index, buff.ordinality());
- buff.kill();
- index = 0;
- started = false;
- CRoxieServerLimitActivity::reset();
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (!started)
- pullInput();
- if (buff.isItem(index))
- {
- const void * next = buff.item(index++);
- if(next)
- processed++;
- return next;
- }
- return NULL;
- }
- protected:
- void pullInput()
- {
- unsigned count = 0;
- loop
- {
- const void * next = inputStream->nextRow();
- if (next == NULL)
- {
- next = inputStream->nextRow();
- if(next == NULL)
- break;
- buff.append(NULL);
- }
- count++;
- if (count > rowLimit)
- {
- ReleaseRoxieRow(next);
- ReleaseRoxieRows(buff);
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- if (transformExtra)
- {
- ensureRowAllocator();
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize = transformExtra->transformOnLimitExceeded(rowBuilder);
- if (outSize)
- buff.append(rowBuilder.finalizeRowClear(outSize));
- }
- break;
- }
- buff.append(next);
- }
- started = true;
- }
- };
- class CRoxieServerSkipLimitActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerSkipLimitActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSkipLimitActivity(_ctx, this, _probeManager, kind==TAKcreaterowlimit);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSkipLimitActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerSkipLimitActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerCatchActivity : public CRoxieServerActivity
- {
- IHThorCatchArg &helper;
- public:
- CRoxieServerCatchActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorCatchArg &)basehelper)
- {
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- try
- {
- const void *ret = inputStream->nextRow();
- if (ret)
- processed++;
- return ret;
- }
- catch (IException *E)
- {
- E->Release();
- helper.onExceptionCaught();
- }
- catch (...)
- {
- helper.onExceptionCaught();
- }
- throwUnexpected(); // onExceptionCaught should have thrown something
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- try
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void * ret = inputStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- if (ret && wasCompleteMatch)
- processed++;
- return ret;
- }
- catch (IException *E)
- {
- E->Release();
- helper.onExceptionCaught();
- }
- catch (...)
- {
- helper.onExceptionCaught();
- }
- throwUnexpected(); // onExceptionCaught should have thrown something
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- inputStream->resetEOF(); // MORE - why not in base class?
- }
- IInputSteppingMeta * querySteppingMeta()
- {
- return input->querySteppingMeta();
- }
- };
- class CRoxieServerSkipCatchActivity : public CRoxieServerActivity
- {
- ConstPointerArray buff;
- bool started;
- unsigned index;
- IHThorCatchArg &helper;
- bool createRow;
- public:
- CRoxieServerSkipCatchActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, bool _createRow)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorCatchArg &)basehelper), createRow(_createRow)
- {
- started = false;
- index = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- started = false;
- index = 0;
- try
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- catch (IException *E)
- {
- onException(E);
- started = true;
- }
- catch (...)
- {
- onException(MakeStringException(ROXIE_INTERNAL_ERROR, "Unknown exception caught"));
- started = true;
- }
- }
- virtual void reset()
- {
- roxiemem::ReleaseRoxieRowRange(buff.getArray(), index, buff.ordinality());
- buff.kill();
- index = 0;
- started = false;
- CRoxieServerActivity::reset();
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (!started)
- pullInput();
- if (buff.isItem(index))
- {
- const void * next = buff.item(index++);
- if(next)
- processed++;
- return next;
- }
- return NULL;
- }
- protected:
- void onException(IException *E)
- {
- inputStream->stop();
- ReleaseRoxieRows(buff);
- if (createRow)
- {
- ensureRowAllocator();
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize = helper.transformOnExceptionCaught(rowBuilder, E);
- if (outSize)
- buff.append(rowBuilder.finalizeRowClear(outSize));
- }
- E->Release();
- }
- void pullInput()
- {
- try
- {
- bool EOGseen = false;
- loop
- {
- const void * next = inputStream->nextRow();
- buff.append(next);
- if (next == NULL)
- {
- if (EOGseen)
- break;
- EOGseen = true;
- }
- else
- EOGseen = false;
- }
- }
- catch (IException *E)
- {
- onException(E);
- }
- catch (...)
- {
- onException(MakeStringException(ROXIE_INTERNAL_ERROR, "Unknown exception caught"));
- }
- started = true;
- }
- };
- class CRoxieServerCatchActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerCatchActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- switch (kind)
- {
- case TAKcatch:
- return new CRoxieServerCatchActivity(_ctx, this, _probeManager);
- case TAKskipcatch:
- return new CRoxieServerSkipCatchActivity(_ctx, this, _probeManager, false);
- case TAKcreaterowcatch:
- return new CRoxieServerSkipCatchActivity(_ctx, this, _probeManager, true);
- default:
- throwUnexpected();
- }
- }
- };
- IRoxieServerActivityFactory *createRoxieServerCatchActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerCatchActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerPullActivity : public CRoxieServerActivity
- {
- ConstPointerArray buff;
- bool started;
- unsigned index;
- public:
- CRoxieServerPullActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager)
- {
- started = false;
- index = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- started = false;
- index = 0;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- while (buff.isItem(index))
- ReleaseRoxieRow(buff.item(index++));
- buff.kill();
- started = false;
- index = 0;
- CRoxieServerActivity::reset();
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (!started)
- pullInput();
- if (buff.isItem(index))
- {
- const void * next = buff.item(index++);
- if (next)
- processed++;
- return next;
- }
- return NULL;
- }
- protected:
- void pullInput()
- {
- bool EOGseen = false;
- loop
- {
- const void * next = inputStream->nextRow();
- buff.append(next);
- if (next == NULL)
- {
- if (EOGseen)
- break;
- EOGseen = true;
- }
- else
- EOGseen = false;
- }
- started = true;
- }
- };
- class CRoxieServerPullActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerPullActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerPullActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerPullActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerPullActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerTraceActivity : public CRoxieServerActivity
- {
- public:
- CRoxieServerTraceActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorTraceArg &) basehelper),
- keepLimit(0), skip(0), sample(0)
- {
- assertex(meta.hasXML());
- traceEnabled = defaultTraceEnabled && !isBlind();
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- if (ctx)
- traceEnabled = ctx->queryOptions().traceEnabled && !isBlind();
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if (traceEnabled && helper.canMatchAny())
- {
- keepLimit = helper.getKeepLimit();
- if (keepLimit==(unsigned) -1)
- keepLimit = ctx->queryOptions().traceLimit;
- skip = helper.getSkip();
- sample = helper.getSample();
- if (sample)
- sample--;
- name.setown(helper.getName());
- if (!name)
- name.set("Row");
- }
- else
- keepLimit = 0;
- }
- virtual void stop()
- {
- name.clear();
- CRoxieServerActivity::stop();
- }
- virtual bool isPassThrough()
- {
- return true;
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- const void *row = inputStream->nextRow();
- if (row)
- {
- onTrace(row);
- processed++;
- }
- return row;
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- // MORE - will need rethinking once we rethink the nextRowGE interface for global smart-stepping.
- ActivityTimer t(totalCycles, timeActivities);
- const void * row = inputStream->nextRowGE(seek, numFields, wasCompleteMatch, stepExtra);
- if (row)
- {
- onTrace(row);
- processed++;
- }
- return row;
- }
- virtual bool gatherConjunctions(ISteppedConjunctionCollector & collector)
- {
- return input->gatherConjunctions(collector);
- }
- virtual void resetEOF()
- {
- inputStream->resetEOF();
- }
- IInputSteppingMeta * querySteppingMeta()
- {
- return input->querySteppingMeta();
- }
- protected:
- void onTrace(const void *row)
- {
- if (keepLimit && helper.isValid(row))
- {
- if (skip)
- skip--;
- else if (sample)
- sample--;
- else
- {
- CommonXmlWriter xmlwrite(XWFnoindent);
- meta.toXML((const byte *) row, xmlwrite);
- CTXLOG("TRACE: <%s>%s<%s>", name.get(), xmlwrite.str(), name.get());
- keepLimit--;
- sample = helper.getSample();
- if (sample)
- sample--;
- }
- }
- }
- OwnedRoxieString name;
- IHThorTraceArg &helper;
- unsigned keepLimit;
- unsigned skip;
- unsigned sample;
- bool traceEnabled;
- };
- class CRoxieServerTraceActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerTraceActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerTraceActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerTraceActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerTraceActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerCaseActivity : public CRoxieServerMultiInputBaseActivity
- {
- IHThorCaseArg &helper;
- unsigned cond;
- bool unusedStopped;
- IEngineRowStream *active;
- public:
- CRoxieServerCaseActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numInputs)
- : CRoxieServerMultiInputBaseActivity(_ctx, _factory, _probeManager, _numInputs), helper((IHThorCaseArg &)basehelper)
- {
- unusedStopped = false;
- cond = 0;
- active = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerMultiInputBaseActivity::start(parentExtractSize, parentExtract, paused);
- cond = helper.getBranch();
- //CHOOSE defaults to the last argument if out of range.
- if (cond >= numInputs)
- cond = numInputs - 1;
- inputArray[cond]->start(parentExtractSize, parentExtract, paused);
- startJunction(junctionArray[cond]);
- assertex(numInputs==numStreams);
- for (unsigned idx = 0; idx < numStreams; idx++)
- {
- if (idx!=cond)
- streamArray[idx]->stop(); // Note: stopping unused branches early helps us avoid buffering splits too long.
- }
- active = streamArray[cond];
- unusedStopped = true;
- }
- virtual void stop()
- {
- for (unsigned idx = 0; idx < numStreams; idx++)
- {
- if (idx==cond || !unusedStopped)
- streamArray[idx]->stop();
- }
- CRoxieServerMultiInputBaseActivity::stop();
- }
- virtual void reset()
- {
- unusedStopped = false;
- active = NULL;
- CRoxieServerMultiInputBaseActivity::reset();
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (active)
- {
- const void *ret = active->nextRow();
- if (ret)
- processed++;
- return ret;
- }
- return NULL;
- }
- };
- class CRoxieServerCaseActivityFactory : public CRoxieServerMultiInputFactory
- {
- bool graphInvariant;
- public:
- CRoxieServerCaseActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _graphInvariant)
- : CRoxieServerMultiInputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- graphInvariant = _graphInvariant;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerCaseActivity(_ctx, this, _probeManager, numInputs());
- }
- virtual bool isGraphInvariant() const
- {
- return graphInvariant;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerCaseActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _graphInvariant)
- {
- return new CRoxieServerCaseActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _graphInvariant);
- }
- //=================================================================================
- class CRoxieServerIfActivity : public CRoxieServerActivity
- {
- IHThorIfArg &helper;
- IFinalRoxieInput *inputTrue;
- IFinalRoxieInput *inputFalse;
- IEngineRowStream *streamTrue;
- IEngineRowStream *streamFalse;
- unsigned sourceIdxFalse = 0;
- unsigned sourceIdxTrue = 0;
- Owned<IStrandJunction> junctionTrue;
- Owned<IStrandJunction> junctionFalse;
- bool cond;
- bool unusedStopped;
- public:
- CRoxieServerIfActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorIfArg &)basehelper)
- {
- inputFalse = NULL;
- inputTrue = NULL;
- streamTrue = NULL;
- streamFalse = NULL;
- unusedStopped = false;
- cond = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- cond = helper.getCondition();
- if (cond)
- {
- inputTrue->start(parentExtractSize, parentExtract, paused);
- startJunction(junctionTrue);
- if (streamFalse)
- streamFalse->stop(); // Note: stopping unused branches early helps us avoid buffering splits too long.
- }
- else
- {
- if (inputFalse)
- {
- inputFalse->start(parentExtractSize, parentExtract, paused);
- startJunction(junctionFalse);
- }
- streamTrue->stop();
- }
- unusedStopped = true;
- }
- virtual void stop()
- {
- if (!unusedStopped || cond)
- streamTrue->stop();
- if (streamFalse && (!unusedStopped || !cond))
- streamFalse->stop();
- CRoxieServerActivity::stop();
- }
- virtual unsigned __int64 queryLocalCycles() const
- {
- __int64 localCycles = totalCycles.totalCycles;
- localCycles -= inputTrue->queryTotalCycles();
- if (inputFalse)
- localCycles -= inputFalse->queryTotalCycles();
- if (localCycles < 0)
- localCycles = 0;
- return localCycles;
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- switch (idx)
- {
- case 0:
- return inputTrue;
- case 1:
- return inputFalse;
- default:
- return NULL;
- }
- }
- virtual IEngineRowStream *queryConcreteOutputStream(unsigned whichInput) { return this; }
- virtual IStrandJunction *queryConcreteOutputJunction(unsigned idx) const { return nullptr; }
- virtual IIndexReadActivityInfo *queryIndexReadActivity()
- {
- IFinalRoxieInput *in = cond ? inputTrue : inputFalse;
- if (in)
- return in->queryIndexReadActivity();
- return NULL;
- }
- virtual void reset()
- {
- CRoxieServerActivity::reset();
- inputTrue->reset();
- if (inputFalse)
- inputFalse->reset();
- resetJunction(junctionTrue);
- resetJunction(junctionFalse);
- unusedStopped = false;
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- if (idx==1)
- {
- inputFalse = _in;
- sourceIdxFalse = _sourceIdx;
- }
- else
- {
- assertex(!idx);
- inputTrue = _in;
- sourceIdxTrue = _sourceIdx;
- }
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- //It should be possible to stream IF(), but the number of strands would need to match on the two branches.
- streamTrue = connectSingleStream(ctx, inputTrue, sourceIdxTrue, junctionTrue, consumerOrdered);
- streamFalse = connectSingleStream(ctx, inputFalse, sourceIdxFalse, junctionFalse, consumerOrdered);
- return CRoxieServerActivity::getOutputStreams(ctx, idx, streams, nullptr, consumerOrdered, nullptr);
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- IEngineRowStream *in = cond ? streamTrue : streamFalse;
- if (in)
- {
- const void * ret;
- if ((ret = in->nextRow()) != NULL)
- processed++;
- return ret;
- }
- return NULL;
- }
- };
- class CRoxieServerIfActivityFactory : public CRoxieServerActivityFactory
- {
- unsigned input2;
- unsigned input2idx;
- bool graphInvariant;
- public:
- CRoxieServerIfActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _graphInvariant)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- graphInvariant = _graphInvariant;
- input2 = (unsigned)-1;
- input2idx = (unsigned)-1;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerIfActivity(_ctx, this, _probeManager);
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- if (idx==1)
- {
- input2 = source;
- input2idx = sourceidx;
- }
- else
- CRoxieServerActivityFactory::setInput(idx, source, sourceidx);
- }
- virtual unsigned getInput(unsigned idx, unsigned &sourceidx) const
- {
- switch (idx)
- {
- case 1:
- sourceidx = input2idx;
- return input2;
- case 0:
- return CRoxieServerActivityFactory::getInput(idx, sourceidx);
- default:
- return (unsigned) -1;
- }
- }
- virtual unsigned numInputs() const { return (input2 == (unsigned)-1) ? 1 : 2; }
- virtual bool isGraphInvariant() const
- {
- return graphInvariant;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerIfActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _graphInvariant)
- {
- return new CRoxieServerIfActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _graphInvariant);
- }
- //=================================================================================
- class CRoxieServerActionBaseActivity : public CRoxieServerActivity
- {
- CriticalSection ecrit;
- Owned<IException> exception;
- bool executed;
- public:
- CRoxieServerActionBaseActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager)
- {
- executed = false;
- }
- virtual void doExecuteAction(unsigned parentExtractSize, const byte * parentExtract) = 0;
- virtual void execute(unsigned parentExtractSize, const byte * parentExtract)
- {
- CriticalBlock b(ecrit);
- if (exception)
- throw(exception.getLink());
- if (!executed)
- {
- try
- {
- executed = true;
- start(parentExtractSize, parentExtract, false);
- doExecuteAction(parentExtractSize, parentExtract);
- stop();
- }
- catch (IException * E)
- {
- ctx->notifyAbort(E);
- abort();
- exception.set(E);
- throw;
- }
- }
- }
- virtual void reset()
- {
- executed = false;
- exception.clear();
- CRoxieServerActivity::reset();
- }
- virtual const void *nextRow()
- {
- throwUnexpected(); // I am nobody's input
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- return NULL;
- }
- };
- class CRoxieServerIfActionActivity : public CRoxieServerActionBaseActivity
- {
- IHThorIfArg &helper;
- public:
- CRoxieServerIfActionActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActionBaseActivity(_ctx, _factory, _probeManager), helper((IHThorIfArg &)basehelper)
- {
- }
- virtual void doExecuteAction(unsigned parentExtractSize, const byte * parentExtract)
- {
- bool cond;
- {
- ActivityTimer t(totalCycles, timeActivities);
- cond = helper.getCondition();
- }
- stopDependencies(parentExtractSize, parentExtract, cond ? 2 : 1);
- executeDependencies(parentExtractSize, parentExtract, cond ? 1 : 2);
- }
- };
- class CRoxieServerIfActionActivityFactory : public CRoxieServerActivityFactory
- {
- bool isRoot;
- public:
- CRoxieServerIfActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), isRoot(_isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerIfActionActivity(_ctx, this, _probeManager);
- }
- virtual bool isSink() const
- {
- return isRoot;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerIfActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerIfActionActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //=================================================================================
- class CRoxieServerParallelActionActivity : public CRoxieServerActionBaseActivity
- {
- public:
- CRoxieServerParallelActionActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActionBaseActivity(_ctx, _factory, _probeManager)
- {
- }
- virtual void doExecuteAction(unsigned parentExtractSize, const byte * parentExtract)
- {
- #ifdef PARALLEL_EXECUTE
- CParallelActivityExecutor afor(dependencies, parentExtractSize, parentExtract);
- afor.For(dependencies.ordinality(), dependencies.ordinality(), true);
- #else
- ForEachItemIn(idx, dependencies)
- {
- dependencies.item(idx).execute(parentExtractSize, parentExtract);
- }
- #endif
- }
- };
- class CRoxieServerParallelActionActivityFactory : public CRoxieServerActivityFactory
- {
- bool isRoot;
- public:
- CRoxieServerParallelActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), isRoot(_isRoot)
- {
- assertex(!isRoot); // non-internal should be expanded out..
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerParallelActionActivity(_ctx, this, _probeManager);
- }
- virtual bool isSink() const
- {
- return isRoot;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerParallelActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerParallelActionActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //=================================================================================
- class CRoxieServerSequentialActionActivity : public CRoxieServerActionBaseActivity
- {
- IHThorSequentialArg &helper;
- public:
- CRoxieServerSequentialActionActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActionBaseActivity(_ctx, _factory, _probeManager), helper((IHThorSequentialArg &)basehelper)
- {
- }
- virtual void doExecuteAction(unsigned parentExtractSize, const byte * parentExtract)
- {
- unsigned numBranches = helper.numBranches();
- for (unsigned branch=1; branch <= numBranches; branch++)
- executeDependencies(parentExtractSize, parentExtract, branch);
- }
- };
- class CRoxieServerSequentialActionActivityFactory : public CRoxieServerActivityFactory
- {
- bool isRoot;
- public:
- CRoxieServerSequentialActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), isRoot(_isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSequentialActionActivity(_ctx, this, _probeManager);
- }
- virtual bool isSink() const
- {
- return isRoot;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSequentialActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerSequentialActionActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //=================================================================================
- class CRoxieServerWhenActivity : public CRoxieServerActivity
- {
- public:
- CRoxieServerWhenActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager)
- {
- savedExtractSize = 0;
- savedExtract = NULL;
- eofseen = false;
- eogseen = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- savedExtractSize = parentExtractSize;
- savedExtract = parentExtract;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- executeDependencies(parentExtractSize, parentExtract, WhenBeforeId);
- executeDependencies(parentExtractSize, parentExtract, WhenParallelId); // MORE: This should probably be done in parallel!
- eofseen = false;
- eogseen = false;
- }
- virtual void stop()
- {
- if (state == STATEreset || (!aborted && !eofseen)) // If someone else aborted, then WHEN clauses don't trigger, whatever
- {
- stopDependencies(savedExtractSize, savedExtract, WhenSuccessId);
- stopDependencies(savedExtractSize, savedExtract, WhenFailureId);
- }
- else if (state != STATEstopped)
- {
- stopDependencies(savedExtractSize, savedExtract, aborted ? WhenSuccessId : WhenFailureId); // These ones don't get executed
- executeDependencies(savedExtractSize, savedExtract, aborted ? WhenFailureId : WhenSuccessId); // These ones do
- }
- CRoxieServerActivity::stop();
- }
- virtual const void *nextRow()
- {
- try
- {
- ActivityTimer t(totalCycles, timeActivities); // bit of a waste of time....
- const void *ret = inputStream->nextRow();
- if (!ret)
- {
- if (eogseen)
- eofseen = true;
- else
- eogseen = true;
- }
- else
- eogseen = false;
- return ret;
- }
- catch (...)
- {
- aborted = true;
- throw;
- }
- }
- protected:
- const byte *savedExtract;
- unsigned savedExtractSize;
- bool eofseen;
- bool eogseen;
- };
- class CRoxieServerWhenActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerWhenActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerWhenActivity(_ctx, this, _probeManager);
- }
- };
- extern IRoxieServerActivityFactory *createRoxieServerWhenActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerWhenActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=================================================================================
- class CRoxieServerWhenActionActivity : public CRoxieServerActionBaseActivity
- {
- public:
- CRoxieServerWhenActionActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActionBaseActivity(_ctx, _factory, _probeManager)
- {
- savedExtractSize = 0;
- savedExtract = NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- savedExtractSize = parentExtractSize;
- savedExtract = parentExtract;
- CRoxieServerActionBaseActivity::start(parentExtractSize, parentExtract, paused);
- executeDependencies(parentExtractSize, parentExtract, WhenBeforeId);
- executeDependencies(parentExtractSize, parentExtract, WhenParallelId); // MORE: This should probably be done in parallel!
- }
- virtual void stop()
- {
- if (state != STATEstopped)
- {
- stopDependencies(savedExtractSize, savedExtract, aborted ? WhenSuccessId : WhenFailureId); // these are NOT going to execute
- executeDependencies(savedExtractSize, savedExtract, aborted ? WhenFailureId : WhenSuccessId);
- }
- CRoxieServerActionBaseActivity::stop();
- }
- virtual void doExecuteAction(unsigned parentExtractSize, const byte * parentExtract)
- {
- executeDependencies(parentExtractSize, parentExtract, 1);
- }
- protected:
- unsigned savedExtractSize;
- const byte *savedExtract;
- };
- class CRoxieServerWhenActionActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerWhenActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), isRoot(_isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerWhenActionActivity(_ctx, this, _probeManager);
- }
- virtual bool isSink() const
- {
- return isRoot;
- }
- private:
- bool isRoot;
- };
- extern IRoxieServerActivityFactory *createRoxieServerWhenActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerWhenActionActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //=================================================================================
- class CRoxieServerStrandedParseActivity : public CRoxieServerStrandedActivity
- {
- INlpParseAlgorithm * algorithm;
- class ParseProcessor : public StrandProcessor, implements IMatchedAction
- {
- protected:
- IHThorParseArg &helper;
- INlpParser * parser;
- INlpResultIterator * rowIter;
- const void * in;
- char * curSearchText;
- size32_t curSearchTextLen;
- bool processRecord(const void * inRec)
- {
- if (helper.searchTextNeedsFree())
- rtlFree(curSearchText);
- curSearchTextLen = 0;
- curSearchText = NULL;
- helper.getSearchText(curSearchTextLen, curSearchText, inRec);
- return parser->performMatch(*this, in, curSearchTextLen, curSearchText);
- }
- public:
- ParseProcessor(CRoxieServerActivity &_parent, IEngineRowStream *_inputStream, IHThorParseArg &_helper, INlpParseAlgorithm * _algorithm)
- : StrandProcessor(_parent, _inputStream, false), helper(_helper)
- {
- parser = _algorithm->createParser(parent.queryCodeContext(), parent.queryId(), helper.queryHelper(), &helper);
- rowIter = parser->queryResultIter();
- in = NULL;
- curSearchText = NULL;
- curSearchTextLen = 0;
- }
- ~ParseProcessor()
- {
- ::Release(parser);
- }
- virtual void start()
- {
- StrandProcessor::start();
- numProcessedLastGroup = 0;
- curSearchTextLen = 0;
- curSearchText = NULL;
- in = NULL;
- parser->reset();
- }
- virtual void reset()
- {
- if (helper.searchTextNeedsFree())
- rtlFree(curSearchText);
- curSearchText = NULL;
- ReleaseClearRoxieRow(in);
- StrandProcessor::reset();
- }
- virtual const void * nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- if (rowIter->isValid())
- {
- OwnedConstRoxieRow out = rowIter->getRow();
- rowIter->next();
- processed++;
- return out.getClear();
- }
- ReleaseClearRoxieRow(in);
- in = inputStream->nextRow();
- if (!in)
- {
- if (processed == numProcessedLastGroup)
- in = inputStream->nextRow();
- if (!in)
- {
- processed = numProcessedLastGroup;
- return NULL;
- }
- in = inputStream->nextRow();
- if (!in)
- return NULL;
- }
- processRecord(in);
- rowIter->first();
- }
- }
- virtual unsigned onMatch(ARowBuilder & self, const void * curRecord, IMatchedResults * results, IMatchWalker * walker)
- {
- try
- {
- return helper.transform(self, curRecord, results, walker);
- }
- catch (IException *E)
- {
- throw parent.makeWrappedException(E);
- }
- }
- };
- public:
- CRoxieServerStrandedParseActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, INlpParseAlgorithm * _algorithm, const StrandOptions &_strandOptions)
- : CRoxieServerStrandedActivity(_ctx, _factory, _probeManager, _strandOptions), algorithm(_algorithm)
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerStrandedActivity::start(parentExtractSize, parentExtract, paused);
- onStartStrands();
- }
- virtual StrandProcessor *createStrandProcessor(IEngineRowStream *instream)
- {
- return new ParseProcessor(*this, instream, (IHThorParseArg &) basehelper, algorithm);
- }
- virtual StrandProcessor *createStrandSourceProcessor(bool inputOrdered) { throwUnexpected(); }
- };
- class CRoxieServerParseActivityFactory : public CRoxieServerActivityFactory
- {
- Owned<INlpParseAlgorithm> algorithm;
- Owned<IHThorParseArg> helper;
- StrandOptions strandOptions;
- public:
- CRoxieServerParseActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, IResourceContext *rc)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), strandOptions(_graphNode)
- {
- helper.setown((IHThorParseArg *) helperFactory());
- algorithm.setown(createThorParser(rc, *helper));
- optStableInput = false;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerStrandedParseActivity(_ctx, this, _probeManager, algorithm, strandOptions);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerParseActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, IResourceContext *rc)
- {
- return new CRoxieServerParseActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, rc);
- }
- //=====================================================================================================
- class CRoxieServerWorkUnitWriteActivity : public CRoxieServerInternalSinkActivity
- {
- IHThorWorkUnitWriteArg &helper;
- bool isReread;
- bool grouped;
- IRoxieServerContext *serverContext;
- public:
- CRoxieServerWorkUnitWriteActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, bool _isReread, unsigned _numOutputs)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, _numOutputs), helper((IHThorWorkUnitWriteArg &)basehelper), isReread(_isReread)
- {
- grouped = (helper.getFlags() & POFgrouped) != 0;
- serverContext = NULL;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerInternalSinkActivity::onCreate(_colocalParent);
- serverContext = ctx->queryServerContext();
- if (!serverContext)
- {
- throw MakeStringException(ROXIE_PIPE_ERROR, "Pipe output activity cannot be executed in slave context");
- }
- }
- virtual bool needsAllocator() const { return true; }
- virtual void onExecute()
- {
- int sequence = helper.getSequence();
- const char *storedName = helper.queryName();
- if (!storedName)
- storedName = "Dataset";
- MemoryBuffer result;
- bool saveInContext = (int) sequence < 0 || isReread;
- if (!meta.queryOriginal()) // this is a bit of a hack - don't know why no meta on an output....
- meta.set(input->queryOutputMeta());
- Owned<IOutputRowSerializer> rowSerializer;
- Owned<IXmlWriter> xmlwriter;
- bool appendRaw = false;
- IHpccProtocolResultsWriter *results = NULL;
- IHpccNativeProtocolResultsWriter *nativeResults = NULL;
- IHpccProtocolResponse *protocol = serverContext->queryProtocol();
- unsigned protocolFlags = (protocol) ? protocol->getFlags() : 0;
- if ((int) sequence >= 0)
- {
- results = (protocol) ? protocol->queryHpccResultsSection() : NULL;
- if (results)
- {
- nativeResults = dynamic_cast<IHpccNativeProtocolResultsWriter*>(results);
- xmlwriter.setown(results->addDataset(helper.queryName(), sequence, "Dataset", appendRaw, serverContext->getXmlFlags(), (helper.getFlags() & POFextend) != 0, serverContext->queryXmlns(sequence))); //xmlwriter only returned if needed
- }
- }
- size32_t outputLimitBytes = 0;
- IConstWorkUnit *workunit = serverContext->queryWorkUnit();
- if (workunit)
- {
- size32_t outputLimit;
- if (helper.getFlags() & POFmaxsize)
- outputLimit = helper.getMaxSize();
- else
- {
- // In absense of OPT_OUTPUTLIMIT check pre 5.2 legacy name OPT_OUTPUTLIMIT_LEGACY
- outputLimit = workunit->getDebugValueInt(OPT_OUTPUTLIMIT, workunit->getDebugValueInt(OPT_OUTPUTLIMIT_LEGACY, defaultDaliResultLimit));
- }
- if (outputLimit>defaultDaliResultOutputMax)
- throw MakeStringException(0, "Dali result outputs are restricted to a maximum of %d MB, the current limit is %d MB. A huge dali result usually indicates the ECL needs altering.", defaultDaliResultOutputMax, defaultDaliResultLimit);
- assertex(outputLimit<=0x1000); // 32bit limit because MemoryBuffer/CMessageBuffers involved etc.
- outputLimitBytes = outputLimit * 0x100000;
- }
- if (workunit != NULL || (results && protocol->getFlags() & HPCC_PROTOCOL_NATIVE_RAW))
- {
- ensureRowAllocator();
- rowSerializer.setown(rowAllocator->createDiskSerializer(ctx->queryCodeContext()));
- }
- __int64 initialProcessed = processed;
- RtlLinkedDatasetBuilder builder(rowAllocator);
- loop
- {
- const void *row = inputStream->nextRow();
- if (saveInContext)
- {
- if (row || grouped)
- builder.append(row);
- }
- if (grouped && (processed != initialProcessed))
- {
- if (workunit)
- result.append(row == NULL);
- if (results)
- {
- if (protocolFlags & HPCC_PROTOCOL_NATIVE_RAW && nativeResults)
- {
- char val = (char)(row == NULL);
- nativeResults->appendRaw(sequence, 1, &val);
- }
- else if (xmlwriter)
- {
- xmlwriter->outputBeginNested("Row", false);
- xmlwriter->outputCString("1", "@__GroupBreak__");
- xmlwriter->outputEndNested("Row");
- }
- }
- }
- if (!row)
- {
- row = inputStream->nextRow();
- if (!row)
- break;
- if (saveInContext)
- builder.append(row);
- }
- processed++;
- if (workunit)
- {
- CThorDemoRowSerializer serializerTarget(result);
- rowSerializer->serialize(serializerTarget, (const byte *) row);
- }
- if ((int) sequence >= 0)
- {
- if (appendRaw && nativeResults)
- {
- // MORE - should be able to serialize straight to the response...
- MemoryBuffer rowbuff;
- CThorDemoRowSerializer serializerTarget(rowbuff);
- rowSerializer->serialize(serializerTarget, (const byte *) row);
- nativeResults->appendRawRow(sequence, rowbuff.length(), rowbuff.toByteArray());
- }
- else if (xmlwriter)
- {
- xmlwriter->outputBeginNested(DEFAULTXMLROWTAG, false);
- helper.serializeXml((byte *) row, *xmlwriter);
- xmlwriter->outputEndNested(DEFAULTXMLROWTAG);
- results->finalizeXmlRow(sequence);
- }
- else if (nativeResults)
- {
- SimpleOutputWriter x;
- helper.serializeXml((byte *) row, x);
- x.newline();
- nativeResults->appendSimpleRow(sequence, x.str());
- }
- }
- ReleaseRoxieRow(row);
- if (outputLimitBytes && result.length() > outputLimitBytes)
- {
- StringBuffer errMsg("Dataset too large to output to workunit (limit ");
- errMsg.append(outputLimitBytes/0x100000).append(" megabytes), in result (");
- const char *name = helper.queryName();
- if (name)
- errMsg.append("name=").append(name);
- else
- errMsg.append("sequence=").append(helper.getSequence());
- errMsg.append(")");
- throw MakeStringExceptionDirect(0, errMsg.str());
- }
- }
- if (xmlwriter)
- xmlwriter->outputEndArray(DEFAULTXMLROWTAG);
- if (saveInContext)
- serverContext->appendResultDeserialized(storedName, sequence, builder.getcount(), builder.linkrows(), (helper.getFlags() & POFextend) != 0, LINK(meta.queryOriginal()));
- if (workunit)
- serverContext->appendResultRawContext(storedName, sequence, result.length(), result.toByteArray(), processed, (helper.getFlags() & POFextend) != 0, false); // MORE - shame to do extra copy...
- }
- };
- class CRoxieServerWorkUnitWriteActivityFactory : public CRoxieServerInternalSinkFactory
- {
- bool isReread;
- public:
- CRoxieServerWorkUnitWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, bool _isRoot)
- : CRoxieServerInternalSinkFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot)
- {
- isReread = usageCount > 0;
- Owned<IHThorWorkUnitWriteArg> helper = (IHThorWorkUnitWriteArg *) helperFactory();
- isInternal = (helper->getSequence()==ResultSequenceInternal);
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerWorkUnitWriteActivity(_ctx, this, _probeManager, isReread, usageCount);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerWorkUnitWriteActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, bool _isRoot)
- {
- return new CRoxieServerWorkUnitWriteActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot);
- }
- //=====================================================================================================
- class CRoxieServerWorkUnitWriteDictActivity : public CRoxieServerInternalSinkActivity
- {
- IHThorDictionaryWorkUnitWriteArg &helper;
- IRoxieServerContext *serverContext;
- public:
- CRoxieServerWorkUnitWriteDictActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _usageCount)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, _usageCount), helper((IHThorDictionaryWorkUnitWriteArg &)basehelper)
- {
- serverContext = NULL;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerInternalSinkActivity::onCreate(_colocalParent);
- serverContext = ctx->queryServerContext();
- if (!serverContext)
- {
- throw MakeStringException(ROXIE_PIPE_ERROR, "Write Dictionary activity cannot be executed in slave context");
- }
- }
- virtual bool needsAllocator() const { return true; }
- virtual void onExecute()
- {
- int sequence = helper.getSequence();
- const char *storedName = helper.queryName();
- assertex(storedName && *storedName);
- assertex(sequence < 0);
- RtlLinkedDictionaryBuilder builder(rowAllocator, helper.queryHashLookupInfo());
- loop
- {
- const void *row = inputStream->ungroupedNextRow();
- if (!row)
- break;
- builder.appendOwn(row);
- processed++;
- }
- serverContext->appendResultDeserialized(storedName, sequence, builder.getcount(), builder.linkrows(), (helper.getFlags() & POFextend) != 0, LINK(meta.queryOriginal()));
- }
- };
- class CRoxieServerWorkUnitWriteDictActivityFactory : public CRoxieServerInternalSinkFactory
- {
- public:
- CRoxieServerWorkUnitWriteDictActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, bool _isRoot)
- : CRoxieServerInternalSinkFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot)
- {
- Owned<IHThorDictionaryWorkUnitWriteArg> helper = (IHThorDictionaryWorkUnitWriteArg *) helperFactory();
- isInternal = (helper->getSequence()==ResultSequenceInternal);
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerWorkUnitWriteDictActivity(_ctx, this, _probeManager, usageCount);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerWorkUnitWriteDictActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, bool _isRoot)
- {
- return new CRoxieServerWorkUnitWriteDictActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot);
- }
- //=================================================================================
- class CRoxieServerRemoteResultActivity : public CRoxieServerInternalSinkActivity
- {
- IHThorRemoteResultArg &helper;
- public:
- CRoxieServerRemoteResultActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, unsigned _numOutputs)
- : CRoxieServerInternalSinkActivity(_ctx, _factory, _probeManager, _numOutputs), helper((IHThorRemoteResultArg &)basehelper)
- {
- }
- virtual void onExecute()
- {
- OwnedConstRoxieRow row = inputStream->nextRow();
- helper.sendResult(row); // should be only one row or something has gone wrong!
- }
- };
- class CRoxieServerRemoteResultActivityFactory : public CRoxieServerInternalSinkFactory
- {
- public:
- CRoxieServerRemoteResultActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, bool _isRoot)
- : CRoxieServerInternalSinkFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot)
- {
- Owned<IHThorRemoteResultArg> helper = (IHThorRemoteResultArg *) helperFactory();
- isInternal = (helper->getSequence()==ResultSequenceInternal);
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerRemoteResultActivity(_ctx, this, _probeManager, usageCount);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerRemoteResultActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, unsigned _usageCount, bool _isRoot)
- {
- return new CRoxieServerRemoteResultActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _usageCount, _isRoot);
- }
- //=================================================================================
- class CRoxieServerXmlParseActivity : public CRoxieServerActivity, implements IXMLSelect
- {
- IHThorXmlParseArg &helper;
- Owned<IXMLParse> xmlParser;
- const void * in;
- char * srchStr;
- unsigned numProcessedLastGroup;
- bool srchStrNeedsFree;
- Owned<IColumnProvider> lastMatch;
- public:
- IMPLEMENT_IINTERFACE_USING(CRoxieServerActivity)
- CRoxieServerXmlParseActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- helper((IHThorXmlParseArg &)basehelper)
- {
- srchStrNeedsFree = helper.searchTextNeedsFree();
- numProcessedLastGroup = 0;
- srchStr = NULL;
- in = NULL;
- }
- ~CRoxieServerXmlParseActivity()
- {
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- numProcessedLastGroup = 0;
- srchStr = NULL;
- in = NULL;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- if (helper.searchTextNeedsFree())
- rtlFree(srchStr);
- srchStr = NULL;
- ReleaseClearRoxieRow(in);
- xmlParser.clear();
- CRoxieServerActivity::reset();
- }
- virtual void match(IColumnProvider &entry, offset_t startOffset, offset_t endOffset)
- {
- lastMatch.set(&entry);
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- if(xmlParser)
- {
- loop
- {
- if(!xmlParser->next())
- {
- if (srchStrNeedsFree)
- {
- rtlFree(srchStr);
- srchStr = NULL;
- }
- xmlParser.clear();
- break;
- }
- if(lastMatch)
- {
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize = helper.transform(rowBuilder, in, lastMatch);
- lastMatch.clear();
- if (outSize)
- {
- processed++;
- return rowBuilder.finalizeRowClear(outSize);
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- }
- }
- ReleaseClearRoxieRow(in);
- in = inputStream->nextRow();
- if(!in)
- {
- if(numProcessedLastGroup == processed)
- in = inputStream->nextRow();
- if(!in)
- {
- numProcessedLastGroup = processed;
- return NULL;
- }
- }
- size32_t srchLen;
- helper.getSearchText(srchLen, srchStr, in);
- OwnedRoxieString xmlIteratorPath(helper.getXmlIteratorPath());
- xmlParser.setown(createXMLParse(srchStr, srchLen, xmlIteratorPath, *this));
- }
- }
- };
- class CRoxieServerXmlParseActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerXmlParseActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerXmlParseActivity(_ctx, this, _probeManager);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerXmlParseActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerXmlParseActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //=====================================================================================================
- class CRoxieServerDiskReadBaseActivity : public CRoxieServerActivity, implements IRoxieServerErrorHandler, implements IIndexReadContext
- {
- protected:
- IHThorDiskReadBaseArg &helper;
- IHThorCompoundExtra * compoundHelper;
- RemoteActivityId remoteId; // Note we copy it rather than reference
- Owned<CSkippableRemoteResultAdaptor> remote;
- unsigned numParts;
- unsigned __int64 rowLimit;
- unsigned __int64 stopAfter;
- Linked<IInMemoryIndexManager> manager;
- Owned<IInMemoryIndexCursor> cursor;
- Owned<IDirectReader> reader;
- CThorContiguousRowBuffer deserializeSource;
- Owned<ISourceRowPrefetcher> prefetcher;
- bool eof;
- bool isKeyed;
- bool variableFileName;
- bool isOpt;
- bool sorted;
- bool maySkip;
- bool isLocal;
- bool forceRemote;
- CachedOutputMetaData diskSize;
- Owned<const IResolvedFile> varFileInfo;
- Owned<IFileIOArray> varFiles;
- inline bool useRemote()
- {
- return remote != NULL && (forceRemote || numParts > 1);
- }
- public:
- IMPLEMENT_IINTERFACE_USING(CRoxieServerActivity)
- CRoxieServerDiskReadBaseActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, unsigned _numParts, bool _isLocal, bool _sorted, bool _maySkip, IInMemoryIndexManager *_manager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- helper((IHThorDiskReadBaseArg &)basehelper),
- numParts(_numParts),
- remoteId(_remoteId),
- manager(_manager),
- isLocal(_isLocal),
- sorted(_sorted),
- maySkip(_maySkip),
- deserializeSource(NULL)
- {
- forceRemote = factory->queryQueryFactory().queryOptions().disableLocalOptimizations;
- if ((forceRemote || numParts != 1) && !isLocal) // NOTE : when numParts == 0 (variable case) we create, even though we may not use
- remote.setown(new CSkippableRemoteResultAdaptor(_ctx, this, remoteId, meta.queryOriginal(), helper, *this, sorted, false, _maySkip));
- compoundHelper = NULL;
- eof = false;
- rowLimit = (unsigned __int64) -1;
- isKeyed = false;
- stopAfter = I64C(0x7FFFFFFFFFFFFFFF);
- diskSize.set(helper.queryDiskRecordSize());
- variableFileName = allFilesDynamic || factory->queryQueryFactory().isDynamic() || ((helper.getFlags() & (TDXvarfilename|TDXdynamicfilename)) != 0);
- isOpt = (helper.getFlags() & TDRoptional) != 0;
- }
- virtual const IResolvedFile *queryVarFileInfo() const
- {
- return varFileInfo;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- if (remote)
- remote->onCreate(_colocalParent);
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if (compoundHelper)
- {
- rowLimit = compoundHelper->getRowLimit();
- stopAfter = compoundHelper->getChooseNLimit();
- }
- if (!helper.canMatchAny())
- eof = true;
- else
- {
- if (variableFileName)
- {
- OwnedRoxieString fileName(helper.getFileName());
- varFileInfo.setown(resolveLFN(fileName, isOpt));
- numParts = 0;
- if (varFileInfo)
- numParts = varFileInfo->getNumParts();
- }
- if (!numParts)
- {
- eof = true;
- }
- else if (useRemote())
- {
- remote->onStart(parentExtractSize, parentExtract);
- remote->setLimits(rowLimit, (unsigned __int64) -1, stopAfter);
- unsigned fileNo = 0;
- // Translation into a message per channel done elsewhere....
- remote->getMem(0, fileNo, 0);
- remote->flush();
- remote->senddone();
- }
- else
- {
- if (variableFileName)
- {
- unsigned channel = isLocal ? factory->queryQueryFactory().queryChannel() : 0;
- varFiles.setown(varFileInfo->getIFileIOArray(isOpt, channel));
- manager.setown(varFileInfo->getIndexManager(isOpt, channel, varFiles, diskSize, false, 0));
- }
- assertex(manager != NULL);
- helper.createSegmentMonitors(this);
- if (cursor)
- {
- isKeyed = cursor->selectKey();
- cursor->reset();
- }
- if (!isKeyed)
- {
- reader.setown(manager->createReader(0, 0, 1));
- deserializeSource.setStream(reader);
- prefetcher.setown(diskSize.queryOriginal()->createDiskPrefetcher(ctx->queryCodeContext(), activityId));
- }
- helper.setCallback(reader ? reader->queryThorDiskCallback() : cursor);
- }
- }
- }
- virtual void append(IKeySegmentMonitor *segment)
- {
- if (!segment->isWild())
- {
- if (!cursor)
- cursor.setown(manager->createCursor());
- cursor->append(segment);
- }
- }
- virtual unsigned ordinality() const
- {
- return cursor ? cursor->ordinality() : 0;
- }
- virtual IKeySegmentMonitor *item(unsigned idx) const
- {
- return cursor ? cursor->item(idx) : 0;
- }
- virtual void setMergeBarrier(unsigned barrierOffset)
- {
- // no merging so no issue...
- }
- virtual void stop()
- {
- if (useRemote())
- remote->onStop();
- CRoxieServerActivity::stop();
- }
- virtual void reset()
- {
- if (useRemote())
- {
- processed = remote->processed;
- remote->processed = 0;
- remote->onReset();
- }
- varFileInfo.clear();
- eof = false;
- if (cursor)
- cursor->reset();
- deserializeSource.clearStream();
- CRoxieServerActivity::reset();
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() called for source activity");
- }
- virtual void onLimitExceeded(bool isKeyed)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- assertex(compoundHelper);
- if (isKeyed) // MORE does this exist for diskread? should it?
- {
- if (helper.getFlags() & (TDRkeyedlimitskips|TDRkeyedlimitcreates))
- {
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- throw makeLimitSkipException(true);
- }
- else
- {
- UNIMPLEMENTED;
- //compoundHelper->onKeyedLimitExceeded(); Doesn't exist - though the flags do... interesting...
- }
- }
- else
- {
- if (helper.getFlags() & (TDRlimitskips|TDRlimitcreates))
- {
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- throw makeLimitSkipException(false);
- }
- else
- compoundHelper->onLimitExceeded();
- }
- }
- virtual const void * createLimitFailRow(bool isKeyed)
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- IHThorSourceLimitTransformExtra *limitTransformExtra = static_cast<IHThorSourceLimitTransformExtra *>(helper.selectInterface(TAIsourcelimittransformextra_1));
- assertex(limitTransformExtra);
- size32_t outSize = isKeyed ? limitTransformExtra->transformOnKeyedLimitExceeded(rowBuilder) : limitTransformExtra->transformOnLimitExceeded(rowBuilder);
- if (outSize)
- return rowBuilder.finalizeRowClear(outSize);
- return NULL;
- }
- };
- class CRoxieServerDiskReadActivity : public CRoxieServerDiskReadBaseActivity
- {
- IHThorCompoundReadExtra * readHelper;
- ConstPointerArray readrows;
- bool readAheadDone;
- unsigned readIndex;
- public:
- CRoxieServerDiskReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, unsigned _numParts, bool _isLocal, bool _sorted, bool _maySkip, IInMemoryIndexManager *_manager)
- : CRoxieServerDiskReadBaseActivity(_ctx, _factory, _probeManager, _remoteId, _numParts, _isLocal, _sorted, _maySkip, _manager)
- {
- compoundHelper = (IHThorDiskReadArg *)&helper;
- readHelper = (IHThorDiskReadArg *)&helper;
- readAheadDone = false;
- readIndex = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- rowLimit = compoundHelper->getRowLimit();
- stopAfter = compoundHelper->getChooseNLimit();
- CRoxieServerDiskReadBaseActivity::start(parentExtractSize, parentExtract, paused);
- readAheadDone = false;
- readIndex = 0;
- }
- virtual void reset()
- {
- roxiemem::ReleaseRoxieRowRange(readrows.getArray(), readIndex, readrows.ordinality());
- readrows.kill();
- readAheadDone = false;
- readIndex = 0;
- CRoxieServerDiskReadBaseActivity::reset();
- }
- virtual const void *nextRow()
- {
- if (eof)
- return NULL;
- else if (useRemote())
- return remote->nextRow();
- else if (maySkip)
- {
- if (!readAheadDone)
- {
- unsigned preprocessed = 0;
- while (!eof)
- {
- const void *row = _nextRow();
- if (row)
- preprocessed++;
- if (preprocessed > rowLimit)
- {
- ReleaseRoxieRow(row);
- roxiemem::ReleaseRoxieRowRange(readrows.getArray(), readIndex, readrows.ordinality());
- readrows.kill();
- readIndex = 0;
- eof = true;
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- if (helper.getFlags() & TDRlimitskips)
- return NULL;
- else if (helper.getFlags() & TDRlimitcreates)
- return createLimitFailRow(false);
- else
- throwUnexpected();
- }
- if (preprocessed > stopAfter) // MORE - bit of a strange place to check
- {
- eof = true;
- ReleaseRoxieRow(row);
- break;
- }
- readrows.append(row);
- }
- readAheadDone = true;
- }
- if (readrows.isItem(readIndex))
- {
- const void *ret = readrows.item(readIndex++);
- if (ret)
- processed++;
- return ret;
- }
- else
- {
- eof = true;
- return NULL;
- }
- }
- else
- {
- const void *ret = _nextRow();
- if (ret)
- {
- processed++;
- if (processed > rowLimit)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- ReleaseRoxieRow(ret);
- compoundHelper->onLimitExceeded();
- throwUnexpected(); // onLimitExceeded is not supposed to return
- }
- if (processed > stopAfter) // MORE - bit of a strange place to check
- {
- eof = true;
- ReleaseRoxieRow(ret);
- return NULL;
- }
- }
- return ret;
- }
- }
- const void *_nextRow()
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned transformedSize = 0;
- if (isKeyed)
- {
- loop
- {
- const void *nextCandidate = cursor->nextMatch();
- if (!nextCandidate)
- {
- eof = true;
- return NULL;
- }
- transformedSize = readHelper->transform(rowBuilder, nextCandidate);
- if (transformedSize)
- break;
- }
- }
- else // use reader...
- {
- assertex(reader != NULL);
- loop
- {
- if (deserializeSource.eos())
- {
- eof = true;
- return NULL;
- }
- prefetcher->readAhead(deserializeSource);
- const byte *nextRec = deserializeSource.queryRow();
- if (cursor && cursor->isFiltered(nextRec))
- transformedSize = 0;
- else
- transformedSize = readHelper->transform(rowBuilder, nextRec);
- deserializeSource.finishedRow();
- if (transformedSize)
- break;
- }
- }
- return rowBuilder.finalizeRowClear(transformedSize);
- }
- };
- class CRoxieServerXmlReadActivity : public CRoxieServerDiskReadBaseActivity, implements IXMLSelect, implements IThorDiskCallback
- {
- IHThorXmlReadArg * readHelper;
- Owned<IXmlToRowTransformer> rowTransformer;
- Owned<IXMLParse> xmlParser;
- Owned<IColumnProvider> lastMatch;
- unsigned __int64 fileoffset;
- public:
- IMPLEMENT_IINTERFACE_USING(CRoxieServerDiskReadBaseActivity)
- CRoxieServerXmlReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, unsigned _numParts, bool _isLocal, bool _sorted, bool _maySkip, IInMemoryIndexManager *_manager)
- : CRoxieServerDiskReadBaseActivity(_ctx, _factory, _probeManager, _remoteId, _numParts, _isLocal, _sorted, _maySkip, _manager)
- {
- compoundHelper = NULL;
- readHelper = (IHThorXmlReadArg *)&helper;
- fileoffset = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- rowLimit = readHelper->getRowLimit();
- stopAfter = readHelper->getChooseNLimit();
- CRoxieServerDiskReadBaseActivity::start(parentExtractSize, parentExtract, paused);
- if (!useRemote())
- {
- rowTransformer.set(readHelper->queryTransformer());
- assertex(reader != NULL);
- OwnedRoxieString xmlIterator(readHelper->getXmlIteratorPath());
- if (factory->getKind()==TAKjsonread)
- xmlParser.setown(createJSONParse(*reader->querySimpleStream(), xmlIterator, *this, (0 != (TDRxmlnoroot & readHelper->getFlags()))?ptr_noRoot:ptr_none, (readHelper->getFlags() & TDRusexmlcontents) != 0));
- else
- xmlParser.setown(createXMLParse(*reader->querySimpleStream(), xmlIterator, *this, (0 != (TDRxmlnoroot & readHelper->getFlags()))?ptr_noRoot:ptr_none, (readHelper->getFlags() & TDRusexmlcontents) != 0));
- }
- }
- virtual void reset()
- {
- CRoxieServerDiskReadBaseActivity::reset();
- rowTransformer.clear();
- xmlParser.clear();
- }
- virtual void match(IColumnProvider &entry, offset_t startOffset, offset_t endOffset)
- {
- fileoffset = startOffset;
- lastMatch.set(&entry);
- }
- //interface IThorDiskCallback
- virtual unsigned __int64 getFilePosition(const void * row)
- {
- return fileoffset;
- }
- virtual unsigned __int64 getLocalFilePosition(const void * row)
- {
- return reader->makeFilePositionLocal(fileoffset);
- }
- virtual const char * queryLogicalFilename(const void * row)
- {
- return reader->queryThorDiskCallback()->queryLogicalFilename(row);
- }
- virtual const void *nextRow()
- {
- if (eof)
- return NULL;
- else if (useRemote())
- return remote->nextRow();
- assertex(xmlParser != NULL);
- try
- {
- while (!eof)
- {
- //call to next() will callback on the IXmlSelect interface
- bool gotNext = false;
- gotNext = xmlParser->next();
- if(!gotNext)
- eof = true;
- else if (lastMatch)
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned sizeGot = rowTransformer->transform(rowBuilder, lastMatch, this);
- lastMatch.clear();
- fileoffset = 0;
- if (sizeGot)
- {
- OwnedConstRoxieRow ret = rowBuilder.finalizeRowClear(sizeGot);
- if (processed > rowLimit)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- readHelper->onLimitExceeded();
- throwUnexpected(); // onLimitExceeded is not supposed to return
- }
- processed++;
- if (processed > stopAfter) // MORE - bit of a strange place to check
- {
- eof = true;
- return NULL;
- }
- return ret.getClear();
- }
- }
- }
- return NULL;
- }
- catch(IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- };
- class CRoxieServerCsvReadActivity : public CRoxieServerDiskReadBaseActivity
- {
- IHThorCsvReadArg *readHelper;
- ICsvParameters * csvInfo;
- unsigned headerLines;
- unsigned maxDiskSize;
- CSVSplitter csvSplitter;
- unsigned __int64 localOffset;
- const char *quotes;
- const char *separators;
- const char *terminators;
- const char *escapes;
- size32_t maxRowSize;
- public:
- CRoxieServerCsvReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId,
- unsigned _numParts, bool _isLocal, bool _sorted, bool _maySkip, IInMemoryIndexManager *_manager,
- const char *_quotes, const char *_separators, const char *_terminators, const char *_escapes, size32_t _maxRowSize)
- : CRoxieServerDiskReadBaseActivity(_ctx, _factory, _probeManager, _remoteId, _numParts, _isLocal, _sorted, _maySkip, _manager),
- quotes(_quotes), separators(_separators), terminators(_terminators), escapes(_escapes), maxRowSize(_maxRowSize)
- {
- compoundHelper = NULL;
- readHelper = (IHThorCsvReadArg *)&helper;
- rowLimit = readHelper->getRowLimit();
- stopAfter = readHelper->getChooseNLimit();
- csvInfo = readHelper->queryCsvParameters();
- maxDiskSize = csvInfo->queryMaxSize();
- localOffset = 0;
- headerLines = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- rowLimit = readHelper->getRowLimit();
- stopAfter = readHelper->getChooseNLimit();
- CRoxieServerDiskReadBaseActivity::start(parentExtractSize, parentExtract, paused);
- if (!useRemote())
- {
- headerLines = csvInfo->queryHeaderLen();
- if (headerLines && isLocal && reader->queryFilePart() != 1)
- headerLines = 0; // MORE - you could argue that if SINGLE not specified, should skip from all parts. But it would be painful since we have already concatenated and no-one else does...
- if (!eof)
- {
- if (varFileInfo)
- {
- const IPropertyTree *options = varFileInfo->queryProperties();
- if (options)
- {
- quotes = options->queryProp("@csvQuote");
- separators = options->queryProp("@csvSeparate");
- terminators = options->queryProp("@csvTerminate");
- escapes = options->queryProp("@csvEscape");
- }
- }
- csvSplitter.init(readHelper->getMaxColumns(), csvInfo, quotes, separators, terminators, escapes);
- }
- }
- }
- virtual const void *nextRow()
- {
- if (eof)
- return NULL;
- else if (useRemote())
- return remote->nextRow();
- try
- {
- while (!eof)
- {
- if (reader->eos())
- {
- eof = true;
- break;
- }
- // MORE - there are rumours of a csvSplitter that operates on a stream... if/when it exists, this should use it
- size32_t rowSize = 4096; // MORE - make configurable
- size32_t thisLineLength;
- loop
- {
- size32_t avail;
- const void *peek = reader->peek(rowSize, avail);
- thisLineLength = csvSplitter.splitLine(avail, (const byte *)peek);
- if (thisLineLength < rowSize || avail < rowSize)
- break;
- if (rowSize == maxRowSize)
- throw MakeStringException(0, "File contained a line of length greater than %d bytes.", maxRowSize);
- if (rowSize >= maxRowSize/2)
- rowSize = maxRowSize;
- else
- rowSize += rowSize;
- }
- if (headerLines)
- {
- headerLines--;
- reader->skip(thisLineLength);
- }
- else
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned transformedSize = readHelper->transform(rowBuilder, csvSplitter.queryLengths(), (const char * *)csvSplitter.queryData());
- reader->skip(thisLineLength);
- if (transformedSize)
- {
- OwnedConstRoxieRow ret = rowBuilder.finalizeRowClear(transformedSize);
- if (processed > rowLimit)
- {
- readHelper->onLimitExceeded();
- throwUnexpected(); // onLimitExceeded is not supposed to return
- }
- processed++;
- if (processed > stopAfter) // MORE - bit of a strange place to check
- {
- eof = true;
- return NULL;
- }
- return ret.getClear();
- }
- }
- }
- return NULL;
- }
- catch(IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- virtual void reset()
- {
- CRoxieServerDiskReadBaseActivity::reset();
- csvSplitter.reset();
- localOffset = 0;
- }
- virtual unsigned __int64 getFilePosition(const void * row)
- {
- UNIMPLEMENTED; // we know offset in the reader but not sure it helps us much
- }
- virtual unsigned __int64 getLocalFilePosition(const void * row)
- {
- UNIMPLEMENTED;
- }
- };
- class CRoxieServerDiskNormalizeActivity : public CRoxieServerDiskReadBaseActivity
- {
- IHThorDiskNormalizeArg *normalizeHelper;
- bool firstPending;
- public:
- CRoxieServerDiskNormalizeActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, unsigned _numParts, bool _isLocal, bool _sorted, IInMemoryIndexManager *_manager)
- : CRoxieServerDiskReadBaseActivity(_ctx, _factory, _probeManager, _remoteId, _numParts, _isLocal, _sorted, false, _manager)
- {
- compoundHelper = (IHThorDiskNormalizeArg *)&helper;
- normalizeHelper = (IHThorDiskNormalizeArg *)&helper;
- firstPending = true;
- }
- virtual void reset()
- {
- firstPending = true;
- CRoxieServerDiskReadBaseActivity::reset();
- }
- virtual const void *nextRow()
- {
- if (eof)
- return NULL;
- else if (useRemote())
- return remote->nextRow();
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned transformedSize = 0;
- if (isKeyed)
- {
- loop
- {
- while (firstPending)
- {
- const void *nextCandidate = cursor->nextMatch();
- if (!nextCandidate)
- {
- eof = true;
- return NULL;
- }
- if (normalizeHelper->first(nextCandidate))
- {
- firstPending = false;
- break;
- }
- }
- transformedSize = normalizeHelper->transform(rowBuilder);
- firstPending = !normalizeHelper->next();
- if (transformedSize)
- break;
- }
- }
- else
- {
- assertex(reader != NULL);
- loop
- {
- while (firstPending)
- {
- if (deserializeSource.eos())
- {
- eof = true;
- return NULL;
- }
- prefetcher->readAhead(deserializeSource);
- const byte *nextRec = deserializeSource.queryRow();
- if (!cursor || !cursor->isFiltered(nextRec))
- {
- if (normalizeHelper->first(nextRec))
- firstPending = false;
- }
- deserializeSource.finishedRow();
- }
- transformedSize = normalizeHelper->transform(rowBuilder);
- firstPending = !normalizeHelper->next();
- if (transformedSize)
- break;
- }
- }
- OwnedConstRoxieRow recBuffer = rowBuilder.finalizeRowClear(transformedSize);
- processed++;
- if (processed > rowLimit)
- {
- compoundHelper->onLimitExceeded();
- throwUnexpected(); // onLimitExceeded is not supposed to return
- }
- if (processed > stopAfter) // MORE - bit of a strange place to check
- {
- eof = true;
- return NULL;
- }
- return recBuffer.getClear();
- }
- };
- class CRoxieServerDiskAggregateBaseActivity : public CRoxieServerDiskReadBaseActivity
- {
- protected:
- bool done;
- public:
- CRoxieServerDiskAggregateBaseActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, unsigned _numParts, bool _isLocal, IInMemoryIndexManager *_manager)
- : CRoxieServerDiskReadBaseActivity(_ctx, _factory, _probeManager, _remoteId, _numParts, _isLocal, false, false, _manager),
- done(false)
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- done = false;
- CRoxieServerDiskReadBaseActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- if (idx==(unsigned)-1)
- idx = 0;
- return idx ? NULL: this;
- }
- };
- class CRoxieServerDiskCountActivity : public CRoxieServerDiskAggregateBaseActivity
- {
- IHThorDiskCountArg & countHelper;
- unsigned __int64 choosenLimit;
- IHThorSourceCountLimit *limitHelper;
- unsigned __int64 getSkippedCount()
- {
- unsigned flags = countHelper.getFlags();
- if (flags & TDRlimitskips)
- return 0;
- else if (flags & TDRlimitcreates)
- return 1;
- else
- {
- assertex(limitHelper);
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- limitHelper->onLimitExceeded();
- throwUnexpected(); // onLimitExceeded should always throw exception
- }
- }
- public:
- CRoxieServerDiskCountActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, unsigned _numParts, bool _isLocal, IInMemoryIndexManager *_manager)
- : CRoxieServerDiskAggregateBaseActivity(_ctx, _factory, _probeManager, _remoteId, _numParts, _isLocal, _manager),
- countHelper((IHThorDiskCountArg &)basehelper)
- {
- limitHelper = static_cast<IHThorSourceCountLimit *>(basehelper.selectInterface(TAIsourcecountlimit_1));
- choosenLimit = 0;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- choosenLimit = countHelper.getChooseNLimit();
- if (limitHelper)
- {
- rowLimit = limitHelper->getRowLimit();
- // keyedLimit = limitHelper->getKeyedLimit(); // more - should there be one?
- }
- CRoxieServerDiskAggregateBaseActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (done) return NULL;
- done = true;
- unsigned __int64 totalCount = 0;
- if (helper.canMatchAny() && !eof)
- {
- if (useRemote())
- {
- loop
- {
- const void * next = remote->nextRow();
- if (!next)
- break;
- if (meta.getFixedSize() == 1)
- totalCount += *(byte *)next;
- else
- totalCount += *(unsigned __int64 *)next;
- ReleaseRoxieRow(next);
- if (totalCount > rowLimit)
- {
- totalCount = getSkippedCount();
- break;
- }
- else if (totalCount >= choosenLimit)
- {
- totalCount = choosenLimit;
- break;
- }
- }
- }
- else
- {
- if (isKeyed)
- {
- loop
- {
- const void *nextCandidate = cursor->nextMatch();
- if (!nextCandidate)
- break;
- totalCount += countHelper.numValid(nextCandidate);
- if (totalCount > rowLimit)
- {
- totalCount = getSkippedCount();
- break;
- }
- else if (totalCount >= choosenLimit)
- {
- totalCount = choosenLimit;
- break;
- }
- }
- }
- else
- {
- assertex(reader != NULL);
- while (!deserializeSource.eos())
- {
- prefetcher->readAhead(deserializeSource);
- const byte *nextRec = deserializeSource.queryRow();
- if (!cursor || !cursor->isFiltered(nextRec))
- {
- totalCount += countHelper.numValid(nextRec);
- }
- deserializeSource.finishedRow();
- if (totalCount > rowLimit)
- {
- totalCount = getSkippedCount();
- break;
- }
- else if (totalCount >= choosenLimit)
- {
- totalCount = choosenLimit;
- break;
- }
- }
- }
- }
- }
- size32_t rowSize = meta.getFixedSize();
- void * result = rowAllocator->createRow();
- if (rowSize == 1)
- *(byte *)result = (byte)totalCount;
- else
- {
- assertex(rowSize == sizeof(unsigned __int64));
- *(unsigned __int64 *)result = totalCount;
- }
- return rowAllocator->finalizeRow(rowSize, result, rowSize);
- }
- };
- class CRoxieServerDiskAggregateActivity : public CRoxieServerDiskAggregateBaseActivity
- {
- IHThorCompoundAggregateExtra & aggregateHelper;
- public:
- CRoxieServerDiskAggregateActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId,
- unsigned _numParts, bool _isLocal, IInMemoryIndexManager *_manager)
- : CRoxieServerDiskAggregateBaseActivity(_ctx, _factory, _probeManager, _remoteId, _numParts, _isLocal, _manager),
- aggregateHelper((IHThorDiskAggregateArg &)basehelper)
- {
- }
- const void * gatherMerged()
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator, false);
- size32_t finalSize = 0;
- if (useRemote())
- {
- const void * firstRow = remote->nextRow();
- if (!firstRow)
- {
- rowBuilder.ensureRow();
- finalSize = aggregateHelper.clearAggregate(rowBuilder);
- }
- else
- {
- // NOTE need to clone this because going to modify below, could special case 1 row only
- finalSize = cloneRow(rowBuilder, firstRow, meta);
- ReleaseRoxieRow(firstRow);
- }
- loop
- {
- const void * next = remote->nextRow();
- if (!next)
- break;
- finalSize = aggregateHelper.mergeAggregate(rowBuilder, next);
- ReleaseRoxieRow(next);
- }
- }
- else
- {
- aggregateHelper.clearAggregate(rowBuilder);
- if (helper.canMatchAny() && !eof)
- {
- if (isKeyed)
- {
- loop
- {
- const void *next = cursor->nextMatch();
- if (!next)
- break;
- aggregateHelper.processRow(rowBuilder, next);
- }
- }
- else
- {
- assertex(reader != NULL);
- while (!deserializeSource.eos())
- {
- prefetcher->readAhead(deserializeSource);
- const byte *nextRec = deserializeSource.queryRow();
- if (!cursor || !cursor->isFiltered(nextRec))
- {
- aggregateHelper.processRow(rowBuilder, nextRec);
- }
- deserializeSource.finishedRow();
- }
- }
- }
- finalSize = meta.getRecordSize(rowBuilder.getSelf());
- }
- return rowBuilder.finalizeRowClear(finalSize);
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (done) return NULL;
- const void * ret = gatherMerged();
- done = true;
- return ret;
- }
- };
- class CRoxieServerDiskGroupAggregateActivity : public CRoxieServerDiskAggregateBaseActivity
- {
- IHThorDiskGroupAggregateArg & aggregateHelper;
- RowAggregator resultAggregator;
- bool gathered;
- public:
- CRoxieServerDiskGroupAggregateActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, unsigned _numParts, bool _isLocal, IInMemoryIndexManager *_manager)
- : CRoxieServerDiskAggregateBaseActivity(_ctx, _factory, _probeManager, _remoteId, _numParts, _isLocal, _manager),
- aggregateHelper((IHThorDiskGroupAggregateArg &)basehelper),
- resultAggregator(aggregateHelper, aggregateHelper),
- gathered(false)
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- gathered= false;
- CRoxieServerDiskAggregateBaseActivity::start(parentExtractSize, parentExtract, paused);
- resultAggregator.start(rowAllocator);
- }
- virtual void reset()
- {
- resultAggregator.reset();
- CRoxieServerDiskAggregateBaseActivity::reset();
- }
- void gatherMerged()
- {
- if (useRemote())
- {
- loop
- {
- const void * next = remote->nextRow();
- if (!next)
- break;
- resultAggregator.mergeElement(next);
- ReleaseRoxieRow(next);
- }
- }
- else
- {
- if (helper.canMatchAny() && !eof)
- {
- Owned<IInMemoryFileProcessor> processor = isKeyed ?
- createKeyedGroupAggregateRecordProcessor(cursor, resultAggregator, aggregateHelper) :
- createUnkeyedGroupAggregateRecordProcessor(cursor, resultAggregator, aggregateHelper, manager->createReader(0, 0, 1),
- ctx->queryCodeContext(), activityId);
- processor->doQuery(NULL, 0, 0, 0);
- }
- }
- gathered = true;
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (done)
- return NULL;
- if (!gathered)
- gatherMerged();
- Owned<AggregateRowBuilder> next = resultAggregator.nextResult();
- if (next)
- {
- processed++;
- return next->finalizeRowClear();
- }
- done = true;
- return NULL;
- }
- };
- class CRoxieServerDiskReadActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- RemoteActivityId remoteId;
- bool isLocal;
- bool sorted;
- bool maySkip;
- bool variableFileName;
- Owned<IFileIOArray> files;
- Owned<IInMemoryIndexManager> manager;
- Owned<const IResolvedFile> datafile;
- const char *quotes;
- const char *separators;
- const char *terminators;
- const char *escapes;
- size32_t maxCsvRowSize;
- CRoxieServerDiskReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), remoteId(_remoteId)
- {
- isLocal = _graphNode.getPropBool("att[@name='local']/@value") && queryFactory.queryChannel()!=0;
- Owned<IHThorDiskReadBaseArg> helper = (IHThorDiskReadBaseArg *) helperFactory();
- sorted = (helper->getFlags() & TDRunsorted) == 0;
- variableFileName = allFilesDynamic || _queryFactory.isDynamic() || ((helper->getFlags() & (TDXvarfilename|TDXdynamicfilename)) != 0);
- maySkip = (helper->getFlags() & (TDRkeyedlimitskips|TDRkeyedlimitcreates|TDRlimitskips|TDRlimitcreates)) != 0;
- quotes = separators = terminators = escapes = NULL;
- if (!variableFileName)
- {
- bool isOpt = (helper->getFlags() & TDRoptional) != 0;
- OwnedRoxieString fileName(helper->getFileName());
- datafile.setown(_queryFactory.queryPackage().lookupFileName(fileName, isOpt, true, true, _queryFactory.queryWorkUnit(), true));
- bool isSimple = (datafile && datafile->getNumParts()==1 && !_queryFactory.queryOptions().disableLocalOptimizations);
- if (isLocal || isSimple)
- {
- if (datafile)
- {
- unsigned channel = isLocal ? queryFactory.queryChannel() : 0;
- files.setown(datafile->getIFileIOArray(isOpt, channel));
- manager.setown(datafile->getIndexManager(isOpt, channel, files, helper->queryDiskRecordSize(), _graphNode.getPropBool("att[@name=\"preload\"]/@value", false), _graphNode.getPropInt("att[@name=\"_preloadSize\"]/@value", 0)));
- const IPropertyTree *options = datafile->queryProperties();
- if (options)
- {
- quotes = options->queryProp("@csvQuote");
- separators = options->queryProp("@csvSeparate");
- terminators = options->queryProp("@csvTerminate");
- escapes = options->queryProp("@csvEscape");
- }
- }
- else
- manager.setown(getEmptyIndexManager());
- }
- }
- switch (kind)
- {
- case TAKcsvread:
- {
- maxCsvRowSize = defaultMaxCsvRowSize * 1024 * 1024;
- IConstWorkUnit *workunit = _queryFactory.queryWorkUnit();
- if (workunit)
- maxCsvRowSize = workunit->getDebugValueInt(OPT_MAXCSVROWSIZE, defaultMaxCsvRowSize) * 1024 * 1024;
- break;
- }
- default:
- maxCsvRowSize = 0; // unused
- }
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- unsigned numParts = datafile ? datafile->getNumParts() : 0;
- switch (kind)
- {
- case TAKcsvread:
- return new CRoxieServerCsvReadActivity(_ctx, this, _probeManager, remoteId, numParts, isLocal, sorted, maySkip, manager,
- quotes, separators, terminators, escapes, maxCsvRowSize);
- case TAKxmlread:
- case TAKjsonread:
- return new CRoxieServerXmlReadActivity(_ctx, this, _probeManager, remoteId, numParts, isLocal, sorted, maySkip, manager);
- case TAKdiskread:
- return new CRoxieServerDiskReadActivity(_ctx, this, _probeManager, remoteId, numParts, isLocal, sorted, maySkip, manager);
- case TAKdisknormalize:
- return new CRoxieServerDiskNormalizeActivity(_ctx, this, _probeManager, remoteId, numParts, isLocal, sorted, manager);
- case TAKdiskcount:
- return new CRoxieServerDiskCountActivity(_ctx, this, _probeManager, remoteId, numParts, isLocal, manager);
- case TAKdiskaggregate:
- return new CRoxieServerDiskAggregateActivity(_ctx, this, _probeManager, remoteId, numParts, isLocal, manager);
- case TAKdiskgroupaggregate:
- return new CRoxieServerDiskGroupAggregateActivity(_ctx, this, _probeManager, remoteId, numParts, isLocal, manager);
- }
- throwUnexpected();
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for %s activity", getActivityText(kind));
- }
- virtual void getXrefInfo(IPropertyTree &reply, const IRoxieContextLogger &logctx) const
- {
- if (datafile)
- addXrefFileInfo(reply, datafile);
- else
- {
- // Temporarily resolve the file
- Owned<IHThorDiskReadBaseArg> helper = (IHThorDiskReadBaseArg *) helperFactory();
- if ((helper->getFlags() & (TDXvarfilename|TDXdynamicfilename)) == 0)
- {
- OwnedRoxieString fileName(helper->getFileName());
- Owned<const IResolvedFile> temp = queryFactory.queryPackage().lookupFileName(fileName, true, true, false, queryFactory.queryWorkUnit(), true);
- if (temp)
- addXrefFileInfo(reply, temp);
- }
- }
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return diskStatistics;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerDiskReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- {
- return new CRoxieServerDiskReadActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId);
- }
- //=================================================================================
- class CRoxieServerIndexActivity : public CRoxieServerActivity, implements IRoxieServerErrorHandler
- {
- protected:
- IHThorIndexReadBaseArg &indexHelper;
- IHThorSteppedSourceExtra * steppedExtra;
- Linked<IKeyArray> keySet;
- Linked<TranslatorArray> translators;
- CSkippableRemoteResultAdaptor remote;
- CIndexTransformCallback callback;
- bool sorted;
- bool variableFileName;
- bool variableInfoPending;
- bool isOpt;
- bool isLocal;
- unsigned __int64 rowLimit;
- unsigned __int64 keyedLimit;
- unsigned __int64 choosenLimit;
- unsigned accepted;
- unsigned rejected;
- unsigned seekGEOffset;
- Owned<IKeyManager> tlk;
- Owned<const IResolvedFile> varFileInfo;
- const RemoteActivityId &remoteId;
- void setVariableFileInfo()
- {
- OwnedRoxieString indexName(indexHelper.getFileName());
- varFileInfo.setown(resolveLFN(indexName, isOpt));
- if (varFileInfo)
- {
- translators.setown(new TranslatorArray) ;
- keySet.setown(varFileInfo->getKeyArray(factory->queryActivityMeta(), translators, isOpt, isLocal ? factory->queryQueryFactory().queryChannel() : 0, factory->queryQueryFactory().queryOptions().enableFieldTranslation));
- }
- variableInfoPending = false;
- }
- public:
- CRoxieServerIndexActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId,
- IKeyArray * _keySet, TranslatorArray *_translators, bool _sorted, bool _isLocal, bool _maySkip)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- keySet(_keySet),
- translators(_translators),
- indexHelper((IHThorIndexReadBaseArg &)basehelper),
- remote(_ctx, this, _remoteId, meta.queryOriginal(), indexHelper, *this, _sorted, false, _maySkip),
- remoteId(_remoteId),
- sorted(_sorted),
- isLocal(_isLocal)
- {
- indexHelper.setCallback(&callback);
- steppedExtra = static_cast<IHThorSteppedSourceExtra *>(indexHelper.selectInterface(TAIsteppedsourceextra_1));
- variableFileName = allFilesDynamic || factory->queryQueryFactory().isDynamic() || ((indexHelper.getFlags() & (TIRvarfilename|TIRdynamicfilename)) != 0);
- variableInfoPending = false;
- isOpt = (indexHelper.getFlags() & TIRoptional) != 0;
- seekGEOffset = 0;
- // started = false;
- rejected = accepted = 0;
- rowLimit = choosenLimit = keyedLimit = 0;
- }
- virtual const IResolvedFile *queryVarFileInfo() const
- {
- return varFileInfo;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- remote.onCreate(_colocalParent);
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- accepted = 0;
- rejected = 0;
- rowLimit = (unsigned __int64) -1;
- keyedLimit = (unsigned __int64 ) -1;
- choosenLimit = I64C(0x7fffffffffffffff);
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- remote.onStart(parentExtractSize, parentExtract);
- variableInfoPending = variableFileName;
- }
- void processAllKeys()
- {
- try
- {
- if (indexHelper.canMatchAny())
- {
- if (variableInfoPending)
- setVariableFileInfo();
- remote.setLimits(rowLimit, keyedLimit, choosenLimit);
- if (keySet)
- {
- // MORE - this recreates the segmonitors per part but not per fileno (which is a little backwards).
- // With soft layout support may need to recreate per fileno too (i.e. different keys in a superkey have different layout) but never per partno
- // However order is probably better to iterate fileno's inside partnos
- // MORE - also not properly supporting STEPPED I fear.
- // A superkey that mixes single and multipart or tlk and roroot keys might be hard
- for (unsigned partNo = 0; partNo < keySet->length(); partNo++)
- {
- IKeyIndexBase *thisBase = keySet->queryKeyPart(partNo);
- if (thisBase)
- {
- unsigned fileNo = 0;
- IKeyIndex *thisKey = thisBase->queryPart(fileNo);
- if (!thisKey->isTopLevelKey())
- {
- if (keyedLimit != (unsigned __int64) -1)
- {
- if ((indexHelper.getFlags() & TIRcountkeyedlimit) != 0)
- {
- Owned<IKeyManager> countKey;
- countKey.setown(createKeyManager(thisKey, 0, this));
- countKey->setLayoutTranslator(translators->item(fileNo));
- createSegmentMonitors(countKey);
- unsigned __int64 count = countKey->checkCount(keyedLimit);
- if (count > keyedLimit)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d line = %d", activityId, __LINE__);
- onLimitExceeded(true);
- }
- }
- }
- }
- if (seekGEOffset && !thisKey->isTopLevelKey())
- {
- tlk.setown(createSingleKeyMerger(thisKey, 0, seekGEOffset, this));
- }
- else
- {
- tlk.setown(createKeyManager(thisKey, 0, this));
- tlk->setLayoutTranslator(translators->item(fileNo));
- }
- createSegmentMonitors(tlk);
- if (queryTraceLevel() > 3 || ctx->queryProbeManager())
- {
- StringBuffer out;
- printKeyedValues(out, tlk, indexHelper.queryDiskRecordSize());
- CTXLOG("Using filter %s", out.str());
- if (ctx->queryProbeManager())
- ctx->queryProbeManager()->setNodeProperty(this, "filter", out.str());
- }
- tlk->reset();
- loop // for each file part
- {
- //block for TransformCallbackAssociation
- {
- TransformCallbackAssociation associate(callback, tlk);
- if (thisKey->isTopLevelKey())
- {
- if (thisKey->isFullySorted())
- {
- while (tlk->lookup(false))
- {
- unsigned slavePart = (unsigned) tlk->queryFpos();
- if (slavePart)
- {
- accepted++;
- remote.getMem(slavePart, fileNo, 0); // the cached context is all we need to send
- if (sorted && numChannels>1)
- remote.flush(); // don't combine parts if we need result sorted, except on a 1-way
- }
- }
- }
- else
- {
- // MORE - we could check whether there are any matching parts if we wanted.
- // If people are in the habit of sending null values that would be worthwhile
- remote.getMem(0, fileNo, 0);
- }
- }
- else
- {
- if (processSingleKey(thisKey, translators->item(fileNo)))
- break;
- }
- }
- if (++fileNo < thisBase->numParts())
- {
- thisKey = thisBase->queryPart(fileNo);
- tlk->setKey(thisKey);
- tlk->setLayoutTranslator(translators->item(fileNo));
- tlk->reset();
- }
- else
- break;
- }
- tlk->releaseSegmentMonitors();
- tlk->setKey(NULL);
- }
- }
- }
- }
- remote.flush();
- remote.senddone();
- }
- catch (IException *E)
- {
- remote.setException(E);
- }
- }
- virtual void createSegmentMonitors(IKeyManager *key)
- {
- indexHelper.createSegmentMonitors(key);
- key->finishSegmentMonitors();
- }
- virtual bool processSingleKey(IKeyIndex *key, IRecordLayoutTranslator * trans) = 0;
- virtual void reset()
- {
- if (accepted)
- noteStatistic(StNumIndexAccepted, accepted);
- if (rejected)
- noteStatistic(StNumIndexRejected, rejected);
- remote.onReset();
- CRoxieServerActivity::reset();
- if (varFileInfo)
- {
- keySet.clear();
- varFileInfo.clear();
- }
- variableInfoPending = false;
- }
- virtual void stop()
- {
- remote.onStop();
- CRoxieServerActivity::stop();
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() called for source activity");
- }
- };
- class CRoxieServerIndexReadBaseActivity : public CRoxieServerIndexActivity
- {
- IHThorSourceLimitTransformExtra * limitTransformExtra;
- public:
- CRoxieServerIndexReadBaseActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId,
- IKeyArray * _keySet, TranslatorArray *_translators, bool _sorted, bool _isLocal, bool _maySkip)
- : CRoxieServerIndexActivity(_ctx, _factory, _probeManager, _remoteId, _keySet, _translators, _sorted, _isLocal, _maySkip)
- {
- limitTransformExtra = static_cast<IHThorSourceLimitTransformExtra *>(indexHelper.selectInterface(TAIsourcelimittransformextra_1));
- }
- virtual void reset()
- {
- remote.onReset();
- CRoxieServerIndexActivity::reset();
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- try
- {
- const void *ret = remote.nextRow();
- if (ret)
- processed++;
- return ret;
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- protected:
- virtual const void * createLimitFailRow(bool isKeyed)
- {
- ensureRowAllocator();
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize = isKeyed ? limitTransformExtra->transformOnKeyedLimitExceeded(rowBuilder) : limitTransformExtra->transformOnLimitExceeded(rowBuilder);
- if (outSize)
- return rowBuilder.finalizeRowClear(outSize);
- return NULL;
- }
- };
- class CRoxieServerIndexReadActivity : public CRoxieServerIndexReadBaseActivity, implements IIndexReadActivityInfo
- {
- protected:
- IHThorCompoundReadExtra & readHelper;
- ISteppingMeta *rawMeta = nullptr;
- CSteppingMeta steppingMeta;
- unsigned * seekSizes = nullptr;
- bool optimizeSteppedPostFilter;
- ISteppingMeta * projectedMeta = nullptr;
- unsigned maxSeekLookahead;
- public:
- CRoxieServerIndexReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId,
- IKeyArray * _keySet, TranslatorArray *_translators, bool _sorted, bool _isLocal, bool _maySkip, unsigned _maxSeekLookahead)
- : CRoxieServerIndexReadBaseActivity(_ctx, _factory, _probeManager, _remoteId, _keySet, _translators, _sorted, _isLocal, _maySkip),
- readHelper((IHThorIndexReadArg &)basehelper)
- {
- rawMeta = readHelper.queryRawSteppingMeta();
- unsigned flags = indexHelper.getFlags();
- optimizeSteppedPostFilter = (flags & TIRunfilteredtransform) != 0;
- seekSizes = NULL;
- maxSeekLookahead = _maxSeekLookahead;
- if (rawMeta)
- {
- const CFieldOffsetSize * fields = rawMeta->queryFields();
- unsigned maxFields = rawMeta->getNumFields();
- seekGEOffset = fields[0].offset;
- seekSizes = new unsigned[maxFields];
- seekSizes[0] = fields[0].size;
- for (unsigned i=1; i < maxFields; i++)
- seekSizes[i] = seekSizes[i-1] + fields[i].size;
- projectedMeta = readHelper.queryProjectedSteppingMeta();
- ISteppingMeta *useMeta = projectedMeta ? projectedMeta : rawMeta;
- remote.setMergeInfo(useMeta); // also need to consider superfile case where there is a mix of multiway and singleparts.. ?
- bool hasPostFilter = readHelper.transformMayFilter() && optimizeSteppedPostFilter;
- steppingMeta.init(useMeta, hasPostFilter);
- }
- }
- ~CRoxieServerIndexReadActivity()
- {
- delete [] seekSizes;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerIndexReadBaseActivity::start(parentExtractSize, parentExtract, paused);
- steppingMeta.setDistributed();
- if (steppedExtra)
- steppingMeta.setExtra(steppedExtra);
- rowLimit = readHelper.getRowLimit();
- keyedLimit = readHelper.getKeyedLimit();
- choosenLimit = readHelper.getChooseNLimit();
- if (!paused)
- processAllKeys();
- }
- class LazyLocalKeyReader : implements IMessageUnpackCursor, implements IMessageResult, public CInterface
- {
- public:
- IMPLEMENT_IINTERFACE;
- virtual IMessageUnpackCursor *getCursor(roxiemem::IRowManager *rowMgr) const
- {
- Link();
- return const_cast<LazyLocalKeyReader*> (this);
- }
- virtual const void *getMessageHeader(unsigned &length) const
- {
- length = 0;
- return NULL;
- }
- virtual const void *getMessageMetadata(unsigned &length) const
- {
- length = 0;
- return NULL;
- }
- virtual void discard() const
- {
- // nothing to do.
- }
- unsigned keyedCount;
- unsigned matched;
- bool EOFseen;
- Owned<IKeyIndexSet> keySet;
- Owned<IKeyManager> tlk;
- CRoxieServerIndexReadActivity &owner;
- LazyLocalKeyReader(CRoxieServerIndexReadActivity &_owner, IKeyIndex *key, IRecordLayoutTranslator * trans)
- : owner(_owner)
- {
- keyedCount = 0;
- matched = 0;
- EOFseen = false;
- keySet.setown(createKeyIndexSet());
- keySet->addIndex(LINK(key));
- if (owner.seekGEOffset)
- tlk.setown(createKeyMerger(keySet, 0, owner.seekGEOffset, &owner));
- else
- tlk.setown(createKeyManager(keySet->queryPart(0), 0, &owner));
- tlk->setLayoutTranslator(trans);
- owner.indexHelper.createSegmentMonitors(tlk);
- tlk->finishSegmentMonitors();
- tlk->reset();
- }
- virtual const void *getNext(int length)
- {
- TransformCallbackAssociation associate(owner.callback, tlk);
- while (tlk->lookup(true))
- {
- keyedCount++;
- if (keyedCount > owner.keyedLimit)
- {
- owner.onLimitExceeded(true); // Should throw exception
- throwUnexpected();
- }
- size32_t transformedSize;
- RtlDynamicRowBuilder rowBuilder(owner.rowAllocator);
- byte const * keyRow = tlk->queryKeyBuffer(owner.callback.getFPosRef());
- try
- {
- transformedSize = owner.readHelper.transform(rowBuilder, keyRow);
- owner.callback.finishedRow();
- }
- catch (IException *E)
- {
- throw owner.makeWrappedException(E);
- }
- if (transformedSize)
- {
- OwnedConstRoxieRow result = rowBuilder.finalizeRowClear(transformedSize);
- matched++;
- if (matched > owner.rowLimit)
- {
- owner.onLimitExceeded(false); // Should throw exception
- throwUnexpected();
- }
- if (matched > owner.choosenLimit) // MORE - bit of a strange place to check
- {
- break;
- }
- owner.accepted++;
- return result.getClear();
- }
- else
- owner.rejected++;
- }
- EOFseen = true;
- return NULL;
- }
- virtual bool atEOF() const
- {
- return EOFseen;
- }
- virtual bool isSerialized() const
- {
- return false;
- }
- };
- virtual bool processSingleKey(IKeyIndex *key, IRecordLayoutTranslator * trans)
- {
- ensureRowAllocator();
- remote.injectResult(new LazyLocalKeyReader(*this, key, trans));
- return false;
- }
- virtual void onLimitExceeded(bool isKeyed)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- if (isKeyed)
- {
- if (indexHelper.getFlags() & (TIRkeyedlimitskips|TIRkeyedlimitcreates))
- {
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- throw makeLimitSkipException(true);
- }
- else
- readHelper.onKeyedLimitExceeded();
- }
- else
- {
- if (indexHelper.getFlags() & (TIRlimitskips|TIRlimitcreates))
- {
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- throw makeLimitSkipException(false);
- }
- else
- readHelper.onLimitExceeded();
- }
- }
- virtual void serializeSkipInfo(MemoryBuffer &out, unsigned seekLen, const void *rawSeek, unsigned numFields, const void * seek, const SmartStepExtra &stepExtra) const
- {
- out.append((unsigned short) numFields);
- out.append((unsigned short) seekLen);
- out.append((unsigned short) stepExtra.queryFlags());
- IMultipleStepSeekInfo *seeks = stepExtra.queryExtraSeeks();
- if (seeks)
- {
- unsigned lookahead = 40000/seekLen;
- if (maxSeekLookahead && (lookahead > maxSeekLookahead))
- lookahead = maxSeekLookahead;
- seeks->ensureFilled(seek, numFields, lookahead);
- out.append(lookahead != seeks->ordinality()); // seeksAreEof flag
- unsigned serialized = 1; // rawseek is always serialized...
- unsigned patchLength = out.length();
- out.append(serialized); // NOTE - we come back and patch with the actual value...
- out.append(seekLen, rawSeek);
- if (seeks->ordinality())
- {
- const void *lastSeek = rawSeek;
- byte *nextSeek = NULL;
- if (projectedMeta)
- nextSeek = (byte *) alloca(seekLen);
- for (unsigned i = 0; i < seeks->ordinality(); i++)
- {
- if (projectedMeta)
- {
- RtlStaticRowBuilder rowBuilder(nextSeek-seekGEOffset, seekGEOffset+seekLen);
- readHelper.mapOutputToInput(rowBuilder, seeks->querySeek(i), numFields); // NOTE - weird interface to mapOutputToInput means that it STARTS writing at seekGEOffset...
- }
- else
- nextSeek = (byte *) seeks->querySeek(i)+seekGEOffset;
- int diff = memcmp(nextSeek, lastSeek, seekLen);
- if (diff > 0)
- {
- serialized++;
- out.append(seekLen, nextSeek);
- lastSeek = (const byte *) out.reserve(0) - seekLen;
- }
- }
- unsigned length = out.length();
- out.setWritePos(patchLength);
- out.append(serialized);
- out.setWritePos(length);
- }
- }
- else
- {
- out.append(false);
- out.append(1);
- out.append(seekLen, rawSeek);
- }
- }
- virtual const void * nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- try
- {
- unsigned seeklen = 0;
- const void *rawSeek = NULL;
- if (seek && numFields)
- {
- seeklen = seekSizes[numFields-1];
- rawSeek = (const byte *)seek + seekGEOffset;
- if (projectedMeta)
- {
- byte * temp = (byte *) alloca(seeklen);
- RtlStaticRowBuilder rawBuilder(temp-seekGEOffset, seekGEOffset+seeklen);
- readHelper.mapOutputToInput(rawBuilder, seek, numFields); // NOTE - weird interface to mapOutputToInput means that it STARTS writing at seekGEOffset...
- rawSeek = (byte *)temp;
- }
- }
- const void *ret = remote.nextRowGE(seek, rawSeek, numFields, seeklen, wasCompleteMatch, stepExtra);
- if (ret && wasCompleteMatch) // GH pleas confirm the wasCompleteMatch I just added here is right
- processed++;
- return ret;
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- virtual IInputSteppingMeta * querySteppingMeta()
- {
- if (rawMeta && steppingEnabled && ((indexHelper.getFlags() & (TIRlimitskips|TIRlimitcreates|TIRkeyedlimitskips|TIRkeyedlimitcreates)) == 0))
- return &steppingMeta;
- return NULL;
- }
- virtual IIndexReadActivityInfo *queryIndexReadActivity()
- {
- if (variableInfoPending)
- setVariableFileInfo();
- return this;
- }
- virtual IKeyArray *getKeySet() const
- {
- return keySet.getLink();
- }
- virtual const IResolvedFile *getVarFileInfo() const
- {
- return varFileInfo.getLink();
- }
- virtual TranslatorArray *getTranslators() const
- {
- return translators.getLink();
- }
- virtual void mergeSegmentMonitors(IIndexReadContext *irc) const
- {
- indexHelper.createSegmentMonitors(irc); // NOTE: they will merge
- }
- virtual IRoxieServerActivity *queryActivity() { return this; }
- virtual const RemoteActivityId& queryRemoteId() const
- {
- return remoteId;
- }
- };
- class CRoxieServerSimpleIndexReadActivity : public CRoxieServerActivity, implements IIndexReadActivityInfo
- {
- IHThorCompoundReadExtra & readHelper;
- IHThorIndexReadBaseArg & indexHelper;
- IHThorSourceLimitTransformExtra * limitTransformExtra;
- IHThorSteppedSourceExtra * steppedExtra;
- bool eof;
- Linked<IKeyArray>keySet;
- Owned<IKeyIndexSet>keyIndexSet;
- Owned<IKeyManager> tlk;
- Linked<TranslatorArray> translators;
- CIndexTransformCallback callback;
- unsigned __int64 keyedLimit;
- unsigned rowLimit;
- unsigned chooseNLimit;
- unsigned accepted;
- unsigned rejected;
- unsigned keyedCount;
- ISteppingMeta * rawMeta;
- ISteppingMeta * projectedMeta;
- size32_t seekGEOffset;
- unsigned * seekSizes;
- CSteppingMeta steppingMeta;
- Owned<const IResolvedFile> varFileInfo;
- const RemoteActivityId &remoteId;
- bool firstRead;
- bool variableFileName;
- bool variableInfoPending;
- bool isOpt;
- bool isLocal;
- bool optimizeSteppedPostFilter;
- // MORE there may be enough in common between this and CRoxieServerIndexActivity to warrant some refactoring
- void initKeySet()
- {
- if ((keySet->length() > 1 || rawMeta != NULL) && translators->needsTranslation())
- {
- throw MakeStringException(ROXIE_UNIMPLEMENTED_ERROR, "Layout translation is not available when merging key parts or smart-stepping, as it may change record order");
- }
- keyIndexSet.setown(createKeyIndexSet());
- for (unsigned part = 0; part < keySet->length(); part++)
- {
- IKeyIndexBase *kib = keySet->queryKeyPart(part);
- if (kib)
- {
- for (unsigned subpart = 0; subpart < kib->numParts(); subpart++)
- {
- IKeyIndex *k = kib->queryPart(subpart);
- if (k)
- {
- assertex(!k->isTopLevelKey());
- keyIndexSet->addIndex(LINK(k));
- }
- }
- }
- }
- }
- void setVariableFileInfo()
- {
- OwnedRoxieString indexName(indexHelper.getFileName());
- varFileInfo.setown(resolveLFN(indexName, isOpt));
- translators.setown(new TranslatorArray) ;
- keySet.setown(varFileInfo->getKeyArray(factory->queryActivityMeta(), translators, isOpt, isLocal ? factory->queryQueryFactory().queryChannel() : 0, factory->queryQueryFactory().queryOptions().enableFieldTranslation));
- initKeySet();
- variableInfoPending = false;
- }
- void onEOF()
- {
- callback.setManager(NULL);
- eof = true;
- tlk.clear();
- }
- public:
- CRoxieServerSimpleIndexReadActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId,
- IKeyArray *_keyArray, TranslatorArray *_translatorArray, bool _isLocal)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- readHelper((IHThorIndexReadArg &)basehelper),
- indexHelper((IHThorIndexReadArg &)basehelper),
- translators(_translatorArray),
- keySet(_keyArray),
- isLocal(_isLocal),
- remoteId(_remoteId)
- {
- rowLimit = 0;
- keyedLimit = 0;
- chooseNLimit = 0;
- indexHelper.setCallback(&callback);
- steppedExtra = static_cast<IHThorSteppedSourceExtra *>(indexHelper.selectInterface(TAIsteppedsourceextra_1));
- limitTransformExtra = static_cast<IHThorSourceLimitTransformExtra *>(indexHelper.selectInterface(TAIsourcelimittransformextra_1));
- unsigned flags = indexHelper.getFlags();
- variableFileName = allFilesDynamic || factory->queryQueryFactory().isDynamic() || ((flags & (TIRvarfilename|TIRdynamicfilename)) != 0);
- variableInfoPending = false;
- isOpt = (flags & TIRoptional) != 0;
- optimizeSteppedPostFilter = (flags & TIRunfilteredtransform) != 0;
- firstRead = true;
- accepted = 0;
- rejected = 0;
- keyedCount = 0;
- eof = false;
- rawMeta = readHelper.queryRawSteppingMeta();
- projectedMeta = readHelper.queryProjectedSteppingMeta();
- seekGEOffset = 0;
- seekSizes = NULL;
- if (rawMeta)
- {
- // MORE - should check all keys in maxFields list can actually be keyed.
- const CFieldOffsetSize * fields = rawMeta->queryFields();
- unsigned maxFields = rawMeta->getNumFields();
- seekGEOffset = fields[0].offset;
- seekSizes = new unsigned[maxFields];
- seekSizes[0] = fields[0].size;
- for (unsigned i=1; i < maxFields; i++)
- seekSizes[i] = seekSizes[i-1] + fields[i].size;
- bool hasPostFilter = readHelper.transformMayFilter() && optimizeSteppedPostFilter;
- if (projectedMeta)
- steppingMeta.init(projectedMeta, hasPostFilter);
- else
- steppingMeta.init(rawMeta, hasPostFilter);
- }
- }
- virtual const IResolvedFile *queryVarFileInfo() const
- {
- return varFileInfo;
- }
- ~CRoxieServerSimpleIndexReadActivity()
- {
- delete [] seekSizes;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- firstRead = true;
- accepted = 0;
- rejected = 0;
- keyedCount = 0;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if (steppedExtra)
- steppingMeta.setExtra(steppedExtra);
- eof = !indexHelper.canMatchAny();
- if (variableFileName)
- variableInfoPending = true;
- else
- {
- variableInfoPending = false;
- if (!keyIndexSet)
- initKeySet();
- }
- }
- virtual IIndexReadActivityInfo *queryIndexReadActivity()
- {
- if (variableInfoPending)
- setVariableFileInfo();
- return this;
- }
- virtual IKeyArray *getKeySet() const
- {
- return keySet.getLink();
- }
- virtual const IResolvedFile *getVarFileInfo() const
- {
- return varFileInfo.getLink();
- }
- virtual TranslatorArray *getTranslators() const
- {
- return translators.getLink();
- }
- virtual void mergeSegmentMonitors(IIndexReadContext *irc) const
- {
- indexHelper.createSegmentMonitors(irc); // NOTE: they will merge
- }
- virtual IRoxieServerActivity *queryActivity() { return this; }
- virtual const RemoteActivityId& queryRemoteId() const
- {
- return remoteId;
- }
-
- const void *nextRow()
- {
- bool matched = true;
- return nextRowGE(NULL, 0, matched, dummySmartStepExtra);
- }
- unsigned __int64 checkCount(unsigned __int64 limit)
- {
- unsigned numParts = keyIndexSet->numParts();
- unsigned __int64 result = 0;
- for (unsigned i = 0; i < numParts; i++)
- {
- Owned<IKeyManager> countTlk = createKeyManager(keyIndexSet->queryPart(i), 0, this);
- countTlk->setLayoutTranslator(translators->item(i));
- indexHelper.createSegmentMonitors(countTlk);
- countTlk->finishSegmentMonitors();
- result += countTlk->checkCount(limit-result);
- if (result > limit)
- break;
- }
- return result;
- }
- virtual const void *nextRowGE(const void * seek, unsigned numFields, bool &wasCompleteMatch, const SmartStepExtra & stepExtra)
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- if (firstRead)
- {
- if (variableInfoPending)
- setVariableFileInfo();
- rowLimit = (unsigned) readHelper.getRowLimit();
- chooseNLimit = (unsigned) readHelper.getChooseNLimit();
- unsigned numParts = keyIndexSet->numParts();
- if (!numParts)
- {
- onEOF();
- return NULL;
- }
- if (numParts > 1 || seekGEOffset)
- {
- tlk.setown(createKeyMerger(keyIndexSet, 0, seekGEOffset, this));
- // note that we don't set up translator because we don't support it. If that ever changes...
- }
- else
- {
- tlk.setown(createKeyManager(keyIndexSet->queryPart(0), 0, this));
- tlk->setLayoutTranslator(translators->item(0));
- }
- indexHelper.createSegmentMonitors(tlk);
- tlk->finishSegmentMonitors();
- if (queryTraceLevel() > 3 || ctx->queryProbeManager())
- {
- StringBuffer out;
- printKeyedValues(out, tlk, indexHelper.queryDiskRecordSize());
- CTXLOG("Using filter %s", out.str());
- if (ctx->queryProbeManager())
- ctx->queryProbeManager()->setNodeProperty(this, "filter", out.str());
- }
- tlk->reset();
- callback.setManager(tlk);
- keyedLimit = readHelper.getKeyedLimit();
- if (keyedLimit != (unsigned __int64) -1)
- {
- if ((indexHelper.getFlags() & TIRcountkeyedlimit) != 0)
- {
- unsigned __int64 count = checkCount(keyedLimit);
- if (count > keyedLimit)
- {
- if ((indexHelper.getFlags() & (TIRkeyedlimitskips|TIRkeyedlimitcreates)) == 0)
- readHelper.onKeyedLimitExceeded();
- const void * ret = NULL;
- if (indexHelper.getFlags() & TIRkeyedlimitcreates)
- ret = createKeyedLimitOnFailRow();
- onEOF();
- return ret;
- }
- keyedLimit = (unsigned __int64) -1;
- }
- }
- firstRead = false;
- }
- if (accepted == chooseNLimit)
- {
- onEOF();
- return NULL;
- }
-
- const byte * rawSeek = NULL;
- unsigned seekSize = 0;
- if (seek)
- {
- seekSize = seekSizes[numFields-1];
- rawSeek = (const byte *)seek + seekGEOffset;
- if (projectedMeta)
- {
- byte *temp = (byte *) alloca(seekSize);
- RtlStaticRowBuilder rawBuilder(temp-seekGEOffset, seekGEOffset+seekSize);
- readHelper.mapOutputToInput(rawBuilder, seek, numFields);// NOTE - weird interface to mapOutputToInput means that it STARTS writing at seekGEOffset...
- rawSeek = (byte *)temp;
- }
- #ifdef _DEBUG
- // StringBuffer seekStr;
- // for (unsigned i = 0; i < seekSize; i++)
- // {
- // seekStr.appendf("%02x ", ((unsigned char *) rawSeek)[i]);
- // }
- // DBGLOG("nextRowGE can skip offset %d size %d value %s", seekGEOffset, seekSize, seekStr.str());
- #endif
- }
- const byte * originalRawSeek = rawSeek;
- RtlDynamicRowBuilder rowBuilder(rowAllocator, false);
- while (rawSeek ? tlk->lookupSkip(rawSeek, seekGEOffset, seekSize) : tlk->lookup(true))
- {
- checkAbort();
- keyedCount++;
- if (keyedCount > keyedLimit)
- {
- readHelper.onKeyedLimitExceeded();
- break;
- }
- byte const * keyRow = tlk->queryKeyBuffer(callback.getFPosRef());
- #ifdef _DEBUG
- // StringBuffer recstr;
- // unsigned size = (tlk->queryRecordSize()<80) ? tlk->queryRecordSize() : 80;
- // for (unsigned i = 0; i < size; i++)
- // {
- // recstr.appendf("%02x ", ((unsigned char *) keyRow)[i]);
- // }
- // DBGLOG("nextRowGE Got %s", recstr.str());
- if (originalRawSeek && memcmp(keyRow + seekGEOffset, originalRawSeek, seekSize) < 0)
- assertex(!"smart seek failure");
- #endif
- size32_t transformedSize;
- rowBuilder.ensureRow();
- try
- {
- transformedSize = readHelper.transform(rowBuilder, keyRow);
- //if the post filter causes a mismatch, and the stepping condition no longer matches
- //then return a mismatch record - so the join code can start seeking on the other input.
- if (transformedSize == 0 && optimizeSteppedPostFilter && stepExtra.returnMismatches())
- {
- if (memcmp(keyRow + seekGEOffset, originalRawSeek, seekSize) != 0)
- {
- transformedSize = readHelper.unfilteredTransform(rowBuilder, keyRow);
- if (transformedSize != 0)
- wasCompleteMatch = false;
- }
- }
- callback.finishedRow();
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- if (transformedSize)
- {
- accepted++;
- if (accepted > rowLimit)
- {
- if ((indexHelper.getFlags() & (TIRlimitskips|TIRlimitcreates)) != 0)
- {
- throwUnexpected(); // should not have used simple variant if maySkip set...
- }
- if (traceLevel > 4)
- DBGLOG("activityid = %d line = %d", activityId, __LINE__);
- readHelper.onLimitExceeded();
- break;
- }
- processed++;
- #ifdef _DEBUG
- // const byte *ret = (const byte *) out.get();
- // CommonXmlWriter xmlwrite(XWFnoindent|XWFtrim|XWFopt);
- // queryOutputMeta()->toXML(ret, xmlwrite);
- // DBGLOG("ROW: {%p} %s", ret, xmlwrite.str());
- #endif
- return rowBuilder.finalizeRowClear(transformedSize);
- }
- else
- rejected++;
- rawSeek = NULL;
- }
- onEOF();
- return NULL;
- }
- virtual void reset()
- {
- onEOF();
- if (accepted)
- noteStatistic(StNumIndexAccepted, accepted);
- if (rejected)
- noteStatistic(StNumIndexRejected, rejected);
- if (variableFileName)
- {
- varFileInfo.clear();
- translators.clear();
- }
- variableInfoPending = false;
- CRoxieServerActivity::reset();
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() called for source activity");
- }
- virtual IInputSteppingMeta * querySteppingMeta()
- {
- if (rawMeta && steppingEnabled && ((indexHelper.getFlags() & (TIRlimitskips|TIRlimitcreates|TIRkeyedlimitskips|TIRkeyedlimitcreates)) == 0))
- return &steppingMeta;
- return NULL;
- }
- protected:
- const void * createKeyedLimitOnFailRow()
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t outSize = limitTransformExtra->transformOnKeyedLimitExceeded(rowBuilder);
- if (outSize)
- return rowBuilder.finalizeRowClear(outSize);
- return NULL;
- }
- };
- class CRoxieServerBaseIndexActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- Owned<IKeyArray> keySet;
- Owned<TranslatorArray> translatorArray;
- Owned<IDefRecordMeta> activityMeta;
- RemoteActivityId remoteId;
- bool isSimple;
- bool isLocal;
- bool maySkip;
- bool sorted;
- bool variableFileName;
- bool enableFieldTranslation;
- unsigned maxSeekLookahead;
- Owned<const IResolvedFile> indexfile;
- CRoxieServerSideCache *cache;
- CRoxieServerBaseIndexActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), remoteId(_remoteId)
- {
- Owned<IHThorIndexReadBaseArg> indexHelper = (IHThorIndexReadBaseArg *) helperFactory();
- unsigned flags = indexHelper->getFlags();
- sorted = (flags & TIRunordered) == 0;
- isLocal = _graphNode.getPropBool("att[@name='local']/@value") && queryFactory.queryChannel()!=0;
- rtlDataAttr indexLayoutMeta;
- size32_t indexLayoutSize;
- if(!indexHelper->getIndexLayout(indexLayoutSize, indexLayoutMeta.refdata()))
- assertex(indexLayoutSize==0);
- MemoryBuffer m;
- m.setBuffer(indexLayoutSize, indexLayoutMeta.getdata());
- activityMeta.setown(deserializeRecordMeta(m, true));
- enableFieldTranslation = queryFactory.queryOptions().enableFieldTranslation;
- translatorArray.setown(new TranslatorArray);
- variableFileName = allFilesDynamic || _queryFactory.isDynamic() || ((flags & (TIRvarfilename|TIRdynamicfilename)) != 0);
- if (!variableFileName)
- {
- bool isOpt = (flags & TIRoptional) != 0;
- OwnedRoxieString indexName(indexHelper->getFileName());
- indexfile.setown(queryFactory.queryPackage().lookupFileName(indexName, isOpt, true, true, queryFactory.queryWorkUnit(), true));
- if (indexfile)
- keySet.setown(indexfile->getKeyArray(activityMeta, translatorArray, isOpt, isLocal ? queryFactory.queryChannel() : 0, enableFieldTranslation));
- }
- isSimple = isLocal;
- maySkip = (flags & (TIRkeyedlimitskips|TIRlimitskips|TIRlimitcreates|TIRkeyedlimitcreates)) != 0;
- if (keySet && keySet->length()==1 && !isLocal && (flags & (TIRlimitskips|TIRlimitcreates|TIRkeyedlimitskips|TIRkeyedlimitcreates))==0)
- {
- IKeyIndexBase *thisBase = keySet->queryKeyPart(0);
- if (thisBase->numParts()==1 && !thisBase->queryPart(0)->isTopLevelKey() && !_queryFactory.queryOptions().disableLocalOptimizations)
- isSimple = true;
- }
- int cacheSize = _graphNode.getPropInt("hint[@name='cachehits']/@value", serverSideCacheSize);
- cache = cacheSize ? new CRoxieServerSideCache(cacheSize) : NULL;
- maxSeekLookahead = _graphNode.getPropInt("hint[@name='maxseeklookahead']/@value", 0);
- }
- ~CRoxieServerBaseIndexActivityFactory()
- {
- delete cache;
- }
- virtual void getXrefInfo(IPropertyTree &reply, const IRoxieContextLogger &logctx) const
- {
- if (indexfile)
- addXrefFileInfo(reply, indexfile);
- else
- {
- Owned<IHThorIndexReadBaseArg> indexHelper = (IHThorIndexReadBaseArg *) helperFactory();
- if ((indexHelper->getFlags() & (TIRvarfilename|TIRdynamicfilename)) == 0)
- {
- OwnedRoxieString indexName(indexHelper->getFileName());
- Owned<const IResolvedFile> temp = queryFactory.queryPackage().lookupFileName(indexName, true, true, false, queryFactory.queryWorkUnit(), true);
- if (temp)
- addXrefFileInfo(reply, temp);
- }
- }
- }
- virtual void setInput(unsigned idx, unsigned source, unsigned sourceidx)
- {
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() should not be called for indexread activity");
- }
- virtual IRoxieServerSideCache *queryServerSideCache() const
- {
- return cache;
- }
- virtual bool getEnableFieldTranslation() const
- {
- return enableFieldTranslation;
- }
- virtual IDefRecordMeta *queryActivityMeta() const
- {
- return activityMeta;
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return indexStatistics;
- }
- };
- class CRoxieServerIndexReadActivityFactory : public CRoxieServerBaseIndexActivityFactory
- {
- public:
- CRoxieServerIndexReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- : CRoxieServerBaseIndexActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- if (!variableFileName && (keySet==NULL || keySet->length()==0))
- return new CRoxieServerNullActivity(_ctx, this, _probeManager);
- else if (isSimple && !maySkip)
- return new CRoxieServerSimpleIndexReadActivity(_ctx, this, _probeManager, remoteId, keySet, translatorArray, isLocal);
- else
- return new CRoxieServerIndexReadActivity(_ctx, this, _probeManager, remoteId, keySet, translatorArray, sorted, isLocal, maySkip, maxSeekLookahead);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerIndexReadActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- {
- return new CRoxieServerIndexReadActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId);
- }
- //--------------------------------------------------------------------------------------------------------------------------
- class CRoxieServerNullCountActivity : public CRoxieServerActivity
- {
- bool done;
- public:
- CRoxieServerNullCountActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager)
- {
- done = false;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- done = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (done) return NULL;
- done = true;
- size32_t rowSize = meta.getFixedSize();
- void * nullRow = rowAllocator->createRow();
- if (rowSize == 1)
- *(byte *)nullRow = 0;
- else
- {
- assertex(rowSize == sizeof(unsigned __int64));
- *(unsigned __int64 *)nullRow = 0;
- }
- return rowAllocator->finalizeRow(rowSize, nullRow, rowSize);
- }
- };
- class CRoxieServerIndexCountActivity : public CRoxieServerIndexActivity
- {
- IHThorCompoundCountExtra & countHelper;
- IHThorSourceCountLimit * limitHelper;
- bool done;
- public:
- CRoxieServerIndexCountActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, IKeyArray * _keySet, TranslatorArray *_translators, bool _isLocal)
- : CRoxieServerIndexActivity(_ctx, _factory, _probeManager, _remoteId, _keySet, _translators, false, _isLocal, false),
- countHelper((IHThorIndexCountArg &)basehelper),
- done(false)
- {
- limitHelper = static_cast<IHThorSourceCountLimit *>(basehelper.selectInterface(TAIsourcecountlimit_1));
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- done = false;
- CRoxieServerIndexActivity::start(parentExtractSize, parentExtract, paused);
- choosenLimit = countHelper.getChooseNLimit();
- if (limitHelper)
- {
- rowLimit = limitHelper->getRowLimit();
- keyedLimit = limitHelper->getKeyedLimit();
- }
- if (!paused)
- processAllKeys();
- }
- virtual bool processSingleKey(IKeyIndex *key, IRecordLayoutTranslator * trans)
- {
- unsigned __int64 count = 0;
- if (countHelper.hasFilter())
- {
- while (tlk->lookup(true))
- {
- try
- {
- count += countHelper.numValid(tlk->queryKeyBuffer(callback.getFPosRef()));
- callback.finishedRow();
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- accepted++;
- if (count >= choosenLimit) // MORE - what about limit?
- break;
- }
- }
- else
- count = tlk->getCount(); //MORE: GH->RKC There should be value in providing a max limit to getCount()
- if (count)
- {
- Owned<CRowArrayMessageResult> result = new CRowArrayMessageResult(ctx->queryRowManager(), false);
- if (count > choosenLimit)
- count = choosenLimit;
- void * recBuffer = rowAllocator->createRow();
- if (meta.getFixedSize() == 1)
- *(byte *)recBuffer = (byte)count;
- else
- {
- assertex(meta.getFixedSize() == sizeof(unsigned __int64));
- *(unsigned __int64 *)recBuffer = count;
- }
- recBuffer = rowAllocator->finalizeRow(meta.getFixedSize(), recBuffer, meta.getFixedSize());
- result->append(recBuffer);
- remote.injectResult(result.getClear());
- //GH->RKC for count(,choosen)/exists passing in the previous count would short-circuit this much earlier
- if (count >= choosenLimit)
- return true;
- }
- return false;
- }
- virtual void onLimitExceeded(bool isKeyed)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- if (isKeyed)
- {
- if (indexHelper.getFlags() & (TIRkeyedlimitskips|TIRkeyedlimitcreates))
- {
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- throw makeLimitSkipException(true);
- }
- else
- {
- assertex(limitHelper); // Should not be able to generate exception if there was not one...
- limitHelper->onKeyedLimitExceeded();
- }
- }
- else
- {
- if (indexHelper.getFlags() & (TIRlimitskips|TIRlimitcreates))
- {
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- throw makeLimitSkipException(false);
- }
- else
- {
- assertex(limitHelper);
- limitHelper->onLimitExceeded();
- }
- }
- }
- virtual const void *createLimitFailRow(bool isKeyed)
- {
- throwUnexpected();
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (done) return NULL;
- done = true;
- unsigned __int64 totalCount = 0;
- bool hasLimit = rowLimit != (unsigned __int64) -1;
- try
- {
- loop
- {
- const void * next = remote.nextRow();
- if (!next)
- break;
- if (meta.getFixedSize() == 1)
- totalCount += *(byte *)next;
- else
- totalCount += *(unsigned __int64 *) next;
- ReleaseRoxieRow(next);
- if (totalCount > rowLimit || (totalCount > choosenLimit && !hasLimit)) // can't break out early if there is a possibility of later slave throwing limit exception
- break;
- }
- if (totalCount > rowLimit)
- {
- unsigned flags = indexHelper.getFlags();
- if (flags & TIRlimitskips)
- totalCount = 0;
- else if (flags & TIRlimitcreates)
- totalCount = 1;
- else
- {
- assertex(limitHelper);
- limitHelper->onLimitExceeded();
- }
- }
- else if (totalCount > choosenLimit)
- totalCount = choosenLimit;
- }
- catch (IException *E)
- {
- if (QUERYINTERFACE(E, LimitSkipException))
- {
- totalCount = 0;
- unsigned flags = indexHelper.getFlags();
- if (E->errorCode() == KeyedLimitSkipErrorCode)
- {
- if (flags & TIRkeyedlimitcreates)
- totalCount++;
- }
- else
- {
- if (flags & TIRlimitcreates)
- totalCount++;
- }
- if (totalCount > choosenLimit)
- totalCount = choosenLimit; // would have to be weird code (and escape the optimizer...)
- E->Release();
- }
- else
- throw ;
- }
- void * result = rowAllocator->createRow();
- if (meta.getFixedSize() == 1)
- *(byte *)result = (byte)totalCount;
- else
- {
- assertex(meta.getFixedSize() == sizeof(unsigned __int64));
- *(unsigned __int64 *)result = totalCount;
- }
- return rowAllocator->finalizeRow(meta.getFixedSize(), result, meta.getFixedSize());
- }
- };
- class CRoxieServerIndexCountActivityFactory : public CRoxieServerBaseIndexActivityFactory
- {
- public:
- CRoxieServerIndexCountActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- : CRoxieServerBaseIndexActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- if (!variableFileName && (keySet==NULL || keySet->length()==0))
- return new CRoxieServerNullCountActivity(_ctx, this, _probeManager);
- // else if (isSimple)
- // return new CRoxieServerSimpleIndexCountActivity(this, keySet->queryKeyPart(0)->queryPart(0));
- else
- return new CRoxieServerIndexCountActivity(_ctx, this, _probeManager, remoteId, keySet, translatorArray, isLocal);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerIndexCountActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- {
- return new CRoxieServerIndexCountActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId);
- }
- //--------------------------------------------------------------------------------------------------------------------------
- class CRoxieServerNullIndexAggregateActivity : public CRoxieServerActivity
- {
- IHThorIndexAggregateArg &aggregateHelper;
- bool done;
- public:
- CRoxieServerNullIndexAggregateActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- aggregateHelper((IHThorIndexAggregateArg &)basehelper)
- {
- done = false;
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- done = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (done) return NULL;
- done = true;
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t thisSize = aggregateHelper.clearAggregate(rowBuilder);
- return rowBuilder.finalizeRowClear(thisSize);
- }
- };
- class CRoxieServerIndexAggregateActivity : public CRoxieServerIndexActivity
- {
- IHThorCompoundAggregateExtra & aggregateHelper;
- bool done;
- public:
- CRoxieServerIndexAggregateActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId,
- IKeyArray * _keySet, TranslatorArray *_translators, bool _isLocal)
- : CRoxieServerIndexActivity(_ctx, _factory, _probeManager, _remoteId, _keySet, _translators, false, _isLocal, false),
- aggregateHelper((IHThorIndexAggregateArg &)basehelper),
- done(false)
- {
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- done = false;
- CRoxieServerIndexActivity::start(parentExtractSize, parentExtract, paused);
- if (!paused)
- processAllKeys();
- }
- virtual bool needsAllocator() const { return true; }
- virtual bool processSingleKey(IKeyIndex *key, IRecordLayoutTranslator * trans)
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator, false);
- while (tlk->lookup(true))
- {
- if (!rowBuilder.exists())
- {
- rowBuilder.ensureRow();
- aggregateHelper.clearAggregate(rowBuilder);
- }
- try
- {
- aggregateHelper.processRow(rowBuilder, tlk->queryKeyBuffer(callback.getFPosRef()));
- callback.finishedRow();
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- accepted++;
- }
- if (aggregateHelper.processedAnyRows())
- {
- size32_t size = meta.getRecordSize(rowBuilder.getSelf());
- const void * recBuffer = rowBuilder.finalizeRowClear(size);
- Owned<CRowArrayMessageResult> result = new CRowArrayMessageResult(ctx->queryRowManager(), meta.isVariableSize());
- result->append(recBuffer);
- remote.injectResult(result.getClear());
- }
- return false;
- }
- virtual void onLimitExceeded(bool isKeyed)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- throwUnexpected();
- }
- virtual const void *createLimitFailRow(bool isKeyed)
- {
- throwUnexpected();
- }
- const void * gatherMerged()
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator, false);
- const void * firstRow = remote.nextRow();
- size32_t finalSize = 0;
- if (!firstRow)
- {
- rowBuilder.ensureRow();
- finalSize = aggregateHelper.clearAggregate(rowBuilder);
- }
- else
- {
- // NOTE need to clone this because going to modify below, could special case 1 row only
- finalSize = cloneRow(rowBuilder, firstRow, meta);
- ReleaseRoxieRow(firstRow);
- }
- loop
- {
- const void * next = remote.nextRow();
- if (!next)
- break;
- finalSize = aggregateHelper.mergeAggregate(rowBuilder, next);
- ReleaseRoxieRow(next);
- }
- return rowBuilder.finalizeRowClear(finalSize);
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (done) return NULL;
- const void * ret = gatherMerged();
- done = true;
- return ret;
- }
- };
- class CRoxieServerIndexAggregateActivityFactory : public CRoxieServerBaseIndexActivityFactory
- {
- public:
- CRoxieServerIndexAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- : CRoxieServerBaseIndexActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- if (!variableFileName && (keySet==NULL || keySet->length()==0))
- return new CRoxieServerNullIndexAggregateActivity(_ctx, this, _probeManager);
- // else if (isSimple)
- // return new CRoxieServerSimpleIndexAggregateActivity(this, keySet->queryKeyPart(0)->queryPart(0));
- else
- return new CRoxieServerIndexAggregateActivity(_ctx, this, _probeManager, remoteId, keySet, translatorArray, isLocal);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerIndexAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- {
- return new CRoxieServerIndexAggregateActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId);
- }
- //--------------------------------------------------------------------------------------------------------------------------
- class CRoxieServerIndexGroupAggregateActivity : public CRoxieServerIndexActivity, implements IHThorGroupAggregateCallback
- {
- IHThorCompoundGroupAggregateExtra & aggregateHelper;
- RowAggregator singleAggregator;
- RowAggregator resultAggregator;
- unsigned groupSegCount;
- bool gathered;
- bool eof;
- public:
- CRoxieServerIndexGroupAggregateActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId,
- IKeyArray * _keySet, TranslatorArray *_translators, bool _isLocal)
- : CRoxieServerIndexActivity(_ctx, _factory, _probeManager, _remoteId, _keySet, _translators, false, _isLocal, false),
- aggregateHelper((IHThorIndexGroupAggregateArg &)basehelper),
- singleAggregator(aggregateHelper, aggregateHelper),
- resultAggregator(aggregateHelper, aggregateHelper),
- gathered(false), eof(true)
- {
- groupSegCount = 0;
- }
- IMPLEMENT_IINTERFACE_USING(CRoxieServerIndexActivity)
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- gathered= false;
- CRoxieServerIndexActivity::start(parentExtractSize, parentExtract, paused);
- groupSegCount = 0;
- if (!paused)
- processAllKeys();
- resultAggregator.start(rowAllocator);
- }
- virtual bool needsAllocator() const { return true; }
- virtual void reset()
- {
- resultAggregator.reset();
- CRoxieServerIndexActivity::reset();
- }
- virtual void processRow(const void * next)
- {
- singleAggregator.addRow(next);
- }
- virtual void createSegmentMonitors(IKeyManager *key)
- {
- unsigned groupSegSize;
- ThorActivityKind kind = factory->getKind();
- if ((kind==TAKindexgroupcount || kind==TAKindexgroupexists))
- groupSegSize = aggregateHelper.getGroupSegmentMonitorsSize();
- else
- groupSegSize = 0;
- if (groupSegSize)
- {
- key->setMergeBarrier(groupSegSize);
- CRoxieServerIndexActivity::createSegmentMonitors(key);
- unsigned numSegs = tlk->ordinality();
- for (unsigned segNo = 0; segNo < numSegs; segNo++)
- {
- IKeySegmentMonitor *seg = tlk->item(segNo);
- if (seg->getOffset()+seg->getSize()==groupSegSize)
- {
- groupSegCount = segNo+1;
- break;
- }
- }
- assertex(groupSegCount);
- }
- else
- CRoxieServerIndexActivity::createSegmentMonitors(key);
- }
- virtual bool processSingleKey(IKeyIndex *key, IRecordLayoutTranslator * trans)
- {
- Owned<CRowArrayMessageResult> result = new CRowArrayMessageResult(ctx->queryRowManager(), meta.isVariableSize());
- singleAggregator.start(rowAllocator);
- ThorActivityKind kind = factory->getKind();
- while (tlk->lookup(true))
- {
- try
- {
- if (groupSegCount && !trans)
- {
- AggregateRowBuilder &rowBuilder = singleAggregator.addRow(tlk->queryKeyBuffer(callback.getFPosRef()));
- callback.finishedRow();
- if (kind==TAKindexgroupcount)
- {
- unsigned __int64 count = tlk->getCurrentRangeCount(groupSegCount);
- aggregateHelper.processCountGrouping(rowBuilder, count-1);
- }
- if (!tlk->nextRange(groupSegCount))
- break;
- }
- else
- {
- aggregateHelper.processRow(tlk->queryKeyBuffer(callback.getFPosRef()), this);
- callback.finishedRow();
- }
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- accepted++;
- }
- loop
- {
- Owned<AggregateRowBuilder> next = singleAggregator.nextResult();
- if (!next)
- break;
- size32_t size = next->querySize();
- result->append(next->finalizeRowClear());
- }
- remote.injectResult(result.getClear());
- singleAggregator.reset();
- return false;
- }
- virtual void onLimitExceeded(bool isKeyed)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- throwUnexpected();
- }
- virtual const void *createLimitFailRow(bool isKeyed)
- {
- throwUnexpected();
- }
- void gatherMerged()
- {
- gathered = true;
- loop
- {
- const void * next = remote.nextRow();
- if (!next)
- break;
- resultAggregator.mergeElement(next);
- ReleaseRoxieRow(next);
- }
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if (eof)
- return NULL;
- if (!gathered)
- gatherMerged();
- Owned<AggregateRowBuilder> next = resultAggregator.nextResult();
- if (next)
- {
- processed++;
- return next->finalizeRowClear();
- }
- eof = true;
- return NULL;
- }
- };
- class CRoxieServerIndexGroupAggregateActivityFactory : public CRoxieServerBaseIndexActivityFactory
- {
- public:
- CRoxieServerIndexGroupAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- : CRoxieServerBaseIndexActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- if (!variableFileName && (keySet==NULL || keySet->length()==0))
- return new CRoxieServerNullActivity(_ctx, this, _probeManager);
- // else if (isSimple)
- // return new CRoxieServerSimpleIndexGroupAggregateActivity(this, keySet->queryKeyPart(0)->queryPart(0));
- else
- return new CRoxieServerIndexGroupAggregateActivity(_ctx, this, _probeManager, remoteId, keySet, translatorArray, isLocal);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerIndexGroupAggregateActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- {
- return new CRoxieServerIndexGroupAggregateActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId);
- }
- //--------------------------------------------------------------------------------------------------------------------------
- class CRoxieServerIndexNormalizeActivity : public CRoxieServerIndexReadBaseActivity
- {
- IHThorCompoundNormalizeExtra & readHelper;
- public:
- CRoxieServerIndexNormalizeActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId,
- IKeyArray * _keySet, TranslatorArray *_translators, bool _sorted, bool _isLocal)
- : CRoxieServerIndexReadBaseActivity(_ctx, _factory, _probeManager, _remoteId, _keySet, _translators, _sorted, _isLocal, false),
- readHelper((IHThorIndexNormalizeArg &)basehelper)
- {
- }
- virtual bool needsAllocator() const { return true; }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerIndexReadBaseActivity::start(parentExtractSize, parentExtract, paused);
- rowLimit = readHelper.getRowLimit();
- keyedLimit = readHelper.getKeyedLimit();
- choosenLimit = readHelper.getChooseNLimit();
- if (!paused)
- processAllKeys();
- }
- virtual bool processSingleKey(IKeyIndex *key, IRecordLayoutTranslator * trans)
- {
- unsigned keyedCount = 0;
- RtlDynamicRowBuilder rowBuilder(rowAllocator, false);
- while (tlk->lookup(true))
- {
- keyedCount++;
- if (keyedCount > keyedLimit)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d line = %d", activityId, __LINE__);
- onLimitExceeded(true);
- break;
- }
- size32_t transformedSize;
-
- if (readHelper.first(tlk->queryKeyBuffer(callback.getFPosRef())))
- {
- Owned<CRowArrayMessageResult> result = new CRowArrayMessageResult(ctx->queryRowManager(), meta.isVariableSize());
- do
- {
- if (accepted>=choosenLimit)
- break;
- rowBuilder.ensureRow();
- try
- {
- transformedSize = readHelper.transform(rowBuilder);
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- if (transformedSize)
- {
- // MORE - would be a good idea to stop these asap if rowlimit exceeded
- result->append(rowBuilder.finalizeRowClear(transformedSize));
- accepted++;
- }
- else
- rejected++;
- } while (readHelper.next());
- remote.injectResult(result.getClear());
- callback.finishedRow();
- if (accepted>=choosenLimit)
- return true;
- }
- }
- return false;
- }
- virtual void onLimitExceeded(bool isKeyed)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- if (isKeyed)
- {
- if (indexHelper.getFlags() & (TIRkeyedlimitskips|TIRkeyedlimitcreates))
- {
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- throw makeLimitSkipException(true);
- }
- else
- readHelper.onKeyedLimitExceeded();
- }
- else
- {
- if (indexHelper.getFlags() & (TIRlimitskips||TIRlimitcreates))
- {
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- throw makeLimitSkipException(false);
- }
- else
- readHelper.onLimitExceeded();
- }
- }
- virtual const void *createLimitFailRow(bool isKeyed)
- {
- UNIMPLEMENTED;
- }
- };
- class CRoxieServerIndexNormalizeActivityFactory : public CRoxieServerBaseIndexActivityFactory
- {
- public:
- CRoxieServerIndexNormalizeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- : CRoxieServerBaseIndexActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- if (!variableFileName && (keySet==NULL || keySet->length()==0))
- return new CRoxieServerNullActivity(_ctx, this, _probeManager);
- else
- return new CRoxieServerIndexNormalizeActivity(_ctx, this, _probeManager, remoteId, keySet, translatorArray, sorted, isLocal);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerIndexNormalizeActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- {
- return new CRoxieServerIndexNormalizeActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId);
- }
- //=================================================================================
- class CRoxieServerFetchActivity : public CRoxieServerActivity, implements IRecordPullerCallback, implements IRoxieServerErrorHandler
- {
- IHThorFetchBaseArg &helper;
- IHThorFetchContext * fetchContext;
- Linked<IFilePartMap> map;
- CRemoteResultAdaptor remote;
- RecordPullerThread puller;
- bool needsRHS;
- bool variableFileName;
- bool isOpt;
- Owned<const IResolvedFile> varFileInfo;
- public:
- CRoxieServerFetchActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, IFilePartMap *_map)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorFetchBaseArg &)basehelper), map(_map), remote(_ctx, this, _remoteId, meta.queryOriginal(), helper, *this, true, true), puller(false)
- {
- fetchContext = static_cast<IHThorFetchContext *>(helper.selectInterface(TAIfetchcontext_1));
- needsRHS = helper.transformNeedsRhs();
- variableFileName = allFilesDynamic || factory->queryQueryFactory().isDynamic() || ((fetchContext->getFetchFlags() & (FFvarfilename|FFdynamicfilename)) != 0);
- isOpt = (fetchContext->getFetchFlags() & FFdatafileoptional) != 0;
- }
- virtual const IResolvedFile *queryVarFileInfo() const
- {
- return varFileInfo;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- remote.onCreate(_colocalParent);
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- if (idx)
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() parameter out of bounds at %s(%d)", __FILE__, __LINE__);
- puller.setInput(this, _sourceIdx, _in);
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- puller.connectInputStreams(ctx, consumerOrdered);
- return CRoxieServerActivity::getOutputStreams(ctx, idx, streams, consumerOptions, consumerOrdered, orderedCallbacks);
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (idx==0)
- return puller.queryInput();
- else
- return NULL;
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- remote.onStart(parentExtractSize, parentExtract);
- remote.setLimits(helper.getRowLimit(), (unsigned __int64) -1, I64C(0x7FFFFFFFFFFFFFFF));
- if (variableFileName)
- {
- OwnedRoxieString fname(fetchContext->getFileName());
- varFileInfo.setown(resolveLFN(fname, isOpt));
- if (varFileInfo)
- map.setown(varFileInfo->getFileMap());
- }
- puller.start(parentExtractSize, parentExtract, paused, ctx->queryOptions().fetchPreload, false, ctx);
- }
- virtual void stop()
- {
- // Called from remote, so no need to call back to it....
- puller.stop();
- CRoxieServerActivity::stop();
- }
- virtual void reset()
- {
- processed = remote.processed;
- remote.processed = 0;
- puller.reset();
- if (variableFileName)
- {
- varFileInfo.clear();
- map.clear();
- }
- CRoxieServerActivity::reset();
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- if (idx==(unsigned)-1)
- idx = 0;
- return idx ? NULL : &remote;
- }
- virtual void processRow(const void *row)
- {
- // called from puller thread
- offset_t rp = fetchContext->extractPosition(row);
- unsigned partNo;
- if (isLocalFpos(rp))
- partNo = getLocalFposPart(rp) + 1;
- else
- partNo = map->mapOffset(rp);
- if (needsRHS)
- {
- Owned<IEngineRowAllocator> extractAllocator = createRowAllocator(helper.queryExtractedSize());
- RtlDynamicRowBuilder rb(extractAllocator, true);
- unsigned rhsSize = helper.extractJoinFields(rb, row);
- char * block = (char *) remote.getMem(partNo, 0, sizeof(rp) + sizeof(rhsSize) + rhsSize); // MORE - superfiles
- *(offset_t *) block = rp;
- block += sizeof(rp);
- *(unsigned *) block = rhsSize;
- block += sizeof(rhsSize);
- memcpy(block, rb.row(), rhsSize);
- }
- else
- *(offset_t *) remote.getMem(partNo, 0, sizeof(rp)) = rp; // MORE - superfiles
- ReleaseRoxieRow(row);
- }
- void processEOG()
- {
- #ifdef FETCH_PRESERVES_GROUPING
- UNIMPLEMENTED;
- #endif
- // else discard is correct
- }
- void processGroup(const ConstPointerArray &)
- {
- throwUnexpected();
- }
- void processDone()
- {
- // called from puller thread
- remote.flush();
- remote.senddone();
- }
- virtual bool fireException(IException *e)
- {
- return remote.fireException(e);
- }
- virtual void onLimitExceeded(bool isKeyed)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- if (isKeyed)
- throwUnexpected();
- helper.onLimitExceeded();
- }
- virtual const void *createLimitFailRow(bool isKeyed)
- {
- UNIMPLEMENTED;
- }
- virtual const void *nextRow()
- {
- throwUnexpected(); // I am nobody's input
- }
- };
- class CRoxieServerFetchActivityFactory : public CRoxieServerActivityFactory
- {
- RemoteActivityId remoteId;
- Owned<IFilePartMap> map;
- bool variableFileName;
- Owned<const IResolvedFile> datafile;
- public:
- CRoxieServerFetchActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), remoteId(_remoteId)
- {
- Owned<IHThorFetchBaseArg> helper = (IHThorFetchBaseArg *) helperFactory();
- IHThorFetchContext *fetchContext = static_cast<IHThorFetchContext *>(helper->selectInterface(TAIfetchcontext_1));
- variableFileName = allFilesDynamic || _queryFactory.isDynamic() || ((fetchContext->getFetchFlags() & (FFvarfilename|FFdynamicfilename)) != 0);
- if (!variableFileName)
- {
- OwnedRoxieString fname(fetchContext->getFileName());
- datafile.setown(_queryFactory.queryPackage().lookupFileName(fname,
- (fetchContext->getFetchFlags() & FFdatafileoptional) != 0,
- true, true,
- _queryFactory.queryWorkUnit(), true));
- if (datafile)
- map.setown(datafile->getFileMap());
- }
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerFetchActivity(_ctx, this, _probeManager, remoteId, map);
- }
- virtual void getXrefInfo(IPropertyTree &reply, const IRoxieContextLogger &logctx) const
- {
- if (datafile)
- addXrefFileInfo(reply, datafile);
- else
- {
- // Temporarily resolve the file
- Owned<IHThorFetchBaseArg> helper = (IHThorFetchBaseArg *) helperFactory();
- IHThorFetchContext *fetchContext = static_cast<IHThorFetchContext *>(helper->selectInterface(TAIfetchcontext_1));
- if ((fetchContext->getFetchFlags() & (FFvarfilename|FFdynamicfilename)) == 0)
- {
- OwnedRoxieString fileName(fetchContext->getFileName());
- Owned<const IResolvedFile> temp = queryFactory.queryPackage().lookupFileName(fileName, true, true, false, queryFactory.queryWorkUnit(), true);
- if (temp)
- addXrefFileInfo(reply, temp);
- }
- }
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return diskStatistics;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerFetchActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId)
- {
- return new CRoxieServerFetchActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId);
- }
- // MORE - is there any point keeping this now?
- class CRoxieServerDummyActivityFactory : public CRoxieServerActivityFactory // not a real activity - just used to properly link files
- {
- public:
- Owned<const IResolvedFile> indexfile;
- Owned<const IResolvedFile> datafile;
- StringAttr fileName;
- StringAttr indexName;
- Owned<IKeyArray> keySet;
- Owned<IFileIOArray> files;
- TranslatorArray layoutTranslators;
- CRoxieServerDummyActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool isLoadDataOnly)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- try // does not want any missing file errors to be fatal, or throw traps - just log it
- {
- if (_graphNode.getPropBool("att[@name='_isSpill']/@value", false) || _graphNode.getPropBool("att[@name='_isSpillGlobal']/@value", false))
- return; // ignore 'spills'
- bool isLocal = _graphNode.getPropBool("att[@name='local']/@value") && queryFactory.queryChannel()!=0;
- ThorActivityKind kind = getActivityKind(_graphNode);
- if (kind != TAKdiskwrite && kind != TAKindexwrite && kind != TAKpiperead && kind != TAKpipewrite)
- {
- fileName.set(queryNodeFileName(_graphNode, kind));
- indexName.set(queryNodeIndexName(_graphNode, kind));
- if (indexName && !allFilesDynamic && !queryFactory.isDynamic())
- {
- bool isOpt = pretendAllOpt || _graphNode.getPropBool("att[@name='_isIndexOpt']/@value");
- indexfile.setown(queryFactory.queryPackage().lookupFileName(indexName, isOpt, true, true, queryFactory.queryWorkUnit(), true));
- if (indexfile)
- keySet.setown(indexfile->getKeyArray(NULL, &layoutTranslators, isOpt, isLocal ? queryFactory.queryChannel() : 0, false));
- }
- if (fileName && !allFilesDynamic && !queryFactory.isDynamic())
- {
- bool isOpt = pretendAllOpt || _graphNode.getPropBool("att[@name='_isOpt']/@value");
- datafile.setown(_queryFactory.queryPackage().lookupFileName(fileName, isOpt, true, true, queryFactory.queryWorkUnit(), true));
- if (datafile)
- {
- if (isLocal)
- files.setown(datafile->getIFileIOArray(isOpt, queryFactory.queryChannel()));
- }
- }
- }
- }
- catch(IException *E)
- {
- StringBuffer errors;
- E->errorMessage(errors);
- DBGLOG("%s File error = %s", (isLoadDataOnly) ? "LOADDATAONLY" : "SUSPENDED QUERY", errors.str());
- E->Release();
- }
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const { throw MakeStringException(ROXIE_INTERNAL_ERROR, "%s query %s is suspended and cannot be executed - error occurred at %s(%d)", (queryFactory.isQueryLibrary()) ? "Library" : " ", queryFactory.queryQueryName(), __FILE__, __LINE__); }
- virtual void getXrefInfo(IPropertyTree &reply, const IRoxieContextLogger &logctx) const
- {
- if (!allFilesDynamic)
- {
- if (datafile)
- addXrefFileInfo(reply, datafile);
- if (indexfile)
- addXrefFileInfo(reply, indexfile);
- }
- else
- {
- Owned<const IResolvedFile> temp;
- if (fileName.length())
- {
- temp.setown(queryFactory.queryPackage().lookupFileName(fileName, true, true, false, queryFactory.queryWorkUnit(), true));
- if (temp)
- addXrefFileInfo(reply, temp);
- }
- if (indexName.length())
- {
- temp.setown(queryFactory.queryPackage().lookupFileName(indexName, true, true, false, queryFactory.queryWorkUnit(), true));
- if (temp)
- addXrefFileInfo(reply, temp);
- }
- }
- }
- };
- IRoxieServerActivityFactory *createRoxieServerDummyActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool isLoadDataOnly)
- {
- return new CRoxieServerDummyActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, isLoadDataOnly);
- }
- //=====================================================================================================
- // Keyed joins...
- //
- // Input records are pulled by a puller thread, which checks each LHS record to determine which (if any) channels it
- // may have RHS matches on, and sends the relevant fields to the relevant slaves.
- // A separate thread (the caller's thread) is waiting on slave replies, and once it has all replies for a given LHS record or group of records, calls
- // the transform and returns rows that are created.
- // For a full-keyed join, there is a third thread that is pulling replies from index part and passing them to fetch part (check this is true)
- //
- //=====================================================================================================
- class CJoinGroup;
- interface IJoinProcessor
- {
- virtual void processEOG() = 0;
- virtual CJoinGroup *createJoinGroup(const void *row) = 0;
- virtual void noteEndReceived(CJoinGroup *jg, unsigned candidateCount) = 0;
- virtual bool fireException(IException *E) = 0;
- virtual void processCompletedGroups() = 0;
- };
- //------------------------------------------------------------------------------------------------------
- // Class CJoinGroup has a record per LHS row, plus (if preserving grouping) a 'head of group' record
- // It gathers all the corresponding RHS rows, keeping track of how may slave transactions are pending in endMarkersPending
- // If preserving groups, the 'head of group' record keeps track of how many LHS records in the group are still incomplete.
- // CJoinGroup records are allocated out of the Roxie row memory manager by overloading operator new, so that they are included in the
- // per-query limits etc (Note that the pointer array block is not though).
- // Because of that, the exact size is significant - especially whether fit just under or just over a chunking threshold...
- //
- // There are two phases to the life of a JoinGroup - it is created by the puller thread that is also firing off slave requests
- // notePending will be called once for every slave request. Puller thread calls noteEndReceived(0) once when done - this corresponds to the
- // initial count when created.
- // Slave replies and are noted by the consumer thread calling addRightMatch() and noteEndReceived(n).
- // Once endMarkersPending reaches 0, JoinGroup is complete. Last thread to call noteEndReceived will process the rows and destroy the group.
- // There is no need for a critsec because although multiple threads will access at different times, only the consumer thread will
- // access any modifiable member variables while endMarkersPending != 0 (i.e. complete() is false). Once complete returns true there is a single
- // remaining reference and the JoinGroup will be processed and destroyed.
- //
- //------------------------------------------------------------------------------------------------------
- class CJoinGroup : public CInterface
- {
- protected:
- const void *left; // LHS row
- PointerArrayOf<KeyedJoinHeader> rows; // matching RHS rows
- atomic_t endMarkersPending; // How many slave responses still waiting for
- CJoinGroup *groupStart; // Head of group, or NULL if not grouping
- unsigned lastPartNo;
- unsigned pos;
- unsigned candidates; // Number of RHS keyed candidates - note this may not be the same as rows.ordinality()
- public:
- #undef new
- void *operator new(size_t size, IRowManager *a, unsigned activityId)
- {
- return a->allocate(size, activityId);
- }
- #if defined(_DEBUG) && defined(_WIN32) && !defined(USING_MPATROL)
- #define new new(_NORMAL_BLOCK, __FILE__, __LINE__)
- #endif
- void operator delete(void *ptr, IRowManager *a, unsigned activityId)
- {
- ReleaseRoxieRow(ptr);
- }
- void operator delete(void *ptr)
- {
- ReleaseRoxieRow(ptr);
- }
- public:
- CJoinGroup(const void *_left, CJoinGroup *_groupStart)
- {
- #ifdef TRACE_JOINGROUPS
- DBGLOG("Creating joinGroup %p, groupstart %p", this, _groupStart);
- #endif
- candidates = 0;
- lastPartNo = 0;
- pos = 0;
- left = _left;
- groupStart = _groupStart;
- if (_groupStart)
- {
- atomic_inc(&_groupStart->endMarkersPending);
- }
- atomic_set(&endMarkersPending, 1);
- }
- ~CJoinGroup()
- {
- #ifdef TRACE_JOINGROUPS
- DBGLOG("Destroying joinGroup %p", this);
- #endif
- if (left)
- {
- ReleaseRoxieRow(left);
- roxiemem::ReleaseRoxieRowArray(rows.ordinality(), (const void * *)rows.getArray());
- rows.kill();
- }
- }
- inline bool isHeadRecord() const
- {
- return left==NULL;
- }
- inline bool complete() const
- {
- return atomic_read(&endMarkersPending) == 0;
- }
- #ifdef TRACE_JOINGROUPS
- inline void notePending(unsigned lineNo)
- #else
- inline void notePending()
- #endif
- {
- assert(!complete());
- atomic_inc(&endMarkersPending);
- #ifdef TRACE_JOINGROUPS
- DBGLOG("CJoinGroup::notePending %p from %d, count became %d group count %d", this, lineNo, atomic_read(&endMarkersPending), groupStart ? atomic_read(&groupStart->endMarkersPending) : 0);
- #endif
- }
- inline bool inGroup(CJoinGroup *leader) const
- {
- return groupStart==leader;
- }
- inline const KeyedJoinHeader *queryRow(unsigned idx) const
- {
- // Single threaded by now
- assert(complete());
- return rows.item(idx);
- }
- #ifdef TRACE_JOINGROUPS
- bool noteEndReceived(unsigned candidateCount, unsigned lineNo)
- #else
- bool noteEndReceived(unsigned candidateCount)
- #endif
- {
- assert(!complete());
- if (candidateCount)
- {
- candidates += candidateCount;
- }
- #ifdef TRACE_JOINGROUPS
- DBGLOG("CJoinGroup::noteEndReceived %p from %d, candidates %d + %d, my count was %d, group count was %d", this, lineNo, candidates, candidateCount, atomic_read(&endMarkersPending), groupStart ? atomic_read(&groupStart->endMarkersPending) : 0);
- #endif
- // NOTE - as soon as endMarkersPending and groupStart->endMarkersPending are decremented to zero this object may get released asynchronously by other threads
- // There must therefore be nothing in this method after them that acceses member variables. Think of it as a delete this...
- // In particular, we can't safely reference groupStart after the dec_and_test of endMarkersPending, hence copy local first
- CJoinGroup *localGroupStart = groupStart;
- if (atomic_dec_and_test(&endMarkersPending))
- {
- if (localGroupStart)
- return atomic_dec_and_test(&localGroupStart->endMarkersPending);
- else
- return true;
- }
- else
- return false;
- }
- inline const void *queryLeft() const
- {
- return left;
- }
- void addRightMatch(KeyedJoinHeader *right)
- {
- assert(!complete());
- unsigned short partNo = right->partNo;
- if (partNo != lastPartNo)
- {
- // MORE - should we binchop? If we did we would need to be careful to find LAST match
- if (partNo > lastPartNo)
- pos = rows.length();
- while (pos>0)
- {
- if (rows.item(pos-1)->partNo <= partNo)
- break;
- pos--;
- }
- lastPartNo = partNo;
- }
- rows.add(right, pos);
- pos++;
- }
- inline unsigned rowsSeen() const
- {
- assert(complete());
- return rows.length();
- }
- inline unsigned candidateCount() const
- {
- assert(complete());
- return candidates;
- }
- };
- #ifdef TRACE_JOINGROUPS
- #define notePending() notePending(__LINE__)
- #define noteEndReceived(a) noteEndReceived(a, __LINE__)
- #endif
- class KeyedJoinRemoteAdaptor : public CRemoteResultAdaptor // MORE - not sure it should be derived from this - makes processed all wrong, for example
- {
- private:
- SafeQueueOf<const void, true> ready;
- public:
- IHThorKeyedJoinArg &helper;
- unsigned joinProcessed;
- bool isFullKey;
- bool eof;
- bool isSimple;
- bool allPulled;
- ActivityTimeAccumulator totalCycles;
- unsigned activityId;
- RecordPullerThread &puller;
- SafeQueueOf<const void, true> injected; // Used in isSimple mode
- Owned<IEngineRowAllocator> ccdRecordAllocator;
- IJoinProcessor &processor;
- KeyedJoinRemoteAdaptor(IRoxieSlaveContext *_ctx, IRoxieServerErrorHandler *_errorHandler, const RemoteActivityId &_remoteId, IHThorKeyedJoinArg &_helper,
- IRoxieServerActivity &_activity, bool _isFullKey, bool _isSimple,
- RecordPullerThread &_puller, IJoinProcessor &_processor)
- : helper(_helper),
- CRemoteResultAdaptor(_ctx, _errorHandler, _remoteId, 0, _helper, _activity, true, true),
- isFullKey(_isFullKey),
- isSimple(_isSimple),
- puller(_puller),
- processor(_processor),
- activityId(_activity.queryId())
- {
- joinProcessed = 0;
- allPulled = false;
- eof = false;
- }
- virtual void onCreate(IHThorArg *_colocalArg)
- {
- CRemoteResultAdaptor::onCreate(_colocalArg);
- ccdRecordAllocator.setown(activity.createRowAllocator(helper.queryJoinFieldsRecordSize()));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- joinProcessed = 0;
- totalCycles.totalCycles = 0;
- allPulled = false;
- assertex(ready.ordinality()==0);
- CRemoteResultAdaptor::start(parentExtractSize, parentExtract, paused);
- }
- virtual void reset()
- {
- CRemoteResultAdaptor::reset();
- while (ready.ordinality())
- {
- const void *goer = ready.dequeue();
- if (goer)
- ReleaseRoxieRow(goer);
- }
- while (injected.ordinality())
- {
- const void *goer = injected.dequeue();
- if (goer)
- ReleaseRoxieRow(goer);
- }
- }
- inline void addResult(const void *row)
- {
- ready.enqueue(row);
- }
- virtual unsigned __int64 queryTotalCycles() const
- {
- return totalCycles.totalCycles;
- }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- loop
- {
- if (eof)
- return NULL;
- processSlaveResults();
- if (ready.ordinality())
- {
- const void *result = ready.dequeue();
- if (result)
- joinProcessed++;
- return result;
- }
- else
- eof = true;
- }
- }
- private:
- void processSlaveResults()
- {
- while (!ready.ordinality())
- {
- KeyedJoinHeader *fetchedData;
- if (isSimple)
- {
- while (!allPulled && !injected.ordinality())
- {
- if (!puller.pullRecords(1))
- {
- puller.done();
- allPulled = true;
- }
- }
- fetchedData = (KeyedJoinHeader *) injected.dequeue();
- }
- else
- fetchedData = (KeyedJoinHeader *) CRemoteResultAdaptor::nextRow();
- if (fetchedData)
- {
- CJoinGroup *thisGroup = fetchedData->thisGroup;
- if (fetchedData->partNo == (unsigned short) -1)
- {
- #ifdef TRACE_JOINGROUPS
- CTXLOG("Got end for group %p", thisGroup);
- #endif
- unsigned candidateCount = (unsigned) fetchedData->fpos;
- ReleaseRoxieRow(fetchedData);
- processor.noteEndReceived(thisGroup, candidateCount); // note - this can throw exception. So release fetchdata before calling
- }
- else
- {
- #ifdef TRACE_JOINGROUPS
- CTXLOG("Reading another %d bytes for group %p data", ccdRecordSize, thisGroup);
- #endif
- thisGroup->addRightMatch(fetchedData);
- if (isFullKey)
- {
- #ifdef TRACE_JOINGROUPS
- CTXLOG("Calling noteEndReceived for record returned from FETCH of full keyed join");
- #endif
- processor.noteEndReceived(thisGroup, 0); // note - this can throw exception. So release fetchdata before calling
- }
- }
- }
- else
- break;
- }
- }
- };
- class CRoxieServerFullKeyedJoinHead: public CRoxieServerActivity, implements IRecordPullerCallback, implements IRoxieServerErrorHandler
- {
- IHThorKeyedJoinArg &helper;
- Owned<IKeyManager> tlk;
- Linked<IKeyArray> keySet;
- Linked<TranslatorArray> translators;
- CRemoteResultAdaptor remote;
- RecordPullerThread puller;
- IOutputMetaData *indexReadMeta;
- IJoinProcessor *joinHandler;
- bool variableIndexFileName;
- bool indexReadInputRecordVariable;
- bool isLocal;
- Owned<IEngineRowAllocator> indexReadAllocator;
- Owned<const IResolvedFile> varFileInfo;
- IFinalRoxieInput *indexReadInput;
- unsigned indexReadIdx = 0;
- IIndexReadActivityInfo *rootIndex;
- public:
- CRoxieServerFullKeyedJoinHead(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, IKeyArray * _keySet, TranslatorArray *_translators, IOutputMetaData *_indexReadMeta, IJoinProcessor *_joinHandler, bool _isLocal)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- helper((IHThorKeyedJoinArg &)basehelper),
- tlk(createKeyManager(NULL, 0, this)),
- translators(_translators),
- keySet(_keySet),
- remote(_ctx, this, _remoteId, 0, helper, *this, true, true),
- indexReadMeta(_indexReadMeta),
- joinHandler(_joinHandler),
- puller(false),
- isLocal(_isLocal)
- {
- variableIndexFileName = allFilesDynamic || factory->queryQueryFactory().isDynamic() || ((helper.getJoinFlags() & (JFvarindexfilename|JFdynamicindexfilename|JFindexfromactivity)) != 0);
- indexReadInputRecordVariable = indexReadMeta->isVariableSize();
- indexReadInput = NULL;
- rootIndex = NULL;
- }
- virtual const IResolvedFile *queryVarFileInfo() const
- {
- return varFileInfo;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- remote.onCreate(_colocalParent);
- indexReadAllocator.setown(createRowAllocator(indexReadMeta));
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- if (!idx)
- puller.setInput(this, _sourceIdx, _in);
- else if (idx==1)
- {
- indexReadInput = _in;
- indexReadIdx = _sourceIdx;
- }
- else
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() parameter out of bounds at %s(%d)", __FILE__, __LINE__);
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- puller.connectInputStreams(ctx, consumerOrdered);
- //No rows are read from indexReadInput, so no need to extract the streams
- return CRoxieServerActivity::getOutputStreams(ctx, idx, streams, consumerOptions, consumerOrdered, orderedCallbacks);
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- switch (idx)
- {
- case 0:
- return puller.queryInput();
- case 1:
- return indexReadInput;
- default:
- return NULL;
- }
- }
- virtual void serializeExtra(MemoryBuffer &out)
- {
- if (helper.getJoinFlags() & JFindexfromactivity)
- {
- assertex(rootIndex);
- const RemoteActivityId& indexId = rootIndex->queryRemoteId();
- indexId.serialize(out);
- // could mess about reserving space for length then patching it again, to avoid copy, but probably not worth it
- MemoryBuffer tmp;
- rootIndex->queryActivity()->serializeCreateStartContext(tmp);
- if (rootIndex->queryActivity()->queryVarFileInfo())
- {
- rootIndex->queryActivity()->queryVarFileInfo()->queryTimeStamp().serialize(tmp);
- tmp.append(rootIndex->queryActivity()->queryVarFileInfo()->queryCheckSum());
- }
- unsigned ctxlen = tmp.length();
- out.append(ctxlen).append(tmp);
- }
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if (indexReadInput)
- {
- indexReadInput->start(parentExtractSize, parentExtract, true); // paused=true because we don't want to actually run the index read
- rootIndex = indexReadInput->queryIndexReadActivity();
- if (!rootIndex)
- throw MakeStringException(ROXIE_INTERNAL_ERROR,"Index in keyed join %d could not be resolved", queryId());
- }
- remote.onStart(parentExtractSize, parentExtract);
- remote.setLimits(helper.getRowLimit(), (unsigned __int64) -1, I64C(0x7FFFFFFFFFFFFFFF));
- if (rootIndex)
- {
- varFileInfo.setown(rootIndex->getVarFileInfo());
- translators.setown(rootIndex->getTranslators());
- keySet.setown(rootIndex->getKeySet());
- }
- else if (variableIndexFileName)
- {
- OwnedRoxieString indexFileName(helper.getIndexFileName());
- varFileInfo.setown(resolveLFN(indexFileName, (helper.getJoinFlags() & JFindexoptional) != 0));
- if (varFileInfo)
- {
- translators.setown(new TranslatorArray);
- keySet.setown(varFileInfo->getKeyArray(factory->queryActivityMeta(), translators, false, isLocal ? factory->queryQueryFactory().queryChannel() : 0, factory->queryQueryFactory().queryOptions().enableFieldTranslation)); // MORE - isLocal?
- }
- }
- puller.start(parentExtractSize, parentExtract, paused, ctx->queryOptions().fullKeyedJoinPreload, false, ctx);
- }
- virtual void stop()
- {
- puller.stop();
- CRoxieServerActivity::stop();
- }
- virtual void reset()
- {
- CRoxieServerActivity::reset();
- puller.reset();
- if (varFileInfo)
- {
- keySet.clear();
- varFileInfo.clear();
- }
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- if (idx==(unsigned)-1)
- idx = 0;
- return idx ? NULL : &remote;
- }
- virtual void processRow(const void *row)
- {
- // MORE - this code seems to be pretty much duplicated below in half-keyed....
- // called from front puller thread
- // buffer up an IndexRead request
- if (helper.leftCanMatch(row))
- {
- RtlDynamicRowBuilder extractedBuilder(indexReadAllocator);
- unsigned indexReadSize = helper.extractIndexReadFields(extractedBuilder, row);
- OwnedConstRoxieRow extracted;
- if (indexReadSize)
- extracted.setown(extractedBuilder.finalizeRowClear(indexReadSize));
- CJoinGroup *jg = joinHandler->createJoinGroup(row);
- for (unsigned partNo = 0; partNo < keySet->length(); partNo++)
- {
- IKeyIndexBase *thisBase = keySet->queryKeyPart(partNo);
- if (thisBase)
- {
- unsigned fileNo = 0;
- IKeyIndex *thisKey = thisBase->queryPart(fileNo);
- try
- {
- tlk->setKey(thisKey);
- tlk->setLayoutTranslator(translators->item(fileNo));
- helper.createSegmentMonitors(tlk, extracted);
- if (rootIndex)
- rootIndex->mergeSegmentMonitors(tlk);
- tlk->finishSegmentMonitors();
- tlk->reset();
- loop
- {
- typedef const void * cvp;
- if (thisKey->isTopLevelKey())
- {
- bool locallySorted = !thisKey->isFullySorted();
- while (locallySorted || tlk->lookup(false))
- {
- unsigned slavePart = locallySorted ? 0 : (unsigned) tlk->queryFpos();
- if (locallySorted || slavePart)
- {
- cvp *outputBuffer = (cvp *) remote.getMem(slavePart, fileNo, indexReadSize + sizeof(cvp) + (indexReadInputRecordVariable ? sizeof(unsigned) : 0));
- *outputBuffer++ = jg;
- if (indexReadInputRecordVariable)
- {
- *(unsigned *) outputBuffer = indexReadSize;
- outputBuffer = (cvp*) (((unsigned *) outputBuffer) + 1);
- }
- jg->notePending();
- memcpy(outputBuffer, extracted, indexReadSize);
- if (locallySorted)
- {
- for (unsigned i = 1; i < numChannels; i++)
- jg->notePending();
- break;
- }
- }
- }
- }
- else
- {
- // MORE - this code seems to be duplicated in half keyed
- unsigned accepted = 0;
- unsigned rejected = 0;
- Owned<CRowArrayMessageResult> result = new CRowArrayMessageResult(ctx->queryRowManager(), true);
- jg->notePending();
- unsigned candidateCount = 0;
- while (tlk->lookup(true))
- {
- candidateCount++;
- atomic_inc(&indexRecordsRead);
- KLBlobProviderAdapter adapter(tlk);
- offset_t recptr;
- const byte *indexRow = tlk->queryKeyBuffer(recptr);
- if (helper.indexReadMatch(extracted, indexRow, recptr, &adapter))
- {
- KeyedJoinHeader *rhs = (KeyedJoinHeader *) ctx->queryRowManager().allocate(KEYEDJOIN_RECORD_SIZE(0), activityId);
- rhs->fpos = recptr;
- rhs->thisGroup = jg;
- rhs->partNo = partNo;
- result->append(rhs);
- }
- else
- {
- rejected++;
- atomic_inc(&postFiltered);
- }
- }
- // output an end marker for the matches to this group
- KeyedJoinHeader *endMarker = (KeyedJoinHeader *) ctx->queryRowManager().allocate(KEYEDJOIN_RECORD_SIZE(0), activityId);
- endMarker->fpos = (offset_t) candidateCount;
- endMarker->thisGroup = jg;
- endMarker->partNo = (unsigned short) -1;
- result->append(endMarker);
- remote.injectResult(result.getClear());
- if (accepted)
- noteStatistic(StNumIndexAccepted, accepted);
- if (rejected)
- noteStatistic(StNumIndexRejected, rejected);
- }
- if (++fileNo < thisBase->numParts())
- {
- thisKey = thisBase->queryPart(fileNo);
- tlk->setKey(thisKey);
- tlk->setLayoutTranslator(translators->item(fileNo));
- tlk->reset();
- }
- else
- break;
- }
- tlk->releaseSegmentMonitors();
- tlk->setKey(NULL);
- }
- catch (...)
- {
- tlk->releaseSegmentMonitors();
- tlk->setKey(NULL);
- throw;
- }
- }
- }
- joinHandler->noteEndReceived(jg, 0);
- }
- else
- {
- joinHandler->noteEndReceived(joinHandler->createJoinGroup(row), 0);
- }
- }
- void processGroup(const ConstPointerArray &)
- {
- throwUnexpected();
- }
- virtual void processEOG()
- {
- joinHandler->processEOG();
- }
- virtual void processDone()
- {
- // called from puller thread
- remote.flush();
- remote.senddone();
- }
- virtual void onLimitExceeded(bool isKeyed)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- if (isKeyed)
- throwUnexpected();
- helper.onLimitExceeded();
- }
- virtual const void *createLimitFailRow(bool isKeyed)
- {
- throwUnexpected();
- }
- virtual bool fireException(IException *e)
- {
- // called from puller thread on failure
- remote.fireException(LINK(e));
- return joinHandler->fireException(e);
- }
- virtual const void *nextRow()
- {
- throwUnexpected(); // I am nobody's input
- }
- };
- class CRoxieServerKeyedJoinBase : public CRoxieServerActivity, implements IRecordPullerCallback, implements IRoxieServerErrorHandler, implements IJoinProcessor
- {
- protected:
- IHThorKeyedJoinArg &helper;
- KeyedJoinRemoteAdaptor remote;
- RecordPullerThread puller;
- OwnedConstRoxieRow defaultRight;
- Owned<IEngineRowAllocator> defaultRightAllocator;
- unsigned joinFlags;
- unsigned atMost;
- unsigned atmostsTriggered;
- unsigned abortLimit;
- unsigned keepLimit;
- bool limitFail;
- bool limitOnFail;
- bool preserveGroups;
- bool cloneLeft;
- bool isSimple;
- bool isLocal;
- ThorActivityKind activityKind;
- CJoinGroup *groupStart;
- CriticalSection groupsCrit;
- QueueOf<CJoinGroup, false> groups;
- IFinalRoxieInput *indexReadInput;
- unsigned indexReadIdx = 0;
- Owned<IStrandJunction> indexReadJunction;
- IEngineRowStream *indexReadStream = NULL; // Never actually pulled
- IIndexReadActivityInfo *rootIndex;
- void createDefaultRight()
- {
- if (!defaultRight)
- {
- if (!defaultRightAllocator)
- defaultRightAllocator.setown(createRowAllocator(helper.queryJoinFieldsRecordSize()));
- RtlDynamicRowBuilder rowBuilder(defaultRightAllocator);
- size32_t thisSize = helper.createDefaultRight(rowBuilder);
- defaultRight.setown(rowBuilder.finalizeRowClear(thisSize));
- }
- }
- public:
- CRoxieServerKeyedJoinBase(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, unsigned _joinFlags
- , bool isFull, bool _isSimple, bool _isLocal)
- : CRoxieServerActivity(_ctx, _factory, _probeManager),
- helper((IHThorKeyedJoinArg &)basehelper),
- remote(_ctx, this, _remoteId, helper, *this, isFull, _isSimple, puller, *this),
- joinFlags(_joinFlags),
- preserveGroups(meta.isGrouped()),
- puller(false),
- isSimple(_isSimple),
- isLocal(_isLocal),
- abortLimit(0),
- keepLimit(0),
- atMost(0),
- limitFail(false),
- limitOnFail(false),
- cloneLeft(false)
- {
- groupStart = NULL;
- activityKind = _factory->getKind();
- indexReadInput = NULL;
- rootIndex = NULL;
- atmostsTriggered = 0;
- // MORE - code would be easier to read if I got more values from helper rather than passing from factory
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerActivity::onCreate(_colocalParent);
- remote.onCreate(_colocalParent);
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- if (!idx)
- puller.setInput(this, _sourceIdx, _in);
- else if (idx==1)
- {
- indexReadInput = _in;
- indexReadIdx = _sourceIdx;
- }
- else
- throw MakeStringException(ROXIE_SET_INPUT, "Internal error: setInput() parameter out of bounds at %s(%d)", __FILE__, __LINE__);
- }
- virtual void connectInputStreams(bool consumerOrdered)
- {
- CRoxieServerActivity::connectInputStreams(consumerOrdered);
- puller.connectInputStreams(ctx, consumerOrdered);
- if (indexReadInput)
- indexReadStream = connectSingleStream(ctx, indexReadInput, indexReadIdx, indexReadJunction, consumerOrdered); // We never actually pull the stream
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- if (indexReadInput)
- {
- indexReadInput->start(parentExtractSize, parentExtract, true); // paused=true because we don't want to actually run the index read
- rootIndex = indexReadInput->queryIndexReadActivity();
- if (!rootIndex)
- throw MakeStringException(ROXIE_INTERNAL_ERROR,"Index in keyed join %d could not be resolved", queryId());
- }
- remote.onStart(parentExtractSize, parentExtract);
- remote.setLimits(helper.getRowLimit(), (unsigned __int64) -1, I64C(0x7FFFFFFFFFFFFFFF));
- atmostsTriggered = 0;
- atMost = helper.getJoinLimit();
- if (atMost == 0) atMost = (unsigned)-1;
- abortLimit = helper.getMatchAbortLimit();
- if (abortLimit == 0 || atMost != (unsigned) -1) abortLimit = (unsigned)-1;
- keepLimit = helper.getKeepLimit();
- if (keepLimit == 0) keepLimit = (unsigned)-1;
- getLimitType(joinFlags, limitFail, limitOnFail);
- cloneLeft = (joinFlags & JFtransformmatchesleft) != 0;
- if (joinFlags & JFleftouter || limitOnFail)
- createDefaultRight();
- }
- virtual void stop()
- {
- puller.stop();
- if (indexReadStream)
- indexReadStream->stop(); // Could probably do this as soon as re have fetched rootIndex?
- CRoxieServerActivity::stop();
- }
- virtual unsigned __int64 queryLocalCycles() const
- {
- __int64 localCycles = remote.totalCycles.totalCycles;
- localCycles -= puller.queryTotalCycles(); // MORE - debatable... but probably fair.
- if (localCycles < 0)
- localCycles = 0;
- return localCycles;
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- if (idx==0)
- return puller.queryInput();
- else if (idx==1)
- return indexReadInput;
- else
- return NULL;
- }
- virtual void reset()
- {
- totalCycles.totalCycles = remote.totalCycles.totalCycles;
- remote.totalCycles.totalCycles = 0;
- processed = remote.joinProcessed;
- remote.joinProcessed = 0;
- defaultRight.clear();
- if (indexReadInput)
- indexReadInput->reset();
- if (atmostsTriggered)
- noteStatistic(StNumAtmostTriggered, atmostsTriggered);
- CRoxieServerActivity::reset();
- puller.reset();
- while (groups.ordinality())
- {
- ::Release(groups.dequeue());
- }
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- if (idx==(unsigned)-1)
- idx = 0;
- return idx ? NULL : &remote;
- }
- #undef new
- virtual CJoinGroup *createJoinGroup(const void *row)
- {
- // NOTE - we need to protect access to queue, since it's also modified by consumer thread. Groupstart is only modified by puller thread.
- CriticalBlock c(groupsCrit);
- if (preserveGroups && !groupStart)
- {
- groupStart = new (&ctx->queryRowManager(), activityId) CJoinGroup(NULL, NULL);
- groups.enqueue(groupStart);
- }
- CJoinGroup *jg = new (&ctx->queryRowManager(), activityId) CJoinGroup(row, groupStart);
- groups.enqueue(jg);
- return jg;
- }
- #if defined(_DEBUG) && defined(_WIN32) && !defined(USING_MPATROL)
- #define new new(_NORMAL_BLOCK, __FILE__, __LINE__)
- #endif
- void endGroup()
- {
- CriticalBlock c(groupsCrit);
- if (groupStart)
- noteEndReceived(groupStart, 0);
- groupStart = NULL;
- }
- virtual void noteEndReceived(CJoinGroup *jg, unsigned candidateCount)
- {
- if (jg->noteEndReceived(candidateCount))
- processCompletedGroups();
- }
- void processCompletedGroups()
- {
- CriticalBlock c(groupsCrit);
- while (groups.ordinality())
- {
- if (!groups.head()->complete())
- break;
- Owned<CJoinGroup> head = groups.dequeue();
- if (preserveGroups)
- {
- assert(head->isHeadRecord());
- assert(groups.head()->inGroup(head));
- unsigned joinGroupSize = 0;
- while (groups.ordinality() && groups.head()->inGroup(head))
- {
- Owned<CJoinGroup> finger = groups.dequeue();
- joinGroupSize += doJoinGroup(finger);
- }
- if (joinGroupSize)
- remote.addResult(NULL);
- }
- else
- doJoinGroup(head);
- }
- }
- void failLimit(const void * left)
- {
- helper.onMatchAbortLimitExceeded();
- CommonXmlWriter xmlwrite(0);
- if (input && input->queryOutputMeta() && input->queryOutputMeta()->hasXML())
- {
- input->queryOutputMeta()->toXML((byte *) left, xmlwrite);
- }
- throw MakeStringException(ROXIE_JOIN_ERROR, "More than %d match candidates in keyed join %d for row %s", abortLimit, queryId(), xmlwrite.str());
- }
- virtual bool needsAllocator() const { return true; }
- unsigned doTransform(const void *left, const void *right, offset_t fpos_or_count, IException *except, const void **group, unsigned counter)
- {
- if (cloneLeft && !except)
- {
- LinkRoxieRow(left);
- remote.addResult((void *) left);
- return 1;
- }
-
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- unsigned outSize;
- try
- {
- outSize = except ? helper.onFailTransform(rowBuilder, left, right, fpos_or_count, except) :
- (activityKind == TAKkeyeddenormalizegroup) ? helper.transform(rowBuilder, left, right, (unsigned) fpos_or_count, group) :
- helper.transform(rowBuilder, left, right, fpos_or_count, counter);
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- if (outSize)
- {
- const void *shrunk = rowBuilder.finalizeRowClear(outSize);
- remote.addResult(shrunk);
- return 1;
- }
- else
- return 0;
- }
- unsigned doJoinGroup(CJoinGroup *jg)
- {
- unsigned matched = jg->rowsSeen();
- unsigned added = 0;
- const void *left = jg->queryLeft();
- if (jg->candidateCount() > abortLimit)
- {
- if (limitFail)
- failLimit(left);
- if (ctx->queryDebugContext())
- ctx->queryDebugContext()->checkBreakpoint(DebugStateLimit, NULL, static_cast<IActivityBase *>(this));
- if (limitOnFail)
- {
- Owned<IException> except;
- try
- {
- failLimit(left);
- }
- catch(IException * e)
- {
- except.setown(e);
- }
- added = doTransform(left, defaultRight, 0, except, NULL, 0);
- }
- }
- else if (!matched || jg->candidateCount() > atMost)
- {
- if (jg->candidateCount() > atMost)
- atmostsTriggered++;
- switch (joinFlags & JFtypemask)
- {
- case JFleftouter:
- case JFleftonly:
- switch (activityKind)
- {
- case TAKkeyedjoin:
- case TAKkeyeddenormalizegroup:
- added = doTransform(left, defaultRight, 0, NULL, NULL, 0);
- break;
- case TAKkeyeddenormalize:
- LinkRoxieRow(left);
- remote.addResult((void *) left);
- added++;
- break;
- }
- break;
- }
- }
- else if (!(joinFlags & JFexclude))
- {
- unsigned idx = 0;
- switch (activityKind)
- {
- case TAKkeyedjoin:
- while (idx < matched)
- {
- const KeyedJoinHeader *rhs = jg->queryRow(idx);
- added += doTransform(left, &rhs->rhsdata, rhs->fpos, NULL, NULL, idx+1);
- if (added==keepLimit)
- break;
- idx++;
- }
- break;
- case TAKkeyeddenormalize:
- {
- OwnedConstRoxieRow newLeft;
- newLeft.set(left);
- unsigned rowSize = 0;
- unsigned rightAdded = 0;
- while (idx < matched)
- {
- const KeyedJoinHeader *rhs = jg->queryRow(idx);
- try
- {
- RtlDynamicRowBuilder rowBuilder(rowAllocator);
- size32_t transformedSize = helper.transform(rowBuilder, newLeft, &rhs->rhsdata, rhs->fpos, idx+1);
- if (transformedSize)
- {
- rowSize = transformedSize;
- newLeft.setown(rowBuilder.finalizeRowClear(rowSize));
- rightAdded++;
- if (rightAdded==keepLimit)
- break;
- }
- idx++;
- }
- catch (IException *E)
- {
- throw makeWrappedException(E);
- }
- }
- if (rowSize)
- {
- remote.addResult(newLeft.getClear());
- added++;
- }
- }
- break;
- case TAKkeyeddenormalizegroup:
- {
- ConstPointerArray extractedRows;
- while (idx < matched && idx < keepLimit)
- {
- const KeyedJoinHeader *rhs = jg->queryRow(idx);
- extractedRows.append((void *) &rhs->rhsdata);
- idx++;
- }
- added += doTransform(left, extractedRows.item(0), extractedRows.ordinality(), NULL, (const void * *)extractedRows.getArray(), 0);
- }
- break;
- }
- }
- return added;
- }
- virtual void processDone()
- {
- // called from puller thread
- remote.flush();
- remote.senddone();
- }
- virtual void onLimitExceeded(bool isKeyed)
- {
- if (traceLevel > 4)
- DBGLOG("activityid = %d isKeyed = %d line = %d", activityId, isKeyed, __LINE__);
- if (isKeyed)
- throwUnexpected();
- helper.onLimitExceeded();
- }
- virtual const void *createLimitFailRow(bool isKeyed)
- {
- throwUnexpected();
- }
- virtual bool fireException(IException *e)
- {
- // called from puller thread on failure
- return remote.fireException(e);
- }
- virtual const void *nextRow()
- {
- throwUnexpected(); // I am nobody's input
- }
- };
- #ifdef _MSC_VER
- #pragma warning ( push )
- #pragma warning ( disable: 4355 )
- #endif
- class CRoxieServerKeyedJoinActivity : public CRoxieServerKeyedJoinBase
- {
- CRoxieServerFullKeyedJoinHead head;
- Owned<IEngineRowAllocator> fetchInputAllocator;
- Linked<IFilePartMap> map;
- bool variableFetchFileName;
- Owned<const IResolvedFile> varFetchFileInfo;
- CachedOutputMetaData fetchInputFields;
- public:
- CRoxieServerKeyedJoinActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_headId, IKeyArray * _key, TranslatorArray *_translators, IOutputMetaData *_indexReadMeta,
- const RemoteActivityId &_tailId, IFilePartMap *_map, unsigned _joinFlags, bool _isLocal)
- : CRoxieServerKeyedJoinBase(_ctx, _factory, _probeManager, _tailId, _joinFlags, true, false, _isLocal),
- head(_ctx, _factory, _probeManager, _headId, _key, _translators, _indexReadMeta, this, _isLocal),
- map(_map)
- {
- CRoxieServerKeyedJoinBase::setInput(0, 0, head.queryOutput(0));
- variableFetchFileName = allFilesDynamic || factory->queryQueryFactory().isDynamic() || ((helper.getFetchFlags() & (FFvarfilename|FFdynamicfilename)) != 0);
- }
-
- virtual const IResolvedFile *queryVarFileInfo() const
- {
- return varFetchFileInfo;
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerKeyedJoinBase::onCreate(_colocalParent);
- head.onCreate(_colocalParent);
- fetchInputFields.set(helper.queryFetchInputRecordSize());
- fetchInputAllocator.setown(createRowAllocator(helper.queryFetchInputRecordSize()));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerKeyedJoinBase::start(parentExtractSize, parentExtract, paused);
- if (variableFetchFileName)
- {
- bool isFetchOpt = (helper.getFetchFlags() & FFdatafileoptional) != 0;
- OwnedRoxieString fname(helper.getFileName());
- varFetchFileInfo.setown(resolveLFN(fname, isFetchOpt));
- if (varFetchFileInfo)
- map.setown(varFetchFileInfo->getFileMap());
- }
- puller.start(parentExtractSize, parentExtract, paused, ctx->queryOptions().keyedJoinPreload, false, ctx);
- }
- virtual void setInput(unsigned idx, unsigned _sourceIdx, IFinalRoxieInput *_in)
- {
- head.setInput(idx, _sourceIdx, _in);
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- connectInputStreams(consumerOrdered);
- return head.getOutputStreams(ctx, idx, streams, consumerOptions, consumerOrdered, orderedCallbacks);
- }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- return head.queryInput(idx);
- }
- virtual void processRow(const void *_rhs)
- {
- // called from puller thread
- KeyedJoinHeader *rhs = (KeyedJoinHeader *) _rhs;
- CJoinGroup *jg = rhs->thisGroup;
- if (rhs->partNo != (unsigned short) -1)
- {
- unsigned partNo = map->mapOffset(rhs->fpos);
- RtlDynamicRowBuilder fetchBuilder(fetchInputAllocator, true);
- size32_t fisize = helper.extractFetchFields(fetchBuilder, jg->queryLeft());
- if (fetchInputFields.isVariableSize())
- {
- KeyedJoinHeader *outRow = (KeyedJoinHeader *) remote.getMem(partNo, 0, KEYEDJOIN_RECORD_SIZE(fisize + sizeof(fisize)));
- memcpy(outRow, rhs, KEYEDJOIN_RECORD_SIZE(0)); // MORE - copy constructor might be more appropriate....
- ReleaseRoxieRow(rhs);
- jg->notePending();
- memcpy(&outRow->rhsdata, &fisize, sizeof(fisize));
- memcpy((&outRow->rhsdata)+sizeof(fisize), fetchBuilder.row(), fisize);
- }
- else
- {
- KeyedJoinHeader *outRow = (KeyedJoinHeader *) remote.getMem(partNo, 0, KEYEDJOIN_RECORD_SIZE(fisize));
- memcpy(outRow, rhs, KEYEDJOIN_RECORD_SIZE(0)); // MORE - copy constructor might be more appropriate....
- ReleaseRoxieRow(rhs);
- jg->notePending();
- memcpy(&outRow->rhsdata, fetchBuilder.row(), fisize);
- }
- }
- else
- {
- unsigned candidateCount = (unsigned) rhs->fpos;
- // CTXLOG("Full keyed join - all results back from index");
- ReleaseRoxieRow(rhs);
- noteEndReceived(jg, candidateCount); // may throw exception - so release row before calling
- }
- }
- virtual void processEOG()
- {
- // called from front puller thread
- if (preserveGroups)
- endGroup();
- }
- void processGroup(const ConstPointerArray &)
- {
- throwUnexpected();
- }
- };
- #ifdef _MSC_VER
- #pragma warning ( pop )
- #endif
- class CRoxieServerHalfKeyedJoinActivity : public CRoxieServerKeyedJoinBase
- {
- IOutputMetaData *indexReadMeta;
- Owned<IEngineRowAllocator> indexReadAllocator;
- Owned<IKeyManager> tlk;
- bool variableIndexFileName;
- bool indexReadInputRecordVariable;
- Owned<const IResolvedFile> varFileInfo;
- Linked<TranslatorArray> translators;
- Linked<IKeyArray> keySet;
- Owned<IOutputMetaData> joinPrefixedMeta;
- Owned<IEngineRowAllocator> joinFieldsAllocator;
- public:
- CRoxieServerHalfKeyedJoinActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager, const RemoteActivityId &_remoteId, IKeyArray * _keySet, TranslatorArray *_translators,
- IOutputMetaData *_indexReadMeta, unsigned _joinFlags, bool _isSimple, bool _isLocal)
- : CRoxieServerKeyedJoinBase(_ctx, _factory, _probeManager, _remoteId, _joinFlags, false, _isSimple, _isLocal),
- indexReadMeta(_indexReadMeta),
- tlk(createKeyManager(NULL, 0, this)),
- keySet(_keySet),
- translators(_translators)
- {
- variableIndexFileName = allFilesDynamic || factory->queryQueryFactory().isDynamic() || ((helper.getJoinFlags() & (JFvarindexfilename|JFdynamicindexfilename|JFindexfromactivity)) != 0);
- indexReadInputRecordVariable = indexReadMeta->isVariableSize();
- }
- virtual void serializeExtra(MemoryBuffer &out)
- {
- if (helper.getJoinFlags() & JFindexfromactivity)
- {
- assertex(rootIndex);
- const RemoteActivityId& indexId = rootIndex->queryRemoteId();
- indexId.serialize(out);
- // could mess about reserving space for length then patching it again, to avoid copy, but probably not worth it
- MemoryBuffer tmp;
- rootIndex->queryActivity()->serializeCreateStartContext(tmp);
- if (rootIndex->queryActivity()->queryVarFileInfo())
- {
- rootIndex->queryActivity()->queryVarFileInfo()->queryTimeStamp().serialize(tmp);
- tmp.append(rootIndex->queryActivity()->queryVarFileInfo()->queryCheckSum());
- }
- unsigned ctxlen = tmp.length();
- out.append(ctxlen).append(tmp);
- }
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- CRoxieServerKeyedJoinBase::onCreate(_colocalParent);
- indexReadAllocator.setown(createRowAllocator(indexReadMeta));
- IOutputMetaData *joinFieldsMeta = helper.queryJoinFieldsRecordSize();
- joinPrefixedMeta.setown(new CPrefixedOutputMeta(KEYEDJOIN_RECORD_SIZE(0), joinFieldsMeta)); // MORE - not sure if we really need this
- joinFieldsAllocator.setown(createRowAllocator(joinPrefixedMeta));
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- CRoxieServerKeyedJoinBase::start(parentExtractSize, parentExtract, paused);
- if (rootIndex)
- {
- varFileInfo.setown(rootIndex->getVarFileInfo());
- translators.setown(rootIndex->getTranslators());
- keySet.setown(rootIndex->getKeySet());
- }
- else if (variableIndexFileName)
- {
- OwnedRoxieString indexFileName(helper.getIndexFileName());
- varFileInfo.setown(resolveLFN(indexFileName, (helper.getJoinFlags() & JFindexoptional) != 0));
- if (varFileInfo)
- {
- translators.setown(new TranslatorArray);
- keySet.setown(varFileInfo->getKeyArray(factory->queryActivityMeta(), translators, false, isLocal ? factory->queryQueryFactory().queryChannel() : 0, factory->queryQueryFactory().queryOptions().enableFieldTranslation));
- }
- }
- puller.start(parentExtractSize, parentExtract, paused, ctx->queryOptions().keyedJoinPreload, isSimple, ctx);
- }
- virtual const IResolvedFile *queryVarFileInfo() const
- {
- return varFileInfo;
- }
- virtual void reset()
- {
- CRoxieServerKeyedJoinBase::reset();
- if (varFileInfo)
- {
- keySet.clear();
- varFileInfo.clear();
- }
- }
- virtual void processRow(const void *row)
- {
- // called from front puller thread
- // buffer up an IndexRead request
- if (helper.leftCanMatch(row) && keySet)
- {
- RtlDynamicRowBuilder extractBuilder(indexReadAllocator);
- unsigned indexReadRecordSize = helper.extractIndexReadFields(extractBuilder, row);
- OwnedConstRoxieRow extracted;
- if (indexReadRecordSize)
- extracted.setown(extractBuilder.finalizeRowClear(indexReadRecordSize));
- CJoinGroup *jg = createJoinGroup(row);
- for (unsigned partNo = 0; partNo < keySet->length(); partNo++)
- {
- IKeyIndexBase *thisBase = keySet->queryKeyPart(partNo);
- if (thisBase)
- {
- unsigned fileNo = 0;
- IKeyIndex *thisKey = thisBase->queryPart(fileNo);
- tlk->setKey(thisKey);
- tlk->setLayoutTranslator(translators->item(fileNo));
- helper.createSegmentMonitors(tlk, extracted);
- if (rootIndex)
- rootIndex->mergeSegmentMonitors(tlk);
- tlk->finishSegmentMonitors();
- try
- {
- tlk->reset();
- loop
- {
- typedef const void * cvp;
- if (thisKey && thisKey->isTopLevelKey())
- {
- bool locallySorted = (!thisKey->isFullySorted());
- while (locallySorted || tlk->lookup(false))
- {
- unsigned slavePart = locallySorted ? 0 : (unsigned) tlk->queryFpos();
- if (locallySorted || slavePart)
- {
- cvp *outputBuffer = (cvp *) remote.getMem(slavePart, fileNo, indexReadRecordSize + sizeof(cvp) + (indexReadInputRecordVariable ? sizeof(unsigned) : 0));
- *outputBuffer++ = jg;
- if (indexReadInputRecordVariable)
- {
- *(unsigned *) outputBuffer = indexReadRecordSize;
- outputBuffer = (cvp *) (((unsigned *) outputBuffer) + 1);
- }
- jg->notePending();
- memcpy(outputBuffer, extracted, indexReadRecordSize);
- if (locallySorted)
- {
- for (unsigned i = 1; i < numChannels; i++)
- jg->notePending();
- break;
- }
- }
- }
- }
- else
- {
- unsigned accepted = 0;
- unsigned rejected = 0;
- Owned<CRowArrayMessageResult> result;
- if (!isSimple)
- result.setown(new CRowArrayMessageResult(ctx->queryRowManager(), true));
- // MORE - This code seems to be duplicated in keyedJoinHead
- jg->notePending();
- unsigned candidateCount = 0;
- while (tlk->lookup(true))
- {
- candidateCount++;
- atomic_inc(&indexRecordsRead);
- KLBlobProviderAdapter adapter(tlk);
- offset_t recptr;
- const byte *indexRow = tlk->queryKeyBuffer(recptr);
- if (helper.indexReadMatch(extracted, indexRow, recptr, &adapter))
- {
- RtlDynamicRowBuilder rb(joinFieldsAllocator, true);
- CPrefixedRowBuilder pb(KEYEDJOIN_RECORD_SIZE(0), rb);
- accepted++;
- KLBlobProviderAdapter adapter(tlk);
- size32_t joinFieldsSize = helper.extractJoinFields(pb, indexRow, recptr, &adapter);
- KeyedJoinHeader *rec = (KeyedJoinHeader *) rb.getUnfinalizedClear(); // lack of finalize ok as unserialized data here.
- rec->fpos = recptr;
- rec->thisGroup = jg;
- rec->partNo = partNo;
- if (isSimple)
- remote.injected.enqueue(rec);
- else
- result->append(rec);
- }
- else
- {
- rejected++;
- atomic_inc(&postFiltered);
- }
- }
- // output an end marker for the matches to this group
- KeyedJoinHeader *rec = (KeyedJoinHeader *) ctx->queryRowManager().allocate(KEYEDJOIN_RECORD_SIZE(0), activityId);
- rec->fpos = (offset_t) candidateCount;
- rec->thisGroup = jg;
- rec->partNo = (unsigned short) -1;
- if (isSimple)
- remote.injected.enqueue(rec);
- else
- {
- result->append(rec);
- remote.injectResult(result.getClear());
- }
- if (accepted)
- noteStatistic(StNumIndexAccepted, accepted);
- if (rejected)
- noteStatistic(StNumIndexRejected, rejected);
- }
- if (++fileNo < thisBase->numParts())
- {
- thisKey = thisBase->queryPart(fileNo);
- tlk->setKey(thisKey);
- tlk->setLayoutTranslator(translators->item(fileNo));
- tlk->reset();
- }
- else
- break;
- }
- tlk->releaseSegmentMonitors();
- tlk->setKey(NULL);
- }
- catch (...)
- {
- tlk->releaseSegmentMonitors();
- tlk->setKey(NULL);
- throw;
- }
- }
- }
- noteEndReceived(jg, 0);
- }
- else
- {
- noteEndReceived(createJoinGroup(row), 0);
- }
- }
- virtual void processEOG()
- {
- // called from front puller thread
- if (preserveGroups)
- endGroup();
- }
- void processGroup(const ConstPointerArray &)
- {
- throwUnexpected();
- }
- };
- class CRoxieServerKeyedJoinActivityFactory : public CRoxieServerMultiInputFactory
- {
- Owned<const IResolvedFile> indexfile;
- Owned<const IResolvedFile> datafile;
- Owned<IKeyArray> keySet;
- Owned<TranslatorArray> translatorArray;
- Owned<IDefRecordMeta> activityMeta;
- RemoteActivityId headId;
- RemoteActivityId tailId;
- IOutputMetaData *indexReadMeta;
- Owned<IFilePartMap> map;
- Owned<IFileIOArray> files;
- unsigned joinFlags;
- bool isHalfKeyed;
- bool isLocal;
- bool enableFieldTranslation;
- bool variableFetchFileName;
- bool variableIndexFileName;
- bool isSimple;
- public:
- CRoxieServerKeyedJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_headId, const RemoteActivityId &_tailId)
- : CRoxieServerMultiInputFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), headId(_headId), tailId(_tailId)
- {
- Owned<IHThorKeyedJoinArg> helper = (IHThorKeyedJoinArg *) helperFactory();
- isLocal = _graphNode.getPropBool("att[@name='local']/@value") && queryFactory.queryChannel()!=0;
- isSimple = isLocal;
- rtlDataAttr indexLayoutMeta;
- size32_t indexLayoutSize;
- if(!helper->getIndexLayout(indexLayoutSize, indexLayoutMeta.refdata()))
- assertex(indexLayoutSize== 0);
- MemoryBuffer m;
- m.setBuffer(indexLayoutSize, indexLayoutMeta.getdata());
- activityMeta.setown(deserializeRecordMeta(m, true));
- enableFieldTranslation = queryFactory.queryOptions().enableFieldTranslation;
- translatorArray.setown(new TranslatorArray);
- joinFlags = helper->getJoinFlags();
- variableIndexFileName = allFilesDynamic || _queryFactory.isDynamic() || ((joinFlags & (JFvarindexfilename|JFdynamicindexfilename|JFindexfromactivity)) != 0);
- variableFetchFileName = allFilesDynamic || _queryFactory.isDynamic() || ((helper->getFetchFlags() & (FFvarfilename|FFdynamicfilename)) != 0);
- if (!variableIndexFileName)
- {
- bool isOpt = (joinFlags & JFindexoptional) != 0;
- OwnedRoxieString indexFileName(helper->getIndexFileName());
- indexfile.setown(queryFactory.queryPackage().lookupFileName(indexFileName, isOpt, true, true, queryFactory.queryWorkUnit(), true));
- if (indexfile)
- keySet.setown(indexfile->getKeyArray(activityMeta, translatorArray, isOpt, isLocal ? queryFactory.queryChannel() : 0, enableFieldTranslation));
- }
- if (keySet && keySet->length()==1 && !isSimple)
- {
- IKeyIndexBase *thisBase = keySet->queryKeyPart(0);
- if (thisBase->numParts()==1 && !thisBase->queryPart(0)->isTopLevelKey() && !_queryFactory.queryOptions().disableLocalOptimizations)
- isSimple = true;
- // MORE - if it's a variable filename then it MAY be simple, we don't know. Tough.
- }
- if (!simpleLocalKeyedJoins)
- isSimple = false;
- isHalfKeyed = !helper->diskAccessRequired();
- indexReadMeta = QUERYINTERFACE(helper->queryIndexReadInputRecordSize(), IOutputMetaData);
- if (!isHalfKeyed && !variableFetchFileName)
- {
- bool isFetchOpt = (helper->getFetchFlags() & FFdatafileoptional) != 0;
- datafile.setown(_queryFactory.queryPackage().lookupFileName(queryNodeFileName(_graphNode, _kind), isFetchOpt, true, true, _queryFactory.queryWorkUnit(), true));
- if (datafile)
- {
- if (isLocal)
- files.setown(datafile->getIFileIOArray(isFetchOpt, queryFactory.queryChannel()));
- else
- map.setown(datafile->getFileMap());
- }
- }
- }
- virtual bool getEnableFieldTranslation() const
- {
- return enableFieldTranslation;
- }
- virtual IDefRecordMeta *queryActivityMeta() const
- {
- return activityMeta;
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- if (isHalfKeyed)
- return new CRoxieServerHalfKeyedJoinActivity(_ctx, this, _probeManager,
- headId, keySet, translatorArray, indexReadMeta, joinFlags, isSimple, isLocal);
- else
- return new CRoxieServerKeyedJoinActivity(_ctx, this, _probeManager,
- headId, keySet, translatorArray, indexReadMeta,
- tailId, map, joinFlags, isLocal);
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return keyedJoinStatistics;
- }
- virtual void getXrefInfo(IPropertyTree &reply, const IRoxieContextLogger &logctx) const
- {
- if (datafile)
- addXrefFileInfo(reply, datafile);
- if (indexfile)
- addXrefFileInfo(reply, indexfile);
- }
- };
- IRoxieServerActivityFactory *createRoxieServerKeyedJoinActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, const RemoteActivityId &_remoteId, const RemoteActivityId &_remoteId2)
- {
- return new CRoxieServerKeyedJoinActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _remoteId, _remoteId2);
- }
- //=================================================================================
- class CRoxieServerSoapActivityBase : public CRoxieServerActivity, implements ISoapCallRowProvider, implements IRoxieAbortMonitor
- {
- protected:
- Owned<ISoapCallHelper> soaphelper;
- IHThorSoapActionArg & helper;
- StringBuffer authToken;
- bool eof;
- CriticalSection crit;
- ClientCertificate *pClientCert;
- public:
- IMPLEMENT_IINTERFACE_USING(CRoxieServerActivity)
- CRoxieServerSoapActivityBase(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerActivity(_ctx, _factory, _probeManager), helper((IHThorSoapActionArg &)basehelper)
- {
- eof = false;
- if (clientCert.certificate.length() > 0 && clientCert.privateKey.length() > 0 && clientCert.passphrase.length() > 0)
- pClientCert = &clientCert;
- else
- pClientCert = NULL;
- }
- // ISoapCallRowProvider
- virtual IHThorSoapActionArg * queryActionHelper() { return &helper; };
- virtual IHThorSoapCallArg * queryCallHelper() { return NULL; };
- virtual const void * getNextRow() { return NULL; };
- virtual void releaseRow(const void * r) { ReleaseRoxieRow(r); };
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- eof = false;
- CRoxieServerActivity::start(parentExtractSize, parentExtract, paused);
- authToken.append(ctx->queryAuthToken());
- }
- virtual void reset()
- {
- // MORE - Shouldn't we make sure thread is stopped etc???
- soaphelper.clear();
- CRoxieServerActivity::reset();
- }
- // IRoxieAbortMonitor
- virtual void checkForAbort() { checkAbort(); }
- };
- //---------------------------------------------------------------------------
- class CRoxieServerSoapRowCallActivity : public CRoxieServerSoapActivityBase
- {
- IHThorSoapCallArg & callHelper;
- public:
- CRoxieServerSoapRowCallActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerSoapActivityBase(_ctx, _factory, _probeManager), callHelper((IHThorSoapCallArg &)basehelper)
- {
- }
- virtual IHThorSoapCallArg * queryCallHelper()
- {
- return &callHelper;
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if(eof) return NULL;
- if (soaphelper == NULL)
- {
- if (factory->getKind()==TAKhttp_rowdataset)
- soaphelper.setown(createHttpCallHelper(this, rowAllocator, authToken.str(), SCrow, pClientCert, *this, this));
- else
- soaphelper.setown(createSoapCallHelper(this, rowAllocator, authToken.str(), SCrow, pClientCert, *this, this));
- soaphelper->start();
- }
- OwnedConstRoxieRow ret = soaphelper->getRow();
- if (!ret)
- {
- eof = true;
- return NULL;
- }
- ++processed;
- return ret.getClear();
- }
- };
- class CRoxieServerSoapRowCallActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerSoapRowCallActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSoapRowCallActivity(_ctx, this, _probeManager);
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return soapStatistics;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSoapRowCallActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerSoapRowCallActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //---------------------------------------------------------------------------
- class CRoxieServerSoapRowActionActivity : public CRoxieServerSoapActivityBase
- {
- public:
- CRoxieServerSoapRowActionActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerSoapActivityBase(_ctx, _factory, _probeManager)
- {}
- virtual void execute(unsigned parentExtractSize, const byte * parentExtract)
- {
- //MORE: parentExtract not passed to start - although shouldn't be a problem.
- soaphelper.setown(createSoapCallHelper(this, NULL, ctx->queryAuthToken(), SCrow, pClientCert, *this, this));
- soaphelper->start();
- soaphelper->waitUntilDone();
- IException *e = soaphelper->getError();
- soaphelper.clear();
- if (e)
- throw e;
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- return NULL;
- }
- virtual const void *nextRow()
- {
- throwUnexpected(); // I am nobody's input
- }
- };
- class CRoxieServerSoapRowActionActivityFactory : public CRoxieServerActivityFactory
- {
- bool isRoot;
- public:
- CRoxieServerSoapRowActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), isRoot(_isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSoapRowActionActivity(_ctx, this, _probeManager);
- }
- virtual bool isSink() const
- {
- return isRoot;
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return soapStatistics;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSoapRowActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerSoapRowActionActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //---------------------------------------------------------------------------
- class CRoxieServerSoapDatasetCallActivity : public CRoxieServerSoapActivityBase
- {
- IHThorSoapCallArg & callHelper;
- public:
- CRoxieServerSoapDatasetCallActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerSoapActivityBase(_ctx, _factory, _probeManager), callHelper((IHThorSoapCallArg &)basehelper)
- {
- }
- virtual IHThorSoapCallArg * queryCallHelper()
- {
- return &callHelper;
- }
- virtual const void *getNextRow()
- {
- CriticalBlock b(crit); // MORE - why ?
- return inputStream->ungroupedNextRow();
- }
- virtual bool needsAllocator() const { return true; }
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, timeActivities);
- if(eof) return NULL;
- if (soaphelper == NULL)
- {
- soaphelper.setown(createSoapCallHelper(this, rowAllocator, authToken.str(), SCdataset, pClientCert, *this, this));
- soaphelper->start();
- }
- OwnedConstRoxieRow ret = soaphelper->getRow();
- if (!ret)
- {
- eof = true;
- return NULL;
- }
- ++processed;
- return ret.getClear();
- }
- };
- class CRoxieServerSoapDatasetCallActivityFactory : public CRoxieServerActivityFactory
- {
- public:
- CRoxieServerSoapDatasetCallActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSoapDatasetCallActivity(_ctx, this, _probeManager);
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return soapStatistics;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSoapDatasetCallActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode)
- {
- return new CRoxieServerSoapDatasetCallActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode);
- }
- //---------------------------------------------------------------------------
- class CRoxieServerSoapDatasetActionActivity : public CRoxieServerSoapActivityBase
- {
- public:
- CRoxieServerSoapDatasetActionActivity(IRoxieSlaveContext *_ctx, const IRoxieServerActivityFactory *_factory, IProbeManager *_probeManager)
- : CRoxieServerSoapActivityBase(_ctx, _factory, _probeManager)
- {}
- virtual const void *getNextRow()
- {
- CriticalBlock b(crit); // MORE - Why?
- const void *nextrec = inputStream->ungroupedNextRow();
- if (nextrec)
- processed++;
- return nextrec;
- }
- virtual void execute(unsigned parentExtractSize, const byte * parentExtract)
- {
- try
- {
- start(parentExtractSize, parentExtract, false);
- soaphelper.setown(createSoapCallHelper(this, NULL, ctx->queryAuthToken(), SCdataset, pClientCert, *this, this));
- soaphelper->start();
- soaphelper->waitUntilDone();
- IException *e = soaphelper->getError();
- soaphelper.clear();
- if (e)
- throw e;
- stop();
- }
- catch (IException *E)
- {
- ctx->notifyAbort(E);
- abort();
- throw;
- }
- catch(...)
- {
- Owned<IException> E = MakeStringException(ROXIE_INTERNAL_ERROR, "Unknown exception caught at %s:%d", __FILE__, __LINE__);
- ctx->notifyAbort(E);
- abort();
- throw;
- }
- }
- virtual IFinalRoxieInput *queryOutput(unsigned idx)
- {
- return NULL;
- }
- virtual const void *nextRow()
- {
- throwUnexpected(); // I am nobody's input
- }
- };
- class CRoxieServerSoapDatasetActionActivityFactory : public CRoxieServerActivityFactory
- {
- bool isRoot;
- public:
- CRoxieServerSoapDatasetActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- : CRoxieServerActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode), isRoot(_isRoot)
- {
- }
- virtual IRoxieServerActivity *createActivity(IRoxieSlaveContext *_ctx, IProbeManager *_probeManager) const
- {
- return new CRoxieServerSoapDatasetActionActivity(_ctx, this, _probeManager);
- }
- virtual bool isSink() const
- {
- return isRoot;
- }
- virtual const StatisticsMapping &queryStatsMapping() const
- {
- return soapStatistics;
- }
- };
- IRoxieServerActivityFactory *createRoxieServerSoapDatasetActionActivityFactory(unsigned _id, unsigned _subgraphId, IQueryFactory &_queryFactory, HelperFactory *_helperFactory, ThorActivityKind _kind, IPropertyTree &_graphNode, bool _isRoot)
- {
- return new CRoxieServerSoapDatasetActionActivityFactory(_id, _subgraphId, _queryFactory, _helperFactory, _kind, _graphNode, _isRoot);
- }
- //=====================================================================================================
- class CGraphResults : implements IRoxieGraphResults, public CInterface
- {
- IArrayOf<IGraphResult> results;
- CriticalSection cs;
- IGraphResult & select(unsigned idx)
- {
- CriticalBlock procedure(cs);
- if (idx >= results.ordinality())
- throw MakeStringException(ROXIE_GRAPH_PROCESSING_ERROR, "Error reading graph result %d before it is calculated", idx);
- return results.item(idx);
- }
- public:
- IMPLEMENT_IINTERFACE
- void clear()
- {
- CriticalBlock procedure(cs);
- results.kill();
- }
- virtual IEngineRowStream * createIterator(unsigned id)
- {
- return select(id).createIterator();
- }
- virtual void getLinkedResult(unsigned & count, byte * * & ret, unsigned id)
- {
- select(id).getLinkedResult(count, ret);
- }
- virtual void getDictionaryResult(unsigned & count, byte * * & ret, unsigned id)
- {
- select(id).getLinkedResult(count, ret);
- }
- virtual const void * getLinkedRowResult(unsigned id)
- {
- return select(id).getLinkedRowResult();
- }
- void setResult(unsigned id, IGraphResult * result)
- {
- CriticalBlock procedure(cs);
- if (results.ordinality() <= id)
- {
- while (results.ordinality() < id)
- results.append(*new CGraphResult);
- results.append(*LINK(result));
- }
- else
- results.replace(*LINK(result), id);
- }
- void appendResult(IGraphResult * result)
- {
- CriticalBlock procedure(cs);
- results.append(*LINK(result));
- }
- };
- //===================================================================================================================
- class CPseudoArg : implements IHThorArg, public CInterface
- {
- public:
- IMPLEMENT_IINTERFACE
- virtual IOutputMetaData * queryOutputMeta() { return NULL; }
- };
- class CPseudoActivity : public CRoxieServerActivity
- {
- public:
- CPseudoActivity(IRoxieSlaveContext *_ctx, IHThorArg & _helper) : CRoxieServerActivity(_ctx, _helper) {}
- virtual const void *nextRow()
- {
- throwUnexpected(); // I am nobody's input
- }
- };
- class CActivityGraph : implements IActivityGraph, implements IThorChildGraph, implements ILocalGraphEx, implements IRoxieServerChildGraph, public CInterface
- {
- protected:
- class ActivityGraphSlaveContext : public IndirectSlaveContext
- {
- SpinLock abortLock;
- bool aborted;
- Owned<IException> exception;
- public:
- ActivityGraphSlaveContext(IRoxieSlaveContext *_ctx, const IRoxieContextLogger &_logctx) : IndirectSlaveContext(_ctx), logctx(_logctx), loopCounter(0), codeContext(NULL)
- {
- aborted = false;
- }
- // Note - we must track exceptions at the child level in case there is a CATCH in parent
- virtual void notifyAbort(IException *E)
- {
- SpinBlock b(abortLock);
- if (!aborted && QUERYINTERFACE(E, InterruptedSemaphoreException) == NULL)
- {
- aborted = true;
- exception.set(E);
- }
- }
- virtual void checkAbort()
- {
- if (aborted) // NOTE - don't bother getting lock before reading this (for speed) - a false read is very unlikely and not a problem
- {
- SpinBlock b(abortLock);
- if (!exception)
- exception.setown(MakeStringException(ROXIE_INTERNAL_ERROR, "Query was aborted"));
- throw exception.getLink();
- }
- IndirectSlaveContext::checkAbort();
- }
- virtual ICodeContext *queryCodeContext()
- {
- return codeContext;
- }
- void setCodeContext(ICodeContext * _codeContext)
- {
- codeContext = _codeContext;
- }
- void setLoopCounter(unsigned _loopCounter)
- {
- loopCounter = _loopCounter;
- }
- virtual void noteChildGraph(unsigned id, IActivityGraph *childGraph)
- {
- childGraphs.setValue(id, childGraph);
- }
- virtual IActivityGraph * queryChildGraph(unsigned id)
- {
- if (queryTraceLevel() > 10)
- CTXLOG("resolveChildGraph %d", id);
- IActivityGraph *childGraph = childGraphs.getValue(id);
- assertex(childGraph);
- return childGraph;
- }
- // MORE should really redirect the other log context ones too (though mostly doesn't matter). Really should refactor to have a queryLogContext() method in IRoxieSlaveContext I think
- virtual StringBuffer &getLogPrefix(StringBuffer &ret) const
- {
- logctx.getLogPrefix(ret);
- if (loopCounter)
- ret.appendf("{%u}", loopCounter);
- return ret;
- }
- protected:
- const IRoxieContextLogger &logctx;
- unsigned loopCounter;
- ICodeContext * codeContext;
- MapXToMyClass<unsigned, unsigned, IActivityGraph> childGraphs;
- } graphSlaveContext;
- class ActivityGraphCodeContext : public IndirectCodeContext
- {
- public:
- virtual IEclGraphResults * resolveLocalQuery(__int64 activityId)
- {
- if ((unsigned) activityId == container->queryId())
- return container;
- IActivityGraph * match = slaveContext->queryChildGraph((unsigned) activityId);
- if (match)
- return match->queryLocalGraph();
- return IndirectCodeContext::resolveLocalQuery(activityId);
- }
- virtual IThorChildGraph * resolveChildQuery(__int64 activityId, IHThorArg * colocal)
- {
- IActivityGraph * match = slaveContext->queryChildGraph((unsigned) activityId);
- return LINK(match->queryChildGraph());
- }
- virtual unsigned getGraphLoopCounter() const
- {
- return container->queryLoopCounter(); // only called if value is valid
- }
- void setContainer(IRoxieSlaveContext * _slaveContext, CActivityGraph * _container)
- {
- slaveContext = _slaveContext;
- container = _container;
- }
- protected:
- IRoxieSlaveContext * slaveContext;
- CActivityGraph * container;
- } graphCodeContext;
- // NOTE - destructor order is significant - need to destroy graphCodeContext and graphSlaveContext last
- IArrayOf<IRoxieServerActivity> activities;
- IArrayOf<IRoxieProbe> probes;
- IRoxieServerActivityCopyArray sinks;
- StringAttr graphName;
- Owned<CGraphResults> results;
- CGraphResults graphLoopResults;
- ActivityArray & graphDefinition;
- CriticalSection evaluateCrit;
- IProbeManager *probeManager;
- unsigned id;
- unsigned loopCounter;
- public:
- IMPLEMENT_IINTERFACE;
- CActivityGraph(IRoxieSlaveContext *_ctx, const char *_graphName, unsigned _id, ActivityArray &x, IProbeManager *_probeManager, const IRoxieContextLogger &_logctx)
- : probeManager(_probeManager), graphDefinition(x), graphName(_graphName), graphSlaveContext(_ctx, _logctx)
- {
- id = x.getLibraryGraphId();
- if (!id)
- id = _id;
- loopCounter = 0;
- graphSlaveContext.setCodeContext(&graphCodeContext);
- graphCodeContext.setContainer(&graphSlaveContext, this);
- }
- ~CActivityGraph()
- {
- if (probeManager)
- probeManager->deleteGraph((IArrayOf<IActivityBase>*)&activities, (IArrayOf<IInputBase>*)&probes);
- }
- virtual const char *queryName() const
- {
- return graphName.get();
- }
- virtual IRoxieServerActivity *queryActivity(unsigned _activityId)
- {
- unsigned idx = graphDefinition.recursiveFindActivityIndex(_activityId);
- if (idx==NotFound)
- return nullptr;
- assertex(activities.isItem(idx));
- IRoxieServerActivity *activity = &activities.item(idx);
- if (activity->queryId() == _activityId)
- return activity;
- return activity->queryChildActivity(_activityId);
- }
- void createGraph(IRoxieSlaveContext *_ctx)
- {
- if (graphDefinition.isMultiInstance())
- {
- graphCodeContext.set(_ctx->queryCodeContext());
- _ctx = &graphSlaveContext;
- }
- ForEachItemIn(idx, graphDefinition)
- {
- IRoxieServerActivityFactory &donor = graphDefinition.serverItem(idx);
- IRoxieServerActivity &activity = *donor.createActivity(_ctx, probeManager);
- activities.append(activity);
- if (donor.isSink())
- {
- sinks.append(activity);
- if (probeManager)
- probeManager->noteSink(&activity);
- }
- }
- ForEachItemIn(idx1, graphDefinition)
- {
- IRoxieServerActivityFactory &donor = graphDefinition.serverItem(idx1);
- IRoxieServerActivity &activity = activities.item(idx1);
- unsigned inputidx = 0;
- loop
- {
- unsigned outputidx;
- unsigned source = donor.getInput(inputidx, outputidx);
- if (source==(unsigned) -1)
- break;
- connectInput(idx1, inputidx, source, outputidx, 0);
- inputidx++;
- }
- IntArray &dependencies = donor.queryDependencies();
- IntArray &dependencyIndexes = donor.queryDependencyIndexes();
- IntArray &dependencyControlIds = donor.queryDependencyControlIds();
- StringArray &dependencyEdgeIds = donor.queryDependencyEdgeIds();
- ForEachItemIn(idx2, dependencies)
- {
- IRoxieServerActivity &dependencySourceActivity = activities.item(dependencies.item(idx2));
- unsigned dependencySourceIndex = dependencyIndexes.item(idx2);
- unsigned dependencyControlId = dependencyControlIds.item(idx2);
- activity.addDependency(dependencySourceActivity, dependencySourceIndex, dependencyControlId);
- if (probeManager)
- probeManager->noteDependency( &dependencySourceActivity, dependencySourceIndex, dependencyControlId, dependencyEdgeIds.item(idx2), &activity);
- }
- }
- ForEachItemIn(idx2, sinks)
- {
- IRoxieServerActivity &sink = sinks.item(idx2);
- sink.connectInputStreams(true);
- }
- }
- void connectInput(unsigned target, unsigned targetIdx, unsigned source, unsigned sourceIdx, unsigned iteration)
- {
- IRoxieServerActivity &targetActivity = activities.item(target);
- IRoxieServerActivity &sourceActivity = activities.item(source);
- IFinalRoxieInput * output = sourceActivity.queryOutput(sourceIdx);
- if (probeManager)
- {
- IRoxieProbe * inputProbe = probeManager->createProbe(static_cast<IInputBase*>(output), &sourceActivity, &targetActivity, sourceIdx, targetIdx, iteration);
- probes.append(*LINK(inputProbe));
- output = &dynamic_cast<IFinalRoxieInput &>(inputProbe->queryInput());
- }
- targetActivity.setInput(targetIdx, sourceIdx, output);
- }
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- ForEachItemIn(idx, activities)
- {
- IRoxieServerActivity *activity = &activities.item(idx);
- if (activity)
- activity->onCreate(_colocalParent);
- }
- }
- virtual void abort()
- {
- ForEachItemIn(idx, sinks)
- {
- IRoxieServerActivity &sink = sinks.item(idx);
- sink.abort();
- }
- }
- virtual void reset()
- {
- ForEachItemIn(idx, sinks)
- {
- IRoxieServerActivity &sink = sinks.item(idx);
- sink.reset();
- }
- }
- Linked<IException> exception;
- CriticalSection eCrit;
- virtual void noteException(IException *E)
- {
- CriticalBlock b(eCrit);
- if (!exception)
- {
- if (graphSlaveContext.queryDebugContext())
- {
- graphSlaveContext.queryDebugContext()->checkBreakpoint(DebugStateException, NULL, exception);
- }
- exception.set(E);
- }
- }
- virtual void checkAbort()
- {
- CriticalBlock b(eCrit);
- if (exception)
- throw exception.getLink();
- }
- virtual void execute()
- {
- results.setown(new CGraphResults);
- doExecute(0, NULL);
- }
- //New child query code...
- virtual IThorChildGraph * queryChildGraph()
- {
- return this;
- }
- virtual IEclGraphResults * queryLocalGraph()
- {
- return this;
- }
- virtual IRoxieServerChildGraph * queryLoopGraph()
- {
- return this;
- }
- inline unsigned queryId() const
- {
- return id;
- }
- inline unsigned queryLoopCounter() const
- {
- return loopCounter;
- }
- void doExecute(unsigned parentExtractSize, const byte * parentExtract)
- {
- if (sinks.ordinality()==1)
- sinks.item(0).execute(parentExtractSize, parentExtract);
- #ifdef PARALLEL_EXECUTE
- else if (!probeManager && !graphDefinition.isSequential())
- {
- class casyncfor: public CAsyncFor
- {
- public:
- IActivityGraph &parent;
- unsigned parentExtractSize;
- const byte * parentExtract;
- casyncfor(IRoxieServerActivityCopyArray &_sinks, IActivityGraph &_parent, unsigned _parentExtractSize, const byte * _parentExtract) :
- sinks(_sinks), parent(_parent), parentExtractSize(_parentExtractSize), parentExtract(_parentExtract) { }
- void Do(unsigned i)
- {
- try
- {
- sinks.item(i).execute(parentExtractSize, parentExtract);
- }
- catch (IException *E)
- {
- parent.noteException(E);
- throw;
- }
- }
- private:
- IRoxieServerActivityCopyArray &sinks;
- } afor(sinks, *this, parentExtractSize, parentExtract);
- afor.For(sinks.ordinality(), sinks.ordinality());
- }
- #endif
- else
- {
- ForEachItemIn(idx, sinks)
- {
- IRoxieServerActivity &sink = sinks.item(idx);
- sink.execute(parentExtractSize, parentExtract);
- }
- }
- }
- virtual IEclGraphResults *evaluate(unsigned parentExtractSize, const byte * parentExtract)
- {
- CriticalBlock block(evaluateCrit);
- results.setown(new CGraphResults);
- try
- {
- doExecute(parentExtractSize, parentExtract);
- }
- catch (...)
- {
- DBGLOG("Exception thrown in child query - cleaning up");
- reset();
- throw;
- }
- reset();
- return results.getClear();
- }
- //interface IRoxieServerChildGraph
- virtual void beforeExecute()
- {
- results.setown(new CGraphResults);
- }
- virtual IFinalRoxieInput * startOutput(unsigned id, unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- IFinalRoxieInput * ret = selectOutput(id);
- ret->start(parentExtractSize, parentExtract, paused);
- return ret;
- }
- virtual IFinalRoxieInput * selectOutput(unsigned id)
- {
- ForEachItemIn(i, sinks)
- {
- IFinalRoxieInput * ret = sinks.item(i).querySelectOutput(id);
- if (ret)
- return ret;
- }
- throwUnexpected();
- return NULL;
- }
- virtual void setInputResult(unsigned id, IGraphResult * result)
- {
- results->setResult(id, result);
- }
- virtual bool querySetInputResult(unsigned id, unsigned _sourceIdx, IFinalRoxieInput * input)
- {
- ForEachItemIn(i, activities)
- {
- if (activities.item(i).querySetStreamInput(id, _sourceIdx, input))
- return true;
- }
- return false;
- }
- virtual void stopUnusedOutputs()
- {
- //Hmm not sure how to do this...
- }
- virtual void afterExecute()
- {
- ForEachItemIn(i, sinks)
- {
- sinks.item(i).stop();
- }
- if (graphSlaveContext.queryDebugContext())
- {
- graphSlaveContext.queryDebugContext()->checkBreakpoint(DebugStateGraphFinished, NULL, NULL);
- }
- reset();
- }
- virtual IRoxieGraphResults * execute(size32_t parentExtractSize, const byte *parentExtract)
- {
- doExecute(parentExtractSize, parentExtract);
- return LINK(results);
- }
- virtual void getLinkedResult(unsigned & count, byte * * & ret, unsigned id)
- {
- results->getLinkedResult(count, ret, id);
- }
- virtual void getDictionaryResult(unsigned & count, byte * * & ret, unsigned id)
- {
- results->getLinkedResult(count, ret, id);
- }
- virtual const void * getLinkedRowResult(unsigned id)
- {
- return results->getLinkedRowResult(id);
- }
- virtual void setResult(unsigned id, IGraphResult * result)
- {
- results->setResult(id, result);
- }
- virtual IEngineRowStream * createResultIterator(unsigned id)
- {
- return results->createIterator(id);
- }
- virtual void setGraphLoopResult(IGraphResult * result)
- {
- graphLoopResults.appendResult(result);
- }
- virtual IEngineRowStream * createGraphLoopResultIterator(unsigned id)
- {
- try
- {
- return graphLoopResults.createIterator(id);
- }
- catch (IException * e)
- {
- e->Release();
- throw MakeStringException(ROXIE_GRAPH_PROCESSING_ERROR, "Error reading graph result %d before it is calculated", id);
- }
- }
- virtual void clearGraphLoopResults()
- {
- graphLoopResults.clear();
- }
- virtual void executeGraphLoop(size32_t parentExtractSize, const byte *parentExtract)
- {
- doExecute(parentExtractSize, parentExtract);
- }
- virtual void setGraphLoopResult(unsigned id, IGraphResult * result)
- {
- graphLoopResults.setResult(id, result);
- }
- virtual IEngineRowStream * getGraphLoopResult(unsigned id)
- {
- return graphLoopResults.createIterator(id);
- }
- virtual void getProbeResponse(IPropertyTree *query)
- {
- if (probeManager)
- probeManager->getProbeResponse(query);
- }
- virtual IRoxieServerChildGraph * createGraphLoopInstance(IRoxieSlaveContext *ctx, unsigned loopCounter, unsigned parentExtractSize, const byte * parentExtract, const IRoxieContextLogger &logctx)
- {
- throwUnexpected();
- }
- virtual CGraphIterationInfo *selectGraphLoopOutput()
- {
- return NULL;
- }
- virtual void gatherIterationUsage(IRoxieServerLoopResultProcessor & processor)
- {
- throwUnexpected();
- }
- virtual void associateIterationOutputs(IRoxieServerLoopResultProcessor & processor)
- {
- throwUnexpected();
- }
- };
- class CProxyActivityGraph : implements IActivityGraph, implements IThorChildGraph, public CInterface
- {
- public:
- IMPLEMENT_IINTERFACE;
- CProxyActivityGraph(IRoxieSlaveContext *_ctx, const char *_graphName, unsigned _id, ActivityArray &_graphDefinition, const IRoxieContextLogger &_logctx, unsigned _numParallel)
- : ctx(_ctx), graphName(_graphName), id(_id), graphDefinition(_graphDefinition), logctx(_logctx), numParallel(_numParallel)
- {
- }
- virtual void abort() { throwUnexpected(); }
- virtual void reset() { throwUnexpected(); }
- virtual void execute() { throwUnexpected(); }
- virtual void getProbeResponse(IPropertyTree *query) { throwUnexpected(); }
- virtual void onCreate(IHThorArg *_colocalArg)
- {
- colocalArg = _colocalArg;
- }
- virtual void noteException(IException *E) { throwUnexpected(); }
- virtual void checkAbort() { throwUnexpected(); }
- virtual IThorChildGraph * queryChildGraph() { return this; }
- virtual IEclGraphResults * queryLocalGraph() { throwUnexpected(); }
- virtual IRoxieServerChildGraph * queryLoopGraph() { throwUnexpected(); }
- virtual IRoxieServerChildGraph * createGraphLoopInstance(IRoxieSlaveContext *ctx, unsigned loopCounter, unsigned parentExtractSize, const byte * parentExtract, const IRoxieContextLogger &logctx) { throwUnexpected(); }
- virtual const char *queryName() const { throwUnexpected(); }
- virtual IRoxieServerActivity *queryActivity(unsigned _activityId) { return nullptr; } // MORE - may need something here!?
- virtual IEclGraphResults * evaluate(unsigned parentExtractSize, const byte * parentExtract)
- {
- Owned<CActivityGraph> realGraph;
- {
- CriticalBlock b(graphCrit);
- if (stack.length())
- realGraph.setown(&stack.popGet());
- }
- if (!realGraph)
- {
- realGraph.setown(new CActivityGraph(ctx, graphName, id, graphDefinition, NULL, logctx));
- realGraph->createGraph(ctx);
- realGraph->onCreate(colocalArg);
- }
- Owned<IEclGraphResults> results = realGraph->evaluate(parentExtractSize, parentExtract);
- {
- CriticalBlock b(graphCrit);
- stack.append(*realGraph.getClear());
- }
- return results.getClear();
- }
- protected:
- IRoxieSlaveContext *ctx;
- StringAttr graphName;
- unsigned id;
- ActivityArray &graphDefinition;
- const IRoxieContextLogger &logctx;
- unsigned numParallel;
- IHThorArg *colocalArg = nullptr;
- CriticalSection graphCrit;
- CIArrayOf<CActivityGraph> stack;
- };
- class CIterationActivityGraph : public CActivityGraph
- {
- IHThorArg * colocalParent;
- unsigned fixedParentExtractSize;
- const byte * fixedParentExtract;
- unsigned graphOutputActivityIndex;
- public:
- CIterationActivityGraph(const char *_graphName, unsigned _id, ActivityArray &x, IProbeManager *_probeManager,
- unsigned _loopCounter, IRoxieSlaveContext *_ctx, IHThorArg * _colocalParent, unsigned parentExtractSize, const byte * parentExtract, const IRoxieContextLogger &_logctx)
- : CActivityGraph(_ctx, _graphName, _id, x, _probeManager, _logctx)
- {
- graphOutputActivityIndex = 0;
- loopCounter = _loopCounter;
- colocalParent = _colocalParent;
- graphSlaveContext.setLoopCounter(loopCounter);
- graphCodeContext.set(_ctx->queryCodeContext());
- fixedParentExtractSize = parentExtractSize;
- fixedParentExtract = parentExtract;
- }
- void createIterationGraph(IRoxieSlaveContext *_ctx)
- {
- Owned<IRoxieServerActivity> pseudoActivity = new CPseudoActivity(_ctx, *new CPseudoArg);
- ForEachItemIn(idx1, graphDefinition)
- activities.append(*LINK(pseudoActivity));
- graphOutputActivityIndex = queryGraphOutputIndex();
- recursiveCreateGraph(_ctx, graphOutputActivityIndex);
- }
- unsigned queryGraphOutputIndex() const
- {
- ForEachItemIn(i, graphDefinition)
- if (graphDefinition.serverItem(i).getKind() == TAKgraphloopresultwrite)
- return i;
- throwUnexpected();
- }
- void recursiveCreateGraph(IRoxieSlaveContext *_ctx, unsigned whichActivity)
- {
- //Check to see if already created
- IRoxieServerActivity & prevActivity = activities.item(whichActivity);
- if (prevActivity.queryId() != 0)
- {
- prevActivity.noteOutputUsed(); //We need to patch up the number of outputs for splitters.
- return;
- }
- IRoxieServerActivityFactory &donor = graphDefinition.serverItem(whichActivity);
- IRoxieServerActivity * activity = NULL;
- if (donor.isGraphInvariant())
- {
- ThorActivityKind kind = donor.getKind();
- switch (kind)
- {
- case TAKif:
- case TAKchildif:
- case TAKcase:
- case TAKchildcase: // MORE RKC->GH - what about FILTER with a graph-invariant condition and other latestart activities?
- {
- Owned<IHThorArg> helper = &donor.getHelper();
- helper->onCreate(&graphCodeContext, colocalParent, NULL);
- helper->onStart(fixedParentExtract, NULL);
- unsigned branch;
- switch (kind)
- {
- case TAKif: case TAKchildif:
- branch = static_cast<IHThorIfArg *>(helper.get())->getCondition() ? 0 : 1;
- break;
- case TAKcase: case TAKchildcase:
- branch = static_cast<IHThorCaseArg *>(helper.get())->getBranch();
- if (branch >= donor.numInputs())
- branch = donor.numInputs() - 1;
- break;
- default:
- throwUnexpected();
- }
- helper.clear();
- unsigned outputidx;
- unsigned source = donor.getInput(branch, outputidx);
- if (source ==(unsigned) -1)
- activity = createRoxieServerNullActivity(&graphSlaveContext, &donor, probeManager);
- else
- activity = createRoxieServerPassThroughActivity(&graphSlaveContext, &donor, probeManager);
- activities.replace(*activity, whichActivity);
- activity->onCreate(colocalParent);
- if (source ==(unsigned) -1)
- return;
- recursiveCreateGraph(_ctx, source);
- connectInput(whichActivity, 0, source, outputidx, loopCounter);
- break;
- }
- }
- }
- if (!activity)
- {
- activity = donor.createActivity(&graphSlaveContext, probeManager);
- activities.replace(*activity, whichActivity);
- activity->onCreate(colocalParent);
- activity->resetOutputsUsed();
- unsigned inputidx = 0;
- loop
- {
- unsigned outputidx;
- unsigned source = donor.getInput(inputidx, outputidx);
- if (source==(unsigned) -1)
- break;
- recursiveCreateGraph(_ctx, source);
- connectInput(whichActivity, inputidx, source, outputidx, loopCounter);
- inputidx++;
- }
- }
- IntArray &dependencies = donor.queryDependencies();
- IntArray &dependencyIndexes = donor.queryDependencyIndexes();
- IntArray &dependencyControlIds = donor.queryDependencyControlIds();
- ForEachItemIn(idx2, dependencies)
- {
- unsigned input = dependencies.item(idx2);
- recursiveCreateGraph(_ctx, input);
- activity->addDependency(activities.item(input),dependencyIndexes.item(idx2),dependencyControlIds.item(idx2));
- }
- }
- virtual CGraphIterationInfo *selectGraphLoopOutput()
- {
- IRoxieServerActivity &sourceActivity = activities.item(graphOutputActivityIndex);
- IFinalRoxieInput *sourceInput = sourceActivity.queryOutput(0);
- return new CGraphIterationInfo(&sourceActivity, sourceInput, 0, loopCounter);
- }
- virtual void gatherIterationUsage(IRoxieServerLoopResultProcessor & processor)
- {
- ForEachItemIn(i, activities)
- activities.item(i).gatherIterationUsage(processor, fixedParentExtractSize, fixedParentExtract);
- }
- virtual void associateIterationOutputs(IRoxieServerLoopResultProcessor & processor)
- {
- ForEachItemIn(i, activities)
- activities.item(i).associateIterationOutputs(processor, fixedParentExtractSize, fixedParentExtract, probeManager, probes);
- }
- };
- class CDelayedActivityGraph : implements IActivityGraph, public CInterface
- {
- StringAttr graphName;
- ActivityArray & graphDefinition;
- IProbeManager *probeManager;
- unsigned id;
- IRoxieSlaveContext * ctx;
- IHThorArg * colocalParent;
- public:
- IMPLEMENT_IINTERFACE;
- CDelayedActivityGraph(const char *_graphName, unsigned _id, ActivityArray &x, IProbeManager *_probeManager)
- : probeManager(_probeManager), graphDefinition(x)
- {
- graphName.set(_graphName);
- id = _id;
- ctx = NULL;
- colocalParent = NULL;
- }
- virtual const char *queryName() const { return graphName.get(); }
- virtual void abort() { throwUnexpected(); }
- virtual void reset() { }
- virtual void execute() { throwUnexpected(); }
- virtual void getProbeResponse(IPropertyTree *query) { throwUnexpected(); }
- virtual void noteException(IException *E) { throwUnexpected(); }
- virtual void checkAbort() { throwUnexpected(); }
- virtual IThorChildGraph * queryChildGraph() { throwUnexpected(); }
- virtual IEclGraphResults * queryLocalGraph() { throwUnexpected(); }
- virtual IRoxieServerChildGraph * queryLoopGraph() { throwUnexpected(); }
- virtual IRoxieServerActivity *queryActivity(unsigned _activityId) { return nullptr; } // MORE - may need something here!?
- virtual void onCreate(IHThorArg *_colocalParent)
- {
- colocalParent = _colocalParent;
- }
- virtual IRoxieServerChildGraph * createGraphLoopInstance(IRoxieSlaveContext *ctx, unsigned loopCounter, unsigned parentExtractSize, const byte * parentExtract, const IRoxieContextLogger &logctx)
- {
- Owned<CIterationActivityGraph> ret = new CIterationActivityGraph(graphName, id, graphDefinition, probeManager, loopCounter, ctx, colocalParent, parentExtractSize, parentExtract, logctx);
- ret->createIterationGraph(ctx);
- return ret.getClear();
- }
- };
- IActivityGraph *createActivityGraph(IRoxieSlaveContext *ctx, const char *_graphName, unsigned id, ActivityArray &childFactories, IRoxieServerActivity *parentActivity, IProbeManager *_probeManager, const IRoxieContextLogger &_logctx, unsigned numParallel)
- {
- if (childFactories.isDelayed())
- {
- assertex(numParallel==1);
- return new CDelayedActivityGraph(_graphName, id, childFactories, _probeManager);
- }
- else
- {
- if (numParallel==1 || _probeManager != nullptr)
- {
- Owned<IProbeManager> childProbe;
- if (_probeManager)
- childProbe.setown(_probeManager->startChildGraph(id, parentActivity));
- Owned<CActivityGraph> ret = new CActivityGraph(ctx, _graphName, id, childFactories, childProbe, _logctx);
- ret->createGraph(ctx);
- if (_probeManager)
- _probeManager->endChildGraph(childProbe, parentActivity);
- return ret.getClear();
- }
- else
- return new CProxyActivityGraph(ctx, _graphName, id, childFactories, _logctx, numParallel);
- }
- }
- //================================================================================================================
- #ifdef _USE_CPPUNIT
- #include "unittests.hpp"
- // There is a bug in VC6 implemetation of protected which prevents nested classes from accessing owner's data. It can be tricky to work around - hence...
- #if _MSC_VER==1200
- #undef protected
- #endif
- static const char *sortAlgorithm;
- class TestMetaData : public CInterface, implements IOutputMetaData
- {
- public:
- IMPLEMENT_IINTERFACE;
- virtual size32_t getRecordSize(const void *) { return 10; }
- virtual size32_t getMinRecordSize() const { return 10; }
- virtual size32_t getFixedSize() const { return 10; }
- virtual void toXML(const byte * self, IXmlWriter & out) {}
- virtual unsigned getVersion() const { return OUTPUTMETADATA_VERSION; }
- virtual unsigned getMetaFlags() { return 0; }
- virtual void destruct(byte * self) {}
- virtual IOutputRowSerializer * createDiskSerializer(ICodeContext * ctx, unsigned activityId) { return NULL; }
- virtual IOutputRowDeserializer * createDiskDeserializer(ICodeContext * ctx, unsigned activityId) { return NULL; }
- virtual ISourceRowPrefetcher * createDiskPrefetcher(ICodeContext * ctx, unsigned activityId) { return NULL; }
- virtual IOutputMetaData * querySerializedDiskMeta() { return NULL; }
- virtual IOutputRowSerializer * createInternalSerializer(ICodeContext * ctx, unsigned activityId) { return NULL; }
- virtual IOutputRowDeserializer * createInternalDeserializer(ICodeContext * ctx, unsigned activityId) { return NULL; }
- virtual void walkIndirectMembers(const byte * self, IIndirectMemberVisitor & visitor) {}
- virtual IOutputMetaData * queryChildMeta(unsigned i) { return NULL; }
- } testMeta;
- class TestInput : public CInterface, implements IFinalRoxieInput, implements IEngineRowStream
- {
- char const * const *input;
- IRoxieSlaveContext *ctx;
- unsigned endSeen;
- bool eof;
- unsigned count;
- ActivityTimeAccumulator totalCycles;
- size32_t recordSize;
- unsigned activityId;
- public:
- enum { STATEreset, STATEstarted, STATEstopped } state;
- bool allRead;
- unsigned repeat;
- IMPLEMENT_IINTERFACE;
- TestInput(IRoxieSlaveContext *_ctx, char const * const *_input)
- {
- ctx = _ctx;
- input = _input;
- count = 0;
- eof = false;
- allRead = false;
- endSeen = 0;
- recordSize = testMeta.getFixedSize();
- state = STATEreset;
- activityId = 1;
- repeat = 0;
- }
- virtual IOutputMetaData * queryOutputMeta() const { return &testMeta; }
- virtual void prestart(unsigned parentExtractSize, const byte *parentExtract)
- {
- ASSERT(state == STATEreset);
- }
- virtual void start(unsigned parentExtractSize, const byte *parentExtract, bool paused)
- {
- ASSERT(state == STATEreset);
- state = STATEstarted;
- }
- virtual IStrandJunction *getOutputStreams(IRoxieSlaveContext *ctx, unsigned idx, PointerArrayOf<IEngineRowStream> &streams, const StrandOptions * consumerOptions, bool consumerOrdered, IOrderedCallbackCollection * orderedCallbacks)
- {
- streams.append(this);
- return NULL;
- }
- virtual IEngineRowStream *queryConcreteOutputStream(unsigned whichInput) { assertex(whichInput==0); return this; }
- virtual IStrandJunction *queryConcreteOutputJunction(unsigned idx) const { assertex(idx==0); return nullptr; }
- virtual IRoxieServerActivity *queryActivity()
- {
- throwUnexpected();
- }
- virtual IIndexReadActivityInfo *queryIndexReadActivity()
- {
- throwUnexpected();
- }
- virtual void stop()
- {
- state = STATEstopped;
- }
- virtual void reset()
- {
- ASSERT(state == STATEstopped);
- eof = false; count = 0; endSeen = 0; allRead = false; state = STATEreset;
- }
- virtual void resetEOF()
- {
- throwUnexpected();
- }
- virtual void checkAbort() {}
- virtual unsigned queryId() const { return activityId; };
- virtual const void *nextRow()
- {
- ActivityTimer t(totalCycles, ctx->queryOptions().timeActivities);
- ASSERT(state == STATEstarted);
- ASSERT(allRead || !eof);
- if (eof)
- return NULL;
- again:
- const char *nextSource = input[count++];
- if (nextSource)
- {
- endSeen = 0;
- void *ret = ctx->queryRowManager().ALLOCATE(recordSize);
- memset(ret, 0, recordSize);
- strncpy((char *) ret, nextSource, recordSize);
- return ret;
- }
- else if (repeat)
- {
- repeat--;
- count = 0;
- goto again;
- }
- else
- {
- endSeen++;
- if (endSeen==2)
- eof = true;
- return NULL;
- }
- }
- virtual bool nextGroup(ConstPointerArray & group)
- {
- const void * next;
- while ((next = nextRow()) != NULL)
- group.append(next);
- if (group.ordinality())
- return true;
- return false;
- }
- virtual unsigned __int64 queryTotalCycles() const { return totalCycles.totalCycles; }
- virtual unsigned __int64 queryLocalCycles() const { return totalCycles.totalCycles; }
- virtual IFinalRoxieInput *queryInput(unsigned idx) const
- {
- return NULL;
- }
- };
- struct SortActivityTest : public ccdserver_hqlhelper::CThorSortArg {
- public:
- struct CompareClass : public ICompare {
- virtual int docompare(const void * _left, const void * _right) const {
- return memcmp(_left, _right, 5);
- }
- } compare;
- virtual ICompare * queryCompare() { return &compare; }
- virtual IOutputMetaData * queryOutputMeta()
- {
- return &testMeta;
- }
- virtual unsigned getAlgorithmFlags() { return TAFunstable; }
- virtual const char * getAlgorithm() { return sortAlgorithm; }
- };
- extern "C" IHThorArg * sortActivityTestFactory() { return new SortActivityTest; }
- struct MergeActivityTest : public ccdserver_hqlhelper::CThorMergeArg {
- static bool isDedup;
- public:
- struct CompareClass : public ICompare {
- virtual int docompare(const void * _left, const void * _right) const {
- return memcmp(_left, _right, 5);
- }
- } compare;
- virtual ICompare * queryCompare() { return &compare; }
- virtual IOutputMetaData * queryOutputMeta()
- {
- return &testMeta;
- }
- virtual bool dedup() { return isDedup; }
- };
- bool MergeActivityTest::isDedup = false;
- extern "C" IHThorArg * mergeActivityTestFactory() { return new MergeActivityTest; }
- struct SplitActivityTest : public ccdserver_hqlhelper::CThorSplitArg {
- public:
- virtual unsigned numBranches() { return 2; }
- virtual IOutputMetaData * queryOutputMeta()
- {
- return &testMeta;
- }
- };
- extern "C" IHThorArg * splitActivityTestFactory() { return new SplitActivityTest; }
- class CcdServerTest : public CppUnit::TestFixture
- {
- CPPUNIT_TEST_SUITE(CcdServerTest);
- CPPUNIT_TEST(testSetup);
- CPPUNIT_TEST(testHeapSort);
- CPPUNIT_TEST(testQuickSort);
- CPPUNIT_TEST(testMerge);
- CPPUNIT_TEST(testMergeDedup);
- CPPUNIT_TEST(testMiscellaneous);
- CPPUNIT_TEST(testSplitter);
- CPPUNIT_TEST(testCleanup);
- CPPUNIT_TEST_SUITE_END();
- protected:
- SlaveContextLogger logctx;
- Owned<const IQueryDll> queryDll;
- Owned<IRoxiePackage> package;
- Owned<IPropertyTree> stateInfo;
- Owned<IQueryFactory> queryFactory;
- Owned<IRoxieSlaveContext> ctx;
- void testSetup()
- {
- selfTestMode = true;
- roxiemem::setTotalMemoryLimit(false, true, false, 100 * 1024 * 1024, 0, NULL, NULL);
- }
- void testCleanup()
- {
- roxiemem::releaseRoxieHeap();
- }
- void init()
- {
- package.setown(createRoxiePackage(NULL, NULL));
- queryDll.setown(createExeQueryDll("roxie"));
- stateInfo.setown(createPTreeFromXMLString("<test memoryLimit='50000000'/>"));
- queryFactory.setown(createServerQueryFactory("test", queryDll.getLink(), *package, stateInfo, false, false));
- ctx.setown(createSlaveContext(queryFactory, logctx, NULL, false));
- queryActiveTimer()->reset();
- }
- void testActivity(IRoxieServerActivity *activity, char const * const *input, char const * const *output)
- {
- testActivity(activity, input, NULL, output);
- }
- void testActivity(IRoxieServerActivity *activity, char const * const *input, char const * const *input2, char const * const *output, unsigned outputIdx = 0)
- {
- TestInput in(ctx, input);
- TestInput in2(ctx, input2);
- activity->setInput(0, 0, &in);
- if (input2)
- activity->setInput(1, 0, &in2);
- IFinalRoxieInput *out = activity->queryOutput(outputIdx);
- IEngineRowStream *outStream = connectSingleStream(ctx, out, outputIdx, true);
- IOutputMetaData *meta = out->queryOutputMeta();
- void *buf = alloca(meta->getFixedSize());
- for (unsigned iteration = 0; iteration < 8; iteration++)
- {
- // All activities should be able to be restarted multiple times in the same context (for child queries) or in a new context (for graph pooling, if we ever wanted it)
- // This should be true whether we read all, some, or none of the data.
- // Should not matter if an activity is not started
- if (iteration % 4 == 0)
- activity->onCreate(NULL);
- unsigned count = 0;
- if (iteration % 4 != 3)
- {
- activity->start(0, NULL, false);
- ASSERT(in.state == TestInput::STATEstarted);
- ASSERT(!input2 || in2.state == TestInput::STATEstarted);
- loop
- {
- const void *next = outStream->nextRow();
- if (!next)
- {
- ASSERT(output[count++] == NULL);
- next = outStream->nextRow();
- if (!next)
- {
- ASSERT(output[count++] == NULL);
- break;
- }
- }
- ASSERT(output[count] != NULL);
- unsigned outsize = meta->getRecordSize(next);
- memset(buf, 0, outsize);
- strncpy((char *) buf, output[count++], outsize);
- ASSERT(memcmp(next, buf, outsize) == 0);
- ReleaseRoxieRow(next);
- if (iteration % 4 == 2)
- break;
- }
- if (iteration % 4 != 2)
- {
- // Check that reading after end is harmless
- in.allRead = true;
- const void *next = outStream->nextRow();
- ASSERT(next == NULL);
- }
- }
- activity->stop();
- ASSERT(in.state == TestInput::STATEstopped);
- ASSERT(!input2 || in2.state == TestInput::STATEstopped);
- activity->reset();
- ASSERT(in.state == TestInput::STATEreset);
- ASSERT(!input2 || in2.state == TestInput::STATEreset);
- ctx->queryRowManager().reportLeaks();
- ASSERT(ctx->queryRowManager().numPagesAfterCleanup(true) == 0);
- }
- }
- void testSplitActivity(IRoxieServerActivityFactory *factory, char const * const *input, char const * const *output, unsigned numOutputs, unsigned repeat)
- {
- Owned <IRoxieServerActivity> activity = factory->createActivity(ctx, NULL);
- TestInput in(ctx, input);
- in.repeat = repeat;
- activity->setInput(0, 0, &in);
- ArrayOf<IFinalRoxieInput *> out;
- for (unsigned i = 0; i < numOutputs; i++)
- {
- out.append(activity->queryOutput(i));
- }
- activity->onCreate(NULL);
- class casyncfor: public CAsyncFor
- {
- public:
- casyncfor(CcdServerTest *_parent, IRoxieServerActivity *_activity, ArrayOf<IFinalRoxieInput *> &_outputs, char const * const *_input, char const * const *_output, unsigned _repeat)
- : parent(_parent), activity(_activity), outputs(_outputs), input(_input), output(_output), repeat(_repeat)
- {}
- void Do(unsigned i)
- {
- IFinalRoxieInput *out = outputs.item(i);
- IEngineRowStream *outStream = connectSingleStream(NULL, out, 0, true);
- out->start(0, NULL, false);
- IOutputMetaData *meta = out->queryOutputMeta();
- void *buf = alloca(meta->getFixedSize());
- unsigned count = 0;
- unsigned repeats = repeat;
- loop
- {
- const void *next = outStream->nextRow();
- if (!next)
- {
- ASSERT(repeats==0);
- ASSERT(output[count] == NULL);
- count++;
- next = outStream->nextRow();
- if (!next)
- {
- ASSERT(output[count] == NULL);
- count++;
- break;
- }
- }
- if (output[count] == NULL && repeats)
- {
- repeats--;
- count = 0;
- }
- ASSERT(output[count] != NULL);
- unsigned outsize = meta->getRecordSize(next);
- memset(buf, 0, outsize);
- strncpy((char *) buf, output[count++], outsize);
- ASSERT(memcmp(next, buf, outsize) == 0);
- ReleaseRoxieRow(next);
- }
- outStream->stop();
- }
- private:
- IRoxieServerActivity *activity;
- ArrayOf<IFinalRoxieInput *> &outputs;
- char const * const *input;
- char const * const *output;
- unsigned repeat;
- CcdServerTest *parent;
- } afor(this, activity, out, input, output, repeat);
- afor.For(numOutputs, numOutputs);
- ASSERT(in.state == TestInput::STATEstopped);
- for (unsigned i2 = 0; i2 < numOutputs; i2++)
- {
- out.item(i2)->reset();
- }
- ASSERT(in.state == TestInput::STATEreset);
- ctx->queryRowManager().reportLeaks();
- ASSERT(ctx->queryRowManager().numPagesAfterCleanup(true) == 0);
- }
- static int compareFunc(const void *l, const void *r)
- {
- return strcmp(*(char **) l, *(char **) r);
- }
- void testSort(unsigned type)
- {
- init();
- sortAlgorithm = NULL;
- if (type==2)
- sortAlgorithm = "heapSort";
- else
- sortAlgorithm = "quickSort";
- DBGLOG("Testing %s activity", sortAlgorithm);
- Owned<IPropertyTree> dummyNode = createPTree("node");
- Owned <IRoxieServerActivityFactory> factory = createRoxieServerSortActivityFactory(1, 1, *queryFactory, sortActivityTestFactory, TAKsort, *dummyNode);
- Owned <IRoxieServerActivity> activity = factory->createActivity(ctx, NULL);
- const char * test[] = { NULL, NULL };
- const char * test12345[] = { "1", "2", "3", "4", "5", NULL, NULL };
- const char * test54321[] = { "5", "4", "3", "2", "1", NULL, NULL };
- const char * test11111[] = { "1", "1", "1", "1", "1", NULL, NULL };
- const char * test11111_12345[] = { "1", "1", "1", "1", "1", NULL, "1", "2", "3", "4", "5", NULL, NULL };
- const char * test11111_54321[] = { "1", "1", "1", "1", "1", NULL, "5", "4", "3", "2", "1", NULL, NULL };
- const char * test54321_54321[] = { "5", "4", "3", "2", "1", NULL, "5", "4", "3", "2", "1", NULL, NULL };
- const char * test12345_12345[] = { "1", "2", "3", "4", "5", NULL, "1", "2", "3", "4", "5", NULL, NULL };
- testActivity(activity, test, test);
- testActivity(activity, test12345, test12345);
- testActivity(activity, test54321, test12345);
- testActivity(activity, test11111, test11111);
- testActivity(activity, test11111_12345, test11111_12345);
- testActivity(activity, test11111_54321, test11111_12345);
- testActivity(activity, test54321_54321, test12345_12345);
- // A few larger tests
- char *input[2002];
- char *output[2002];
- input[2000] = input[2001] = output[2000] = output[2001] = NULL;
- unsigned i;
- // identical
- for (i=0; i<2000; i++)
- {
- input[i] = new char[11];
- output[i] = new char[11];
- sprintf(input[i], "1");
- sprintf(output[i], "1");
- }
- testActivity(activity, input, output);
- // Ascending
- for (i=0; i<2000; i++)
- {
- sprintf(input[i], "%04d", i);
- sprintf(output[i], "%04d", i);
- }
- testActivity(activity, input, output);
- // Almost sorted
- for (i=0; i<20; i++)
- {
- unsigned h = i*100;
- sprintf(input[h], "%04d", 1900-h);
- }
- testActivity(activity, input, output);
- // Descending
- for (i=0; i<2000; i++)
- {
- sprintf(input[i], "%04d", 1999-i);
- sprintf(output[i], "%04d", i);
- }
- testActivity(activity, input, output);
- // Random
- for (i=0; i<2000; i++)
- {
- unsigned r = rand() % 1500;
- sprintf(input[i], "%04d", r);
- sprintf(output[i], "%04d", r);
- }
- qsort(output, 2000, sizeof(output[0]), compareFunc);
- testActivity(activity, input, output);
- unsigned __int64 us = cycle_to_nanosec(factory->queryLocalCycles()/1000);
- DBGLOG("Simple %s sorts: activity time %u.%u ms", type==2?"Heap" : (type==1 ? "Insertion" : "Quick"), (int)(us/1000), (int)(us%1000));
- factory->resetNodeProgressInfo();
- if (type)
- {
- // Other than quicksort, it's supposed to be stable. Let's check that it is
- // All sort identical
- for (i=0; i<2000; i++)
- {
- sprintf(input[i], "1 %d", i);
- sprintf(output[i], "1 %d", i);
- }
- testActivity(activity, input, output);
- // Already sorted
- for (i=0; i<2000; i++)
- {
- sprintf(input[i], "%04d %d", i / 10, i);
- sprintf(output[i], "%04d %d", i / 10, i);
- }
- testActivity(activity, input, output);
- // Reverse order
- for (i=0; i<2000; i++)
- {
- sprintf(input[i], "%04d %d", 199 - (i / 10), i%10);
- sprintf(output[i], "%04d %d", i / 10, i%10);
- }
- testActivity(activity, input, output);
- }
- for (i=0; i<2000; i++)
- {
- delete [] input[i];
- delete [] output[i];
- }
- DBGLOG("Finished testing %s sort", type==2?"Heap" : (type==1 ? "Insertion" : "Quick"));
- }
- void testQuickSort()
- {
- testSort(0);
- }
- void testHeapSort()
- {
- testSort(2);
- }
- void testMerge()
- {
- DBGLOG("testMerge");
- init();
- Owned<IPropertyTree> splitterNode = createPTree();
- Owned <IRoxieServerActivityFactory> factory = createRoxieServerMergeActivityFactory(1, 1, *queryFactory, mergeActivityTestFactory, TAKmerge, *splitterNode);
- factory->setInput(0,0,0);
- factory->setInput(1,0,0);
- Owned <IRoxieServerActivity> activity = factory->createActivity(ctx, NULL);
- const char * test[] = { NULL, NULL };
- const char * test12345[] = { "1", "2", "3", "4", "5", NULL, NULL };
- const char * test1122334455[] = { "1", "1", "2", "2", "3", "3", "4", "4", "5", "5", NULL, NULL };
- const char * test11111[] = { "1", "1", "1", "1", "1", NULL, NULL };
- const char * test1111111111[] = { "1", "1", "1", "1", "1", "1", "1", "1", "1", "1", NULL, NULL };
- const char * test11111_12345[] = { "1", "1", "1", "1", "1", NULL, "1", "2", "3", "4", "5", NULL, NULL };
- const char * test1111112345[] = { "1", "1", "1", "1", "1", "1", "2", "3", "4", "5", NULL, NULL };
- const char * test11111111111122334455[] = { "1", "1", "1", "1", "1", "1", "1", "1", "1", "1", "1", "1", "2", "2", "3", "3", "4", "4", "5", "5", NULL, NULL };
-
- testActivity(activity, test, test, test);
- testActivity(activity, test12345, test, test12345);
- testActivity(activity, test, test12345, test12345);
- testActivity(activity, test12345, test12345, test1122334455);
- testActivity(activity, test11111, test, test11111);
- testActivity(activity, test, test11111, test11111);
- testActivity(activity, test11111, test11111, test1111111111);
- testActivity(activity, test11111_12345, test, test1111112345);
- testActivity(activity, test11111_12345, test11111_12345, test11111111111122334455);
- // Should really test WHICH side gets kept...
- // Should test with more than 2 inputs...
- DBGLOG("testMerge done");
- }
- void testMergeDedup()
- {
- DBGLOG("testMergeDedup");
- init();
- MergeActivityTest::isDedup = true;
- Owned<IPropertyTree> splitterNode = createPTree();
- Owned <IRoxieServerActivityFactory> factory = createRoxieServerMergeActivityFactory(1, 1, *queryFactory, mergeActivityTestFactory, TAKmerge, *splitterNode);
- factory->setInput(0,0,0);
- factory->setInput(1,0,0);
- Owned <IRoxieServerActivity> activity = factory->createActivity(ctx, NULL);
- const char * test[] = { NULL, NULL };
- const char * test12345[] = { "1", "2", "3", "4", "5", NULL, NULL };
- const char * test11111[] = { "1", "1", "1", "1", "1", NULL, NULL };
- const char * test11111_12345[] = { "1", "1", "1", "1", "1", NULL, "1", "2", "3", "4", "5", NULL, NULL };
- const char * test1111112345[] = { "1", "1", "1", "1", "1", "1", "2", "3", "4", "5", NULL, NULL };
-
- testActivity(activity, test11111, test, test11111); // No dedup within a stream
- testActivity(activity, test11111, test11111, test11111); // No dedup within a stream
- testActivity(activity, test, test11111, test11111);
- testActivity(activity, test, test, test);
- testActivity(activity, test12345, test, test12345);
- testActivity(activity, test, test12345, test12345);
- testActivity(activity, test12345, test12345, test12345);
- testActivity(activity, test11111_12345, test, test1111112345);
- testActivity(activity, test11111_12345, test11111_12345, test1111112345);
- // Should really test WHICH side gets kept...
- // Should test with more than 2 inputs...
- DBGLOG("testMergeDedup done");
- }
- void doTestSplitter(unsigned numOutputs)
- {
- DBGLOG("testSplit %d", numOutputs);
- init();
- Owned<IPropertyTree> splitterNode = createPTree();
- Owned <IRoxieServerActivityFactory> factory = createRoxieServerThroughSpillActivityFactory(*queryFactory, splitActivityTestFactory, numOutputs, *splitterNode);
- factory->setInput(0,0,0);
- Owned <IRoxieServerActivity> activity = factory->createActivity(ctx, NULL);
- const char * test[] = { NULL, NULL };
- const char * test12345[] = { "1", "2", "3", "4", "5", NULL, NULL };
- unsigned start = msTick();
- testSplitActivity(factory, test, test, numOutputs, 0);
- testSplitActivity(factory, test12345, test12345, numOutputs, 0);
- testSplitActivity(factory, test12345, test12345, numOutputs, 100);
- unsigned elapsed = msTick() - start;
- DBGLOG("testSplit %d done in %dms", numOutputs, elapsed);
- }
- void testSplitter()
- {
- doTestSplitter(1);
- doTestSplitter(2);
- doTestSplitter(3);
- doTestSplitter(10);
- }
- void testMiscellaneous()
- {
- DBGLOG("sizeof(CriticalSection)=%u", (unsigned) sizeof(CriticalSection));
- DBGLOG("sizeof(SpinLock)=%u", (unsigned) sizeof(SpinLock));
- DBGLOG("sizeof(CJoinGroup)=%u", (unsigned) sizeof(CJoinGroup));
- ASSERT(sizeof(CJoinGroup) <= 120);
- }
- };
- CPPUNIT_TEST_SUITE_REGISTRATION( CcdServerTest );
- CPPUNIT_TEST_SUITE_NAMED_REGISTRATION( CcdServerTest, "CcdServerTest" );
- #endif
|