bb494a3be8387467d892f10a8dbba4d2e1a68a3b,python/ray/serve/router.py,Router,set_traffic,#Router#Any#Any#,261

Before Change

        logger.debug("Setting traffic for endpoint %s to %s", endpoint,
                     traffic_dict)
        self.traffic[endpoint] = traffic_dict
        await self.flush()

    async def remove_endpoint(self, endpoint):
        logger.debug("Removing endpoint {}".format(endpoint))

After Change

    async def set_traffic(self, endpoint, traffic_dict):
        logger.debug("Setting traffic for endpoint %s to %s", endpoint,
                     traffic_dict)
        async with self.flush_lock:
            self.traffic[endpoint] = self.policy(traffic_dict,
                                                 **self.policy_kwargs)
            await self.flush_endpoint_queue(endpoint)

    async def remove_endpoint(self, endpoint):
        logger.debug("Removing endpoint {}".format(endpoint))
        async with self.flush_lock:
            await self.flush_endpoint_queue(endpoint)

Instances