51 lines
898 B
Go
51 lines
898 B
Go
package main
|
|
|
|
import (
|
|
"bufio"
|
|
"log"
|
|
"os"
|
|
"strings"
|
|
"sync"
|
|
)
|
|
|
|
var (
|
|
whiteList = make(map[string]struct{})
|
|
whiteListLock = sync.RWMutex{}
|
|
)
|
|
|
|
func LoadWhiteList(filename string) error {
|
|
// 如果文件不存在，直接返回 nil 表示无需加载
|
|
if _, err := os.Stat(filename); os.IsNotExist(err) {
|
|
log.Printf(" 白名单文件 %s 不存在，跳过加载。", filename)
|
|
return nil
|
|
}
|
|
|
|
file, err := os.Open(filename)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
scanner := bufio.NewScanner(file)
|
|
tempList := make(map[string]struct{})
|
|
|
|
for scanner.Scan() {
|
|
ip := strings.TrimSpace(scanner.Text())
|
|
if ip != "" {
|
|
tempList[ip] = struct{}{}
|
|
}
|
|
}
|
|
|
|
if err := scanner.Err(); err != nil {
|
|
return err
|
|
}
|
|
|
|
// 用写锁更新全局白名单
|
|
whiteListLock.Lock()
|
|
whiteList = tempList
|
|
whiteListLock.Unlock()
|
|
|
|
log.Println("白名单已加载完成")
|
|
return nil
|
|
}
|