有条件地运行连续的 Go 例程

各位小伙伴们，大家好呀！看看今天我又给各位带来了什么文章？本文标题是《有条件地运行连续的 Go 例程》，很明显是关于Golang的文章哈哈哈，其中内容主要会涉及到等等，如果能帮到你，觉得很不错的话，欢迎各位多多点评和分享！

我有以下代码。我试图同时运行 3 个 go 例程，但不要超过三个。这按预期工作，但代码应该运行以更新数据库中的表。

因此，第一个例程处理第一个 50，然后是第二个 50，然后是第三个 50，如此重复。我不希望两个例程同时处理相同的行，并且由于更新需要多长时间，这种情况几乎每次都会发生。

为了解决这个问题，我开始使用新列 processing 来标记行，该列是一个布尔值。我将其设置为 true，以便在例程启动时更新所有行，并使脚本休眠 6 秒以允许更新标志。

这适用于随机的时间，但时不时地，我会看到 2-3 个作业再次处理相同的行。我觉得我用来防止重复更新的方法有点笨拙，想知道是否有更好的方法。

stopper := make(chan struct{}, 3)
var counter int
for {
    counter++
    stopper <- struct{}{}
    go func(db *sqlx.db, c int) {
        fmt.println("start")
        updatetables(db)
        fmt.println("stop"b)
        <-stopper
    }(db, counter)
    time.sleep(6 * time.second)

}

in updatetables

var ids[]string
err := sqlx.Select(db, &data, `select * from table_data where processing = false `)
    if err != nil {
        panic(err)
    }

    for _, row:= range data{
        list = append(ids, row.Id)
    }
    if len(rows) == 0 {
        return
    }

    for _, row:= range data{
        _, err = db.Exec(`update table_data set processing = true where id = $1, row.Id)
        if err != nil {
            panic(err)
        }
    }
    // Additional row processing

正确答案

我认为在这种情况下，对 go 例程的方法存在误解。

执行此类工作的 go 例程应该像工作线程一样处理，使用通道作为主例程（将执行同步）和工作 go 例程（将执行实际工作）之间的通信方法）。

package main

import (
    "log"
    "sync"
    "time"
)

type record struct {
    id int
}

func main() {
    const WORKER_COUNT = 10

    recordschan := make(chan record)

    var wg sync.WaitGroup
    for k := 0; k < WORKER_COUNT; k++ {
        wg.Add(1)
        // Create the worker which will be doing the updates
        go func(workerID int) {
            defer wg.Done() // Marking the worker as done
            for record := range recordschan {
                updateRecord(record)
                log.Printf("req %d processed by worker %d", record.id, workerID)
            }
        }(k)
    }

    // Feeding the records channel
    for _, record := range fetchRecords() {
        recordschan <- record
    }

    // Closing our channel as we're not using it anymore
    close(recordschan)

    // Waiting for all the go routines to finish
    wg.Wait()

    log.Println("we're done!")
}

func fetchRecords() []record {
    result := []record{}
    for k := 0; k < 100; k++ {
        result = append(result, record{k})
    }
    return result
}

func updateRecord(req record) {
    time.Sleep(200 * time.Millisecond)
}

如果您需要一次更新所有 50 个表，您甚至可以在主 go 例程中缓冲内容。

理论要掌握，实操不能落！以上关于《有条件地运行连续的 Go 例程》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！