com.xti.spring.cloud.heroku.discovery.instance.HerokuSpaceTopologyInstanceProvider Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-cloud-heroku-registry-discovery Show documentation
The newest version!
package com.xti.spring.cloud.heroku.discovery.instance;

import com.xti.spring.cloud.heroku.discovery.instance.port.PortSelectorChain;
import com.xti.spring.cloud.heroku.discovery.topology.HerokuSpaceTopologyDyno;
import com.xti.spring.cloud.heroku.discovery.topology.HerokuSpaceTopologyListener;
import com.xti.spring.cloud.heroku.discovery.topology.HerokuSpaceTopologyV1;
import org.springframework.cloud.client.ServiceInstance;

import java.util.ArrayList;
import java.util.List;
import java.util.Objects;

public class HerokuSpaceTopologyInstanceProvider implements HerokuInstanceProvider {

    private PortSelectorChain portSelectorChain;
    private HerokuSpaceTopologyListener listener;

    public HerokuSpaceTopologyInstanceProvider(PortSelectorChain portSelectorChain, HerokuSpaceTopologyListener listener) {
        this.portSelectorChain = portSelectorChain;
        this.listener = listener;
    }

    @Override
    public List getServiceInstances(String appProcess) {

        List serviceInstances = new ArrayList<>();

        final String[] herokuParts = appProcess.split("\\.");
        final String processName = herokuParts[0];
        final String appName = herokuParts[1];

        HerokuSpaceTopologyV1 topology = listener.getTopology();

        if(topology != null){
            topology.getApps().stream()
                    .filter(app -> Objects.equals(app.getAppName(), appName))
                    .forEach(app -> app.getFormation().stream()
                            .filter(process -> Objects.equals(process.getProcessType(), processName))
                            .forEach(process -> {
                                        for (HerokuSpaceTopologyDyno dyno : process.getDynos()) {
                                            ServiceInstance remoteServiceInstance = new DynoProcessServiceInstanceBuilder()
                                                    .appProcess(appProcess)
                                                    .host(dyno.getPrivateIp())
                                                    .portSelectorChain(portSelectorChain)
                                                    .build();
                                            serviceInstances.add(remoteServiceInstance);
                                        }
                                    }
                            ));
        }

        return serviceInstances;
    }

    @Override
    public ServiceInstance getLocalServiceInstance() {
        return new DynoProcessServiceInstanceBuilder().portSelectorChain(portSelectorChain).local(true).build();
    }
}