1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929293029312932293329342935293629372938293929402941294229432944294529462947294829492950295129522953295429552956295729582959296029612962296329642965296629672968296929702971297229732974297529762977297829792980298129822983298429852986298729882989299029912992299329942995299629972998299930003001300230033004300530063007300830093010301130123013301430153016301730183019302030213022302330243025302630273028302930303031303230333034303530363037303830393040304130423043304430453046304730483049305030513052305330543055305630573058305930603061306230633064306530663067306830693070307130723073307430753076307730783079308030813082308330843085308630873088308930903091309230933094309530963097309830993100310131023103310431053106310731083109311031113112311331143115311631173118311931203121312231233124312531263127312831293130313131323133313431353136313731383139314031413142314331443145314631473148314931503151315231533154315531563157315831593160316131623163316431653166316731683169317031713172317331743175317631773178317931803181318231833184318531863187318831893190319131923193319431953196319731983199320032013202320332043205320632073208320932103211321232133214321532163217321832193220322132223223322432253226322732283229323032313232323332343235323632373238323932403241324232433244324532463247324832493250325132523253325432553256325732583259326032613262326332643265326632673268326932703271327232733274327532763277327832793280328132823283328432853286328732883289329032913292329332943295329632973298329933003301330233033304330533063307330833093310331133123313331433153316331733183319332033213322332333243325332633273328332933303331333233333334333533363337333833393340334133423343334433453346334733483349335033513352335333543355335633573358335933603361336233633364336533663367336833693370337133723373337433753376337733783379338033813382338333843385338633873388338933903391339233933394339533963397339833993400340134023403340434053406340734083409341034113412341334143415341634173418341934203421342234233424342534263427342834293430343134323433343434353436343734383439344034413442344334443445344634473448344934503451345234533454345534563457345834593460346134623463346434653466346734683469347034713472347334743475347634773478347934803481348234833484348534863487348834893490349134923493349434953496349734983499350035013502350335043505350635073508350935103511351235133514351535163517351835193520352135223523352435253526352735283529353035313532353335343535353635373538353935403541354235433544354535463547354835493550355135523553355435553556355735583559356035613562356335643565356635673568356935703571357235733574357535763577357835793580358135823583358435853586358735883589359035913592359335943595359635973598359936003601360236033604360536063607360836093610361136123613361436153616361736183619362036213622362336243625362636273628362936303631363236333634363536363637363836393640364136423643364436453646364736483649365036513652365336543655365636573658365936603661366236633664366536663667366836693670367136723673367436753676367736783679368036813682368336843685368636873688368936903691369236933694369536963697369836993700370137023703370437053706370737083709371037113712371337143715371637173718371937203721372237233724372537263727372837293730373137323733373437353736373737383739374037413742374337443745374637473748374937503751375237533754375537563757375837593760376137623763376437653766376737683769377037713772377337743775377637773778377937803781378237833784378537863787378837893790379137923793379437953796379737983799380038013802380338043805380638073808380938103811381238133814381538163817381838193820382138223823382438253826382738283829383038313832383338343835383638373838383938403841384238433844384538463847384838493850385138523853385438553856385738583859386038613862386338643865386638673868386938703871387238733874387538763877387838793880388138823883388438853886388738883889389038913892389338943895389638973898389939003901390239033904390539063907390839093910391139123913391439153916391739183919392039213922392339243925392639273928392939303931393239333934393539363937393839393940394139423943394439453946394739483949395039513952395339543955395639573958395939603961396239633964396539663967396839693970397139723973397439753976397739783979398039813982398339843985398639873988398939903991399239933994399539963997399839994000400140024003 |
- /*##############################################################################
- HPCC SYSTEMS software Copyright (C) 2012 HPCC Systems.
- Licensed under the Apache License, Version 2.0 (the "License");
- you may not use this file except in compliance with the License.
- You may obtain a copy of the License at
- http://www.apache.org/licenses/LICENSE-2.0
- Unless required by applicable law or agreed to in writing, software
- distributed under the License is distributed on an "AS IS" BASIS,
- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
- See the License for the specific language governing permissions and
- limitations under the License.
- ############################################################################## */
- #include "ws_workunitsService.hpp"
- #include "ws_fs.hpp"
- #include "jlib.hpp"
- #include "daclient.hpp"
- #include "dalienv.hpp"
- #include "dadfs.hpp"
- #include "daaudit.hpp"
- #include "exception_util.hpp"
- #include "wujobq.hpp"
- #include "eventqueue.hpp"
- #include "fileview.hpp"
- #include "hqlerror.hpp"
- #include "sacmd.hpp"
- #include "wuwebview.hpp"
- #include "portlist.h"
- #include "dllserver.hpp"
- #include "schedulectrl.hpp"
- #include "scheduleread.hpp"
- #include "dadfs.hpp"
- #include "dfuwu.hpp"
- #include "thorplugin.hpp"
- #include "roxiecontrol.hpp"
- #include "package.h"
- #ifdef _USE_ZLIB
- #include "zcrypt.hpp"
- #endif
- #define ESP_WORKUNIT_DIR "workunits/"
- class ExecuteExistingQueryInfo
- {
- public:
- ExecuteExistingQueryInfo(IConstWorkUnit *cw)
- {
- SCMStringBuffer isv;
- cw->getJobName(isv);
- const char *name = isv.str();
- const char *div = strchr(name, '.');
- if (div)
- {
- queryset.set(name, div-name);
- query.set(div+1);
- }
- }
- public:
- StringAttr queryset;
- StringAttr query;
- };
- typedef enum _WuActionType
- {
- ActionDelete=0,
- ActionProtect,
- ActionAbort,
- ActionRestore,
- ActionEventSchedule,
- ActionEventDeschedule,
- ActionChangeState,
- ActionPause,
- ActionPauseNow,
- ActionResume,
- ActionUnknown
- } WsWuActionType;
- void setActionResult(const char* wuid, int action, const char* result, StringBuffer& strAction, IArrayOf<IConstWUActionResult>* results)
- {
- if (!results || !wuid || !*wuid || !result || !*result)
- return;
- switch(action)
- {
- case ActionDelete:
- {
- strAction = "Delete";
- break;
- }
- case ActionProtect:
- {
- strAction = "Protect";
- break;
- }
- case ActionAbort:
- {
- strAction = "Abort";
- break;
- }
- case ActionRestore:
- {
- strAction = "Restore";
- break;
- }
- case ActionEventSchedule:
- {
- strAction = "EventSchedule";
- break;
- }
- case ActionEventDeschedule:
- {
- strAction = "EventDeschedule";
- break;
- }
- case ActionChangeState:
- {
- strAction = "ChangeState";
- break;
- }
- case ActionPause:
- {
- strAction = "Pause";
- break;
- }
- case ActionPauseNow:
- {
- strAction = "PauseNow";
- break;
- }
- case ActionResume:
- {
- strAction = "Resume";
- break;
- }
- default:
- {
- strAction = "Unknown";
- }
- }
- Owned<IEspWUActionResult> res = createWUActionResult("", "");
- res->setWuid(wuid);
- res->setAction(strAction.str());
- res->setResult(result);
- results->append(*res.getClear());
- }
- bool doAction(IEspContext& context, StringArray& wuids, int action, IProperties* params, IArrayOf<IConstWUActionResult>* results)
- {
- if (!wuids.length())
- return true;
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- bool bAllSuccess = true;
- for(aindex_t i=0; i<wuids.length();i++)
- {
- StringBuffer strAction;
- StringBuffer wuidStr = wuids.item(i);
- const char* wuid = wuidStr.trim().str();
- if (isEmpty(wuid))
- {
- WARNLOG("Empty Workunit ID");
- continue;
- }
- try
- {
- if (!looksLikeAWuid(wuid))
- throw MakeStringException(ECLWATCH_INVALID_INPUT, "Invalid Workunit ID: %s", wuid);
- if ((action == ActionRestore) || (action == ActionEventDeschedule))
- {
- switch(action)
- {
- case ActionRestore:
- {
- SocketEndpoint ep;
- if (params->hasProp("sashaServerIP"))
- ep.set(params->queryProp("sashaServerIP"), params->getPropInt("sashaServerPort"));
- else
- getSashaNode(ep);
- Owned<ISashaCommand> cmd = createSashaCommand();
- cmd->setAction(SCA_RESTORE);
- cmd->addId(wuid);
- Owned<INode> node = createINode(ep);
- if (!node)
- throw MakeStringException(ECLWATCH_INODE_NOT_FOUND,"INode not found.");
- StringBuffer s;
- if (!cmd->send(node, 1*60*1000))
- throw MakeStringException(ECLWATCH_CANNOT_CONNECT_ARCHIVE_SERVER,"Cannot connect to Archive server at %s.", ep.getUrlStr(s).str());
- if (cmd->numIds()==0)
- throw MakeStringException(ECLWATCH_CANNOT_UPDATE_WORKUNIT,"Could not Archive/restore %s",wuid);
- StringBuffer reply;
- cmd->getId(0,reply);
- AuditSystemAccess(context.queryUserId(), true, "Updated %s", wuid);
- ensureWsWorkunitAccess(context, wuid, SecAccess_Write);
- break;
- }
- case ActionEventDeschedule:
- if (!context.validateFeatureAccess(OWN_WU_ACCESS, SecAccess_Full, false)
- || !context.validateFeatureAccess(OTHERS_WU_ACCESS, SecAccess_Full, false))
- ensureWsWorkunitAccess(context, wuid, SecAccess_Full);
- descheduleWorkunit(wuid);
- AuditSystemAccess(context.queryUserId(), true, "Updated %s", wuid);
- break;
- }
- }
- else
- {
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid, false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_OPEN_WORKUNIT,"Cannot open workunit %s.",wuid);
- if ((action == ActionDelete) && (cw->getState() == WUStateWait))
- throw MakeStringException(ECLWATCH_CANNOT_DELETE_WORKUNIT,"Cannot delete a workunit which is in a 'Wait' status.");
- switch(action)
- {
- case ActionPause:
- {
- ensureWsWorkunitAccess(context, *cw, SecAccess_Full);
- WorkunitUpdate wu(&cw->lock());
- wu->setAction(WUActionPause);
- break;
- }
- case ActionPauseNow:
- {
- ensureWsWorkunitAccess(context, *cw, SecAccess_Full);
- WorkunitUpdate wu(&cw->lock());
- wu->setAction(WUActionPauseNow);
- break;
- }
- case ActionResume:
- {
- ensureWsWorkunitAccess(context, *cw, SecAccess_Full);
- WorkunitUpdate wu(&cw->lock());
- wu->setAction(WUActionResume);
- break;
- }
- case ActionDelete:
- ensureWsWorkunitAccess(context, *cw, SecAccess_Full);
- {
- int state = cw->getState();
- switch (state)
- {
- case WUStateWait:
- case WUStateAborted:
- case WUStateCompleted:
- case WUStateFailed:
- case WUStateArchived:
- case WUStateCompiled:
- case WUStateUploadingFiles:
- break;
- default:
- {
- WorkunitUpdate wu(&cw->lock());
- wu->setState(WUStateFailed);
- }
- }
- cw.clear();
- factory->deleteWorkUnit(wuid);
- AuditSystemAccess(context.queryUserId(), true, "Deleted %s", wuid);
- }
- break;
- case ActionAbort:
- ensureWsWorkunitAccess(context, *cw, SecAccess_Full);
- {
- if (cw->getState() == WUStateWait)
- {
- WorkunitUpdate wu(&cw->lock());
- wu->deschedule();
- wu->setState(WUStateAborted);
- }
- else
- secAbortWorkUnit(wuid, *context.querySecManager(), *context.queryUser());
- AuditSystemAccess(context.queryUserId(), true, "Aborted %s", wuid);
- }
- break;
- case ActionProtect:
- cw->protect(!params || params->getPropBool("Protect",true));
- AuditSystemAccess(context.queryUserId(), true, "Updated %s", wuid);
- break;
- case ActionChangeState:
- {
- if (params)
- {
- WUState state = (WUState) params->getPropInt("State");
- if (state > WUStateUnknown && state < WUStateSize)
- {
- WorkunitUpdate wu(&cw->lock());
- wu->setState(state);
- AuditSystemAccess(context.queryUserId(), true, "Updated %s", wuid);
- }
- }
- }
- break;
- case ActionEventSchedule:
- {
- WorkunitUpdate wu(&cw->lock());
- wu->schedule();
- AuditSystemAccess(context.queryUserId(), true, "Updated %s", wuid);
- }
- break;
- }
- }
- setActionResult(wuid, action, "Success", strAction, results);
- }
- catch (IException *e)
- {
- bAllSuccess = false;
- StringBuffer eMsg;
- StringBuffer failedMsg("Failed: ");
- setActionResult(wuid, action, failedMsg.append(e->errorMessage(eMsg)).str(), strAction, results);
- WARNLOG("Failed to %s for workunit: %s, %s", strAction.str(), wuid, eMsg.str());
- AuditSystemAccess(context.queryUserId(), false, "Failed to %s %s", strAction.str(), wuid);
- e->Release();
- continue;
- }
- catch (...)
- {
- bAllSuccess = false;
- StringBuffer failedMsg;
- failedMsg.appendf("Unknown exception");
- setActionResult(wuid, action, failedMsg.str(), strAction, results);
- WARNLOG("Failed to %s for workunit: %s, %s", strAction.str(), wuid, failedMsg.str());
- AuditSystemAccess(context.queryUserId(), false, "Failed to %s %s", strAction.str(), wuid);
- continue;
- }
- }
- int timeToWait = 0;
- if (params)
- timeToWait = params->getPropInt("BlockTillFinishTimer");
- if (timeToWait != 0)
- {
- for(aindex_t i=0; i<wuids.length();i++)
- {
- const char* wuid=wuids.item(i);
- if (isEmpty(wuid))
- continue;
- waitForWorkUnitToComplete(wuid, timeToWait);
- }
- }
- return bAllSuccess;
- }
- MapStringTo<int> wuActionTable;
- void CWsWorkunitsEx::init(IPropertyTree *cfg, const char *process, const char *service)
- {
- if (!daliClientActive())
- {
- ERRLOG("No Dali Connection Active.");
- throw MakeStringException(-1, "No Dali Connection Active. Please Specify a Dali to connect to in you configuration file");
- }
- setPasswordsFromSDS();
- DBGLOG("Initializing %s service [process = %s]", service, process);
- refreshValidClusters();
- daliServers.set(cfg->queryProp("Software/EspProcess/@daliServers"));
- wuActionTable.setValue("delete", ActionDelete);
- wuActionTable.setValue("abort", ActionAbort);
- wuActionTable.setValue("pausenow", ActionPauseNow);
- wuActionTable.setValue("pause", ActionPause);
- wuActionTable.setValue("resume", ActionResume);
- wuActionTable.setValue("protect", ActionProtect);
- wuActionTable.setValue("unprotect", ActionProtect);
- wuActionTable.setValue("restore", ActionRestore);
- wuActionTable.setValue("reschedule", ActionEventSchedule);
- wuActionTable.setValue("deschedule", ActionEventDeschedule);
- wuActionTable.setValue("settofailed", ActionChangeState);
- awusCacheMinutes = AWUS_CACHE_MIN_DEFAULT;
- VStringBuffer xpath("Software/EspProcess[@name=\"%s\"]/EspService[@name=\"%s\"]/AWUsCacheMinutes", process, service);
- cfg->getPropInt(xpath.str(), awusCacheMinutes);
- xpath.setf("Software/EspProcess[@name=\"%s\"]/EspService[@name=\"%s\"]/serverForArchivedECLWU/@netAddress", process, service);
- if (cfg->hasProp(xpath.str()))
- {
- sashaServerIp.set(cfg->queryProp(xpath.str()));
- xpath.setf("Software/EspProcess[@name=\"%s\"]/EspService[@name=\"%s\"]/serverForArchivedECLWU/@port", process, service);
- sashaServerPort = cfg->getPropInt(xpath.str(), DEFAULT_SASHA_PORT);
- }
- directories.set(cfg->queryPropTree("Software/Directories"));
- const char *name = cfg->queryProp("Software/EspProcess/@name");
- getConfigurationDirectory(directories, "query", "esp", name ? name : "esp", queryDirectory);
- recursiveCreateDirectory(queryDirectory.str());
- dataCache.setown(new DataCache(DATA_SIZE));
- archivedWuCache.setown(new ArchivedWuCache(AWUS_CACHE_SIZE));
- //Create a folder for temporarily holding gzip files by WUResultBin()
- Owned<IFile> tmpdir = createIFile(TEMPZIPDIR);
- if(!tmpdir->exists())
- tmpdir->createDirectory();
- recursiveCreateDirectory(ESP_WORKUNIT_DIR);
- m_sched.start();
- filesInUse.subscribe();
- QueryFilesInUseUpdateThread *updateFilesInUse = new QueryFilesInUseUpdateThread(filesInUse);
- updateFilesInUse->startRelease();
- }
- void CWsWorkunitsEx::refreshValidClusters()
- {
- validClusters.kill();
- Owned<IStringIterator> it = getTargetClusters(NULL, NULL);
- ForEach(*it)
- {
- SCMStringBuffer s;
- IStringVal &val = it->str(s);
- if (!validClusters.getValue(val.str()))
- validClusters.setValue(val.str(), true);
- }
- }
- bool CWsWorkunitsEx::isValidCluster(const char *cluster)
- {
- if (!cluster || !*cluster)
- return false;
- CriticalBlock block(crit);
- if (validClusters.getValue(cluster))
- return true;
- if (validateTargetClusterName(cluster))
- {
- refreshValidClusters();
- return true;
- }
- return false;
- }
- bool CWsWorkunitsEx::onWUCreate(IEspContext &context, IEspWUCreateRequest &req, IEspWUCreateResponse &resp)
- {
- try
- {
- if (!context.validateFeatureAccess(OWN_WU_ACCESS, SecAccess_Write, false))
- throw MakeStringException(ECLWATCH_ECL_WU_ACCESS_DENIED, "Failed to create workunit. Permission denied.");
- NewWsWorkunit wu(context);
- SCMStringBuffer wuid;
- resp.updateWorkunit().setWuid(wu->getWuid(wuid).str());
- AuditSystemAccess(context.queryUserId(), true, "Updated %s", wuid.str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- static bool origValueChanged(const char *newValue, const char *origValue, StringBuffer &s, bool nillable=true)
- {
- if (!nillable && isEmpty(newValue))
- return false;
- if(newValue && origValue)
- {
- if (!streq(origValue, newValue))
- {
- s.append(newValue).trim();
- return true;
- }
- return false;
- }
- if (newValue)
- {
- s.append(newValue).trim();
- return true;
- }
- return (origValue!=NULL);
- }
- bool CWsWorkunitsEx::onWUUpdate(IEspContext &context, IEspWUUpdateRequest &req, IEspWUUpdateResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUUpdate", wuid);
- ensureWsWorkunitAccess(context, wuid.str(), SecAccess_Write);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_UPDATE_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- if(req.getProtected() != req.getProtectedOrig())
- {
- cw->protect(req.getProtected());
- cw.clear();
- cw.setown(factory->openWorkUnit(wuid.str(), false));
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_UPDATE_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- }
- if ((req.getState() == WUStateRunning)||(req.getState() == WUStateDebugPaused)||(req.getState() == WUStateDebugRunning))
- {
- WsWuInfo winfo(context, cw);
- winfo.getInfo(resp.updateWorkunit(), WUINFO_All);
- resp.setRedirectUrl(StringBuffer("/WsWorkunits/WUInfo?Wuid=").append(wuid).str());
- AuditSystemAccess(context.queryUserId(), true, "Updated %s", wuid.str());
- return true;
- }
- WorkunitUpdate wu(&cw->lock());
- if(!req.getState_isNull() && (req.getStateOrig_isNull() || req.getState() != req.getStateOrig()))
- {
- if (!req.getStateOrig_isNull() && cw->getState() != (WUState) req.getStateOrig())
- throw MakeStringException(ECLWATCH_CANNOT_UPDATE_WORKUNIT, "Cannot update workunit %s because its state has been changed internally. Please refresh the page and try again.", wuid.str());
- WUState state = (WUState) req.getState();
- if(state < WUStateSize)
- wu->setState(state);
- }
- StringBuffer s;
- if (origValueChanged(req.getJobname(), req.getJobnameOrig(), s))
- wu->setJobName(s.trim().str());
- if (origValueChanged(req.getDescription(), req.getDescriptionOrig(), s.clear()))
- wu->setDebugValue("description", (req.getDescription() && *req.getDescription()) ? s.trim().str() : NULL, true);
- double version = context.getClientVersion();
- if (version > 1.04)
- {
- if (origValueChanged(req.getClusterSelection(), req.getClusterOrig(), s.clear(), false))
- {
- if (!isValidCluster(s.str()))
- throw MakeStringException(ECLWATCH_INVALID_CLUSTER_NAME, "Invalid cluster name: %s", s.str());
- if (req.getState() == WUStateBlocked)
- switchWorkUnitQueue(wu.get(), s.str());
- else if ((req.getState() != WUStateSubmitted) && (req.getState() != WUStateRunning) && (req.getState() != WUStateDebugPaused) && (req.getState() != WUStateDebugRunning))
- wu->setClusterName(s.str());
- }
- }
- WsWuHelpers::setXmlParameters(wu, req.getXmlParams(), (req.getAction()==WUActionExecuteExisting));
- if (notEmpty(req.getQueryText()))
- {
- Owned<IWUQuery> query=wu->updateQuery();
- query->setQueryText(req.getQueryText());
- }
- if (version > 1.34 && notEmpty(req.getQueryMainDefinition()))
- {
- Owned<IWUQuery> query=wu->updateQuery();
- query->setQueryMainDefinition(req.getQueryMainDefinition());
- }
- if (!req.getResultLimit_isNull())
- wu->setResultLimit(req.getResultLimit());
- if (!req.getAction_isNull())
- {
- WUAction action = (WUAction) req.getAction();
- if(action < WUActionSize)
- wu->setAction(action);
- }
- if (!req.getPriorityClass_isNull())
- {
- WUPriorityClass priority = (WUPriorityClass) req.getPriorityClass();
- if(priority<PriorityClassSize)
- wu->setPriority(priority);
- }
- if (!req.getPriorityLevel_isNull())
- wu->setPriorityLevel(req.getPriorityLevel());
- if (origValueChanged(req.getScope(), req.getScopeOrig(), s.clear(), false))
- wu->setWuScope(s.str());
- ForEachItemIn(di, req.getDebugValues())
- {
- IConstDebugValue& item = req.getDebugValues().item(di);
- if (notEmpty(item.getName()))
- wu->setDebugValue(item.getName(), item.getValue(), true);
- }
- ForEachItemIn(ai, req.getApplicationValues())
- {
- IConstApplicationValue& item=req.getApplicationValues().item(ai);
- if(notEmpty(item.getApplication()) && notEmpty(item.getName()))
- wu->setApplicationValue(item.getApplication(), item.getName(), item.getValue(), true);
- }
- wu->commit();
- wu.clear();
- WsWuInfo winfo(context, cw);
- winfo.getInfo(resp.updateWorkunit(), WUINFO_All);
- StringBuffer thorSlaveIP;
- if (version > 1.24 && notEmpty(req.getThorSlaveIP()))
- thorSlaveIP = req.getThorSlaveIP();
- if (thorSlaveIP.length() > 0)
- {
- StringBuffer url;
- url.appendf("/WsWorkunits/WUInfo?Wuid=%s&ThorSlaveIP=%s", wuid.str(), thorSlaveIP.str());
- resp.setRedirectUrl(url.str());
- }
- else
- resp.setRedirectUrl(StringBuffer("/WsWorkunits/WUInfo?Wuid=").append(wuid).str());
- AuditSystemAccess(context.queryUserId(), true, "Updated %s", wuid.str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUCreateAndUpdate(IEspContext &context, IEspWUUpdateRequest &req, IEspWUUpdateResponse &resp)
- {
- try
- {
- if (!context.validateFeatureAccess(OWN_WU_ACCESS, SecAccess_Write, false))
- throw MakeStringException(ECLWATCH_ECL_WU_ACCESS_DENIED, "Failed to create workunit. Permission denied.");
- NewWsWorkunit wu(context);
- SCMStringBuffer wuid;
- wu->getWuid(wuid);
- req.setWuid(wuid.str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return onWUUpdate(context, req, resp);
- }
- static inline StringBuffer &appendUrlParameter(StringBuffer &url, const char *name, const char *value, bool &first)
- {
- if (notEmpty(value))
- {
- url.append(first ? '?' : '&').append(name).append('=').append(value);
- first=false;
- }
- return url;
- }
- bool CWsWorkunitsEx::onWUAction(IEspContext &context, IEspWUActionRequest &req, IEspWUActionResponse &resp)
- {
- try
- {
- StringBuffer sAction(req.getActionType());
- if (!sAction.length())
- throw MakeStringException(ECLWATCH_INVALID_INPUT,"Action not defined.");
- int *action=wuActionTable.getValue(sAction.toLowerCase().str());
- if (!action)
- throw MakeStringException(ECLWATCH_INVALID_INPUT,"Invalid Action '%s'.", sAction.str());
- Owned<IProperties> params = createProperties(true);
- params->setProp("BlockTillFinishTimer", req.getBlockTillFinishTimer());
- if (*action==ActionProtect)
- params->setProp("Protect", streq(sAction.str(), "protect"));
- if (*action==ActionChangeState && streq(sAction.str(), "settofailed"))
- params->setProp("State",4);
- if ((*action==ActionRestore) && !sashaServerIp.isEmpty())
- {
- params->setProp("sashaServerIP", sashaServerIp.get());
- params->setProp("sashaServerPort", sashaServerPort);
- }
- IArrayOf<IConstWUActionResult> results;
- if (doAction(context, req.getWuids(), *action, params, &results) && *action!=ActionDelete && checkRedirect(context))
- {
- StringBuffer redirect;
- if(req.getPageFrom() && strieq(req.getPageFrom(), "wuid"))
- redirect.append("/WsWorkunits/WUInfo?Wuid=").append(req.getWuids().item(0));
- else if (req.getPageFrom() && strieq(req.getPageFrom(), "scheduler"))
- {
- redirect.set("/WsWorkunits/WUShowScheduled");
- bool first=true;
- appendUrlParameter(redirect, "Cluster", req.getEventServer(), first);
- appendUrlParameter(redirect, "EventName", req.getEventName(), first);
- }
- else
- {
- redirect.append("/WsWorkunits/WUQuery");
- bool first=true;
- appendUrlParameter(redirect, "PageSize", req.getPageSize(), first);
- appendUrlParameter(redirect, "PageStartFrom", req.getCurrentPage(), first);
- appendUrlParameter(redirect, "Sortby", req.getSortby(), first);
- appendUrlParameter(redirect, "Descending", req.getDescending() ? "1" : "0", first);
- appendUrlParameter(redirect, "State", req.getState(), first);
- appendUrlParameter(redirect, "Cluster", req.getCluster(), first);
- appendUrlParameter(redirect, "Owner", req.getOwner(), first);
- appendUrlParameter(redirect, "StartDate", req.getStartDate(), first);
- appendUrlParameter(redirect, "EndDate", req.getEndDate(), first);
- appendUrlParameter(redirect, "ECL", req.getECL(), first);
- appendUrlParameter(redirect, "Jobname", req.getJobname(), first);
- }
- resp.setRedirectUrl(redirect.str());
- }
- else
- resp.setActionResults(results);
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUDelete(IEspContext &context, IEspWUDeleteRequest &req, IEspWUDeleteResponse &resp)
- {
- try
- {
- IArrayOf<IConstWUActionResult> results;
- Owned<IProperties> params = createProperties(true);
- params->setProp("BlockTillFinishTimer", req.getBlockTillFinishTimer());
- if (!doAction(context,req.getWuids(), ActionDelete, params, &results))
- resp.setActionResults(results);
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUAbort(IEspContext &context, IEspWUAbortRequest &req, IEspWUAbortResponse &resp)
- {
- try
- {
- IArrayOf<IConstWUActionResult> results;
- Owned<IProperties> params = createProperties(true);
- params->setProp("BlockTillFinishTimer", req.getBlockTillFinishTimer());
- if (!doAction(context,req.getWuids(), ActionAbort, params, &results))
- resp.setActionResults(results);
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUProtect(IEspContext &context, IEspWUProtectRequest &req, IEspWUProtectResponse &resp)\
- {
- try
- {
- IArrayOf<IConstWUActionResult> results;
- Owned<IProperties> params(createProperties(true));
- params->setProp("Protect", req.getProtect());
- params->setProp("BlockTillFinishTimer", 0);
- if (!doAction(context,req.getWuids(), ActionProtect, params, &results))
- resp.setActionResults(results);
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUResubmit(IEspContext &context, IEspWUResubmitRequest &req, IEspWUResubmitResponse &resp)
- {
- try
- {
- Owned<IMultiException> me = MakeMultiException();
- SCMStringBuffer wuid;
- StringArray wuids;
- double version = context.getClientVersion();
- IArrayOf<IEspResubmittedWU> resubmittedWUs;
- for(aindex_t i=0; i<req.getWuids().length();i++)
- {
- StringBuffer requestWuid = req.getWuids().item(i);
- WsWuHelpers::checkAndTrimWorkunit("WUResubmit", requestWuid);
- ensureWsWorkunitAccess(context, requestWuid.str(), SecAccess_Write);
- wuid.set(requestWuid.str());
- try
- {
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- if(req.getCloneWorkunit() || req.getRecompile())
- {
- Owned<IConstWorkUnit> src(factory->openWorkUnit(wuid.str(), false));
- NewWsWorkunit wu(factory, context);
- wu->getWuid(wuid);
- queryExtendedWU(wu)->copyWorkUnit(src, false);
- SCMStringBuffer token;
- wu->setSecurityToken(createToken(wuid.str(), context.queryUserId(), context.queryPassword(), token).str());
- }
- wuids.append(wuid.str());
- Owned<IConstWorkUnit> cw(factory->openWorkUnit(wuid.str(), false));
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_OPEN_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- //Dont allow resubmit of someone else's workunit
- if (context.querySecManager())
- {
- IUserDescriptor * owner = cw->queryUserDescriptor();
- if (!owner)
- throw MakeStringException(ECLWATCH_CANNOT_SUBMIT_WORKUNIT,"Workunit User Descriptor missing on %s", wuid.str());
- StringBuffer ownerUserName;
- owner->getUserName(ownerUserName);
- if (strcmp(context.queryUser()->getName(), ownerUserName.str()))
- throw MakeStringException(ECLWATCH_CANNOT_SUBMIT_WORKUNIT,"Cannot resubmit another user's workunit %s.", wuid.str());
- }
- WsWuHelpers::submitWsWorkunit(context, cw, NULL, NULL, 0, req.getRecompile(), req.getResetWorkflow(), false);
- if (version < 1.40)
- continue;
- Owned<IEspResubmittedWU> resubmittedWU = createResubmittedWU();
- resubmittedWU->setWUID(wuid.str());
- if (!streq(requestWuid.str(), wuid.str()))
- resubmittedWU->setParentWUID(requestWuid.str());
- resubmittedWUs.append(*resubmittedWU.getClear());
- }
- catch (IException *E)
- {
- me->append(*E);
- }
- catch (...)
- {
- me->append(*MakeStringException(0,"Unknown exception submitting %s",wuid.str()));
- }
- }
- if(me->ordinality())
- throw me.getLink();
- int timeToWait = req.getBlockTillFinishTimer();
- if (timeToWait != 0)
- {
- for(aindex_t i=0; i<wuids.length(); i++)
- waitForWorkUnitToComplete(wuids.item(i), timeToWait);
- }
- if (version >= 1.40)
- resp.setWUs(resubmittedWUs);
- if(wuids.length()==1)
- {
- resp.setRedirectUrl(StringBuffer("/WsWorkunits/WUInfo?Wuid=").append(wuids.item(0)));
- }
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUPushEvent(IEspContext &context, IEspWUPushEventRequest &req, IEspWUPushEventResponse &resp)
- {
- try
- {
- const char *name = req.getEventName();
- const char *text = req.getEventText();
- const char *target = NULL;
- if (notEmpty(name) && notEmpty(text))
- {
- Owned<IScheduleEventPusher> pusher(getScheduleEventPusher());
- pusher->push(name, text, target);
- StringBuffer redirect("/WsWorkunits/WUShowScheduled");
- bool first=true;
- appendUrlParameter(redirect, "PushEventName", name, first);
- appendUrlParameter(redirect, "PushEventText", text, first);
- resp.setRedirectUrl(redirect.str());
- return true;
- }
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return false;
- }
- bool CWsWorkunitsEx::onWUSchedule(IEspContext &context, IEspWUScheduleRequest &req, IEspWUScheduleResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUSchedule", wuid);
- const char* cluster = req.getCluster();
- if (isEmpty(cluster))
- throw MakeStringException(ECLWATCH_INVALID_INPUT,"No Cluster defined.");
- if (!isValidCluster(cluster))
- throw MakeStringException(ECLWATCH_INVALID_CLUSTER_NAME, "Invalid cluster name: %s", cluster);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- WorkunitUpdate wu(factory->updateWorkUnit(wuid.str()));
- ensureWsWorkunitAccess(context, *wu.get(), SecAccess_Write);
- switch(wu->getState())
- {
- case WUStateDebugPaused:
- case WUStateDebugRunning:
- case WUStateRunning:
- case WUStateAborting:
- case WUStateBlocked:
- {
- SCMStringBuffer descr;
- throw MakeStringException(ECLWATCH_CANNOT_SCHEDULE_WORKUNIT, "Cannot schedule the workunit. Workunit state is '%s'.", wu->getStateDesc(descr).str());
- }
- }
- wu->clearExceptions();
- wu->setClusterName(cluster);
- if (notEmpty(req.getWhen()))
- {
- WsWuDateTime dt;
- dt.setString(req.getWhen());
- wu->setTimeScheduled(dt);
- }
- if(notEmpty(req.getSnapshot()))
- wu->setSnapshot(req.getSnapshot());
- wu->setState(WUStateScheduled);
- if (req.getMaxRunTime())
- wu->setDebugValueInt("maxRunTime", req.getMaxRunTime(), true);
- SCMStringBuffer token;
- wu->setSecurityToken(createToken(wuid.str(), context.queryUserId(), context.queryPassword(), token).str());
- AuditSystemAccess(context.queryUserId(), true, "Scheduled %s", wuid.str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUSubmit(IEspContext &context, IEspWUSubmitRequest &req, IEspWUSubmitResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUSubmit", wuid);
- const char *cluster = req.getCluster();
- if (isEmpty(cluster))
- throw MakeStringException(ECLWATCH_INVALID_INPUT,"No Cluster defined.");
- if (!isValidCluster(cluster))
- throw MakeStringException(ECLWATCH_INVALID_CLUSTER_NAME, "Invalid cluster name: %s", cluster);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_OPEN_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- if (cw->getAction()==WUActionExecuteExisting)
- {
- ExecuteExistingQueryInfo info(cw);
- if (info.queryset.isEmpty() || info.query.isEmpty())
- {
- WorkunitUpdate wu(&cw->lock());
- throw WsWuHelpers::noteException(wu, MakeStringException(ECLWATCH_INVALID_INPUT,"Queryset and/or query not specified"));
- }
- WsWuHelpers::runWsWuQuery(context, cw, info.queryset.sget(), info.query.sget(), cluster, NULL);
- }
- else
- WsWuHelpers::submitWsWorkunit(context, cw, cluster, req.getSnapshot(), req.getMaxRunTime(), true, false, false);
- if (req.getBlockTillFinishTimer() != 0)
- waitForWorkUnitToComplete(wuid.str(), req.getBlockTillFinishTimer());
- resp.setRedirectUrl(StringBuffer("/WsWorkunits/WUInfo?Wuid=").append(wuid).str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWURun(IEspContext &context, IEspWURunRequest &req, IEspWURunResponse &resp)
- {
- try
- {
- const char *cluster = req.getCluster();
- if (notEmpty(cluster) && !isValidCluster(cluster))
- throw MakeStringException(ECLWATCH_INVALID_CLUSTER_NAME, "Invalid cluster name: %s", cluster);
- StringBuffer wuidStr = req.getWuid();
- const char* runWuid = wuidStr.trim().str();
- StringBuffer wuid;
- if (runWuid && *runWuid)
- {
- if (!looksLikeAWuid(runWuid))
- throw MakeStringException(ECLWATCH_INVALID_INPUT, "Invalid Workunit ID: %s", runWuid);
- if (req.getCloneWorkunit())
- WsWuHelpers::runWsWorkunit(context, wuid, runWuid, cluster, req.getInput(), &req.getVariables(), &req.getDebugValues());
- else
- {
- WsWuHelpers::submitWsWorkunit(context, runWuid, cluster, NULL, 0, false, true, true, req.getInput(), &req.getVariables(), &req.getDebugValues());
- wuid.set(runWuid);
- }
- }
- else if (notEmpty(req.getQuerySet()) && notEmpty(req.getQuery()))
- WsWuHelpers::runWsWuQuery(context, wuid, req.getQuerySet(), req.getQuery(), cluster, req.getInput());
- else
- throw MakeStringException(ECLWATCH_MISSING_PARAMS,"Workunit or Query required");
- int timeToWait = req.getWait();
- if (timeToWait != 0)
- waitForWorkUnitToComplete(wuid.str(), timeToWait);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if (!cw)
- throw MakeStringException(ECLWATCH_CANNOT_UPDATE_WORKUNIT,"Cannot open workunit %s.", wuid.str());
- SCMStringBuffer stateDesc;
- resp.setState(cw->getStateDesc(stateDesc).str());
- resp.setWuid(wuid.str());
- switch (cw->getState())
- {
- case WUStateCompleted:
- case WUStateFailed:
- case WUStateUnknown:
- {
- SCMStringBuffer result;
- unsigned flags = WorkUnitXML_SeverityTags;
- if (req.getNoRootTag())
- flags |= WorkUnitXML_NoRoot;
- getFullWorkUnitResultsXML(context.queryUserId(), context.queryPassword(), cw.get(), result, flags);
- resp.setResults(result.str());
- break;
- }
- default:
- break;
- }
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUWaitCompiled(IEspContext &context, IEspWUWaitRequest &req, IEspWUWaitResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUWaitCompiled", wuid);
- secWaitForWorkUnitToCompile(wuid.str(), *context.querySecManager(), *context.queryUser(), req.getWait());
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_OPEN_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- resp.setStateID(cw->getState());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUWaitComplete(IEspContext &context, IEspWUWaitRequest &req, IEspWUWaitResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUWaitComplete", wuid);
- resp.setStateID(secWaitForWorkUnitToComplete(wuid.str(), *context.querySecManager(), *context.queryUser(), req.getWait(), req.getReturnOnWait()));
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUCDebug(IEspContext &context, IEspWUDebugRequest &req, IEspWUDebugResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUCDebug", wuid);
- StringBuffer result;
- secDebugWorkunit(wuid.str(), *context.querySecManager(), *context.queryUser(), req.getCommand(), result);
- resp.setResult(result);
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUSyntaxCheckECL(IEspContext &context, IEspWUSyntaxCheckRequest &req, IEspWUSyntaxCheckResponse &resp)
- {
- try
- {
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- NewWsWorkunit wu(factory, context);
- wu->setAction(WUActionCheck);
- if(notEmpty(req.getModuleName()) && notEmpty(req.getAttributeName()))
- {
- wu->setApplicationValue("SyntaxCheck", "ModuleName", req.getModuleName(), true);
- wu->setApplicationValue("SyntaxCheck", "AttributeName", req.getAttributeName(), true);
- }
- ForEachItemIn(di, req.getDebugValues())
- {
- IConstDebugValue& item=req.getDebugValues().item(di);
- if(notEmpty(item.getName()))
- wu->setDebugValue(item.getName(), item.getValue(), true);
- }
- wu.setQueryText(req.getECL());
- SCMStringBuffer wuid;
- wu->getWuid(wuid);
- wu->commit();
- wu.clear();
- WsWuHelpers::submitWsWorkunit(context, wuid.str(), req.getCluster(), req.getSnapshot(), 0, true, false, false);
- waitForWorkUnitToComplete(wuid.str(), req.getTimeToWait());
- Owned<IConstWorkUnit> cw(factory->openWorkUnit(wuid.str(), false));
- WsWUExceptions errors(*cw);
- resp.setErrors(errors);
- cw.clear();
- factory->deleteWorkUnit(wuid.str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUCompileECL(IEspContext &context, IEspWUCompileECLRequest &req, IEspWUCompileECLResponse &resp)
- {
- try
- {
- ensureWsCreateWorkunitAccess(context);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- NewWsWorkunit wu(factory, context);
- if(req.getIncludeComplexity())
- {
- wu->setAction(WUActionCompile);
- wu->setDebugValueInt("calculateComplexity",1,true);
- }
- else
- wu->setAction(WUActionCheck);
- if(req.getModuleName() && req.getAttributeName())
- {
- wu->setApplicationValue("SyntaxCheck","ModuleName",req.getModuleName(),true);
- wu->setApplicationValue("SyntaxCheck","AttributeName",req.getAttributeName(),true);
- }
- if(req.getIncludeDependencies())
- wu->setApplicationValueInt("SyntaxCheck","IncludeDependencies",1,true);
- wu.setQueryText(req.getECL());
- SCMStringBuffer wuid;
- wu->getWuid(wuid);
- wu.clear();
- WsWuHelpers::submitWsWorkunit(context, wuid.str(), req.getCluster(), req.getSnapshot(), 0, true, false, false);
- waitForWorkUnitToComplete(wuid.str(),req.getTimeToWait());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- SCMStringBuffer s;
- cw->getDebugValue("__Calculated__Complexity__",s);
- if(s.length())
- resp.setComplexity(s.str());
- WsWUExceptions errors(*cw);
- resp.setErrors(errors);
- if(!errors.ErrCount())
- {
- IArrayOf<IEspWUECLAttribute> dependencies;
- for(unsigned count=1;;count++)
- {
- SCMStringBuffer xml;
- cw->getApplicationValue("SyntaxCheck",StringBuffer("Dependency").append(count).str(),xml);
- if(!xml.length())
- break;
- Owned<IPropertyTree> dep=createPTreeFromXMLString(xml.str(), ipt_caseInsensitive);
- if(!dep)
- continue;
- Owned<IEspWUECLAttribute> att = createWUECLAttribute("","");
- att->setModuleName(dep->queryProp("@module"));
- att->setAttributeName(dep->queryProp("@name"));
- int flags = dep->getPropInt("@flags",0);
- if(flags & ob_locked)
- {
- if(flags & ob_lockedself)
- att->setIsCheckedOut(true);
- else
- att->setIsLocked(true);
- }
- if(flags & ob_sandbox)
- att->setIsSandbox(true);
- if(flags & ob_orphaned)
- att->setIsOrphaned(true);
- dependencies.append(*att.getLink());
- }
- resp.setDependencies(dependencies);
- }
- cw.clear();
- factory->deleteWorkUnit(wuid.str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUGetDependancyTrees(IEspContext& context, IEspWUGetDependancyTreesRequest& req, IEspWUGetDependancyTreesResponse& resp)
- {
- try
- {
- DBGLOG("WUGetDependancyTrees");
- unsigned int timeMilliSec = 500;
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- NewWsWorkunit wu(factory, context);
- wu->setAction(WUActionCheck);
- if (notEmpty(req.getCluster()))
- wu->setClusterName(req.getCluster());
- if (notEmpty(req.getSnapshot()))
- wu->setSnapshot(req.getSnapshot());
- wu->setDebugValue("gatherDependenciesSelection",notEmpty(req.getItems()) ? req.getItems() : NULL,true);
- if (context.getClientVersion() > 1.12)
- {
- wu->setDebugValueInt("gatherDependencies", 1, true);
- const char *timeout = req.getTimeoutMilliSec();
- if (notEmpty(timeout))
- {
- const char *finger = timeout;
- while (*finger)
- {
- if (!isdigit(*finger++))
- throw MakeStringException(ECLWATCH_INVALID_INPUT, "Incorrect timeout value");
- }
- timeMilliSec = atol(timeout);
- }
- }
- SCMStringBuffer wuid;
- wu->getWuid(wuid);
- wu->commit();
- wu.clear();
- ensureWsWorkunitAccess(context, wuid.str(), SecAccess_Read);
- WsWuHelpers::submitWsWorkunit(context, wuid.str(), req.getCluster(), req.getSnapshot(), 0, true, false, false);
- int state = waitForWorkUnitToComplete(wuid.str(), timeMilliSec);
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- WsWUExceptions errors(*cw);
- resp.setErrors(errors);
- MemoryBuffer temp;
- MemoryBuffer2IDataVal xmlresult(temp);
- Owned<IConstWUResult> result = cw->getResultBySequence(0);
- if (result)
- {
- result->getResultRaw(xmlresult, NULL, NULL);
- resp.setDependancyTrees(temp);
- }
- wu.setown(&cw->lock());
- wu->setState(WUStateAborted);
- wu->commit();
- wu.clear();
- factory->deleteWorkUnit(wuid.str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool getWsWuInfoFromSasha(IEspContext &context, SocketEndpoint &ep, const char* wuid, IEspECLWorkunit *info)
- {
- Owned<INode> node = createINode(ep);
- Owned<ISashaCommand> cmd = createSashaCommand();
- cmd->addId(wuid);
- cmd->setAction(SCA_GET);
- if (!cmd->send(node, 1*60*1000))
- {
- StringBuffer url;
- DBGLOG("Could not connect to Sasha server at %s", ep.getUrlStr(url).str());
- throw MakeStringException(ECLWATCH_CANNOT_CONNECT_ARCHIVE_SERVER,"Cannot connect to archive server at %s.", url.str());
- }
- if (cmd->numIds()==0)
- {
- DBGLOG("Could not read archived workunit %s",wuid);
- throw MakeStringException(ECLWATCH_CANNOT_GET_WORKUNIT,"Cannot read workunit %s.",wuid);
- }
- unsigned num = cmd->numResults();
- if (num < 1)
- return false;
- StringBuffer res;
- cmd->getResult(0, res);
- if(res.length() < 1)
- return false;
- Owned<IPropertyTree> wpt = createPTreeFromXMLString(res.str());
- if (!wpt)
- return false;
- const char * owner = wpt->queryProp("@submitID");
- ensureWsWorkunitAccessByOwnerId(context, owner, SecAccess_Read);
- info->setWuid(wuid);
- info->setArchived(true);
- if (notEmpty(owner))
- info->setOwner(owner);
- const char * state = wpt->queryProp("@state");
- if (notEmpty(state))
- info->setState(state);
- const char * cluster = wpt->queryProp("@clusterName");
- if (notEmpty(cluster))
- info->setCluster(cluster);
- const char * scope = wpt->queryProp("@scope");
- if (notEmpty(scope))
- info->setScope(scope);
- const char * jobName = wpt->queryProp("@jobName");
- if (notEmpty(jobName))
- info->setJobname(jobName);
- const char * description = wpt->queryProp("Debug/description");
- if (notEmpty(description))
- info->setDescription(description);
- const char * queryText = wpt->queryProp("Query/Text");
- if (notEmpty(queryText))
- info->updateQuery().setText(queryText);
- const char * protectedWU = wpt->queryProp("@protected");
- info->setProtected((protectedWU && *protectedWU!='0'));
- return true;
- }
- #define WUDETAILS_REFRESH_MINS 1
- void getArchivedWUInfo(IEspContext &context, const char* sashaServerIP, unsigned sashaServerPort, const char *wuid, IEspWUInfoResponse &resp)
- {
- SocketEndpoint ep;
- if (sashaServerIP && *sashaServerIP)
- ep.set(sashaServerIP, sashaServerPort);
- else
- getSashaNode(ep);
- if (getWsWuInfoFromSasha(context, ep, wuid, &resp.updateWorkunit()))
- {
- resp.setAutoRefresh(WUDETAILS_REFRESH_MINS);
- resp.setCanCompile(false);
- return;
- }
- throw MakeStringException(ECLWATCH_CANNOT_GET_WORKUNIT, "Cannot find workunit %s.", wuid);
- return;
- }
- #define WUDETAILS_REFRESH_MINS 1
- bool CWsWorkunitsEx::onWUInfo(IEspContext &context, IEspWUInfoRequest &req, IEspWUInfoResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUInfo", wuid);
- double version = context.getClientVersion();
- if (req.getType() && strieq(req.getType(), "archived workunits"))
- getArchivedWUInfo(context, sashaServerIp.get(), sashaServerPort, wuid.str(), resp);
- else
- {
- try
- {
- //The access is checked here because getArchivedWUInfo() has its own access check.
- ensureWsWorkunitAccess(context, wuid.str(), SecAccess_Read);
- unsigned flags=0;
- if (req.getTruncateEclTo64k())
- flags|=WUINFO_TruncateEclTo64k;
- if (req.getIncludeExceptions())
- flags|=WUINFO_IncludeExceptions;
- if (req.getIncludeGraphs())
- flags|=WUINFO_IncludeGraphs;
- if (req.getIncludeSourceFiles())
- flags|=WUINFO_IncludeSourceFiles;
- if (req.getIncludeResults())
- flags|=WUINFO_IncludeResults;
- if (req.getIncludeVariables())
- flags|=WUINFO_IncludeVariables;
- if (req.getIncludeTimers())
- flags|=WUINFO_IncludeTimers;
- if (req.getIncludeDebugValues())
- flags|=WUINFO_IncludeDebugValues;
- if (req.getIncludeApplicationValues())
- flags|=WUINFO_IncludeApplicationValues;
- if (req.getIncludeWorkflows())
- flags|=WUINFO_IncludeWorkflows;
- if (!req.getSuppressResultSchemas())
- flags|=WUINFO_IncludeEclSchemas;
- if (req.getIncludeXmlSchemas())
- flags|=WUINFO_IncludeXmlSchema;
- if (req.getIncludeResultsViewNames())
- flags|=WUINFO_IncludeResultsViewNames;
- if (req.getIncludeResourceURLs())
- flags|=WUINFO_IncludeResourceURLs;
- WsWuInfo winfo(context, wuid.str());
- winfo.getInfo(resp.updateWorkunit(), flags);
- if (req.getIncludeResultsViewNames()||req.getIncludeResourceURLs()||(version >= 1.50))
- {
- StringArray views, urls;
- winfo.getResourceInfo(views, urls, WUINFO_IncludeResultsViewNames|WUINFO_IncludeResourceURLs);
- IEspECLWorkunit& eclWU = resp.updateWorkunit();
- if (req.getIncludeResultsViewNames())
- resp.setResultViews(views);
- if (req.getIncludeResourceURLs())
- eclWU.setResourceURLs(urls);
- if (version >= 1.50)
- {
- eclWU.setResultViewCount(views.length());
- eclWU.setResourceURLCount(urls.length());
- }
- }
- }
- catch (IException *e)
- {
- if (e->errorCode() != ECLWATCH_CANNOT_OPEN_WORKUNIT)
- throw e;
- getArchivedWUInfo(context, sashaServerIp.get(), sashaServerPort, wuid.str(), resp);
- e->Release();
- }
- switch (resp.getWorkunit().getStateID())
- {
- case WUStateCompiling:
- case WUStateCompiled:
- case WUStateScheduled:
- case WUStateSubmitted:
- case WUStateRunning:
- case WUStateAborting:
- case WUStateWait:
- case WUStateUploadingFiles:
- case WUStateDebugPaused:
- case WUStateDebugRunning:
- resp.setAutoRefresh(WUDETAILS_REFRESH_MINS);
- break;
- case WUStateBlocked:
- resp.setAutoRefresh(WUDETAILS_REFRESH_MINS*5);
- break;
- }
- resp.setCanCompile(notEmpty(context.queryUserId()));
- if (context.getClientVersion() > 1.24 && notEmpty(req.getThorSlaveIP()))
- resp.setThorSlaveIP(req.getThorSlaveIP());
- }
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUInfoDetails(IEspContext &context, IEspWUInfoRequest &req, IEspWUInfoResponse &resp)
- {
- return onWUInfo(context, req, resp);
- }
- bool CWsWorkunitsEx::onWUResultView(IEspContext &context, IEspWUResultViewRequest &req, IEspWUResultViewResponse &resp)
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUResultView", wuid);
- ensureWsWorkunitAccess(context, wuid.str(), SecAccess_Read);
- Owned<IWuWebView> wv = createWuWebView(wuid.str(), NULL, getCFD(), true);
- StringBuffer html;
- wv->renderSingleResult(req.getViewName(), req.getResultName(), html);
- resp.setResult(html.str());
- resp.setResult_mimetype("text/html");
- return true;
- }
- void doWUQueryBySingleWuid(IEspContext &context, const char *wuid, IEspWUQueryResponse &resp)
- {
- Owned<IEspECLWorkunit> info= createECLWorkunit("","");
- WsWuInfo winfo(context, wuid);
- winfo.getCommon(*info, 0);
- IArrayOf<IEspECLWorkunit> results;
- results.append(*info.getClear());
- resp.setWorkunits(results);
- resp.setPageSize(1);
- resp.setCount(1);
- }
- void doWUQueryByFile(IEspContext &context, const char *logicalFile, IEspWUQueryResponse &resp)
- {
- StringBuffer wuid;
- getWuidFromLogicalFileName(context, logicalFile, wuid);
- if (!wuid.length())
- throw MakeStringException(ECLWATCH_CANNOT_GET_WORKUNIT,"Cannot find the workunit for file %s.", logicalFile);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw= factory->openWorkUnit(wuid.str(), false);
- if (!cw)
- throw MakeStringException(ECLWATCH_CANNOT_OPEN_WORKUNIT,"Cannot find the workunit for file %s.", logicalFile);
- if (getWsWorkunitAccess(context, *cw) < SecAccess_Read)
- throw MakeStringException(ECLWATCH_ECL_WU_ACCESS_DENIED,"Cannot access the workunit for file %s.",logicalFile);
- SCMStringBuffer parent;
- if (!cw->getParentWuid(parent).length())
- doWUQueryBySingleWuid(context, wuid.str(), resp);
- resp.setFirst(false);
- resp.setPageSize(1);
- resp.setCount(1);
- }
- void doWUQueryByXPath(IEspContext &context, IEspWUQueryRequest & req, IEspWUQueryResponse & resp)
- {
- IArrayOf<IEspECLWorkunit> results;
- WsWuSearch wlist(context,req.getOwner(),req.getState(),req.getCluster(),req.getStartDate(),req.getEndDate(),req.getECL(),req.getJobname(),req.getApplicationName(),req.getApplicationKey(),req.getApplicationData());
- int count=(int)req.getPageSize();
- if (!count)
- count=100;
- if (wlist.getSize() < 1)
- {
- resp.setNumWUs(0);
- return;
- }
- if (wlist.getSize() < count)
- count = (int) wlist.getSize() - 1;
- WsWuSearch::iterator begin, end;
- if(notEmpty(req.getAfter()))
- {
- begin=wlist.locate(req.getAfter());
- end=min(begin+count,wlist.end());
- }
- else if (notEmpty(req.getBefore()))
- {
- end=wlist.locate(req.getBefore());
- begin=max(end-count,wlist.begin());
- }
- else
- {
- begin=wlist.begin();
- end=min(begin+count,wlist.end());
- }
- if(begin>wlist.begin() && begin<wlist.end())
- resp.setCurrent(begin->c_str());
- if (context.getClientVersion() > 1.02)
- {
- resp.setPageStartFrom(begin - wlist.begin() + 1);
- resp.setNumWUs((int)wlist.getSize());
- resp.setCount(end - begin);
- }
- if(end<wlist.end())
- resp.setNext(end->c_str());
- for(;begin!=end;begin++)
- {
- Owned<IEspECLWorkunit> info = createECLWorkunit("","");
- WsWuInfo winfo(context, begin->c_str());
- winfo.getCommon(*info, 0);
- results.append(*info.getClear());
- }
- resp.setPageSize(abs(count));
- resp.setWorkunits(results);
- return;
- }
- bool addWUQueryFilter(WUSortField *filters, unsigned short &count, MemoryBuffer &buff, const char *name, WUSortField value)
- {
- if (isEmpty(name))
- return false;
- filters[count++] = value;
- buff.append(name);
- return true;
- }
- bool addWUQueryFilterTime(WUSortField *filters, unsigned short &count, MemoryBuffer &buff, const char *stime, WUSortField value)
- {
- if (isEmpty(stime))
- return false;
- CDateTime dt;
- dt.setString(stime, NULL, true);
- unsigned year, month, day, hour, minute, second, nano;
- dt.getDate(year, month, day, true);
- dt.getTime(hour, minute, second, nano, true);
- VStringBuffer wuid("W%4d%02d%02d-%02d%02d%02d",year,month,day,hour,minute,second);
- filters[count++] = value;
- buff.append(wuid.str());
- return true;
- }
- bool addWUQueryFilterApplication(WUSortField *filters, unsigned short &count, MemoryBuffer &buff, const char *appname, const char *appkey, const char *appdata)
- {
- if (isEmpty(appname) && isEmpty(appkey) && isEmpty(appdata)) //no application filter
- return false;
- VStringBuffer path("Application/%s/%s", appname && *appname ? appname : "*", appkey && *appkey ? appkey : "*");
- if(appdata && *appdata)
- path.append("=?~\"").append(appdata).append("\"");
- filters[count++] = WUSFcustom;
- buff.append(path.str());
- return true;
- }
- void doWUQueryWithSort(IEspContext &context, IEspWUQueryRequest & req, IEspWUQueryResponse & resp)
- {
- SecAccessFlags accessOwn;
- SecAccessFlags accessOthers;
- getUserWuAccessFlags(context, accessOwn, accessOthers, true);
- double version = context.getClientVersion();
- IArrayOf<IEspECLWorkunit> results;
- int begin = 0;
- unsigned int count = 100;
- int pagesize = 100;
- if (version > 1.01)
- {
- pagesize = (int)req.getPageSize();
- if (!req.getCount_isNull())
- pagesize = req.getCount();
- if(pagesize < 1)
- pagesize = 100;
- begin = (int)req.getPageStartFrom();
- }
- else
- {
- count=(unsigned)req.getCount();
- if(!count)
- count=100;
- if (notEmpty(req.getAfter()))
- begin=atoi(req.getAfter());
- else if (notEmpty(req.getBefore()))
- begin=atoi(req.getBefore())-count;
- if (begin < 0)
- begin = 0;
- pagesize = count;
- }
- WUSortField sortorder[2] = {(WUSortField) (WUSFwuid | WUSFreverse), WUSFterm};
- if(notEmpty(req.getSortby()))
- {
- const char *sortby = req.getSortby();
- if (strieq(sortby, "Owner"))
- sortorder[0] = WUSFuser;
- else if (strieq(sortby, "JobName"))
- sortorder[0] = WUSFjob;
- else if (strieq(sortby, "Cluster"))
- sortorder[0] = WUSFcluster;
- else if (strieq(sortby, "RoxieCluster"))
- sortorder[0] = WUSFroxiecluster;
- else if (strieq(sortby, "Protected"))
- sortorder[0] = WUSFprotected;
- else if (strieq(sortby, "State"))
- sortorder[0] = WUSFstate;
- else if (strieq(sortby, "ThorTime"))
- sortorder[0] = (WUSortField) (WUSFtotalthortime+WUSFnumeric);
- else
- sortorder[0] = WUSFwuid;
- sortorder[0] = (WUSortField) (sortorder[0] | WUSFnocase);
- bool descending = req.getDescending();
- if (descending)
- sortorder[0] = (WUSortField) (sortorder[0] | WUSFreverse);
- }
- WUSortField filters[10];
- unsigned short filterCount = 0;
- MemoryBuffer filterbuf;
- bool bDoubleCheckState = false;
- if(req.getState() && *req.getState())
- {
- filters[filterCount++] = WUSFstate;
- if (!strieq(req.getState(), "unknown"))
- filterbuf.append(req.getState());
- else
- filterbuf.append("");
- if (strieq(req.getState(), "submitted"))
- bDoubleCheckState = true;
- }
- addWUQueryFilter(filters, filterCount, filterbuf, req.getWuid(), WUSFwildwuid);
- addWUQueryFilter(filters, filterCount, filterbuf, req.getCluster(), WUSFcluster);
- if(version > 1.07)
- addWUQueryFilter(filters, filterCount, filterbuf, req.getRoxieCluster(), WUSFroxiecluster);
- addWUQueryFilter(filters, filterCount, filterbuf, req.getLogicalFile(), WUSFfileread);
- addWUQueryFilter(filters, filterCount, filterbuf, req.getOwner(), (WUSortField) (WUSFuser | WUSFnocase));
- addWUQueryFilter(filters, filterCount, filterbuf, req.getJobname(), (WUSortField) (WUSFjob | WUSFnocase));
- addWUQueryFilter(filters, filterCount, filterbuf, req.getECL(), (WUSortField) (WUSFecl | WUSFwild));
- addWUQueryFilterTime(filters, filterCount, filterbuf, req.getStartDate(), WUSFwuid);
- addWUQueryFilterTime(filters, filterCount, filterbuf, req.getEndDate(), WUSFwuidhigh);
- addWUQueryFilterApplication(filters, filterCount, filterbuf, req.getApplicationName(), req.getApplicationKey(), req.getApplicationData());
- filters[filterCount] = WUSFterm;
- __int64 cacheHint = 0;
- if (!req.getCacheHint_isNull())
- cacheHint = req.getCacheHint();
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- unsigned numWUs;
- Owned<IConstWorkUnitIterator> it = factory->getWorkUnitsSorted(sortorder, filters, filterbuf.bufferBase(), begin, pagesize+1, "", &cacheHint, &numWUs);
- if (version >= 1.41)
- resp.setCacheHint(cacheHint);
- unsigned actualCount = 0;
- ForEach(*it)
- {
- IConstWorkUnit& cw = it->query();
- if (chooseWuAccessFlagsByOwnership(context.queryUserId(), cw, accessOwn, accessOthers) < SecAccess_Read)
- {
- numWUs--;
- continue;
- }
- if (bDoubleCheckState && (cw.getState() != WUStateSubmitted))
- {
- numWUs--;
- continue;
- }
- SCMStringBuffer parent;
- if (!cw.getParentWuid(parent).length())
- {
- const char* wuid = cw.getWuid(parent).str();
- if (!looksLikeAWuid(wuid))
- {
- numWUs--;
- continue;
- }
- actualCount++;
- Owned<IEspECLWorkunit> info = createECLWorkunit("","");
- WsWuInfo winfo(context, wuid);
- winfo.getCommon(*info, 0);
- results.append(*info.getClear());
- }
- }
- if (version > 1.02)
- {
- resp.setPageStartFrom(begin+1);
- resp.setNumWUs(numWUs);
- if (results.length() > (aindex_t)pagesize)
- results.pop();
- if(unsigned (begin + pagesize) < numWUs)
- {
- resp.setNextPage(begin + pagesize);
- resp.setPageEndAt(begin + pagesize);
- int last = begin + pagesize;
- while (numWUs > (unsigned) last + pagesize)
- last += pagesize;
- resp.setLastPage(last);
- }
- else
- {
- resp.setNextPage(-1);
- resp.setPageEndAt(numWUs);
- }
- if(begin > 0)
- {
- resp.setFirst(false);
- if (begin - pagesize > 0)
- resp.setPrevPage(begin - pagesize);
- else
- resp.setPrevPage(0);
- }
- resp.setPageSize(pagesize);
- }
- else
- {
- if(begin>0 && actualCount > 0)
- {
- char buf[10];
- itoa(begin, buf, 10);
- resp.setCurrent(buf);
- }
- if(count<actualCount)
- {
- char buf[10];
- itoa(begin+count, buf, 10);
- resp.setNext(buf);
- resp.setNumWUs(numWUs);
- if (results.length() > count)
- results.pop();
- }
- if(begin == 0 && actualCount <= count)
- resp.setFirst(false);
- resp.setCount(count);
- }
- resp.setWorkunits(results);
- return;
- }
- void doWUQueryFromArchive(IEspContext &context, const char* sashaServerIP, unsigned sashaServerPort,
- ArchivedWuCache &archivedWuCache, unsigned cacheMinutes, IEspWUQueryRequest & req, IEspWUQueryResponse & resp)
- {
- class CArchivedWUsReader : public CInterface, implements IArchivedWUsReader
- {
- IEspContext& context;
- IEspWUQueryRequest& req;
- unsigned pageFrom, pageSize;
- StringAttr sashaServerIP;
- unsigned sashaServerPort;
- unsigned cacheMinutes;
- StringBuffer filterStr;
- ArchivedWuCache& archivedWuCache;
- unsigned numberOfWUsReturned;
- bool hasMoreWU;
- void readDateFilters(StringBuffer& from, StringBuffer& to)
- {
- CDateTime timeFrom, timeTo;
- if(notEmpty(req.getEndDate()))
- timeTo.setString(req.getEndDate(), NULL, true);
- else
- timeTo.setNow();
- unsigned year, month, day, hour, minute, second, nano;
- timeTo.getDate(year, month, day, true);
- timeTo.getTime(hour, minute, second, nano, true);
- to.setf("%4d%02d%02d%02d%02d", year, month, day, hour, minute);
- if(!notEmpty(req.getStartDate()))
- return;
- timeFrom.setString(req.getStartDate(), NULL, true);
- if (timeFrom >= timeTo)
- return;
- unsigned year0, month0, day0, hour0, minute0, second0, nano0;
- timeFrom.getDate(year0, month0, day0, true);
- timeFrom.getTime(hour0, minute0, second0, nano0, true);
- from.setf("%4d%02d%02d%02d%02d", year0, month0, day0, hour0, minute0);
- return;
- }
- bool addToFilterString(const char *name, const char *value)
- {
- if (isEmpty(name) || isEmpty(value))
- return false;
- if (filterStr.length())
- filterStr.append(';');
- filterStr.append(name).append("=").append(value);
- return true;
- }
- bool addToFilterString(const char *name, unsigned value)
- {
- if (isEmpty(name))
- return false;
- if (filterStr.length())
- filterStr.append(';');
- filterStr.append(name).append("=").append(value);
- return true;
- }
- void setFilterString()
- {
- addToFilterString("cluster", req.getCluster());
- addToFilterString("owner", req.getOwner());
- addToFilterString("jobName", req.getJobname());
- addToFilterString("state", req.getState());
- addToFilterString("timeFrom", req.getStartDate());
- addToFilterString("timeTo", req.getEndDate());
- addToFilterString("pageStart", pageFrom);
- addToFilterString("pageSize", pageSize);
- if (sashaServerIP && *sashaServerIP)
- {
- addToFilterString("sashaServerIP", sashaServerIP.get());
- addToFilterString("sashaServerPort", sashaServerPort);
- }
- }
- void setSashaCommand(INode* sashaserver, ISashaCommand* cmd)
- {
- cmd->setAction(SCA_LIST);
- cmd->setOutputFormat("owner,jobname,cluster,state");
- cmd->setOnline(false);
- cmd->setArchived(true);
- cmd->setStart(pageFrom);
- cmd->setLimit(pageSize+1); //read an extra WU to check hasMoreWU
- if (notEmpty(req.getCluster()))
- cmd->setCluster(req.getCluster());
- if (notEmpty(req.getOwner()))
- cmd->setOwner(req.getOwner());
- if (notEmpty(req.getJobname()))
- cmd->setJobName(req.getJobname());
- if (notEmpty(req.getState()))
- cmd->setState(req.getState());
- StringBuffer timeFrom, timeTo;
- readDateFilters(timeFrom, timeTo);
- if (timeFrom.length())
- cmd->setAfter(timeFrom.str());
- if (timeTo.length())
- cmd->setBefore(timeTo.str());
- return;
- }
- void addArchivedWU(IArrayOf<IEspECLWorkunit>& archivedWUs, StringArray& wuDataArray, bool canAccess)
- {
- Owned<IEspECLWorkunit> info= createECLWorkunit("","");
- const char* wuid = wuDataArray.item(0);
- info->setWuid(wuid);
- if (!canAccess)
- info->setState("<Hidden>");
- else
- {
- if (notEmpty(wuDataArray.item(1)))
- info->setOwner(wuDataArray.item(1));
- if (notEmpty(wuDataArray.item(2)))
- info->setJobname(wuDataArray.item(2));
- if (notEmpty(wuDataArray.item(3)))
- info->setCluster(wuDataArray.item(3));
- if (notEmpty(wuDataArray.item(4)))
- info->setState(wuDataArray.item(4));
- }
- //Sort WUs by WUID
- ForEachItemIn(i, archivedWUs)
- {
- IEspECLWorkunit& w = archivedWUs.item(i);
- if (!isEmpty(w.getWuid()) && strcmp(wuid, w.getWuid())>0)
- {
- archivedWUs.add(*info.getClear(), (aindex_t) i);
- return;
- }
- }
- archivedWUs.append(*info.getClear());
- return;
- }
- public:
- IMPLEMENT_IINTERFACE_USING(CInterface);
- CArchivedWUsReader(IEspContext& _context, const char* _sashaServerIP, unsigned _sashaServerPort, ArchivedWuCache& _archivedWuCache,
- unsigned _cacheMinutes, unsigned _pageFrom, unsigned _pageSize, IEspWUQueryRequest& _req)
- : context(_context), sashaServerIP(_sashaServerIP), sashaServerPort(_sashaServerPort),
- archivedWuCache(_archivedWuCache), cacheMinutes(_cacheMinutes), pageFrom(_pageFrom), pageSize(_pageSize), req(_req)
- {
- hasMoreWU = false;
- numberOfWUsReturned = 0;
- };
- void getArchivedWUs(IArrayOf<IEspECLWorkunit>& archivedWUs)
- {
- setFilterString();
- Owned<ArchivedWuCacheElement> cachedResults = archivedWuCache.lookup(context, filterStr, "AddWhenAvailable", cacheMinutes);
- if (cachedResults)
- {
- hasMoreWU = cachedResults->m_hasNextPage;
- numberOfWUsReturned = cachedResults->numWUsReturned;
- if (cachedResults->m_results.length())
- {
- ForEachItemIn(ai, cachedResults->m_results)
- archivedWUs.append(*LINK(&cachedResults->m_results.item(ai)));
- }
- }
- else
- {
- SocketEndpoint ep;
- if (sashaServerIP && *sashaServerIP)
- ep.set(sashaServerIP, sashaServerPort);
- else
- getSashaNode(ep);
- Owned<INode> sashaserver = createINode(ep);
- Owned<ISashaCommand> cmd = createSashaCommand();
- setSashaCommand(sashaserver, cmd);
- if (!cmd->send(sashaserver))
- {
- StringBuffer msg("Cannot connect to archive server at ");
- sashaserver->endpoint().getUrlStr(msg);
- throw MakeStringException(ECLWATCH_CANNOT_CONNECT_ARCHIVE_SERVER, "%s", msg.str());
- }
- numberOfWUsReturned = cmd->numIds();
- hasMoreWU = (numberOfWUsReturned > pageSize);
- if (hasMoreWU)
- numberOfWUsReturned--;
- if (numberOfWUsReturned > 0)
- {
- SecAccessFlags accessOwn, accessOthers;
- getUserWuAccessFlags(context, accessOwn, accessOthers, true);
- for (unsigned i=0; i<numberOfWUsReturned; i++)
- {
- const char *csline = cmd->queryId(i);
- if (!csline || !*csline)
- continue;
- StringArray wuDataArray;
- wuDataArray.appendList(csline, ",");
- const char* wuid = wuDataArray.item(0);
- if (isEmpty(wuid))
- {
- WARNLOG("Empty WUID in SCA_LIST response");
- continue;
- }
- addArchivedWU(archivedWUs, wuDataArray, chooseWuAccessFlagsByOwnership(context.queryUserId(), wuDataArray.item(1), accessOwn, accessOthers) >= SecAccess_Read);
- }
- archivedWuCache.add(filterStr, "AddWhenAvailable", hasMoreWU, numberOfWUsReturned, archivedWUs);
- }
- }
- return;
- };
- bool getHasMoreWU() { return hasMoreWU; };
- unsigned getNumberOfWUsReturned() { return numberOfWUsReturned; };
- };
- unsigned pageStart = (unsigned) req.getPageStartFrom();
- unsigned pageSize = (unsigned) req.getPageSize();
- if(pageSize < 1)
- pageSize=500;
- IArrayOf<IEspECLWorkunit> archivedWUs;
- Owned<IArchivedWUsReader> archiveWUsReader = new CArchivedWUsReader(context, sashaServerIP, sashaServerPort, archivedWuCache,
- cacheMinutes, pageStart, pageSize, req);
- archiveWUsReader->getArchivedWUs(archivedWUs);
- resp.setWorkunits(archivedWUs);
- resp.setNumWUs(archiveWUsReader->getNumberOfWUsReturned());
- resp.setType("archived only");
- resp.setPageSize(pageSize);
- resp.setPageStartFrom(pageStart+1);
- resp.setPageEndAt(pageStart + archiveWUsReader->getNumberOfWUsReturned());
- if(pageStart > 0)
- { //This is not the first page;
- resp.setFirst(false);
- resp.setPrevPage((pageStart > pageSize) ? pageStart - pageSize: 0);
- }
- if (archiveWUsReader->getHasMoreWU())
- resp.setNextPage(pageStart + pageSize);
- return;
- }
- bool CWsWorkunitsEx::onWUQuery(IEspContext &context, IEspWUQueryRequest & req, IEspWUQueryResponse & resp)
- {
- try
- {
- StringBuffer wuidStr = req.getWuid();
- const char* wuid = wuidStr.trim().str();
- if (req.getType() && strieq(req.getType(), "archived workunits"))
- doWUQueryFromArchive(context, sashaServerIp.get(), sashaServerPort, *archivedWuCache, awusCacheMinutes, req, resp);
- else if(notEmpty(wuid) && looksLikeAWuid(wuid))
- doWUQueryBySingleWuid(context, wuid, resp);
- else if (notEmpty(req.getLogicalFile()) && req.getLogicalFileSearchType() && strieq(req.getLogicalFileSearchType(), "Created"))
- doWUQueryByFile(context, req.getLogicalFile(), resp);
- else
- doWUQueryWithSort(context, req, resp);
- resp.setState(req.getState());
- resp.setCluster(req.getCluster());
- resp.setRoxieCluster(req.getRoxieCluster());
- resp.setOwner(req.getOwner());
- resp.setStartDate(req.getStartDate());
- resp.setEndDate(req.getEndDate());
- double version = context.getClientVersion();
- StringBuffer basicQuery;
- addToQueryString(basicQuery, "State", req.getState());
- addToQueryString(basicQuery, "Cluster", req.getCluster());
- if (version > 1.07)
- addToQueryString(basicQuery, "RoxieCluster", req.getRoxieCluster());
- addToQueryString(basicQuery, "Owner", req.getOwner());
- addToQueryString(basicQuery, "StartDate", req.getStartDate());
- addToQueryString(basicQuery, "EndDate", req.getEndDate());
- if (version > 1.25 && req.getLastNDays() > -1)
- addToQueryString(basicQuery, "LastNDays", StringBuffer().append(req.getLastNDays()).str());
- addToQueryString(basicQuery, "ECL", req.getECL());
- addToQueryString(basicQuery, "Jobname", req.getJobname());
- addToQueryString(basicQuery, "Type", req.getType());
- if (addToQueryString(basicQuery, "LogicalFile", req.getLogicalFile()))
- addToQueryString(basicQuery, "LogicalFileSearchType", req.getLogicalFileSearchType());
- resp.setFilters(basicQuery.str());
- if (notEmpty(req.getSortby()) && !strstr(basicQuery.str(), StringBuffer(req.getSortby()).append('=').str()))
- {
- resp.setSortby(req.getSortby());
- addToQueryString(basicQuery, "Sortby", req.getSortby());
- if (req.getDescending())
- {
- resp.setDescending(req.getDescending());
- addToQueryString(basicQuery, "Descending", "1");
- }
- }
- resp.setBasicQuery(basicQuery.str());
- StringBuffer s;
- if(notEmpty(req.getECL()))
- resp.setECL(Utils::url_encode(req.getECL(), s).str());
- if(notEmpty(req.getJobname()))
- resp.setJobname(Utils::url_encode(req.getJobname(), s.clear()).str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- void appendResultSet(MemoryBuffer& mb, INewResultSet* result, const char *name, __int64 start, unsigned& count, __int64& total, bool bin, bool xsd, ESPSerializationFormat fmt, const IProperties *xmlns)
- {
- if (!result)
- return;
- Owned<IResultSetCursor> cursor(result->createCursor());
- total=result->getNumRows();
- if(bin)
- count = getResultBin(mb, result, (unsigned)start, count);
- else
- {
- struct MemoryBuffer2IStringVal : public CInterface, implements IStringVal
- {
- MemoryBuffer2IStringVal(MemoryBuffer & _buffer) : buffer(_buffer) {}
- IMPLEMENT_IINTERFACE;
- virtual const char * str() const { UNIMPLEMENTED; }
- virtual void set(const char *val) { buffer.append(strlen(val),val); }
- virtual void clear() { } // support appending only
- virtual void setLen(const char *val, unsigned length) { buffer.append(length, val); }
- virtual unsigned length() const { return buffer.length(); };
- MemoryBuffer & buffer;
- } adaptor(mb);
- if (fmt==ESPSerializationJSON)
- count = getResultJSON(adaptor, result, name, (unsigned) start, count, (xsd) ? "myschema" : NULL);
- else
- count = getResultXml(adaptor, result, name, (unsigned) start, count, (xsd) ? "myschema" : NULL, xmlns);
- }
- }
- INewResultSet* createFilteredResultSet(INewResultSet* result, IArrayOf<IConstNamedValue>* filterBy)
- {
- if (!result || !filterBy || !filterBy->length())
- return NULL;
- Owned<IFilteredResultSet> filter = result->createFiltered();
- const IResultSetMetaData &meta = result->getMetaData();
- unsigned columnCount = meta.getColumnCount();
- ForEachItemIn(i, *filterBy)
- {
- IConstNamedValue &item = filterBy->item(i);
- const char *name = item.getName();
- const char *value = item.getValue();
- if (!name || !*name || !value || !*value)
- continue;
- for(unsigned col = 0; col < columnCount; col++)
- {
- SCMStringBuffer scmbuf;
- meta.getColumnLabel(scmbuf, col);
- if (strieq(scmbuf.str(), name))
- {
- filter->addFilter(col, value);
- break;
- }
- }
- }
- return filter->create();
- }
- void getWsWuResult(IEspContext &context, const char* wuid, const char *name, const char *logical, unsigned index, __int64 start,
- unsigned& count, __int64& total, IStringVal& resname, bool bin, IArrayOf<IConstNamedValue>* filterBy, MemoryBuffer& mb, bool xsd=true)
- {
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid, false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_OPEN_WORKUNIT,"Cannot open workunit %s.",wuid);
- Owned<IConstWUResult> result;
- if (notEmpty(name))
- result.setown(cw->getResultByName(name));
- else if (notEmpty(logical))
- {
- Owned<IConstWUResultIterator> it = &cw->getResults();
- ForEach(*it)
- {
- IConstWUResult &r = it->query();
- SCMStringBuffer filename;
- if(strieq(r.getResultLogicalName(filename).str(), logical))
- {
- result.setown(LINK(&r));
- break;
- }
- }
- }
- else
- result.setown(cw->getResultBySequence(index));
- if (!result)
- throw MakeStringException(ECLWATCH_CANNOT_GET_WU_RESULT,"Cannot open the workunit result.");
- if (!resname.length())
- result->getResultName(resname);
- Owned<IResultSetFactory> resultSetFactory = getSecResultSetFactory(context.querySecManager(), context.queryUser(), context.queryUserId(), context.queryPassword());
- SCMStringBuffer logicalName;
- result->getResultLogicalName(logicalName);
- Owned<INewResultSet> rs;
- if (logicalName.length())
- {
- SCMStringBuffer cluster; //MORE is this wrong cluster?
- rs.setown(resultSetFactory->createNewFileResultSet(logicalName.str(), cw->getClusterName(cluster).str()));
- }
- else
- rs.setown(resultSetFactory->createNewResultSet(result, wuid));
- if (!filterBy || !filterBy->length())
- appendResultSet(mb, rs, name, start, count, total, bin, xsd, context.getResponseFormat(), result->queryXmlns());
- else
- {
- Owned<INewResultSet> filteredResult = createFilteredResultSet(rs, filterBy);
- appendResultSet(mb, filteredResult, name, start, count, total, bin, xsd, context.getResponseFormat(), result->queryXmlns());
- }
- }
- void openSaveFile(IEspContext &context, int opt, const char* filename, const char* origMimeType, MemoryBuffer& buf, IEspWULogFileResponse &resp)
- {
- if (opt < 1)
- {
- resp.setThefile(buf);
- resp.setThefile_mimetype(origMimeType);
- }
- else if (opt < 2)
- {
- StringBuffer headerStr("attachment;");
- if (filename && *filename)
- {
- const char* pFileName = strrchr(filename, PATHSEPCHAR);
- if (pFileName)
- headerStr.appendf("filename=%s", pFileName+1);
- else
- headerStr.appendf("filename=%s", filename);
- }
- MemoryBuffer buf0;
- unsigned i = 0;
- char* p = (char*) buf.toByteArray();
- while (i < buf.length())
- {
- if (p[0] != 10)
- buf0.append(p[0]);
- else
- buf0.append(0x0d);
- p++;
- i++;
- }
- resp.setThefile(buf);
- resp.setThefile_mimetype(origMimeType);
- context.addCustomerHeader("Content-disposition", headerStr.str());
- }
- else
- {
- #ifndef _USE_ZLIB
- throw MakeStringException(ECLWATCH_CANNOT_COMPRESS_DATA,"The data cannot be compressed.");
- #else
- StringBuffer fileNameStr, headerStr("attachment;");
- if (notEmpty(filename))
- {
- fileNameStr.append(filename);
- headerStr.append("filename=").append(filename).append((opt>2) ? ".gz" : ".zip");
- }
- else
- fileNameStr.append("file");
- StringBuffer ifname;
- ifname.appendf("%s%sT%xAT%x", TEMPZIPDIR, PATHSEPSTR, (unsigned)(memsize_t)GetCurrentThreadId(), msTick()).append((opt>2)? "" : ".zip");
- IZZIPor* Zipor = createZZIPor();
- int ret = 0;
- if (opt > 2)
- ret = Zipor->gzipToFile(buf.length(), (void*)buf.toByteArray(), ifname.str());
- else
- ret = Zipor->zipToFile(buf.length(), (void*)buf.toByteArray(), fileNameStr.str(), ifname.str());
- releaseIZ(Zipor);
- if (ret < 0)
- {
- Owned<IFile> rFile = createIFile(ifname.str());
- if (rFile->exists())
- rFile->remove();
- throw MakeStringException(ECLWATCH_CANNOT_COMPRESS_DATA,"The data cannot be compressed.");
- }
- Owned <IFile> rf = createIFile(ifname.str());
- if (!rf->exists())
- throw MakeStringException(ECLWATCH_CANNOT_COMPRESS_DATA,"The data cannot be compressed.");
- MemoryBuffer out;
- Owned <IFileIO> fio = rf->open(IFOread);
- read(fio, 0, (size32_t) rf->size(), out);
- resp.setThefile(out);
- fio.clear();
- rf->remove();
- resp.setThefile_mimetype((opt > 2) ? "application/x-gzip" : "application/zip");
- context.addCustomerHeader("Content-disposition", headerStr.str());
- #endif
- }
- }
- bool CWsWorkunitsEx::onWUFile(IEspContext &context,IEspWULogFileRequest &req, IEspWULogFileResponse &resp)
- {
- try
- {
- StringBuffer wuidStr = req.getWuid();
- const char* wuidIn = wuidStr.trim().str();
- if (wuidIn && *wuidIn)
- {
- if (!looksLikeAWuid(wuidIn))
- throw MakeStringException(ECLWATCH_INVALID_INPUT, "Invalid Workunit ID");
- ensureWsWorkunitAccess(context, wuidIn, SecAccess_Read);
- }
- StringAttr wuid(wuidIn);
- if (wuid.isEmpty() && notEmpty(req.getQuerySet()) && notEmpty(req.getQuery()))
- {
- Owned<IPropertyTree> registry = getQueryRegistry(req.getQuerySet(), false);
- if (!registry)
- throw MakeStringException(ECLWATCH_QUERYSET_NOT_FOUND, "Queryset %s not found", req.getQuerySet());
- Owned<IPropertyTree> query = resolveQueryAlias(registry, req.getQuery());
- if (!query)
- throw MakeStringException(ECLWATCH_QUERYID_NOT_FOUND, "Query %s not found", req.getQuery());
- resp.setQuerySet(req.getQuerySet());
- resp.setQueryName(query->queryProp("@name"));
- resp.setQueryId(query->queryProp("@id"));
- wuid.set(query->queryProp("@wuid"));
- }
- int opt = req.getOption();
- if (!wuid.isEmpty())
- {
- resp.setWuid(wuid.get());
- MemoryBuffer mb;
- WsWuInfo winfo(context, wuid);
- if (strieq(File_ArchiveQuery, req.getType()))
- {
- winfo.getWorkunitArchiveQuery(mb);
- openSaveFile(context, opt, "ArchiveQuery.xml", HTTP_TYPE_APPLICATION_XML, mb, resp);
- }
- else if (strieq(File_Cpp,req.getType()) && notEmpty(req.getName()))
- {
- winfo.getWorkunitCpp(req.getName(), req.getDescription(), req.getIPAddress(),mb, opt > 0);
- openSaveFile(context, opt, req.getName(), HTTP_TYPE_TEXT_PLAIN, mb, resp);
- }
- else if (strieq(File_DLL,req.getType()))
- {
- StringBuffer name;
- winfo.getWorkunitDll(name, mb);
- resp.setFileName(name.str());
- resp.setDaliServer(daliServers.get());
- openSaveFile(context, opt, req.getName(), HTTP_TYPE_OCTET_STREAM, mb, resp);
- }
- else if (strieq(File_Res,req.getType()))
- {
- winfo.getWorkunitResTxt(mb);
- openSaveFile(context, opt, "res.txt", HTTP_TYPE_TEXT_PLAIN, mb, resp);
- }
- else if (strncmp(req.getType(), File_ThorLog, 7) == 0)
- {
- winfo.getWorkunitThorLog(req.getName(), mb);
- openSaveFile(context, opt, "thormaster.log", HTTP_TYPE_TEXT_PLAIN, mb, resp);
- }
- else if (strieq(File_ThorSlaveLog,req.getType()))
- {
- StringBuffer logDir;
- getConfigurationDirectory(directories, "log", "thor", req.getProcess(), logDir);
- winfo.getWorkunitThorSlaveLog(req.getClusterGroup(), req.getIPAddress(), req.getLogDate(), logDir.str(), req.getSlaveNumber(), mb, false);
- openSaveFile(context, opt, "ThorSlave.log", HTTP_TYPE_TEXT_PLAIN, mb, resp);
- }
- else if (strieq(File_EclAgentLog,req.getType()))
- {
- winfo.getWorkunitEclAgentLog(req.getName(), req.getProcess(), mb);
- openSaveFile(context, opt, "eclagent.log", HTTP_TYPE_TEXT_PLAIN, mb, resp);
- }
- else if (strieq(File_XML,req.getType()) && notEmpty(req.getName()))
- {
- const char* name = req.getName();
- const char* ptr = strrchr(name, '/');
- if (ptr)
- ptr++;
- else
- ptr = name;
- winfo.getWorkunitAssociatedXml(name, req.getIPAddress(), req.getPlainText(), req.getDescription(), opt > 0, mb);
- openSaveFile(context, opt, ptr, HTTP_TYPE_APPLICATION_XML, mb, resp);
- }
- else if (strieq(File_XML,req.getType()))
- {
- winfo.getWorkunitXml(req.getPlainText(), mb);
- if (opt < 2)
- {
- resp.setThefile(mb);
- const char* plainText = req.getPlainText();
- if (plainText && (!stricmp(plainText, "yes")))
- resp.setThefile_mimetype(HTTP_TYPE_TEXT_PLAIN);
- else
- resp.setThefile_mimetype(HTTP_TYPE_APPLICATION_XML);
- }
- else
- {
- VStringBuffer xmlName("%s.xml", wuid.get());
- openSaveFile(context, 2, xmlName.str(), HTTP_TYPE_APPLICATION_XML, mb, resp);
- }
- }
- }
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUResultBin(IEspContext &context,IEspWUResultBinRequest &req, IEspWUResultBinResponse &resp)
- {
- try
- {
- StringBuffer wuidStr = req.getWuid();
- const char* wuidIn = wuidStr.trim().str();
- if (wuidIn && *wuidIn)
- {
- if (!looksLikeAWuid(wuidIn))
- throw MakeStringException(ECLWATCH_INVALID_INPUT, "Invalid Workunit ID: %s", wuidIn);
- ensureWsWorkunitAccess(context, wuidIn, SecAccess_Read);
- }
- MemoryBuffer mb;
- __int64 total=0;
- __int64 start = req.getStart() > 0 ? req.getStart() : 0;
- unsigned count = req.getCount(), requested=count;
- IArrayOf<IConstNamedValue>* filterBy = &req.getFilterBy();
- SCMStringBuffer name;
- bool bin = (req.getFormat() && strieq(req.getFormat(),"raw"));
- if (notEmpty(wuidIn) && notEmpty(req.getResultName()))
- getWsWuResult(context, wuidIn, req.getResultName(), NULL, 0, start, count, total, name, bin, filterBy, mb);
- else if (notEmpty(wuidIn) && (req.getSequence() >= 0))
- getWsWuResult(context, wuidIn, NULL, NULL, req.getSequence(), start, count, total, name, bin,filterBy, mb);
- else if (notEmpty(req.getLogicalName()))
- {
- const char* logicalName = req.getLogicalName();
- StringBuffer wuid;
- getWuidFromLogicalFileName(context, logicalName, wuid);
- if (!wuid.length())
- throw MakeStringException(ECLWATCH_CANNOT_GET_WORKUNIT,"Cannot find the workunit for file %s.",logicalName);
- getWsWuResult(context, wuid.str(), NULL, logicalName, 0, start, count, total, name, bin, filterBy, mb);
- }
- else
- throw MakeStringException(ECLWATCH_CANNOT_GET_WU_RESULT,"Cannot open the workunit result.");
- if(stricmp(req.getFormat(),"xls")==0)
- {
- Owned<IProperties> params(createProperties());
- params->setProp("showCount",0);
- StringBuffer xml;
- xml.append("<WUResultExcel><Result>").append(mb.length(), mb.toByteArray()).append("</Result></WUResultExcel>");
- if (xml.length() > MAXXLSTRANSFER)
- throw MakeStringException(ECLWATCH_TOO_BIG_DATA_SET, "The data set is too big to be converted to an Excel file. Please use the gzip link to download a compressed XML data file.");
- StringBuffer xls;
- xsltTransform(xml.str(), StringBuffer(getCFD()).append("./smc_xslt/result.xslt").str(), params, xls);
- MemoryBuffer out;
- out.setBuffer(xls.length(), (void*)xls.str());
- resp.setResult(out);
- resp.setResult_mimetype("application/vnd.ms-excel");
- }
- #ifdef _USE_ZLIB
- else if(strieq(req.getFormat(),"zip") || strieq(req.getFormat(),"gzip"))
- {
- bool gzip = strieq(req.getFormat(),"gzip");
- StringBuffer xml("<?xml version=\"1.0\" encoding=\"UTF-8\"?>");
- xml.append("<Result>").append(mb.length(),mb.toByteArray()).append("</Result>");
- VStringBuffer ifname("%s%sT%xAT%x%s", TEMPZIPDIR, PATHSEPSTR, (unsigned)(memsize_t)GetCurrentThreadId(), msTick(), gzip ? "" : ".zip");
- IZZIPor* Zipor = createZZIPor();
- int ret = 0;
- if (gzip)
- ret = Zipor->gzipToFile(xml.length(), (void*)xml.str(), ifname.str());
- else
- ret = Zipor->zipToFile(xml.length(), (void*)xml.str(), "WUResult.xml", ifname.str());
- releaseIZ(Zipor);
- if (ret < 0)
- {
- Owned<IFile> rFile = createIFile(ifname.str());
- if (rFile->exists())
- rFile->remove();
- throw MakeStringException(ECLWATCH_CANNOT_COMPRESS_DATA, "The data cannot be compressed.");
- }
- MemoryBuffer out;
- Owned <IFile> rf = createIFile(ifname.str());
- if (rf->exists())
- {
- Owned <IFileIO> fio = rf->open(IFOread);
- read(fio, 0, (size32_t) rf->size(), out);
- resp.setResult(out);
- }
- if (gzip)
- {
- resp.setResult_mimetype("application/x-gzip");
- context.addCustomerHeader("Content-disposition", "attachment;filename=WUResult.xml.gz");
- }
- else
- {
- resp.setResult_mimetype("application/zip");
- context.addCustomerHeader("Content-disposition", "attachment;filename=WUResult.xml.zip");
- }
- Owned<IFile> rFile = createIFile(ifname.str());
- if (rFile->exists())
- rFile->remove();
- }
- #endif
- else
- {
- resp.setResult(mb);
- }
- resp.setName(name.str());
- resp.setWuid(wuidIn);
- resp.setSequence(req.getSequence());
- resp.setStart(start);
- if (requested > total)
- requested = (unsigned)total;
- resp.setRequested(requested);
- resp.setCount(count);
- resp.setTotal(total);
- resp.setFormat(req.getFormat());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUResultSummary(IEspContext &context, IEspWUResultSummaryRequest &req, IEspWUResultSummaryResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUResultSummary", wuid);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_OPEN_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- ensureWsWorkunitAccess(context, *cw, SecAccess_Read);
- resp.setWuid(wuid.str());
- resp.setSequence(req.getSequence());
- IArrayOf<IEspECLResult> results;
- Owned<IConstWUResult> r = cw->getResultBySequence(req.getSequence());
- if (r)
- {
- WsWuInfo winfo(context, cw);
- winfo.getResult(*r, results, 0);
- resp.setFormat(r->getResultFormat());
- resp.setResult(results.item(0));
- }
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- void getFileResults(IEspContext &context, const char* logicalName, const char* cluster,__int64 start, unsigned& count,__int64& total,
- IStringVal& resname,bool bin, IArrayOf<IConstNamedValue>* filterBy, MemoryBuffer& buf, bool xsd)
- {
- Owned<IResultSetFactory> resultSetFactory = getSecResultSetFactory(context.querySecManager(), context.queryUser(), context.queryUserId(), context.queryPassword());
- Owned<INewResultSet> result(resultSetFactory->createNewFileResultSet(logicalName, cluster));
- if (!filterBy || !filterBy->length())
- appendResultSet(buf, result, resname.str(), start, count, total, bin, xsd, context.getResponseFormat(), NULL);
- else
- {
- Owned<INewResultSet> filteredResult = createFilteredResultSet(result, filterBy);
- appendResultSet(buf, filteredResult, resname.str(), start, count, total, bin, xsd, context.getResponseFormat(), NULL);
- }
- }
- void getWorkunitCluster(IEspContext &context, const char* wuid, SCMStringBuffer& cluster, bool checkArchiveWUs)
- {
- if (isEmpty(wuid))
- return;
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid, false);
- if (cw)
- cw->getClusterName(cluster);
- else if (checkArchiveWUs)
- {
- Owned<IPropertyTree> wuProps;// = getArchivedWorkUnitProperties(wuid);
- if (wuProps)
- cluster.set(wuProps->queryProp("@clusterName"));
- }
- }
- bool CWsWorkunitsEx::onWUResult(IEspContext &context, IEspWUResultRequest &req, IEspWUResultResponse &resp)
- {
- try
- {
- StringBuffer wuidStr = req.getWuid();
- const char* wuid = wuidStr.trim().str();
- if (wuid && *wuid)
- {
- if (!looksLikeAWuid(wuid))
- throw MakeStringException(ECLWATCH_INVALID_INPUT, "Invalid Workunit ID: %s", wuid);
- ensureWsWorkunitAccess(context, wuid, SecAccess_Read);
- }
- MemoryBuffer mb;
- SCMStringBuffer name;
- __int64 total=0;
- __int64 start = req.getStart() > 0 ? req.getStart() : 0;
- unsigned count=req.getCount() ? req.getCount() : 100, requested=count;
- unsigned seq = req.getSequence();
- bool inclXsd = !req.getSuppressXmlSchema();
- VStringBuffer filter("start=%"I64F"d;count=%d", start, count);
- addToQueryString(filter, "clusterName", req.getCluster(), ';');
- addToQueryString(filter, "logicalName", req.getLogicalName(), ';');
- if (wuid && *wuid)
- addToQueryString(filter, "wuid", wuid, ';');
- addToQueryString(filter, "resultName", req.getResultName(), ';');
- filter.appendf(";seq=%d;", seq);
- if (inclXsd)
- filter.append("xsd;");
- if (context.getResponseFormat()==ESPSerializationJSON)
- filter.append("json;");
- IArrayOf<IConstNamedValue>* filterBy = &req.getFilterBy();
- ForEachItemIn(i, *filterBy)
- {
- IConstNamedValue &item = filterBy->item(i);
- const char *name = item.getName();
- const char *value = item.getValue();
- if (name && *name && value && *value)
- addToQueryString(filter, name, value, ';');
- }
- const char* logicalName = req.getLogicalName();
- const char* clusterName = req.getCluster();
- const char* resultName = req.getResultName();
- Owned<DataCacheElement> data = dataCache->lookup(context, filter, awusCacheMinutes);
- if (data)
- {
- mb.append(data->m_data.c_str());
- name.set(data->m_name.c_str());
- logicalName = data->m_logicalName.c_str();
- wuid = data->m_wuid.c_str();
- resultName = data->m_resultName.c_str();
- seq = data->m_seq;
- start = data->m_start;
- count = data->m_rowcount;
- requested = (unsigned)data->m_requested;
- total = data->m_total;
- if (notEmpty(logicalName))
- resp.setLogicalName(logicalName);
- else
- {
- if (notEmpty(wuid))
- resp.setWuid(wuid);
- resp.setSequence(seq);
- }
- }
- else
- {
- if(logicalName && *logicalName)
- {
- StringBuffer lwuid;
- getWuidFromLogicalFileName(context, logicalName, lwuid);
- SCMStringBuffer cluster;
- if (lwuid.length())
- getWorkunitCluster(context, lwuid.str(), cluster, true);
- if (cluster.length())
- {
- getFileResults(context, logicalName, cluster.str(), start, count, total, name, false, filterBy, mb, inclXsd);
- resp.setLogicalName(logicalName);
- }
- else if (notEmpty(clusterName))
- {
- getFileResults(context, logicalName, clusterName, start, count, total, name, false, filterBy, mb, inclXsd);
- resp.setLogicalName(logicalName);
- }
- else
- throw MakeStringException(ECLWATCH_INVALID_INPUT,"Need valid target cluster to browse file %s.",logicalName);
- }
- else if (notEmpty(wuid) && notEmpty(resultName))
- {
- name.set(resultName);
- getWsWuResult(context, wuid, resultName, NULL, 0, start, count, total, name, false, filterBy, mb, inclXsd);
- resp.setWuid(wuid);
- resp.setSequence(seq);
- }
- else
- {
- getWsWuResult(context, wuid, NULL, NULL, seq, start, count, total, name, false, filterBy, mb, inclXsd);
- resp.setWuid(wuid);
- resp.setSequence(seq);
- }
- mb.append(0);
- if (requested > total)
- requested = (unsigned)total;
- dataCache->add(filter, mb.toByteArray(), name.str(), logicalName, wuid, resultName, seq, start, count, requested, total);
- }
- resp.setName(name.str());
- resp.setStart(start);
- if (clusterName && *clusterName)
- resp.setCluster(clusterName);
- resp.setRequested(requested);
- resp.setCount(count);
- resp.setTotal(total);
- resp.setResult(mb.toByteArray());
- context.queryXslParameters()->setProp("escapeResults","1");
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- void getScheduledWUs(IEspContext &context, const char *serverName, const char *eventName, IArrayOf<IEspScheduledWU> & results)
- {
- if (notEmpty(serverName))
- {
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IScheduleReader> reader = getScheduleReader(serverName, eventName);
- Owned<IScheduleReaderIterator> it(reader->getIterator());
- while(it->isValidEventName())
- {
- StringBuffer ieventName;
- it->getEventName(ieventName);
- while(it->isValidEventText())
- {
- StringBuffer ieventText;
- it->getEventText(ieventText);
- while(it->isValidWuid())
- {
- StringBuffer wuid;
- it->getWuid(wuid);
- if (wuid.length())
- {
- Owned<IEspScheduledWU> scheduledWU = createScheduledWU("");
- scheduledWU->setWuid(wuid.str());
- scheduledWU->setCluster(serverName);
- if (ieventName.length())
- scheduledWU->setEventName(ieventName.str());
- if (ieventText.str())
- scheduledWU->setEventText(ieventText.str());
- try
- {
- SCMStringBuffer s;
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if (cw)
- scheduledWU->setJobName(cw->getJobName(s).str());
- }
- catch (IException *e)
- {
- e->Release();
- }
- results.append(*scheduledWU.getLink());
- }
- it->nextWuid();
- }
- it->nextEventText();
- }
- it->nextEventName();
- }
- }
- return;
- }
- bool CWsWorkunitsEx::onWUShowScheduled(IEspContext &context, IEspWUShowScheduledRequest & req, IEspWUShowScheduledResponse & resp)
- {
- try
- {
- DBGLOG("WUShowScheduled");
- const char *clusterName = req.getCluster();
- const char *eventName = req.getEventName();
- IArrayOf<IEspScheduledWU> results;
- if(notEmpty(req.getPushEventName()))
- resp.setPushEventName(req.getPushEventName());
- if(notEmpty(req.getPushEventText()))
- resp.setPushEventText(req.getPushEventText());
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory();
- Owned<IConstEnvironment> environment = factory->openEnvironmentByFile();
- Owned<IPropertyTree> root = &environment->getPTree();
- if (!root)
- throw MakeStringException(ECLWATCH_CANNOT_GET_ENV_INFO, "Failed to get environment information.");
- unsigned i = 0;
- Owned<IPropertyTreeIterator> ic = root->getElements("Software/Topology/Cluster");
- IArrayOf<IEspServerInfo> servers;
- ForEach(*ic)
- {
- IPropertyTree &cluster = ic->query();
- const char *iclusterName = cluster.queryProp("@name");
- if (isEmpty(iclusterName))
- continue;
- if(isEmpty(clusterName))
- getScheduledWUs(context, iclusterName, eventName, results);
- else if (strieq(clusterName, iclusterName))
- {
- getScheduledWUs(context, clusterName, eventName, results);
- resp.setClusterSelected(i+1);
- }
- Owned<IEspServerInfo> server = createServerInfo("");
- server->setName(iclusterName);
- servers.append(*server.getLink());
- i++;
- }
- if (servers.length())
- resp.setClusters(servers);
- if (results.length())
- resp.setWorkunits(results);
- bool first=false;
- StringBuffer Query("PageFrom=Scheduler");
- appendUrlParameter(Query, "EventName", eventName, first);
- appendUrlParameter(Query, "ECluster", clusterName, first);
- resp.setQuery(Query.str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUExport(IEspContext &context, IEspWUExportRequest &req, IEspWUExportResponse &resp)
- {
- try
- {
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- WsWuSearch ws(context, req.getOwner(), req.getState(), req.getCluster(), req.getStartDate(), req.getEndDate(), req.getECL(), req.getJobname());
- StringBuffer xml("<?xml version=\"1.0\" encoding=\"UTF-8\"?><Workunits>");
- for(WsWuSearch::iterator it=ws.begin(); it!=ws.end(); it++)
- {
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(it->c_str(), false);
- if (cw)
- exportWorkUnitToXML(cw, xml, true, false);
- }
- xml.append("</Workunits>");
- MemoryBuffer mb;
- mb.setBuffer(xml.length(),(void*)xml.str());
- resp.setExportData(mb);
- resp.setExportData_mimetype(HTTP_TYPE_APPLICATION_XML);
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUListLocalFileRequired(IEspContext& context, IEspWUListLocalFileRequiredRequest& req, IEspWUListLocalFileRequiredResponse& resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUListLocalFileRequired", wuid);
- ensureWsWorkunitAccess(context, wuid.str(), SecAccess_Read);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if (!cw)
- throw MakeStringException(ECLWATCH_CANNOT_UPDATE_WORKUNIT, "Workunit %s not found.", wuid.str());
- IArrayOf<IEspLogicalFileUpload> localFiles;
- Owned<IConstLocalFileUploadIterator> it = cw->getLocalFileUploads();
- ForEach(*it)
- {
- Owned<IConstLocalFileUpload> file = it->get();
- if(!file)
- continue;
- Owned<IEspLogicalFileUpload> up = createLogicalFileUpload();
- SCMStringBuffer s;
- up->setType(file->queryType());
- up->setSource(file->getSource(s).str());
- up->setDestination(file->getDestination(s).str());
- up->setEventTag(file->getEventTag(s).str());
- localFiles.append(*up.getLink());
- }
- resp.setLocalFileUploads(localFiles);
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- typedef enum wsEclTypes_
- {
- wsEclTypeUnknown,
- xsdString,
- xsdBoolean,
- xsdDecimal,
- xsdFloat,
- xsdDouble,
- xsdDuration,
- xsdDateTime,
- xsdTime,
- xsdDate,
- xsdYearMonth,
- xsdYear,
- xsdMonthDay,
- xsdDay,
- xsdMonth,
- xsdHexBinary,
- xsdBase64Binary,
- xsdAnyURI,
- xsdQName,
- xsdNOTATION,
- xsdNormalizedString,
- xsdToken,
- xsdLanguage,
- xsdNMTOKEN,
- xsdNMTOKENS,
- xsdName,
- xsdNCName,
- xsdID,
- xsdIDREF,
- xsdIDREFS,
- xsdENTITY,
- xsdENTITIES,
- xsdInteger,
- xsdNonPositiveInteger,
- xsdNegativeInteger,
- xsdLong,
- xsdInt,
- xsdShort,
- xsdByte,
- xsdNonNegativeInteger,
- xsdUnsignedLong,
- xsdUnsignedInt,
- xsdUnsignedShort,
- xsdUnsignedByte,
- xsdPositiveInteger,
- tnsRawDataFile,
- tnsCsvDataFile,
- tnsEspStringArray,
- tnsEspIntArray,
- tnsXmlDataSet,
- maxWsEclType
- } wsEclType;
- bool CWsWorkunitsEx::onWUAddLocalFileToWorkunit(IEspContext& context, IEspWUAddLocalFileToWorkunitRequest& req, IEspWUAddLocalFileToWorkunitResponse& resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUAddLocalFileToWorkunit", wuid);
- ensureWsWorkunitAccess(context, wuid.str(), SecAccess_Write);
- resp.setWuid(wuid.str());
- const char* varname = req.getName();
- if (isEmpty(varname))
- {
- resp.setResult("Name is not defined!");
- return true;
- }
- resp.setName(varname);
- wsEclType type = (wsEclType) req.getType();
- const char *val = req.getVal();
- unsigned len = req.getLength();
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- WorkunitUpdate wu(factory->updateWorkUnit(wuid.str()));
- if (!wu)
- {
- resp.setResult("Workunit not found!");
- return true;
- }
- Owned<IWUResult> wuRslt = wu->updateResultByName(varname);
- if (isEmpty(val))
- val=req.getDefVal();
- if (notEmpty(val))
- {
- switch (type)
- {
- case xsdBoolean:
- wuRslt->setResultBool((strieq(val, "1") || strieq(val, "true") || strieq(val, "on")));
- wuRslt->setResultStatus(ResultStatusSupplied);
- break;
- case xsdDecimal:
- case xsdFloat:
- case xsdDouble:
- wuRslt->setResultReal(atof(val));
- wuRslt->setResultStatus(ResultStatusSupplied);
- break;
- case xsdInteger:
- case xsdNonPositiveInteger:
- case xsdNegativeInteger:
- case xsdLong:
- case xsdInt:
- case xsdShort:
- case xsdByte:
- case xsdNonNegativeInteger:
- case xsdUnsignedLong:
- case xsdUnsignedInt:
- case xsdUnsignedShort:
- case xsdUnsignedByte:
- case xsdPositiveInteger:
- wuRslt->setResultInt(_atoi64(val));
- wuRslt->setResultStatus(ResultStatusSupplied);
- break;
- case tnsEspIntArray:
- case tnsEspStringArray:
- wuRslt->setResultRaw(len, val, ResultFormatXmlSet);
- break;
- case tnsRawDataFile:
- wuRslt->setResultRaw(len, val, ResultFormatRaw);
- break;
- case tnsXmlDataSet:
- wuRslt->setResultRaw(len, val, ResultFormatXml);
- break;
- case tnsCsvDataFile:
- case xsdBase64Binary: //tbd
- case xsdHexBinary:
- break;
- default:
- wuRslt->setResultString(val, len);
- wuRslt->setResultStatus(ResultStatusSupplied);
- break;
- }
- }
- resp.setResult("Result has been set as required!");
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- void getClusterConfig(char const * clusterType, char const * clusterName, char const * processName, StringBuffer& netAddress)
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory();
- Owned<IConstEnvironment> environment = factory->openEnvironmentByFile();
- Owned<IPropertyTree> pRoot = &environment->getPTree();
- VStringBuffer xpath("Software/%s[@name='%s']", clusterType, clusterName);
- IPropertyTree* pCluster = pRoot->queryPropTree(xpath.str());
- if (!pCluster)
- throw MakeStringException(ECLWATCH_CLUSTER_NOT_IN_ENV_INFO, "'%s %s' is not defined.", clusterType, clusterName);
- const char* port = pCluster->queryProp(xpath.set(processName).append("@port").str());
- const char* computer = pCluster->queryProp(xpath.set(processName).append("@computer").str());
- if (isEmpty(computer))
- throw MakeStringException(ECLWATCH_INVALID_CLUSTER_INFO, "'%s %s: %s' is not defined.", clusterType, clusterName, processName);
- Owned<IConstMachineInfo> pMachine = environment->getMachine(computer);
- if (pMachine)
- {
- StringBufferAdaptor s(netAddress);
- pMachine->getNetAddress(s);
- #ifdef MACHINE_IP
- if (streq(netAddress.str(), "."))
- netAddress = MACHINE_IP;
- #endif
- netAddress.append(':').append(port);
- }
- return;
- }
- bool CWsWorkunitsEx::onWUProcessGraph(IEspContext &context,IEspWUProcessGraphRequest &req, IEspWUProcessGraphResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUProcessGraph", wuid);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_OPEN_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- ensureWsWorkunitAccess(context, *cw, SecAccess_Read);
- Owned <IConstWUGraph> graph = cw->getGraph(req.getName());
- Owned <IPropertyTree> xgmml = graph->getXGMMLTree(true); // merge in graph progress information
- StringBuffer xml;
- resp.setTheGraph(toXML(xgmml.get(), xml).str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool isRunning(IConstWorkUnit &cw)
- {
- // MORE - move into workunit interface
- switch (cw.getState())
- {
- case WUStateFailed:
- case WUStateAborted:
- case WUStateCompleted:
- return false;
- default:
- return true;
- }
- }
- bool CWsWorkunitsEx::onWUGetGraph(IEspContext& context, IEspWUGetGraphRequest& req, IEspWUGetGraphResponse& resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUGetGraph", wuid);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_OPEN_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- ensureWsWorkunitAccess(context, *cw, SecAccess_Read);
- WUGraphIDType id;
- SCMStringBuffer runningGraph;
- bool running= (isRunning(*cw) && cw->getRunningGraph(runningGraph,id));
- IArrayOf<IEspECLGraphEx> graphs;
- Owned<IConstWUGraphIterator> it = &cw->getGraphs(GraphTypeAny);
- ForEach(*it)
- {
- IConstWUGraph &graph = it->query();
- if(!graph.isValid())
- continue;
- SCMStringBuffer name, label, type;
- graph.getName(name);
- graph.getLabel(label);
- graph.getTypeName(type);
- if(isEmpty(req.getGraphName()) || strieq(name.str(), req.getGraphName()))
- {
- Owned<IEspECLGraphEx> g = createECLGraphEx("","");
- g->setName(name.str());
- g->setLabel(label.str());
- g->setType(type.str());
- if(running && streq(name.str(), runningGraph.str()))
- {
- g->setRunning(true);
- g->setRunningId(id);
- }
- Owned<IPropertyTree> xgmml = graph.getXGMMLTree(true);
- // New functionality, if a subgraph id is specified and we only want to load the xgmml for that subgraph
- // then we need to conditionally pull a propertytree from the xgmml graph one and use that for the xgmml.
- StringBuffer xml;
- if (notEmpty(req.getSubGraphId()))
- {
- VStringBuffer xpath("//node[@id='%s']", req.getSubGraphId());
- toXML(xgmml->queryPropTree(xpath.str()), xml);
- }
- else
- toXML(xgmml, xml);
- g->setGraph(xml.str());
- if (context.getClientVersion() > 1.20)
- {
- Owned<IConstWUGraphProgress> progress = cw->getGraphProgress(name.str());
- if (progress)
- {
- WUGraphState graphstate= progress->queryGraphState();
- if (graphstate == WUGraphComplete)
- g->setComplete(true);
- else if (graphstate == WUGraphFailed)
- g->setFailed(true);
- }
- }
- graphs.append(*g.getClear());
- }
- }
- resp.setGraphs(graphs);
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onGVCAjaxGraph(IEspContext &context, IEspGVCAjaxGraphRequest &req, IEspGVCAjaxGraphResponse &resp)
- {
- try
- {
- resp.setName(req.getName());
- resp.setGraphName(req.getGraphName());
- resp.setGraphType("eclwatch");
- double version = context.getClientVersion();
- if (version > 1.19)
- resp.setSubGraphId(req.getSubGraphId());
- if (version > 1.20)
- resp.setSubGraphOnly(req.getSubGraphOnly());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUGraphInfo(IEspContext &context,IEspWUGraphInfoRequest &req, IEspWUGraphInfoResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUGraphInfo", wuid);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_UPDATE_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- ensureWsWorkunitAccess(context, *cw, SecAccess_Write);
- resp.setWuid(wuid.str());
- resp.setName(req.getName());
- resp.setRunning(isRunning(*cw));
- if (notEmpty(req.getGID()))
- resp.setGID(req.getGID());
- if(!req.getBatchWU_isNull())
- resp.setBatchWU(req.getBatchWU());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUGVCGraphInfo(IEspContext &context,IEspWUGVCGraphInfoRequest &req, IEspWUGVCGraphInfoResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUGVCGraphInfo", wuid);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_UPDATE_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- ensureWsWorkunitAccess(context, *cw, SecAccess_Read);
- resp.setWuid(wuid.str());
- resp.setName(req.getName());
- resp.setRunning(isRunning(*cw));
- if (notEmpty(req.getGID()))
- resp.setGID(req.getGID());
- if(!req.getBatchWU_isNull())
- resp.setBatchWU(req.getBatchWU());
- StringBuffer xml("<Control><Endpoint><Query id=\"Gordon.Extractor.0\">");
- xml.appendf("<Graph id=\"%s\">", req.getName());
- if (context.getClientVersion() > 1.17)
- {
- xml.append("<Subgraph>");
- xml.append(req.getSubgraphId_isNull() ? 0 : req.getSubgraphId());
- xml.append("</Subgraph>");
- }
- xml.append("</Graph></Query></Endpoint></Control>");
- resp.setTheGraph(xml.str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUGraphTiming(IEspContext &context, IEspWUGraphTimingRequest &req, IEspWUGraphTimingResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWuid();
- WsWuHelpers::checkAndTrimWorkunit("WUGraphTiming", wuid);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid.str(), false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_UPDATE_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- ensureWsWorkunitAccess(context, *cw, SecAccess_Read);
- resp.updateWorkunit().setWuid(wuid.str());
- WsWuInfo winfo(context, cw);
- IArrayOf<IConstECLTimingData> timingData;
- winfo.getGraphTimingData(timingData, 0);
- resp.updateWorkunit().setTimingData(timingData);
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- int CWsWorkunitsSoapBindingEx::onGetForm(IEspContext &context, CHttpRequest* request, CHttpResponse* response, const char *service, const char *method)
- {
- try
- {
- StringBuffer xml;
- StringBuffer xslt;
- if(strieq(method,"WUQuery") || strieq(method,"WUJobList"))
- {
- Owned<IEnvironmentFactory> factory = getEnvironmentFactory();
- Owned<IConstEnvironment> environment = factory->openEnvironmentByFile();
- Owned<IPropertyTree> root = &environment->getPTree();
- if (!root)
- throw MakeStringException(ECLWATCH_CANNOT_GET_ENV_INFO, "Failed to get environment information.");
- if(strieq(method,"WUQuery"))
- {
- SecAccessFlags accessOwn;
- SecAccessFlags accessOthers;
- getUserWuAccessFlags(context, accessOwn, accessOthers, false);
- xml.append("<WUQuery>");
- if ((accessOwn == SecAccess_None) && (accessOthers == SecAccess_None))
- xml.appendf("<ErrorMessage>Access to workunit is denied.</ErrorMessage>");
- else
- {
- MapStringTo<bool> added;
- Owned<IPropertyTreeIterator> it = root->getElements("Software/Topology/Cluster");
- ForEach(*it)
- {
- const char *name = it->query().queryProp("@name");
- if (notEmpty(name) && !added.getValue(name))
- {
- added.setValue(name, true);
- appendXMLTag(xml, "Cluster", name);
- }
- }
- }
- xml.append("</WUQuery>");
- xslt.append(getCFD()).append("./smc_xslt/wuid_search.xslt");
- }
- else if (strieq(method,"WUJobList"))
- {
- StringBuffer cluster, defaultProcess, range;
- request->getParameter("Cluster", cluster);
- request->getParameter("Process",defaultProcess);
- request->getParameter("Range",range);
- Owned<IConstWUClusterInfo> clusterInfo = getTargetClusterInfo(cluster);
- xml.append("<WUJobList>");
- if (range.length())
- appendXMLTag(xml, "Range", range.str());
- if (clusterInfo)
- {
- const StringArray &thorInstances = clusterInfo->getThorProcesses();
- ForEachItemIn(i, thorInstances)
- {
- const char* instance = thorInstances.item(i);
- if (defaultProcess.length() && strieq(instance, defaultProcess.str()))
- xml.append("<Cluster selected=\"1\">").append(instance).append("</Cluster>");
- else
- xml.append("<Cluster>").append(instance).append("</Cluster>");
- }
- }
- xml.append("<TargetCluster>").append(cluster).append("</TargetCluster>");
- xml.append("</WUJobList>");
- xslt.append(getCFD()).append("./smc_xslt/jobs_search.xslt");
- response->addHeader("Expires", "0");
- }
- }
- if (xslt.length() && xml.length())
- {
- StringBuffer html;
- xsltTransform(xml.str(), xslt.str(), NULL, html);
- response->setContent(html.str());
- response->setContentType(HTTP_TYPE_TEXT_HTML_UTF8);
- response->send();
- return 0;
- }
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return onGetNotFound(context, request, response, service);
- }
- void deployEclOrArchive(IEspContext &context, IEspWUDeployWorkunitRequest & req, IEspWUDeployWorkunitResponse & resp)
- {
- NewWsWorkunit wu(context);
- SCMStringBuffer wuid;
- wu->getWuid(wuid);
- wu->setAction(WUActionCompile);
- StringBuffer name(req.getName());
- if (!name.trim().length() && notEmpty(req.getFileName()))
- splitFilename(req.getFileName(), NULL, NULL, &name, NULL);
- if (name.length())
- wu->setJobName(name.str());
- if (req.getObject().length())
- {
- StringBuffer text(req.getObject().length(), req.getObject().toByteArray());
- wu.setQueryText(text.str());
- }
- if (req.getQueryMainDefinition())
- wu.setQueryMain(req.getQueryMainDefinition());
- if (req.getSnapshot())
- wu->setSnapshot(req.getSnapshot());
- if (!req.getResultLimit_isNull())
- wu->setResultLimit(req.getResultLimit());
- wu->commit();
- wu.clear();
- WsWuHelpers::submitWsWorkunit(context, wuid.str(), req.getCluster(), NULL, 0, true, false, false, NULL, NULL, &req.getDebugValues());
- waitForWorkUnitToCompile(wuid.str(), req.getWait());
- WsWuInfo winfo(context, wuid.str());
- winfo.getCommon(resp.updateWorkunit(), WUINFO_All);
- winfo.getExceptions(resp.updateWorkunit(), WUINFO_All);
- name.clear();
- if (notEmpty(resp.updateWorkunit().getJobname()))
- origValueChanged(req.getName(), resp.updateWorkunit().getJobname(), name, false);
- if (name.length()) //non generated user specified name, so override #Workunit('name')
- {
- WorkunitUpdate wx(&winfo.cw->lock());
- wx->setJobName(name.str());
- resp.updateWorkunit().setJobname(name.str());
- }
- AuditSystemAccess(context.queryUserId(), true, "Updated %s", wuid.str());
- }
- StringBuffer &sharedObjectFileName(StringBuffer &filename, const char *name, const char *ext, unsigned copy)
- {
- filename.append((name && *name) ? name : "workunit");
- if (copy)
- filename.append('-').append(copy);
- if (notEmpty(ext))
- filename.append(ext);
- return filename;
- }
- inline StringBuffer &buildFullDllPath(StringBuffer &dllpath, StringBuffer &dllname, const char *dir, const char *name, const char *ext, unsigned copy)
- {
- return addPathSepChar(dllpath.set(dir)).append(sharedObjectFileName(dllname, name, ext, copy));
- }
- void writeSharedObject(const char *srcpath, const MemoryBuffer &obj, const char *dir, StringBuffer &dllpath, StringBuffer &dllname)
- {
- StringBuffer name, ext;
- if (srcpath && *srcpath)
- splitFilename(srcpath, NULL, NULL, &name, &ext);
- unsigned copy=0;
- buildFullDllPath(dllpath.clear(), dllname.clear(), dir, name.str(), ext.str(), copy);
- while (checkFileExists(dllpath.str()))
- buildFullDllPath(dllpath.clear(), dllname.clear(), dir, name.str(), ext.str(), ++copy);
- DBGLOG("Writing workunit dll: %s", dllpath.str());
- Owned<IFile> f = createIFile(dllpath.str());
- Owned<IFileIO> io = f->open(IFOcreate);
- io->write(0, obj.length(), obj.toByteArray());
- }
- void CWsWorkunitsEx::deploySharedObject(IEspContext &context, StringBuffer &wuid, const char *filename, const char *cluster, const char *name, const MemoryBuffer &obj, const char *dir, const char *xml)
- {
- StringBuffer dllpath, dllname;
- StringBuffer srcname(filename);
- if (!srcname.length())
- srcname.append(name).append(SharedObjectExtension);
- writeSharedObject(srcname.str(), obj, dir, dllpath, dllname);
- NewWsWorkunit wu(context);
- StringBufferAdaptor isvWuid(wuid);
- wu->getWuid(isvWuid);
- wu->setClusterName(cluster);
- wu->commit();
- StringBuffer dllXML;
- if (getWorkunitXMLFromFile(dllpath.str(), dllXML))
- {
- Owned<ILocalWorkUnit> embeddedWU = createLocalWorkUnit();
- embeddedWU->loadXML(dllXML.str());
- queryExtendedWU(wu)->copyWorkUnit(embeddedWU, true);
- }
- wu.associateDll(dllpath.str(), dllname.str());
- if (name && *name)
- wu->setJobName(name);
- //clean slate, copy only select items from processed workunit xml
- if (xml && *xml)
- {
- Owned<IPropertyTree> srcxml = createPTreeFromXMLString(xml);
- if (srcxml->hasProp("@jobName"))
- wu->setJobName(srcxml->queryProp("@jobName"));
- if (srcxml->hasProp("@token"))
- wu->setSecurityToken(srcxml->queryProp("@token"));
- if (srcxml->hasProp("Query/Text"))
- wu.setQueryText(srcxml->queryProp("Query/Text"));
- }
- wu->setState(WUStateCompiled);
- wu->commit();
- wu.clear();
- AuditSystemAccess(context.queryUserId(), true, "Updated %s", wuid.str());
- }
- void CWsWorkunitsEx::deploySharedObject(IEspContext &context, IEspWUDeployWorkunitRequest & req, IEspWUDeployWorkunitResponse & resp, const char *dir, const char *xml)
- {
- if (isEmpty(req.getFileName()))
- throw MakeStringException(ECLWATCH_INVALID_INPUT, "File name required when deploying a shared object.");
- const char *cluster = req.getCluster();
- if (isEmpty(cluster))
- throw MakeStringException(ECLWATCH_INVALID_INPUT, "Cluster name required when deploying a shared object.");
- StringBuffer wuid;
- deploySharedObject(context, wuid, req.getFileName(), cluster, req.getName(), req.getObject(), dir, xml);
- WsWuInfo winfo(context, wuid.str());
- winfo.getCommon(resp.updateWorkunit(), WUINFO_All);
- AuditSystemAccess(context.queryUserId(), true, "Updated %s", wuid.str());
- }
- bool CWsWorkunitsEx::onWUDeployWorkunit(IEspContext &context, IEspWUDeployWorkunitRequest & req, IEspWUDeployWorkunitResponse & resp)
- {
- const char *type = req.getObjType();
- try
- {
- if (!context.validateFeatureAccess(OWN_WU_ACCESS, SecAccess_Write, false))
- throw MakeStringException(ECLWATCH_ECL_WU_ACCESS_DENIED, "Failed to create workunit. Permission denied.");
- if (notEmpty(req.getCluster()) && !isValidCluster(req.getCluster()))
- throw MakeStringException(ECLWATCH_INVALID_CLUSTER_NAME, "Invalid cluster name: %s", req.getCluster());
- if (strieq(type, "archive")|| strieq(type, "ecl_text"))
- deployEclOrArchive(context, req, resp);
- else if (strieq(type, "shared_object"))
- deploySharedObject(context, req, resp, queryDirectory.str());
- else
- throw MakeStringException(ECLWATCH_INVALID_INPUT, "WUDeployWorkunit '%s' unkown object type.", type);
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- #ifdef _USE_ZLIB
- void CWsWorkunitsEx::addProcessLogfile(IZZIPor* zipper, Owned<IConstWorkUnit> &cwu, WsWuInfo &winfo, const char * process, PointerArray &mbArr)
- {
- Owned<IPropertyTreeIterator> procs = cwu->getProcesses(process, NULL);
- ForEach (*procs)
- {
- StringBuffer logSpec;
- IPropertyTree& proc = procs->query();
- proc.getProp("@log",logSpec);
- if (!logSpec.length())
- continue;
- StringBuffer pid;
- pid.appendf("%d",proc.getPropInt("@pid"));
- MemoryBuffer * pMB = NULL;
- try
- {
- pMB = new MemoryBuffer;
- if (0 == stricmp(process, "EclAgent"))
- winfo.getWorkunitEclAgentLog(logSpec.str(), pid.str(), *pMB);
- else if (0 == stricmp(process, "Thor"))
- winfo.getWorkunitThorLog(logSpec.str(), *pMB);
- else
- {
- delete pMB;
- return;
- }
- mbArr.append(pMB);
- }
- catch(IException *e)
- {
- StringBuffer s;
- e->errorMessage(s);
- pMB->append(s.str());
- e->Release();
- mbArr.append(pMB);
- }
- if (pMB && pMB->length())
- {
- const char * logName = logSpec.str();
- for (const char * p=logSpec; *p; p++)
- {
- if (*p == '\\' || *p == '/')
- logName = p+1;
- }
- zipper->addContentToZIP(pMB->length(), pMB->bufferBase(), (char*)logName, true);
- }
- }
- }
- #endif
- bool CWsWorkunitsEx::onWUCreateZAPInfo(IEspContext &context, IEspWUCreateZAPInfoRequest &req, IEspWUCreateZAPInfoResponse &resp)
- {
- try
- {
- #ifndef _USE_ZLIB
- throw MakeStringException(ECLWATCH_CANNOT_COMPRESS_DATA,"The data cannot be compressed.");
- #else
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cwu = factory->openWorkUnit(req.getWuid(), false);
- if(!cwu.get())
- throw MakeStringException(ECLWATCH_CANNOT_OPEN_WORKUNIT, "Cannot open workunit %s.", req.getWuid());
- //Create output report file
- StringBuffer zipFile;
- StringBuffer userName;
- if (context.queryUser())
- userName.append(context.queryUser()->getName());
- zipFile.append("ZAPReport_").append(req.getWuid()).append('_').append(userName.str()).append(".zip");
- SCMStringBuffer temp;
- StringBuffer sb;
- sb.append("Workunit: ").append(cwu->getWuid(temp)).append("\r\n");
- sb.append("User: ").append(cwu->getUser(temp).str()).append("\r\n");
- sb.append("Build Version:").append(req.getBuildVersion()).append("\r\n");
- sb.append("Cluster: ").append(cwu->getClusterName(temp).str()).append("\r\n");
- if (req.getESPIPAddress())
- sb.append("ESP: ").append(req.getESPIPAddress()).append("\r\n");
- if (req.getThorIPAddress())
- sb.append("Thor: ").append(req.getThorIPAddress()).append("\r\n");
- //Exceptions/Warnings/Info
- Owned<IConstWUExceptionIterator> exceptions = &cwu->getExceptions();
- StringBuffer info, warn, err;
- ForEach(*exceptions)
- {
- switch (exceptions->query().getSeverity())
- {
- case ExceptionSeverityInformation:
- info.append("\t").append(exceptions->query().getExceptionMessage(temp)).append("\r\n\r\n");
- break;
- case ExceptionSeverityWarning:
- warn.append("\t").append(exceptions->query().getExceptionMessage(temp)).append("\r\n\r\n");
- break;
- case ExceptionSeverityError:
- err.append("\t").append(exceptions->query().getExceptionMessage(temp)).append("\r\n\r\n");
- break;
- }
- }
- if (err.length())
- sb.append("Exceptions: ").append("\r\n").append(err);
- if (warn.length())
- sb.append("Warnings: ").append("\r\n").append(warn);
- if (info.length())
- sb.append("Information: ").append("\r\n").append(info);
- //User provided Information
- sb.append("Problem: ").append(req.getProblemDescription()).append("\r\n\r\n");
- sb.append("What Changed: ").append(req.getWhatChanged()).append("\r\n\r\n");
- sb.append("Timing: ").append(req.getWhereSlow()).append("\r\n\r\n");
- //Zip all files together
- {
- IZZIPor* zipper = createZZIPor();
- #ifdef _DEBUG
- zipper->setTraceLevel(100);
- #endif
- StringBuffer fs;
- //add report file to ZIP
- fs.append("ZAPReport_").append(req.getWuid()).append('_').append(userName.str()).append(".txt");
- zipper->addContentToZIP(sb.length(), (void*)sb.str(), (char*)fs.str(), false);
- //add ECL query/archive to zip
- Owned<IConstWUQuery> query = cwu->getQuery();
- StringBuffer eclContents;//String buffers containing file contents must persist until ziptofile is called !
- StringBuffer archiveContents;//String buffers containing file contents must persist until ziptofile is called !
- if(query)
- {
- //Add archive if present
- Owned<IConstWUAssociatedFileIterator> iter = &query->getAssociatedFiles();
- ForEach(*iter)
- {
- IConstWUAssociatedFile & cur = iter->query();
- SCMStringBuffer ssb;
- cur.getDescription(ssb);
- if (0 == stricmp(ssb.str(), "archive"))
- {
- cur.getName(ssb);
- if (ssb.length())
- {
- fs.clear().append("ZAPReport_").append(req.getWuid()).append('_').append(userName.str()).append(".archive");
- try
- {
- archiveContents.loadFile(ssb.str());
- zipper->addContentToZIP(archiveContents.length(), (void*)archiveContents.str(), (char*)fs.str(), true);
- }
- catch (IException *E)
- {
- DBGLOG("Error accessing archive file %s", ssb.str());
- E->Release();
- }
- break;
- }
- }
- }
- //Add Query
- query->getQueryText(temp);
- if (temp.length())
- {
- fs.clear().append("ZAPReport_").append(req.getWuid()).append('_').append(userName.str()).append(".ecl");
- eclContents.append(temp.str());
- zipper->addContentToZIP(eclContents.length(), (void*)eclContents.str(), (char*)fs.str(), true);
- }
- }
- //Add logfiles to ZIP
- WsWuInfo winfo(context, cwu);
- PointerArray eclAgentLogs;//array of dynamically allocated MemoryBuffers
- PointerArray thorLogs;
- addProcessLogfile(zipper, cwu, winfo, "EclAgent", eclAgentLogs);
- addProcessLogfile(zipper, cwu, winfo, "Thor", thorLogs);
- //Add Workunit XML file
- MemoryBuffer wuXmlMB;
- winfo.getWorkunitXml(NULL, wuXmlMB);
- fs.clear().append("ZAPReport_").append(req.getWuid()).append('_').append(userName.str()).append(".xml");
- zipper->addContentToZIP(wuXmlMB.length(), (void*)wuXmlMB.toByteArray(), (char*)fs.str(), true);
- //Write out ZIP file
- zipper->zipToFile(zipFile.str());
- for (aindex_t x=0; x<eclAgentLogs.length(); x++)
- delete (MemoryBuffer*)eclAgentLogs.item(x);
- for (aindex_t x=0; x<thorLogs.length(); x++)
- delete (MemoryBuffer*)thorLogs.item(x);
- }
- //Download ZIP file to user
- Owned<IFile> f = createIFile(zipFile.str());
- Owned<IFileIO> io = f->open(IFOread);
- MemoryBuffer mb;
- void * data = mb.reserve((unsigned)io->size());
- size32_t read = io->read(0, (unsigned)io->size(), data);
- mb.setLength(read);
- resp.setThefile(mb);
- resp.setThefile_mimetype(HTTP_TYPE_OCTET_STREAM);
- StringBuffer headerStr("attachment;filename=");
- headerStr.append(zipFile.str());
- context.addCustomerHeader("Content-disposition", headerStr.str());
- io->close();
- f->remove();
- #endif
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
- bool CWsWorkunitsEx::onWUGetZAPInfo(IEspContext &context, IEspWUGetZAPInfoRequest &req, IEspWUGetZAPInfoResponse &resp)
- {
- try
- {
- StringBuffer wuid = req.getWUID();
- WsWuHelpers::checkAndTrimWorkunit("WUGetZAPInfo", wuid);
- Owned<IWorkUnitFactory> factory = getWorkUnitFactory(context.querySecManager(), context.queryUser());
- Owned<IConstWorkUnit> cw = factory->openWorkUnit(wuid, false);
- if(!cw)
- throw MakeStringException(ECLWATCH_CANNOT_OPEN_WORKUNIT,"Cannot open workunit %s.",wuid.str());
- StringBuffer EspIP, ThorIP;
- resp.setWUID(wuid.str());
- resp.setBuildVersion(getBuildVersion());
- IpAddress ipaddr = queryHostIP();
- ipaddr.getIpText(EspIP);
- resp.setESPIPAddress(EspIP.str());
- //Get Archive
- Owned<IConstWUQuery> query = cw->getQuery();
- if(query)
- {
- SCMStringBuffer queryText;
- query->getQueryText(queryText);
- if (queryText.length() && isArchiveQuery(queryText.str()))
- resp.setArchive(queryText.str());
- }
- //Get Thor IP
- BoolHash uniqueProcesses, uniqueThorIPs;
- Owned<IStringIterator> thorInstances = cw->getProcesses("Thor");
- ForEach (*thorInstances)
- {
- SCMStringBuffer processName;
- thorInstances->str(processName);
- if ((processName.length() < 1) || uniqueProcesses.getValue(processName.str()))
- continue;
- uniqueProcesses.setValue(processName.str(), true);
- Owned<IStringIterator> thorLogs = cw->getLogs("Thor", processName.str());
- ForEach (*thorLogs)
- {
- SCMStringBuffer logName;
- thorLogs->str(logName);
- if (!logName.length())
- continue;
- const char* thorIPPtr = NULL;
- const char* ptr = logName.str();
- while (ptr)
- {
- if (!thorIPPtr && (*ptr != '/'))
- thorIPPtr = ptr;
- else if (thorIPPtr && (*ptr == '/'))
- break;
- ptr++;
- }
- if (!thorIPPtr)
- continue;
- //Found a thor IP
- if (ThorIP.length())
- ThorIP.append(",");
- if (!*ptr)
- ThorIP.append(thorIPPtr);
- else
- ThorIP.append(ptr-thorIPPtr, thorIPPtr);
- }
- }
- if (ThorIP.length())
- resp.setThorIPAddress(ThorIP.str());
- }
- catch(IException* e)
- {
- FORWARDEXCEPTION(context, e, ECLWATCH_INTERNAL_ERROR);
- }
- return true;
- }
|