1 /*
2  * Copyright (c) 2021 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 #ifndef OHOS_ABILITY_RUNTIME_REVERSE_CONTINUATION_SCHEDULER_PRIMARY_H
16 #define OHOS_ABILITY_RUNTIME_REVERSE_CONTINUATION_SCHEDULER_PRIMARY_H
17 
18 #include <memory>
19 #include "reverse_continuation_scheduler_primary_handler_interface.h"
20 #include "reverse_continuation_scheduler_primary_stub.h"
21 #include "ability_handler.h"
22 
23 namespace OHOS {
24 namespace AppExecFwk {
25 class ReverseContinuationSchedulerPrimary : public ReverseContinuationSchedulerPrimaryStub {
26 public:
27     ReverseContinuationSchedulerPrimary(
28         const std::weak_ptr<IReverseContinuationSchedulerPrimaryHandler> &continuationHandler,
29         const std::shared_ptr<AbilityHandler> &mainHandler);
30     virtual ~ReverseContinuationSchedulerPrimary() = default;
31 
32     /**
33      * @brief Replica call this method when it terminated.
34      */
35     void NotifyReplicaTerminated() override;
36 
37     /**
38      * @brief Replica call this method to notify primary go on.
39      *
40      * @param want Contains data to be restore.
41      * @return True if success, otherwise false.
42      */
43     bool ContinuationBack(const AAFwk::Want &want) override;
44 
45 private:
46     void HandlerNotifyReplicaTerminated();
47     void HandlerContinuationBack(const AAFwk::Want &want);
48 
49     std::weak_ptr<IReverseContinuationSchedulerPrimaryHandler> continuationHandler_;
50     const std::shared_ptr<AbilityHandler> mainHandler_;
51 };
52 }  // namespace AppExecFwk
53 }  // namespace OHOS
54 #endif /* OHOS_ABILITY_RUNTIME_REVERSE_CONTINUATION_SCHEDULER_PRIMARY_H */
55