hqlecl.cpp 24 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709
  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 "build-config.h"
  14. #include "jliball.hpp"
  15. #include "jmisc.hpp"
  16. #include "jstream.hpp"
  17. #include "hql.hpp"
  18. #include "hqlexpr.hpp"
  19. #include "hqlecl.hpp"
  20. #include "hqlthql.hpp"
  21. #include "hqlcerrors.hpp"
  22. #include "hqlcatom.hpp"
  23. #include "hqllib.ipp"
  24. #include "hqlutil.hpp"
  25. #include "hqlhtcpp.ipp"
  26. #include "hqlwcpp.hpp"
  27. #include "hqllib.ipp"
  28. #include "hqlttcpp.ipp"
  29. #include "workunit.hpp"
  30. #include "thorplugin.hpp"
  31. #ifdef _DEBUG
  32. #define IS_DEBUG_BUILD true
  33. #else
  34. #define IS_DEBUG_BUILD false
  35. #endif
  36. #define MAIN_MODULE_TEMPLATE "thortpl.cpp"
  37. #define HEADER_TEMPLATE "thortpl.hpp"
  38. #define CHILD_MODULE_TEMPLATE "childtpl.cpp"
  39. #define PROTO_TEMPLATE "prototpl.cpp"
  40. class NullContextCallback : public CInterface, implements ICodegenContextCallback
  41. {
  42. IMPLEMENT_IINTERFACE
  43. virtual void noteCluster(const char *clusterName) {}
  44. virtual void registerFile(const char * filename, const char * description) {}
  45. virtual bool allowAccess(const char * category) { return true; }
  46. };
  47. class HqlDllGenerator : public CInterface, implements IHqlExprDllGenerator, implements IAbortRequestCallback
  48. {
  49. public:
  50. HqlDllGenerator(IErrorReceiver * _errs, const char * _wuname, const char * _targetdir, IWorkUnit * _wu, const char * _template_dir, ClusterType _targetClusterType, ICodegenContextCallback * _ctxCallback, bool _checkForLocalFileUploads, bool _okToAbort) :
  51. errs(_errs), wuname(_wuname), targetDir(_targetdir), wu(_wu), template_dir(_template_dir), targetClusterType(_targetClusterType), ctxCallback(_ctxCallback), checkForLocalFileUploads(_checkForLocalFileUploads), okToAbort(_okToAbort)
  52. {
  53. if (!ctxCallback)
  54. ctxCallback.setown(new NullContextCallback);
  55. noOutput = true;
  56. defaultMaxCompileThreads = 1;
  57. generateTarget = EclGenerateNone;
  58. code.setown(createCppInstance(wu, wuname));
  59. deleteGenerated = false;
  60. totalGeneratedSize = 0;
  61. }
  62. IMPLEMENT_IINTERFACE
  63. virtual void addLibrary(const char * name);
  64. virtual bool processQuery(OwnedHqlExpr & parsedQuery, EclGenerateTarget _generateTarget);
  65. virtual bool generateDll(ICppCompiler * compiler);
  66. virtual bool generateExe(ICppCompiler * compiler);
  67. virtual bool generatePackage(const char * packageName);
  68. virtual void setMaxCompileThreads(unsigned value) { defaultMaxCompileThreads = value; }
  69. virtual void addManifest(const char *filename) { code->addManifest(filename); }
  70. virtual void addManifestFromArchive(IPropertyTree *archive) { code->addManifestFromArchive(archive); }
  71. virtual void addWebServiceInfo(IPropertyTree *wsinfo){ code->addWebServiceInfo(wsinfo); }
  72. virtual double getECLcomplexity(IHqlExpression * exprs);
  73. virtual void setSaveGeneratedFiles(bool value) { deleteGenerated = !value; }
  74. protected:
  75. void addCppName(const char * filename);
  76. void addLibrariesToCompiler();
  77. void addWorkUnitAsResource();
  78. void calculateHash(IHqlExpression * expr);
  79. bool doCompile(ICppCompiler * compiler);
  80. void doExpand(HqlCppTranslator & translator);
  81. void expandCode(const char * templateName, const char * ext, IHqlCppInstance * code, bool multiFile, unsigned pass, CompilerType compiler);
  82. void flushResources();
  83. bool generateCode(HqlQueryContext & query);
  84. bool generateFullFieldUsageStatistics(HqlCppTranslator & translator, IHqlExpression * query);
  85. IPropertyTree * generateSingleFieldUsageStatistics(IHqlExpression * expr, const char * variety, const IPropertyTree * exclude);
  86. offset_t getGeneratedSize() const;
  87. void insertStandAloneCode();
  88. void setWuState(bool ok);
  89. inline bool abortRequested();
  90. protected:
  91. Linked<IErrorReceiver> errs;
  92. const char * wuname;
  93. StringAttr targetDir;
  94. Linked<IWorkUnit> wu;
  95. Owned<IHqlCppInstance> code;
  96. const char * template_dir;
  97. ClusterType targetClusterType;
  98. Linked<ICodegenContextCallback> ctxCallback;
  99. unsigned defaultMaxCompileThreads;
  100. StringArray sourceFiles;
  101. StringArray libraries;
  102. offset_t totalGeneratedSize;
  103. bool checkForLocalFileUploads;
  104. bool noOutput;
  105. EclGenerateTarget generateTarget;
  106. bool deleteGenerated;
  107. bool okToAbort;
  108. };
  109. //---------------------------------------------------------------------------
  110. static void processMetaCommands(HqlCppTranslator & translator, IWorkUnit * wu, HqlQueryContext & query, ICodegenContextCallback *ctxCallback);
  111. void HqlDllGenerator::addCppName(const char * filename)
  112. {
  113. if (wu)
  114. {
  115. Owned<IWUQuery> query = wu->updateQuery();
  116. associateLocalFile(query, FileTypeCpp, filename, pathTail(filename), 0);
  117. ctxCallback->registerFile(filename, "Workunit CPP");
  118. }
  119. }
  120. void HqlDllGenerator::addLibrary(const char * name)
  121. {
  122. libraries.append(name);
  123. }
  124. void HqlDllGenerator::addLibrariesToCompiler()
  125. {
  126. unsigned idx=0;
  127. loop
  128. {
  129. const char * lib = code->queryLibrary(idx);
  130. if (!lib)
  131. break;
  132. PrintLog("Adding library: %s", lib);
  133. addLibrary(lib);
  134. idx++;
  135. }
  136. }
  137. void HqlDllGenerator::expandCode(const char * templateName, const char * ext, IHqlCppInstance * code, bool multiFile, unsigned pass, CompilerType compiler)
  138. {
  139. StringBuffer fullname;
  140. addDirectoryPrefix(fullname, targetDir).append(wuname).append(ext);
  141. Owned<IFile> out = createIFile(fullname.str());
  142. Owned<ITemplateExpander> expander = createTemplateExpander(out, templateName, template_dir);
  143. if (!expander)
  144. throwError2(HQLERR_CouldNotOpenTemplateXatY, templateName, template_dir ? template_dir : ".");
  145. Owned<ISectionWriter> writer = createCppWriter(*code, compiler);
  146. Owned<IProperties> props = createProperties(true);
  147. if (multiFile)
  148. {
  149. props->setProp("multiFile", true);
  150. props->setProp("pass", pass);
  151. }
  152. StringBuffer headerName;
  153. headerName.append(wuname).append(".hpp");
  154. props->setProp("headerName", headerName.str());
  155. props->setProp("outputName", fullname.str());
  156. expander->generate(*writer, pass, props);
  157. totalGeneratedSize += out->size();
  158. if (!deleteGenerated)
  159. addCppName(fullname);
  160. }
  161. //---------------------------------------------------------------------------
  162. bool HqlDllGenerator::processQuery(OwnedHqlExpr & parsedQuery, EclGenerateTarget _generateTarget)
  163. {
  164. generateTarget = _generateTarget;
  165. assertex(wu->getHash());
  166. unsigned prevCount = errs->errCount();
  167. HqlQueryContext query;
  168. query.expr.setown(parsedQuery.getClear());
  169. bool ok = generateCode(query);
  170. code->flushHints();
  171. wu->commit();
  172. if (!ok)
  173. return false;
  174. if (errs->errCount() != prevCount)
  175. return false;
  176. switch (generateTarget)
  177. {
  178. case EclGenerateNone:
  179. case EclGenerateCpp:
  180. return true;
  181. }
  182. flushResources();
  183. addLibrariesToCompiler();
  184. // Free up memory before the c++ compile occurs
  185. code.clear();
  186. return true;
  187. }
  188. bool HqlDllGenerator::generateDll(ICppCompiler * compiler)
  189. {
  190. if (noOutput || !compiler)
  191. return true;
  192. bool ok = doCompile(compiler);
  193. setWuState(ok);
  194. return ok;
  195. }
  196. bool HqlDllGenerator::generateExe(ICppCompiler * compiler)
  197. {
  198. if (noOutput || !compiler)
  199. return true;
  200. compiler->setCreateExe(true);
  201. bool ok = doCompile(compiler);
  202. setWuState(ok);
  203. return ok;
  204. }
  205. bool HqlDllGenerator::generatePackage(const char * packageName)
  206. {
  207. return false;
  208. }
  209. static bool isSetMeta(IHqlExpression * expr)
  210. {
  211. switch (expr->getOperator())
  212. {
  213. case no_compound:
  214. case no_comma:
  215. return isSetMeta(expr->queryChild(0)) && isSetMeta(expr->queryChild(1));
  216. case no_setmeta:
  217. return true;
  218. default:
  219. return false;
  220. }
  221. }
  222. IPropertyTree * HqlDllGenerator::generateSingleFieldUsageStatistics(IHqlExpression * expr, const char * variety, const IPropertyTree * exclude)
  223. {
  224. //Generate each into a new workunit
  225. // Owned<IWorkUnit> localWu = createLocalWorkUnit();
  226. IWorkUnit * localWu = wu;
  227. //Generate the code into a new instance each time so it doesn't hang around eating memory
  228. Owned<IHqlCppInstance> localCode = createCppInstance(localWu, wuname);
  229. HqlQueryContext query;
  230. query.expr.set(expr);
  231. resetUniqueId();
  232. wu->resetBeforeGeneration();
  233. HqlCppTranslator translator(errs, wuname, localCode, targetClusterType, ctxCallback);
  234. try
  235. {
  236. translator.buildCpp(*localCode, query);
  237. return translator.gatherFieldUsage(variety, exclude);
  238. }
  239. catch (IException * e)
  240. {
  241. if (e->errorCode() != HQLERR_ErrorAlreadyReported)
  242. {
  243. unsigned errcode = e->errorCode() ? e->errorCode() : 1;
  244. StringBuffer s;
  245. errs->reportError(errcode, e->errorMessage(s).str(), NULL, 0, 0, 1);
  246. }
  247. e->Release();
  248. return NULL;
  249. }
  250. catch (RELEASE_CATCH_ALL)
  251. {
  252. errs->reportError(99, "Unknown error", NULL, 0, 0, 1);
  253. return NULL;
  254. }
  255. }
  256. bool HqlDllGenerator::generateFullFieldUsageStatistics(HqlCppTranslator & translator, IHqlExpression* query)
  257. {
  258. LinkedHqlExpr savedQuery = query;
  259. //Strip any #options etc. and check that the query consists of a single OUTPUT() action
  260. while ((savedQuery->getOperator() == no_compound) && isSetMeta(savedQuery->queryChild(0)))
  261. savedQuery.set(savedQuery->queryChild(1));
  262. if (savedQuery->getOperator() != no_output)
  263. throw MakeStringException(0, "#option('GenerateFullFieldUsage') requires the query to be a single OUTPUT()");
  264. Owned<IPropertyTree> allFields = createPTree("usage");
  265. IHqlExpression * dataset = savedQuery->queryChild(0);
  266. //First calculate which files are used if no fields are extracted from the output i.e. shared by all output fields
  267. //Use COUNT(dataset) as the query to test that
  268. OwnedHqlExpr countDataset = createValue(no_count, makeIntType(8, false), LINK(dataset));
  269. Owned<IPropertyTree> countFilesUsed = generateSingleFieldUsageStatistics(countDataset, "__shared__", NULL);
  270. if (countFilesUsed)
  271. allFields->addPropTree(countFilesUsed->queryName(), LINK(countFilesUsed));
  272. //Now project the output dataset down to a single field, and generate the file/field dependencies for each of those
  273. OwnedHqlExpr selSeq = createUniqueSelectorSequence();
  274. OwnedHqlExpr left = createSelector(no_left, dataset, selSeq);
  275. RecordSelectIterator iter(dataset->queryRecord(), left);
  276. ForEach(iter)
  277. {
  278. IHqlExpression * cur = iter.query();
  279. IHqlExpression * field = cur->queryChild(1);
  280. OwnedHqlExpr record = createRecord(field);
  281. OwnedHqlExpr self = getSelf(record);
  282. OwnedHqlExpr assign = createAssign(createSelectExpr(LINK(self), LINK(field)), LINK(cur));
  283. OwnedHqlExpr transform = createValue(no_transform, makeTransformType(record->getType()), LINK(assign), NULL);
  284. HqlExprArray args;
  285. args.append(*LINK(dataset));
  286. args.append(*LINK(transform));
  287. args.append(*LINK(selSeq));
  288. OwnedHqlExpr project = createDataset(no_hqlproject,args);
  289. OwnedHqlExpr projectedOutput = replaceChildDataset(savedQuery, project, 0);
  290. StringBuffer variety;
  291. getExprIdentifier(variety, cur);
  292. //Generate each into a new property tree, and only include fields not shared by all other output fields.
  293. Owned<IPropertyTree> filesUsed = generateSingleFieldUsageStatistics(projectedOutput, variety.str(), countFilesUsed);
  294. if (filesUsed)
  295. allFields->addPropTree(filesUsed->queryName(), LINK(filesUsed));
  296. //Generate progress so far
  297. translator.writeFieldUsage(targetDir, allFields, NULL);
  298. }
  299. translator.writeFieldUsage(targetDir, allFields, NULL);
  300. return false;
  301. }
  302. bool HqlDllGenerator::generateCode(HqlQueryContext & query)
  303. {
  304. noOutput = true;
  305. {
  306. // ensure warnings/errors are available before we do the processing...
  307. wu->commit();
  308. MTIME_SECTION (timer, "Generate_code");
  309. unsigned time = msTick();
  310. HqlCppTranslator translator(errs, wuname, code, targetClusterType, ctxCallback);
  311. processMetaCommands(translator, wu, query, ctxCallback);
  312. if (wu->getDebugValueBool("generateFullFieldUsage", false))
  313. {
  314. //Ensure file information is generated. It only partially works with field information at the moment.
  315. //(The merging/difference logic would need to improve, but would be relatively simple if it was useful.)
  316. wu->setDebugValueInt("reportFileUsage", 1, true);
  317. return generateFullFieldUsageStatistics(translator, query.expr);
  318. }
  319. bool ok = false;
  320. try
  321. {
  322. if (!translator.buildCpp(*code, query))
  323. {
  324. wu->setState(WUStateCompleted);
  325. return true;
  326. }
  327. translator.generateStatistics(targetDir, NULL);
  328. translator.finalizeResources();
  329. translator.expandFunctions(true);
  330. }
  331. catch (IECLError * e)
  332. {
  333. if (e->errorCode() != HQLERR_ErrorAlreadyReported)
  334. {
  335. StringBuffer s;
  336. errs->reportError(e->errorCode(), e->errorMessage(s).str(), e->getFilename(), e->getLine(), e->getColumn(), e->getPosition());
  337. }
  338. e->Release();
  339. return false;
  340. }
  341. catch (IException * e)
  342. {
  343. if (e->errorCode() != HQLERR_ErrorAlreadyReported)
  344. {
  345. unsigned errcode = e->errorCode() ? e->errorCode() : 1;
  346. StringBuffer s;
  347. errs->reportError(errcode, e->errorMessage(s).str(), NULL, 0, 0, 1);
  348. }
  349. e->Release();
  350. return false;
  351. }
  352. catch (RELEASE_CATCH_ALL)
  353. {
  354. errs->reportError(99, "Unknown error", NULL, 0, 0, 1);
  355. return false;
  356. }
  357. if (generateTarget == EclGenerateExe)
  358. insertStandAloneCode();
  359. wu->commit();
  360. //Commit work unit so can view graphs etc. while compiling the C++
  361. if ((generateTarget == EclGenerateNone) || wu->getDebugValueBool("OnlyCheckQuery", false))
  362. {
  363. wu->setState(WUStateCompleted);
  364. return false;
  365. }
  366. if (wu->getDebugValueBool("saveEclTempFiles", false) || wu->getDebugValueBool("saveCppTempFiles", false))
  367. setSaveGeneratedFiles(true);
  368. doExpand(translator);
  369. if (wu->getDebugValueBool("addTimingToWorkunit", true))
  370. {
  371. unsigned elapsed = msTick()-time;
  372. updateWorkunitTimeStat(wu, "eclcc", "workunit", "GenerateCpp", "eclcc: generate code", milliToNano(elapsed), 1, 0);
  373. }
  374. wu->commit();
  375. addWorkUnitAsResource();
  376. }
  377. noOutput = false;
  378. return true;
  379. }
  380. void HqlDllGenerator::addWorkUnitAsResource()
  381. {
  382. SCMStringBuffer wuXML;
  383. exportWorkUnitToXML(wu, wuXML, false, false);
  384. code->addCompressResource("WORKUNIT", wuXML.length(), wuXML.str(), NULL, 1000);
  385. }
  386. void HqlDllGenerator::insertStandAloneCode()
  387. {
  388. BuildCtx ctx(static_cast<HqlCppInstance &>(*code), goAtom);
  389. ctx.addQuotedCompound("int main(int argc, const char *argv[])");
  390. ctx.addQuoted("return start_query(argc, argv);\n");
  391. }
  392. void HqlDllGenerator::doExpand(HqlCppTranslator & translator)
  393. {
  394. unsigned startExpandTime = msTick();
  395. unsigned numExtraFiles = translator.getNumExtraCppFiles();
  396. bool isMultiFile = translator.spanMultipleCppFiles() && (numExtraFiles != 0);
  397. expandCode(MAIN_MODULE_TEMPLATE, ".cpp", code, isMultiFile, 0, translator.queryOptions().targetCompiler);
  398. if (isMultiFile)
  399. {
  400. expandCode(HEADER_TEMPLATE, ".hpp", code, true, 0, translator.queryOptions().targetCompiler);
  401. for (unsigned i= 0; i < translator.getNumExtraCppFiles(); i++)
  402. {
  403. StringBuffer fullext;
  404. fullext.append("_").append(i+1).append(".cpp");
  405. expandCode(CHILD_MODULE_TEMPLATE, fullext, code, true, i+1, translator.queryOptions().targetCompiler);
  406. StringBuffer fullname;
  407. fullname.append(wuname).append("_").append(i+1);
  408. sourceFiles.append(fullname);
  409. }
  410. }
  411. unsigned endExpandTime = msTick();
  412. if (wu->getDebugValueBool("addTimingToWorkunit", true))
  413. updateWorkunitTimeStat(wu, "eclcc", "workunit", "writeCpp", "eclcc: Time to write c++", milliToNano(endExpandTime-startExpandTime), 1, 0);
  414. }
  415. bool HqlDllGenerator::abortRequested()
  416. {
  417. return (wu && wu->aborting());
  418. }
  419. bool HqlDllGenerator::doCompile(ICppCompiler * compiler)
  420. {
  421. ForEachItemIn(i, sourceFiles)
  422. compiler->addSourceFile(sourceFiles.item(i));
  423. unsigned maxThreads = wu->getDebugValueInt("maxCompileThreads", defaultMaxCompileThreads);
  424. compiler->setMaxCompileThreads(maxThreads);
  425. bool debug = wu->getDebugValueBool("debugQuery", false);
  426. bool debugLibrary = debug; // should be wu->getDebugValueBool("debugLibrary", IS_DEBUG_BUILD); change for 3.8
  427. compiler->setDebug(debug);
  428. compiler->setDebugLibrary(debugLibrary);
  429. if (!debug)
  430. {
  431. int optimizeLevel = wu->getDebugValueInt("optimizeLevel", targetClusterType == RoxieCluster ? 3 : -1);
  432. if (optimizeLevel != -1)
  433. compiler->setOptimizeLevel(optimizeLevel);
  434. }
  435. #ifdef __64BIT__
  436. // ARMFIX: Map all the uses of this property and make sure
  437. // they're not used to mean x86_64 (it shouldn't, though)
  438. bool target64bit = wu->getDebugValueBool("target64bit", true);
  439. #else
  440. bool target64bit = wu->getDebugValueBool("target64bit", false);
  441. #endif
  442. compiler->setTargetBitLength(target64bit ? 64 : 32);
  443. ForEachItemIn(idx, libraries)
  444. compiler->addLibrary(libraries.item(idx));
  445. StringBuffer options;
  446. StringBufferAdaptor linkOptionAdaptor(options);
  447. wu->getDebugValue("linkOptions", linkOptionAdaptor);
  448. compiler->addLinkOption(options.str());
  449. options.clear();
  450. StringBufferAdaptor optionAdaptor(options);
  451. wu->getDebugValue("compileOptions", optionAdaptor);
  452. compiler->addCompileOption(options.str());
  453. if (okToAbort)
  454. compiler->setAbortChecker(this);
  455. MTIME_SECTION (timer, "Compile_code");
  456. unsigned time = msTick();
  457. PrintLog("Compiling %s", wuname);
  458. bool ok = compiler->compile();
  459. if(ok)
  460. PrintLog("Compiled %s", wuname);
  461. else
  462. PrintLog("Failed to compile %s", wuname);
  463. time = msTick()-time;
  464. if (wu->getDebugValueBool("addTimingToWorkunit", true))
  465. updateWorkunitTimeStat(wu, "eclcc", "workunit", "compile", "eclcc: compile code", milliToNano(time), 1, 0);
  466. //Keep the files if there was a compile error.
  467. if (ok && deleteGenerated)
  468. {
  469. StringBuffer temp;
  470. remove(temp.clear().append(wuname).append(".cpp").str());
  471. remove(temp.clear().append(wuname).append(".hpp").str());
  472. ForEachItemIn(i, sourceFiles)
  473. {
  474. temp.clear().append(sourceFiles.item(i)).append(".cpp");
  475. remove(temp.str());
  476. }
  477. }
  478. return ok;
  479. }
  480. void HqlDllGenerator::flushResources()
  481. {
  482. StringBuffer resname(wuname);
  483. #ifdef _WIN32
  484. resname.append(".res");
  485. #else
  486. resname.append(".res.o");
  487. #endif
  488. if (code)
  489. code->flushResources(resname.str(), ctxCallback);
  490. }
  491. void HqlDllGenerator::setWuState(bool ok)
  492. {
  493. if(ok)
  494. {
  495. if(checkForLocalFileUploads && wu->requiresLocalFileUpload())
  496. wu->setState(WUStateUploadingFiles);
  497. else
  498. wu->setState(WUStateCompiled);
  499. }
  500. else
  501. wu->setState(WUStateFailed);
  502. }
  503. double HqlDllGenerator::getECLcomplexity(IHqlExpression * exprs)
  504. {
  505. Owned<IHqlCppInstance> code = createCppInstance(wu, NULL);
  506. HqlCppTranslator translator(errs, "temp", code, targetClusterType, NULL);
  507. HqlQueryContext query;
  508. query.expr.set(exprs);
  509. processMetaCommands(translator, wu, query, ctxCallback);
  510. return translator.getComplexity(*code, query.expr);
  511. }
  512. offset_t HqlDllGenerator::getGeneratedSize() const
  513. {
  514. return totalGeneratedSize;
  515. }
  516. extern HQLCPP_API double getECLcomplexity(IHqlExpression * exprs, IErrorReceiver * errs, IWorkUnit *wu, ClusterType targetClusterType)
  517. {
  518. HqlDllGenerator generator(errs, "unknown", NULL, wu, NULL, targetClusterType, NULL, false, false);
  519. return generator.getECLcomplexity(exprs);
  520. }
  521. extern HQLCPP_API IHqlExprDllGenerator * createDllGenerator(IErrorReceiver * errs, const char *wuname, const char * targetdir, IWorkUnit *wu, const char * template_dir, ClusterType targetClusterType, ICodegenContextCallback *ctxCallback, bool checkForLocalFileUploads, bool okToAbort)
  522. {
  523. return new HqlDllGenerator(errs, wuname, targetdir, wu, template_dir, targetClusterType, ctxCallback, checkForLocalFileUploads, okToAbort);
  524. }
  525. /*
  526. extern HQLCPP_API void addLibraryReference(IWorkUnit * wu, IHqlLibraryInfo * library, const char * lid)
  527. {
  528. StringBuffer libraryName;
  529. library->getName(libraryName);
  530. Owned<IWULibrary> entry = wu->updateLibraryByName(libraryName.str());
  531. entry->setMajorVersion(library->queryMajorVersion());
  532. entry->setMinorVersion(library->queryMinorVersion());
  533. entry->setCrc(library->queryInterfaceCrc());
  534. if (lid && *lid)
  535. entry->setLID(lid);
  536. }
  537. */
  538. extern HQLCPP_API ClusterType queryClusterType(IConstWorkUnit * wu, ClusterType prevType)
  539. {
  540. ClusterType targetClusterType = prevType;
  541. if (wu->getDebugValueBool("forceRoxie", false))
  542. targetClusterType = RoxieCluster;
  543. else if (prevType == NoCluster)
  544. targetClusterType = ThorLCRCluster;
  545. SCMStringBuffer targetText;
  546. wu->getDebugValue("targetClusterType", targetText);
  547. targetClusterType = getClusterType(targetText.s.str(), targetClusterType);
  548. if ((targetClusterType != RoxieCluster) && wu->getDebugValueBool("forceFakeThor", false))
  549. targetClusterType = HThorCluster;
  550. return targetClusterType;
  551. }
  552. static void processMetaCommands(HqlCppTranslator & translator, IWorkUnit * wu, HqlQueryContext & query, ICodegenContextCallback *ctxCallback)
  553. {
  554. NewThorStoredReplacer transformer(translator, wu, ctxCallback);
  555. translator.traceExpression("before process meta commands", query.expr);
  556. transformer.analyse(query.expr);
  557. if (transformer.needToTransform())
  558. query.expr.setown(transformer.transform(query.expr));
  559. }
  560. extern HQLCPP_API unsigned getLibraryCRC(IHqlExpression * library)
  561. {
  562. assertex(library->getOperator() == no_funcdef);
  563. return getExpressionCRC(library->queryChild(0));
  564. }
  565. void setWorkunitHash(IWorkUnit * wu, IHqlExpression * expr)
  566. {
  567. //Assuming builds come from different branches this will change the crc for each one.
  568. unsigned cacheCRC = crc32(BUILD_TAG, strlen(BUILD_TAG), ACTIVITY_INTERFACE_VERSION);
  569. cacheCRC += getExpressionCRC(expr);
  570. #ifdef _WIN32
  571. cacheCRC++; // make sure CRC is different in windows/linux
  572. #endif
  573. // make sure CRC is different for different host platform
  574. // shouldn't really matter if cross-compiling working properly,
  575. // but fairly harmless.
  576. #ifdef _ARCH_X86_
  577. cacheCRC += 1;
  578. #endif
  579. #ifdef _ARCH_X86_64_
  580. cacheCRC += 2;
  581. #endif
  582. // In theory, ARM and x86 workunits should be totally different, but...
  583. #ifdef _ARCH_ARM32_
  584. cacheCRC += 3;
  585. #endif
  586. #ifdef _ARCH_ARM64_
  587. cacheCRC += 4;
  588. #endif
  589. IExtendedWUInterface *ewu = queryExtendedWU(wu);
  590. cacheCRC = ewu->calculateHash(cacheCRC);
  591. wu->setHash(cacheCRC);
  592. }