如何让子类返回自己的类型？

Question

这里我们有多个使用相关语法的自定义查询-DSL.我正在创建一个创建一个AbstractBuilder所有共性可以写在一个地方.问题是这会导致方法链接问题.当我尝试从一个方法链接到AbstractBuilder一个子类时,如果没有强制转换它就无法工作.

有了这些课程:

class AbstractBuilder{
    protected final StringBuilder bldr = new StringBuilder();

    AbstractBuilder addValue( String name, String value ){
        bldr.append( name ).append( '=' ).append( value )append( ',' );
        return this;
    }

    String toString(){
        return bldr.toString();
    }
}

class IntBuilder extends AbstractBuilder{

    IntBuilder addValue( String name, int value ){
        bldr.append( name ).append( '=' ).append( value )append( ',' );
        return this;
    }
}

这有效

new IntBuilder().addValue( "age", 12 ).addValue( "name", "Bruce" ).toString();` but `new IntBuilder().addValue( "name", "Bruce" ).addValue( "age", 12 ).toString();

除非你做一个丑陋的演员,否则不会:

((IntBuilder) (new IntBuilder().addValue( "name", "Bruce" ))).addValue( "age", 12 ).toString();

现在我想我可以覆盖每个方法并通过调用父母(通过super.addValue( name, value );)来实现它们,但这真的很难看.

我怎么能让每个方法都返回当前的类而不是它所定义的类？

Answer 1

你可以使用泛型来实现这一点.

class BaseBuilder<B extends BaseBuilder>{
    protected final StringBuilder bldr = new StringBuilder();
    private Class<B> builderImplementationClass;

    protected BaseBuilder( Class<B> builderImpl ){
        builderImplementationClass = builderImpl;

        if( this.getClass() != builderImpl )
            throw new IllegalStateException( "Should match." );
    }

    B addValue( String name, String value ){
        bldr.append( name ).append( '=' ).append( value )append( ',' );
        return builderImplementationClass.cast( this );
    }

    String toString(){
        return bldr.toString();
    }
}

class IntBuilder extends AbstractBuilder<IntBuilder>{

    public IntBuilder(){
        super( IntBuilder.class );
    }

    IntBuilder addValue( String name, int value ){
        bldr.append( name ).append( '=' ).append( value )append( ',' );
        return this;
    }
}

基本上,您使所有方法都返回一个类型的值,B并确保将B其设置为当前的实现