feast/sdk/python/feast/ui_server.py at swapyaml · feast-dev/feast · GitHub

History

122 lines (104 loc) · 3.25 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

import json

import threading

from importlib import resources as importlib_resources

from typing import Callable, Optional

import uvicorn

from fastapi import FastAPI, Response

from fastapi.middleware.cors import CORSMiddleware

from fastapi.staticfiles import StaticFiles

import feast

def get_app(

store: "feast.FeatureStore",

project_id: str,

registry_ttl_secs: int,

root_path: str = "",

app = FastAPI()

app.add_middleware(

CORSMiddleware,

allow_origins=["*"],

allow_credentials=True,

allow_methods=["*"],

allow_headers=["*"],

)

# Asynchronously refresh registry, notifying shutdown and canceling the active timer if the app is shutting down

registry_proto = None

shutting_down = False

active_timer: Optional[threading.Timer] = None

def async_refresh():

store.refresh_registry()

nonlocal registry_proto

registry_proto = store.registry.proto()

if shutting_down:

return

nonlocal active_timer

active_timer = threading.Timer(registry_ttl_secs, async_refresh)

active_timer.start()

@app.on_event("shutdown")

def shutdown_event():

nonlocal shutting_down

shutting_down = True

if active_timer:

active_timer.cancel()

async_refresh()

ui_dir_ref = importlib_resources.files(__spec__.parent) / "ui/build/" # type: ignore[name-defined, arg-type]

with importlib_resources.as_file(ui_dir_ref) as ui_dir:

# Initialize with the projects-list.json file

with ui_dir.joinpath("projects-list.json").open(mode="w") as f:

projects_dict = {

"projects": [

{

"name": "Project",

"description": "Test project",

"id": project_id,

"registryPath": f"{root_path}/registry",

}

]

}

f.write(json.dumps(projects_dict))

@app.get("/registry")

def read_registry():

return Response(

content=registry_proto.SerializeToString(),

media_type="application/octet-stream",

)

# For all other paths (such as paths that would otherwise be handled by react router), pass to React

@app.api_route("/p/{path_name:path}", methods=["GET"])

def catch_all():

filename = ui_dir.joinpath("index.html")

with open(filename) as f:

content = f.read()

return Response(content, media_type="text/html")

app.mount(

"/",

StaticFiles(directory=ui_dir, html=True),

name="site",

)

return app

def start_server(

store: "feast.FeatureStore",

host: str,

port: int,

get_registry_dump: Callable,

project_id: str,

registry_ttl_sec: int,

root_path: str = "",

tls_key_path: str = "",

tls_cert_path: str = "",

app = get_app(

store,

project_id,

registry_ttl_sec,

root_path,

)

if tls_key_path and tls_cert_path:

uvicorn.run(

app,

host=host,

port=port,

ssl_keyfile=tls_key_path,

ssl_certfile=tls_cert_path,

)

else:

uvicorn.run(app, host=host, port=port)

feast/sdk/python/feast/ui_server.py at swapyaml · feast-dev/feast · GitHub

Uh oh!

Expand file tree

Breadcrumbs

ui_server.py

Latest commit

History

Breadcrumbs

ui_server.py

File metadata and controls