com.netflix.eureka2.testkit.embedded.server.EmbeddedDashboardServer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of eureka-testkit Show documentation
eureka-testkit developed by Netflix
The newest version!
package com.netflix.eureka2.testkit.embedded.server;

import com.google.inject.Module;
import com.netflix.eureka2.DashboardEurekaClientBuilder;
import com.netflix.eureka2.EurekaDashboardModule;
import com.netflix.eureka2.client.EurekaClient;
import com.netflix.eureka2.client.EurekaClientBuilder;
import com.netflix.eureka2.client.resolver.ServerResolver;
import com.netflix.eureka2.config.EurekaDashboardConfig;
import com.netflix.eureka2.registry.datacenter.LocalDataCenterInfo.DataCenterType;
import com.netflix.eureka2.testkit.embedded.server.EmbeddedDashboardServer.DashboardServerReport;
import com.netflix.eureka2.transport.EurekaTransports.Codec;

/**
 * @author Tomasz Bak
 */
public class EmbeddedDashboardServer extends EmbeddedEurekaServer {

    private static final String DASHBOARD_SERVER_NAME = "eureka2-dashboard";
    private static final int DASHBOARD_SERVER_PORTS_FROM = 16000;

    private final ServerResolver registrationServerResolver;
    private final ServerResolver discoveryServerResolver;

    public EmbeddedDashboardServer(EurekaDashboardConfig config,
                                   ServerResolver registrationServerResolver,
                                   ServerResolver discoveryServerResolver,
                                   boolean withExt,
                                   boolean withDashboard) {
        super(config, withExt, withDashboard);
        this.registrationServerResolver = registrationServerResolver;
        this.discoveryServerResolver = discoveryServerResolver;
    }

    @Override
    public void start() {
        final EurekaClient eurekaClient = new EurekaClientBuilder(discoveryServerResolver, registrationServerResolver).build();
        Module[] modules = {
                new EurekaDashboardModule(config) {
                    @Override
                    protected void configure() {
                        super.configure();
                        bind(DashboardEurekaClientBuilder.class).toInstance(new DashboardEurekaClientBuilder(eurekaClient));
                    }
                }
        };

        setup(modules);
    }

    @Override
    public DashboardServerReport serverReport() {
        String dashboardURI = "http://localhost:" + config.getDashboardPort() + "/dashboard.html";
        return new DashboardServerReport(
                dashboardURI,
                formatAdminURI()
        );
    }

    public static EmbeddedDashboardServer newDashboard(ServerResolver registrationServerResolver,
                                                       ServerResolver discoveryServerResolver,
                                                       boolean withExt,
                                                       boolean withAdminUI) {
        EurekaDashboardConfig config = EurekaDashboardConfig.newBuilder()
                .withAppName(DASHBOARD_SERVER_NAME)
                .withVipAddress(DASHBOARD_SERVER_NAME)
                .withDataCenterType(DataCenterType.Basic)
                .withCodec(Codec.Avro)
                .withShutDownPort(DASHBOARD_SERVER_PORTS_FROM + 3)
                .withWebAdminPort(DASHBOARD_SERVER_PORTS_FROM + 4)
                .build();
        return new EmbeddedDashboardServer(config, registrationServerResolver, discoveryServerResolver, withExt, withAdminUI);
    }

    public static class DashboardServerReport {

        private final String dashboardURI;
        private final String adminURI;

        public DashboardServerReport(String dashboardURI, String adminURI) {
            this.dashboardURI = dashboardURI;
            this.adminURI = adminURI;
        }

        public String getDashboardURI() {
            return dashboardURI;
        }

        public String getAdminURI() {
            return adminURI;
        }
    }
}