作为一名计算机专业的毕业生,你在面试中遇到了这样一道业务上BUG的定位和修复。假设你正在开发一个在线书店系统,用户可以在系统中添加书籍到购物车。系统要求在添加书籍时,用户已经将同一本书添加到购物车中,则系统应该增加该书籍的数量而不是添加新的条目。是一个简化版的代码片段,存在一个BUG,请找出这个BUG并解释如何修复它。
python
class ShoppingCart:
def __init__(self):
self.items = {}
def add_item(self, item_id, quantity):
if item_id in self.items:
self.items[item_id] += quantity
else:
self.items[item_id] = quantity
# 示例使用
cart = ShoppingCart()
cart.add_item(1, 1)
print(cart.items) # 应该输出:{1: 1}
cart.add_item(1, 1)
print(cart.items) # 应该输出:{1: 2}
BUG定位与修复
我们需要理解代码的逻辑。这段代码定义了一个`ShoppingCart`类,包含一个字典`items`用于存储购物车中的商品和对应数量。`add_item`方法用于向购物车中添加商品,商品已存在,则增加其数量;商品不存在,则添加新的商品条目。
在这个例子中,BUG并不明显。代码似乎按照预期工作。我们注意到在添加商品时,没有对`quantity`参数进行检查,它可以是任何数字,包括负数。`quantity`是一个负数,这可能会导致一个逻辑错误,因为我们应该不会减少购物车中的商品数量。
分析BUG
假设我们调用`cart.add_item(1, -1)`,按照当前的代码逻辑,这将导致`cart.items`变为`{1: 0}`,这是不正确的,因为我们应该允许用户减少商品数量,而不是直接将其从购物车中移除。
修复BUG
为了修复这个BUG,我们需要对`add_item`方法进行修改,确保`quantity`是一个非负数,尝试减少商品数量,我们可以将其设置为一个合理的默认值,0。
是修复后的代码:
python
class ShoppingCart:
def __init__(self):
self.items = {}
def add_item(self, item_id, quantity):
if quantity < 0:
print("Quantity cannot be negative.")
return
if item_id in self.items:
self.items[item_id] += quantity
if self.items[item_id] <= 0:
del self.items[item_id]
else:
self.items[item_id] = quantity
# 示例使用
cart = ShoppingCart()
cart.add_item(1, 1)
print(cart.items) # 应该输出:{1: 1}
cart.add_item(1, 1)
print(cart.items) # 应该输出:{1: 2}
cart.add_item(1, -1)
print(cart.items) # 应该输出:{1: 1}
cart.add_item(1, -2)
print(cart.items) # 应该输出:{}
在这个修复版本中,我们添加了一个检查来确保`quantity`不是负数,商品的数量减少到0或我们将从购物车中删除该商品。这样,我们就确保了购物车的逻辑是一致的,可以正确处理负数的数量情况。
还没有评论呢,快来抢沙发~