aixiao/DenyIP-go
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

优化日志打印格式

Browse Source
This commit is contained in:
aixiao
2025-08-21 14:51:48 +08:00
parent f6adc5b2a1
commit 7761fb1343
6 changed files with 26 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
main.go
View File

@@ -67,7 +67,7 @@ func StartChildProcess() (*exec.Cmd, error) {
if err := cmd.Start(); err != nil {
return nil, fmt.Errorf("启动子进程失败: %w", err)
}
fmt.Printf("子进程已启动, PID: %d\n", cmd.Process.Pid)
log.Printf(" 子进程已启动, PID: %d\n", cmd.Process.Pid)
return cmd, nil
}
@@ -224,7 +224,7 @@ func loadFromFile(filePath string, logMessage string) error {
}
func InitMap() {
if err := loadFromFile("cn.json", " Map 已成功从文件加载"); err != nil {
if err := loadFromFile("cn.json", " Map 已成功从文件加载"); err != nil {
log.Fatalf(" 加载 Map 失败: %v", err)
}
@@ -257,7 +257,7 @@ func WriteLocalAddr() {
func RunMainProcess() { // 主进程逻辑
fmt.Println("主进程启动...")
log.Println(" 主进程启动...")
WriteLocalAddr() // 将本机外网地址加入到已处理集合中
@@ -302,11 +302,11 @@ func RunMainProcess() { // 主进程逻辑
setName, err := RemoveIPIfInSets("root", MAX_IPSET_NAME, ipStr)
if err != nil {
log.Printf(" %s 删除 IP 出错: %v\n", ipStr, err)
log.Printf(" %s 删除 IP 出错: %v\n", ipStr, err)
} else if setName != "" {
log.Printf(" %s 已从 %s 中移除 \n", ipStr, setName)
log.Printf(" %s 已从 %s 中移除 \n", ipStr, setName)
} else {
log.Printf(" %s 不在任何 IPSet 中，无需移除\n", ipStr)
log.Printf(" %s 不在任何 IPSet 中，无需移除\n", ipStr)
}
goto next
@@ -316,7 +316,7 @@ func RunMainProcess() { // 主进程逻辑
if Is_Ip_Ipset(ipStr) != 0 { // IP 不在 ipset 集合中
//if !strings.Contains(region, "中国") && !strings.Contains(region, "内网") { // 离线库判断不在中国内
//if !strings.Contains(region, "中国") && !strings.Contains(region, "内网") { // 离线库判断不在中国内
if !ContainsPart(region, REGION) {
log.Printf("\033[33m [%s %s] 离线库为国外, 进一步API判断\033[0m\n", ipStr, region)
@@ -325,13 +325,12 @@ func RunMainProcess() { // 主进程逻辑
} else {
log.Printf("\033[31m [%s %s]\033[0m\n\n", ipStr, position) // 打印地域
//if !strings.Contains(position, "中国") && !strings.Contains(position, "内网") { // API 判断为国外
//if !strings.Contains(position, "中国") && !strings.Contains(position, "内网") { // API 判断为国外
if !ContainsPart(region, REGION) {
AddIPSet(IPSET_NAME, ipStr) // 添加 IP 到 ipset 集合
// 钉钉告警，废弃！钉钉可能限制文本长度，和发送次数！
// warning_ding(ipStr, position) // 警告 IP 地域
} else {
log.Printf("\033[33m %s 离线库为国外, API 判断为国内, 标记为已处理\033[0m\n", ipStr)
@@ -342,7 +341,7 @@ func RunMainProcess() { // 主进程逻辑
// 写入json文件
if err := saveMapToFile("cn.json"); err != nil {
log.Printf("实时保存 Map 失败: %v", err)
log.Printf(" 实时保存 Map 失败: %v", err)
}
}
}
@@ -399,7 +398,7 @@ func RunMainProcess() { // 主进程逻辑
ProcessedMutex.Unlock()
// 打印日志，表示本次清理已完成
log.Println(" 已清理过期 ProcessedIPMap 项")
log.Println(" 已清理过期 ProcessedIPMap 项")
}
}()
@@ -408,7 +407,7 @@ func RunMainProcess() { // 主进程逻辑
for {
time.Sleep(1 * time.Minute) // 每 10 分钟自动刷新
if err := LoadWhiteList("whitelist.txt"); err != nil {
log.Printf(" 刷新白名单失败: %v", err)
log.Printf(" 刷新白名单失败: %v", err)
}
}
}()
@@ -511,9 +510,9 @@ func main() {
// 加载白名单
err := LoadWhiteList("whitelist.txt")
if err != nil {
log.Fatalf(" whiteList Map 加载白名单失败: %v", err)
log.Fatalf(" whiteList Map 加载白名单失败: %v", err)
} else {
log.Println(" whiteList Map 白名单加载成功")
log.Println(" whiteList Map 白名单加载成功")
}
// 守护进程模式