Return value from array on subscription (#262)

polkadot-js · Oct 12, 2018 · a47485c · a47485c
1 parent dc5802a
commit a47485c
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/packages/api/src/promise/index.ts b/packages/api/src/promise/index.ts
@@ -206,7 +206,8 @@ export default class ApiPromise extends ApiBase<Rpc, QueryableStorage, Submittab
 
       return this.rpc.state.storage(
         [[method, args.length === 1 ? undefined : args[0]]],
-        args[args.length - 1]
+        (result: Array<Base | null | undefined> = []) =>
+          args[args.length - 1](result[0])
       );
     };
 

diff --git a/packages/types/src/codec/UInt.ts b/packages/types/src/codec/UInt.ts
@@ -116,6 +116,12 @@ export default class UInt extends Base<BN> {
       : this.raw.div(bnToBn(other));
   }
 
+  eq (other: UInt | BN | number): boolean {
+    return other instanceof UInt
+      ? this.raw.eq(other.raw)
+      : this.raw.eq(bnToBn(other));
+  }
+
   isZero (): boolean {
     return this.raw.isZero();
   }