🎨 优化登录逻辑,新增后端ip判断
This commit is contained in:
20
utils/ip.go
20
utils/ip.go
@@ -4,6 +4,7 @@ import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"net/http"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
@@ -13,6 +14,13 @@ type ipAdcodeResp struct {
|
||||
} `json:"adcode"`
|
||||
}
|
||||
|
||||
// adcodes 是允许的地区编码列表
|
||||
var adcodes = []string{
|
||||
"重庆",
|
||||
"海南",
|
||||
"广东",
|
||||
}
|
||||
|
||||
func GetIPAdcode(ip string) string {
|
||||
url := fmt.Sprintf("https://api.vore.top/api/IPdata?ip=%s", ip)
|
||||
client := &http.Client{Timeout: 5 * time.Second}
|
||||
@@ -27,3 +35,15 @@ func GetIPAdcode(ip string) string {
|
||||
}
|
||||
return result.Adcode.O
|
||||
}
|
||||
|
||||
// CheckIPInAdcodes 检测用户IP是否在指定的地区编码范围内
|
||||
func CheckIPInAdcodes(ipAdcode string) bool {
|
||||
lower := strings.ToLower(ipAdcode)
|
||||
for _, code := range adcodes {
|
||||
// 检测是否包含
|
||||
if strings.Contains(lower, strings.ToLower(code)) {
|
||||
return true
|
||||
}
|
||||
}
|
||||
return false
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user