Workload Class — netty Architecture

Source Code

microbench/src/main/java/io/netty/buffer/AllocationPatternSimulator.java lines 606–662

    private final class Workload implements Runnable {
        private final CountDownLatch startLatch;
        private final ByteBufAllocator allocator;
        private final SplittableRandom rng;
        private final AtomicBoolean stopCondition;
        private final int avgLiveBuffers;

        private Workload(CountDownLatch startLatch,
                         ByteBufAllocator allocator,
                         SplittableRandom rng,
                         AtomicBoolean stopCondition,
                         int avgLiveBuffers) {
            this.startLatch = startLatch;
            this.allocator = allocator;
            this.rng = rng;
            this.stopCondition = stopCondition;
            this.avgLiveBuffers = avgLiveBuffers;
        }

        Thread start(String name) {
            Thread thread = new Thread(this, name + '-' + THREAD_NAMES.compute(name, (n, c) -> c == null ? 1 : c + 1));
            thread.start();
            return thread;
        }

        @SuppressWarnings("BusyWait")
        @Override
        public void run() {
            try {
                startLatch.await();
                List<ByteBuf> buffers = new ArrayList<>();
                while (!stopCondition.get()) {
                    Thread.sleep(1);

                    int freqChoice = rng.nextInt(0, sumFrequency);
                    int choiceIndex = Arrays.binarySearch(cumulativeFrequency, freqChoice);
                    if (choiceIndex < 0) {
                        choiceIndex = -(choiceIndex + 1);
                    }
                    int chosenSize = size[choiceIndex];
                    ByteBuf buf = allocator.heapBuffer(chosenSize);
                    buffers.add(buf);
                    while (buffers.size() > rng.nextInt(0, avgLiveBuffers << 1)) {
                        int deallocChoice = rng.nextInt(0, buffers.size());
                        ByteBuf toDealloc = buffers.get(deallocChoice);
                        ByteBuf lastBuffer = buffers.remove(buffers.size() - 1);
                        if (toDealloc != lastBuffer) {
                            buffers.set(deallocChoice, lastBuffer);
                        }
                        toDealloc.release();
                    }
                }
            } catch (InterruptedException e) {
                throw new RuntimeException(e);
            }
        }
    }