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

Question

我创建了一个最小的可复制示例

package main

import (

    "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 例程退出。