Examples with AdditionalServletWithPulsarService org.apache.pulsar.broker.web.plugin.servlet.AdditionalServletWithPulsarService used on opensource projects

Example 1 with AdditionalServletWithPulsarService

use of org.apache.pulsar.broker.web.plugin.servlet.AdditionalServletWithPulsarService in project pulsar by yahoo.

the class PulsarService method addBrokerAdditionalServlets.

private void addBrokerAdditionalServlets(WebService webService, Map<String, Object> attributeMap, ServiceConfiguration config) {
    if (this.getBrokerAdditionalServlets() != null) {
        Collection<AdditionalServletWithClassLoader> additionalServletCollection = this.getBrokerAdditionalServlets().getServlets().values();
        for (AdditionalServletWithClassLoader servletWithClassLoader : additionalServletCollection) {
            servletWithClassLoader.loadConfig(config);
            AdditionalServlet additionalServlet = servletWithClassLoader.getServlet();
            if (additionalServlet instanceof AdditionalServletWithPulsarService) {
                ((AdditionalServletWithPulsarService) additionalServlet).setPulsarService(this);
            }
            webService.addServlet(servletWithClassLoader.getBasePath(), servletWithClassLoader.getServletHolder(), config.isAuthenticationEnabled(), attributeMap);
            LOG.info("Broker add additional servlet basePath {} ", servletWithClassLoader.getBasePath());
        }
    }
}

Example 2 with AdditionalServletWithPulsarService

use of org.apache.pulsar.broker.web.plugin.servlet.AdditionalServletWithPulsarService in project pulsar by yahoo.

the class BrokerAdditionalServletTest method mockAdditionalServlet.

private void mockAdditionalServlet(PulsarService pulsar) {
    Servlet servlet = new OrdinaryServlet();
    AdditionalServlet brokerAdditionalServlet = Mockito.mock(AdditionalServlet.class);
    Mockito.when(brokerAdditionalServlet.getBasePath()).thenReturn(BASE_PATH);
    Mockito.when(brokerAdditionalServlet.getServletHolder()).thenReturn(new ServletHolder(servlet));
    AdditionalServletWithPulsarService brokerAdditionalServletWithPulsarService = new AdditionalServletWithPulsarService() {

        private PulsarService pulsarService;

        @Override
        public void setPulsarService(PulsarService pulsarService) {
            this.pulsarService = pulsarService;
        }

        @Override
        public void loadConfig(PulsarConfiguration pulsarConfiguration) {
        // No-op
        }

        @Override
        public String getBasePath() {
            return WITH_PULSAR_SERVICE_BASE_PATH;
        }

        @Override
        public ServletHolder getServletHolder() {
            return new ServletHolder(new WithPulsarServiceServlet(pulsarService));
        }

        @Override
        public void close() {
        // No-op
        }
    };
    AdditionalServlets brokerAdditionalServlets = Mockito.mock(AdditionalServlets.class);
    Map<String, AdditionalServletWithClassLoader> map = new HashMap<>();
    map.put("broker-additional-servlet", new AdditionalServletWithClassLoader(brokerAdditionalServlet, null));
    map.put("broker-additional-servlet-with-pulsar-service", new AdditionalServletWithClassLoader(brokerAdditionalServletWithPulsarService, null));
    Mockito.when(brokerAdditionalServlets.getServlets()).thenReturn(map);
    Mockito.when(pulsar.getBrokerAdditionalServlets()).thenReturn(brokerAdditionalServlets);
}

Example 3 with AdditionalServletWithPulsarService

use of org.apache.pulsar.broker.web.plugin.servlet.AdditionalServletWithPulsarService in project incubator-pulsar by apache.

the class PulsarService method addBrokerAdditionalServlets.

