Golang生成器模式

FredricZhu · · 1178 次点击 · · 开始浏览

这是一个创建于的文章，其中的信息可能已经有所发展或是发生改变。

builder.go

// builder
package builder

type VehicleProduct struct {
    Wheels    int
    Seats     int
    Structure string
}

type BuildProcess interface {
    SetWheels() BuildProcess
    SetSeats() BuildProcess
    SetStructure() BuildProcess
    GetVehicle() VehicleProduct
}

//MenufacturingDirector Begin
type ManufacturingDiretcor struct {
    builder BuildProcess
}

func (f *ManufacturingDiretcor) Construct() {
    f.builder.SetWheels().SetSeats().SetStructure()
}

func (f *ManufacturingDiretcor) SetBuilder(b BuildProcess) {
    f.builder = b
}

//MenufacturingDirector End

//Car Builder Start implements BuildProcess
type CarBuilder struct {
    v VehicleProduct
}

func (c *CarBuilder) SetWheels() BuildProcess {
    c.v.Wheels = 4
    return c
}

func (c *CarBuilder) SetSeats() BuildProcess {
    c.v.Seats = 5
    return c
}

func (c *CarBuilder) SetStructure() BuildProcess {
    c.v.Structure = "Car"
    return c
}

func (c *CarBuilder) GetVehicle() VehicleProduct {
    return c.v
}

//Car Builder End

//Bike Builder Start implements BuildProcess
type BikeBuilder struct {
    v VehicleProduct
}

func (b *BikeBuilder) SetWheels() BuildProcess {
    b.v.Wheels = 2
    return b
}

func (b *BikeBuilder) SetSeats() BuildProcess {
    b.v.Seats = 2
    return b
}

func (b *BikeBuilder) SetStructure() BuildProcess {
    b.v.Structure = "Motorbike"
    return b
}

func (b *BikeBuilder) GetVehicle() VehicleProduct {
    return b.v
}

//Bike Builder End

builder_test.go

package builder

import (
    "testing"
)

func TestBuilderPattern(t *testing.T) {
    manufactureComplex := ManufacturingDiretcor{}

    carBuilder := &CarBuilder{}
    manufactureComplex.SetBuilder(carBuilder)
    manufactureComplex.Construct()
    car := carBuilder.GetVehicle()

    if car.Wheels != 4 {
        t.Errorf("Wheels on a car must be 4, but it is %d \n", car.Wheels)
    }

    if car.Structure != "Car" {
        t.Errorf("Structure Of a car must be 'Car', but it is %v \n", car.Structure)
    }

    if car.Seats != 5 {
        t.Errorf("Seats on a car must be 5, but it is %d \n", car.Seats)
    }

    bikeBuilder := &BikeBuilder{}
    manufactureComplex.SetBuilder(bikeBuilder)
    manufactureComplex.Construct()
    motoBike := bikeBuilder.GetVehicle()

    motoBike.Seats = 1
    if motoBike.Wheels != 2 {
        t.Errorf("Wheels on a motobike must be 2 but found %d \n", motoBike.Wheels)
    }

    if motoBike.Structure != "Motorbike" {
        t.Errorf("Structure of a motobike must be 'Motorbike' but found %v \n", motoBike.Structure)
    }
}

程序输出如下，