workunit.cpp 303 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034203520362037203820392040204120422043204420452046204720482049205020512052205320542055205620572058205920602061206220632064206520662067206820692070207120722073207420752076207720782079208020812082208320842085208620872088208920902091209220932094209520962097209820992100210121022103210421052106210721082109211021112112211321142115211621172118211921202121212221232124212521262127212821292130213121322133213421352136213721382139214021412142214321442145214621472148214921502151215221532154215521562157215821592160216121622163216421652166216721682169217021712172217321742175217621772178217921802181218221832184218521862187218821892190219121922193219421952196219721982199220022012202220322042205220622072208220922102211221222132214221522162217221822192220222122222223222422252226222722282229223022312232223322342235223622372238223922402241224222432244224522462247224822492250225122522253225422552256225722582259226022612262226322642265226622672268226922702271227222732274227522762277227822792280228122822283228422852286228722882289229022912292229322942295229622972298229923002301230223032304230523062307230823092310231123122313231423152316231723182319232023212322232323242325232623272328232923302331233223332334233523362337233823392340234123422343234423452346234723482349235023512352235323542355235623572358235923602361236223632364236523662367236823692370237123722373237423752376237723782379238023812382238323842385238623872388238923902391239223932394239523962397239823992400240124022403240424052406240724082409241024112412241324142415241624172418241924202421242224232424242524262427242824292430243124322433243424352436243724382439244024412442244324442445244624472448244924502451245224532454245524562457245824592460246124622463246424652466246724682469247024712472247324742475247624772478247924802481248224832484248524862487248824892490249124922493249424952496249724982499250025012502250325042505250625072508250925102511251225132514251525162517251825192520252125222523252425252526252725282529253025312532253325342535253625372538253925402541254225432544254525462547254825492550255125522553255425552556255725582559256025612562256325642565256625672568256925702571257225732574257525762577257825792580258125822583258425852586258725882589259025912592259325942595259625972598259926002601260226032604260526062607260826092610261126122613261426152616261726182619262026212622262326242625262626272628262926302631263226332634263526362637263826392640264126422643264426452646264726482649265026512652265326542655265626572658265926602661266226632664266526662667266826692670267126722673267426752676267726782679268026812682268326842685268626872688268926902691269226932694269526962697269826992700270127022703270427052706270727082709271027112712271327142715271627172718271927202721272227232724272527262727272827292730273127322733273427352736273727382739274027412742274327442745274627472748274927502751275227532754275527562757275827592760276127622763276427652766276727682769277027712772277327742775277627772778277927802781278227832784278527862787278827892790279127922793279427952796279727982799280028012802280328042805280628072808280928102811281228132814281528162817281828192820282128222823282428252826282728282829283028312832283328342835283628372838283928402841284228432844284528462847284828492850285128522853285428552856285728582859286028612862286328642865286628672868286928702871287228732874287528762877287828792880288128822883288428852886288728882889289028912892289328942895289628972898289929002901290229032904290529062907290829092910291129122913291429152916291729182919292029212922292329242925292629272928292929302931293229332934293529362937293829392940294129422943294429452946294729482949295029512952295329542955295629572958295929602961296229632964296529662967296829692970297129722973297429752976297729782979298029812982298329842985298629872988298929902991299229932994299529962997299829993000300130023003300430053006300730083009301030113012301330143015301630173018301930203021302230233024302530263027302830293030303130323033303430353036303730383039304030413042304330443045304630473048304930503051305230533054305530563057305830593060306130623063306430653066306730683069307030713072307330743075307630773078307930803081308230833084308530863087308830893090309130923093309430953096309730983099310031013102310331043105310631073108310931103111311231133114311531163117311831193120312131223123312431253126312731283129313031313132313331343135313631373138313931403141314231433144314531463147314831493150315131523153315431553156315731583159316031613162316331643165316631673168316931703171317231733174317531763177317831793180318131823183318431853186318731883189319031913192319331943195319631973198319932003201320232033204320532063207320832093210321132123213321432153216321732183219322032213222322332243225322632273228322932303231323232333234323532363237323832393240324132423243324432453246324732483249325032513252325332543255325632573258325932603261326232633264326532663267326832693270327132723273327432753276327732783279328032813282328332843285328632873288328932903291329232933294329532963297329832993300330133023303330433053306330733083309331033113312331333143315331633173318331933203321332233233324332533263327332833293330333133323333333433353336333733383339334033413342334333443345334633473348334933503351335233533354335533563357335833593360336133623363336433653366336733683369337033713372337333743375337633773378337933803381338233833384338533863387338833893390339133923393339433953396339733983399340034013402340334043405340634073408340934103411341234133414341534163417341834193420342134223423342434253426342734283429343034313432343334343435343634373438343934403441344234433444344534463447344834493450345134523453345434553456345734583459346034613462346334643465346634673468346934703471347234733474347534763477347834793480348134823483348434853486348734883489349034913492349334943495349634973498349935003501350235033504350535063507350835093510351135123513351435153516351735183519352035213522352335243525352635273528352935303531353235333534353535363537353835393540354135423543354435453546354735483549355035513552355335543555355635573558355935603561356235633564356535663567356835693570357135723573357435753576357735783579358035813582358335843585358635873588358935903591359235933594359535963597359835993600360136023603360436053606360736083609361036113612361336143615361636173618361936203621362236233624362536263627362836293630363136323633363436353636363736383639364036413642364336443645364636473648364936503651365236533654365536563657365836593660366136623663366436653666366736683669367036713672367336743675367636773678367936803681368236833684368536863687368836893690369136923693369436953696369736983699370037013702370337043705370637073708370937103711371237133714371537163717371837193720372137223723372437253726372737283729373037313732373337343735373637373738373937403741374237433744374537463747374837493750375137523753375437553756375737583759376037613762376337643765376637673768376937703771377237733774377537763777377837793780378137823783378437853786378737883789379037913792379337943795379637973798379938003801380238033804380538063807380838093810381138123813381438153816381738183819382038213822382338243825382638273828382938303831383238333834383538363837383838393840384138423843384438453846384738483849385038513852385338543855385638573858385938603861386238633864386538663867386838693870387138723873387438753876387738783879388038813882388338843885388638873888388938903891389238933894389538963897389838993900390139023903390439053906390739083909391039113912391339143915391639173918391939203921392239233924392539263927392839293930393139323933393439353936393739383939394039413942394339443945394639473948394939503951395239533954395539563957395839593960396139623963396439653966396739683969397039713972397339743975397639773978397939803981398239833984398539863987398839893990399139923993399439953996399739983999400040014002400340044005400640074008400940104011401240134014401540164017401840194020402140224023402440254026402740284029403040314032403340344035403640374038403940404041404240434044404540464047404840494050405140524053405440554056405740584059406040614062406340644065406640674068406940704071407240734074407540764077407840794080408140824083408440854086408740884089409040914092409340944095409640974098409941004101410241034104410541064107410841094110411141124113411441154116411741184119412041214122412341244125412641274128412941304131413241334134413541364137413841394140414141424143414441454146414741484149415041514152415341544155415641574158415941604161416241634164416541664167416841694170417141724173417441754176417741784179418041814182418341844185418641874188418941904191419241934194419541964197419841994200420142024203420442054206420742084209421042114212421342144215421642174218421942204221422242234224422542264227422842294230423142324233423442354236423742384239424042414242424342444245424642474248424942504251425242534254425542564257425842594260426142624263426442654266426742684269427042714272427342744275427642774278427942804281428242834284428542864287428842894290429142924293429442954296429742984299430043014302430343044305430643074308430943104311431243134314431543164317431843194320432143224323432443254326432743284329433043314332433343344335433643374338433943404341434243434344434543464347434843494350435143524353435443554356435743584359436043614362436343644365436643674368436943704371437243734374437543764377437843794380438143824383438443854386438743884389439043914392439343944395439643974398439944004401440244034404440544064407440844094410441144124413441444154416441744184419442044214422442344244425442644274428442944304431443244334434443544364437443844394440444144424443444444454446444744484449445044514452445344544455445644574458445944604461446244634464446544664467446844694470447144724473447444754476447744784479448044814482448344844485448644874488448944904491449244934494449544964497449844994500450145024503450445054506450745084509451045114512451345144515451645174518451945204521452245234524452545264527452845294530453145324533453445354536453745384539454045414542454345444545454645474548454945504551455245534554455545564557455845594560456145624563456445654566456745684569457045714572457345744575457645774578457945804581458245834584458545864587458845894590459145924593459445954596459745984599460046014602460346044605460646074608460946104611461246134614461546164617461846194620462146224623462446254626462746284629463046314632463346344635463646374638463946404641464246434644464546464647464846494650465146524653465446554656465746584659466046614662466346644665466646674668466946704671467246734674467546764677467846794680468146824683468446854686468746884689469046914692469346944695469646974698469947004701470247034704470547064707470847094710471147124713471447154716471747184719472047214722472347244725472647274728472947304731473247334734473547364737473847394740474147424743474447454746474747484749475047514752475347544755475647574758475947604761476247634764476547664767476847694770477147724773477447754776477747784779478047814782478347844785478647874788478947904791479247934794479547964797479847994800480148024803480448054806480748084809481048114812481348144815481648174818481948204821482248234824482548264827482848294830483148324833483448354836483748384839484048414842484348444845484648474848484948504851485248534854485548564857485848594860486148624863486448654866486748684869487048714872487348744875487648774878487948804881488248834884488548864887488848894890489148924893489448954896489748984899490049014902490349044905490649074908490949104911491249134914491549164917491849194920492149224923492449254926492749284929493049314932493349344935493649374938493949404941494249434944494549464947494849494950495149524953495449554956495749584959496049614962496349644965496649674968496949704971497249734974497549764977497849794980498149824983498449854986498749884989499049914992499349944995499649974998499950005001500250035004500550065007500850095010501150125013501450155016501750185019502050215022502350245025502650275028502950305031503250335034503550365037503850395040504150425043504450455046504750485049505050515052505350545055505650575058505950605061506250635064506550665067506850695070507150725073507450755076507750785079508050815082508350845085508650875088508950905091509250935094509550965097509850995100510151025103510451055106510751085109511051115112511351145115511651175118511951205121512251235124512551265127512851295130513151325133513451355136513751385139514051415142514351445145514651475148514951505151515251535154515551565157515851595160516151625163516451655166516751685169517051715172517351745175517651775178517951805181518251835184518551865187518851895190519151925193519451955196519751985199520052015202520352045205520652075208520952105211521252135214521552165217521852195220522152225223522452255226522752285229523052315232523352345235523652375238523952405241524252435244524552465247524852495250525152525253525452555256525752585259526052615262526352645265526652675268526952705271527252735274527552765277527852795280528152825283528452855286528752885289529052915292529352945295529652975298529953005301530253035304530553065307530853095310531153125313531453155316531753185319532053215322532353245325532653275328532953305331533253335334533553365337533853395340534153425343534453455346534753485349535053515352535353545355535653575358535953605361536253635364536553665367536853695370537153725373537453755376537753785379538053815382538353845385538653875388538953905391539253935394539553965397539853995400540154025403540454055406540754085409541054115412541354145415541654175418541954205421542254235424542554265427542854295430543154325433543454355436543754385439544054415442544354445445544654475448544954505451545254535454545554565457545854595460546154625463546454655466546754685469547054715472547354745475547654775478547954805481548254835484548554865487548854895490549154925493549454955496549754985499550055015502550355045505550655075508550955105511551255135514551555165517551855195520552155225523552455255526552755285529553055315532553355345535553655375538553955405541554255435544554555465547554855495550555155525553555455555556555755585559556055615562556355645565556655675568556955705571557255735574557555765577557855795580558155825583558455855586558755885589559055915592559355945595559655975598559956005601560256035604560556065607560856095610561156125613561456155616561756185619562056215622562356245625562656275628562956305631563256335634563556365637563856395640564156425643564456455646564756485649565056515652565356545655565656575658565956605661566256635664566556665667566856695670567156725673567456755676567756785679568056815682568356845685568656875688568956905691569256935694569556965697569856995700570157025703570457055706570757085709571057115712571357145715571657175718571957205721572257235724572557265727572857295730573157325733573457355736573757385739574057415742574357445745574657475748574957505751575257535754575557565757575857595760576157625763576457655766576757685769577057715772577357745775577657775778577957805781578257835784578557865787578857895790579157925793579457955796579757985799580058015802580358045805580658075808580958105811581258135814581558165817581858195820582158225823582458255826582758285829583058315832583358345835583658375838583958405841584258435844584558465847584858495850585158525853585458555856585758585859586058615862586358645865586658675868586958705871587258735874587558765877587858795880588158825883588458855886588758885889589058915892589358945895589658975898589959005901590259035904590559065907590859095910591159125913591459155916591759185919592059215922592359245925592659275928592959305931593259335934593559365937593859395940594159425943594459455946594759485949595059515952595359545955595659575958595959605961596259635964596559665967596859695970597159725973597459755976597759785979598059815982598359845985598659875988598959905991599259935994599559965997599859996000600160026003600460056006600760086009601060116012601360146015601660176018601960206021602260236024602560266027602860296030603160326033603460356036603760386039604060416042604360446045604660476048604960506051605260536054605560566057605860596060606160626063606460656066606760686069607060716072607360746075607660776078607960806081608260836084608560866087608860896090609160926093609460956096609760986099610061016102610361046105610661076108610961106111611261136114611561166117611861196120612161226123612461256126612761286129613061316132613361346135613661376138613961406141614261436144614561466147614861496150615161526153615461556156615761586159616061616162616361646165616661676168616961706171617261736174617561766177617861796180618161826183618461856186618761886189619061916192619361946195619661976198619962006201620262036204620562066207620862096210621162126213621462156216621762186219622062216222622362246225622662276228622962306231623262336234623562366237623862396240624162426243624462456246624762486249625062516252625362546255625662576258625962606261626262636264626562666267626862696270627162726273627462756276627762786279628062816282628362846285628662876288628962906291629262936294629562966297629862996300630163026303630463056306630763086309631063116312631363146315631663176318631963206321632263236324632563266327632863296330633163326333633463356336633763386339634063416342634363446345634663476348634963506351635263536354635563566357635863596360636163626363636463656366636763686369637063716372637363746375637663776378637963806381638263836384638563866387638863896390639163926393639463956396639763986399640064016402640364046405640664076408640964106411641264136414641564166417641864196420642164226423642464256426642764286429643064316432643364346435643664376438643964406441644264436444644564466447644864496450645164526453645464556456645764586459646064616462646364646465646664676468646964706471647264736474647564766477647864796480648164826483648464856486648764886489649064916492649364946495649664976498649965006501650265036504650565066507650865096510651165126513651465156516651765186519652065216522652365246525652665276528652965306531653265336534653565366537653865396540654165426543654465456546654765486549655065516552655365546555655665576558655965606561656265636564656565666567656865696570657165726573657465756576657765786579658065816582658365846585658665876588658965906591659265936594659565966597659865996600660166026603660466056606660766086609661066116612661366146615661666176618661966206621662266236624662566266627662866296630663166326633663466356636663766386639664066416642664366446645664666476648664966506651665266536654665566566657665866596660666166626663666466656666666766686669667066716672667366746675667666776678667966806681668266836684668566866687668866896690669166926693669466956696669766986699670067016702670367046705670667076708670967106711671267136714671567166717671867196720672167226723672467256726672767286729673067316732673367346735673667376738673967406741674267436744674567466747674867496750675167526753675467556756675767586759676067616762676367646765676667676768676967706771677267736774677567766777677867796780678167826783678467856786678767886789679067916792679367946795679667976798679968006801680268036804680568066807680868096810681168126813681468156816681768186819682068216822682368246825682668276828682968306831683268336834683568366837683868396840684168426843684468456846684768486849685068516852685368546855685668576858685968606861686268636864686568666867686868696870687168726873687468756876687768786879688068816882688368846885688668876888688968906891689268936894689568966897689868996900690169026903690469056906690769086909691069116912691369146915691669176918691969206921692269236924692569266927692869296930693169326933693469356936693769386939694069416942694369446945694669476948694969506951695269536954695569566957695869596960696169626963696469656966696769686969697069716972697369746975697669776978697969806981698269836984698569866987698869896990699169926993699469956996699769986999700070017002700370047005700670077008700970107011701270137014701570167017701870197020702170227023702470257026702770287029703070317032703370347035703670377038703970407041704270437044704570467047704870497050705170527053705470557056705770587059706070617062706370647065706670677068706970707071707270737074707570767077707870797080708170827083708470857086708770887089709070917092709370947095709670977098709971007101710271037104710571067107710871097110711171127113711471157116711771187119712071217122712371247125712671277128712971307131713271337134713571367137713871397140714171427143714471457146714771487149715071517152715371547155715671577158715971607161716271637164716571667167716871697170717171727173717471757176717771787179718071817182718371847185718671877188718971907191719271937194719571967197719871997200720172027203720472057206720772087209721072117212721372147215721672177218721972207221722272237224722572267227722872297230723172327233723472357236723772387239724072417242724372447245724672477248724972507251725272537254725572567257725872597260726172627263726472657266726772687269727072717272727372747275727672777278727972807281728272837284728572867287728872897290729172927293729472957296729772987299730073017302730373047305730673077308730973107311731273137314731573167317731873197320732173227323732473257326732773287329733073317332733373347335733673377338733973407341734273437344734573467347734873497350735173527353735473557356735773587359736073617362736373647365736673677368736973707371737273737374737573767377737873797380738173827383738473857386738773887389739073917392739373947395739673977398739974007401740274037404740574067407740874097410741174127413741474157416741774187419742074217422742374247425742674277428742974307431743274337434743574367437743874397440744174427443744474457446744774487449745074517452745374547455745674577458745974607461746274637464746574667467746874697470747174727473747474757476747774787479748074817482748374847485748674877488748974907491749274937494749574967497749874997500750175027503750475057506750775087509751075117512751375147515751675177518751975207521752275237524752575267527752875297530753175327533753475357536753775387539754075417542754375447545754675477548754975507551755275537554755575567557755875597560756175627563756475657566756775687569757075717572757375747575757675777578757975807581758275837584758575867587758875897590759175927593759475957596759775987599760076017602760376047605760676077608760976107611761276137614761576167617761876197620762176227623762476257626762776287629763076317632763376347635763676377638763976407641764276437644764576467647764876497650765176527653765476557656765776587659766076617662766376647665766676677668766976707671767276737674767576767677767876797680768176827683768476857686768776887689769076917692769376947695769676977698769977007701770277037704770577067707770877097710771177127713771477157716771777187719772077217722772377247725772677277728772977307731773277337734773577367737773877397740774177427743774477457746774777487749775077517752775377547755775677577758775977607761776277637764776577667767776877697770777177727773777477757776777777787779778077817782778377847785778677877788778977907791779277937794779577967797779877997800780178027803780478057806780778087809781078117812781378147815781678177818781978207821782278237824782578267827782878297830783178327833783478357836783778387839784078417842784378447845784678477848784978507851785278537854785578567857785878597860786178627863786478657866786778687869787078717872787378747875787678777878787978807881788278837884788578867887788878897890789178927893789478957896789778987899790079017902790379047905790679077908790979107911791279137914791579167917791879197920792179227923792479257926792779287929793079317932793379347935793679377938793979407941794279437944794579467947794879497950795179527953795479557956795779587959796079617962796379647965796679677968796979707971797279737974797579767977797879797980798179827983798479857986798779887989799079917992799379947995799679977998799980008001800280038004800580068007800880098010801180128013801480158016801780188019802080218022802380248025802680278028802980308031803280338034803580368037803880398040804180428043804480458046804780488049805080518052805380548055805680578058805980608061806280638064806580668067806880698070807180728073807480758076807780788079808080818082808380848085808680878088808980908091809280938094809580968097809880998100810181028103810481058106810781088109811081118112811381148115811681178118811981208121812281238124812581268127812881298130813181328133813481358136813781388139814081418142814381448145814681478148814981508151815281538154815581568157815881598160816181628163816481658166816781688169817081718172817381748175817681778178817981808181818281838184818581868187818881898190819181928193819481958196819781988199820082018202820382048205820682078208820982108211821282138214821582168217821882198220822182228223822482258226822782288229823082318232823382348235823682378238823982408241824282438244824582468247824882498250825182528253825482558256825782588259826082618262826382648265826682678268826982708271827282738274827582768277827882798280828182828283828482858286828782888289829082918292829382948295829682978298829983008301830283038304830583068307830883098310831183128313831483158316831783188319832083218322832383248325832683278328832983308331833283338334833583368337833883398340834183428343834483458346834783488349835083518352835383548355835683578358835983608361836283638364836583668367836883698370837183728373837483758376837783788379838083818382838383848385838683878388838983908391839283938394839583968397839883998400840184028403840484058406840784088409841084118412841384148415841684178418841984208421842284238424842584268427842884298430843184328433843484358436843784388439844084418442844384448445844684478448844984508451845284538454845584568457845884598460846184628463846484658466846784688469847084718472847384748475847684778478847984808481848284838484848584868487848884898490849184928493849484958496849784988499850085018502850385048505850685078508850985108511851285138514851585168517851885198520852185228523852485258526852785288529853085318532853385348535853685378538853985408541854285438544854585468547854885498550855185528553855485558556855785588559856085618562856385648565856685678568856985708571857285738574857585768577857885798580858185828583858485858586858785888589859085918592859385948595859685978598859986008601860286038604860586068607860886098610861186128613861486158616861786188619862086218622862386248625862686278628862986308631863286338634863586368637863886398640864186428643864486458646864786488649865086518652865386548655865686578658865986608661866286638664866586668667866886698670867186728673867486758676867786788679868086818682868386848685868686878688868986908691869286938694869586968697869886998700870187028703870487058706870787088709871087118712871387148715871687178718871987208721872287238724872587268727872887298730873187328733873487358736873787388739874087418742874387448745874687478748874987508751875287538754875587568757875887598760876187628763876487658766876787688769877087718772877387748775877687778778877987808781878287838784878587868787878887898790879187928793879487958796879787988799880088018802880388048805880688078808880988108811881288138814881588168817881888198820882188228823882488258826882788288829883088318832883388348835883688378838883988408841884288438844884588468847884888498850885188528853885488558856885788588859886088618862886388648865886688678868886988708871887288738874887588768877887888798880888188828883888488858886888788888889889088918892889388948895889688978898889989008901890289038904890589068907890889098910891189128913891489158916891789188919892089218922892389248925892689278928892989308931893289338934893589368937893889398940894189428943894489458946894789488949895089518952895389548955895689578958895989608961896289638964896589668967896889698970897189728973897489758976897789788979898089818982898389848985898689878988898989908991899289938994899589968997899889999000900190029003900490059006900790089009901090119012901390149015901690179018901990209021902290239024902590269027902890299030903190329033903490359036903790389039904090419042904390449045904690479048904990509051905290539054905590569057905890599060906190629063906490659066906790689069907090719072907390749075907690779078907990809081908290839084908590869087908890899090909190929093909490959096909790989099910091019102910391049105910691079108910991109111911291139114911591169117911891199120912191229123912491259126912791289129913091319132913391349135913691379138913991409141914291439144914591469147914891499150915191529153915491559156915791589159916091619162916391649165916691679168916991709171917291739174917591769177917891799180918191829183918491859186918791889189919091919192919391949195919691979198919992009201920292039204920592069207920892099210921192129213921492159216921792189219922092219222922392249225922692279228922992309231923292339234923592369237923892399240924192429243924492459246924792489249925092519252925392549255925692579258925992609261926292639264926592669267926892699270927192729273927492759276927792789279928092819282928392849285928692879288928992909291929292939294929592969297929892999300930193029303930493059306930793089309931093119312931393149315931693179318931993209321932293239324932593269327932893299330933193329333933493359336933793389339934093419342934393449345934693479348
  1. /*##############################################################################
  2. HPCC SYSTEMS software Copyright (C) 2012 HPCC Systems.
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. ############################################################################## */
  13. #include "jlib.hpp"
  14. #include "workunit.hpp"
  15. #include "jprop.hpp"
  16. #include "jmisc.hpp"
  17. #include "jexcept.hpp"
  18. #include "jiter.ipp"
  19. #include "jptree.hpp"
  20. #include "jtime.ipp"
  21. #include "jencrypt.hpp"
  22. #include "junicode.hpp"
  23. #include "eclrtl.hpp"
  24. #include "deftype.hpp"
  25. #include <time.h>
  26. #include "mpbase.hpp"
  27. #include "daclient.hpp"
  28. #include "dadfs.hpp"
  29. #include "dafdesc.hpp"
  30. #include "dasds.hpp"
  31. #include "danqs.hpp"
  32. #include "dautils.hpp"
  33. #include "dllserver.hpp"
  34. #include "thorhelper.hpp"
  35. #include "workflow.hpp"
  36. #include "nbcd.hpp"
  37. #include "seclib.hpp"
  38. #include "wuerror.hpp"
  39. #include "wujobq.hpp"
  40. #define GLOBAL_WORKUNIT "global"
  41. #define SDS_LOCK_TIMEOUT (5*60*1000) // 5mins, 30s a bit short
  42. static int workUnitTraceLevel = 1;
  43. static StringBuffer &getXPath(StringBuffer &wuRoot, const char *wuid)
  44. {
  45. // MORE - can fold in the date
  46. return wuRoot.append("/WorkUnits/").append(wuid);
  47. }
  48. //To be called by eclserver, but esp etc. won't know, so we need to store it.
  49. static StringBuffer & appendLibrarySuffix(StringBuffer & suffix)
  50. {
  51. #ifdef _WIN32
  52. suffix.append("W");
  53. #else
  54. suffix.append("L");
  55. #endif
  56. #ifdef __64BIT__
  57. suffix.append("64");
  58. #else
  59. suffix.append("32");
  60. #endif
  61. return suffix;
  62. }
  63. typedef MapStringTo<bool> UniqueScopes;
  64. static void wuAccessError(const char *username, const char *action, const char *wuscope, const char *wuid, bool excpt, bool log)
  65. {
  66. StringBuffer err;
  67. err.append("Workunit Access Denied - action: ").append(action).append(" user:").append(username ? username : "<Unknown>");
  68. if (wuid)
  69. err.append(" workunit:").append(wuid);
  70. if (wuscope)
  71. err.append(" scope:").append(wuscope);
  72. //MORE - we would need more information passed in from outside if we want to make the audit message format the same as from higher level ESP calls
  73. SYSLOG(AUDIT_TYPE_ACCESS_FAILURE, err.str());
  74. if (log)
  75. LOG(MCuserError, "%s", err.str());
  76. if (excpt)
  77. throw MakeStringException(WUERR_AccessError, "%s", err.str());
  78. }
  79. static bool checkWuScopeSecAccess(const char *wuscope, ISecManager &secmgr, ISecUser *secuser, int required, const char *action, bool excpt, bool log)
  80. {
  81. bool ret=(!secuser) ? true : (secmgr.authorizeEx(RT_WORKUNIT_SCOPE, *secuser, wuscope)>=required);
  82. if (!ret && (log || excpt))
  83. wuAccessError(secuser ? secuser->getName() : NULL, action, wuscope, NULL, excpt, log);
  84. return ret;
  85. }
  86. static bool checkWuScopeListSecAccess(const char *wuscope, ISecResourceList *scopes, int required, const char *action, bool excpt, bool log)
  87. {
  88. if (!scopes)
  89. return true;
  90. bool ret=true;
  91. if (wuscope)
  92. {
  93. Owned<ISecResource> res=scopes->getResource(wuscope);
  94. if (!res || res->getAccessFlags()<required)
  95. ret=false;
  96. }
  97. else
  98. {
  99. for (int seq=0; ret && seq<scopes->count(); seq++)
  100. {
  101. ISecResource *res=scopes->queryResource(seq);
  102. if (res && res->getAccessFlags()<required)
  103. return false;
  104. }
  105. }
  106. if (!ret && (log || excpt))
  107. wuAccessError(NULL, action, wuscope, NULL, excpt, log);
  108. return ret;
  109. }
  110. static bool checkWuSecAccess(IConstWorkUnit &cw, ISecManager &secmgr, ISecUser *secuser, int required, const char *action, bool excpt, bool log)
  111. {
  112. SCMStringBuffer wuscope;
  113. bool ret=(!secuser) ? true : (secmgr.authorizeEx(RT_WORKUNIT_SCOPE, *secuser, cw.getWuScope(wuscope).str())>=required);
  114. if (!ret && (log || excpt))
  115. {
  116. SCMStringBuffer wuid;
  117. wuAccessError(secuser ? secuser->getName() : NULL, action, wuscope.str(), cw.getWuid(wuid).str(), excpt, log);
  118. }
  119. return ret;
  120. }
  121. static bool checkWuSecAccess(const char *wuid, ISecManager &secmgr, ISecUser *secuser, int required, const char *action, bool excpt, bool log)
  122. {
  123. StringBuffer wuRoot;
  124. Owned<IRemoteConnection> conn = querySDS().connect(getXPath(wuRoot, wuid).str(), myProcessSession(), 0, SDS_LOCK_TIMEOUT);
  125. if (conn)
  126. {
  127. Owned<IPropertyTree> ptree=conn->getRoot();
  128. return checkWuScopeSecAccess(ptree->queryProp("@scope"), secmgr, secuser, required, action, excpt, log);
  129. }
  130. if (log || excpt)
  131. wuAccessError(secuser ? secuser->getName() : NULL, action, "Unknown", NULL, excpt, log);
  132. return false;
  133. }
  134. void doDescheduleWorkkunit(char const * wuid)
  135. {
  136. StringBuffer xpath;
  137. xpath.append("*/*/*/");
  138. ncnameEscape(wuid, xpath);
  139. Owned<IRemoteConnection> conn = querySDS().connect("/Schedule", myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
  140. if(!conn) return;
  141. Owned<IPropertyTree> root = conn->getRoot();
  142. bool more;
  143. do more = root->removeProp(xpath.str()); while(more);
  144. }
  145. #define PROGRESS_FORMAT_V 2
  146. class CConstGraphProgress : public CInterface, implements IConstWUGraphProgress
  147. {
  148. class CGraphProgress : public CInterface, implements IWUGraphProgress
  149. {
  150. CConstGraphProgress &parent;
  151. public:
  152. IMPLEMENT_IINTERFACE;
  153. CGraphProgress(CConstGraphProgress &_parent) : parent(_parent)
  154. {
  155. parent.lockWrite();
  156. }
  157. ~CGraphProgress()
  158. {
  159. parent.unlock();
  160. }
  161. virtual IPropertyTree * queryProgressTree() { return parent.queryProgressTree(); }
  162. virtual WUGraphState queryGraphState() { return parent.queryGraphState(); }
  163. virtual WUGraphState queryNodeState(WUGraphIDType nodeId) { return parent.queryNodeState(nodeId); }
  164. virtual IWUGraphProgress * update() { return parent.update(); }
  165. virtual unsigned queryFormatVersion() { return parent.queryFormatVersion(); }
  166. virtual IPropertyTree & updateEdge(WUGraphIDType nodeId, const char *edgeId)
  167. {
  168. return parent.updateEdge(nodeId, edgeId);
  169. }
  170. virtual IPropertyTree & updateNode(WUGraphIDType nodeId, WUNodeIDType id)
  171. {
  172. return parent.updateNode(nodeId, id);
  173. }
  174. virtual void setGraphState(WUGraphState state)
  175. {
  176. parent.setGraphState(state);
  177. }
  178. virtual void setNodeState(WUGraphIDType nodeId, WUGraphState state)
  179. {
  180. parent.setNodeState(nodeId, state);
  181. }
  182. };
  183. IPropertyTree &updateElement(WUGraphIDType nodeId, const char *elemName, const char *id)
  184. {
  185. IPropertyTree *elem = NULL;
  186. if (!connectedWrite) lockWrite();
  187. StringBuffer path;
  188. path.append("node[@id=\"").append(nodeId).append("\"]");
  189. IPropertyTree *node = progress->queryPropTree(path.str());
  190. if (!node)
  191. {
  192. node = progress->addPropTree("node", createPTree());
  193. node->setPropInt("@id", (int)nodeId);
  194. elem = node->addPropTree(elemName, createPTree());
  195. elem->setProp("@id", id);
  196. }
  197. else
  198. {
  199. path.clear().append(elemName).append("[@id=\"").append(id).append("\"]");
  200. elem = node->queryPropTree(path.str());
  201. if (!elem)
  202. {
  203. elem = node->addPropTree(elemName, createPTree());
  204. elem->setProp("@id", id);
  205. }
  206. }
  207. return *elem;
  208. }
  209. public:
  210. IMPLEMENT_IINTERFACE;
  211. static void deleteWuidProgress(const char *wuid)
  212. {
  213. StringBuffer path("/GraphProgress/");
  214. path.append(wuid);
  215. Owned<IRemoteConnection> conn = querySDS().connect(path.str(), myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
  216. if (conn)
  217. conn->close(true);
  218. }
  219. CConstGraphProgress(const char *_wuid, const char *_graphName) : wuid(_wuid), graphName(_graphName)
  220. {
  221. rootPath.append("/GraphProgress/").append(wuid).append('/').append(graphName).append('/');
  222. connected = connectedWrite = false;
  223. formatVersion = 0;
  224. progress = NULL;
  225. }
  226. void connect()
  227. {
  228. conn.clear();
  229. packProgress(wuid,false);
  230. conn.setown(querySDS().connect(rootPath.str(), myProcessSession(), RTM_LOCK_READ|RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT));
  231. progress = conn->queryRoot();
  232. formatVersion = progress->getPropInt("@format");
  233. connected = true;
  234. }
  235. void lockWrite()
  236. {
  237. if (connectedWrite) return;
  238. // JCSMORE - look at using changeMode here.
  239. if (conn)
  240. conn.clear();
  241. else
  242. packProgress(wuid,false);
  243. conn.setown(querySDS().connect(rootPath.str(), myProcessSession(), RTM_LOCK_WRITE|RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT));
  244. progress = conn->queryRoot();
  245. if (!progress->hasChildren()) // i.e. blank.
  246. {
  247. formatVersion = PROGRESS_FORMAT_V;
  248. progress->setPropInt("@format", PROGRESS_FORMAT_V);
  249. }
  250. else
  251. formatVersion = progress->getPropInt("@format");
  252. connected = connectedWrite = true;
  253. }
  254. void unlock()
  255. {
  256. connected = false;
  257. connectedWrite = false;
  258. conn.clear();
  259. }
  260. IPropertyTree &updateNode(WUGraphIDType nodeId, WUNodeIDType id)
  261. {
  262. StringBuffer s;
  263. return updateElement(nodeId, "node", s.append(id).str());
  264. }
  265. IPropertyTree &updateEdge(WUGraphIDType nodeId, const char *edgeId)
  266. {
  267. return updateElement(nodeId, "edge", edgeId);
  268. }
  269. static bool getRunningGraph(const char *wuid, IStringVal &graphName, WUGraphIDType &subId)
  270. {
  271. StringBuffer path;
  272. Owned<IRemoteConnection> conn = querySDS().connect(path.append("/GraphProgress/").append(wuid).str(), myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT);
  273. if (!conn) return false;
  274. const char *name = conn->queryRoot()->queryProp("Running/@graph");
  275. if (name)
  276. {
  277. graphName.set(name);
  278. subId = conn->queryRoot()->getPropInt64("Running/@subId");
  279. return true;
  280. }
  281. else
  282. return false;
  283. }
  284. void setGraphState(WUGraphState state)
  285. {
  286. progress->setPropInt("@_state", (unsigned)state);
  287. }
  288. void setNodeState(WUGraphIDType nodeId, WUGraphState state)
  289. {
  290. if (!connectedWrite) lockWrite();
  291. StringBuffer path;
  292. path.append("node[@id=\"").append(nodeId).append("\"]");
  293. IPropertyTree *node = progress->queryPropTree(path.str());
  294. if (!node)
  295. {
  296. node = progress->addPropTree("node", createPTree());
  297. node->setPropInt("@id", (int)nodeId);
  298. }
  299. node->setPropInt("@_state", (unsigned)state);
  300. switch (state)
  301. {
  302. case WUGraphRunning:
  303. {
  304. StringBuffer path;
  305. Owned<IRemoteConnection> conn = querySDS().connect(path.append("/GraphProgress/").append(wuid).str(), myProcessSession(), RTM_LOCK_WRITE|RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT);
  306. IPropertyTree *running = conn->queryRoot()->setPropTree("Running", createPTree());
  307. running->setProp("@graph", graphName);
  308. running->setPropInt64("@subId", nodeId);
  309. break;
  310. }
  311. case WUGraphComplete:
  312. {
  313. StringBuffer path;
  314. Owned<IRemoteConnection> conn = querySDS().connect(path.append("/GraphProgress/").append(wuid).str(), myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
  315. conn->queryRoot()->removeProp("Running"); // only one thing running at any given time and one thing with lockWrite access
  316. break;
  317. }
  318. }
  319. }
  320. virtual IPropertyTree * queryProgressTree()
  321. {
  322. if (!connected) connect();
  323. return progress;
  324. }
  325. virtual WUGraphState queryGraphState()
  326. {
  327. return (WUGraphState)queryProgressTree()->getPropInt("@_state", (unsigned)WUGraphUnknown);
  328. }
  329. virtual WUGraphState queryNodeState(WUGraphIDType nodeId)
  330. {
  331. StringBuffer path;
  332. path.append("node[@id=\"").append(nodeId).append("\"]/@_state");
  333. return (WUGraphState)queryProgressTree()->getPropInt(path.str(), (unsigned)WUGraphUnknown);
  334. }
  335. virtual IWUGraphProgress * update()
  336. {
  337. return new CGraphProgress(*this);
  338. }
  339. virtual unsigned queryFormatVersion()
  340. {
  341. if (!connected) connect();
  342. return formatVersion;
  343. }
  344. static bool packProgress(const char *wuid,bool pack)
  345. {
  346. StringBuffer path;
  347. path.append("/GraphProgress/").append(wuid);
  348. Owned<IRemoteConnection> conn(querySDS().connect(path.str(), myProcessSession(), RTM_LOCK_WRITE|RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT));
  349. if (!conn)
  350. return false;
  351. Owned<IPropertyTree> newt;
  352. MemoryBuffer buf;
  353. IPropertyTree *root = conn->queryRoot();
  354. if (root->getPropBin("Packed",buf)) {
  355. if (pack)
  356. return true;
  357. newt.setown(createPTree(buf));
  358. IPropertyTree *running = root->queryPropTree("Running");
  359. if (running)
  360. newt->setPropTree("Running",createPTreeFromIPT(running));
  361. }
  362. else {
  363. if (!pack)
  364. return true;
  365. newt.setown(createPTree(wuid));
  366. IPropertyTree *running = root->queryPropTree("Running");
  367. if (running) {
  368. newt->setPropTree("Running",createPTreeFromIPT(running));
  369. root->removeTree(running);
  370. }
  371. root->serialize(buf);
  372. newt->setPropBin("Packed",buf.length(),buf.bufferBase());
  373. }
  374. root->setPropTree(NULL,newt.getClear());
  375. return true;
  376. }
  377. private:
  378. Owned<IRemoteConnection> conn;
  379. IPropertyTree* progress;
  380. StringAttr wuid, graphName;
  381. StringBuffer rootPath;
  382. bool connected, connectedWrite;
  383. unsigned formatVersion;
  384. };
  385. class CLocalWUTimeStamp : public CInterface, implements IConstWUTimeStamp
  386. {
  387. Owned<IPropertyTree> p;
  388. public:
  389. IMPLEMENT_IINTERFACE;
  390. CLocalWUTimeStamp(IPropertyTree *p);
  391. virtual IStringVal & getApplication(IStringVal & str) const;
  392. virtual IStringVal & getEvent(IStringVal & str) const;
  393. virtual IStringVal & getDate(IStringVal & dt) const;
  394. };
  395. class CLocalWUAppValue : public CInterface, implements IConstWUAppValue
  396. {
  397. Owned<IPropertyTree> p;
  398. StringBuffer prop;
  399. public:
  400. IMPLEMENT_IINTERFACE;
  401. CLocalWUAppValue(IPropertyTree *p,unsigned child);
  402. virtual IStringVal & getApplication(IStringVal & str) const;
  403. virtual IStringVal & getName(IStringVal & str) const;
  404. virtual IStringVal & getValue(IStringVal & dt) const;
  405. };
  406. template <typename T, typename IT> struct CachedTags
  407. {
  408. CachedTags(): cached(false) {}
  409. void load(IPropertyTree* p,const char* xpath)
  410. {
  411. if (!cached)
  412. {
  413. assertex(tags.length() == 0);
  414. Owned<IPropertyTreeIterator> r = p->getElements(xpath);
  415. for (r->first(); r->isValid(); r->next())
  416. {
  417. IPropertyTree *rp = &r->query();
  418. rp->Link();
  419. tags.append(*new T(rp));
  420. }
  421. cached = true;
  422. }
  423. }
  424. operator IArrayOf<IT>&() { return tags; }
  425. void kill()
  426. {
  427. cached = false;
  428. tags.kill();
  429. }
  430. bool cached;
  431. IArrayOf<IT> tags;
  432. };
  433. template <> struct CachedTags<CLocalWUAppValue, IConstWUAppValue>
  434. {
  435. CachedTags(): cached(false) {}
  436. void load(IPropertyTree* p,const char* xpath)
  437. {
  438. if (!cached)
  439. {
  440. assertex(tags.length() == 0);
  441. Owned<IPropertyTreeIterator> r = p->getElements(xpath);
  442. for (r->first(); r->isValid(); r->next())
  443. {
  444. IPropertyTree *rp = &r->query();
  445. Owned<IPropertyTreeIterator> v = rp->getElements("*");
  446. unsigned pos = 1;
  447. for (v->first(); v->isValid(); v->next())
  448. {
  449. rp->Link();
  450. tags.append(*new CLocalWUAppValue(rp,pos++));
  451. }
  452. }
  453. cached = true;
  454. }
  455. }
  456. operator IArrayOf<IConstWUAppValue>&() { return tags; }
  457. void kill()
  458. {
  459. cached = false;
  460. tags.kill();
  461. }
  462. bool cached;
  463. IArrayOf<IConstWUAppValue> tags;
  464. };
  465. class CLocalWorkUnit : public CInterface, implements IConstWorkUnit , implements ISDSSubscription, implements IExtendedWUInterface
  466. {
  467. friend StringBuffer &exportWorkUnitToXML(const IConstWorkUnit *wu, StringBuffer &str);
  468. friend void exportWorkUnitToXMLFile(const IConstWorkUnit *wu, const char * filename, unsigned extraXmlFlags);
  469. // NOTE - order is important - we need to construct connection before p and (especially) destruct after p
  470. Owned<IRemoteConnection> connection;
  471. Owned<IPropertyTree> p;
  472. bool dirty;
  473. bool connectAtRoot;
  474. mutable bool abortDirty;
  475. mutable bool abortState;
  476. mutable CriticalSection crit;
  477. mutable Owned<IWUQuery> query;
  478. mutable Owned<IWUWebServicesInfo> webServicesInfo;
  479. mutable Owned<IWURoxieQueryInfo> roxieQueryInfo;
  480. mutable Owned<IWorkflowItemIterator> workflowIterator;
  481. mutable bool workflowIteratorCached;
  482. mutable bool resultsCached;
  483. mutable bool temporariesCached;
  484. mutable bool variablesCached;
  485. mutable bool exceptionsCached;
  486. mutable bool pluginsCached;
  487. mutable bool librariesCached;
  488. mutable bool activitiesCached;
  489. mutable bool webServicesInfoCached;
  490. mutable bool roxieQueryInfoCached;
  491. mutable IArrayOf<IWUActivity> activities;
  492. mutable IArrayOf<IWUPlugin> plugins;
  493. mutable IArrayOf<IWULibrary> libraries;
  494. mutable IArrayOf<IWUException> exceptions;
  495. mutable IArrayOf<IWUGraph> graphs;
  496. mutable IArrayOf<IWUResult> results;
  497. mutable IArrayOf<IWUResult> temporaries;
  498. mutable IArrayOf<IWUResult> variables;
  499. mutable CachedTags<CLocalWUTimeStamp,IConstWUTimeStamp> timestamps;
  500. mutable CachedTags<CLocalWUAppValue,IConstWUAppValue> appvalues;
  501. mutable Owned<IUserDescriptor> userDesc;
  502. Mutex locked;
  503. Owned<ISecManager> secMgr;
  504. Owned<ISecUser> secUser;
  505. mutable Owned<IPropertyTree> cachedGraphs;
  506. public:
  507. IMPLEMENT_IINTERFACE;
  508. CLocalWorkUnit(IRemoteConnection *_conn, ISecManager *secmgr, ISecUser *secuser, const char *parentWuid = NULL);
  509. CLocalWorkUnit(IRemoteConnection *_conn, IPropertyTree* root, ISecManager *secmgr, ISecUser *secuser);
  510. ~CLocalWorkUnit();
  511. CLocalWorkUnit(const char *dummyWuid, const char *parentWuid, ISecManager *secmgr, ISecUser *secuser);
  512. ISecManager *querySecMgr(){return secMgr.get();}
  513. ISecUser *querySecUser(){return secUser.get();}
  514. void setSecIfcs(ISecManager *mgr, ISecUser*usr){secMgr.set(mgr); secUser.set(usr);}
  515. virtual bool aborting() const;
  516. virtual void forceReload();
  517. virtual WUAction getAction() const;
  518. virtual IStringVal& getActionEx(IStringVal & str) const;
  519. virtual IStringVal & getApplicationValue(const char * application, const char * propname, IStringVal & str) const;
  520. virtual int getApplicationValueInt(const char * application, const char * propname, int defVal) const;
  521. virtual IConstWUAppValueIterator & getApplicationValues() const;
  522. virtual bool hasWorkflow() const;
  523. virtual unsigned queryEventScheduledCount() const;
  524. virtual IPropertyTree * queryWorkflowTree() const;
  525. virtual IConstWorkflowItemIterator * getWorkflowItems() const;
  526. virtual IWorkflowItemArray * getWorkflowClone() const;
  527. virtual IConstLocalFileUploadIterator * getLocalFileUploads() const;
  528. virtual bool requiresLocalFileUpload() const;
  529. virtual bool getIsQueryService() const;
  530. virtual IStringVal & getClusterName(IStringVal & str) const;
  531. virtual unsigned getCombineQueries() const;
  532. virtual WUCompareMode getCompareMode() const;
  533. virtual IStringVal & getCustomerId(IStringVal & str) const;
  534. virtual bool hasDebugValue(const char * propname) const;
  535. virtual IStringVal & getDebugValue(const char * propname, IStringVal & str) const;
  536. virtual IStringIterator & getDebugValues() const;
  537. virtual IStringIterator & getDebugValues(const char *prop) const;
  538. virtual int getDebugValueInt(const char * propname, int defVal) const;
  539. virtual __int64 getDebugValueInt64(const char * propname, __int64 defVal) const;
  540. virtual bool getDebugValueBool(const char * propname, bool defVal) const;
  541. virtual unsigned getExceptionCount() const;
  542. virtual IConstWUExceptionIterator & getExceptions() const;
  543. virtual IConstWUResult * getGlobalByName(const char * name) const;
  544. virtual unsigned getGraphCount() const;
  545. virtual unsigned getSourceFileCount() const;
  546. virtual unsigned getResultCount() const;
  547. virtual unsigned getVariableCount() const;
  548. virtual unsigned getTimerCount() const;
  549. virtual unsigned getApplicationValueCount() const;
  550. virtual IConstWUGraphIterator & getGraphs(WUGraphType type) const;
  551. virtual IConstWUGraph * getGraph(const char *name) const;
  552. virtual IConstWUGraphProgress * getGraphProgress(const char * name) const;
  553. virtual IStringVal & getJobName(IStringVal & str) const;
  554. virtual IStringVal & getParentWuid(IStringVal & str) const;
  555. virtual IConstWUPlugin * getPluginByName(const char * name) const;
  556. virtual IConstWUPluginIterator & getPlugins() const;
  557. virtual IConstWULibraryIterator & getLibraries() const;
  558. virtual WUPriorityClass getPriority() const;
  559. virtual int getPriorityLevel() const;
  560. virtual int getPriorityValue() const;
  561. virtual IConstWUQuery * getQuery() const;
  562. virtual bool getRescheduleFlag() const;
  563. virtual IConstWUResult * getResultByName(const char * name) const;
  564. virtual IConstWUResult * getResultBySequence(unsigned seq) const;
  565. virtual unsigned getResultLimit() const;
  566. virtual IConstWUResultIterator & getResults() const;
  567. virtual IConstWUActivityIterator& getActivities() const;
  568. virtual IConstWUActivity* getActivity(__int64 id) const;
  569. virtual IStringVal & getScope(IStringVal & str) const;
  570. virtual IStringVal & getSecurityToken(IStringVal & str) const;
  571. virtual WUState getState() const;
  572. virtual IStringVal & getStateEx(IStringVal & str) const;
  573. virtual __int64 getAgentSession() const;
  574. virtual unsigned getAgentPID() const;
  575. virtual IStringVal & getStateDesc(IStringVal & str) const;
  576. virtual IConstWUResult * getTemporaryByName(const char * name) const;
  577. virtual IConstWUResultIterator & getTemporaries() const;
  578. virtual unsigned getTimerCount(const char * timerName, const char * instance) const;
  579. virtual unsigned getTimerDuration(const char * timerName, const char * instance) const;
  580. virtual IStringIterator & getTimers() const;
  581. virtual IConstWUTimeStampIterator & getTimeStamps() const;
  582. virtual IConstWUWebServicesInfo * getWebServicesInfo() const;
  583. virtual IConstWURoxieQueryInfo * getRoxieQueryInfo() const;
  584. virtual IStringVal & getXmlParams(IStringVal & params) const;
  585. virtual const IPropertyTree *getXmlParams() const;
  586. virtual unsigned __int64 getHash() const;
  587. virtual IStringIterator *getLogs(const char *type, const char *component) const;
  588. virtual IStringIterator *getProcesses(const char *type) const;
  589. virtual bool getWuDate(unsigned & year, unsigned & month, unsigned& day);
  590. virtual IStringVal & getSnapshot(IStringVal & str) const;
  591. virtual IStringVal & getTimeStamp(const char * name, const char * instance, IStringVal & str) const;
  592. virtual IStringVal & getUser(IStringVal & str) const;
  593. virtual IStringVal & getWuScope(IStringVal & str) const;
  594. virtual IConstWUResult * getVariableByName(const char * name) const;
  595. virtual IConstWUResultIterator & getVariables() const;
  596. virtual IStringVal & getWuid(IStringVal & str) const;
  597. virtual bool getRunningGraph(IStringVal &graphName, WUGraphIDType &subId) const;
  598. virtual bool isProtected() const;
  599. virtual bool isPausing() const;
  600. virtual bool isBilled() const;
  601. virtual IWorkUnit& lock();
  602. virtual bool reload();
  603. virtual void requestAbort();
  604. virtual void subscribe(WUSubscribeOptions options);
  605. virtual unsigned calculateHash(unsigned prevHash);
  606. virtual void copyWorkUnit(IConstWorkUnit *cached, bool all);
  607. virtual unsigned queryFileUsage(const char *filename) const;
  608. virtual bool getCloneable() const;
  609. virtual IUserDescriptor * queryUserDescriptor() const;
  610. virtual unsigned getCodeVersion() const;
  611. virtual unsigned getWuidVersion() const;
  612. virtual void getBuildVersion(IStringVal & buildVersion, IStringVal & eclVersion) const;
  613. virtual IPropertyTree * getDiskUsageStats();
  614. virtual IPropertyTreeIterator & getFileIterator() const;
  615. virtual bool archiveWorkUnit(const char *base,bool del,bool ignoredllerrors,bool deleteOwned);
  616. virtual void packWorkUnit(bool pack=true);
  617. virtual IJlibDateTime & getTimeScheduled(IJlibDateTime &val) const;
  618. virtual IPropertyTreeIterator & getFilesReadIterator() const;
  619. virtual void protect(bool protectMode);
  620. virtual IConstWULibrary * getLibraryByName(const char * name) const;
  621. virtual unsigned getDebugAgentListenerPort() const;
  622. virtual IStringVal & getDebugAgentListenerIP(IStringVal &ip) const;
  623. void clearExceptions();
  624. void commit();
  625. IWUException *createException();
  626. void setTimeStamp(const char *name, const char *instance, const char *event);
  627. void addTimeStamp(const char * name, const char * instance, const char *event);
  628. void addProcess(const char *type, const char *instance, const char *log);
  629. void setAction(WUAction action);
  630. void setApplicationValue(const char * application, const char * propname, const char * value, bool overwrite);
  631. void setApplicationValueInt(const char * application, const char * propname, int value, bool overwrite);
  632. void incEventScheduledCount();
  633. void setIsQueryService(bool value);
  634. void setCloneable(bool value);
  635. void setIsClone(bool value);
  636. void setClusterName(const char * value);
  637. void setCodeVersion(unsigned version, const char * buildVersion, const char * eclVersion);
  638. void setCombineQueries(unsigned combine);
  639. void setCompareMode(WUCompareMode value);
  640. void setCustomerId(const char * value);
  641. void setDebugValue(const char * propname, const char * value, bool overwrite);
  642. void setDebugValueInt(const char * propname, int value, bool overwrite);
  643. void setJobName(const char * value);
  644. void setPriority(WUPriorityClass cls);
  645. void setPriorityLevel(int level);
  646. void setRescheduleFlag(bool value);
  647. void setResultLimit(unsigned value);
  648. void setState(WUState state);
  649. void setStateEx(const char * text);
  650. void setAgentSession(__int64 sessionId);
  651. void setAgentPID(unsigned pid);
  652. void setSecurityToken(const char *value);
  653. void setTimerInfo(const char * name, const char * instance, unsigned ms, unsigned count, unsigned __int64 max);
  654. void setTracingValue(const char * propname, const char * value);
  655. void setTracingValueInt(const char * propname, int value);
  656. void setUser(const char * value);
  657. void setWuScope(const char * value);
  658. void setBilled(bool billed);
  659. void setSnapshot(const char * value);
  660. void setTimeStamp(const char *application, const char *instance, const char *event, bool add);
  661. void setDebugAgentListenerPort(unsigned port);
  662. void setDebugAgentListenerIP(const char * ip);
  663. void setXmlParams(const char *params);
  664. void setXmlParams(IPropertyTree *tree);
  665. void setHash(unsigned __int64 hash);
  666. IWorkflowItem* addWorkflowItem(unsigned wfid, WFType type, WFMode mode, unsigned success, unsigned failure, unsigned recovery, unsigned retriesAllowed, unsigned contingencyFor);
  667. IWorkflowItemIterator * updateWorkflowItems();
  668. void syncRuntimeWorkflow(IWorkflowItemArray * array);
  669. void resetWorkflow();
  670. void schedule();
  671. void deschedule();
  672. unsigned addLocalFileUpload(LocalFileUploadType type, char const * source, char const * destination, char const * eventTag);
  673. IWUResult * updateGlobalByName(const char * name);
  674. IWUGraph * updateGraph(const char * name);
  675. IWUQuery * updateQuery();
  676. IWUWebServicesInfo* updateWebServicesInfo(bool create);
  677. IWURoxieQueryInfo* updateRoxieQueryInfo(const char *wuid, const char *roxieClusterName);
  678. IWUActivity* updateActivity(__int64 id);
  679. IWUPlugin * updatePluginByName(const char * name);
  680. IWULibrary * updateLibraryByName(const char * name);
  681. IWUResult * updateResultByName(const char * name);
  682. IWUResult * updateResultBySequence(unsigned seq);
  683. IWUResult * updateTemporaryByName(const char * name);
  684. IWUResult * updateVariableByName(const char * name);
  685. void addFile(const char *fileName, StringArray *clusters, unsigned usageCount, WUFileKind fileKind, const char *graphOwner);
  686. void noteFileRead(IDistributedFile *file);
  687. void releaseFile(const char *fileName);
  688. void clearGraphProgress();
  689. void resetBeforeGeneration();
  690. void deleteTempFiles(const char *graph, bool deleteOwned, bool deleteJobOwned);
  691. void addDiskUsageStats(__int64 avgNodeUsage, unsigned minNode, __int64 minNodeUsage, unsigned maxNode, __int64 maxNodeUsage, __int64 graphId);
  692. void setTimeScheduled(const IJlibDateTime &val);
  693. // ILocalWorkUnit - used for debugging etc
  694. void loadXML(const char *xml);
  695. void serialize(MemoryBuffer &tgt);
  696. void deserialize(MemoryBuffer &src);
  697. IWorkUnit &lockRemote(bool commit);
  698. void unlockRemote(bool closing);
  699. void notify(SubscriptionId id, const char *xpath, SDSNotifyFlags flags, unsigned valueLen=0, const void *valueData=NULL);
  700. void abort();
  701. void cleanupAndDelete(bool deldll,bool deleteOwned);
  702. bool switchThorQueue(const char *cluster, IQueueSwitcher *qs);
  703. void setAllowedClusters(const char *value);
  704. IStringVal & getAllowedClusters(IStringVal & str) const;
  705. void remoteCheckAccess(IUserDescriptor *user, bool writeaccess) const;
  706. void setAllowAutoQueueSwitch(bool val);
  707. bool getAllowAutoQueueSwitch() const;
  708. void setLibraryInformation(const char * name, unsigned interfaceHash, unsigned definitionHash);
  709. private:
  710. void init();
  711. IWUGraph *createGraph();
  712. IWUResult *createResult();
  713. void loadGraphs() const;
  714. void loadResults() const;
  715. void loadTemporaries() const;
  716. void loadVariables() const;
  717. void loadExceptions() const;
  718. void loadPlugins() const;
  719. void loadLibraries() const;
  720. void loadClusters() const;
  721. void loadActivities() const;
  722. void unsubscribe();
  723. void checkAgentRunning(WUState & state);
  724. // MORE - the two could be a bit more similar...
  725. class CWorkUnitWatcher : public CInterface, implements ISDSSubscription
  726. {
  727. ISDSSubscription *parent; // not linked - it links me
  728. SubscriptionId change;
  729. bool sub;
  730. public:
  731. IMPLEMENT_IINTERFACE;
  732. CWorkUnitWatcher(ISDSSubscription *_parent, const char *wuid, bool _sub) : parent(_parent), sub(_sub)
  733. {
  734. StringBuffer wuRoot;
  735. getXPath(wuRoot, wuid);
  736. change = querySDS().subscribe(wuRoot.str(), *this, sub);
  737. }
  738. ~CWorkUnitWatcher()
  739. {
  740. assertex(change==0);
  741. }
  742. bool watchingChildren()
  743. {
  744. return sub;
  745. }
  746. void unsubscribe()
  747. {
  748. querySDS().unsubscribe(change);
  749. change = 0;
  750. }
  751. void notify(SubscriptionId id, const char *xpath, SDSNotifyFlags flags, unsigned valueLen, const void *valueData)
  752. {
  753. parent->notify(id, xpath, flags, valueLen, valueData);
  754. }
  755. };
  756. class CWorkUnitAbortWatcher : public CInterface, implements ISDSSubscription
  757. {
  758. CLocalWorkUnit *parent; // not linked - it links me
  759. SubscriptionId abort;
  760. public:
  761. IMPLEMENT_IINTERFACE;
  762. CWorkUnitAbortWatcher(CLocalWorkUnit *_parent, const char *wuid) : parent(_parent)
  763. {
  764. StringBuffer wuRoot;
  765. wuRoot.append("/WorkUnitAborts/").append(wuid);
  766. abort = querySDS().subscribe(wuRoot.str(), *this);
  767. }
  768. ~CWorkUnitAbortWatcher()
  769. {
  770. assertex(abort==0);
  771. }
  772. void unsubscribe()
  773. {
  774. querySDS().unsubscribe(abort);
  775. abort = 0;
  776. }
  777. void notify(SubscriptionId id, const char *xpath, SDSNotifyFlags flags, unsigned valueLen, const void *valueData)
  778. {
  779. parent->abort();
  780. }
  781. };
  782. Owned<CWorkUnitAbortWatcher> abortWatcher;
  783. Owned<CWorkUnitWatcher> changeWatcher;
  784. void ensureGraphsUnpacked ()
  785. {
  786. IPropertyTree *t = p->queryPropTree("PackedGraphs");
  787. MemoryBuffer buf;
  788. if (t&&t->getPropBin(NULL,buf)) {
  789. cachedGraphs.clear();
  790. IPropertyTree *st = createPTree(buf);
  791. if (st) {
  792. p->setPropTree("Graphs",st);
  793. p->removeTree(t);
  794. }
  795. }
  796. }
  797. };
  798. class CLockedWorkUnit : public CInterface, implements ILocalWorkUnit, implements IExtendedWUInterface
  799. {
  800. public:
  801. Owned<CLocalWorkUnit> c;
  802. IMPLEMENT_IINTERFACE;
  803. CLockedWorkUnit(CLocalWorkUnit *_c) : c(_c) {}
  804. ~CLockedWorkUnit()
  805. {
  806. if (workUnitTraceLevel > 1)
  807. {
  808. StringAttr x;
  809. StringAttrAdaptor strval(x);
  810. getWuid(strval);
  811. PrintLog("Releasing locked workunit %s", x.get());
  812. }
  813. if (c)
  814. c->unlockRemote(c->IsShared());
  815. }
  816. void setSecIfcs(ISecManager *mgr, ISecUser*usr){c->setSecIfcs(mgr, usr);}
  817. virtual IConstWorkUnit * unlock()
  818. {
  819. c->unlockRemote(c->IsShared());
  820. return c.getClear();
  821. }
  822. virtual bool aborting() const
  823. { return c->aborting(); }
  824. virtual void forceReload()
  825. { UNIMPLEMENTED; }
  826. virtual WUAction getAction() const
  827. { return c->getAction(); }
  828. virtual IStringVal& getActionEx(IStringVal & str) const
  829. { return c->getActionEx(str); }
  830. virtual IStringVal & getApplicationValue(const char * application, const char * propname, IStringVal & str) const
  831. { return c->getApplicationValue(application, propname, str); }
  832. virtual int getApplicationValueInt(const char * application, const char * propname, int defVal) const
  833. { return c->getApplicationValueInt(application, propname, defVal); }
  834. virtual IConstWUAppValueIterator & getApplicationValues() const
  835. { return c->getApplicationValues(); }
  836. virtual bool hasWorkflow() const
  837. { return c->hasWorkflow(); }
  838. virtual unsigned queryEventScheduledCount() const
  839. { return c->queryEventScheduledCount(); }
  840. virtual IPropertyTree * queryWorkflowTree() const
  841. { return c->queryWorkflowTree(); }
  842. virtual IConstWorkflowItemIterator * getWorkflowItems() const
  843. { return c->getWorkflowItems(); }
  844. virtual IWorkflowItemArray * getWorkflowClone() const
  845. { return c->getWorkflowClone(); }
  846. virtual bool requiresLocalFileUpload() const
  847. { return c->requiresLocalFileUpload(); }
  848. virtual IConstLocalFileUploadIterator * getLocalFileUploads() const
  849. { return c->getLocalFileUploads(); }
  850. virtual bool getIsQueryService() const
  851. { return c->getIsQueryService(); }
  852. virtual bool getCloneable() const
  853. { return c->getCloneable(); }
  854. virtual IUserDescriptor * queryUserDescriptor() const
  855. { return c->queryUserDescriptor(); }
  856. virtual IStringVal & getClusterName(IStringVal & str) const
  857. { return c->getClusterName(str); }
  858. virtual unsigned getCodeVersion() const
  859. { return c->getCodeVersion(); }
  860. virtual unsigned getWuidVersion() const
  861. { return c->getWuidVersion(); }
  862. virtual void getBuildVersion(IStringVal & buildVersion, IStringVal & eclVersion) const
  863. { c->getBuildVersion(buildVersion, eclVersion); }
  864. virtual unsigned getCombineQueries() const
  865. { return c->getCombineQueries(); }
  866. virtual WUCompareMode getCompareMode() const
  867. { return c->getCompareMode(); }
  868. virtual IStringVal & getCustomerId(IStringVal & str) const
  869. { return c->getCustomerId(str); }
  870. virtual bool hasDebugValue(const char * propname) const
  871. { return c->hasDebugValue(propname); }
  872. virtual IStringVal & getDebugValue(const char * propname, IStringVal & str) const
  873. { return c->getDebugValue(propname, str); }
  874. virtual int getDebugValueInt(const char * propname, int defVal) const
  875. { return c->getDebugValueInt(propname, defVal); }
  876. virtual __int64 getDebugValueInt64(const char * propname, __int64 defVal) const
  877. { return c->getDebugValueInt64(propname, defVal); }
  878. virtual bool getDebugValueBool(const char * propname, bool defVal) const
  879. { return c->getDebugValueBool(propname, defVal); }
  880. virtual IStringIterator & getDebugValues() const
  881. { return c->getDebugValues(NULL); }
  882. virtual IStringIterator & getDebugValues(const char *prop) const
  883. { return c->getDebugValues(prop); }
  884. virtual unsigned getExceptionCount() const
  885. { return c->getExceptionCount(); }
  886. virtual IConstWUExceptionIterator & getExceptions() const
  887. { return c->getExceptions(); }
  888. virtual unsigned getGraphCount() const
  889. { return c->getGraphCount(); }
  890. virtual unsigned getSourceFileCount() const
  891. { return c->getSourceFileCount(); }
  892. virtual unsigned getResultCount() const
  893. { return c->getResultCount(); }
  894. virtual unsigned getVariableCount() const
  895. { return c->getVariableCount(); }
  896. virtual unsigned getTimerCount() const
  897. { return c->getTimerCount(); }
  898. virtual unsigned getApplicationValueCount() const
  899. { return c->getApplicationValueCount(); }
  900. virtual IConstWUGraphIterator & getGraphs(WUGraphType type) const
  901. { return c->getGraphs(type); }
  902. virtual IConstWUGraph * getGraph(const char *name) const
  903. { return c->getGraph(name); }
  904. virtual IConstWUGraphProgress * getGraphProgress(const char * name) const
  905. { return c->getGraphProgress(name); }
  906. virtual IStringVal & getJobName(IStringVal & str) const
  907. { return c->getJobName(str); }
  908. virtual IStringVal & getParentWuid(IStringVal & str) const
  909. { return c->getParentWuid(str); }
  910. virtual IConstWUPlugin * getPluginByName(const char * name) const
  911. { return c->getPluginByName(name); }
  912. virtual IConstWUPluginIterator & getPlugins() const
  913. { return c->getPlugins(); }
  914. virtual IConstWULibrary* getLibraryByName(const char *name) const
  915. { return c->getLibraryByName(name); }
  916. virtual IConstWULibraryIterator & getLibraries() const
  917. { return c->getLibraries(); }
  918. virtual WUPriorityClass getPriority() const
  919. { return c->getPriority(); }
  920. virtual int getPriorityLevel() const
  921. { return c->getPriorityLevel(); }
  922. virtual int getPriorityValue() const
  923. { return c->getPriorityValue(); }
  924. virtual IConstWUQuery * getQuery() const
  925. { return c->getQuery(); }
  926. virtual IConstWUWebServicesInfo * getWebServicesInfo() const
  927. { return c->getWebServicesInfo(); }
  928. virtual IConstWURoxieQueryInfo* getRoxieQueryInfo() const
  929. { return c->getRoxieQueryInfo(); }
  930. virtual bool getRescheduleFlag() const
  931. { return c->getRescheduleFlag(); }
  932. virtual IConstWUResult * getResultByName(const char * name) const
  933. { return c->getResultByName(name); }
  934. virtual IConstWUResult * getResultBySequence(unsigned seq) const
  935. { return c->getResultBySequence(seq); }
  936. virtual unsigned getResultLimit() const
  937. { return c->getResultLimit(); }
  938. virtual IConstWUResultIterator & getResults() const
  939. { return c->getResults(); }
  940. virtual IConstWUActivityIterator & getActivities() const
  941. { return c->getActivities(); }
  942. virtual IConstWUActivity * getActivity(__int64 id) const
  943. { return c->getActivity(id); }
  944. virtual IStringVal & getScope(IStringVal & str) const
  945. { return c->getScope(str); }
  946. virtual IStringVal & getSecurityToken(IStringVal & str) const
  947. { return c->getSecurityToken(str); }
  948. virtual WUState getState() const
  949. { return c->getState(); }
  950. virtual IStringVal & getStateEx(IStringVal & str) const
  951. { return c->getStateEx(str); }
  952. virtual __int64 getAgentSession() const
  953. { return c->getAgentSession(); }
  954. virtual unsigned getAgentPID() const
  955. { return c->getAgentPID(); }
  956. virtual IStringVal & getStateDesc(IStringVal & str) const
  957. { return c->getStateDesc(str); }
  958. virtual bool getRunningGraph(IStringVal & graphName, WUGraphIDType & subId) const
  959. { return c->getRunningGraph(graphName, subId); }
  960. virtual unsigned getTimerCount(const char * timerName, const char * instance) const
  961. { return c->getTimerCount(timerName, instance); }
  962. virtual unsigned getTimerDuration(const char * timerName, const char * instance) const
  963. { return c->getTimerDuration(timerName, instance); }
  964. virtual IStringVal & getTimeStamp(const char * name, const char * instance, IStringVal & str) const
  965. { return c->getTimeStamp(name, instance, str); }
  966. virtual IStringIterator & getTimers() const
  967. { return c->getTimers(); }
  968. virtual IConstWUTimeStampIterator & getTimeStamps() const
  969. { return c->getTimeStamps(); }
  970. virtual bool getWuDate(unsigned & year, unsigned & month, unsigned& day)
  971. { return c->getWuDate(year,month,day);}
  972. virtual IStringVal & getSnapshot(IStringVal & str) const
  973. { return c->getSnapshot(str); }
  974. virtual IStringVal & getUser(IStringVal & str) const
  975. { return c->getUser(str); }
  976. virtual IStringVal & getWuScope(IStringVal & str) const
  977. { return c->getWuScope(str); }
  978. virtual IStringVal & getWuid(IStringVal & str) const
  979. { return c->getWuid(str); }
  980. virtual IConstWUResult * getGlobalByName(const char * name) const
  981. { return c->getGlobalByName(name); }
  982. virtual IConstWUResult * getTemporaryByName(const char * name) const
  983. { return c->getTemporaryByName(name); }
  984. virtual IConstWUResultIterator & getTemporaries() const
  985. { return c->getTemporaries(); }
  986. virtual IConstWUResult * getVariableByName(const char * name) const
  987. { return c->getVariableByName(name); }
  988. virtual IConstWUResultIterator & getVariables() const
  989. { return c->getVariables(); }
  990. virtual bool isProtected() const
  991. { return c->isProtected(); }
  992. virtual bool isPausing() const
  993. { return c->isPausing(); }
  994. virtual bool isBilled() const
  995. { return c->isBilled(); }
  996. virtual IWorkUnit & lock()
  997. { ((CInterface *)this)->Link(); return (IWorkUnit &) *this; }
  998. virtual bool reload()
  999. { UNIMPLEMENTED; }
  1000. virtual void subscribe(WUSubscribeOptions options)
  1001. { c->subscribe(options); }
  1002. virtual void requestAbort()
  1003. { c->requestAbort(); }
  1004. virtual unsigned calculateHash(unsigned prevHash)
  1005. { return c->calculateHash(prevHash); }
  1006. virtual void copyWorkUnit(IConstWorkUnit *cached, bool all)
  1007. { c->copyWorkUnit(cached, all); }
  1008. virtual bool archiveWorkUnit(const char *base,bool del,bool deldll,bool deleteOwned)
  1009. { return c->archiveWorkUnit(base,del,deldll,deleteOwned); }
  1010. virtual void packWorkUnit(bool pack)
  1011. { c->packWorkUnit(pack); }
  1012. virtual unsigned queryFileUsage(const char *filename) const
  1013. { return c->queryFileUsage(filename); }
  1014. virtual IJlibDateTime & getTimeScheduled(IJlibDateTime &val) const
  1015. { return c->getTimeScheduled(val); }
  1016. virtual unsigned getDebugAgentListenerPort() const
  1017. { return c->getDebugAgentListenerPort(); }
  1018. virtual IStringVal & getDebugAgentListenerIP(IStringVal &ip) const
  1019. { return c->getDebugAgentListenerIP(ip); }
  1020. virtual IStringVal & getXmlParams(IStringVal & params) const
  1021. { return c->getXmlParams(params); }
  1022. virtual const IPropertyTree *getXmlParams() const
  1023. { return c->getXmlParams(); }
  1024. virtual unsigned __int64 getHash() const
  1025. { return c->getHash(); }
  1026. virtual IStringIterator *getLogs(const char *type, const char *instance) const
  1027. { return c->getLogs(type, instance); }
  1028. virtual IStringIterator *getProcesses(const char *type) const
  1029. { return c->getProcesses(type); }
  1030. virtual void clearExceptions()
  1031. { c->clearExceptions(); }
  1032. virtual void commit()
  1033. { c->commit(); }
  1034. virtual IWUException * createException()
  1035. { return c->createException(); }
  1036. virtual void setTimeStamp(const char * name, const char * instance, const char *event)
  1037. { c->setTimeStamp(name, instance, event); }
  1038. virtual void addTimeStamp(const char * name, const char * instance, const char *event)
  1039. { c->addTimeStamp(name, instance, event); }
  1040. virtual void addProcess(const char *type, const char *instance, const char *log)
  1041. { c->addProcess(type, instance, log); }
  1042. virtual void protect(bool protectMode)
  1043. { c->protect(protectMode); }
  1044. virtual void setBilled(bool billed)
  1045. { c->setBilled(billed); }
  1046. virtual void setAction(WUAction action)
  1047. { c->setAction(action); }
  1048. virtual void setApplicationValue(const char * application, const char * propname, const char * value, bool overwrite)
  1049. { c->setApplicationValue(application, propname, value, overwrite); }
  1050. virtual void setApplicationValueInt(const char * application, const char * propname, int value, bool overwrite)
  1051. { c->setApplicationValueInt(application, propname, value, overwrite); }
  1052. virtual void incEventScheduledCount()
  1053. { c->incEventScheduledCount(); }
  1054. virtual void setIsQueryService(bool value)
  1055. { c->setIsQueryService(value); }
  1056. virtual void setCloneable(bool value)
  1057. { c->setCloneable(value); }
  1058. virtual void setIsClone(bool value)
  1059. { c->setIsClone(value); }
  1060. virtual void setClusterName(const char * value)
  1061. { c->setClusterName(value); }
  1062. virtual void setCodeVersion(unsigned version, const char * buildVersion, const char * eclVersion)
  1063. { c->setCodeVersion(version, buildVersion, eclVersion); }
  1064. virtual void setCombineQueries(unsigned combine)
  1065. { c->setCombineQueries(combine); }
  1066. virtual void setCompareMode(WUCompareMode value)
  1067. { c->setCompareMode(value); }
  1068. virtual void setCustomerId(const char * value)
  1069. { c->setCustomerId(value); }
  1070. virtual void setDebugValue(const char * propname, const char * value, bool overwrite)
  1071. { c->setDebugValue(propname, value, overwrite); }
  1072. virtual void setDebugValueInt(const char * propname, int value, bool overwrite)
  1073. { c->setDebugValueInt(propname, value, overwrite); }
  1074. virtual void setJobName(const char * value)
  1075. { c->setJobName(value); }
  1076. virtual void setPriority(WUPriorityClass cls)
  1077. { c->setPriority(cls); }
  1078. virtual void setPriorityLevel(int level)
  1079. { c->setPriorityLevel(level); }
  1080. virtual void setRescheduleFlag(bool value)
  1081. { c->setRescheduleFlag(value); }
  1082. virtual void setResultLimit(unsigned value)
  1083. { c->setResultLimit(value); }
  1084. virtual void setSecurityToken(const char *value)
  1085. { c->setSecurityToken(value); }
  1086. virtual void setState(WUState state)
  1087. { c->setState(state); }
  1088. virtual void setStateEx(const char * text)
  1089. { c->setStateEx(text); }
  1090. virtual void setAgentSession(__int64 sessionId)
  1091. { c->setAgentSession(sessionId); }
  1092. virtual void setAgentPID(unsigned pid)
  1093. { c->setAgentPID(pid); }
  1094. virtual void setTimerInfo(const char * name, const char * instance, unsigned ms, unsigned count, unsigned __int64 max)
  1095. { c->setTimerInfo(name, instance, ms, count, max); }
  1096. virtual void setTracingValue(const char * propname, const char * value)
  1097. { c->setTracingValue(propname, value); }
  1098. virtual void setTracingValueInt(const char * propname, int value)
  1099. { c->setTracingValueInt(propname, value); }
  1100. virtual void setUser(const char * value)
  1101. { c->setUser(value); }
  1102. virtual void setWuScope(const char * value)
  1103. {
  1104. if (value && *value)
  1105. {
  1106. ISecManager *secmgr=c->querySecMgr();
  1107. ISecUser *secusr=c->querySecUser();
  1108. if (!secmgr || !secusr)
  1109. throw MakeStringException(WUERR_SecurityNotAvailable, "Trying to change workunit scope without security interfaces available");
  1110. if (checkWuScopeSecAccess(value, *secmgr, secusr, SecAccess_Write, "Change Scope", true, true))
  1111. c->setWuScope(value);
  1112. }
  1113. }
  1114. virtual IWorkflowItem* addWorkflowItem(unsigned wfid, WFType type, WFMode mode, unsigned success, unsigned failure, unsigned recovery, unsigned retriesAllowed, unsigned contingencyFor)
  1115. { return c->addWorkflowItem(wfid, type, mode, success, failure, recovery, retriesAllowed, contingencyFor); }
  1116. virtual void syncRuntimeWorkflow(IWorkflowItemArray * array)
  1117. { c->syncRuntimeWorkflow(array); }
  1118. virtual IWorkflowItemIterator * updateWorkflowItems()
  1119. { return c->updateWorkflowItems(); }
  1120. virtual void resetWorkflow()
  1121. { c->resetWorkflow(); }
  1122. virtual void schedule()
  1123. { c->schedule(); }
  1124. virtual void deschedule()
  1125. { c->deschedule(); }
  1126. virtual unsigned addLocalFileUpload(LocalFileUploadType type, char const * source, char const * destination, char const * eventTag)
  1127. { return c->addLocalFileUpload(type, source, destination, eventTag); }
  1128. virtual IWUResult * updateGlobalByName(const char * name)
  1129. { return c->updateGlobalByName(name); }
  1130. virtual IWUGraph * updateGraph(const char * name)
  1131. { return c->updateGraph(name); }
  1132. virtual IWUQuery * updateQuery()
  1133. { return c->updateQuery(); }
  1134. virtual IWUWebServicesInfo * updateWebServicesInfo(bool create)
  1135. { return c->updateWebServicesInfo(create); }
  1136. virtual IWURoxieQueryInfo * updateRoxieQueryInfo(const char *wuid, const char *roxieClusterName)
  1137. { return c->updateRoxieQueryInfo(wuid, roxieClusterName); }
  1138. virtual IWUActivity * updateActivity(__int64 id)
  1139. { return c->updateActivity(id); }
  1140. virtual IWUPlugin * updatePluginByName(const char * name)
  1141. { return c->updatePluginByName(name); }
  1142. virtual IWULibrary * updateLibraryByName(const char * name)
  1143. { return c->updateLibraryByName(name); }
  1144. virtual IWUResult * updateResultByName(const char * name)
  1145. { return c->updateResultByName(name); }
  1146. virtual IWUResult * updateResultBySequence(unsigned seq)
  1147. { return c->updateResultBySequence(seq); }
  1148. virtual IWUResult * updateTemporaryByName(const char * name)
  1149. { return c->updateTemporaryByName(name); }
  1150. virtual IWUResult * updateVariableByName(const char * name)
  1151. { return c->updateVariableByName(name); }
  1152. virtual void addFile(const char *fileName, StringArray *clusters, unsigned usageCount, WUFileKind fileKind, const char *graphOwner)
  1153. { c->addFile(fileName, clusters, usageCount, fileKind, graphOwner); }
  1154. virtual void noteFileRead(IDistributedFile *file)
  1155. { c->noteFileRead(file); }
  1156. virtual void releaseFile(const char *fileName)
  1157. { c->releaseFile(fileName); }
  1158. virtual void clearGraphProgress()
  1159. { c->clearGraphProgress(); }
  1160. virtual void resetBeforeGeneration()
  1161. { c->resetBeforeGeneration(); }
  1162. virtual void deleteTempFiles(const char *graph, bool deleteOwned, bool deleteJobOwned)
  1163. { c->deleteTempFiles(graph, deleteOwned, deleteJobOwned); }
  1164. virtual void addDiskUsageStats(__int64 avgNodeUsage, unsigned minNode, __int64 minNodeUsage, unsigned maxNode, __int64 maxNodeUsage, __int64 graphId)
  1165. { c->addDiskUsageStats(avgNodeUsage, minNode, minNodeUsage, maxNode, maxNodeUsage, graphId); }
  1166. virtual IPropertyTree * getDiskUsageStats()
  1167. { return c->getDiskUsageStats(); }
  1168. virtual IPropertyTreeIterator & getFileIterator() const
  1169. { return c->getFileIterator(); }
  1170. virtual IPropertyTreeIterator & getFilesReadIterator() const
  1171. { return c->getFilesReadIterator(); }
  1172. virtual void setSnapshot(const char * value)
  1173. { c->setSnapshot(value); }
  1174. virtual void setTimeScheduled(const IJlibDateTime &val)
  1175. { c->setTimeScheduled(val); }
  1176. virtual void setDebugAgentListenerPort(unsigned port)
  1177. { c->setDebugAgentListenerPort(port); }
  1178. virtual void setDebugAgentListenerIP(const char * ip)
  1179. { c->setDebugAgentListenerIP(ip); }
  1180. virtual void setXmlParams(const char *params)
  1181. { c->setXmlParams(params); }
  1182. virtual void setXmlParams(IPropertyTree *tree)
  1183. { c->setXmlParams(tree); }
  1184. virtual void setHash(unsigned __int64 hash)
  1185. { c->setHash(hash); }
  1186. // ILocalWorkUnit - used for debugging etc
  1187. virtual void loadXML(const char *xml)
  1188. { c->loadXML(xml); }
  1189. virtual void serialize(MemoryBuffer &tgt)
  1190. { c->serialize(tgt); }
  1191. virtual void deserialize(MemoryBuffer &src)
  1192. { c->deserialize(src); }
  1193. virtual bool switchThorQueue(const char *cluster, IQueueSwitcher *qs)
  1194. { return c->switchThorQueue(cluster,qs); }
  1195. virtual void setAllowedClusters(const char *value)
  1196. { c->setAllowedClusters(value); }
  1197. virtual IStringVal& getAllowedClusters(IStringVal &str) const
  1198. { return c->getAllowedClusters(str); }
  1199. virtual void remoteCheckAccess(IUserDescriptor *user, bool writeaccess) const
  1200. { c->remoteCheckAccess(user,writeaccess); }
  1201. virtual void setAllowAutoQueueSwitch(bool val)
  1202. { c->setAllowAutoQueueSwitch(val); }
  1203. virtual bool getAllowAutoQueueSwitch() const
  1204. { return c->getAllowAutoQueueSwitch(); }
  1205. virtual void setLibraryInformation(const char * name, unsigned interfaceHash, unsigned definitionHash)
  1206. { c->setLibraryInformation(name, interfaceHash, definitionHash); }
  1207. virtual void setResultInt(const char * name, unsigned sequence, __int64 val)
  1208. {
  1209. Owned<IWUResult> r = updateResult(name, sequence);
  1210. if (r)
  1211. {
  1212. r->setResultInt(val);
  1213. r->setResultStatus(ResultStatusCalculated);
  1214. }
  1215. }
  1216. virtual void setResultUInt(const char * name, unsigned sequence, unsigned __int64 val)
  1217. {
  1218. Owned<IWUResult> r = updateResult(name, sequence);
  1219. if (r)
  1220. {
  1221. r->setResultUInt(val);
  1222. r->setResultStatus(ResultStatusCalculated);
  1223. }
  1224. }
  1225. virtual void setResultReal(const char *name, unsigned sequence, double val)
  1226. {
  1227. Owned<IWUResult> r = updateResult(name, sequence);
  1228. if (r)
  1229. {
  1230. r->setResultReal(val);
  1231. r->setResultStatus(ResultStatusCalculated);
  1232. }
  1233. }
  1234. virtual void setResultVarString(const char * stepname, unsigned sequence, const char *val)
  1235. {
  1236. setResultString(stepname, sequence, strlen(val), val);
  1237. }
  1238. virtual void setResultVarUnicode(const char * stepname, unsigned sequence, UChar const *val)
  1239. {
  1240. setResultUnicode(stepname, sequence, rtlUnicodeStrlen(val), val);
  1241. }
  1242. virtual void setResultString(const char * stepname, unsigned sequence, int len, const char *val)
  1243. {
  1244. doSetResultString(type_string, stepname, sequence, len, val);
  1245. }
  1246. virtual void setResultData(const char * stepname, unsigned sequence, int len, const void *val)
  1247. {
  1248. doSetResultString(type_data, stepname, sequence, len, (const char *)val);
  1249. }
  1250. virtual void setResultRaw(const char * name, unsigned sequence, int len, const void *val)
  1251. {
  1252. Owned<IWUResult> r = updateResult(name, sequence);
  1253. if (r)
  1254. {
  1255. r->setResultRaw(len, val, ResultFormatRaw);
  1256. r->setResultStatus(ResultStatusCalculated);
  1257. }
  1258. }
  1259. virtual void setResultSet(const char * name, unsigned sequence, bool isAll, size32_t len, const void *val, ISetToXmlTransformer *)
  1260. {
  1261. Owned<IWUResult> r = updateResult(name, sequence);
  1262. if (r)
  1263. {
  1264. r->setResultIsAll(isAll);
  1265. r->setResultRaw(len, val, ResultFormatRaw);
  1266. r->setResultStatus(ResultStatusCalculated);
  1267. }
  1268. }
  1269. virtual void setResultUnicode(const char * name, unsigned sequence, int len, UChar const * val)
  1270. {
  1271. Owned<IWUResult> r = updateResult(name, sequence);
  1272. if (r)
  1273. {
  1274. r->setResultUnicode((char const *)val, len);
  1275. r->setResultStatus(ResultStatusCalculated);
  1276. }
  1277. }
  1278. virtual void setResultBool(const char *name, unsigned sequence, bool val)
  1279. {
  1280. Owned<IWUResult> r = updateResult(name, sequence);
  1281. if (r)
  1282. {
  1283. r->setResultBool(val);
  1284. r->setResultStatus(ResultStatusCalculated);
  1285. }
  1286. }
  1287. virtual void setResultDecimal(const char *name, unsigned sequence, int len, int precision, bool isSigned, const void *val)
  1288. {
  1289. Owned<IWUResult> r = updateResult(name, sequence);
  1290. if (r)
  1291. {
  1292. r->setResultDecimal(val, len);
  1293. r->setResultStatus(ResultStatusCalculated);
  1294. }
  1295. }
  1296. virtual void setResultDataset(const char * name, unsigned sequence, size32_t len, const void *val, unsigned numRows, bool extend)
  1297. {
  1298. Owned<IWUResult> r = updateResult(name, sequence);
  1299. if (r)
  1300. {
  1301. __int64 totalRows = numRows;
  1302. if (extend)
  1303. {
  1304. totalRows += r->getResultRowCount();
  1305. r->addResultRaw(len, val, ResultFormatRaw);
  1306. }
  1307. else
  1308. r->setResultRaw(len, val, ResultFormatRaw);
  1309. r->setResultStatus(ResultStatusCalculated);
  1310. r->setResultRowCount(totalRows);
  1311. r->setResultTotalRowCount(totalRows);
  1312. }
  1313. }
  1314. protected:
  1315. IWUResult *updateResult(const char *name, unsigned sequence)
  1316. {
  1317. Owned <IWUResult> result = updateWorkUnitResult(this, name, sequence);
  1318. if (result)
  1319. {
  1320. SCMStringBuffer rname;
  1321. if (!result->getResultName(rname).length())
  1322. result->setResultName(name);
  1323. }
  1324. return result.getClear();
  1325. }
  1326. void doSetResultString(type_t type, const char *name, unsigned sequence, int len, const char *val)
  1327. {
  1328. Owned<IWUResult> r = updateResult(name, sequence);
  1329. if (r)
  1330. {
  1331. r->setResultString(val, len);
  1332. r->setResultStatus(ResultStatusCalculated);
  1333. }
  1334. }
  1335. };
  1336. class CLocalWUAssociated : public CInterface, implements IConstWUAssociatedFile
  1337. {
  1338. Owned<IPropertyTree> p;
  1339. public:
  1340. IMPLEMENT_IINTERFACE;
  1341. CLocalWUAssociated(IPropertyTree *p);
  1342. virtual WUFileType getType() const;
  1343. virtual IStringVal & getDescription(IStringVal & ret) const;
  1344. virtual IStringVal & getIp(IStringVal & ret) const;
  1345. virtual IStringVal & getName(IStringVal & ret) const;
  1346. virtual IStringVal & getNameTail(IStringVal & ret) const;
  1347. virtual unsigned getCrc() const;
  1348. };
  1349. class CLocalWUQuery : public CInterface, implements IWUQuery
  1350. {
  1351. Owned<IPropertyTree> p;
  1352. mutable IArrayOf<IConstWUAssociatedFile> associated;
  1353. mutable CriticalSection crit;
  1354. mutable bool associatedCached;
  1355. private:
  1356. void addSpecialCaseAssociated(WUFileType type, const char * propname, unsigned crc) const;
  1357. void loadAssociated() const;
  1358. public:
  1359. IMPLEMENT_IINTERFACE;
  1360. CLocalWUQuery(IPropertyTree *p);
  1361. virtual WUQueryType getQueryType() const;
  1362. virtual IStringVal& getQueryText(IStringVal &str) const;
  1363. virtual IStringVal& getQueryShortText(IStringVal &str) const;
  1364. virtual IStringVal& getQueryName(IStringVal &str) const;
  1365. virtual IStringVal & getQueryMainDefinition(IStringVal & str) const;
  1366. virtual IStringVal& getQueryDllName(IStringVal &str) const;
  1367. virtual unsigned getQueryDllCrc() const;
  1368. virtual IStringVal& getQueryCppName(IStringVal &str) const;
  1369. virtual IStringVal& getQueryResTxtName(IStringVal &str) const;
  1370. virtual IConstWUAssociatedFile * getAssociatedFile(WUFileType type, unsigned index) const;
  1371. virtual IConstWUAssociatedFileIterator& getAssociatedFiles() const;
  1372. virtual void setQueryType(WUQueryType qt);
  1373. virtual void setQueryText(const char *pstr);
  1374. virtual void setQueryName(const char *);
  1375. virtual void setQueryMainDefinition(const char * str);
  1376. virtual void addAssociatedFile(WUFileType type, const char * name, const char * ip, const char * desc, unsigned crc);
  1377. virtual void removeAssociatedFiles();
  1378. };
  1379. class CLocalWUWebServicesInfo : public CInterface, implements IWUWebServicesInfo
  1380. {
  1381. Owned<IPropertyTree> p;
  1382. mutable CriticalSection crit;
  1383. private:
  1384. public:
  1385. IMPLEMENT_IINTERFACE;
  1386. CLocalWUWebServicesInfo(IPropertyTree *p);
  1387. virtual IStringVal& getModuleName(IStringVal &str) const;
  1388. virtual IStringVal& getAttributeName(IStringVal &str) const;
  1389. virtual IStringVal& getDefaultName(IStringVal &str) const;
  1390. virtual IStringVal& getInfo(const char *name, IStringVal &str) const;
  1391. virtual unsigned getWebServicesCRC() const;
  1392. virtual void setModuleName(const char *);
  1393. virtual void setAttributeName(const char *);
  1394. virtual void setDefaultName(const char *);
  1395. virtual void setInfo(const char *name, const char *info);
  1396. virtual void setWebServicesCRC(unsigned);
  1397. };
  1398. class CLocalWURoxieQueryInfo : public CInterface, implements IWURoxieQueryInfo
  1399. {
  1400. Owned<IPropertyTree> p;
  1401. mutable CriticalSection crit;
  1402. private:
  1403. public:
  1404. IMPLEMENT_IINTERFACE;
  1405. CLocalWURoxieQueryInfo(IPropertyTree *p);
  1406. virtual IStringVal& getQueryInfo(IStringVal &str) const;
  1407. virtual IStringVal& getDefaultPackageInfo(IStringVal &str) const;
  1408. virtual IStringVal& getRoxieClusterName(IStringVal &str) const;
  1409. virtual IStringVal& getWuid(IStringVal &str) const;
  1410. virtual void setQueryInfo(const char *info);
  1411. virtual void setDefaultPackageInfo(const char *, int len);
  1412. virtual void setRoxieClusterName(const char *str);
  1413. virtual void setWuid(const char *str);
  1414. };
  1415. class CLocalWUResult : public CInterface, implements IWUResult
  1416. {
  1417. friend class CLocalWorkUnit;
  1418. Owned<IPropertyTree> p;
  1419. void getSchema(TypeInfoArray &types, StringAttrArray &names, IStringVal * ecl=NULL) const;
  1420. public:
  1421. IMPLEMENT_IINTERFACE;
  1422. CLocalWUResult(IPropertyTree *props);
  1423. ~CLocalWUResult() { try { p.clear(); } catch (IException *E) {E->Release();}}
  1424. virtual WUResultStatus getResultStatus() const;
  1425. virtual IStringVal& getResultName(IStringVal &str) const;
  1426. virtual int getResultSequence() const;
  1427. virtual bool isResultScalar() const;
  1428. virtual IStringVal& getResultXml(IStringVal &str) const;
  1429. virtual unsigned getResultFetchSize() const;
  1430. virtual __int64 getResultTotalRowCount() const;
  1431. virtual __int64 getResultRowCount() const;
  1432. virtual void getResultDataset(IStringVal & ecl, IStringVal & defs) const;
  1433. virtual IStringVal& getResultLogicalName(IStringVal &ecl) const;
  1434. virtual IStringVal& getResultKeyField(IStringVal& ecl) const;
  1435. virtual unsigned getResultRequestedRows() const;
  1436. virtual __int64 getResultInt() const;
  1437. virtual bool getResultBool() const;
  1438. virtual double getResultReal() const;
  1439. virtual IStringVal& getResultString(IStringVal & str) const;
  1440. virtual IDataVal& getResultRaw(IDataVal & data, IXmlToRawTransformer * xmlTransformer, ICsvToRawTransformer * csvTransformer) const;
  1441. virtual IDataVal& getResultUnicode(IDataVal & data) const;
  1442. virtual void getResultDecimal(void * val, unsigned length, unsigned precision, bool isSigned) const;
  1443. virtual IStringVal& getResultEclSchema(IStringVal & str) const;
  1444. virtual __int64 getResultRawSize(IXmlToRawTransformer * xmlTransformer, ICsvToRawTransformer * csvTransformer) const;
  1445. virtual IDataVal& getResultRaw(IDataVal & data, __int64 from, __int64 length, IXmlToRawTransformer * xmlTransformer, ICsvToRawTransformer * csvTransformer) const;
  1446. virtual IStringVal& getResultRecordSizeEntry(IStringVal & str) const;
  1447. virtual IStringVal& getResultTransformerEntry(IStringVal & str) const;
  1448. virtual __int64 getResultRowLimit() const;
  1449. virtual IStringVal& getResultFilename(IStringVal & str) const;
  1450. virtual WUResultFormat getResultFormat() const;
  1451. virtual unsigned getResultHash() const;
  1452. virtual bool getResultIsAll() const;
  1453. // interface IWUResult
  1454. virtual void setResultStatus(WUResultStatus status);
  1455. virtual void setResultName(const char *name);
  1456. virtual void setResultSequence(unsigned seq);
  1457. virtual void setResultSchemaRaw(unsigned len, const void *schema);
  1458. virtual void setResultScalar(bool isScalar);
  1459. virtual void setResultRaw(unsigned len, const void *xml, WUResultFormat format);
  1460. virtual void setResultFetchSize(unsigned rows); // 0 means file-loaded
  1461. virtual void setResultTotalRowCount(__int64 rows); // -1 means unknown
  1462. virtual void setResultRowCount(__int64 rows);
  1463. virtual void setResultDataset(const char *ecl, const char *defs);
  1464. virtual void setResultLogicalName(const char *logicalName);
  1465. virtual void setResultKeyField(const char * name);
  1466. virtual void setResultRequestedRows(unsigned req);
  1467. virtual void setResultRecordSizeEntry(const char * val);
  1468. virtual void setResultTransformerEntry(const char * val);
  1469. virtual void setResultInt(__int64 val);
  1470. virtual void setResultReal(double val);
  1471. virtual void setResultBool(bool val);
  1472. virtual void setResultString(const char * val, unsigned length);
  1473. virtual void setResultUnicode(const void * val, unsigned length);
  1474. virtual void setResultData(const void * val, unsigned length);
  1475. virtual void setResultDecimal(const void * val, unsigned length);
  1476. virtual void addResultRaw(unsigned len, const void * data, WUResultFormat format);
  1477. virtual void setResultRowLimit(__int64 value);
  1478. virtual void setResultFilename(const char * name);
  1479. virtual void setResultUInt(unsigned __int64 val);
  1480. virtual void setResultIsAll(bool value);
  1481. virtual void setResultFormat(WUResultFormat format);
  1482. virtual void setResultXML(const char *val);
  1483. virtual void setResultRow(unsigned len, const void * data);
  1484. };
  1485. class CLocalWUPlugin : public CInterface, implements IWUPlugin
  1486. {
  1487. Owned<IPropertyTree> p;
  1488. public:
  1489. IMPLEMENT_IINTERFACE;
  1490. CLocalWUPlugin(IPropertyTree *p);
  1491. virtual IStringVal& getPluginName(IStringVal &str) const;
  1492. virtual IStringVal& getPluginVersion(IStringVal &str) const;
  1493. virtual bool getPluginThor() const;
  1494. virtual bool getPluginHole() const;
  1495. virtual void setPluginName(const char *str);
  1496. virtual void setPluginVersion(const char *str);
  1497. virtual void setPluginThor(bool on);
  1498. virtual void setPluginHole(bool on);
  1499. };
  1500. class CLocalWULibrary : public CInterface, implements IWULibrary
  1501. {
  1502. Owned<IPropertyTree> p;
  1503. public:
  1504. IMPLEMENT_IINTERFACE;
  1505. CLocalWULibrary(IPropertyTree *p);
  1506. virtual IStringVal & getName(IStringVal & str) const;
  1507. virtual IConstWULibraryActivityIterator * getActivities() const;
  1508. virtual void setName(const char * str);
  1509. virtual void addActivity(unsigned id);
  1510. };
  1511. class CLocalWUGraph : public CInterface, implements IWUGraph
  1512. {
  1513. Owned<IPropertyTree> p;
  1514. mutable Linked<IConstWUGraphProgress> progress;
  1515. StringAttr wuid;
  1516. void mergeProgress(IPropertyTree &tree, IPropertyTree &progressTree, const unsigned &progressV) const;
  1517. public:
  1518. IMPLEMENT_IINTERFACE;
  1519. CLocalWUGraph(IPropertyTree *p, const char *wuid);
  1520. virtual IStringVal & getXGMML(IStringVal & ret, bool mergeProgress) const;
  1521. virtual IStringVal & getDOT(IStringVal & ret) const;
  1522. virtual IStringVal & getName(IStringVal & ret) const;
  1523. virtual IStringVal & getLabel(IStringVal & ret) const;
  1524. virtual IStringVal & getTypeName(IStringVal & ret) const;
  1525. virtual WUGraphType getType() const;
  1526. virtual IPropertyTree * getXGMMLTree(bool mergeProgress) const;
  1527. virtual bool isValid() const;
  1528. virtual void setName(const char *str);
  1529. virtual void setType(WUGraphType type);
  1530. virtual void setXGMML(const char *str);
  1531. virtual void setXGMMLTree(IPropertyTree * tree);
  1532. };
  1533. class CLocalWUActivity : public CInterface, implements IWUActivity
  1534. {
  1535. Owned<IPropertyTree> p;
  1536. public:
  1537. IMPLEMENT_IINTERFACE;
  1538. CLocalWUActivity(IPropertyTree *p, __int64 id = 0);
  1539. virtual __int64 getId() const;
  1540. virtual unsigned getKind() const;
  1541. virtual IStringVal & getHelper(IStringVal & ret) const;
  1542. virtual void setKind(unsigned id);
  1543. virtual void setHelper(const char * str);
  1544. };
  1545. class CLocalWUException : public CInterface, implements IWUException
  1546. {
  1547. Owned<IPropertyTree> p;
  1548. public:
  1549. IMPLEMENT_IINTERFACE;
  1550. CLocalWUException(IPropertyTree *p);
  1551. virtual IStringVal& getExceptionSource(IStringVal &str) const;
  1552. virtual IStringVal& getExceptionMessage(IStringVal &str) const;
  1553. virtual unsigned getExceptionCode() const;
  1554. virtual WUExceptionSeverity getSeverity() const;
  1555. virtual IStringVal & getTimeStamp(IStringVal & dt) const;
  1556. virtual IStringVal & getExceptionFileName(IStringVal & str) const;
  1557. virtual unsigned getExceptionLineNo() const;
  1558. virtual unsigned getExceptionColumn() const;
  1559. virtual void setExceptionSource(const char *str);
  1560. virtual void setExceptionMessage(const char *str);
  1561. virtual void setExceptionCode(unsigned code);
  1562. virtual void setSeverity(WUExceptionSeverity level);
  1563. virtual void setTimeStamp(const char * dt);
  1564. virtual void setExceptionFileName(const char *str);
  1565. virtual void setExceptionLineNo(unsigned r);
  1566. virtual void setExceptionColumn(unsigned c);
  1567. };
  1568. //==========================================================================================
  1569. extern WORKUNIT_API bool isSpecialResultSequence(unsigned sequence)
  1570. {
  1571. switch (sequence)
  1572. {
  1573. case ResultSequenceInternal:
  1574. case ResultSequenceOnce:
  1575. case ResultSequencePersist:
  1576. case ResultSequenceStored:
  1577. return true;
  1578. default:
  1579. assertex(sequence <= INT_MAX);
  1580. return false;
  1581. }
  1582. }
  1583. struct mapEnums { int val; const char *str; };
  1584. const char *getEnumText(int value, mapEnums *map)
  1585. {
  1586. const char *defval = map->str;
  1587. while (map->str)
  1588. {
  1589. if (value==map->val)
  1590. return map->str;
  1591. map++;
  1592. }
  1593. assertex(!"Unexpected value in setEnum");
  1594. return defval;
  1595. }
  1596. void setEnum(IPropertyTree *p, const char *propname, int value, mapEnums *map)
  1597. {
  1598. const char *defval = map->str;
  1599. while (map->str)
  1600. {
  1601. if (value==map->val)
  1602. {
  1603. p->setProp(propname, map->str);
  1604. return;
  1605. }
  1606. map++;
  1607. }
  1608. assertex(!"Unexpected value in setEnum");
  1609. p->setProp(propname, defval);
  1610. }
  1611. static int getEnum(const char *v, mapEnums *map)
  1612. {
  1613. if (v)
  1614. {
  1615. while (map->str)
  1616. {
  1617. if (stricmp(v, map->str)==0)
  1618. return map->val;
  1619. map++;
  1620. }
  1621. assertex(!"Unexpected value in getEnum");
  1622. }
  1623. return 0;
  1624. }
  1625. static int getEnum(IPropertyTree *p, const char *propname, mapEnums *map)
  1626. {
  1627. return getEnum(p->queryProp(propname),map);
  1628. }
  1629. //==========================================================================================
  1630. class CConstWUArrayIterator : public CInterface, implements IConstWorkUnitIterator
  1631. {
  1632. IArrayOf<IConstWorkUnit> w;
  1633. CArrayIteratorOf<IConstWorkUnit,IConstWorkUnitIterator> it;
  1634. public:
  1635. IMPLEMENT_IINTERFACE;
  1636. CConstWUArrayIterator(IRemoteConnection *conn, IArrayOf<IPropertyTree> &trees, ISecManager *secmgr=NULL, ISecUser *secuser=NULL)
  1637. : it(w)
  1638. {
  1639. ForEachItemIn(i,trees) {
  1640. IPropertyTree &tree = trees.item(i);
  1641. tree.Link();
  1642. w.append(*(IConstWorkUnit *) new CLocalWorkUnit(LINK(conn), &tree, secmgr, secuser));
  1643. }
  1644. }
  1645. bool first() { return it.first(); }
  1646. bool isValid() { return it.isValid(); }
  1647. bool next() { return it.next(); }
  1648. IConstWorkUnit & query() { return it.query();}
  1649. };
  1650. //==========================================================================================
  1651. class CStringArrayIterator : public CInterface, implements IStringIterator
  1652. {
  1653. StringArray strings;
  1654. unsigned idx;
  1655. public:
  1656. IMPLEMENT_IINTERFACE;
  1657. CStringArrayIterator() { idx = 0; };
  1658. void append(const char *str) { strings.append(str); }
  1659. virtual bool first() { idx = 0; return strings.isItem(idx); }
  1660. virtual bool next() { idx ++; return strings.isItem(idx); }
  1661. virtual bool isValid() { return strings.isItem(idx); }
  1662. virtual IStringVal & str(IStringVal &s) { s.set(strings.item(idx)); return s; }
  1663. };
  1664. class CCachedJobNameIterator : public CInterface, implements IStringIterator
  1665. {
  1666. Owned<IPropertyTreeIterator> it;
  1667. public:
  1668. IMPLEMENT_IINTERFACE;
  1669. CCachedJobNameIterator(IPropertyTreeIterator *p) : it(p) {};
  1670. virtual bool first() { return it->first(); }
  1671. virtual bool next() { return it->next(); }
  1672. virtual bool isValid() { return it->isValid(); }
  1673. virtual IStringVal & str(IStringVal &s) { s.set(it->query().queryName()+1); return s; }
  1674. };
  1675. class CEmptyStringIterator : public CInterface, implements IStringIterator
  1676. {
  1677. public:
  1678. IMPLEMENT_IINTERFACE;
  1679. virtual bool first() { return false; }
  1680. virtual bool next() { return false; }
  1681. virtual bool isValid() { return false; }
  1682. virtual IStringVal & str(IStringVal &s) { s.clear(); return s; }
  1683. };
  1684. mapEnums sortFields[] =
  1685. {
  1686. { WUSFuser, "@submitID" },
  1687. { WUSFcluster, "@clusterName" },
  1688. { WUSFjob, "@jobName" },
  1689. { WUSFstate, "@state" },
  1690. { WUSFpriority, "@priorityClass" },
  1691. { WUSFprotected, "@protected" },
  1692. { WUSFwuid, "@" },
  1693. { WUSFfileread, "FilesRead/File/@name" },
  1694. { WUSFroxiecluster, "RoxieQueryInfo/@roxieClusterName" },
  1695. { WUSFbatchloginid, "Application/Dispatcher/FTPUserID" },
  1696. { WUSFbatchcustomername, "Application/Dispatcher/CustomerName" },
  1697. { WUSFbatchpriority, "Application/Dispatcher/JobPriority" },
  1698. { WUSFbatchinputreccount, "Application/Dispatcher/InputRecords" },
  1699. { WUSFbatchtimeuploaded, "Application/Dispatcher/TimeUploaded" },
  1700. { WUSFbatchtimecompleted, "Application/Dispatcher/TimeCompleted" },
  1701. { WUSFbatchmachine, "Application/Dispatcher/Machine" },
  1702. { WUSFbatchinputfile, "Application/Dispatcher/InputFileName" },
  1703. { WUSFbatchoutputfile, "Application/Dispatcher/OutputFileName" },
  1704. { WUSFtotalthortime, "Timings/Timing[@name='Total thor time']/@duration" },
  1705. { WUSFterm, NULL }
  1706. };
  1707. class asyncRemoveDllWorkItem: public CInterface, implements IWorkQueueItem // class only used in asyncRemoveDll
  1708. {
  1709. StringAttr name;
  1710. bool removeDlls;
  1711. bool removeDirectory;
  1712. public:
  1713. IMPLEMENT_IINTERFACE;
  1714. asyncRemoveDllWorkItem(const char * _name, bool _removeDlls, bool _removeDirectory)
  1715. : name(_name)
  1716. {
  1717. removeDlls = _removeDlls;
  1718. removeDirectory = _removeDirectory;
  1719. }
  1720. void execute()
  1721. {
  1722. PROGLOG("WU removeDll %s",name.get());
  1723. queryDllServer().removeDll(name,removeDlls,removeDirectory);
  1724. }
  1725. };
  1726. class asyncRemoveRemoteFileWorkItem: public CInterface, implements IWorkQueueItem // class only used in asyncRemoveDll
  1727. {
  1728. RemoteFilename name;
  1729. public:
  1730. IMPLEMENT_IINTERFACE;
  1731. asyncRemoveRemoteFileWorkItem(const char * _ip, const char * _name)
  1732. {
  1733. SocketEndpoint ep(_ip);
  1734. name.setPath(ep, _name);
  1735. }
  1736. void execute()
  1737. {
  1738. Owned<IFile> file = createIFile(name);
  1739. PROGLOG("WU removeDll %s",file->queryFilename());
  1740. file->remove();
  1741. }
  1742. };
  1743. #define WUID_VERSION 1 // recorded in each wuid created, useful for bkwd compat. checks
  1744. class CWorkUnitFactory : public CInterface, implements IWorkUnitFactory, implements IDaliClientShutdown
  1745. {
  1746. Owned<IWorkQueueThread> deletedllworkq;
  1747. public:
  1748. IMPLEMENT_IINTERFACE;
  1749. CWorkUnitFactory()
  1750. {
  1751. // Assumes dali client configuration has already been done
  1752. sdsManager = &querySDS();
  1753. session = myProcessSession();
  1754. deletedllworkq.setown(createWorkQueueThread());
  1755. addShutdownHook(*this);
  1756. }
  1757. ~CWorkUnitFactory()
  1758. {
  1759. removeShutdownHook(*this);
  1760. // deletepool->joinAll();
  1761. }
  1762. void clientShutdown();
  1763. SessionId startSession()
  1764. {
  1765. // Temporary placeholder until startSession is implemented
  1766. #ifdef _WIN32
  1767. return GetTickCount();
  1768. #else
  1769. struct timeval tm;
  1770. gettimeofday(&tm,NULL);
  1771. return tm.tv_usec;
  1772. #endif
  1773. }
  1774. IWorkUnit* ensureNamedWorkUnit(const char *name)
  1775. {
  1776. if (workUnitTraceLevel > 1)
  1777. PrintLog("ensureNamedWorkUnit created %s", name);
  1778. StringBuffer wuRoot;
  1779. getXPath(wuRoot, name);
  1780. IRemoteConnection* conn = sdsManager->connect(wuRoot.str(), session, RTM_LOCK_WRITE|RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT);
  1781. conn->queryRoot()->setProp("@xmlns:xsi", "http://www.w3.org/1999/XMLSchema-instance");
  1782. Owned<CLocalWorkUnit> cw = new CLocalWorkUnit(conn, (ISecManager *)NULL, NULL, (const char *)NULL);
  1783. return &cw->lockRemote(false);
  1784. }
  1785. virtual IWorkUnit* createNamedWorkUnit(const char *wuid,const char *parentWuid, const char *app, const char *user)
  1786. {
  1787. StringBuffer wuRoot;
  1788. getXPath(wuRoot, wuid);
  1789. IRemoteConnection *conn;
  1790. if (queryDaliServerVersion().compare("2.0") >= 0)
  1791. conn = sdsManager->connect(wuRoot.str(), session, RTM_LOCK_WRITE|RTM_CREATE_UNIQUE, SDS_LOCK_TIMEOUT);
  1792. else
  1793. conn = sdsManager->connect(wuRoot.str(), session, RTM_LOCK_WRITE|RTM_CREATE, SDS_LOCK_TIMEOUT);
  1794. conn->queryRoot()->setProp("@xmlns:xsi", "http://www.w3.org/1999/XMLSchema-instance");
  1795. conn->queryRoot()->setPropInt("@wuidVersion", WUID_VERSION);
  1796. Owned<CLocalWorkUnit> cw = new CLocalWorkUnit(conn, (ISecManager*)NULL, NULL, parentWuid);
  1797. IWorkUnit* ret = &cw->lockRemote(false);
  1798. ret->setDebugValue("CREATED_BY", app, true);
  1799. ret->setDebugValue("CREATED_FOR", user, true);
  1800. if (user)
  1801. cw->setWuScope(user);
  1802. return ret;
  1803. }
  1804. virtual IWorkUnit* createWorkUnit(const char *parentWuid, const char *app, const char *user)
  1805. {
  1806. StringBuffer wuid("W");
  1807. char result[32];
  1808. time_t ltime;
  1809. time( &ltime );
  1810. tm *today = localtime( &ltime ); // MORE - this is not threadsafe. But I probably don't care that much!
  1811. strftime(result, sizeof(result), "%Y%m%d-%H%M%S", today);
  1812. wuid.append(result);
  1813. if (queryDaliServerVersion().compare("2.0") < 0)
  1814. wuid.append('-').append(startSession());
  1815. if (workUnitTraceLevel > 1)
  1816. PrintLog("createWorkUnit created %s", wuid.str());
  1817. IWorkUnit* ret = createNamedWorkUnit(wuid.str(),parentWuid, app, user);
  1818. if (workUnitTraceLevel > 1)
  1819. {
  1820. SCMStringBuffer wuidName;
  1821. ret->getWuid(wuidName);
  1822. PrintLog("createWorkUnit created %s", wuidName.str());
  1823. }
  1824. ret->addTimeStamp("workunit", NULL, "Created");
  1825. return ret;
  1826. }
  1827. bool secDeleteWorkUnit(const char * wuid, ISecManager *secmgr, ISecUser *secuser, bool raiseexceptions)
  1828. {
  1829. if (workUnitTraceLevel > 1)
  1830. PrintLog("deleteWorkUnit %s", wuid);
  1831. StringBuffer wuRoot;
  1832. getXPath(wuRoot, wuid);
  1833. IRemoteConnection *conn = sdsManager->connect(wuRoot.str(), session, RTM_LOCK_WRITE|RTM_LOCK_SUB, SDS_LOCK_TIMEOUT);
  1834. if (!conn)
  1835. {
  1836. if (workUnitTraceLevel > 0)
  1837. PrintLog("deleteWorkUnit %s not found", wuid);
  1838. return false;
  1839. }
  1840. Owned<CLocalWorkUnit> cw = new CLocalWorkUnit(conn, secmgr, secuser); // takes ownership of conn
  1841. if (secmgr && !checkWuSecAccess(*cw.get(), *secmgr, secuser, SecAccess_Full, "delete", true, true)) {
  1842. if (raiseexceptions) {
  1843. // perhaps raise exception here?
  1844. }
  1845. return false;
  1846. }
  1847. if (raiseexceptions) {
  1848. try
  1849. {
  1850. cw->cleanupAndDelete(true,true);
  1851. }
  1852. catch (IException *E)
  1853. {
  1854. StringBuffer s;
  1855. LOG(MCexception(E, MSGCLS_warning), E, s.append("Exception during deleteWorkUnit: ").append(wuid).str());
  1856. E->Release();
  1857. return false;
  1858. }
  1859. }
  1860. else
  1861. cw->cleanupAndDelete(true,true);
  1862. removeWorkUnitFromAllQueues(wuid); //known active workunits wouldn't make it this far
  1863. return true;
  1864. }
  1865. virtual bool deleteWorkUnitEx(const char * wuid)
  1866. {
  1867. return secDeleteWorkUnit(wuid,NULL,NULL,true);
  1868. }
  1869. virtual bool deleteWorkUnit(const char * wuid)
  1870. {
  1871. return secDeleteWorkUnit(wuid,NULL,NULL,false);
  1872. }
  1873. virtual IConstWorkUnitIterator * getWorkUnitsByOwner(const char * owner)
  1874. {
  1875. // Not sure what to do about customerID vs user etc
  1876. StringBuffer path("*");
  1877. if (owner && *owner)
  1878. path.append("[@submitID=\"").append(owner).append("\"]");
  1879. return getWorkUnitsByXPath(path.str());
  1880. }
  1881. virtual IConstWorkUnitIterator * getWorkUnitsByState(WUState state);
  1882. virtual IConstWorkUnitIterator * getWorkUnitsByECL(const char* ecl)
  1883. {
  1884. StringBuffer path("*");
  1885. if (ecl && *ecl)
  1886. path.append("[Query/Text=~\"*").append(ecl).append("*\"]");
  1887. return getWorkUnitsByXPath(path.str());
  1888. }
  1889. virtual IConstWorkUnitIterator * getWorkUnitsByCluster(const char* cluster)
  1890. {
  1891. StringBuffer path("*");
  1892. if (cluster && *cluster)
  1893. path.append("[@clusterName=\"").append(cluster).append("\"]");
  1894. return getWorkUnitsByXPath(path.str());
  1895. }
  1896. virtual IConstWorkUnitIterator * getChildWorkUnits(const char *parent)
  1897. {
  1898. StringBuffer path("*[@parent=\"");
  1899. path.append(parent).append("\"]");
  1900. return getWorkUnitsByXPath(path.str());
  1901. }
  1902. virtual IConstWorkUnit* secOpenWorkUnit(const char *wuid, bool lock, ISecManager *secmgr=NULL, ISecUser *secuser=NULL)
  1903. {
  1904. if (workUnitTraceLevel > 1)
  1905. PrintLog("openWorkUnit %s", wuid);
  1906. StringBuffer wuRoot;
  1907. getXPath(wuRoot, wuid);
  1908. IRemoteConnection* conn = sdsManager->connect(wuRoot.str(), session, lock ? RTM_LOCK_READ|RTM_LOCK_SUB : 0, SDS_LOCK_TIMEOUT);
  1909. if (conn)
  1910. {
  1911. CLocalWorkUnit *wu = new CLocalWorkUnit(conn, secmgr, secuser);
  1912. if (secmgr && wu)
  1913. {
  1914. if (!checkWuSecAccess(*wu, *secmgr, secuser, SecAccess_Read, "opening", true, true))
  1915. {
  1916. delete wu;
  1917. return NULL;
  1918. }
  1919. }
  1920. return wu;
  1921. }
  1922. else
  1923. {
  1924. if (workUnitTraceLevel > 0)
  1925. PrintLog("openWorkUnit %s not found", wuid);
  1926. return NULL;
  1927. }
  1928. }
  1929. virtual IConstWorkUnit* openWorkUnit(const char *wuid, bool lock)
  1930. {
  1931. return secOpenWorkUnit(wuid, lock);
  1932. }
  1933. virtual IWorkUnit* secUpdateWorkUnit(const char *wuid, ISecManager *secmgr=NULL, ISecUser *secuser=NULL)
  1934. {
  1935. if (workUnitTraceLevel > 1)
  1936. PrintLog("updateWorkUnit %s", wuid);
  1937. StringBuffer wuRoot;
  1938. getXPath(wuRoot, wuid);
  1939. IRemoteConnection* conn = sdsManager->connect(wuRoot.str(), session, RTM_LOCK_WRITE|RTM_LOCK_SUB, SDS_LOCK_TIMEOUT);
  1940. if (conn)
  1941. {
  1942. Owned<CLocalWorkUnit> cw = new CLocalWorkUnit(conn, secmgr, secuser);
  1943. if (secmgr && cw)
  1944. {
  1945. if (!checkWuSecAccess(*cw.get(), *secmgr, secuser, SecAccess_Write, "updating", true, true))
  1946. return NULL;
  1947. }
  1948. return &cw->lockRemote(false);
  1949. }
  1950. else
  1951. {
  1952. if (workUnitTraceLevel > 0)
  1953. PrintLog("updateWorkUnit %s not found", wuid);
  1954. return NULL;
  1955. }
  1956. }
  1957. virtual IWorkUnit* updateWorkUnit(const char *wuid)
  1958. {
  1959. return secUpdateWorkUnit(wuid);
  1960. }
  1961. virtual int setTracingLevel(int newLevel)
  1962. {
  1963. if (newLevel)
  1964. PrintLog("Setting workunit trace level to %d", newLevel);
  1965. int level = workUnitTraceLevel;
  1966. workUnitTraceLevel = newLevel;
  1967. return level;
  1968. }
  1969. IConstWorkUnitIterator * getWorkUnitsByXPath(const char *xpath)
  1970. {
  1971. return getWorkUnitsByXPath(xpath,NULL,NULL);
  1972. }
  1973. void descheduleAllWorkUnits()
  1974. {
  1975. Owned<IRemoteConnection> conn = querySDS().connect("/Schedule", myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
  1976. if(!conn) return;
  1977. Owned<IPropertyTree> root(conn->queryRoot()->getBranch("."));
  1978. KeptAtomTable entries;
  1979. Owned<IPropertyTreeIterator> iter(root->getElements("*/*/*/*"));
  1980. StringBuffer wuid;
  1981. for(iter->first(); iter->isValid(); iter->next())
  1982. {
  1983. char const * entry = iter->query().queryName();
  1984. if(!entries.find(entry))
  1985. {
  1986. entries.addAtom(entry);
  1987. ncnameUnescape(entry, wuid.clear());
  1988. Owned<IWorkUnit> wu = updateWorkUnit(wuid);
  1989. if(wu && (wu->getState() == WUStateWait))
  1990. wu->setState(WUStateCompleted);
  1991. }
  1992. }
  1993. bool more;
  1994. do more = root->removeProp("*"); while(more);
  1995. }
  1996. IConstWorkUnitIterator * getWorkUnitsByXPath(const char *xpath, ISecManager *secmgr, ISecUser *secuser)
  1997. {
  1998. Owned<IRemoteConnection> conn = sdsManager->connect("/WorkUnits", session, 0, SDS_LOCK_TIMEOUT);
  1999. if (conn)
  2000. {
  2001. CDaliVersion serverVersionNeeded("3.2");
  2002. Owned<IPropertyTreeIterator> iter(queryDaliServerVersion().compare(serverVersionNeeded) < 0 ?
  2003. conn->queryRoot()->getElements(xpath) :
  2004. conn->getElements(xpath));
  2005. return new CConstWUIterator(conn, NULL, iter, secmgr, secuser);
  2006. }
  2007. else
  2008. return NULL;
  2009. }
  2010. IConstWorkUnitIterator* getWorkUnitsSorted( WUSortField *sortorder, // list of fields to sort by (terminated by WUSFterm)
  2011. WUSortField *filters, // NULL or list of fields to folteron (terminated by WUSFterm)
  2012. const void *filterbuf, // (appended) string values for filters
  2013. unsigned startoffset,
  2014. unsigned maxnum,
  2015. const char *queryowner,
  2016. __int64 *cachehint,
  2017. ISecManager *secmgr,
  2018. ISecUser *secuser)
  2019. {
  2020. class cScopeChecker: implements ISortedElementsTreeFilter
  2021. {
  2022. UniqueScopes done;
  2023. ISecManager *secmgr;
  2024. ISecUser *secuser;
  2025. public:
  2026. cScopeChecker(ISecManager *_secmgr,ISecUser *_secuser)
  2027. {
  2028. secmgr = _secmgr;
  2029. secuser = _secuser;
  2030. }
  2031. bool isOK(IPropertyTree &tree)
  2032. {
  2033. const char *scopename = tree.queryProp("@scope");
  2034. if (!scopename||!*scopename)
  2035. return true;
  2036. const bool *b = done.getValue(scopename);
  2037. if (b)
  2038. return *b;
  2039. bool ret = checkWuScopeSecAccess(scopename,*secmgr,secuser,SecAccess_Read,"iterating",false,false);
  2040. done.setValue(scopename,ret);
  2041. return ret;
  2042. }
  2043. } sc(secmgr,secuser);
  2044. StringBuffer query("*");
  2045. StringBuffer so;
  2046. StringAttr namefilterlo;
  2047. StringAttr namefilterhi;
  2048. if (filters) {
  2049. const char *fv = (const char *)filterbuf;
  2050. for (unsigned i=0;filters[i]!=WUSFterm;i++) {
  2051. int fmt = filters[i];
  2052. int subfmt = (fmt&0xff);
  2053. if (subfmt==WUSFwuid)
  2054. namefilterlo.set(fv);
  2055. else if (subfmt==WUSFwuidhigh)
  2056. namefilterhi.set(fv);
  2057. else {
  2058. query.append('[').append(getEnumText(subfmt,sortFields)).append('=');
  2059. if (fmt&WUSFnocase)
  2060. query.append('?');
  2061. if (fmt&WUSFwild)
  2062. query.append('~');
  2063. query.append('"').append(fv).append("\"]");
  2064. }
  2065. fv = fv + strlen(fv)+1;
  2066. }
  2067. }
  2068. if (sortorder) {
  2069. for (unsigned i=0;sortorder[i]!=WUSFterm;i++) {
  2070. if (so.length())
  2071. so.append(',');
  2072. int fmt = sortorder[i];
  2073. if (fmt&WUSFreverse)
  2074. so.append('-');
  2075. if (fmt&WUSFnocase)
  2076. so.append('~');
  2077. if (fmt&WUSFnumeric)
  2078. so.append('#');
  2079. so.append(getEnumText(fmt&0xff,sortFields));
  2080. }
  2081. }
  2082. IArrayOf<IPropertyTree> results;
  2083. Owned<IRemoteConnection> conn=getElementsPaged( "WorkUnits", query.str(), so.length()?so.str():NULL,startoffset,maxnum,
  2084. secmgr?&sc:NULL,queryowner,cachehint,namefilterlo.get(),namefilterhi.get(),results);
  2085. return new CConstWUArrayIterator(conn, results, secmgr, secuser);
  2086. }
  2087. IConstWorkUnitIterator* getWorkUnitsSorted( WUSortField *sortorder, // list of fields to sort by (terminated by WUSFterm)
  2088. WUSortField *filters, // NULL or list of fields to filter on (terminated by WUSFterm)
  2089. const void *filterbuf, // (appended) string values for filters
  2090. unsigned startoffset,
  2091. unsigned maxnum,
  2092. const char *queryowner,
  2093. __int64 *cachehint)
  2094. {
  2095. return getWorkUnitsSorted(sortorder,filters,filterbuf,startoffset,maxnum,queryowner,cachehint, NULL, NULL);
  2096. }
  2097. virtual unsigned numWorkUnits()
  2098. {
  2099. Owned<IRemoteConnection> conn = sdsManager->connect("/WorkUnits", session, 0, SDS_LOCK_TIMEOUT);
  2100. if (!conn)
  2101. return 0;
  2102. IPropertyTree *root = conn->queryRoot();
  2103. return root->numChildren();
  2104. }
  2105. virtual unsigned numWorkUnitsFiltered(WUSortField *filters,
  2106. const void *filterbuf,
  2107. ISecManager *secmgr,
  2108. ISecUser *secuser)
  2109. {
  2110. Owned<IConstWorkUnitIterator> iter = getWorkUnitsSorted( NULL,filters,filterbuf,0,0x7fffffff,NULL,NULL,secmgr,secuser);
  2111. // this is rather slow but necessarily so (for security check)
  2112. unsigned ret = 0;
  2113. ForEach(*iter)
  2114. ret++;
  2115. return ret;
  2116. }
  2117. virtual unsigned numWorkUnitsFiltered(WUSortField *filters,const void *filterbuf)
  2118. {
  2119. if (!filters)
  2120. return numWorkUnits();
  2121. return numWorkUnitsFiltered(filters,filterbuf,NULL,NULL);
  2122. }
  2123. void asyncRemoveDll(const char * name, bool removeDlls, bool removeDirectory)
  2124. {
  2125. const char * tail = pathTail(name);
  2126. deletedllworkq->post(new asyncRemoveDllWorkItem(tail,removeDlls,removeDirectory));
  2127. }
  2128. void asyncRemoveFile(const char * ip, const char * name)
  2129. {
  2130. deletedllworkq->post(new asyncRemoveRemoteFileWorkItem(ip, name));
  2131. }
  2132. ISDSManager *sdsManager;
  2133. SessionId session;
  2134. ISecManager *secMgr;
  2135. private:
  2136. void deleteChildren(IPropertyTree *root, const char *wuid)
  2137. {
  2138. StringBuffer kids("*[@parent=\"");
  2139. kids.append(wuid).append("\"]");
  2140. Owned<IPropertyTreeIterator> it = root->getElements(kids.str());
  2141. ForEach (*it)
  2142. {
  2143. deleteChildren(root, it->query().queryName());
  2144. }
  2145. root->removeProp(wuid);
  2146. }
  2147. class CConstWUIterator : public CInterface, implements IConstWorkUnitIterator
  2148. {
  2149. IArrayOf<IConstWorkUnit> w;
  2150. CArrayIteratorOf<IConstWorkUnit,IConstWorkUnitIterator> it;
  2151. public:
  2152. IMPLEMENT_IINTERFACE;
  2153. CConstWUIterator() : it(w)
  2154. {
  2155. }
  2156. CConstWUIterator(IRemoteConnection *conn, IPropertyTree *, IPropertyTreeIterator *_it, ISecManager *secmgr=NULL, ISecUser *secuser=NULL) : it(w)
  2157. {
  2158. UniqueScopes us;
  2159. Owned<ISecResourceList> scopes;
  2160. if (secmgr /* && secmgr->authTypeRequired(RT_WORKUNIT_SCOPE) tbd */)
  2161. {
  2162. scopes.setown(secmgr->createResourceList("wuscopes"));
  2163. for (_it->first(); _it->isValid(); _it->next())
  2164. {
  2165. const char *scopename = _it->query().queryProp("@scope");
  2166. if (scopename && *scopename && !us.getValue(scopename))
  2167. {
  2168. scopes->addResource(scopename);
  2169. us.setValue(scopename, true);
  2170. }
  2171. }
  2172. if (scopes->count())
  2173. {
  2174. secmgr->authorizeEx(RT_WORKUNIT_SCOPE, *secuser, scopes);
  2175. if (checkWuScopeListSecAccess(NULL, scopes, SecAccess_Read, "iterating", false, false))
  2176. scopes.clear(); //if no scopes restricted, no need to check later
  2177. }
  2178. else
  2179. scopes.clear();
  2180. }
  2181. for (_it->first(); _it->isValid(); _it->next())
  2182. {
  2183. IPropertyTree *rp = &_it->query();
  2184. const char *scopename=rp->queryProp("@scope");
  2185. if (!scopename || !*scopename || !scopes || checkWuScopeListSecAccess(rp->queryProp("@scope"), scopes, SecAccess_Read, "iterating", false, false))
  2186. w.append(*(IConstWorkUnit *) new CLocalWorkUnit(LINK(conn), LINK(rp), secmgr, secuser));
  2187. }
  2188. }
  2189. bool first() { return it.first(); }
  2190. bool isValid() { return it.isValid(); }
  2191. bool next() { return it.next(); }
  2192. IConstWorkUnit & query() { return it.query();}
  2193. };
  2194. IRemoteConnection* connect(const char *xpath, unsigned flags)
  2195. {
  2196. return sdsManager->connect(xpath, session, flags, SDS_LOCK_TIMEOUT);
  2197. }
  2198. };
  2199. static Owned<CWorkUnitFactory> factory;
  2200. void CWorkUnitFactory::clientShutdown()
  2201. {
  2202. factory.clear();
  2203. }
  2204. void clientShutdownWorkUnit()
  2205. {
  2206. factory.clear();
  2207. }
  2208. extern WORKUNIT_API IWorkUnitFactory * getWorkUnitFactory()
  2209. {
  2210. if (!factory)
  2211. factory.setown(new CWorkUnitFactory());
  2212. return factory.getLink();
  2213. }
  2214. class CSecureWorkUnitFactory : public CInterface, implements IWorkUnitFactory
  2215. {
  2216. public:
  2217. IMPLEMENT_IINTERFACE;
  2218. CSecureWorkUnitFactory(ISecManager &secmgr, ISecUser &secuser)
  2219. {
  2220. if (!factory)
  2221. factory.setown(new CWorkUnitFactory());
  2222. secMgr.set(&secmgr);
  2223. secUser.set(&secuser);
  2224. }
  2225. virtual IWorkUnit* createNamedWorkUnit(const char *wuid,const char *parentWuid, const char *app, const char *user)
  2226. {
  2227. checkWuScopeSecAccess(user, *secMgr.get(), secUser.get(), SecAccess_Write, "Create", true, true);
  2228. IWorkUnit *wu=factory->createNamedWorkUnit(wuid, parentWuid, app, user);
  2229. if (wu)
  2230. {
  2231. CLockedWorkUnit* lw = dynamic_cast<CLockedWorkUnit*>(wu);
  2232. if (lw)
  2233. lw->setSecIfcs(secMgr.get(), secUser.get());
  2234. }
  2235. return wu;
  2236. }
  2237. virtual IWorkUnit* createWorkUnit(const char *parentWuid, const char *app, const char *user)
  2238. {
  2239. checkWuScopeSecAccess(user, *secMgr.get(), secUser.get(), SecAccess_Write, "Create", true, true);
  2240. IWorkUnit *wu=factory->createWorkUnit(parentWuid, app, user);
  2241. if (wu)
  2242. {
  2243. CLockedWorkUnit* lw = dynamic_cast<CLockedWorkUnit*>(wu);
  2244. if (lw)
  2245. lw->setSecIfcs(secMgr.get(), secUser.get());
  2246. }
  2247. return wu;
  2248. }
  2249. virtual bool deleteWorkUnitEx(const char * wuid)
  2250. {
  2251. return factory->secDeleteWorkUnit(wuid, secMgr.get(), secUser.get(), true);
  2252. }
  2253. virtual bool deleteWorkUnit(const char * wuid)
  2254. {
  2255. return factory->secDeleteWorkUnit(wuid, secMgr.get(), secUser.get(), false);
  2256. }
  2257. virtual IConstWorkUnit* openWorkUnit(const char *wuid, bool lock)
  2258. {
  2259. return factory->secOpenWorkUnit(wuid, lock, secMgr.get(), secUser.get());
  2260. }
  2261. virtual IWorkUnit* updateWorkUnit(const char *wuid)
  2262. {
  2263. return factory->secUpdateWorkUnit(wuid, secMgr.get(), secUser.get());
  2264. }
  2265. //make cached workunits a non secure pass through for now.
  2266. virtual IConstWorkUnitIterator * getWorkUnitsByOwner(const char * owner)
  2267. {
  2268. // Not sure what to do about customerID vs user etc
  2269. StringBuffer path("*");
  2270. if (owner && *owner)
  2271. path.append("[@submitID=\"").append(owner).append("\"]");
  2272. return factory->getWorkUnitsByXPath(path.str(), secMgr.get(), secUser.get());
  2273. }
  2274. virtual IConstWorkUnitIterator * getWorkUnitsByState(WUState state);
  2275. virtual IConstWorkUnitIterator * getWorkUnitsByECL(const char* ecl)
  2276. {
  2277. return factory->getWorkUnitsByECL(ecl);
  2278. }
  2279. virtual IConstWorkUnitIterator * getWorkUnitsByCluster(const char* cluster)
  2280. {
  2281. return factory->getWorkUnitsByCluster(cluster);
  2282. }
  2283. virtual IConstWorkUnitIterator * getWorkUnitsByXPath(const char * xpath)
  2284. {
  2285. return factory->getWorkUnitsByXPath(xpath, secMgr.get(), secUser.get());
  2286. }
  2287. virtual void descheduleAllWorkUnits()
  2288. {
  2289. factory->descheduleAllWorkUnits();
  2290. }
  2291. virtual IConstWorkUnitIterator * getChildWorkUnits(const char *parent)
  2292. {
  2293. StringBuffer path("*[@parent=\"");
  2294. path.append(parent).append("\"]");
  2295. return factory->getWorkUnitsByXPath(path.str(), secMgr.get(), secUser.get());
  2296. }
  2297. virtual int setTracingLevel(int newLevel)
  2298. {
  2299. return factory->setTracingLevel(newLevel);
  2300. }
  2301. virtual IConstWorkUnitIterator* getWorkUnitsSorted( WUSortField *sortorder, // list of fields to sort by (terminated by WUSFterm)
  2302. WUSortField *filters, // NULL or list of fields to filter on (terminated by WUSFterm)
  2303. const void *filterbuf, // (appended) string values for filters
  2304. unsigned startoffset,
  2305. unsigned maxnum,
  2306. const char *queryowner,
  2307. __int64 *cachehint)
  2308. {
  2309. return factory->getWorkUnitsSorted(sortorder,filters,filterbuf,startoffset,maxnum,queryowner,cachehint, secMgr.get(), secUser.get());
  2310. }
  2311. virtual unsigned numWorkUnits()
  2312. {
  2313. return factory->numWorkUnits();
  2314. }
  2315. virtual unsigned numWorkUnitsFiltered(WUSortField *filters,
  2316. const void *filterbuf)
  2317. {
  2318. return factory->numWorkUnitsFiltered(filters,filterbuf,secMgr.get(), secUser.get());
  2319. }
  2320. private:
  2321. Owned<CWorkUnitFactory> base_factory;
  2322. Owned<ISecManager> secMgr;
  2323. Owned<ISecUser> secUser;
  2324. };
  2325. extern WORKUNIT_API IWorkUnitFactory * getSecWorkUnitFactory(ISecManager &secmgr, ISecUser &secuser)
  2326. {
  2327. return new CSecureWorkUnitFactory(secmgr, secuser);
  2328. }
  2329. extern WORKUNIT_API IWorkUnitFactory * getWorkUnitFactory(ISecManager *secmgr, ISecUser *secuser)
  2330. {
  2331. if (secmgr && secuser)
  2332. return getSecWorkUnitFactory(*secmgr, *secuser);
  2333. else
  2334. return getWorkUnitFactory();
  2335. }
  2336. //==========================================================================================
  2337. class CStringPTreeIterator : public CInterface, implements IStringIterator
  2338. {
  2339. Owned<IPropertyTreeIterator> it;
  2340. public:
  2341. IMPLEMENT_IINTERFACE;
  2342. CStringPTreeIterator(IPropertyTreeIterator *p) : it(p) {};
  2343. virtual bool first() { return it->first(); }
  2344. virtual bool next() { return it->next(); }
  2345. virtual bool isValid() { return it->isValid(); }
  2346. virtual IStringVal & str(IStringVal &s) { s.set(it->query().queryProp(NULL)); return s; }
  2347. };
  2348. class CStringPTreeTagIterator : public CInterface, implements IStringIterator
  2349. {
  2350. Owned<IPropertyTreeIterator> it;
  2351. public:
  2352. IMPLEMENT_IINTERFACE;
  2353. CStringPTreeTagIterator(IPropertyTreeIterator *p) : it(p) {};
  2354. virtual bool first() { return it->first(); }
  2355. virtual bool next() { return it->next(); }
  2356. virtual bool isValid() { return it->isValid(); }
  2357. virtual IStringVal & str(IStringVal &s) { s.set(it->query().queryName()); return s; }
  2358. };
  2359. class CStringPTreeAttrIterator : public CInterface, implements IStringIterator
  2360. {
  2361. Owned<IPropertyTreeIterator> it;
  2362. StringAttr name;
  2363. public:
  2364. IMPLEMENT_IINTERFACE;
  2365. CStringPTreeAttrIterator(IPropertyTreeIterator *p, const char *_name) : it(p), name(_name) {};
  2366. virtual bool first() { return it->first(); }
  2367. virtual bool next() { return it->next(); }
  2368. virtual bool isValid() { return it->isValid(); }
  2369. virtual IStringVal & str(IStringVal &s) { s.set(it->query().queryProp(name)); return s; }
  2370. };
  2371. //==========================================================================================
  2372. CLocalWorkUnit::CLocalWorkUnit(IRemoteConnection *_conn, ISecManager *secmgr, ISecUser *secuser, const char *parentWuid) : connection(_conn)
  2373. {
  2374. connectAtRoot = true;
  2375. init();
  2376. p.setown(connection->getRoot());
  2377. if (parentWuid)
  2378. p->setProp("@parent", parentWuid);
  2379. secMgr.set(secmgr);
  2380. secUser.set(secuser);
  2381. }
  2382. CLocalWorkUnit::CLocalWorkUnit(IRemoteConnection *_conn, IPropertyTree* root, ISecManager *secmgr, ISecUser *secuser) : connection(_conn)
  2383. {
  2384. connectAtRoot = false;
  2385. init();
  2386. p.setown(root);
  2387. secMgr.set(secmgr);
  2388. secUser.set(secuser);
  2389. }
  2390. void CLocalWorkUnit::init()
  2391. {
  2392. p.clear();
  2393. cachedGraphs.clear();
  2394. workflowIterator.clear();
  2395. query.clear();
  2396. graphs.kill();
  2397. results.kill();
  2398. variables.kill();
  2399. plugins.kill();
  2400. libraries.kill();
  2401. activities.kill();
  2402. exceptions.kill();
  2403. temporaries.kill();
  2404. roxieQueryInfo.clear();
  2405. webServicesInfo.clear();
  2406. workflowIteratorCached = false;
  2407. resultsCached = false;
  2408. temporariesCached = false;
  2409. variablesCached = false;
  2410. exceptionsCached = false;
  2411. pluginsCached = false;
  2412. librariesCached = false;
  2413. activitiesCached = false;
  2414. webServicesInfoCached = false;
  2415. roxieQueryInfoCached = false;
  2416. dirty = false;
  2417. abortDirty = true;
  2418. abortState = false;
  2419. }
  2420. // Dummy workunit support
  2421. CLocalWorkUnit::CLocalWorkUnit(const char *_wuid, const char *parentWuid, ISecManager *secmgr, ISecUser *secuser)
  2422. {
  2423. connectAtRoot = true;
  2424. init();
  2425. p.setown(createPTree(_wuid));
  2426. p->setProp("@xmlns:xsi", "http://www.w3.org/1999/XMLSchema-instance");
  2427. if (parentWuid)
  2428. p->setProp("@parentWuid", parentWuid);
  2429. secMgr.set(secmgr);
  2430. secUser.set(secuser);
  2431. }
  2432. CLocalWorkUnit::~CLocalWorkUnit()
  2433. {
  2434. if (workUnitTraceLevel > 1)
  2435. {
  2436. PrintLog("Releasing workunit %s mode %x", p->queryName(), connection ? connection->queryMode() :0);
  2437. }
  2438. try
  2439. {
  2440. unsubscribe();
  2441. query.clear();
  2442. webServicesInfo.clear();
  2443. roxieQueryInfo.clear();
  2444. workflowIterator.clear();
  2445. activities.kill();
  2446. plugins.kill();
  2447. libraries.kill();
  2448. exceptions.kill();
  2449. graphs.kill();
  2450. results.kill();
  2451. temporaries.kill();
  2452. variables.kill();
  2453. timestamps.kill();
  2454. appvalues.kill();
  2455. userDesc.clear();
  2456. secMgr.clear();
  2457. secUser.clear();
  2458. cachedGraphs.clear();
  2459. p.clear();
  2460. connection.clear();
  2461. }
  2462. catch (IException *E) { LOG(MCexception(E, MSGCLS_warning), E, "Exception during ~CLocalWorkUnit"); E->Release(); }
  2463. }
  2464. void CLocalWorkUnit::cleanupAndDelete(bool deldll,bool deleteOwned)
  2465. {
  2466. TIME_SECTION("WUDELETE cleanupAndDelete total");
  2467. // Delete any related things in SDS etc that might otherwise be forgotten
  2468. assertex(connectAtRoot); // make sure we don't delete entire workunit tree!
  2469. if (p->getPropBool("@protected", false))
  2470. throw MakeStringException(WUERR_WorkunitProtected, "%s: Workunit is protected",p->queryName());
  2471. switch (getState())
  2472. {
  2473. case WUStateAborted:
  2474. case WUStateCompleted:
  2475. case WUStateFailed:
  2476. case WUStateArchived:
  2477. break;
  2478. case WUStateCompiled:
  2479. if (getAction()==WUActionRun || getAction()==WUActionUnknown)
  2480. throw MakeStringException(WUERR_WorkunitActive, "%s: Workunit is active",p->queryName());
  2481. break;
  2482. case WUStateWait:
  2483. throw MakeStringException(WUERR_WorkunitScheduled, "%s: Workunit is scheduled",p->queryName());
  2484. default:
  2485. throw MakeStringException(WUERR_WorkunitActive, "%s: Workunit is active",p->queryName());
  2486. break;
  2487. }
  2488. if (getIsQueryService())
  2489. {
  2490. Owned<IPropertyTree> registry = getQueryRegistryRoot();
  2491. if (registry)
  2492. {
  2493. VStringBuffer xpath("QuerySet/Query[@wuid='%s']", p->queryName());
  2494. if (registry->hasProp(xpath.str()))
  2495. throw MakeStringException(WUERR_WorkunitPublished, "%s: Workunit is published",p->queryName());
  2496. }
  2497. }
  2498. try
  2499. {
  2500. if (deldll && !p->getPropBool("@isClone", false))
  2501. {
  2502. Owned<IConstWUQuery> q = getQuery();
  2503. if (q)
  2504. {
  2505. Owned<IConstWUAssociatedFileIterator> iter = &q->getAssociatedFiles();
  2506. SCMStringBuffer name;
  2507. SCMStringBuffer ip;
  2508. ForEach(*iter)
  2509. {
  2510. IConstWUAssociatedFile & cur = iter->query();
  2511. cur.getName(name);
  2512. if (cur.getType() == FileTypeDll)
  2513. {
  2514. bool removeDir = true; // this is to keep the code the same as before, but I don't know why it only does it for the dll.
  2515. factory->asyncRemoveDll(name.str(), true, removeDir);
  2516. }
  2517. else
  2518. {
  2519. cur.getIp(ip);
  2520. factory->asyncRemoveFile(ip.str(), name.str());
  2521. }
  2522. }
  2523. }
  2524. }
  2525. StringBuffer apath;
  2526. // PROGLOG("wuid dll files removed");
  2527. {
  2528. apath.append("/WorkUnitAborts/").append(p->queryName());
  2529. Owned<IRemoteConnection> acon = factory->sdsManager->connect(apath.str(), factory->session, RTM_LOCK_WRITE | RTM_DELETE_ON_DISCONNECT, SDS_LOCK_TIMEOUT);
  2530. acon.clear();
  2531. }
  2532. // PROGLOG("wuid WorkUnitAborts entry removed");
  2533. deleteTempFiles(NULL, deleteOwned, true); // all, any remaining.
  2534. }
  2535. catch(IException *E)
  2536. {
  2537. StringBuffer s;
  2538. LOG(MCexception(E, MSGCLS_warning), E, s.append("Exception during cleanupAndDelete: ").append(p->queryName()).str());
  2539. E->Release();
  2540. }
  2541. catch (...)
  2542. {
  2543. WARNLOG("Unknown exception during cleanupAndDelete: %s", p->queryName());
  2544. }
  2545. CConstGraphProgress::deleteWuidProgress(p->queryName());
  2546. connection->close(true);
  2547. PROGLOG("WUID %s removed",p->queryName());
  2548. connection.clear();
  2549. }
  2550. void CLocalWorkUnit::setTimeScheduled(const IJlibDateTime &val)
  2551. {
  2552. SCMStringBuffer strval;
  2553. val.getGmtString(strval);
  2554. p->setProp("@timescheduled",strval.str());
  2555. }
  2556. IJlibDateTime & CLocalWorkUnit::getTimeScheduled(IJlibDateTime &val) const
  2557. {
  2558. StringBuffer str;
  2559. p->getProp("@timescheduled",str);
  2560. if(str.length())
  2561. val.setGmtString(str.str());
  2562. return val;
  2563. }
  2564. bool modifyAndWriteWorkUnitXML(char const * wuid, StringBuffer & buf, StringBuffer & extra, IFileIO * fileio)
  2565. {
  2566. // kludge in extra chunks of XML such as GraphProgress and GeneratedDlls
  2567. if(extra.length())
  2568. {
  2569. size32_t l = (size32_t)strlen(wuid);
  2570. size32_t p = buf.length()-l-4; // bit of a kludge
  2571. assertex(memcmp(buf.str()+p+2,wuid,l)==0);
  2572. StringAttr tail(buf.str()+p);
  2573. buf.setLength(p);
  2574. buf.append(extra);
  2575. buf.append(tail);
  2576. }
  2577. return (fileio->write(0,buf.length(),buf.str()) == buf.length());
  2578. }
  2579. bool CLocalWorkUnit::archiveWorkUnit(const char *base,bool del,bool ignoredllerrors,bool deleteOwned)
  2580. {
  2581. CriticalBlock block(crit);
  2582. StringBuffer path(base);
  2583. if (!p)
  2584. return false;
  2585. const char *wuid = p->queryName();
  2586. if (!wuid||!*wuid)
  2587. return false;
  2588. addPathSepChar(path).append(wuid).append(".xml");
  2589. Owned<IFile> file = createIFile(path.str());
  2590. if (!file)
  2591. return false;
  2592. Owned<IFileIO> fileio = file->open(IFOcreate);
  2593. if (!fileio)
  2594. return false;
  2595. StringBuffer buf;
  2596. exportWorkUnitToXML(this, buf);
  2597. StringBuffer extraWorkUnitXML;
  2598. StringBuffer xpath("/GraphProgress/");
  2599. xpath.append(wuid);
  2600. Owned<IRemoteConnection> conn = querySDS().connect(xpath.str(), myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
  2601. if (conn) {
  2602. Owned<IPropertyTree> tmp = createPTree("GraphProgress");
  2603. mergePTree(tmp,conn->queryRoot());
  2604. toXML(tmp,extraWorkUnitXML,1,XML_Format);
  2605. conn->close();
  2606. }
  2607. Owned<IConstWUQuery> q = getQuery();
  2608. if (!q) {
  2609. if(!modifyAndWriteWorkUnitXML(wuid, buf, extraWorkUnitXML, fileio))
  2610. return false;
  2611. if (del) {
  2612. if (getState()==WUStateUnknown)
  2613. setState(WUStateArchived); // to allow delete
  2614. cleanupAndDelete(false,deleteOwned); // no query, may as well delete
  2615. }
  2616. return false;
  2617. }
  2618. Owned<IConstWUAssociatedFileIterator> iter = &q->getAssociatedFiles();
  2619. SCMStringBuffer name;
  2620. Owned<IException> exception;
  2621. Owned<IDllLocation> loc;
  2622. StringBuffer dst, locpath;
  2623. Owned<IPropertyTree> generatedDlls = createPTree("GeneratedDlls");
  2624. ForEach(*iter) {
  2625. IConstWUAssociatedFile & cur = iter->query();
  2626. cur.getName(name);
  2627. if (name.length()) {
  2628. Owned<IDllEntry> entry = queryDllServer().getEntry(name.str());
  2629. if (entry.get()) {
  2630. Owned<IPropertyTree> generatedDllBranch = createPTree();
  2631. generatedDllBranch->setProp("@name", entry->queryName());
  2632. generatedDllBranch->setProp("@kind", entry->queryKind());
  2633. bool removeDllFiles = true;
  2634. exception.clear();
  2635. try
  2636. {
  2637. loc.setown(entry->getBestLocation()); //throws exception if no readable locations
  2638. }
  2639. catch(IException * e)
  2640. {
  2641. exception.setown(e);
  2642. loc.setown(entry->getBestLocationCandidate()); //this will be closest of the unreadable locations
  2643. }
  2644. RemoteFilename filename;
  2645. loc->getDllFilename(filename);
  2646. if(!exception)
  2647. {
  2648. Owned<IFile> srcfile = createIFile(filename);
  2649. addPathSepChar(dst.clear().append(base));
  2650. filename.getTail(dst);
  2651. Owned<IFile> dstfile = createIFile(dst.str());
  2652. try
  2653. {
  2654. copyFile(dstfile,srcfile);
  2655. makeAbsolutePath(dstfile->queryFilename(), locpath.clear());
  2656. }
  2657. catch(IException * e)
  2658. {
  2659. exception.setown(e);
  2660. }
  2661. }
  2662. if(exception)
  2663. {
  2664. if(ignoredllerrors)
  2665. {
  2666. EXCLOG(exception.get(), "archiveWorkUnit (copying associated file)");
  2667. //copy failed, so store original (best) location and don't delete the files
  2668. filename.getRemotePath(locpath.clear());
  2669. removeDllFiles = false;
  2670. }
  2671. else
  2672. {
  2673. throw exception.getLink();
  2674. }
  2675. }
  2676. generatedDllBranch->setProp("@location", locpath.str());
  2677. generatedDlls->addPropTree("GeneratedDll", generatedDllBranch.getClear());
  2678. if (del) {
  2679. bool removeDir = (cur.getType() == FileTypeDll); // copied from cleanupAndDelete code, above
  2680. if (!p->getPropBool("@isClone", false)) // Leak to protect against cloned WUs
  2681. entry->remove(removeDllFiles, removeDir);
  2682. }
  2683. }
  2684. }
  2685. }
  2686. iter.clear();
  2687. if(generatedDlls->numChildren())
  2688. toXML(generatedDlls, extraWorkUnitXML, 1, XML_Format);
  2689. if(!modifyAndWriteWorkUnitXML(wuid, buf, extraWorkUnitXML, fileio))
  2690. return false;
  2691. if (del) {
  2692. //setState(WUStateArchived); // this isn't useful as about to delete it!
  2693. q.clear();
  2694. //deldll false as should have deleted all those we successfully copied, and archived and removed SDS entries, above
  2695. cleanupAndDelete(false, deleteOwned);
  2696. }
  2697. return true;
  2698. }
  2699. void CLocalWorkUnit::packWorkUnit(bool pack)
  2700. {
  2701. // only packs Graph info currently
  2702. CriticalBlock block(crit);
  2703. if (!p)
  2704. return;
  2705. const char *wuid = p->queryName();
  2706. if (!wuid||!*wuid)
  2707. return;
  2708. if (pack) {
  2709. if (!p->hasProp("PackedGraphs")) {
  2710. cachedGraphs.clear();
  2711. IPropertyTree *t = p->queryPropTree("Graphs");
  2712. if (t) {
  2713. MemoryBuffer buf;
  2714. t->serialize(buf);
  2715. p->setPropBin("PackedGraphs",buf.length(),buf.bufferBase());
  2716. p->removeTree(t);
  2717. }
  2718. }
  2719. }
  2720. else {
  2721. ensureGraphsUnpacked();
  2722. }
  2723. CConstGraphProgress::packProgress(wuid,pack);
  2724. }
  2725. IPropertyTree * pruneBranch(IPropertyTree * from, char const * xpath)
  2726. {
  2727. Owned<IPropertyTree> ret;
  2728. IPropertyTree * branch = from->queryPropTree(xpath);
  2729. if(branch) {
  2730. ret.setown(createPTreeFromIPT(branch));
  2731. from->removeTree(branch);
  2732. }
  2733. return ret.getClear();
  2734. }
  2735. bool restoreWorkUnit(const char *base,const char *wuid)
  2736. {
  2737. StringBuffer path(base);
  2738. if (!wuid||!*wuid)
  2739. return false;
  2740. addPathSepChar(path).append(wuid).append(".xml");
  2741. Owned<IFile> file = createIFile(path.str());
  2742. if (!file)
  2743. return false;
  2744. Owned<IFileIO> fileio = file->open(IFOread);
  2745. if (!fileio)
  2746. return false;
  2747. Owned<IPropertyTree> pt = createPTree(*fileio);
  2748. if (!pt)
  2749. return false;
  2750. CDateTime dt;
  2751. dt.setNow();
  2752. StringBuffer dts;
  2753. dt.getString(dts);
  2754. pt->setProp("@restoredDate", dts.str());
  2755. Owned<IRemoteConnection> conn = querySDS().connect("/WorkUnits", myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
  2756. if (!conn) {
  2757. ERRLOG("restoreWorkUnit could not connect to /WorkUnits");
  2758. return false;
  2759. }
  2760. IPropertyTree *root = conn->queryRoot();
  2761. if (root->hasProp(wuid)) {
  2762. ERRLOG("restoreWorkUnit WUID %s already exists",wuid);
  2763. return false;
  2764. }
  2765. Owned<IPropertyTree> gprogress = pruneBranch(pt, "GraphProgress[1]");
  2766. Owned<IPropertyTree> generatedDlls = pruneBranch(pt, "GeneratedDlls[1]");
  2767. root->setPropTree(wuid,pt.getClear());
  2768. conn.clear();
  2769. // now kludge back GraphProgress and GeneratedDlls
  2770. if (gprogress) {
  2771. StringBuffer xpath("/GraphProgress/");
  2772. conn.setown(querySDS().connect("/GraphProgress", myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT));
  2773. if (conn) {
  2774. IPropertyTree *groot = conn->queryRoot();
  2775. if (groot->hasProp(wuid)) {
  2776. ERRLOG("restoreWorkUnit WUID %s graphprogress already exists, removing",wuid);
  2777. groot->removeProp(wuid);
  2778. }
  2779. groot->setPropTree(wuid,gprogress.getClear());
  2780. }
  2781. }
  2782. if(generatedDlls)
  2783. {
  2784. Owned<IPropertyTreeIterator> dlls = generatedDlls->getElements("GeneratedDll");
  2785. for(dlls->first(); dlls->isValid(); dlls->next())
  2786. {
  2787. IPropertyTree & dll = dlls->query();
  2788. char const * name = dll.queryProp("@name");
  2789. char const * kind = dll.queryProp("@kind");
  2790. char const * location = dll.queryProp("@location");
  2791. Owned<IDllEntry> got = queryDllServer().getEntry(name);
  2792. if(!got)
  2793. queryDllServer().registerDll(name, kind, location);
  2794. }
  2795. }
  2796. return true;
  2797. }
  2798. void CLocalWorkUnit::loadXML(const char *xml)
  2799. {
  2800. CriticalBlock block(crit);
  2801. init();
  2802. assertex(xml);
  2803. p.setown(createPTreeFromXMLString(xml));
  2804. }
  2805. void CLocalWorkUnit::serialize(MemoryBuffer &tgt)
  2806. {
  2807. CriticalBlock block(crit);
  2808. StringBuffer x;
  2809. tgt.append(exportWorkUnitToXML(this, x).str());
  2810. }
  2811. void CLocalWorkUnit::deserialize(MemoryBuffer &src)
  2812. {
  2813. CriticalBlock block(crit);
  2814. StringAttr value;
  2815. src.read(value);
  2816. loadXML(value);
  2817. }
  2818. void CLocalWorkUnit::notify(SubscriptionId id, const char *xpath, SDSNotifyFlags flags, unsigned valueLen, const void *valueData)
  2819. {
  2820. dirty = true;
  2821. }
  2822. void CLocalWorkUnit::abort()
  2823. {
  2824. abortDirty = true;
  2825. }
  2826. void CLocalWorkUnit::requestAbort()
  2827. {
  2828. CriticalBlock block(crit);
  2829. abortWorkUnit(p->queryName());
  2830. }
  2831. void CLocalWorkUnit::subscribe(WUSubscribeOptions options)
  2832. {
  2833. CriticalBlock block(crit);
  2834. bool subscribeAbort = false;
  2835. bool subscribeChange = false;
  2836. bool needChildren = true;
  2837. switch (options)
  2838. {
  2839. case SubscribeOptionAbort:
  2840. subscribeAbort = true;
  2841. break;
  2842. case SubscribeOptionRunningState:
  2843. needChildren = false;
  2844. case SubscribeOptionAnyState:
  2845. subscribeAbort = true;
  2846. subscribeChange = true;
  2847. break;
  2848. case SubscribeOptionProgress:
  2849. case SubscribeOptionAll:
  2850. subscribeChange = true;
  2851. break;
  2852. }
  2853. if (subscribeChange)
  2854. {
  2855. if (changeWatcher && changeWatcher->watchingChildren() != needChildren)
  2856. {
  2857. changeWatcher->unsubscribe();
  2858. changeWatcher.clear();
  2859. }
  2860. if (!changeWatcher)
  2861. {
  2862. changeWatcher.setown(new CWorkUnitWatcher(this, p->queryName(), needChildren));
  2863. dirty = true;
  2864. }
  2865. }
  2866. if (subscribeAbort && !abortWatcher)
  2867. {
  2868. abortWatcher.setown(new CWorkUnitAbortWatcher(this, p->queryName()));
  2869. abortDirty = true;
  2870. }
  2871. }
  2872. #if 0 // I don't think this is used (I grepped the source), am leaving here just in case I've missed somewhere (PG)
  2873. WUState CLocalWorkUnit::waitComplete(int timeout, bool returnOnWaitState)
  2874. {
  2875. class WorkUnitWaiter : public CInterface, implements ISDSSubscription, implements IAbortHandler
  2876. {
  2877. Semaphore changed;
  2878. CLocalWorkUnit *parent;
  2879. public:
  2880. IMPLEMENT_IINTERFACE;
  2881. WorkUnitWaiter(CLocalWorkUnit *_parent) : parent(_parent) { aborted = false; };
  2882. void notify(SubscriptionId id, const char *xpath, SDSNotifyFlags flags, unsigned valueLen, const void *valueData)
  2883. {
  2884. parent->notify(id, xpath, flags, valueLen, valueData);
  2885. changed.signal();
  2886. }
  2887. bool wait(unsigned timeout)
  2888. {
  2889. return changed.wait(timeout) && !aborted;
  2890. }
  2891. bool onAbort()
  2892. {
  2893. aborted = true;
  2894. changed.signal();
  2895. return false;
  2896. }
  2897. bool aborted;
  2898. } waiter(this);
  2899. Owned<CWorkUnitWatcher> w = new CWorkUnitWatcher(&waiter, p->queryName(), false);
  2900. LocalIAbortHandler abortHandler(waiter);
  2901. forceReload(); // or may miss changes that already happened, between load of wu and now.
  2902. unsigned start = msTick();
  2903. WUState ret;
  2904. loop
  2905. {
  2906. ret = getState();
  2907. switch (ret)
  2908. {
  2909. case WUStateWait:
  2910. if(!returnOnWaitState)
  2911. break;
  2912. //fall thru
  2913. case WUStateCompleted:
  2914. case WUStateFailed:
  2915. case WUStateAborted:
  2916. w->unsubscribe();
  2917. return ret;
  2918. }
  2919. unsigned waited = msTick() - start;
  2920. if (timeout==-1)
  2921. {
  2922. waiter.wait(20000);
  2923. if (waiter.aborted)
  2924. {
  2925. ret = WUStateUnknown; // MORE - throw an exception?
  2926. break;
  2927. }
  2928. }
  2929. else if (waited > timeout || !waiter.wait(timeout-waited))
  2930. {
  2931. ret = WUStateUnknown; // MORE - throw an exception?
  2932. break;
  2933. }
  2934. reload();
  2935. }
  2936. w->unsubscribe();
  2937. return ret;
  2938. }
  2939. #endif
  2940. void CLocalWorkUnit::forceReload()
  2941. {
  2942. dirty = true;
  2943. reload();
  2944. }
  2945. bool CLocalWorkUnit::reload()
  2946. {
  2947. CriticalBlock block(crit);
  2948. if (dirty)
  2949. {
  2950. if (!connectAtRoot)
  2951. {
  2952. StringBuffer wuRoot;
  2953. getXPath(wuRoot, p->queryName());
  2954. IRemoteConnection *newconn = factory->sdsManager->connect(wuRoot.str(), factory->session, 0, SDS_LOCK_TIMEOUT);
  2955. if (!newconn)
  2956. throw MakeStringException(WUERR_ConnectFailed, "Could not connect to workunit %s (deleted?)",p->queryName());
  2957. connection.setown(newconn);
  2958. connectAtRoot = true;
  2959. }
  2960. else
  2961. connection->reload();
  2962. init();
  2963. p.setown(connection->getRoot());
  2964. return true;
  2965. }
  2966. return false;
  2967. }
  2968. void CLocalWorkUnit::unsubscribe()
  2969. {
  2970. CriticalBlock block(crit);
  2971. if (abortWatcher)
  2972. {
  2973. abortWatcher->unsubscribe();
  2974. abortWatcher.clear();
  2975. }
  2976. if (changeWatcher)
  2977. {
  2978. changeWatcher->unsubscribe();
  2979. changeWatcher.clear();
  2980. }
  2981. }
  2982. void CLocalWorkUnit::unlockRemote(bool commit)
  2983. {
  2984. CriticalBlock block(crit);
  2985. MTIME_SECTION(timer, "WorkUnit_unlockRemote");
  2986. locked.unlock();
  2987. if (commit)
  2988. {
  2989. try
  2990. {
  2991. assertex(connectAtRoot);
  2992. setTimeStamp("workunit", NULL, "Modified",false);
  2993. try { connection->commit(); }
  2994. catch (IException *e)
  2995. {
  2996. EXCLOG(e, "Error during workunit commit");
  2997. connection->rollback();
  2998. connection->changeMode(0, SDS_LOCK_TIMEOUT);
  2999. throw;
  3000. }
  3001. connection->changeMode(0, SDS_LOCK_TIMEOUT);
  3002. }
  3003. catch (IException *E)
  3004. {
  3005. StringBuffer s;
  3006. PrintLog("Failed to release write lock on workunit: %s", E->errorMessage(s).str());
  3007. throw;
  3008. }
  3009. }
  3010. }
  3011. IWorkUnit &CLocalWorkUnit::lockRemote(bool commit)
  3012. {
  3013. if (secMgr)
  3014. checkWuSecAccess(*this, *secMgr.get(), secUser.get(), SecAccess_Write, "write lock", true, true);
  3015. locked.lock();
  3016. CriticalBlock block(crit);
  3017. MTIME_SECTION(timer, "WorkUnit_lockRemote");
  3018. if (commit)
  3019. {
  3020. try
  3021. {
  3022. StringBuffer wuRoot;
  3023. getXPath(wuRoot, p->queryName());
  3024. if (connection&&connectAtRoot)
  3025. connection->changeMode(RTM_LOCK_WRITE,SDS_LOCK_TIMEOUT);
  3026. else
  3027. connection.setown(factory->sdsManager->connect(wuRoot.str(), factory->session, RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT));
  3028. if (!connection)
  3029. throw MakeStringException(WUERR_LockFailed, "Failed to get connection for xpath %s", wuRoot.str());
  3030. connectAtRoot = true;
  3031. init();
  3032. p.setown(connection->getRoot());
  3033. }
  3034. catch (IException *E)
  3035. {
  3036. StringBuffer s;
  3037. PrintLog("Failed to get write lock on workunit: %s", E->errorMessage(s).str());
  3038. locked.unlock();
  3039. throw;
  3040. }
  3041. }
  3042. return *new CLockedWorkUnit(LINK(this));
  3043. }
  3044. void CLocalWorkUnit::commit()
  3045. {
  3046. CriticalBlock block(crit);
  3047. assertex(connectAtRoot);
  3048. if (connection)
  3049. connection->commit();
  3050. }
  3051. IWorkUnit& CLocalWorkUnit::lock()
  3052. {
  3053. return lockRemote(true);
  3054. }
  3055. IStringVal& CLocalWorkUnit::getWuid(IStringVal &str) const
  3056. {
  3057. CriticalBlock block(crit);
  3058. str.set(p->queryName());
  3059. return str;
  3060. }
  3061. unsigned CLocalWorkUnit::getDebugAgentListenerPort() const
  3062. {
  3063. CriticalBlock block(crit);
  3064. return p->getPropInt("@DebugListenerPort", 0);
  3065. }
  3066. void CLocalWorkUnit::setDebugAgentListenerPort(unsigned port)
  3067. {
  3068. CriticalBlock block(crit);
  3069. p->setPropInt("@DebugListenerPort", port);
  3070. }
  3071. IStringVal& CLocalWorkUnit::getDebugAgentListenerIP(IStringVal &ip) const
  3072. {
  3073. CriticalBlock block(crit);
  3074. ip.set(p->queryProp("@DebugListenerIP"));
  3075. return ip;
  3076. }
  3077. void CLocalWorkUnit::setDebugAgentListenerIP(const char * ip)
  3078. {
  3079. CriticalBlock block(crit);
  3080. p->setProp("@DebugListenerIP", ip);
  3081. }
  3082. IStringVal& CLocalWorkUnit::getSecurityToken(IStringVal &str) const
  3083. {
  3084. CriticalBlock block(crit);
  3085. str.set(p->queryProp("@token"));
  3086. return str;
  3087. }
  3088. void CLocalWorkUnit::setSecurityToken(const char *value)
  3089. {
  3090. CriticalBlock block(crit);
  3091. p->setProp("@token", value);
  3092. }
  3093. bool CLocalWorkUnit::getRunningGraph(IStringVal &graphName, WUGraphIDType &subId) const
  3094. {
  3095. return CConstGraphProgress::getRunningGraph(p->queryName(), graphName, subId);
  3096. }
  3097. void CLocalWorkUnit::setJobName(const char *value)
  3098. {
  3099. CriticalBlock block(crit);
  3100. p->setProp("@jobName", value);
  3101. }
  3102. IStringVal& CLocalWorkUnit::getJobName(IStringVal &str) const
  3103. {
  3104. CriticalBlock block(crit);
  3105. str.set(p->queryProp("@jobName"));
  3106. return str;
  3107. }
  3108. void CLocalWorkUnit::setClusterName(const char *value)
  3109. {
  3110. CriticalBlock block(crit);
  3111. p->setProp("@clusterName", value);
  3112. }
  3113. IStringVal& CLocalWorkUnit::getClusterName(IStringVal &str) const
  3114. {
  3115. CriticalBlock block(crit);
  3116. str.set(p->queryProp("@clusterName"));
  3117. return str;
  3118. }
  3119. void CLocalWorkUnit::setAllowedClusters(const char *value)
  3120. {
  3121. setDebugValue("allowedclusters",value, true);
  3122. }
  3123. IStringVal& CLocalWorkUnit::getAllowedClusters(IStringVal &str) const
  3124. {
  3125. CriticalBlock block(crit);
  3126. getDebugValue("allowedclusters",str);
  3127. if (str.length()!=0)
  3128. return str;
  3129. str.set(p->queryProp("@clusterName"));
  3130. return str;
  3131. }
  3132. void CLocalWorkUnit::setAllowAutoQueueSwitch(bool val)
  3133. {
  3134. setDebugValueInt("allowautoqueueswitch",val?1:0,true);
  3135. }
  3136. bool CLocalWorkUnit::getAllowAutoQueueSwitch() const
  3137. {
  3138. CriticalBlock block(crit);
  3139. return getDebugValueBool("allowautoqueueswitch",false);
  3140. }
  3141. void CLocalWorkUnit::setLibraryInformation(const char * name, unsigned interfaceHash, unsigned definitionHash)
  3142. {
  3143. StringBuffer suffix;
  3144. if (name && *name)
  3145. setApplicationValue("LibraryModule", "name", name, true);
  3146. setApplicationValueInt("LibraryModule", "interfaceHash", interfaceHash, true);
  3147. setApplicationValueInt("LibraryModule", "definitionHash", definitionHash, true);
  3148. setApplicationValue("LibraryModule", "platform", appendLibrarySuffix(suffix).str(), true);
  3149. }
  3150. void CLocalWorkUnit::remoteCheckAccess(IUserDescriptor *user, bool writeaccess) const
  3151. {
  3152. unsigned auditflags = DALI_LDAP_AUDIT_REPORT|DALI_LDAP_READ_WANTED;
  3153. if (writeaccess)
  3154. auditflags |= DALI_LDAP_WRITE_WANTED;
  3155. int perm = 255;
  3156. const char *scopename = p->queryProp("@scope");
  3157. if (scopename&&*scopename) {
  3158. if (!user)
  3159. user = queryUserDescriptor();
  3160. perm = querySessionManager().getPermissionsLDAP("workunit",scopename,user,auditflags);
  3161. if (perm<0) {
  3162. if (perm==-1)
  3163. perm = 255;
  3164. else
  3165. perm = 0;
  3166. }
  3167. }
  3168. if (!HASREADPERMISSION(perm)) {
  3169. SCMStringBuffer wuid;
  3170. getWuid(wuid);
  3171. throw MakeStringException(WUERR_WorkunitAccessDenied, "Read access denied for workunit %s",wuid.s.str());
  3172. }
  3173. if (writeaccess&&!HASWRITEPERMISSION(perm)) {
  3174. SCMStringBuffer wuid;
  3175. getWuid(wuid);
  3176. throw MakeStringException(WUERR_WorkunitAccessDenied, "Write access denied for workunit %s",wuid.s.str());
  3177. }
  3178. }
  3179. IStringVal& CLocalWorkUnit::getParentWuid(IStringVal &str) const
  3180. {
  3181. CriticalBlock block(crit);
  3182. str.set(p->queryProp("@parent"));
  3183. return str;
  3184. }
  3185. void CLocalWorkUnit::setUser(const char * value)
  3186. {
  3187. CriticalBlock block(crit);
  3188. p->setProp("@submitID", value);
  3189. }
  3190. IStringVal& CLocalWorkUnit::getUser(IStringVal &str) const
  3191. {
  3192. CriticalBlock block(crit);
  3193. str.set(p->queryProp("@submitID"));
  3194. return str;
  3195. }
  3196. void CLocalWorkUnit::setWuScope(const char * value)
  3197. {
  3198. CriticalBlock block(crit);
  3199. p->setProp("@scope", value);
  3200. }
  3201. IStringVal& CLocalWorkUnit::getWuScope(IStringVal &str) const
  3202. {
  3203. CriticalBlock block(crit);
  3204. str.set(p->queryProp("@scope"));
  3205. return str;
  3206. }
  3207. void CLocalWorkUnit::setCustomerId(const char * value)
  3208. {
  3209. CriticalBlock block(crit);
  3210. p->setProp("CustomerID", value);
  3211. }
  3212. IStringVal& CLocalWorkUnit::getCustomerId(IStringVal &str) const
  3213. {
  3214. CriticalBlock block(crit);
  3215. str.set(p->queryProp("CustomerID"));
  3216. return str;
  3217. }
  3218. mapEnums priorityClasses[] = {
  3219. { PriorityClassUnknown, "unknown" },
  3220. { PriorityClassLow, "low" },
  3221. { PriorityClassNormal, "normal" },
  3222. { PriorityClassHigh, "high" },
  3223. { PriorityClassSize, NULL },
  3224. };
  3225. void CLocalWorkUnit::setPriority(WUPriorityClass cls)
  3226. {
  3227. CriticalBlock block(crit);
  3228. setEnum(p, "@priorityClass", cls, priorityClasses);
  3229. }
  3230. WUPriorityClass CLocalWorkUnit::getPriority() const
  3231. {
  3232. CriticalBlock block(crit);
  3233. return (WUPriorityClass) getEnum(p, "@priorityClass", priorityClasses);
  3234. }
  3235. mapEnums states[] = {
  3236. { WUStateUnknown, "unknown" },
  3237. { WUStateCompiled, "compiled" },
  3238. { WUStateRunning, "running" },
  3239. { WUStateCompleted, "completed" },
  3240. { WUStateFailed, "failed" },
  3241. { WUStateArchived, "archived" },
  3242. { WUStateAborting, "aborting" },
  3243. { WUStateAborted, "aborted" },
  3244. { WUStateBlocked, "blocked" },
  3245. { WUStateSubmitted, "submitted" },
  3246. { WUStateScheduled, "scheduled" },
  3247. { WUStateCompiling, "compiling" },
  3248. { WUStateWait, "wait" },
  3249. { WUStateUploadingFiles, "uploading_files" },
  3250. { WUStateDebugPaused, "debugging" },
  3251. { WUStateDebugRunning, "debug_running" },
  3252. { WUStatePaused, "paused" },
  3253. { WUStateSize, NULL }
  3254. };
  3255. IConstWorkUnitIterator * CWorkUnitFactory::getWorkUnitsByState(WUState state)
  3256. {
  3257. StringBuffer path("*");
  3258. path.append("[@state=\"").append(getEnumText(state, states)).append("\"]");
  3259. return getWorkUnitsByXPath(path.str());
  3260. }
  3261. IConstWorkUnitIterator * CSecureWorkUnitFactory::getWorkUnitsByState(WUState state)
  3262. {
  3263. StringBuffer path("*");
  3264. path.append("[@state=\"").append(getEnumText(state, states)).append("\"]");
  3265. return factory->getWorkUnitsByXPath(path.str(), secMgr.get(), secUser.get());
  3266. }
  3267. void CLocalWorkUnit::setState(WUState value)
  3268. {
  3269. CriticalBlock block(crit);
  3270. if (value==WUStateAborted || value==WUStatePaused || value==WUStateCompleted || value==WUStateFailed || value==WUStateSubmitted || value==WUStateWait)
  3271. {
  3272. if (abortWatcher)
  3273. {
  3274. abortWatcher->unsubscribe();
  3275. abortWatcher.clear();
  3276. }
  3277. StringBuffer apath;
  3278. apath.append("/WorkUnitAborts/").append(p->queryName());
  3279. if(factory)
  3280. {
  3281. Owned<IRemoteConnection> acon = factory->sdsManager->connect(apath.str(), factory->session, RTM_LOCK_WRITE|RTM_LOCK_SUB, SDS_LOCK_TIMEOUT);
  3282. if (acon)
  3283. acon->close(true);
  3284. }
  3285. }
  3286. setEnum(p, "@state", value, states);
  3287. if (getDebugValueBool("monitorWorkunit", false))
  3288. {
  3289. switch(value)
  3290. {
  3291. case WUStateAborted:
  3292. FLLOG(MCoperatorWarning, "Workunit %s aborted", p->queryName());
  3293. break;
  3294. case WUStateCompleted:
  3295. FLLOG(MCoperatorProgress, "Workunit %s completed", p->queryName());
  3296. break;
  3297. case WUStateFailed:
  3298. FLLOG(MCoperatorProgress, "Workunit %s failed", p->queryName());
  3299. break;
  3300. }
  3301. }
  3302. p->removeProp("@stateEx");
  3303. }
  3304. void CLocalWorkUnit::setStateEx(const char * text)
  3305. {
  3306. CriticalBlock block(crit);
  3307. p->setProp("@stateEx", text);
  3308. }
  3309. void CLocalWorkUnit::setAgentSession(__int64 sessionId)
  3310. {
  3311. CriticalBlock block(crit);
  3312. p->setPropInt64("@agentSession", sessionId);
  3313. }
  3314. void CLocalWorkUnit::setAgentPID(unsigned pid)
  3315. {
  3316. CriticalBlock block(crit);
  3317. p->setPropInt("@agentPID", pid);
  3318. }
  3319. bool CLocalWorkUnit::aborting() const
  3320. {
  3321. CriticalBlock block(crit);
  3322. if (abortDirty)
  3323. {
  3324. if (factory)
  3325. {
  3326. StringBuffer apath;
  3327. apath.append("/WorkUnitAborts/").append(p->queryName());
  3328. Owned<IRemoteConnection> acon = factory->sdsManager->connect(apath.str(), factory->session, 0, SDS_LOCK_TIMEOUT);
  3329. if (acon)
  3330. abortState = acon->queryRoot()->getPropInt(NULL)!=0;
  3331. else
  3332. abortState = false;
  3333. }
  3334. abortDirty = false;
  3335. }
  3336. return abortState;
  3337. }
  3338. bool CLocalWorkUnit::getIsQueryService() const
  3339. {
  3340. CriticalBlock block(crit);
  3341. return p->getPropBool("@isQueryService", false);
  3342. }
  3343. void CLocalWorkUnit::setIsQueryService(bool value)
  3344. {
  3345. CriticalBlock block(crit);
  3346. p->setPropBool("@isQueryService", value);
  3347. }
  3348. void CLocalWorkUnit::checkAgentRunning(WUState & state)
  3349. {
  3350. if (queryDaliServerVersion().compare("2.1")<0)
  3351. return;
  3352. switch(state)
  3353. {
  3354. case WUStateRunning:
  3355. case WUStateDebugPaused:
  3356. case WUStateDebugRunning:
  3357. case WUStateBlocked:
  3358. case WUStateAborting:
  3359. case WUStateCompiling:
  3360. case WUStatePaused:
  3361. {
  3362. SessionId agent = getAgentSession();
  3363. if((agent>0) && querySessionManager().sessionStopped(agent, 0))
  3364. {
  3365. forceReload();
  3366. state = (WUState) getEnum(p, "@state", states);
  3367. bool isecl=state==WUStateCompiling;
  3368. if (aborting())
  3369. state = WUStateAborted;
  3370. else if (state==WUStateRunning || state==WUStatePaused || state==WUStateDebugPaused || state==WUStateDebugRunning || state==WUStateBlocked || state==WUStateCompiling)
  3371. state = WUStateFailed;
  3372. else
  3373. return;
  3374. WARNLOG("checkAgentRunning terminated: %"I64F"d state = %d",(__int64)agent,(int)state);
  3375. Owned<IWorkUnit> w = &lock();
  3376. w->setState(state);
  3377. Owned<IWUException> e = w->createException();
  3378. WUAction action = w->getAction();
  3379. switch (action)
  3380. {
  3381. case WUActionPause:
  3382. case WUActionPauseNow:
  3383. case WUActionResume:
  3384. w->setAction(WUActionUnknown);
  3385. }
  3386. if(isecl)
  3387. {
  3388. e->setExceptionCode(1001);
  3389. e->setExceptionMessage("EclServer terminated unexpectedly");
  3390. }
  3391. else
  3392. {
  3393. e->setExceptionCode(1000);
  3394. e->setExceptionMessage("Workunit terminated unexpectedly");
  3395. }
  3396. }
  3397. }
  3398. }
  3399. }
  3400. WUState CLocalWorkUnit::getState() const
  3401. {
  3402. CriticalBlock block(crit);
  3403. WUState state = (WUState) getEnum(p, "@state", states);
  3404. switch (state)
  3405. {
  3406. case WUStateRunning:
  3407. case WUStateDebugPaused:
  3408. case WUStateDebugRunning:
  3409. case WUStateBlocked:
  3410. case WUStateCompiling:
  3411. if (aborting())
  3412. state = WUStateAborting;
  3413. break;
  3414. case WUStateSubmitted:
  3415. if (aborting())
  3416. state = WUStateAborted;
  3417. break;
  3418. }
  3419. const_cast<CLocalWorkUnit *>(this)->checkAgentRunning(state); //need const_cast as will change state if agent has died
  3420. return state;
  3421. }
  3422. IStringVal& CLocalWorkUnit::getStateEx(IStringVal & str) const
  3423. {
  3424. CriticalBlock block(crit);
  3425. str.set(p->queryProp("@stateEx"));
  3426. return str;
  3427. }
  3428. __int64 CLocalWorkUnit::getAgentSession() const
  3429. {
  3430. CriticalBlock block(crit);
  3431. return p->getPropInt64("@agentSession", -1);
  3432. }
  3433. unsigned CLocalWorkUnit::getAgentPID() const
  3434. {
  3435. CriticalBlock block(crit);
  3436. return p->getPropInt("@agentPID", -1);
  3437. }
  3438. IStringVal& CLocalWorkUnit::getStateDesc(IStringVal &str) const
  3439. {
  3440. // MORE - not sure about this - may prefer a separate interface
  3441. CriticalBlock block(crit);
  3442. try
  3443. {
  3444. str.set(getEnumText(getState(), states));
  3445. }
  3446. catch (...)
  3447. {
  3448. str.set("???");
  3449. }
  3450. return str;
  3451. }
  3452. mapEnums actions[] = {
  3453. { WUActionUnknown, "unknown" },
  3454. { WUActionCompile, "compile" },
  3455. { WUActionCheck, "check" },
  3456. { WUActionRun, "run" },
  3457. { WUActionExecuteExisting, "execute" },
  3458. { WUActionPause, "pause" },
  3459. { WUActionPauseNow, "pausenow" },
  3460. { WUActionResume, "resume" },
  3461. { WUActionSize, NULL },
  3462. };
  3463. void CLocalWorkUnit::setAction(WUAction value)
  3464. {
  3465. CriticalBlock block(crit);
  3466. setEnum(p, "Action", value, actions);
  3467. }
  3468. WUAction CLocalWorkUnit::getAction() const
  3469. {
  3470. CriticalBlock block(crit);
  3471. return (WUAction) getEnum(p, "Action", actions);
  3472. }
  3473. IStringVal& CLocalWorkUnit::getActionEx(IStringVal & str) const
  3474. {
  3475. CriticalBlock block(crit);
  3476. str.set(p->queryProp("Action"));
  3477. return str;
  3478. }
  3479. IStringVal& CLocalWorkUnit::getApplicationValue(const char *app, const char *propname, IStringVal &str) const
  3480. {
  3481. CriticalBlock block(crit);
  3482. StringBuffer prop("Application/");
  3483. prop.append(app).append('/').append(propname);
  3484. str.set(p->queryProp(prop.str()));
  3485. return str;
  3486. }
  3487. int CLocalWorkUnit::getApplicationValueInt(const char *app, const char *propname, int defVal) const
  3488. {
  3489. CriticalBlock block(crit);
  3490. StringBuffer prop("Application/");
  3491. prop.append(app).append('/').append(propname);
  3492. return p->getPropInt(prop.str(), defVal);
  3493. }
  3494. IConstWUAppValueIterator& CLocalWorkUnit::getApplicationValues() const
  3495. {
  3496. CriticalBlock block(crit);
  3497. appvalues.load(p,"Application/*");
  3498. return *new CArrayIteratorOf<IConstWUAppValue,IConstWUAppValueIterator> (appvalues, 0, (IConstWorkUnit *) this);
  3499. }
  3500. void CLocalWorkUnit::setApplicationValue(const char *app, const char *propname, const char *value, bool overwrite)
  3501. {
  3502. CriticalBlock block(crit);
  3503. StringBuffer prop("Application/");
  3504. prop.append(app).append('/').append(propname);
  3505. if (overwrite || !p->hasProp(prop.str()))
  3506. {
  3507. // MORE - not sure these lines should be needed....
  3508. StringBuffer sp;
  3509. p->setProp(sp.append("Application").str(), "");
  3510. p->setProp(sp.append('/').append(app).str(), "");
  3511. p->setProp(prop.str(), value);
  3512. }
  3513. }
  3514. void CLocalWorkUnit::setApplicationValueInt(const char *app, const char *propname, int value, bool overwrite)
  3515. {
  3516. CriticalBlock block(crit);
  3517. StringBuffer prop("Application/");
  3518. prop.append(app).append('/').append(propname);
  3519. if (overwrite || !p->hasProp(prop.str()))
  3520. {
  3521. // MORE - not sure these lines should be needed....
  3522. StringBuffer sp;
  3523. p->setProp(sp.append("Application").str(), "");
  3524. p->setProp(sp.append('/').append(app).str(), "");
  3525. p->setPropInt(prop.str(), value);
  3526. }
  3527. }
  3528. void CLocalWorkUnit::setPriorityLevel(int level)
  3529. {
  3530. CriticalBlock block(crit);
  3531. p->setPropInt("PriorityFlag", level);
  3532. }
  3533. int CLocalWorkUnit::getPriorityLevel() const
  3534. {
  3535. CriticalBlock block(crit);
  3536. return p->getPropInt("PriorityFlag");
  3537. }
  3538. int CLocalWorkUnit::getPriorityValue() const
  3539. {
  3540. CriticalBlock block(crit);
  3541. int priority = p->getPropInt("PriorityFlag");
  3542. switch((WUPriorityClass) getEnum(p, "@priorityClass", priorityClasses))
  3543. {
  3544. case PriorityClassLow:
  3545. priority -= 100;
  3546. break;
  3547. case PriorityClassHigh:
  3548. priority += 100;
  3549. break;
  3550. }
  3551. return priority;
  3552. }
  3553. void CLocalWorkUnit::setRescheduleFlag(bool value)
  3554. {
  3555. CriticalBlock block(crit);
  3556. p->setPropInt("RescheduleFlag", (int) value);
  3557. }
  3558. bool CLocalWorkUnit::getRescheduleFlag() const
  3559. {
  3560. CriticalBlock block(crit);
  3561. return p->getPropInt("RescheduleFlag") != 0;
  3562. }
  3563. class NullIStringIterator : public CInterface, extends IStringIterator
  3564. {
  3565. public:
  3566. IMPLEMENT_IINTERFACE;
  3567. bool first() { return false; }
  3568. bool next() { return false; }
  3569. bool isValid() { return false; }
  3570. IStringVal & str(IStringVal & str) { return str; }
  3571. };
  3572. ClusterType getClusterType(const char * platform, ClusterType dft)
  3573. {
  3574. if (stricmp(platform, "thor") == 0)
  3575. return ThorCluster;
  3576. if (stricmp(platform, "thorlcr") == 0)
  3577. return ThorLCRCluster;
  3578. if (stricmp(platform, "hthor") == 0)
  3579. return HThorCluster;
  3580. if (stricmp(platform, "roxie") == 0)
  3581. return RoxieCluster;
  3582. return dft;
  3583. }
  3584. const char *clusterTypeString(ClusterType clusterType, bool lcrSensitive)
  3585. {
  3586. switch (clusterType)
  3587. {
  3588. case ThorLCRCluster:
  3589. if (lcrSensitive)
  3590. return "thorlcr";
  3591. // fall through
  3592. case ThorCluster:
  3593. return "thor";
  3594. case RoxieCluster:
  3595. return "roxie";
  3596. case HThorCluster:
  3597. return "hthor";
  3598. }
  3599. throwUnexpected();
  3600. }
  3601. IPropertyTree *queryRoxieProcessTree(IPropertyTree *environment, const char *process)
  3602. {
  3603. if (!process || !*process)
  3604. return NULL;
  3605. VStringBuffer xpath("Software/RoxieCluster[@name=\"%s\"]", process);
  3606. return environment->queryPropTree(xpath.str());
  3607. }
  3608. void getRoxieProcessServers(IPropertyTree *roxie, SocketEndpointArray &endpoints)
  3609. {
  3610. if (!roxie)
  3611. return;
  3612. Owned<IPropertyTreeIterator> servers = roxie->getElements("RoxieServerProcess");
  3613. ForEach(*servers)
  3614. {
  3615. IPropertyTree &server = servers->query();
  3616. const char *netAddress = server.queryProp("@netAddress");
  3617. if (netAddress && *netAddress)
  3618. {
  3619. SocketEndpoint ep(netAddress, server.getPropInt("@port", 9876));
  3620. endpoints.append(ep);
  3621. }
  3622. }
  3623. }
  3624. void getRoxieProcessServers(const char *process, SocketEndpointArray &servers)
  3625. {
  3626. Owned<IRemoteConnection> conn = querySDS().connect("Environment", myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT);
  3627. if (!conn)
  3628. return;
  3629. getRoxieProcessServers(queryRoxieProcessTree(conn->queryRoot(), process), servers);
  3630. }
  3631. class CEnvironmentClusterInfo: public CInterface, implements IConstWUClusterInfo
  3632. {
  3633. StringAttr name;
  3634. StringAttr serverQueue;
  3635. StringAttr agentQueue;
  3636. StringAttr roxieProcess;
  3637. SocketEndpointArray roxieServers;
  3638. StringAttr thorQueue;
  3639. StringArray thorProcesses;
  3640. StringAttr prefix;
  3641. ClusterType platform;
  3642. unsigned clusterWidth;
  3643. public:
  3644. IMPLEMENT_IINTERFACE;
  3645. CEnvironmentClusterInfo(const char *_name, const char *_prefix, IPropertyTree *agent, IArrayOf<IPropertyTree> &thors, IPropertyTree *roxie)
  3646. : name(_name), prefix(_prefix)
  3647. {
  3648. StringBuffer queue;
  3649. if (thors.ordinality())
  3650. {
  3651. thorQueue.set(getClusterThorQueueName(queue.clear(), name));
  3652. clusterWidth = 0;
  3653. bool lcr = false;
  3654. ForEachItemIn(i,thors)
  3655. {
  3656. IPropertyTree &thor = thors.item(i);
  3657. thorProcesses.append(thor.queryProp("@name"));
  3658. unsigned nodes = thor.getCount("ThorSlaveProcess");
  3659. if (!nodes)
  3660. throw MakeStringException(WUERR_MismatchClusterSize,"CEnvironmentClusterInfo: Thor cluster can not have 0 slave processes");
  3661. unsigned ts = nodes * thor.getPropInt("@slavesPerNode", 1);
  3662. if (clusterWidth && (ts!=clusterWidth))
  3663. throw MakeStringException(WUERR_MismatchClusterSize,"CEnvironmentClusterInfo: mismatched thor sizes in cluster");
  3664. clusterWidth = ts;
  3665. bool islcr = !thor.getPropBool("@Legacy");
  3666. if (i==0)
  3667. lcr = islcr;
  3668. else if (lcr!=islcr)
  3669. throw MakeStringException(WUERR_MismatchThorType,"CEnvironmentClusterInfo: mismatched thor Legacy in cluster");
  3670. }
  3671. platform = lcr ? ThorLCRCluster : ThorCluster;
  3672. }
  3673. else if (roxie)
  3674. {
  3675. roxieProcess.set(roxie->queryProp("@name"));
  3676. clusterWidth = roxie->getPropInt("@numChannels", 1);
  3677. platform = RoxieCluster;
  3678. getRoxieProcessServers(roxie, roxieServers);
  3679. }
  3680. else
  3681. {
  3682. clusterWidth = 1;
  3683. platform = HThorCluster;
  3684. }
  3685. if (agent)
  3686. {
  3687. assertex(!roxie);
  3688. agentQueue.set(getClusterEclAgentQueueName(queue.clear(), name));
  3689. }
  3690. else if (roxie)
  3691. agentQueue.set(getClusterRoxieQueueName(queue.clear(), name));
  3692. // MORE - does this need to be conditional?
  3693. serverQueue.set(getClusterEclCCServerQueueName(queue.clear(), name));
  3694. }
  3695. IStringVal & getName(IStringVal & str) const
  3696. {
  3697. str.set(name.get());
  3698. return str;
  3699. }
  3700. IStringVal & getScope(IStringVal & str) const
  3701. {
  3702. str.set(prefix.get());
  3703. return str;
  3704. }
  3705. IStringVal & getAgentQueue(IStringVal & str) const
  3706. {
  3707. str.set(agentQueue);
  3708. return str;
  3709. }
  3710. virtual IStringVal & getServerQueue(IStringVal & str) const
  3711. {
  3712. str.set(serverQueue);
  3713. return str;
  3714. }
  3715. IStringVal & getThorQueue(IStringVal & str) const
  3716. {
  3717. str.set(thorQueue);
  3718. return str;
  3719. }
  3720. unsigned getSize() const
  3721. {
  3722. return clusterWidth;
  3723. }
  3724. virtual ClusterType getPlatform() const
  3725. {
  3726. return platform;
  3727. }
  3728. IStringVal & getRoxieProcess(IStringVal & str) const
  3729. {
  3730. str.set(roxieProcess.get());
  3731. return str;
  3732. }
  3733. const StringArray & getThorProcesses() const
  3734. {
  3735. return thorProcesses;
  3736. }
  3737. const SocketEndpointArray & getRoxieServers() const
  3738. {
  3739. return roxieServers;
  3740. }
  3741. };
  3742. IStringVal &getProcessQueueNames(IStringVal &ret, const char *process, const char *type, const char *suffix)
  3743. {
  3744. if (process)
  3745. {
  3746. Owned<IRemoteConnection> conn = querySDS().connect("Environment", myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT);
  3747. if (conn)
  3748. {
  3749. StringBuffer queueNames;
  3750. StringBuffer xpath;
  3751. xpath.appendf("%s[@process=\"%s\"]", type, process);
  3752. Owned<IPropertyTreeIterator> targets = conn->queryRoot()->getElements("Software/Topology/Cluster");
  3753. ForEach(*targets)
  3754. {
  3755. IPropertyTree &target = targets->query();
  3756. if (target.hasProp(xpath))
  3757. {
  3758. if (queueNames.length())
  3759. queueNames.append(',');
  3760. queueNames.append(target.queryProp("@name")).append(suffix);
  3761. }
  3762. }
  3763. ret.set(queueNames);
  3764. }
  3765. }
  3766. return ret;
  3767. }
  3768. #define ROXIE_QUEUE_EXT ".roxie"
  3769. #define THOR_QUEUE_EXT ".thor"
  3770. #define ECLCCSERVER_QUEUE_EXT ".eclserver"
  3771. #define ECLSERVER_QUEUE_EXT ECLCCSERVER_QUEUE_EXT
  3772. #define ECLSCHEDULER_QUEUE_EXT ".eclscheduler"
  3773. #define ECLAGENT_QUEUE_EXT ".agent"
  3774. extern WORKUNIT_API IStringVal &getEclCCServerQueueNames(IStringVal &ret, const char *process)
  3775. {
  3776. return getProcessQueueNames(ret, process, "EclCCServerProcess", ECLCCSERVER_QUEUE_EXT);
  3777. }
  3778. extern WORKUNIT_API IStringVal &getEclServerQueueNames(IStringVal &ret, const char *process)
  3779. {
  3780. return getProcessQueueNames(ret, process, "EclServerProcess", ECLSERVER_QUEUE_EXT); // shares queue name with EclCCServer
  3781. }
  3782. extern WORKUNIT_API IStringVal &getEclSchedulerQueueNames(IStringVal &ret, const char *process)
  3783. {
  3784. return getProcessQueueNames(ret, process, "EclSchedulerProcess", ECLSCHEDULER_QUEUE_EXT); // Shares deployment/config with EclCCServer
  3785. }
  3786. extern WORKUNIT_API IStringVal &getAgentQueueNames(IStringVal &ret, const char *process)
  3787. {
  3788. return getProcessQueueNames(ret, process, "EclAgentProcess", ECLAGENT_QUEUE_EXT);
  3789. }
  3790. extern WORKUNIT_API IStringVal &getRoxieQueueNames(IStringVal &ret, const char *process)
  3791. {
  3792. return getProcessQueueNames(ret, process, "RoxieCluster", ROXIE_QUEUE_EXT);
  3793. }
  3794. extern WORKUNIT_API IStringVal &getThorQueueNames(IStringVal &ret, const char *process)
  3795. {
  3796. return getProcessQueueNames(ret, process, "ThorCluster", THOR_QUEUE_EXT);
  3797. }
  3798. extern WORKUNIT_API StringBuffer &getClusterThorQueueName(StringBuffer &ret, const char *cluster)
  3799. {
  3800. return ret.append(cluster).append(THOR_QUEUE_EXT);
  3801. }
  3802. extern WORKUNIT_API StringBuffer &getClusterThorGroupName(StringBuffer &ret, const char *cluster)
  3803. {
  3804. StringBuffer path;
  3805. Owned<IRemoteConnection> conn = querySDS().connect(path.append("Environment/Software/ThorCluster[@name=\"").append(cluster).append("\"]").str(), myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT);
  3806. if (conn)
  3807. {
  3808. getClusterGroupName(*conn->queryRoot(), ret);
  3809. }
  3810. return ret;
  3811. }
  3812. extern WORKUNIT_API StringBuffer &getClusterRoxieQueueName(StringBuffer &ret, const char *cluster)
  3813. {
  3814. return ret.append(cluster).append(ROXIE_QUEUE_EXT);
  3815. }
  3816. extern WORKUNIT_API StringBuffer &getClusterEclCCServerQueueName(StringBuffer &ret, const char *cluster)
  3817. {
  3818. return ret.append(cluster).append(ECLCCSERVER_QUEUE_EXT);
  3819. }
  3820. extern WORKUNIT_API StringBuffer &getClusterEclServerQueueName(StringBuffer &ret, const char *cluster)
  3821. {
  3822. return ret.append(cluster).append(ECLSERVER_QUEUE_EXT);
  3823. }
  3824. extern WORKUNIT_API StringBuffer &getClusterEclAgentQueueName(StringBuffer &ret, const char *cluster)
  3825. {
  3826. return ret.append(cluster).append(ECLAGENT_QUEUE_EXT);
  3827. }
  3828. extern WORKUNIT_API IStringIterator *getTargetClusters(const char *processType, const char *processName)
  3829. {
  3830. Owned<IRemoteConnection> conn = querySDS().connect("Environment", myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT);
  3831. Owned<CStringArrayIterator> ret = new CStringArrayIterator;
  3832. if (conn)
  3833. {
  3834. StringBuffer xpath;
  3835. xpath.appendf("%s", processType ? processType : "*");
  3836. if (processName)
  3837. xpath.appendf("[@process=\"%s\"]", processName);
  3838. Owned<IPropertyTreeIterator> targets = conn->queryRoot()->getElements("Software/Topology/Cluster");
  3839. ForEach(*targets)
  3840. {
  3841. IPropertyTree &target = targets->query();
  3842. if (target.hasProp(xpath))
  3843. {
  3844. ret->append(target.queryProp("@name"));
  3845. }
  3846. }
  3847. }
  3848. return ret.getClear();
  3849. }
  3850. IConstWUClusterInfo* getTargetClusterInfo(IPropertyTree *environment, IPropertyTree *cluster)
  3851. {
  3852. const char *clustname = cluster->queryProp("@name");
  3853. // MORE - at the moment configenf specifies eclagent and thor queues by (in effect) placing an 'example' thor or eclagent in the topology
  3854. // that uses the queue that will be used.
  3855. // We should and I hope will change that, at which point the code below gets simpler
  3856. StringBuffer prefix(cluster->queryProp("@prefix"));
  3857. prefix.toLowerCase();
  3858. StringBuffer xpath;
  3859. StringBuffer querySetName;
  3860. IPropertyTree *agent = NULL;
  3861. const char *agentName = cluster->queryProp("EclAgentProcess/@process");
  3862. if (agentName)
  3863. {
  3864. xpath.clear().appendf("Software/EclAgentProcess[@name=\"%s\"]", agentName);
  3865. agent = environment->queryPropTree(xpath.str());
  3866. }
  3867. Owned<IPropertyTreeIterator> ti = cluster->getElements("ThorCluster");
  3868. IArrayOf<IPropertyTree> thors;
  3869. ForEach(*ti)
  3870. {
  3871. const char *thorName = ti->query().queryProp("@process");
  3872. if (thorName)
  3873. {
  3874. xpath.clear().appendf("Software/ThorCluster[@name=\"%s\"]", thorName);
  3875. thors.append(*environment->getPropTree(xpath.str()));
  3876. }
  3877. }
  3878. const char *roxieName = cluster->queryProp("RoxieCluster/@process");
  3879. return new CEnvironmentClusterInfo(clustname, prefix, agent, thors, queryRoxieProcessTree(environment, roxieName));
  3880. }
  3881. IPropertyTree* getTopologyCluster(Owned<IRemoteConnection> &conn, const char *clustname)
  3882. {
  3883. if (!clustname || !*clustname)
  3884. return NULL;
  3885. conn.setown(querySDS().connect("Environment", myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT));
  3886. if (!conn)
  3887. return NULL;
  3888. StringBuffer xpath;
  3889. xpath.appendf("Software/Topology/Cluster[@name=\"%s\"]", clustname);
  3890. return conn->queryRoot()->getPropTree(xpath.str());
  3891. }
  3892. bool validateTargetClusterName(const char *clustname)
  3893. {
  3894. Owned<IRemoteConnection> conn;
  3895. Owned<IPropertyTree> cluster = getTopologyCluster(conn, clustname);
  3896. return (cluster.get()!=NULL);
  3897. }
  3898. IConstWUClusterInfo* getTargetClusterInfo(const char *clustname)
  3899. {
  3900. Owned<IRemoteConnection> conn;
  3901. Owned<IPropertyTree> cluster = getTopologyCluster(conn, clustname);
  3902. if (!cluster)
  3903. return NULL;
  3904. return getTargetClusterInfo(conn->queryRoot(), cluster);
  3905. }
  3906. unsigned getEnvironmentClusterInfo(CConstWUClusterInfoArray &clusters)
  3907. {
  3908. Owned<IRemoteConnection> conn = querySDS().connect("Environment", myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT);
  3909. if (!conn)
  3910. return 0;
  3911. Owned<IPropertyTreeIterator> clusterIter = conn->queryRoot()->getElements("Software/Topology/Cluster");
  3912. ForEach(*clusterIter)
  3913. {
  3914. IPropertyTree &node = clusterIter->query();
  3915. Owned<IConstWUClusterInfo> cluster = getTargetClusterInfo(conn->queryRoot(), &node);
  3916. clusters.append(*cluster.getClear());
  3917. }
  3918. return clusters.ordinality();
  3919. }
  3920. const char *getTargetClusterComponentName(const char *clustname, const char *processType, StringBuffer &name)
  3921. {
  3922. if (!clustname)
  3923. return NULL;
  3924. Owned<IRemoteConnection> conn = querySDS().connect("Environment", myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT);
  3925. if (!conn)
  3926. return NULL;
  3927. StringBuffer xpath;
  3928. xpath.appendf("Software/Topology/Cluster[@name=\"%s\"]", clustname);
  3929. Owned<IPropertyTree> cluster = conn->queryRoot()->getPropTree(xpath.str());
  3930. if (!cluster)
  3931. return NULL;
  3932. StringBuffer xpath1;
  3933. xpath1.appendf("%s/@process", processType);
  3934. name.append(cluster->queryProp(xpath1.str()));
  3935. return name.str();
  3936. }
  3937. unsigned getEnvironmentThorClusterNames(StringArray &thorNames, StringArray &groupNames, StringArray &targetNames, StringArray &queueNames)
  3938. {
  3939. Owned<IRemoteConnection> conn = querySDS().connect("Environment", myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT);
  3940. if (!conn)
  3941. return 0;
  3942. Owned<IPropertyTreeIterator> allTargets = conn->queryRoot()->getElements("Software/Topology/Cluster");
  3943. ForEach(*allTargets)
  3944. {
  3945. IPropertyTree &target = allTargets->query();
  3946. const char *targetName = target.queryProp("@name");
  3947. if (targetName && *targetName)
  3948. {
  3949. Owned<IPropertyTreeIterator> thorClusters = target.getElements("ThorCluster");
  3950. ForEach(*thorClusters)
  3951. {
  3952. const char *thorName = thorClusters->query().queryProp("@process");
  3953. VStringBuffer query("Software/ThorCluster[@name=\"%s\"]",thorName);
  3954. IPropertyTree *thorCluster = conn->queryRoot()->queryPropTree(query.str());
  3955. if (thorCluster)
  3956. {
  3957. const char *groupName = thorCluster->queryProp("@nodeGroup");
  3958. if (!groupName||!*groupName)
  3959. groupName = thorName;
  3960. thorNames.append(thorName);
  3961. groupNames.append(groupName);
  3962. targetNames.append(targetName);
  3963. StringBuffer queueName(targetName);
  3964. queueNames.append(queueName.append(THOR_QUEUE_EXT));
  3965. }
  3966. }
  3967. }
  3968. }
  3969. return thorNames.ordinality();
  3970. }
  3971. unsigned getEnvironmentHThorClusterNames(StringArray &eclAgentNames, StringArray &groupNames, StringArray &targetNames)
  3972. {
  3973. Owned<IRemoteConnection> conn = querySDS().connect("Environment", myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT);
  3974. if (!conn)
  3975. return 0;
  3976. Owned<IPropertyTreeIterator> allEclAgents = conn->queryRoot()->getElements("Software/EclAgentProcess");
  3977. ForEach(*allEclAgents)
  3978. {
  3979. IPropertyTree &eclAgent = allEclAgents->query();
  3980. const char *eclAgentName = eclAgent.queryProp("@name");
  3981. if (eclAgentName && *eclAgentName)
  3982. {
  3983. Owned<IPropertyTreeIterator> allTargets = conn->queryRoot()->getElements("Software/Topology/Cluster");
  3984. ForEach(*allTargets)
  3985. {
  3986. IPropertyTree &target = allTargets->query();
  3987. const char *targetName = target.queryProp("@name");
  3988. if (targetName && *targetName)
  3989. {
  3990. StringBuffer xpath;
  3991. xpath.appendf("EclAgentProcess[@process=\"%s\"]", eclAgentName);
  3992. if (target.hasProp(xpath) && !target.hasProp("ThorCluster"))
  3993. {
  3994. StringBuffer groupName("hthor__");
  3995. groupName.append(eclAgentName);
  3996. groupNames.append(groupName);
  3997. eclAgentNames.append(eclAgentName);
  3998. targetNames.append(targetName);
  3999. }
  4000. }
  4001. }
  4002. }
  4003. }
  4004. return eclAgentNames.ordinality();
  4005. }
  4006. IStringVal& CLocalWorkUnit::getScope(IStringVal &str) const
  4007. {
  4008. CriticalBlock block(crit);
  4009. if (p->hasProp("Debug/ForceScope"))
  4010. {
  4011. StringBuffer prefix(p->queryProp("Debug/ForceScope"));
  4012. str.set(prefix.toLowerCase().str());
  4013. }
  4014. else
  4015. {
  4016. Owned <IConstWUClusterInfo> ci = getTargetClusterInfo(p->queryProp("@clusterName"));
  4017. if (ci)
  4018. ci->getScope(str);
  4019. else
  4020. str.clear();
  4021. }
  4022. return str;
  4023. }
  4024. //Queries
  4025. void CLocalWorkUnit::setCodeVersion(unsigned codeVersion, const char * buildVersion, const char * eclVersion)
  4026. {
  4027. CriticalBlock block(crit);
  4028. p->setPropInt("@codeVersion", codeVersion);
  4029. p->setProp("@buildVersion", buildVersion);
  4030. p->setProp("@eclVersion", eclVersion);
  4031. }
  4032. unsigned CLocalWorkUnit::getCodeVersion() const
  4033. {
  4034. CriticalBlock block(crit);
  4035. return p->getPropInt("@codeVersion");
  4036. }
  4037. unsigned CLocalWorkUnit::getWuidVersion() const
  4038. {
  4039. CriticalBlock block(crit);
  4040. return p->getPropInt("@wuidVersion");
  4041. }
  4042. void CLocalWorkUnit::getBuildVersion(IStringVal & buildVersion, IStringVal & eclVersion) const
  4043. {
  4044. CriticalBlock block(crit);
  4045. buildVersion.set(p->queryProp("@buildVersion"));
  4046. eclVersion.set(p->queryProp("@eclVersion"));
  4047. }
  4048. void CLocalWorkUnit::setCloneable(bool value)
  4049. {
  4050. CriticalBlock block(crit);
  4051. p->setPropInt("@cloneable", value);
  4052. }
  4053. void CLocalWorkUnit::setIsClone(bool value)
  4054. {
  4055. CriticalBlock block(crit);
  4056. p->setPropInt("@isClone", value);
  4057. }
  4058. bool CLocalWorkUnit::getCloneable() const
  4059. {
  4060. CriticalBlock block(crit);
  4061. return p->getPropBool("@cloneable", false);
  4062. }
  4063. IUserDescriptor *CLocalWorkUnit::queryUserDescriptor() const
  4064. {
  4065. CriticalBlock block(crit);
  4066. if (!userDesc)
  4067. {
  4068. SCMStringBuffer token, user, password;
  4069. getSecurityToken(token);
  4070. SCMStringBuffer wuid;
  4071. getWuid(wuid);
  4072. extractToken(token.str(), wuid.str(), user, password);
  4073. userDesc.setown(createUserDescriptor());
  4074. userDesc->set(user.str(), password.str());
  4075. }
  4076. return userDesc;
  4077. }
  4078. void CLocalWorkUnit::setCombineQueries(unsigned combine)
  4079. {
  4080. CriticalBlock block(crit);
  4081. p->setPropInt("COMBINE_QUERIES", combine);
  4082. }
  4083. unsigned CLocalWorkUnit::getCombineQueries() const
  4084. {
  4085. CriticalBlock block(crit);
  4086. return p->getPropInt("COMBINE_QUERIES");
  4087. }
  4088. bool CLocalWorkUnit::isProtected() const
  4089. {
  4090. CriticalBlock block(crit);
  4091. return p->getPropBool("@protected", false);
  4092. }
  4093. bool CLocalWorkUnit::isPausing() const
  4094. {
  4095. CriticalBlock block(crit);
  4096. if (WUActionPause == getAction())
  4097. {
  4098. switch (getState())
  4099. {
  4100. case WUStateRunning:
  4101. case WUStateAborting:
  4102. return true;
  4103. }
  4104. }
  4105. return false;
  4106. }
  4107. void CLocalWorkUnit::protect(bool protectMode)
  4108. {
  4109. CriticalBlock block(crit);
  4110. p->setPropBool("@protected", protectMode);
  4111. }
  4112. bool CLocalWorkUnit::isBilled() const
  4113. {
  4114. CriticalBlock block(crit);
  4115. return p->getPropBool("@billed", false);
  4116. }
  4117. void CLocalWorkUnit::setBilled(bool value)
  4118. {
  4119. CriticalBlock block(crit);
  4120. p->setPropBool("@billed", value);
  4121. }
  4122. void CLocalWorkUnit::setResultLimit(unsigned value)
  4123. {
  4124. CriticalBlock block(crit);
  4125. p->setPropInt("resultLimit", value);
  4126. }
  4127. unsigned CLocalWorkUnit::getResultLimit() const
  4128. {
  4129. CriticalBlock block(crit);
  4130. return p->getPropInt("resultLimit");
  4131. }
  4132. void CLocalWorkUnit::setCompareMode(WUCompareMode value)
  4133. {
  4134. CriticalBlock block(crit);
  4135. p->setPropInt("comparemode", (int)value);
  4136. }
  4137. WUCompareMode CLocalWorkUnit::getCompareMode() const
  4138. {
  4139. CriticalBlock block(crit);
  4140. return (WUCompareMode) p->getPropInt("comparemode");
  4141. }
  4142. IStringVal & CLocalWorkUnit::getSnapshot(IStringVal & str) const
  4143. {
  4144. CriticalBlock block(crit);
  4145. str.set(p->queryProp("SNAPSHOT"));
  4146. return str;
  4147. }
  4148. void CLocalWorkUnit::setSnapshot(const char * val)
  4149. {
  4150. CriticalBlock block(crit);
  4151. p->setProp("SNAPSHOT", val);
  4152. }
  4153. static int comparePropTrees(IInterface **ll, IInterface **rr)
  4154. {
  4155. IPropertyTree *l = (IPropertyTree *) *ll;
  4156. IPropertyTree *r = (IPropertyTree *) *rr;
  4157. return stricmp(l->queryName(), r->queryName());
  4158. };
  4159. unsigned CLocalWorkUnit::calculateHash(unsigned crc)
  4160. {
  4161. // Any other values in the WU that could affect generated code should be crc'ed here
  4162. IPropertyTree *tree = p->queryBranch("Debug");
  4163. if (tree)
  4164. {
  4165. Owned<IPropertyTreeIterator> sub = tree->getElements("*");
  4166. ICopyArrayOf<IPropertyTree> subs;
  4167. for(sub->first(); sub->isValid(); sub->next())
  4168. subs.append(sub->query());
  4169. subs.sort(comparePropTrees);
  4170. ForEachItemIn(idx, subs)
  4171. {
  4172. const char *name = subs.item(idx).queryName();
  4173. const char *val = subs.item(idx).queryProp(NULL);
  4174. crc = crc32(name, (size32_t)strlen(name), crc);
  4175. if (val)
  4176. crc = crc32(val, (size32_t)strlen(val), crc);
  4177. }
  4178. }
  4179. Owned<IConstWUPluginIterator> plugins = &getPlugins();
  4180. for (plugins->first();plugins->isValid();plugins->next())
  4181. {
  4182. IConstWUPlugin &thisplugin = plugins->query();
  4183. SCMStringBuffer version;
  4184. thisplugin.getPluginVersion(version);
  4185. crc = crc32(version.str(), version.length(), crc);
  4186. }
  4187. return crc;
  4188. }
  4189. static void updateProp(IPropertyTree * to, const IPropertyTree * from, const char * xpath)
  4190. {
  4191. if (!to->hasProp(xpath) && from->hasProp(xpath))
  4192. to->setProp(xpath, from->queryProp(xpath));
  4193. }
  4194. static void copyTree(IPropertyTree * to, const IPropertyTree * from, const char * xpath)
  4195. {
  4196. IPropertyTree * match = from->getBranch(xpath);
  4197. if (match)
  4198. to->setPropTree(xpath, match);
  4199. }
  4200. void CLocalWorkUnit::copyWorkUnit(IConstWorkUnit *cached, bool all)
  4201. {
  4202. CLocalWorkUnit *from = QUERYINTERFACE(cached, CLocalWorkUnit);
  4203. if (!from)
  4204. {
  4205. CLockedWorkUnit *fl = QUERYINTERFACE(cached, CLockedWorkUnit);
  4206. if (!fl)
  4207. throw MakeStringException(WUERR_InternalUnknownImplementation, "Cached workunit not created using workunit dll");
  4208. from = fl->c;
  4209. }
  4210. // Need to copy the query, the results, and the graphs from the cached query.
  4211. // The cache is made before the query is executed so there is no need to clear them.
  4212. if (!cached->getCloneable())
  4213. throw MakeStringException(WUERR_CannotCloneWorkunit, "Source work unit not marked as clonable");
  4214. const IPropertyTree * fromP = from->p;
  4215. IPropertyTree *pt;
  4216. CriticalBlock block(crit);
  4217. query.clear();
  4218. updateProp(p, fromP, "@jobName");
  4219. copyTree(p, fromP, "Query");
  4220. pt = fromP->getBranch("Application/LibraryModule");
  4221. if (pt)
  4222. {
  4223. ensurePTree(p, "Application");
  4224. p->setPropTree("Application/LibraryModule", pt);
  4225. }
  4226. pt = fromP->queryBranch("Debug");
  4227. if (pt)
  4228. {
  4229. IPropertyTree *curDebug = p->queryPropTree("Debug");
  4230. if (curDebug)
  4231. {
  4232. Owned<IPropertyTreeIterator> elems = pt->getElements("*");
  4233. ForEach(*elems)
  4234. {
  4235. IPropertyTree *elem = &elems->query();
  4236. if (!curDebug->hasProp(elem->queryName()))
  4237. curDebug->setPropTree(elem->queryName(),LINK(elem));
  4238. }
  4239. }
  4240. else
  4241. p->setPropTree("Debug", LINK(pt));
  4242. }
  4243. copyTree(p, fromP, "Plugins");
  4244. copyTree(p, fromP, "Libraries");
  4245. copyTree(p, fromP, "Results");
  4246. copyTree(p, fromP, "Graphs");
  4247. copyTree(p, fromP, "Workflow");
  4248. if (all)
  4249. {
  4250. // Merge timing info from both branches
  4251. pt = fromP->getBranch("Timings");
  4252. if (pt)
  4253. {
  4254. IPropertyTree *tgtTimings = ensurePTree(p, "Timings");
  4255. mergePTree(tgtTimings, pt);
  4256. pt->Release();
  4257. }
  4258. }
  4259. updateProp(p, fromP, "@clusterName");
  4260. updateProp(p, fromP, "allowedclusters");
  4261. updateProp(p, fromP, "@submitID");
  4262. updateProp(p, fromP, "CustomerID");
  4263. updateProp(p, fromP, "SNAPSHOT");
  4264. //Variables may have been set up as parameters to the query - so need to preserve any values that were supplied.
  4265. pt = fromP->getBranch("Variables");
  4266. if (pt)
  4267. {
  4268. IPropertyTree *ptTgtVariables = ensurePTree(p, "Variables");
  4269. Owned<IPropertyTreeIterator> ptiVariable = pt->getElements("Variable");
  4270. for (ptiVariable->first(); ptiVariable->isValid(); ptiVariable->next())
  4271. {
  4272. IPropertyTree *ptSrcVariable = &ptiVariable->query();
  4273. const char *name = ptSrcVariable->queryProp("@name");
  4274. assertex(name);
  4275. StringBuffer xpath;
  4276. xpath.append("Variable[@name='").append(name).append("']");
  4277. IPropertyTree *ptTgtVariable = ptTgtVariables->queryPropTree(xpath.str());
  4278. IPropertyTree *merged = createPTreeFromIPT(ptSrcVariable); // clone entire source info...
  4279. merged->removeProp("Value"); // except value and status
  4280. merged->setProp("@status", "undefined");
  4281. if (!merged->getPropBool("@isScalar"))
  4282. merged->removeProp("totalRowCount");
  4283. merged->removeProp("rowCount");
  4284. // If there are any other fields that get set ONLY by eclagent, strip them out here...
  4285. if (ptTgtVariable)
  4286. {
  4287. // copy status and Value from what is already set in target
  4288. merged->setProp("@status", ptTgtVariable->queryProp("@status"));
  4289. MemoryBuffer value;
  4290. if (ptTgtVariable->getPropBin("Value", value))
  4291. merged->setPropBin("Value", value.length(), value.toByteArray());
  4292. ptTgtVariable->removeProp(xpath.str());
  4293. // If there are any other fields in a variable that get set by ws_ecl before submitting, copy them across here...
  4294. }
  4295. ptTgtVariables->addPropTree("Variable", merged);
  4296. }
  4297. pt->Release();
  4298. }
  4299. p->setProp("@codeVersion", fromP->queryProp("@codeVersion"));
  4300. p->setPropBool("@cloneable", true);
  4301. p->setPropBool("@isClone", true);
  4302. resetWorkflow(); // the source Workflow section may have had some parts already executed...
  4303. // resetResults(); // probably should be resetting the results as well... rather than waiting for the rerun to overwrite them
  4304. }
  4305. bool CLocalWorkUnit::hasDebugValue(const char *propname) const
  4306. {
  4307. StringBuffer lower;
  4308. lower.append(propname).toLowerCase();
  4309. CriticalBlock block(crit);
  4310. StringBuffer prop("Debug/");
  4311. return p->hasProp(prop.append(lower));
  4312. }
  4313. IStringVal& CLocalWorkUnit::getDebugValue(const char *propname, IStringVal &str) const
  4314. {
  4315. StringBuffer lower;
  4316. lower.append(propname).toLowerCase();
  4317. CriticalBlock block(crit);
  4318. StringBuffer prop("Debug/");
  4319. str.set(p->queryProp(prop.append(lower).str()));
  4320. return str;
  4321. }
  4322. IStringIterator& CLocalWorkUnit::getDebugValues() const
  4323. {
  4324. return getDebugValues(NULL);
  4325. }
  4326. IStringIterator& CLocalWorkUnit::getDebugValues(const char *prop) const
  4327. {
  4328. CriticalBlock block(crit);
  4329. StringBuffer path("Debug/");
  4330. if (prop)
  4331. {
  4332. StringBuffer lower;
  4333. lower.append(prop).toLowerCase();
  4334. path.append(lower);
  4335. }
  4336. else
  4337. path.append("*");
  4338. return *new CStringPTreeTagIterator(p->getElements(path.str()));
  4339. }
  4340. int CLocalWorkUnit::getDebugValueInt(const char *propname, int defVal) const
  4341. {
  4342. StringBuffer lower;
  4343. lower.append(propname).toLowerCase();
  4344. CriticalBlock block(crit);
  4345. StringBuffer prop("Debug/");
  4346. prop.append(lower);
  4347. return p->getPropInt(prop.str(), defVal);
  4348. }
  4349. __int64 CLocalWorkUnit::getDebugValueInt64(const char *propname, __int64 defVal) const
  4350. {
  4351. StringBuffer lower;
  4352. lower.append(propname).toLowerCase();
  4353. CriticalBlock block(crit);
  4354. StringBuffer prop("Debug/");
  4355. prop.append(lower);
  4356. return p->getPropInt64(prop.str(), defVal);
  4357. }
  4358. bool CLocalWorkUnit::getDebugValueBool(const char * propname, bool defVal) const
  4359. {
  4360. StringBuffer lower;
  4361. lower.append(propname).toLowerCase();
  4362. CriticalBlock block(crit);
  4363. StringBuffer prop("Debug/");
  4364. prop.append(lower);
  4365. return p->getPropBool(prop.str(), defVal);
  4366. }
  4367. IStringIterator *CLocalWorkUnit::getLogs(const char *type, const char *instance) const
  4368. {
  4369. VStringBuffer xpath("Process/%s/", type);
  4370. if (instance)
  4371. xpath.append(instance);
  4372. else
  4373. xpath.append("*");
  4374. CriticalBlock block(crit);
  4375. if (p->getPropInt("@wuidVersion") < 1) // legacy wuid
  4376. {
  4377. // NB: instance unused
  4378. if (streq("EclAgent", type))
  4379. return new CStringPTreeIterator(p->getElements("Debug/eclagentlog"));
  4380. else if (streq("Thor", type))
  4381. return new CStringPTreeIterator(p->getElements("Debug/thorlog*"));
  4382. VStringBuffer xpath("Debug/%s", type);
  4383. return new CStringPTreeIterator(p->getElements(xpath.str()));
  4384. }
  4385. else
  4386. return new CStringPTreeAttrIterator(p->getElements(xpath.str()), "@log");
  4387. }
  4388. IStringIterator *CLocalWorkUnit::getProcesses(const char *type) const
  4389. {
  4390. VStringBuffer xpath("Process/%s/*", type);
  4391. CriticalBlock block(crit);
  4392. return new CStringPTreeTagIterator(p->getElements(xpath.str()));
  4393. }
  4394. void CLocalWorkUnit::addProcess(const char *type, const char *instance, const char *log)
  4395. {
  4396. VStringBuffer processType("Process/%s", type);
  4397. VStringBuffer xpath("%s/%s", processType.str(), instance);
  4398. if (log)
  4399. xpath.appendf("[@log=\"%s\"]", log);
  4400. CriticalBlock block(crit);
  4401. if (!p->hasProp(xpath))
  4402. {
  4403. IPropertyTree *node = ensurePTree(p, processType.str());
  4404. node = node->addPropTree(instance, createPTree());
  4405. node->setProp("@log", log);
  4406. }
  4407. }
  4408. void CLocalWorkUnit::setDebugValue(const char *propname, const char *value, bool overwrite)
  4409. {
  4410. StringBuffer lower;
  4411. lower.append(propname).toLowerCase();
  4412. CriticalBlock block(crit);
  4413. StringBuffer prop("Debug/");
  4414. prop.append(lower);
  4415. if (overwrite || !p->hasProp(prop.str()))
  4416. {
  4417. // MORE - not sure this line should be needed....
  4418. p->setProp("Debug", "");
  4419. p->setProp(prop.str(), value);
  4420. }
  4421. }
  4422. void CLocalWorkUnit::setDebugValueInt(const char *propname, int value, bool overwrite)
  4423. {
  4424. StringBuffer lower;
  4425. lower.append(propname).toLowerCase();
  4426. CriticalBlock block(crit);
  4427. StringBuffer prop("Debug/");
  4428. prop.append(lower);
  4429. if (overwrite || !p->hasProp(prop.str()))
  4430. {
  4431. // MORE - not sure this line should be needed....
  4432. p->setProp("Debug", "");
  4433. p->setPropInt(prop.str(), value);
  4434. }
  4435. }
  4436. void CLocalWorkUnit::setTracingValue(const char *propname, const char *value)
  4437. {
  4438. CriticalBlock block(crit);
  4439. // MORE - not sure this line should be needed....
  4440. p->setProp("Tracing", "");
  4441. StringBuffer prop("Tracing/");
  4442. p->setProp(prop.append(propname).str(), value);
  4443. }
  4444. void CLocalWorkUnit::setTracingValueInt(const char *propname, int value)
  4445. {
  4446. CriticalBlock block(crit);
  4447. StringBuffer prop("Tracing/");
  4448. p->setPropInt(prop.append(propname).str(), value);
  4449. }
  4450. IConstWUQuery* CLocalWorkUnit::getQuery() const
  4451. {
  4452. // For this to be legally called, we must have the read-able interface. So we are already locked for (at least) read.
  4453. CriticalBlock block(crit);
  4454. if (!query)
  4455. {
  4456. IPropertyTree *s = p->getPropTree("Query");
  4457. if (s)
  4458. query.setown(new CLocalWUQuery(s));
  4459. }
  4460. return query.getLink();
  4461. }
  4462. IWUQuery* CLocalWorkUnit::updateQuery()
  4463. {
  4464. // For this to be legally called, we must have the write-able interface. So we are already locked for write.
  4465. CriticalBlock block(crit);
  4466. if (!query)
  4467. {
  4468. IPropertyTree *s = p->queryPropTree("Query");
  4469. if (!s)
  4470. s = p->addPropTree("Query", createPTreeFromXMLString("<Query fetchEntire='1'/>"));
  4471. s->Link();
  4472. query.setown(new CLocalWUQuery(s));
  4473. }
  4474. return query.getLink();
  4475. }
  4476. void CLocalWorkUnit::loadPlugins() const
  4477. {
  4478. CriticalBlock block(crit);
  4479. if (!pluginsCached)
  4480. {
  4481. assertex(plugins.length() == 0);
  4482. Owned<IPropertyTreeIterator> r = p->getElements("Plugins/Plugin");
  4483. for (r->first(); r->isValid(); r->next())
  4484. {
  4485. IPropertyTree *rp = &r->query();
  4486. rp->Link();
  4487. plugins.append(*new CLocalWUPlugin(rp));
  4488. }
  4489. pluginsCached = true;
  4490. }
  4491. }
  4492. IConstWUPluginIterator& CLocalWorkUnit::getPlugins() const
  4493. {
  4494. CriticalBlock block(crit);
  4495. loadPlugins();
  4496. return *new CArrayIteratorOf<IConstWUPlugin,IConstWUPluginIterator> (plugins, 0, (IConstWorkUnit *) this);
  4497. }
  4498. void CLocalWorkUnit::loadLibraries() const
  4499. {
  4500. CriticalBlock block(crit);
  4501. if (!librariesCached)
  4502. {
  4503. assertex(libraries.length() == 0);
  4504. Owned<IPropertyTreeIterator> r = p->getElements("Libraries/Library");
  4505. ForEach(*r)
  4506. {
  4507. IPropertyTree *rp = &r->query();
  4508. rp->Link();
  4509. libraries.append(*new CLocalWULibrary(rp));
  4510. }
  4511. librariesCached = true;
  4512. }
  4513. }
  4514. IConstWULibraryIterator& CLocalWorkUnit::getLibraries() const
  4515. {
  4516. CriticalBlock block(crit);
  4517. loadLibraries();
  4518. return *new CArrayIteratorOf<IConstWULibrary,IConstWULibraryIterator> (libraries, 0, (IConstWorkUnit *) this);
  4519. }
  4520. IConstWULibrary * CLocalWorkUnit::getLibraryByName(const char * search) const
  4521. {
  4522. CriticalBlock block(crit);
  4523. loadLibraries();
  4524. ForEachItemIn(idx, libraries)
  4525. {
  4526. SCMStringBuffer name;
  4527. IConstWULibrary &cur = libraries.item(idx);
  4528. cur.getName(name);
  4529. if (stricmp(name.str(), search)==0)
  4530. return &OLINK(cur);
  4531. }
  4532. return NULL;
  4533. }
  4534. unsigned CLocalWorkUnit::getTimerDuration(const char *name, const char *_unused) const
  4535. {
  4536. CriticalBlock block(crit);
  4537. StringBuffer pname;
  4538. pname.appendf("Timings/Timing[@name=\"%s\"]/@duration", name);
  4539. return p->getPropInt(pname.str(), 0);
  4540. }
  4541. unsigned CLocalWorkUnit::getTimerCount(const char *name, const char *_unused) const
  4542. {
  4543. CriticalBlock block(crit);
  4544. StringBuffer pname;
  4545. pname.appendf("Timings/Timing[@name=\"%s\"]/@count", name);
  4546. return p->getPropInt(pname.str(), 0);
  4547. }
  4548. IStringIterator& CLocalWorkUnit::getTimers() const
  4549. {
  4550. CriticalBlock block(crit);
  4551. return *new CStringPTreeAttrIterator(p->getElements("Timings/Timing"), "@name");
  4552. }
  4553. StringBuffer &formatGraphTimerLabel(StringBuffer &str, const char *graphName, unsigned subGraphNum, unsigned __int64 subId)
  4554. {
  4555. str.append("Graph ").append(graphName);
  4556. if (subGraphNum) str.append(" - ").append(subGraphNum).append(" (").append(subId).append(")");
  4557. else if (subId) str.append(" - id(").append(subId).append(")");
  4558. return str;
  4559. }
  4560. bool parseGraphTimerLabel(const char *label, StringBuffer &graphName, unsigned &subGraphNum, unsigned __int64 &subId)
  4561. {
  4562. // expects format: "Graph <graphname>[ - <subgraphnum> (<subgraphid>)]"
  4563. unsigned len = (size32_t)strlen(label);
  4564. if (len < 6 || (0 != memcmp(label, "Graph ", 6)))
  4565. return false;
  4566. subGraphNum = 0;
  4567. subId = 0;
  4568. const char *finger = label+6;
  4569. const char *finger2 = strchr(finger, '-');
  4570. if (NULL == finger2) // just graphName
  4571. graphName.append(finger);
  4572. else
  4573. {
  4574. graphName.append((size32_t)((finger2-1)-finger), finger);
  4575. finger = finger2+2; // skip '-' and space
  4576. finger2 = strchr(finger, ' ');
  4577. if (finger2)
  4578. {
  4579. subGraphNum = atoi_l(finger, (size32_t)(finger2-finger));
  4580. finger = finger2+2; // skip space and '('
  4581. finger2 = strchr(finger, ')');
  4582. if (finger2)
  4583. subId = atoi64_l(finger, (size32_t)(finger2-finger));
  4584. }
  4585. else if (((len-(finger-label))>3) && 0 == memcmp(finger, "id(", 3)) // subgraph id only, new format.
  4586. {
  4587. finger += 3;
  4588. finger2 = strchr(finger, ')');
  4589. if (finger2)
  4590. subId = atoi64_l(finger, (size32_t)(finger2-finger));
  4591. }
  4592. }
  4593. return true;
  4594. }
  4595. void CLocalWorkUnit::setTimerInfo(const char *name, const char *subname, unsigned ms, unsigned count, unsigned __int64 max)
  4596. {
  4597. CriticalBlock block(crit);
  4598. IPropertyTree *timings = p->queryPropTree("Timings");
  4599. if (!timings)
  4600. timings = p->addPropTree("Timings", createPTree("Timings"));
  4601. StringBuffer fullname;
  4602. if (subname)
  4603. fullname.append('.').append(subname);
  4604. fullname.append(name);
  4605. StringBuffer xpath;
  4606. xpath.append("Timing[@name=\"").append(fullname).append("\"]");
  4607. IPropertyTree *timing = timings->queryPropTree(xpath.str());
  4608. if (!timing)
  4609. {
  4610. timing = timings->addPropTree("Timing", createPTree("Timing"));
  4611. timing->setProp("@name", fullname.str());
  4612. }
  4613. timing->setPropInt("@count", count);
  4614. timing->setPropInt("@duration", ms);
  4615. if (!max && 1==count) max = ms * 1000000; // max is in nanoseconds
  4616. if (max)
  4617. timing->setPropInt64("@max", max);
  4618. }
  4619. void CLocalWorkUnit::setTimeStamp(const char *application, const char *instance, const char *event, bool add)
  4620. {
  4621. CriticalBlock block(crit);
  4622. char timeStamp[64];
  4623. time_t tNow;
  4624. time(&tNow);
  4625. #ifdef _WIN32
  4626. struct tm *gmtNow;
  4627. gmtNow = gmtime(&tNow);
  4628. strftime(timeStamp, 64, "%Y-%m-%dT%H:%M:%SZ", gmtNow);
  4629. #else
  4630. struct tm gmtNow;
  4631. gmtime_r(&tNow, &gmtNow);
  4632. strftime(timeStamp, 64, "%Y-%m-%dT%H:%M:%SZ", &gmtNow);
  4633. #endif //_WIN32
  4634. IPropertyTree *ts = p->queryPropTree("TimeStamps");
  4635. if (!ts) {
  4636. ts = p->addPropTree("TimeStamps", createPTree("TimeStamps"));
  4637. add = true;
  4638. }
  4639. IPropertyTree *t=NULL;
  4640. if (!add) {
  4641. StringBuffer path;
  4642. path.appendf("TimeStamp[@application=\"%s\"]",application);
  4643. t = ts->queryBranch(path.str());
  4644. }
  4645. if (!t) {
  4646. t = createPTree("TimeStamp");
  4647. t->setProp("@application", application);
  4648. add = true;
  4649. }
  4650. if (instance)
  4651. t->setProp("@instance", instance);
  4652. t->setProp(event, timeStamp);
  4653. IPropertyTree *et = t->queryPropTree(event);
  4654. if(et)
  4655. et->setPropInt("@ts",(int)tNow);
  4656. if (add)
  4657. ts->addPropTree("TimeStamp", t);
  4658. }
  4659. void CLocalWorkUnit::setTimeStamp(const char *application, const char *instance, const char *event)
  4660. {
  4661. setTimeStamp(application,instance,event,false);
  4662. }
  4663. void CLocalWorkUnit::addTimeStamp(const char *application, const char *instance, const char *event)
  4664. {
  4665. setTimeStamp(application,instance,event,true);
  4666. }
  4667. IStringVal &CLocalWorkUnit::getTimeStamp(const char *name, const char *application, IStringVal &str) const
  4668. {
  4669. CriticalBlock block(crit);
  4670. str.clear();
  4671. StringBuffer pname("TimeStamps/TimeStamp");
  4672. if (application)
  4673. pname.appendf("[@application=\"%s\"]", application);
  4674. pname.appendf("/%s", name);
  4675. Owned<IPropertyTreeIterator> stamps = p->getElements(pname.str());
  4676. if (stamps && stamps->first())
  4677. str.set(stamps->query().queryProp(NULL));
  4678. return str;
  4679. }
  4680. IConstWUTimeStampIterator& CLocalWorkUnit::getTimeStamps() const
  4681. {
  4682. CriticalBlock block(crit);
  4683. timestamps.load(p,"TimeStamps/*");
  4684. return *new CArrayIteratorOf<IConstWUTimeStamp,IConstWUTimeStampIterator> (timestamps, 0, (IConstWorkUnit *) this);
  4685. }
  4686. bool CLocalWorkUnit::getWuDate(unsigned & year, unsigned & month, unsigned& day)
  4687. {
  4688. CriticalBlock block(crit);
  4689. SCMStringBuffer wuidstr;
  4690. const char *wuid = getWuid(wuidstr).str();
  4691. if (sscanf(wuid, "W%4u%2u%2u", &year, &month, &day)==3)
  4692. {
  4693. }
  4694. return false;
  4695. }
  4696. IWUPlugin* CLocalWorkUnit::updatePluginByName(const char *qname)
  4697. {
  4698. CriticalBlock block(crit);
  4699. IConstWUPlugin *existing = getPluginByName(qname);
  4700. if (existing)
  4701. return (IWUPlugin *) existing;
  4702. if (!plugins.length())
  4703. p->addPropTree("Plugins", createPTree("Plugins"));
  4704. IPropertyTree *pl = p->queryPropTree("Plugins");
  4705. IPropertyTree *s = pl->addPropTree("Plugin", createPTree("Plugin"));
  4706. s->Link();
  4707. IWUPlugin* q = new CLocalWUPlugin(s);
  4708. q->Link();
  4709. plugins.append(*q);
  4710. q->setPluginName(qname);
  4711. return q;
  4712. }
  4713. IConstWUPlugin* CLocalWorkUnit::getPluginByName(const char *qname) const
  4714. {
  4715. CriticalBlock block(crit);
  4716. loadPlugins();
  4717. ForEachItemIn(idx, plugins)
  4718. {
  4719. SCMStringBuffer name;
  4720. IConstWUPlugin &cur = plugins.item(idx);
  4721. cur.getPluginName(name);
  4722. if (stricmp(name.str(), qname)==0)
  4723. {
  4724. cur.Link();
  4725. return &cur;
  4726. }
  4727. }
  4728. return NULL;
  4729. }
  4730. IWULibrary* CLocalWorkUnit::updateLibraryByName(const char *qname)
  4731. {
  4732. CriticalBlock block(crit);
  4733. IConstWULibrary *existing = getLibraryByName(qname);
  4734. if (existing)
  4735. return (IWULibrary *) existing;
  4736. if (!libraries.length())
  4737. p->addPropTree("Libraries", createPTree("Libraries"));
  4738. IPropertyTree *pl = p->queryPropTree("Libraries");
  4739. IPropertyTree *s = pl->addPropTree("Library", createPTree("Library"));
  4740. s->Link();
  4741. IWULibrary* q = new CLocalWULibrary(s);
  4742. q->Link();
  4743. libraries.append(*q);
  4744. q->setName(qname);
  4745. return q;
  4746. }
  4747. void CLocalWorkUnit::loadExceptions() const
  4748. {
  4749. CriticalBlock block(crit);
  4750. if (!exceptionsCached)
  4751. {
  4752. assertex(exceptions.length() == 0);
  4753. Owned<IPropertyTreeIterator> r = p->getElements("Exceptions/Exception");
  4754. for (r->first(); r->isValid(); r->next())
  4755. {
  4756. IPropertyTree *rp = &r->query();
  4757. rp->Link();
  4758. exceptions.append(*new CLocalWUException(rp));
  4759. }
  4760. exceptionsCached = true;
  4761. }
  4762. }
  4763. IConstWUExceptionIterator& CLocalWorkUnit::getExceptions() const
  4764. {
  4765. CriticalBlock block(crit);
  4766. loadExceptions();
  4767. return *new CArrayIteratorOf<IConstWUException,IConstWUExceptionIterator> (exceptions, 0, (IConstWorkUnit *) this);
  4768. }
  4769. unsigned CLocalWorkUnit::getExceptionCount() const
  4770. {
  4771. CriticalBlock block(crit);
  4772. loadExceptions();
  4773. return exceptions.length();
  4774. }
  4775. void CLocalWorkUnit::clearExceptions()
  4776. {
  4777. CriticalBlock block(crit);
  4778. // For this to be legally called, we must have the write-able interface. So we are already locked for write.
  4779. exceptions.kill();
  4780. exceptionsCached = true;
  4781. p->removeProp("Exceptions");
  4782. }
  4783. IWUException* CLocalWorkUnit::createException()
  4784. {
  4785. CriticalBlock block(crit);
  4786. // For this to be legally called, we must have the write-able interface. So we are already locked for write.
  4787. loadExceptions();
  4788. if (!exceptions.length())
  4789. p->addPropTree("Exceptions", createPTree("Exceptions"));
  4790. IPropertyTree *r = p->queryPropTree("Exceptions");
  4791. IPropertyTree *s = r->addPropTree("Exception", createPTree("Exception"));
  4792. IWUException* q = new CLocalWUException(LINK(s));
  4793. exceptions.append(*LINK(q));
  4794. Owned<IJlibDateTime> now = createDateTimeNow();
  4795. SCMStringBuffer temp;
  4796. now->getString(temp);
  4797. q->setTimeStamp(temp.str());
  4798. return q;
  4799. }
  4800. IConstWUWebServicesInfo* CLocalWorkUnit::getWebServicesInfo() const
  4801. {
  4802. // For this to be legally called, we must have the read-able interface. So we are already locked for (at least) read.
  4803. CriticalBlock block(crit);
  4804. if (!webServicesInfoCached)
  4805. {
  4806. assertex(!webServicesInfo);
  4807. IPropertyTree *s = p->getPropTree("WebServicesInfo");
  4808. if (s)
  4809. webServicesInfo.setown(new CLocalWUWebServicesInfo(s));
  4810. webServicesInfoCached = true;
  4811. }
  4812. return webServicesInfo.getLink();
  4813. }
  4814. IWUWebServicesInfo* CLocalWorkUnit::updateWebServicesInfo(bool create)
  4815. {
  4816. // For this to be legally called, we must have the write-able interface. So we are already locked for write.
  4817. CriticalBlock block(crit);
  4818. if (!webServicesInfoCached)
  4819. {
  4820. IPropertyTree *s = p->queryPropTree("WebServicesInfo");
  4821. if (!s)
  4822. {
  4823. if (create)
  4824. s = p->addPropTree("WebServicesInfo", createPTreeFromXMLString("<WebServicesInfo />"));
  4825. else
  4826. return NULL;
  4827. }
  4828. s->Link();
  4829. webServicesInfo.setown(new CLocalWUWebServicesInfo(s));
  4830. webServicesInfoCached = true;
  4831. }
  4832. return webServicesInfo.getLink();
  4833. }
  4834. IConstWURoxieQueryInfo* CLocalWorkUnit::getRoxieQueryInfo() const
  4835. {
  4836. // For this to be legally called, we must have the read-able interface. So we are already locked for (at least) read.
  4837. CriticalBlock block(crit);
  4838. if (!roxieQueryInfoCached)
  4839. {
  4840. assertex(!roxieQueryInfo);
  4841. IPropertyTree *s = p->getPropTree("RoxieQueryInfo");
  4842. if (s)
  4843. roxieQueryInfo.setown(new CLocalWURoxieQueryInfo(s));
  4844. roxieQueryInfoCached = true;
  4845. }
  4846. return roxieQueryInfo.getLink();
  4847. }
  4848. IWURoxieQueryInfo* CLocalWorkUnit::updateRoxieQueryInfo(const char *wuid, const char *roxieClusterName)
  4849. {
  4850. // For this to be legally called, we must have the write-able interface. So we are already locked for write.
  4851. CriticalBlock block(crit);
  4852. if (!roxieQueryInfoCached)
  4853. {
  4854. IPropertyTree *s = p->queryPropTree("RoxieQueryInfo");
  4855. if (!s)
  4856. s = p->addPropTree("RoxieQueryInfo", createPTreeFromXMLString("<RoxieQueryInfo />"));
  4857. if (wuid && *wuid)
  4858. s->addProp("@wuid", wuid);
  4859. if (roxieClusterName && *roxieClusterName)
  4860. s->addProp("@roxieClusterName", roxieClusterName);
  4861. s->Link();
  4862. roxieQueryInfo.setown(new CLocalWURoxieQueryInfo(s));
  4863. roxieQueryInfoCached = true;
  4864. }
  4865. return roxieQueryInfo.getLink();
  4866. }
  4867. static int compareResults(IInterface **ll, IInterface **rr)
  4868. {
  4869. CLocalWUResult *l = (CLocalWUResult *) *ll;
  4870. CLocalWUResult *r = (CLocalWUResult *) *rr;
  4871. return l->getResultSequence() - r->getResultSequence();
  4872. }
  4873. void CLocalWorkUnit::loadResults() const
  4874. {
  4875. CriticalBlock block(crit);
  4876. if (!resultsCached)
  4877. {
  4878. assertex(results.length() == 0);
  4879. Owned<IPropertyTreeIterator> r = p->getElements("Results/Result");
  4880. for (r->first(); r->isValid(); r->next())
  4881. {
  4882. IPropertyTree *rp = &r->query();
  4883. rp->Link();
  4884. results.append(*new CLocalWUResult(rp));
  4885. }
  4886. results.sort(compareResults);
  4887. resultsCached = true;
  4888. }
  4889. }
  4890. void CLocalWorkUnit::loadVariables() const
  4891. {
  4892. CriticalBlock block(crit);
  4893. if (!variablesCached)
  4894. {
  4895. assertex(variables.length() == 0);
  4896. Owned<IPropertyTreeIterator> r = p->getElements("Variables/Variable");
  4897. for (r->first(); r->isValid(); r->next())
  4898. {
  4899. IPropertyTree *rp = &r->query();
  4900. rp->Link();
  4901. variables.append(*new CLocalWUResult(rp));
  4902. }
  4903. variablesCached = true;
  4904. }
  4905. }
  4906. void CLocalWorkUnit::loadTemporaries() const
  4907. {
  4908. CriticalBlock block(crit);
  4909. if (!temporariesCached)
  4910. {
  4911. assertex(temporaries.length() == 0);
  4912. Owned<IPropertyTreeIterator> r = p->getElements("Temporaries/Variable");
  4913. for (r->first(); r->isValid(); r->next())
  4914. {
  4915. IPropertyTree *rp = &r->query();
  4916. rp->Link();
  4917. temporaries.append(*new CLocalWUResult(rp));
  4918. }
  4919. temporariesCached = true;
  4920. }
  4921. }
  4922. IWUResult* CLocalWorkUnit::createResult()
  4923. {
  4924. CriticalBlock block(crit);
  4925. // For this to be legally called, we must have the write-able interface. So we are already locked for write.
  4926. loadResults();
  4927. if (!results.length())
  4928. p->addPropTree("Results", createPTree("Results"));
  4929. IPropertyTree *r = p->queryPropTree("Results");
  4930. IPropertyTree *s = r->addPropTree("Result", createPTreeFromXMLString("<Result fetchEntire='1'/>"));
  4931. s->Link();
  4932. IWUResult* q = new CLocalWUResult(s);
  4933. q->Link();
  4934. results.append(*q);
  4935. return q;
  4936. }
  4937. IWUResult* CLocalWorkUnit::updateResultByName(const char *qname)
  4938. {
  4939. CriticalBlock block(crit);
  4940. IConstWUResult *existing = getResultByName(qname);
  4941. if (existing)
  4942. return (IWUResult *) existing;
  4943. IWUResult* q = createResult();
  4944. q->setResultName(qname);
  4945. return q;
  4946. }
  4947. IWUResult* CLocalWorkUnit::updateResultBySequence(unsigned seq)
  4948. {
  4949. CriticalBlock block(crit);
  4950. IConstWUResult *existing = getResultBySequence(seq);
  4951. if (existing)
  4952. return (IWUResult *) existing;
  4953. IWUResult* q = createResult();
  4954. q->setResultSequence(seq);
  4955. return q;
  4956. }
  4957. IConstWUResultIterator& CLocalWorkUnit::getResults() const
  4958. {
  4959. CriticalBlock block(crit);
  4960. loadResults();
  4961. return *new CArrayIteratorOf<IConstWUResult,IConstWUResultIterator> (results, 0, (IConstWorkUnit *) this);
  4962. }
  4963. IConstWUResult* CLocalWorkUnit::getResultByName(const char *qname) const
  4964. {
  4965. CriticalBlock block(crit);
  4966. loadResults();
  4967. ForEachItemIn(idx, results)
  4968. {
  4969. SCMStringBuffer name;
  4970. IConstWUResult &cur = results.item(idx);
  4971. cur.getResultName(name);
  4972. if (stricmp(name.str(), qname)==0)
  4973. {
  4974. cur.Link();
  4975. return &cur;
  4976. }
  4977. }
  4978. return NULL;
  4979. }
  4980. IConstWUResult* CLocalWorkUnit::getResultBySequence(unsigned seq) const
  4981. {
  4982. CriticalBlock block(crit);
  4983. loadResults();
  4984. ForEachItemIn(idx, results)
  4985. {
  4986. IConstWUResult &cur = results.item(idx);
  4987. if (cur.getResultSequence() == seq)
  4988. {
  4989. cur.Link();
  4990. return &cur;
  4991. }
  4992. }
  4993. return NULL;
  4994. }
  4995. IConstWUResultIterator& CLocalWorkUnit::getVariables() const
  4996. {
  4997. CriticalBlock block(crit);
  4998. loadVariables();
  4999. return *new CArrayIteratorOf<IConstWUResult,IConstWUResultIterator> (variables, 0, (IConstWorkUnit *) this);
  5000. }
  5001. IConstWUResult* CLocalWorkUnit::getGlobalByName(const char *qname) const
  5002. {
  5003. CriticalBlock block(crit);
  5004. if (strcmp(p->queryName(), GLOBAL_WORKUNIT)==0)
  5005. return getVariableByName(qname);
  5006. Owned <IWorkUnit> global = factory->ensureNamedWorkUnit(GLOBAL_WORKUNIT);
  5007. return global->getVariableByName(qname);
  5008. }
  5009. IWUResult* CLocalWorkUnit::updateGlobalByName(const char *qname)
  5010. {
  5011. CriticalBlock block(crit);
  5012. if (strcmp(p->queryName(), GLOBAL_WORKUNIT)==0)
  5013. return updateVariableByName(qname);
  5014. Owned <IWorkUnit> global = factory->ensureNamedWorkUnit(GLOBAL_WORKUNIT);
  5015. return global->updateVariableByName(qname);
  5016. }
  5017. IConstWUResult* CLocalWorkUnit::getVariableByName(const char *qname) const
  5018. {
  5019. CriticalBlock block(crit);
  5020. loadVariables();
  5021. ForEachItemIn(idx, variables)
  5022. {
  5023. SCMStringBuffer name;
  5024. IConstWUResult &cur = variables.item(idx);
  5025. cur.getResultName(name);
  5026. if (stricmp(name.str(), qname)==0)
  5027. {
  5028. cur.Link();
  5029. return &cur;
  5030. }
  5031. }
  5032. return NULL;
  5033. }
  5034. IConstWUResult* CLocalWorkUnit::getTemporaryByName(const char *qname) const
  5035. {
  5036. CriticalBlock block(crit);
  5037. loadTemporaries();
  5038. ForEachItemIn(idx, temporaries)
  5039. {
  5040. SCMStringBuffer name;
  5041. IConstWUResult &cur = temporaries.item(idx);
  5042. cur.getResultName(name);
  5043. if (stricmp(name.str(), qname)==0)
  5044. {
  5045. cur.Link();
  5046. return &cur;
  5047. }
  5048. }
  5049. return NULL;
  5050. }
  5051. IConstWUResultIterator& CLocalWorkUnit::getTemporaries() const
  5052. {
  5053. CriticalBlock block(crit);
  5054. loadTemporaries();
  5055. return *new CArrayIteratorOf<IConstWUResult,IConstWUResultIterator> (temporaries, 0, (IConstWorkUnit *) this);
  5056. }
  5057. IWUResult* CLocalWorkUnit::updateTemporaryByName(const char *qname)
  5058. {
  5059. CriticalBlock block(crit);
  5060. IConstWUResult *existing = getTemporaryByName(qname);
  5061. if (existing)
  5062. return (IWUResult *) existing;
  5063. if (!temporaries.length())
  5064. p->addPropTree("Temporaries", createPTree("Temporaries"));
  5065. IPropertyTree *vars = p->queryPropTree("Temporaries");
  5066. IPropertyTree *s = vars->addPropTree("Variable", createPTree("Variable"));
  5067. s->Link();
  5068. IWUResult* q = new CLocalWUResult(s);
  5069. q->Link();
  5070. temporaries.append(*q);
  5071. q->setResultName(qname);
  5072. return q;
  5073. }
  5074. IWUResult* CLocalWorkUnit::updateVariableByName(const char *qname)
  5075. {
  5076. CriticalBlock block(crit);
  5077. IConstWUResult *existing = getVariableByName(qname);
  5078. if (existing)
  5079. return (IWUResult *) existing;
  5080. if (!variables.length())
  5081. p->addPropTree("Variables", createPTree("Variables"));
  5082. IPropertyTree *vars = p->queryPropTree("Variables");
  5083. IPropertyTree *s = vars->addPropTree("Variable", createPTree("Variable"));
  5084. s->Link();
  5085. IWUResult* q = new CLocalWUResult(s);
  5086. q->Link();
  5087. variables.append(*q);
  5088. q->setResultName(qname);
  5089. return q;
  5090. }
  5091. void CLocalWorkUnit::deleteTempFiles(const char *graph, bool deleteOwned, bool deleteJobOwned)
  5092. {
  5093. CriticalBlock block(crit);
  5094. IPropertyTree *files = p->queryPropTree("Files");
  5095. if (!files) return;
  5096. Owned<IPropertyTreeIterator> iter = files->getElements("File");
  5097. ICopyArrayOf<IPropertyTree> toRemove;
  5098. ForEach (*iter)
  5099. {
  5100. IPropertyTree &file = iter->query();
  5101. WUFileKind fileKind = (WUFileKind) file.getPropInt("@kind", WUFileStandard);
  5102. if(file.getPropBool("@temporary")) fileKind = WUFileTemporary; // @temporary, legacy check
  5103. bool needDelete;
  5104. switch(fileKind)
  5105. {
  5106. case WUFileTemporary:
  5107. if(graph==NULL)
  5108. needDelete = true;
  5109. else
  5110. {
  5111. const char *graphOwner = file.queryProp("@graph");
  5112. needDelete = ((graphOwner==NULL) || (strcmp(graph, graphOwner)==0));
  5113. }
  5114. break;
  5115. case WUFileJobOwned:
  5116. needDelete = ((graph==NULL) && deleteJobOwned);
  5117. break;
  5118. case WUFileOwned:
  5119. needDelete = ((graph==NULL) && deleteOwned);
  5120. break;
  5121. default:
  5122. needDelete = false;
  5123. }
  5124. if(needDelete)
  5125. {
  5126. const char *name = file.queryProp("@name");
  5127. LOG(MCdebugProgress, unknownJob, "Removing workunit file %s from DFS", name);
  5128. queryDistributedFileDirectory().removeEntry(name, queryUserDescriptor());
  5129. toRemove.append(file);
  5130. }
  5131. }
  5132. ForEachItemIn(r, toRemove) files->removeTree(&toRemove.item(r));
  5133. }
  5134. static void _noteFileRead(IDistributedFile *file, IPropertyTree *filesRead)
  5135. {
  5136. IDistributedSuperFile *super = file->querySuperFile();
  5137. StringBuffer fname;
  5138. file->getLogicalName(fname);
  5139. StringBuffer path("File[@name=\"");
  5140. path.append(fname).append("\"]");
  5141. IPropertyTree *fileTree = filesRead->queryPropTree(path.str());
  5142. if (fileTree)
  5143. fileTree->setPropInt("@useCount", fileTree->getPropInt("@useCount")+1);
  5144. else
  5145. {
  5146. StringBuffer cluster;
  5147. file->getClusterName(0,cluster);
  5148. fileTree = createPTree();
  5149. fileTree->setProp("@name", fname.str());
  5150. fileTree->setProp("@cluster", cluster.str());
  5151. fileTree->setPropInt("@useCount", 1);
  5152. fileTree = filesRead->addPropTree("File", fileTree);
  5153. }
  5154. if (super)
  5155. {
  5156. Owned<IDistributedFileIterator> iter = super->getSubFileIterator(false);
  5157. ForEach (*iter)
  5158. {
  5159. IDistributedFile &file = iter->query();
  5160. StringBuffer fname;
  5161. file.getLogicalName(fname);
  5162. Owned<IPropertyTree> subfile = createPTree();
  5163. subfile->setProp("@name", fname.str());
  5164. fileTree->addPropTree("Subfile", subfile.getClear());
  5165. _noteFileRead(&file, filesRead);
  5166. }
  5167. }
  5168. }
  5169. void CLocalWorkUnit::noteFileRead(IDistributedFile *file)
  5170. {
  5171. CriticalBlock block(crit);
  5172. IPropertyTree *files = p->queryPropTree("FilesRead");
  5173. if (!files)
  5174. files = p->addPropTree("FilesRead", createPTree());
  5175. _noteFileRead(file, files);
  5176. }
  5177. static void addFile(IPropertyTree *files, const char *fileName, const char *cluster, unsigned usageCount, WUFileKind fileKind, const char *graphOwner)
  5178. {
  5179. StringBuffer path("File[@name=\"");
  5180. path.append(fileName).append("\"]");
  5181. if (cluster)
  5182. path.append("[@cluster=\"").append(cluster).append("\"]");
  5183. IPropertyTree *file = files->queryPropTree(path.str());
  5184. if (file) files->removeTree(file);
  5185. file = createPTree();
  5186. file->setProp("@name", fileName);
  5187. if (cluster)
  5188. file->setProp("@cluster", cluster);
  5189. if (graphOwner)
  5190. file->setProp("@graph", graphOwner);
  5191. file->setPropInt("@kind", (unsigned)fileKind);
  5192. if (WUFileTemporary == fileKind)
  5193. file->setPropInt("@usageCount", usageCount);
  5194. files->addPropTree("File", file);
  5195. }
  5196. void CLocalWorkUnit::addFile(const char *fileName, StringArray *clusters, unsigned usageCount, WUFileKind fileKind, const char *graphOwner)
  5197. {
  5198. CriticalBlock block(crit);
  5199. IPropertyTree *files = p->queryPropTree("Files");
  5200. if (!files)
  5201. files = p->addPropTree("Files", createPTree());
  5202. if (!clusters)
  5203. addFile(fileName, NULL, usageCount, fileKind, graphOwner);
  5204. else
  5205. {
  5206. ForEachItemIn(c, *clusters)
  5207. ::addFile(files, fileName, clusters->item(c), usageCount, fileKind, graphOwner);
  5208. }
  5209. }
  5210. void CLocalWorkUnit::releaseFile(const char *fileName)
  5211. {
  5212. StringBuffer path("File[@name=\"");
  5213. path.append(fileName).append("\"]");
  5214. CriticalBlock block(crit);
  5215. IPropertyTree *files = p->queryPropTree("Files");
  5216. if (!files) return;
  5217. Owned<IPropertyTreeIterator> fiter = files->getElements(path.str());
  5218. ForEach (*fiter)
  5219. {
  5220. IPropertyTree *file = &fiter->query();
  5221. unsigned usageCount = file->getPropInt("@usageCount");
  5222. if (usageCount > 1)
  5223. file->setPropInt("@usageCount", usageCount-1);
  5224. else
  5225. {
  5226. StringAttr name(file->queryProp("@name"));
  5227. files->removeTree(file);
  5228. if (!name.isEmpty()&&(1 == usageCount))
  5229. {
  5230. if (queryDistributedFileDirectory().removeEntry(fileName, queryUserDescriptor()))
  5231. LOG(MCdebugProgress, unknownJob, "Removed (released) file %s from DFS", name.get());
  5232. }
  5233. }
  5234. }
  5235. }
  5236. void CLocalWorkUnit::clearGraphProgress()
  5237. {
  5238. CConstGraphProgress::deleteWuidProgress(p->queryName());
  5239. }
  5240. void CLocalWorkUnit::resetBeforeGeneration()
  5241. {
  5242. CriticalBlock block(crit);
  5243. //Remove all associated files
  5244. Owned<IWUQuery> q = updateQuery();
  5245. q->removeAssociatedFiles();
  5246. //Remove any pre-existing workflow information
  5247. workflowIterator.clear();
  5248. p->removeProp("Workflow");
  5249. }
  5250. unsigned CLocalWorkUnit::queryFileUsage(const char *fileName) const
  5251. {
  5252. StringBuffer path("Files/File[@name=\"");
  5253. path.append(fileName).append("\"]/@usageCount");
  5254. CriticalBlock block(crit);
  5255. return p->getPropInt(path.str());
  5256. }
  5257. IPropertyTree *CLocalWorkUnit::getDiskUsageStats()
  5258. {
  5259. return p->getPropTree("DiskUsageStats");
  5260. }
  5261. void CLocalWorkUnit::addDiskUsageStats(__int64 _avgNodeUsage, unsigned _minNode, __int64 _minNodeUsage, unsigned _maxNode, __int64 _maxNodeUsage, __int64 _graphId)
  5262. {
  5263. IPropertyTree *stats = p->queryPropTree("DiskUsageStats");
  5264. offset_t maxNodeUsage;
  5265. if (stats)
  5266. maxNodeUsage = stats->getPropInt64("@maxNodeUsage");
  5267. else
  5268. {
  5269. stats = p->addPropTree("DiskUsageStats", createPTree());
  5270. maxNodeUsage = 0;
  5271. }
  5272. if ((offset_t)_maxNodeUsage > maxNodeUsage)
  5273. {
  5274. // record all details at time of max node usage.
  5275. stats->setPropInt("@minNode", _minNode);
  5276. stats->setPropInt("@maxNode", _maxNode);
  5277. stats->setPropInt64("@minNodeUsage", _minNodeUsage);
  5278. stats->setPropInt64("@maxNodeUsage", _maxNodeUsage);
  5279. stats->setPropInt64("@graphId", _graphId);
  5280. if (_avgNodeUsage)
  5281. {
  5282. unsigned _skewHi = (unsigned)((100 * (_maxNodeUsage-_avgNodeUsage))/_avgNodeUsage);
  5283. unsigned _skewLo = (unsigned)((100 * (_avgNodeUsage-_minNodeUsage))/_avgNodeUsage);
  5284. stats->setPropInt("@skewHi", _skewHi);
  5285. stats->setPropInt("@skewLo", _skewLo);
  5286. }
  5287. }
  5288. }
  5289. IPropertyTreeIterator & CLocalWorkUnit::getFileIterator() const
  5290. {
  5291. CriticalBlock block(crit);
  5292. return * p->getElements("Files/File");
  5293. }
  5294. IPropertyTreeIterator & CLocalWorkUnit::getFilesReadIterator() const
  5295. {
  5296. CriticalBlock block(crit);
  5297. return * p->getElements("FilesRead/File");
  5298. }
  5299. //=================================================================================================
  5300. IWUActivity * CLocalWorkUnit::updateActivity(__int64 id)
  5301. {
  5302. CriticalBlock block(crit);
  5303. IConstWUActivity *existing = getActivity(id);
  5304. if (existing)
  5305. return (IWUActivity *) existing;
  5306. if (!activities.length())
  5307. p->addPropTree("Activities", createPTree("Activities"));
  5308. IPropertyTree *pl = p->queryPropTree("Activities");
  5309. IPropertyTree *s = pl->addPropTree("Activity", createPTree("Activity"));
  5310. IWUActivity * q = new CLocalWUActivity(LINK(s), id);
  5311. activities.append(*LINK(q));
  5312. return q;
  5313. }
  5314. IConstWUActivity * CLocalWorkUnit::getActivity(__int64 id) const
  5315. {
  5316. CriticalBlock block(crit);
  5317. loadActivities();
  5318. ForEachItemIn(idx, activities)
  5319. {
  5320. IConstWUActivity &cur = activities.item(idx);
  5321. if (cur.getId() == id)
  5322. return &OLINK(cur);
  5323. }
  5324. return NULL;
  5325. }
  5326. void CLocalWorkUnit::loadActivities() const
  5327. {
  5328. CriticalBlock block(crit);
  5329. if (!activitiesCached)
  5330. {
  5331. assertex(activities.length() == 0);
  5332. Owned<IPropertyTreeIterator> r = p->getElements("Activities/Activity");
  5333. for (r->first(); r->isValid(); r->next())
  5334. {
  5335. IPropertyTree *rp = &r->query();
  5336. rp->Link();
  5337. activities.append(*new CLocalWUActivity(rp));
  5338. }
  5339. activitiesCached = true;
  5340. }
  5341. }
  5342. IConstWUActivityIterator& CLocalWorkUnit::getActivities() const
  5343. {
  5344. CriticalBlock block(crit);
  5345. loadActivities();
  5346. return *new CArrayIteratorOf<IConstWUActivity,IConstWUActivityIterator> (activities, 0, (IConstWorkUnit *) this);
  5347. }
  5348. //=================================================================================================
  5349. bool CLocalWorkUnit::switchThorQueue(const char *cluster, IQueueSwitcher *qs)
  5350. {
  5351. CriticalBlock block(crit);
  5352. if (qs->isAuto()&&!getAllowAutoQueueSwitch())
  5353. return false;
  5354. Owned<IConstWUClusterInfo> newci = getTargetClusterInfo(cluster);
  5355. if (!newci)
  5356. return false;
  5357. StringBuffer currentcluster;
  5358. if (!p->getProp("@clusterName",currentcluster))
  5359. return false;
  5360. Owned<IConstWUClusterInfo> curci = getTargetClusterInfo(currentcluster.str());
  5361. if (!curci)
  5362. return false;
  5363. SCMStringBuffer curqname;
  5364. curci->getThorQueue(curqname);
  5365. const char *wuid = p->queryName();
  5366. void *qi = qs->getQ(curqname.str(),wuid);
  5367. if (!qi)
  5368. return false;
  5369. setClusterName(cluster);
  5370. SCMStringBuffer newqname;
  5371. newci->getThorQueue(newqname);
  5372. qs->putQ(newqname.str(),wuid,qi);
  5373. return true;
  5374. }
  5375. //=================================================================================================
  5376. void CLocalWorkUnit::loadGraphs() const
  5377. {
  5378. CriticalBlock block(crit);
  5379. if (!cachedGraphs.get())
  5380. {
  5381. MemoryBuffer buf;
  5382. IPropertyTree *t = p->queryPropTree("PackedGraphs");
  5383. if (t&&t->getPropBin(NULL,buf)) {
  5384. cachedGraphs.setown(createPTree(buf));
  5385. }
  5386. else
  5387. cachedGraphs.set(p->queryPropTree("Graphs"));
  5388. if (cachedGraphs.get()) {
  5389. Owned<IPropertyTreeIterator> r = cachedGraphs->getElements("Graph");
  5390. for (r->first(); r->isValid(); r->next())
  5391. {
  5392. IPropertyTree *rp = &r->query();
  5393. rp->Link();
  5394. graphs.append(*new CLocalWUGraph(rp, p->queryName()));
  5395. }
  5396. }
  5397. }
  5398. }
  5399. mapEnums graphTypes[] = {
  5400. { GraphTypeAny, "unknown" },
  5401. { GraphTypeProgress, "progress" },
  5402. { GraphTypeEcl, "ECL" },
  5403. { GraphTypeActivities, "activities" },
  5404. { GraphTypeSubProgress, "subgraph" },
  5405. { GraphTypeSize, NULL },
  5406. };
  5407. CLocalWUGraph::CLocalWUGraph(IPropertyTree *props, const char *_wuid) : p(props), wuid(_wuid)
  5408. {
  5409. }
  5410. IStringVal& CLocalWUGraph::getName(IStringVal &str) const
  5411. {
  5412. str.set(p->queryProp("@name"));
  5413. return str;
  5414. }
  5415. IStringVal& CLocalWUGraph::getLabel(IStringVal &str) const
  5416. {
  5417. Owned<IPropertyTree> xgmml = getXGMMLTree(false);
  5418. str.set(xgmml->queryProp("@label"));
  5419. return str;
  5420. }
  5421. IStringVal& CLocalWUGraph::getXGMML(IStringVal &str, bool mergeProgress) const
  5422. {
  5423. Owned<IPropertyTree> xgmml = getXGMMLTree(mergeProgress);
  5424. if (xgmml)
  5425. {
  5426. StringBuffer x;
  5427. toXML(xgmml, x);
  5428. str.set(x.str());
  5429. }
  5430. return str;
  5431. }
  5432. unsigned CLocalWorkUnit::getGraphCount() const
  5433. {
  5434. CriticalBlock block(crit);
  5435. if (p->hasProp("Graphs"))
  5436. {
  5437. return p->queryPropTree("Graphs")->numChildren();
  5438. }
  5439. return 0;
  5440. }
  5441. unsigned CLocalWorkUnit::getSourceFileCount() const
  5442. {
  5443. CriticalBlock block(crit);
  5444. if (p->hasProp("FilesRead"))
  5445. {
  5446. return p->queryPropTree("FilesRead")->numChildren();
  5447. }
  5448. return 0;
  5449. }
  5450. unsigned CLocalWorkUnit::getResultCount() const
  5451. {
  5452. CriticalBlock block(crit);
  5453. if (p->hasProp("Results"))
  5454. {
  5455. return p->queryPropTree("Results")->numChildren();
  5456. }
  5457. return 0;
  5458. }
  5459. unsigned CLocalWorkUnit::getVariableCount() const
  5460. {
  5461. CriticalBlock block(crit);
  5462. if (p->hasProp("Variables"))
  5463. {
  5464. return p->queryPropTree("Variables")->numChildren();
  5465. }
  5466. return 0;
  5467. }
  5468. unsigned CLocalWorkUnit::getTimerCount() const
  5469. {
  5470. CriticalBlock block(crit);
  5471. if (p->hasProp("Timings"))
  5472. {
  5473. return p->queryPropTree("Timings")->numChildren();
  5474. }
  5475. return 0;
  5476. }
  5477. unsigned CLocalWorkUnit::getApplicationValueCount() const
  5478. {
  5479. CriticalBlock block(crit);
  5480. if (p->hasProp("Application"))
  5481. {
  5482. return p->queryPropTree("Application")->numChildren();
  5483. }
  5484. return 0;
  5485. }
  5486. IStringVal &CLocalWorkUnit::getXmlParams(IStringVal &str) const
  5487. {
  5488. CriticalBlock block(crit);
  5489. IPropertyTree *paramTree = p->queryPropTree("Parameters");
  5490. if (paramTree)
  5491. {
  5492. StringBuffer temp;
  5493. toXML(paramTree, temp);
  5494. str.set(temp.str());
  5495. }
  5496. return str;
  5497. }
  5498. const IPropertyTree *CLocalWorkUnit::getXmlParams() const
  5499. {
  5500. CriticalBlock block(crit);
  5501. return p->getPropTree("Parameters");
  5502. }
  5503. void CLocalWorkUnit::setXmlParams(const char *params)
  5504. {
  5505. CriticalBlock block(crit);
  5506. p->setPropTree("Parameters", createPTreeFromXMLString(params));
  5507. }
  5508. void CLocalWorkUnit::setXmlParams(IPropertyTree *tree)
  5509. {
  5510. CriticalBlock block(crit);
  5511. p->setPropTree("Parameters", tree);
  5512. }
  5513. unsigned __int64 CLocalWorkUnit::getHash() const
  5514. {
  5515. CriticalBlock block(crit);
  5516. return p->getPropInt64("@hash");
  5517. }
  5518. void CLocalWorkUnit::setHash(unsigned __int64 hash)
  5519. {
  5520. CriticalBlock block(crit);
  5521. p->setPropInt64("@hash", hash);
  5522. }
  5523. IConstWUGraphIterator& CLocalWorkUnit::getGraphs(WUGraphType type) const
  5524. {
  5525. CriticalBlock block(crit);
  5526. loadGraphs();
  5527. IConstWUGraphIterator *giter = new CArrayIteratorOf<IConstWUGraph,IConstWUGraphIterator> (graphs, 0, (IConstWorkUnit *) this);
  5528. if (type!=GraphTypeAny) {
  5529. class CConstWUGraphIterator: public CInterface, implements IConstWUGraphIterator
  5530. {
  5531. WUGraphType type;
  5532. Owned<IConstWUGraphIterator> base;
  5533. bool match()
  5534. {
  5535. return base->query().getType()==type;
  5536. }
  5537. public:
  5538. IMPLEMENT_IINTERFACE;
  5539. CConstWUGraphIterator(IConstWUGraphIterator *_base,WUGraphType _type)
  5540. : base(_base)
  5541. {
  5542. type = _type;
  5543. }
  5544. bool first()
  5545. {
  5546. if (!base->first())
  5547. return false;
  5548. if (match())
  5549. return true;
  5550. return next();
  5551. }
  5552. bool next()
  5553. {
  5554. while (base->next())
  5555. if (match())
  5556. return true;
  5557. return false;
  5558. }
  5559. virtual bool isValid()
  5560. {
  5561. return base->isValid();
  5562. }
  5563. IConstWUGraph & query()
  5564. {
  5565. return base->query();
  5566. }
  5567. };
  5568. giter = new CConstWUGraphIterator(giter,type);
  5569. }
  5570. return *giter;
  5571. }
  5572. IConstWUGraph* CLocalWorkUnit::getGraph(const char *qname) const
  5573. {
  5574. CriticalBlock block(crit);
  5575. loadGraphs();
  5576. ForEachItemIn(idx, graphs)
  5577. {
  5578. SCMStringBuffer name;
  5579. IConstWUGraph &cur = graphs.item(idx);
  5580. cur.getName(name);
  5581. if (stricmp(name.str(), qname)==0)
  5582. {
  5583. cur.Link();
  5584. return &cur;
  5585. }
  5586. }
  5587. return NULL;
  5588. }
  5589. IWUGraph* CLocalWorkUnit::createGraph()
  5590. {
  5591. // For this to be legally called, we must have the write-able interface. So we are already locked for write.
  5592. CriticalBlock block(crit);
  5593. ensureGraphsUnpacked();
  5594. loadGraphs();
  5595. if (!graphs.length())
  5596. p->addPropTree("Graphs", createPTree("Graphs"));
  5597. IPropertyTree *r = p->queryPropTree("Graphs");
  5598. IPropertyTree *s = r->addPropTree("Graph", createPTreeFromXMLString("<Graph fetchEntire='1'/>"));
  5599. s->Link();
  5600. IWUGraph* q = new CLocalWUGraph(s, p->queryName());
  5601. q->Link();
  5602. graphs.append(*q);
  5603. return q;
  5604. }
  5605. IWUGraph * CLocalWorkUnit::updateGraph(const char * name)
  5606. {
  5607. CriticalBlock block(crit);
  5608. ensureGraphsUnpacked();
  5609. IConstWUGraph *existing = getGraph(name);
  5610. if (existing)
  5611. return (IWUGraph *) existing;
  5612. IWUGraph * q = createGraph();
  5613. q->setName(name);
  5614. return q;
  5615. }
  5616. IConstWUGraphProgress *CLocalWorkUnit::getGraphProgress(const char *name) const
  5617. {
  5618. CriticalBlock block(crit);
  5619. return new CConstGraphProgress(p->queryName(), name);
  5620. }
  5621. IStringVal& CLocalWUGraph::getDOT(IStringVal &str) const
  5622. {
  5623. UNIMPLEMENTED;
  5624. }
  5625. void CLocalWUGraph::setName(const char *str)
  5626. {
  5627. p->setProp("@name", str);
  5628. progress.clear();
  5629. progress.setown(new CConstGraphProgress(wuid, str));
  5630. }
  5631. void CLocalWUGraph::setXGMML(const char *str)
  5632. {
  5633. setXGMMLTree(createPTreeFromXMLString(str));
  5634. }
  5635. void CLocalWUGraph::setXGMMLTree(IPropertyTree *graph)
  5636. {
  5637. assertex(strcmp(graph->queryName(), "graph")==0);
  5638. IPropertyTree *xgmml = createPTree("xgmml");
  5639. xgmml->setPropTree("graph", graph);
  5640. p->setPropTree("xgmml", xgmml);
  5641. }
  5642. void CLocalWUGraph::mergeProgress(IPropertyTree &rootNode, IPropertyTree &progressTree, const unsigned &progressV) const
  5643. {
  5644. IPropertyTree *graphNode = rootNode.queryPropTree("att/graph");
  5645. if (!graphNode) return;
  5646. unsigned nodeId = rootNode.getPropInt("@id");
  5647. StringBuffer progressNodePath("node[@id=\"");
  5648. progressNodePath.append(nodeId).append("\"]");
  5649. IPropertyTree *progressNode = progressTree.queryPropTree(progressNodePath.str());
  5650. if (progressNode)
  5651. {
  5652. Owned<IPropertyTreeIterator> edges = progressNode->getElements("edge");
  5653. ForEach (*edges)
  5654. {
  5655. IPropertyTree &edge = edges->query();
  5656. StringBuffer edgePath("edge[@id=\"");
  5657. edgePath.append(edge.queryProp("@id")).append("\"]");
  5658. IPropertyTree *graphEdge = graphNode->queryPropTree(edgePath.str());
  5659. if (graphEdge)
  5660. {
  5661. if (progressV < 1)
  5662. mergePTree(graphEdge, &edge);
  5663. else
  5664. { // must translate to XGMML format
  5665. Owned<IAttributeIterator> aIter = edge.getAttributes();
  5666. ForEach (*aIter)
  5667. {
  5668. const char *aName = aIter->queryName()+1;
  5669. if (0 != stricmp("id", aName)) // "id" reserved.
  5670. {
  5671. IPropertyTree *att = graphEdge->addPropTree("att", createPTree());
  5672. att->setProp("@name", aName);
  5673. att->setProp("@value", aIter->queryValue());
  5674. }
  5675. }
  5676. // This is really only here, so that our progress format can use non-attribute values, which have different efficiency qualifies (e.g. can be external by dali)
  5677. Owned<IPropertyTreeIterator> iter = edge.getElements("*");
  5678. ForEach (*iter)
  5679. {
  5680. IPropertyTree &t = iter->query();
  5681. IPropertyTree *att = graphEdge->addPropTree("att", createPTree());
  5682. att->setProp("@name", t.queryName());
  5683. att->setProp("@value", t.queryProp(NULL));
  5684. }
  5685. }
  5686. }
  5687. }
  5688. Owned<IPropertyTreeIterator> nodes = progressNode->getElements("node");
  5689. ForEach (*nodes)
  5690. {
  5691. IPropertyTree &node = nodes->query();
  5692. StringBuffer nodePath("node[@id=\"");
  5693. nodePath.append(node.queryProp("@id")).append("\"]");
  5694. IPropertyTree *_node = graphNode->queryPropTree(nodePath.str());
  5695. if (_node)
  5696. {
  5697. if (progressV < 1)
  5698. mergePTree(_node, &node);
  5699. else
  5700. { // must translate to XGMML format
  5701. Owned<IAttributeIterator> aIter = node.getAttributes();
  5702. ForEach (*aIter)
  5703. {
  5704. const char *aName = aIter->queryName()+1;
  5705. if (0 != stricmp("id", aName)) // "id" reserved.
  5706. {
  5707. IPropertyTree *att = _node->addPropTree("att", createPTree());
  5708. att->setProp("@name", aName);
  5709. att->setProp("@value", aIter->queryValue());
  5710. }
  5711. }
  5712. }
  5713. }
  5714. }
  5715. }
  5716. Owned<IPropertyTreeIterator> iter = graphNode->getElements("node");
  5717. ForEach (*iter)
  5718. mergeProgress(iter->query(), progressTree, progressV);
  5719. }
  5720. IPropertyTree * CLocalWUGraph::getXGMMLTree(bool doMergeProgress) const
  5721. {
  5722. if (!doMergeProgress)
  5723. return p->getPropTree("xgmml/graph");
  5724. else
  5725. {
  5726. IPropertyTree *src = p->queryPropTree("xgmml/graph");
  5727. if (!src) return NULL;
  5728. Owned<IPropertyTree> copy = createPTreeFromIPT(src);
  5729. Owned<IConstWUGraphProgress> _progress;
  5730. if (progress) _progress.set(progress);
  5731. else
  5732. _progress.setown(new CConstGraphProgress(wuid, p->queryProp("@name")));
  5733. unsigned progressV = _progress->queryFormatVersion();
  5734. IPropertyTree *progressTree = _progress->queryProgressTree();
  5735. Owned<IPropertyTreeIterator> nodeIterator = copy->getElements("node");
  5736. ForEach (*nodeIterator)
  5737. mergeProgress(nodeIterator->query(), *progressTree, progressV);
  5738. return LINK(copy);
  5739. }
  5740. }
  5741. bool CLocalWUGraph::isValid() const
  5742. {
  5743. return p->hasProp("xgmml/graph/node");
  5744. }
  5745. WUGraphType CLocalWUGraph::getType() const
  5746. {
  5747. return (WUGraphType) getEnum(p, "@type", graphTypes);
  5748. }
  5749. IStringVal & CLocalWUGraph::getTypeName(IStringVal &str) const
  5750. {
  5751. str.set(p->queryProp("@type"));
  5752. if (!str.length())
  5753. str.set("unknown");
  5754. return str;
  5755. }
  5756. void CLocalWUGraph::setType(WUGraphType _type)
  5757. {
  5758. setEnum(p, "@type", _type, graphTypes);
  5759. }
  5760. //=================================================================================================
  5761. mapEnums queryFileTypes[] = {
  5762. { FileTypeCpp, "cpp" },
  5763. { FileTypeDll, "dll" },
  5764. { FileTypeResText, "res" },
  5765. { FileTypeHintXml, "hint" },
  5766. { FileTypeXml, "xml" },
  5767. { FileTypeSize, NULL },
  5768. };
  5769. CLocalWUAssociated::CLocalWUAssociated(IPropertyTree *props) : p(props)
  5770. {
  5771. }
  5772. WUFileType CLocalWUAssociated::getType() const
  5773. {
  5774. return (WUFileType)getEnum(p, "@type", queryFileTypes);
  5775. }
  5776. IStringVal & CLocalWUAssociated::getDescription(IStringVal & str) const
  5777. {
  5778. str.set(p->queryProp("@desc"));
  5779. return str;
  5780. }
  5781. IStringVal & CLocalWUAssociated::getIp(IStringVal & str) const
  5782. {
  5783. str.set(p->queryProp("@ip"));
  5784. return str;
  5785. }
  5786. IStringVal & CLocalWUAssociated::getName(IStringVal & str) const
  5787. {
  5788. str.set(p->queryProp("@filename"));
  5789. return str;
  5790. }
  5791. IStringVal & CLocalWUAssociated::getNameTail(IStringVal & str) const
  5792. {
  5793. str.set(pathTail(p->queryProp("@filename")));
  5794. return str;
  5795. }
  5796. unsigned CLocalWUAssociated::getCrc() const
  5797. {
  5798. return p->getPropInt("@crc", 0);
  5799. }
  5800. //=================================================================================================
  5801. CLocalWUQuery::CLocalWUQuery(IPropertyTree *props) : p(props)
  5802. {
  5803. associatedCached = false;
  5804. }
  5805. mapEnums queryTypes[] = {
  5806. { QueryTypeUnknown, "unknown" },
  5807. { QueryTypeEcl, "ECL" },
  5808. { QueryTypeSql, "SQL" },
  5809. { QueryTypeXml, "XML" },
  5810. { QueryTypeAttribute, "Attribute" },
  5811. { QueryTypeSize, NULL },
  5812. };
  5813. WUQueryType CLocalWUQuery::getQueryType() const
  5814. {
  5815. return (WUQueryType) getEnum(p, "@type", queryTypes);
  5816. }
  5817. void CLocalWUQuery::setQueryType(WUQueryType qt)
  5818. {
  5819. setEnum(p, "@type", qt, queryTypes);
  5820. }
  5821. IStringVal& CLocalWUQuery::getQueryText(IStringVal &str) const
  5822. {
  5823. str.set(p->queryProp("Text"));
  5824. return str;
  5825. }
  5826. IStringVal& CLocalWUQuery::getQueryShortText(IStringVal &str) const
  5827. {
  5828. const char * text = p->queryProp("Text");
  5829. if (isArchiveQuery(text))
  5830. {
  5831. Owned<IPropertyTree> xml = createPTreeFromXMLString(text, ipt_caseInsensitive);
  5832. const char * path = xml->queryProp("Query/@attributePath");
  5833. if (path)
  5834. {
  5835. IPropertyTree * resolved = resolveDefinitionInArchive(xml, path);
  5836. if (resolved)
  5837. str.set(resolved->queryProp(NULL));
  5838. }
  5839. else
  5840. str.set(xml->queryProp("Query"));
  5841. }
  5842. else
  5843. str.set(text);
  5844. return str;
  5845. }
  5846. IStringVal& CLocalWUQuery::getQueryName(IStringVal &str) const
  5847. {
  5848. str.set(p->queryProp("@name"));
  5849. return str;
  5850. }
  5851. IStringVal & CLocalWUQuery::getQueryMainDefinition(IStringVal & str) const
  5852. {
  5853. str.set(p->queryProp("@main"));
  5854. return str;
  5855. }
  5856. IStringVal& CLocalWUQuery::getQueryDllName(IStringVal &str) const
  5857. {
  5858. Owned<IConstWUAssociatedFile> entry = getAssociatedFile(FileTypeDll, 0);
  5859. if (entry)
  5860. entry->getNameTail(str);
  5861. return str;
  5862. }
  5863. IStringVal& CLocalWUQuery::getQueryCppName(IStringVal &str) const
  5864. {
  5865. Owned<IConstWUAssociatedFile> entry = getAssociatedFile(FileTypeCpp, 0);
  5866. if (entry)
  5867. entry->getName(str);
  5868. return str;
  5869. }
  5870. IStringVal& CLocalWUQuery::getQueryResTxtName(IStringVal &str) const
  5871. {
  5872. Owned<IConstWUAssociatedFile> entry = getAssociatedFile(FileTypeResText, 0);
  5873. if (entry)
  5874. entry->getName(str);
  5875. return str;
  5876. }
  5877. unsigned CLocalWUQuery::getQueryDllCrc() const
  5878. {
  5879. Owned<IConstWUAssociatedFile> entry = getAssociatedFile(FileTypeDll, 0);
  5880. if (entry)
  5881. return entry->getCrc();
  5882. return 0;
  5883. }
  5884. void CLocalWUQuery::setQueryText(const char *text)
  5885. {
  5886. p->setProp("Text", text);
  5887. }
  5888. void CLocalWUQuery::setQueryName(const char *qname)
  5889. {
  5890. p->setProp("@name", qname);
  5891. }
  5892. void CLocalWUQuery::setQueryMainDefinition(const char * str)
  5893. {
  5894. p->setProp("@main", str);
  5895. }
  5896. void CLocalWUQuery::addAssociatedFile(WUFileType type, const char * name, const char * ip, const char * desc, unsigned crc)
  5897. {
  5898. CriticalBlock block(crit);
  5899. loadAssociated();
  5900. if (!associated.length())
  5901. p->addPropTree("Associated", createPTree("Associated"));
  5902. IPropertyTree *pl = p->queryPropTree("Associated");
  5903. IPropertyTree *s = pl->addPropTree("File", createPTree("File"));
  5904. setEnum(s, "@type", type, queryFileTypes);
  5905. s->setProp("@filename", name);
  5906. s->setProp("@ip", ip);
  5907. s->setProp("@desc", desc);
  5908. if (crc)
  5909. s->setPropInt("@crc", crc);
  5910. IConstWUAssociatedFile * q = new CLocalWUAssociated(LINK(s));
  5911. associated.append(*q);
  5912. }
  5913. void CLocalWUQuery::removeAssociatedFiles()
  5914. {
  5915. associatedCached = false;
  5916. associated.kill();
  5917. p->removeProp("Associated");
  5918. }
  5919. IConstWUAssociatedFile * CLocalWUQuery::getAssociatedFile(WUFileType type, unsigned index) const
  5920. {
  5921. CriticalBlock block(crit);
  5922. loadAssociated();
  5923. ForEachItemIn(idx, associated)
  5924. {
  5925. CLocalWUAssociated &cur = static_cast<CLocalWUAssociated &>(associated.item(idx));
  5926. if (cur.getType() == type)
  5927. {
  5928. if (index-- == 0)
  5929. return &OLINK(cur);
  5930. }
  5931. }
  5932. return NULL;
  5933. }
  5934. void CLocalWUQuery::addSpecialCaseAssociated(WUFileType type, const char * propname, unsigned crc) const
  5935. {
  5936. const char * name = p->queryProp(propname);
  5937. if (name)
  5938. {
  5939. IPropertyTree *s = createPTree("File");
  5940. setEnum(s, "@type", type, queryFileTypes);
  5941. s->setProp("@filename", name);
  5942. if (crc)
  5943. s->setPropInt("@crc", crc);
  5944. associated.append(*new CLocalWUAssociated(s));
  5945. }
  5946. }
  5947. void CLocalWUQuery::loadAssociated() const
  5948. {
  5949. CriticalBlock block(crit);
  5950. if (!associatedCached)
  5951. {
  5952. assertex(associated.length() == 0);
  5953. addSpecialCaseAssociated(FileTypeDll, "DllName", p->getPropInt("DllCrc", 0));
  5954. addSpecialCaseAssociated(FileTypeCpp, "CppName", 0);
  5955. addSpecialCaseAssociated(FileTypeResText, "ResTxtName", 0);
  5956. Owned<IPropertyTreeIterator> r = p->getElements("Associated/File");
  5957. for (r->first(); r->isValid(); r->next())
  5958. {
  5959. IPropertyTree *rp = &r->query();
  5960. rp->Link();
  5961. associated.append(*new CLocalWUAssociated(rp));
  5962. }
  5963. associatedCached = true;
  5964. }
  5965. }
  5966. IConstWUAssociatedFileIterator& CLocalWUQuery::getAssociatedFiles() const
  5967. {
  5968. CriticalBlock block(crit);
  5969. loadAssociated();
  5970. return *new CArrayIteratorOf<IConstWUAssociatedFile,IConstWUAssociatedFileIterator> (associated, 0, (IConstWUQuery *) this);
  5971. }
  5972. //========================================================================================
  5973. CLocalWUWebServicesInfo::CLocalWUWebServicesInfo(IPropertyTree *props) : p(props)
  5974. {
  5975. }
  5976. IStringVal& CLocalWUWebServicesInfo::getModuleName(IStringVal &str) const
  5977. {
  5978. str.set(p->queryProp("@module"));
  5979. return str;
  5980. }
  5981. IStringVal& CLocalWUWebServicesInfo::getAttributeName(IStringVal &str) const
  5982. {
  5983. str.set(p->queryProp("@attribute"));
  5984. return str;
  5985. }
  5986. IStringVal& CLocalWUWebServicesInfo::getDefaultName(IStringVal &str) const
  5987. {
  5988. str.set(p->queryProp("@defaultName"));
  5989. return str;
  5990. }
  5991. unsigned CLocalWUWebServicesInfo::getWebServicesCRC() const
  5992. {
  5993. return (unsigned) p->getPropInt("@crc");
  5994. }
  5995. IStringVal& CLocalWUWebServicesInfo::getInfo(const char *name, IStringVal &str) const
  5996. {
  5997. if (!name)
  5998. {
  5999. StringBuffer ws_info;
  6000. ws_info.appendf("<%s ", p->queryName());
  6001. Owned<IAttributeIterator> attrs = p->getAttributes();
  6002. for(attrs->first(); attrs->isValid(); attrs->next())
  6003. {
  6004. const char *name = attrs->queryName()+1;
  6005. const char *value = attrs->queryValue();
  6006. ws_info.appendf("%s='%s' ", name, value);
  6007. }
  6008. ws_info.append("> \n");
  6009. Owned<IPropertyTreeIterator> info = p->getElements("*");
  6010. ForEach(*info)
  6011. {
  6012. IPropertyTree &item = info->query();
  6013. const char *name = item.queryName();
  6014. if (name)
  6015. {
  6016. MemoryBuffer mb;
  6017. bool isbin = p->isBinary(name);
  6018. if (isbin)
  6019. {
  6020. p->getPropBin(name,mb);
  6021. if (mb.length())
  6022. {
  6023. unsigned len = 0;
  6024. mb.read(len);
  6025. StringBuffer encodedString;
  6026. StringBuffer val(len, (const char *) mb.readDirect(len));
  6027. encodeXML(val, encodedString);
  6028. ws_info.appendf("<%s>%s</%s>", name, encodedString.str(), name);
  6029. }
  6030. }
  6031. else
  6032. {
  6033. StringBuffer tmp;
  6034. toXML(&item, tmp);
  6035. ws_info.append(tmp.str());
  6036. }
  6037. }
  6038. }
  6039. ws_info.appendf("</%s>", p->queryName());
  6040. str.setLen(ws_info.str(), ws_info.length());
  6041. }
  6042. else
  6043. {
  6044. MemoryBuffer mb;
  6045. p->getPropBin(name,mb);
  6046. if (mb.length())
  6047. {
  6048. unsigned len;
  6049. mb.read(len);
  6050. str.setLen((const char *) mb.readDirect(len), len);
  6051. }
  6052. }
  6053. return str;
  6054. }
  6055. void CLocalWUWebServicesInfo::setModuleName(const char *mname)
  6056. {
  6057. p->setProp("@module", mname);
  6058. }
  6059. void CLocalWUWebServicesInfo::setAttributeName(const char *aname)
  6060. {
  6061. p->setProp("@attribute", aname);
  6062. }
  6063. void CLocalWUWebServicesInfo::setDefaultName(const char *dname)
  6064. {
  6065. p->setProp("@defaultName", dname);
  6066. }
  6067. void CLocalWUWebServicesInfo::setWebServicesCRC(unsigned crc)
  6068. {
  6069. p->setPropInt("@crc", crc);
  6070. }
  6071. void CLocalWUWebServicesInfo::setInfo(const char *name, const char *info)
  6072. {
  6073. MemoryBuffer m;
  6074. unsigned len = (size32_t)strlen(info);
  6075. serializeLPString(len, info, m);
  6076. p->setPropBin(name, m.length(), m.toByteArray());
  6077. }
  6078. //========================================================================================
  6079. CLocalWURoxieQueryInfo::CLocalWURoxieQueryInfo(IPropertyTree *props) : p(props)
  6080. {
  6081. }
  6082. IStringVal& CLocalWURoxieQueryInfo::getQueryInfo(IStringVal &str) const
  6083. {
  6084. IPropertyTree *queryTree = p->queryPropTree("query");
  6085. if (queryTree)
  6086. {
  6087. StringBuffer temp;
  6088. toXML(queryTree, temp);
  6089. str.set(temp.str());
  6090. }
  6091. return str;
  6092. }
  6093. IStringVal& CLocalWURoxieQueryInfo::getDefaultPackageInfo(IStringVal &str) const
  6094. {
  6095. MemoryBuffer mb;
  6096. p->getPropBin("RoxiePackages",mb);
  6097. if (mb.length())
  6098. {
  6099. unsigned len;
  6100. mb.read(len);
  6101. str.setLen((const char *) mb.readDirect(len), len);
  6102. }
  6103. return str;
  6104. }
  6105. IStringVal& CLocalWURoxieQueryInfo::getRoxieClusterName(IStringVal &str) const
  6106. {
  6107. const char *val = p->queryProp("@roxieClusterName");
  6108. if (val)
  6109. str.set(val);
  6110. return str;
  6111. }
  6112. IStringVal& CLocalWURoxieQueryInfo::getWuid(IStringVal &str) const
  6113. {
  6114. const char *val = p->queryProp("@wuid");
  6115. if (val)
  6116. str.set(val);
  6117. return str;
  6118. }
  6119. void CLocalWURoxieQueryInfo::setQueryInfo(const char *info)
  6120. {
  6121. IPropertyTree *queryTree = p->queryPropTree("query");
  6122. if (queryTree)
  6123. p->removeTree(queryTree);
  6124. IPropertyTree * tempTree = p->addPropTree("query", createPTreeFromXMLString(info));
  6125. if (!p->hasProp("@roxieClusterName"))
  6126. {
  6127. const char *roxieClusterName = tempTree->queryProp("@roxieName");
  6128. if (roxieClusterName && *roxieClusterName)
  6129. p->addProp("@roxieClusterName", roxieClusterName);
  6130. }
  6131. if (!p->hasProp("@wuid"))
  6132. {
  6133. const char *wuid = tempTree->queryProp("Query/@wuid");
  6134. if (wuid && *wuid)
  6135. p->addProp("@wuid", wuid);
  6136. }
  6137. }
  6138. void CLocalWURoxieQueryInfo::setDefaultPackageInfo(const char *info, int len)
  6139. {
  6140. MemoryBuffer m;
  6141. serializeLPString(len, info, m);
  6142. p->setPropBin("RoxiePackages", m.length(), m.toByteArray());
  6143. }
  6144. void CLocalWURoxieQueryInfo::setRoxieClusterName(const char *info)
  6145. {
  6146. p->setProp("@roxieClusterName", info);
  6147. }
  6148. void CLocalWURoxieQueryInfo::setWuid(const char *info)
  6149. {
  6150. p->setProp("@wuid", info);
  6151. }
  6152. //========================================================================================
  6153. CLocalWUResult::CLocalWUResult(IPropertyTree *props) : p(props)
  6154. {
  6155. }
  6156. mapEnums resultStatuses[] = {
  6157. { ResultStatusUndefined, "undefined" },
  6158. { ResultStatusCalculated, "calculated" },
  6159. { ResultStatusSupplied, "supplied" },
  6160. { ResultStatusFailed, "failed" },
  6161. { ResultStatusPartial, "partial" },
  6162. { ResultStatusSize, NULL }
  6163. };
  6164. WUResultStatus CLocalWUResult::getResultStatus() const
  6165. {
  6166. return (WUResultStatus ) getEnum(p, "@status", resultStatuses);
  6167. }
  6168. IStringVal& CLocalWUResult::getResultName(IStringVal &str) const
  6169. {
  6170. str.set(p->queryProp("@name"));
  6171. return str;
  6172. }
  6173. int CLocalWUResult::getResultSequence() const
  6174. {
  6175. return p->getPropInt("@sequence", -1);
  6176. }
  6177. bool CLocalWUResult::isResultScalar() const
  6178. {
  6179. return p->getPropInt("@isScalar", 1) != 0;
  6180. }
  6181. bool findSize(int size, IntArray &sizes)
  6182. {
  6183. ForEachItemIn(idx, sizes)
  6184. {
  6185. if (sizes.item(idx)==size)
  6186. return true;
  6187. }
  6188. return false;
  6189. }
  6190. void CLocalWUResult::getSchema(TypeInfoArray &types, StringAttrArray &names, IStringVal * eclText) const
  6191. {
  6192. MemoryBuffer schema;
  6193. p->getPropBin("SchemaRaw", schema);
  6194. if (schema.length())
  6195. {
  6196. for (;;)
  6197. {
  6198. StringAttr name;
  6199. schema.read(name);
  6200. if (*schema.readDirect(0)==type_void)
  6201. break;
  6202. names.append(*new StringAttrItem(name));
  6203. types.append(*deserializeType(schema)); // MORE - nested records!
  6204. }
  6205. schema.skip(1);
  6206. if (schema.length() != schema.getPos())
  6207. {
  6208. unsigned eclLen;
  6209. schema.read(eclLen);
  6210. const char * schemaData = (const char *)schema.readDirect(eclLen);
  6211. if (eclText)
  6212. {
  6213. eclText->setLen(schemaData, eclLen);
  6214. if ((eclLen == 0) && names.ordinality())
  6215. {
  6216. const char * firstName = names.item(0).text;
  6217. StringBuffer temp;
  6218. temp.append("RECORD ");
  6219. types.item(0).getECLType(temp);
  6220. temp.append(" value{NAMED('").append(firstName).append("')}").append("; END;");
  6221. eclText->set(temp.str());
  6222. }
  6223. }
  6224. }
  6225. }
  6226. }
  6227. void readRow(StringBuffer &out, MemoryBuffer &in, TypeInfoArray &types, StringAttrArray &names)
  6228. {
  6229. ForEachItemIn(idx, types)
  6230. {
  6231. StringAttrItem &name = names.item(idx);
  6232. ITypeInfo &type = types.item(idx);
  6233. unsigned size = type.getSize();
  6234. switch(type.getTypeCode())
  6235. {
  6236. case type_data:
  6237. if (size==UNKNOWN_LENGTH)
  6238. {
  6239. if (in.remaining() < sizeof(int))
  6240. throw MakeStringException(WUERR_CorruptResult, "corrupt workunit information");
  6241. in.read(size);
  6242. }
  6243. outputXmlData(size, in.readDirect(size), name.text, out);
  6244. break;
  6245. case type_string:
  6246. if (size==UNKNOWN_LENGTH)
  6247. {
  6248. if (in.remaining() < sizeof(int))
  6249. throw MakeStringException(WUERR_CorruptResult, "corrupt workunit information");
  6250. in.read(size);
  6251. }
  6252. outputXmlString(size, (const char *) in.readDirect(size), name.text, out);
  6253. break;
  6254. case type_varstring:
  6255. {
  6256. if (size == UNKNOWN_LENGTH)
  6257. size = (size32_t)strlen((const char *) in.readDirect(0))+1;
  6258. const char * text = (const char *) in.readDirect(size);
  6259. outputXmlString((size32_t)strlen(text), text, name.text, out);
  6260. break;
  6261. }
  6262. case type_unicode:
  6263. {
  6264. unsigned len = type.getStringLen();
  6265. if (size==UNKNOWN_LENGTH)
  6266. in.read(len);
  6267. outputXmlUnicode(len, (UChar const *) in.readDirect(len*2), name.text, out);
  6268. }
  6269. break;
  6270. case type_utf8:
  6271. {
  6272. unsigned len = type.getStringLen();
  6273. if (size==UNKNOWN_LENGTH)
  6274. {
  6275. in.read(len);
  6276. size = rtlUtf8Size(len, in.readDirect(0));
  6277. }
  6278. outputXmlUtf8(len, (const char *) in.readDirect(size), name.text, out);
  6279. }
  6280. break;
  6281. case type_qstring:
  6282. {
  6283. unsigned len = type.getStringLen();
  6284. if (size==UNKNOWN_LENGTH)
  6285. in.read(len);
  6286. unsigned outlen;
  6287. char *outstr;
  6288. rtlQStrToStrX(outlen, outstr, len, (const char *) in.readDirect(rtlQStrSize(len)));
  6289. outputXmlString(outlen, outstr, name.text, out);
  6290. free(outstr);
  6291. break;
  6292. }
  6293. case type_int:
  6294. case type_swapint:
  6295. if (type.isSigned())
  6296. {
  6297. const unsigned char *raw = (const unsigned char *) in.readDirect(size);
  6298. unsigned __int64 cval8 = 0;
  6299. //MORE: I think this is wrong - swapped doesn't mean little/big/
  6300. if (type.isSwappedEndian())
  6301. {
  6302. unsigned idx = 0;
  6303. if (raw[idx] & 0x80)
  6304. cval8 = (__int64)-1;
  6305. while (size--)
  6306. cval8 = (cval8 << 8) | raw[idx++];
  6307. }
  6308. else
  6309. {
  6310. if (raw[size-1] & 0x80)
  6311. cval8 = (__int64)-1;
  6312. while (size--)
  6313. cval8 = (cval8 << 8) | raw[size];
  6314. }
  6315. outputXmlInt((__int64) cval8, name.text, out);
  6316. }
  6317. else
  6318. {
  6319. const unsigned char *raw = (const unsigned char *) in.readDirect(size);
  6320. unsigned __int64 cval8 = 0;
  6321. if (type.isSwappedEndian())
  6322. {
  6323. unsigned idx = 0;
  6324. while (size--)
  6325. cval8 = (cval8 << 8) | raw[idx++];
  6326. }
  6327. else
  6328. {
  6329. while (size--)
  6330. cval8 = (cval8 << 8) | raw[size];
  6331. }
  6332. outputXmlUInt(cval8, name.text, out);
  6333. }
  6334. break;
  6335. case type_boolean:
  6336. bool cvalb;
  6337. in.read(cvalb);
  6338. outputXmlBool(cvalb, name.text, out);
  6339. break;
  6340. case type_decimal:
  6341. if (type.isSigned())
  6342. outputXmlDecimal(in.readDirect(size), size, type.getPrecision(), name.text, out);
  6343. else
  6344. outputXmlUDecimal(in.readDirect(size), size, type.getPrecision(), name.text, out);
  6345. break;
  6346. case type_real:
  6347. double cvald;
  6348. switch(size)
  6349. {
  6350. case 4:
  6351. float cvalf;
  6352. in.read(cvalf);
  6353. cvald = cvalf;
  6354. break;
  6355. case 8:
  6356. in.read(cvald);
  6357. break;
  6358. }
  6359. outputXmlReal(cvald, name.text, out);
  6360. break;
  6361. default:
  6362. assertex(!"unexpected type in raw record");
  6363. break;
  6364. }
  6365. }
  6366. }
  6367. IStringVal& CLocalWUResult::getResultXml(IStringVal &str) const
  6368. {
  6369. TypeInfoArray types;
  6370. StringAttrArray names;
  6371. getSchema(types, names);
  6372. StringBuffer xml;
  6373. MemoryBuffer raw;
  6374. p->getPropBin("Value", raw);
  6375. const char * name = p->queryProp("@name");
  6376. if (name)
  6377. xml.appendf("<Dataset name=\'%s\'>\n", name);
  6378. else
  6379. xml.append("<Dataset>\n");
  6380. unsigned __int64 numrows = getResultRowCount();
  6381. while (numrows--)
  6382. {
  6383. xml.append(" <Row>");
  6384. readRow(xml, raw, types, names);
  6385. xml.append("</Row>\n");
  6386. }
  6387. xml.append("</Dataset>\n");
  6388. str.set(xml.str());
  6389. return str;
  6390. }
  6391. unsigned CLocalWUResult::getResultFetchSize() const
  6392. {
  6393. return p->getPropInt("fetchSize", 100);
  6394. }
  6395. __int64 CLocalWUResult::getResultTotalRowCount() const
  6396. {
  6397. return p->getPropInt64("totalRowCount", -1);
  6398. }
  6399. __int64 CLocalWUResult::getResultRowCount() const
  6400. {
  6401. return p->getPropInt64("rowCount", 0);
  6402. }
  6403. void CLocalWUResult::getResultDataset(IStringVal & ecl, IStringVal & defs) const
  6404. {
  6405. ecl.set(p->queryProp("datasetEcl"));
  6406. defs.set(p->queryProp("datasetEclDefs"));
  6407. }
  6408. IStringVal& CLocalWUResult::getResultLogicalName(IStringVal & val) const
  6409. {
  6410. val.set(p->queryProp("logicalName"));
  6411. return val;
  6412. }
  6413. IStringVal& CLocalWUResult::getResultKeyField(IStringVal & ecl) const
  6414. {
  6415. ecl.set(p->queryProp("keyField"));
  6416. return ecl;
  6417. }
  6418. unsigned CLocalWUResult::getResultRequestedRows() const
  6419. {
  6420. return p->getPropInt("requestedRows", 1);
  6421. }
  6422. IStringVal& CLocalWUResult::getResultEclSchema(IStringVal & str) const
  6423. {
  6424. TypeInfoArray types;
  6425. StringAttrArray names;
  6426. getSchema(types, names, &str);
  6427. return str;
  6428. }
  6429. IStringVal& CLocalWUResult::getResultRecordSizeEntry(IStringVal & str) const
  6430. {
  6431. str.set(p->queryProp("@recordSizeEntry"));
  6432. return str;
  6433. }
  6434. IStringVal& CLocalWUResult::getResultTransformerEntry(IStringVal & str) const
  6435. {
  6436. str.set(p->queryProp("@transformerEntry"));
  6437. return str;
  6438. }
  6439. __int64 CLocalWUResult::getResultRowLimit() const
  6440. {
  6441. return p->getPropInt64("@rowLimit");
  6442. }
  6443. IStringVal& CLocalWUResult::getResultFilename(IStringVal & str) const
  6444. {
  6445. str.set(p->queryProp("@tempFilename"));
  6446. return str;
  6447. }
  6448. void CLocalWUResult::setResultStatus(WUResultStatus status)
  6449. {
  6450. setEnum(p, "@status", status, resultStatuses);
  6451. if (status==ResultStatusUndefined)
  6452. p->removeProp("Value");
  6453. }
  6454. void CLocalWUResult::setResultName(const char *s)
  6455. {
  6456. p->setProp("@name", s);
  6457. }
  6458. void CLocalWUResult::setResultSequence(unsigned seq)
  6459. {
  6460. p->setPropInt("@sequence", seq);
  6461. }
  6462. void CLocalWUResult::setResultSchemaRaw(unsigned size, const void *schema)
  6463. {
  6464. p->setPropBin("SchemaRaw", size, schema);
  6465. }
  6466. void CLocalWUResult::setResultScalar(bool isScalar)
  6467. {
  6468. p->setPropInt("@isScalar", (int) isScalar);
  6469. if (isScalar)
  6470. setResultTotalRowCount(1);
  6471. }
  6472. void CLocalWUResult::setResultRaw(unsigned len, const void *data, WUResultFormat format)
  6473. {
  6474. p->setPropBin("Value", len, data);
  6475. setResultStatus(ResultStatusSupplied);
  6476. setResultFormat(format);
  6477. }
  6478. void CLocalWUResult::setResultFormat(WUResultFormat format)
  6479. {
  6480. switch (format)
  6481. {
  6482. case ResultFormatXml:
  6483. p->setProp("@format","xml");
  6484. break;
  6485. case ResultFormatXmlSet:
  6486. p->setProp("@format","xmlSet");
  6487. break;
  6488. case ResultFormatCsv:
  6489. p->setProp("@format","csv");
  6490. break;
  6491. default:
  6492. p->removeProp("@format");
  6493. break;
  6494. }
  6495. }
  6496. void CLocalWUResult::setResultXML(const char *val)
  6497. {
  6498. p->setProp("xmlValue", val);
  6499. }
  6500. void CLocalWUResult::addResultRaw(unsigned len, const void *data, WUResultFormat format)
  6501. {
  6502. p->appendPropBin("Value", len, data);
  6503. setResultStatus(ResultStatusPartial);
  6504. const char *existingFormat = p->queryProp("@format");
  6505. const char *formatStr = NULL;
  6506. switch (format)
  6507. {
  6508. case ResultFormatXml:
  6509. formatStr = "xml";
  6510. break;
  6511. case ResultFormatXmlSet:
  6512. formatStr = "xmlSet";
  6513. break;
  6514. case ResultFormatCsv:
  6515. formatStr = "csv";
  6516. break;
  6517. default:
  6518. p->removeProp("@format");
  6519. break;
  6520. }
  6521. if (format)
  6522. {
  6523. if (existingFormat)
  6524. {
  6525. if (0 != stricmp(formatStr, existingFormat))
  6526. throw MakeStringException(WUERR_ResultFormatMismatch, "addResult format %s, does not match existing format %s", formatStr, existingFormat);
  6527. }
  6528. else
  6529. p->setProp("@format", formatStr);
  6530. }
  6531. }
  6532. void CLocalWUResult::setResultFetchSize(unsigned rows)
  6533. {
  6534. p->setPropInt("fetchSize", rows);
  6535. }
  6536. void CLocalWUResult::setResultTotalRowCount(__int64 rows)
  6537. {
  6538. p->setPropInt64("totalRowCount", rows);
  6539. }
  6540. void CLocalWUResult::setResultRowCount(__int64 rows)
  6541. {
  6542. p->setPropInt64("rowCount", rows);
  6543. }
  6544. void CLocalWUResult::setResultDataset(const char *ecl, const char *defs)
  6545. {
  6546. p->setProp("datasetEcl", ecl);
  6547. p->setProp("datasetEclDefs", defs);
  6548. }
  6549. void CLocalWUResult::setResultLogicalName(const char *logicalName)
  6550. {
  6551. p->setProp("logicalName", logicalName);
  6552. }
  6553. void CLocalWUResult::setResultKeyField(const char *ecl)
  6554. {
  6555. p->setProp("keyField", ecl);
  6556. }
  6557. void CLocalWUResult::setResultRequestedRows(unsigned rows)
  6558. {
  6559. p->setPropInt("requestedRows", rows);
  6560. }
  6561. void CLocalWUResult::setResultRecordSizeEntry(const char * entry)
  6562. {
  6563. p->setProp("@recordSizeEntry", entry);
  6564. }
  6565. void CLocalWUResult::setResultTransformerEntry(const char * entry)
  6566. {
  6567. p->setProp("@transformerEntry", entry);
  6568. }
  6569. void CLocalWUResult::setResultRowLimit(__int64 value)
  6570. {
  6571. p->setPropInt64("@rowLimit", value);
  6572. }
  6573. void CLocalWUResult::setResultFilename(const char * name)
  6574. {
  6575. p->setProp("@tempFilename", name);
  6576. }
  6577. // MORE - it's an undetected error if we call getResult... of a type that does not match schema
  6578. __int64 CLocalWUResult::getResultInt() const
  6579. {
  6580. __int64 result = 0;
  6581. MemoryBuffer s;
  6582. p->getPropBin("Value", s);
  6583. if (s.length())
  6584. s.read(result);
  6585. else
  6586. result = p->getPropInt64("xmlValue");
  6587. return result;
  6588. }
  6589. bool CLocalWUResult::getResultBool() const
  6590. {
  6591. bool result = false;
  6592. MemoryBuffer s;
  6593. p->getPropBin("Value", s);
  6594. if (s.length())
  6595. s.read(result);
  6596. else
  6597. result = p->getPropBool("xmlValue");
  6598. return result;
  6599. }
  6600. double CLocalWUResult::getResultReal() const
  6601. {
  6602. double result = 0;
  6603. MemoryBuffer s;
  6604. p->getPropBin("Value", s);
  6605. if (s.length())
  6606. s.read(result);
  6607. else
  6608. {
  6609. const char *xmlVal = p->queryProp("xmlValue");
  6610. if (xmlVal)
  6611. result = atof(xmlVal);
  6612. }
  6613. return result;
  6614. }
  6615. void CLocalWUResult::getResultDecimal(void * val, unsigned len, unsigned precision, bool isSigned) const
  6616. {
  6617. MemoryBuffer s;
  6618. p->getPropBin("Value", s);
  6619. if (s.length())
  6620. {
  6621. assertex(s.length() == len);
  6622. s.read(len, val);
  6623. }
  6624. else
  6625. {
  6626. const char *xmlVal = p->queryProp("xmlValue");
  6627. if (xmlVal)
  6628. {
  6629. Decimal d;
  6630. d.setString(strlen(xmlVal), xmlVal);
  6631. if (isSigned)
  6632. d.getDecimal(len, precision, val);
  6633. else
  6634. d.getUDecimal(len, precision, val);
  6635. }
  6636. else
  6637. memset(val, 0, len);
  6638. }
  6639. }
  6640. IStringVal& CLocalWUResult::getResultString(IStringVal & str) const
  6641. {
  6642. MemoryBuffer s;
  6643. p->getPropBin("Value", s);
  6644. if (s.length())
  6645. {
  6646. unsigned len;
  6647. s.read(len);
  6648. str.setLen((const char *) s.readDirect(len), len);
  6649. }
  6650. else
  6651. {
  6652. p->getPropBin("xmlValue", s);
  6653. if (p->isBinary("xmlValue"))
  6654. str.setLen(s.toByteArray(), s.length());
  6655. else
  6656. {
  6657. char *ascii = rtlUtf8ToVStr(rtlUtf8Length(s.length(), s.toByteArray()), s.toByteArray());
  6658. str.set(ascii);
  6659. rtlFree(ascii);
  6660. }
  6661. }
  6662. return str;
  6663. }
  6664. WUResultFormat CLocalWUResult::getResultFormat() const
  6665. {
  6666. const char * format = p->queryProp("@format");
  6667. if (!format)
  6668. return ResultFormatRaw;
  6669. else if (strcmp(format, "xml") == 0)
  6670. return ResultFormatXml;
  6671. else if (strcmp(format, "xmlSet") == 0)
  6672. return ResultFormatXmlSet;
  6673. else if (strcmp(format, "csv") == 0)
  6674. return ResultFormatCsv;
  6675. else
  6676. throw MakeStringException(WUERR_InvalidResultFormat, "Unrecognised result format %s", format);
  6677. }
  6678. IDataVal& CLocalWUResult::getResultRaw(IDataVal & data, IXmlToRawTransformer * xmlTransformer, ICsvToRawTransformer * csvTransformer) const
  6679. {
  6680. MemoryBuffer s;
  6681. p->getPropBin("Value", s);
  6682. unsigned len = s.length();
  6683. if (len)
  6684. {
  6685. WUResultFormat format = getResultFormat();
  6686. if (format == ResultFormatXml || format == ResultFormatXmlSet)
  6687. {
  6688. if (!xmlTransformer)
  6689. throw MakeStringException(WUERR_MissingFormatTranslator, "No transformer supplied to translate XML format result");
  6690. xmlTransformer->transform(data, len, s.readDirect(len), format == ResultFormatXml);
  6691. }
  6692. else if (format == ResultFormatCsv)
  6693. {
  6694. if (!csvTransformer)
  6695. throw MakeStringException(WUERR_MissingFormatTranslator, "No transformer supplied to translate Csv format result");
  6696. csvTransformer->transform(data, len, s.readDirect(len), true);
  6697. }
  6698. else
  6699. data.setLen(s.readDirect(len), len);
  6700. }
  6701. else
  6702. data.clear();
  6703. return data;
  6704. }
  6705. unsigned CLocalWUResult::getResultHash() const
  6706. {
  6707. MemoryBuffer s;
  6708. p->getPropBin("Value", s);
  6709. unsigned len = s.length();
  6710. const byte * data = (const byte *)s.toByteArray();
  6711. return ~hashc(data, len, ~0);
  6712. }
  6713. IDataVal& CLocalWUResult::getResultUnicode(IDataVal & data) const
  6714. {
  6715. MemoryBuffer s;
  6716. p->getPropBin("Value", s);
  6717. if (s.length())
  6718. {
  6719. unsigned len;
  6720. s.read(len);
  6721. data.setLen(s.readDirect(len*2), len*2);
  6722. }
  6723. else
  6724. {
  6725. StringBuffer utf8;
  6726. if (p->getProp("xmlValue", utf8))
  6727. {
  6728. unsigned outlen;
  6729. UChar *out;
  6730. rtlUtf8ToUnicodeX(outlen, out, utf8.length(), utf8.str());
  6731. data.setLen(out, outlen*2);
  6732. rtlFree(out);
  6733. }
  6734. else
  6735. data.clear();
  6736. }
  6737. return data;
  6738. }
  6739. __int64 CLocalWUResult::getResultRawSize(IXmlToRawTransformer * xmlTransformer, ICsvToRawTransformer * csvTransformer) const
  6740. {
  6741. WUResultFormat format = getResultFormat();
  6742. if (format == ResultFormatRaw)
  6743. {
  6744. //MORE: This should not load the whole property...
  6745. MemoryBuffer s;
  6746. p->getPropBin("Value", s);
  6747. return s.length();
  6748. }
  6749. else
  6750. {
  6751. MemoryBuffer temp;
  6752. MemoryBuffer2IDataVal adaptor(temp);
  6753. getResultRaw(adaptor, xmlTransformer, csvTransformer);
  6754. return temp.length();
  6755. }
  6756. }
  6757. IDataVal& CLocalWUResult::getResultRaw(IDataVal & data, __int64 from, __int64 length, IXmlToRawTransformer * xmlTransformer, ICsvToRawTransformer * csvTransformer) const
  6758. {
  6759. WUResultFormat format = getResultFormat();
  6760. if (format != ResultFormatRaw)
  6761. {
  6762. MemoryBuffer temp;
  6763. MemoryBuffer2IDataVal adaptor(temp);
  6764. getResultRaw(adaptor, xmlTransformer, csvTransformer);
  6765. unsigned len = temp.length();
  6766. if (from > len) from = len;
  6767. if (from + length > len) length = len - from;
  6768. data.setLen(temp.readDirect(len) + from, (size32_t)length);
  6769. return data;
  6770. }
  6771. else
  6772. {
  6773. //MORE: This should not load the whole property, and should be different from the code above...
  6774. MemoryBuffer s;
  6775. p->getPropBin("Value", s);
  6776. unsigned len = s.length();
  6777. if (from > len) from = len;
  6778. if (from + length > len) length = len - from;
  6779. data.setLen(s.readDirect(len) + from, (size32_t)length);
  6780. return data;
  6781. }
  6782. }
  6783. bool CLocalWUResult::getResultIsAll() const
  6784. {
  6785. return p->getPropBool("@isAll", false);
  6786. }
  6787. // MORE - it's an undetected error if we call setResult... of a type that does not match schema
  6788. void CLocalWUResult::setResultInt(__int64 val)
  6789. {
  6790. // Note: we always serialize scalar integer results as int8, and schema must reflect this
  6791. MemoryBuffer m;
  6792. serializeInt8(val, m);
  6793. p->setPropBin("Value", m.length(), m.toByteArray());
  6794. setResultRowCount(1);
  6795. setResultTotalRowCount(1);
  6796. }
  6797. void CLocalWUResult::setResultUInt(unsigned __int64 val)
  6798. {
  6799. setResultInt((__int64) val);
  6800. }
  6801. void CLocalWUResult::setResultReal(double val)
  6802. {
  6803. // Note: we always serialize scalar real results as real8, and schema must reflect this
  6804. MemoryBuffer m;
  6805. serializeReal8(val, m);
  6806. p->setPropBin("Value", m.length(), m.toByteArray());
  6807. setResultRowCount(1);
  6808. setResultTotalRowCount(1);
  6809. }
  6810. void CLocalWUResult::setResultBool(bool val)
  6811. {
  6812. MemoryBuffer m;
  6813. serializeBool(val, m);
  6814. p->setPropBin("Value", m.length(), m.toByteArray());
  6815. setResultRowCount(1);
  6816. setResultTotalRowCount(1);
  6817. }
  6818. void CLocalWUResult::setResultString(const char *val, unsigned len)
  6819. {
  6820. // Note: we always serialize scalar strings with length prefix, and schema must reflect this
  6821. MemoryBuffer m;
  6822. serializeLPString(len, val, m);
  6823. p->setPropBin("Value", m.length(), m.toByteArray());
  6824. setResultRowCount(1);
  6825. setResultTotalRowCount(1);
  6826. }
  6827. void CLocalWUResult::setResultUnicode(const void *val, unsigned len)
  6828. {
  6829. // Note: we always serialize scalar strings with length prefix, and schema must reflect this
  6830. MemoryBuffer m;
  6831. m.append(len).append(len*2, val);
  6832. p->setPropBin("Value", m.length(), m.toByteArray());
  6833. setResultRowCount(1);
  6834. setResultTotalRowCount(1);
  6835. }
  6836. void CLocalWUResult::setResultData(const void *val, unsigned len)
  6837. {
  6838. // Note: we always serialize scalar data with length prefix, and schema must reflect this
  6839. MemoryBuffer m;
  6840. serializeLPString(len, (const char *)val, m);
  6841. p->setPropBin("Value", m.length(), m.toByteArray());
  6842. setResultRowCount(1);
  6843. setResultTotalRowCount(1);
  6844. }
  6845. void CLocalWUResult::setResultDecimal(const void *val, unsigned len)
  6846. {
  6847. // Note: serialized as data but with length known from schema
  6848. MemoryBuffer m;
  6849. serializeFixedData(len, val, m);
  6850. p->setPropBin("Value", m.length(), m.toByteArray());
  6851. setResultRowCount(1);
  6852. setResultTotalRowCount(1);
  6853. }
  6854. void CLocalWUResult::setResultRow(unsigned len, const void * data)
  6855. {
  6856. p->setPropBin("Value", len, data);
  6857. setResultRowCount(1);
  6858. setResultTotalRowCount(1);
  6859. setResultFormat(ResultFormatRaw);
  6860. }
  6861. void CLocalWUResult::setResultIsAll(bool value)
  6862. {
  6863. p->setPropBool("@isAll", value);
  6864. }
  6865. //==========================================================================================
  6866. CLocalWUPlugin::CLocalWUPlugin(IPropertyTree *props) : p(props)
  6867. {
  6868. }
  6869. IStringVal& CLocalWUPlugin::getPluginName(IStringVal &str) const
  6870. {
  6871. str.set(p->queryProp("@dllname"));
  6872. return str;
  6873. }
  6874. IStringVal& CLocalWUPlugin::getPluginVersion(IStringVal &str) const
  6875. {
  6876. str.set(p->queryProp("@version"));
  6877. return str;
  6878. }
  6879. bool CLocalWUPlugin::getPluginThor() const
  6880. {
  6881. return p->getPropInt("@thor") != 0;
  6882. }
  6883. bool CLocalWUPlugin::getPluginHole() const
  6884. {
  6885. return p->getPropInt("@hole") != 0;
  6886. }
  6887. void CLocalWUPlugin::setPluginName(const char *str)
  6888. {
  6889. p->setProp("@dllname", str);
  6890. }
  6891. void CLocalWUPlugin::setPluginVersion(const char *str)
  6892. {
  6893. p->setProp("@version", str);
  6894. }
  6895. void CLocalWUPlugin::setPluginThor(bool on)
  6896. {
  6897. p->setPropInt("@thor", (int) on);
  6898. }
  6899. void CLocalWUPlugin::setPluginHole(bool on)
  6900. {
  6901. p->setPropInt("@hole", (int) on);
  6902. }
  6903. //==========================================================================================
  6904. class WULibraryActivityIterator : public CInterface, implements IConstWULibraryActivityIterator
  6905. {
  6906. public:
  6907. WULibraryActivityIterator(IPropertyTree * tree) { iter.setown(tree->getElements("activity")); }
  6908. IMPLEMENT_IINTERFACE;
  6909. bool first() { return iter->first(); }
  6910. bool isValid() { return iter->isValid(); }
  6911. bool next() { return iter->next(); }
  6912. unsigned query() const { return iter->query().getPropInt("@id"); }
  6913. private:
  6914. Owned<IPropertyTreeIterator> iter;
  6915. };
  6916. CLocalWULibrary::CLocalWULibrary(IPropertyTree *props) : p(props)
  6917. {
  6918. }
  6919. IStringVal& CLocalWULibrary::getName(IStringVal &str) const
  6920. {
  6921. str.set(p->queryProp("@name"));
  6922. return str;
  6923. }
  6924. IConstWULibraryActivityIterator * CLocalWULibrary::getActivities() const
  6925. {
  6926. return new WULibraryActivityIterator(p);
  6927. }
  6928. void CLocalWULibrary::setName(const char *str)
  6929. {
  6930. p->setProp("@name", str);
  6931. }
  6932. void CLocalWULibrary::addActivity(unsigned id)
  6933. {
  6934. StringBuffer s;
  6935. s.append("activity[@id=\"").append(id).append("\"]");
  6936. if (!p->hasProp(s.str()))
  6937. p->addPropTree("activity", createPTree())->setPropInt("@id", id);
  6938. }
  6939. //==========================================================================================
  6940. CLocalWUActivity::CLocalWUActivity(IPropertyTree *props, __int64 id) : p(props)
  6941. {
  6942. if (id)
  6943. p->setPropInt64("@id", id);
  6944. }
  6945. __int64 CLocalWUActivity::getId() const
  6946. {
  6947. return p->getPropInt64("@id");
  6948. }
  6949. unsigned CLocalWUActivity::getKind() const
  6950. {
  6951. return (unsigned)p->getPropInt("@kind");
  6952. }
  6953. IStringVal & CLocalWUActivity::getHelper(IStringVal & str) const
  6954. {
  6955. str.set(p->queryProp("@helper"));
  6956. return str;
  6957. }
  6958. void CLocalWUActivity::setKind(unsigned kind)
  6959. {
  6960. p->setPropInt64("@kind", kind);
  6961. }
  6962. void CLocalWUActivity::setHelper(const char * str)
  6963. {
  6964. p->setProp("@helper", str);
  6965. }
  6966. //==========================================================================================
  6967. CLocalWUException::CLocalWUException(IPropertyTree *props) : p(props)
  6968. {
  6969. }
  6970. IStringVal& CLocalWUException::getExceptionSource(IStringVal &str) const
  6971. {
  6972. str.set(p->queryProp("@source"));
  6973. return str;
  6974. }
  6975. IStringVal& CLocalWUException::getExceptionMessage(IStringVal &str) const
  6976. {
  6977. str.set(p->queryProp(NULL));
  6978. return str;
  6979. }
  6980. unsigned CLocalWUException::getExceptionCode() const
  6981. {
  6982. return p->getPropInt("@code", 0);
  6983. }
  6984. WUExceptionSeverity CLocalWUException::getSeverity() const
  6985. {
  6986. return (WUExceptionSeverity)p->getPropInt("@severity", ExceptionSeverityError);
  6987. }
  6988. IStringVal & CLocalWUException::getTimeStamp(IStringVal & dt) const
  6989. {
  6990. dt.set(p->queryProp("@time"));
  6991. return dt;
  6992. }
  6993. IStringVal & CLocalWUException::getExceptionFileName(IStringVal & str) const
  6994. {
  6995. str.set(p->queryProp("@filename"));
  6996. return str;
  6997. }
  6998. unsigned CLocalWUException::getExceptionLineNo() const
  6999. {
  7000. return p->getPropInt("@row", 0);
  7001. }
  7002. unsigned CLocalWUException::getExceptionColumn() const
  7003. {
  7004. return p->getPropInt("@col", 0);
  7005. }
  7006. void CLocalWUException::setExceptionSource(const char *str)
  7007. {
  7008. p->setProp("@source", str);
  7009. }
  7010. void CLocalWUException::setExceptionMessage(const char *str)
  7011. {
  7012. p->setProp(NULL, str);
  7013. }
  7014. void CLocalWUException::setExceptionCode(unsigned code)
  7015. {
  7016. p->setPropInt("@code", code);
  7017. }
  7018. void CLocalWUException::setSeverity(WUExceptionSeverity level)
  7019. {
  7020. p->setPropInt("@severity", level);
  7021. }
  7022. void CLocalWUException::setTimeStamp(const char *str)
  7023. {
  7024. p->setProp("@time", str);
  7025. }
  7026. void CLocalWUException::setExceptionFileName(const char *str)
  7027. {
  7028. p->setProp("@filename", str);
  7029. }
  7030. void CLocalWUException::setExceptionLineNo(unsigned r)
  7031. {
  7032. p->setPropInt("@row", r);
  7033. }
  7034. void CLocalWUException::setExceptionColumn(unsigned c)
  7035. {
  7036. p->setPropInt("@col", c);
  7037. }
  7038. CLocalWUTimeStamp::CLocalWUTimeStamp(IPropertyTree *props) : p(props)
  7039. {
  7040. }
  7041. IStringVal & CLocalWUTimeStamp::getApplication(IStringVal & str) const
  7042. {
  7043. str.set(p->queryProp("@application"));
  7044. return str;
  7045. }
  7046. IStringVal & CLocalWUTimeStamp::getEvent(IStringVal & str) const
  7047. {
  7048. IPropertyTree* evt=p->queryPropTree("*[1]");
  7049. if(evt)
  7050. str.set(evt->queryName());
  7051. return str;
  7052. }
  7053. IStringVal & CLocalWUTimeStamp::getDate(IStringVal & str) const
  7054. {
  7055. str.set(p->queryProp("*[1]"));
  7056. return str;
  7057. }
  7058. CLocalWUAppValue::CLocalWUAppValue(IPropertyTree *props,unsigned child): p(props)
  7059. {
  7060. prop.append("*[").append(child).append("]");
  7061. }
  7062. IStringVal & CLocalWUAppValue::getApplication(IStringVal & str) const
  7063. {
  7064. str.set(p->queryName());
  7065. return str;
  7066. }
  7067. IStringVal & CLocalWUAppValue::getName(IStringVal & str) const
  7068. {
  7069. IPropertyTree* val=p->queryPropTree(prop.str());
  7070. if(val)
  7071. str.set(val->queryName());
  7072. return str;
  7073. }
  7074. IStringVal & CLocalWUAppValue::getValue(IStringVal & str) const
  7075. {
  7076. str.set(p->queryProp(prop.str()));
  7077. return str;
  7078. }
  7079. extern WORKUNIT_API ILocalWorkUnit * createLocalWorkUnit()
  7080. {
  7081. Owned<CLocalWorkUnit> cw = new CLocalWorkUnit("W_LOCAL", NULL, (ISecManager*)NULL, NULL);
  7082. ILocalWorkUnit* ret = QUERYINTERFACE(&cw->lockRemote(false), ILocalWorkUnit);
  7083. return ret;
  7084. }
  7085. extern WORKUNIT_API StringBuffer &exportWorkUnitToXML(const IConstWorkUnit *wu, StringBuffer &str)
  7086. {
  7087. const CLocalWorkUnit *w = QUERYINTERFACE(wu, const CLocalWorkUnit);
  7088. if (!w)
  7089. {
  7090. const CLockedWorkUnit *wl = QUERYINTERFACE(wu, const CLockedWorkUnit);
  7091. if (wl)
  7092. w = wl->c;
  7093. }
  7094. if (w)
  7095. toXML(w->p, str, 0, XML_Format|XML_SortTags);
  7096. else
  7097. str.append("Unrecognized workunit format");
  7098. return str;
  7099. }
  7100. extern WORKUNIT_API IStringVal& exportWorkUnitToXML(const IConstWorkUnit *wu, IStringVal &str)
  7101. {
  7102. StringBuffer x;
  7103. str.set(exportWorkUnitToXML(wu,x).str());
  7104. return str;
  7105. }
  7106. extern WORKUNIT_API void exportWorkUnitToXMLFile(const IConstWorkUnit *wu, const char * filename, unsigned extraXmlFlags)
  7107. {
  7108. const CLocalWorkUnit *w = QUERYINTERFACE(wu, const CLocalWorkUnit);
  7109. if (!w)
  7110. {
  7111. const CLockedWorkUnit *wl = QUERYINTERFACE(wu, const CLockedWorkUnit);
  7112. if (wl)
  7113. w = wl->c;
  7114. }
  7115. if (w)
  7116. saveXML(filename, w->p, 0, XML_Format|XML_SortTags|extraXmlFlags);
  7117. }
  7118. extern WORKUNIT_API void submitWorkUnit(const char *wuid, const char *username, const char *password)
  7119. {
  7120. MemoryBuffer buffer;
  7121. Owned<INamedQueueConnection> conn = createNamedQueueConnection(0); // MORE - security token?
  7122. Owned<IWorkUnitFactory> factory = getWorkUnitFactory();
  7123. Owned<IWorkUnit> workunit = factory->updateWorkUnit(wuid);
  7124. assertex(workunit);
  7125. SCMStringBuffer token;
  7126. createToken(wuid, username, password, token);
  7127. workunit->setSecurityToken(token.str());
  7128. SCMStringBuffer clusterName;
  7129. workunit->getClusterName(clusterName);
  7130. if (!clusterName.length())
  7131. throw MakeStringException(WUERR_InvalidCluster, "No target cluster specified");
  7132. workunit->commit();
  7133. workunit.clear();
  7134. Owned<IConstWUClusterInfo> clusterInfo = getTargetClusterInfo(clusterName.str());
  7135. if (!clusterInfo)
  7136. throw MakeStringException(WUERR_InvalidCluster, "Unknown cluster %s", clusterName.str());
  7137. SCMStringBuffer serverQueue;
  7138. clusterInfo->getServerQueue(serverQueue);
  7139. assertex(serverQueue.length());
  7140. Owned<IJobQueue> queue = createJobQueue(serverQueue.str());
  7141. if (!queue.get())
  7142. throw MakeStringException(WUERR_InvalidQueue, "Could not create workunit queue");
  7143. IJobQueueItem *item = createJobQueueItem(wuid);
  7144. queue->enqueue(item);
  7145. }
  7146. extern WORKUNIT_API void abortWorkUnit(const char *wuid)
  7147. {
  7148. StringBuffer xpath("/WorkUnitAborts/");
  7149. xpath.append(wuid);
  7150. Owned<IRemoteConnection> acon = querySDS().connect(xpath.str(), myProcessSession(), RTM_LOCK_WRITE|RTM_CREATE, SDS_LOCK_TIMEOUT);
  7151. acon->queryRoot()->setPropInt(NULL, 1);
  7152. }
  7153. extern WORKUNIT_API void secSubmitWorkUnit(const char *wuid, ISecManager &secmgr, ISecUser &secuser)
  7154. {
  7155. if (checkWuSecAccess(wuid, secmgr, &secuser, SecAccess_Write, "Submit", true, true))
  7156. submitWorkUnit(wuid, secuser.getName(), secuser.credentials().getPassword());
  7157. }
  7158. extern WORKUNIT_API void secAbortWorkUnit(const char *wuid, ISecManager &secmgr, ISecUser &secuser)
  7159. {
  7160. if (checkWuSecAccess(wuid, secmgr, &secuser, SecAccess_Write, "Submit", true, true))
  7161. abortWorkUnit(wuid);
  7162. }
  7163. extern WORKUNIT_API void submitWorkUnit(const char *wuid, ISecManager *secmgr, ISecUser *secuser)
  7164. {
  7165. if (secmgr && secuser)
  7166. return secSubmitWorkUnit(wuid, *secmgr, *secuser);
  7167. if (secuser)
  7168. return submitWorkUnit(wuid, secuser->getName(), secuser->credentials().getPassword());
  7169. submitWorkUnit(wuid, "", "");
  7170. }
  7171. extern WORKUNIT_API void abortWorkUnit(const char *wuid, ISecManager *secmgr, ISecUser *secuser)
  7172. {
  7173. if (secmgr && secuser)
  7174. return secAbortWorkUnit(wuid, *secmgr, *secuser);
  7175. abortWorkUnit(wuid);
  7176. }
  7177. bool CLocalWorkUnit::hasWorkflow() const
  7178. {
  7179. return p->hasProp("Workflow");
  7180. }
  7181. unsigned CLocalWorkUnit::queryEventScheduledCount() const
  7182. {
  7183. CriticalBlock block(crit);
  7184. return p->getPropInt("Workflow/@eventScheduledCount", 0);
  7185. }
  7186. void CLocalWorkUnit::incEventScheduledCount()
  7187. {
  7188. CriticalBlock block(crit);
  7189. p->setPropInt("Workflow/@eventScheduledCount", p->getPropInt("Workflow/@eventScheduledCount", 0)+1);
  7190. }
  7191. IPropertyTree * CLocalWorkUnit::queryWorkflowTree() const
  7192. {
  7193. CriticalBlock block(crit);
  7194. return p->queryPropTree("Workflow");
  7195. }
  7196. IConstWorkflowItemIterator* CLocalWorkUnit::getWorkflowItems() const
  7197. {
  7198. // For this to be legally called, we must have the read-able interface. So we are already locked for (at least) read.
  7199. CriticalBlock block(crit);
  7200. if(!workflowIteratorCached)
  7201. {
  7202. assertex(!workflowIterator);
  7203. Owned<IPropertyTree> s = p->getPropTree("Workflow");
  7204. if(s)
  7205. workflowIterator.setown(createWorkflowItemIterator(s));
  7206. workflowIteratorCached = true;
  7207. }
  7208. return workflowIterator.getLink();
  7209. }
  7210. IWorkflowItemArray * CLocalWorkUnit::getWorkflowClone() const
  7211. {
  7212. unsigned count = 0;
  7213. Owned<IConstWorkflowItemIterator> iter = getWorkflowItems();
  7214. for(iter->first(); iter->isValid(); iter->next())
  7215. count++;
  7216. Owned<IWorkflowItemArray> array = createWorkflowItemArray(count);
  7217. for(iter->first(); iter->isValid(); iter->next())
  7218. array->addClone(iter->query());
  7219. return array.getLink();
  7220. }
  7221. IWorkflowItem * CLocalWorkUnit::addWorkflowItem(unsigned wfid, WFType type, WFMode mode, unsigned success, unsigned failure, unsigned recovery, unsigned retriesAllowed, unsigned contingencyFor)
  7222. {
  7223. // For this to be legally called, we must have the write-able interface. So we are already locked for write.
  7224. CriticalBlock block(crit);
  7225. workflowIterator.clear();
  7226. workflowIteratorCached = false;
  7227. IPropertyTree * s = p->queryPropTree("Workflow");
  7228. if(!s)
  7229. s = p->addPropTree("Workflow", createPTree("Workflow"));
  7230. return createWorkflowItem(s, wfid, type, mode, success, failure, recovery, retriesAllowed, contingencyFor);
  7231. }
  7232. IWorkflowItemIterator * CLocalWorkUnit::updateWorkflowItems()
  7233. {
  7234. // For this to be legally called, we must have the write-able interface. So we are already locked for write.
  7235. CriticalBlock block(crit);
  7236. if(!workflowIterator)
  7237. {
  7238. IPropertyTree * s = p->queryPropTree("Workflow");
  7239. if(!s)
  7240. s = p->addPropTree("Workflow", createPTree("Workflow"));
  7241. workflowIterator.setown(createWorkflowItemIterator(s));
  7242. workflowIteratorCached = true;
  7243. }
  7244. return workflowIterator.getLink();
  7245. }
  7246. void CLocalWorkUnit::syncRuntimeWorkflow(IWorkflowItemArray * array)
  7247. {
  7248. Owned<IWorkflowItemIterator> iter = updateWorkflowItems();
  7249. Owned<IWorkflowItem> item;
  7250. for(iter->first(); iter->isValid(); iter->next())
  7251. {
  7252. item.setown(iter->get());
  7253. item->syncRuntimeData(array->queryWfid(item->queryWfid()));
  7254. }
  7255. workflowIterator.clear();
  7256. workflowIteratorCached = false;
  7257. }
  7258. void CLocalWorkUnit::resetWorkflow()
  7259. {
  7260. if (hasWorkflow())
  7261. {
  7262. Owned<IWorkflowItemIterator> iter = updateWorkflowItems();
  7263. Owned<IWorkflowItem> wf;
  7264. for(iter->first(); iter->isValid(); iter->next())
  7265. {
  7266. wf.setown(iter->get());
  7267. wf->reset();
  7268. }
  7269. workflowIterator.clear();
  7270. workflowIteratorCached = false;
  7271. }
  7272. }
  7273. void CLocalWorkUnit::schedule()
  7274. {
  7275. CriticalBlock block(crit);
  7276. if(queryEventScheduledCount() == 0) return;
  7277. switch(getState())
  7278. {
  7279. case WUStateCompleted:
  7280. setState(WUStateWait);
  7281. break;
  7282. case WUStateFailed:
  7283. case WUStateArchived:
  7284. case WUStateAborting:
  7285. case WUStateAborted:
  7286. case WUStateScheduled:
  7287. throw MakeStringException(WUERR_CannotSchedule, "Cannot schedule workunit in this state");
  7288. }
  7289. StringBuffer rootPath;
  7290. SCMStringBuffer clusterName;
  7291. getClusterName(clusterName);
  7292. rootPath.append("/Schedule/").append(clusterName.str());
  7293. Owned<IRemoteConnection> conn = querySDS().connect(rootPath.str(), myProcessSession(), RTM_LOCK_WRITE | RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT);
  7294. Owned<IPropertyTree> root = conn->getRoot();
  7295. if(!root->hasChildren())
  7296. {
  7297. StringBuffer addPath;
  7298. addPath.append("/Schedulers/").append(clusterName.str());
  7299. Owned<IRemoteConnection> addConn = querySDS().connect(addPath.str(), myProcessSession(), RTM_LOCK_WRITE | RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT);
  7300. }
  7301. char const * wuid = p->queryName();
  7302. StringBuffer xpath("*/*/");
  7303. ncnameEscape(wuid, xpath);
  7304. bool more;
  7305. do more = root->removeProp(xpath.str()); while(more);
  7306. Owned<IConstWorkflowItemIterator> iter = getWorkflowItems();
  7307. Owned<IWorkflowEvent> event;
  7308. Owned<IPropertyTree> branch1, branch2;
  7309. for(iter->first(); iter->isValid(); iter->next())
  7310. {
  7311. event.setown(iter->query()->getScheduleEvent());
  7312. if(!event) continue;
  7313. ncnameEscape(event->queryName(), xpath.clear());
  7314. ensurePTree(root, xpath.str());
  7315. branch1.setown(root->getPropTree(xpath.str()));
  7316. ncnameEscape(event->queryText(), xpath.clear());
  7317. ensurePTree(branch1, xpath.str());
  7318. branch2.setown(branch1->getPropTree(xpath.str()));
  7319. ncnameEscape(wuid, xpath.clear());
  7320. ensurePTree(branch2, xpath.str());
  7321. }
  7322. }
  7323. void CLocalWorkUnit::deschedule()
  7324. {
  7325. if(queryEventScheduledCount() == 0) return;
  7326. if(getState() == WUStateWait)
  7327. setState(WUStateCompleted);
  7328. doDescheduleWorkkunit(p->queryName());
  7329. }
  7330. mapEnums localFileUploadTypes[] = {
  7331. { UploadTypeFileSpray, "FileSpray" },
  7332. { UploadTypeWUResult, "WUResult" },
  7333. { UploadTypeWUResultCsv, "WUResultCsv" },
  7334. { UploadTypeWUResultXml, "WUResultXml" },
  7335. { UploadTypeSize, NULL }
  7336. };
  7337. class CLocalFileUpload : public CInterface, implements IConstLocalFileUpload
  7338. {
  7339. public:
  7340. CLocalFileUpload(IPropertyTree * _tree) : tree(_tree) {}
  7341. CLocalFileUpload(unsigned id, LocalFileUploadType type, char const * source, char const * destination, char const * eventTag)
  7342. {
  7343. tree.setown(createPTree());
  7344. tree->setPropInt("@id", id);
  7345. setEnum(tree, "@type", type, localFileUploadTypes);
  7346. tree->setProp("@source", source);
  7347. tree->setProp("@destination", destination);
  7348. if (eventTag)
  7349. tree->setProp("@eventTag", eventTag);
  7350. }
  7351. IMPLEMENT_IINTERFACE;
  7352. IPropertyTree * getTree() { return tree.getLink(); }
  7353. virtual unsigned queryID() const { return tree->getPropInt("@id"); }
  7354. virtual LocalFileUploadType queryType() const { return (LocalFileUploadType)getEnum(tree, "@type", localFileUploadTypes); }
  7355. virtual IStringVal & getSource(IStringVal & ret) const { ret.set(tree->queryProp("@source")); return ret; }
  7356. virtual IStringVal & getDestination(IStringVal & ret) const { ret.set(tree->queryProp("@destination")); return ret; }
  7357. virtual IStringVal & getEventTag(IStringVal & ret) const { if(tree->hasProp("@eventTag")) ret.set(tree->queryProp("@eventTag")); else ret.clear(); return ret; }
  7358. private:
  7359. Owned<IPropertyTree> tree;
  7360. };
  7361. class CLocalFileUploadIterator : public CInterface, implements IConstLocalFileUploadIterator
  7362. {
  7363. public:
  7364. CLocalFileUploadIterator(IPropertyTree * _tree) : tree(_tree), iter(tree->getElements("LocalFileUpload")) {}
  7365. IMPLEMENT_IINTERFACE;
  7366. bool first() { return iter->first(); }
  7367. bool isValid() { return iter->isValid(); }
  7368. bool next() { return iter->next(); }
  7369. IConstLocalFileUpload * get() { return new CLocalFileUpload(&iter->get()); }
  7370. private:
  7371. Owned<IPropertyTree> tree;
  7372. Owned<IPropertyTreeIterator> iter;
  7373. };
  7374. IConstLocalFileUploadIterator * CLocalWorkUnit::getLocalFileUploads() const
  7375. {
  7376. // For this to be legally called, we must have the read-able interface. So we are already locked for (at least) read.
  7377. CriticalBlock block(crit);
  7378. Owned<IPropertyTree> s = p->getPropTree("LocalFileUploads");
  7379. if(s)
  7380. return new CLocalFileUploadIterator(s.getClear());
  7381. else
  7382. return NULL;
  7383. }
  7384. bool CLocalWorkUnit::requiresLocalFileUpload() const
  7385. {
  7386. SCMStringBuffer dest;
  7387. Owned<IConstWUResult> result;
  7388. Owned<IConstLocalFileUploadIterator> iter(getLocalFileUploads());
  7389. if(!iter)
  7390. return false;
  7391. for(iter->first(); iter->isValid(); iter->next())
  7392. {
  7393. Owned<IConstLocalFileUpload> upload(iter->get());
  7394. switch(upload->queryType())
  7395. {
  7396. case UploadTypeWUResult:
  7397. case UploadTypeWUResultCsv:
  7398. case UploadTypeWUResultXml:
  7399. upload->getDestination(dest);
  7400. result.setown(getResultByName(dest.str()));
  7401. if(!result)
  7402. return true;
  7403. break;
  7404. default:
  7405. throw MakeStringException(WUERR_InvalidUploadFormat, "Unsupported local file upload type %s", getEnumText(upload->queryType(), localFileUploadTypes));
  7406. }
  7407. }
  7408. return false;
  7409. }
  7410. unsigned CLocalWorkUnit::addLocalFileUpload(LocalFileUploadType type, char const * source, char const * destination, char const * eventTag)
  7411. {
  7412. // For this to be legally called, we must have the write-able interface. So we are already locked for write.
  7413. CriticalBlock block(crit);
  7414. IPropertyTree * s = p->queryPropTree("LocalFileUploads");
  7415. if(!s)
  7416. s = p->addPropTree("LocalFileUploads", createPTree());
  7417. unsigned id = s->numChildren();
  7418. Owned<CLocalFileUpload> upload = new CLocalFileUpload(id, type, source, destination, eventTag);
  7419. s->addPropTree("LocalFileUpload", upload->getTree());
  7420. return id;
  7421. }
  7422. #if 0
  7423. void testConstWorkflow(IConstWorkflowItem * cwf, bool * okay, bool * dep)
  7424. {
  7425. DBGLOG("Test workflow const iface %u", cwf->queryWfid());
  7426. unsigned deps = 0;
  7427. Owned<IWorkflowDependencyIterator> diter;
  7428. switch(cwf->queryWfid())
  7429. {
  7430. case 1:
  7431. assertex(!cwf->isScheduled());
  7432. assertex(cwf->queryType() == WFTypeNormal);
  7433. assertex(cwf->queryState() == WFStateNull);
  7434. diter.setown(cwf->getDependencies());
  7435. for(diter->first(); diter->isValid(); diter->next())
  7436. deps++;
  7437. assertex(deps==0);
  7438. okay[0] = true;
  7439. break;
  7440. case 2:
  7441. assertex(!cwf->isScheduled());
  7442. assertex(cwf->queryType() == WFTypeRecovery);
  7443. assertex(cwf->queryState() == WFStateSkip);
  7444. okay[1] = true;
  7445. break;
  7446. case 3:
  7447. assertex(cwf->queryContingencyFor() == 4);
  7448. okay[2] = true;
  7449. break;
  7450. case 4:
  7451. assertex(cwf->isScheduled());
  7452. assertex(cwf->queryType() == WFTypeNormal);
  7453. assertex(cwf->queryState() == WFStateReqd);
  7454. assertex(cwf->querySuccess() == 0);
  7455. assertex(cwf->queryFailure() == 3);
  7456. assertex(cwf->queryRecovery() == 2);
  7457. assertex(cwf->queryRetriesAllowed() == 10);
  7458. assertex(cwf->queryRetriesRemaining() == 10);
  7459. diter.setown(cwf->getDependencies());
  7460. for(diter->first(); diter->isValid(); diter->next())
  7461. {
  7462. dep[diter->query()-1] = true;
  7463. deps++;
  7464. }
  7465. assertex(deps==2);
  7466. assertex(dep[0]);
  7467. assertex(dep[1]);
  7468. okay[3] = true;
  7469. break;
  7470. case 5:
  7471. assertex(cwf->isScheduled());
  7472. assertex(!cwf->isScheduledNow());
  7473. assertex(cwf->querySchedulePriority() == 75);
  7474. assertex(cwf->queryScheduleCount() == 5);
  7475. assertex(cwf->queryScheduleCountRemaining() == 5);
  7476. okay[4] = true;
  7477. break;
  7478. case 6:
  7479. assertex(cwf->isScheduled());
  7480. assertex(!cwf->isScheduledNow());
  7481. assertex(cwf->querySchedulePriority() == 25);
  7482. assertex(!cwf->hasScheduleCount());
  7483. okay[5] = true;
  7484. break;
  7485. default:
  7486. assertex(!"unknown wfid in test");
  7487. }
  7488. }
  7489. void testRuntimeWorkflow(IRuntimeWorkflowItem * rwf, bool * okay)
  7490. {
  7491. DBGLOG("Test workflow runtime iface %u", rwf->queryWfid());
  7492. switch(rwf->queryWfid())
  7493. {
  7494. case 1:
  7495. case 2:
  7496. case 3:
  7497. okay[rwf->queryWfid()-1] = true;
  7498. break;
  7499. case 4:
  7500. {
  7501. unsigned tries = 0;
  7502. while(rwf->testAndDecRetries())
  7503. tries++;
  7504. assertex(tries == 10);
  7505. assertex(rwf->queryRetriesRemaining() == 0);
  7506. rwf->setState(WFStateFail);
  7507. assertex(rwf->queryState() == WFStateFail);
  7508. rwf->reset();
  7509. assertex(rwf->queryRetriesRemaining() == 10);
  7510. assertex(rwf->queryState() == WFStateReqd);
  7511. }
  7512. okay[3] = true;
  7513. break;
  7514. case 5:
  7515. {
  7516. assertex(rwf->queryScheduleCountRemaining() == 5);
  7517. unsigned count = 0;
  7518. do count++; while(rwf->decAndTestScheduleCountRemaining());
  7519. assertex(count == 5);
  7520. assertex(rwf->queryScheduleCountRemaining() == 0);
  7521. rwf->reset();
  7522. assertex(rwf->queryScheduleCountRemaining() == 5);
  7523. }
  7524. okay[4] = true;
  7525. break;
  7526. case 6:
  7527. {
  7528. assertex(!rwf->hasScheduleCount());
  7529. unsigned count;
  7530. for(count=0; count<20; count++)
  7531. assertex(rwf->decAndTestScheduleCountRemaining());
  7532. }
  7533. okay[5] = true;
  7534. break;
  7535. default:
  7536. assertex(!"unknown wfid in test");
  7537. }
  7538. }
  7539. void testWorkflow()
  7540. {
  7541. DBGLOG("workunit.cpp : testWorkflow");
  7542. CLocalWorkUnit wu("W-WF-TEST", 0, 0, 0);
  7543. Owned<IWorkflowItem> wf;
  7544. wf.setown(wu.addWorkflowItem(1, WFTypeNormal, 0, 0, 0, 0, 0));
  7545. wf.setown(wu.addWorkflowItem(2, WFTypeRecovery, 0, 0, 0, 0, 0));
  7546. wf.setown(wu.addWorkflowItem(3, WFTypeFailure, 0, 0, 0, 0, 4));
  7547. wf.setown(wu.addWorkflowItem(4, WFTypeNormal, 0, 3, 2, 10, 0));
  7548. wf->setScheduledNow();
  7549. wf->addDependency(1);
  7550. wf.setown(wu.addWorkflowItem(5, WFTypeNormal, 0, 0, 0, 0, 0));
  7551. wf->setScheduledOn("test", "foo*");
  7552. wf->setSchedulePriority(75);
  7553. wf->setScheduleCount(5);
  7554. wf.setown(wu.addWorkflowItem(6, WFTypeNormal, 0, 0, 0, 0, 0));
  7555. wf->setScheduledOn("test", "bar*");
  7556. wf->setSchedulePriority(25);
  7557. unsigned const n = 6;
  7558. bool okay[n];
  7559. bool dep[n];
  7560. unsigned i;
  7561. for(i=0; i<n; i++)
  7562. okay[i] = dep[i] = 0;
  7563. Owned<IConstWorkflowItemIterator> citer(wu.getWorkflowItems());
  7564. for(citer->first(); citer->isValid(); citer->next())
  7565. testConstWorkflow(citer->query(), okay, dep);
  7566. for(i=0; i<n; i++)
  7567. {
  7568. assertex(okay[i]);
  7569. okay[i] = false;
  7570. }
  7571. Owned<IWorkflowItemIterator> miter(wu.updateWorkflowItems());
  7572. for(miter->first(); miter->isValid(); miter->next())
  7573. {
  7574. Owned<IRuntimeWorkflowItem> rwf(miter->get());
  7575. testRuntimeWorkflow(rwf, okay);
  7576. }
  7577. for(i=0; i<n; i++)
  7578. {
  7579. assertex(okay[i]);
  7580. okay[i] = dep[i] = false;
  7581. }
  7582. Owned<IWorkflowItemArray> array(wu.getWorkflowClone());
  7583. unsigned wfid;
  7584. for(wfid = 1; array->isValid(wfid); wfid++)
  7585. testConstWorkflow(&array->queryWfid(wfid), okay, dep);
  7586. for(i=0; i<n; i++)
  7587. {
  7588. assertex(okay[i]);
  7589. okay[i] = false;
  7590. }
  7591. for(wfid = 1; array->isValid(wfid); wfid++)
  7592. testRuntimeWorkflow(&array->queryWfid(wfid), okay);
  7593. for(i=0; i<n; i++)
  7594. {
  7595. assertex(okay[i]);
  7596. okay[i] = false;
  7597. }
  7598. }
  7599. #endif
  7600. //------------------------------------------------------------------------------------------
  7601. class WorkUnitWaiter : public CInterface, implements ISDSSubscription, implements IAbortHandler
  7602. {
  7603. Semaphore changed;
  7604. SubscriptionId change;
  7605. public:
  7606. IMPLEMENT_IINTERFACE;
  7607. WorkUnitWaiter(const char *xpath)
  7608. {
  7609. change = querySDS().subscribe(xpath, *this, false);
  7610. aborted = false;
  7611. }
  7612. ~WorkUnitWaiter()
  7613. {
  7614. assertex(change==0);
  7615. }
  7616. void notify(SubscriptionId id, const char *xpath, SDSNotifyFlags flags, unsigned valueLen, const void *valueData)
  7617. {
  7618. changed.signal();
  7619. }
  7620. bool wait(unsigned timeout)
  7621. {
  7622. return changed.wait(timeout) && !aborted;
  7623. }
  7624. bool onAbort()
  7625. {
  7626. aborted = true;
  7627. changed.signal();
  7628. return false;
  7629. }
  7630. void unsubscribe()
  7631. {
  7632. querySDS().unsubscribe(change);
  7633. change = 0;
  7634. }
  7635. bool aborted;
  7636. };
  7637. static WUState _waitForWorkUnit(const char * wuid, unsigned timeout, bool compiled, bool returnOnWaitState)
  7638. {
  7639. StringBuffer wuRoot;
  7640. getXPath(wuRoot, wuid);
  7641. Owned<WorkUnitWaiter> waiter = new WorkUnitWaiter(wuRoot.str());
  7642. LocalIAbortHandler abortHandler(*waiter);
  7643. WUState ret = WUStateUnknown;
  7644. Owned<IRemoteConnection> conn = querySDS().connect(wuRoot.str(), myProcessSession(), 0, SDS_LOCK_TIMEOUT);
  7645. if (conn)
  7646. {
  7647. unsigned start = msTick();
  7648. loop
  7649. {
  7650. ret = (WUState) getEnum(conn->queryRoot(), "@state", states);
  7651. switch (ret)
  7652. {
  7653. case WUStateCompiled:
  7654. case WUStateUploadingFiles:
  7655. if (!compiled)
  7656. break;
  7657. // fall into
  7658. case WUStateCompleted:
  7659. case WUStateFailed:
  7660. case WUStateAborted:
  7661. waiter->unsubscribe();
  7662. return ret;
  7663. case WUStateWait:
  7664. if(returnOnWaitState)
  7665. {
  7666. waiter->unsubscribe();
  7667. return ret;
  7668. }
  7669. break;
  7670. case WUStateCompiling:
  7671. case WUStateRunning:
  7672. case WUStateDebugPaused:
  7673. case WUStateDebugRunning:
  7674. case WUStateBlocked:
  7675. case WUStateAborting:
  7676. if (queryDaliServerVersion().compare("2.1")>=0)
  7677. {
  7678. SessionId agent = conn->queryRoot()->getPropInt64("@agentSession", -1);
  7679. if((agent>0) && querySessionManager().sessionStopped(agent, 0))
  7680. {
  7681. waiter->unsubscribe();
  7682. conn->reload();
  7683. ret = (WUState) getEnum(conn->queryRoot(), "@state", states);
  7684. bool isEcl = false;
  7685. switch (ret)
  7686. {
  7687. case WUStateCompiling:
  7688. isEcl = true;
  7689. // drop into
  7690. case WUStateRunning:
  7691. case WUStateBlocked:
  7692. ret = WUStateFailed;
  7693. break;
  7694. case WUStateAborting:
  7695. ret = WUStateAborted;
  7696. break;
  7697. default:
  7698. return ret;
  7699. }
  7700. WARNLOG("_waitForWorkUnit terminated: %"I64F"d state = %d",(__int64)agent,(int)ret);
  7701. Owned<IWorkUnitFactory> factory = getWorkUnitFactory();
  7702. Owned<IWorkUnit> wu = factory->updateWorkUnit(wuid);
  7703. wu->setState(ret);
  7704. Owned<IWUException> e = wu->createException();
  7705. e->setExceptionCode(isEcl ? 1001 : 1000);
  7706. e->setExceptionMessage(isEcl ? "EclServer terminated unexpectedly" : "Workunit terminated unexpectedly");
  7707. return ret;
  7708. }
  7709. }
  7710. break;
  7711. }
  7712. unsigned waited = msTick() - start;
  7713. if (timeout==-1)
  7714. {
  7715. waiter->wait(20000); // recheck state every 20 seconds even if no timeout, in case eclagent has crashed.
  7716. if (waiter->aborted)
  7717. {
  7718. ret = WUStateUnknown; // MORE - throw an exception?
  7719. break;
  7720. }
  7721. }
  7722. else if (waited > timeout || !waiter->wait(timeout-waited))
  7723. {
  7724. ret = WUStateUnknown; // MORE - throw an exception?
  7725. break;
  7726. }
  7727. conn->reload();
  7728. }
  7729. }
  7730. waiter->unsubscribe();
  7731. return ret;
  7732. }
  7733. extern WUState waitForWorkUnitToComplete(const char * wuid, int timeout, bool returnOnWaitState)
  7734. {
  7735. return _waitForWorkUnit(wuid, (unsigned)timeout, false, returnOnWaitState);
  7736. }
  7737. extern WORKUNIT_API WUState secWaitForWorkUnitToComplete(const char * wuid, ISecManager &secmgr, ISecUser &secuser, int timeout, bool returnOnWaitState)
  7738. {
  7739. if (checkWuSecAccess(wuid, secmgr, &secuser, SecAccess_Read, "Wait for Complete", false, true))
  7740. return waitForWorkUnitToComplete(wuid, timeout, returnOnWaitState);
  7741. return WUStateUnknown;
  7742. }
  7743. extern bool waitForWorkUnitToCompile(const char * wuid, int timeout)
  7744. {
  7745. switch(_waitForWorkUnit(wuid, (unsigned)timeout, true, true))
  7746. {
  7747. case WUStateCompiled:
  7748. case WUStateCompleted:
  7749. case WUStateWait:
  7750. case WUStateUploadingFiles:
  7751. return true;
  7752. default:
  7753. return false;
  7754. }
  7755. }
  7756. extern WORKUNIT_API bool secWaitForWorkUnitToCompile(const char * wuid, ISecManager &secmgr, ISecUser &secuser, int timeout)
  7757. {
  7758. if (checkWuSecAccess(wuid, secmgr, &secuser, SecAccess_Read, "Wait for Compile", false, true))
  7759. return waitForWorkUnitToCompile(wuid, timeout);
  7760. return false;
  7761. }
  7762. extern WORKUNIT_API bool secDebugWorkunit(const char * wuid, ISecManager &secmgr, ISecUser &secuser, const char *command, StringBuffer &response)
  7763. {
  7764. if (strnicmp(command, "<debug:", 7) == 0 && checkWuSecAccess(wuid, secmgr, &secuser, SecAccess_Read, "Debug", false, true))
  7765. {
  7766. Owned<IConstWorkUnit> wu = factory->secOpenWorkUnit(wuid, false, &secmgr, &secuser);
  7767. SCMStringBuffer ip;
  7768. unsigned port;
  7769. port = wu->getDebugAgentListenerPort();
  7770. wu->getDebugAgentListenerIP(ip);
  7771. SocketEndpoint debugEP(ip.str(), port);
  7772. Owned<ISocket> socket = ISocket::connect_timeout(debugEP, 1000);
  7773. unsigned len = (size32_t)strlen(command);
  7774. unsigned revlen = len;
  7775. _WINREV(revlen);
  7776. socket->write(&revlen, sizeof(revlen));
  7777. socket->write(command, len);
  7778. for (;;)
  7779. {
  7780. socket->read(&len, sizeof(len));
  7781. _WINREV(len);
  7782. if (len == 0)
  7783. break;
  7784. if (len & 0x80000000)
  7785. {
  7786. throwUnexpected();
  7787. }
  7788. char * mem = (char*) response.reserve(len);
  7789. socket->read(mem, len);
  7790. }
  7791. return true;
  7792. }
  7793. return false;
  7794. }
  7795. IWUResult * updateWorkUnitResult(IWorkUnit * w, const char *name, unsigned sequence)
  7796. {
  7797. switch ((int)sequence)
  7798. {
  7799. case ResultSequenceStored:
  7800. return w->updateVariableByName(name);
  7801. case ResultSequencePersist:
  7802. return w->updateGlobalByName(name);
  7803. case ResultSequenceInternal:
  7804. case ResultSequenceOnce:
  7805. return w->updateTemporaryByName(name);
  7806. default:
  7807. return w->updateResultBySequence(sequence);
  7808. }
  7809. }
  7810. IConstWUResult * getWorkUnitResult(IConstWorkUnit * w, const char *name, unsigned sequence)
  7811. {
  7812. switch ((int)sequence)
  7813. {
  7814. case ResultSequenceStored:
  7815. return w->getVariableByName(name);
  7816. case ResultSequencePersist:
  7817. return w->getGlobalByName(name);
  7818. case ResultSequenceInternal:
  7819. case ResultSequenceOnce:
  7820. return w->getTemporaryByName(name);
  7821. default:
  7822. if (name && name[0])
  7823. return w->getResultByName(name);//name takes precedence over sequence
  7824. else
  7825. return w->getResultBySequence(sequence);
  7826. }
  7827. }
  7828. extern WORKUNIT_API bool getWorkUnitCreateTime(const char *wuid,CDateTime &time)
  7829. {
  7830. if (wuid) {
  7831. char prefchar;
  7832. unsigned year,month,day,hour,min,sec;
  7833. if (sscanf(wuid, "%c%4u%2u%2u-%2u%2u%2u", &prefchar, &year, &month, &day, &hour, &min, &sec)==7) {
  7834. time.set(year, month, day, hour, min, sec, 0, true);
  7835. // time.setDate(year, month, day);
  7836. // time.setTime(hour, min, sec, 0, true); // for some reason time is local
  7837. return true;
  7838. }
  7839. }
  7840. return false;
  7841. }
  7842. extern WORKUNIT_API IStringVal& createToken(const char *wuid, const char *user, const char *password, IStringVal &str)
  7843. {
  7844. StringBuffer wu, token("X");
  7845. wu.append(wuid).append(';').append(user).append(';').append(password);
  7846. encrypt(token,wu.str());
  7847. str.set(token.str());
  7848. return str;
  7849. }
  7850. // This will be replaced by something more secure!
  7851. extern WORKUNIT_API void extractToken(const char *token, const char *wuid, IStringVal &user, IStringVal &password)
  7852. {
  7853. if (token && *token)
  7854. {
  7855. StringBuffer wu;
  7856. decrypt(wu, token+1);
  7857. const char *finger = strchr(wu.str(),';');
  7858. if (finger && strnicmp(wuid, wu.str(), finger-wu.str())==0)
  7859. {
  7860. const char *finger1 = strchr(++finger,';');
  7861. if(finger1)
  7862. {
  7863. user.setLen(finger, (size32_t)(finger1-finger));
  7864. password.setLen(++finger1, (size32_t)(wu.str() + wu.length() - finger1));
  7865. return;
  7866. }
  7867. }
  7868. throw MakeStringException(WUERR_InvalidSecurityToken, "Invalid call to extractToken");
  7869. }
  7870. }
  7871. extern WORKUNIT_API WUState getWorkUnitState(const char* state)
  7872. {
  7873. return (WUState) getEnum(state, states);
  7874. }
  7875. const LogMsgCategory MCschedconn = MCprogress(1000); // Category used to inform about schedule synchronization
  7876. class CWorkflowScheduleConnection : public CInterface, implements IWorkflowScheduleConnection
  7877. {
  7878. public:
  7879. CWorkflowScheduleConnection(char const * wuid)
  7880. {
  7881. basexpath.append("/WorkflowSchedule/").append(wuid);
  7882. flagxpath.append(basexpath.str()).append("/Active");
  7883. }
  7884. IMPLEMENT_IINTERFACE;
  7885. virtual void lock()
  7886. {
  7887. LOG(MCschedconn, "Locking base schedule connection");
  7888. baseconn.setown(querySDS().connect(basexpath.str(), myProcessSession(), RTM_CREATE_QUERY | RTM_LOCK_WRITE, INFINITE));
  7889. if(!baseconn)
  7890. throw MakeStringException(WUERR_ScheduleLockFailed, "Could not get base workflow schedule lock");
  7891. }
  7892. virtual void unlock()
  7893. {
  7894. LOG(MCschedconn, "Unlocking base schedule connection");
  7895. baseconn.clear();
  7896. }
  7897. virtual void setActive()
  7898. {
  7899. LOG(MCschedconn, "Setting active flag in schedule connection");
  7900. flagconn.setown(querySDS().connect(flagxpath.str(), myProcessSession(), RTM_CREATE | RTM_LOCK_WRITE | RTM_DELETE_ON_DISCONNECT, INFINITE));
  7901. if(!flagconn)
  7902. throw MakeStringException(WUERR_ScheduleLockFailed, "Could not get active workflow schedule lock");
  7903. }
  7904. virtual void resetActive()
  7905. {
  7906. LOG(MCschedconn, "Resetting active flag in schedule connection");
  7907. flagconn.clear();
  7908. }
  7909. virtual bool queryActive()
  7910. {
  7911. return baseconn->queryRoot()->hasProp("Active");
  7912. }
  7913. virtual bool pull(IWorkflowItemArray * workflow)
  7914. {
  7915. assertex(baseconn);
  7916. Owned<IPropertyTree> root = baseconn->getRoot();
  7917. Owned<IPropertyTree> eventQueue = root->getPropTree("EventQueue");
  7918. if(!eventQueue) return false;
  7919. if(!eventQueue->hasProp("Item")) return false;
  7920. {
  7921. Owned<IPropertyTreeIterator> eventItems = eventQueue->getElements("Item");
  7922. Owned<IPropertyTree> eventItem;
  7923. Owned<IRuntimeWorkflowItemIterator> wfItems = workflow->getSequenceIterator();
  7924. Owned<IRuntimeWorkflowItem> wfItem;
  7925. for(eventItems->first(); eventItems->isValid(); eventItems->next())
  7926. {
  7927. eventItem.setown(&eventItems->get());
  7928. const char * eventName = eventItem->queryProp("@name");
  7929. const char * eventText = eventItem->queryProp("@text");
  7930. for(wfItems->first(); wfItems->isValid(); wfItems->next())
  7931. {
  7932. wfItem.setown(wfItems->get());
  7933. if(wfItem->queryState() != WFStateWait)
  7934. continue;
  7935. Owned<IWorkflowEvent> targetEvent = wfItem->getScheduleEvent();
  7936. if(!targetEvent || !targetEvent->matches(eventName, eventText))
  7937. continue;
  7938. wfItem->setEvent(eventName, eventText);
  7939. wfItem->setState(WFStateReqd);
  7940. resetDepState(workflow, *wfItem);
  7941. }
  7942. }
  7943. }
  7944. bool more;
  7945. do
  7946. more = eventQueue->removeProp("Item");
  7947. while(more);
  7948. return true;
  7949. }
  7950. virtual void push(char const * name, char const * text)
  7951. {
  7952. assertex(baseconn);
  7953. Owned<IPropertyTree> root = baseconn->getRoot();
  7954. ensurePTree(root, "EventQueue");
  7955. Owned<IPropertyTree> eventQueue = root->getPropTree("EventQueue");
  7956. Owned<IPropertyTree> eventItem = createPTree();
  7957. eventItem->setProp("@name", name);
  7958. eventItem->setProp("@text", text);
  7959. eventQueue->addPropTree("Item", eventItem.getLink());
  7960. }
  7961. private:
  7962. void resetDepState(IWorkflowItemArray * workflow, IRuntimeWorkflowItem & item) const
  7963. {
  7964. Owned<IWorkflowDependencyIterator> iter(item.getDependencies());
  7965. for(iter->first(); iter->isValid(); iter->next())
  7966. {
  7967. IRuntimeWorkflowItem & dep = workflow->queryWfid(iter->query());
  7968. switch(dep.queryState())
  7969. {
  7970. case WFStateDone:
  7971. case WFStateFail:
  7972. dep.setState(WFStateNull);
  7973. resetDepState(workflow, dep);
  7974. break;
  7975. }
  7976. }
  7977. }
  7978. private:
  7979. StringBuffer basexpath;
  7980. StringBuffer flagxpath;
  7981. Owned<IRemoteConnection> baseconn;
  7982. Owned<IRemoteConnection> flagconn;
  7983. };
  7984. extern WORKUNIT_API IWorkflowScheduleConnection * getWorkflowScheduleConnection(char const * wuid)
  7985. {
  7986. return new CWorkflowScheduleConnection(wuid);
  7987. }
  7988. extern WORKUNIT_API IExtendedWUInterface * queryExtendedWU(IWorkUnit * wu)
  7989. {
  7990. return QUERYINTERFACE(wu, IExtendedWUInterface);
  7991. }
  7992. extern WORKUNIT_API void addExceptionToWorkunit(IWorkUnit * wu, WUExceptionSeverity severity, const char * source, unsigned code, const char * text, const char * filename, unsigned lineno, unsigned column)
  7993. {
  7994. Owned<IWUException> we = wu->createException();
  7995. we->setSeverity(severity);
  7996. we->setExceptionMessage(text);
  7997. if (source)
  7998. we->setExceptionSource(source);
  7999. if (code)
  8000. we->setExceptionCode(code);
  8001. if (filename)
  8002. we->setExceptionFileName(filename);
  8003. if (lineno)
  8004. {
  8005. we->setExceptionLineNo(lineno);
  8006. if (column)
  8007. we->setExceptionColumn(lineno);
  8008. }
  8009. }
  8010. const char * skipLeadingXml(const char * text)
  8011. {
  8012. if (!text)
  8013. return NULL;
  8014. //skip utf8 BOM, probably excessive
  8015. if (memcmp(text, UTF8_BOM, 3) == 0)
  8016. text += 3;
  8017. loop
  8018. {
  8019. if (isspace(*text))
  8020. text++;
  8021. else if (text[0] == '<' && text[1] == '?')
  8022. {
  8023. text += 2;
  8024. loop
  8025. {
  8026. if (!*text) break;
  8027. if (text[0] == '?' && text[1] == '>')
  8028. {
  8029. text += 2;
  8030. break;
  8031. }
  8032. text++;
  8033. }
  8034. }
  8035. else if (text[0] == '<' && text[1] == '!' && text[2] == '-' && text[3] == '-')
  8036. {
  8037. text += 4;
  8038. loop
  8039. {
  8040. if (!*text) break;
  8041. if (text[0] == '-' && text[1] == '-' && text[2] == '>')
  8042. {
  8043. text += 3;
  8044. break;
  8045. }
  8046. text++;
  8047. }
  8048. }
  8049. else
  8050. break;
  8051. }
  8052. return text;
  8053. }
  8054. extern WORKUNIT_API bool isArchiveQuery(const char * text)
  8055. {
  8056. text = skipLeadingXml(text);
  8057. if (!text)
  8058. return false;
  8059. const char * archivePrefix = "<Archive";
  8060. return memicmp(text, archivePrefix, strlen(archivePrefix)) == 0;
  8061. }
  8062. extern WORKUNIT_API bool isQueryManifest(const char * text)
  8063. {
  8064. text = skipLeadingXml(text);
  8065. if (!text)
  8066. return false;
  8067. const char * manifestPrefix = "<Manifest";
  8068. return memicmp(text, manifestPrefix, strlen(manifestPrefix)) == 0;
  8069. }
  8070. //------------------------------------------------------------------------------
  8071. // Named Alias helper function
  8072. static IPropertyTree * resolveQueryByDll(IPropertyTree * queryRegistry, const char * dll)
  8073. {
  8074. StringBuffer xpath;
  8075. xpath.append("Query[@dll=\"").append(dll).append("\"]");
  8076. return queryRegistry->getPropTree(xpath);
  8077. }
  8078. static IPropertyTree * resolveQueryByWuid(IPropertyTree * queryRegistry, const char * wuid)
  8079. {
  8080. StringBuffer xpath;
  8081. xpath.append("Query[@wuid=\"").append(wuid).append("\"]");
  8082. return queryRegistry->getPropTree(xpath);
  8083. }
  8084. static void clearAliases(IPropertyTree * queryRegistry, const char * id)
  8085. {
  8086. StringBuffer lcId(id);
  8087. lcId.toLowerCase();
  8088. StringBuffer xpath;
  8089. xpath.append("Alias[@id=\"").append(lcId).append("\"]");
  8090. Owned<IPropertyTreeIterator> iter = queryRegistry->getElements(xpath);
  8091. ForEach(*iter)
  8092. {
  8093. queryRegistry->removeProp(xpath.str());
  8094. }
  8095. }
  8096. IPropertyTree * addNamedQuery(IPropertyTree * queryRegistry, const char * name, const char * wuid, const char * dll, bool library)
  8097. {
  8098. StringBuffer lcName(name);
  8099. lcName.toLowerCase();
  8100. StringBuffer xpath;
  8101. xpath.append("Query[@name=\"").append(lcName.str()).append("\"]");
  8102. Owned<IPropertyTreeIterator> iter = queryRegistry->getElements(xpath);
  8103. unsigned seq = 1;
  8104. ForEach(*iter)
  8105. {
  8106. IPropertyTree &item = iter->query();
  8107. const char *thisWuid = item.queryProp("@wuid");
  8108. if (strieq(wuid, thisWuid))
  8109. return &item;
  8110. unsigned thisSeq = item.getPropInt("@seq");
  8111. if (thisSeq >= seq)
  8112. seq = thisSeq + 1;
  8113. }
  8114. StringBuffer id;
  8115. id.append(lcName).append(".").append(seq);
  8116. IPropertyTree * newEntry = createPTree("Query", ipt_caseInsensitive);
  8117. newEntry->setProp("@name", lcName);
  8118. newEntry->setProp("@wuid", wuid);
  8119. newEntry->setProp("@dll", dll);
  8120. newEntry->setProp("@id", id);
  8121. newEntry->setPropInt("@seq", seq);
  8122. if (library)
  8123. newEntry->setPropBool("@isLibrary", true);
  8124. return queryRegistry->addPropTree("Query", newEntry);
  8125. }
  8126. void removeNamedQuery(IPropertyTree * queryRegistry, const char * id)
  8127. {
  8128. StringBuffer lcId(id);
  8129. lcId.toLowerCase();
  8130. clearAliases(queryRegistry, lcId);
  8131. StringBuffer xpath;
  8132. xpath.append("Query[@id=\"").append(lcId).append("\"]");
  8133. queryRegistry->removeProp(xpath);
  8134. }
  8135. void removeDllFromNamedQueries(IPropertyTree * queryRegistry, const char * dll)
  8136. {
  8137. Owned<IPropertyTree> match = resolveQueryByDll(queryRegistry, dll);
  8138. if (!match)
  8139. return;
  8140. clearAliases(queryRegistry, match->queryProp("@id"));
  8141. queryRegistry->removeTree(match);
  8142. }
  8143. void removeWuidFromNamedQueries(IPropertyTree * queryRegistry, const char * wuid)
  8144. {
  8145. Owned<IPropertyTree> match = resolveQueryByWuid(queryRegistry, wuid);
  8146. if (!match)
  8147. return;
  8148. clearAliases(queryRegistry, match->queryProp("@id"));
  8149. queryRegistry->removeTree(match);
  8150. }
  8151. void removeAliasesFromNamedQuery(IPropertyTree * queryRegistry, const char * id)
  8152. {
  8153. clearAliases(queryRegistry, id);
  8154. }
  8155. void setQueryAlias(IPropertyTree * queryRegistry, const char * name, const char * value)
  8156. {
  8157. StringBuffer lcName(name);
  8158. lcName.toLowerCase();
  8159. StringBuffer xpath;
  8160. xpath.append("Alias[@name=\"").append(lcName).append("\"]");
  8161. IPropertyTree * match = queryRegistry->queryPropTree(xpath);
  8162. if (!match)
  8163. {
  8164. IPropertyTree * newEntry = createPTree("Alias");
  8165. newEntry->setProp("@name", lcName);
  8166. match = queryRegistry->addPropTree("Alias", newEntry);
  8167. }
  8168. match->setProp("@id", value);
  8169. }
  8170. extern WORKUNIT_API IPropertyTree * getQueryById(IPropertyTree * queryRegistry, const char *queryid)
  8171. {
  8172. if (!queryRegistry || !queryid)
  8173. return NULL;
  8174. StringBuffer xpath;
  8175. xpath.append("Query[@id=\"").append(queryid).append("\"]");
  8176. return queryRegistry->getPropTree(xpath);
  8177. }
  8178. extern WORKUNIT_API IPropertyTree * getQueryById(const char *queryset, const char *queryid, bool readonly)
  8179. {
  8180. Owned<IPropertyTree> queryRegistry = getQueryRegistry(queryset, readonly);
  8181. return getQueryById(queryRegistry, queryid);
  8182. }
  8183. extern WORKUNIT_API IPropertyTree * resolveQueryAlias(IPropertyTree * queryRegistry, const char * alias)
  8184. {
  8185. if (!queryRegistry || !alias)
  8186. return NULL;
  8187. StringBuffer xpath;
  8188. unsigned cnt = 0;
  8189. StringBuffer lc(alias);
  8190. const char * search = lc.toLowerCase().str();
  8191. loop
  8192. {
  8193. xpath.set("Alias[@name='").append(search).append("']/@id");
  8194. const char * queryId = queryRegistry->queryProp(xpath);
  8195. if (!queryId)
  8196. break;
  8197. //Check for too many alias indirections.
  8198. if (cnt++ > 10)
  8199. return NULL;
  8200. search = lc.set(queryId).toLowerCase().str();
  8201. }
  8202. return getQueryById(queryRegistry, search);
  8203. }
  8204. extern WORKUNIT_API IPropertyTree * resolveQueryAlias(const char *queryset, const char *alias, bool readonly)
  8205. {
  8206. Owned<IPropertyTree> queryRegistry = getQueryRegistry(queryset, readonly);
  8207. return resolveQueryAlias(queryRegistry, alias);
  8208. }
  8209. void setQuerySuspendedState(IPropertyTree * queryRegistry, const char *id, bool suspend)
  8210. {
  8211. StringBuffer lcId(id);
  8212. lcId.toLowerCase();
  8213. StringBuffer xpath;
  8214. xpath.append("Query[@id=\"").append(lcId).append("\"]");
  8215. IPropertyTree *tree = queryRegistry->queryPropTree(xpath);
  8216. if (tree)
  8217. {
  8218. if (suspend)
  8219. tree->addPropBool("@suspended", true);
  8220. else
  8221. tree->removeProp("@suspended");
  8222. }
  8223. else
  8224. throw MakeStringException((suspend)? QUERRREG_SUSPEND : QUERRREG_UNSUSPEND, "Modifying query suspended state failed. Could not find query %s", id);
  8225. }
  8226. void setQueryCommentForNamedQuery(IPropertyTree * queryRegistry, const char *id, const char *queryComment)
  8227. {
  8228. if (queryComment)
  8229. {
  8230. StringBuffer lcId(id);
  8231. lcId.toLowerCase();
  8232. StringBuffer xpath;
  8233. xpath.append("Query[@id=\"").append(lcId).append("\"]");
  8234. IPropertyTree *tree = queryRegistry->queryPropTree(xpath);
  8235. if (tree)
  8236. tree->setProp("@queryComment", queryComment);
  8237. else
  8238. throw MakeStringException(QUERRREG_COMMENT, "Could not find query %s", id);
  8239. }
  8240. }
  8241. extern WORKUNIT_API IPropertyTree * getQueryRegistryRoot()
  8242. {
  8243. Owned<IRemoteConnection> globalLock = querySDS().connect("/QuerySets/", myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT);
  8244. if (!globalLock)
  8245. return NULL;
  8246. //Only lock the branch for the target we're interested in.
  8247. StringBuffer xpath;
  8248. xpath.append("/QuerySets");
  8249. Owned<IRemoteConnection> conn = querySDS().connect(xpath.str(), myProcessSession(), RTM_LOCK_READ, SDS_LOCK_TIMEOUT);
  8250. if (conn)
  8251. return conn->getRoot();
  8252. else
  8253. return NULL;
  8254. }
  8255. extern WORKUNIT_API IPropertyTree * getQueryRegistry(const char * wsEclId, bool readonly)
  8256. {
  8257. Owned<IRemoteConnection> globalLock = querySDS().connect("/QuerySets/", myProcessSession(), RTM_LOCK_WRITE|RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT);
  8258. //Only lock the branch for the target we're interested in.
  8259. StringBuffer xpath;
  8260. xpath.append("/QuerySets/QuerySet[@id=\"").append(wsEclId).append("\"]");
  8261. Owned<IRemoteConnection> conn = querySDS().connect(xpath.str(), myProcessSession(), readonly ? RTM_LOCK_READ : RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
  8262. if (!conn)
  8263. {
  8264. if (readonly)
  8265. return NULL;
  8266. Owned<IPropertyTree> querySet = createPTree();
  8267. querySet->setProp("@id", wsEclId);
  8268. globalLock->queryRoot()->addPropTree("QuerySet", querySet.getClear());
  8269. globalLock->commit();
  8270. conn.setown(querySDS().connect(xpath.str(), myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT));
  8271. if (!conn)
  8272. throwUnexpected();
  8273. }
  8274. return conn->getRoot();
  8275. }
  8276. IPropertyTree * addNamedPackageSet(IPropertyTree * packageRegistry, const char * name, IPropertyTree *packageInfo, bool overWrite)
  8277. {
  8278. StringBuffer xpath;
  8279. StringBuffer lcName(name);
  8280. lcName.toLowerCase();
  8281. // see if "name" already exists
  8282. xpath.append("Package[@id='").append(name).append("']");
  8283. IPropertyTree *pkgTree = packageRegistry->queryPropTree(xpath.str());
  8284. if (pkgTree)
  8285. {
  8286. if (overWrite)
  8287. packageRegistry->removeTree(pkgTree);
  8288. else
  8289. throw MakeStringException(WUERR_PackageAlreadyExists, "Package name %s already exists, either delete it or specify overwrite",name);
  8290. }
  8291. IPropertyTree *tree = packageRegistry->addPropTree("Package", packageInfo);
  8292. tree->setProp("@id", lcName);
  8293. return tree;
  8294. }
  8295. void removeNamedPackage(IPropertyTree * packageRegistry, const char * id)
  8296. {
  8297. StringBuffer lcId(id);
  8298. lcId.toLowerCase();
  8299. StringBuffer xpath;
  8300. xpath.append("Package[@id=\"").append(lcId).append("\"]");
  8301. packageRegistry->removeProp(xpath);
  8302. }
  8303. extern WORKUNIT_API IPropertyTree * getPackageSetRegistry(const char * wsEclId, bool readonly)
  8304. {
  8305. Owned<IRemoteConnection> globalLock = querySDS().connect("/PackageSets/", myProcessSession(), RTM_LOCK_WRITE|RTM_CREATE_QUERY, SDS_LOCK_TIMEOUT);
  8306. //Only lock the branch for the target we're interested in.
  8307. StringBuffer xpath;
  8308. xpath.append("/PackageSets/PackageSet[@id=\"").append(wsEclId).append("\"]");
  8309. Owned<IRemoteConnection> conn = querySDS().connect(xpath.str(), myProcessSession(), readonly ? RTM_LOCK_READ : RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT);
  8310. if (!conn)
  8311. {
  8312. if (readonly)
  8313. return NULL;
  8314. Owned<IPropertyTree> querySet = createPTree();
  8315. querySet->setProp("@id", wsEclId);
  8316. globalLock->queryRoot()->addPropTree("PackageSet", querySet.getClear());
  8317. globalLock->commit();
  8318. conn.setown(querySDS().connect(xpath.str(), myProcessSession(), RTM_LOCK_WRITE, SDS_LOCK_TIMEOUT));
  8319. if (!conn)
  8320. throwUnexpected();
  8321. }
  8322. return conn->getRoot();
  8323. }
  8324. void addQueryToQuerySet(IWorkUnit *workunit, const char *querySetName, const char *queryName, IPropertyTree *packageInfo, WUQueryActivationOptions activateOption, StringBuffer &newQueryId)
  8325. {
  8326. StringBuffer cleanQueryName;
  8327. appendUtf8XmlName(cleanQueryName, strlen(queryName), queryName);
  8328. SCMStringBuffer dllName;
  8329. Owned<IConstWUQuery> q = workunit->getQuery();
  8330. q->getQueryDllName(dllName);
  8331. if (!dllName.length())
  8332. throw MakeStringException(WUERR_InvalidDll, "Cannot deploy query - no associated dll.");
  8333. SCMStringBuffer wuid;
  8334. workunit->getWuid(wuid);
  8335. Owned<IPropertyTree> queryRegistry = getQueryRegistry(querySetName, false);
  8336. StringBuffer currentTargetClusterType;
  8337. queryRegistry->getProp("@targetclustertype", currentTargetClusterType);
  8338. SCMStringBuffer targetClusterType;
  8339. workunit->getDebugValue("targetclustertype", targetClusterType);
  8340. if (currentTargetClusterType.length() < 1)
  8341. {
  8342. queryRegistry->setProp("@targetclustertype", targetClusterType.str());
  8343. }
  8344. else
  8345. {
  8346. if (strcmp(currentTargetClusterType.str(), "roxie") == 0 && strcmp(currentTargetClusterType.str(), targetClusterType.str())!=0)
  8347. {
  8348. throw MakeStringException(WUERR_MismatchClusterType, "TargetClusterTypes of workunit and queryset do not match.");
  8349. }
  8350. }
  8351. IPropertyTree *newEntry = addNamedQuery(queryRegistry, cleanQueryName, wuid.str(), dllName.str(), isLibrary(workunit));
  8352. newQueryId.append(newEntry->queryProp("@id"));
  8353. workunit->setIsQueryService(true); //will check querysets before delete
  8354. workunit->commit();
  8355. if (activateOption == ACTIVATE_SUSPEND_PREVIOUS|| activateOption == ACTIVATE_DELETE_PREVIOUS)
  8356. {
  8357. Owned<IPropertyTree> aliasTree = resolveQueryAlias(queryRegistry, cleanQueryName);
  8358. setQueryAlias(queryRegistry, cleanQueryName, newQueryId);
  8359. if (aliasTree)
  8360. {
  8361. if (activateOption == ACTIVATE_SUSPEND_PREVIOUS)
  8362. setQuerySuspendedState(queryRegistry, cleanQueryName, true);
  8363. else
  8364. removeNamedQuery(queryRegistry, aliasTree->queryProp("@id"));
  8365. }
  8366. }
  8367. else if (activateOption == MAKE_ACTIVATE || activateOption == MAKE_ACTIVATE_LOAD_DATA_ONLY)
  8368. setQueryAlias(queryRegistry, cleanQueryName, newQueryId.str());
  8369. }
  8370. bool removeQuerySetAlias(const char *querySetName, const char *alias)
  8371. {
  8372. Owned<IPropertyTree> queryRegistry = getQueryRegistry(querySetName, true);
  8373. StringBuffer xpath;
  8374. xpath.appendf("Alias[@name='%s']", alias);
  8375. IPropertyTree *t = queryRegistry->queryPropTree(xpath);
  8376. return queryRegistry->removeTree(t);
  8377. }
  8378. void addQuerySetAlias(const char *querySetName, const char *alias, const char *id)
  8379. {
  8380. Owned<IPropertyTree> queryRegistry = getQueryRegistry(querySetName, false);
  8381. setQueryAlias(queryRegistry, alias, id);
  8382. }
  8383. void setSuspendQuerySetQuery(const char *querySetName, const char *id, bool suspend)
  8384. {
  8385. Owned<IPropertyTree> queryRegistry = getQueryRegistry(querySetName, true);
  8386. setQuerySuspendedState(queryRegistry, id, suspend);
  8387. }
  8388. void deleteQuerySetQuery(const char *querySetName, const char *id)
  8389. {
  8390. Owned<IPropertyTree> queryRegistry = getQueryRegistry(querySetName, true);
  8391. removeNamedQuery(queryRegistry, id);
  8392. }
  8393. void removeQuerySetAliasesFromNamedQuery(const char *querySetName, const char * id)
  8394. {
  8395. Owned<IPropertyTree> queryRegistry = getQueryRegistry(querySetName, true);
  8396. clearAliases(queryRegistry, id);
  8397. }
  8398. void setQueryCommentForNamedQuery(const char *querySetName, const char *id, const char *queryComment)
  8399. {
  8400. Owned<IPropertyTree> queryRegistry = getQueryRegistry(querySetName, true);
  8401. setQueryCommentForNamedQuery(queryRegistry, id, queryComment);
  8402. }
  8403. const char *queryIdFromQuerySetWuid(const char *querySetName, const char *wuid, IStringVal &id)
  8404. {
  8405. Owned<IPropertyTree> queryRegistry = getQueryRegistry(querySetName, false);
  8406. StringBuffer xpath;
  8407. xpath.appendf("Query[@wuid='%s']", wuid);
  8408. IPropertyTree *q = queryRegistry->queryPropTree(xpath.str());
  8409. if (q)
  8410. {
  8411. id.set(q->queryProp("@id"));
  8412. }
  8413. return id.str();
  8414. }
  8415. bool looksLikeAWuid(const char * wuid)
  8416. {
  8417. if (!wuid)
  8418. return false;
  8419. if (wuid[0] != 'W')
  8420. return false;
  8421. if (!isdigit(wuid[1]) || !isdigit(wuid[2]) || !isdigit(wuid[3]) || !isdigit(wuid[4]))
  8422. return false;
  8423. if (!isdigit(wuid[5]) || !isdigit(wuid[6]) || !isdigit(wuid[7]) || !isdigit(wuid[8]))
  8424. return false;
  8425. return (wuid[9]=='-');
  8426. }
  8427. IPropertyTree * resolveDefinitionInArchive(IPropertyTree * archive, const char * path)
  8428. {
  8429. IPropertyTree * module = archive;
  8430. const char * dot = strrchr(path, '.');
  8431. StringBuffer xpath;
  8432. if (dot)
  8433. {
  8434. xpath.clear().append("Module[@key='").appendLower(dot-path, path).append("']");
  8435. module = archive->queryPropTree(xpath);
  8436. path = dot+1;
  8437. }
  8438. else
  8439. module = archive->queryPropTree("Module[@key='']");
  8440. if (!module)
  8441. return NULL;
  8442. xpath.clear().append("Attribute[@key='").appendLower(strlen(path), path).append("']");
  8443. return module->queryPropTree(xpath);
  8444. }
  8445. extern WORKUNIT_API void associateLocalFile(IWUQuery * query, WUFileType type, const char * name, const char * description, unsigned crc)
  8446. {
  8447. StringBuffer hostname;
  8448. queryHostIP().getIpText(hostname);
  8449. StringBuffer fullPathname;
  8450. makeAbsolutePath(name, fullPathname);
  8451. query->addAssociatedFile(type, fullPathname, hostname, description, crc);
  8452. }
  8453. extern WORKUNIT_API void descheduleWorkunit(char const * wuid)
  8454. {
  8455. Owned<IWorkUnitFactory> factory = getWorkUnitFactory();
  8456. Owned<IWorkUnit> workunit = factory->updateWorkUnit(wuid);
  8457. if(workunit)
  8458. workunit->deschedule();
  8459. else
  8460. doDescheduleWorkkunit(wuid);
  8461. }