1 /*
2 * Copyright (c) 2020 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 "ability_connect_task.h"
17
18 #include "ability_connect_record.h"
19 #include "utils.h"
20
21 namespace OHOS {
AbilityConnectTask(AbilityMgrContext * context,const Want & want,const SvcIdentity & identity,uint64_t token)22 AbilityConnectTask::AbilityConnectTask(AbilityMgrContext *context, const Want &want, const SvcIdentity& identity,
23 uint64_t token) : AbilityTask(context), want_(want), connectSid_(identity), token_(token)
24 {
25 }
26
Execute()27 AbilityMsStatus AbilityConnectTask::Execute()
28 {
29 PRINTD("AbilityConnectTask", "start");
30 if (abilityMgrContext_ == nullptr || want_.element == nullptr) {
31 return AbilityMsStatus::TaskStatus("connectTask", "invalid argument");
32 }
33 AbilityStackManager &stackManager = AbilityStackManager::GetInstance();
34 PageAbilityRecord *service = stackManager.FindServiceAbility(*abilityMgrContext_,
35 want_.element->bundleName, want_.element->abilityName);
36 if (service == nullptr) {
37 return AbilityMsStatus::TaskStatus("connectTask", "service ability does not exists");
38 }
39 if (service->IsPerformStop()) {
40 return AbilityMsStatus::TaskStatus("connectTask", "service has stopped");
41 }
42 return PerformConnectTask(service);
43 }
44
PerformConnectTask(PageAbilityRecord * service)45 AbilityMsStatus AbilityConnectTask::PerformConnectTask(PageAbilityRecord *service)
46 {
47 ConnectStatus serviceConnectStatus = service->GetConnectStatus();
48 if (serviceConnectStatus == ConnectStatus::DISCONNECTING || serviceConnectStatus == ConnectStatus::DISCONNECT ||
49 serviceConnectStatus == ConnectStatus::STOPPING || serviceConnectStatus == ConnectStatus::STOPPED) {
50 PRINTW("AbilityConnectTask", "service status has disconnected");
51 return AbilityMsStatus::TaskStatus("AbilityConnectTask", "service status has disconnected");
52 }
53 auto connectRecord = service->GetConnectRecord(connectSid_);
54 if (connectRecord == nullptr) {
55 connectRecord = new AbilityConnectRecord(connectSid_, token_);
56 connectRecord->SetStatus(ConnectStatus::CONNECTING);
57 service->pushConnectRecord(connectRecord);
58 }
59 AbilityMsStatus abilityMsStatus = AbilityMsStatus::Ok();
60 switch (serviceConnectStatus) {
61 case ConnectStatus::INIT: {
62 if (service->GetCurrentState() == STATE_UNINITIALIZED) {
63 PRINTI("AbilityConnectTask", "service status is initializing, waiting connecting");
64 service->SetConnectStatus(ConnectStatus::WAIT_CONNECT);
65 } else {
66 PRINTI("AbilityConnectTask", "service status is init, connection is connecting");
67 service->SetConnectStatus(ConnectStatus::CONNECTING);
68 abilityMsStatus = service->ConnectAbility();
69 }
70 break;
71 }
72 case ConnectStatus::CONNECTED: {
73 if (connectRecord->GetStatus() == ConnectStatus::CONNECTING) {
74 PRINTI("AbilityConnectTask", "service status has connected, connection is connecting");
75 abilityMsStatus = service->ConnectAbilityDone();
76 } else {
77 PRINTI("AbilityConnectTask", "connection has connected");
78 }
79 break;
80 }
81 default:
82 break;
83 }
84 return abilityMsStatus;
85 }
86 } // namespace OHOS
87