Как выполнить некоторый код, когда блок выходит из-за «ожидания»?

Если вы хотите регистрировать, когда Something завершает какие-либо синхронные действия и возвращает задачу, это легко сделать:

var task = Something(); 
/* log as you like */ 
await task;

Удивительно, но это можно сделать, используя AsyncLocal. AsyncLocal похож на ThreadLocal, за исключением того, что он проходит через асинхронный код, который может переключать потоки. У него есть конструктор, который позволяет вам прослушивать изменения значений и даже сообщает вам, почему значение изменилось. Его можно изменить либо потому, что вы явно установили Value, либо если происходит асинхронное переключение контекста (в этом случае Value изменяется на ноль/по умолчанию, когда управление уходит, и возвращается к исходному значению, когда управление возвращается). Это позволяет нам определить, когда будет достигнуто первое ожидание, и не только первое ожидание, но и ожидание, которое введет переключение контекста (например, await Task.CompletedTask не вызовет переключение контекста). Таким образом, при первом таком переключении Task будет возвращен вызывающему абоненту.

Вот пример кода:

public class Program {
    public static void Main() {
        var task = Test();
        Console.WriteLine("Control flow got out of Test");
        task.Wait();
    }

    static async Task Test() {
        using (LogMethodCall()) {
            await Task.Delay(1000);
            Console.WriteLine("Finished using block");
        }
    }

    static IDisposable LogMethodCall([CallerMemberName] string methodName = null) {        
        return new Logger(methodName);
    }

    private class Logger : IDisposable {
        private readonly string _methodName;
        private AsyncLocal<object> _alocal;        
        private bool _disposed;
        public Logger(string methodName) {
            Console.WriteLine($"{methodName} entered");
            _methodName = methodName;            
            _alocal = new AsyncLocal<object>(OnChanged);
            _alocal.Value = new object();
        }

        private void OnChanged(AsyncLocalValueChangedArgs<object> args) {
            if (_disposed)
                return;
            // this property tells us that value changed because of context switch
            if (args.ThreadContextChanged) {                                
                Dispose();
            }
        }

        public void Dispose() {
            // prevent multiple disposal
            if (_disposed)
                return;                        
            _disposed = true;
            _alocal = null;
            Console.WriteLine($"{_methodName} exited");
        }
    }
}

Он выводит:

Test entered
Test exited
Control flow got out of Test
Finished using block

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