workunit.cpp 344 KB

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