lc_garage/server/service/parking/booth.go

package parking

import (
	"errors"
	"server/dao"
	"server/global"
	"server/model/parking/request"
)

type boothService struct{}

func NewBoothService() *boothService {
	return &boothService{}
}

// CreateBooth 创建岗亭
func (s *boothService) CreateBooth(req request.BoothCreate) error {
	// 检查岗亭编码是否已存在
	var existingBooth dao.Booth
	result := global.GVA_DB.Where("booth_code = ?", req.BoothCode).First(&existingBooth)
	if result.RowsAffected > 0 {
		return errors.New("岗亭编码已存在")
	}

	booth := dao.Booth{
		BoothCode:   req.BoothCode,
		BoothName:   req.BoothName,
		IPAddress:   req.IPAddress,
		Description: req.Description,
	}

	return global.GVA_DB.Create(&booth).Error
}

// UpdateBooth 更新岗亭
func (s *boothService) UpdateBooth(req request.BoothUpdate) error {
	var booth dao.Booth
	result := global.GVA_DB.First(&booth, req.ID)
	if result.RowsAffected == 0 {
		return errors.New("岗亭不存在")
	}

	// 检查岗亭编码是否被其他岗亭使用
	var existingBooth dao.Booth
	result = global.GVA_DB.Where("booth_code = ? AND id != ?", req.BoothCode, req.ID).First(&existingBooth)
	if result.RowsAffected > 0 {
		return errors.New("岗亭编码已被其他岗亭使用")
	}

	booth.BoothCode = req.BoothCode
	booth.BoothName = req.BoothName
	booth.IPAddress = req.IPAddress
	booth.Description = req.Description

	return global.GVA_DB.Save(&booth).Error
}

// GetBoothByID 根据ID获取岗亭
func (s *boothService) GetBoothByID(id uint) (dao.Booth, error) {
	var booth dao.Booth
	result := global.GVA_DB.First(&booth, id)
	if result.RowsAffected == 0 {
		return dao.Booth{}, errors.New("岗亭不存在")
	}
	return booth, nil
}

// GetBoothByCode 根据编码获取岗亭
func (s *boothService) GetBoothByCode(code string) (dao.Booth, error) {
	var booth dao.Booth
	result := global.GVA_DB.Where("booth_code = ?", code).First(&booth)
	if result.RowsAffected == 0 {
		return dao.Booth{}, errors.New("岗亭不存在")
	}
	return booth, nil
}

// ListBooths 获取岗亭列表
func (s *boothService) ListBooths(req request.BoothQuery) (int64, []dao.Booth, error) {
	var booths []dao.Booth
	var total int64

	query := global.GVA_DB.Model(&dao.Booth{})

	if req.BoothCode != "" {
		query = query.Where("booth_code LIKE ?", "%"+req.BoothCode+"%")
	}

	if req.BoothName != "" {
		query = query.Where("booth_name LIKE ?", "%"+req.BoothName+"%")
	}

	// 计算总数
	query.Count(&total)

	// 分页查询
	offset := (req.Page - 1) * req.PageSize
	result := query.Offset(offset).Limit(req.PageSize).Find(&booths)
	if result.Error != nil {
		return 0, nil, result.Error
	}

	return total, booths, nil
}

// DeleteBooth 删除岗亭
func (s *boothService) DeleteBooth(id uint) error {
	result := global.GVA_DB.Delete(&dao.Booth{}, id)
	if result.RowsAffected == 0 {
		return errors.New("岗亭不存在")
	}
	return result.Error
}