refactor: Allow socket register by each handler separately