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 IAM_SCOPE_GUARD_H 17 #define IAM_SCOPE_GUARD_H 18 19 #include <functional> 20 #include <string> 21 22 #include "iam_logger.h" 23 #include "iam_defines.h" 24 #include "nocopyable.h" 25 26 namespace OHOS { 27 namespace UserIam { 28 namespace Common { 29 class IamScopeGuard : public NoCopyable { 30 public: 31 using ExitAction = std::function<void()>; IamScopeGuard(ExitAction && action)32 explicit IamScopeGuard(ExitAction &&action) : action_(std::forward<ExitAction>(action)) 33 { 34 } ~IamScopeGuard()35 ~IamScopeGuard() override 36 { 37 if (action_ != nullptr) { 38 IAM_TRY(action_(), return); 39 } 40 } 41 42 private: 43 ExitAction action_; 44 }; 45 46 #define ON_SCOPE_EXIT_ID(id, ...) auto exitGuard##id = OHOS::UserIam::Common::IamScopeGuard([]() { __VA_ARGS__; }) 47 48 #define ON_SCOPE_EXIT(...) ON_SCOPE_EXIT_ID(__LINE__, __VA_ARGS__) 49 } // namespace Common 50 } // namespace UserIam 51 } // namespace OHOS 52 53 #endif // IAM_SCOPE_GUARD_H