34 lines
715 B
Go
34 lines
715 B
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
|
|
"cidservice/service"
|
|
)
|
|
|
|
func main() {
|
|
ctx := context.Background()
|
|
|
|
// 检查租户请求次数限制
|
|
allowed, err := service.RateLimit.CheckTenantRequestLimit(ctx, 1, nil)
|
|
if err != nil {
|
|
fmt.Printf("错误: %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("租户请求是否允许: %v\n", allowed)
|
|
|
|
// 获取租户当前使用情况
|
|
current, max, err := service.RateLimit.GetTenantCurrentUsage(ctx, 1, nil)
|
|
if err != nil {
|
|
fmt.Printf("错误: %v\n", err)
|
|
return
|
|
}
|
|
|
|
fmt.Printf("租户请求使用情况:\n")
|
|
fmt.Printf(" 当前使用: %d\n", current)
|
|
fmt.Printf(" 最大允许: %d\n", max)
|
|
fmt.Printf(" 使用率: %.2f%%\n", float64(current)/float64(max)*100)
|
|
}
|