# SimpleMapCounter.java

    /*
     ** Module   : SimpleMapCounter.java
     ** Abstract : MBean supporting counting
     **
     ** Copyright (c) 2020-2024, Golden Code Development Corporation.
     **
     ** -#- -I- --Date-- ---------------------------------Description----------------------------------
     ** 001 HC  20201010 Created initial version.
     ** 002 HC  20240103 Added map counts. The map counter now accumulates counts and the provided values.
     */

    /*
     ** This program is free software: you can redistribute it and/or modify
     ** it under the terms of the GNU Affero General Public License as
     ** published by the Free Software Foundation, either version 3 of the
     ** License, or (at your option) any later version.
     **
     ** This program is distributed in the hope that it will be useful,
     ** but WITHOUT ANY WARRANTY; without even the implied warranty of
     ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
     ** GNU Affero General Public License for more details.
     **
     ** You may find a copy of the GNU Affero GPL version 3 at the following
     ** location: https://www.gnu.org/licenses/agpl-3.0.en.html
     **
     ** Additional terms under GNU Affero GPL version 3 section 7:
     **
     **   Under Section 7 of the GNU Affero GPL version 3, the following additional
     **   terms apply to the works covered under the License.  These additional terms
     **   are non-permissive additional terms allowed under Section 7 of the GNU
     **   Affero GPL version 3 and may not be removed by you.
     **
     **   0. Attribution Requirement.
     **
     **     You must preserve all legal notices or author attributions in the covered
     **     work or Appropriate Legal Notices displayed by works containing the covered
     **     work.  You may not remove from the covered work any author or developer
     **     credit already included within the covered work.
     **
     **   1. No License To Use Trademarks.
     **
     **     This license does not grant any license or rights to use the trademarks
     **     Golden Code, FWD, any Golden Code or FWD logo, or any other trademarks
     **     of Golden Code Development Corporation. You are not authorized to use the
     **     name Golden Code, FWD, or the names of any author or contributor, for
     **     publicity purposes without written authorization.
     **
     **   2. No Misrepresentation of Affiliation.
     **
     **     You may not represent yourself as Golden Code Development Corporation or FWD.
     **
     **     You may not represent yourself for publicity purposes as associated with
     **     Golden Code Development Corporation, FWD, or any author or contributor to
     **     the covered work, without written authorization.
     **
     **   3. No Misrepresentation of Source or Origin.
     **
     **     You may not represent the covered work as solely your work.  All modified
     **     versions of the covered work must be marked in a reasonable way to make it
     **     clear that the modified work is not originating from Golden Code Development
     **     Corporation or FWD.  All modified versions must contain the notices of
     **     attribution required in this license.
     */

    package com.goldencode.p2j.jmx;

    import java.util.*;
    import java.util.concurrent.*;
    import java.util.concurrent.atomic.*;

    /** MBean supporting counting */
    public class SimpleMapCounter
    extends SimpleLongCounter
    implements SimpleMapCounterMBean,
               MapCounter
    {
       /** Accumulated counts */
       private Map<String, AtomicLong> mapCounts = new ConcurrentHashMap<>();

       /** Accumulated values */
       private Map<String, AtomicLong> mapValues = new ConcurrentHashMap<>();

       /**
        * {@inheritDoc}
        */
       @Override
       public void update(String valueName, long value)
       {
          mapCounts.computeIfAbsent(valueName, k -> new AtomicLong(0)).incrementAndGet();
          mapValues.computeIfAbsent(valueName, k -> new AtomicLong(0)).addAndGet(value);
          super.update(value);
       }

       /**
        * Returns accumulated counts.
        *
        * @return  See above.
        */
       @Override
       public Map<String, Long> getMapCounts()
       {
          return getValues(mapCounts);
       }

       /**
        * Returns accumulated values.
        *
        * @return  See above.
        */
       @Override
       public Map<String, Long> getMapValues()
       {
          return getValues(mapValues);
       }

       /**
        * Returns accumulated values.
        *
        * @param   inValues
        *          The values to process.
        *
        * @return  See above.
        */
       private Map<String, Long> getValues(Map<String, AtomicLong> inValues)
       {
          Map<Long, List<String>> sortedMap = new TreeMap<>();
          for (Map.Entry<String, AtomicLong> e : inValues.entrySet())
          {
             List<String> values = sortedMap.get(e.getValue().get());
             if (values == null)
             {
                values = new ArrayList<>();
                sortedMap.put(e.getValue().get(), values);
             }
             values.add(e.getKey());
          }

          Map<String, Long> result = new LinkedHashMap<>();
          for (Map.Entry<Long, List<String>> e : sortedMap.entrySet())
          {
             for (String name : e.getValue())
             {
                result.put(name, e.getKey());
             }
          }

          return result;
       }

       @Override
       /**
        * Reset counters
        */
       public void reset()
       {
          mapCounts.clear();
          mapValues.clear();
          super.reset();
       }
    }