Как изменить существующий класс с классом, содержащим 2 метода?

Question

Как создать класс Inventory, который имеет 2 метода (добавить и вычесть), который добавляет или вычитает сумму в класс Item и возвращает новый элемент с правильно настроенным счетчиком?

Язык: Scala

Что код должен делать:

val Shirts = Item("Hanes", 12) 

val Inv = new Inventory 

scala> Inv.subtract(5, Shirts) 

output: Item(Hanes,7) 

scala> Inv.add(5, Shirts) 

output: Item(Hanes,17) 

Код у меня есть:

case class Item(val brand: String, val count: Int) 

class Inventory { 
    def add(amount:Int):Int={ 
    count+=amount 
    } 

    def subtract(amount:Int):Int={ 
    count-= amount 
    } 
} 

Примечание: Я не могу понять, как изменить класс Item с классом Inventory, содержащим 2 метода. Любая помощь приветствуется.

Answer 1

Это должно сделать работу:

class Inventory { 
    def add(amount:Int, item: Item): Item = { 
    item.copy(count = item.count+amount)  
    } 

    def subtract(amount:Int, item: Item): Item = { 
    item.copy(count = item.count-amount) 
    } 
} 

EDIT: за свой комментарий, чтобы добавить проверку, что сумма> 0 (если сумма < = 0, я просто оставить этот пункт без изменений) :

class Inventory { 
    def add(amount:Int, item: Item): Item = { 
    if (amount > 0) item.copy(count = item.count+amount) else item 
    } 

    def subtract(amount:Int, item: Item): Item = { 
    if (amount > 0) item.copy(count = item.count-amount) else item 
    } 
}