mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-20 15:31:15 -05:00
48 lines
1.7 KiB
Python
48 lines
1.7 KiB
Python
from datetime import datetime
|
|
|
|
from sqlalchemy.dialects.postgresql import insert
|
|
|
|
from app import db
|
|
from app.dao.dao_utils import autocommit
|
|
from app.models import FactProcessingTime
|
|
|
|
|
|
@autocommit
|
|
def insert_update_processing_time(processing_time):
|
|
'''
|
|
This uses the Postgres upsert to avoid race conditions when two threads try and insert
|
|
at the same row. The excluded object refers to values that we tried to insert but were
|
|
rejected.
|
|
http://docs.sqlalchemy.org/en/latest/dialects/postgresql.html#insert-on-conflict-upsert
|
|
'''
|
|
table = FactProcessingTime.__table__
|
|
stmt = insert(table).values(
|
|
bst_date=processing_time.bst_date,
|
|
messages_total=processing_time.messages_total,
|
|
messages_within_10_secs=processing_time.messages_within_10_secs
|
|
)
|
|
stmt = stmt.on_conflict_do_update(
|
|
index_elements=[table.c.bst_date],
|
|
set_={
|
|
'messages_total': stmt.excluded.messages_total,
|
|
'messages_within_10_secs': stmt.excluded.messages_within_10_secs,
|
|
'updated_at': datetime.utcnow()
|
|
}
|
|
)
|
|
db.session.connection().execute(stmt)
|
|
|
|
|
|
def get_processing_time_percentage_for_date_range(start_date, end_date):
|
|
query = db.session.query(
|
|
FactProcessingTime.bst_date.cast(db.Text).label("date"),
|
|
FactProcessingTime.messages_total,
|
|
FactProcessingTime.messages_within_10_secs,
|
|
((FactProcessingTime.messages_within_10_secs / FactProcessingTime.messages_total.cast(
|
|
db.Float)) * 100).label("percentage")
|
|
).filter(
|
|
FactProcessingTime.bst_date >= start_date,
|
|
FactProcessingTime.bst_date <= end_date
|
|
).order_by(FactProcessingTime.bst_date)
|
|
|
|
return query.all()
|