package org.apache.helix.monitoring;

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.
 */

import java.io.IOException;
import java.lang.management.ManagementFactory;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import javax.management.AttributeNotFoundException;
import javax.management.InstanceNotFoundException;
import javax.management.JMException;
import javax.management.MBeanException;
import javax.management.MBeanServer;
import javax.management.MalformedObjectNameException;
import javax.management.ObjectInstance;
import javax.management.ObjectName;
import javax.management.ReflectionException;
import org.apache.helix.controller.stages.BestPossibleStateCalcStage;
import org.apache.helix.controller.stages.IntermediateStateCalcStage;
import org.apache.helix.controller.stages.ReadClusterDataStage;
import org.apache.helix.controller.stages.resource.ResourceMessageDispatchStage;
import org.apache.helix.monitoring.mbeans.ClusterEventMonitor;
import org.apache.helix.monitoring.mbeans.ClusterStatusMonitor;
import org.testng.Assert;
import org.testng.annotations.Test;

public class TestClusterEventStatusMonitor {
  private static final int TEST_SLIDING_WINDOW_MS = 2000; // 2s window for testing

  private class ClusterStatusMonitorForTest extends ClusterStatusMonitor {
    public ClusterStatusMonitorForTest(String clusterName) {
      super(clusterName);
      active();
    }
    public ConcurrentHashMap<String, ClusterEventMonitor> getClusterEventMonitors() {
      return _clusterEventMonitorMap;
    }
  }

  @Test()
  public void test()
      throws InstanceNotFoundException, MalformedObjectNameException, NullPointerException,
      IOException, InterruptedException, MBeanException, AttributeNotFoundException,
      ReflectionException{
    System.out.println("START TestClusterEventStatusMonitor");
    String clusterName = "TestCluster";
    ClusterStatusMonitorForTest monitor = new ClusterStatusMonitorForTest(clusterName);

    MBeanServer _server = ManagementFactory.getPlatformMBeanServer();
    Set<ObjectInstance> mbeans =
        _server.queryMBeans(new ObjectName("ClusterStatus:Cluster=TestCluster,eventName=ClusterEvent,*"), null);
    Assert.assertEquals(mbeans.size(), 0);

    // Customize event monitors for testing
    try {
      this.addTestEventMonitor(monitor, ClusterEventMonitor.PhaseName.Callback.name());
      this.addTestEventMonitor(monitor, ClusterEventMonitor.PhaseName.InQueue.name());
      this.addTestEventMonitor(monitor, BestPossibleStateCalcStage.class.getSimpleName());
      this.addTestEventMonitor(monitor, ReadClusterDataStage.class.getSimpleName());
      this.addTestEventMonitor(monitor, IntermediateStateCalcStage.class.getSimpleName());
      this.addTestEventMonitor(monitor, ResourceMessageDispatchStage.class.getSimpleName());
    } catch (JMException jme) {
      Assert.assertTrue(false, "Failed to customize event monitors");
    }

    int count = 5;
    Long totalDuration = 0L;
    for (int i = 1; i <= count; i++) {
      monitor.updateClusterEventDuration(ClusterEventMonitor.PhaseName.Callback.name(), 100 * i);
      monitor.updateClusterEventDuration(ClusterEventMonitor.PhaseName.InQueue.name(), 100 * i);
      monitor.updateClusterEventDuration(BestPossibleStateCalcStage.class.getSimpleName(), 100 * i);
      monitor.updateClusterEventDuration(ReadClusterDataStage.class.getSimpleName(), 100 * i);
      monitor.updateClusterEventDuration(IntermediateStateCalcStage.class.getSimpleName(), 100 * i);
      monitor.updateClusterEventDuration(ResourceMessageDispatchStage.class.getSimpleName(), 100 * i);
      totalDuration += 100 * i;
    }

    mbeans =
        _server.queryMBeans(
            new ObjectName("ClusterStatus:cluster=TestCluster,eventName=ClusterEvent,*"), null);
    Assert.assertEquals(mbeans.size(), 6);

    for (ObjectInstance mbean : mbeans) {
      Long duration = (Long) _server.getAttribute(mbean.getObjectName(), "TotalDurationCounter");
      Long maxDuration = (Long) _server.getAttribute(mbean.getObjectName(), "MaxSingleDurationGauge");
      Long eventCount = (Long) _server.getAttribute(mbean.getObjectName(), "EventCounter");

      Double pct75th = (Double) _server.getAttribute(mbean.getObjectName(), "DurationGauge.Pct75th");
      Double pct95th = (Double) _server.getAttribute(mbean.getObjectName(), "DurationGauge.Pct95th");
      Double pct99th = (Double) _server.getAttribute(mbean.getObjectName(), "DurationGauge.Pct99th");
      Long max = (Long) _server.getAttribute(mbean.getObjectName(), "DurationGauge.Max");
      Double stddev = (Double) _server.getAttribute(mbean.getObjectName(), "DurationGauge.StdDev");

      Assert.assertEquals(duration, totalDuration);
      Assert.assertEquals(maxDuration, Long.valueOf(100 * count));
      Assert.assertEquals(eventCount, Long.valueOf(count));
      Assert.assertTrue(Math.abs(pct75th - 450.0) < 1);
      Assert.assertTrue(Math.abs(pct95th - 500.0) < 1);
      Assert.assertTrue(Math.abs(pct99th - 500.0) < 1);
      Assert.assertTrue(max == 500);
      Assert.assertTrue(Math.abs(stddev - 158.0) < 0.2);
    }

    System.out.println("\nWaiting for time window to expire\n");
    Thread.sleep(TEST_SLIDING_WINDOW_MS);

    // Since sliding window has expired, just make sure histograms have its values reset
    for (ObjectInstance mbean : mbeans) {
      Double pct75th = (Double) _server.getAttribute(mbean.getObjectName(), "DurationGauge.Pct75th");
      Double pct95th = (Double) _server.getAttribute(mbean.getObjectName(), "DurationGauge.Pct95th");
      Double pct99th = (Double) _server.getAttribute(mbean.getObjectName(), "DurationGauge.Pct99th");
      Long max = (Long) _server.getAttribute(mbean.getObjectName(), "DurationGauge.Max");
      Double stddev = (Double) _server.getAttribute(mbean.getObjectName(), "DurationGauge.StdDev");

      Assert.assertTrue(pct75th == 0.0);
      Assert.assertTrue(pct95th == 0.0);
      Assert.assertTrue(pct99th == 0.0);
      Assert.assertTrue(max == 0);
      Assert.assertTrue(stddev == 0.0);

    }

    monitor.reset();

    mbeans =
        _server.queryMBeans(
            new ObjectName("ClusterStatus:cluster=TestCluster,eventName=ClusterEvent,*"), null);
    Assert.assertEquals(mbeans.size(), 0);

    System.out.println("END TestParticipantMonitor");
  }

  private void addTestEventMonitor(ClusterStatusMonitorForTest monitor, String phaseName) throws
      JMException {
    ConcurrentHashMap<String, ClusterEventMonitor> monitors = monitor.getClusterEventMonitors();
    ClusterEventMonitor eventMonitor = new ClusterEventMonitor(monitor, phaseName,
        TEST_SLIDING_WINDOW_MS);
    eventMonitor.register();
    monitors.put(phaseName, eventMonitor);
  }

}
