Decorated can return number (for subscription) (#521)

This commit is contained in:
Jaco Greeff
2018-12-30 15:32:43 +01:00
committed by GitHub
parent 5dcbf1848e
commit f8e7388ac2
+2 -2
View File
@@ -237,7 +237,7 @@ export default class ApiPromise extends ApiBase<Rpc, QueryableStorage, Submittab
}
private decorateStorageEntry (method: StorageFunction): QueryableStorageFunction {
const decorated: any = (...args: Array<any>): Promise<Codec | null | undefined> => {
const decorated: any = (...args: Array<any>): Promise<Codec | null | undefined> | Promise<number> => {
if (args.length === 0 || !isFunction(args[args.length - 1])) {
return this.rpc.state.getStorage([method, args[0]]);
} else if (!this.hasSubscriptions && isFunction(args[args.length - 1])) {
@@ -255,7 +255,7 @@ export default class ApiPromise extends ApiBase<Rpc, QueryableStorage, Submittab
decorated.at = (hash: Hash, arg?: any): Promise<Codec | null | undefined> =>
this.rpc.state.getStorage([method, arg], hash);
decorated.unsubscribe = (subscriptionId: any): Promise<any> =>
decorated.unsubscribe = (subscriptionId: number): Promise<any> =>
this.rpc.state.subscribeStorage.unsubscribe(subscriptionId);
return this.decorateFunctionMeta(method, decorated) as QueryableStorageFunction;