perms-system-server/internal/middleware/adminloginratelimitMiddleware.go

// Code scaffolded by goctl. Safe to edit.
// goctl 1.10.1

package middleware

import (
	"net/http"

	"github.com/zeromicro/go-zero/core/stores/redis"
)

type AdminLoginRateLimitMiddleware struct {
	rl *RateLimitMiddleware
}

// NewAdminLoginRateLimitMiddleware 管理后台登录限流：每 IP 每分钟最多 20 次。
func NewAdminLoginRateLimitMiddleware(rds *redis.Redis, keyPrefix string, behindProxy bool) *AdminLoginRateLimitMiddleware {
	return &AdminLoginRateLimitMiddleware{
		rl: NewRateLimitMiddleware(rds, 60, 20, keyPrefix+":rl:login:admin", behindProxy),
	}
}

func (m *AdminLoginRateLimitMiddleware) Handle(next http.HandlerFunc) http.HandlerFunc {
	return m.rl.Handle(next)
}