/*
    * Licensed under the GPL License. You may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *   https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
    *
    * THIS PACKAGE IS PROVIDED "AS IS" AND WITHOUT ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING,
    * WITHOUT LIMITATION, THE IMPLIED WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR
    * PURPOSE.
   */
  package psiprobe.beans.stats.collectors;
  
  import jakarta.inject.Inject;
  
  import javax.naming.NamingException;
  
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  import org.springframework.beans.factory.annotation.Value;
  
  import psiprobe.beans.ContainerWrapperBean;
  import psiprobe.model.ApplicationResource;
  import psiprobe.model.DataSourceInfo;
  import psiprobe.tools.TimeExpression;
  
  /**
   * The Class DatasourceStatsCollectorBean.
   */
  public class DatasourceStatsCollectorBean extends AbstractStatsCollectorBean {
  
    /** The Constant PREFIX_ESTABLISHED. */
    private static final String PREFIX_ESTABLISHED = "ds.est.";
  
    /** The Constant PREFIX_BUSY. */
    private static final String PREFIX_BUSY = "ds.busy.";
  
    /** The Constant logger. */
    private static final Logger logger = LoggerFactory.getLogger(DatasourceStatsCollectorBean.class);
  
    /** The container wrapper. */
    @Inject
    private ContainerWrapperBean containerWrapper;
  
    /**
     * Gets the container wrapper.
     *
     * @return the container wrapper
     */
    public ContainerWrapperBean getContainerWrapper() {
      return containerWrapper;
    }
  
    /**
     * Sets the container wrapper.
     *
     * @param containerWrapper the new container wrapper
     */
    public void setContainerWrapper(ContainerWrapperBean containerWrapper) {
      this.containerWrapper = containerWrapper;
    }
  
    @Override
    public void collect() throws NamingException, InterruptedException {
      long currentTime = System.currentTimeMillis();
      if (containerWrapper == null) {
        logger.error("Cannot collect data source stats. Container wrapper is not set.");
      } else {
        for (ApplicationResource ds : getContainerWrapper().getDataSources()) {
          String appName = ds.getApplicationName();
          String name = (appName == null ? "" : appName) + '/' + ds.getName();
          DataSourceInfo dsi = ds.getDataSourceInfo();
          int numEstablished = dsi.getEstablishedConnections();
          int numBusy = dsi.getBusyConnections();
          logger.trace("Collecting stats for datasource: {}", name);
          buildAbsoluteStats(PREFIX_ESTABLISHED + name, numEstablished, currentTime);
          buildAbsoluteStats(PREFIX_BUSY + name, numBusy, currentTime);
        }
        logger.debug("datasource stats collected in {}ms", System.currentTimeMillis() - currentTime);
      }
    }
  
    /**
     * Reset.
     *
     * @throws NamingException the naming exception
     */
    public void reset() throws NamingException {
      if (containerWrapper == null) {
        logger.error("Cannot reset application stats. Container wrapper is not set.");
      } else {
        for (ApplicationResource ds : getContainerWrapper().getDataSources()) {
          reset(ds.getName());
        }
      }
    }
  
    /**
     * Reset.
     *
    * @param name the name
    */
   public void reset(String name) {
     resetStats(PREFIX_ESTABLISHED + name);
     resetStats(PREFIX_BUSY + name);
   }
 
   /**
    * Sets the max series expression.
    *
    * @param period the period
    * @param span the span
    */
   public void setMaxSeries(
       @Value("${psiprobe.beans.stats.collectors.datasource.period}") long period,
       @Value("${psiprobe.beans.stats.collectors.datasource.span}") long span) {
     super.setMaxSeries((int) TimeExpression.dataPoints(period, span));
   }
 
 }