Implement a startup script which runs once on microservice start.
This commit is contained in:
@@ -3,6 +3,15 @@ from slingshot_microservice import Microservice
|
||||
from typing import Generator
|
||||
from sqlalchemy.engine.base import Connection
|
||||
|
||||
|
||||
def startup(
|
||||
read_file: ReadFileFn,
|
||||
write_file: WriteFileFn,
|
||||
_connection: Connection,
|
||||
) -> Generator[tuple[int, bool | int | str], None, None]:
|
||||
if False:
|
||||
yield (0, True)
|
||||
|
||||
def process(
|
||||
request: int,
|
||||
read_file: ReadFileFn,
|
||||
@@ -16,5 +25,5 @@ def process(
|
||||
yield (request, True)
|
||||
|
||||
|
||||
microservice = Microservice("simple-py-microservice", "sys-map.slingshot.cv", process)
|
||||
microservice = Microservice("simple-py-microservice", "sys-map.slingshot.cv", startup, process)
|
||||
microservice.start()
|
||||
|
||||
Reference in New Issue
Block a user