1package observer
2
3type Subject interface {
4    Register(o Observer)
5    Deregister(o Observer) error
6    NotifyObservers()
7}

1package observer
2
3type Observer interface {
4    Update(name, status string)
5    GetID() int
6}

 1package observer
 2
 3import (
 4    "fmt"
 5    "sync"
 6)
 7
 8const (
 9    TimeIsUp  = "time is up"
10    IsEnd     = "is end"
11    NotAtTime = "not at time"
12)
13
14type shirt struct {
15    sync.Mutex
16    customers []Observer
17    status    string
18    name      string
19}
20
21//NewShirt create a shirt, its default status is not at time.
22func NewShirt() *shirt {
23    return &shirt{status: NotAtTime, name: "shirt"}
24}
25
26// Register registers the observer (customer) of this shirt.
27func (s *shirt) Register(o Observer) {
28    s.Lock()
29    defer s.Unlock()
30    s.customers = append(s.customers, o)
31    fmt.Printf("[%s] registered a new customer with ID[%d]\n", s.name, o.GetID())
32}
33
34// Deregister removes the observer (customer) from the customers list.
35// The removed observer (customer) won't get notifications any more.
36func (s *shirt) Deregister(o Observer) error {
37    s.Lock()
38    defer s.Unlock()
39    var index int
40    var found bool
41    id := o.GetID()
42    for i, c := range s.customers {
43        if c.GetID() == id {
44            index = i
45            found = true
46            break
47        }
48    }
49    if !found {
50        return fmt.Errorf("Customer %d not found\n", id)
51    }
52    s.customers = append(s.customers[:index], s.customers[index+1:]...)
53    fmt.Printf("Removed the customer with ID[%d]\n", id)
54    return nil
55}
56
57// NotifyObservers notifies the customers (customers) when the shirt has in status "time is up".
58func (s *shirt) NotifyObservers() {
59    s.Lock()
60    defer s.Unlock()
61    wg := sync.WaitGroup{}
62    for _, c := range s.customers {
63        wg.Add(1)
64        go func(c Observer) {
65            defer wg.Done()
66            c.Update(s.name, s.status)
67        }(c)
68    }
69    wg.Wait()
70    fmt.Println("Finished notify customers")
71}

 1package observer
 2
 3import "fmt"
 4
 5type customer struct {
 6    ID             int
 7    wantItemStatus string
 8}
 9
10// NewCustomer creates a new customer with an ID
11func NewCustomers(id int) *customer {
12    return &customer{ID: id}
13}
14
15// Update function updates the item status of the customer's want.
16func (c *customer) Update(name, status string) {
17    c.wantItemStatus = status
18    fmt.Printf("Update: hi customer %d, the item[%s] you want is [%v] now\n", c.ID, name, c.wantItemStatus)
19}
20
21// GetID returns the ID of the customer.
22func (c customer) GetID() int {
23    return c.ID
24}

 1package observer
 2
 3import (
 4    "fmt"
 5    "testing"
 6)
 7
 8// UpdateShirtStatusForTest updates the status of a shirt.
 9// This function is for testing only.
10func (s *shirt) UpdateShirtStatusForTest(status string) {
11    if s.status != status {
12        fmt.Printf("Update status of the [shirt]: previous is [%s], current is [%s]\n", s.status, status)
13        s.status = status
14    }
15}
16func TestRushToBuy(t *testing.T) {
17    c1 := NewCustomers(1)
18    c2 := NewCustomers(2)
19    c3 := NewCustomers(3)
20
21    s := NewShirt()
22    s.Register(c1)
23    s.Register(c2)
24    s.Register(c3)
25
26    s.UpdateShirtStatusForTest(TimeIsUp)
27    s.NotifyObservers()
28}

 1=== RUN   TestRushToBuy
 2[shirt] registered a new customer with ID[1]
 3[shirt] registered a new customer with ID[2]
 4[shirt] registered a new customer with ID[3]
 5Update status of the [shirt]: previous is [not at time], current is [time is up]
 6Update: hi customer 1, the item[shirt] you want is [time is up] now
 7Update: hi customer 2, the item[shirt] you want is [time is up] now
 8Update: hi customer 3, the item[shirt] you want is [time is up] now
 9Finished notify customers
10--- PASS: TestRushToBuy (0.00s)
11PASS