Correctly handle drain with new netty api