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

时间:2012-12-14 21:38:29

标签: java generics

这里我们有多个使用相关语法的自定义查询-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 );),但这真的很难看。

我怎样才能让每个方法都返回当前类而不是返回它的类?

2 个答案:

答案 0 :(得分:7)

您可以使用泛型来实现此目的。

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设置为当前的实现

答案 1 :(得分:1)

ArtB的解决方案很好,但您可以通过以下方式使其更简单:

private static class AbstractBuilder <T extends AbstractBuilder<T>> {
    protected StringBuilder bldr = new StringBuilder();

    public T addValue( String name, String value ){
        bldr.append( name ).append( '=' ).append( value ).append(',');
        @SuppressWarnings("unchecked")
        T result = (T)this;
        return result;
    }

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

private static class IntBuilder extends AbstractBuilder<IntBuilder> {

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

注意我还修复了两个私有StringBuilder变量的bug,这几乎肯定不是你想要的。