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 OHOS_ZIDL_WORK_SCHED_STUB_IMP_H
17 #define OHOS_ZIDL_WORK_SCHED_STUB_IMP_H
18 
19 #include <memory>
20 #include "js_work_scheduler_extension.h"
21 #include "work_scheduler_stub.h"
22 #include "work_sched_hilog.h"
23 
24 namespace OHOS {
25 namespace WorkScheduler {
26 class WorkSchedulerStubImp : public WorkSchedulerStub {
27 public:
WorkSchedulerStubImp(const std::shared_ptr<JsWorkSchedulerExtension> & extension)28     explicit WorkSchedulerStubImp(const std::shared_ptr<JsWorkSchedulerExtension>& extension)
29         : extension_(extension) {}
30 
~WorkSchedulerStubImp()31     ~WorkSchedulerStubImp() override {}
32 
33     /**
34      * @brief The OnWorkStart callback.
35      *
36      * @param workInfo The info of work.
37      */
38     void OnWorkStart(WorkInfo& workInfo) override;
39 
40     /**
41      * @brief The OnWorkStop callback.
42      *
43      * @param workInfo The info of work.
44      */
45     void OnWorkStop(WorkInfo& workInfo) override;
46 private:
47     std::weak_ptr<JsWorkSchedulerExtension> extension_;
48 };
49 } // namespace WorkScheduler
50 } // namespace OHOS
51 #endif // OHOS_ZIDL_WORK_SCHED_STUB_IMP_H
52 
53