From ff412f94ec4447e060ce2c2dded2d081b865fa8e Mon Sep 17 00:00:00 2001 From: wjqserver <114663932+WJQSERVER@users.noreply.github.com> Date: Thu, 3 Apr 2025 16:39:56 +0800 Subject: [PATCH] add docs --- config/config.go | 12 +- config/config.toml | 5 +- docker/compose/docker-compose.yml | 3 +- docs/config.md | 328 ++++++++++++++++++++++++++++++ docs/flag.md | 24 +++ docs/menu.md | 13 ++ go.mod | 2 +- main.go | 14 +- middleware/loggin/loggin.go | 13 +- proxy/chunkreq.go | 17 -- proxy/error.go | 3 +- proxy/gitreq.go | 24 +-- proxy/reqheader.go | 30 --- rate/rate.go | 2 +- 14 files changed, 386 insertions(+), 104 deletions(-) create mode 100644 docs/config.md create mode 100644 docs/flag.md create mode 100644 docs/menu.md diff --git a/config/config.go b/config/config.go index 76e14df..1ec7707 100644 --- a/config/config.go +++ b/config/config.go @@ -192,10 +192,10 @@ func DefaultConfig() *Config { GitClone: GitCloneConfig{ Mode: "bypass", SmartGitAddr: "http://127.0.0.1:8080", - ForceH2C: true, + ForceH2C: false, }, Shell: ShellConfig{ - Editor: true, + Editor: false, RewriteAPI: false, }, Pages: PagesConfig{ @@ -205,7 +205,7 @@ func DefaultConfig() *Config { }, Log: LogConfig{ LogFilePath: "/data/ghproxy/log/ghproxy.log", - MaxLogSize: 100, + MaxLogSize: 10, Level: "info", }, Auth: AuthConfig{ @@ -214,7 +214,7 @@ func DefaultConfig() *Config { Key: "", Token: "token", PassThrough: false, - ForceAllowApi: true, + ForceAllowApi: false, }, Blacklist: BlacklistConfig{ Enabled: false, @@ -227,8 +227,8 @@ func DefaultConfig() *Config { RateLimit: RateLimitConfig{ Enabled: false, RateMethod: "total", - RatePerMinute: 1000, - Burst: 100, + RatePerMinute: 100, + Burst: 10, }, Outbound: OutboundConfig{ Enabled: false, diff --git a/config/config.toml b/config/config.toml index 9ae7000..ef27c6d 100644 --- a/config/config.toml +++ b/config/config.toml @@ -32,8 +32,9 @@ maxLogSize = 5 # MB level = "info" # dump, debug, info, warn, error, none [auth] -authMethod = "parameters" # "header" or "parameters" -authToken = "token" +method = "parameters" # "header" or "parameters" +token = "token" +key = "" enabled = false passThrough = false ForceAllowApi = false diff --git a/docker/compose/docker-compose.yml b/docker/compose/docker-compose.yml index dc33343..c120f35 100644 --- a/docker/compose/docker-compose.yml +++ b/docker/compose/docker-compose.yml @@ -4,8 +4,7 @@ services: image: 'wjqserver/ghproxy:latest' restart: always volumes: - - './ghproxy/log/run:/data/ghproxy/log' - - './ghproxy/log/caddy:/data/caddy/log' + - './ghproxy/log:/data/ghproxy/log' - './ghproxy/config:/data/ghproxy/config' ports: - '7210:8080' diff --git a/docs/config.md b/docs/config.md new file mode 100644 index 0000000..a9cb18f --- /dev/null +++ b/docs/config.md @@ -0,0 +1,328 @@ +# ghproxy 用户配置文档 + +`ghproxy` 的配置主要通过修改 `config` 目录下的 `config.toml`、`blacklist.json` 和 `whitelist.json` 文件来实现。本文档将详细介绍这些配置文件的作用以及用户可以自定义的配置选项。 + +## `config.toml` - 主配置文件 + +`config.toml` 是 `ghproxy` 的主配置文件，采用 TOML 格式。您可以通过修改此文件来定制 `ghproxy` 的各项功能，例如服务器端口、连接设置、Git 克隆模式、日志级别、认证方式、黑白名单以及限速策略等。 + +以下是 `config.toml` 文件的详细配置项说明： + +```toml name=config/config.toml +[server] +host = "0.0.0.0" +port = 8080 +sizeLimit = 125 # MB +H2C = true +cors = "*" # "*"/"" -> "*" ; "nil" -> "" ; +debug = false + +[httpc] +mode = "auto" # "auto" or "advanced" +maxIdleConns = 100 # only for advanced mode +maxIdleConnsPerHost = 60 # only for advanced mode +maxConnsPerHost = 0 # only for advanced mode + +[gitclone] +mode = "bypass" # bypass / cache +smartGitAddr = "http://127.0.0.1:8080" +ForceH2C = false + +[shell] +editor = false +rewriteAPI = false + +[pages] +mode = "internal" # "internal" or "external" +theme = "bootstrap" # "bootstrap" or "nebula" +staticDir = "/data/www" + +[log] +logFilePath = "/data/ghproxy/log/ghproxy.log" +maxLogSize = 5 # MB +level = "info" # dump, debug, info, warn, error, none + +[auth] +method = "parameters" # "header" or "parameters" +token = "token" +key = "" +enabled = false +passThrough = false +ForceAllowApi = false + +[blacklist] +blacklistFile = "/data/ghproxy/config/blacklist.json" +enabled = false + +[whitelist] +enabled = false +whitelistFile = "/data/ghproxy/config/whitelist.json" + +[rateLimit] +enabled = false +rateMethod = "total" # "ip" or "total" +ratePerMinute = 180 +burst = 5 + +[outbound] +enabled = false +url = "socks5://127.0.0.1:1080" # "http://127.0.0.1:7890" +``` + +### 配置项详细说明 + +* **`[server]` - 服务器配置** + + * `host`: 监听地址。 + * 类型: 字符串 (`string`) + * 默认值: `"0.0.0.0"` (监听所有) + * 说明: 设置 `ghproxy` 监听的网络地址。通常设置为 `"0.0.0.0"` 以监听所有可用的网络接口。 + * `port`: 监听端口。 + * 类型: 整数 (`int`) + * 默认值: `8080` + * 说明: 设置 `ghproxy` 监听的端口号。 + * `sizeLimit`: 请求体大小限制。 + * 类型: 整数 (`int`) + * 默认值: `125` (MB) + * 说明: 限制允许接收的请求体最大大小，单位为 MB。用于防止过大的请求导致服务压力过大。 + * `H2C`: 是否启用 H2C (HTTP/2 Cleartext) 传输。 + * 类型: 布尔值 (`bool`) + * 默认值: `true` (启用) + * 说明: 启用后，允许客户端使用 HTTP/2 协议进行无加密传输，提升性能。 + * `cors`: CORS (跨域资源共享) 设置。 + * 类型: 字符串 (`string`) + * 默认值: `"*"` (允许所有来源) + * 可选值: + * `""` 或`"*"`: 允许所有来源跨域访问。 + * `"nil"`: 禁用 CORS。 + * 具体的域名: 例如 `"https://example.com"`，只允许来自指定域名的跨域请求。 + * 说明: 配置 CORS 策略，用于控制哪些域名可以跨域访问 `ghproxy` 服务。 + * `debug`: 是否启用调试模式。 + * 类型: 布尔值 (`bool`) + * 默认值: `false` (禁用) + * 说明: 启用后，`ghproxy` 会输出更详细的日志信息，用于开发和调试。 + +* **`[httpc]` - HTTP 客户端配置** + + * `mode`: HTTP 客户端模式。 + * 类型: 字符串 (`string`) + * 默认值: `"auto"` (自动模式) + * 可选值: + * `"auto"`: 自动模式，使用默认的 HTTP 客户端配置，适用于大多数场景。 + * `"advanced"`: 高级模式，允许自定义连接池参数，可以更精细地控制 HTTP 客户端的行为。 + * 说明: 选择 HTTP 客户端的运行模式。 + * `maxIdleConns`: 最大空闲连接数 (仅在高级模式下生效)。 + * 类型: 整数 (`int`) + * 默认值: `100` + * 说明: 设置 HTTP 客户端连接池中保持的最大空闲连接数。 + * `maxIdleConnsPerHost`: 每个主机最大空闲连接数 (仅在高级模式下生效)。 + * 类型: 整数 (`int`) + * 默认值: `60` + * 说明: 设置 HTTP 客户端连接池中，每个主机允许保持的最大空闲连接数。 + * `maxConnsPerHost`: 每个主机最大连接数 (仅在高级模式下生效)。 + * 类型: 整数 (`int`) + * 默认值: `0` (不限制) + * 说明: 设置 HTTP 客户端连接池中，每个主机允许建立的最大连接数。设置为 `0` 表示不限制。 + +* **`[gitclone]` - Git 克隆配置** + + * `mode`: Git 克隆模式。 + * 类型: 字符串 (`string`) + * 默认值: `"bypass"` (绕过模式) + * 可选值: + * `"bypass"`: 绕过模式，直接克隆 GitHub 仓库，不使用任何缓存加速。 + * `"cache"`: 缓存模式，使用智能 Git 服务加速克隆，需要配置 `smartGitAddr`。 + * 说明: 选择 Git 克隆的模式。 + * `smartGitAddr`: 智能 Git 服务地址 (仅在缓存模式下生效)。 + * 类型: 字符串 (`string`) + * 默认值: `"http://127.0.0.1:8080"` + * 说明: 当 `mode` 设置为 `"cache"` 时，需要配置智能 Git 服务的地址，用于加速 Git 克隆。 + * `ForceH2C`: 是否强制使用 H2C 连接到智能 Git 服务。 + * 类型: 布尔值 (`bool`) + * 默认值: `false` (不强制) + * 说明: 如果智能 Git 服务支持 H2C，可以设置为 `true` 以强制使用 H2C 连接，提升性能。 + +* **`[shell]` - Shell 嵌套加速功能配置** + + * `editor`: 是否启用编辑(嵌套加速)功能。 + * 类型: 布尔值 (`bool`) + * 默认值: `false` (禁用) + * 说明: 启用后, 会修改`.sh`文件内容以实现嵌套加速 + * `rewriteAPI`: 是否重写 API 地址。 + * 类型: 布尔值 (`bool`) + * 默认值: `false` (禁用) + * 说明: 启用后，`ghproxy` 会重写脚本内的Github API地址。 + +* **`[pages]` - Pages 服务配置** + + * `mode`: Pages 服务模式。 + * 类型: 字符串 (`string`) + * 默认值: `"internal"` (内置 Pages 服务) + * 可选值: + * `"internal"`: 使用 `ghproxy` 内置的 Pages 服务。 + * `"external"`: 使用外部 Pages 位置。 + * 说明: 选择 Pages 服务的运行模式。 + * `theme`: Pages 主题。 + * 类型: 字符串 (`string`) + * 默认值: `"bootstrap"` + * 可选值: 参看[GHProxy项目前端仓库](https://github.com/WJQSERVER-STUDIO/GHProxy-Frontend) + * 说明: 设置内置 Pages 服务使用的主题。 + * `staticDir`: 静态文件目录。 + * 类型: 字符串 (`string`) + * 默认值: `"/data/www"` + * 说明: 指定外置 Pages 服务使用的静态文件目录。 + +* **`[log]` - 日志配置** + + * `logFilePath`: 日志文件路径。 + * 类型: 字符串 (`string`) + * 默认值: `"/data/ghproxy/log/ghproxy.log"` + * 说明: 设置 `ghproxy` 日志文件的存储路径。 + * `maxLogSize`: 最大日志文件大小。 + * 类型: 整数 (`int`) + * 默认值: `5` (MB) + * 说明: 设置单个日志文件的最大大小，单位为 MB。当日志文件大小超过此限制时，会进行日志轮转。 + * `level`: 日志级别。 + * 类型: 字符串 (`string`) + * 默认值: `"info"` + * 可选值: `"dump"`, `"debug"`, `"info"`, `"warn"`, `"error"`, `"none"` + * 说明: 设置日志输出的级别。级别越高，输出的日志信息越少。 + * `"dump"`: 输出所有日志，包括最详细的调试信息。 + * `"debug"`: 输出调试信息、信息、警告和错误日志。 + * `"info"`: 输出信息、警告和错误日志。 + * `"warn"`: 输出警告和错误日志。 + * `"error"`: 仅输出错误日志。 + * `"none"`: 禁用所有日志输出。 + +* **`[auth]` - 认证配置** + + * `enabled`: 是否启用认证。 + * 类型: 布尔值 (`bool`) + * 默认值: `false` (禁用) + * 说明: 启用后，需要提供正确的认证信息才能访问 `ghproxy` 服务。 + * `method`: 认证方法。 + * 类型: 字符串 (`string`) + * 默认值: `"parameters"` (URL 参数) + * 可选值: `"header"` 或 `"parameters"` + * `"header"`: 通过请求头 `GH-Auth` 或自定义请求头 (通过 `key` 配置) 传递认证 Token。 + * `"parameters"`: 通过 URL 参数 `auth_token` 或自定义 URL 参数名 (通过 `Key` 配置) 传递认证 Token。 + * 说明: 选择认证信息传递的方式。 + * `key`: 自定义认证 Key。 + * 类型: 字符串 (`string`) + * 默认值: `""` (空字符串，使用默认的 `GH-Auth` 请求头或 `auth_token` URL 参数名) + * 说明: 可以自定义认证时使用的请求头名称或 URL 参数名。如果为空，则使用默认的 `GH-Auth` 请求头或 `auth_token` URL 参数名。 + * `token`: 认证 Token。 + * 类型: 字符串 (`string`) + * 默认值: `"token"` + * 说明: 设置认证时需要提供的 Token 值。 + * `passThrough`: 是否认证参数透穿到Github。 + * 类型: 布尔值 (`bool`) + * 默认值: `false` (不允许) + * 说明: 如果设置为 `true`，相关参数会被透穿到Github。 + * `ForceAllowApi`: 是否强制允许 API 访问。 + * 类型: 布尔值 (`bool`) + * 默认值: `false` (不强制允许) + * 说明: 如果设置为 `true`，则强制允许对 GitHub API 的访问，即使未启用认证或认证失败。 + +* **`[blacklist]` - 黑名单配置** + + * `enabled`: 是否启用黑名单。 + * 类型: 布尔值 (`bool`) + * 默认值: `false` (禁用) + * 说明: 启用后，`ghproxy` 将根据 `blacklist.json` 文件中的规则阻止对特定用户或仓库的访问。 + * `blacklistFile`: 黑名单文件路径。 + * 类型: 字符串 (`string`) + * 默认值: `"/data/ghproxy/config/blacklist.json"` + * 说明: 指定黑名单配置文件的路径。 + +* **`[whitelist]` - 白名单配置** + + * `enabled`: 是否启用白名单。 + * 类型: 布尔值 (`bool`) + * 默认值: `false` (禁用) + * 说明: 启用后，`ghproxy` 将只允许访问 `whitelist.json` 文件中规则指定的用户或仓库。白名单的优先级高于黑名单。 + * `whitelistFile`: 白名单文件路径。 + * 类型: 字符串 (`string`) + * 默认值: `"/data/ghproxy/config/whitelist.json"` + * 说明: 指定白名单配置文件的路径。 + +* **`[rateLimit]` - 限速配置** + + * `enabled`: 是否启用限速。 + * 类型: 布尔值 (`bool`) + * 默认值: `false` (禁用) + * 说明: 启用后，`ghproxy` 将根据配置的策略限制请求速率，防止服务被滥用。 + * `rateMethod`: 限速方法。 + * 类型: 字符串 (`string`) + * 默认值: `"total"` (全局限速) + * 可选值: `"ip"` 或 `"total"` + * `"ip"`: 基于客户端 IP 地址进行限速，每个 IP 地址都有独立的速率限制。 + * `"total"`: 全局限速，所有客户端共享同一个速率限制。 + * 说明: 选择限速的策略。 + * `ratePerMinute`: 每分钟允许的请求数。 + * 类型: 整数 (`int`) + * 默认值: `180` + * 说明: 设置每分钟允许通过的最大请求数。 + * `burst`: 突发请求数。 + * 类型: 整数 (`int`) + * 默认值: `5` + * 说明: 允许在短时间内超过 `ratePerMinute` 的突发请求数。 + +* **`[outbound]` - 出站代理配置** + + * `enabled`: 是否启用出站代理。 + * 类型: 布尔值 (`bool`) + * 默认值: `false` (禁用) + * 说明: 启用后，`ghproxy` 将通过配置的代理服务器转发所有出站请求。 + * `url`: 出站代理 URL。 + * 类型: 字符串 (`string`) + * 默认值: `"socks5://127.0.0.1:1080"` + * 支持协议: `socks5://` 和 `http://` + * 说明: 设置出站代理服务器的 URL。支持 SOCKS5 和 HTTP 代理协议。 + +## `blacklist.json` - 黑名单配置 + +`blacklist.json` 文件用于配置黑名单规则，阻止对特定用户或仓库的访问。 + +```json name=config/blacklist.json +{ + "blacklist": [ + "eviluser", + "spamuser/bad-repo", + "malwareuser/*" + ] +} +``` + +### 黑名单规则说明 + +* `blacklist`: 一个 JSON 数组，包含黑名单规则，每条规则为一个字符串。 + * **用户名**: 例如 `"eviluser"`，阻止所有名为 `eviluser` 的用户的访问。 + * **仓库名**: 例如 `"spamuser/bad-repo"`，阻止访问 `spamuser` 用户下的 `bad-repo` 仓库。 + * **通配符**: 例如 `"malwareuser/*"`，使用 `*` 通配符，阻止访问 `malwareuser` 用户下的所有仓库。 + * **缩略写法**: 例如 `"example"`, 等同于 `"example/*"`， 允许访问 `example` 用户下的所有仓库。 + +## `whitelist.json` - 白名单配置 + +`whitelist.json` 文件用于配置白名单规则，只允许访问白名单中指定的用户或仓库。白名单的优先级高于黑名单，如果一个请求同时匹配黑名单和白名单，则白名单生效，请求将被允许。 + +```json name=config/whitelist.json +{ + "whitelist": [ + "white/list", + "white/test1", + "example/*", + "example" + ] +} +``` + +### 白名单规则说明 + +* `whitelist`: 一个 JSON 数组，包含白名单规则，每条规则为一个字符串。 + * **仓库名**: 例如 `"white/list"`，允许访问 `white` 用户下的 `list` 仓库。 + * **仓库名**: 例如 `"white/test1"`，允许访问 `white` 用户下的 `test1` 仓库。 + * **通配符**: 例如 `"example/*"`，使用 `*` 通配符，允许访问 `example` 用户下的所有仓库。 + * **缩略写法**: 例如 `"example"`, 等同于 `"example/*"`， 允许访问 `example` 用户下的所有仓库。 + +--- \ No newline at end of file diff --git a/docs/flag.md b/docs/flag.md new file mode 100644 index 0000000..cc09c4c --- /dev/null +++ b/docs/flag.md @@ -0,0 +1,24 @@ +# Flag + +GHProxy接受以下flag传入 + +```bash +root@root:/data/ghproxy$ ghproxy -h + -c string + config file path (default "/data/ghproxy/config/config.toml") + -cfg value + exit + -h show help message and exit + -v show version and exit +``` + +- `-c` + 类型: `string` + 默认值: `/data/ghproxy/config/config.toml` + 示例: `ghproxy -c /data/ghproxy/demo.toml` +- `-cfg` + 已弃用, 被`-c`替代 +- `-h` + 显示帮助信息 +- `-v` + 显示版本号 diff --git a/docs/menu.md b/docs/menu.md new file mode 100644 index 0000000..bbb10e3 --- /dev/null +++ b/docs/menu.md @@ -0,0 +1,13 @@ +## GHProxy 文档 + +### 配置文件 + +https://github.com/WJQSERVER-STUDIO/ghproxy/blob/main/docs/config.md + +### Flag + +https://github.com/WJQSERVER-STUDIO/ghproxy/blob/main/docs/flag.md + +### 部署 + +参看 https://blog.wjqserver.com/post/ghproxy-deploy-with-smart-git/ \ No newline at end of file diff --git a/go.mod b/go.mod index 7a1d994..fe29791 100644 --- a/go.mod +++ b/go.mod @@ -1,6 +1,6 @@ module ghproxy -go 1.24.1 +go 1.24.2 require ( github.com/BurntSushi/toml v1.5.0 diff --git a/main.go b/main.go index 4a7d009..aabb054 100644 --- a/main.go +++ b/main.go @@ -36,21 +36,13 @@ var ( runMode string limiter *rate.RateLimiter iplimiter *rate.IPRateLimiter - showVersion bool // 新增的版本号标志 - showHelp bool // 新增的帮助标志 + showVersion bool + showHelp bool ) var ( //go:embed pages/* pagesFS embed.FS - /* - //go:embed pages/bootstrap/* - BootstrapPagesFS embed.FS - //go:embed pages/nebula/* - NebulaPagesFS embed.FS - //go:embed pages/design/* - DesignPagesFS embed.FS - */ ) var ( @@ -101,7 +93,7 @@ func readFlag() { fmt.Fprintf(os.Stderr, " %s\n", flag) } if len(invalidFlags) > 0 { - os.Exit(2) // 使用非零状态码退出，表示有错误 + os.Exit(2) } } diff --git a/middleware/loggin/loggin.go b/middleware/loggin/loggin.go index 95f6668..15f17ee 100644 --- a/middleware/loggin/loggin.go +++ b/middleware/loggin/loggin.go @@ -20,18 +20,13 @@ var ( // 日志中间件 func Middleware() app.HandlerFunc { return func(ctx context.Context, c *app.RequestContext) { - startTime := time.Now() // 请求开始处理前记录当前时间作为开始时间 + startTime := time.Now() - c.Next(ctx) // 调用 Next() 执行后续的 Handler + c.Next(ctx) - endTime := time.Now() // 请求处理完成后记录当前时间作为结束时间 - timingResults := endTime.Sub(startTime) // 计算时间差，得到请求处理耗时 (Duration 类型) + endTime := time.Now() + timingResults := endTime.Sub(startTime) - // 记录日志 IP METHOD URL USERAGENT PROTOCOL STATUS TIMING - // %s %s %s %s %s %d %s 分别对应: ClientIP, Method, Protolcol, Path, UserAgent, StatusCode, timingResults (需要格式化) - // %v 可以通用地格式化 time.Duration 类型 logInfo("%s %s %s %s %s %d %v ", c.ClientIP(), c.Method(), c.Request.Header.GetProtocol(), string(c.Path()), c.Request.Header.UserAgent(), c.Response.StatusCode(), timingResults) - - //logInfo("%s %s %s %s %d %v ", c.ClientIP(), c.Method(), c.Path(), c.Request.Header.UserAgent(), c.Response.StatusCode(), timingResults) } } diff --git a/proxy/chunkreq.go b/proxy/chunkreq.go index fdd54a5..dd2b1ee 100644 --- a/proxy/chunkreq.go +++ b/proxy/chunkreq.go @@ -10,7 +10,6 @@ import ( "strconv" "github.com/cloudwego/hertz/pkg/app" - //hresp "github.com/cloudwego/hertz/pkg/protocol/http1/resp" ) func ChunkedProxyRequest(ctx context.Context, c *app.RequestContext, u string, cfg *config.Config, matcher string) { @@ -31,7 +30,6 @@ func ChunkedProxyRequest(ctx context.Context, c *app.RequestContext, u string, c HandleError(c, fmt.Sprintf("Failed to send request: %v", err)) return } - //defer headResp.Body.Close() defer func(Body io.ReadCloser) { if err := Body.Close(); err != nil { logError("Failed to close response body: %v", err) @@ -68,7 +66,6 @@ func ChunkedProxyRequest(ctx context.Context, c *app.RequestContext, u string, c HandleError(c, fmt.Sprintf("Failed to send request: %v", err)) return } - //defer resp.Body.Close() // 错误处理(404) if resp.StatusCode == 404 { @@ -115,7 +112,6 @@ func ChunkedProxyRequest(ctx context.Context, c *app.RequestContext, u string, c } c.Status(resp.StatusCode) - //c.Response.HijackWriter(hresp.NewChunkedBodyWriter(&c.Response, c.GetWriter())) if MatcherShell(u) && matchString(matcher, matchedMatchers) && cfg.Shell.Editor { // 判断body是不是gzip @@ -127,7 +123,6 @@ func ChunkedProxyRequest(ctx context.Context, c *app.RequestContext, u string, c logInfo("Is Shell: %s %s %s %s %s", c.ClientIP(), method, u, c.Request.Header.Get("User-Agent"), c.Request.Header.GetProtocol()) c.Header("Content-Length", "") - //err := ProcessLinksAndWriteChunked(resp.Body, compress, string(c.Request.Host()), cfg, c) reader, _, err := processLinks(resp.Body, compress, string(c.Request.Host()), cfg) c.SetBodyStream(reader, -1) @@ -136,18 +131,6 @@ func ChunkedProxyRequest(ctx context.Context, c *app.RequestContext, u string, c return } } else { - //err = hwriter.Writer(resp.Body, c) - //writer := c.Response.BodyWriter() - - /* - _, err := copyb.Copy(c.Response.BodyWriter(), resp.Body) - if err != nil { - logError("%s %s %s %s %s Failed to copy response body: %v", c.ClientIP(), method, u, c.Request.Header.Get("User-Agent"), c.Request.Header.GetProtocol(), err) - return - } else { - c.Flush() // 确保刷入 - } - */ c.SetBodyStream(resp.Body, -1) } diff --git a/proxy/error.go b/proxy/error.go index fc6470f..d8b27c5 100644 --- a/proxy/error.go +++ b/proxy/error.go @@ -1,7 +1,6 @@ package proxy import ( - "fmt" "net/http" "github.com/WJQSERVER-STUDIO/go-utils/logger" @@ -19,6 +18,6 @@ var ( ) func HandleError(c *app.RequestContext, message string) { - c.String(http.StatusInternalServerError, fmt.Sprintf("server error %v", message)) + c.JSON(http.StatusInternalServerError, map[string]string{"error": message}) logError(message) } diff --git a/proxy/gitreq.go b/proxy/gitreq.go index 6e15e89..0826a58 100644 --- a/proxy/gitreq.go +++ b/proxy/gitreq.go @@ -44,7 +44,6 @@ func GitReq(ctx context.Context, c *app.RequestContext, u string, cfg *config.Co } setRequestHeaders(c, req) removeWSHeader(req) - //reWriteEncodeHeader(req) AuthPassThrough(c, cfg, req) resp, err = gitclient.Do(req) @@ -60,7 +59,6 @@ func GitReq(ctx context.Context, c *app.RequestContext, u string, cfg *config.Co } setRequestHeaders(c, req) removeWSHeader(req) - //reWriteEncodeHeader(req) AuthPassThrough(c, cfg, req) resp, err = client.Do(req) @@ -69,14 +67,6 @@ func GitReq(ctx context.Context, c *app.RequestContext, u string, cfg *config.Co return } } - /* - //defer resp.Body.Close() - defer func(Body io.ReadCloser) { - if err := Body.Close(); err != nil { - logError("Failed to close response body: %v", err) - } - }(resp.Body) - */ contentLength := resp.Header.Get("Content-Length") if contentLength != "" { @@ -123,18 +113,6 @@ func GitReq(ctx context.Context, c *app.RequestContext, u string, cfg *config.Co c.Response.Header.Set("Pragma", "no-cache") c.Response.Header.Set("Expires", "0") } + c.SetBodyStream(resp.Body, -1) - //err = hwriter.Writer(resp.Body, c) - /* - _, err = copyb.Copy(c.Response.BodyWriter(), resp.Body) - - if err != nil { - logError("%s %s %s %s %s Failed to copy response body: %v", c.ClientIP(), method, u, c.Request.Header.Get("User-Agent"), c.Request.Header.GetProtocol(), err) - return - } else { - - c.Flush() // 确保刷入 - } - */ - } diff --git a/proxy/reqheader.go b/proxy/reqheader.go index 5e19109..3421722 100644 --- a/proxy/reqheader.go +++ b/proxy/reqheader.go @@ -17,33 +17,3 @@ func removeWSHeader(req *http.Request) { req.Header.Del("Upgrade") req.Header.Del("Connection") } - -/* -func reWriteEncodeHeader(req *http.Request) { - - if isGzipAccepted(req.Header) { - req.Header.Set("Content-Encoding", "gzip") - req.Header.Set("Accept-Encoding", "gzip") - } else { - req.Header.Del("Content-Encoding") - req.Header.Del("Accept-Encoding") - } - -} - -// isGzipAccepted 检查 Accept-Encoding 头部中是否包含 gzip -func isGzipAccepted(header http.Header) bool { - // 获取 Accept-Encoding 的值 - encodings := header["Accept-Encoding"] - for _, encoding := range encodings { - // 将 encoding 字符串拆分为多个编码 - for _, enc := range strings.Split(encoding, ",") { - // 去除空格并检查是否为 gzip - if strings.TrimSpace(enc) == "gzip" { - return true - } - } - } - return false -} -*/ diff --git a/rate/rate.go b/rate/rate.go index 409b2c1..a1237b2 100644 --- a/rate/rate.go +++ b/rate/rate.go @@ -10,7 +10,7 @@ import ( // 日志输出 var ( logw = logger.Logw - LogDump = logger.LogDump + logDump = logger.LogDump logDebug = logger.LogDebug logInfo = logger.LogInfo logWarning = logger.LogWarning