should be it

external/duckdb/third_party/tpce-tool/include/main/FlatFileLoader.h

@@ -0,0 +1,164 @@
+/*
+ * Legal Notice
+ *
+ * This document and associated source code (the "Work") is a part of a
+ * benchmark specification maintained by the TPC.
+ *
+ * The TPC reserves all right, title, and interest to the Work as provided
+ * under U.S. and international laws, including without limitation all patent
+ * and trademark rights therein.
+ *
+ * No Warranty
+ *
+ * 1.1 TO THE MAXIMUM EXTENT PERMITTED BY APPLICABLE LAW, THE INFORMATION
+ *     CONTAINED HEREIN IS PROVIDED "AS IS" AND WITH ALL FAULTS, AND THE
+ *     AUTHORS AND DEVELOPERS OF THE WORK HEREBY DISCLAIM ALL OTHER
+ *     WARRANTIES AND CONDITIONS, EITHER EXPRESS, IMPLIED OR STATUTORY,
+ *     INCLUDING, BUT NOT LIMITED TO, ANY (IF ANY) IMPLIED WARRANTIES,
+ *     DUTIES OR CONDITIONS OF MERCHANTABILITY, OF FITNESS FOR A PARTICULAR
+ *     PURPOSE, OF ACCURACY OR COMPLETENESS OF RESPONSES, OF RESULTS, OF
+ *     WORKMANLIKE EFFORT, OF LACK OF VIRUSES, AND OF LACK OF NEGLIGENCE.
+ *     ALSO, THERE IS NO WARRANTY OR CONDITION OF TITLE, QUIET ENJOYMENT,
+ *     QUIET POSSESSION, CORRESPONDENCE TO DESCRIPTION OR NON-INFRINGEMENT
+ *     WITH REGARD TO THE WORK.
+ * 1.2 IN NO EVENT WILL ANY AUTHOR OR DEVELOPER OF THE WORK BE LIABLE TO
+ *     ANY OTHER PARTY FOR ANY DAMAGES, INCLUDING BUT NOT LIMITED TO THE
+ *     COST OF PROCURING SUBSTITUTE GOODS OR SERVICES, LOST PROFITS, LOSS
+ *     OF USE, LOSS OF DATA, OR ANY INCIDENTAL, CONSEQUENTIAL, DIRECT,
+ *     INDIRECT, OR SPECIAL DAMAGES WHETHER UNDER CONTRACT, TORT, WARRANTY,
+ *     OR OTHERWISE, ARISING IN ANY WAY OUT OF THIS OR ANY OTHER AGREEMENT
+ *     RELATING TO THE WORK, WHETHER OR NOT SUCH AUTHOR OR DEVELOPER HAD
+ *     ADVANCE NOTICE OF THE POSSIBILITY OF SUCH DAMAGES.
+ *
+ * Contributors
+ * - Doug Johnson
+ */
+
+/*
+ *   Class representing a flat file loader.
+ */
+#ifndef FLAT_FILE_LOADER_H
+#define FLAT_FILE_LOADER_H
+
+#include <cstdio>
+#include "BaseLoader.h"
+#include "unusedflag.h"
+
+using namespace std;
+
+namespace TPCE {
+
+// EGen Formatting Defaults
+#ifndef DATETIME_FORMAT
+#define DATETIME_FORMAT 12 // YYYY-MM-DD HH:MM:SS.mmm
+#endif
+
+#ifndef TIME_FORMAT
+#define TIME_FORMAT 01 // hh:mm:ss
+#endif
+
+#ifndef DATE_FORMAT
+#define DATE_FORMAT 10 // YYYY-MM-DD
+#endif
+
+#ifndef BOOLEAN_TRUE
+#define BOOLEAN_TRUE "1"
+#endif
+
+#ifndef BOOLEAN_FALSE
+#define BOOLEAN_FALSE "0"
+#endif
+
+#ifndef BUFFER_SIZE
+#define BUFFER_SIZE 0
+#endif
+
+#ifndef FILE_OPEN_MODE_OVERWRITE
+#ifdef WIN32
+#define FILE_OPEN_MODE_OVERWRITE "w+"
+#else
+#define FILE_OPEN_MODE_OVERWRITE "w"
+#endif
+#endif
+
+#ifndef FILE_OPEN_MODE_APPEND
+#ifdef WIN32
+#define FILE_OPEN_MODE_APPEND "a+"
+#else
+#define FILE_OPEN_MODE_APPEND "a"
+#endif
+#endif
+
+// EGen Formatting
+const int FlatFileDateTimeFormat = DATETIME_FORMAT;
+const int FlatFileTimeFormat = TIME_FORMAT;
+const int FlatFileDateFormat = DATE_FORMAT;
+const char *const FlatFileBoolTrue = BOOLEAN_TRUE;
+const char *const FlatFileBoolFalse = BOOLEAN_FALSE;
+
+// EGen Buffering
+const int FlatFileBufferSize = BUFFER_SIZE;
+
+// EGen File Open Modes
+const char *const FlatFileOpenModeOverwrite = FILE_OPEN_MODE_OVERWRITE;
+const char *const FlatFileOpenModeAppend = FILE_OPEN_MODE_APPEND;
+
+// Overwrite vs. append functionality for output flat files.
+enum FlatFileOutputModes { FLAT_FILE_OUTPUT_APPEND = 0, FLAT_FILE_OUTPUT_OVERWRITE };
+
+/*
+ *   FlatLoader class.
+ */
+template <typename T> class CFlatFileLoader : public CBaseLoader<T> {
+protected:
+	FILE *hOutFile;
+
+public:
+	CFlatFileLoader(char *szFileName, FlatFileOutputModes FlatFileOutputMode);
+	~CFlatFileLoader(void);
+
+	// virtual void WriteNextRecord(const T* next_record UNUSED) {};
+	// virtual void WriteNextRecord(const T& next_record UNUSED) {};
+	void FinishLoad(); // finish load
+};
+
+/*
+ *       The constructor.
+ */
+template <typename T> CFlatFileLoader<T>::CFlatFileLoader(char *szFileName, FlatFileOutputModes flatFileOutputMode) {
+	if (FLAT_FILE_OUTPUT_APPEND == flatFileOutputMode) {
+		hOutFile = fopen(szFileName, FlatFileOpenModeAppend);
+	} else if (FLAT_FILE_OUTPUT_OVERWRITE == flatFileOutputMode) {
+		hOutFile = fopen(szFileName, FlatFileOpenModeOverwrite);
+	}
+
+	if (!hOutFile) {
+		throw CSystemErr(CSystemErr::eCreateFile, "CFlatFileLoader<T>::CFlatFileLoader");
+	}
+
+	if (FlatFileBufferSize > 0) {
+		if (setvbuf(hOutFile, NULL, _IOFBF, FlatFileBufferSize)) {
+			throw CSystemErr(CSystemErr::eCreateFile, "CFlatFileLoader<T>::CFlatFileLoader");
+		}
+	}
+}
+
+/*
+ *       Destructor.
+ */
+template <typename T> CFlatFileLoader<T>::~CFlatFileLoader() {
+	fclose(hOutFile);
+}
+
+/*
+ *       Commit sent rows. This needs to be called after the last row has been
+ * sent and before the object is destructed. Otherwise all rows will be
+ * discarded.
+ */
+template <typename T> void CFlatFileLoader<T>::FinishLoad() {
+	fflush(hOutFile);
+}
+
+} // namespace TPCE
+
+#endif // FLAT_FILE_LOADER_H