Go语言网络爬虫下载器接口

与 ModuleInternal 接口一样，下载器接口 Downloader 也内嵌了 Module 接口，它额外声明了一个 Download 方法。有了 ModuleInternal 接口及其实现类型，实现下载器时只需关注它的特色功能，其他的都交给内嵌的 stub.ModuleInternal 就可以了。

下载器的实现类型名为 myDownloader，其声明如下：

//下载器的实现类型
type myDownloader struet {
    //组件基础实例
    stub.ModuleInternal
    //下载用的HTTP客户端
    httpClient http.Client
}

可以看到，我匿名地嵌入了一个 stub.ModuleInternal 类型的字段，这种只有类型而没有名称的字段称为匿名字段。如此一来，myDownloader 类型的方法集合中就包含了 stub.ModuleInternal 类型的所有方法。因而，*myDownloader 类型已经实现了 Module 接口。

另一个 http.Client 类型的字段用于对目标服务器发送 HTTP 请求并接收响应。http.Client 类型是做 HTTP 客户端程序的必选，它开箱即用，同时又很开放，有很多可定制的地方。

myDownloader 类型的这两个字段的值都需要使用方直接或间接提供。关于第一个字段的值，可以很容易通过 stub.NewModuleInternal 函数生成。而第二个字段的值，可以直接通过复合字面量 http.Client{} 生成。

不过，我强烈建议你对它进行一些定制，如果你想让下载器跑得更快的话。后面讲网络爬虫程序示例的时候，我们会给出一些建议。

代码包 gopcp.v2/chapter6/webcrawler/module/local/downloader 中存放了所有与下载器实现有关的代码，大家可以从我的网盘中下载相关代码包（链接：https://pan.baidu.com/s/1yzWHnK1t2jLDIcTPFMLPCA 提取码：slm5）。为了方便使用方创建下载器的实例，可以在其中编写了一个名为 New 的函数：

//用于创建一个下载器实例
func New(
    mid module.MID,client *http.Client,scoreCalculator module.Calculatescore) (module.Downloader,error) {
    moduleBase,err := stub.NewModuleInternal(mid,scoreCalculator)
    if err != nil {
        return nil,err
    }
    if client == nil {
        return nil,genParameterError( "nil http client")
    }
    return &myDownloader{
        ModuleInternal: moduleBase,httpClient:    *client,},nil
}

上述代码中，stub.NewModuleInternal 函数需要组件 ID 和组件评分计算器来生成组件内部基础类型的值，那我就让 New 函数的参数声明列表包含它们。对这两个参数的校验由 stub.NewModuleInternal 函数全权负责。

注意，这里还隐藏着一个 Go语言的命名惯例。由于下载器的实现代码独占一个代码包，所以可以让这个函数的名称足够简单，只有一个单词 New。这不同于前面提到的函数 NewPool 和 NewMultipleReader，这两个函数所创建的实例的含义无法由其所在代码包的名称 buffer 和 reader 表达。

另外，虽然函数 NewBuffer 所创建的实例的含义可以由其所在的代码包 buffer 表达，但是该包中用于创建实例的函数不止它一个。如果把它们的名称简化为 New，恐怕会造成表达上的不清晰。而 downloader 包中唯一用于创建实例的函数 New，可以让你马上明白它就是用于创建下载器实例的，并不需要过多解释。这就是命名方面的惯用法，也是一种技巧。

下面来看下载器的 Download 方法的实现：

func (downloader *myDownloader) Download(req *module.Request) (*module.Response,error) {
    downloader.ModuleInternal.IncrHandlingNumber()
    defer downloader.ModuleInternal.DecrHandlingNumber()
    downloader.ModuleInternal.IncrCalledCount()
    if req == nil {
        return nil,genParameterError("nil request")
    }
    httpReq := req.HTTPReq()
    if httpReq == nil {
        return nil,genParameterError("nil HTTP request")
    }
    downloader.ModuleInternal.IncrAcceptedCount()
    logger.Infof("Do the request (URL: %s,depth: %d)... \n",httpReq.URL,req.Depth()) httpResp,err := downloader.httpClient.Do(httpReq)
    if err != nil {
        return nil,err
    }
    downloader.ModuleInternal.IncrCompletedCount()
    return module.NewResponse(httpResp,req.Depth()),nil
}

这个方法的功能实现起来很简单，不过要注意对那 4 个组件计数的操作。在方法的开始处，要递增实时处理数，并利用 defer 语句保证方法执行结束时递减这个计数。同时,还要递增调用计数。

在所有参数检查都通过后，要递增接受计数以表明该方法接受了这次调用。一旦目标服务器发回了 HTTP 响应并且未发生错误，就可以递增成功完成计数 To 这代表当前组件实例又有效地为使用者提供了一次服务。

Go语言网络爬虫下载器接口

相关推荐