heyujie
2021-05-20 6ebdefb4a5b2be82a8c452c0bb4624f3d85a17b7
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
'use strict';
 
var clear          = require('es5-ext/object/clear')
  , setPrototypeOf = require('es5-ext/object/set-prototype-of')
  , validValue     = require('es5-ext/object/valid-value')
  , callable       = require('es5-ext/object/valid-callable')
  , d              = require('d')
  , iterator       = require('es6-iterator/valid-iterable')
  , forOf          = require('es6-iterator/for-of')
  , isNative       = require('../is-native-implemented')
  , MapPolyfill    = require('../polyfill')
  , Iterator       = require('../lib/primitive-iterator')
 
  , call = Function.prototype.call
  , create = Object.create, defineProperty = Object.defineProperty
  , defineProperties = Object.defineProperties, getPrototypeOf = Object.getPrototypeOf
  , hasOwnProperty = Object.prototype.hasOwnProperty
  , PrimitiveMap;
 
module.exports = PrimitiveMap = function (/*iterable, serialize*/) {
    var iterable = arguments[0], serialize = arguments[1], self;
    if (!(this instanceof PrimitiveMap)) throw new TypeError('Constructor requires \'new\'');
    if (isNative && setPrototypeOf && (Map !== MapPolyfill)) {
        self = setPrototypeOf(new Map(), getPrototypeOf(this));
    } else {
        self = this;
    }
    if (iterable != null) iterator(iterable);
    if (serialize !== undefined) {
        callable(serialize);
        defineProperty(self, '_serialize', d('', serialize));
    }
    defineProperties(self, {
        __mapKeysData__: d('c', create(null)),
        __mapValuesData__: d('c', create(null)),
        __size__: d('w', 0)
    });
    if (!iterable) return self;
    forOf(iterable, function (value) {
        var key = validValue(value)[0], sKey = self._serialize(key);
        if (sKey == null) throw new TypeError(key + " cannot be serialized");
        value = value[1];
        if (hasOwnProperty.call(self.__mapKeysData__, sKey)) {
            if (self.__mapValuesData__[sKey] === value) return;
        } else {
            ++self.__size__;
        }
        self.__mapKeysData__[sKey] = key;
        self.__mapValuesData__[sKey] = value;
    });
    return self;
};
if (setPrototypeOf) setPrototypeOf(PrimitiveMap, MapPolyfill);
 
PrimitiveMap.prototype = create(MapPolyfill.prototype, {
    constructor: d(PrimitiveMap),
    _serialize: d(function (value) {
        if (value && (typeof value.toString !== 'function')) return null;
        return String(value);
    }),
    clear: d(function () {
        if (!this.__size__) return;
        clear(this.__mapKeysData__);
        clear(this.__mapValuesData__);
        this.__size__ = 0;
        this.emit('_clear');
    }),
    delete: d(function (key) {
        var sKey = this._serialize(key);
        if (sKey == null) return false;
        if (!hasOwnProperty.call(this.__mapKeysData__, sKey)) return false;
        delete this.__mapKeysData__[sKey];
        delete this.__mapValuesData__[sKey];
        --this.__size__;
        this.emit('_delete', sKey);
        return true;
    }),
    entries: d(function () { return new Iterator(this, 'key+value'); }),
    forEach: d(function (cb/*, thisArg*/) {
        var thisArg = arguments[1], iterator, result, sKey;
        callable(cb);
        iterator = this.entries();
        result = iterator._next();
        while (result !== undefined) {
            sKey = iterator.__list__[result];
            call.call(cb, thisArg, this.__mapValuesData__[sKey],
                this.__mapKeysData__[sKey], this);
            result = iterator._next();
        }
    }),
    get: d(function (key) {
        var sKey = this._serialize(key);
        if (sKey == null) return;
        return this.__mapValuesData__[sKey];
    }),
    has: d(function (key) {
        var sKey = this._serialize(key);
        if (sKey == null) return false;
        return hasOwnProperty.call(this.__mapKeysData__, sKey);
    }),
    keys: d(function () { return new Iterator(this, 'key'); }),
    size: d.gs(function () { return this.__size__; }),
    set: d(function (key, value) {
        var sKey = this._serialize(key);
        if (sKey == null) throw new TypeError(key + " cannot be serialized");
        if (hasOwnProperty.call(this.__mapKeysData__, sKey)) {
            if (this.__mapValuesData__[sKey] === value) return this;
        } else {
            ++this.__size__;
        }
        this.__mapKeysData__[sKey] = key;
        this.__mapValuesData__[sKey] = value;
        this.emit('_add', sKey);
        return this;
    }),
    values: d(function () { return new Iterator(this, 'value'); })
});