versioned_csv_file.h

Go to the documentation of this file.

// Copyright (C) 2015,2017 Internet Systems Consortium, Inc. ("ISC")
//
// This Source Code Form is subject to the terms of the Mozilla Public
// License, v. 2.0. If a copy of the MPL was not distributed with this
// file, You can obtain one at http://mozilla.org/MPL/2.0/.
 
#ifndef VERSIONED_CSV_FILE_H
#define VERSIONED_CSV_FILE_H
 
#include <util/csv_file.h>
 
namespace isc {
namespace util {
 
class VersionedCSVFileError : public Exception {
public:
    VersionedCSVFileError(const char* file, size_t line, const char* what) :
        isc::Exception(file, line, what) { };
};
 
class VersionedColumn {
public:
    VersionedColumn(const std::string& name, const std::string& version,
               const std::string& default_value = "")
        : name_(name), version_(version), default_value_(default_value) {
    };
 
    virtual ~VersionedColumn(){};
 
    std::string name_;
 
    std::string version_;
 
    std::string default_value_;
};
 
typedef boost::shared_ptr<VersionedColumn> VersionedColumnPtr;
 
class VersionedCSVFile : public CSVFile {
public:
 
    enum InputSchemaState {
        CURRENT,
        NEEDS_UPGRADE,
        NEEDS_DOWNGRADE
    };
 
    VersionedCSVFile(const std::string& filename);
 
    virtual ~VersionedCSVFile();
 
    void addColumn(const std::string& col_name, const std::string& version,
                   const std::string& default_value = "");
 
    void setMinimumValidColumns(const std::string& column_name);
 
    size_t getMinimumValidColumns() const;
 
    size_t getInputHeaderCount() const;
 
    size_t getValidColumnCount() const;
 
    virtual void open(const bool seek_to_end = false);
 
    virtual void recreate();
 
    bool next(CSVRow& row);
 
    std::string getInputSchemaVersion() const;
 
    std::string getSchemaVersion() const;
 
    const VersionedColumnPtr& getVersionedColumn(const size_t index) const;
 
    enum InputSchemaState getInputSchemaState() const;
 
    bool needsConversion() const;
 
protected:
 
    virtual bool validateHeader(const CSVRow& header);
 
    void columnCountError(const CSVRow& row, const std::string& reason);
 
private:
    std::vector<VersionedColumnPtr> columns_;
 
    size_t valid_column_count_;
 
    size_t minimum_valid_columns_;
 
    size_t input_header_count_;
 
    enum InputSchemaState input_schema_state_;
};
 
 
} // namespace isc::util
} // namespace isc
 
#endif // VERSIONED_CSV_FILE_H