View Javadoc
1   /*
2    * Licensed to the Apache Software Foundation (ASF) under one or more
3    * contributor license agreements. See the NOTICE file distributed with
4    * this work for additional information regarding copyright ownership.
5    * The ASF licenses this file to You under the Apache license, Version 2.0
6    * (the "License"); you may not use this file except in compliance with
7    * the License. You may obtain a copy of the License at
8    *
9    *      http://www.apache.org/licenses/LICENSE-2.0
10   *
11   * Unless required by applicable law or agreed to in writing, software
12   * distributed under the License is distributed on an "AS IS" BASIS,
13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14   * See the license for the specific language governing permissions and
15   * limitations under the license.
16   */
17  package org.apache.logging.log4j.audit.service.config;
18  
19  import org.apache.logging.log4j.catalog.api.util.ProfileUtil;
20  import org.springframework.web.WebApplicationInitializer;
21  import org.springframework.web.context.ContextLoaderListener;
22  import org.springframework.web.context.support.AnnotationConfigWebApplicationContext;
23  import org.springframework.web.servlet.DispatcherServlet;
24  
25  import javax.servlet.ServletContext;
26  import javax.servlet.ServletException;
27  import javax.servlet.ServletRegistration;
28  
29  public class WebAppInitializer implements WebApplicationInitializer {
30      private static final String APPLICATION_NAME = "AuditService";
31  
32      @Override
33      public void onStartup(ServletContext servletContext) throws ServletException {
34          servletContext.setInitParameter("applicationName", APPLICATION_NAME);
35          System.setProperty("applicationName", APPLICATION_NAME);
36          ProfileUtil.setActiveProfile(servletContext);
37          AnnotationConfigWebApplicationContext rootContext = new AnnotationConfigWebApplicationContext();
38          rootContext.setDisplayName(APPLICATION_NAME);
39          rootContext.register(ApplicationConfiguration.class);
40          servletContext.addListener(new ContextLoaderListener(rootContext));
41  
42          // MVC Context
43          AnnotationConfigWebApplicationContext mvcContext = new AnnotationConfigWebApplicationContext();
44          mvcContext.register(WebMvcAppContext.class);
45  
46          ServletRegistration.Dynamic restServlet = servletContext.addServlet("restServlet", new DispatcherServlet(mvcContext));
47          restServlet.setLoadOnStartup(1);
48          restServlet.addMapping("/*");
49      }
50  
51  
52  
53  }