Need variants of ExecuteInTx and ExecuteTx that take a `fn(context.Context, ...)` as arg

The ExecuteTx and ExecuteInTx APIs take a function callback that only takes either no argument or a tx object.

We want to pass a callback instead that takes the context as argument.

cc @rafiss