Don't short circuit seeking
[WebKit-https.git] / Source / WebCore / platform / GenericTaskQueue.h
1 /*
2  * Copyright (C) 2015 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. ``AS IS'' AND ANY
14  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
15  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL APPLE INC. OR
17  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
18  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
19  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
20  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
21  * OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
23  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 
24  */
25
26 #ifndef GenericTaskQueue_h
27 #define GenericTaskQueue_h
28
29 #include "Timer.h"
30 #include <wtf/Deque.h>
31 #include <wtf/WeakPtr.h>
32
33 namespace WebCore {
34
35 template <typename T>
36 class TaskDispatcher {
37 public:
38     TaskDispatcher(T& context)
39         : m_context(context)
40     {
41     }
42
43     void postTask(std::function<void()> f)
44     {
45         m_context.postTask(f);
46     }
47
48 private:
49     T& m_context;
50 };
51
52 template<>
53 class TaskDispatcher<Timer> {
54 public:
55     TaskDispatcher()
56         : m_timer(*this, &TaskDispatcher<Timer>::timerFired)
57     {
58     }
59
60     void postTask(std::function<void()> function)
61     {
62         m_queue.append(function);
63         m_timer.startOneShot(0);
64     }
65
66     void timerFired()
67     {
68         Deque<std::function<void()>> queue;
69         queue.swap(m_queue);
70         for (std::function<void()>& function : queue)
71             function();
72     }
73
74     Timer m_timer;
75     Deque<std::function<void()>> m_queue;
76 };
77
78 template <typename T>
79 class GenericTaskQueue {
80 public:
81     GenericTaskQueue()
82         : m_weakPtrFactory(this)
83         , m_dispatcher()
84     {
85     }
86
87     GenericTaskQueue(T& t)
88         : m_weakPtrFactory(this)
89         , m_dispatcher(t)
90     {
91     }
92
93     typedef std::function<void()> TaskFunction;
94
95     void enqueueTask(TaskFunction task)
96     {
97         if (m_isClosed)
98             return;
99
100         ++m_pendingTasks;
101         auto weakThis = m_weakPtrFactory.createWeakPtr();
102         m_dispatcher.postTask([weakThis, task] {
103             if (!weakThis)
104                 return;
105             ASSERT(weakThis->m_pendingTasks);
106             --weakThis->m_pendingTasks;
107             task();
108         });
109     }
110
111     void close()
112     {
113         cancelAllTasks();
114         m_isClosed = true;
115     }
116
117     void cancelAllTasks()
118     {
119         m_weakPtrFactory.revokeAll();
120         m_pendingTasks = 0;
121     }
122     bool hasPendingTasks() const { return m_pendingTasks; }
123
124 private:
125     WeakPtrFactory<GenericTaskQueue> m_weakPtrFactory;
126     TaskDispatcher<T> m_dispatcher;
127     unsigned m_pendingTasks { 0 };
128     bool m_isClosed { false };
129 };
130
131 }
132
133 #endif