9aa4e4c9b57d681609fa0c8231b02aa120e6a4d3
[so.git] / common / src / main / java / org / onap / so / logging / spring / interceptor / LoggingInterceptor.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP - SO
4  * ================================================================================
5  * Copyright (C) 2017 - 2018 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  * 
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  * 
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.onap.so.logging.spring.interceptor;
22
23 import java.util.Collections;
24 import java.util.Map;
25 import java.util.UUID;
26 import java.util.stream.Collectors;
27 import javax.servlet.http.HttpServletRequest;
28 import javax.servlet.http.HttpServletResponse;
29 import javax.ws.rs.core.Context;
30 import javax.ws.rs.core.Response;
31 import javax.ws.rs.ext.Providers;
32 import org.onap.logging.ref.slf4j.ONAPLogConstants;
33 import org.onap.so.logger.LogConstants;
34 import org.onap.so.logging.jaxrs.filter.MDCSetup;
35 import org.slf4j.Logger;
36 import org.slf4j.LoggerFactory;
37 import org.slf4j.MDC;
38 import org.springframework.beans.factory.annotation.Autowired;
39 import org.springframework.stereotype.Component;
40 import org.springframework.web.servlet.ModelAndView;
41 import org.springframework.web.servlet.handler.HandlerInterceptorAdapter;
42
43 @Component
44 public class LoggingInterceptor extends HandlerInterceptorAdapter {
45
46     Logger logger = LoggerFactory.getLogger(LoggingInterceptor.class);
47
48     @Autowired
49     private MDCSetup mdcSetup;
50     
51     @Context 
52     private Providers providers;
53
54     @Override
55     public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler)
56             throws Exception {
57         
58         Map<String, String> headers = Collections.list((request).getHeaderNames())
59                 .stream()
60                 .collect(Collectors.toMap(h -> h, request::getHeader));
61         setRequestId(headers);
62         setInvocationId(headers);
63         setServiceName(request);
64         setMDCPartnerName(headers);
65         mdcSetup.setClientIPAddress(request);
66         mdcSetup.setEntryTimeStamp();
67         mdcSetup.setInstanceUUID();
68         mdcSetup.setServerFQDN();
69         MDC.put(ONAPLogConstants.MDCs.RESPONSE_STATUS_CODE, ONAPLogConstants.ResponseStatus.INPROGRESS.toString());
70         logger.info(ONAPLogConstants.Markers.ENTRY, "Entering");
71         if (logger.isDebugEnabled()) 
72                 logRequestInformation(request);
73         return true;
74     }
75     
76     protected void logRequestInformation(HttpServletRequest request) {
77         Map<String, String> headers = Collections.list((request).getHeaderNames())
78                     .stream()
79                     .collect(Collectors.toMap(h -> h, request::getHeader));
80
81         logger.debug("===========================request begin================================================");
82         logger.debug("URI         : {}", request.getRequestURI());
83         logger.debug("Method      : {}", request.getMethod());
84         logger.debug("Headers     : {}", headers);
85         logger.debug("==========================request end================================================");
86                 
87         }
88
89         @Override
90     public void postHandle(
91             HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView)
92             throws Exception {
93         setResponseStatusCode(response);
94         MDC.put(ONAPLogConstants.MDCs.RESPONSE_DESCRIPTION,"");      
95         MDC.put(ONAPLogConstants.MDCs.RESPONSE_CODE,String.valueOf(response.getStatus()));
96         logger.info(ONAPLogConstants.Markers.EXIT, "Exiting.");
97         MDC.clear();
98     }
99
100         protected void setResponseStatusCode(HttpServletResponse response) {
101         String statusCode;
102         if(Response.Status.Family.familyOf(response.getStatus()).equals(Response.Status.Family.SUCCESSFUL)){     
103             statusCode=ONAPLogConstants.ResponseStatus.COMPLETED.toString();
104         }else{                          
105             statusCode= ONAPLogConstants.ResponseStatus.ERROR.toString();               
106         }           
107         MDC.put(ONAPLogConstants.MDCs.RESPONSE_STATUS_CODE, statusCode);
108     }
109
110         protected void setServiceName(HttpServletRequest request) {
111         MDC.put(ONAPLogConstants.MDCs.SERVICE_NAME, request.getRequestURI());
112     }
113         
114         protected void setRequestId(Map<String, String> headers) {
115         String requestId=headers.get(ONAPLogConstants.Headers.REQUEST_ID.toLowerCase());      
116         if(requestId == null || requestId.isEmpty())
117                 requestId = UUID.randomUUID().toString();    
118         MDC.put(ONAPLogConstants.MDCs.REQUEST_ID,requestId);
119     }
120
121         protected void setInvocationId(Map<String, String> headers) {
122         String invocationId = headers.get(ONAPLogConstants.Headers.INVOCATION_ID.toLowerCase());
123         if(invocationId == null || invocationId.isEmpty())
124             invocationId =UUID.randomUUID().toString();
125         MDC.put(ONAPLogConstants.MDCs.INVOCATION_ID, invocationId);
126     }
127
128         protected void setMDCPartnerName(Map<String, String> headers) {
129         String partnerName=headers.get(ONAPLogConstants.Headers.PARTNER_NAME.toLowerCase());
130         if(partnerName == null || partnerName.isEmpty())
131             partnerName = "";
132         MDC.put(ONAPLogConstants.MDCs.PARTNER_NAME,partnerName);
133     }
134     
135
136 }