1 /* 2 * Copyright (c) 2023 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 UTIL_SINGLETON_H 17 #define UTIL_SINGLETON_H 18 19 #include "cb_func.h" 20 namespace ffrt { 21 template <typename T> 22 class SingletonRegister { 23 public: Instance_()24 static SingletonRegister<T> &Instance_() 25 { 26 static SingletonRegister<T> ins; 27 return ins; 28 } 29 Instance()30 static T &Instance() 31 { 32 return Instance_().cb_(); 33 } 34 RegistInsCb(typename SingleInsCB<T>::Instance && cb)35 static void RegistInsCb(typename SingleInsCB<T>::Instance &&cb) 36 { 37 Instance_().cb_ = std::move(cb); 38 } 39 40 private: 41 typename SingleInsCB<T>::Instance cb_; 42 }; 43 } // namespace ffrt 44 #endif