Examples with Message io.scalecube.transport.Message used on opensource projects

Example 51 with Message

use of io.scalecube.transport.Message in project scalecube by scalecube.

the class RoundRobinServiceRouter method route.

@Override
public Optional<ServiceInstance> route(Message request) {
    String serviceName = request.header(ServiceHeaders.SERVICE_REQUEST);
    List<ServiceInstance> serviceInstances = serviceRegistry.serviceLookup(serviceName).stream().filter(instance -> instance.methodExists(request.header(ServiceHeaders.METHOD))).collect(Collectors.toList());
    if (serviceInstances.size() > 1) {
        AtomicInteger counter = counterByServiceName.computeIfAbsent(serviceName, or -> new AtomicInteger());
        int index = counter.incrementAndGet() % serviceInstances.size();
        return Optional.of(serviceInstances.get(index));
    } else if (serviceInstances.size() == 1) {
        return Optional.of(serviceInstances.get(0));
    } else {
        LOGGER.warn("route selection return null since no service instance was found for {}", serviceName);
        return Optional.empty();
    }
}