Golang 性能优化：提高与外部 API 交互性能的技巧-Golang-PHP中文网

Golang 性能优化：提高与外部 API 交互性能的技巧

王林

发布： 2024-09-29 20:48:02

原创

1040人浏览过

与外部 api 交互的优化技巧包括：使用连接池复用连接、设置超时防止阻塞、并行发送请求提升并发性、启用 http/2 优化网络性能，以及使用缓存存储响应避免重复调用。

Golang 性能优化：提高与外部 API 交互性能的技巧

Go 性能优化：提升与外部 API 交互性能的技巧

在 Go 中与外部 API 交互通常是一个相对耗时的操作，会阻碍应用程序的性能。为了减轻这种情况，本文将介绍多种优化技术，可以显着提升与外部 API 交互的性能。

1. 使用连接池

立即学习“go语言免费学习笔记（深入）”；

连接池是一个预先建立的连接池，可以复用，避免了在每次请求时都建立和关闭连接的开销。在 Go 中，可以利用 net/http 包中的 Client 类型来创建连接池：

import (
    "net/http"
)

var client = &http.Client{
    Transport: &http.Transport{
        MaxIdleConns:          100,
        MaxIdleConnsPerHost: 10,
    },
}

登录后复制

2. 设置超时

为发送和接收请求设置超时可以防止 API 交互无限期地阻塞。建议将超时设置为合理的持续时间，例如 5-10 秒：

req, _ := http.NewRequest(http.MethodGet, "https://example.com/api/v1/users", nil)
req.Header.Set("Content-Type", "application/json")

ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
defer cancel()

resp, err := client.Do(req.WithContext(ctx))

登录后复制

3. 并行请求

MOKI

MOKI是美图推出的一款AI短片创作工具，旨在通过AI技术自动生成分镜图并转为视频素材。

375

查看详情

如果应用程序需要同时向多个 API 端点发送请求，可以利用 Go 的并发功能来并行这些请求。可以通过创建等待所有请求完成的通道来实现：

type result struct {
    resp *http.Response
    err  error
}

func makeRequests(urls []string) <-chan result {
    results := make(chan result)
    for _, url := range urls {
        go func(url string) {
            resp, err := client.Get(url)
            results <- result{resp, err}
        }(url)
    }
    return results
}

登录后复制

4. 启用 HTTP/2

HTTP/2 是一种比 HTTP/1.1 更高效的协议，它通过多路复用和服务器推送等特性来优化网络性能。可以在 Go 中通过启用 http2 传输来使用它：

import (
    "net/http"
    "net/http/http2"
)

var client = &http.Client{
    Transport: &http2.Transport{},
}

登录后复制

5. 使用缓存

如果应用程序需要重复向相同的 API 端点发送请求，则可以考虑使用缓存来存储响应并避免重复的网络调用。在 Go 中，可以使用 sync.Map 类型或第三方库（如 github.com/hashicorp/golang-lru）来实现缓存：

var cache sync.Map

func cachedRequest(url string) (*http.Response, error) {
    if cached, ok := cache.Load(url); ok {
        return cached.(*http.Response), nil
    }

    resp, err := client.Get(url)
    if err != nil {
        return nil, err
    }

    cache.Store(url, resp)
    return resp, nil
}

登录后复制

实战案例

以下示例展示了如何将这些优化技术应用于实际应用程序：

import (
    "fmt"
    "net/http"
    "net/http/http2"
    "sync"
    "time"

    "github.com/hashicorp/golang-lru"
)

var cache *lru.ARCCache

func main() {
    var err error

    // 设置连接池和超时
    client := &http.Client{
        Transport: &http2.Transport{},
    }

    ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
    defer cancel()

    // 并行发送请求
    urls := []string{"https://example.com/api/v1/users", "https://example.com/api/v1/posts"}
    results := make(chan result)
    for _, url := range urls {
        go func(url string) {
            req, _ := http.NewRequest(http.MethodGet, url, nil)
            req.Header.Set("Content-Type", "application/json")
            resp, err := client.Do(req.WithContext(ctx))
            results <- result{resp, err}
        }(url)
    }

    // 处理结果
    for result := range results {
        if result.err != nil {
            fmt.Println(result.err)
            continue
        }

        // 将响应添加到缓存
        cache.Add(result.resp.Request.URL.String(), result.resp)
    }
}

登录后复制

以上就是Golang 性能优化：提高与外部 API 交互性能的技巧的详细内容，更多请关注php中文网其它相关文章！