1/*
2 * Copyright (C) 2017 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. AND ITS CONTRIBUTORS ``AS IS''
14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23 * THE POSSIBILITY OF SUCH DAMAGE.
24 */
25
26#pragma once
27
28#include <wtf/CrossThreadQueue.h>
29#include <wtf/CrossThreadTask.h>
30#include <wtf/Lock.h>
31#include <wtf/Threading.h>
32
33namespace WTF {
34
35class RegistrationStore;
36class SQLiteDatabase;
37
38class CrossThreadTaskHandler {
39public:
40 WTF_EXPORT_PRIVATE virtual ~CrossThreadTaskHandler();
41 enum class AutodrainedPoolForRunLoop { DoNotUse, Use };
42
43protected:
44 WTF_EXPORT_PRIVATE CrossThreadTaskHandler(const char* threadName, AutodrainedPoolForRunLoop = AutodrainedPoolForRunLoop::DoNotUse);
45
46 WTF_EXPORT_PRIVATE void postTask(CrossThreadTask&&);
47 WTF_EXPORT_PRIVATE void postTaskReply(CrossThreadTask&&);
48 WTF_EXPORT_PRIVATE void suspendAndWait();
49 WTF_EXPORT_PRIVATE void resume();
50
51private:
52 void handleTaskRepliesOnMainThread();
53 void taskRunLoop();
54
55 AutodrainedPoolForRunLoop m_useAutodrainedPool { AutodrainedPoolForRunLoop::DoNotUse };
56
57 Lock m_taskThreadCreationLock;
58 Lock m_mainThreadReplyLock;
59 bool m_mainThreadReplyScheduled { false };
60
61 bool m_shouldSuspend { false };
62 Condition m_shouldSuspendCondition;
63 Lock m_shouldSuspendLock;
64
65 bool m_suspended { false };
66 Condition m_suspendedCondition;
67 Lock m_suspendedLock;
68
69 CrossThreadQueue<CrossThreadTask> m_taskQueue;
70 CrossThreadQueue<CrossThreadTask> m_taskReplyQueue;
71};
72
73} // namespace WTF
74
75using WTF::CrossThreadTaskHandler;
76
77