分段輸出到同一個Stream（.NET）

某項目需要輸出一個數(shù)據(jù)文件，該文件由2部分組成，即文件頭信息和數(shù)據(jù)。

專業(yè)領(lǐng)域包括網(wǎng)站建設(shè)、成都網(wǎng)站建設(shè)、商城系統(tǒng)網(wǎng)站開發(fā)、微信營銷、系統(tǒng)平臺開發(fā)， 與其他網(wǎng)站設(shè)計及系統(tǒng)開發(fā)公司不同，成都創(chuàng)新互聯(lián)公司的整合解決方案結(jié)合了幫做網(wǎng)絡(luò)品牌建設(shè)經(jīng)驗和互聯(lián)網(wǎng)整合營銷的理念，并將策略和執(zhí)行緊密結(jié)合，為客戶提供全網(wǎng)互聯(lián)網(wǎng)整合方案。

項目是使用C#語言在.NET Framework 4上創(chuàng)建的。

拿到這個需求，首先想到的是定義一個Writer類，在寫入方法中創(chuàng)建一個文件流，使用BinaryWriter封裝，寫入所需要的各種數(shù)據(jù)。看起來就像這樣：

public void Write()
{
    using (Stream stream = OpenFileStream())
    using (BinaryWriter writer = new BinaryWriter(stream))
    {
        WriteHeader(writer);
        WriteData(writer);
    }
}
private void WriteHeader(BinaryWriter writer)
{
    // ......
}
private void WriteData(BinaryWriter writer)
{
    // ......
}

不過隨后需求就發(fā)生了變化，因為數(shù)據(jù)敏感，需要加密。于是想到對文件頭部分使用BinaryWriter寫入，而后面的數(shù)據(jù)部分，先使用CryptoStream包裝流，再使用BinaryWriter寫入。于是改成這樣：

public void Write()
{
    using (Stream stream = OpenFileStream())
    {
        using (BinaryWriter headerWriter = new BinaryWriter(stream))
        {
            WriteHeader(headerWriter);
        }
        using (CryptoStream cryptoStream
            = new CryptoStream(stream, GetCryptoTransform(),
                CryptoStreamMode.Write))
        using (BinaryWriter dataWriter = new BinaryWriter(cryptoStream))
        {
            WriteData(dataWriter);
        }
    }
}

改過之后，問題產(chǎn)生了——在使用headerWriter的using語句結(jié)束時，會自動調(diào)用headerWriter.Dispose()，而這個方法會調(diào)用BaseStream.Close()，也就是說，文件流被關(guān)閉了，那么后面嘗試寫入數(shù)據(jù)時就會拋出異常。

雖然BinaryWriter有一個構(gòu)造方法可以申明不關(guān)閉流：

public BinaryWriter(
    Stream output,
    Encoding encoding,
    bool leaveOpen
)

但這個構(gòu)造方法是在.NET 4.5才加入的，項目是用的4.0的Framework，所以必須另外想辦法。而且后面的CryptoStream也存在同樣的問題，而它可沒有提供不關(guān)閉流的構(gòu)造。

這里可以想到兩個辦法來處理：

1. 在所有內(nèi)容都寫完之后再統(tǒng)一Dispose各種操作對象和流對象。

2. 定義一個從Stream繼承的StreamWrapper，將Close和Dispose都重載并實現(xiàn)為空方法，再定義一個ReallyClose方法來真正關(guān)閉封裝的流。

使用第1種方法，就像這樣：

public void Write()
{
    using (Stream s = OpenFileStream())
    using (BinaryWriter headWriter = new BinaryWriter(s))
    using (CryptoStream cs = new CryptoStream(s, GetCryptoTransform(),
        CryptoStreamMode.Write))
    using (BinaryWriter dataWriter = new BinaryWriter(cs))
    {
        WriterHeader(headWriter);
        WriteData(dataWriter);
    }
}

而且如果一個文件分成了很多很多段的話，這個using列表就太長了。但這不是問題，問題是如果WriteHeader中拋出異常，那么cs和dataWriter這兩個對象就浪費了。所以，可以考慮使用try {...} finally {...} 來實現(xiàn)using，

