xfy
3cc2dc7625
- 修复 server.go 缩进问题 - 添加文件末尾换行符 - 格式化测试文件表格对齐 - 更新 update-prompts.md 添加新任务 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
77 lines
2.1 KiB
Go
77 lines
2.1 KiB
Go
// Package netutil 提供网络相关的通用工具函数。
|
||
//
|
||
// 该包包含 URL 解析、客户端 IP 提取等网络操作的工具函数，
|
||
// 供 proxy、middleware、server 等模块共享使用。
|
||
//
|
||
// 作者：xfy
|
||
package netutil
|
||
|
||
import "strings"
|
||
|
||
// ParseTargetURL 解析目标 URL，提取主机地址和 TLS 标志。
|
||
//
|
||
// 该函数用于统一处理代理模块中的 URL 解析逻辑，支持 http:// 和 https:// 前缀。
|
||
//
|
||
// 参数：
|
||
// - targetURL: 目标 URL 字符串（如 "http://backend:8080/path" 或 "https://api.example.com"）
|
||
// - addDefaultPort: 是否在没有端口时添加默认端口（:80 或 :443）
|
||
//
|
||
// 返回值：
|
||
// - addr: 主机地址（格式 host:port）
|
||
// - isTLS: 是否使用 TLS（HTTPS）
|
||
//
|
||
// 示例：
|
||
//
|
||
// addr, isTLS := ParseTargetURL("https://api.example.com/api", true)
|
||
// // addr = "api.example.com:443", isTLS = true
|
||
//
|
||
// addr, isTLS := ParseTargetURL("http://backend:8080", false)
|
||
// // addr = "backend:8080", isTLS = false
|
||
func ParseTargetURL(targetURL string, addDefaultPort bool) (addr string, isTLS bool) {
|
||
addr = targetURL
|
||
|
||
// 处理协议前缀
|
||
if strings.HasPrefix(targetURL, "http://") {
|
||
addr = targetURL[7:]
|
||
} else if strings.HasPrefix(targetURL, "https://") {
|
||
addr = targetURL[8:]
|
||
isTLS = true
|
||
}
|
||
|
||
// 移除路径部分，只保留 host:port
|
||
if idx := strings.Index(addr, "/"); idx != -1 {
|
||
addr = addr[:idx]
|
||
}
|
||
|
||
// 如果需要，添加默认端口
|
||
if addDefaultPort && !strings.Contains(addr, ":") {
|
||
if isTLS {
|
||
addr = addr + ":443"
|
||
} else {
|
||
addr = addr + ":80"
|
||
}
|
||
}
|
||
|
||
return addr, isTLS
|
||
}
|
||
|
||
// ExtractHost 从 URL 提取主机地址（host:port）。
|
||
//
|
||
// 该函数是 ParseTargetURL 的简化版本，始终添加默认端口，
|
||
// 用于需要完整地址但不需要 TLS 标志的场景。
|
||
//
|
||
// 参数：
|
||
// - targetURL: 目标 URL 字符串
|
||
//
|
||
// 返回值：
|
||
// - string: 主机地址（格式 host:port）
|
||
//
|
||
// 示例：
|
||
//
|
||
// host := ExtractHost("https://api.example.com/api")
|
||
// // host = "api.example.com:443"
|
||
func ExtractHost(targetURL string) string {
|
||
addr, _ := ParseTargetURL(targetURL, true)
|
||
return addr
|
||
}
|