首页 > 代码库 > cglib源码分析（三）：Class生成策略

cglib源码分析（三）：Class生成策略

2024-07-13 15:12:55   219人阅读

cglib中生成类的工作是由AbstractClassGenerator的create方法使用相应的生成策略完成，具体代码如下：

private GeneratorStrategy strategy = DefaultGeneratorStrategy.INSTANCE;byte[] b = strategy.generate(this);

GeneratorStrategy是一个接口，它负责调用ClassGenerator 的generateClass方法来生成类。DefaultGeneratorStrategy是cglib提供的一个默认的生成策略，部分代码如下：

public class DefaultGeneratorStrategy implements GeneratorStrategy {    public static final DefaultGeneratorStrategy INSTANCE = new DefaultGeneratorStrategy();        public byte[] generate(ClassGenerator cg) throws Exception {        DebuggingClassWriter cw = getClassVisitor();        transform(cg).generateClass(cw);        return transform(cw.toByteArray());    }    protected DebuggingClassWriter getClassVisitor() throws Exception {      return new DebuggingClassWriter(ClassWriter.COMPUTE_MAXS);    }
    ... ...
    protected byte[] transform(byte[] b) throws Exception {        return b;    }    protected ClassGenerator transform(ClassGenerator cg) throws Exception {        return cg;    }}

DefaultGeneratorStrategy实现了ClassGenerator接口，并提供两个transform方法。用户可以通过继承DefaultGeneratorStrategy并重写相关的方法来检查或者修改生成类。

类生成策略和命名策略一样，使用了策略设计模式，用户可以通过实现GeneratorStrategy接口来自定义生成策略，并使用AbstractClassGenerator的setStrategy方法来设置自定义策略。


联系
我们