mirror of
https://github.com/GSA/notifications-api.git
synced 2025-12-21 07:51:13 -05:00
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from datetime import datetime
|
|
|
|
from sqlalchemy.dialects.postgresql import insert
|
|
|
|
from app import db
|
|
from app.dao.dao_utils import transactional
|
|
from app.models import FactProcessingTime
|
|
|
|
|
|
@transactional
|
|
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)
|