26 lines
726 B
Go
26 lines
726 B
Go
package application
|
|
|
|
import (
|
|
//"gitlab.com/tedge/edgex/internal/pkg/limit"
|
|
//"gitlab.com/tedge/edgex/internal/tedge/resource/config"
|
|
"github.com/winc-link/hummingbird/internal/hummingbird/core/config"
|
|
"github.com/winc-link/hummingbird/internal/pkg/limit"
|
|
)
|
|
|
|
type LimitMethodConf struct {
|
|
methods map[string]struct{}
|
|
}
|
|
|
|
//TODO: 接口限流功能需要重构
|
|
func NewLimitMethodConf(configuration config.ConfigurationStruct) limit.LimitMethodConf {
|
|
var conf = &LimitMethodConf{methods: make(map[string]struct{})}
|
|
for _, method := range configuration.Writable.LimitMethods {
|
|
conf.methods[method] = struct{}{}
|
|
}
|
|
return conf
|
|
}
|
|
|
|
func (lmc *LimitMethodConf) GetLimitMethods() map[string]struct{} {
|
|
return lmc.methods
|
|
}
|