mirror of
https://github.com/sot-tech/mochi.git
synced 2026-04-26 15:40:01 -07:00
middleware: add varinterval
This commit is contained in:
@@ -12,6 +12,7 @@ import (
|
||||
"github.com/chihaya/chihaya/middleware"
|
||||
"github.com/chihaya/chihaya/middleware/clientapproval"
|
||||
"github.com/chihaya/chihaya/middleware/jwt"
|
||||
"github.com/chihaya/chihaya/middleware/varinterval"
|
||||
"github.com/chihaya/chihaya/storage/memory"
|
||||
)
|
||||
|
||||
@@ -94,6 +95,17 @@ func (cfg ConfigFile) CreateHooks() (preHooks, postHooks []middleware.Hook, err
|
||||
return nil, nil, errors.New("invalid client approval middleware config: " + err.Error())
|
||||
}
|
||||
preHooks = append(preHooks, hook)
|
||||
case "interval variation":
|
||||
var viCfg varinterval.Config
|
||||
err := yaml.Unmarshal(cfgBytes, &viCfg)
|
||||
if err != nil {
|
||||
return nil, nil, errors.New("invalid interval variation middleware config: " + err.Error())
|
||||
}
|
||||
hook, err := varinterval.New(viCfg)
|
||||
if err != nil {
|
||||
return nil, nil, errors.New("invalid interval variation middleware config: " + err.Error())
|
||||
}
|
||||
preHooks = append(preHooks, hook)
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user