策略模式:使用参数对象传递参数

2023-08-16 19:43:42 浏览数 (2)

以下是一个使用参数对象的策略模式的例子,该例子中,我们将创建两种不同的支付策略,它们需要不同的参数:

代码语言:javascript复制
package main


import "fmt"


// Strategy Interface
type PaymentStrategy interface {
  Pay(data interface{}) string
}


// Concrete Strategy
type PayPalStrategy struct{}


func (p *PayPalStrategy) Pay(data interface{}) string {
  payPalData, ok := data.(*PayPalData)
  if !ok {
    return "Invalid data for PayPalStrategy"
  }
  return fmt.Sprintf("Paid %d using PayPal. PayPal user: %s", payPalData.Amount, payPalData.User)
}


// Concrete Strategy
type CreditCardStrategy struct{}


func (c *CreditCardStrategy) Pay(data interface{}) string {
  creditCardData, ok := data.(*CreditCardData)
  if !ok {
    return "Invalid data for CreditCardStrategy"
  }
  return fmt.Sprintf("Paid %d using Credit Card. Card Number: %s", creditCardData.Amount, creditCardData.CardNumber)
}


// Context
type ShoppingCart struct {
  strategy PaymentStrategy
  data     interface{}
}


func (s *ShoppingCart) checkout() string {
  return s.strategy.Pay(s.data)
}


// Parameter objects for different strategies
type PayPalData struct {
  Amount int
  User   string
}


type CreditCardData struct {
  Amount     int
  CardNumber string
}


func main() {
  data := &PayPalData{Amount: 200, User: "user@example.com"}
  cart := &ShoppingCart{strategy: &PayPalStrategy{}, data: data}
  fmt.Println(cart.checkout()) // Paid 200 using PayPal. PayPal user: user@example.com


  data = &CreditCardData{Amount: 300, CardNumber: "1234-5678-9012-3456"}
  cart = &ShoppingCart{strategy: &CreditCardStrategy{}, data: data}
  fmt.Println(cart.checkout()) // Paid 300 using Credit Card. Card Number: 1234-5678-9012-3456
}

在这个例子中,我们创建了两个不同的参数对象(PayPalData 和 CreditCardData),分别对应两种不同的支付策略。这两个参数对象都有一个 Amount 字段,但是其他的字段则根据支付方式的需要来定。我们的 ShoppingCart 需要一个策略和一个对应的参数对象。在运行时,我们将参数对象作为一个 interface{} 类型的值传递给 Pay 方法,然后在 Pay 方法中将其转换为正确的类型。

注意,这只是一种可能的解决方案,具体的解决方案取决于你的应用场景和需求。在一些情况下,可能需要采用其他的方法来处理不同策略需要不同参数的问题。

0 人点赞