Function DownloadGeoLite2DB

utils.go:428–464

DownloadGeoLite2DB 下载GeoLite2-Country.mmdb文件

(filePath string)

// DownloadGeoLite2DB 下载GeoLite2-Country.mmdb文件
func DownloadGeoLite2DB(filePath string) error {
	// MaxMind的免费GeoLite2数据库下载链接
	// 注意：这个链接可能需要注册账户才能使用，这里使用一个公开的镜像链接
	url := "https://github.com/P3TERX/GeoLite.mmdb/raw/download/GeoLite2-Country.mmdb"
	
	printInfo("正在下载GeoLite2-Country.mmdb数据库...")
	
	// 创建HTTP请求
	resp, err := http.Get(url)
	if err != nil {
		return fmt.Errorf("下载请求失败: %v", err)
	}
	defer resp.Body.Close()
	
	// 检查HTTP状态码
	if resp.StatusCode != http.StatusOK {
		return fmt.Errorf("下载失败，HTTP状态码: %d", resp.StatusCode)
	}
	
	// 创建目标文件
	file, err := os.Create(filePath)
	if err != nil {
		return fmt.Errorf("创建文件失败: %v", err)
	}
	defer file.Close()
	
	// 复制数据到文件
	_, err = io.Copy(file, resp.Body)
	if err != nil {
		// 如果下载失败，删除不完整的文件
		os.Remove(filePath)
		return fmt.Errorf("写入文件失败: %v", err)
	}
	
	printSuccess(fmt.Sprintf("GeoLite2数据库下载成功: %s", filePath))
	return nil
}

// TryDownloadGeoLite2DB 尝试下载GeoLite2数据库，失败时不报错
func TryDownloadGeoLite2DB(filePath string) bool {