def METHOD_NAME(self, address):
    if self._args.get("mock"):
        # if mock, the ps open a file and set
        # the file name like ps_address_1.1.1.1:1001
        file_name = "ps_address_{}".format(address)
        with open(file_name, "w") as f:
            f.write("")
    else:
        # if not mock, the ps report its service_addr to dlrover master
        ps_nodes, _ = master_client.query_ps_nodes()
        logger.info("ps nodes is {}".format(ps_nodes))
        res = master_client.update_node_addr(
            self.task_type, self.task_id, address
        )
        logger.info(
            "{}:{} updating ps address {}".format(
                self.task_type, self.task_id, res
            )
        )