thgraphmaster.cpp 99 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134213521362137213821392140214121422143214421452146214721482149215021512152215321542155215621572158215921602161216221632164216521662167216821692170217121722173217421752176217721782179218021812182218321842185218621872188218921902191219221932194219521962197219821992200220122022203220422052206220722082209221022112212221322142215221622172218221922202221222222232224222522262227222822292230223122322233223422352236223722382239224022412242224322442245224622472248224922502251225222532254225522562257225822592260226122622263226422652266226722682269227022712272227322742275227622772278227922802281228222832284228522862287228822892290229122922293229422952296229722982299230023012302230323042305230623072308230923102311231223132314231523162317231823192320232123222323232423252326232723282329233023312332233323342335233623372338233923402341234223432344234523462347234823492350235123522353235423552356235723582359236023612362236323642365236623672368236923702371237223732374237523762377237823792380238123822383238423852386238723882389239023912392239323942395239623972398239924002401240224032404240524062407240824092410241124122413241424152416241724182419242024212422242324242425242624272428242924302431243224332434243524362437243824392440244124422443244424452446244724482449245024512452245324542455245624572458245924602461246224632464246524662467246824692470247124722473247424752476247724782479248024812482248324842485248624872488248924902491249224932494249524962497249824992500250125022503250425052506250725082509251025112512251325142515251625172518251925202521252225232524252525262527252825292530253125322533253425352536253725382539254025412542254325442545254625472548254925502551255225532554255525562557255825592560256125622563256425652566256725682569257025712572257325742575257625772578257925802581258225832584258525862587258825892590259125922593259425952596259725982599260026012602260326042605260626072608260926102611261226132614261526162617261826192620262126222623262426252626262726282629263026312632263326342635263626372638263926402641264226432644264526462647264826492650265126522653265426552656265726582659266026612662266326642665266626672668266926702671267226732674267526762677267826792680268126822683268426852686268726882689269026912692269326942695269626972698269927002701270227032704270527062707270827092710271127122713271427152716271727182719272027212722272327242725272627272728272927302731273227332734273527362737273827392740274127422743274427452746274727482749275027512752275327542755275627572758275927602761276227632764276527662767276827692770277127722773277427752776277727782779278027812782278327842785278627872788278927902791279227932794279527962797279827992800280128022803280428052806280728082809281028112812281328142815281628172818281928202821282228232824282528262827282828292830283128322833283428352836283728382839284028412842284328442845284628472848284928502851285228532854285528562857285828592860286128622863286428652866286728682869287028712872287328742875287628772878287928802881288228832884288528862887288828892890289128922893289428952896289728982899290029012902290329042905290629072908290929102911291229132914291529162917291829192920292129222923292429252926292729282929
  1. /*##############################################################################
  2. HPCC SYSTEMS software Copyright (C) 2012 HPCC Systems®.
  3. Licensed under the Apache License, Version 2.0 (the "License");
  4. you may not use this file except in compliance with the License.
  5. You may obtain a copy of the License at
  6. http://www.apache.org/licenses/LICENSE-2.0
  7. Unless required by applicable law or agreed to in writing, software
  8. distributed under the License is distributed on an "AS IS" BASIS,
  9. WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  10. See the License for the specific language governing permissions and
  11. limitations under the License.
  12. ############################################################################## */
  13. #include <limits.h>
  14. #include <stdlib.h>
  15. #include "jprop.hpp"
  16. #include "jexcept.hpp"
  17. #include "jiter.ipp"
  18. #include "jlzw.hpp"
  19. #include "jsocket.hpp"
  20. #include "jset.hpp"
  21. #include "jsort.hpp"
  22. #include "portlist.h"
  23. #include "jhtree.hpp"
  24. #include "mputil.hpp"
  25. #include "dllserver.hpp"
  26. #include "dautils.hpp"
  27. #include "danqs.hpp"
  28. #include "daclient.hpp"
  29. #include "daaudit.hpp"
  30. #include "wujobq.hpp"
  31. #include "thorport.hpp"
  32. #include "commonext.hpp"
  33. #include "thorxmlread.hpp"
  34. #include "thorplugin.hpp"
  35. #include "thormisc.hpp"
  36. #include "thgraphmaster.ipp"
  37. #include "thdemonserver.hpp"
  38. #include "rtlds_imp.hpp"
  39. #include "eclhelper.hpp"
  40. #include "thexception.hpp"
  41. #include "thactivitymaster.ipp"
  42. #include "thmem.hpp"
  43. #include "thcompressutil.hpp"
  44. using roxiemem::OwnedRoxieString;
  45. static CriticalSection *jobManagerCrit;
  46. MODULE_INIT(INIT_PRIORITY_STANDARD)
  47. {
  48. jobManagerCrit = new CriticalSection;
  49. return true;
  50. }
  51. MODULE_EXIT()
  52. {
  53. delete jobManagerCrit;
  54. }
  55. unsigned uniqGraphId = 1;
  56. #define FATAL_TIMEOUT 60
  57. class CFatalHandler : public CTimeoutTrigger, implements IFatalHandler
  58. {
  59. public:
  60. IMPLEMENT_IINTERFACE;
  61. CFatalHandler(unsigned timeout) : CTimeoutTrigger(timeout, "EXCEPTION")
  62. {
  63. }
  64. virtual bool action()
  65. {
  66. StringBuffer s("FAILED TO RECOVER FROM EXCEPTION, STOPPING THOR");
  67. FLLOG(MCoperatorWarning, thorJob, exception, s.str());
  68. Owned<IJobManager> jobManager = getJobManager();
  69. if (jobManager)
  70. {
  71. jobManager->fatal(exception);
  72. jobManager.clear();
  73. }
  74. return true;
  75. }
  76. // IFatalHandler
  77. virtual void inform(IException *e)
  78. {
  79. CTimeoutTrigger::inform(e);
  80. }
  81. virtual void clear()
  82. {
  83. CTimeoutTrigger::clear();
  84. }
  85. };
  86. /////
  87. CSlaveMessageHandler::CSlaveMessageHandler(CJobMaster &_job, mptag_t _mptag) : threaded("CSlaveMessageHandler"), job(_job), mptag(_mptag)
  88. {
  89. stopped = false;
  90. threaded.init(this);
  91. }
  92. CSlaveMessageHandler::~CSlaveMessageHandler()
  93. {
  94. stop();
  95. }
  96. void CSlaveMessageHandler::stop()
  97. {
  98. if (!stopped)
  99. {
  100. stopped = true;
  101. job.queryJobChannel(0).queryJobComm().cancel(0, mptag);
  102. threaded.join();
  103. }
  104. }
  105. void CSlaveMessageHandler::main()
  106. {
  107. try
  108. {
  109. loop
  110. {
  111. rank_t sender;
  112. CMessageBuffer msg;
  113. if (stopped || !job.queryJobChannel(0).queryJobComm().recv(msg, RANK_ALL, mptag, &sender))
  114. break;
  115. SlaveMsgTypes msgType;
  116. msg.read((int &)msgType);
  117. switch (msgType)
  118. {
  119. case smt_errorMsg:
  120. {
  121. unsigned slave;
  122. msg.read(slave);
  123. Owned<IThorException> e = deserializeThorException(msg);
  124. e->setSlave(slave+1);
  125. Owned<CGraphBase> graph = job.queryJobChannel(0).getGraph(e->queryGraphId());
  126. if (graph)
  127. {
  128. activity_id id = e->queryActivityId();
  129. if (id)
  130. {
  131. CGraphElementBase *elem = graph->queryElement(id);
  132. CActivityBase *act = elem->queryActivity();
  133. if (act)
  134. act->fireException(e);
  135. else
  136. graph->fireException(e);
  137. }
  138. else
  139. graph->fireException(e);
  140. }
  141. else
  142. job.fireException(e);
  143. if (msg.getReplyTag() <= TAG_REPLY_BASE)
  144. {
  145. msg.clear();
  146. job.queryJobChannel(0).queryJobComm().reply(msg);
  147. }
  148. break;
  149. }
  150. case smt_dataReq:
  151. {
  152. graph_id gid;
  153. activity_id id;
  154. unsigned slave;
  155. msg.read(slave);
  156. msg.read(gid);
  157. msg.read(id);
  158. msg.clear();
  159. Owned<CGraphBase> graph = job.queryJobChannel(0).getGraph(gid);
  160. if (graph)
  161. {
  162. CMasterGraphElement *e = (CMasterGraphElement *)graph->queryElement(id);
  163. e->queryActivity()->getInitializationData(slave, msg);
  164. }
  165. job.queryJobChannel(0).queryJobComm().reply(msg);
  166. break;
  167. }
  168. case smt_initGraphReq:
  169. {
  170. graph_id gid;
  171. msg.read(gid);
  172. Owned<CMasterGraph> graph = (CMasterGraph *)job.queryJobChannel(0).getGraph(gid);
  173. assertex(graph);
  174. {
  175. CriticalBlock b(graph->queryCreateLock());
  176. Owned<IThorActivityIterator> iter = graph->getIterator();
  177. // onCreate all
  178. ForEach (*iter)
  179. {
  180. CMasterGraphElement &element = (CMasterGraphElement &)iter->query();
  181. element.onCreate();
  182. }
  183. }
  184. msg.clear();
  185. graph->serializeCreateContexts(msg);
  186. job.queryJobChannel(0).queryJobComm().reply(msg);
  187. break;
  188. }
  189. case smt_initActDataReq:
  190. {
  191. graph_id gid;
  192. msg.read(gid);
  193. unsigned slave;
  194. msg.read(slave);
  195. Owned<CMasterGraph> graph = (CMasterGraph *)job.queryJobChannel(0).getGraph(gid);
  196. assertex(graph);
  197. CGraphElementArray toSerialize;
  198. CriticalBlock b(graph->queryCreateLock());
  199. size32_t parentExtractSz;
  200. msg.read(parentExtractSz);
  201. const byte *parentExtract = NULL;
  202. if (parentExtractSz)
  203. {
  204. parentExtract = msg.readDirect(parentExtractSz);
  205. StringBuffer msg("Graph(");
  206. msg.append(graph->queryGraphId()).append(") - initializing master graph with parentExtract ").append(parentExtractSz).append(" bytes");
  207. DBGLOG("%s", msg.str());
  208. parentExtract = graph->setParentCtx(parentExtractSz, parentExtract);
  209. }
  210. Owned<IException> exception;
  211. loop
  212. {
  213. activity_id id;
  214. msg.read(id);
  215. if (!id)
  216. break;
  217. CMasterGraphElement *element = (CMasterGraphElement *)graph->queryElement(id);
  218. assertex(element);
  219. try
  220. {
  221. element->reset();
  222. size32_t startCtxLen;
  223. msg.read(startCtxLen);
  224. element->doCreateActivity(parentExtractSz, parentExtract, startCtxLen ? &msg : nullptr);
  225. if (element->queryActivity())
  226. element->preStart(parentExtractSz, parentExtract);
  227. }
  228. catch (IException *e)
  229. {
  230. EXCLOG(e, NULL);
  231. exception.setown(e);
  232. break;
  233. }
  234. CActivityBase *activity = element->queryActivity();
  235. if (activity && activity->needReInit())
  236. element->sentActInitData->set(slave, 0); // clear to permit serializeActivityInitData to resend
  237. toSerialize.append(*LINK(element));
  238. }
  239. msg.clear();
  240. mptag_t replyTag = job.queryJobChannel(0).queryMPServer().createReplyTag();
  241. msg.append(replyTag); // second reply
  242. if (exception)
  243. {
  244. msg.append(true);
  245. serializeException(exception, msg);
  246. }
  247. else
  248. {
  249. msg.append(false);
  250. CGraphElementArrayIterator iter(toSerialize);
  251. graph->serializeActivityInitData(slave, msg, iter);
  252. }
  253. job.queryJobChannel(0).queryJobComm().reply(msg);
  254. if (!job.queryJobChannel(0).queryJobComm().recv(msg, slave+1, replyTag, NULL, MEDIUMTIMEOUT))
  255. throwUnexpected();
  256. if (exception)
  257. throw exception.getClear();
  258. bool error;
  259. msg.read(error);
  260. if (error)
  261. {
  262. Owned<IThorException> e = deserializeThorException(msg);
  263. e->setSlave(slave);
  264. StringBuffer tmpStr("Slave ");
  265. job.queryJobGroup().queryNode(slave).endpoint().getUrlStr(tmpStr);
  266. GraphPrintLog(graph, e, "%s", tmpStr.append(": slave initialization error").str());
  267. throw e.getClear();
  268. }
  269. break;
  270. }
  271. case smt_getPhysicalName:
  272. {
  273. LOG(MCdebugProgress, unknownJob, "getPhysicalName called from node %d", sender-1);
  274. StringAttr logicalName;
  275. unsigned partNo;
  276. bool create;
  277. msg.read(logicalName);
  278. msg.read(partNo);
  279. msg.read(create);
  280. msg.clear();
  281. StringBuffer phys;
  282. if (create && !job.queryCreatedFile(logicalName)) // not sure who would do this ever??
  283. queryThorFileManager().getPublishPhysicalName(job, logicalName, partNo, phys);
  284. else
  285. queryThorFileManager().getPhysicalName(job, logicalName, partNo, phys);
  286. msg.append(phys);
  287. job.queryJobChannel(0).queryJobComm().reply(msg);
  288. break;
  289. }
  290. case smt_getFileOffset:
  291. {
  292. LOG(MCdebugProgress, unknownJob, "getFileOffset called from node %d", sender-1);
  293. StringAttr logicalName;
  294. unsigned partNo;
  295. msg.read(logicalName);
  296. msg.read(partNo);
  297. msg.clear();
  298. offset_t offset = queryThorFileManager().getFileOffset(job, logicalName, partNo);
  299. msg.append(offset);
  300. job.queryJobChannel(0).queryJobComm().reply(msg);
  301. break;
  302. }
  303. case smt_actMsg:
  304. {
  305. LOG(MCdebugProgress, unknownJob, "smt_actMsg called from node %d", sender-1);
  306. graph_id gid;
  307. msg.read(gid);
  308. activity_id id;
  309. msg.read(id);
  310. Owned<CMasterGraph> graph = (CMasterGraph *)job.queryJobChannel(0).getGraph(gid);
  311. assertex(graph);
  312. CMasterGraphElement *container = (CMasterGraphElement *)graph->queryElement(id);
  313. assertex(container);
  314. CMasterActivity *activity = (CMasterActivity *)container->queryActivity();
  315. assertex(activity);
  316. activity->handleSlaveMessage(msg); // don't block
  317. break;
  318. }
  319. case smt_getresult:
  320. {
  321. unsigned slave;
  322. msg.read(slave);
  323. LOG(MCdebugProgress, unknownJob, "smt_getresult called from slave %d", slave);
  324. graph_id gid;
  325. msg.read(gid);
  326. activity_id ownerId;
  327. msg.read(ownerId);
  328. unsigned resultId;
  329. msg.read(resultId);
  330. CJobChannel &jobChannel = job.queryJobChannel(0);
  331. mptag_t replyTag = jobChannel.deserializeMPTag(msg);
  332. Owned<IThorResult> result = jobChannel.getOwnedResult(gid, ownerId, resultId);
  333. Owned<IRowStream> resultStream = result->getRowStream();
  334. sendInChunks(job.queryJobChannel(0).queryJobComm(), slave+1, replyTag, resultStream, result->queryRowInterfaces());
  335. break;
  336. }
  337. }
  338. }
  339. }
  340. catch (IException *e)
  341. {
  342. job.fireException(e);
  343. e->Release();
  344. }
  345. }
  346. //////////////////////
  347. CMasterActivity::CMasterActivity(CGraphElementBase *_container) : CActivityBase(_container), threaded("CMasterActivity", this), timingInfo(_container->queryJob())
  348. {
  349. notedWarnings = createThreadSafeBitSet();
  350. mpTag = TAG_NULL;
  351. data = new MemoryBuffer[container.queryJob().querySlaves()];
  352. asyncStart = false;
  353. if (container.isSink())
  354. progressInfo.append(*new ProgressInfo(queryJob()));
  355. else
  356. {
  357. unsigned o=0;
  358. for (; o<container.getOutputs(); o++)
  359. progressInfo.append(*new ProgressInfo(queryJob()));
  360. }
  361. }
  362. CMasterActivity::~CMasterActivity()
  363. {
  364. if (asyncStart)
  365. threaded.join();
  366. notedWarnings->Release();
  367. queryJob().freeMPTag(mpTag);
  368. delete [] data;
  369. }
  370. void CMasterActivity::addReadFile(IDistributedFile *file, bool temp)
  371. {
  372. readFiles.append(*LINK(file));
  373. if (!temp) // NB: Temps not listed in workunit
  374. queryThorFileManager().noteFileRead(container.queryJob(), file);
  375. }
  376. IDistributedFile *CMasterActivity::queryReadFile(unsigned f)
  377. {
  378. if (f>=readFiles.ordinality())
  379. return NULL;
  380. return &readFiles.item(f);
  381. }
  382. void CMasterActivity::preStart(size32_t parentExtractSz, const byte *parentExtract)
  383. {
  384. CActivityBase::preStart(parentExtractSz, parentExtract);
  385. IArrayOf<IDistributedFile> tmpFiles;
  386. tmpFiles.swapWith(readFiles);
  387. ForEachItemIn(f, tmpFiles)
  388. {
  389. IDistributedFile &file = tmpFiles.item(f);
  390. IDistributedSuperFile *super = file.querySuperFile();
  391. if (super)
  392. getSuperFileSubs(super, readFiles, true);
  393. else
  394. readFiles.append(*LINK(&file));
  395. }
  396. }
  397. MemoryBuffer &CMasterActivity::queryInitializationData(unsigned slave) const
  398. { // NB: not intended to be called by multiple threads.
  399. return data[slave].reset();
  400. }
  401. MemoryBuffer &CMasterActivity::getInitializationData(unsigned slave, MemoryBuffer &dst) const
  402. {
  403. return dst.append(data[slave]);
  404. }
  405. void CMasterActivity::main()
  406. {
  407. try
  408. {
  409. process();
  410. }
  411. catch (IException *e)
  412. {
  413. Owned<IException> e2;
  414. if (QUERYINTERFACE(e, ISEH_Exception))
  415. e2.setown(MakeThorFatal(e, TE_SEH, "(SEH)"));
  416. else
  417. e2.setown(MakeActivityException(this, e, "Master exception"));
  418. e->Release();
  419. ActPrintLog(e2, "In CMasterActivity::main");
  420. fireException(e2);
  421. }
  422. catch (CATCHALL)
  423. {
  424. Owned<IException> e = MakeThorFatal(NULL, TE_MasterProcessError, "FATAL: Unknown master process exception kind=%s, id=%" ACTPF "d", activityKindStr(container.getKind()), container.queryId());
  425. ActPrintLog(e, "In CMasterActivity::main");
  426. fireException(e);
  427. }
  428. }
  429. void CMasterActivity::init()
  430. {
  431. readFiles.kill();
  432. }
  433. void CMasterActivity::startProcess(bool async)
  434. {
  435. if (async)
  436. {
  437. asyncStart = true;
  438. threaded.start();
  439. }
  440. else
  441. main();
  442. }
  443. bool CMasterActivity::wait(unsigned timeout)
  444. {
  445. if (!asyncStart)
  446. return true;
  447. return threaded.join(timeout);
  448. }
  449. void CMasterActivity::kill()
  450. {
  451. CActivityBase::kill();
  452. readFiles.kill();
  453. }
  454. bool CMasterActivity::fireException(IException *_e)
  455. {
  456. IThorException *e = QUERYINTERFACE(_e, IThorException);
  457. if (!e) return false;
  458. switch (e->errorCode())
  459. {
  460. case TE_LargeBufferWarning:
  461. case TE_MoxieIndarOverflow:
  462. case TE_BuildIndexFewExcess:
  463. case TE_FetchMisaligned:
  464. case TE_FetchOutOfRange:
  465. case TE_CouldNotCreateLookAhead:
  466. case TE_SpillAdded:
  467. case TE_ReadPartialFromPipe:
  468. case TE_LargeAggregateTable:
  469. {
  470. if (!notedWarnings->testSet(e->errorCode()))
  471. CActivityBase::fireException(e);
  472. return true;
  473. }
  474. }
  475. return container.queryOwner().fireException(e);
  476. }
  477. void CMasterActivity::reset()
  478. {
  479. asyncStart = false;
  480. CActivityBase::reset();
  481. }
  482. void CMasterActivity::deserializeStats(unsigned node, MemoryBuffer &mb)
  483. {
  484. CriticalBlock b(progressCrit); // don't think needed
  485. unsigned __int64 localTimeNs;
  486. mb.read(localTimeNs);
  487. timingInfo.set(node, localTimeNs);
  488. rowcount_t count;
  489. ForEachItemIn(p, progressInfo)
  490. {
  491. mb.read(count);
  492. progressInfo.item(p).set(node, count);
  493. }
  494. }
  495. void CMasterActivity::getActivityStats(IStatisticGatherer & stats)
  496. {
  497. timingInfo.getStats(stats);
  498. }
  499. void CMasterActivity::getEdgeStats(IStatisticGatherer & stats, unsigned idx)
  500. {
  501. CriticalBlock b(progressCrit);
  502. if (progressInfo.isItem(idx))
  503. progressInfo.item(idx).getStats(stats);
  504. }
  505. void CMasterActivity::done()
  506. {
  507. CActivityBase::done();
  508. ForEachItemIn(s, readFiles)
  509. {
  510. IDistributedFile &file = readFiles.item(s);
  511. file.setAccessed();
  512. }
  513. }
  514. //////////////////////
  515. // CMasterGraphElement impl.
  516. //
  517. CMasterGraphElement::CMasterGraphElement(CGraphBase &_owner, IPropertyTree &_xgmml) : CGraphElementBase(_owner, _xgmml)
  518. {
  519. }
  520. bool CMasterGraphElement::checkUpdate()
  521. {
  522. if (!onlyUpdateIfChanged)
  523. return false;
  524. if (!globals->getPropBool("@updateEnabled", true) || 0 != queryJob().getWorkUnitValueInt("disableUpdate", 0))
  525. return false;
  526. bool doCheckUpdate = false;
  527. OwnedRoxieString filename;
  528. unsigned eclCRC;
  529. unsigned __int64 totalCRC;
  530. bool temporary = false;
  531. switch (getKind())
  532. {
  533. case TAKindexwrite:
  534. {
  535. IHThorIndexWriteArg *helper = (IHThorIndexWriteArg *)queryHelper();
  536. doCheckUpdate = 0 != (helper->getFlags() & TIWupdate);
  537. filename.set(helper->getFileName());
  538. helper->getUpdateCRCs(eclCRC, totalCRC);
  539. break;
  540. }
  541. case TAKdiskwrite:
  542. case TAKcsvwrite:
  543. case TAKxmlwrite:
  544. case TAKjsonwrite:
  545. {
  546. IHThorDiskWriteArg *helper = (IHThorDiskWriteArg *)queryHelper();
  547. doCheckUpdate = 0 != (helper->getFlags() & TDWupdate);
  548. filename.set(helper->getFileName());
  549. helper->getUpdateCRCs(eclCRC, totalCRC);
  550. if (TAKdiskwrite == getKind())
  551. temporary = 0 != (helper->getFlags() & (TDXtemporary|TDXjobtemp));
  552. break;
  553. }
  554. }
  555. if (doCheckUpdate)
  556. {
  557. StringAttr lfn;
  558. Owned<IDistributedFile> file = queryThorFileManager().lookup(queryJob(), filename, temporary, true);
  559. if (file)
  560. {
  561. IPropertyTree &props = file->queryAttributes();
  562. if ((eclCRC == props.getPropInt("@eclCRC")) && (totalCRC == props.getPropInt64("@totalCRC")))
  563. {
  564. // so this needs pruning
  565. Owned<IThorException> e = MakeActivityWarning(this, TE_UpToDate, "output file = '%s' - is up to date - it will not be rebuilt", file->queryLogicalName());
  566. queryOwner().fireException(e);
  567. return true;
  568. }
  569. }
  570. }
  571. return false;
  572. }
  573. void CMasterGraphElement::initActivity()
  574. {
  575. CriticalBlock b(crit);
  576. if (!initialized)
  577. {
  578. initialized = true;
  579. ((CMasterActivity *)queryActivity())->init();
  580. }
  581. }
  582. void CMasterGraphElement::reset()
  583. {
  584. CGraphElementBase::reset();
  585. if (activity && activity->needReInit())
  586. initialized = false;
  587. }
  588. void CMasterGraphElement::doCreateActivity(size32_t parentExtractSz, const byte *parentExtract, MemoryBuffer *startCtx)
  589. {
  590. createActivity();
  591. onStart(parentExtractSz, parentExtract, startCtx);
  592. initActivity();
  593. }
  594. void CMasterGraphElement::slaveDone(size32_t slaveIdx, MemoryBuffer &mb)
  595. {
  596. ((CMasterActivity *)queryActivity())->slaveDone(slaveIdx, mb);
  597. }
  598. //////
  599. ///////
  600. class CBarrierMaster : public CInterface, implements IBarrier
  601. {
  602. mptag_t tag;
  603. Linked<ICommunicator> comm;
  604. bool receiving;
  605. public:
  606. IMPLEMENT_IINTERFACE;
  607. CBarrierMaster(ICommunicator &_comm, mptag_t _tag) : comm(&_comm), tag(_tag)
  608. {
  609. receiving = false;
  610. }
  611. virtual const mptag_t queryTag() const { return tag; }
  612. virtual bool wait(bool exception, unsigned timeout)
  613. {
  614. Owned<IException> e;
  615. CTimeMon tm(timeout);
  616. unsigned s=comm->queryGroup().ordinality()-1;
  617. bool aborted = false;
  618. CMessageBuffer msg;
  619. Owned<IBitSet> raisedSet = createThreadSafeBitSet();
  620. unsigned remaining = timeout;
  621. while (s--)
  622. {
  623. rank_t sender;
  624. msg.clear();
  625. if (INFINITE != timeout && tm.timedout(&remaining))
  626. {
  627. if (exception)
  628. throw createBarrierAbortException();
  629. else
  630. return false;
  631. }
  632. {
  633. BooleanOnOff onOff(receiving);
  634. if (!comm->recv(msg, RANK_ALL, tag, &sender, remaining))
  635. break;
  636. }
  637. msg.read(aborted);
  638. bool hasExcept;
  639. msg.read(hasExcept);
  640. if (hasExcept && !e.get())
  641. e.setown(deserializeException(msg));
  642. sender = sender - 1; // 0 = master
  643. if (raisedSet->testSet(sender, true) && !aborted)
  644. WARNLOG("CBarrierMaster, raise barrier message on tag %d, already received from slave %d", tag, sender);
  645. if (aborted) break;
  646. }
  647. msg.clear();
  648. msg.append(aborted);
  649. if (e)
  650. {
  651. msg.append(true);
  652. serializeException(e, msg);
  653. }
  654. else
  655. msg.append(false);
  656. if (INFINITE != timeout && tm.timedout(&remaining))
  657. {
  658. if (exception)
  659. throw createBarrierAbortException();
  660. else
  661. return false;
  662. }
  663. if (!comm->send(msg, RANK_ALL_OTHER, tag, INFINITE != timeout ? remaining : LONGTIMEOUT))
  664. throw MakeStringException(0, "CBarrierMaster::wait - Timeout sending to slaves");
  665. if (aborted)
  666. {
  667. if (!exception)
  668. return false;
  669. if (e)
  670. throw e.getClear();
  671. else
  672. throw createBarrierAbortException();
  673. }
  674. return true;
  675. }
  676. virtual void cancel(IException *e)
  677. {
  678. if (receiving)
  679. comm->cancel(RANK_ALL, tag);
  680. CMessageBuffer msg;
  681. msg.append(true);
  682. if (e)
  683. {
  684. msg.append(true);
  685. serializeException(e, msg);
  686. }
  687. else
  688. msg.append(false);
  689. if (!comm->send(msg, RANK_ALL_OTHER, tag, LONGTIMEOUT))
  690. throw MakeStringException(0, "CBarrierMaster::cancel - Timeout sending to slaves");
  691. }
  692. };
  693. /////////////
  694. class CMasterGraphTempHandler : public CGraphTempHandler
  695. {
  696. public:
  697. CMasterGraphTempHandler(CJobBase &job, bool errorOnMissing) : CGraphTempHandler(job, errorOnMissing) { }
  698. virtual bool removeTemp(const char *name)
  699. {
  700. queryThorFileManager().clearCacheEntry(name);
  701. return true;
  702. }
  703. virtual void registerFile(const char *name, graph_id graphId, unsigned usageCount, bool temp, WUFileKind fileKind, StringArray *clusters)
  704. {
  705. if (!temp || job.queryUseCheckpoints())
  706. {
  707. Owned<IWorkUnit> wu = &job.queryWorkUnit().lock();
  708. wu->addFile(name, clusters, usageCount, fileKind, job.queryGraphName());
  709. }
  710. else
  711. CGraphTempHandler::registerFile(name, graphId, usageCount, temp, fileKind, clusters);
  712. }
  713. virtual void deregisterFile(const char *name, bool kept) // NB: only called for temp files
  714. {
  715. if (kept || job.queryUseCheckpoints())
  716. {
  717. Owned<IWorkUnit> wu = &job.queryWorkUnit().lock();
  718. wu->releaseFile(name);
  719. }
  720. else
  721. CGraphTempHandler::deregisterFile(name);
  722. }
  723. virtual void clearTemps()
  724. {
  725. try
  726. {
  727. if (!job.queryPausing()) // temps of completed workunit will have been preserved and want to keep
  728. {
  729. Owned<IWorkUnit> lwu = &job.queryWorkUnit().lock();
  730. lwu->deleteTempFiles(job.queryGraphName(), false, false);
  731. }
  732. }
  733. catch (IException *e)
  734. {
  735. EXCLOG(e, "Problem deleting temp files");
  736. e->Release();
  737. }
  738. CGraphTempHandler::clearTemps();
  739. }
  740. };
  741. static const char * getResultText(StringBuffer & s, const char * stepname, unsigned sequence)
  742. {
  743. switch ((int)sequence)
  744. {
  745. case -1: return s.append("STORED('").append(stepname).append("')");
  746. case -2: return s.append("PERSIST('").append(stepname).append("')");
  747. case -3: return s.append("global('").append(stepname).append("')");
  748. default:
  749. if (stepname)
  750. return s.append(stepname);
  751. return s.append('#').append(sequence);
  752. }
  753. }
  754. class CThorCodeContextMaster : public CThorCodeContextBase
  755. {
  756. Linked<IConstWorkUnit> workunit;
  757. Owned<IDistributedFileTransaction> superfiletransaction;
  758. IConstWUResult * getResult(const char * name, unsigned sequence)
  759. {
  760. return getWorkUnitResult(workunit, name, sequence);
  761. }
  762. #define PROTECTED_GETRESULT(STEPNAME, SEQUENCE, KIND, KINDTEXT, ACTION) \
  763. LOG(MCdebugProgress, unknownJob, "getResult%s(%s,%d)", KIND, STEPNAME?STEPNAME:"", SEQUENCE); \
  764. Owned<IConstWUResult> r = getResultForGet(STEPNAME, SEQUENCE); \
  765. try \
  766. { \
  767. ACTION \
  768. } \
  769. catch (IException * e) { \
  770. StringBuffer s, text; e->errorMessage(text); e->Release(); \
  771. throw MakeStringException(TE_FailedToRetrieveWorkunitValue, "result %s in workunit contains an invalid " KINDTEXT " value [%s]", getResultText(s, STEPNAME, SEQUENCE), text.str()); \
  772. } \
  773. catch (CATCHALL) { StringBuffer s; throw MakeStringException(TE_FailedToRetrieveWorkunitValue, "value %s in workunit contains an invalid " KINDTEXT " value", getResultText(s, STEPNAME, SEQUENCE)); }
  774. public:
  775. CThorCodeContextMaster(CJobChannel &jobChannel, IConstWorkUnit &_workunit, ILoadedDllEntry &querySo, IUserDescriptor &userDesc) : CThorCodeContextBase(jobChannel, querySo, userDesc), workunit(&_workunit)
  776. {
  777. }
  778. // ICodeContext
  779. virtual void setResultBool(const char *name, unsigned sequence, bool result)
  780. {
  781. WorkunitUpdate w(&workunit->lock());
  782. Owned<IWUResult> r = updateWorkUnitResult(w, name, sequence);
  783. if (r)
  784. {
  785. r->setResultBool(result);
  786. r->setResultStatus(ResultStatusCalculated);
  787. }
  788. else
  789. throw MakeStringException(TE_UnexpectedParameters, "Unexpected parameters to setResultBool");
  790. }
  791. virtual void setResultData(const char *name, unsigned sequence, int len, const void *result)
  792. {
  793. WorkunitUpdate w(&workunit->lock());
  794. Owned<IWUResult> r = updateWorkUnitResult(w, name, sequence);
  795. if (r)
  796. {
  797. r->setResultData(result, len);
  798. r->setResultStatus(ResultStatusCalculated);
  799. }
  800. else
  801. throw MakeStringException(TE_UnexpectedParameters, "Unexpected parameters to setResultData");
  802. }
  803. virtual void setResultDecimal(const char * name, unsigned sequence, int len, int precision, bool isSigned, const void *val)
  804. {
  805. WorkunitUpdate w(&workunit->lock());
  806. Owned<IWUResult> r = updateWorkUnitResult(w, name, sequence);
  807. if (r)
  808. {
  809. r->setResultDecimal(val, len);
  810. r->setResultStatus(ResultStatusCalculated);
  811. }
  812. else
  813. throw MakeStringException(TE_UnexpectedParameters, "Unexpected parameters to setResultDecimal");
  814. }
  815. virtual void setResultInt(const char *name, unsigned sequence, __int64 result, unsigned size)
  816. {
  817. WorkunitUpdate w(&workunit->lock());
  818. Owned<IWUResult> r = updateWorkUnitResult(w, name, sequence);
  819. if (r)
  820. {
  821. r->setResultInt(result);
  822. r->setResultStatus(ResultStatusCalculated);
  823. }
  824. else
  825. throw MakeStringException(TE_UnexpectedParameters, "Unexpected parameters to setResultInt");
  826. }
  827. virtual void setResultRaw(const char *name, unsigned sequence, int len, const void *result)
  828. {
  829. WorkunitUpdate w(&workunit->lock());
  830. Owned<IWUResult> r = updateWorkUnitResult(w, name, sequence);
  831. if (r)
  832. {
  833. r->setResultRaw(len, result, ResultFormatRaw);
  834. r->setResultStatus(ResultStatusCalculated);
  835. }
  836. else
  837. throw MakeStringException(TE_UnexpectedParameters, "Unexpected parameters to setResultData");
  838. }
  839. virtual void setResultReal(const char *name, unsigned sequence, double result)
  840. {
  841. WorkunitUpdate w(&workunit->lock());
  842. Owned<IWUResult> r = updateWorkUnitResult(w, name, sequence);
  843. if (r)
  844. {
  845. r->setResultReal(result);
  846. r->setResultStatus(ResultStatusCalculated);
  847. }
  848. else
  849. throw MakeStringException(TE_UnexpectedParameters, "Unexpected parameters to setResultReal");
  850. }
  851. virtual void setResultSet(const char *name, unsigned sequence, bool isAll, size32_t len, const void *result, ISetToXmlTransformer *)
  852. {
  853. WorkunitUpdate w(&workunit->lock());
  854. Owned<IWUResult> r = updateWorkUnitResult(w, name, sequence);
  855. if (r)
  856. {
  857. r->setResultIsAll(isAll);
  858. r->setResultRaw(len, result, ResultFormatRaw);
  859. r->setResultStatus(ResultStatusCalculated);
  860. }
  861. else
  862. throw MakeStringException(TE_UnexpectedParameters, "Unexpected parameters to setResultData");
  863. }
  864. virtual void setResultString(const char *name, unsigned sequence, int len, const char *result)
  865. {
  866. WorkunitUpdate w(&workunit->lock());
  867. Owned<IWUResult> r = updateWorkUnitResult(w, name, sequence);
  868. if (r)
  869. {
  870. r->setResultString(result, len);
  871. r->setResultStatus(ResultStatusCalculated);
  872. }
  873. else
  874. throw MakeStringException(TE_UnexpectedParameters, "Unexpected parameters to setResultString");
  875. }
  876. virtual void setResultUnicode(const char * name, unsigned sequence, int len, UChar const * result)
  877. {
  878. WorkunitUpdate w(&workunit->lock());
  879. Owned<IWUResult> r = updateWorkUnitResult(w, name, sequence);
  880. if (r)
  881. {
  882. r->setResultUnicode(result, len);
  883. r->setResultStatus(ResultStatusCalculated);
  884. }
  885. else
  886. throw MakeStringException(TE_UnexpectedParameters, "Unexpected parameters to setResultUnicode");
  887. }
  888. virtual void setResultVarString(const char * name, unsigned sequence, const char *result)
  889. {
  890. WorkunitUpdate w(&workunit->lock());
  891. Owned<IWUResult> r = updateWorkUnitResult(w, name, sequence);
  892. if (r)
  893. {
  894. r->setResultString(result, strlen(result));
  895. r->setResultStatus(ResultStatusCalculated);
  896. }
  897. else
  898. throw MakeStringException(TE_UnexpectedParameters, "Unexpected parameters to setResultVarString");
  899. }
  900. virtual void setResultUInt(const char *name, unsigned sequence, unsigned __int64 result, unsigned size)
  901. {
  902. WorkunitUpdate w(&workunit->lock());
  903. Owned<IWUResult> r = updateWorkUnitResult(w, name, sequence);
  904. if (r)
  905. {
  906. r->setResultUInt(result);
  907. r->setResultStatus(ResultStatusCalculated);
  908. }
  909. else
  910. throw MakeStringException(TE_UnexpectedParameters, "Unexpected parameters to setResultUInt");
  911. }
  912. virtual void setResultVarUnicode(const char * stepname, unsigned sequence, UChar const *val)
  913. {
  914. setResultUnicode(stepname, sequence, rtlUnicodeStrlen(val), val);
  915. }
  916. virtual bool getResultBool(const char * stepname, unsigned sequence)
  917. {
  918. PROTECTED_GETRESULT(stepname, sequence, "Bool", "bool",
  919. return r->getResultBool();
  920. );
  921. }
  922. virtual void getResultData(unsigned & tlen, void * & tgt, const char * stepname, unsigned sequence)
  923. {
  924. PROTECTED_GETRESULT(stepname, sequence, "Data", "data",
  925. SCMStringBuffer result;
  926. r->getResultString(result, false);
  927. tlen = result.length();
  928. tgt = (char *)result.s.detach();
  929. );
  930. }
  931. virtual void getResultDecimal(unsigned tlen, int precision, bool isSigned, void * tgt, const char * stepname, unsigned sequence)
  932. {
  933. PROTECTED_GETRESULT(stepname, sequence, "Decimal", "decimal",
  934. r->getResultDecimal(tgt, tlen, precision, isSigned);
  935. );
  936. }
  937. virtual void getResultRaw(unsigned & tlen, void * & tgt, const char * stepname, unsigned sequence, IXmlToRowTransformer * xmlTransformer, ICsvToRowTransformer * csvTransformer)
  938. {
  939. tgt = NULL;
  940. PROTECTED_GETRESULT(stepname, sequence, "Raw", "raw",
  941. Variable2IDataVal result(&tlen, &tgt);
  942. Owned<IXmlToRawTransformer> rawXmlTransformer = createXmlRawTransformer(xmlTransformer);
  943. Owned<ICsvToRawTransformer> rawCsvTransformer = createCsvRawTransformer(csvTransformer);
  944. r->getResultRaw(result, rawXmlTransformer, rawCsvTransformer);
  945. );
  946. }
  947. virtual void getResultSet(bool & isAll, unsigned & tlen, void * & tgt, const char * stepname, unsigned sequence, IXmlToRowTransformer * xmlTransformer, ICsvToRowTransformer * csvTransformer)
  948. {
  949. tgt = NULL;
  950. PROTECTED_GETRESULT(stepname, sequence, "Raw", "raw",
  951. Variable2IDataVal result(&tlen, &tgt);
  952. Owned<IXmlToRawTransformer> rawXmlTransformer = createXmlRawTransformer(xmlTransformer);
  953. Owned<ICsvToRawTransformer> rawCsvTransformer = createCsvRawTransformer(csvTransformer);
  954. isAll = r->getResultIsAll();
  955. r->getResultRaw(result, rawXmlTransformer, rawCsvTransformer);
  956. );
  957. }
  958. virtual __int64 getResultInt(const char * name, unsigned sequence)
  959. {
  960. PROTECTED_GETRESULT(name, sequence, "Int", "integer",
  961. return r->getResultInt();
  962. );
  963. }
  964. virtual double getResultReal(const char * name, unsigned sequence)
  965. {
  966. PROTECTED_GETRESULT(name, sequence, "Real", "real",
  967. return r->getResultReal();
  968. );
  969. }
  970. virtual void getResultString(unsigned & tlen, char * & tgt, const char * stepname, unsigned sequence)
  971. {
  972. PROTECTED_GETRESULT(stepname, sequence, "String", "string",
  973. SCMStringBuffer result;
  974. r->getResultString(result, false);
  975. tlen = result.length();
  976. tgt = (char *)result.s.detach();
  977. );
  978. }
  979. virtual void getResultStringF(unsigned tlen, char * tgt, const char * stepname, unsigned sequence)
  980. {
  981. PROTECTED_GETRESULT(stepname, sequence, "String", "string",
  982. SCMStringBuffer result;
  983. r->getResultString(result, false);
  984. rtlStrToStr(tlen, tgt, result.length(), result.s.str());
  985. );
  986. }
  987. virtual void getResultUnicode(unsigned & tlen, UChar * & tgt, const char * stepname, unsigned sequence)
  988. {
  989. PROTECTED_GETRESULT(stepname, sequence, "Unicode", "unicode",
  990. MemoryBuffer result;
  991. r->getResultUnicode(MemoryBuffer2IDataVal(result));
  992. tlen = result.length()/2;
  993. tgt = (UChar *)malloc(tlen*2);
  994. memcpy(tgt, result.toByteArray(), tlen*2);
  995. );
  996. }
  997. virtual char * getResultVarString(const char * stepname, unsigned sequence)
  998. {
  999. PROTECTED_GETRESULT(stepname, sequence, "VarString", "string",
  1000. SCMStringBuffer result;
  1001. r->getResultString(result, false);
  1002. return result.s.detach();
  1003. );
  1004. }
  1005. virtual UChar * getResultVarUnicode(const char * stepname, unsigned sequence)
  1006. {
  1007. PROTECTED_GETRESULT(stepname, sequence, "VarUnicode", "unicode",
  1008. MemoryBuffer result;
  1009. r->getResultUnicode(MemoryBuffer2IDataVal(result));
  1010. unsigned tlen = result.length()/2;
  1011. result.append((UChar)0);
  1012. return (UChar *)result.detach();
  1013. );
  1014. }
  1015. virtual unsigned getResultHash(const char * name, unsigned sequence)
  1016. {
  1017. PROTECTED_GETRESULT(name, sequence, "Hash", "hash",
  1018. return r->getResultHash();
  1019. );
  1020. }
  1021. virtual unsigned getExternalResultHash(const char * wuid, const char * stepname, unsigned sequence)
  1022. {
  1023. try
  1024. {
  1025. LOG(MCdebugProgress, unknownJob, "getExternalResultRaw %s", stepname);
  1026. Owned<IConstWUResult> r = getExternalResult(wuid, stepname, sequence);
  1027. return r->getResultHash();
  1028. }
  1029. catch (CATCHALL)
  1030. {
  1031. throw MakeStringException(TE_FailedToRetrieveWorkunitValue, "Failed to retrieve external data hash %s from workunit %s", stepname, wuid);
  1032. }
  1033. }
  1034. virtual void getResultRowset(size32_t & tcount, byte * * & tgt, const char * stepname, unsigned sequence, IEngineRowAllocator * _rowAllocator, bool isGrouped, IXmlToRowTransformer * xmlTransformer, ICsvToRowTransformer * csvTransformer)
  1035. {
  1036. tgt = NULL;
  1037. PROTECTED_GETRESULT(stepname, sequence, "Rowset", "rowset",
  1038. MemoryBuffer datasetBuffer;
  1039. MemoryBuffer2IDataVal result(datasetBuffer);
  1040. Owned<IXmlToRawTransformer> rawXmlTransformer = createXmlRawTransformer(xmlTransformer);
  1041. Owned<ICsvToRawTransformer> rawCsvTransformer = createCsvRawTransformer(csvTransformer);
  1042. r->getResultRaw(result, rawXmlTransformer, rawCsvTransformer);
  1043. Owned<IOutputRowDeserializer> deserializer = _rowAllocator->createDiskDeserializer(this);
  1044. rtlDataset2RowsetX(tcount, tgt, _rowAllocator, deserializer, datasetBuffer.length(), datasetBuffer.toByteArray(), isGrouped);
  1045. );
  1046. }
  1047. virtual void getResultDictionary(size32_t & tcount, byte * * & tgt, IEngineRowAllocator * _rowAllocator, const char * stepname, unsigned sequence, IXmlToRowTransformer * xmlTransformer, ICsvToRowTransformer * csvTransformer, IHThorHashLookupInfo * hasher)
  1048. {
  1049. tcount = 0;
  1050. tgt = NULL;
  1051. PROTECTED_GETRESULT(stepname, sequence, "Dictionary", "dictionary",
  1052. MemoryBuffer datasetBuffer;
  1053. MemoryBuffer2IDataVal result(datasetBuffer);
  1054. Owned<IXmlToRawTransformer> rawXmlTransformer = createXmlRawTransformer(xmlTransformer);
  1055. Owned<ICsvToRawTransformer> rawCsvTransformer = createCsvRawTransformer(csvTransformer);
  1056. r->getResultRaw(result, rawXmlTransformer, rawCsvTransformer);
  1057. Owned<IOutputRowDeserializer> deserializer = _rowAllocator->createDiskDeserializer(this);
  1058. rtlDeserializeDictionary(tcount, tgt, _rowAllocator, deserializer, datasetBuffer.length(), datasetBuffer.toByteArray());
  1059. );
  1060. }
  1061. virtual void getExternalResultRaw(unsigned & tlen, void * & tgt, const char * wuid, const char * stepname, unsigned sequence, IXmlToRowTransformer * xmlTransformer, ICsvToRowTransformer * csvTransformer)
  1062. {
  1063. tgt = NULL;
  1064. try
  1065. {
  1066. LOG(MCdebugProgress, unknownJob, "getExternalResultRaw %s", stepname);
  1067. Variable2IDataVal result(&tlen, &tgt);
  1068. Owned<IConstWUResult> r = getExternalResult(wuid, stepname, sequence);
  1069. Owned<IXmlToRawTransformer> rawXmlTransformer = createXmlRawTransformer(xmlTransformer);
  1070. Owned<ICsvToRawTransformer> rawCsvTransformer = createCsvRawTransformer(csvTransformer);
  1071. r->getResultRaw(result, rawXmlTransformer, rawCsvTransformer);
  1072. }
  1073. catch (CATCHALL)
  1074. {
  1075. throw MakeStringException(TE_FailedToRetrieveWorkunitValue, "Failed to retrieve external data value %s from workunit %s", stepname, wuid);
  1076. }
  1077. }
  1078. virtual void addWuException(const char * text, unsigned code, unsigned severity, const char * source)
  1079. {
  1080. DBGLOG("%s", text);
  1081. try
  1082. {
  1083. Owned<IWorkUnit> w = updateWorkUnit();
  1084. Owned<IWUException> we = w->createException();
  1085. we->setSeverity((ErrorSeverity)severity);
  1086. we->setExceptionMessage(text);
  1087. we->setExceptionSource(source);
  1088. if (code)
  1089. we->setExceptionCode(code);
  1090. }
  1091. catch (IException *E)
  1092. {
  1093. StringBuffer m;
  1094. E->errorMessage(m);
  1095. DBGLOG("Unable to record exception in workunit: %s", m.str());
  1096. E->Release();
  1097. }
  1098. catch (...)
  1099. {
  1100. DBGLOG("Unable to record exception in workunit: unknown exception");
  1101. }
  1102. }
  1103. virtual void addWuAssertFailure(unsigned code, const char * text, const char * filename, unsigned lineno, unsigned column, bool isAbort)
  1104. {
  1105. DBGLOG("%s", text);
  1106. try
  1107. {
  1108. Owned<IWorkUnit> w = updateWorkUnit();
  1109. unsigned activity = 0;
  1110. addExceptionToWorkunit(w, SeverityError, "user", code, text, filename, lineno, column, activity);
  1111. }
  1112. catch (IException *E)
  1113. {
  1114. StringBuffer m;
  1115. E->errorMessage(m);
  1116. DBGLOG("Unable to record exception in workunit: %s", m.str());
  1117. E->Release();
  1118. }
  1119. catch (...)
  1120. {
  1121. DBGLOG("Unable to record exception in workunit: unknown exception");
  1122. }
  1123. if (isAbort)
  1124. rtlFailOnAssert(); // minimal implementation
  1125. }
  1126. virtual unsigned __int64 getFileOffset(const char *logicalName) { assertex(false); return 0; }
  1127. virtual unsigned getNodes() { return jobChannel.queryJob().querySlaves(); }
  1128. virtual unsigned getNodeNum() { throw MakeThorException(0, "Unsupported. getNodeNum() called in master"); return (unsigned)-1; }
  1129. virtual char *getFilePart(const char *logicalName, bool create=false) { assertex(false); return NULL; }
  1130. virtual unsigned __int64 getDatasetHash(const char * name, unsigned __int64 hash)
  1131. {
  1132. unsigned checkSum = 0;
  1133. Owned<IDistributedFile> iDfsFile = queryThorFileManager().lookup(jobChannel.queryJob(), name, false, true, false); // NB: do not update accessed
  1134. if (iDfsFile.get())
  1135. {
  1136. if (iDfsFile->getFileCheckSum(checkSum))
  1137. hash ^= checkSum;
  1138. else
  1139. {
  1140. StringBuffer modifiedStr;
  1141. if (iDfsFile->queryAttributes().getProp("@modified", modifiedStr))
  1142. hash = rtlHash64Data(modifiedStr.length(), modifiedStr.str(), hash);
  1143. // JCS->GH - what's the best thing to do here, if [for some reason] neither are available..
  1144. }
  1145. }
  1146. return hash;
  1147. }
  1148. virtual IDistributedFileTransaction *querySuperFileTransaction()
  1149. {
  1150. if (!superfiletransaction.get())
  1151. superfiletransaction.setown(createDistributedFileTransaction(userDesc, this));
  1152. return superfiletransaction.get();
  1153. }
  1154. virtual char *getJobName()
  1155. {
  1156. return strdup(workunit->queryJobName());
  1157. }
  1158. virtual char *getClusterName()
  1159. {
  1160. return strdup(workunit->queryClusterName());
  1161. }
  1162. virtual char *getGroupName()
  1163. {
  1164. StringBuffer out;
  1165. if (globals)
  1166. globals->getProp("@nodeGroup",out);
  1167. return out.detach();
  1168. }
  1169. // ICodeContextExt impl.
  1170. virtual IConstWUResult *getExternalResult(const char * wuid, const char *name, unsigned sequence)
  1171. {
  1172. Owned<IWorkUnitFactory> factory = getWorkUnitFactory();
  1173. Owned<IConstWorkUnit> externalWU = factory->openWorkUnit(wuid);
  1174. externalWU->remoteCheckAccess(userDesc, false);
  1175. return getWorkUnitResult(externalWU, name, sequence);
  1176. }
  1177. virtual IConstWUResult *getResultForGet(const char *name, unsigned sequence)
  1178. {
  1179. Owned<IConstWUResult> r = getResult(name, sequence);
  1180. if (!r || (r->getResultStatus() == ResultStatusUndefined))
  1181. {
  1182. StringBuffer s;
  1183. throw MakeStringException(TE_FailedToRetrieveWorkunitValue, "value %s in workunit is undefined", getResultText(s,name,sequence));
  1184. }
  1185. return r.getClear();
  1186. }
  1187. };
  1188. class CThorCodeContextMasterSharedMem : public CThorCodeContextMaster
  1189. {
  1190. IThorAllocator *sharedAllocator;
  1191. public:
  1192. CThorCodeContextMasterSharedMem(CJobChannel &jobChannel, IThorAllocator *_sharedAllocator, IConstWorkUnit &_workunit, ILoadedDllEntry &querySo, IUserDescriptor &userDesc)
  1193. : CThorCodeContextMaster(jobChannel, _workunit, querySo, userDesc)
  1194. {
  1195. sharedAllocator = _sharedAllocator;
  1196. }
  1197. virtual IEngineRowAllocator *getRowAllocator(IOutputMetaData * meta, unsigned activityId) const
  1198. {
  1199. return sharedAllocator->getRowAllocator(meta, activityId);
  1200. }
  1201. };
  1202. /////////////
  1203. //
  1204. // CJobMaster
  1205. //
  1206. void loadPlugin(SafePluginMap *pluginMap, const char *_path, const char *name)
  1207. {
  1208. StringBuffer path(_path);
  1209. path.append(name);
  1210. pluginMap->addPlugin(path.str(), name);
  1211. }
  1212. CJobMaster::CJobMaster(IConstWorkUnit &_workunit, const char *graphName, ILoadedDllEntry *querySo, bool _sendSo, const SocketEndpoint &_agentEp)
  1213. : CJobBase(querySo, graphName), workunit(&_workunit), sendSo(_sendSo), agentEp(_agentEp)
  1214. {
  1215. SCMStringBuffer _token, _scope;
  1216. workunit->getScope(_scope);
  1217. workunit->getSecurityToken(_token);
  1218. wuid.set(workunit->queryWuid());
  1219. user.set(workunit->queryUser());
  1220. token.append(_token.str());
  1221. scope.append(_scope.str());
  1222. globalMemoryMB = globals->getPropInt("@masterMemorySize", globals->getPropInt("@globalMemorySize")); // in MB
  1223. numChannels = 1;
  1224. init();
  1225. resumed = WUActionResume == workunit->getAction();
  1226. fatalHandler.setown(new CFatalHandler(globals->getPropInt("@fatal_timeout", FATAL_TIMEOUT)));
  1227. querySent = spillsSaved = false;
  1228. nodeDiskUsageCached = false;
  1229. StringBuffer pluginsDir;
  1230. globals->getProp("@pluginsPath", pluginsDir);
  1231. if (pluginsDir.length())
  1232. addPathSepChar(pluginsDir);
  1233. Owned<IConstWUPluginIterator> pluginIter = &workunit->getPlugins();
  1234. ForEach(*pluginIter)
  1235. {
  1236. IConstWUPlugin &plugin = pluginIter->query();
  1237. SCMStringBuffer name;
  1238. plugin.getPluginName(name);
  1239. loadPlugin(pluginMap, pluginsDir.str(), name.str());
  1240. }
  1241. sharedAllocator.setown(::createThorAllocator(globalMemoryMB, 0, 1, memorySpillAtPercentage, *logctx, crcChecking, usePackedAllocator));
  1242. Owned<IMPServer> mpServer = getMPServer();
  1243. addChannel(mpServer);
  1244. mpJobTag = allocateMPTag();
  1245. slavemptag = allocateMPTag();
  1246. slaveMsgHandler = new CSlaveMessageHandler(*this, slavemptag);
  1247. tmpHandler.setown(createTempHandler(true));
  1248. xgmml.set(graphXGMML);
  1249. }
  1250. CJobMaster::~CJobMaster()
  1251. {
  1252. if (slaveMsgHandler)
  1253. delete slaveMsgHandler;
  1254. freeMPTag(mpJobTag);
  1255. freeMPTag(slavemptag);
  1256. tmpHandler.clear();
  1257. }
  1258. void CJobMaster::addChannel(IMPServer *mpServer)
  1259. {
  1260. jobChannels.append(*new CJobMasterChannel(*this, mpServer, jobChannels.ordinality()));
  1261. }
  1262. static IException *createBCastException(unsigned slave, const char *errorMsg)
  1263. {
  1264. // think this should always be fatal, could check link down here, or in general and flag as _shutdown.
  1265. StringBuffer msg("General failure communicating to slave");
  1266. if (slave)
  1267. msg.append("(").append(slave).append(") ");
  1268. else
  1269. msg.append("s ");
  1270. Owned<IThorException> e = MakeThorException(0, "%s", msg.append(" [").append(errorMsg).append("]").str());
  1271. e->setAction(tea_shutdown);
  1272. return e.getClear();
  1273. }
  1274. mptag_t CJobMaster::allocateMPTag()
  1275. {
  1276. mptag_t tag = allocateClusterMPTag();
  1277. queryJobChannel(0).queryJobComm().flush(tag);
  1278. PROGLOG("allocateMPTag: tag = %d", (int)tag);
  1279. return tag;
  1280. }
  1281. void CJobMaster::freeMPTag(mptag_t tag)
  1282. {
  1283. if (TAG_NULL != tag)
  1284. {
  1285. freeClusterMPTag(tag);
  1286. PROGLOG("freeMPTag: tag = %d", (int)tag);
  1287. queryJobChannel(0).queryJobComm().flush(tag);
  1288. }
  1289. }
  1290. void CJobMaster::broadcast(ICommunicator &comm, CMessageBuffer &msg, mptag_t mptag, unsigned timeout, const char *errorMsg, CReplyCancelHandler *msgHandler, bool sendOnly)
  1291. {
  1292. unsigned groupSizeExcludingMaster = comm.queryGroup().ordinality() - 1;
  1293. mptag_t replyTag = TAG_NULL;
  1294. if (!sendOnly)
  1295. {
  1296. replyTag = queryJobChannel(0).queryMPServer().createReplyTag();
  1297. msg.setReplyTag(replyTag);
  1298. }
  1299. if (globals->getPropBool("@broadcastSendAsync", true)) // only here in case of problems/debugging.
  1300. {
  1301. class CSendAsyncfor : public CAsyncFor
  1302. {
  1303. CMessageBuffer &msg;
  1304. mptag_t mptag;
  1305. unsigned timeout;
  1306. StringAttr errorMsg;
  1307. ICommunicator &comm;
  1308. public:
  1309. CSendAsyncfor(ICommunicator &_comm, CMessageBuffer &_msg, mptag_t _mptag, unsigned _timeout, const char *_errorMsg)
  1310. : comm(_comm), msg(_msg), mptag(_mptag), timeout(_timeout), errorMsg(_errorMsg)
  1311. {
  1312. }
  1313. void Do(unsigned i)
  1314. {
  1315. if (!comm.send(msg, i+1, mptag, timeout))
  1316. throw createBCastException(i+1, errorMsg);
  1317. }
  1318. } afor(comm, msg, mptag, timeout, errorMsg);
  1319. try
  1320. {
  1321. afor.For(groupSizeExcludingMaster, groupSizeExcludingMaster);
  1322. }
  1323. catch (IException *e)
  1324. {
  1325. EXCLOG(e, "broadcastSendAsync");
  1326. abort(e);
  1327. throw;
  1328. }
  1329. }
  1330. else if (!comm.send(msg, RANK_ALL_OTHER, mptag, timeout))
  1331. {
  1332. Owned<IException> e = createBCastException(0, errorMsg);
  1333. EXCLOG(e, NULL);
  1334. abort(e);
  1335. throw e.getClear();
  1336. }
  1337. if (sendOnly) return;
  1338. unsigned respondents = 0;
  1339. Owned<IBitSet> bitSet = createThreadSafeBitSet();
  1340. loop
  1341. {
  1342. rank_t sender;
  1343. CMessageBuffer msg;
  1344. bool r = msgHandler ? msgHandler->recv(comm, msg, RANK_ALL, replyTag, &sender, LONGTIMEOUT)
  1345. : comm.recv(msg, RANK_ALL, replyTag, &sender, LONGTIMEOUT);
  1346. if (!r)
  1347. {
  1348. StringBuffer tmpStr;
  1349. if (errorMsg)
  1350. tmpStr.append(": ").append(errorMsg).append(" - ");
  1351. tmpStr.append("Timeout receiving from slaves - no reply from: [");
  1352. unsigned s = bitSet->scan(0, false);
  1353. assertex(s<querySlaves()); // must be at least one
  1354. tmpStr.append(s+1);
  1355. loop
  1356. {
  1357. s = bitSet->scan(s+1, false);
  1358. if (s>=querySlaves())
  1359. break;
  1360. tmpStr.append(",").append(s+1);
  1361. }
  1362. tmpStr.append("]");
  1363. Owned<IException> e = MakeThorFatal(NULL, 0, " %s", tmpStr.str());
  1364. EXCLOG(e, NULL);
  1365. throw e.getClear();
  1366. }
  1367. bool error;
  1368. msg.read(error);
  1369. if (error)
  1370. {
  1371. Owned<IThorException> e = deserializeThorException(msg);
  1372. e->setSlave(sender);
  1373. throw e.getClear();
  1374. }
  1375. ++respondents;
  1376. bitSet->set((unsigned)sender-1);
  1377. if (respondents == groupSizeExcludingMaster)
  1378. break;
  1379. }
  1380. }
  1381. void CJobMaster::initNodeDUCache()
  1382. {
  1383. if (!nodeDiskUsageCached)
  1384. {
  1385. nodeDiskUsageCached = true;
  1386. Owned<IPropertyTreeIterator> fileIter = &workunit->getFileIterator();
  1387. ForEach (*fileIter)
  1388. {
  1389. Owned<IDistributedFile> f = queryDistributedFileDirectory().lookup(fileIter->query().queryProp("@name"), userDesc);
  1390. if (f)
  1391. {
  1392. unsigned n = f->numParts();
  1393. for (unsigned i=0;i<n;i++)
  1394. {
  1395. Owned<IDistributedFilePart> part = f->getPart(i);
  1396. offset_t sz = part->getFileSize(false, false);
  1397. if (i>=nodeDiskUsage.ordinality())
  1398. nodeDiskUsage.append(sz);
  1399. else
  1400. {
  1401. sz += nodeDiskUsage.item(i);
  1402. nodeDiskUsage.add(sz, i);
  1403. }
  1404. }
  1405. }
  1406. }
  1407. }
  1408. }
  1409. IPropertyTree *CJobMaster::prepareWorkUnitInfo()
  1410. {
  1411. Owned<IPropertyTree> workUnitInfo = createPTree("workUnitInfo");
  1412. workUnitInfo->setProp("wuid", wuid);
  1413. workUnitInfo->setProp("user", user);
  1414. workUnitInfo->setProp("token", token);
  1415. workUnitInfo->setProp("scope", scope);
  1416. Owned<IConstWUPluginIterator> pluginIter = &queryWorkUnit().getPlugins();
  1417. IPropertyTree *plugins = NULL;
  1418. ForEach(*pluginIter)
  1419. {
  1420. IConstWUPlugin &thisplugin = pluginIter->query();
  1421. if (!plugins)
  1422. plugins = workUnitInfo->addPropTree("plugins", createPTree());
  1423. SCMStringBuffer name;
  1424. thisplugin.getPluginName(name);
  1425. IPropertyTree *plugin = plugins->addPropTree("plugin", createPTree());
  1426. plugin->setProp("@name", name.str());
  1427. }
  1428. IPropertyTree *debug = workUnitInfo->addPropTree("Debug", createPTree(ipt_caseInsensitive));
  1429. SCMStringBuffer debugStr, valueStr;
  1430. Owned<IStringIterator> debugIter = &queryWorkUnit().getDebugValues();
  1431. ForEach (*debugIter)
  1432. {
  1433. debugIter->str(debugStr);
  1434. queryWorkUnit().getDebugValue(debugStr.str(), valueStr);
  1435. debug->setProp(debugStr.str(), valueStr.str());
  1436. }
  1437. return workUnitInfo.getClear();
  1438. }
  1439. void CJobMaster::sendQuery()
  1440. {
  1441. CriticalBlock b(sendQueryCrit);
  1442. if (querySent) return;
  1443. CMessageBuffer tmp;
  1444. tmp.append(mpJobTag);
  1445. tmp.append(slavemptag);
  1446. tmp.append(queryWuid());
  1447. tmp.append(graphName);
  1448. const char *soName = queryDllEntry().queryName();
  1449. PROGLOG("Query dll: %s", soName);
  1450. tmp.append(soName);
  1451. tmp.append(sendSo);
  1452. if (sendSo)
  1453. {
  1454. CTimeMon atimer;
  1455. OwnedIFile iFile = createIFile(soName);
  1456. OwnedIFileIO iFileIO = iFile->open(IFOread);
  1457. size32_t sz = (size32_t)iFileIO->size();
  1458. tmp.append(sz);
  1459. read(iFileIO, 0, sz, tmp);
  1460. PROGLOG("Loading query for serialization to slaves took %d ms", atimer.elapsed());
  1461. }
  1462. Owned<IPropertyTree> deps = createPTree(queryXGMML()->queryName());
  1463. Owned<IPropertyTreeIterator> edgeIter = queryXGMML()->getElements("edge"); // JCSMORE trim to those actually needed
  1464. ForEach (*edgeIter)
  1465. {
  1466. IPropertyTree &edge = edgeIter->query();
  1467. deps->addPropTree("edge", LINK(&edge));
  1468. }
  1469. Owned<IPropertyTree> workUnitInfo = prepareWorkUnitInfo();
  1470. workUnitInfo->serialize(tmp);
  1471. deps->serialize(tmp);
  1472. CMessageBuffer msg;
  1473. msg.append(QueryInit);
  1474. compressToBuffer(msg, tmp.length(), tmp.toByteArray());
  1475. CTimeMon queryToSlavesTimer;
  1476. broadcast(queryNodeComm(), msg, masterSlaveMpTag, LONGTIMEOUT, "sendQuery");
  1477. PROGLOG("Serialization of query init info (%d bytes) to slaves took %d ms", msg.length(), queryToSlavesTimer.elapsed());
  1478. queryJobManager().addCachedSo(soName);
  1479. querySent = true;
  1480. }
  1481. void CJobMaster::jobDone()
  1482. {
  1483. if (!querySent) return;
  1484. CMessageBuffer msg;
  1485. msg.append(QueryDone);
  1486. msg.append(queryKey());
  1487. broadcast(queryNodeComm(), msg, masterSlaveMpTag, LONGTIMEOUT, "jobDone");
  1488. }
  1489. void CJobMaster::saveSpills()
  1490. {
  1491. CriticalBlock b(spillCrit);
  1492. if (spillsSaved)
  1493. return;
  1494. spillsSaved = true;
  1495. PROGLOG("Paused, saving spills..");
  1496. assertex(!queryUseCheckpoints()); // JCSMORE - checkpoints probably need revisiting
  1497. unsigned numSavedSpills = 0;
  1498. // stash away spills ready for resume, make them owned by workunit in event of abort/delete
  1499. Owned<IFileUsageIterator> iter = queryTempHandler()->getIterator();
  1500. ForEach(*iter)
  1501. {
  1502. CFileUsageEntry &entry = iter->query();
  1503. StringAttr tmpName = entry.queryName();
  1504. if (WUGraphComplete == workunit->queryNodeState(queryGraphName(), entry.queryGraphId()))
  1505. {
  1506. IArrayOf<IGroup> groups;
  1507. StringArray clusters;
  1508. fillClusterArray(*this, tmpName, clusters, groups);
  1509. Owned<IFileDescriptor> fileDesc = queryThorFileManager().create(*this, tmpName, clusters, groups, true, TDXtemporary|TDWnoreplicate);
  1510. fileDesc->queryProperties().setPropBool("@pausefile", true); // JCSMORE - mark to keep, may be able to distinguish via other means
  1511. fileDesc->queryProperties().setProp("@kind", "flat");
  1512. IPropertyTree &props = fileDesc->queryProperties();
  1513. props.setPropBool("@owned", true);
  1514. bool blockCompressed=true; // JCSMORE, should come from helper really
  1515. if (blockCompressed)
  1516. props.setPropBool("@blockCompressed", true);
  1517. Owned<IDistributedFile> file = queryDistributedFileDirectory().createNew(fileDesc);
  1518. // NB: This is renaming/moving from temp path
  1519. StringBuffer newName;
  1520. queryThorFileManager().addScope(*this, tmpName, newName, true, true);
  1521. verifyex(file->renamePhysicalPartFiles(newName.str(), NULL, NULL, queryBaseDirectory(grp_unknown)));
  1522. file->attach(newName,userDesc);
  1523. Owned<IWorkUnit> wu = &queryWorkUnit().lock();
  1524. wu->addFile(newName, &clusters, entry.queryUsage(), entry.queryKind(), queryGraphName());
  1525. ++numSavedSpills;
  1526. }
  1527. }
  1528. PROGLOG("Paused, %d spill(s) saved.", numSavedSpills);
  1529. }
  1530. bool CJobMaster::go()
  1531. {
  1532. class CWorkunitPauseHandler : public CInterface, implements IWorkUnitSubscriber
  1533. {
  1534. CJobMaster &job;
  1535. IConstWorkUnit &wu;
  1536. Owned<IWorkUnitWatcher> watcher;
  1537. CriticalSection crit;
  1538. public:
  1539. IMPLEMENT_IINTERFACE;
  1540. CWorkunitPauseHandler(CJobMaster &_job, IConstWorkUnit &_wu) : job(_job), wu(_wu)
  1541. {
  1542. Owned<IWorkUnitFactory> factory = getWorkUnitFactory();
  1543. watcher.setown(factory->getWatcher(this, (WUSubscribeOptions) (SubscribeOptionAction | SubscribeOptionAbort), wu.queryWuid()));
  1544. }
  1545. ~CWorkunitPauseHandler() { stop(); }
  1546. void stop()
  1547. {
  1548. CriticalBlock b(crit);
  1549. if (watcher)
  1550. {
  1551. watcher->unsubscribe();
  1552. watcher.clear();
  1553. }
  1554. }
  1555. void notify(WUSubscribeOptions flags)
  1556. {
  1557. CriticalBlock b(crit);
  1558. if (!watcher)
  1559. return;
  1560. if (flags & SubscribeOptionAbort)
  1561. {
  1562. Owned<IWorkUnitFactory> factory = getWorkUnitFactory();
  1563. if (factory->isAborting(wu.queryWuid()))
  1564. {
  1565. LOG(MCwarning, thorJob, "ABORT detected from user");
  1566. Owned <IException> e = MakeThorException(TE_WorkUnitAborting, "User signalled abort");
  1567. job.fireException(e);
  1568. }
  1569. else
  1570. PROGLOG("CWorkunitPauseHandler [SubscribeOptionAbort] notifier called, workunit was not aborting");
  1571. }
  1572. if (flags & SubscribeOptionAction)
  1573. {
  1574. job.markWuDirty();
  1575. bool abort = false;
  1576. bool pause = false;
  1577. wu.forceReload();
  1578. WUAction action = wu.getAction();
  1579. if (action==WUActionPause)
  1580. {
  1581. // pause after current subgraph
  1582. pause = true;
  1583. }
  1584. else if (action==WUActionPauseNow)
  1585. {
  1586. // abort current subgraph
  1587. abort = true;
  1588. pause = true;
  1589. }
  1590. if (pause)
  1591. {
  1592. PROGLOG("Pausing job%s", abort?" [now]":"");
  1593. job.pause(abort);
  1594. }
  1595. }
  1596. }
  1597. } workunitPauseHandler(*this, *workunit);
  1598. class CQueryTimeoutHandler : public CTimeoutTrigger
  1599. {
  1600. CJobMaster &job;
  1601. public:
  1602. CQueryTimeoutHandler(CJobMaster &_job, unsigned timeout) : CTimeoutTrigger(timeout, "QUERY"), job(_job)
  1603. {
  1604. inform(MakeThorException(TE_QueryTimeoutError, "Query took greater than %d seconds", timeout));
  1605. }
  1606. virtual bool action()
  1607. {
  1608. job.fireException(exception);
  1609. return true;
  1610. }
  1611. private:
  1612. graph_id graphId;
  1613. };
  1614. Owned<CTimeoutTrigger> qtHandler;
  1615. int guillotineTimeout = workunit->getDebugValueInt("maxRunTime", 0);
  1616. if (guillotineTimeout > 0)
  1617. qtHandler.setown(new CQueryTimeoutHandler(*this, guillotineTimeout));
  1618. else if (guillotineTimeout < 0)
  1619. {
  1620. Owned<IException> e = MakeStringException(0, "Ignoring negative maxRunTime: %d", guillotineTimeout);
  1621. reportExceptionToWorkunit(*workunit, e);
  1622. }
  1623. if (WUActionPause == workunit->getAction() || WUActionPauseNow == workunit->getAction())
  1624. throw MakeStringException(0, "Job paused at start, exiting");
  1625. bool allDone = true;
  1626. unsigned concurrentSubGraphs = (unsigned)getWorkUnitValueInt("concurrentSubGraphs", globals->getPropInt("@concurrentSubGraphs", 1));
  1627. try
  1628. {
  1629. startJob();
  1630. workunit->setGraphState(queryGraphName(), WUGraphRunning);
  1631. Owned<IThorGraphIterator> iter = queryJobChannel(0).getSubGraphs();
  1632. CICopyArrayOf<CMasterGraph> toRun;
  1633. ForEach(*iter)
  1634. {
  1635. CMasterGraph &graph = (CMasterGraph &)iter->query();
  1636. if ((queryResumed() || queryUseCheckpoints()) && WUGraphComplete == workunit->queryNodeState(queryGraphName(), graph.queryGraphId()))
  1637. graph.setCompleteEx();
  1638. else
  1639. toRun.append(graph);
  1640. }
  1641. ForEachItemInRev(g, toRun)
  1642. {
  1643. if (aborted) break;
  1644. CMasterGraph &graph = toRun.item(g);
  1645. if (graph.isSink())
  1646. graph.execute(0, NULL, true, concurrentSubGraphs>1);
  1647. if (queryPausing()) break;
  1648. }
  1649. queryJobChannel(0).wait();
  1650. workunitPauseHandler.stop();
  1651. ForEachItemIn(tr, toRun)
  1652. {
  1653. CMasterGraph &graph = toRun.item(tr);
  1654. if (!graph.isComplete())
  1655. {
  1656. allDone = false;
  1657. break;
  1658. }
  1659. }
  1660. }
  1661. catch (IException *e) { fireException(e); e->Release(); }
  1662. catch (CATCHALL) { Owned<IException> e = MakeThorException(0, "Unknown exception running sub graphs"); fireException(e); }
  1663. workunit->setGraphState(queryGraphName(), aborted?WUGraphFailed:(allDone?WUGraphComplete:(pausing?WUGraphPaused:WUGraphComplete)));
  1664. if (queryPausing())
  1665. saveSpills();
  1666. Owned<IException> jobDoneException;
  1667. try { jobDone(); }
  1668. catch (IException *e)
  1669. {
  1670. EXCLOG(e, NULL);
  1671. jobDoneException.setown(e);
  1672. }
  1673. fatalHandler->clear();
  1674. queryTempHandler()->clearTemps();
  1675. slaveMsgHandler->stop();
  1676. if (jobDoneException.get())
  1677. throw LINK(jobDoneException);
  1678. return allDone;
  1679. }
  1680. void CJobMaster::pause(bool doAbort)
  1681. {
  1682. pausing = true;
  1683. if (doAbort)
  1684. {
  1685. // reply will trigger DAMP_THOR_REPLY_PAUSED to agent, unless all graphs are already complete.
  1686. queryJobManager().replyException(*this, NULL);
  1687. // abort current graph asynchronously.
  1688. // After spill files have been saved, trigger timeout handler in case abort doesn't succeed.
  1689. Owned<IException> e = MakeThorException(0, "Unable to recover from pausenow");
  1690. class CAbortThread : implements IThreaded
  1691. {
  1692. CJobMaster &owner;
  1693. CThreaded threaded;
  1694. Linked<IException> exception;
  1695. public:
  1696. CAbortThread(CJobMaster &_owner, IException *_exception) : owner(_owner), exception(_exception), threaded("SaveSpillThread", this)
  1697. {
  1698. threaded.start();
  1699. }
  1700. ~CAbortThread()
  1701. {
  1702. threaded.join();
  1703. }
  1704. // IThreaded
  1705. virtual void main()
  1706. {
  1707. owner.abort(exception);
  1708. }
  1709. } abortThread(*this, e);
  1710. saveSpills();
  1711. fatalHandler->inform(e.getClear());
  1712. }
  1713. }
  1714. __int64 CJobMaster::queryNodeDiskUsage(unsigned node)
  1715. {
  1716. initNodeDUCache();
  1717. if (!nodeDiskUsage.isItem(node)) return 0;
  1718. return nodeDiskUsage.item(node);
  1719. }
  1720. void CJobMaster::setNodeDiskUsage(unsigned node, __int64 sz)
  1721. {
  1722. initNodeDUCache();
  1723. while (nodeDiskUsage.ordinality() <= node)
  1724. nodeDiskUsage.append(0);
  1725. nodeDiskUsage.replace(sz, node);
  1726. }
  1727. __int64 CJobMaster::addNodeDiskUsage(unsigned node, __int64 sz)
  1728. {
  1729. sz += queryNodeDiskUsage(node);
  1730. setNodeDiskUsage(node, sz);
  1731. return sz;
  1732. }
  1733. bool CJobMaster::queryCreatedFile(const char *file)
  1734. {
  1735. StringBuffer scopedName;
  1736. queryThorFileManager().addScope(*this, file, scopedName, false);
  1737. return (NotFound != createdFiles.find(scopedName.str()));
  1738. }
  1739. void CJobMaster::addCreatedFile(const char *file)
  1740. {
  1741. StringBuffer scopedName;
  1742. queryThorFileManager().addScope(*this, file, scopedName, false);
  1743. createdFiles.append(scopedName.str());
  1744. }
  1745. __int64 CJobMaster::getWorkUnitValueInt(const char *prop, __int64 defVal) const
  1746. {
  1747. return queryWorkUnit().getDebugValueInt64(prop, defVal);
  1748. }
  1749. bool CJobMaster::getWorkUnitValueBool(const char *prop, bool defVal) const
  1750. {
  1751. return queryWorkUnit().getDebugValueBool(prop, defVal);
  1752. }
  1753. StringBuffer &CJobMaster::getWorkUnitValue(const char *prop, StringBuffer &str) const
  1754. {
  1755. SCMStringBuffer scmStr;
  1756. queryWorkUnit().getDebugValue(prop, scmStr);
  1757. return str.append(scmStr.str());
  1758. }
  1759. IGraphTempHandler *CJobMaster::createTempHandler(bool errorOnMissing)
  1760. {
  1761. return new CMasterGraphTempHandler(*this, errorOnMissing);
  1762. }
  1763. bool CJobMaster::fireException(IException *e)
  1764. {
  1765. IThorException *te = QUERYINTERFACE(e, IThorException);
  1766. ThorExceptionAction action;
  1767. if (!te) action = tea_null;
  1768. else action = te->queryAction();
  1769. if (QUERYINTERFACE(e, IMP_Exception) && MPERR_link_closed==e->errorCode())
  1770. action = tea_shutdown;
  1771. else if (QUERYINTERFACE(e, ISEH_Exception))
  1772. action = tea_shutdown;
  1773. CriticalBlock b(exceptCrit);
  1774. switch (action)
  1775. {
  1776. case tea_warning:
  1777. {
  1778. LOG(MCwarning, thorJob, e);
  1779. ErrorSeverity mappedSeverity = workunit->getWarningSeverity(e->errorCode(), SeverityWarning);
  1780. if (mappedSeverity != SeverityIgnore)
  1781. reportExceptionToWorkunit(*workunit, e);
  1782. break;
  1783. }
  1784. default:
  1785. {
  1786. LOG(MCerror, thorJob, e);
  1787. queryJobManager().replyException(*this, e);
  1788. fatalHandler->inform(LINK(e));
  1789. try { abort(e); }
  1790. catch (IException *e)
  1791. {
  1792. Owned<IThorException> te = ThorWrapException(e, "Error aborting job, will cause thor restart");
  1793. e->Release();
  1794. reportExceptionToWorkunit(*workunit, te);
  1795. action = tea_shutdown;
  1796. }
  1797. if (tea_shutdown == action)
  1798. queryJobManager().stop();
  1799. }
  1800. }
  1801. return true;
  1802. }
  1803. // CJobMasterChannel
  1804. CJobMasterChannel::CJobMasterChannel(CJobBase &job, IMPServer *mpServer, unsigned channel) : CJobChannel(job, mpServer, channel)
  1805. {
  1806. codeCtx.setown(new CThorCodeContextMaster(*this, job.queryWorkUnit(), job.queryDllEntry(), *job.queryUserDescriptor()));
  1807. sharedMemCodeCtx.setown(new CThorCodeContextMasterSharedMem(*this, job.querySharedAllocator(), job.queryWorkUnit(), job.queryDllEntry(), *job.queryUserDescriptor()));
  1808. }
  1809. CGraphBase *CJobMasterChannel::createGraph()
  1810. {
  1811. return new CMasterGraph(*this);
  1812. }
  1813. IBarrier *CJobMasterChannel::createBarrier(mptag_t tag)
  1814. {
  1815. return new CBarrierMaster(*jobComm, tag);
  1816. }
  1817. ///////////////////
  1818. class CCollatedResult : implements IThorResult, public CSimpleInterface
  1819. {
  1820. CMasterGraph &graph;
  1821. CActivityBase &activity;
  1822. IThorRowInterfaces *rowIf;
  1823. unsigned id;
  1824. CriticalSection crit;
  1825. PointerArrayOf<CThorExpandingRowArray> results;
  1826. Owned<IThorResult> result;
  1827. unsigned spillPriority;
  1828. activity_id ownerId;
  1829. void ensure()
  1830. {
  1831. CriticalBlock b(crit);
  1832. if (result)
  1833. return;
  1834. mptag_t replyTag = graph.queryMPServer().createReplyTag();
  1835. CMessageBuffer msg;
  1836. msg.append(GraphGetResult);
  1837. msg.append(activity.queryJob().queryKey());
  1838. msg.append(graph.queryGraphId());
  1839. msg.append(ownerId);
  1840. msg.append(id);
  1841. msg.append(replyTag);
  1842. ((CJobMaster &)graph.queryJob()).broadcast(queryNodeComm(), msg, masterSlaveMpTag, LONGTIMEOUT, "CCollectResult", NULL, true);
  1843. unsigned numSlaves = graph.queryJob().querySlaves();
  1844. for (unsigned n=0; n<numSlaves; n++)
  1845. results.item(n)->kill();
  1846. rank_t sender;
  1847. MemoryBuffer mb;
  1848. Owned<ISerialStream> stream = createMemoryBufferSerialStream(mb);
  1849. CThorStreamDeserializerSource rowSource(stream);
  1850. unsigned todo = numSlaves;
  1851. loop
  1852. {
  1853. loop
  1854. {
  1855. if (activity.queryAbortSoon())
  1856. return;
  1857. msg.clear();
  1858. if (activity.receiveMsg(msg, RANK_ALL, replyTag, &sender, 60*1000))
  1859. break;
  1860. ActPrintLog(&activity, "WARNING: tag %d timedout, retrying", (unsigned)replyTag);
  1861. }
  1862. sender = sender - 1; // 0 = master
  1863. if (!msg.length())
  1864. {
  1865. --todo;
  1866. if (0 == todo)
  1867. break; // done
  1868. }
  1869. else
  1870. {
  1871. bool error;
  1872. msg.read(error);
  1873. if (error)
  1874. {
  1875. Owned<IThorException> e = deserializeThorException(msg);
  1876. e->setSlave(sender);
  1877. throw e.getClear();
  1878. }
  1879. ThorExpand(msg, mb.clear());
  1880. CThorExpandingRowArray *slaveResults = results.item(sender);
  1881. while (!rowSource.eos())
  1882. {
  1883. RtlDynamicRowBuilder rowBuilder(rowIf->queryRowAllocator());
  1884. size32_t sz = rowIf->queryRowDeserializer()->deserialize(rowBuilder, rowSource);
  1885. slaveResults->append(rowBuilder.finalizeRowClear(sz));
  1886. }
  1887. }
  1888. }
  1889. Owned<IThorResult> _result = ::createResult(activity, rowIf, false, spillPriority);
  1890. Owned<IRowWriter> resultWriter = _result->getWriter();
  1891. for (unsigned s=0; s<numSlaves; s++)
  1892. {
  1893. CThorExpandingRowArray *slaveResult = results.item(s);
  1894. ForEachItemIn(r, *slaveResult)
  1895. {
  1896. const void *row = slaveResult->query(r);
  1897. LinkThorRow(row);
  1898. resultWriter->putRow(row);
  1899. }
  1900. }
  1901. result.setown(_result.getClear());
  1902. }
  1903. public:
  1904. IMPLEMENT_IINTERFACE_USING(CSimpleInterface);
  1905. CCollatedResult(CMasterGraph &_graph, CActivityBase &_activity, IThorRowInterfaces *_rowIf, unsigned _id, activity_id _ownerId, unsigned _spillPriority)
  1906. : graph(_graph), activity(_activity), rowIf(_rowIf), id(_id), ownerId(_ownerId), spillPriority(_spillPriority)
  1907. {
  1908. for (unsigned n=0; n<graph.queryJob().querySlaves(); n++)
  1909. results.append(new CThorExpandingRowArray(activity, rowIf));
  1910. }
  1911. ~CCollatedResult()
  1912. {
  1913. ForEachItemIn(l, results)
  1914. {
  1915. CThorExpandingRowArray *result = results.item(l);
  1916. delete result;
  1917. }
  1918. }
  1919. void setId(unsigned _id)
  1920. {
  1921. id = _id;
  1922. }
  1923. // IThorResult
  1924. virtual IRowWriter *getWriter() { throwUnexpected(); }
  1925. virtual void setResultStream(IRowWriterMultiReader *stream, rowcount_t count)
  1926. {
  1927. throwUnexpected();
  1928. }
  1929. virtual IRowStream *getRowStream()
  1930. {
  1931. ensure();
  1932. return result->getRowStream();
  1933. }
  1934. virtual IThorRowInterfaces *queryRowInterfaces()
  1935. {
  1936. return rowIf;
  1937. }
  1938. virtual CActivityBase *queryActivity()
  1939. {
  1940. return &activity;
  1941. }
  1942. virtual bool isDistributed() const { return false; }
  1943. virtual void serialize(MemoryBuffer &mb)
  1944. {
  1945. ensure();
  1946. result->serialize(mb);
  1947. }
  1948. virtual void getLinkedResult(unsigned & count, byte * * & ret)
  1949. {
  1950. ensure();
  1951. result->getLinkedResult(count, ret);
  1952. }
  1953. virtual const void * getLinkedRowResult()
  1954. {
  1955. ensure();
  1956. return result->getLinkedRowResult();
  1957. }
  1958. };
  1959. ///////////////////
  1960. //
  1961. // CMasterGraph impl.
  1962. //
  1963. CMasterGraph::CMasterGraph(CJobChannel &jobChannel) : CGraphBase(jobChannel)
  1964. {
  1965. jobM = (CJobMaster *)&jobChannel.queryJob();
  1966. mpTag = queryJob().allocateMPTag();
  1967. startBarrierTag = queryJob().allocateMPTag();
  1968. waitBarrierTag = queryJob().allocateMPTag();
  1969. startBarrier = jobChannel.createBarrier(startBarrierTag);
  1970. waitBarrier = jobChannel.createBarrier(waitBarrierTag);
  1971. }
  1972. CMasterGraph::~CMasterGraph()
  1973. {
  1974. queryJob().freeMPTag(mpTag);
  1975. queryJob().freeMPTag(startBarrierTag);
  1976. queryJob().freeMPTag(waitBarrierTag);
  1977. if (TAG_NULL != doneBarrierTag)
  1978. queryJob().freeMPTag(doneBarrierTag);
  1979. if (TAG_NULL != executeReplyTag)
  1980. queryJob().freeMPTag(executeReplyTag);
  1981. }
  1982. void CMasterGraph::init()
  1983. {
  1984. CGraphBase::init();
  1985. if (queryOwner() && isGlobal())
  1986. {
  1987. doneBarrierTag = queryJob().allocateMPTag();
  1988. doneBarrier = jobChannel.createBarrier(doneBarrierTag);
  1989. }
  1990. }
  1991. bool CMasterGraph::fireException(IException *e)
  1992. {
  1993. IThorException *te = QUERYINTERFACE(e, IThorException);
  1994. ThorExceptionAction action;
  1995. if (!te) action = tea_null;
  1996. else action = te->queryAction();
  1997. if (QUERYINTERFACE(e, IMP_Exception) && MPERR_link_closed==e->errorCode())
  1998. action = tea_shutdown;
  1999. else if (QUERYINTERFACE(e, ISEH_Exception))
  2000. action = tea_shutdown;
  2001. CriticalBlock b(exceptCrit);
  2002. switch (action)
  2003. {
  2004. case tea_warning:
  2005. {
  2006. LOG(MCwarning, thorJob, e);
  2007. ErrorSeverity mappedSeverity = job.queryWorkUnit().getWarningSeverity(e->errorCode(), SeverityWarning);
  2008. if (mappedSeverity != SeverityIgnore)
  2009. reportExceptionToWorkunit(job.queryWorkUnit(), e);
  2010. break;
  2011. }
  2012. case tea_abort:
  2013. {
  2014. EXCLOG(e, NULL);
  2015. abort(e);
  2016. // fall through
  2017. }
  2018. default:
  2019. {
  2020. LOG(MCerror, thorJob, e);
  2021. if (NULL != fatalHandler)
  2022. fatalHandler->inform(LINK(e));
  2023. if (owner)
  2024. owner->fireException(e);
  2025. else
  2026. queryJobChannel().fireException(e);
  2027. }
  2028. }
  2029. return true;
  2030. }
  2031. void CMasterGraph::reset()
  2032. {
  2033. CGraphBase::reset();
  2034. bcastMsgHandler.reset();
  2035. activityInitMsgHandler.reset();
  2036. executeReplyMsgHandler.reset();
  2037. }
  2038. void CMasterGraph::abort(IException *e)
  2039. {
  2040. if (aborted) return;
  2041. bool _graphDone = graphDone; // aborting master activities can trigger master graphDone, but want to fire GraphAbort to slaves if graphDone=false at start.
  2042. try { CGraphBase::abort(e); }
  2043. catch (IException *e)
  2044. {
  2045. GraphPrintLog(e, "Aborting master graph");
  2046. e->Release();
  2047. }
  2048. bcastMsgHandler.cancel(0);
  2049. activityInitMsgHandler.cancel(RANK_ALL);
  2050. executeReplyMsgHandler.cancel(RANK_ALL);
  2051. if (started && !_graphDone)
  2052. {
  2053. try
  2054. {
  2055. CMessageBuffer msg;
  2056. msg.append(GraphAbort);
  2057. msg.append(job.queryKey());
  2058. msg.append(queryGraphId());
  2059. jobM->broadcast(queryNodeComm(), msg, masterSlaveMpTag, LONGTIMEOUT, "abort");
  2060. }
  2061. catch (IException *e)
  2062. {
  2063. GraphPrintLog(e, "Aborting slave graph");
  2064. if (abortException)
  2065. {
  2066. e->Release();
  2067. throw LINK(abortException);
  2068. }
  2069. throw;
  2070. }
  2071. }
  2072. if (!queryOwner())
  2073. {
  2074. if (globals->getPropBool("@watchdogProgressEnabled"))
  2075. queryJobManager().queryDeMonServer()->endGraph(this, !queryAborted());
  2076. }
  2077. }
  2078. bool CMasterGraph::serializeActivityInitData(unsigned slave, MemoryBuffer &mb, IThorActivityIterator &iter)
  2079. {
  2080. CriticalBlock b(createdCrit);
  2081. DelayedSizeMarker sizeMark1(mb);
  2082. ForEach (iter)
  2083. {
  2084. CMasterGraphElement &element = (CMasterGraphElement &)iter.query();
  2085. if (!element.sentActInitData->testSet(slave))
  2086. {
  2087. CMasterActivity *activity = (CMasterActivity *)element.queryActivity();
  2088. if (activity)
  2089. {
  2090. mb.append(element.queryId());
  2091. DelayedSizeMarker sizeMark2(mb);
  2092. activity->serializeSlaveData(mb, slave);
  2093. sizeMark2.write();
  2094. }
  2095. }
  2096. }
  2097. if (0 == sizeMark1.size())
  2098. return false;
  2099. mb.append((activity_id)0); // terminator
  2100. sizeMark1.write();
  2101. return true;
  2102. }
  2103. void CMasterGraph::readActivityInitData(MemoryBuffer &mb, unsigned slave)
  2104. {
  2105. loop
  2106. {
  2107. activity_id id;
  2108. mb.read(id);
  2109. if (0 == id)
  2110. break;
  2111. size32_t dataLen;
  2112. mb.read(dataLen);
  2113. CGraphElementBase *element = queryElement(id);
  2114. MemoryBuffer &mbDst = element->queryActivity()->queryInitializationData(slave);
  2115. mbDst.append(dataLen, mb.readDirect(dataLen));
  2116. }
  2117. }
  2118. bool CMasterGraph::prepare(size32_t parentExtractSz, const byte *parentExtract, bool checkDependencies, bool shortCircuit, bool async)
  2119. {
  2120. if (!CGraphBase::prepare(parentExtractSz, parentExtract, checkDependencies, shortCircuit, async)) return false;
  2121. if (aborted) return false;
  2122. return true;
  2123. }
  2124. void CMasterGraph::execute(size32_t _parentExtractSz, const byte *parentExtract, bool checkDependencies, bool async)
  2125. {
  2126. if (isComplete())
  2127. return;
  2128. if (!queryOwner()) // owning graph sends query+child graphs
  2129. jobM->sendQuery(); // if not previously sent
  2130. CGraphBase::execute(parentExtractSz, parentExtract, checkDependencies, async);
  2131. }
  2132. void CMasterGraph::start()
  2133. {
  2134. if (!queryOwner())
  2135. {
  2136. if (globals->getPropBool("@watchdogProgressEnabled"))
  2137. queryJobManager().queryDeMonServer()->startGraph(this);
  2138. }
  2139. Owned<IThorActivityIterator> iter = getConnectedIterator();
  2140. ForEach (*iter)
  2141. iter->query().queryActivity()->startProcess();
  2142. }
  2143. void CMasterGraph::sendActivityInitData()
  2144. {
  2145. CMessageBuffer msg;
  2146. mptag_t replyTag = queryJobChannel().queryMPServer().createReplyTag();
  2147. msg.setReplyTag(replyTag);
  2148. unsigned pos = msg.length();
  2149. unsigned w=0;
  2150. unsigned sentTo = 0;
  2151. for (; w<queryJob().querySlaves(); w++)
  2152. {
  2153. unsigned needActInit = 0;
  2154. Owned<IThorActivityIterator> iter = getConnectedIterator();
  2155. ForEach(*iter)
  2156. {
  2157. CGraphElementBase &element = iter->query();
  2158. CActivityBase *activity = element.queryActivity();
  2159. if (activity && activity->needReInit())
  2160. element.sentActInitData->set(w, false); // force act init to be resent
  2161. if (!element.sentActInitData->test(w)) // has it been sent
  2162. ++needActInit;
  2163. }
  2164. if (needActInit)
  2165. {
  2166. try
  2167. {
  2168. msg.rewrite(pos);
  2169. Owned<IThorActivityIterator> iter = getConnectedIterator();
  2170. serializeActivityInitData(w, msg, *iter);
  2171. }
  2172. catch (IException *e)
  2173. {
  2174. GraphPrintLog(e);
  2175. throw;
  2176. }
  2177. if (!queryJobChannel().queryJobComm().send(msg, w+1, mpTag, LONGTIMEOUT))
  2178. {
  2179. StringBuffer epStr;
  2180. throw MakeStringException(0, "Timeout sending to slave %s", job.querySlaveGroup().queryNode(w).endpoint().getUrlStr(epStr).str());
  2181. }
  2182. ++sentTo;
  2183. }
  2184. }
  2185. if (sentTo)
  2186. {
  2187. assertex(sentTo == queryJob().querySlaves());
  2188. w=0;
  2189. Owned<IException> e;
  2190. // now get back initialization data from graph tag
  2191. for (; w<queryJob().querySlaves(); w++)
  2192. {
  2193. rank_t sender;
  2194. msg.clear();
  2195. if (!activityInitMsgHandler.recv(queryJobChannel().queryJobComm(), msg, w+1, replyTag, &sender, LONGTIMEOUT))
  2196. throw MakeGraphException(this, 0, "Timeout receiving from slaves after graph sent");
  2197. bool error;
  2198. msg.read(error);
  2199. if (error)
  2200. {
  2201. Owned<IThorException> se = deserializeThorException(msg);
  2202. se->setSlave(sender);
  2203. if (!e.get())
  2204. {
  2205. StringBuffer tmpStr("Slave ");
  2206. queryJob().queryJobGroup().queryNode(sender).endpoint().getUrlStr(tmpStr);
  2207. GraphPrintLog(se, "%s", tmpStr.append(": slave initialization error").str());
  2208. e.setown(se.getClear());
  2209. }
  2210. continue; // to read other slave responses.
  2211. }
  2212. readActivityInitData(msg, w);
  2213. }
  2214. if (e.get())
  2215. throw LINK(e);
  2216. }
  2217. }
  2218. void CMasterGraph::serializeGraphInit(MemoryBuffer &mb)
  2219. {
  2220. mb.append(graphId);
  2221. serializeMPtag(mb, mpTag);
  2222. mb.append((int)startBarrierTag);
  2223. mb.append((int)waitBarrierTag);
  2224. mb.append((int)doneBarrierTag);
  2225. mb.append(queryChildGraphCount());
  2226. Owned<IThorGraphIterator> childIter = getChildGraphIterator();
  2227. ForEach (*childIter)
  2228. {
  2229. CMasterGraph &childGraph = (CMasterGraph &)childIter->query();
  2230. childGraph.serializeGraphInit(mb);
  2231. }
  2232. }
  2233. // IThorChildGraph impl.
  2234. IEclGraphResults *CMasterGraph::evaluate(unsigned _parentExtractSz, const byte *parentExtract)
  2235. {
  2236. throw MakeGraphException(this, 0, "Thor master does not support the execution of child queries");
  2237. }
  2238. void CMasterGraph::executeSubGraph(size32_t parentExtractSz, const byte *parentExtract)
  2239. {
  2240. if (job.queryResumed()) // skip complete subgraph if resuming. NB: temp spill have been tucked away for this purpose when paused.
  2241. {
  2242. if (!queryOwner())
  2243. {
  2244. if (WUGraphComplete == job.queryWorkUnit().queryNodeState(job.queryGraphName(), graphId))
  2245. setCompleteEx();
  2246. }
  2247. }
  2248. if (isComplete())
  2249. return;
  2250. if (!sentGlobalInit)
  2251. {
  2252. sentGlobalInit = true;
  2253. if (!queryOwner())
  2254. sendGraph();
  2255. else
  2256. {
  2257. if (isGlobal())
  2258. {
  2259. CMessageBuffer msg;
  2260. serializeCreateContexts(msg);
  2261. try
  2262. {
  2263. jobM->broadcast(queryJobChannel().queryJobComm(), msg, mpTag, LONGTIMEOUT, "serializeCreateContexts", &bcastMsgHandler);
  2264. }
  2265. catch (IException *e)
  2266. {
  2267. GraphPrintLog(e, "Aborting graph create(2)");
  2268. if (abortException)
  2269. {
  2270. e->Release();
  2271. throw LINK(abortException);
  2272. }
  2273. throw;
  2274. }
  2275. }
  2276. }
  2277. }
  2278. fatalHandler.clear();
  2279. fatalHandler.setown(new CFatalHandler(globals->getPropInt("@fatal_timeout", FATAL_TIMEOUT)));
  2280. CGraphBase::executeSubGraph(parentExtractSz, parentExtract);
  2281. if (TAG_NULL != executeReplyTag)
  2282. {
  2283. rank_t sender;
  2284. unsigned s=0;
  2285. for (; s<queryJob().querySlaves(); s++)
  2286. {
  2287. CMessageBuffer msg;
  2288. if (!executeReplyMsgHandler.recv(queryJobChannel().queryJobComm(), msg, RANK_ALL, executeReplyTag, &sender))
  2289. break;
  2290. bool error;
  2291. msg.read(error);
  2292. if (error)
  2293. {
  2294. Owned<IThorException> exception = deserializeThorException(msg);
  2295. exception->setSlave(sender);
  2296. GraphPrintLog(exception, "slave execute reply exception");
  2297. throw exception.getClear();
  2298. }
  2299. }
  2300. if (fatalHandler)
  2301. fatalHandler->clear();
  2302. }
  2303. fatalHandler.clear();
  2304. Owned<IWorkUnit> wu = &job.queryWorkUnit().lock();
  2305. queryJobManager().updateWorkUnitLog(*wu);
  2306. }
  2307. void CMasterGraph::sendGraph()
  2308. {
  2309. CTimeMon atimer;
  2310. CMessageBuffer msg;
  2311. msg.append(GraphInit);
  2312. msg.append(job.queryKey());
  2313. if (TAG_NULL == executeReplyTag)
  2314. executeReplyTag = jobM->allocateMPTag();
  2315. serializeMPtag(msg, executeReplyTag);
  2316. serializeCreateContexts(msg);
  2317. serializeGraphInit(msg);
  2318. // slave graph data
  2319. try
  2320. {
  2321. jobM->broadcast(queryNodeComm(), msg, masterSlaveMpTag, LONGTIMEOUT, "sendGraph", &bcastMsgHandler);
  2322. }
  2323. catch (IException *e)
  2324. {
  2325. GraphPrintLog(e, "Aborting sendGraph");
  2326. if (abortException)
  2327. {
  2328. e->Release();
  2329. throw LINK(abortException);
  2330. }
  2331. throw;
  2332. }
  2333. GraphPrintLog("sendGraph took %d ms", atimer.elapsed());
  2334. }
  2335. bool CMasterGraph::preStart(size32_t parentExtractSz, const byte *parentExtract)
  2336. {
  2337. GraphPrintLog("Processing graph");
  2338. if (syncInitData())
  2339. {
  2340. sendActivityInitData(); // has to be done at least once
  2341. // NB: At this point, on the slaves, the graphs will start
  2342. }
  2343. CGraphBase::preStart(parentExtractSz, parentExtract);
  2344. if (isGlobal())
  2345. {
  2346. if (!startBarrier->wait(true)) // ensure all graphs are at start point at same time, as some may request data from each other.
  2347. return false;
  2348. }
  2349. if (!queryOwner())
  2350. job.queryWorkUnit().setNodeState(job.queryGraphName(), graphId, WUGraphRunning);
  2351. return true;
  2352. }
  2353. void CMasterGraph::handleSlaveDone(unsigned node, MemoryBuffer &mb)
  2354. {
  2355. graph_id gid;
  2356. mb.read(gid);
  2357. assertex(gid == queryGraphId());
  2358. unsigned count;
  2359. mb.read(count);
  2360. while (count--)
  2361. {
  2362. activity_id activityId;
  2363. mb.read(activityId);
  2364. CMasterGraphElement *act = (CMasterGraphElement *)queryElement(activityId);
  2365. unsigned len;
  2366. mb.read(len);
  2367. const void *d = mb.readDirect(len);
  2368. MemoryBuffer sdMb;
  2369. sdMb.setBuffer(len, (void *)d);
  2370. act->slaveDone(node, sdMb);
  2371. }
  2372. }
  2373. void CMasterGraph::getFinalProgress()
  2374. {
  2375. offset_t totalDiskUsage = 0;
  2376. offset_t minNodeDiskUsage = 0, maxNodeDiskUsage = 0;
  2377. unsigned maxNode = (unsigned)-1, minNode = (unsigned)-1;
  2378. CMessageBuffer msg;
  2379. mptag_t replyTag = queryJobChannel().queryMPServer().createReplyTag();
  2380. msg.setReplyTag(replyTag);
  2381. msg.append((unsigned)GraphEnd);
  2382. msg.append(job.queryKey());
  2383. msg.append(queryGraphId());
  2384. jobM->broadcast(queryNodeComm(), msg, masterSlaveMpTag, LONGTIMEOUT, "graphEnd", NULL, true);
  2385. unsigned n=queryJob().queryNodes();
  2386. while (n--)
  2387. {
  2388. rank_t sender;
  2389. if (!queryNodeComm().recv(msg, RANK_ALL, replyTag, &sender, LONGTIMEOUT))
  2390. {
  2391. GraphPrintLog("Timeout receiving final progress from slaves, %d slaves did not respond", n+1);
  2392. return;
  2393. }
  2394. bool error;
  2395. msg.read(error);
  2396. if (error)
  2397. {
  2398. Owned<IThorException> e = deserializeThorException(msg);
  2399. e->setSlave(sender);
  2400. throw e.getClear();
  2401. }
  2402. if (0 == msg.remaining())
  2403. continue;
  2404. unsigned channelsPerSlave = globals->getPropInt("@channelsPerSlave", 1); // JCSMORE - should move somewhere common
  2405. for (unsigned sc=0; sc<channelsPerSlave; sc++)
  2406. {
  2407. unsigned slave;
  2408. msg.read(slave);
  2409. handleSlaveDone(slave, msg);
  2410. if (!queryOwner())
  2411. {
  2412. if (globals->getPropBool("@watchdogProgressEnabled"))
  2413. {
  2414. try
  2415. {
  2416. size32_t progressLen;
  2417. msg.read(progressLen);
  2418. MemoryBuffer progressData;
  2419. progressData.setBuffer(progressLen, (void *)msg.readDirect(progressLen));
  2420. queryJobManager().queryDeMonServer()->takeHeartBeat(progressData);
  2421. }
  2422. catch (IException *e)
  2423. {
  2424. GraphPrintLog(e, "Failure whilst deserializing stats/progress");
  2425. e->Release();
  2426. }
  2427. }
  2428. }
  2429. offset_t nodeDiskUsage;
  2430. msg.read(nodeDiskUsage);
  2431. jobM->setNodeDiskUsage(n, nodeDiskUsage);
  2432. if (nodeDiskUsage > maxNodeDiskUsage)
  2433. {
  2434. maxNodeDiskUsage = nodeDiskUsage;
  2435. maxNode = n;
  2436. }
  2437. if ((unsigned)-1 == minNode || nodeDiskUsage < minNodeDiskUsage)
  2438. {
  2439. minNodeDiskUsage = nodeDiskUsage;
  2440. minNode = n;
  2441. }
  2442. totalDiskUsage += nodeDiskUsage;
  2443. Owned<ITimeReporter> slaveReport = createStdTimeReporter(msg);
  2444. queryJobChannel().queryTimeReporter().merge(*slaveReport);
  2445. }
  2446. }
  2447. if (totalDiskUsage)
  2448. {
  2449. Owned<IWorkUnit> wu = &job.queryWorkUnit().lock();
  2450. wu->addDiskUsageStats(totalDiskUsage/queryJob().querySlaves(), minNode, minNodeDiskUsage, maxNode, maxNodeDiskUsage, queryGraphId());
  2451. }
  2452. }
  2453. void CMasterGraph::done()
  2454. {
  2455. if (started)
  2456. {
  2457. if (!aborted && (!queryOwner() || isGlobal()))
  2458. getFinalProgress(); // waiting for slave graph to finish and send final progress update + extra act end info.
  2459. }
  2460. CGraphBase::done();
  2461. if (started && !queryOwner())
  2462. {
  2463. if (globals->getPropBool("@watchdogProgressEnabled"))
  2464. queryJobManager().queryDeMonServer()->endGraph(this, true);
  2465. }
  2466. if (!queryOwner())
  2467. {
  2468. if (queryJobChannel().queryTimeReporter().numSections())
  2469. {
  2470. if (globals->getPropBool("@reportTimingsToWorkunit", true))
  2471. {
  2472. struct CReport : implements ITimeReportInfo
  2473. {
  2474. Owned<IWorkUnit> wu;
  2475. CGraphBase &graph;
  2476. CReport(CGraphBase &_graph) : graph(_graph)
  2477. {
  2478. wu.setown(&graph.queryJob().queryWorkUnit().lock());
  2479. }
  2480. virtual void report(const char * timerScope, const char *description, const __int64 totaltime, const __int64 maxtime, const unsigned count)
  2481. {
  2482. StringBuffer timerStr(graph.queryJob().queryGraphName());
  2483. timerStr.append("(").append(graph.queryGraphId()).append("): ");
  2484. timerStr.append(description);
  2485. StringBuffer scope;
  2486. //GH-.JCS is this correct queryGraphId() is a subgraph?
  2487. formatGraphTimerScope(scope, graph.queryJob().queryGraphName(), graph.queryGraphId(), 0);
  2488. scope.append(":").append(timerScope);
  2489. wu->setStatistic(queryStatisticsComponentType(), queryStatisticsComponentName(), SSTsection, scope, StTimeElapsed, timerStr.str(), totaltime, count, maxtime, StatsMergeReplace);
  2490. }
  2491. } wureport(*this);
  2492. queryJobChannel().queryTimeReporter().report(wureport);
  2493. }
  2494. else
  2495. queryJobChannel().queryTimeReporter().printTimings();
  2496. }
  2497. }
  2498. }
  2499. void CMasterGraph::setComplete(bool tf)
  2500. {
  2501. CGraphBase::setComplete(tf);
  2502. if (tf && !queryOwner())
  2503. {
  2504. job.queryWorkUnit().setNodeState(job.queryGraphName(), graphId, graphDone?WUGraphComplete:WUGraphFailed);
  2505. }
  2506. }
  2507. bool CMasterGraph::deserializeStats(unsigned node, MemoryBuffer &mb)
  2508. {
  2509. CriticalBlock b(createdCrit);
  2510. unsigned count, _count;
  2511. mb.read(count);
  2512. if (count)
  2513. setProgressUpdated();
  2514. _count = count;
  2515. while (count--)
  2516. {
  2517. activity_id activityId;
  2518. mb.read(activityId);
  2519. CMasterActivity *activity = NULL;
  2520. CMasterGraphElement *element = (CMasterGraphElement *)queryElement(activityId);
  2521. if (element)
  2522. {
  2523. activity = (CMasterActivity *)element->queryActivity();
  2524. if (!activity)
  2525. {
  2526. CGraphBase *parentGraph = element->queryOwner().queryOwner(); // i.e. am I in a childgraph
  2527. if (!parentGraph)
  2528. {
  2529. GraphPrintLog("Activity id=%" ACTPF "d not created in master and not a child query activity", activityId);
  2530. return false; // don't know if or how this could happen, but all bets off with packet if did.
  2531. }
  2532. Owned<IException> e;
  2533. try
  2534. {
  2535. element->createActivity();
  2536. activity = (CMasterActivity *)element->queryActivity();
  2537. }
  2538. catch (IException *_e)
  2539. {
  2540. e.setown(_e);
  2541. GraphPrintLog(_e, "In deserializeStats");
  2542. }
  2543. if (!activity || e.get())
  2544. {
  2545. GraphPrintLog("Activity id=%" ACTPF "d failed to created child query activity ready for progress", activityId);
  2546. return false;
  2547. }
  2548. }
  2549. if (activity)
  2550. activity->deserializeStats(node, mb);
  2551. }
  2552. else
  2553. {
  2554. GraphPrintLog("Failed to find activity, during progress deserialization, id=%" ACTPF "d", activityId);
  2555. return false; // don't know if or how this could happen, but all bets off with packet if did.
  2556. }
  2557. }
  2558. unsigned subs, _subs;
  2559. mb.read(subs);
  2560. _subs = subs;
  2561. while (subs--)
  2562. {
  2563. graph_id subId;
  2564. mb.read(subId);
  2565. Owned<CMasterGraph> graph = (CMasterGraph *)job.queryJobChannel(0).getGraph(subId);
  2566. if (NULL == graph.get())
  2567. return false;
  2568. if (!graph->deserializeStats(node, mb))
  2569. return false;
  2570. }
  2571. return true;
  2572. }
  2573. IThorResult *CMasterGraph::createResult(CActivityBase &activity, unsigned id, IThorGraphResults *results, IThorRowInterfaces *rowIf, bool distributed, unsigned spillPriority)
  2574. {
  2575. Owned<CCollatedResult> result = new CCollatedResult(*this, activity, rowIf, id, results->queryOwnerId(), spillPriority);
  2576. results->setResult(id, result);
  2577. return result;
  2578. }
  2579. IThorResult *CMasterGraph::createResult(CActivityBase &activity, unsigned id, IThorRowInterfaces *rowIf, bool distributed, unsigned spillPriority)
  2580. {
  2581. Owned<CCollatedResult> result = new CCollatedResult(*this, activity, rowIf, id, localResults->queryOwnerId(), spillPriority);
  2582. localResults->setResult(id, result);
  2583. return result;
  2584. }
  2585. IThorResult *CMasterGraph::createGraphLoopResult(CActivityBase &activity, IThorRowInterfaces *rowIf, bool distributed, unsigned spillPriority)
  2586. {
  2587. Owned<CCollatedResult> result = new CCollatedResult(*this, activity, rowIf, 0, localResults->queryOwnerId(), spillPriority);
  2588. unsigned id = graphLoopResults->addResult(result);
  2589. result->setId(id);
  2590. return result;
  2591. }
  2592. ///////////////////////////////////////////////////
  2593. static bool suppressStatisticIfZero(StatisticKind kind)
  2594. {
  2595. switch (kind)
  2596. {
  2597. case StNumSpills:
  2598. case StSizeSpillFile:
  2599. case StTimeSpillElapsed:
  2600. case StNumDiskRetries:
  2601. return true;
  2602. }
  2603. return false;
  2604. }
  2605. ///////////////////////////////////////////////////
  2606. CThorStats::CThorStats(CJobBase &_ctx, StatisticKind _kind) : ctx(_ctx), kind(_kind)
  2607. {
  2608. unsigned c = queryClusterWidth();
  2609. while (c--) counts.append(0);
  2610. reset();
  2611. }
  2612. void CThorStats::extract(unsigned node, const CRuntimeStatisticCollection & stats)
  2613. {
  2614. set(node, stats.getStatisticValue(kind));
  2615. }
  2616. void CThorStats::set(unsigned node, unsigned __int64 count)
  2617. {
  2618. counts.replace(count, node);
  2619. }
  2620. void CThorStats::reset()
  2621. {
  2622. tot = max = avg = 0;
  2623. min = (unsigned __int64) -1;
  2624. minNode = maxNode = maxSkew = minSkew = maxNode = minNode = 0;
  2625. }
  2626. void CThorStats::calculateSkew()
  2627. {
  2628. unsigned count = counts.ordinality();
  2629. double _avg = (double)tot/count;
  2630. if (_avg)
  2631. {
  2632. if (max > ctx.querySlaves()) // i.e. if small count, suppress skew stats.
  2633. {
  2634. //MORE: Range protection on maxSkew?
  2635. maxSkew = (unsigned)(10000.0 * (((double)max-_avg)/_avg));
  2636. minSkew = (unsigned)(10000.0 * ((_avg-(double)min)/_avg));
  2637. }
  2638. avg = (unsigned __int64)_avg;
  2639. }
  2640. }
  2641. void CThorStats::tallyValue(unsigned __int64 thiscount, unsigned n)
  2642. {
  2643. tot += thiscount;
  2644. if (thiscount > max)
  2645. {
  2646. max = thiscount;
  2647. maxNode = n;
  2648. }
  2649. if (thiscount < min)
  2650. {
  2651. min = thiscount;
  2652. minNode = n;
  2653. }
  2654. }
  2655. void CThorStats::processInfo()
  2656. {
  2657. reset();
  2658. ForEachItemIn(n, counts)
  2659. {
  2660. unsigned __int64 thiscount = counts.item(n);
  2661. tallyValue(thiscount, n+1);
  2662. }
  2663. calculateSkew();
  2664. }
  2665. void CThorStats::getStats(IStatisticGatherer & stats, bool suppressMinMaxWhenEqual)
  2666. {
  2667. processInfo();
  2668. if ((0 == tot) && suppressStatisticIfZero(kind))
  2669. return;
  2670. //MORE: For most measures (not time stamps etc.) it would be sensible to output the total here....
  2671. if (!suppressMinMaxWhenEqual || (maxSkew != minSkew))
  2672. {
  2673. stats.addStatistic((StatisticKind)(kind|StMinX), min);
  2674. stats.addStatistic((StatisticKind)(kind|StMaxX), max);
  2675. stats.addStatistic((StatisticKind)(kind|StAvgX), avg);
  2676. }
  2677. if (maxSkew != minSkew)
  2678. {
  2679. stats.addStatistic((StatisticKind)(kind|StSkewMin), -(__int64)minSkew); // Save minimum as a negative value so consistent
  2680. stats.addStatistic((StatisticKind)(kind|StSkewMax), maxSkew);
  2681. stats.addStatistic((StatisticKind)(kind|StNodeMin), minNode);
  2682. stats.addStatistic((StatisticKind)(kind|StNodeMax), maxNode);
  2683. }
  2684. }
  2685. ///////////////////////////////////////////////////
  2686. CTimingInfo::CTimingInfo(CJobBase &ctx) : CThorStats(ctx, StTimeLocalExecute)
  2687. {
  2688. }
  2689. ///////////////////////////////////////////////////
  2690. ProgressInfo::ProgressInfo(CJobBase &ctx) : CThorStats(ctx, StNumRowsProcessed)
  2691. {
  2692. startcount = stopcount = 0;
  2693. }
  2694. void ProgressInfo::processInfo() // reimplement as counts have special flags (i.e. stop/start)
  2695. {
  2696. reset();
  2697. startcount = stopcount = 0;
  2698. ForEachItemIn(n, counts)
  2699. {
  2700. unsigned __int64 thiscount = counts.item(n);
  2701. if (thiscount & THORDATALINK_STARTED)
  2702. startcount++;
  2703. if (thiscount & THORDATALINK_STOPPED)
  2704. stopcount++;
  2705. thiscount = thiscount & THORDATALINK_COUNT_MASK;
  2706. tallyValue(thiscount, n+1);
  2707. }
  2708. calculateSkew();
  2709. }
  2710. void ProgressInfo::getStats(IStatisticGatherer & stats)
  2711. {
  2712. CThorStats::getStats(stats, true);
  2713. stats.addStatistic(kind, tot);
  2714. stats.addStatistic(StNumSlaves, counts.ordinality());
  2715. stats.addStatistic(StNumStarted, startcount);
  2716. stats.addStatistic(StNumStopped, stopcount);
  2717. }
  2718. ///////////////////////////////////////////////////
  2719. CJobMaster *createThorGraph(const char *graphName, IConstWorkUnit &workunit, ILoadedDllEntry *querySo, bool sendSo, const SocketEndpoint &agentEp)
  2720. {
  2721. Owned<CJobMaster> jobMaster = new CJobMaster(workunit, graphName, querySo, sendSo, agentEp);
  2722. IPropertyTree *graphXGMML = jobMaster->queryGraphXGMML();
  2723. Owned<IPropertyTreeIterator> iter = graphXGMML->getElements("node");
  2724. ForEach(*iter)
  2725. jobMaster->addSubGraph(iter->query());
  2726. jobMaster->addDependencies(graphXGMML);
  2727. return LINK(jobMaster);
  2728. }
  2729. static IJobManager *jobManager = NULL;
  2730. void setJobManager(IJobManager *_jobManager)
  2731. {
  2732. CriticalBlock b(*jobManagerCrit);
  2733. jobManager = _jobManager;
  2734. }
  2735. IJobManager *getJobManager()
  2736. {
  2737. CriticalBlock b(*jobManagerCrit);
  2738. return LINK(jobManager);
  2739. }
  2740. IJobManager &queryJobManager()
  2741. {
  2742. CriticalBlock b(*jobManagerCrit);
  2743. assertex(jobManager);
  2744. return *jobManager;
  2745. }