CreationRecordLeakDetector Class — netty Architecture

Source Code

common/src/test/java/io/netty/util/ResourceLeakDetectorTest.java lines 249–304

    private static final class CreationRecordLeakDetector<T> extends ResourceLeakDetector<T> {
        String canaryString;
        Object initialHint;
        boolean failOnUntraced = true;

        private final AtomicReference<Throwable> error = new AtomicReference<Throwable>();
        private final AtomicInteger leaksFound = new AtomicInteger(0);

        CreationRecordLeakDetector(Class<?> resourceType, int samplingInterval) {
            super(resourceType, samplingInterval);
        }

        public void initialise() {
            canaryString = "creation-canary-" + UUID.randomUUID();
            initialHint = canaryString;
            leaksFound.set(0);
        }

        @Override
        protected boolean needReport() {
            return true;
        }

        @Override
        protected void reportTracedLeak(String resourceType, String records) {
            if (!records.contains(canaryString)) {
                reportError(new AssertionError("Leak records did not contain canary string"));
            }
            leaksFound.incrementAndGet();
        }

        @Override
        protected void reportUntracedLeak(String resourceType) {
            if (failOnUntraced) {
                reportError(new AssertionError("Got untraced leak w/o canary string"));
            }
            leaksFound.incrementAndGet();
        }

        private void reportError(AssertionError cause) {
            error.compareAndSet(null, cause);
        }

        @Override
        protected Object getInitialHint(String resourceType) {
            return initialHint;
        }

        int getLeaksFound() {
            return leaksFound.get();
        }

        void assertNoErrors() throws Throwable {
            ResourceLeakDetectorTest.assertNoErrors(error);
        }
    }