一个APP动态代理N个服务器 · WSH032/fastapi-proxy-lib · Discussion #35

shizheng163
Apr 11, 2024

您好,这个工具库支持这几种种场景吗?

时间比较匆忙,未能仔细阅读您的源码就提出问题,希望理解。

1. 一个fastapi实现类似nginx那种同一个app里的路由代理到不同的服务器。

例如

/api/users/ -> 代理到用户服务器
/api/tasks/ -> 代理到任务服务器

2. 根据请求启动服务器,并提供反向代理

/api/viewers/1/start 启动一个viewer服务器
/viewers/1/ -> 代理到上一步启动的服务器

3. 同时支持代理http和ws

一个fastapi的app同时支持http和ws

2024年04月11日10:28:38更新
ForwardHttpProxy 好像可以解决动态代理HTTP请求的需求
但是websocket转发似乎不可以?

Replies: 1 comment 2 replies

WSH032
Apr 11, 2024
Maintainer

Hi @shizheng163

一个fastapi实现类似nginx那种同一个app里的路由代理到不同的服务器

支持,参考这部分文档

from contextlib import asynccontextmanager
from typing import AsyncIterator
from fastapi import FastAPI
from fastapi_proxy_lib.core.http import ReverseHttpProxy
from httpx import AsyncClient
from starlette.requests import Request
user_proxy = ReverseHttpProxy(AsyncClient(), base_url="http://www.user-server.com/")
task_proxy = ReverseHttpProxy(AsyncClient(), base_url="http://www.task-server.com/")
@asynccontextmanager
async def close_proxy_event(_: FastAPI) -> AsyncIterator[None]:
 """Close proxy."""
 yield
 await user_proxy.aclose()
 await task_proxy.aclose()
app = FastAPI(lifespan=close_proxy_event)
@app.get("/api/users/{path:path}")
async def _(request: Request):
 return await user_proxy.proxy(request=request)
@app.get("/api/tasks/{path:path}")
async def _(request: Request):
 return await task_proxy.proxy(request=request)