com.github.romanqed.jeflect.transform.CheckedTransformer Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jeflect-transform Show documentation

The bytecode transformers.

The newest version!

package com.github.romanqed.jeflect.transform;

import java.lang.instrument.ClassFileTransformer;
import java.security.ProtectionDomain;
import java.util.Collections;
import java.util.LinkedList;
import java.util.List;
import java.util.function.Supplier;

public abstract class CheckedTransformer implements ClassFileTransformer {
    private final Supplier> supplier;
    private List problems;

    protected CheckedTransformer(Supplier> supplier) {
        this.supplier = supplier;
        this.problems = Collections.synchronizedList(supplier.get());
    }

    protected CheckedTransformer() {
        this(LinkedList::new);
    }

    public synchronized void validate() {
        if (!problems.isEmpty()) {
            throw new ClassTransformException(problems);
        }
        this.problems = Collections.synchronizedList(supplier.get());
    }

    protected abstract byte[] checkedTransform(ClassLoader loader,
                                               String className,
                                               Class classBeingRedefined,
                                               ProtectionDomain protectionDomain,
                                               byte[] classfileBuffer) throws Throwable;

    @Override
    public byte[] transform(ClassLoader loader,
                            String className,
                            Class classBeingRedefined,
                            ProtectionDomain protectionDomain,
                            byte[] classFileBuffer) {
        try {
            return checkedTransform(loader, className, classBeingRedefined, protectionDomain, classFileBuffer);
        } catch (Throwable e) {
            problems.add(e);
            return classFileBuffer;
        }
    }
}