18.4 使用Socket流

使用wxWidgets的流,你仅使用很少的代码就可以很容易传输很大量的数据.现在,假设我们要通过socket来传输一个文件.你可能使用的方法是: 打开这个文件,将所有的内容读入内存,然后将这块内存写入到socket.这种方法对于小文件来说没什么问题,但是如果这个文件是一个很多兆的大文件,将其完整读入内存对于一个速度和内存都很小的电脑来说,显得有些不太现实.而且通常我们需要对大文件进行压缩然后才发往socket以便降低网络流量.怎么办法呢,把大文件读入内存,对其整个进行压缩,然后再一次性发送socket,这样得作法在效率和实用性方面都值得怀疑.

OK,我们来想另外一种办法,每次从文件里读入一小段数据,比如几K数据,然后将其压缩,然后发往socket,如此反复.不幸得是, 小段压缩比起整个文件一起压缩来,压缩效率是大打折扣的.因此我们需要更进一步,维护一个压缩的状态,以便后面的小段数据可以使用前面的压缩信息,也可以避免多次传递压缩头信息.可是到目前为止,你的代码已经变的很庞大了,要分段读取文件,维护压缩数据,压缩并且写入socket. wxWidgets提供了一种更简便的方法.

因为wxWidgets提供了wxSocketInputStream和wxSocketOutputStream类,通过别的流来将数据读出或者写入socket是非常方便的.因为wxWidgets提供了基于文件,字符串,文本,内存以及zlib压缩的流操作,将这些流和 socket流结合起来使用,可以实现很有趣也是很强大的socket数据操作方法.现在,回过头来看看我们刚才说的通过socket压缩传输大文件的问题,我们可能已经找到了一个更方便的途径.要发送一个文件,我们可以现将来自文件的数据流通过zlib的压缩流以后发送到socket的发送流,这样我们一下在就有了强大的支持大文件,支持压缩的,每次只需要读几K的socket文件传输方法了.而在接收端,我们同样可以使用流操作将来自socket流的数据通过zlib解压缩流发送到文件输出流,最后还原为原来的文件.所有这些可能几行代码就足够了.

我们将使用线程来处理整个过程,以便我们可以既不占用100%的CPU,又不阻塞GUI(正如上一小节讨论的那样),要知道在使用socket传输大型的数据的时候,(如果不使用多线程,)这种阻塞几乎是不可避免的.

完整的例子可以在光盘的examples/chap18目录中找到.

文件发送线程

下面的例子中演示了流对象在堆上创建,FileSendThread派生自wxThread.

FileSendThread::FileSendThread(wxString Filename,
                                 wxSocketBase* Socket)
{
    m_Filename = Filename;
    m_Socket = Socket;
    Create();
    Run();
}
void* FileSendThread::Entry()
{
    // 如果10秒之内我们什么数据都发送不了,就超时退出
    m_Socket->SetTimeout(10);
    // 在所有数据发送完成之前,阻塞一切非socket操作
    m_Socket->SetFlags(wxSOCKET_WAITALL | wxSOCKET_BLOCK);
    // 从特定的文件流中读入数据
    wxFileInputStream* FileInputStream =
        new wxFileInputStream(m_Filename);
    // 用来写入socket的流对象
    wxSocketOutputStream* SocketOutputStream =
        new wxSocketOutputStream(*m_Socket);
    // 我们写入的将是压缩以后的数据
    wxZlibOutputStream* ZlibOutputStream =
        new wxZlibOutputStream(*SocketOutputStream);
    // 将文件的内容写入压缩流
    ZlibOutputStream->Write(*FileInputStream);
    // 写所有的数据
    ZlibOutputStream->Sync();
    // 释放ZlibOutputStream将导致发送zlib的压缩结束标记
    delete ZlibOutputStream;
    // 释放资源
    delete SocketOutputStream;
    delete FileInputStream;
    return NULL;
}

文件接收线程

接收例子演示了相关流对象也可以在栈上创建.FileReceiveThread派生自wxThread.

FileReceiveThread::FileReceiveThread(wxString Filename,
                                    wxSocketBase* Socket)
{
    m_Filename = Filename;
    m_Socket = Socket;
    Create();
    Run();
}
void* FileReceiveThread::Entry()
{
    // 如果10秒内什么也收不到,中止接收
    m_Socket->SetTimeout(10);
    // 在我们成功接收完数据之前,阻塞一切其它的代码
    m_Socket->SetFlags(wxSOCKET_WAITALL | wxSOCKET_BLOCK);
    // 用于输出数据到文件的流对象
    wxFileOutputStream FileOutputStream(m_Filename);
    // 从socket接收数据的流对象
    wxSocketInputStream SocketInputStream(*m_Socket);
    // zlib解压缩流对象
    wxZlibInputStream ZlibInputStream(SocketInputStream);
    // 将解压缩以后的结果写入文件
    FileOutputStream.Write(ZlibInputStream);
    return NULL;
}