[리팩터링 2판] 12.10 서브클래스를 위임으로 바꾸기(예제 2)

위임 포인트 : 클래스 내의 스태틱한 팩토리 메서드 or 최상의 function, 자식-부모 연결을 제거 한 후 각각 인스턴스 생성

 

2. 예시2: 서브 클래스가 여러 개일 때 

 

아래와 같은 코드가 있다고 가정한다.

이 코드는 곧 야생 조류와 사육 조류를 구분 짓기 위해 크게 수정할 예정이다. (이 차이를 WildBird와 CaptiveBird라는 두 서브클래스로 모델링하는 방법도 있다)

상속은 한번만 쓸 수 있으니 야생과 사육을 기준으로 나누려면 종에 따른 분류를 포기해야 한다.

function createBird(data) {
	switch (data.type) {
		case '유럽 제비':
			return new EuropeanSwallow(data);
			break;
		case '아프리카 제비':
			return new AfricanSwallow(data);
			break;
		case '노르웨이 파랑 앵무':
			return new NorwegianBlueSwallow(data);
			break;

		default:
			return new Bird(data);
	}
}

class Bird {
	constructor(data) {
		this._name = data.name;
		this._plumage = data.plumage;
	}

	get name() {
		return this._name;
	}
	get plumage() {
		return this._plumage || '보통이다';
	}
	get airSpeedVelocity() {
		return null;
	}
}

class EuropeanSwallow extends Bird {
	get airSpeedVelocity() {
		return 35;
	}
}

class AfricanSwallow extends Bird {
	constructor(data) {
		super(data);
		this._numberOfCoconuts = data.numberOfCoconuts;
	}
	get airSpeedVelocity() {
		return 40 - 2 * this._numberOfCoconuts;
	}
}

class NorwegianBlueSwallow extends Bird {
	constructor(data) {
		super(data);
		this._voltage = data.voltage;
		this._isNailed = data.isNailed;
	}
	get plumage() {
		if (this._voltage > 100) return '그을렸다';
		else return this._plumage || '예쁘다';
	}
	get airSpeedVelocity() {
		return (this._isNailde) ? 0 : 10 + this._voltage / 10;
	}
}

 

예시2 : 상속을 위임으로 변경하기 위한 코드 리팩토링

*순서는 책과 다르게 기재하였습니다

 

(1) 역시, 팩토리 메서드를 활용해보자

function createBird(data) {
	return new Bird(data);
}

(2) Bird 클래스의 생성자를 아래와 같이 변경

	constructor(data) {
		this._name = data.name;
		this._plumage = data.plumage;
		this._speciesPlumge = this.selectSpeciesDelegate(data);
	}

(3) Bird 클래스 내 깃털 상태나 공속 게터도 스페셜 델리게이트에서 값을 가져오도록 변경

	get plumage() {
		return this._speciesDelegate.plumage;
	}
	get airSpeedVelocity() {
		return this._speciesDelegate.airSpeedVelocity;
	}

(4) Bird 클래스 내 selectSpecialDelegate를 작성해준다.

selectSpecialDelegate를 호출해야지 해당 data에 해당하는 델리게이트를 만들 수 있도록

	selectSpecialDelegate(data) {
		switch (data.type) {
			case '유럽 제비':
				return new EuropeanSwallowDelegate(data, this);
				break;
			case '아프리카 제비':
				return new AfricanSwallowDelegate(data, this);
				break;
			case '노르웨이 파랑 앵무':
				return new NorwegianBlueSwallowDelegate(data, this);
				break;

			default:
				return new SpeciesDelegate(data, this);
		}
	}

(6) 클래스 SpeciesDelegate 생성

class SpeciesDelegate {
	constructor(data, bird) {
		this._bird = bird;
	}

	get plumge() {
		return this._bird._plumage || '보통이다';
	}

	get airSpeedVelocity() {
		return null;
	}

}

(7) 서브 클래스들이 SpeciesDelegate를 확장받도록 하고, 클래스명 적절하게 변경

