Почему мои сопрограммы застряли после обработки? - PullRequest
Новая
       40

Почему мои сопрограммы застряли после обработки?

0 голосов
/ 27 августа 2018

Я новичок в Голанге. Я использовал 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
}

1 Ответ

0 голосов
/ 27 августа 2018

Получение из небуферизованного канала (такого как ваш), где никто не готов отправить блоки значений. Это то, что вы испытываете. Спецификация: Оператор получения:

Выражение [<-c] блокируется до тех пор, пока значение не станет доступным.

Распространенным способом подачи сигнала «EOF» в случае каналов является закрытие канала со стороны отправителя, когда больше нет значений для отправки, с помощью встроенной функции close().

Попытка получения из закрытого канала может быть продолжена немедленно, что дает нулевое значение типа элемента канала. Чтобы обнаружить это «закрытое» состояние, используйте специальную идиому comma-ok: 

value, ok := <- c

Если канал закрыт, ok будет false (в противном случае это true).

Простой и правильный способ «стравить» канал до его закрытия - использовать цикл for range, например: 

for value := range c {
    fmt.Println("Received:", value)
}

for range завершается после получения всех значений из канала c, которые были отправлены на него до его закрытия.

Итак, внутри 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")
}
...