thorxmlwrite.cpp 30 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001
  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::outputBeginNested(const char *fieldname, bool nestChildren)
  220. {
  221. const char * sep = strchr(fieldname, '/');
  222. if (sep)
  223. {
  224. StringAttr leading(fieldname, sep-fieldname);
  225. outputBeginNested(leading, nestChildren);
  226. outputBeginNested(sep+1, nestChildren);
  227. return;
  228. }
  229. closeTag();
  230. if (!nestLimit)
  231. out.pad(indent);
  232. out.append('<').append(fieldname);
  233. indent += 1;
  234. if (!nestChildren && !nestLimit)
  235. nestLimit = indent;
  236. tagClosed = false;
  237. }
  238. void CommonXmlWriter::outputEndNested(const char *fieldname)
  239. {
  240. const char * sep = strchr(fieldname, '/');
  241. if (sep)
  242. {
  243. StringAttr leading(fieldname, sep-fieldname);
  244. outputEndNested(sep+1);
  245. outputEndNested(leading);
  246. return;
  247. }
  248. if (flags & XWFexpandempty)
  249. closeTag();
  250. if (!tagClosed)
  251. {
  252. out.append("/>");
  253. tagClosed = true;
  254. }
  255. else
  256. {
  257. if (!nestLimit)
  258. out.pad(indent-1);
  259. out.append("</").append(fieldname).append('>');
  260. }
  261. if (indent==nestLimit)
  262. nestLimit = 0;
  263. indent -= 1;
  264. if (!nestLimit)
  265. out.newline();
  266. }
  267. void CommonXmlWriter::outputSetAll()
  268. {
  269. closeTag();
  270. if (!nestLimit)
  271. out.pad(indent);
  272. outputXmlSetAll(out);
  273. if (!nestLimit)
  274. out.newline();
  275. }
  276. //=====================================================================================
  277. inline void outputEncodedXmlString(unsigned len, const char *field, const char *fieldname, StringBuffer &out)
  278. {
  279. if (fieldname)
  280. out.append('<').append(fieldname).append(" xsi:type=\"xsd:string\">");
  281. encodeXML(field, out, 0, len);
  282. if (fieldname)
  283. out.append("</").append(fieldname).append('>');
  284. }
  285. inline void outputEncodedXmlBool(bool field, const char *fieldname, StringBuffer &out)
  286. {
  287. const char * text = field ? "true" : "false";
  288. if (fieldname)
  289. out.append('<').append(fieldname).append(" xsi:type=\"xsd:boolean\">").append(text).append("</").append(fieldname).append('>');
  290. else
  291. out.append(text);
  292. }
  293. static char thorHelperhexchar[] = "0123456789ABCDEF";
  294. inline void outputEncodedXmlData(unsigned len, const void *_field, const char *fieldname, StringBuffer &out)
  295. {
  296. const unsigned char *field = (const unsigned char *) _field;
  297. if (fieldname)
  298. out.append('<').append(fieldname).append(" xsi:type=\"xsd:hexBinary\">");
  299. for (unsigned int i = 0; i < len; i++)
  300. {
  301. out.append(thorHelperhexchar[field[i] >> 4]).append(thorHelperhexchar[field[i] & 0x0f]);
  302. }
  303. if (fieldname)
  304. out.append("</").append(fieldname).append('>');
  305. }
  306. inline void outputEncoded64XmlData(unsigned len, const void *_field, const char *fieldname, StringBuffer &out)
  307. {
  308. if (fieldname)
  309. out.append('<').append(fieldname).append(" xsi:type=\"xsd:base64Binary\">");
  310. JBASE64_Encode(_field, len, out, false);
  311. if (fieldname)
  312. out.append("</").append(fieldname).append('>');
  313. }
  314. inline void outputEncodedXmlInt(__int64 field, const char *fieldname, StringBuffer &out)
  315. {
  316. if (fieldname)
  317. out.append('<').append(fieldname).append(" xsi:type=\"xsd:integer\">").append(field).append("</").append(fieldname).append('>');
  318. else
  319. out.append(field);
  320. }
  321. inline void outputEncodedXmlUInt(unsigned __int64 field, const char *fieldname, StringBuffer &out)
  322. {
  323. if (fieldname)
  324. out.append('<').append(fieldname).append(" xsi:type=\"xsd:nonNegativeInteger\">").append(field).append("</").append(fieldname).append('>');
  325. else
  326. out.append(field);
  327. }
  328. inline void outputEncodedXmlReal(double field, const char *fieldname, StringBuffer &out)
  329. {
  330. if (fieldname)
  331. out.append('<').append(fieldname).append(" xsi:type=\"xsd:double\">").append(field).append("</").append(fieldname).append('>');
  332. else
  333. out.append(field);
  334. }
  335. inline void outputEncodedXmlDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname, StringBuffer &out)
  336. {
  337. char dec[50];
  338. if (fieldname)
  339. out.append('<').append(fieldname).append(" xsi:type=\"xsd:decimal\">");
  340. DecLock();
  341. if (DecValid(true, size*2-1, field))
  342. {
  343. DecPushDecimal(field, size, precision);
  344. DecPopCString(sizeof(dec), dec);
  345. const char *finger = dec;
  346. while(isspace(*finger)) finger++;
  347. out.append(finger);
  348. }
  349. else
  350. out.append("####");
  351. DecUnlock();
  352. if (fieldname)
  353. out.append("</").append(fieldname).append('>');
  354. }
  355. inline void outputEncodedXmlUDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname, StringBuffer &out)
  356. {
  357. char dec[50];
  358. if (fieldname)
  359. out.append('<').append(fieldname).append(" xsi:type=\"xsd:decimal\">");
  360. DecLock();
  361. if (DecValid(false, size*2, field))
  362. {
  363. DecPushUDecimal(field, size, precision);
  364. DecPopCString(sizeof(dec), dec);
  365. const char *finger = dec;
  366. while(isspace(*finger)) finger++;
  367. out.append(finger);
  368. }
  369. else
  370. out.append("####");
  371. DecUnlock();
  372. if (fieldname)
  373. out.append("</").append(fieldname).append('>');
  374. }
  375. inline void outputEncodedXmlUnicode(unsigned len, const UChar *field, const char *fieldname, StringBuffer &out)
  376. {
  377. char * buff = 0;
  378. unsigned bufflen = 0;
  379. rtlUnicodeToCodepageX(bufflen, buff, len, field, "utf-8");
  380. if (fieldname)
  381. out.append('<').append(fieldname).append(" xsi:type=\"xsd:string\">");
  382. encodeXML(buff, out, 0, bufflen, true); // output as UTF-8
  383. if (fieldname)
  384. out.append("</").append(fieldname).append('>');
  385. rtlFree(buff);
  386. }
  387. inline void outputEncodedXmlUtf8(unsigned len, const char *field, const char *fieldname, StringBuffer &out)
  388. {
  389. if (fieldname)
  390. out.append('<').append(fieldname).append(" xsi:type=\"xsd:string\">");
  391. encodeXML(field, out, 0, rtlUtf8Size(len, field), true); // output as UTF-8
  392. if (fieldname)
  393. out.append("</").append(fieldname).append('>');
  394. }
  395. //=====================================================================================
  396. CommonEncodedXmlWriter::CommonEncodedXmlWriter(unsigned _flags, unsigned initialIndent, IXmlStreamFlusher *_flusher)
  397. : CommonXmlWriter(_flags, initialIndent, _flusher)
  398. {
  399. }
  400. void CommonEncodedXmlWriter::outputString(unsigned len, const char *field, const char *fieldname)
  401. {
  402. if (flags & XWFtrim)
  403. len = rtlTrimStrLen(len, field);
  404. if ((flags & XWFopt) && (rtlTrimStrLen(len, field) == 0))
  405. return;
  406. if (checkForAttribute(fieldname))
  407. outputXmlAttrString(len, field, fieldname+1, out);
  408. else
  409. {
  410. if (!nestLimit)
  411. out.pad(indent);
  412. outputEncodedXmlString(len, field, fieldname, out);
  413. if (!nestLimit)
  414. out.newline();
  415. }
  416. }
  417. void CommonEncodedXmlWriter::outputBool(bool field, const char *fieldname)
  418. {
  419. if (checkForAttribute(fieldname))
  420. outputXmlAttrBool(field, fieldname+1, out);
  421. else
  422. {
  423. if (!nestLimit)
  424. out.pad(indent);
  425. outputEncodedXmlBool(field, fieldname, out);
  426. if (!nestLimit)
  427. out.newline();
  428. }
  429. }
  430. void CommonEncodedXmlWriter::outputData(unsigned len, const void *field, const char *fieldname)
  431. {
  432. if (checkForAttribute(fieldname))
  433. outputXmlAttrData(len, field, fieldname+1, out);
  434. else
  435. {
  436. if (!nestLimit)
  437. out.pad(indent);
  438. outputEncodedXmlData(len, field, fieldname, out);
  439. if (!nestLimit)
  440. out.newline();
  441. }
  442. }
  443. void CommonEncodedXmlWriter::outputInt(__int64 field, const char *fieldname)
  444. {
  445. if (checkForAttribute(fieldname))
  446. outputXmlAttrInt(field, fieldname+1, out);
  447. else
  448. {
  449. if (!nestLimit)
  450. out.pad(indent);
  451. outputEncodedXmlInt(field, fieldname, out);
  452. if (!nestLimit)
  453. out.newline();
  454. }
  455. }
  456. void CommonEncodedXmlWriter::outputUInt(unsigned __int64 field, const char *fieldname)
  457. {
  458. if (checkForAttribute(fieldname))
  459. outputXmlAttrUInt(field, fieldname+1, out);
  460. else
  461. {
  462. if (!nestLimit)
  463. out.pad(indent);
  464. outputEncodedXmlUInt(field, fieldname, out);
  465. if (!nestLimit)
  466. out.newline();
  467. }
  468. }
  469. void CommonEncodedXmlWriter::outputReal(double field, const char *fieldname)
  470. {
  471. if (checkForAttribute(fieldname))
  472. outputXmlAttrReal(field, fieldname+1, out);
  473. else
  474. {
  475. if (!nestLimit)
  476. out.pad(indent);
  477. outputEncodedXmlReal(field, fieldname, out);
  478. if (!nestLimit)
  479. out.newline();
  480. }
  481. }
  482. void CommonEncodedXmlWriter::outputDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname)
  483. {
  484. if (checkForAttribute(fieldname))
  485. outputXmlAttrDecimal(field, size, precision, fieldname+1, out);
  486. else
  487. {
  488. if (!nestLimit)
  489. out.pad(indent);
  490. outputEncodedXmlDecimal(field, size, precision, fieldname, out);
  491. if (!nestLimit)
  492. out.newline();
  493. }
  494. }
  495. void CommonEncodedXmlWriter::outputUDecimal(const void *field, unsigned size, unsigned precision, const char *fieldname)
  496. {
  497. if (checkForAttribute(fieldname))
  498. outputXmlAttrUDecimal(field, size, precision, fieldname+1, out);
  499. else
  500. {
  501. if (!nestLimit)
  502. out.pad(indent);
  503. outputEncodedXmlUDecimal(field, size, precision, fieldname, out);
  504. if (!nestLimit)
  505. out.newline();
  506. }
  507. }
  508. void CommonEncodedXmlWriter::outputUnicode(unsigned len, const UChar *field, const char *fieldname)
  509. {
  510. if (flags & XWFtrim)
  511. len = rtlTrimUnicodeStrLen(len, field);
  512. if ((flags & XWFopt) && (rtlTrimUnicodeStrLen(len, field) == 0))
  513. return;
  514. if (checkForAttribute(fieldname))
  515. outputXmlAttrUnicode(len, field, fieldname+1, out);
  516. else
  517. {
  518. if (!nestLimit)
  519. out.pad(indent);
  520. outputEncodedXmlUnicode(len, field, fieldname, out);
  521. if (!nestLimit)
  522. out.newline();
  523. }
  524. }
  525. void CommonEncodedXmlWriter::outputUtf8(unsigned len, const char *field, const char *fieldname)
  526. {
  527. if (flags & XWFtrim)
  528. len = rtlTrimUtf8StrLen(len, field);
  529. if ((flags & XWFopt) && (rtlTrimUtf8StrLen(len, field) == 0))
  530. return;
  531. if (checkForAttribute(fieldname))
  532. outputXmlAttrUtf8(len, field, fieldname+1, out);
  533. else
  534. {
  535. if (!nestLimit)
  536. out.pad(indent);
  537. outputEncodedXmlUtf8(len, field, fieldname, out);
  538. if (!nestLimit)
  539. out.newline();
  540. }
  541. }
  542. //=====================================================================================
  543. CommonEncoded64XmlWriter::CommonEncoded64XmlWriter(unsigned _flags, unsigned initialIndent, IXmlStreamFlusher *_flusher)
  544. : CommonEncodedXmlWriter(_flags, initialIndent, _flusher)
  545. {
  546. }
  547. void CommonEncoded64XmlWriter::outputData(unsigned len, const void *field, const char *fieldname)
  548. {
  549. if (checkForAttribute(fieldname))
  550. outputXmlAttrData(len, field, fieldname+1, out);
  551. else
  552. {
  553. if (!nestLimit)
  554. out.pad(indent);
  555. outputEncoded64XmlData(len, field, fieldname, out);
  556. if (!nestLimit)
  557. out.newline();
  558. }
  559. }
  560. //=====================================================================================
  561. CommonXmlWriter * CreateCommonXmlWriter(unsigned _flags, unsigned _initialIndent, IXmlStreamFlusher *_flusher, XMLWriterType xmlType)
  562. {
  563. switch (xmlType)
  564. {
  565. case WTStandard:
  566. return new CommonXmlWriter(_flags, _initialIndent, _flusher);//standard XML writer
  567. case WTEncodingData64:
  568. return new CommonEncoded64XmlWriter(_flags, _initialIndent, _flusher);//writes xsd type attributes, and all data as base64binary
  569. case WTEncoding:
  570. return new CommonEncodedXmlWriter(_flags, _initialIndent, _flusher);//writes xsd type attributes, and all data as hexBinary
  571. default:
  572. assertex(false);
  573. return NULL;
  574. }
  575. }
  576. //=====================================================================================
  577. SimpleOutputWriter::SimpleOutputWriter()
  578. {
  579. separatorNeeded = false;
  580. }
  581. void SimpleOutputWriter::outputFieldSeparator()
  582. {
  583. if (separatorNeeded)
  584. out.append(',');
  585. separatorNeeded = true;
  586. }
  587. SimpleOutputWriter & SimpleOutputWriter::clear()
  588. {
  589. out.clear();
  590. separatorNeeded = false;
  591. return *this;
  592. }
  593. void SimpleOutputWriter::outputQuoted(const char *text)
  594. {
  595. out.append(text);
  596. }
  597. void SimpleOutputWriter::outputString(unsigned len, const char *field, const char *)
  598. {
  599. outputFieldSeparator();
  600. out.append(len, field);
  601. }
  602. void SimpleOutputWriter::outputQString(unsigned len, const char *field, const char *fieldname)
  603. {
  604. MemoryAttr tempBuffer;
  605. char * temp;
  606. if (len <= 100)
  607. temp = (char *)alloca(len);
  608. else
  609. temp = (char *)tempBuffer.allocate(len);
  610. rtlQStrToStr(len, temp, len, field);
  611. outputString(len, temp, fieldname);
  612. }
  613. void SimpleOutputWriter::outputBool(bool field, const char *)
  614. {
  615. outputFieldSeparator();
  616. outputXmlBool(field, NULL, out);
  617. }
  618. void SimpleOutputWriter::outputData(unsigned len, const void *field, const char *)
  619. {
  620. outputFieldSeparator();
  621. outputXmlData(len, field, NULL, out);
  622. }
  623. void SimpleOutputWriter::outputInt(__int64 field, const char *)
  624. {
  625. outputFieldSeparator();
  626. outputXmlInt(field, NULL, out);
  627. }
  628. void SimpleOutputWriter::outputUInt(unsigned __int64 field, const char *)
  629. {
  630. outputFieldSeparator();
  631. outputXmlUInt(field, NULL, out);
  632. }
  633. void SimpleOutputWriter::outputReal(double field, const char *)
  634. {
  635. outputFieldSeparator();
  636. outputXmlReal(field, NULL, out);
  637. }
  638. void SimpleOutputWriter::outputDecimal(const void *field, unsigned size, unsigned precision, const char *)
  639. {
  640. outputFieldSeparator();
  641. outputXmlDecimal(field, size, precision, NULL, out);
  642. }
  643. void SimpleOutputWriter::outputUDecimal(const void *field, unsigned size, unsigned precision, const char *)
  644. {
  645. outputFieldSeparator();
  646. outputXmlUDecimal(field, size, precision, NULL, out);
  647. }
  648. void SimpleOutputWriter::outputUnicode(unsigned len, const UChar *field, const char *)
  649. {
  650. outputFieldSeparator();
  651. outputXmlUnicode(len, field, NULL, out);
  652. }
  653. void SimpleOutputWriter::outputUtf8(unsigned len, const char *field, const char *)
  654. {
  655. outputFieldSeparator();
  656. outputXmlUtf8(len, field, NULL, out);
  657. }
  658. void SimpleOutputWriter::outputBeginNested(const char *, bool)
  659. {
  660. outputFieldSeparator();
  661. out.append('[');
  662. separatorNeeded = false;
  663. }
  664. void SimpleOutputWriter::outputEndNested(const char *)
  665. {
  666. out.append(']');
  667. separatorNeeded = true;
  668. }
  669. void SimpleOutputWriter::outputSetAll()
  670. {
  671. out.append('*');
  672. }
  673. void SimpleOutputWriter::newline()
  674. {
  675. out.append('\n');
  676. }
  677. //=====================================================================================
  678. CommonFieldProcessor::CommonFieldProcessor(StringBuffer &_result, bool _trim) : result(_result), trim(_trim)
  679. {
  680. }
  681. void CommonFieldProcessor::processString(unsigned len, const char *value, const RtlFieldInfo * field)
  682. {
  683. if (trim)
  684. len = rtlTrimStrLen(len, value);
  685. result.append("'");
  686. outputXmlString(len, value, NULL, result);
  687. result.append("'");
  688. }
  689. void CommonFieldProcessor::processBool(bool value, const RtlFieldInfo * field)
  690. {
  691. outputXmlBool(value, NULL, result);
  692. }
  693. void CommonFieldProcessor::processData(unsigned len, const void *value, const RtlFieldInfo * field)
  694. {
  695. outputXmlData(len, value, NULL, result);
  696. }
  697. void CommonFieldProcessor::processInt(__int64 value, const RtlFieldInfo * field)
  698. {
  699. outputXmlInt(value, NULL, result);
  700. }
  701. void CommonFieldProcessor::processUInt(unsigned __int64 value, const RtlFieldInfo * field)
  702. {
  703. outputXmlUInt(value, NULL, result);
  704. }
  705. void CommonFieldProcessor::processReal(double value, const RtlFieldInfo * field)
  706. {
  707. outputXmlReal(value, NULL, result);
  708. }
  709. void CommonFieldProcessor::processDecimal(const void *value, unsigned digits, unsigned precision, const RtlFieldInfo * field)
  710. {
  711. outputXmlDecimal(value, digits, precision, NULL, result);
  712. }
  713. void CommonFieldProcessor::processUDecimal(const void *value, unsigned digits, unsigned precision, const RtlFieldInfo * field)
  714. {
  715. outputXmlUDecimal(value, digits, precision, NULL, result);
  716. }
  717. void CommonFieldProcessor::processUnicode(unsigned len, const UChar *value, const RtlFieldInfo * field)
  718. {
  719. if (trim)
  720. len = rtlTrimUnicodeStrLen(len, value);
  721. outputXmlUnicode(len, value, NULL, result);
  722. }
  723. void CommonFieldProcessor::processQString(unsigned len, const char *value, const RtlFieldInfo * field)
  724. {
  725. MemoryAttr tempBuffer;
  726. char * temp;
  727. if (len <= 100)
  728. temp = (char *)alloca(len);
  729. else
  730. temp = (char *)tempBuffer.allocate(len);
  731. rtlQStrToStr(len, temp, len, value);
  732. processString(len, temp, field);
  733. }
  734. void CommonFieldProcessor::processSetAll(const RtlFieldInfo * field)
  735. {
  736. result.append("ALL");
  737. }
  738. void CommonFieldProcessor::processUtf8(unsigned len, const char *value, const RtlFieldInfo * field)
  739. {
  740. if (trim)
  741. len = rtlTrimUtf8StrLen(len, value);
  742. outputXmlUtf8(len, value, NULL, result);
  743. }
  744. bool CommonFieldProcessor::processBeginSet(const RtlFieldInfo * field)
  745. {
  746. result.append('[');
  747. return true;
  748. }
  749. bool CommonFieldProcessor::processBeginDataset(const RtlFieldInfo * field)
  750. {
  751. result.append('[');
  752. return true;
  753. }
  754. bool CommonFieldProcessor::processBeginRow(const RtlFieldInfo * field)
  755. {
  756. result.append('{');
  757. return true;
  758. }
  759. void CommonFieldProcessor::processEndSet(const RtlFieldInfo * field)
  760. {
  761. result.append(']');
  762. }
  763. void CommonFieldProcessor::processEndDataset(const RtlFieldInfo * field)
  764. {
  765. result.append(']');
  766. }
  767. void CommonFieldProcessor::processEndRow(const RtlFieldInfo * field)
  768. {
  769. result.append('}');
  770. }
  771. //=============================================================================================
  772. void printKeyedValues(StringBuffer &out, IIndexReadContext *segs, IOutputMetaData *rowMeta)
  773. {
  774. unsigned totalKeyedSize = 0;
  775. unsigned numSegs = segs->ordinality();
  776. while (numSegs)
  777. {
  778. IKeySegmentMonitor &seg = *segs->item(numSegs-1);
  779. if (!seg.isWild())
  780. {
  781. totalKeyedSize = seg.getOffset() + seg.getSize();
  782. break;
  783. }
  784. numSegs--;
  785. }
  786. if (numSegs)
  787. {
  788. byte *tempRow = (byte *) alloca(totalKeyedSize);
  789. byte *savedRow = (byte *) alloca(totalKeyedSize);
  790. const RtlFieldInfo * const *fields = rowMeta->queryTypeInfo()->queryFields();
  791. unsigned fieldOffset = 0;
  792. bool inKeyed = false;
  793. bool inWild = false;
  794. for (unsigned segNo = 0; segNo < numSegs; segNo++)
  795. {
  796. IKeySegmentMonitor &seg = *segs->item(segNo);
  797. unsigned segOffset = seg.getOffset();
  798. unsigned segSize = seg.getSize();
  799. while (fieldOffset < segOffset + segSize) // This is trying to cope with the combined case but not sure it completely does
  800. {
  801. assertex(fields[0]->type->isFixedSize());
  802. unsigned curFieldSize = fields[0]->type->size(NULL, NULL);
  803. if (seg.isWild())
  804. {
  805. if (!inWild)
  806. {
  807. if (inKeyed)
  808. {
  809. out.append("),");
  810. inKeyed = false;
  811. }
  812. out.append("WILD(");
  813. inWild = true;
  814. }
  815. else
  816. out.append(',');
  817. out.append(fields[0]->name);
  818. }
  819. else
  820. {
  821. StringBuffer setValues;
  822. CommonFieldProcessor setProcessor(setValues, true);
  823. unsigned numValues = 0;
  824. unsigned subStringLength = 0;
  825. if (!seg.isEmpty())
  826. {
  827. seg.setLow(tempRow);
  828. loop
  829. {
  830. if (numValues)
  831. setValues.append(",");
  832. memcpy(savedRow+segOffset, tempRow+segOffset, segSize);
  833. seg.endRange(tempRow);
  834. if (memcmp(savedRow+segOffset, tempRow+segOffset, segSize) != 0)
  835. {
  836. // Special case - if they differ only in trailing values that are 0 vs 0xff, then it's a substring match...
  837. if (numValues==0 && (fields[0]->type->fieldType & (RFTMkind | RFTMebcdic)) == type_string)
  838. {
  839. unsigned pos;
  840. for (pos = 0; pos < segSize; pos++)
  841. {
  842. if (savedRow[segOffset+pos] != tempRow[segOffset+pos])
  843. break;
  844. }
  845. subStringLength = pos;
  846. for (; pos < segSize; pos++)
  847. {
  848. if (savedRow[segOffset+pos] != 0 || tempRow[segOffset+pos] != 0xff)
  849. {
  850. subStringLength = 0;
  851. break;
  852. }
  853. }
  854. }
  855. fields[0]->process(savedRow + fieldOffset, tempRow, setProcessor);
  856. setValues.append("..");
  857. fields[0]->process(tempRow + fieldOffset, tempRow, setProcessor);
  858. numValues+=2;
  859. }
  860. else
  861. {
  862. fields[0]->process(tempRow + fieldOffset, tempRow, setProcessor);
  863. numValues++;
  864. }
  865. if (!seg.increment(tempRow))
  866. break;
  867. }
  868. }
  869. if (!inKeyed)
  870. {
  871. if (inWild)
  872. {
  873. out.append("),");
  874. inWild = false;
  875. }
  876. out.append("KEYED(");
  877. inKeyed = true;
  878. }
  879. else
  880. out.append(',');
  881. out.append(fields[0]->name);
  882. if (numValues==1)
  883. out.append("=").append(setValues);
  884. else if (subStringLength)
  885. out.appendf("[1..%d]='", subStringLength).append(subStringLength, (char *) savedRow+fieldOffset).append("'");
  886. else
  887. out.append(" IN [").append(setValues).append("]");
  888. }
  889. fieldOffset += curFieldSize;
  890. fields++;
  891. if (!fields[0])
  892. break;
  893. }
  894. }
  895. if (inKeyed || inWild)
  896. out.append(")");
  897. }
  898. else
  899. out.append("UNKEYED");
  900. }
  901. extern thorhelper_decl void convertRowToXML(size32_t & lenResult, char * & result, IOutputMetaData & info, const void * row, unsigned flags)
  902. {
  903. const byte * self = (const byte *)row;
  904. if (flags == (unsigned)-1)
  905. flags = XWFtrim|XWFopt|XWFnoindent;
  906. CommonXmlWriter writer(flags);
  907. info.toXML(self, writer);
  908. //could use detach...
  909. unsigned sizeResult;
  910. rtlStrToStrX(sizeResult, result, writer.length(), writer.str());
  911. lenResult = rtlUtf8Length(sizeResult, result);
  912. }