reference_sink.hpp

Go to the documentation of this file.

/*
 * Copyright 2015, alex at staticlibs.net
 *
 * Licensed 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
 *
 * http://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.
 */

/* 
 * File:   reference_sink.hpp
 * Author: alex
 *
 * Created on December 13, 2015, 2:18 PM
 */

#ifndef STATICLIB_IO_REFERENCE_SINK_HPP
#define STATICLIB_IO_REFERENCE_SINK_HPP

#include <ios>
#include <functional>

#include "staticlib/config.hpp"

#include "staticlib/io/span.hpp"

namespace staticlib {
namespace io {

template <typename Sink>
class reference_sink {
    std::reference_wrapper<Sink> sink;

public:

    explicit reference_sink(Sink& sink) :
    sink(sink) { }

    reference_sink(const reference_sink& other) :
    sink(other.sink) { }

    reference_sink& operator=(const reference_sink& other) {
        this->sink = other.sink;
        return *this;
    }

    reference_sink(reference_sink&& other) STATICLIB_NOEXCEPT :
    sink(std::move(other.sink)) { }

    reference_sink& operator=(reference_sink&& other) STATICLIB_NOEXCEPT {
        sink = std::move(other.sink);
        return *this;
    }

    std::streamsize write(span<const char> span) {
        return sink.get().write(span);
    }

    std::streamsize flush() {
        return sink.get().flush();
    }

    Sink& get_sink() {
        return sink.get();
    }

};

template <typename Sink>
reference_sink<Sink> make_reference_sink(Sink& sink) {
    return reference_sink<Sink>(sink);
}

} // namespace
}

#endif /* STATICLIB_IO_REFERENCE_SINK_HPP */