如何每天创建 .csv

编程并不是一个机械性的工作，而是需要有思考，有创新的工作，语法是固定的，但解决问题的思路则是依靠人的思维，这就需要我们坚持学习和更新自己的知识。今天golang学习网就整理分享《如何每天创建 .csv》，文章讲解的知识点主要包括，如果你对Golang方面的知识点感兴趣，就不要错过golang学习网，在这可以对大家的知识积累有所帮助，助力开发能力的提升。

在我的应用程序中，我监听 nats，得到的输出是包含数据的地图。我需要使用地图中的数据生成每日 csv 文件。 现在地图中的数据被连续写入文件。如何仅在每天结束时将数据写入文件？

type Hits struct {
    HitMap      map[string]map[SearchRequestKey]uint32 
    Mu          sync.RWMutex
    Log         zerolog.Logger
    TimeStart   time.Time
    MinHits     int
    WasDelete   map[string]int64
    Cnt1        float32
    Version     int
    QQMap       map[string]struct{}
    GitlabToken string
}

type SearchRequestKey struct {
    Query string
    Date  string
}

func FileAdd(hits *models.Hits) {
    file, err := os.OpenFile("query-hits.csv", os.O_CREATE|os.O_RDWR, 0644)
    if err != nil {
        fmt.Println(err)
    }

    writer := csv.NewWriter(file)
    writer.Comma = '|'

    for key, value := range hits.HitMap {
        for query, shard := range value {
            err := writer.Write([]string{query.Query, key, strconv.Itoa(int(shard))})
            if err != nil {
                fmt.Println(err)
            }
        }
    }
}

正确答案

为了每天创建 .csv 文件，您需要循环运行代码并以当前时间为起点。 24 小时后，该函数将再次生成该文件。最后time.AfterFunc()的方式有所帮助。不要忘记使用互斥体。

type Hits struct {
    HitMap      map[string]map[SearchRequestKey]uint32 
    Mu          sync.RWMutex
    Log         zerolog.Logger
    TimeStart   time.Time
    MinHits     int
    WasDelete   map[string]int64
    Cnt1        float32
    Version     int
    QQMap       map[string]struct{}
    GitlabToken string
}

type SearchRequestKey struct {
    Query string
    Date  string
}

func FileAdd(hits *models.Hits) {
    hits.Mu.Lock()
    defer hits.Mu.Unlock()

    for {
        year, month, day := time.Now().Date()
        file := strconv.Itoa(year) + month.String() + strconv.Itoa(day)

        fileName, err := os.Create(file)
        if err != nil {
            fmt.Println(err)
        }

        writer := csv.NewWriter(fileName)
        writer.Comma = '|'

        for key, value := range hits.HitMap {
            for query, shard := range value {
                err := writer.Write([]string{query.Query, key, strconv.Itoa(int(shard))})
                if err != nil {
                    fmt.Println(err)
                }
            }
        }
        time.Sleep(24 * time.Hour) //only after 24 hours
    }

}

func FuncThatStart (hits *models.Hits) {
    hits.Mu.Lock()
    defer hits.Mu.Unlock()

    // generate for one day
    today := time.Now()
    tomorrow := today.Add(24 * time.Hour) // same time as in time.Sleep()

    time.AfterFunc(time.Until(tomorrow), func() {
        go storage.FileAdd(hits) //goroutine handle one time per day
    })
}

谢谢评论

好了，本文到此结束，带大家了解了《如何每天创建 .csv》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！