Go語言開發點餐系統中的菜色供應商管理功能詳解

// 定义供应商模型
type Supplier struct {
    gorm.Model
    Name        string
    Contact     string
    ContactInfo string
    Address     string
}

// 创建供应商
func CreateSupplier(name, contact, contactInfo, address string) (*Supplier, error) {
    supplier := &Supplier{
        Name:        name,
        Contact:     contact,
        ContactInfo: contactInfo,
        Address:     address,
    }
    if err := db.Create(supplier).Error; err != nil {
        return nil, err
    }
    return supplier, nil
}

// 根据ID获取供应商
func GetSupplierByID(id uint) (*Supplier, error) {
    supplier := &Supplier{}
    if err := db.First(supplier, id).Error; err != nil {
        return nil, err
    }
    return supplier, nil
}

// 根据供应商名称查询供应商
func GetSupplierByName(name string) ([]*Supplier, error) {
    suppliers := []*Supplier{}
    if err := db.Where("name = ?", name).Find(&suppliers).Error; err != nil {
        return nil, err
    }
    return suppliers, nil
}

// 更新供应商信息
func UpdateSupplier(supplier *Supplier, name, contact, contactInfo, address string) error {
    supplier.Name = name
    supplier.Contact = contact
    supplier.ContactInfo = contactInfo
    supplier.Address = address
    if err := db.Save(supplier).Error; err != nil {
        return err
    }
    return nil
}

// 删除供应商
func DeleteSupplier(supplier *Supplier) error {
    if err := db.Delete(supplier).Error; err != nil {
        return err
    }
    return nil
}