1 | /* |
2 | * Legal Notice |
3 | * |
4 | * This document and associated source code (the "Work") is a part of a |
5 | * benchmark specification maintained by the TPC. |
6 | * |
7 | * The TPC reserves all right, title, and interest to the Work as provided |
8 | * under U.S. and international laws, including without limitation all patent |
9 | * and trademark rights therein. |
10 | * |
11 | * No Warranty |
12 | * |
13 | * 1.1 TO THE MAXIMUM EXTENT PERMITTED BY APPLICABLE LAW, THE INFORMATION |
14 | * CONTAINED HEREIN IS PROVIDED "AS IS" AND WITH ALL FAULTS, AND THE |
15 | * AUTHORS AND DEVELOPERS OF THE WORK HEREBY DISCLAIM ALL OTHER |
16 | * WARRANTIES AND CONDITIONS, EITHER EXPRESS, IMPLIED OR STATUTORY, |
17 | * INCLUDING, BUT NOT LIMITED TO, ANY (IF ANY) IMPLIED WARRANTIES, |
18 | * DUTIES OR CONDITIONS OF MERCHANTABILITY, OF FITNESS FOR A PARTICULAR |
19 | * PURPOSE, OF ACCURACY OR COMPLETENESS OF RESPONSES, OF RESULTS, OF |
20 | * WORKMANLIKE EFFORT, OF LACK OF VIRUSES, AND OF LACK OF NEGLIGENCE. |
21 | * ALSO, THERE IS NO WARRANTY OR CONDITION OF TITLE, QUIET ENJOYMENT, |
22 | * QUIET POSSESSION, CORRESPONDENCE TO DESCRIPTION OR NON-INFRINGEMENT |
23 | * WITH REGARD TO THE WORK. |
24 | * 1.2 IN NO EVENT WILL ANY AUTHOR OR DEVELOPER OF THE WORK BE LIABLE TO |
25 | * ANY OTHER PARTY FOR ANY DAMAGES, INCLUDING BUT NOT LIMITED TO THE |
26 | * COST OF PROCURING SUBSTITUTE GOODS OR SERVICES, LOST PROFITS, LOSS |
27 | * OF USE, LOSS OF DATA, OR ANY INCIDENTAL, CONSEQUENTIAL, DIRECT, |
28 | * INDIRECT, OR SPECIAL DAMAGES WHETHER UNDER CONTRACT, TORT, WARRANTY, |
29 | * OR OTHERWISE, ARISING IN ANY WAY OUT OF THIS OR ANY OTHER AGREEMENT |
30 | * RELATING TO THE WORK, WHETHER OR NOT SUCH AUTHOR OR DEVELOPER HAD |
31 | * ADVANCE NOTICE OF THE POSSIBILITY OF SUCH DAMAGES. |
32 | * |
33 | * Contributors |
34 | * - Doug Johnson |
35 | */ |
36 | |
37 | #include "input/StreamSplitter.h" |
38 | |
39 | #include <sstream> |
40 | #include <stdexcept> |
41 | |
42 | using namespace TPCE; |
43 | |
44 | StreamSplitter::StreamSplitter(std::istream &textStream, char recordDelimiter, char fieldDelimiter) |
45 | : recordDelim(recordDelimiter), fieldDelim(fieldDelimiter), stream(textStream) { |
46 | } |
47 | |
48 | // |
49 | // ITextSplitter implementation |
50 | // |
51 | |
52 | bool StreamSplitter::eof() const { |
53 | return stream.eof(); |
54 | } |
55 | |
56 | std::deque<std::string> StreamSplitter::getNextRecord() { |
57 | std::string recordString; |
58 | std::getline(stream, recordString, recordDelim); |
59 | |
60 | std::deque<std::string> fields; |
61 | std::istringstream recordStream(recordString); |
62 | |
63 | // Either or both of eof and fail can occur so check |
64 | // for both. |
65 | while (!recordStream.eof() && !recordStream.fail()) { |
66 | std::string field; |
67 | std::getline(recordStream, field, fieldDelim); |
68 | fields.push_back(field); |
69 | } |
70 | |
71 | // Fail can get set on eof which we don't want to treat |
72 | // as a failure. So instead of testing for fail we test |
73 | // for !eof. |
74 | if (!recordStream.eof()) { |
75 | throw std::runtime_error("Error reading record from stream." ); |
76 | } |
77 | |
78 | return fields; |
79 | } |
80 | |