對工廠模式的形狀實例進(jìn)行擴(kuò)展，增加顏色填充功能。

類圖

• 形狀和顏色接口

public interface Color {
    void fill();
}
------------------------------------------------------
public interface Shape {
    void draw();
}

• 顏色實現(xiàn)類（形狀的實現(xiàn)類上一篇有）

public class Blue implements Color{
    @Override
    public void fill() {
        System.out.println("fill Blue color!");
    }
}
------------------------------------------------------
public class Red implements Color{
    @Override
    public void fill() {
        System.out.println("fill Red color!");
    }
}
------------------------------------------------------
public class Yellow implements Color{
    @Override
    public void fill() {
        System.out.println("fill Yellow color!");
    }
}

• 為 Color 和 Shape 對象創(chuàng)建抽象類來獲取工廠

public interface AbstractFactory {
    Color getColor(String color);
    Shape getShape(String shape);
}

• 創(chuàng)建擴(kuò)展了 AbstractFactory 的工廠類，基于給定的信息生成實體類的對象

public class ColorFactory implements AbstractFactory{

    @Override
    public Color getColor(String color) {
        if(color == null){
            return null;
        }
        if(color.equalsIgnoreCase("RED")){
            return new Red();
        } else if(color.equalsIgnoreCase("YELLOW")){
            return new Yellow();
        } else if(color.equalsIgnoreCase("BLUE")){
            return new Blue();
        }
        return null;
    }

    @Override
    public Shape getShape(String shape) {
        return null;
    }
}
----------------------------------------------------------------------------------
public class ShapeFactory implements AbstractFactory{

    @Override
    public Color getColor(String color) {
        return null;
    }

    @Override
    public Shape getShape(String shape) {
        if(shape == null){
            return null;
        }
        if(shape.equalsIgnoreCase("CIRCLE")){
            return new Circle();
        } else if(shape.equalsIgnoreCase("RECTANGLE")){
            return new Rectangle();
        } else if(shape.equalsIgnoreCase("SQUARE")){
            return new Square();
        }
        return null;
    }
}

• 創(chuàng)建一個工廠創(chuàng)造器/生成器類，通過傳遞形狀或顏色信息來獲取工廠

public class FactoryProducer {
    public static AbstractFactory getFactory(String choice){
        if(choice.equalsIgnoreCase("SHAPE")){
            return new ShapeFactory();
        } else if(choice.equalsIgnoreCase("COLOR")){
            return new ColorFactory();
        }
        return null;
    }
}

• 使用 FactoryProducer 來獲取 AbstractFactory，通過傳遞類型信息來獲取實體類的對象

public class AbstractFactoryPatternDemo {
    public static void main(String[] args) {
        AbstractFactory colorFatory = FactoryProducer.getFactory("color");
        Color red = colorFatory.getColor("red");
        red.fill();
        Color yellow = colorFatory.getColor("yellow");
        yellow.fill();
        Color blue = colorFatory.getColor("blue");
        blue.fill();

        AbstractFactory shapeFactory = FactoryProducer.getFactory("shape");
        Shape rectangle = shapeFactory.getShape("rectangle");
        rectangle.draw();
        Shape circle = shapeFactory.getShape("circle");
        circle.draw();
        Shape square = shapeFactory.getShape("square");
        square.draw();
    }
}

• 輸出結(jié)果：

fill Red color!
fill Yellow color!
fill Blue color!
Draw Rectangle!
Draw Circle!
Draw Square!