/*
 * Copyright 2019, OpenTelemetry Authors
 *
 * Licensed 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 io.opentelemetry.sdk.trace;

import io.opentelemetry.trace.SpanId;
import io.opentelemetry.trace.TraceId;
import java.util.Random;

final class RandomIdsGenerator implements IdsGenerator {

  private static final ThreadLocal<Random> threadLocalRandom =
      new ThreadLocal<Random>() {
        @Override
        protected Random initialValue() {
          return new Random();
        }
      };
  private static final long INVALID_ID = 0;

  @Override
  public SpanId generateSpanId() {
    long id;
    do {
      id = threadLocalRandom.get().nextLong();
    } while (id == INVALID_ID);
    return new SpanId(id);
  }

  @Override
  public TraceId generateTraceId() {
    long idHi;
    long idLo;
    do {
      Random random = threadLocalRandom.get();
      idHi = random.nextLong();
      idLo = random.nextLong();
    } while (idHi == INVALID_ID && idLo == INVALID_ID);
    return new TraceId(idHi, idLo);
  }
}

@@ -731,16 +731,38 @@
 va.util.
+concurrent.ThreadLocal
 Random;%0A
@@ -822,216 +822,8 @@
 or %7B
-%0A%0A  private static final ThreadLocal%3CRandom%3E threadLocalRandom =%0A      new ThreadLocal%3CRandom%3E() %7B%0A        @Override%0A        protected Random initialValue() %7B%0A          return new Random();%0A        %7D%0A      %7D;
 %0A  p
@@ -932,25 +932,36 @@
     
-do %7B%0A      id = t
+ThreadLocalRandom random = T
 hrea
@@ -973,21 +973,53 @@
 lRandom.
-get()
+current();%0A    do %7B%0A      id = random
 .nextLon
@@ -1170,27 +1170,27 @@
 Lo;%0A    
-do %7B%0A      
+ThreadLocal
 Random r
@@ -1197,17 +1197,17 @@
 andom = 
-t
+T
 hreadLoc
@@ -1219,14 +1219,27 @@
 dom.
-get();
+current();%0A    do %7B
 %0A   

Use ThreadLocalRandom insted of our own class. (#846)