junte/Model
1
0
Fork 0
mirror of https://github.com/Hestia-Homes/Model.git synced 2026-07-12 13:29:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

relaxing connections and temp connection for lamnda

Browse source
This commit is contained in:
Khalim Conn-Kowlessar 2025-12-14 23:36:52 +08:00
parent 873ca92948
commit 35085d25d3
2 changed files with 38 additions and 25 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
backend/app/db/connection.py
View file

@ -18,8 +18,8 @@ db_string = connection_string.format(
# each lambda doesn't hog all connections # each lambda doesn't hog all connections
db_engine = create_engine( db_engine = create_engine(
db_string, db_string,
pool_size=1, pool_size=3,
max_overflow=0, # Limit the number of extra connections. With this and pool size, we allow 1 connection per lambda max_overflow=5, # Limit the number of extra connections. With this and pool size, we allow 1 connection per lambda
pool_pre_ping=True, pool_pre_ping=True,
pool_recycle=300, # Forces SQLAlchemy to close and reopen any connection older than 300 seconds pool_recycle=300, # Forces SQLAlchemy to close and reopen any connection older than 300 seconds
) )

59
backend/app/plan/router.py
View file

@ -2,7 +2,8 @@ import boto3
import json import json
import math import math
import asyncio import asyncio
import random from contextlib import contextmanager
from sqlmodel import Session
from datetime import datetime from datetime import datetime
@ -29,6 +30,19 @@ router = APIRouter(
sqs_client = boto3.client("sqs") sqs_client = boto3.client("sqs")
@contextmanager
def db_session():
session = Session(db_engine)
try:
yield session
session.commit()
except Exception:
session.rollback()
raise
finally:
session.close()
@router.post("/trigger", status_code=202) @router.post("/trigger", status_code=202)
async def trigger_plan_entrypoint(body: PlanTriggerRequest): async def trigger_plan_entrypoint(body: PlanTriggerRequest):
""" """
@ -57,28 +71,27 @@ async def trigger_plan_entrypoint(body: PlanTriggerRequest):
scenario_id = data.get("scenario_id") scenario_id = data.get("scenario_id")
if not scenario_id: if not scenario_id:
created_at = datetime.now().isoformat() created_at = datetime.now().isoformat()
session = sessionmaker(bind=db_engine)() with db_session() as session:
# Create a new scenario
# Create a new scenario new_scenario = create_scenario(
new_scenario = create_scenario( session=session,
session=session, scenario={
scenario={ "name": body.scenario_name,
"name": body.scenario_name, "created_at": created_at,
"created_at": created_at, "budget": body.budget,
"budget": body.budget, "portfolio_id": body.portfolio_id,
"portfolio_id": body.portfolio_id, "housing_type": body.housing_type,
"housing_type": body.housing_type, "goal": body.goal,
"goal": body.goal, "goal_value": body.goal_value,
"goal_value": body.goal_value, "trigger_file_path": body.trigger_file_path,
"trigger_file_path": body.trigger_file_path, "already_installed_file_path": body.already_installed_file_path,
"already_installed_file_path": body.already_installed_file_path, "patches_file_path": body.patches_file_path,
"patches_file_path": body.patches_file_path, "non_invasive_recommendations_file_path": body.non_invasive_recommendations_file_path,
"non_invasive_recommendations_file_path": body.non_invasive_recommendations_file_path, "exclusions": body.exclusions,
"exclusions": body.exclusions, "multi_plan": body.multi_plan
"multi_plan": body.multi_plan }
} )
) scenario_id = new_scenario.id
scenario_id = new_scenario.id
# Insert the scenario ID into the data payload # Insert the scenario ID into the data payload
data["scenario_id"] = scenario_id data["scenario_id"] = scenario_id