class EuropeanSwallowDelegate extends SpeciesDelegate {
	get airSpeedVelocity() {
		return 35;
	}
}

(8) 역참조 추가

NorwegianBlueSwallowDelegate는 다른 서브 클래스와는 다르게 깃털 상태, 즉 plumage()를 오버라이드 한다. 생성자에 Bird로의 역참조를 추가해야 한다.

class AfricanSwallowDelegate extends SpeciesDelegate {
	constructor(data, bird) {
		super(data, bird);
		this._numberOfCoconuts = data.numberOfCoconuts;
	}
	get airSpeedVelocity() {
		return 40 - 2 * this._numberOfCoconuts;
	}
}

class NorwegianBlueSwallowDelegate extends SpeciesDelegate {
	constructor(data, bird) {
		super(data, bird);
		this._voltage = data.voltage;
		this._isNailed = data.isNailed;
	}
	get airSpeedVelocity() {
		return (this._isNailde) ? 0 : 10 + this._voltage / 10;
	}
	get plumge() {
		if (this._voltage > 100) return '그을렸다';
		else return thisthis._bird._plumage || '예쁘다';
	}
}

 

예시2 리팩토링 후 전체 코드

function createBird(data) {
	return new Bird(data);
}

class Bird {
	constructor(data) {
		this._name = data.name;
		this._plumage = data.plumage;
		this._speciesPlumge = this.selectSpeciesDelegate(data);
	}

	get name() { return this._name; }
	get plumage() {
		return this._speciesDelegate.plumage;
	}
	get airSpeedVelocity() {
		return this._speciesDelegate.airSpeedVelocity;
	}

	selectSpeciesDelegate(data) {
		switch (data.type) {
			case '유럽 제비':
				return new EuropeanSwallowDelegate(data, this);
				break;
			case '아프리카 제비':
				return new AfricanSwallowDelegate(data, this);
				break;
			case '노르웨이 파랑 앵무':
				return new NorwegianBlueSwallowDelegate(data, this);
				break;

			default:
				return new SpeciesDelegate(data, this);
		}
	}
}

class SpeciesDelegate {
	constructor(data, bird) {
		this._bird = bird;
	}

	get plumge() {
		return this._bird._plumage || '보통이다';
	}

	get airSpeedVelocity() {
		return null;
	}

}


class EuropeanSwallowDelegate extends SpeciesDelegate {
	get airSpeedVelocity() {
		return 35;
	}
}

class AfricanSwallowDelegate extends SpeciesDelegate {
	constructor(data, bird) {
		super(data, bird);
		this._numberOfCoconuts = data.numberOfCoconuts;
	}
	get airSpeedVelocity() {
		return 40 - 2 * this._numberOfCoconuts;
	}
}

class NorwegianBlueSwallowDelegate extends SpeciesDelegate {
	constructor(data, bird) {
		super(data, bird);
		this._voltage = data.voltage;
		this._isNailed = data.isNailed;
	}
	get airSpeedVelocity() {
		return (this._isNailde) ? 0 : 10 + this._voltage / 10;
	}
	get plumge() {
		if (this._voltage > 100) return '그을렸다';
		else return thisthis._bird._plumage || '예쁘다';
	}
}

 

예시2 리팩토링 : Bird를 상속으로부터 구제한 것 외에 이 리팩터링에서 얻은 것

예시2 리팩토링은 원래의 서브클래스들을 위임으로 교체했지만 SpeciesDelegate에는 여전히 처음 구조와 매우 비슷한 계층 구조가 존재한다. 위임으로 옮겨진 종 계층구조는 더 엄격하게 종과 관련한 내용만을 다루게 되었다. 위임 클래스들은 종에 따라 달라지는 데이터와 메서드만을 담게 되고 종과 상관없는 공통 코드는 Bird 자체와 미래의 서브클래스들에 남는다.$

 

 

책과 순서대로 리팩토링을 참고하고 싶다면! : https://eunjin3786.tistory.com/405

리팩터링 (7) - 타입코드를 서브클래스로 / 서브클래스를 위임으로