-
Notifications
You must be signed in to change notification settings - Fork 389
/
ratelimit.go
81 lines (67 loc) · 2.12 KB
/
ratelimit.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
/**
* Tencent is pleased to support the open source community by making Polaris available.
*
* Copyright (C) 2019 THL A29 Limited, a Tencent company. All rights reserved.
*
* Licensed under the BSD 3-Clause License (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://opensource.org/licenses/BSD-3-Clause
*
* Unless required by applicable law or agreed to in writing, software distributed
* under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR
* CONDITIONS OF ANY KIND, either express or implied. See the License for the
* specific language governing permissions and limitations under the License.
*/
package plugin
import (
"os"
"sync"
)
// RatelimitType rate limit type
type RatelimitType int
const (
// IPRatelimit Based on IP flow control
IPRatelimit RatelimitType = iota + 1
// APIRatelimit Based on interface-level flow control
APIRatelimit
// ServiceRatelimit Based on Service flow control
ServiceRatelimit
// InstanceRatelimit Based on Instance flow control
InstanceRatelimit
)
// RatelimitStr rate limit string map
var RatelimitStr = map[RatelimitType]string{
IPRatelimit: "ip-limit",
APIRatelimit: "api-limit",
ServiceRatelimit: "service-limit",
InstanceRatelimit: "instance-limit",
}
var (
rateLimitOnce sync.Once
)
// Ratelimit Ratelimit plugin interface
type Ratelimit interface {
Plugin
// Allow Whether to allow access, true: allow, FALSE: not allowing Todo
// - Parameter ratingype is the type of current limits, and the ID is the key that limits the current
// - If RateType is Ratelimitip, the ID is IP, RateType is Ratelimitservice, and the ID is
// IP_NAMESPACE_SERVICE or IP_SERVICEID
Allow(typ RatelimitType, key string) bool
}
// GetRatelimit Get the Ratelimit plugin
func GetRatelimit() Ratelimit {
c := &config.RateLimit
plugin, exist := pluginSet[c.Name]
if !exist {
return nil
}
rateLimitOnce.Do(func() {
if err := plugin.Initialize(c); err != nil {
log.Errorf("Ratelimit plugin init err: %s", err.Error())
os.Exit(-1)
}
})
return plugin.(Ratelimit)
}