happyz
/
fastapi
mirror of https://github.com/tiangolo/fastapi.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

⚡ Make Swagger UI, ReDoc and OpenAPI handlers be coroutines to improve performance (#241)

This commit is contained in:
Trim21 2019-05-20 22:34:33 +08:00 committed by Sebastián Ramírez
parent 3cf92a156c
commit 01b43e6e25
1 changed files with 21 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
fastapi/applications.py
View File

@ -9,7 +9,7 @@ from starlette.applications import Starlette
from starlette.exceptions import ExceptionMiddleware, HTTPException from starlette.exceptions import ExceptionMiddleware, HTTPException
from starlette.middleware.errors import ServerErrorMiddleware from starlette.middleware.errors import ServerErrorMiddleware
from starlette.requests import Request from starlette.requests import Request
from starlette.responses import JSONResponse, Response from starlette.responses import HTMLResponse, JSONResponse, Response
from starlette.routing import BaseRoute from starlette.routing import BaseRoute
@ -79,29 +79,28 @@ class FastAPI(Starlette):
def setup(self) -> None: def setup(self) -> None:
if self.openapi_url: if self.openapi_url:
self.add_route(
self.openapi_url, async def openapi(req: Request) -> JSONResponse:
lambda req: JSONResponse(self.openapi()), return JSONResponse(self.openapi())
include_in_schema=False,
) self.add_route(self.openapi_url, openapi, include_in_schema=False)
openapi_url = self.openapi_prefix + self.openapi_url
if self.openapi_url and self.docs_url: if self.openapi_url and self.docs_url:
self.add_route(
self.docs_url, async def swagger_ui_html(req: Request) -> HTMLResponse:
lambda r: get_swagger_ui_html( return get_swagger_ui_html(
openapi_url=self.openapi_prefix + self.openapi_url, openapi_url=openapi_url, title=self.title + " - Swagger UI"
title=self.title + " - Swagger UI",
),
include_in_schema=False,
) )
self.add_route(self.docs_url, swagger_ui_html, include_in_schema=False)
if self.openapi_url and self.redoc_url: if self.openapi_url and self.redoc_url:
self.add_route(
self.redoc_url, async def redoc_html(req: Request) -> HTMLResponse:
lambda r: get_redoc_html( return get_redoc_html(
openapi_url=self.openapi_prefix + self.openapi_url, openapi_url=openapi_url, title=self.title + " - ReDoc"
title=self.title + " - ReDoc",
),
include_in_schema=False,
) )
self.add_route(self.redoc_url, redoc_html, include_in_schema=False)
self.add_exception_handler(HTTPException, http_exception) self.add_exception_handler(HTTPException, http_exception)
def add_api_route( def add_api_route(