Go HTTP 服务器与全局变量的并发安全

本文探讨了在 Go HTTP 服务器中使用全局变量时可能出现的并发安全问题。通过示例代码，展示了如何使用 channel 和 goroutine 来安全地更新全局变量，避免竞态条件，并提供了一种通用的并发安全计数器实现方案。

在 Go 语言中，编写 HTTP 服务器是一个常见的任务。然而，当涉及到在 HTTP 请求处理程序中使用全局变量时，必须特别注意并发安全问题。由于 Go 的 HTTP 服务器会为每个新的请求启动一个独立的 goroutine 来处理，因此多个请求可能同时访问和修改同一个全局变量，从而导致竞态条件和数据不一致。

并发安全问题

考虑以下示例代码：

package main

import (
    "fmt"
    "net/http"
    "runtime"
)

var cur = 0

func handler(w http.ResponseWriter, r *http.Request) {
    cur = cur + 1
    fmt.Fprintf(w, "Current value: %d\n", cur)
}

func main() {
    runtime.GOMAXPROCS(runtime.NumCPU())
    http.HandleFunc("/", handler)
    http.ListenAndServe(":9010", nil)
}

这段代码看似简单，但实际上存在并发安全问题。cur 是一个全局变量，handler 函数会增加它的值。由于多个 goroutine 可能同时执行 handler 函数，因此 cur = cur + 1 这个操作不是原子性的，会导致竞态条件。例如，两个 goroutine 同时读取 cur 的值，然后各自加 1，最后写回 cur，可能导致 cur 的值只增加 1，而不是 2。

使用 Channel 和 Goroutine 解决并发安全问题

为了解决这个问题，可以使用 channel 和 goroutine 来同步对全局变量的访问。以下是一个改进后的示例：

phpList

phpList提供开源电子邮件营销服务，包括分析、列表分割、内容个性化和退信处理。丰富的技术功能和安全稳定的代码基础是17年持续开发的结果。在95个国家使用，在20多种语言中可用，并用于去年发送了250亿封电子邮件活动。您可以使用自己的SMTP服务器部署它，或在http://phplist.com上获得免费的托管帐户。

14

查看详情

package main

import (
    "fmt"
    "net/http"
    "runtime"
)

var counterInput = make(chan int)

func handler(w http.ResponseWriter, r *http.Request) {
    counterInput <- 1
    fmt.Fprintln(w, "Request processed")
}

func counter(c <-chan int) {
    cur := 0
    for v := range c {
        cur += v
        fmt.Printf("Counter updated: %d\n", cur)
    }
}

func main() {
    runtime.GOMAXPROCS(runtime.NumCPU())
    go counter(counterInput)
    http.HandleFunc("/", handler)
    http.ListenAndServe(":9010", nil)
}

在这个改进后的代码中，我们创建了一个 counterInput channel。handler 函数不再直接修改 cur，而是将一个值（这里是 1）发送到 counterInput channel。counter 函数运行在一个独立的 goroutine 中，它从 counterInput channel 接收值，并更新 cur。

由于 channel 的发送和接收操作是同步的，因此可以保证对 cur 的更新是原子性的，从而避免了竞态条件。

通用并发安全计数器

上面的例子展示了一种通用的并发安全计数器实现方案。可以根据需要修改 handler 函数发送到 channel 的值，以实现不同的计数逻辑。

注意事项

• Channel 的容量： 在上面的例子中，我们创建的是一个无缓冲的 channel。这意味着发送操作会阻塞，直到有接收者准备好接收。如果希望 handler 函数不阻塞，可以创建一个带缓冲的 channel，例如 counterInput := make(chan int, 100)。但是，带缓冲的 channel 可能会导致数据丢失，因此需要根据实际情况选择合适的容量。
• 错误处理： 在实际应用中，应该添加适当的错误处理机制，例如检查 channel 是否关闭，以及处理 goroutine 的 panic。
• 性能： 虽然 channel 和 goroutine 可以解决并发安全问题，但它们也会带来一定的性能开销。在性能要求较高的场景中，可以考虑使用其他并发控制机制，例如 sync.Mutex 或 sync.Atomic。

总结

在 Go HTTP 服务器中使用全局变量时，必须特别注意并发安全问题。使用 channel 和 goroutine 是一种常用的解决并发安全问题的方法。通过将对全局变量的访问同步化，可以避免竞态条件，并保证数据的一致性。在实际应用中，需要根据具体情况选择合适的并发控制机制，并添加适当的错误处理机制。

以上就是Go HTTP 服务器与全局变量的并发安全的详细内容，更多请关注php中文网其它相关文章！