Migrate storageReader to es6 syntax

s1na · holgerd77 · commit f84dd0570b3c · 2019-02-26T10:37:28.000+01:00
diff --git a/lib/state/storageReader.js b/lib/state/storageReader.js
@@ -1,40 +1,37 @@
-module.exports = StorageReader
-
-function StorageReader (stateManager) {
-  this._stateManager = stateManager
-  this._storageCache = new Map()
-}
-
-const proto = StorageReader.prototype
-
-proto.getContractStorage = function getContractStorage (address, key, cb) {
-  const self = this
-  const addressHex = address.toString('hex')
-  const keyHex = key.toString('hex')
-
-  self._stateManager.getContractStorage(address, key, function (err, current) {
-    if (err) return cb(err)
-
-    let map = null
-    if (!self._storageCache.has(addressHex)) {
-      map = new Map()
-      self._storageCache.set(addressHex, map)
-    } else {
-      map = self._storageCache.get(addressHex)
-    }
-
-    let original = null
-
-    if (map.has(keyHex)) {
-      original = map.get(keyHex)
-    } else {
-      map.set(keyHex, current)
-      original = current
-    }
-
-    cb(null, {
-      original,
-      current
+module.exports = class StorageReader {
+  constructor (stateManager) {
+    this._stateManager = stateManager
+    this._storageCache = new Map()
+  }
+
+  getContractStorage (address, key, cb) {
+    const addressHex = address.toString('hex')
+    const keyHex = key.toString('hex')
+
+    this._stateManager.getContractStorage(address, key, (err, current) => {
+      if (err) return cb(err)
+
+      let map = null
+      if (!this._storageCache.has(addressHex)) {
+        map = new Map()
+        this._storageCache.set(addressHex, map)
+      } else {
+        map = this._storageCache.get(addressHex)
+      }
+
+      let original = null
+
+      if (map.has(keyHex)) {
+        original = map.get(keyHex)
+      } else {
+        map.set(keyHex, current)
+        original = current
+      }
+
+      cb(null, {
+        original,
+        current
+      })
     })
-  })
+  }
 }
