/*
 * ============LICENSE_START=======================================================
 * VES Collector
 * ================================================================================
 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
 * Copyright (C) 2018,2020 Nokia. All rights reserved.
 * ================================================================================
 * 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.
 * ============LICENSE_END=========================================================
 */
package org.onap.dcae.common;

import com.att.nsa.clock.SaClock;
import com.att.nsa.logging.LoggingContext;
import com.att.nsa.logging.log4j.EcompFields;
import io.vavr.collection.Map;
import org.onap.dcae.common.model.VesEvent;
import org.onap.dcae.common.publishing.DMaaPEventPublisher;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.util.List;

public class EventSender {

  private static final Logger metriclog = LoggerFactory.getLogger("com.att.ecomp.metrics");
  private Map<String, String[]> streamIdToDmaapIds;
  private DMaaPEventPublisher eventPublisher;
  private static final Logger log = LoggerFactory.getLogger(EventSender.class);

  public EventSender(DMaaPEventPublisher eventPublisher, Map<String, String[]> streamIdToDmaapIds) {
    this.eventPublisher = eventPublisher;
    this.streamIdToDmaapIds = streamIdToDmaapIds;
  }

  public void send(List<VesEvent> vesEvents) {
    for (VesEvent vesEvent : vesEvents) {
      metriclog.info("EVENT_PUBLISH_START");
      setLoggingContext(vesEvent);
      streamIdToDmaapIds.get(vesEvent.getStreamId())
          .onEmpty(() -> log.error("No StreamID defined for publish - Message dropped" + vesEvent.asJsonObject()))
          .forEach(dmaapIds -> sendEventsToStreams(vesEvent, dmaapIds));
      log.debug("Message published" + vesEvent.asJsonObject());
    }
    log.debug("CommonStartup.handleEvents:EVENTS has been published successfully!");
    metriclog.info("EVENT_PUBLISH_END");
  }

  private void sendEventsToStreams(VesEvent vesEvent, String[] dmaapIds) {
    for (String dmaapId : dmaapIds) {
      log.info("Invoking publisher for streamId/domain:" + dmaapId);
      eventPublisher.sendEvent(vesEvent, dmaapId);
    }
  }

  private void setLoggingContext(VesEvent vesEvent) {
    LoggingContext localLC = VESLogger.getLoggingContextForThread(vesEvent.getUniqueId().toString());
    localLC.put(EcompFields.kBeginTimestampMs, SaClock.now());
    log.debug("event.VESuniqueId" + vesEvent.getUniqueId() + "event.commonEventHeader.domain:" + vesEvent.getDomain());
  }
}