message_multiplexer: improve multiplexer code

Signed-off-by: Sébastien Boisvert <boisvert@anl.gov>