@stack-spot/portal-network 0.203.0 → 0.204.0-beta.1

package/src/utils/StreamedJson.tsx

@@ -21,6 +21,11 @@ interface ConstructorParams<T> {
    * @default 50
    */
   minChangeIntervalMS?: number,
+  /**
+   * Allows overriding the minimum interval by event.
+   * Useful when some events must flush immediately (agent_info events).
+   */
+  minChangeIntervalMSFromEvent?: (event: Partial<T>, data: Partial<T>) => number | undefined,
   /**
    * Optional. If set, this function will be called with every streaming event and must transform the current data object according to the
    * message received.
@@ -31,6 +36,7 @@ interface ConstructorParams<T> {
    * of merged.
    */
   ignoreKeys?: (keyof T)[],
+  textFromErrorEvent?: (data: Partial<T>) => string,
 }
 
 /**
@@ -44,16 +50,20 @@ export class StreamedJson<T> {
   private abortController: AbortController | undefined
   private transform?: (event: Partial<T>, data: Partial<T>) => void | Promise<void>
   private ignoreKeys?: (keyof T)[]
+  private minChangeIntervalMSFromEvent?: (event: Partial<T>, data: Partial<T>) => number | undefined
+  private textFromErrorEvent: (data: Partial<T>) => string
 
   /**
    * @param response the fetch response.
    * @param minChangeIntervalMS a stream can be too fast. This sets a minimum interval between running the listeners. The default is 50ms.
    */
-  constructor({ eventsPromise, abortController, minChangeIntervalMS = 50, transform, ignoreKeys }: ConstructorParams<T>) {
+  constructor({ eventsPromise, abortController, minChangeIntervalMS = 50, minChangeIntervalMSFromEvent, transform, ignoreKeys, textFromErrorEvent }: ConstructorParams<T>) {
     this.abortController = abortController
     this.transform = transform
     this.ignoreKeys = ignoreKeys
+    this.minChangeIntervalMSFromEvent = minChangeIntervalMSFromEvent
     this.run(eventsPromise, minChangeIntervalMS)
+    this.textFromErrorEvent = textFromErrorEvent ?? ((data) => JSON.stringify(data))
   }
 
   private async run(eventsPromise: Promise<FetchEventStream>, minChangeIntervalMS: number) {
@@ -74,7 +84,9 @@ export class StreamedJson<T> {
           }
           await this.transform?.(json, this.data)
           this.merge(json, this.data)
-          if (new Date().getTime() - lastChangeCall >= minChangeIntervalMS) {
+
+          const intervalMS = this.minChangeIntervalMSFromEvent?.(json, this.data) ?? minChangeIntervalMS
+          if (event.event !== 'error' && new Date().getTime() - lastChangeCall >= intervalMS) {
             this.onChangeListeners.forEach(l => l(this.data))
             lastChangeCall = new Date().getTime()
             flushed = true
@@ -82,6 +94,11 @@ export class StreamedJson<T> {
             flushed = false
           }
         }
+        if (event.event === 'error') {
+          const error = new Error(this.textFromErrorEvent(this.data))
+          this.data = {}
+          throw error
+        }
       }
       if (!flushed) this.onChangeListeners.forEach(l => l(this.data))
     } catch (error: any) {