/*
 * Copyright (c) 2003, 2018 Oracle and/or its affiliates. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package com.sun.s1asdev.admin.mbeanapi.monitoring.ejb;

import java.io.IOException;

import java.util.Map;
import java.util.Set;
import java.util.Iterator;

import javax.management.j2ee.statistics.Statistic;
import javax.management.j2ee.statistics.Stats;

import com.sun.appserv.management.DomainRoot;
import com.sun.appserv.management.monitor.MonitoringRoot;
import com.sun.appserv.management.monitor.ServerRootMonitor;
import com.sun.appserv.management.monitor.ApplicationMonitor;
import com.sun.appserv.management.monitor.EJBModuleMonitor;
import com.sun.appserv.management.monitor.EJBMonitor;
import com.sun.appserv.management.monitor.BeanPoolMonitor;
import com.sun.appserv.management.monitor.BeanCacheMonitor;
import com.sun.appserv.management.monitor.BeanMethodMonitor;
import com.sun.appserv.management.monitor.MonitoringStats;

import com.sun.appserv.management.util.misc.ExceptionUtil;

import com.sun.appserv.management.util.stringifier.SmartStringifier;
import com.sun.appserv.management.util.stringifier.StringifierRegistryIniterImpl;
import com.sun.appserv.management.util.stringifier.StringifierRegistryImpl;
import com.sun.enterprise.admin.mbeanapi.common.AMXConnector;


/**
 * This test print the statistics for ejb module and stateless bean
 *
 *  EJBMonitoring</B>
 *
 * @author <a href=mailto:satish.viswanatham@sun.com>Satish Viswanatham</a>
 *         Date: Aug 24, 2004
 * @version $Revision: 1.5 $
 */
public class EJBMonitoring {

    private final DomainRoot    mDomainRoot;

    private String SERVER_NAME = "server";

    //private String APP_NAME = "cmpcustomer1";
    //private String APP_NAME = "bmp-simple";
    private String APP_NAME = "stateless-simple";

    private String STATELESS_EJB_NAME = "stateless-simpleEjb_jar";

    public void  testEJBPoolStats()
    {
        MonitoringRoot monitorRoot = mDomainRoot.getMonitoringRoot() ;
        assert(monitorRoot !=null);
        ServerRootMonitor svrRootMtr = (ServerRootMonitor) monitorRoot.
                        getServerRootMonitorMap().  get(SERVER_NAME);
        Map appsMtrMgr = svrRootMtr.getApplicationMonitorMap();

        ApplicationMonitor appMtr =
            (ApplicationMonitor) appsMtrMgr.get(APP_NAME);

        Map ejbMgr = appMtr.getEJBModuleMonitorMap();

        Stats stats = null;
        Statistic[] sts = null;

        System.out.println(" size  is " + ejbMgr.size());
        Iterator itr = ejbMgr.values().iterator();

        while (itr.hasNext()) {
            Object tmp = itr.next();
            System.out.println("Ejb is " + tmp);
            EJBModuleMonitor beanMgr =  (EJBModuleMonitor) tmp;

            Map beans = beanMgr.getEJBMonitorMap();

            Iterator it = beans.values().iterator();

            while (it.hasNext()) {
                EJBMonitor em = (EJBMonitor) it.next();
                listStats(em);

                Map methodMap = em.getBeanMethodMonitorMap();
                itr = methodMap.values().iterator();

                while (itr.hasNext()) {
                    Object o = itr.next();
                    System.out.println(
                        " Looking at stats for this method " + o);
                    BeanMethodMonitor bmm = (BeanMethodMonitor) o;
                    listStats(bmm);

                }
            }

            BeanPoolMonitor bPool = beanMgr.getBeanPoolMonitor(
                                STATELESS_EJB_NAME);
            if (bPool == null) {
                System.out.println("Error: BeanPool monitor is null");
            } else {
                listStats(bPool);
            }

            BeanCacheMonitor bCache = beanMgr.getBeanCacheMonitor(
                                STATELESS_EJB_NAME);
            if (bCache == null) {
                System.out.println("Error: BeanCache monitor is null");
            } else {
                listStats(bCache);
            }

       }


    }

    public void listStats (MonitoringStats ms)
    {
        Stats stats = ms.getStats();
        Statistic[] sts = stats.getStatistics();
        printStats(sts);
    }

    public void printStats(Statistic[] stats)
    {
        if (stats == null)
            return;

        for ( int i=0; i < stats.length; i++)
        {
            printStat(stats[i]);
        }

    }

    public void printStat(Statistic stat)
    {
        if (stat == null)
            return;
        else
            System.out.println(" Stat name is " + stat.getName() +
                " description: " + stat.getDescription() + " start time "
                + stat.getStartTime() + " last sample time "
                + stat.getLastSampleTime() + " unit " + stat.getUnit());
    }

    public EJBMonitoring(final String host,
                                   final int port,
                                   final String adminUser,
                                   final String adminPassword,
                                   final boolean useTLS)
                                    throws IOException
    {
        final AMXConnector ct    =
            new AMXConnector( host, port, adminUser, adminPassword, useTLS );

        mDomainRoot    = ct.getDomainRoot();

    }


    public static void   main( final String[] args )
    {
        new StringifierRegistryIniterImpl( StringifierRegistryImpl.DEFAULT );

        try
        {
            EJBMonitoring ejbMtr = new EJBMonitoring(
            System.getProperty("HOST", "localhost"),
            Integer.parseInt(System.getProperty("PORT","8686")),
            System.getProperty("ADMIN_USER", "admin"),
            System.getProperty("ADMIN_PASSWORD", "adminadmin"),
            Boolean.getBoolean(System.getProperty("USE_TLS", "false")));

           ejbMtr.testEJBPoolStats();
        }
        catch( Throwable t )
        {
            ExceptionUtil.getRootCause( t ).printStackTrace();
        }
    }
}