GZCSVPathWriter.cpp

/*
** libutil++
** $Id: GZCSVPathWriter.cpp 1653 2016-02-28 19:54:59Z sella $
** Copyright (c) 2011-2016 Digital Genesis, LLC. All Rights Reserved.
** Released under the LGPL Version 2.1 License.
** http://www.digitalgenesis.com
*/

static const char rcsid[] __attribute__((used)) = "$Id: GZCSVPathWriter.cpp 1653 2016-02-28 19:54:59Z sella $";

#include "GZCSVPathWriter.h"
#include "GZCSVFileWriter.h"

#include <fcntl.h>
#include <dirent.h>

using namespace sella::file;

const char * GZCSVPathWriter::FileSuffix = ".csv.gz";

GZCSVPathWriter::GZCSVPathWriter(const std::string & path, char delimiter, char seperator) THROWS(IOException) :
        path(path),

        delimiter(delimiter),
        seperator(seperator),

        sequence(::time(NULL) << 16)
{
        /* Ensure the supplied path is valid. */
        DIR * directory = ::opendir(this->path.c_str());

        if (directory == NULL) {
                THROW(IOException, "failed to open directory '%s' for writing", this->path.c_str());
        }
        ::closedir(directory);

        /* Remove any trailing slash characters from the path. */
        while (*(this->path.rbegin()) == '/') {
                this->path.erase(this->path.length() - 1, 1);
        }
}

void GZCSVPathWriter::write(const std::vector<std::vector<std::string>> & data) THROWS(IOException) {
        char file[2][4096];

        snprintf(file[0], sizeof(file[0]), "%s/%08lx.%s-incomplete", this->path.c_str(), this->sequence, GZCSVPathWriter::FileSuffix);
        snprintf(file[1], sizeof(file[1]), "%s/%08lx.%s", this->path.c_str(), this->sequence++, GZCSVPathWriter::FileSuffix);

        GZCSVFileWriter writer(file[0], this->delimiter, this->seperator);

        for (auto iter = data.begin(); iter != data.end(); ++iter) {
                writer.write(*iter);
        }
        writer.close(file[1]);
}

/*
** vim: noet ts=3 sw=3
*/