我需要为采用来自文本文件的流的方法编写单元测试。我想做这样的事情:
Stream s = GenerateStreamFromString("a,b \n c,d");
StringReaderStream
public static Stream GenerateStreamFromString(string s)
{
var stream = new MemoryStream();
var writer = new StreamWriter(stream);
writer.Write(s);
writer.Flush();
stream.Position = 0;
return stream;
}
不要忘记使用使用:
using (var stream = GenerateStreamFromString("a,b \n c,d"))
{
// ... Do stuff to stream
}
关于未处置的 StreamWriter
。 StreamWriter
只是基本流的包装器,不使用任何需要处置的资源。 Dispose
方法将关闭 StreamWriter
正在写入的底层 Stream
。在这种情况下,这就是我们想要返回的 MemoryStream
。
在 .NET 4.5 中,现在 StreamWriter
有一个重载,它在编写器被处置后保持底层流打开,但此代码做同样的事情,并且也适用于其他版本的 .NET。
请参阅Is there any way to close a StreamWriter without closing its BaseStream?
另一种解决方案:
public static MemoryStream GenerateStreamFromString(string value)
{
return new MemoryStream(Encoding.UTF8.GetBytes(value ?? ""));
}
new MemoryStream(Encoding.UTF8.GetBytes("\ufeff" + (value ?? ""))
如果您需要在流的开头包含 BOM
new MemoryStream( value, false )
。如果必须使用流编写器编写流,则不能将流设为只读。
将此添加到静态字符串实用程序类:
public static Stream ToStream(this string str)
{
MemoryStream stream = new MemoryStream();
StreamWriter writer = new StreamWriter(stream);
writer.Write(str);
writer.Flush();
stream.Position = 0;
return stream;
}
这添加了一个扩展功能,因此您可以简单地:
using (var stringStream = "My string".ToStream())
{
// use stringStream
}
StreamWriter
时,返回的流被关闭(导致半随机异常)。解决方法是使用不同的构造函数 - 一个允许我指定 leaveOpen 的构造函数。
StreamWriter
吗?
public Stream GenerateStreamFromString(string s)
{
return new MemoryStream(Encoding.UTF8.GetBytes(s));
}
ToStream
的扩展方法的现代化和稍微修改的版本:
public static Stream ToStream(this string value) => ToStream(value, Encoding.UTF8);
public static Stream ToStream(this string value, Encoding encoding)
=> new MemoryStream(encoding.GetBytes(value ?? string.Empty));
按照@Palec 对@Shaun Bowe 答案的评论中的建议进行修改。
或作为单线(@satnhak 建议):
public static Stream ToStream(this string value, Encoding encoding = null)
=> new MemoryStream((encoding ?? Encoding.UTF8).GetBytes(value ?? string.Empty));
public static Stream ToStream(this string value, Encoding encoding = null) => new MemoryStream((encoding ?? Encoding.UTF8).GetBytes(value ?? string.Empty));
我使用了这样的混合答案:
public static Stream ToStream(this string str, Encoding enc = null)
{
enc = enc ?? Encoding.UTF8;
return new MemoryStream(enc.GetBytes(str ?? ""));
}
然后我像这样使用它:
String someStr="This is a Test";
Encoding enc = getEncodingFromSomeWhere();
using (Stream stream = someStr.ToStream(enc))
{
// Do something with the stream....
}
public static Stream ToStream(this string str, Encoding enc = Encoding.UTF8) { return new MemoryStream(enc.GetBytes(str ?? "")); }
使用 MemoryStream
类,首先调用 Encoding.GetBytes
将您的字符串转换为字节数组。
您随后需要在流中添加 TextReader
吗?如果是这样,您可以直接提供 StringReader
,并绕过 MemoryStream
和 Encoding
步骤。
我们使用下面列出的扩展方法。我认为您应该让开发人员决定编码,因此涉及的魔法更少。
public static class StringExtensions {
public static Stream ToStream(this string s) {
return s.ToStream(Encoding.UTF8);
}
public static Stream ToStream(this string s, Encoding encoding) {
return new MemoryStream(encoding.GetBytes(s ?? ""));
}
}
return ToStream(s, Encoding.UTF8);
。在当前的实现中(return s.ToStream(Encoding.UTF8);
,开发人员被迫更加思考以掌握代码,并且似乎s == null
的情况未处理并抛出NullReferenceException
。
如果您需要更改编码,我会投票支持 @ShaunBowe 的解决方案。但是这里的每个答案都至少将整个字符串复制到内存中一次。 ToCharArray
+ BlockCopy
组合的答案执行两次。
如果这很重要,那么这里是原始 UTF-16 字符串的简单 Stream
包装器。如果与 StreamReader
一起使用,请为其选择 Encoding.Unicode
:
public class StringStream : Stream
{
private readonly string str;
public override bool CanRead => true;
public override bool CanSeek => true;
public override bool CanWrite => false;
public override long Length => str.Length * 2;
public override long Position { get; set; } // TODO: bounds check
public StringStream(string s) => str = s ?? throw new ArgumentNullException(nameof(s));
public override long Seek(long offset, SeekOrigin origin)
{
switch (origin)
{
case SeekOrigin.Begin:
Position = offset;
break;
case SeekOrigin.Current:
Position += offset;
break;
case SeekOrigin.End:
Position = Length - offset;
break;
}
return Position;
}
private byte this[int i] => (i & 1) == 0 ? (byte)(str[i / 2] & 0xFF) : (byte)(str[i / 2] >> 8);
public override int Read(byte[] buffer, int offset, int count)
{
// TODO: bounds check
var len = Math.Min(count, Length - Position);
for (int i = 0; i < len; i++)
buffer[offset++] = this[(int)(Position++)];
return (int)len;
}
public override int ReadByte() => Position >= Length ? -1 : this[(int)Position++];
public override void Flush() { }
public override void SetLength(long value) => throw new NotSupportedException();
public override void Write(byte[] buffer, int offset, int count) => throw new NotSupportedException();
public override string ToString() => str; // ;)
}
here 是一个更完整的解决方案,具有必要的绑定检查(源自 MemoryStream
,因此它也具有 ToArray
和 WriteTo
方法)。
干得好:
private Stream GenerateStreamFromString(String p)
{
Byte[] bytes = UTF8Encoding.GetBytes(p);
MemoryStream strm = new MemoryStream();
strm.Write(bytes, 0, bytes.Length);
return strm;
}
字符串扩展的良好组合:
public static byte[] GetBytes(this string str)
{
byte[] bytes = new byte[str.Length * sizeof(char)];
System.Buffer.BlockCopy(str.ToCharArray(), 0, bytes, 0, bytes.Length);
return bytes;
}
public static Stream ToStream(this string str)
{
Stream StringStream = new MemoryStream();
StringStream.Read(str.GetBytes(), 0, str.Length);
return StringStream;
}
GenerateStreamFromString
方法中,您没有将 Using 与 StreamWriter 一起使用。是否有一个原因?StreamWriter
可能正在执行您在内部所说的操作。优点是封装和更简单的代码,但代价是抽象编码之类的东西。这取决于您要达到的目标。