1/*
2 * Copyright (c) 2022 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 *     http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16#ifndef JS_CONCURRENT_MODULE_WORKER_MESSAGE_QUEUE_H
17#define JS_CONCURRENT_MODULE_WORKER_MESSAGE_QUEUE_H
18
19#include <mutex>
20#include <queue>
21#include "napi/native_api.h"
22#include "napi/native_node_api.h"
23
24namespace Commonlibrary::Concurrent::WorkerModule {
25using MessageDataType = void*;
26class MessageQueue final {
27public:
28    void EnQueue(MessageDataType data);
29    bool DeQueue(MessageDataType *data);
30    bool IsEmpty() const;
31    void Clear(napi_env env);
32    size_t GetSize() const
33    {
34        return queue_.size();
35    }
36
37private:
38    std::mutex queueLock_;
39    std::queue<MessageDataType> queue_;
40};
41
42class MarkedMessageQueue final {
43public:
44    void Push(uint32_t id, MessageDataType data);
45    void Pop();
46    std::pair<uint32_t, MessageDataType> Front();
47    bool IsEmpty();
48    void Clear(napi_env env);
49    size_t GetSize()
50    {
51        std::unique_lock<std::mutex> lock(queueLock_);
52        return queue_.size();
53    }
54
55private:
56    std::mutex queueLock_;
57    std::queue<std::pair<uint32_t, MessageDataType>> queue_;
58};
59}  // namespace Commonlibrary::Concurrent::WorkerModule
60#endif // JS_CONCURRENT_MODULE_WORKER_MESSAGE_QUEUE_H
61