ENG-2536 Introduce toggle for using mongo persistence

server/index.ts

@@ -64,6 +64,8 @@ export const snapshotHubMongoDb = new MongoClient(
     );
   });
 
+export const useMongoPersistence = false; // replace with launchdarkly flag
+
 /**
  * The upstream implementation relies on @snapshot-labs/snapshot-spaces npm lib to fetch all the available spaces.
  * Since this implementation is used by OpenLaw only, that dependency was removed and the spaces are loaded from the

server/repositories/events-repository.ts

@@ -1,5 +1,7 @@
 import { postgresEventsRepository } from '../helpers/adapters/postgres';
 import { Event } from '../models/event.js';
+import { useMongoPersistence } from '../index.js';
+import { mongoEventsRepository } from './mongo/mongo-events-repository.js';
 
 export type EventsRepository = {
   getExpiredEvents: (timestamp: number) => Promise<Event[]>;
@@ -8,17 +10,19 @@ export type EventsRepository = {
     EVENT_ID: string,
     space: string,
     timestamp: number
-  ) => Promise<void>,
+  ) => Promise<void>;
   insertStartedProposal: (
     EVENT_ID: string,
     space: string,
     timestamp: number
-  ) => Promise<void>
+  ) => Promise<void>;
   insertProposalEnd: (
     EVENT_ID: string,
     space: string,
     timestamp: number
-  ) => Promise<void>
+  ) => Promise<void>;
 };
 
-export const eventsRepository: EventsRepository = postgresEventsRepository;
+export const eventsRepository: EventsRepository = useMongoPersistence
+  ? mongoEventsRepository
+  : postgresEventsRepository;

server/repositories/messages-repository.ts

@@ -1,5 +1,7 @@
 import { postgresMessagesRepository } from '../helpers/adapters/postgres';
 import { Message, MessageWithVotes } from '../models/message.js';
+import { useMongoPersistence } from '../index.js';
+import { mongoMessagesRepository } from './mongo/mongo-messages-repository.js';
 
 export type MessagesRepository = {
   getMessages: (spaces: string, msgType: string) => Promise<Message[]>;
@@ -34,7 +36,7 @@ export type MessagesRepository = {
     authorIpfsHash,
     relayerIpfsHash,
     actionId
-  ) => Promise<void>,
+  ) => Promise<void>;
   storeProposal: (
     space,
     erc712Hash,
@@ -44,20 +46,20 @@ export type MessagesRepository = {
     authorIpfsHash,
     relayerIpfsHash,
     actionId
-  ) => Promise<void>,
-  storeVote:(
+  ) => Promise<void>;
+  storeVote: (
     space,
     erc712Hash,
     token,
     body,
     authorIpfsHash,
     relayerIpfsHash,
     actionId
-  ) => Promise<void>
-  sponsorDraftIfAny: (space, erc712DraftHash) => Promise<number>,
-  findVotesForProposals: (space, proposals) => Promise<MessageWithVotes[]>
+  ) => Promise<void>;
+  sponsorDraftIfAny: (space, erc712DraftHash) => Promise<number>;
+  findVotesForProposals: (space, proposals) => Promise<MessageWithVotes[]>;
 };
 
-
-
-export const messagesRepository: MessagesRepository = postgresMessagesRepository;
+export const messagesRepository: MessagesRepository = useMongoPersistence
+  ? mongoMessagesRepository
+  : postgresMessagesRepository;

server/repositories/offchain-proofs-repository.ts

@@ -2,14 +2,16 @@ import {
   OffchainProof,
   postgresOffchainProofsRepository
 } from '../helpers/adapters/postgres';
+import { useMongoPersistence } from '../index.js';
+import { mongoOffchainProofsRepository } from './mongo/mongo-offchain-proofs-repository.js';
 
 export type OffchainProofsRepository = {
   getOffchainProof: (
     space: string,
     merkleRoot: string
   ) => Promise<OffchainProof[]>;
-  saveOffchainProof: (
-    offchainProof: OffchainProof
-  ) => Promise<void>;
+  saveOffchainProof: (offchainProof: OffchainProof) => Promise<void>;
 };
-export const offchainProofsRepository: OffchainProofsRepository = postgresOffchainProofsRepository;
+export const offchainProofsRepository: OffchainProofsRepository = useMongoPersistence
+  ? mongoOffchainProofsRepository
+  : postgresOffchainProofsRepository;