[compositorworker] Register compositor proxies with proxy client

third_party/WebKit/Source/web/CompositorProxyClientImpl.cpp

Index: third_party/WebKit/Source/web/CompositorProxyClientImpl.cpp
diff --git a/third_party/WebKit/Source/web/CompositorProxyClientImpl.cpp b/third_party/WebKit/Source/web/CompositorProxyClientImpl.cpp
index 785f096b53c210fa47430bbbf0c614d36fad27c0..a233f8ef398d04db2b167368c3c87d4c8a1398ea 100644
--- a/third_party/WebKit/Source/web/CompositorProxyClientImpl.cpp
+++ b/third_party/WebKit/Source/web/CompositorProxyClientImpl.cpp
@@ -23,6 +23,7 @@ DEFINE_TRACE(CompositorProxyClientImpl)
     CompositorProxyClient::trace(visitor);
     visitor->trace(m_mutator);
     visitor->trace(m_globalScope);
+    visitor->trace(m_proxyMap);
 }
 
 void CompositorProxyClientImpl::setGlobalScope(WorkerGlobalScope* scope)
@@ -64,4 +65,27 @@ bool CompositorProxyClientImpl::executeAnimationFrameCallbacks(double monotonicT
     return shouldReinvoke;
 }
 
+void CompositorProxyClientImpl::registerCompositorProxy(CompositorProxy* proxy)
+{
+    uint64_t elementId = proxy->elementId();
+    ProxyMap::AddResult entry = m_proxyMap.add(elementId, nullptr);
+    Member<ProxySet>& proxies = entry.storedValue->value;
+    if (entry.isNewEntry)
+        proxies = new ProxySet;
+
+    proxies->add(proxy);
+}
+
+void CompositorProxyClientImpl::unregisterCompositorProxy(CompositorProxy* proxy)
+{
+    ProxyMap::iterator it = m_proxyMap.find(proxy->elementId());
+    if (it == m_proxyMap.end())
+        return;
+
+    ProxySet* proxies = it->value;
+    proxies->remove(proxy);
+    if (proxies->isEmpty())
+        m_proxyMap.remove(it);
+}
+
 } // namespace blink