当前位置： > > > > 为什么我的 go 协程在处理后卡住了？

为什么我的 go 协程在处理后卡住了？

来源：stackoverflow
2024-04-25 15:00:37
0浏览
收藏

你在学习Golang相关的知识吗？本文《为什么我的 go 协程在处理后卡住了？》，主要介绍的内容就涉及到，如果你想提升自己的开发能力，就不要错过这篇文章，大家要知道编程理论基础和实战操作都是不可或缺的哦！

问题内容

我是 golang 新手。我一直在使用 gorm 和 go 的并发来读取 sqlite 数据库并将其写入 csv 文件。它工作顺利，但是当处理完成时，它并没有结束主程序并退出。我必须打印 command+c 才能退出。我不知道我做错了什么。可能它正在进入某种阻塞或死锁模式或其他模式。此外，它也不打印再见消息。这意味着它仍在尝试从通道读取数据。请帮忙。这是代码。

package main

import (
    "fmt"
    "reflect"

    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/sqlite"
)

type AirQuality struct {
    // gorm.Model
    // ID      uint   `gorm:"column:id"`
    Index   string `gorm:"column:index"`
    BEN     string `gorm:"column:BEN"`
    CH4     string `gorm:"column:CH4"`
    CO      string `gorm:"column:CO"`
    EBE     string `gorm:"column:EBE"`
    MXY     string `gorm:"column:MXY"`
    NMHC    string `gorm:"column:NMHC"`
    NO      string `gorm:"column:NO"`
    NO2     string `gorm:"column:NO_2"`
    NOX     string `gorm:"column:NOx"`
    OXY     string `gorm:"column:OXY"`
    O3      string `gorm:"column:O_3"`
    PM10    string `gorm:"column:PM10"`
    PM25    string `gorm:"column:PM25"`
    PXY     string `gorm:"column:PXY"`
    SO2     string `gorm:"column:SO_2"`
    TCH     string `gorm:"column:TCH"`
    TOL     string `gorm:"column:TOL"`
    Time    string `gorm:"column:date; type:timestamp"`
    Station string `gorm:"column:station"`
}

func (AirQuality) TableName() string {
    return "AQ"
}

func main() {
    c := generateRows("boring!!")
    for {
        fmt.Println(<-c)
        if c == nil {
            fmt.Println("Bye")
            break
        }
    }
}

func generateRows(msg string) <-chan []string {
    c := make(chan []string)
    go func() {
        db, err := gorm.Open("sqlite3", "./load_testing_7.6m.db")
        if err != nil {
            panic("failed to connect database")
        }
        defer db.Close()
        rows, err := db.Model(&AirQuality{}).Limit(20).Rows()
        defer rows.Close()
        if err != nil {
            panic(err)
        }
        for rows.Next() {
            var aq AirQuality
            db.ScanRows(rows, &aq)
            v := reflect.Indirect(reflect.ValueOf(aq))
            var buf []string
            for i := 0; i < v.NumField(); i++ {
                buf = append(buf, v.Field(i).String())
            }
            c <- buf
        }
    }()
    return c
}

解决方案

从无缓冲的通道（例如您的通道）接收，其中没有人准备好发送值块。这就是你所经历的。

表达式 [<-c] 会阻塞，直到有可用值为止。

在通道中发出“eof”信号的常见方法是，当没有更多值要发送时，使用内置的 函数从发送方关闭通道。

尝试从关闭的通道接收数据可以立即进行，产生通道元素类型的 。要检测此“关闭”状态，请使用特殊的逗号-ok 习惯用法：

value, ok := <- c

如果通道关闭，ok 将为 false（否则为 true）。

“耗尽”通道直至其关闭的简单且正确的方法是使用 for range 循环，如下所示：

for value := range c {
    fmt.println("received:", value)
}

一旦从通道 c 接收到关闭之前在其上发送的所有值，for range 就会终止。

所以在 generaterows() 中，执行以下操作：

go func() {
    // // use defer so it will be closed no matter how this function call ends
    defer close(c)
    // ...
}()

还有你的 main():

func main() {
    c := generateRows("boring!!")
    for v := range c {
        fmt.Println(v)
    }
    fmt.Println("Bye")
}

理论要掌握，实操不能落！以上关于《为什么我的 go 协程在处理后卡住了？》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注米云公众号吧！