com.payneteasy.grpc.longpolling.server.servlet.registry.TransportRegistryImpl Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of grpc-long-polling-server Show documentation

There is a newer version: 1.0-9

Show newest version

package com.payneteasy.grpc.longpolling.server.servlet.registry;

import com.payneteasy.grpc.longpolling.common.TransportId;
import com.payneteasy.grpc.longpolling.server.LongPollingServerTransport;
import io.grpc.internal.ServerListener;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

public class TransportRegistryImpl implements ITransportRegistry {

    private static final Logger LOG = LoggerFactory.getLogger(TransportRegistryImpl.class);

    private final ServerListener                    listener;
    private final Map transports;
    private final LongPollingServerTransport        serverTransport;

    public TransportRegistryImpl(ServerListener listener, LongPollingServerTransport aServerTransport) {
        this.listener = listener;
        transports = new ConcurrentHashMap<>();
        serverTransport = aServerTransport;
    }

    @Override
    public TransportHolder findTransportHolder(TransportId transportId) {
        TransportHolder transportHolder = transports.computeIfAbsent(transportId, id -> {
            LOG.debug("Creating new TransportHolder for {}", id);
            return new TransportHolder(listener.transportCreated(serverTransport), id);
        });
        transportHolder.updateAccessTime();
        return transportHolder;
    }

    @Override
    public void cleanInactiveTransports() {
        LOG.debug("Inspecting {} transports...", transports.size());
        transports.values().removeIf(transportHolder -> {
            if(!transportHolder.isActive() || transportHolder.getLastAccessTime() < System.currentTimeMillis() - 600_000) {
                LOG.debug("Removing {} ...", transportHolder);
                return true;
            } else {
                LOG.debug("Active: {}", transportHolder);
                return false;
            }
        });
    }
}