string_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:   string_sink.hpp
 * Author: alex
 *
 * Created on November 6, 2015, 3:59 PM
 */

#ifndef STATICLIB_IO_STRING_SINK_HPP
#define STATICLIB_IO_STRING_SINK_HPP

#include <cstdint>
#include <cstring>
#include <ios>
#include <string>

#include "staticlib/config.hpp"

#include "staticlib/io/io_exception.hpp"
#include "staticlib/io/span.hpp"

namespace staticlib {
namespace io {

class string_sink {
    std::string str;
    
public:
    string_sink() { }

    explicit string_sink(std::string&& str) :
    str(std::move(str)) { }
    
    string_sink(const string_sink&) = delete;

    string_sink& operator=(const string_sink&) = delete;

    string_sink(string_sink&& other) STATICLIB_NOEXCEPT :
    str(std::move(other.str)) { }

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

    std::streamsize write(span<const char> span) {
        size_t ulen = span.size();
        size_t size = str.size();
        if (!sl::support::is_streamsize(size)) throw io_exception(TRACEMSG(
                "Target string size limit exceeded, length: [" + sl::support::to_string(str.size()) + "]"));
        str.resize(size + ulen);
        std::memcpy(std::addressof(str.front()) + size, span.data(), ulen);
        return static_cast<std::streamsize> (ulen);
    }

    std::string& get_string() {
        return str;
    }

    std::streamsize flush() {
        // no-op
        return 0;
    }

};

} // namespace
}


#endif /* STATICLIB_IO_STRING_SINK_HPP */