refactor: simplify the handling of the "drain" event

The two event handlers are merged into one.