[JAVA 디자인패턴] Composite pattern(컴포지트 패턴)

OCP를 만족시키면서 부분-전체 관계를 가지는 객체들 간의 관계를 정의할 때 쓸 수있다. 전체와 부분을 동일한 인터페이스로 관리할 수 있다.

composite pattern을 이해하기 위해서, 하나의 박스가 있다고 하자. 그리고 그 박스에는 컴퓨터의 부품들이 들어간다.

Box클래스를 만들고 Keyboard, Monitor, Speaker 객체들을 멤버로 가진다.

 

public class Box {
    private Keyboard keyboard;
    private Monitor monitor;
    private Speaker speaker;
 
    public int getTotalPrice(){
        return keyboard.getPrice() + monitor.getPrice() + speaker.getPrice();
    }
 
    public int getTotalPower(){
        return keyboard.getPower() + monitor.getPower() + speaker.getPower();
    }
    
    //setter...
}

public class Main {
    public static void main(String[] args) {
        Box box = new Box();
 
        Keyboard keyboard = new Keyboard(100000,20);
        Monitor monitor = new Monitor(200000,40);
        Speaker speaker = new Speaker(70000,50);
 
        box.setKeyboard(keyboard);
        box.setMonitor(monitor);
        box.setSpeaker(speaker);
 
        int totalPrice = box.getTotalPrice();
        int totalPower = box.getTotalPower();
 
        System.out.println("total price : " + totalPrice);
        System.out.println("total power : " + totalPower);
 
    }
}

public class Speaker extends ComputeDevice
public class Keyboard extends ComputeDevice
public class Monitor extends ComputeDevice

public class Main {
    public static void main(String[] args) {
        Box box = new Box();
 
        Keyboard keyboard = new Keyboard(100000,20);
        Monitor monitor = new Monitor(200000,40);
        Speaker speaker = new Speaker(70000,50);
 
        box.addComputerDevice(keyboard);
        box.addComputerDevice(monitor);
        box.addComputerDevice(speaker);
 
        int totalPrice = box.getTotalPrice();
        int totalPower = box.getTotalPower();
 
        System.out.println("total price : " + totalPrice);
        System.out.println("total power : " + totalPower);
 
    }
}

 

 

위와 같은 방식으로 하면, OCP를 만족시키고 여러 ComputerDevice를 추가할 수 있다.

 

하지만, 요구사항을 하나 추가해보도록 하자. 만약, Box안에 Box를 넣고 싶으면 어떻게 해야할까? 여기서 중요한 것은 부분과 전체를 구별하지 않고 같은 인터페이스로 해야한다. 즉, ComputerDevice를 넣는 addComputerDevice 메소드를 통해서 Box도 넣어야 한다. 마찬가지로, ComputerDevice를 제거하는 rmComputerDevice 메소드를 통해서 Box를 제거할 수 있어야한다.

의외로 이 문제는 아주 간단하게 풀린다. Box도 ComputerDevice의 자식클래스이면 된다.

 

 

public class Box extends ComputeDevice {
 
    private List<ComputeDevice> computeDeviceList = new ArrayList<ComputeDevice>();
 
    public int getTotalPrice(){
        return computeDeviceList.stream()
            .flatMapToInt(computeDevice -> IntStream.of(computeDevice.getPrice()))
            .sum();
    }
 
    public int getTotalPower(){
        return computeDeviceList.stream()
            .flatMapToInt(computeDevice -> IntStream.of(computeDevice.getPower()))
            .sum();
    }
 
    public void addComputerDevice(ComputeDevice computeDevice){
        computeDeviceList.add(computeDevice);
    }
 
    public void rmComputerDevice(ComputeDevice computeDevice){
        computeDeviceList.remove(computeDevice);
    }
 
    @Override
    public int getPower() {
        return getTotalPower();
    }
 
    @Override
    public int getPrice() {
        return getTotalPrice();
    }
}

 

참고문헌

[JAVA 객체지향 디자인 패턴, 정인상]