com.logviewer.springboot.LogViewerSpringBootConfig Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of log-viewer-spring-boot Show documentation

Classes for integration with spring-boot

There is a newer version: 1.0.11

Show newest version

package com.logviewer.springboot;

import com.logviewer.config.LogViewerAutoConfig;
import com.logviewer.config.LvConfigBase;
import com.logviewer.web.LogViewerServlet;
import org.springframework.boot.web.servlet.ServletRegistrationBean;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Import;
import org.springframework.core.env.Environment;

import java.util.Collections;

@Import({LogViewerAutoConfig.class, LvConfigBase.class})
@Configuration
public class LogViewerSpringBootConfig {

    public static final String LOG_VIEWER_WEBSOCKET_PATH = "log-viewer.websocket.path";

    public static final String DEFAULT_LOG_PATH = "/logs/*";

    public static final String LOG_VIEWER_URL_MAPPING = "log-viewer.url-mapping";

    @Bean
    public ServletRegistrationBean logViewerServlet(Environment environment) {
        ServletRegistrationBean servlet = new ServletRegistrationBean<>();
        servlet.setName("logViewerServlet");
        servlet.setAsyncSupported(true);
        servlet.setServlet(new LogViewerServlet());

        String logServletPath = environment.getProperty(LOG_VIEWER_URL_MAPPING, DEFAULT_LOG_PATH);
        if (!logServletPath.endsWith("*")) {
            if (!logServletPath.endsWith("/"))
                logServletPath += "/";

            logServletPath += "*";
        }

        servlet.setUrlMappings(Collections.singletonList(logServletPath));

        String websocketPath = environment.getProperty(LOG_VIEWER_WEBSOCKET_PATH);
        if (websocketPath != null && !websocketPath.isEmpty()) {
            servlet.addInitParameter("web-socket-path", websocketPath);
        }

        return servlet;
    }
}