Round out stateDB get/set pairs

polkadot-js · Mar 23, 2018 · 6aed7d0 · 6aed7d0
1 parent c5cee7f
commit 6aed7d0
Show file tree

Hide file tree

Showing 12 changed files with 41 additions and 13 deletions.
diff --git a/packages/client-db-chain/src/state/session/length.js b/packages/client-db-chain/src/state/session/length.js
@@ -11,6 +11,8 @@ const { SESSION_LENGTH } = require('./keys');
 
 module.exports = function length (db: WrapDbInterface): ChainDb$State$Session$Length {
   return {
+    get: (): BN =>
+      db.getBn64(SESSION_LENGTH()),
     set: (length: BN | number): void =>
       db.setBn64(SESSION_LENGTH(), length)
   };

diff --git a/packages/client-db-chain/src/state/session/types.js b/packages/client-db-chain/src/state/session/types.js
@@ -6,15 +6,18 @@
 import type BN from 'bn.js';
 
 export type ChainDb$State$Session$Length = {
-  set (length: BN | number): void,
+  get: () => BN,
+  set: (length: BN | number) => void
 };
 
 export type ChainDb$State$Session$Value = {
-  set (id: BN | number, publicKey: Uint8Array): void
+  get: (id: BN | number) => Uint8Array,
+  set: (id: BN | number, publicKey: Uint8Array) => void
 };
 
 export type ChainDb$State$Session$ValueCount = {
-  set (count: BN | number): void
+  get: () => BN,
+  set: (count: BN | number) => void
 }
 
 export type ChainDb$State$Session = {

diff --git a/packages/client-db-chain/src/state/session/value.js b/packages/client-db-chain/src/state/session/value.js
@@ -11,12 +11,15 @@ const bnToU8a = require('@polkadot/util/bn/toU8a');
 
 const { VALUE } = require('./keys');
 
+function key (id: BN | number): Uint8Array {
+  return VALUE(bnToU8a(id, 32, true));
+}
+
 module.exports = function value (db: WrapDbInterface): ChainDb$State$Session$Value {
   return {
+    get: (id: BN | number): Uint8Array =>
+      db.get(key(id)),
     set: (id: BN | number, publicKey: Uint8Array): void =>
-      db.set(
-        VALUE(bnToU8a(id, 32, true)),
-        publicKey
-      )
+      db.set(key(id), publicKey)
   };
 };
diff --git a/packages/client-db-chain/src/state/session/valueCount.js b/packages/client-db-chain/src/state/session/valueCount.js
@@ -11,6 +11,8 @@ const { VALUE_LENGTH } = require('./keys');
 
 module.exports = function valueCount (db: WrapDbInterface): ChainDb$State$Session$ValueCount {
   return {
+    get: (): BN =>
+      db.getBn32(VALUE_LENGTH()),
     set: (count: BN | number): void =>
       db.setBn32(VALUE_LENGTH(), count)
   };

diff --git a/packages/client-db-chain/src/state/staking/currentEra.js b/packages/client-db-chain/src/state/staking/currentEra.js
@@ -11,6 +11,8 @@ const { CURRENT_ERA } = require('./keys');
 
 module.exports = function currentEra (db: WrapDbInterface): ChainDb$State$Staking$CurrentEra {
   return {
+    get: (): BN =>
+      db.getBn32(CURRENT_ERA()),
     set: (era: BN | number): void =>
       db.setBn64(CURRENT_ERA(), era)
   };

diff --git a/packages/client-db-chain/src/state/staking/intent.js b/packages/client-db-chain/src/state/staking/intent.js
@@ -17,6 +17,8 @@ function key (id: BN | number): Uint8Array {
 
 module.exports = function intent (db: WrapDbInterface): ChainDb$State$Staking$Intent {
   return {
+    get: (id: BN | number): Uint8Array =>
+      db.get(key(id)),
     set: (id: BN | number, publicKey: Uint8Array): void =>
       db.set(key(id), publicKey)
   };

diff --git a/packages/client-db-chain/src/state/staking/intentLength.js b/packages/client-db-chain/src/state/staking/intentLength.js
@@ -11,6 +11,8 @@ const { INTENT_WILL_LENGTH } = require('./keys');
 
 module.exports = function setIntentLength (db: WrapDbInterface): ChainDb$State$Staking$IntentLength {
   return {
+    get: (): BN =>
+      db.getBn32(INTENT_WILL_LENGTH()),
     set: (length: BN | number): void =>
       db.setBn32(INTENT_WILL_LENGTH(), length)
   };

diff --git a/packages/client-db-chain/src/state/staking/sessionsPerEra.js b/packages/client-db-chain/src/state/staking/sessionsPerEra.js
@@ -11,6 +11,8 @@ const { SESSIONS_PER_ERA } = require('./keys');
 
 module.exports = function sessionsPerEra (db: WrapDbInterface): ChainDb$State$Staking$SessionsPerEra {
   return {
+    get: (): BN =>
+      db.getBn64(SESSIONS_PER_ERA()),
     set: (count: BN | number): void =>
       db.setBn64(SESSIONS_PER_ERA(), count)
   };

diff --git a/packages/client-db-chain/src/state/staking/types.js b/packages/client-db-chain/src/state/staking/types.js
@@ -11,22 +11,27 @@ export type ChainDb$State$Staking$Balance = {
 }
 
 export type ChainDb$State$Staking$CurrentEra = {
-  set (era: BN | number): void,
+  get: () => BN,
+  set: (era: BN | number) => void
 }
 
 export type ChainDb$State$Staking$Intent = {
-  set (id: BN | number, publicKey: Uint8Array): void,
+  get: (id: BN | number) => Uint8Array,
+  set: (id: BN | number, publicKey: Uint8Array) => void,
 }
 
 export type ChainDb$State$Staking$IntentLength = {
-  set (length: BN | number): void,
+  get: () => BN,
+  set: (length: BN | number) => void
 }
 
 export type ChainDb$State$Staking$SessionsPerEra = {
-  set (count: BN | number): void
+  get: () => BN,
+  set: (count: BN | number) => void
 }
 
 export type ChainDb$State$Staking$ValidatorCount = {
+  get: () => BN,
   set: (count: BN | number) => void
 }
 

diff --git a/packages/client-db-chain/src/state/staking/validatorCount.js b/packages/client-db-chain/src/state/staking/validatorCount.js
@@ -11,6 +11,8 @@ const { VALIDATOR_COUNT } = require('./keys');
 
 module.exports = function validatorCount (db: WrapDbInterface): ChainDb$State$Staking$ValidatorCount {
   return {
+    get: (): BN =>
+      db.getBn64(VALIDATOR_COUNT()),
     set: (count: BN | number): void =>
       db.setBn64(VALIDATOR_COUNT(), count)
   };

diff --git a/packages/client-db-chain/src/state/system/nonce.js b/packages/client-db-chain/src/state/system/nonce.js
@@ -12,6 +12,8 @@ const { NONCE_OF } = require('./keys');
 module.exports = function nonce (db: WrapDbInterface): ChainDb$State$System$Nonce {
   return {
     get: (publicKey: Uint8Array): BN =>
-      db.getBn64(NONCE_OF(publicKey))
+      db.getBn64(NONCE_OF(publicKey)),
+    set: (value: number | BN, publicKey: Uint8Array): void =>
+      db.detBn64(NONCE_OF(publicKey), value)
   };
 };
diff --git a/packages/client-db-chain/src/state/system/types.js b/packages/client-db-chain/src/state/system/types.js
@@ -16,7 +16,8 @@ export type ChainDb$State$System$Code = {
 };
 
 export type ChainDb$State$System$Nonce = {
-  get: (publicKey: Uint8Array) => BN
+  get: (publicKey: Uint8Array) => BN,
+  set: (nonce: number | BN, publicKey: Uint8Array) => void
 };
 
 export type ChainDb$State$System = {