如何使用http4s Server与Client实现带请求校验的代理服务

我来帮你梳理清楚用http4s实现反向代理的正确姿势，同时解决你遇到的连接拒绝和EOF错误问题。

核心思路回顾 #

你的需求很明确：http4s服务端接收请求后先执行校验，校验通过就转发到目标服务器，失败则返回403。这里的关键是正确管理HTTP客户端资源、修改请求目标地址与头信息，以及避免流处理异常。

先分析你之前代码的问题 #

1. HttpClient初始化错误：你最初在请求处理逻辑里每次都创建HttpClient，这不仅低效，还会导致连接池混乱。HttpClient是重量级资源，应该全局单例管理，用Resource来确保资源正确释放。
2. 请求地址修改不彻底：之前的代码可能没完全替换原请求的目标地址，导致客户端还是尝试连接localhost（原服务地址），引发连接拒绝。
3. 请求头处理不完善：没有正确过滤掉与连接相关的头（比如Connection、Transfer-Encoding），也没确保Host头指向目标服务器，这会导致目标服务器处理异常，出现EOF错误。

完整解决方案代码 #

下面是可以直接运行的完整实现，包含所有最佳实践：

import org.http4s._
import org.http4s.dsl.io._
import org.http4s.client._
import org.http4s.client.blaze.Http1Client
import org.http4s.headers._
import org.http4s.Uri.{Authority, RegName}
import cats.effect.{IO, IOApp, Resource}

object ProxyServer extends IOApp {
  // 全局单例HttpClient，用Resource管理生命周期，确保资源正确释放
  private val httpClient: Resource[IO, Client[IO]] = Http1Client[IO]()

  // 自定义请求校验逻辑：这里替换成你的实际业务校验（比如token验证、IP白名单等）
  // 用IO包裹，支持异步校验操作（比如查数据库、调用鉴权接口）
  private def verifyRequest(req: Request[IO]): IO[Boolean] = IO {
    // 示例：简单返回true，实际业务逻辑写在这里
    true
  }

  // 构建转发请求：修改目标地址、过滤无效头、设置正确Host头
  private def buildProxiedRequest(
    req: Request[IO],
    targetHost: String,
    targetPort: Int,
    targetScheme: Uri.Scheme = Uri.Scheme.http
  ): Request[IO] = {
    // 替换请求的目标地址
    val targetAuthority = Authority(
      host = RegName(targetHost),
      port = Some(targetPort)
    )
    val proxiedUri = req.uri.copy(
      scheme = Some(targetScheme),
      authority = Some(targetAuthority)
    )

    // 过滤掉不需要转发的头：这些头与当前连接绑定，不应该传递给目标服务器
    val filteredHeaders = req.headers.filterNot { h =>
      val headerName = h.name
      headerName == CaseInsensitiveString("Connection") ||
      headerName == CaseInsensitiveString("Keep-Alive") ||
      headerName == CaseInsensitiveString("Proxy-Authenticate") ||
      headerName == CaseInsensitiveString("Proxy-Authorization") ||
      headerName == CaseInsensitiveString("TE") ||
      headerName == CaseInsensitiveString("Trailer") ||
      headerName == CaseInsensitiveString("Transfer-Encoding") ||
      headerName == CaseInsensitiveString("Upgrade")
    }
    // 设置目标服务器的Host头
    val proxiedHeaders = filteredHeaders.put(Host(targetHost, targetPort))

    // 构建最终转发请求
    req.withUri(proxiedUri).withHeaders(proxiedHeaders)
  }

  // 定义代理服务路由
  private val proxyService: HttpRoutes[IO] = HttpRoutes.of[IO] { case req =>
    verifyRequest(req).flatMap {
      case true =>
        // 使用Resource管理HttpClient，确保每次请求都正确使用连接池
        httpClient.use { client =>
          val proxiedReq = buildProxiedRequest(req, "scala-lang.org", 80)
          // 用client.run转发请求，并正确处理响应流
          client.run(proxiedReq).use(IO.pure)
        }
      case false =>
        Forbidden("请求校验失败，拒绝访问")
    }
  }

  // 启动http4s服务器
  override def run(args: List[String]): IO[ExitCode] = {
    import org.http4s.server.blaze.BlazeServerBuilder
    import scala.concurrent.ExecutionContext.global

    BlazeServerBuilder[IO](global)
      .bindHttp(28080, "localhost")
      .withHttpApp(proxyService.orNotFound)
      .resource
      .use(_ => IO.never) // 保持服务器运行
      .as(ExitCode.Success)
  }
}

关键细节解释 #

1. HttpClient资源管理：用Resource包裹HttpClient，确保它在服务器生命周期内只初始化一次，请求结束后正确释放连接。
2. 请求地址修改：完全替换原请求的scheme、authority，确保客户端连接到目标服务器而非原服务地址。
3. 请求头处理：过滤掉与连接相关的头，设置正确的Host头，避免目标服务器因无效头返回异常。
4. 校验逻辑异步化：用IO包裹校验逻辑，支持异步操作（比如查询数据库），符合http4s的IO异步模型。

针对你遇到的错误的修复 #

• 连接拒绝错误：之前的代码可能没完全替换请求的目标地址，现在的buildProxiedRequest函数确保了请求指向目标服务器（比如scala-lang.org）。
• Premature EOF错误：通过过滤无效头、使用client.run正确处理响应流，避免了流提前关闭的问题。

额外注意事项 #

• 如果目标服务器是HTTPS，只需将targetScheme改为Uri.Scheme.https，并确保HttpClient支持HTTPS（http4s的默认Http1Client需要额外配置SSL，或者使用ember-client）。
• 可以根据业务需求调整校验逻辑，比如添加token验证、IP白名单等。
• 对于大文件上传/下载，http4s的异步流处理会自动适配，无需额外配置。

内容的提问来源于stack exchange，提问作者LoranceChen