update auth init

This commit is contained in:
wjqserver 2025-06-16 08:45:47 +08:00
parent a4d324a361
commit 1636bf1548
2 changed files with 9 additions and 4 deletions

View file

@ -7,19 +7,20 @@ import (
"github.com/infinite-iroha/touka" "github.com/infinite-iroha/touka"
) )
func Init(cfg *config.Config) { func ListInit(cfg *config.Config) error {
if cfg.Blacklist.Enabled { if cfg.Blacklist.Enabled {
err := InitBlacklist(cfg) err := InitBlacklist(cfg)
if err != nil { if err != nil {
panic(err.Error()) return err
} }
} }
if cfg.Whitelist.Enabled { if cfg.Whitelist.Enabled {
err := InitWhitelist(cfg) err := InitWhitelist(cfg)
if err != nil { if err != nil {
panic(err.Error()) return err
} }
} }
return nil
} }
func AuthHandler(c *touka.Context, cfg *config.Config) (isValid bool, err error) { func AuthHandler(c *touka.Context, cfg *config.Config) (isValid bool, err error) {

View file

@ -151,7 +151,11 @@ func setMemLimit(cfg *config.Config) {
} }
func loadlist(cfg *config.Config) { func loadlist(cfg *config.Config) {
auth.Init(cfg) err := auth.ListInit(cfg)
if err != nil {
logger.Errorf("Failed to initialize list: %v", err)
}
} }
func setupApi(cfg *config.Config, r *touka.Engine, version string) { func setupApi(cfg *config.Config, r *touka.Engine, version string) {