testcontainers-rs icon indicating copy to clipboard operation
testcontainers-rs copied to clipboard

Use "drop-notifier" pattern for to asynchronously shut down containers with `Http` client

Open thomaseizinger opened this issue 3 years ago • 2 comments

Not sure if this is documented anywhere but I've since learned about this very nifty pattern for doing async work in a Drop implementation:

  1. Have a Void async, oneshot channel
  2. Store the Sender in the struct you intend to eventually Drop
  3. Listen on the receiver in a task
  4. Once you get oneshot::Canceled, you know that the struct has been dropped
  5. Perform the async work

To utilize this, we need to store the Sender in ContainerAsync and the Receiver in Http. The client (Http), needs to create one task per container that it starts and shut it down as soon as it receives oneshot::Canceled.

This should allow us to drop the executor feature on futures.

thomaseizinger avatar Jan 16 '22 23:01 thomaseizinger

Can you please help me understand the idea? I'm failing to come up with a possible implementation for a good half of what you wrote :D

To execute some async work — we'd need to spawn an async task, and to do that — we need an executor. Also, to perform a "proper" drop because of the threads and stuff, we'd need to do a join on all the spawned threads inside the Http's Drop...

What am I missing? Do you have an example of the implementation?

andrey-yantsen avatar Jan 17 '22 20:01 andrey-yantsen

So when writing something up, I realized that actually utilizing the drop notifier here is a bit pointless :D

I'll write it up anyway because I think it is a really interested pattern.

Pattern

What this does not do is synchronize the drop with the async code that executes, i.e. Drop will complete before the async work gets done.

  1. Return a DropListener in addition to ContainterAsync from Http::run.
  2. DropListener would implement Future and probably just store a BoxFuture inside.
  3. The BoxFuture would be constructed from something like:
let client = Http {
    inner: self.inner.clone(),
};

Box::pin(async move {
	match listener.await {
		Ok(void) => void::unreachable(void),
		Err(Canceled {}) => {
			client.rm(id).await;
		}
	}
});

Returning a dedicated future makes us executor agnostic. What users would have to do is:

let (container, drop_listener) = client.run(HelloWorld {}).await;
tokio::spawn(drop_listener);

This is quite a klunky API though and with testcontainers being targeted at making tests ergonomic, this is quite the deal breaker.

There is an alternative to this where we define an Executor trait and users you can pass in an instance when you initially construct Http. That would allow us to spawn tasks onto the user's executor. However, if we do that, we might as well just call

executor.spawn(async move {
	client.rm(id).await;
})

from the container's Drop impl directly, no need to use the drop notifier pattern.

thomaseizinger avatar Jan 17 '22 22:01 thomaseizinger