Bagaimana untuk mengehadkan aliran dalam golang? Penjelasan terperinci tentang algoritma

Dengan perkembangan teknologi Internet yang berterusan, permintaan untuk konkurensi yang tinggi dan trafik yang besar menjadi semakin biasa, yang menjadikan kepentingan teknologi mengehadkan semasa semakin penting. Sebagai bahasa yang pantas dan cekap, golang tidak boleh mengabaikan penggunaannya dalam pengehadan semasa. Jadi, mari kita lihat secara khusus bagaimana golang mengehadkan aliran semasa.

1. Algoritma corong

Algoritma corong ialah algoritma pengehad semasa yang biasa digunakan ialah untuk mengekalkan corong dengan kapasiti tetap, dan kemudian menambah air pada corong pada tahap tertentu kadar. Jika Apabila corong mencapai kapasiti maksimum, air di belakangnya akan melimpah Untuk permintaan yang memasuki corong, air dalam corong perlu dimakan diproses pada masa ini.

Dalam golang, anda boleh menggunakan pakej "kadar" untuk melaksanakan algoritma corong untuk pengehadan semasa, seperti kod berikut:

import (
     "golang.org/x/time/rate"
     "net/http"
)

// 创建一个每秒钟只允许1个请求的漏斗
r := rate.NewLimiter(1, 1)
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
     if r.Method != "GET" {
          http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
          return
     }
     if !r.Limiter.CanAllow() {
          http.Error(w, "Too Many Requests", http.StatusTooManyRequests)
          return
     }
     // 处理业务逻辑
})

2 algoritma baldi Token

Biarkan Algoritma baldi kad juga merupakan algoritma pengehad arus biasa. Idea terasnya ialah untuk mengekalkan baldi berkapasiti tetap dan terus memasukkan token ke dalam baldi pada kadar tertentu digunakan. Jika token tidak mencukupi dalam baldi, permintaan tidak dapat diproses pada masa ini.

Dalam golang, anda boleh menggunakan pakej "golang.org/x/time/rate" untuk melaksanakan algoritma baldi token untuk pengehadan semasa, seperti kod berikut:

import (
     "golang.org/x/time/rate"
     "net/http"
)

// 创建一个每秒钟只允许1个请求的令牌桶
r := rate.NewLimiter(1, 1)
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
     if r.Method != "GET" {
          http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
          return
     }
     if !r.Wait(r.Context()) {
          http.Error(w, "Too Many Requests", http.StatusTooManyRequests)
          return
     }
     // 处理业务逻辑
})

3. Algoritma Tetingkap Slaid

Algoritma tetingkap gelongsor juga merupakan algoritma pengehad semasa yang biasa digunakan. . Setiap kali permintaan dimasukkan, kaunter tempoh masa yang sepadan ditambah satu Untuk permintaan yang bilangan kaunter masuk mencapai had atas, permintaan itu tidak boleh diproses.

Dalam golang, anda boleh menggunakan pakej "github.com/uber-go/ratelimit" untuk melaksanakan algoritma tetingkap gelongsor untuk pengehadan semasa, seperti kod berikut:

import (
     "github.com/uber-go/ratelimit"
     "net/http"
)

// 创建一个每秒最多只允许1个请求的滑动窗口
rl := ratelimit.New(10) // 表示在一个时间段内最多允许处理10个请求
http.HandleFunc("/", func(w http.ResponseWriter, r *http.Request) {
     if r.Method != "GET" {
          http.Error(w, "Method Not Allowed", http.StatusMethodNotAllowed)
          return
     }
     if !rl.TakeAvailable(1) { // 表示当前请求需要消耗1个计数
          http.Error(w, "Too Many Requests", http.StatusTooManyRequests)
          return
     }
     // 处理业务逻辑
})

4. Perbandingan Token bagi algoritma baldi dan baldi bocor

Walaupun kedua-dua baldi token dan algoritma corong boleh digunakan untuk pengehadan semasa, ia masih berbeza dari segi senario aplikasi, kerumitan algoritma, kesukaran pelaksanaan dan kesan. Khususnya:

• Senario aplikasi: Algoritma baldi token lebih sesuai untuk mengehadkan trafik purata dan permintaan pemprosesan pada kelajuan yang stabil manakala algoritma corong lebih sesuai untuk mengehadkan trafik puncak untuk menghalang sistem daripada; terharu dengan permintaan dalam sekelip mata.
• Kerumitan algoritma: Algoritma corong mempunyai kerumitan yang rendah dan hanya perlu mengekalkan pembilang jenis int dan cap waktu manakala algoritma baldi token perlu mengekalkan kapasiti baldi token dan meletakkan kelajuan token dan lebih banyak parameter.
• Kesukaran pelaksanaan: Pelaksanaan algoritma corong adalah agak mudah Anda boleh menggunakan gelung for untuk mensimulasikan proses menambah air dan mengambil air manakala algoritma baldi token perlu mempertimbangkan keselamatan berbilang benang, token tamat tempoh dan butiran lanjut.

Secara amnya, algoritma pengehadan semasa yang berbeza mempunyai senario aplikasi tertentu, kelebihan dan kekurangannya, dan anda boleh memilih algoritma yang sesuai untuk pengehadan semasa mengikut keperluan sebenar.

Atas ialah kandungan terperinci Bagaimana untuk mengehadkan aliran dalam golang? Penjelasan terperinci tentang algoritma. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!