Как выполнить некоторый код при выходе из функции?

Question

[SerJook]

Предположим у меня есть несколько точек выхода (return) из функции и надо обязательно выполнить некий код при выходе из функции. Можно выделить этот код в функцию, но можно забыть ее вызвать. Использовать goto не хочу.
Есть ли в C++ готовое решение для таких случаев? Хотелось бы что-нибудь очень простое вроде:

void function() {
  run_at_return([]{ /*do something */ });
  //...
   if (condition) {
       return;
   }
   //...
   if (condition2){
      return;
   }
}

Comments

[Roman]

В данном случае if else
можно заменить на switch case?
Тогда можно следовать принципу один вход - один выход.

[SerJook]

Roman, нет, код может быть любой, просто для примера написал if else.

Answer 1

[Alexander Movchan]

Можно воспользоватся деструктором. Не забывайте, что нельзя бросать исключения в деструкторе.

#include <iostream>
#include <functional>

template <typename T>
class defer {
private:
    std::function<T()> mFunctor;
public:
    defer(std::function<T()> functor) : mFunctor(functor) {}
    ~defer() { mFunctor(); }
};

int main() {
    defer<void> d([] { std::cout << "Deferred func!" << std::endl; });
    //...
    
    std::cout << "Hello, world!" << std::endl;
    
    if (true) {
        return 0;
    }
    //...
    if (false){
        return 1;
    }
    
    return 0;
}

Comments

[SerJook]

Для чего класс defer сделан шаблонным?

[Alexander Movchan]

SerJook, чтобы можно было передать функцию, которая имеет возвращаемое значение.

[Василий Мельников]

Хорошее решение, именно поставленной задачи.
Но я бы лучше посмотрел на то, что за код надо выполнить на каждом return и почему этих return несколько.
Почти всегда лучше подобные вещи рефакторить так, чтобы не требовалось подобных костылей. Грубо говоря, обернуть каждый используемый ресурс своим классом, стандартным, библиотечным или самописным.

Answer 2

[Victor Vakhturov]

...
auto f = []{ cout << "asdf"; };
...
class invoker
{
public:
    ~invoker()
    {
        fn();
    }
    decltype(f) fn = f;
};

    invoker c;
...

Писал с телефона, сорри если есть опечатки.

Answer 3

[Vitaly]

Не вопрос (C++17, свёртка):

template<typename... Funcs>
auto compose_sq(Funcs&&... funcs)
{
    return [=]{ (std::invoke(funcs), ...); };
}

int main()
{
    auto f1 = []{ std::cout << "f1" << std::endl; };
    auto f2 = []{ std::cout << "f2" << std::endl; };
    auto f3 = compose_sq(f1, f2);
    f3();

    return 0;
}

Ну а если нужно сохранить результат возврата значения первой функции, то придётся ещё немного кода написать. Думаю, что общее направление вам понятно. Если что, то дополнительную информацию можно найти тут: https://en.cppreference.com/w/cpp/language/fold

Comments

[SerJook]

Я к сожалению не понял, как это поможет мне в моем вопросе.

[Vitaly]

SerJook, а код запускать пробовали? Функция 2 всегда выполнится после функции 1 (ну, кроме случаев с исключениями). Т.е. вот функция 2 это тот код, который всегда будет вызван после функции 1.

[Roman]

Vitaly,
в итоге в коде (без лишних подробностей) это будет

так

//...
class __lambda_6_12
//...
public: inline /*constexpr */ void operator()() const
{
    std::invoke(funcs) , std::invoke(funcs);
}
//...

последовательный вызов, а автор хочет вызов второй до/перед возвратом из первой.

[Alexander Movchan]

Vitaly, так вопрос же в другом. Возможно нужно сделать что-то с локальным объектом в функции. Например, закрытыть файлы или вызвать delete для созданных объектов.

[Vitaly]

Roman, если именно до возврата, то я бы предпочёл объект, который вызывает код в деструкторе как советовали сверху.
Хотя, мой код не хуже, если не надо оперировать над объектами из тела функции.

[Vitaly]

Александр Мовчан, да, тогда объект, который делает что-то в деструкторе -- проверенное и рабочее решение.