/*
 * JBoss, the OpenSource J2EE webOS
 *
 * Distributable under LGPL license.
 * See terms of license at gnu.org.
 */
package org.jboss.varia.stats;

import java.io.Serializable;
import java.util.Map;
import java.util.HashMap;
import java.util.Iterator;

/**
 * @author <a href="mailto:alex@jboss.org">Alexey Loubyansky</a>
 * @version <tt>$Revision: 1.2.4.1 $</tt>
 */
public class TxReport
   implements Serializable
{
   static final long serialVersionUID = 1144384523072827681L;
   private static final String DEFAULT_NAME = "UNKNOWN";

   private String name = DEFAULT_NAME;
   private final Map stats = new HashMap();
   private int count = 1;

   public String getName()
   {
      return name;
   }

   public int getCount()
   {
      return count;
   }

   public Map getStats()
   {
      return stats;
   }

   public boolean addItem(StatisticalItem item)
   {
      if(name == DEFAULT_NAME)
      {
         name = item.getValue();
      }

      boolean addedNew = false;
      Map itemMap = (Map) stats.get(item.getName());
      if(itemMap == null)
      {
         itemMap = new HashMap();
         stats.put(item.getName(), itemMap);
         addedNew = true;
      }

      StatisticalItem curItem = (StatisticalItem) itemMap.get(item.getValue());
      if(curItem == null)
      {
         itemMap.put(item.getValue(), item);
      }
      else
      {
         curItem.add(item);
      }
      return addedNew;
   }

   /**
    * This method destroys txReport parameter!!! Not really a nice implementation.
    */
   public void merge(TxReport txReport)
   {
      for(Iterator iter = txReport.stats.entrySet().iterator(); iter.hasNext();)
      {
         Map.Entry entry = (Map.Entry) iter.next();
         String itemName = (String) entry.getKey();

         Map myMap = (Map) stats.get(itemName);
         Map itemMap = (Map) entry.getValue();

         if(myMap == null)
         {
            stats.put(itemName, itemMap);
         }
         else
         {
            // first merge common items
            for(Iterator myItems = myMap.values().iterator(); myItems.hasNext();)
            {
               StatisticalItem myItem = (StatisticalItem) myItems.next();
               StatisticalItem newItem = (StatisticalItem) itemMap.remove(myItem.getValue());

               if(newItem == null)
               {
                  myItem.mergeNull();
               }
               else
               {
                  myItem.merge(newItem);
               }
            }

            // add new items
            if(!itemMap.isEmpty())
            {
               for(Iterator newItems = itemMap.values().iterator(); newItems.hasNext();)
               {
                  StatisticalItem newItem = (StatisticalItem) newItems.next();
                  myMap.put(newItem.getValue(), newItem);
               }
            }
         }
      }

      count += txReport.count;
   }

   // Inner

   public static class MethodStats extends AbstractStatisticalItem
   {
      public static final String NAME = "Method Statistics Per Transaction";

      public MethodStats(String method)
      {
         super(NAME);
         value = method;
      }
   }

   public static class SqlStats extends AbstractStatisticalItem
   {
      public static final String NAME = "SQL Statistics Per Transaction";

      public SqlStats(String sql)
      {
         super(NAME);
         value = sql;
      }
   }
}