[wrapper-tracing] Trace CustomElementReactionStack

third_party/WebKit/Source/core/dom/custom/CustomElementReactionStack.cpp

 // Copyright 2016 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "core/dom/custom/CustomElementReactionStack.h"
 
 #include "bindings/core/v8/Microtask.h"
 #include "core/dom/Element.h"
 #include "core/dom/custom/CEReactionsScope.h"
 #include "core/dom/custom/CustomElementReactionQueue.h"
@@ skipping 16 matching lines @@
 // finalizers, to make short-lived entries fast.
 
 CustomElementReactionStack::CustomElementReactionStack() {}
 
 DEFINE_TRACE(CustomElementReactionStack) {
   visitor->trace(m_map);
   visitor->trace(m_stack);
   visitor->trace(m_backupQueue);
 }
 
+DEFINE_TRACE_WRAPPERS(CustomElementReactionStack) {
+  for (auto key : m_map.keys()) {
+    visitor->traceWrappers(key);
+  }
+}
+
 void CustomElementReactionStack::push() {
   m_stack.append(nullptr);
 }
 
 void CustomElementReactionStack::popInvokingReactions() {
   ElementQueue* queue = m_stack.back();
   if (queue)
     invokeReactions(*queue);
   m_stack.pop_back();
 }
@@ skipping 18 matching lines @@
 void CustomElementReactionStack::enqueue(Member<ElementQueue>& queue,
                                          Element* element,
                                          CustomElementReaction* reaction) {
   if (!queue)
     queue = new ElementQueue();
   queue->append(element);
 
   CustomElementReactionQueue* reactions = m_map.get(element);
   if (!reactions) {
     reactions = new CustomElementReactionQueue();
-    m_map.add(element, reactions);
+    m_map.add(TraceWrapperMember<Element>(this, element), reactions);
   }
 
   reactions->add(reaction);
 }
 
 void CustomElementReactionStack::enqueueToBackupQueue(
     Element* element,
     CustomElementReaction* reaction) {
   // https://html.spec.whatwg.org/multipage/scripting.html#backup-element-queue
 
@@ skipping 29 matching lines @@
 CustomElementReactionStack*
 CustomElementReactionStackTestSupport::setCurrentForTest(
     CustomElementReactionStack* newStack) {
   Persistent<CustomElementReactionStack>& stack = customElementReactionStack();
   CustomElementReactionStack* oldStack = stack.get();
   stack = newStack;
   return oldStack;
 }
 
 }  // namespace blink