使用Go和Goroutines构建高可拓展性的并发物流管理系统

type Warehouse struct {
    lock    sync.Mutex
    stock   map[string]int
}

func (w *Warehouse) Take(item string) {
    w.lock.Lock()
    defer w.lock.Unlock()
    w.stock[item]--
}

func (w *Warehouse) Store(item string) {
    w.lock.Lock()
    defer w.lock.Unlock()
    w.stock[item]++
}

type Vehicle struct {
    id      int
    current string
}

func (v *Vehicle) Run(warehouse *Warehouse, orders <-chan string) {
    for target := range orders {
        v.current = target
        time.Sleep(time.Second * 2) // 模拟配送耗时
        warehouse.Take(target)
        v.current = ""
    }
}

func ProcessOrders(orders []string, dispatch chan<- string) {
    for _, order := range orders {
        dispatch <- order
    }
    close(dispatch)
}

func Schedule(warehouse *Warehouse, dispatch <-chan string) {
    for target := range dispatch {
        vehicles := FindAvailableVehicles(warehouse, target)
        for _, vehicle := range vehicles {
            vehicleOrders[vehicle.id] <- target
        }
    }
}

func FindAvailableVehicles(warehouse *Warehouse, target string) []Vehicle {
    var available []Vehicle
    for _, vehicle := range vehicles {
        if vehicle.current == "" {
            available = append(available, vehicle)
        }
    }
    return available
}