| 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 |  | 
|---|