Java 如何避免类型转换

Question

我过去曾多次遇到过这个问题，但还没有真正找到一个好的解决方案/设计。

以下示例代码将从实体（公司或文章）生成 PDF 文档

public class Entity
{
    int id;
}

public class Company extends Entity
{
    private String HQ;
}

public class Article extends Entity
{
    private String title;
}

public interface EntityPDFGenerator
{
    void generate(Entity entity);
}

public class ArticlePDFGenerator implements EntityPDFGenerator
{
    public void generate(Entity entity)
    {
        Article article = (Article) entity;
        // create Article related PDF from entity
    }
}

public class CompanyPDFGenerator implements EntityPDFGenerator
{
    public void generate(Entity entity)
    {
        Company company = (Company) entity;
        // create Company related PDF
    }
}

主类：

public class PDFGenerator
{
    public void generate(Entity entity)
    {
        EntityPDFGenerator pdfGenerator = getConcretePDFGenerator(entity);
        pdfGenerator.generate(entity);

    }

    // lets make the factory task simple for now
    EntityPDFGenerator getConcretePDFGenerator(Entity entity)
    {
        if(entity instanceof Article){
            return new ArticlePDFGenerator();
        }else{
            return new CompanyPDFGenerator();
        }
    }
}

在上述方法中，问题在于将实体转换为具体类型（在代码的后期进行转换可能很危险）。我试图用泛型来做，但后来我收到了警告

对“生成（T）”的未经检查的调用

我可以改进这段代码吗？

Answer 1

在这里，您可以进行建议的更改：

public interface EntityPDFGenerator<T extends Entity> {
    void generate(T entity);
}


public class ArticlePDFGenerator implements EntityPDFGenerator<Article> {

    public void generate(Article entity)
    {
        // create Article related PDF from entity
    }
}


public class CompanyPDFGenerator implements EntityPDFGenerator<Company> {

    public void generate(Company entity)
    {
        // create Company related PDF
    }
}

Answer 2

简答

泛型在这里不是正确的工具。您可以明确地进行强制转换：

public class CompanyPDFGenerator implements EntityPDFGenerator
{
    public void generate(Entity entity)
    {
        if (! (entity instanceof Company)) {
            throw new IllegalArgumentException("CompanyPDFGenerator works with Company object. You provided " + (entity == null ? "null" : entity.getClass().getName()));
        }
        Company company = (Company) entity;
        System.out.println(company);
        // create Company related PDF
    }
}

或者您可以在实体类中定义某种数据结构并仅在打印机中使用：

public abstract class Entity
{
    int id;
    public abstract EntityPdfData getPdfData();
}

// ...

public class CompanyPDFGenerator implements EntityPDFGenerator
{
    public void generate(Entity entity)
    {
        EntityPdfData entityPdfData = entity.getPdfData();
        // create Company related PDF
    }
}

长答案

如果您在编译时知道类型，泛型很有用。 IE。如果您可以将实际类型写入程序。对于列表，它看起来很简单：

// now you know at compile time that you need a list of integers
List<Integer> list = new ArrayList<>();

在你的例子中你不知道：

public void generate(Entity entity)
{
    // either Article or Company can come it. It's a general method
    EntityPDFGenerator pdfGenerator = getConcretePDFGenerator(entity);
    pdfGenerator.generate(entity);

}

假设您想将类型添加到 EntityPDFGenerator ，如下所示：

public static interface EntityPDFGenerator<T extends Entity>
{
    void generate(T entity);
}

public static class ArticlePDFGenerator implements EntityPDFGenerator<Article>
{
    public void generate(Article entity)
    {
        Article article = (Article) entity;
        // create Article related PDF from entity
    }
}

public static class CompanyPDFGenerator implements EntityPDFGenerator<Company>
{
    public void generate(Company entity)
    {
        Company company = (Company) entity;
        // create Company related PDF
    }
}

这看起来不错。但是，获得正确的生成器将很棘手。 Java 泛型是不变的。甚至ArrayList<Integer> 也不是ArrayList<Number> 的子类。所以，ArticlePdfGenerator 不是EntityPDFGenerator<T extends Entity> 的子类。 IE。 这不会编译：

<T extends Entity> EntityPDFGenerator<T> getConcretePDFGenerator(T entity, Class<T> classToken)
{
    if(entity instanceof Article){
        return new ArticlePDFGenerator();
    }else{
        return new CompanyPDFGenerator();
    }
}

Answer 3

我建议在 Entity 类中移动 getGenerator() 方法并在 Company 和 Article 类中覆盖它。 当然，除非有充分的理由不这样做。