本文探讨了在 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: %dn", 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 来同步对全局变量的访问。以下是一个改进后的示例：
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: %dn", 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 是一种常用的解决并发安全问题的方法。通过将对全局变量的访问同步化，可以避免竞态条件，并保证数据的一致性。在实际应用中，需要根据具体情况选择合适的并发控制机制，并添加适当的错误处理机制。