AbstractByteBuffer.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/io/PeekableStream.h"
 
namespace sese::io {
 
class  AbstractByteBuffer : public Stream, public PeekableStream {
private:
    struct Node {
        void *buffer = nullptr;
        Node *next = nullptr;
        size_t length = 0;
        size_t cap = 0;
        explicit Node(size_t buffer_size);
        ~Node();
    };
 
public:
    explicit AbstractByteBuffer(size_t base_size = STREAM_BYTE_STREAM_SIZE_FACTOR, size_t factor = STREAM_BYTE_STREAM_SIZE_FACTOR);
 
    AbstractByteBuffer(const AbstractByteBuffer &abstract_byte_buffer) noexcept;
    AbstractByteBuffer(AbstractByteBuffer &&abstract_byte_buffer) noexcept;
 
    ~AbstractByteBuffer() override;
    virtual void resetPos();
    virtual bool eof();
 
    [[nodiscard]] virtual size_t getLength() const;
    [[nodiscard]] virtual size_t getCapacity() const;
    [[nodiscard]] virtual size_t getReadableSize() const;
 
    virtual size_t freeCapacity();
 
    virtual void swap(AbstractByteBuffer &other) noexcept;
 
    // [[nodiscard]] size_t getCurrentWritePos() const { return currentWritePos; }
    // [[nodiscard]] size_t getCurrentReadPos() const { return currentReadPos; }
 
public:
    int64_t read(void *buffer, size_t len) override;
    int64_t write(const void *buffer, size_t len) override;
 
    int64_t peek(void *buffer, size_t len) override;
    int64_t trunc(size_t need_read) override;
 
private:
    size_t factor = 0;
 
    Node *root = nullptr;
    Node *currentWriteNode = nullptr;
    size_t currentWritePos = 0;
    Node *currentReadNode = nullptr;
    size_t currentReadPos = 0;
 
    size_t length = 0;
    size_t cap = 0;
};
 
} // namespace sese::io