thorxmlwrite.cpp 38 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308
  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 "platform.h"
  14. #include "jlib.hpp"
  15. #include "thorxmlwrite.hpp"
  16. #include "eclrtl.hpp"
  17. #include "rtlkey.hpp"
  18. #include "eclhelper.hpp"
  19. #include "deftype.hpp"
  20. #include "rtlbcd.hpp"
  21. CommonXmlWriter::CommonXmlWriter(unsigned _flags, unsigned initialIndent, IXmlStreamFlusher *_flusher)
  22. {
  23. flusher = _flusher;
  24. flags = _flags;
  25. indent = initialIndent;
  26. nestLimit = flags & XWFnoindent ? (unsigned) -1 : 0;
  27. tagClosed = true;
  28. }
  29. CommonXmlWriter::~CommonXmlWriter()
  30. {
  31. flush(true);
  32. }
  33. CommonXmlWriter & CommonXmlWriter::clear()
  34. {
  35. out.clear();
  36. indent = 0;
  37. nestLimit = flags & XWFnoindent ? (unsigned) -1 : 0;
  38. tagClosed = true;
  39. return *this;
  40. }
  41. bool CommonXmlWriter::checkForAttribute(const char * fieldname)
  42. {
  43. if (!tagClosed)
  44. {
  45. if (fieldname && (fieldname[0] == '@'))
  46. return true;
  47. closeTag();
  48. }
  49. return false;
  50. }
  51. void CommonXmlWriter::closeTag()
  52. {
  53. if (!tagClosed)
  54. {
  55. out.append(">");
  56. if (!nestLimit)
  57. out.newline();
  58. tagClosed = true;
  59. }
  60. flush(false);
  61. }
  62. void CommonXmlWriter::outputQuoted(const char *text)
  63. {
  64. out.append(text);
  65. }
  66. void CommonXmlWriter::outputString(unsigned len, const char *field, const char *fieldname)
  67. {
  68. if (flags & XWFtrim)
  69. len = rtlTrimStrLen(len, field);
  70. if ((flags & XWFopt) && (rtlTrimStrLen(len, field) == 0))
  71. return;
  72. if (checkForAttribute(fieldname))
  73. outputXmlAttrString(len, field, fieldname+1, out);
  74. else
  75. {
  76. if (!nestLimit)
  77. out.pad(indent);
  78. outputXmlString(len, field, fieldname, out);
  79. if (!nestLimit)
  80. out.newline();
  81. }
  82. }
  83. void CommonXmlWriter::outputQString(unsigned len, const char *field, const char *fieldname)
  84. {
  85. MemoryAttr tempBuffer;
  86. char * temp;
  87. if (len <= 100)
  88. temp = (char *)alloca(len);
  89. else
  90. temp = (char *)tempBuffer.allocate(len);
  91. rtlQStrToStr(len, temp, len, field);
  92. outputString(len, temp, fieldname);
  93. }
  94. void CommonXmlWriter::outputBool(bool field, const char *fieldname)
  95. {
  96. if (checkForAttribute(fieldname))
  97. outputXmlAttrBool(field, fieldname+1, out);
  98. else
  99. {
  100. if (!nestLimit)
  101. out.pad(indent);
  102. outputXmlBool(field, fieldname, out);
  103. if (!nestLimit)
  104. out.newline();
  105. }
  106. }
  107. void CommonXmlWriter::outputData(unsigned len, const void *field, const char *fieldname)
  108. {
  109. if (checkForAttribute(fieldname))
  110. outputXmlAttrData(len, field, fieldname+1, out);
  111. else
  112. {
  113. if (!nestLimit)
  114. out.pad(indent);
  115. outputXmlData(len, field, fieldname, out);
  116. if (!nestLimit)
  117. out.newline();
  118. }
  119. }
  120. void CommonXmlWriter::outputInt(__int64 field, const char *fieldname)
  121. {
  122. if (checkForAttribute(fieldname))
  123. outputXmlAttrInt(field, fieldname+1, out);
  124. else
  125. {
  126. if (!nestLimit)
  127. out.pad(indent);
  128. outputXmlInt(field, fieldname, out);
  129. if (!nestLimit)
  130. out.newline();
  131. }
  132. }
  133. void CommonXmlWriter::outputUInt(unsigned __int64 field, const char *fieldname)
  134. {
  135. if (checkForAttribute(fieldname))
  136. outputXmlAttrUInt(field, fieldname+1, out);
  137. else
  138. {
  139. if (!nestLimit)
  140. out.pad(indent);
  141. outputXmlUInt(field, fieldname, out);
  142. if (!nestLimit)
  143. out.newline();
  144. }
  145. }
  146. void CommonXmlWriter::outputReal(double field, const char *fieldname)
  147. {
  148. if (checkForAttribute(fieldname))
  149. outputXmlAttrReal(field, fieldname+1, out);
  150. else
  151. {
  152. if (!nestLimit)
  153. out.pad(indent);
  154. outputXmlReal(field, fieldname, out);
  155. if (!nestLimit)
  156. out.newline();
  157. }
  158. }
  159. void CommonXmlWriter::outputDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname)
  160. {
  161. if (checkForAttribute(fieldname))
  162. outputXmlAttrDecimal(field, size, precision, fieldname+1, out);
  163. else
  164. {
  165. if (!nestLimit)
  166. out.pad(indent);
  167. outputXmlDecimal(field, size, precision, fieldname, out);
  168. if (!nestLimit)
  169. out.newline();
  170. }
  171. }
  172. void CommonXmlWriter::outputUDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname)
  173. {
  174. if (checkForAttribute(fieldname))
  175. outputXmlAttrUDecimal(field, size, precision, fieldname+1, out);
  176. else
  177. {
  178. if (!nestLimit)
  179. out.pad(indent);
  180. outputXmlUDecimal(field, size, precision, fieldname, out);
  181. if (!nestLimit)
  182. out.newline();
  183. }
  184. }
  185. void CommonXmlWriter::outputUnicode(unsigned len, const UChar *field, const char *fieldname)
  186. {
  187. if (flags & XWFtrim)
  188. len = rtlTrimUnicodeStrLen(len, field);
  189. if ((flags & XWFopt) && (rtlTrimUnicodeStrLen(len, field) == 0))
  190. return;
  191. if (checkForAttribute(fieldname))
  192. outputXmlAttrUnicode(len, field, fieldname+1, out);
  193. else
  194. {
  195. if (!nestLimit)
  196. out.pad(indent);
  197. outputXmlUnicode(len, field, fieldname, out);
  198. if (!nestLimit)
  199. out.newline();
  200. }
  201. }
  202. void CommonXmlWriter::outputUtf8(unsigned len, const char *field, const char *fieldname)
  203. {
  204. if (flags & XWFtrim)
  205. len = rtlTrimUtf8StrLen(len, field);
  206. if ((flags & XWFopt) && (rtlTrimUtf8StrLen(len, field) == 0))
  207. return;
  208. if (checkForAttribute(fieldname))
  209. outputXmlAttrUtf8(len, field, fieldname+1, out);
  210. else
  211. {
  212. if (!nestLimit)
  213. out.pad(indent);
  214. outputXmlUtf8(len, field, fieldname, out);
  215. if (!nestLimit)
  216. out.newline();
  217. }
  218. }
  219. void CommonXmlWriter::outputBeginDataset(const char *dsname, bool nestChildren)
  220. {
  221. outputBeginNested("Dataset", nestChildren, false); //indent row, not dataset for backward compatibility
  222. if (nestChildren && indent==0)
  223. indent++;
  224. if (!dsname || !*dsname)
  225. return;
  226. out.append(" name='"); //single quote for backward compatibility
  227. outputXmlUtf8(strlen(dsname), dsname, NULL, out);
  228. out.append("'");
  229. }
  230. void CommonXmlWriter::outputEndDataset(const char *dsname)
  231. {
  232. outputEndNested("Dataset", false);
  233. }
  234. void CommonXmlWriter::outputBeginNested(const char *fieldname, bool nestChildren, bool doIndent)
  235. {
  236. if (!fieldname || !*fieldname)
  237. return;
  238. const char * sep = strchr(fieldname, '/');
  239. if (sep)
  240. {
  241. StringAttr leading(fieldname, sep-fieldname);
  242. outputBeginNested(leading, nestChildren, doIndent);
  243. outputBeginNested(sep+1, nestChildren, doIndent);
  244. return;
  245. }
  246. closeTag();
  247. if (!nestLimit && doIndent)
  248. out.pad(indent);
  249. out.append('<').append(fieldname);
  250. if (doIndent)
  251. indent += 1;
  252. if (!nestChildren && !nestLimit)
  253. nestLimit = indent;
  254. tagClosed = false;
  255. }
  256. void CommonXmlWriter::outputBeginNested(const char *fieldname, bool nestChildren)
  257. {
  258. outputBeginNested(fieldname, nestChildren, true);
  259. }
  260. void CommonXmlWriter::outputEndNested(const char *fieldname, bool doIndent)
  261. {
  262. if (!fieldname || !*fieldname)
  263. return;
  264. const char * sep = strchr(fieldname, '/');
  265. if (sep)
  266. {
  267. StringAttr leading(fieldname, sep-fieldname);
  268. outputEndNested(sep+1, doIndent);
  269. outputEndNested(leading, doIndent);
  270. return;
  271. }
  272. if (flags & XWFexpandempty)
  273. closeTag();
  274. if (!tagClosed)
  275. {
  276. out.append("/>");
  277. tagClosed = true;
  278. }
  279. else
  280. {
  281. if (!nestLimit && doIndent)
  282. out.pad(indent-1);
  283. out.append("</").append(fieldname).append('>');
  284. }
  285. if (indent==nestLimit)
  286. nestLimit = 0;
  287. if (doIndent)
  288. indent -= 1;
  289. if (!nestLimit)
  290. out.newline();
  291. }
  292. void CommonXmlWriter::outputEndNested(const char *fieldname)
  293. {
  294. outputEndNested(fieldname, true);
  295. }
  296. void CommonXmlWriter::outputSetAll()
  297. {
  298. closeTag();
  299. if (!nestLimit)
  300. out.pad(indent);
  301. outputXmlSetAll(out);
  302. if (!nestLimit)
  303. out.newline();
  304. }
  305. //=====================================================================================
  306. CommonJsonWriter::CommonJsonWriter(unsigned _flags, unsigned initialIndent, IXmlStreamFlusher *_flusher)
  307. {
  308. flusher = _flusher;
  309. flags = _flags;
  310. indent = initialIndent;
  311. nestLimit = flags & XWFnoindent ? (unsigned) -1 : 0;
  312. needDelimiter = false;
  313. }
  314. CommonJsonWriter::~CommonJsonWriter()
  315. {
  316. flush(true);
  317. }
  318. CommonJsonWriter & CommonJsonWriter::clear()
  319. {
  320. out.clear();
  321. indent = 0;
  322. nestLimit = flags & XWFnoindent ? (unsigned) -1 : 0;
  323. return *this;
  324. }
  325. void CommonJsonWriter::checkFormat(bool doDelimit, bool delimitNext, int inc)
  326. {
  327. if (doDelimit)
  328. {
  329. if (needDelimiter)
  330. {
  331. if (!out.length()) //new block
  332. out.append(',');
  333. else
  334. delimitJSON(out);
  335. }
  336. if (!nestLimit)
  337. out.append('\n').pad(indent);
  338. }
  339. indent+=inc;
  340. needDelimiter = delimitNext;
  341. }
  342. void CommonJsonWriter::checkDelimit(int inc)
  343. {
  344. checkFormat(true, true, inc);
  345. }
  346. const char *CommonJsonWriter::checkItemName(CJsonWriterItem *item, const char *name, bool simpleType)
  347. {
  348. if (simpleType && (!name || !*name))
  349. name = "#value"; //xml mixed content
  350. if (item && item->depth==0 && strieq(item->name, name))
  351. return NULL;
  352. return name;
  353. }
  354. const char *CommonJsonWriter::checkItemName(const char *name, bool simpleType)
  355. {
  356. CJsonWriterItem *item = (arrays.length()) ? &arrays.tos() : NULL;
  357. return checkItemName(item, name, simpleType);
  358. }
  359. const char *CommonJsonWriter::checkItemNameBeginNested(const char *name)
  360. {
  361. CJsonWriterItem *item = (arrays.length()) ? &arrays.tos() : NULL;
  362. name = checkItemName(item, name, false);
  363. if (item)
  364. item->depth++;
  365. return name;
  366. }
  367. const char *CommonJsonWriter::checkItemNameEndNested(const char *name)
  368. {
  369. CJsonWriterItem *item = (arrays.length()) ? &arrays.tos() : NULL;
  370. if (item)
  371. item->depth--;
  372. return checkItemName(item, name, false);
  373. }
  374. void CommonJsonWriter::outputQuoted(const char *text)
  375. {
  376. checkDelimit();
  377. appendJSONValue(out, NULL, text);
  378. }
  379. void CommonJsonWriter::outputString(unsigned len, const char *field, const char *fieldname)
  380. {
  381. if (flags & XWFtrim)
  382. len = rtlTrimStrLen(len, field);
  383. if ((flags & XWFopt) && (rtlTrimStrLen(len, field) == 0))
  384. return;
  385. checkDelimit();
  386. appendJSONStringValue(out, checkItemName(fieldname), len, field, true);
  387. }
  388. void CommonJsonWriter::outputQString(unsigned len, const char *field, const char *fieldname)
  389. {
  390. MemoryAttr tempBuffer;
  391. char * temp;
  392. if (len <= 100)
  393. temp = (char *)alloca(len);
  394. else
  395. temp = (char *)tempBuffer.allocate(len);
  396. rtlQStrToStr(len, temp, len, field);
  397. outputString(len, temp, fieldname);
  398. }
  399. void CommonJsonWriter::outputBool(bool field, const char *fieldname)
  400. {
  401. checkDelimit();
  402. appendJSONValue(out, checkItemName(fieldname), field);
  403. }
  404. void CommonJsonWriter::outputData(unsigned len, const void *field, const char *fieldname)
  405. {
  406. checkDelimit();
  407. appendJSONDataValue(out, checkItemName(fieldname), len, field);
  408. }
  409. void CommonJsonWriter::outputInt(__int64 field, const char *fieldname)
  410. {
  411. checkDelimit();
  412. appendJSONValue(out, checkItemName(fieldname), field);
  413. }
  414. void CommonJsonWriter::outputUInt(unsigned __int64 field, const char *fieldname)
  415. {
  416. checkDelimit();
  417. appendJSONValue(out, checkItemName(fieldname), field);
  418. }
  419. void CommonJsonWriter::outputReal(double field, const char *fieldname)
  420. {
  421. checkDelimit();
  422. appendJSONValue(out, checkItemName(fieldname), field);
  423. }
  424. void CommonJsonWriter::outputDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname)
  425. {
  426. checkDelimit();
  427. outputJsonDecimal(field, size, precision, checkItemName(fieldname), out);
  428. }
  429. void CommonJsonWriter::outputUDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname)
  430. {
  431. checkDelimit();
  432. outputJsonUDecimal(field, size, precision, checkItemName(fieldname), out);
  433. }
  434. void CommonJsonWriter::outputUnicode(unsigned len, const UChar *field, const char *fieldname)
  435. {
  436. if (flags & XWFtrim)
  437. len = rtlTrimUnicodeStrLen(len, field);
  438. if ((flags & XWFopt) && (rtlTrimUnicodeStrLen(len, field) == 0))
  439. return;
  440. checkDelimit();
  441. outputJsonUnicode(len, field, checkItemName(fieldname), out);
  442. }
  443. void CommonJsonWriter::outputUtf8(unsigned len, const char *field, const char *fieldname)
  444. {
  445. if (flags & XWFtrim)
  446. len = rtlTrimUtf8StrLen(len, field);
  447. if ((flags & XWFopt) && (rtlTrimUtf8StrLen(len, field) == 0))
  448. return;
  449. checkDelimit();
  450. appendJSONStringValue(out, checkItemName(fieldname), len, field, true);
  451. }
  452. void CommonJsonWriter::outputBeginArray(const char *fieldname)
  453. {
  454. arrays.append(*new CJsonWriterItem(fieldname));
  455. const char * sep = strchr(fieldname, '/');
  456. while (sep)
  457. {
  458. StringAttr leading(fieldname, sep-fieldname);
  459. appendJSONName(out, leading).append(" {");
  460. fieldname = sep+1;
  461. sep = strchr(fieldname, '/');
  462. }
  463. checkFormat(false, false, 1);
  464. appendJSONName(out, fieldname).append('[');
  465. }
  466. void CommonJsonWriter::outputEndArray(const char *fieldname)
  467. {
  468. arrays.pop();
  469. checkFormat(false, true, -1);
  470. out.append(']');
  471. const char * sep = (fieldname) ? strchr(fieldname, '/') : NULL;
  472. while (sep)
  473. {
  474. out.append('}');
  475. sep = strchr(sep+1, '/');
  476. }
  477. }
  478. void CommonJsonWriter::outputBeginDataset(const char *dsname, bool nestChildren)
  479. {
  480. if (dsname && *dsname)
  481. outputBeginNested(dsname, nestChildren);
  482. }
  483. void CommonJsonWriter::outputEndDataset(const char *dsname)
  484. {
  485. if (dsname && *dsname)
  486. outputEndNested(dsname);
  487. }
  488. void CommonJsonWriter::outputBeginNested(const char *fieldname, bool nestChildren)
  489. {
  490. if (!fieldname || !*fieldname)
  491. return;
  492. flush(false);
  493. checkFormat(true, false, 1);
  494. fieldname = checkItemNameBeginNested(fieldname);
  495. if (fieldname)
  496. {
  497. const char * sep = (fieldname) ? strchr(fieldname, '/') : NULL;
  498. while (sep)
  499. {
  500. StringAttr leading(fieldname, sep-fieldname);
  501. appendJSONName(out, leading).append("{");
  502. fieldname = sep+1;
  503. sep = strchr(fieldname, '/');
  504. }
  505. appendJSONName(out, fieldname);
  506. }
  507. out.append("{");
  508. if (!nestChildren && !nestLimit)
  509. nestLimit = indent;
  510. }
  511. void CommonJsonWriter::outputEndNested(const char *fieldname)
  512. {
  513. if (!fieldname || !*fieldname)
  514. return;
  515. flush(false);
  516. checkFormat(false, true, -1);
  517. fieldname = checkItemNameEndNested(fieldname);
  518. if (fieldname)
  519. {
  520. const char * sep = (fieldname) ? strchr(fieldname, '/') : NULL;
  521. while (sep)
  522. {
  523. out.append('}');
  524. sep = strchr(sep+1, '/');
  525. }
  526. }
  527. out.append("}");
  528. if (indent==nestLimit)
  529. nestLimit = 0;
  530. }
  531. void CommonJsonWriter::outputSetAll()
  532. {
  533. flush(false);
  534. checkDelimit();
  535. appendJSONValue(out, "All", true);
  536. }
  537. //=====================================================================================
  538. inline void outputEncodedXmlString(unsigned len, const char *field, const char *fieldname, StringBuffer &out)
  539. {
  540. if (fieldname)
  541. out.append('<').append(fieldname).append(" xsi:type=\"xsd:string\">");
  542. encodeXML(field, out, 0, len);
  543. if (fieldname)
  544. out.append("</").append(fieldname).append('>');
  545. }
  546. inline void outputEncodedXmlBool(bool field, const char *fieldname, StringBuffer &out)
  547. {
  548. const char * text = field ? "true" : "false";
  549. if (fieldname)
  550. out.append('<').append(fieldname).append(" xsi:type=\"xsd:boolean\">").append(text).append("</").append(fieldname).append('>');
  551. else
  552. out.append(text);
  553. }
  554. static char thorHelperhexchar[] = "0123456789ABCDEF";
  555. inline void outputEncodedXmlData(unsigned len, const void *_field, const char *fieldname, StringBuffer &out)
  556. {
  557. const unsigned char *field = (const unsigned char *) _field;
  558. if (fieldname)
  559. out.append('<').append(fieldname).append(" xsi:type=\"xsd:hexBinary\">");
  560. for (unsigned int i = 0; i < len; i++)
  561. {
  562. out.append(thorHelperhexchar[field[i] >> 4]).append(thorHelperhexchar[field[i] & 0x0f]);
  563. }
  564. if (fieldname)
  565. out.append("</").append(fieldname).append('>');
  566. }
  567. inline void outputEncoded64XmlData(unsigned len, const void *_field, const char *fieldname, StringBuffer &out)
  568. {
  569. if (fieldname)
  570. out.append('<').append(fieldname).append(" xsi:type=\"xsd:base64Binary\">");
  571. JBASE64_Encode(_field, len, out, false);
  572. if (fieldname)
  573. out.append("</").append(fieldname).append('>');
  574. }
  575. inline void outputEncodedXmlInt(__int64 field, const char *fieldname, StringBuffer &out)
  576. {
  577. if (fieldname)
  578. out.append('<').append(fieldname).append(" xsi:type=\"xsd:integer\">").append(field).append("</").append(fieldname).append('>');
  579. else
  580. out.append(field);
  581. }
  582. inline void outputEncodedXmlUInt(unsigned __int64 field, const char *fieldname, StringBuffer &out)
  583. {
  584. if (fieldname)
  585. out.append('<').append(fieldname).append(" xsi:type=\"xsd:nonNegativeInteger\">").append(field).append("</").append(fieldname).append('>');
  586. else
  587. out.append(field);
  588. }
  589. inline void outputEncodedXmlReal(double field, const char *fieldname, StringBuffer &out)
  590. {
  591. if (fieldname)
  592. out.append('<').append(fieldname).append(" xsi:type=\"xsd:double\">").append(field).append("</").append(fieldname).append('>');
  593. else
  594. out.append(field);
  595. }
  596. inline void outputEncodedXmlDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname, StringBuffer &out)
  597. {
  598. char dec[50];
  599. if (fieldname)
  600. out.append('<').append(fieldname).append(" xsi:type=\"xsd:decimal\">");
  601. DecLock();
  602. if (DecValid(true, size*2-1, field))
  603. {
  604. DecPushDecimal(field, size, precision);
  605. DecPopCString(sizeof(dec), dec);
  606. const char *finger = dec;
  607. while(isspace(*finger)) finger++;
  608. out.append(finger);
  609. }
  610. else
  611. out.append("####");
  612. DecUnlock();
  613. if (fieldname)
  614. out.append("</").append(fieldname).append('>');
  615. }
  616. inline void outputEncodedXmlUDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname, StringBuffer &out)
  617. {
  618. char dec[50];
  619. if (fieldname)
  620. out.append('<').append(fieldname).append(" xsi:type=\"xsd:decimal\">");
  621. DecLock();
  622. if (DecValid(false, size*2, field))
  623. {
  624. DecPushUDecimal(field, size, precision);
  625. DecPopCString(sizeof(dec), dec);
  626. const char *finger = dec;
  627. while(isspace(*finger)) finger++;
  628. out.append(finger);
  629. }
  630. else
  631. out.append("####");
  632. DecUnlock();
  633. if (fieldname)
  634. out.append("</").append(fieldname).append('>');
  635. }
  636. inline void outputEncodedXmlUnicode(unsigned len, const UChar *field, const char *fieldname, StringBuffer &out)
  637. {
  638. char * buff = 0;
  639. unsigned bufflen = 0;
  640. rtlUnicodeToCodepageX(bufflen, buff, len, field, "utf-8");
  641. if (fieldname)
  642. out.append('<').append(fieldname).append(" xsi:type=\"xsd:string\">");
  643. encodeXML(buff, out, 0, bufflen, true); // output as UTF-8
  644. if (fieldname)
  645. out.append("</").append(fieldname).append('>');
  646. rtlFree(buff);
  647. }
  648. inline void outputEncodedXmlUtf8(unsigned len, const char *field, const char *fieldname, StringBuffer &out)
  649. {
  650. if (fieldname)
  651. out.append('<').append(fieldname).append(" xsi:type=\"xsd:string\">");
  652. encodeXML(field, out, 0, rtlUtf8Size(len, field), true); // output as UTF-8
  653. if (fieldname)
  654. out.append("</").append(fieldname).append('>');
  655. }
  656. //=====================================================================================
  657. CommonEncodedXmlWriter::CommonEncodedXmlWriter(unsigned _flags, unsigned initialIndent, IXmlStreamFlusher *_flusher)
  658. : CommonXmlWriter(_flags, initialIndent, _flusher)
  659. {
  660. }
  661. void CommonEncodedXmlWriter::outputString(unsigned len, const char *field, const char *fieldname)
  662. {
  663. if (flags & XWFtrim)
  664. len = rtlTrimStrLen(len, field);
  665. if ((flags & XWFopt) && (rtlTrimStrLen(len, field) == 0))
  666. return;
  667. if (checkForAttribute(fieldname))
  668. outputXmlAttrString(len, field, fieldname+1, out);
  669. else
  670. {
  671. if (!nestLimit)
  672. out.pad(indent);
  673. outputEncodedXmlString(len, field, fieldname, out);
  674. if (!nestLimit)
  675. out.newline();
  676. }
  677. }
  678. void CommonEncodedXmlWriter::outputBool(bool field, const char *fieldname)
  679. {
  680. if (checkForAttribute(fieldname))
  681. outputXmlAttrBool(field, fieldname+1, out);
  682. else
  683. {
  684. if (!nestLimit)
  685. out.pad(indent);
  686. outputEncodedXmlBool(field, fieldname, out);
  687. if (!nestLimit)
  688. out.newline();
  689. }
  690. }
  691. void CommonEncodedXmlWriter::outputData(unsigned len, const void *field, const char *fieldname)
  692. {
  693. if (checkForAttribute(fieldname))
  694. outputXmlAttrData(len, field, fieldname+1, out);
  695. else
  696. {
  697. if (!nestLimit)
  698. out.pad(indent);
  699. outputEncodedXmlData(len, field, fieldname, out);
  700. if (!nestLimit)
  701. out.newline();
  702. }
  703. }
  704. void CommonEncodedXmlWriter::outputInt(__int64 field, const char *fieldname)
  705. {
  706. if (checkForAttribute(fieldname))
  707. outputXmlAttrInt(field, fieldname+1, out);
  708. else
  709. {
  710. if (!nestLimit)
  711. out.pad(indent);
  712. outputEncodedXmlInt(field, fieldname, out);
  713. if (!nestLimit)
  714. out.newline();
  715. }
  716. }
  717. void CommonEncodedXmlWriter::outputUInt(unsigned __int64 field, const char *fieldname)
  718. {
  719. if (checkForAttribute(fieldname))
  720. outputXmlAttrUInt(field, fieldname+1, out);
  721. else
  722. {
  723. if (!nestLimit)
  724. out.pad(indent);
  725. outputEncodedXmlUInt(field, fieldname, out);
  726. if (!nestLimit)
  727. out.newline();
  728. }
  729. }
  730. void CommonEncodedXmlWriter::outputReal(double field, const char *fieldname)
  731. {
  732. if (checkForAttribute(fieldname))
  733. outputXmlAttrReal(field, fieldname+1, out);
  734. else
  735. {
  736. if (!nestLimit)
  737. out.pad(indent);
  738. outputEncodedXmlReal(field, fieldname, out);
  739. if (!nestLimit)
  740. out.newline();
  741. }
  742. }
  743. void CommonEncodedXmlWriter::outputDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname)
  744. {
  745. if (checkForAttribute(fieldname))
  746. outputXmlAttrDecimal(field, size, precision, fieldname+1, out);
  747. else
  748. {
  749. if (!nestLimit)
  750. out.pad(indent);
  751. outputEncodedXmlDecimal(field, size, precision, fieldname, out);
  752. if (!nestLimit)
  753. out.newline();
  754. }
  755. }
  756. void CommonEncodedXmlWriter::outputUDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname)
  757. {
  758. if (checkForAttribute(fieldname))
  759. outputXmlAttrUDecimal(field, size, precision, fieldname+1, out);
  760. else
  761. {
  762. if (!nestLimit)
  763. out.pad(indent);
  764. outputEncodedXmlUDecimal(field, size, precision, fieldname, out);
  765. if (!nestLimit)
  766. out.newline();
  767. }
  768. }
  769. void CommonEncodedXmlWriter::outputUnicode(unsigned len, const UChar *field, const char *fieldname)
  770. {
  771. if (flags & XWFtrim)
  772. len = rtlTrimUnicodeStrLen(len, field);
  773. if ((flags & XWFopt) && (rtlTrimUnicodeStrLen(len, field) == 0))
  774. return;
  775. if (checkForAttribute(fieldname))
  776. outputXmlAttrUnicode(len, field, fieldname+1, out);
  777. else
  778. {
  779. if (!nestLimit)
  780. out.pad(indent);
  781. outputEncodedXmlUnicode(len, field, fieldname, out);
  782. if (!nestLimit)
  783. out.newline();
  784. }
  785. }
  786. void CommonEncodedXmlWriter::outputUtf8(unsigned len, const char *field, const char *fieldname)
  787. {
  788. if (flags & XWFtrim)
  789. len = rtlTrimUtf8StrLen(len, field);
  790. if ((flags & XWFopt) && (rtlTrimUtf8StrLen(len, field) == 0))
  791. return;
  792. if (checkForAttribute(fieldname))
  793. outputXmlAttrUtf8(len, field, fieldname+1, out);
  794. else
  795. {
  796. if (!nestLimit)
  797. out.pad(indent);
  798. outputEncodedXmlUtf8(len, field, fieldname, out);
  799. if (!nestLimit)
  800. out.newline();
  801. }
  802. }
  803. //=====================================================================================
  804. CommonEncoded64XmlWriter::CommonEncoded64XmlWriter(unsigned _flags, unsigned initialIndent, IXmlStreamFlusher *_flusher)
  805. : CommonEncodedXmlWriter(_flags, initialIndent, _flusher)
  806. {
  807. }
  808. void CommonEncoded64XmlWriter::outputData(unsigned len, const void *field, const char *fieldname)
  809. {
  810. if (checkForAttribute(fieldname))
  811. outputXmlAttrData(len, field, fieldname+1, out);
  812. else
  813. {
  814. if (!nestLimit)
  815. out.pad(indent);
  816. outputEncoded64XmlData(len, field, fieldname, out);
  817. if (!nestLimit)
  818. out.newline();
  819. }
  820. }
  821. //=====================================================================================
  822. CommonXmlWriter * CreateCommonXmlWriter(unsigned _flags, unsigned _initialIndent, IXmlStreamFlusher *_flusher, XMLWriterType xmlType)
  823. {
  824. switch (xmlType)
  825. {
  826. case WTStandard:
  827. return new CommonXmlWriter(_flags, _initialIndent, _flusher);//standard XML writer
  828. case WTEncodingData64:
  829. return new CommonEncoded64XmlWriter(_flags, _initialIndent, _flusher);//writes xsd type attributes, and all data as base64binary
  830. case WTEncoding:
  831. return new CommonEncodedXmlWriter(_flags, _initialIndent, _flusher);//writes xsd type attributes, and all data as hexBinary
  832. default:
  833. assertex(false);
  834. return NULL;
  835. }
  836. }
  837. //=====================================================================================
  838. IXmlWriter * createIXmlWriter(unsigned _flags, unsigned _initialIndent, IXmlStreamFlusher *_flusher, XMLWriterType xmlType)
  839. {
  840. if (xmlType==WTJSON)
  841. return new CommonJsonWriter(_flags, _initialIndent, _flusher);
  842. return CreateCommonXmlWriter(_flags, _initialIndent, _flusher, xmlType);
  843. }
  844. //=====================================================================================
  845. SimpleOutputWriter::SimpleOutputWriter()
  846. {
  847. separatorNeeded = false;
  848. }
  849. void SimpleOutputWriter::outputFieldSeparator()
  850. {
  851. if (separatorNeeded)
  852. out.append(',');
  853. separatorNeeded = true;
  854. }
  855. SimpleOutputWriter & SimpleOutputWriter::clear()
  856. {
  857. out.clear();
  858. separatorNeeded = false;
  859. return *this;
  860. }
  861. void SimpleOutputWriter::outputQuoted(const char *text)
  862. {
  863. out.append(text);
  864. }
  865. void SimpleOutputWriter::outputString(unsigned len, const char *field, const char *)
  866. {
  867. outputFieldSeparator();
  868. out.append(len, field);
  869. }
  870. void SimpleOutputWriter::outputQString(unsigned len, const char *field, const char *fieldname)
  871. {
  872. MemoryAttr tempBuffer;
  873. char * temp;
  874. if (len <= 100)
  875. temp = (char *)alloca(len);
  876. else
  877. temp = (char *)tempBuffer.allocate(len);
  878. rtlQStrToStr(len, temp, len, field);
  879. outputString(len, temp, fieldname);
  880. }
  881. void SimpleOutputWriter::outputBool(bool field, const char *)
  882. {
  883. outputFieldSeparator();
  884. outputXmlBool(field, NULL, out);
  885. }
  886. void SimpleOutputWriter::outputData(unsigned len, const void *field, const char *)
  887. {
  888. outputFieldSeparator();
  889. outputXmlData(len, field, NULL, out);
  890. }
  891. void SimpleOutputWriter::outputInt(__int64 field, const char *)
  892. {
  893. outputFieldSeparator();
  894. outputXmlInt(field, NULL, out);
  895. }
  896. void SimpleOutputWriter::outputUInt(unsigned __int64 field, const char *)
  897. {
  898. outputFieldSeparator();
  899. outputXmlUInt(field, NULL, out);
  900. }
  901. void SimpleOutputWriter::outputReal(double field, const char *)
  902. {
  903. outputFieldSeparator();
  904. outputXmlReal(field, NULL, out);
  905. }
  906. void SimpleOutputWriter::outputDecimal(const void *field, unsigned size, unsigned precision, const char *)
  907. {
  908. outputFieldSeparator();
  909. outputXmlDecimal(field, size, precision, NULL, out);
  910. }
  911. void SimpleOutputWriter::outputUDecimal(const void *field, unsigned size, unsigned precision, const char *)
  912. {
  913. outputFieldSeparator();
  914. outputXmlUDecimal(field, size, precision, NULL, out);
  915. }
  916. void SimpleOutputWriter::outputUnicode(unsigned len, const UChar *field, const char *)
  917. {
  918. outputFieldSeparator();
  919. outputXmlUnicode(len, field, NULL, out);
  920. }
  921. void SimpleOutputWriter::outputUtf8(unsigned len, const char *field, const char *)
  922. {
  923. outputFieldSeparator();
  924. outputXmlUtf8(len, field, NULL, out);
  925. }
  926. void SimpleOutputWriter::outputBeginNested(const char *s, bool)
  927. {
  928. if (!s || !*s)
  929. return;
  930. outputFieldSeparator();
  931. out.append('[');
  932. separatorNeeded = false;
  933. }
  934. void SimpleOutputWriter::outputEndNested(const char *s)
  935. {
  936. if (!s || !*s)
  937. return;
  938. out.append(']');
  939. separatorNeeded = true;
  940. }
  941. void SimpleOutputWriter::outputSetAll()
  942. {
  943. out.append('*');
  944. }
  945. void SimpleOutputWriter::newline()
  946. {
  947. out.append('\n');
  948. }
  949. //=====================================================================================
  950. CommonFieldProcessor::CommonFieldProcessor(StringBuffer &_result, bool _trim) : result(_result), trim(_trim)
  951. {
  952. }
  953. void CommonFieldProcessor::processString(unsigned len, const char *value, const RtlFieldInfo * field)
  954. {
  955. if (trim)
  956. len = rtlTrimStrLen(len, value);
  957. result.append("'");
  958. outputXmlString(len, value, NULL, result);
  959. result.append("'");
  960. }
  961. void CommonFieldProcessor::processBool(bool value, const RtlFieldInfo * field)
  962. {
  963. outputXmlBool(value, NULL, result);
  964. }
  965. void CommonFieldProcessor::processData(unsigned len, const void *value, const RtlFieldInfo * field)
  966. {
  967. outputXmlData(len, value, NULL, result);
  968. }
  969. void CommonFieldProcessor::processInt(__int64 value, const RtlFieldInfo * field)
  970. {
  971. outputXmlInt(value, NULL, result);
  972. }
  973. void CommonFieldProcessor::processUInt(unsigned __int64 value, const RtlFieldInfo * field)
  974. {
  975. outputXmlUInt(value, NULL, result);
  976. }
  977. void CommonFieldProcessor::processReal(double value, const RtlFieldInfo * field)
  978. {
  979. outputXmlReal(value, NULL, result);
  980. }
  981. void CommonFieldProcessor::processDecimal(const void *value, unsigned digits, unsigned precision, const RtlFieldInfo * field)
  982. {
  983. outputXmlDecimal(value, digits, precision, NULL, result);
  984. }
  985. void CommonFieldProcessor::processUDecimal(const void *value, unsigned digits, unsigned precision, const RtlFieldInfo * field)
  986. {
  987. outputXmlUDecimal(value, digits, precision, NULL, result);
  988. }
  989. void CommonFieldProcessor::processUnicode(unsigned len, const UChar *value, const RtlFieldInfo * field)
  990. {
  991. if (trim)
  992. len = rtlTrimUnicodeStrLen(len, value);
  993. outputXmlUnicode(len, value, NULL, result);
  994. }
  995. void CommonFieldProcessor::processQString(unsigned len, const char *value, const RtlFieldInfo * field)
  996. {
  997. MemoryAttr tempBuffer;
  998. char * temp;
  999. if (len <= 100)
  1000. temp = (char *)alloca(len);
  1001. else
  1002. temp = (char *)tempBuffer.allocate(len);
  1003. rtlQStrToStr(len, temp, len, value);
  1004. processString(len, temp, field);
  1005. }
  1006. void CommonFieldProcessor::processUtf8(unsigned len, const char *value, const RtlFieldInfo * field)
  1007. {
  1008. if (trim)
  1009. len = rtlTrimUtf8StrLen(len, value);
  1010. outputXmlUtf8(len, value, NULL, result);
  1011. }
  1012. bool CommonFieldProcessor::processBeginSet(const RtlFieldInfo * field, unsigned numElements, bool isAll, const byte *data)
  1013. {
  1014. result.append('[');
  1015. if (isAll)
  1016. result.append("ALL");
  1017. return true;
  1018. }
  1019. bool CommonFieldProcessor::processBeginDataset(const RtlFieldInfo * field, unsigned numRows)
  1020. {
  1021. result.append('[');
  1022. return true;
  1023. }
  1024. bool CommonFieldProcessor::processBeginRow(const RtlFieldInfo * field)
  1025. {
  1026. result.append('{');
  1027. return true;
  1028. }
  1029. void CommonFieldProcessor::processEndSet(const RtlFieldInfo * field)
  1030. {
  1031. result.append(']');
  1032. }
  1033. void CommonFieldProcessor::processEndDataset(const RtlFieldInfo * field)
  1034. {
  1035. result.append(']');
  1036. }
  1037. void CommonFieldProcessor::processEndRow(const RtlFieldInfo * field)
  1038. {
  1039. result.append('}');
  1040. }
  1041. //=============================================================================================
  1042. void printKeyedValues(StringBuffer &out, IIndexReadContext *segs, IOutputMetaData *rowMeta)
  1043. {
  1044. unsigned totalKeyedSize = 0;
  1045. unsigned numSegs = segs->ordinality();
  1046. while (numSegs)
  1047. {
  1048. IKeySegmentMonitor &seg = *segs->item(numSegs-1);
  1049. if (!seg.isWild())
  1050. {
  1051. totalKeyedSize = seg.getOffset() + seg.getSize();
  1052. break;
  1053. }
  1054. numSegs--;
  1055. }
  1056. if (numSegs)
  1057. {
  1058. byte *tempRow = (byte *) alloca(totalKeyedSize);
  1059. byte *savedRow = (byte *) alloca(totalKeyedSize);
  1060. const RtlFieldInfo * const *fields = rowMeta->queryTypeInfo()->queryFields();
  1061. unsigned fieldOffset = 0;
  1062. bool inKeyed = false;
  1063. bool inWild = false;
  1064. for (unsigned segNo = 0; segNo < numSegs; segNo++)
  1065. {
  1066. IKeySegmentMonitor &seg = *segs->item(segNo);
  1067. unsigned segOffset = seg.getOffset();
  1068. unsigned segSize = seg.getSize();
  1069. while (fieldOffset < segOffset + segSize) // This is trying to cope with the combined case but not sure it completely does
  1070. {
  1071. assertex(fields[0]->type->isFixedSize());
  1072. unsigned curFieldSize = fields[0]->type->size(NULL, NULL);
  1073. if (seg.isWild())
  1074. {
  1075. if (!inWild)
  1076. {
  1077. if (inKeyed)
  1078. {
  1079. out.append("),");
  1080. inKeyed = false;
  1081. }
  1082. out.append("WILD(");
  1083. inWild = true;
  1084. }
  1085. else
  1086. out.append(',');
  1087. out.append(fields[0]->name);
  1088. }
  1089. else
  1090. {
  1091. StringBuffer setValues;
  1092. CommonFieldProcessor setProcessor(setValues, true);
  1093. unsigned numValues = 0;
  1094. unsigned subStringLength = 0;
  1095. if (!seg.isEmpty())
  1096. {
  1097. seg.setLow(tempRow);
  1098. loop
  1099. {
  1100. if (numValues)
  1101. setValues.append(",");
  1102. memcpy(savedRow+segOffset, tempRow+segOffset, segSize);
  1103. seg.endRange(tempRow);
  1104. if (memcmp(savedRow+segOffset, tempRow+segOffset, segSize) != 0)
  1105. {
  1106. // Special case - if they differ only in trailing values that are 0 vs 0xff, then it's a substring match...
  1107. if (numValues==0 && (fields[0]->type->fieldType & (RFTMkind | RFTMebcdic)) == type_string)
  1108. {
  1109. unsigned pos;
  1110. for (pos = 0; pos < segSize; pos++)
  1111. {
  1112. if (savedRow[segOffset+pos] != tempRow[segOffset+pos])
  1113. break;
  1114. }
  1115. subStringLength = pos;
  1116. for (; pos < segSize; pos++)
  1117. {
  1118. if (savedRow[segOffset+pos] != 0 || tempRow[segOffset+pos] != 0xff)
  1119. {
  1120. subStringLength = 0;
  1121. break;
  1122. }
  1123. }
  1124. }
  1125. fields[0]->process(savedRow + fieldOffset, tempRow, setProcessor);
  1126. setValues.append("..");
  1127. fields[0]->process(tempRow + fieldOffset, tempRow, setProcessor);
  1128. numValues+=2;
  1129. }
  1130. else
  1131. {
  1132. fields[0]->process(tempRow + fieldOffset, tempRow, setProcessor);
  1133. numValues++;
  1134. }
  1135. if (!seg.increment(tempRow))
  1136. break;
  1137. }
  1138. }
  1139. if (!inKeyed)
  1140. {
  1141. if (inWild)
  1142. {
  1143. out.append("),");
  1144. inWild = false;
  1145. }
  1146. out.append("KEYED(");
  1147. inKeyed = true;
  1148. }
  1149. else
  1150. out.append(',');
  1151. out.append(fields[0]->name);
  1152. if (numValues==1)
  1153. out.append("=").append(setValues);
  1154. else if (subStringLength)
  1155. out.appendf("[1..%d]='", subStringLength).append(subStringLength, (char *) savedRow+fieldOffset).append("'");
  1156. else
  1157. out.append(" IN [").append(setValues).append("]");
  1158. }
  1159. fieldOffset += curFieldSize;
  1160. fields++;
  1161. if (!fields[0])
  1162. break;
  1163. }
  1164. }
  1165. if (inKeyed || inWild)
  1166. out.append(")");
  1167. }
  1168. else
  1169. out.append("UNKEYED");
  1170. }
  1171. extern thorhelper_decl void convertRowToXML(size32_t & lenResult, char * & result, IOutputMetaData & info, const void * row, unsigned flags)
  1172. {
  1173. const byte * self = (const byte *)row;
  1174. if (flags == (unsigned)-1)
  1175. flags = XWFtrim|XWFopt|XWFnoindent;
  1176. CommonXmlWriter writer(flags);
  1177. info.toXML(self, writer);
  1178. //could use detach...
  1179. unsigned sizeResult;
  1180. rtlStrToStrX(sizeResult, result, writer.length(), writer.str());
  1181. lenResult = rtlUtf8Length(sizeResult, result);
  1182. }