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 #include "event.h"
17
18 namespace STtools {
WaitCompleted(Event & event,const std::string & eventName,const int code,const int timeout)19 int WaitCompleted(Event &event, const std::string &eventName, const int code, const int timeout)
20 {
21 return event.WaitingMessage(std::to_string(code) + eventName, timeout, false);
22 }
23
Completed(Event & event,const std::string & eventName,const int code)24 void Completed(Event &event, const std::string &eventName, const int code)
25 {
26 return event.CompleteMessage(std::to_string(code) + eventName);
27 }
28
CleanMsg(Event & event)29 void CleanMsg(Event &event)
30 {
31 return event.Clean();
32 }
33
Event()34 Event::Event()
35 {
36 waiting_message_ = "";
37 complete_message_.clear();
38 }
39
~Event()40 Event::~Event()
41 {
42 waiting_message_ = "";
43 std::vector<std::string> tmp_vector;
44 tmp_vector.swap(complete_message_);
45 complete_message_.clear();
46 }
47
Compare()48 bool Event::Compare()
49 {
50 if (!waiting_message_.empty()) {
51 for (size_t i = 0; i < complete_message_.size(); i++) {
52 if (!waiting_message_.compare(complete_message_.at(i))) {
53 complete_message_.erase(std::begin(complete_message_) + i, std::begin(complete_message_) + i + 1);
54 waiting_message_ = "";
55 return true;
56 }
57 }
58 }
59 return false;
60 }
61
WaitingMessage(const std::string & message,int timeout_ms,bool locked)62 int Event::WaitingMessage(const std::string &message, int timeout_ms, bool locked)
63 {
64 std::unique_lock<std::mutex> lock(mutex_);
65 waiting_message_ = message;
66 if (Compare()) {
67 return 0;
68 }
69
70 if (locked) {
71 cv_.wait(lock);
72 return 0;
73 }
74
75 if (cv_.wait_for(lock, std::chrono::seconds(timeout_ms)) == std::cv_status::timeout) {
76 waiting_message_ = "";
77 return -1;
78 }
79 return 0;
80 }
81
CompleteMessage(const std::string & message)82 void Event::CompleteMessage(const std::string &message)
83 {
84 std::unique_lock<std::mutex> lock(mutex_);
85 if (!waiting_message_.compare(message)) {
86 waiting_message_ = "";
87 cv_.notify_all();
88 return;
89 }
90 complete_message_.push_back(message);
91 }
92
Clean()93 void Event::Clean()
94 {
95 waiting_message_ = "";
96 complete_message_.clear();
97 }
98 } // namespace STtools