Update api.derive to use api.queryMulti (#911)
* session api.queryMulti * Make a single subscription to retrieve all session info * Adjustments * WIP * Seems mostly comprehensive, need to full test e2e * e2e tests for operations * Allow "create from empty" * skip * Remove comment * Remove ancient ?'s * Update types.ts
This commit is contained in:
@@ -12,8 +12,41 @@ import { idAndIndex } from '../accounts/idAndIndex';
|
||||
import { DerivedBalances } from '../types';
|
||||
import { drr } from '../util/drr';
|
||||
|
||||
type Result = [AccountId | undefined, BlockNumber | undefined, [Balance?, Balance?, Array<BalanceLock>?, Option<VestingSchedule>?]];
|
||||
|
||||
const EMPTY_ACCOUNT = new AccountId();
|
||||
|
||||
function calcBalances ([accountId = EMPTY_ACCOUNT, bestNumber = new BlockNumber(0), [freeBalance = new Balance(0), reservedBalance = new Balance(0), locks = [], vesting = new Option<VestingSchedule>(VestingSchedule, null)]]: Result): DerivedBalances {
|
||||
let lockedBalance = new Balance(0);
|
||||
|
||||
if (Array.isArray(locks)) {
|
||||
// only get the locks that are valid until passed the current block
|
||||
const totals = locks.filter((value) => bestNumber && value.until.gt(bestNumber));
|
||||
// get the maximum of the locks according to https://github.com/paritytech/substrate/blob/master/srml/balances/src/lib.rs#L699
|
||||
lockedBalance = totals[0] ? bnMax(...totals.map(({ amount }) => amount)) as Balance : new Balance(0);
|
||||
}
|
||||
|
||||
const { offset, perBlock } = vesting.unwrapOr({
|
||||
offset: new Balance(0),
|
||||
perBlock: new Balance(0)
|
||||
} as VestingSchedule);
|
||||
|
||||
const vestedBalance = new Balance(perBlock.mul(bestNumber).add(freeBalance.sub(offset)));
|
||||
const availableBalance = new Balance(vestedBalance.sub(lockedBalance));
|
||||
|
||||
return {
|
||||
accountId,
|
||||
freeBalance,
|
||||
reservedBalance,
|
||||
votingBalance: new Balance(
|
||||
freeBalance.add(reservedBalance)
|
||||
),
|
||||
availableBalance,
|
||||
lockedBalance,
|
||||
vestedBalance: vestedBalance.lt(new Balance(0)) ? new Balance(0) : vestedBalance
|
||||
};
|
||||
}
|
||||
|
||||
/**
|
||||
* @name all
|
||||
* @param {( ccountIndex | AccountId | Address | string )} address - An accounts Id in different formats.
|
||||
@@ -38,46 +71,17 @@ export function all (api: ApiInterface$Rx) {
|
||||
? combineLatest([
|
||||
of(accountId),
|
||||
api.derive.chain.bestNumber(),
|
||||
api.query.balances.freeBalance(accountId),
|
||||
api.query.balances.reservedBalance(accountId),
|
||||
api.query.balances.locks(accountId),
|
||||
api.query.balances.vesting(accountId)
|
||||
api.queryMulti([
|
||||
[api.query.balances.freeBalance, accountId],
|
||||
[api.query.balances.reservedBalance, accountId],
|
||||
[api.query.balances.locks, accountId],
|
||||
[api.query.balances.vesting, accountId]
|
||||
])
|
||||
])
|
||||
: of([undefined, undefined, undefined, undefined, undefined, undefined])
|
||||
) as Observable<[AccountId?, BlockNumber?, Balance?, Balance?, Array<BalanceLock>?, Option<VestingSchedule>?]>
|
||||
: of([undefined, undefined, [undefined, undefined, undefined, undefined]])
|
||||
) as any as Observable<Result>
|
||||
),
|
||||
map(([accountId = EMPTY_ACCOUNT, bestNumber = new BlockNumber(0), freeBalance = new Balance(0), reservedBalance = new Balance(0), locks = [], vesting = new Option<VestingSchedule>(VestingSchedule, null)]) => {
|
||||
let lockedBalance = new Balance(0);
|
||||
|
||||
if (Array.isArray(locks)) {
|
||||
// only get the locks that are valid until passed the current block
|
||||
const totals = locks.filter((value) => bestNumber && value.until.gt(bestNumber));
|
||||
// get the maximum of the locks according to https://github.com/paritytech/substrate/blob/master/srml/balances/src/lib.rs#L699
|
||||
lockedBalance = totals[0] ? bnMax(...totals.map(({ amount }) => amount)) as Balance : new Balance(0);
|
||||
}
|
||||
|
||||
const { offset, perBlock } = vesting.unwrapOr({
|
||||
offset: new Balance(0),
|
||||
perBlock: new Balance(0)
|
||||
} as VestingSchedule);
|
||||
|
||||
const vestedBalance = new Balance(perBlock.mul(bestNumber).add(freeBalance.sub(offset)));
|
||||
|
||||
// const availableBalance = lockedBalance.eq(new BlockNumber('0xffffffffffffffffffffffffffffffff')) ? new Balance(0) : freeBalance && new Balance(freeBalance.sub(lockedBalance));
|
||||
const availableBalance = new Balance(vestedBalance.sub(lockedBalance));
|
||||
|
||||
return {
|
||||
accountId,
|
||||
freeBalance,
|
||||
reservedBalance,
|
||||
votingBalance: new Balance(
|
||||
freeBalance.add(reservedBalance)
|
||||
),
|
||||
availableBalance,
|
||||
lockedBalance,
|
||||
vestedBalance: vestedBalance.lt(new Balance(0)) ? new Balance(0) : vestedBalance
|
||||
};
|
||||
}),
|
||||
map(calcBalances),
|
||||
drr()
|
||||
);
|
||||
};
|
||||
|
||||
Reference in New Issue
Block a user