1 /* 2 * Copyright (c) 2021-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 ROSEN_RENDER_SERVICE_BASE_TRANSACTION_RS_RENDER_SERVICE_CONNECT_HUB_H 17 #define ROSEN_RENDER_SERVICE_BASE_TRANSACTION_RS_RENDER_SERVICE_CONNECT_HUB_H 18 19 #include <mutex> 20 #include <platform/ohos/rs_irender_service.h> 21 #include "platform/common/rs_log.h" 22 23 namespace OHOS { 24 namespace Rosen { 25 using OnConnectCallback = std::function<void(sptr<RSIRenderServiceConnection>&)>; 26 class RSRenderServiceConnectHub : public RefBase { 27 public: 28 static sptr<RSIRenderServiceConnection> GetRenderService(); SetOnConnectCallback(OnConnectCallback cb)29 static void SetOnConnectCallback(OnConnectCallback cb) 30 { 31 onConnectCallback_ = cb; 32 // if already connected, call the callback immediately 33 if (instance_ && instance_->conn_ && onConnectCallback_) { 34 onConnectCallback_(instance_->conn_); 35 } 36 } 37 GetConnectHubInstance()38 static sptr<RSRenderServiceConnectHub> GetConnectHubInstance() 39 { 40 return instance_; 41 } 42 GetToken()43 RSIConnectionToken* GetToken() 44 { 45 return token_.GetRefPtr(); 46 } 47 48 private: 49 static sptr<RSRenderServiceConnectHub> GetInstance(); 50 static void Init(); 51 static void Destroy(); 52 53 RSRenderServiceConnectHub(); 54 ~RSRenderServiceConnectHub() noexcept; 55 56 class RenderServiceDeathRecipient final : public IRemoteObject::DeathRecipient { 57 public: RenderServiceDeathRecipient(wptr<RSRenderServiceConnectHub> rsConnHub)58 explicit RenderServiceDeathRecipient(wptr<RSRenderServiceConnectHub> rsConnHub) 59 : rsConnHub_(rsConnHub) {} 60 ~RenderServiceDeathRecipient() noexcept final = default; 61 62 DISALLOW_COPY_AND_MOVE(RenderServiceDeathRecipient); 63 64 void OnRemoteDied(const wptr<IRemoteObject> &remote) final override; 65 66 private: 67 wptr<RSRenderServiceConnectHub> rsConnHub_; 68 }; 69 70 sptr<RSIRenderServiceConnection> GetRenderServiceConnection(); 71 bool Connect(); 72 void ConnectDied(); 73 74 mutable std::mutex mutex_; 75 sptr<RSIRenderService> renderService_; 76 sptr<RSIConnectionToken> token_; 77 sptr<RSIRenderServiceConnection> conn_; 78 sptr<IRemoteObject::DeathRecipient> deathRecipient_; 79 80 static std::once_flag flag_; 81 static sptr<RSRenderServiceConnectHub> instance_; 82 static OnConnectCallback onConnectCallback_; 83 }; 84 } // namespace Rosen 85 } // namespace OHOS 86 87 #endif // ROSEN_RENDER_SERVICE_BASE_TRANSACTION_RS_RENDER_SERVICE_CONNECT_HUB_H 88