如何提升图片链接替换性能？

提升图片链接替换性能

给定包含图片链接的 html 字符串，将图片上传到阿里云 oss 并替换字符串中的旧链接。

为了提升替换速度，可以采取以下优化措施：

1. 使用sync.once 来优化oss 客户端创建

避免重复创建oss 客户端实例，可以通过使用sync.once 来确保它只会被调用一次。

2. 并发上传图片

将 bucket.putobject 函数放入 goroutine 中运行，这样可以并行上传而不阻塞主线程。

Autoppt

Autoppt：打造高效与精美PPT的AI工具

下载

3. 使用缓存

创建一个缓存来存储已经上传的图片链接，这样当遇到重复的链接时，就不需要再次上传图片了。

优化后的代码如下：

var once sync.Once
var client *oss.Client

func initOssClient() {
    once.Do(func() {
        var err error
        client, err = oss.New(endpoint, key, secret)
        if err != nil {
            log.Fatal(err)
        }
    })
}

func Upload(url string) string {
    initOssClient()

    t1 := time.Now().Unix()
    long := 0

    obj := fmt.Sprintf("article/%d%d.png", t1, long)
    bucket, err := client.Bucket(bucketName)
    if err != nil {
        log.Fatal(err)
    }

    r, err := http.Get(url)
    if err != nil {
        log.Fatal(err)
    }
    defer r.Body.Close()

    err = bucket.PutObject(obj, r.Body)
    if err != nil {
        log.Fatal(err)
    }
    return "https://" + bucketName + "." + endpoint + "/" + obj
}

var cache = map[string]string{}

func RepImages(htmls string) string {
    var imgRE = regexp.MustCompile(`]+\bsrc=["']([^"']+)["']`)
    imgs := imgRE.FindAllStringSubmatch(htmls, -1)
    out := make([]string, len(imgs))

    for i := range out {
        url := imgs[i][1]
        if cachedUrl, ok := cache[url]; ok {
            out[i] = imgs[i][1]
            continue
        }

        myImage := Upload(url)
        cache[url] = myImage
        resHtml := strings.Replace(resHtml, imgs[i][1], myImage, -1)
        out[i] = imgs[i][1]
    }

    return resHtml
}

通过这些优化，图片上传和替换的速度将得到显着提升。