aboutsummaryrefslogtreecommitdiffstats
path: root/ms/blueprintsprocessor/modules/commons/message-lib/src/main/kotlin/org/onap/ccsdk/cds/blueprintsprocessor/message/service/MessageLoggerService.kt
blob: 7e5f35d157dfbb64be51705a9a7aa717c7831636 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
/*
 * Copyright © 2018-2019 AT&T Intellectual Property.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.onap.ccsdk.cds.blueprintsprocessor.message.service

import org.apache.kafka.clients.consumer.ConsumerRecord
import org.apache.kafka.common.header.Headers
import org.onap.ccsdk.cds.blueprintsprocessor.core.api.data.CommonHeader
import org.onap.ccsdk.cds.blueprintsprocessor.message.addHeader
import org.onap.ccsdk.cds.blueprintsprocessor.message.toMap
import org.onap.ccsdk.cds.controllerblueprints.core.BlueprintConstants
import org.onap.ccsdk.cds.controllerblueprints.core.defaultToEmpty
import org.onap.ccsdk.cds.controllerblueprints.core.defaultToUUID
import org.onap.ccsdk.cds.controllerblueprints.core.logger
import org.slf4j.MDC
import java.net.InetAddress
import java.time.Instant
import java.time.ZoneOffset
import java.time.ZonedDateTime
import java.time.format.DateTimeFormatter
import java.util.UUID

class MessageLoggerService {

    private val log = logger(MessageLoggerService::class)

    fun messageConsuming(headers: CommonHeader, consumerRecord: ConsumerRecord<*, *>) {
        messageConsuming(
            headers.requestId, headers.subRequestId,
            headers.originatorId, consumerRecord
        )
    }

    fun messageConsuming(consumerRecord: ConsumerRecord<*, *>) {
        val headers = consumerRecord.headers().toMap()
        val requestID = headers[BlueprintConstants.ONAP_REQUEST_ID].defaultToUUID()
        val invocationID = headers[BlueprintConstants.ONAP_INVOCATION_ID].defaultToUUID()
        val partnerName = headers[BlueprintConstants.ONAP_PARTNER_NAME] ?: "UNKNOWN"
        messageConsuming(requestID, invocationID, partnerName, consumerRecord)
    }

    fun messageConsuming(
        requestID: String,
        invocationID: String,
        partnerName: String,
        consumerRecord: ConsumerRecord<*, *>
    ) {
        val headers = consumerRecord.headers().toMap()
        val localhost = InetAddress.getLocalHost()
        MDC.put(
            "InvokeTimestamp",
            ZonedDateTime
                .ofInstant(Instant.ofEpochMilli(consumerRecord.timestamp()), ZoneOffset.UTC)
                .format(DateTimeFormatter.ISO_INSTANT)
        )
        MDC.put("RequestID", requestID)
        MDC.put("InvocationID", invocationID)
        MDC.put("PartnerName", partnerName)
        MDC.put("ClientIPAddress", headers["ClientIPAddress"].defaultToEmpty())
        MDC.put("ServerFQDN", localhost.hostName.defaultToEmpty())
        MDC.put("ServiceName", consumerRecord.topic())
        // Custom MDC for Message Consumers
        MDC.put("Offset", consumerRecord.offset().toString())
        MDC.put("MessageKey", consumerRecord.key()?.toString().defaultToEmpty())
        log.info("Consuming MDC Properties : ${MDC.getCopyOfContextMap()}")
    }

    /** Used before producing message request, Inbound Invocation ID is used as request Id
     * for produced message Request, If invocation Id is missing then default Request Id will be generated.
     */
    fun messageProducing(requestHeader: Headers) {
        val localhost = InetAddress.getLocalHost()
        requestHeader.addHeader(BlueprintConstants.ONAP_REQUEST_ID, MDC.get("InvocationID").defaultToUUID())
        requestHeader.addHeader(BlueprintConstants.ONAP_INVOCATION_ID, UUID.randomUUID().toString())
        requestHeader.addHeader(BlueprintConstants.ONAP_PARTNER_NAME, BlueprintConstants.APP_NAME)
        requestHeader.addHeader("ClientIPAddress", localhost.hostAddress)
    }

    fun messageConsumingExisting() {
        MDC.clear()
    }
}