Enhancements for the aai-common library
[aai/aai-common.git] / aai-els-onap-logging / src / main / java / org / onap / logging / filter / base / AbstractMetricLogFilter.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * ONAP - Logging
4  * ================================================================================
5  * Copyright (C) 2019 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.logging.filter.base;
22
23 import java.time.ZoneOffset;
24 import java.time.ZonedDateTime;
25 import java.time.format.DateTimeFormatter;
26 import java.util.UUID;
27 import org.onap.logging.ref.slf4j.ONAPLogConstants;
28 import org.slf4j.Logger;
29 import org.slf4j.LoggerFactory;
30 import org.slf4j.MDC;
31 import org.slf4j.Marker;
32 import org.slf4j.MarkerFactory;
33
34 public abstract class AbstractMetricLogFilter<Request, Response, RequestHeaders> extends MDCSetup {
35     protected static final Logger logger = LoggerFactory.getLogger(AbstractMetricLogFilter.class);
36     private final String partnerName;
37     private static final Marker INVOKE_RETURN = MarkerFactory.getMarker("INVOKE-RETURN");
38
39     public AbstractMetricLogFilter() {
40         partnerName = getPartnerName();
41     }
42
43     protected abstract void addHeader(RequestHeaders requestHeaders, String headerName, String headerValue);
44
45     protected abstract String getTargetServiceName(Request request);
46
47     protected abstract String getServiceName(Request request);
48
49     protected abstract int getHttpStatusCode(Response response);
50
51     protected abstract String getResponseCode(Response response);
52
53     protected abstract String getTargetEntity(Request request);
54
55     protected void pre(Request request, RequestHeaders requestHeaders) {
56         try {
57             setupMDC(request);
58             setupHeaders(request, requestHeaders);
59             logger.info(ONAPLogConstants.Markers.INVOKE, "Invoke");
60         } catch (Exception e) {
61             logger.warn("Error in AbstractMetricLogFilter pre", e);
62         }
63     }
64
65     protected void setupHeaders(Request clientRequest, RequestHeaders requestHeaders) {
66         String requestId = extractRequestID();
67         addHeader(requestHeaders, ONAPLogConstants.Headers.REQUEST_ID, requestId);
68         addHeader(requestHeaders, Constants.HttpHeaders.HEADER_REQUEST_ID, requestId);
69         addHeader(requestHeaders, Constants.HttpHeaders.TRANSACTION_ID, requestId);
70         addHeader(requestHeaders, Constants.HttpHeaders.ECOMP_REQUEST_ID, requestId);
71         addHeader(requestHeaders, ONAPLogConstants.Headers.INVOCATION_ID, MDC.get(ONAPLogConstants.MDCs.INVOCATION_ID));
72         addHeader(requestHeaders, ONAPLogConstants.Headers.PARTNER_NAME, partnerName);
73     }
74
75     protected void setupMDC(Request request) {
76         MDC.put(ONAPLogConstants.MDCs.INVOKE_TIMESTAMP,
77                 ZonedDateTime.now(ZoneOffset.UTC).format(DateTimeFormatter.ISO_INSTANT));
78         // setup time stamp defaults
79         setLogTimestamp();
80         setElapsedTimeInvokeTimestamp();
81         MDC.put(ONAPLogConstants.MDCs.TARGET_SERVICE_NAME, getTargetServiceName(request));
82         MDC.put(ONAPLogConstants.MDCs.RESPONSE_STATUS_CODE, ONAPLogConstants.ResponseStatus.INPROGRESS.toString());
83         setInvocationIdFromMDC();
84
85         if (MDC.get(ONAPLogConstants.MDCs.TARGET_ENTITY) == null) {
86             String targetEntity = getTargetEntity(request);
87             if (targetEntity != null) {
88                 MDC.put(ONAPLogConstants.MDCs.TARGET_ENTITY, targetEntity);
89             } else {
90                 MDC.put(ONAPLogConstants.MDCs.TARGET_ENTITY, Constants.DefaultValues.UNKNOWN_TARGET_ENTITY);
91             }
92         }
93
94         if (MDC.get(ONAPLogConstants.MDCs.SERVICE_NAME) == null) {
95             MDC.put(ONAPLogConstants.MDCs.SERVICE_NAME, getServiceName(request));
96         }
97         setServerFQDN();
98     }
99
100     protected String extractRequestID() {
101         String requestId = MDC.get(ONAPLogConstants.MDCs.REQUEST_ID);
102         if (requestId == null || requestId.isEmpty()) {
103             requestId = UUID.randomUUID().toString();
104             setLogTimestamp();
105             setElapsedTimeInvokeTimestamp();
106             logger.warn("No value found in MDC when checking key {} value will be set to {}",
107                     ONAPLogConstants.MDCs.REQUEST_ID, requestId);
108             MDC.put(ONAPLogConstants.MDCs.REQUEST_ID, requestId);
109         }
110         return requestId;
111     }
112
113     protected void post(Request request, Response response) {
114         try {
115             setLogTimestamp();
116             setElapsedTimeInvokeTimestamp();
117             setResponseStatusCode(getHttpStatusCode(response));
118             setResponseDescription(getHttpStatusCode(response));
119             MDC.put(ONAPLogConstants.MDCs.RESPONSE_CODE, getResponseCode(response));
120             logger.info(INVOKE_RETURN, "InvokeReturn");
121             clearClientMDCs();
122         } catch (Exception e) {
123             logger.warn("Error in AbstractMetricLogFilter post", e);
124         }
125     }
126
127     protected String getPartnerName() {
128         return getProperty(Constants.Property.PARTNER_NAME);
129     }
130
131     protected void logInvoke() {
132         logger.info(ONAPLogConstants.Markers.INVOKE, "Invoke");
133     }
134
135 }