Writer.hh

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you under the Apache License, Version 2.0 (the
 * "License"); you may not use this file except in compliance
 * with the License.  You may obtain a copy of the License at
 *
 *     https://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#ifndef avro_Writer_hh__
#define avro_Writer_hh__
 
#include <array>
#include <boost/noncopyable.hpp>
 
#include "Config.hh"
#include "Types.hh"
#include "Validator.hh"
#include "Zigzag.hh"
#include "buffer/Buffer.hh"
 
namespace avro {
 
 
template<class ValidatorType>
class WriterImpl : private boost::noncopyable {
 
public:
    WriterImpl() = default;
 
    explicit WriterImpl(const ValidSchema &schema) : validator_(schema) {}
 
    void writeValue(const Null &) {
        validator_.checkTypeExpected(AVRO_NULL);
    }
 
    void writeValue(bool val) {
        validator_.checkTypeExpected(AVRO_BOOL);
        int8_t byte = (val != 0);
        buffer_.writeTo(byte);
    }
 
    void writeValue(int32_t val) {
        validator_.checkTypeExpected(AVRO_INT);
        // NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
        std::array<uint8_t, 5> bytes;
        size_t size = encodeInt32(val, bytes);
        buffer_.writeTo(reinterpret_cast<const char *>(bytes.data()), size);
    }
 
    void writeValue(int64_t val) {
        validator_.checkTypeExpected(AVRO_LONG);
        putLong(val);
    }
 
    void writeValue(float val) {
        validator_.checkTypeExpected(AVRO_FLOAT);
        union {
            float f;
            int32_t i;
        } v;
 
        v.f = val;
        buffer_.writeTo(v.i);
    }
 
    void writeValue(double val) {
        validator_.checkTypeExpected(AVRO_DOUBLE);
        union {
            double d;
            int64_t i;
        } v;
 
        v.d = val;
        buffer_.writeTo(v.i);
    }
 
    void writeValue(const std::string &val) {
        validator_.checkTypeExpected(AVRO_STRING);
        putBytes(val.c_str(), val.size());
    }
 
    void writeBytes(const void *val, size_t size) {
        validator_.checkTypeExpected(AVRO_BYTES);
        putBytes(val, size);
    }
 
    template<size_t N>
    void writeFixed(const uint8_t (&val)[N]) {
        validator_.checkFixedSizeExpected(N);
        buffer_.writeTo(reinterpret_cast<const char *>(val), N);
    }
 
    template<size_t N>
    void writeFixed(const std::array<uint8_t, N> &val) {
        validator_.checkFixedSizeExpected(val.size());
        buffer_.writeTo(reinterpret_cast<const char *>(val.data()), val.size());
    }
 
    void writeRecord() {
        validator_.checkTypeExpected(AVRO_RECORD);
        validator_.checkTypeExpected(AVRO_LONG);
        validator_.setCount(1);
    }
 
    void writeRecordEnd() {
        validator_.checkTypeExpected(AVRO_RECORD);
        validator_.checkTypeExpected(AVRO_LONG);
        validator_.setCount(0);
    }
 
    void writeArrayBlock(int64_t size) {
        validator_.checkTypeExpected(AVRO_ARRAY);
        writeCount(size);
    }
 
    void writeArrayEnd() {
        writeArrayBlock(0);
    }
 
    void writeMapBlock(int64_t size) {
        validator_.checkTypeExpected(AVRO_MAP);
        writeCount(size);
    }
 
    void writeMapEnd() {
        writeMapBlock(0);
    }
 
    void writeUnion(int64_t choice) {
        validator_.checkTypeExpected(AVRO_UNION);
        writeCount(choice);
    }
 
    void writeEnum(int64_t choice) {
        validator_.checkTypeExpected(AVRO_ENUM);
        writeCount(choice);
    }
 
    InputBuffer buffer() const {
        return buffer_;
    }
 
private:
    void putLong(int64_t val) {
        // NOLINTNEXTLINE(cppcoreguidelines-pro-type-member-init)
        std::array<uint8_t, 10> bytes;
        size_t size = encodeInt64(val, bytes);
        buffer_.writeTo(reinterpret_cast<const char *>(bytes.data()), size);
    }
 
    void putBytes(const void *val, size_t size) {
        putLong(size);
        buffer_.writeTo(reinterpret_cast<const char *>(val), size);
    }
 
    void writeCount(int64_t count) {
        validator_.checkTypeExpected(AVRO_LONG);
        validator_.setCount(count);
        putLong(count);
    }
 
    ValidatorType validator_;
    OutputBuffer buffer_;
};
 
using Writer = WriterImpl<NullValidator>;
using ValidatingWriter = WriterImpl<Validator>;
 
} // namespace avro
 
#endif