第一章：重构，第一个案例(读书笔记系列3)

1.4运用多态（polymorphism）取代与价格相关的条件逻辑

 

price.h

#pragma once typedef enum _MovieType { eCHILDREN = 2, //ｶｯﾆｬ eREGUIAR = 0, //ﾆﾕﾍｨﾆｬ eNEW_RELEASE = 1 //ﾐﾂﾆｬ }MovieType; class Price { public: virtual int GetPriceCode(void) = 0; virtual double GetCharge(int daysRented) = 0; virtual int GetFrequentRenterPoints(int daysRented) { return 1; }; }; class ChildrensPrice :public Price { public: int GetPriceCode(void) { return eCHILDREN; }; double GetCharge(int daysRented) { double result = 1.5; if ( daysRented > 3) { result += (daysRented - 3) * 1.5; } return result; }; }; class NewReleasePrice :public Price { public: int GetPriceCode(void) { return eNEW_RELEASE; }; double GetCharge(int daysRented) { return daysRented * 3; }; //overload int GetFrequentRenterPoints(int daysRented) { return (daysRented > 1) ? 2 : 1; }; }; class RegularPrice :public Price { public: int GetPriceCode(void) { return eREGUIAR; }; double GetCharge(int daysRented) { double result = 2.0; if ( daysRented > 2) { result += (daysRented - 2) * 1.5; } return result; }; };

 

movie.h

 

#pragma once #include <string> #include "price.h" class Movie { public: Movie(std::string title, int priceCode) { m_title = title; SetPriceCode(priceCode); }; std::string GetTitle(void) { return m_title; }; // //v1.4 double GetCharge(int daysRented) { return m_priceCode->GetCharge(daysRented); }; int GetFrequentRenterPoints(int daysRented) { return m_priceCode->GetFrequentRenterPoints(daysRented); }; void SetPriceCode(int arg) { switch(arg) { case eCHILDREN://儿童片 m_priceCode = new ChildrensPrice(); break; case eREGUIAR://普通片 m_priceCode = new RegularPrice(); break; case eNEW_RELEASE://新片 m_priceCode = new NewReleasePrice(); break; default: break; } }; int GetPriceCode(void) { return m_priceCode->GetPriceCode(); }; private: std::string m_title;//名称 Price* m_priceCode;//价格代号 };

 

Rental.h

 

#pragma once #include "Movie.h" class Rental { public: Rental(Movie* pMovie,int rented) { m_pMovie = pMovie; m_daysRented = rented; }; int GetDaysRented(void) { return m_daysRented; }; Movie* GetMovie(void) { return m_pMovie; }; // public: double GetCharge(void) { return m_pMovie->GetCharge(m_daysRented); }; //常客积点计算 int GetFrequentRenterPoints(void) { return m_pMovie->GetFrequentRenterPoints(m_daysRented); }; // private: Movie* m_pMovie;//影片 int m_daysRented;//租期 };

 

Customer.h

#pragma once #include <vector> #include "Rental.h" class Customer { // //v1.3分解并重组:迭代器类型的声明 public: typedef std::vector<Rental*>::const_iterator RentalIterator; // public: Customer(std::string name) { m_name = name; }; void AddRental(Rental* arg) { m_Rentals.push_back(arg); }; std::string GetName(void) { return m_name; }; std::string Statement(void); // //v1.3分解并重组 public: //添加功能 std::string HtmlStatement(void); private: //常客积点计算总数 double GetTotalFrequentRenterPoints(void); //常客积点计算总消费金额 double GetTotalCharge(void); // private: std::string m_name; //姓名 std::vector<Rental*> m_Rentals; //租借记录 };

 

 

Customer.cpp

#include "Customer.h" #include <iostream> #include "Common.h" std::string Customer::Statement(void) { RentalIterator iterRentals = m_Rentals.begin(); std::string result = "Rental Record for" + this->GetName() + "/n"; while (iterRentals != m_Rentals.end()) { //显示此笔数据 result += "/t" + (*iterRentals)->GetMovie()->GetTitle() + "/t" + CommonFunction::ConvertToString<double>((*iterRentals)->GetCharge()) + "/n"; ++iterRentals; } //结尾打映 result += "Amount owed is " + CommonFunction::ConvertToString<double>(GetTotalCharge()) + "/n"; result += "You earned " + CommonFunction::ConvertToString<double>(GetTotalFrequentRenterPoints()) + " frequent renter points"; return result; }; //添加功能 std::string Customer::HtmlStatement(void) { RentalIterator iterRentals = m_Rentals.begin(); std::string result = "<H1>Rental Record for<EM>" + this->GetName() + "</EM></H1></P>/n"; while (iterRentals != m_Rentals.end()) { result += (*iterRentals)->GetMovie()->GetTitle() + ": " + CommonFunction::ConvertToString<double>((*iterRentals)->GetCharge()) + "<BR>/n"; ++iterRentals; } result += "<P>Amount owed <EM> " + CommonFunction::ConvertToString<double>(GetTotalCharge()) + "</EM><P>/n"; result += "You earned <EM>" + CommonFunction::ConvertToString<double>(GetTotalFrequentRenterPoints()) + "</EM> frequent renter points<P>"; return result; }; //query method double Customer::GetTotalFrequentRenterPoints(void) { int result = 0; RentalIterator iterRentals = m_Rentals.begin(); while (iterRentals != m_Rentals.end()) { //累加常客积点 result += (*iterRentals)->GetFrequentRenterPoints(); ++iterRentals; } return result; }; //query method double Customer::GetTotalCharge(void) { double result = 0; RentalIterator iterRentals = m_Rentals.begin(); while (iterRentals != m_Rentals.end()) { result += (*iterRentals)->GetCharge(); ++iterRentals; } return result; };

 

Common.h (共同的方法)

#pragma once #include <string> #include <iostream> #include <sstream> class CommonFunction { public: template <class T> static std::string ConvertToString(T value) { std::stringstream ss; ss << value; return ss.str(); }; };

 

Refactoring1.4.cpp 变更的履历写在这个里面

// //v1.4 运用多态 // step1:GetCharge应该转移到Movie类中 // step2:GetFrequentRenterPoints也转移到Movie类中 // step3:Replace type Code with State/Strategy // 增加SetPriceCode和GetPriceCode，和Price类 // step4:对(Movie::getCharge)实施Move Method,移动到了Price类中 // step5:Replace Conditional with Polymorphism 对移动后的(Price::getCharge)进行分解 // // #include "stdafx.h" #include "Customer.h" #include <iostream> int _tmain(int argc, _TCHAR* argv[]) { Movie* movie1 = new Movie("赤壁", 1); Movie* movie2 = new Movie("风声", 0); Movie* movie3 = new Movie("建国大业", 0); Rental* rental1 = new Rental(movie1, 1); Rental* rental2 = new Rental(movie2, 2); Rental* rental3 = new Rental(movie3, 3); Customer* custom = new Customer("ZZ"); custom->AddRental(rental1); custom->AddRental(rental2); custom->AddRental(rental3); std::cout << custom->Statement() << std::endl; delete movie1; delete movie2; delete movie3; delete rental1; delete rental2; delete rental3; delete custom; return 0; }