Possible way to capture tracing for anonymous functions

traceOps.scala

def trace[U, V](
    functionName: String
)(func: U => V)(implicit txnExecutionContext: TxnExecutionContext): U => V =
  new Function1[U, V] {
    // set functionName, call trace async here
    def apply(u: U): V = {
      txnExecutionContext.token.link()
      func(u)
    }
  }


// Function could be called as

Future(1)
  .flatMap(trace("increment")(x => x + 1))
  .map(trace("decrement")(y => y - 1))