public void Write()
{
    Stream stream = null;
    BinaryWriter headWriter = null;
    CryptoStream cs = null;
    BinaryWriter dataWriter = null;
    try
    {
        stream = OpenFileStream();
        headWriter = new BinaryWriter(stream);
        WriterHeader(headWriter);
        cs = new CryptoStream(stream, GetCryptoTransform(), CryptoStreamMode.Write);
        dataWriter = new BinaryWriter(cs);
        WriteData(dataWriter);
    }
    finally
    {
        if (dataWriter != null) { dataWriter.Dispose(); }
        if (cs != null) { cs.Dispose(); }
        if (headWriter != null) { headWriter.Dispose(); }
        if (stream != null) { stream.Dispose(); }
    }
}

解決問題，但代碼量大，而且容易出錯。比如，要記得在WriterHeader里面Flush，這個原本會在Dispose()自動執(zhí)行的操作（對于CryptoStream，需要執(zhí)行FlushFinalBlock()）。

相對來說，寫一個StreamWrapper靠譜多了。不過使用RealClose就失去了IDisposable的意義，所以稍稍改變一下，定義一個變量來允許Dispose時關(guān)閉流。

class StreamWrapper : Stream
{
    private readonly Stream stream;
    public Stream Stream { get { return stream; } }
    public StreamWrapper(Stream stream)
    {
        this.stream = stream;
        IsLeavingOpen = true;
    }
    public override void Flush()
    {
        stream.Flush();
    }
    public override long Seek(long offset, SeekOrigin origin)
    {
        return stream.Seek(offset, origin);
    }
    public override void SetLength(long value)
    {
        stream.SetLength(value);
    }
    public override int Read(byte[] buffer, int offset, int count)
    {
        return stream.Read(buffer, offset, count);
    }
    public override void Write(byte[] buffer, int offset, int count)
    {
        stream.Write(buffer, offset, count);
    }
    public override bool CanRead { get { return stream.CanRead; } }
    public override bool CanSeek { get { return stream.CanSeek; } }
    public override bool CanWrite { get { return stream.CanWrite; } }
    public override long Length { get { return stream.Length; } }
    public override long Position
    {
        get { return stream.Position; }
        set { stream.Position = value; }
    }
    public override void Close()
    {
        if (IsLeavingOpen)
        {
            return;
        }
        stream.Close();
        base.Close();
    }
    protected override void Dispose(bool disposing)
    {
        if (IsLeavingOpen)
        {
            return;
        }
        if (disposing)
        {
            stream.Dispose();
        }
        base.Dispose(disposing);
    }
    public bool IsLeavingOpen { get; set; }
}

如果稍稍改變一下寫入數(shù)據(jù)的接口，使用起來也非常方便

public void Write()
{
    using (Stream fileStream = OpenFileStream())
    using (StreamWrapper stream = new StreamWrapper(fileStream))
    {
        WriteHeader(stream);
        WriteData(stream);
        stream.IsLeavingOpen = false;
    }
}
private void WriteHeader(Stream stream)
{
    using (BinaryWriter writer = new BinaryWriter(stream))
    {
        // ......
    }
}
private void WriteData(Stream stream)
{
    using (CryptoStream cryptoStream = new CryptoStream(stream,
        GetCryptoTransform(), CryptoStreamMode.Write))
    using (BinaryWriter writer = new BinaryWriter(cryptoStream))
    {
        // ......
    }
}

本文名稱：分段輸出到同一個Stream（.NET）
文章源于：http://www.muchs.cn/article46/gjgjeg.html

成都網(wǎng)站建設(shè)公司_創(chuàng)新互聯(lián)，為您提供標(biāo)簽優(yōu)化、云服務(wù)器、定制開發(fā)、品牌網(wǎng)站制作、網(wǎng)站建設(shè)、App開發(fā)

廣告

聲明：本網(wǎng)站發(fā)布的內(nèi)容（圖片、視頻和文字）以用戶投稿、用戶轉(zhuǎn)載內(nèi)容為主，如果涉及侵權(quán)請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網(wǎng)站立場，如需處理請聯(lián)系客服。電話：028-86922220；郵箱：631063699@qq.com。內(nèi)容未經(jīng)允許不得轉(zhuǎn)載，或轉(zhuǎn)載時需注明來源： 創(chuàng)新互聯(lián)

成都網(wǎng)站建設(shè)公司