2011-06-04 55 views
3

我有一个接受Stream的函数。我的数据列在一个大列表中,涉及数百万个项目。在F中包装流中的序列#

是否有一种简单的方法可以将一个序列封装在一个Stream中,并返回序列在流中的块?一种明显的方法是实现我自己的流类,它返回序列的块。例如:

type SeqStream(sequence:seq<'a>) = 
    inherit Stream() 
    default x.Read(buf, offset, count) = 
     // get next chunk 
     // yield chunk 

有没有更简单的方法呢?我没有办法改变接受流的目标函数。

回答

6

我认为你的方法看起来不错。唯一的问题是,Stream是一个相对复杂的类,有相当多的成员,你可能不想实现其中的大部分 - 如果你想将它传递给一些使用一些额外成员的代码,你会需要使实现更复杂。总之,只实现Read一个简单的流可以是这样的:

type SeqStream<'a>(sequence:seq<'a>, formatter:'a -> byte[]) = 
    inherit Stream() 

    // Keeps bytes that were read previously, but were not used  
    let temp = ResizeArray<_>() 
    // Enumerator for reading data from the sequence 
    let en = sequence.GetEnumerator() 

    override x.Read(buffer, offset, size) = 
    // Read next element and add it to temp until we have enough 
    // data or until we reach the end of the sequence 
    while temp.Count < size && en.MoveNext() do 
     temp.AddRange(formatter(en.Current)) 

    // Copy data to the output & return count (may be less then 
    // required (at the end of the sequence) 
    let ret = min size temp.Count 
    temp.CopyTo(0, buffer, offset, ret) 
    temp.RemoveRange(0, ret) 
    ret 

    override x.Seek(offset, dir) = invalidOp "Seek" 
    override x.Flush() = invalidOp "Flush" 
    override x.SetLength(l) = invalidOp "SetLength" 
    override x.Length = invalidOp "Length" 
    override x.Position 
    with get() = invalidOp "Position" 
    and set(p) = invalidOp "Position" 
    override x.Write(buffer, offset, size) = invalidOp "Write" 
    override x.CanWrite = false 
    override x.CanSeek = false 
    override x.CanRead = true 

注意,我增加了一个额外的参数 - 对一般类型的值转换为字节数组的功能。一般来说,很难将任何东西转换为字节(可以使用一些序列化),所以这可能更容易。例如,对于整数,您可以编写:

let stream = new SeqStream<_>([ 1 .. 5 ], System.BitConverter.GetBytes)