/*
    * Licensed to the Apache Software Foundation (ASF) under one
    * or more contributor license agreements.  See the NOTICE file
    * distributed with this work for additional information
    * regarding copyright ownership.  The ASF licenses this file
    * to you under the Apache License, Version 2.0 (the
    * "License"); you may not use this file except in compliance
    * with the License.  You may obtain a copy of the License at
    *
   *   http://www.apache.org/licenses/LICENSE-2.0
   *
   * Unless required by applicable law or agreed to in writing,
   * software distributed under the License is distributed on an
   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
   * KIND, either express or implied.  See the License for the
   * specific language governing permissions and limitations
   * under the License.
   */
  package org.apache.maven.surefire.report;
  
  import java.util.Map;
  import java.util.concurrent.ConcurrentHashMap;
  import java.util.concurrent.atomic.AtomicInteger;
  
  import org.apache.maven.surefire.api.util.internal.ClassMethod;
  
  import static java.util.Objects.requireNonNull;
  
  /**
   * Creates an index for class/method.
   * Returns ThreadLocal index if created before.
   */
  public final class ClassMethodIndexer {
      private final AtomicInteger classIndex = new AtomicInteger(1);
      private final AtomicInteger methodIndex = new AtomicInteger(1);
      private final Map<ClassMethod, Long> testIdMapping = new ConcurrentHashMap<>();
      private final ThreadLocal<Long> testLocalMapping = new ThreadLocal<>();
  
      public long indexClassMethod(String clazz, String method) {
          ClassMethod key = new ClassMethod(requireNonNull(clazz), method);
          return testIdMapping.computeIfAbsent(key, cm -> {
              Long classId = testIdMapping.get(new ClassMethod(requireNonNull(clazz), null));
              long c = classId == null ? (((long) classIndex.getAndIncrement()) << 32) : classId;
              int m = method == null ? 0 : methodIndex.getAndIncrement();
              long id = c | m;
              testLocalMapping.set(id);
              return id;
          });
      }
  
      public long indexClass(String clazz) {
          return indexClassMethod(clazz, null);
      }
  
      public Long getLocalIndex() {
          return testLocalMapping.get();
      }
  }