2 * ============LICENSE_START=======================================================
3 * Copyright (C) 2023 Nordix Foundation
4 * ================================================================================
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
9 * http://www.apache.org/licenses/LICENSE-2.0
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.
17 * SPDX-License-Identifier: Apache-2.0
18 * ============LICENSE_END=========================================================
21 package org.onap.cps.ncmp.api.impl.trustlevel;
23 import static org.onap.cps.ncmp.api.impl.events.mapper.CloudEventMapper.toTargetEvent;
25 import com.hazelcast.collection.ISet;
26 import io.cloudevents.CloudEvent;
27 import io.cloudevents.kafka.impl.KafkaHeaders;
28 import lombok.RequiredArgsConstructor;
29 import lombok.extern.slf4j.Slf4j;
30 import org.apache.kafka.clients.consumer.ConsumerRecord;
31 import org.springframework.kafka.annotation.KafkaListener;
32 import org.springframework.stereotype.Component;
36 @RequiredArgsConstructor
37 public class DeviceHeartbeatConsumer {
39 private final ISet<String> untrustworthyCmHandlesSet;
42 * Listening the device heartbeats.
44 * @param deviceHeartbeatConsumerRecord Device Heartbeat record.
46 @KafkaListener(topics = "${app.dmi.device-heartbeat.topic}",
47 containerFactory = "cloudEventConcurrentKafkaListenerContainerFactory")
48 public void heartbeatListener(final ConsumerRecord<String, CloudEvent> deviceHeartbeatConsumerRecord) {
50 final String cmHandleId = KafkaHeaders.getParsedKafkaHeader(deviceHeartbeatConsumerRecord.headers(), "ce_id");
52 final DeviceTrustLevel deviceTrustLevel =
53 toTargetEvent(deviceHeartbeatConsumerRecord.value(), DeviceTrustLevel.class);
55 if (deviceTrustLevel == null || deviceTrustLevel.getTrustLevel() == null) {
56 log.warn("No or Invalid trust level defined");
60 if (deviceTrustLevel.getTrustLevel().equals(TrustLevel.NONE)) {
61 untrustworthyCmHandlesSet.add(cmHandleId);
62 log.debug("Added cmHandleId to untrustworthy set : {}", cmHandleId);
63 } else if (deviceTrustLevel.getTrustLevel().equals(TrustLevel.COMPLETE) && untrustworthyCmHandlesSet.contains(
65 untrustworthyCmHandlesSet.remove(cmHandleId);
66 log.debug("Removed cmHandleId from untrustworthy set : {}", cmHandleId);