Register worker functions once we are connected

This also re-registers once a server appears ensuring reconnection is
properly set up.

Change-Id: I9b915482494d1b273991cb8bdd1b6ad963cf25f2