Chromium Code Reviews| Index: Source/core/inspector/PromiseTracker.cpp |
| diff --git a/Source/core/inspector/PromiseTracker.cpp b/Source/core/inspector/PromiseTracker.cpp |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..f73fe44b72fe9695e5492c922417617c4436cd25 |
| --- /dev/null |
| +++ b/Source/core/inspector/PromiseTracker.cpp |
| @@ -0,0 +1,115 @@ |
| +// Copyright 2014 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 "config.h" |
| +#include "core/inspector/PromiseTracker.h" |
| + |
| +#include "bindings/core/v8/ScriptCallStackFactory.h" |
| +#include "bindings/core/v8/ScriptState.h" |
| + |
| +namespace blink { |
| + |
| +static int indexOf(PromiseTracker::PromiseDataVector* vector, const ScopedPersistent<v8::Object>& promise) |
| +{ |
| + for (size_t index = 0; index < vector->size(); ++index) { |
| + RefPtr<PromiseTracker::PromiseData> data = vector->at(index); |
| + if (data->promise() == promise) |
|
aandrey
2014/08/25 09:48:37
if (vector->at(index)->promise() == promise)
Alexandra Mikhaylova
2014/08/25 14:25:47
Done.
|
| + return index; |
| + } |
| + return -1; |
| +} |
| + |
| +namespace { |
| + |
| +struct PromiseDataWrapper { |
| + PromiseDataWrapper(WeakPtr<PromiseTracker::PromiseData> data, PromiseTracker::PromiseDataMap* map) |
| + : m_data(data) |
| + , m_promiseDataMap(map) |
| + { |
| + } |
| + |
| + static void didRemovePromise(const v8::WeakCallbackData<v8::Object, PromiseDataWrapper>& data) |
| + { |
| + OwnPtr<PromiseDataWrapper> wrapper = adoptPtr(data.GetParameter()); |
| + WeakPtr<PromiseTracker::PromiseData> promiseData = wrapper->m_data; |
| + if (!promiseData) |
| + return; |
| + PromiseTracker::PromiseDataMap* map = wrapper->m_promiseDataMap; |
| + int promiseHash = promiseData->promiseHash(); |
| + PromiseTracker::PromiseDataVector* vector = &map->find(promiseHash)->value; |
| + int index = indexOf(vector, promiseData->promise()); |
| + ASSERT(index >= 0); |
| + vector->remove(index); |
| + if (vector->size() == 0) |
| + map->remove(promiseHash); |
| + } |
| + |
| + WeakPtr<PromiseTracker::PromiseData> m_data; |
| + PromiseTracker::PromiseDataMap* m_promiseDataMap; |
| +}; |
| + |
| +} |
| + |
| +PromiseTracker::PromiseData::PromiseData(v8::Isolate* isolate, int promiseHash, v8::Handle<v8::Object> promise, v8::Handle<v8::Object> parentPromise, int status, bool captureStack) |
| + : m_promiseHash(promiseHash) |
| + , m_promise(isolate, promise) |
| + , m_parentPromise(isolate, parentPromise) |
| + , m_status(status) |
| + , m_weakPtrFactory(this) |
| +{ |
| + if (captureStack) { |
| + v8::Handle<v8::StackTrace> stackTrace(v8::StackTrace::CurrentStackTrace(isolate, 1)); |
| + RefPtrWillBeRawPtr<ScriptCallStack> stack = createScriptCallStack(stackTrace, 1, isolate); |
| + if (stack->size()) |
| + m_callFrame = stack->at(0); |
| + } |
| +} |
| + |
| +void PromiseTracker::enable() |
| +{ |
| + m_isEnabled = true; |
| +} |
| + |
| +void PromiseTracker::disable() |
| +{ |
| + m_isEnabled = false; |
| + clear(); |
| +} |
| + |
| +void PromiseTracker::clear() |
| +{ |
| + m_promiseDataMap.clear(); |
| +} |
| + |
| +void PromiseTracker::didReceiveV8PromiseEvent(ScriptState* scriptState, v8::Handle<v8::Object> promise, v8::Handle<v8::Value> parentPromise, int status) |
| +{ |
| + ASSERT(isEnabled()); |
| + |
| + int promiseHash = promise->GetIdentityHash(); |
| + PromiseDataVector* vector; |
| + PromiseDataMap::iterator it = m_promiseDataMap.find(promiseHash); |
| + if (it != m_promiseDataMap.end()) |
| + vector = &it->value; |
| + else |
| + vector = &m_promiseDataMap.add(promiseHash, PromiseDataVector()).storedValue->value; |
| + for (size_t index = 0; index < vector->size(); ++index) { |
| + RefPtr<PromiseData> data = vector->at(index); |
| + if (data->m_promise == promise) { |
|
aandrey
2014/08/25 09:48:37
RefPtr<PromiseData> data;
size_t index = indexOf(v
Alexandra Mikhaylova
2014/08/25 14:25:47
Done.
|
| + if (status != 0) { |
| + data->m_status = status; |
| + } else { |
| + ASSERT(parentPromise->IsObject()); |
| + data->m_parentPromise.set(scriptState->isolate(), parentPromise->ToObject()); |
| + } |
| + return; |
| + } |
| + } |
| + bool captureStack = status == 0 && parentPromise.IsEmpty(); |
| + PromiseData* data = new PromiseData(scriptState->isolate(), promiseHash, promise, parentPromise->IsObject() ? parentPromise->ToObject() : v8::Handle<v8::Object>(), 0, captureStack); |
|
aandrey
2014/08/25 09:48:37
parentPromise->IsObject() will crash if parentProm
Alexandra Mikhaylova
2014/08/25 14:25:47
Fixed this method as described above.
|
| + PromiseDataWrapper* wrapper = new PromiseDataWrapper(data->m_weakPtrFactory.createWeakPtr(), &m_promiseDataMap); |
| + data->m_promise.setWeak(wrapper, &PromiseDataWrapper::didRemovePromise); |
| + vector->append(adoptRef(data)); |
| +} |
| + |
| +} // namespace blink |