分享每一天-golang使用 Singleflight包优化你的代码

golang使用 Singleflight包优化你的代码

GoLang 680 0 创建时间：2023-03-16 11:21:43

Singleflight

#### 介绍
有许多方法可以优化代码以提高效率，减少运行的进程就是其中之一。在本文中，我们将了解如何通过使用 Go 包之一 Singleflight 减少重复进程来优化 Go 代码。

#### 问题
假设你有一个 Web 应用，它每秒有 10 个请求 （RPS）。根据数据，您知道其中一些请求具有相同的模式，实际上可以生成相同的结果，这意味着您实际上具有冗余进程。
![](https://www.seedblog.cn/uploads/article/2023-03-16/wtWszan2qG7W7y3a8tNzSFA1cv1FeUIxQhvYYtfg.png)
从上图中，我们知道并想要同样的东西，但最终，我们（大部分）分别处理这两个请求。user 1 user 2

Singleflight 是可以解决这类问题的 Go 包之一，如文档中所述，它提供了重复函数调用抑制机制。

很酷，所以如果我们知道我们要调用的函数是重复的，我们实际上可以减少我们处理的函数数量，让我们看看如何在现实世界中使用它。

#### 实现
我们将创建两个程序，并且.server.goclient.go

server.go— 将充当 Web 服务，可以使用名为/api/v1/get_somethingname
```go
// server.go

package main

import (
 "fmt"
 "net/http"
)

func main() {
 http.HandleFunc("/api/v1/get_something", func(w http.ResponseWriter, r *http.Request) {
  name := r.URL.Query().Get("name")
  response := processingRequest(name)
  fmt.Fprint(w, response)
 })

err := http.ListenAndServe(":8080", nil)
 if err != nil {
  fmt.Println(err)
 }
}

func processingRequest(name string) string {
 fmt.Println("[DEBUG] processing request..")
 return "Hi there! You requested " + name
}
```
client.go 将充当向 Web 服务发出 5 个并发请求的客户端（您可以在变量中设置数字totalRequests)

```go
// client.go

package main

import (
 "io/ioutil"
 "log"
 "net/http"
 "sync"
)

func main() {

var wg sync.WaitGroup

endpoint := "http://localhost:8080/api/v1/get_something?name=something"
 totalRequests := 5

for i := 0; i < totalRequests; i++ {
  wg.Add(1)
  go func(i int) {
   defer wg.Done()
   makeAPICall(endpoint)
  }(i)
 }
 wg.Wait()

}

func makeAPICall(endpoint string) {
 resp, err := http.Get(endpoint)
 if err != nil {
  log.Fatalln(err)
 }

body, err := ioutil.ReadAll(resp.Body)
 if err != nil {
  log.Fatalln(err)
 }

result := string(body)
 log.Printf(result)
}
```
首先，我们可以运行然后继续。 我们将在服务器脚本的终端中看到如下所示的内容：server.goclient.go
```go
[DEBUG] processing request..
[DEBUG] processing request..
[DEBUG] processing request..
[DEBUG] processing request..
[DEBUG] processing request..
```
客户端脚本的终端如下所示：
```go
Hi there! You requested something
Hi there! You requested something
Hi there! You requested something
Hi there! You requested something
Hi there! You requested something
```
这是有道理的，因为我们从客户端发送了五个请求，并在服务器中处理了这五个请求。

现在让我们在代码中实现 Singleflight，以便更高效。
```go
// server.go

package main

import (
 "fmt"
 "net/http"

"golang.org/x/sync/singleflight"
)

var g = singleflight.Group{}

func main() {
 http.HandleFunc("/api/v1/get_something", func(w http.ResponseWriter, r *http.Request) {
  name := r.URL.Query().Get("name")
  response, _, _ := g.Do(name, func() (interface{}, error) {
   result := processingRequest(name)
   return result, nil
  })

fmt.Fprint(w, response)
 })

err := http.ListenAndServe(":8080", nil)
 if err != nil {
  fmt.Println(err)
 }
}

func processingRequest(name string) string {
 fmt.Println("[DEBUG] processing request..")
 return "Hi there! You requested " + name
}
```
在我们重新启动服务器并再次运行客户端程序后，服务器脚本中的终端显示如下：

```go
[DEBUG] processing request..
[DEBUG] processing request..
```
客户端脚本的终端显示如下：

```go
Hi there! You requested something
Hi there! You requested something
Hi there! You requested something
Hi there! You requested something
Hi there! You requested something
```
真棒！所有客户端都得到了预期的响应，但现在我们的服务器只处理了两个请求。想象一下，如果您处理数千个类似的请求，您将带来多大的效率，这真是太神奇了！

#### 结论
在本文中，我们了解了 Singleflight 在优化代码方面的强大功能。不仅可以处理 Web 请求，还可以将其用例扩展到其他内容，例如从数据库中获取数据等。

还有一些我在本文中没有涉及的内容，例如，如果 Singleflight 中的进程失败怎么办，以及我们如何缓解它。这可能是我想在下一篇文章中探讨的主题。

最后，非常感谢您，如果您仍在阅读这一点，如果您觉得它有用或可能认识需要它的人，请分享这篇文章。

[原文链接](https://levelup.gitconnected.com/optimize-your-go-code-using-singleflight-3f11a808324 "原文链接")

golang使用 Singleflight包优化你的代码

商品推荐

广告推荐

推荐文章