@sentio/sdk 2.36.2-rc.1 → 2.37.0-rc.2

package/src/store/store.ts

@@ -0,0 +1,68 @@
+import { Entity, EntityClass } from './entity.js'
+import { StoreContext } from './context.js'
+
+export class Store {
+  constructor(private readonly context: StoreContext) {}
+
+  async get<T extends Entity>(entity: EntityClass<T> | string, id: string): Promise<T | undefined> {
+    const promise = this.context.sendRequest({
+      get: {
+        entity: typeof entity == 'string' ? entity : entity.name,
+        id
+      }
+    })
+
+    const data = (await promise) as any
+    if (data?.['id'] != null) {
+      return this.newEntity(entity as EntityClass<T>, data)
+    }
+    return undefined
+  }
+
+  async delete(entity: EntityClass<any>, id: string | string[]): Promise<void> {
+    await this.context.sendRequest({
+      delete: {
+        entity: entity.name,
+        id: Array.isArray(id) ? id : [id]
+      }
+    })
+  }
+
+  async upsert<T extends Entity>(entity: T | T[]): Promise<T> {
+    const promise = this.context.sendRequest({
+      upsert: {
+        entity: entity.constructor.name,
+        data: Array.isArray(entity) ? entity.map((e) => e.data) : [entity.data]
+      }
+    })
+
+    if (Array.isArray(entity)) {
+      entity.forEach((e) => (e.store = this))
+    } else {
+      entity.store = this
+    }
+
+    return promise as Promise<T>
+  }
+
+  async list<T extends Entity>(entity: EntityClass<T>, limit?: number, offset?: number): Promise<T[]> {
+    const promise = this.context.sendRequest({
+      list: {
+        entity: entity.name,
+        limit,
+        offset
+      }
+    })
+
+    const list = (await promise) as any[]
+    return list.map((data) => {
+      return this.newEntity(entity, data)
+    })
+  }
+
+  private newEntity<T extends Entity>(entity: EntityClass<T>, data: any) {
+    const e = new (entity as EntityClass<T>)(data)
+    e.store = this
+    return e
+  }
+}

package/src/store/types.ts

@@ -0,0 +1,10 @@
+export type { BigDecimal } from '@sentio/bigdecimal'
+
+export type ID = string
+export type String = string
+export type Int = number
+export type Float = number
+export type Boolean = boolean
+export type DateTime = Date
+export type Json = any
+export type Bytes = Uint8Array

package/src/testing/test-processor-server.ts

@@ -2,12 +2,16 @@ import {
   AccountConfig,
   ContractConfig,
   DataBinding,
+  DeepPartial,
   Empty,
   ProcessBindingResponse,
   ProcessBindingsRequest,
   ProcessConfigRequest,
   ProcessConfigResponse,
   ProcessorServiceImplementation,
+  ProcessStreamRequest,
+  ProcessStreamResponse,
+  ServerStreamingMethodResult,
   StartRequest
 } from '@sentio/protos'
 import { CallContext } from 'nice-grpc-common'
@@ -80,7 +84,14 @@ export class TestProcessorServer implements ProcessorServiceImplementation {
     return this.service.processBindings({ bindings: [request] }, context)
   }
 
-  processBindingsStream(request: AsyncIterable<DataBinding>, context: CallContext) {
-    return this.service.processBindingsStream(request, context)
+  processBindingsStream(
+    requests: AsyncIterable<ProcessStreamRequest>,
+    context: CallContext
+  ): ServerStreamingMethodResult<DeepPartial<ProcessStreamResponse>> {
+    throw new Error('Method not implemented.')
   }
+
+  // processBindingsStream(request: AsyncIterable<ProcessStreamRequest>, context: CallContext) {
+  //   return this.service.processBindingsStream(request, context)
+  // }
 }