mirror of
https://github.com/Rain-kl/OpenFlare.git
synced 2026-09-30 06:16:37 +08:00
ryan
87 lines
2.5 KiB
Go
87 lines
2.5 KiB
Go
package geoip
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
"net"
|
|
"net/http"
|
|
"time"
|
|
)
|
|
|
|
// IPAPIService 使用 ip-api.com 服务实现 GeoIPService 接口。
|
|
type IPAPIService struct {
|
|
Client *http.Client
|
|
}
|
|
|
|
// ipAPIResponse 定义了 ip-api.com 服务返回的 JSON 响应的结构。
|
|
type ipAPIResponse struct {
|
|
Status string `json:"status"`
|
|
Message string `json:"message"` // 当 status 为 fail 时出现
|
|
Country string `json:"country"`
|
|
CountryCode string `json:"countryCode"`
|
|
Region string `json:"region"`
|
|
RegionName string `json:"regionName"`
|
|
City string `json:"city"`
|
|
Zip string `json:"zip"`
|
|
Lat float64 `json:"lat"`
|
|
Lon float64 `json:"lon"`
|
|
Timezone string `json:"timezone"`
|
|
ISP string `json:"isp"`
|
|
Org string `json:"org"`
|
|
As string `json:"as"`
|
|
Query string `json:"query"`
|
|
}
|
|
|
|
func (s *IPAPIService) Name() string {
|
|
return "ip-api.com"
|
|
}
|
|
|
|
// NewIPAPIService 创建并返回一个 IPAPIService 的新实例。
|
|
func NewIPAPIService() (*IPAPIService, error) {
|
|
return &IPAPIService{
|
|
Client: &http.Client{
|
|
Timeout: 5 * time.Second, // 设置请求超时
|
|
},
|
|
}, nil
|
|
}
|
|
|
|
// GetGeoInfo 使用 ip-api.com 服务检索给定 IP 地址的地理位置信息。
|
|
func (s *IPAPIService) GetGeoInfo(ip net.IP) (*GeoInfo, error) {
|
|
// API URL, 使用 fields 参数来仅请求需要的字段
|
|
apiURL := fmt.Sprintf("http://ip-api.com/json/%s?fields=status,message,country,countryCode", ip.String())
|
|
|
|
resp, err := s.Client.Get(apiURL)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("failed to get geo info from ip-api.com: %w", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
var apiResp ipAPIResponse
|
|
if err := json.NewDecoder(resp.Body).Decode(&apiResp); err != nil {
|
|
return nil, fmt.Errorf("failed to decode ip-api.com response: %w", err)
|
|
}
|
|
|
|
if apiResp.Status != "success" {
|
|
return nil, fmt.Errorf("ip-api.com returned an error: %s", apiResp.Message)
|
|
}
|
|
|
|
return &GeoInfo{
|
|
ISOCode: apiResp.CountryCode,
|
|
Name: apiResp.Country,
|
|
Latitude: float64Pointer(apiResp.Lat),
|
|
Longitude: float64Pointer(apiResp.Lon),
|
|
}, nil
|
|
}
|
|
|
|
// UpdateDatabase 对于 ip-api.com 是一个空操作，因为它是一个 Web 服务。
|
|
func (s *IPAPIService) UpdateDatabase() error {
|
|
// 无需执行任何操作，因为数据由外部服务提供
|
|
return nil
|
|
}
|
|
|
|
// Close 对于 ip-api.com 是一个空操作，因为没有需要关闭的持久连接。
|
|
func (s *IPAPIService) Close() error {
|
|
// 无需执行任何操作
|
|
return nil
|
|
}
|