1 /*
2 * Copyright (c) 2021-2024 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 "mock_task_factory.h"
17
18 #include "distributed_hardware_log.h"
19 #include "mock_disable_task.h"
20 #include "mock_enable_task.h"
21 #include "mock_meta_disable_task.h"
22 #include "mock_meta_enable_task.h"
23 #include "mock_offline_task.h"
24 #include "mock_online_task.h"
25 #include "task_board.h"
26
27 namespace OHOS {
28 namespace DistributedHardware {
29 IMPLEMENT_SINGLE_INSTANCE(MockTaskFactory);
CreateTask(TaskType taskType,TaskParam taskParam,std::shared_ptr<Task> fatherTask)30 std::shared_ptr<Task> MockTaskFactory::CreateTask(TaskType taskType, TaskParam taskParam,
31 std::shared_ptr<Task> fatherTask)
32 {
33 std::shared_ptr<Task> task = nullptr;
34 switch (taskType) {
35 case TaskType::ENABLE: {
36 task = std::make_shared<MockEnableTask>(taskParam.networkId, taskParam.uuid, taskParam.udid,
37 taskParam.dhId, taskParam.dhType);
38 break;
39 }
40 case TaskType::DISABLE: {
41 task = std::make_shared<MockDisableTask>(taskParam.networkId, taskParam.uuid, taskParam.udid,
42 taskParam.dhId, taskParam.dhType);
43 break;
44 }
45 case TaskType::META_ENABLE: {
46 task = std::make_shared<MockMetaEnableTask>(taskParam.networkId, taskParam.uuid, taskParam.udid,
47 taskParam.dhId, taskParam.dhType);
48 break;
49 }
50 case TaskType::META_DISABLE: {
51 task = std::make_shared<MockMetaDisableTask>(taskParam.networkId, taskParam.uuid, taskParam.udid,
52 taskParam.dhId, taskParam.dhType);
53 break;
54 }
55 case TaskType::ON_LINE: {
56 task = std::make_shared<MockOnLineTask>(taskParam.networkId, taskParam.uuid, taskParam.udid,
57 taskParam.dhId, taskParam.dhType);
58 break;
59 }
60 case TaskType::OFF_LINE: {
61 task = std::make_shared<MockOffLineTask>(taskParam.networkId, taskParam.uuid, taskParam.udid,
62 taskParam.dhId, taskParam.dhType);
63 break;
64 }
65 default: {
66 DHLOGE("CreateTask type invalid, type: %{public}d", taskType);
67 return nullptr;
68 }
69 }
70
71 if (fatherTask != nullptr) {
72 task->SetFatherTask(fatherTask);
73 fatherTask->AddChildrenTask(task);
74 }
75 TaskBoard::GetInstance().AddTask(task);
76
77 return task;
78 }
79 } // namespace DistributedHardware
80 } // namespace OHOS
81