mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-15 15:16:12 -04:00
Janusz Sobczak
1a78b33a89
Read() can return fewer bytes than requested. ReadExactly() will call Read() repeatedly until we have read as many bytes as we need. PiperOrigin-RevId: 518719418
78 lines
2.2 KiB
C++
78 lines
2.2 KiB
C++
// Copyright 2020 Google LLC
|
|
//
|
|
// 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
|
|
//
|
|
// 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.
|
|
|
|
#include "internal/platform/input_stream.h"
|
|
|
|
#include <algorithm>
|
|
#include <cstddef>
|
|
#include <cstdint>
|
|
#include <utility>
|
|
|
|
#include "internal/platform/byte_array.h"
|
|
#include "internal/platform/exception.h"
|
|
|
|
namespace nearby {
|
|
namespace {
|
|
constexpr size_t kSkipBufferSize = 64 * 1024;
|
|
} // namespace
|
|
|
|
ExceptionOr<size_t> InputStream::Skip(size_t offset) {
|
|
size_t bytes_left = offset;
|
|
while (bytes_left > 0) {
|
|
size_t chunk_size = std::min(bytes_left, kSkipBufferSize);
|
|
ExceptionOr<ByteArray> result = Read(chunk_size);
|
|
if (!result.ok()) {
|
|
return result.GetException();
|
|
}
|
|
size_t bytes_read = result.GetResult().size();
|
|
if (bytes_read == 0) {
|
|
return ExceptionOr<size_t>(offset - bytes_left);
|
|
}
|
|
bytes_left -= bytes_read;
|
|
}
|
|
return ExceptionOr<size_t>(offset);
|
|
}
|
|
|
|
ExceptionOr<ByteArray> InputStream::ReadExactly(std::size_t size) {
|
|
ByteArray buffer;
|
|
std::size_t current_pos = 0;
|
|
|
|
while (current_pos < size) {
|
|
ExceptionOr<ByteArray> read_bytes = Read(size - current_pos);
|
|
if (!read_bytes.ok()) {
|
|
return read_bytes;
|
|
}
|
|
const ByteArray& result = read_bytes.result();
|
|
|
|
if (result.Empty()) {
|
|
return ExceptionOr<ByteArray>(Exception::kIo);
|
|
}
|
|
if (current_pos == 0) {
|
|
if (result.size() == size) {
|
|
// We have read the requested `size` bytes in one chunk. We can return
|
|
// it directly.
|
|
return read_bytes;
|
|
} else {
|
|
// Reserve space for in the buffer.
|
|
buffer.SetData(size);
|
|
}
|
|
}
|
|
buffer.CopyAt(current_pos, result);
|
|
current_pos += result.size();
|
|
}
|
|
|
|
return ExceptionOr<ByteArray>(std::move(buffer));
|
|
}
|
|
} // namespace nearby
|