1. package stratege
    3. import (
    4.     "fmt"
    5.     "testing"
    6. )
    8. type travel interface {
    9.     goTravel(name string)
    10. }
    12. type travelByCar struct {
    13. }
    15. func (t *travelByCar) goTravel(name string) {
    16.     fmt.Println(name + " travel by car")
    17. }
    19. type travelByPlane struct {
    20. }
    22. func (t *travelByPlane) goTravel(name string) {
    23.     fmt.Println(name + " travel by plane")
    24. }
    26. type TravelContext struct {
    27.     name string
    28.     travel
    29. }
    31. func NewTravelContext(name string, travelType string) *TravelContext {
    32.     t := &TravelContext{
    33.         name: name,
    34.     }
    35.     switch travelType {
    36.     case "轿车":
    37.         t.travel = &travelByCar{}
    38.     case "飞机":
    39.         t.travel = &travelByPlane{}
    40.     default:
    41.         t.travel = &travelByCar{}
    42.     }
    43.     return t
    44. }
    46. func (t *TravelContext) GoTravel() {
    47.     t.travel.goTravel(t.name)
    48. }
    50. func TestMain(t *testing.T) {
    51.     p1 := NewTravelContext("小明", "轿车")
    52.     p2 := NewTravelContext("小明", "飞机")
    53.     p1.GoTravel()
    54.     p2.GoTravel()
    55. }