code: /* LanguageTool, a natural language style checker
   * Copyright (C) 2018 Daniel Naber (http://www.danielnaber.de)
   *
   * This library is free software; you can redistribute it and/or
   * modify it under the terms of the GNU Lesser General Public
   * License as published by the Free Software Foundation; either
   * version 2.1 of the License, or (at your option) any later version.
   *
   * This library 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
   * Lesser General Public License for more details.
   *
   * You should have received a copy of the GNU Lesser General Public
   * License along with this library; if not, write to the Free Software
   * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301
   * USA
   */
  package org.languagetool.server;
  
  import java.util.HashSet;
  import java.util.Map;
  import java.util.Set;
  import java.util.concurrent.ConcurrentHashMap;
  import java.util.concurrent.atomic.AtomicInteger;
  
  class RequestCounter {
  
    private final AtomicInteger reqCount = new AtomicInteger();
    private final AtomicInteger handleCount = new AtomicInteger();
    private final Map<Integer, String> handleIps = new ConcurrentHashMap<>();
  
    int getHandleCount() {
      return handleCount.get();
    }
  
    int getRequestCount() {
      return reqCount.get();
    }
    
    void incrementHandleCount(String ip, int uniqueId) {
      handleCount.incrementAndGet();
      handleIps.put(uniqueId, ip);
    }
  
    int incrementRequestCount() {
      return reqCount.incrementAndGet();
    }
  
    void decrementHandleCount(int uniqueId) {
      handleCount.decrementAndGet();
      handleIps.remove(uniqueId);
    }
  
    int getDistinctIps() {
      Set<String> ips = new HashSet<>();
      for (Map.Entry<Integer, String> entry : handleIps.entrySet()) {
        ips.add(entry.getValue());
      }
      return ips.size();
    }
  }
  