public static ByteBuffer getCurrentThreadStorage()

in jvmti-access/src/main/java/co/elastic/otel/UniversalProfilingCorrelation.java [77:99]

22 lines of code
6 McCabe index (conditional complexity)


  public static ByteBuffer getCurrentThreadStorage(
      boolean allocateIfRequired, int expectedCapacity) {
    if (!virtualThreadSupportEnabled && isVirtual(Thread.currentThread())) {
      return null;
    }
    ByteBuffer buffer = threadStorage.get();
    if (buffer == null) {
      if (!allocateIfRequired) {
        return null;
      }
      buffer = ByteBuffer.allocateDirect(expectedCapacity);
      buffer.order(ByteOrder.nativeOrder());
      JvmtiAccess.setProfilingCorrelationCurrentThreadStorage(buffer);
      threadStorage.set(buffer);
    }

    int actualCapacity = buffer.capacity();
    if (actualCapacity != expectedCapacity) {
      throw new IllegalArgumentException(
          "Buffer has been allocated with a different capacity: " + actualCapacity);
    }
    return buffer;
  }