BufferedStream.h

Go to the documentation of this file.

// Copyright 2024 libsese
//
// 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.
 
#pragma once
 
#include "sese/io/Stream.h"
#include "sese/util/Noncopyable.h"
 
namespace sese::io {
 
class  BufferedStream final : public Noncopyable, public Stream {
public:
    using Ptr = std::shared_ptr<BufferedStream>;
 
    explicit BufferedStream(const Stream::Ptr &source, size_t buffer_size = STREAM_BYTE_STREAM_SIZE_FACTOR);
    ~BufferedStream() noexcept override;
 
    int64_t read(void *buffer, size_t length) override;
 
    void clear() noexcept;
 
    int64_t write(const void *buffer, size_t length) override;
 
    int64_t flush() noexcept;
 
    void reset(const Stream::Ptr &source) noexcept;
 
    [[nodiscard]] size_t getPosition() const { return pos; }
    [[nodiscard]] size_t getLength() const { return len; }
    [[nodiscard]] size_t getCapacity() const { return cap; }
 
private:
    inline int64_t preRead();
 
    Stream::Ptr source;
    void *buffer = nullptr;
    size_t pos;
    size_t len;
    size_t cap;
};
} // namespace sese::io