forked from steger/pr3-ws202526
112 lines
2.5 KiB
Go
112 lines
2.5 KiB
Go
package main
|
|
|
|
import "fmt"
|
|
|
|
type Category int
|
|
|
|
const (
|
|
Electronics Category = iota
|
|
Groceries
|
|
Clothes
|
|
)
|
|
|
|
func (c Category) String() string {
|
|
switch c {
|
|
case Electronics:
|
|
return "Electronics"
|
|
case Groceries:
|
|
return "Groceries"
|
|
case Clothes:
|
|
return "Clothes"
|
|
default:
|
|
return "Unknown"
|
|
}
|
|
}
|
|
|
|
// Struct for product details
|
|
type Product struct {
|
|
Name string
|
|
Price float64
|
|
Quantity int
|
|
Category Category
|
|
}
|
|
|
|
// Add a new product to the inventory
|
|
func addProduct(inventory *[]Product, name string, price float64, quantity int, category Category) {
|
|
// Check if the product already exists in the inventory
|
|
for _, product := range *inventory {
|
|
if product.Name == name {
|
|
fmt.Println("Product already exists.")
|
|
return
|
|
}
|
|
}
|
|
// Add the new product
|
|
*inventory = append(*inventory, Product{
|
|
Name: name,
|
|
Price: price,
|
|
Quantity: quantity,
|
|
Category: category,
|
|
})
|
|
fmt.Printf("Added product: %s\n", name)
|
|
}
|
|
|
|
// Remove a product from the inventory
|
|
func removeProduct(inventory *[]Product, name string) {
|
|
for i, product := range *inventory {
|
|
if product.Name == name {
|
|
*inventory = append((*inventory)[:i], (*inventory)[i+1:]...)
|
|
fmt.Printf("Removed product: %s\n", name)
|
|
return
|
|
}
|
|
}
|
|
fmt.Println("Product not found.")
|
|
}
|
|
|
|
// Update the quantity of a product
|
|
func updateQuantity(inventory *[]Product, name string, newQuantity int) {
|
|
for i, product := range *inventory {
|
|
if product.Name == name {
|
|
(*inventory)[i].Quantity = newQuantity
|
|
fmt.Printf("Updated quantity for %s: New Quantity = %d\n", name, newQuantity)
|
|
return
|
|
}
|
|
}
|
|
fmt.Println("Product not found.")
|
|
}
|
|
|
|
// Display the inventory
|
|
func displayInventory(inventory []Product) {
|
|
fmt.Println("Inventory:")
|
|
for _, product := range inventory {
|
|
fmt.Printf("%s - %s (Price: $%.2f, Quantity: %d)\n",
|
|
product.Name, product.Category, product.Price, product.Quantity)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
// Initialize inventory
|
|
inventory := []Product{
|
|
{Name: "Laptop", Price: 1000, Quantity: 5, Category: Electronics},
|
|
{Name: "Apples", Price: 2, Quantity: 50, Category: Groceries},
|
|
{Name: "T-shirt", Price: 10, Quantity: 20, Category: Clothes},
|
|
}
|
|
|
|
// Display initial inventory
|
|
displayInventory(inventory)
|
|
|
|
// Add a new product
|
|
addProduct(&inventory, "Phone", 800, 10, Electronics)
|
|
|
|
// Display updated inventory
|
|
displayInventory(inventory)
|
|
|
|
// Update the quantity of an existing product
|
|
updateQuantity(&inventory, "Apples", 30)
|
|
|
|
// Remove a product
|
|
removeProduct(&inventory, "T-shirt")
|
|
|
|
// Display final inventory
|
|
displayInventory(inventory)
|
|
}
|