private void addBrokerAdditionalServlets(WebService webService, Map<String, Object> attributeMap, ServiceConfiguration config) {
    if (this.getBrokerAdditionalServlets() != null) {
        Collection<AdditionalServletWithClassLoader> additionalServletCollection = this.getBrokerAdditionalServlets().getServlets().values();
        for (AdditionalServletWithClassLoader servletWithClassLoader : additionalServletCollection) {
            servletWithClassLoader.loadConfig(config);
            AdditionalServlet additionalServlet = servletWithClassLoader.getServlet();
            if (additionalServlet instanceof AdditionalServletWithPulsarService) {
                ((AdditionalServletWithPulsarService) additionalServlet).setPulsarService(this);
            }
            webService.addServlet(servletWithClassLoader.getBasePath(), servletWithClassLoader.getServletHolder(), config.isAuthenticationEnabled(), attributeMap);
            LOG.info("Broker add additional servlet basePath {} ", servletWithClassLoader.getBasePath());
        }
    }
}

Example 4 with AdditionalServletWithPulsarService

use of org.apache.pulsar.broker.web.plugin.servlet.AdditionalServletWithPulsarService in project incubator-pulsar by apache.

the class BrokerAdditionalServletTest method mockAdditionalServlet.

private void mockAdditionalServlet(PulsarService pulsar) {
    Servlet servlet = new OrdinaryServlet();
    AdditionalServlet brokerAdditionalServlet = Mockito.mock(AdditionalServlet.class);
    Mockito.when(brokerAdditionalServlet.getBasePath()).thenReturn(BASE_PATH);
    Mockito.when(brokerAdditionalServlet.getServletHolder()).thenReturn(new ServletHolder(servlet));
    AdditionalServletWithPulsarService brokerAdditionalServletWithPulsarService = new AdditionalServletWithPulsarService() {

        private PulsarService pulsarService;

        @Override
        public void setPulsarService(PulsarService pulsarService) {
            this.pulsarService = pulsarService;
        }

        @Override
        public void loadConfig(PulsarConfiguration pulsarConfiguration) {
        // No-op
        }

        @Override
        public String getBasePath() {
            return WITH_PULSAR_SERVICE_BASE_PATH;
        }

        @Override
        public ServletHolder getServletHolder() {
            return new ServletHolder(new WithPulsarServiceServlet(pulsarService));
        }

        @Override
        public void close() {
        // No-op
        }
    };
    AdditionalServlets brokerAdditionalServlets = Mockito.mock(AdditionalServlets.class);
    Map<String, AdditionalServletWithClassLoader> map = new HashMap<>();
    map.put("broker-additional-servlet", new AdditionalServletWithClassLoader(brokerAdditionalServlet, null));
    map.put("broker-additional-servlet-with-pulsar-service", new AdditionalServletWithClassLoader(brokerAdditionalServletWithPulsarService, null));
    Mockito.when(brokerAdditionalServlets.getServlets()).thenReturn(map);
    Mockito.when(pulsar.getBrokerAdditionalServlets()).thenReturn(brokerAdditionalServlets);
}

Example 5 with AdditionalServletWithPulsarService

use of org.apache.pulsar.broker.web.plugin.servlet.AdditionalServletWithPulsarService in project pulsar by apache.

the class PulsarService method addBrokerAdditionalServlets.

private void addBrokerAdditionalServlets(WebService webService, Map<String, Object> attributeMap, ServiceConfiguration config) {
    if (this.getBrokerAdditionalServlets() != null) {
        Collection<AdditionalServletWithClassLoader> additionalServletCollection = this.getBrokerAdditionalServlets().getServlets().values();
        for (AdditionalServletWithClassLoader servletWithClassLoader : additionalServletCollection) {
            servletWithClassLoader.loadConfig(config);
            AdditionalServlet additionalServlet = servletWithClassLoader.getServlet();
            if (additionalServlet instanceof AdditionalServletWithPulsarService) {
                ((AdditionalServletWithPulsarService) additionalServlet).setPulsarService(this);
            }
            webService.addServlet(servletWithClassLoader.getBasePath(), servletWithClassLoader.getServletHolder(), config.isAuthenticationEnabled(), attributeMap);
            LOG.info("Broker add additional servlet basePath {} ", servletWithClassLoader.getBasePath());
        }
    }
}