dyaml/source/dyaml/streamcompat.d
2014-07-31 14:53:14 +02:00

63 lines
1.9 KiB
D

// Copyright Ferdinand Majerech 2014.
// Distributed under the Boost Software License, Version 1.0.
// (See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)
/// Backward compatibility with std.stream (which was used by D:YAML API in the past).
module dyaml.streamcompat;
import std.stream;
import tinyendian;
/// A streamToBytes wrapper that allocates memory using the GC.
///
/// See_Also: streamToBytes
ubyte[] streamToBytesGC(Stream stream) @trusted nothrow
{
try return stream.streamToBytes(new ubyte[stream.available]);
catch(Exception e)
{
assert(false, "Unexpected exception in streamToBytesGC: " ~ e.msg);
}
}
/// Read all data from a Stream into an array of bytes.
///
/// Params:
///
/// stream = Stream to read from. Must be readable and seekable.
/// memory = Memory to use. Must be long enough to store the entire stream
/// (memory.length >= stream.available).
///
/// Returns: A slice of memory containing all contents of the stream on success.
/// NULL if unable to read the entire stream.
ubyte[] streamToBytes(Stream stream, ubyte[] memory) @system nothrow
{
try
{
assert(stream.readable && stream.seekable,
"Can't read YAML from a stream that is not readable and seekable");
assert(memory.length >= stream.available,
"Not enough memory passed to streamToBytes");
auto buffer = memory[0 .. stream.available];
size_t bytesRead = 0;
for(; bytesRead < buffer.length;)
{
// Returns 0 on eof
const bytes = stream.readBlock(&buffer[bytesRead], buffer.length - bytesRead);
// Reached EOF before reading buffer.length bytes.
if(bytes == 0) { return null; }
bytesRead += bytes;
}
return buffer;
}
catch(Exception e)
{
assert(false, "Unexpected exception in streamToBytes " ~ e.msg);
}
}