Enhancements for the aai-common library
[aai/aai-common.git] / aai-els-onap-logging / src / main / java / org / onap / logging / filter / base / AuditLogServletFilter.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP - Logging
4  * ================================================================================
5  * Copyright (C) 2019 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Modifications Copyright (C) 2018 IBM.
8  * ================================================================================
9  * Licensed under the Apache License, Version 2.0 (the "License");
10  * you may not use this file except in compliance with the License.
11  * You may obtain a copy of the License at
12  * 
13  *      http://www.apache.org/licenses/LICENSE-2.0
14  * 
15  * Unless required by applicable law or agreed to in writing, software
16  * distributed under the License is distributed on an "AS IS" BASIS,
17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
18  * See the License for the specific language governing permissions and
19  * limitations under the License.
20  * ============LICENSE_END=========================================================
21  */
22
23 package org.onap.logging.filter.base;
24
25 import java.io.IOException;
26 import javax.servlet.Filter;
27 import javax.servlet.FilterChain;
28 import javax.servlet.FilterConfig;
29 import javax.servlet.ServletException;
30 import javax.servlet.ServletRequest;
31 import javax.servlet.ServletResponse;
32 import javax.servlet.http.HttpServletRequest;
33 import javax.servlet.http.HttpServletResponse;
34 import org.onap.logging.ref.slf4j.ONAPLogConstants;
35 import org.slf4j.MDC;
36
37 public class AuditLogServletFilter extends AbstractAuditLogFilter<HttpServletRequest, HttpServletResponse>
38         implements Filter {
39
40     @Override
41     public void destroy() {
42         // this method does nothing
43     }
44
45     @Override
46     public void doFilter(final ServletRequest request, final ServletResponse response, final FilterChain filterChain)
47             throws IOException, ServletException {
48         try {
49             if (request != null && request instanceof HttpServletRequest) {
50                 pre((HttpServletRequest) request);
51             }
52             filterChain.doFilter(request, response);
53         } finally {
54             if (request != null && request instanceof HttpServletRequest) {
55                 post((HttpServletRequest) request, (HttpServletResponse) response);
56             }
57             MDC.clear();
58         }
59     }
60
61     @Override
62     public void init(FilterConfig filterConfig) throws ServletException {
63         // this method does nothing
64     }
65
66     protected void pre(HttpServletRequest request) {
67         SimpleMap headers = new SimpleServletHeadersMap(request);
68         pre(headers, request, request);
69     }
70
71     @Override
72     protected void setServiceName(HttpServletRequest request) {
73         MDC.put(ONAPLogConstants.MDCs.SERVICE_NAME, request.getRequestURI());
74     }
75
76     private void post(HttpServletRequest request, HttpServletResponse response) {
77         post(response);
78     }
79
80     @Override
81     protected int getResponseCode(HttpServletResponse response) {
82         return response.getStatus();
83     }
84
85 }