Optimumkan aplikasi bahasa Go: tetapkan had permintaan dengan sewajarnya

Aplikasi bahasa Optimumkan Go: Tetapkan had permintaan dengan munasabah

Apabila membangunkan aplikasi web, anda biasanya menghadapi keperluan untuk mengehadkan permintaan luaran, seperti mengehadkan kekerapan permintaan setiap pengguna, mencegah serangan berniat jahat atau mengurangkan tekanan Pelayan. Dalam bahasa Go, kami boleh mengoptimumkan prestasi dan keselamatan aplikasi dengan menetapkan had permintaan yang munasabah. Artikel ini akan memperkenalkan cara melaksanakan sekatan permintaan dalam bahasa Go dan memberikan contoh kod khusus.

1. Laksanakan pengehadan permintaan melalui algoritma baldi token

Algoritma baldi token ialah algoritma pengehad semasa klasik yang boleh mengawal kadar permintaan dengan berkesan. Algoritma ini mengekalkan baldi token kapasiti tetap, dan setiap permintaan menggunakan satu token Apabila token tidak mencukupi dalam baldi token, permintaan akan dihadkan. Berikut ialah contoh had permintaan mudah berdasarkan algoritma baldi token:

package main

import (
    "time"
)

type Limiter struct {
    tokens       chan struct{}
    tokenPerSec  int
}

func NewLimiter(tokenPerSec int) *Limiter {
    l := &Limiter{
        tokens:      make(chan struct{}, tokenPerSec),
        tokenPerSec: tokenPerSec,
    }
    go func() {
        ticker := time.NewTicker(time.Second)
        defer ticker.Stop()
        for range ticker.C {
            select {
            case l.tokens <- struct{}{}:
            default:
            }
        }
    }()
    return l
}

func (l *Limiter) Allow() bool {
    select {
    case <-l.tokens:
        return true
    default:
        return false
    }
}

func main() {
    limiter := NewLimiter(10) // 每秒限制10个请求
    for i := 0; i < 20; i++ {
        if limiter.Allow() {
            fmt.Println("Allow request")
        } else {
            fmt.Println("Limit request")
        }
        time.Sleep(100 * time.Millisecond)
    }
}

Dalam contoh di atas, baldi token yang mengehadkan 10 permintaan sesaat dibuat melalui fungsi NewLimiter dan kaedah Allow menentukan sama ada untuk membenarkan permintaan tersebut. Dengan cara ini, kami boleh mengehadkan permintaan dan memastikan kestabilan sistem.

2. Gunakan penyegerakan.Peta untuk melaksanakan pengehadan kekerapan permintaan IP

Selain pengehadan permintaan berdasarkan algoritma baldi token, kami juga boleh menggunakan penyegerakan.Peta untuk melaksanakan pengehadan kekerapan permintaan IP. Berikut ialah contoh mudah:

package main

import (
    "fmt"
    "sync"
    "time"
)

type IPRequest struct {
    Count    int
    LastTime time.Time
}

var requests sync.Map

func LimitIP(ip string, limit int) bool {
    now := time.Now()
    value, ok := requests.Load(ip)
    if !ok {
        requests.Store(ip, &IPRequest{Count: 1, LastTime: now})
        return true
    }

    req := value.(*IPRequest)
    if req.Count >= limit && now.Sub(req.LastTime) < time.Second {
        return false
    }

    req.Count++
    req.LastTime = now
    return true
}

func main() {
    for i := 0; i < 20; i++ {
        ip := "127.0.0.1"
        if LimitIP(ip, 5) {
            fmt.Println("Allow request from ", ip)
        } else {
            fmt.Println("Limit request from ", ip)
        }
        time.Sleep(100 * time.Millisecond)
    }
}

Dalam kod di atas, sync.Map digunakan untuk menyimpan bilangan permintaan dan masa permintaan terakhir setiap IP, dan fungsi LimitIP digunakan untuk mengehadkan kekerapan permintaan setiap IP. Kaedah ini boleh melaksanakan had kekerapan permintaan untuk setiap IP dalam aplikasi dan melindungi pelayan daripada serangan berniat jahat.

Dengan menetapkan had permintaan dengan sewajarnya, kami boleh mengoptimumkan prestasi dan keselamatan aplikasi bahasa Go, mencegah serangan berniat jahat dan mengurangkan beban pelayan. Saya harap contoh kod yang disediakan dalam artikel ini dapat membantu anda melaksanakan fungsi mengehadkan permintaan dengan lebih baik.

Atas ialah kandungan terperinci Optimumkan aplikasi bahasa Go: tetapkan had permintaan dengan sewajarnya. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!