Created AddressSet factory method.

2020-01-23 20:56:51 +00:00
parent 31e9a54c19
commit 79b98a4b54
2 changed files with 12 additions and 1 deletions
--- a/contracts/mocks/EnumerableSetMock.sol
+++ b/contracts/mocks/EnumerableSetMock.sol
@ -14,7 +14,7 @@ contract EnumerableSetMock{
    EnumerableSet.AddressSet private set;

    constructor() public {
-        set = EnumerableSet.AddressSet({values: new address[](0)});
+        set = EnumerableSet.newAddressSet();
    }

    /**
--- a/contracts/utils/EnumerableSet.sol
+++ b/contracts/utils/EnumerableSet.sol
@ -27,6 +27,17 @@ library EnumerableSet {
        address[] values;
    }

+    /**
+     * @dev Add a value. O(1).
+     */
+    function newAddressSet()
+        internal
+        pure
+        returns (AddressSet memory)
+    {
+        return AddressSet({values: new address[](0)});
+    }
+
    /**
     * @dev Add a value. O(1).
     */