ValidatingReader.hh

#ifndef avro_ValidatingReader_hh__
#define avro_ValidatingReader_hh__

#include <stdint.h>
#include <vector>
#include <boost/noncopyable.hpp>

#include "Reader.hh"
#include "Validator.hh"
#include "AvroTraits.hh"

namespace avro {

class ValidSchema;
class InputStreamer;


class ValidatingReader : private boost::noncopyable
{

  public:

    ValidatingReader(const ValidSchema &schema, InputStreamer &in);

    template<typename T>
    void readValue(T &val) {
        checkSafeToGet(type_to_avro<T>::type);
        reader_.readValue(val);
        validator_.advance();
    }

    void readBytes(std::vector<uint8_t> &val) {
        checkSafeToGet(AVRO_BYTES);
        validator_.advance();
        reader_.readBytes(val);
    }

    void readFixed(uint8_t *val, size_t size) {
        checkSafeToGet(AVRO_FIXED);
        checkSizeExpected(size);
        validator_.advance();
        reader_.readFixed(val, size);
    }

    template <size_t N>
    void readFixed(uint8_t (&val)[N]) {
        readFixed(val, N);
    }

    template<size_t N>
    void readFixed(boost::array<uint8_t, N> &val) {
        readFixed(val.c_array(), N);
    }

    void readRecord();

    int64_t readArrayBlockSize();

    int64_t readUnion();

    int64_t readEnum();

    int64_t readMapBlockSize();

    Type nextType() const{
        return validator_.nextTypeExpected();
    }

    bool currentRecordName(std::string &name) const {
        return validator_.getCurrentRecordName(name);
    }

    bool nextFieldName(std::string &name) const {
        return validator_.getNextFieldName(name);
    }

  private:

    int64_t readCount();

    void checkSafeToGet(Type type) const {
        if(validator_.nextTypeExpected() != type) {
            throw Exception("Type does not match");
        }
    }

    void checkSizeExpected(int size) const {
        if(validator_.nextSizeExpected() != size) {
            throw Exception("Wrong size of for fixed");
        }
    }

    Validator validator_;
    Reader reader_;
};

} // namespace avro

#endif