mirror of
https://github.com/osm-search/Nominatim.git
synced 2026-02-16 15:47:58 +00:00
move server route creation into async function
This commit is contained in:
@@ -11,6 +11,7 @@ from typing import Any, Optional, Mapping, Callable, cast, Coroutine, Dict, Awai
|
||||
from pathlib import Path
|
||||
import datetime as dt
|
||||
import asyncio
|
||||
import contextlib
|
||||
|
||||
from starlette.applications import Starlette
|
||||
from starlette.routing import Route
|
||||
@@ -66,7 +67,7 @@ class ParamWrapper(ASGIAdaptor):
|
||||
return cast(Configuration, self.request.app.state.API.config)
|
||||
|
||||
def formatting(self) -> FormatDispatcher:
|
||||
return cast(FormatDispatcher, self.request.app.state.API.formatter)
|
||||
return cast(FormatDispatcher, self.request.app.state.formatter)
|
||||
|
||||
|
||||
def _wrap_endpoint(func: EndpointFunc)\
|
||||
@@ -132,14 +133,6 @@ def get_application(project_dir: Path,
|
||||
"""
|
||||
config = Configuration(project_dir, environ)
|
||||
|
||||
routes = []
|
||||
legacy_urls = config.get_bool('SERVE_LEGACY_URLS')
|
||||
for name, func in api_impl.ROUTES:
|
||||
endpoint = _wrap_endpoint(func)
|
||||
routes.append(Route(f"/{name}", endpoint=endpoint))
|
||||
if legacy_urls:
|
||||
routes.append(Route(f"/{name}.php", endpoint=endpoint))
|
||||
|
||||
middleware = []
|
||||
if config.get_bool('CORS_NOACCESSCONTROL'):
|
||||
middleware.append(Middleware(CORSMiddleware,
|
||||
@@ -156,14 +149,26 @@ def get_application(project_dir: Path,
|
||||
asyncio.TimeoutError: timeout_error
|
||||
}
|
||||
|
||||
async def _shutdown() -> None:
|
||||
@contextlib.asynccontextmanager
|
||||
async def lifespan(app: Starlette) -> None:
|
||||
app.state.API = NominatimAPIAsync(project_dir, environ)
|
||||
config = app.state.API.config
|
||||
|
||||
legacy_urls = config.get_bool('SERVE_LEGACY_URLS')
|
||||
for name, func in api_impl.ROUTES:
|
||||
endpoint = _wrap_endpoint(func)
|
||||
app.routes.append(Route(f"/{name}", endpoint=endpoint))
|
||||
if legacy_urls:
|
||||
app.routes.append(Route(f"/{name}.php", endpoint=endpoint))
|
||||
|
||||
yield
|
||||
|
||||
await app.state.API.close()
|
||||
|
||||
app = Starlette(debug=debug, routes=routes, middleware=middleware,
|
||||
app = Starlette(debug=debug, middleware=middleware,
|
||||
exception_handlers=exceptions,
|
||||
on_shutdown=[_shutdown])
|
||||
lifespan=lifespan)
|
||||
|
||||
app.state.API = NominatimAPIAsync(project_dir, environ)
|
||||
app.state.formatter = load_format_dispatcher('v1', project_dir)
|
||||
|
||||
return app
|
||||
|
||||
Reference in New Issue
Block a user