读取完通道后 main 函数仍在执行

Golang小白一枚，正在不断学习积累知识，现将学习到的知识记录一下，也是将我的所得分享给大家！而今天这篇文章《读取完通道后 main 函数仍在执行》带大家来了解一下##content_title##，希望对大家的知识积累有所帮助，从而弥补自己的不足，助力实战开发！

我正在学习 go，但很难理解为什么我的应用程序无法正常退出。通过下面的代码片段，我得到了正确的结果，但 main 在读取最后一个文件后永远不会退出。据我所知，这是因为它是某个地方的开放通道，但我不明白为什么以及如何可能。

这个应用程序的要点是获取 pdf 文件数组并读出其中的某个字符串。

package main

import (
    "fmt"
    "os"
)

type PDF struct {
    filename string
    code string
    error  error
}

func (p *PDF) GetCode() string {
    return p.code
}

func main() {
    jobs := make(chan PDF)
    results := make(chan PDF)

    for w := 0; w < 8; w++ {
        go worker(jobs, results)
    }

    // PDF files to read.
    pdfs := []string{"test.pdf", "test2.pdf", "test3.pdf"}

    for _, file := range pdfs {
        go func(file string) {
            jobs <- PDF{filename: file}
        }(file)
    }
    defer close(jobs)

    for i := range results {
        fmt.Printf("%s ", i.GetCode())
    }
}

func worker(jobs <-chan PDF, results chan<- PDF) {
    for file := range jobs {
        fmt.Printf("Processing %s\n", file.filename)
        code, err := outputPdfText(file.filename)
        results <- PDF{filename: file.filename, code: code, error: err}
    }
}

func outputPdfText(inputPath string) (string, error) {
    // Code that reads PDF and returns value
    [...]
}

解决方案

您的代码出现死锁。等待处理完成的一个好方法是使用 sync.waitgroup。每次向工作线程添加作业时加 1，然后在 goroutine 中调用 wg.done() 来表示作业已完成。当等待组上的作业计数为 0 时，您的代码将正确退出。这是一个例子：

package main

import (
    "fmt"
    "sync"
)

type PDF struct {
    filename string
    code     string
    error    error
}

func (p *PDF) GetCode() string {
    return p.code
}

var wg sync.WaitGroup

func main() {

    jobs := make(chan PDF)
    results := make(chan PDF)

    for w := 0; w < 8; w++ {
        go worker(jobs, results)
    }

    // PDF files to read.
    pdfs := []string{"test.pdf", "test2.pdf", "test3.pdf"}

    defer func() { close(results) }()
    for _, file := range pdfs {
        wg.Add(1)
        jobs <- PDF{filename: file}
    }

    go func() {
        for i := range results {
            fmt.Printf("%s", i.GetCode())
        }
    }()
    wg.Wait()
    close(jobs)
}

func worker(jobs <-chan PDF, results chan<- PDF) {
    for file := range jobs {
        fmt.Printf("Processing %s\n", file.filename)
        code, err := outputPdfText(file.filename)
        results <- PDF{filename: file.filename, code: code, error: err}
        wg.Done()
    }
}

func outputPdfText(inputPath string) (string, error) {
    // Code that reads PDF and returns value
    return "", nil
}

今天关于《读取完通道后 main 函数仍在执行》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！