Forráskód Böngészése

added file storage as an option, added feature to release specific buffer based on uuid

weiwei 8 hónapja
szülő
commit
f8dad60c35
2 módosított fájl, 102 hozzáadás és 2 törlés
  1. 2 1
      interfaces/general.interface.ts
  2. 100 1
      services/buffer.service.ts

+ 2 - 1
interfaces/general.interface.ts

@@ -3,7 +3,8 @@
 import { Observable, Subject } from "rxjs"
 
 export interface ConnectionState {
-    status: 'BUFFER' | 'DIRECT_PUBLISH' | 'LIMIT_EXCEEDED'
+    uuid?: string | number;
+    status: 'BUFFER' | 'DIRECT_PUBLISH' | 'LIMIT_EXCEEDED' | 'TARGET_PUBLISH' | 'TARGET_BUFFER'
     reason?: string;
     payload?: any;
 }

+ 100 - 1
services/buffer.service.ts

@@ -2,6 +2,7 @@
 import { BehaviorSubject, Observable, Subject, from } from "rxjs";
 import mongoose, { Connection, Model } from "mongoose";
 import { ConnectionState, Message, MessageLog, } from "../interfaces/general.interface";
+import * as fs from 'fs';
 
 export class BufferService {
   private bufferIdentifier!: string
@@ -91,6 +92,7 @@ export class BufferService {
       // do nothing... Let handleConnectionStateChanges deal with this state
     }
     if (this.connectionState.getValue().status === "BUFFER") {
+      this.fileManagement(message, 'buffer');
       this.bufferMessage(message);
     }
     if (this.connectionState.getValue().status === "DIRECT_PUBLISH") {
@@ -122,6 +124,103 @@ export class BufferService {
       // Relese the messages by inserting them into the outgoing Messages together.
       this.releaseBufferedMessages(this.messageStream);
     }
+
+    // testing
+    if(state.status === "TARGET_PUBLISH") {
+      this.fileManagement(state, 'release');
+      this.releaseTargetBufferedMessage(state);
+    }
+  }
+
+  releaseTargetBufferedMessage(message: ConnectionState) {
+    if(this.bufferIdentifier === "file") {
+
+    }
+
+    let target = this.messageBuffered.find((data: Message) => {
+      if(data.id === message.uuid) {
+        return data;
+      }
+    })
+    if(target) {
+      this.messageBuffered = this.messageBuffered.filter((data: Message) => {
+        if(data.id !== message.uuid) {
+          return data;
+        }
+      })
+    }
+    console.log(`TESTING : Releasing buffer Message under ${this.bufferIdentifier}: currently there is ${this.messageBuffered.length} messages to be released`);
+  }
+
+  fileManagement(message: any, type: 'buffer' | 'release') {
+    let filePath = "./_LocalStorage/retransmissionData.json";
+
+    // DECLARATION : status chain
+    let checkStatus: boolean = true;
+    let errorMessage: any;
+    let fileContent: any;
+
+    if(checkStatus === true) {
+      if(this.bufferIdentifier !== "file") {
+        errorMessage = "not file storage, skip.";
+        checkStatus = false;
+      }
+    }
+
+    // CONDITION : check state log file exists or not
+    if(checkStatus === true) {
+        if(fs.existsSync(filePath)) {
+            fileContent = fs.readFileSync(filePath, 'utf-8');
+        }
+        else{
+            fileContent = JSON.stringify({ events: [] });
+        }
+    }
+
+    // CONDITION : fileContent is empty
+    if(checkStatus === true) {
+        if(fileContent === undefined && fileContent === null || fileContent === "") {
+            fileContent = JSON.stringify({ events: [] });
+        }
+    }
+
+    // TASK : json parse fileContent
+    if(checkStatus === true) {
+        fileContent = JSON.parse(fileContent);
+    }
+
+    // TASK : add new message into fileContent
+    if(checkStatus === true) {
+      if(type === "buffer") {
+        fileContent.events.push(message);
+      }
+    }
+    // TASK : add new message into fileContent
+    if(checkStatus === true) {
+      if(type === "release") {
+        fileContent.events = fileContent.events.filter((data) => {
+          if(data.id !== message.uuid) {
+            return data
+          }
+        })
+      }
+    }
+
+    // TASK : write file
+    if(checkStatus === true) {
+        try {
+            fs.writeFileSync(filePath, JSON.stringify(fileContent, null, 4), 'utf-8');
+            console.log('\u001b[' + 32 + 'm' + "RE-TRANSMISSION FILE : ADDED NEW DATA..." + '\u001b[0m');
+        }catch(error) {
+            errorMessage = error;
+            checkStatus = false;
+        }
+    }
+
+    // CONDITION : ERROR
+    if(checkStatus === false) {
+        console.log('\u001b[' + 31 + 'm' + "RE-TRANSMISSION FILE : ERROR -> " + JSON.stringify(errorMessage,null,4) + '\u001b[0m');
+    }
   }
 
   private async bufferMessage(message: any): Promise<void> {
@@ -146,7 +245,7 @@ export class BufferService {
     } else {
       if (this.bufferLimit > this.messageBuffered.length) {
         this.messageBuffered.push(message);
-        console.log(this.messageBuffered) // Fallback to local buffer if model is not defined
+        // console.log(this.messageBuffered) // Fallback to local buffer if model is not defined
         console.log(`pushing into local array buffer under ${this.bufferIdentifier}.... There is now ${this.messageBuffered.length} messages`);
       } else {
         let reportState: ConnectionState = {