Bus Ride, code snippet 04

busride_code_snippet_04.go

// Bus carries Passengers from A to B if they have a valid bus ticket.
type Bus struct {
	name string
	passengers map[string]Passenger
}

// Add adds a single Passenger to the Bus. For brevity, we don't care too much about accidentally adding the same Passenger more than once.
func (b *Bus) Add(p Passenger) {
	if b.passengers == nil {
		b.passengers = make(map[string]Passenger)
	}
	b.passengers[p.SSN] = p
}

// VisitPassengers calls function visitor for each Passenger on the Bus.
func (b *Bus) VisitPassengers(visitor func(Passenger)) {
	for _, p := range b.passengers {
		visitor(p)
	}
}