wip

Author: silesky

packages/signals/signals/src/core/buffer/index.ts

@@ -1,6 +1,7 @@
 import { Signal } from '@segment/analytics-signals-runtime'
 import { openDB, DBSchema, IDBPDatabase } from 'idb'
 import { logger } from '../../lib/logger'
+import { WebStorage } from '../../lib/storage/web-storage'
 
 interface SignalDatabase extends DBSchema {
   signals: {
@@ -20,6 +21,7 @@ export class SignalStore implements SignalPersistentStorage {
   static readonly STORE_NAME = 'signals'
   private db: Promise<IDBPDatabase<SignalDatabase>>
   private maxBufferSize: number
+  private sessionKeyStorage = new WebStorage(window.sessionStorage)
   static deleteDatabase() {
     return indexedDB.deleteDatabase(SignalStore.DB_NAME)
   }
@@ -34,6 +36,14 @@ export class SignalStore implements SignalPersistentStorage {
   }
 
   private async initSignalDB() {
+    // prevent the signals buffer from persisting across sessions (e.g, user closes tab and reopens)
+    const sessionKey = 'segment_signals_db_session_key'
+    if (!sessionStorage.getItem(sessionKey)) {
+      this.sessionKeyStorage.setItem(sessionKey, true)
+      await SignalStore.deleteDatabase()
+      logger.debug('New Session, so signals buffer cleared')
+    }
+
     const db = await openDB<SignalDatabase>(SignalStore.DB_NAME, 1, {
       upgrade(db) {
         db.createObjectStore(SignalStore.STORE_NAME, { autoIncrement: true })