From 64d5ec090b5d54f68647b96032d765c53c43ef1a Mon Sep 17 00:00:00 2001 From: Sebastian McKenzie Date: Mon, 5 Oct 2015 16:43:37 +0100 Subject: [PATCH] extend Store from Map --- packages/babel-core/src/store.js | 16 +++++----------- 1 file changed, 5 insertions(+), 11 deletions(-) diff --git a/packages/babel-core/src/store.js b/packages/babel-core/src/store.js index d4c31300b2..6c1ae4f953 100644 --- a/packages/babel-core/src/store.js +++ b/packages/babel-core/src/store.js @@ -1,26 +1,20 @@ /* @flow */ -export default class Store { +export default class Store extends Map { constructor() { + super(); this.dynamicData = {}; - this.data = {}; } dynamicData: Object; - data: Object; - set(key: string, val: any): any { - return this.data[key] = val; - } - - setDynamic(key: string, fn: Function) { + setDynamic(key, fn) { this.dynamicData[key] = fn; } get(key: string): any { - let data = this.data[key]; - if (data) { - return data; + if (this.has(key)) { + return super.get(key); } else { let dynamic = this.dynamicData[key]; if (dynamic) {