From 79b98a4b54ce9ae4adb9ba784f6384d663070a5a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Alberto=20Cuesta=20Ca=C3=B1ada?= Date: Thu, 23 Jan 2020 20:56:51 +0000 Subject: [PATCH] Created AddressSet factory method. --- contracts/mocks/EnumerableSetMock.sol | 2 +- contracts/utils/EnumerableSet.sol | 11 +++++++++++ 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/contracts/mocks/EnumerableSetMock.sol b/contracts/mocks/EnumerableSetMock.sol index 10c4470de..e4456fba5 100644 --- 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(); } /** diff --git a/contracts/utils/EnumerableSet.sol b/contracts/utils/EnumerableSet.sol index 292ed7078..a8fb56863 100644 --- 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). */