IpAddresses: Adding contains and remove functions (#23)

Author: stcheng

common/ipaddresses.cpp

@@ -28,6 +28,28 @@ void IpAddresses::add(const IpAddress &ip)
     m_ips.insert(ip);
 }
 
+bool IpAddresses::contains(const std::string &ipStr)
+{
+    IpAddress ip(ipStr);
+    return m_ips.find(ip) != m_ips.end();
+}
+
+bool IpAddresses::contains(const IpAddress &ip)
+{
+    return m_ips.find(ip) != m_ips.end();
+}
+
+void IpAddresses::remove(const string &ipStr)
+{
+    IpAddress ip(ipStr);
+    m_ips.erase(ip);
+}
+
+void IpAddresses::remove(const IpAddress &ip)
+{
+    m_ips.erase(ip);
+}
+
 const string IpAddresses::to_string() const
 {
     string ips_str;

common/ipaddresses.h

@@ -44,6 +44,14 @@ class IpAddresses
 
     void add(const IpAddress &ip);
 
+    bool contains(const std::string &ip);
+
+    bool contains(const IpAddress &ip);
+
+    void remove(const std::string &ip);
+
+    void remove(const IpAddress &ip);
+
     const std::string to_string() const;