完善接口方法

Author: enjoyCoding666

src/main/java/demo/invoice/Invoice.java

@@ -21,7 +21,7 @@ public class Invoice {
 	private BigDecimal totalAmount;
 
 	/**
-	 * 火车票的专属字段
+	 * 火车票的专属字段，是放在主类里面，还是单独拆分一个新的类好？
 	 */
 	private String trainNum;
 	private String passengerName;
@@ -31,4 +31,5 @@ public class Invoice {
 	private String electronicTicketNum;
 	private String customerIdentityNum;
 
+
 }

src/main/java/demo/invoice/InvoiceConstant.java

@@ -0,0 +1,9 @@
+package demo.invoice;
+
+public class InvoiceConstant {
+
+	public static final int INVOICE_TRAIN=8;
+
+	public static final int INVOICE_AIR=9;
+
+}

src/main/java/demo/invoice/InvoiceOperation.java

@@ -0,0 +1,37 @@
+package demo.invoice;
+
+public class InvoiceOperation {
+
+	/**
+	 * 计算发票类型。
+	 * 根据发票代码、发票号码计算得到发票类型。
+	 * @param invoiceNo
+	 * @param invoiceCode
+	 * @return
+	 */
+	public static int calculateInvoiceType(String invoiceNo,String invoiceCode){
+		int invoiceType=(int)(Math.random()*2+8);
+		System.out.println("发票类型为："+invoiceType);
+		return invoiceType;
+	}
+
+	/**
+	 * 根据不同的发票类型生成不同的发票服务
+	 * @param invoice
+	 */
+	public static void updateInvoice(Invoice invoice){
+		String invoiceNo=invoice.getInvoiceNo();
+		String invoiceCode=invoice.getInvoiceCode();
+		int invoiceType=calculateInvoiceType(invoiceNo,invoiceCode);
+		InvoiceService invoiceService;
+		if(invoiceType==InvoiceConstant.INVOICE_TRAIN) {
+			invoiceService=new InvoiceServiceAir();
+		}else if(invoiceType==InvoiceConstant.INVOICE_AIR){
+			invoiceService=new InvoiceServiceTrain();
+		}else {
+			invoiceService=new InvoiceServiceOther();
+		}
+		invoiceService.updateInvoice();
+	}
+
+}

src/main/java/demo/invoice/InvoicePatternTest.java

@@ -5,24 +5,10 @@
  */
 public class InvoicePatternTest {
 	public static void main(String[] args) {
-		int invoiceType=9;
-		updateInvoice(invoiceType);
-
-	}
-
-	public static void updateInvoice(int invoiceType){
-		Invoice invoice=new Invoice();
-		//根据不同的发票类型生成不同的发票服务
-		InvoiceService invoiceService=null;
-		if(invoiceType==8) {
-			 invoiceService=new InvoiceServiceAir();
-		}else if(invoiceType==9){
-			 invoiceService=new InvoiceServiceTrain();
-		}else {
-			invoiceService=new InvoiceServiceOther();
-		}
-		invoiceService.updateInvoice();
-
+		String invoiceNo="invoiceNo";
+		String invoiceCode="invoiceCode";
+		Invoice invoice=Invoice.builder().invoiceCode(invoiceCode).invoiceNo(invoiceNo).build();
+		InvoiceOperation.updateInvoice(invoice);
 	}
 
 }

src/main/java/demo/invoice/InvoiceService.java

@@ -1,8 +1,11 @@
 package demo.invoice;
 
-public interface InvoiceService {
 
-	Invoice queryInvoice(String invoiceNo,String invoiceCode);
+/**
+ * 变化的部分，使用接口实现。相同的部分，使用继承复用。
+ */
+public interface InvoiceService {
+	Invoice queryInvoice(Invoice invoice);
 
 	int saveInvoice();
 

src/main/java/demo/invoice/InvoiceServiceAir.java

@@ -1,10 +1,14 @@
 package demo.invoice;
 
-public class InvoiceServiceAir implements InvoiceService{
-
+public class InvoiceServiceAir extends InvoiceOperation implements InvoiceService {
 
+	/**
+	 * 此处的参数类型，应该是Invoice还是InvoiceAir?
+	 * @param invoice
+	 * @return
+	 */
 	@Override
-	public Invoice queryInvoice(String invoiceNo, String invoiceCode) {
+	public Invoice queryInvoice(Invoice invoice) {
 		System.out.println("查询飞机票");
 		return null;
 	}

src/main/java/demo/invoice/InvoiceServiceOther.java

@@ -1,8 +1,8 @@
 package demo.invoice;
 
-public class InvoiceServiceOther implements InvoiceService {
+public class InvoiceServiceOther extends InvoiceOperation implements InvoiceService {
 	@Override
-	public Invoice queryInvoice(String invoiceNo, String invoiceCode) {
+	public Invoice queryInvoice(Invoice invoice) {
 		System.out.println("查询其他发票");
 		return null;
 	}

src/main/java/demo/invoice/InvoiceServiceTrain.java

@@ -1,8 +1,8 @@
 package demo.invoice;
 
-public class InvoiceServiceTrain implements InvoiceService {
+public class InvoiceServiceTrain  extends InvoiceOperation implements InvoiceService {
 	@Override
-	public Invoice queryInvoice(String invoiceNo, String invoiceCode) {
+	public Invoice queryInvoice(Invoice invoice) {
 		System.out.println("查询火车票");
 		return null;
 	}