refactor storageRangeInternal

refactor_remix_debug2c_2
Iuri Matias 5 years ago committed by aniket-engg
parent 5c81c76873
commit e411be3eda
  1. 22
      libs/remix-debug/src/storage/storageResolver.js

@ -26,7 +26,9 @@ class StorageResolver {
* @param {Function} - callback - contains a map: [hashedKey] = {key, hashedKey, value} * @param {Function} - callback - contains a map: [hashedKey] = {key, hashedKey, value}
*/ */
storageRange (tx, stepIndex, address, callback) { storageRange (tx, stepIndex, address, callback) {
this.storageRangeInternal(this, this.zeroSlot, tx, stepIndex, address, callback) this.storageRangeInternal(this, this.zeroSlot, tx, stepIndex, address).then((result) => {
callback(null, result)
}).catch(callback)
} }
/** /**
@ -63,13 +65,9 @@ class StorageResolver {
* @param {Function} - callback - {key, hashedKey, value} - * @param {Function} - callback - {key, hashedKey, value} -
*/ */
storageSlot (slot, tx, stepIndex, address, callback) { storageSlot (slot, tx, stepIndex, address, callback) {
this.storageRangeInternal(this, slot, tx, stepIndex, address, (error, storage) => { this.storageRangeInternal(this, slot, tx, stepIndex, address).then((storage) => {
if (error) {
callback(error)
} else {
callback(null, storage[slot] !== undefined ? storage[slot] : null) callback(null, storage[slot] !== undefined ? storage[slot] : null)
} }).catch(callback)
})
} }
/** /**
@ -88,14 +86,15 @@ class StorageResolver {
* even if the next 1000 items are not in the cache. * even if the next 1000 items are not in the cache.
* - If @arg slot is not cached, the corresponding value will be resolved and the next 1000 slots. * - If @arg slot is not cached, the corresponding value will be resolved and the next 1000 slots.
*/ */
storageRangeInternal (self, slotKey, tx, stepIndex, address, callback) { storageRangeInternal (self, slotKey, tx, stepIndex, address) {
return new Promise((resolve, reject) => {
var cached = this.fromCache(self, address) var cached = this.fromCache(self, address)
if (cached && cached.storage[slotKey]) { // we have the current slot in the cache and maybe the next 1000... if (cached && cached.storage[slotKey]) { // we have the current slot in the cache and maybe the next 1000...
return callback(null, cached.storage) return resolve(cached.storage)
} }
this.storageRangeWeb3Call(tx, address, slotKey, self.maxSize, (error, storage, nextKey) => { this.storageRangeWeb3Call(tx, address, slotKey, self.maxSize, (error, storage, nextKey) => {
if (error) { if (error) {
return callback(error) return reject(error)
} }
if (!storage[slotKey] && slotKey !== self.zeroSlot) { // we don't cache the zero slot (could lead to inconsistency) if (!storage[slotKey] && slotKey !== self.zeroSlot) { // we don't cache the zero slot (could lead to inconsistency)
storage[slotKey] = { storage[slotKey] = {
@ -107,7 +106,8 @@ class StorageResolver {
if (slotKey === self.zeroSlot && !nextKey) { // only working if keys are sorted !! if (slotKey === self.zeroSlot && !nextKey) { // only working if keys are sorted !!
self.storageByAddress[address].complete = true self.storageByAddress[address].complete = true
} }
callback(null, storage) return resolve(storage)
})
}) })
} }

Loading…
Cancel
Save