Merge pull request #29 from mobuild-io/single_row_tracking

Use single row tracking

Author: johnbywater

eventsourcing_sqlalchemy/models.py

@@ -8,6 +8,7 @@
     from sqlalchemy.orm import declarative_base
 except ImportError:
     from sqlalchemy.ext.declarative import declarative_base
+
 from sqlalchemy.orm import Mapped
 from sqlalchemy_utils.types.uuid import UUIDType
 
@@ -87,9 +88,9 @@ class NotificationTrackingRecord(Base):
     __abstract__ = True
 
     # Application name.
-    application_name = Column(String(length=32), primary_key=True)
+    application_name: Mapped[str] = Column(String(length=32), primary_key=True)
 
     # Notification ID.
-    notification_id = Column(
-        BigInteger().with_variant(Integer(), "sqlite"), primary_key=True
+    notification_id: Mapped[int] = Column(
+        BigInteger().with_variant(Integer(), "sqlite")
     )

eventsourcing_sqlalchemy/recorders.py

@@ -430,6 +430,15 @@ def _insert_tracking(self, session: Session, tracking: Tracking) -> None:
                 tracking.application_name, tracking.notification_id
             ):
                 raise IntegrityError
+
+        existing = (
+            session.query(self.tracking_record_cls)
+            .filter_by(application_name=str(tracking.application_name))
+            .first()
+        )
+        if existing:
+            existing.notification_id = tracking.notification_id
+        else:
             record = self.tracking_record_cls(
                 application_name=tracking.application_name,
                 notification_id=tracking.notification_id,