The heartbeat method uses a context.Background() for two of its network operation which effectively lets a heartbeat wait forever. This can cause an issue for disconnecting clients that are waiting to send an empty struct on the done channel.