如何在 Go 中同时写入两个不同的 csv 文件？

Question

我创建了一个最小的可复制示例package mainimport (    "encoding/csv"    "fmt"    "os"    "strconv"    "sync/atomic"    "time")var (    csvOnePath = "test.csv"    csvTwoPath = "test_two.csv")type A struct {    Running     int32 // used atomically    QuitChan    chan struct{}}func NewA() *A {    return &A{        QuitChan: make(chan struct{}),    }}func (a *A) Start() error {    if ok := atomic.CompareAndSwapInt32(&a.Running, 0, 1); !ok {        return fmt.Errorf("Cannot start service A: service already started")    }    go a.record()    return nil}func (a *A) Stop() error {    if ok := atomic.CompareAndSwapInt32(&a.Running, 1, 0); !ok {        return fmt.Errorf("Cannot stop service A: service already stopped")    }    close(a.QuitChan)    return nil}func (a *A) record() {    //file_one, err := os.OpenFile(csvOnePath, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0755)    file_one, err := os.Create(csvOnePath)    if err != nil {        fmt.Println(err)        return    }    writer := csv.NewWriter(file_one)    // writer, closeFileFunc, err := NewCsvWriter(csvOnePath)    if err != nil {        fmt.Println(err)        return    }    header := []string{"this", "is", "a", "test"}    err = writer.Write(header)    if err != nil {        fmt.Println(err)        return    }    ticker := time.NewTicker(10*time.Second)    for {        select {        case t := <-ticker.C:            err = writer.Write([]string{fmt.Sprintf("%2d:%2d:%2d", t.Hour(), t.Minute(), t.Second())})            if err != nil {                fmt.Println(err)                a.QuitChan <- struct{}{}            }        case <-a.QuitChan:            ticker.Stop()            writer.Flush()            file_one.Close()            fmt.Println("Stopped recording.")            break        }    }}本质上，我有两个不同的服务，它们record在两个不同的 goroutine 中运行一个方法。它们各自在不同时间创建并写入不同的 csv 文件。当我运行它时，会创建 csv 文件但永远不会有数据。运行此程序时不会引发任何错误。我读到我应该使用我已经实现的互斥锁，但这也没有奏效。我应该在这里做什么？

Answer 1

如评论中所述，程序将在main()完成后退出；规范声明“它不会等待其他（非主）goroutines 完成。”。

这意味着您的 go 例程不太可能处理关闭文件的代码，这意味着可能不会写入缓冲数据。

我在操场上创建了您的应用程序的简化版本来演示这一点。

有很多方法可以解决这个问题，但最简单的可能是添加一个WaitGroup，这样您的应用程序就可以在终止之前等待 go 例程退出。