diff --git a/modules/smithy/model_trade.go b/modules/smithy/model_trade.go index 94c8c599c..4b9347fe4 100644 --- a/modules/smithy/model_trade.go +++ b/modules/smithy/model_trade.go @@ -109,12 +109,21 @@ func (s *modelTrade) addCustomer(uid string, num int32) (*pb.DBCustomer, error) // 移除顾客 func (s *modelTrade) removeCustomer(cus *pb.DBCustomer, customerId int32) *pb.DBCustomer { - for i, v := range cus.Customers { - if v.CustomerId == customerId { - cus.Customers = append(cus.Customers[:i], cus.Customers[i+1:]...) - i-- + var ( + temp []*pb.CustomerInfo = make([]*pb.CustomerInfo, 0) + ) + for _, v := range cus.Customers { + if v.CustomerId != customerId { + temp = append(temp, v) } } + cus.Customers = temp + // for i, v := range cus.Customers { + // if v.CustomerId == customerId { + // cus.Customers = append(cus.Customers[:i], cus.Customers[i+1:]...) + // i-- + // } + // } return cus }