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 WORK_SCHED_SERVICES_WORK_CONN_MANAGER_H
17 #define WORK_SCHED_SERVICES_WORK_CONN_MANAGER_H
18 
19 #include <map>
20 #include <string>
21 #include <memory>
22 
23 #include "work_scheduler_connection.h"
24 #include "work_status.h"
25 #include "ffrt.h"
26 
27 namespace OHOS {
28 using namespace Utils;
29 namespace WorkScheduler {
30 class WorkConnManager {
31 public:
32     /**
33      * @brief Start work.
34      *
35      * @param workStatus The status of work.
36      * @return True if success,else false.
37      */
38     bool StartWork(std::shared_ptr<WorkStatus> workStatus);
39     /**
40      * @brief Stop work.
41      *
42      * @param workStatus The status of work.
43      * @param isTimeOut If the work is timeout.
44      * @return True if success,else false.
45      */
46     bool StopWork(std::shared_ptr<WorkStatus> workStatus, bool isTimeOut);
47 private:
48     void RemoveConnInfo(std::string &workId);
49     void AddConnInfo(std::string &workId, sptr<WorkSchedulerConnection> &connection);
50     sptr<WorkSchedulerConnection> GetConnInfo(std::string &workId);
51     bool DisConnect(sptr<WorkSchedulerConnection> connect);
52     void WriteStartWorkEvent(std::shared_ptr<WorkStatus> workStatus);
53     ffrt::mutex connMapMutex_;
54     std::map<std::string, sptr<WorkSchedulerConnection>> connMap_;
55     std::map<std::string, int32_t> eventIdMap_;
56 };
57 } // namespace WorkScheduler
58 } // namespace OHOS
59 #endif // WORK_SCHED_SERVICES_WORK_CONN_MANAGER_H