Medical_ERP/services/enterprise.go

package services

import (
	"Medical_ERP/common/actions"
	cDto "Medical_ERP/common/dto"
	"Medical_ERP/common/global"
	db "Medical_ERP/common/initialize"
	"errors"
	"github.com/beego/beego/v2/core/logs"

	"gorm.io/gorm"

	"Medical_ERP/dto"
	"Medical_ERP/models"
)

type Enterprise struct {
}

// GetPage 获取Enterprise列表
func (e *Enterprise) GetPage(c *dto.EnterprisePageReq, list *[]models.Enterprise, count *int64, p *actions.DataPermission) error {
	var err error
	var data models.Enterprise

	err = db.DB.Model(&data).
		Scopes(
			cDto.MakeCondition(c.GetNeedSearch()),
			cDto.Paginate(c.GetPageSize(), c.GetPageIndex()),
			actions.Permission(data.TableName(), p),
		).
		Find(list).Limit(-1).Offset(-1).
		Count(count).Error
	if err != nil {
		logs.Error("db error: %s ", err)
		return global.GetFailedErr
	}
	return nil
}

// Get 获取Enterprise对象
func (e *Enterprise) Get(d *dto.EnterpriseGetReq, productModel *models.Enterprise, p *actions.DataPermission) error {
	err := db.DB.
		Scopes(actions.Permission(productModel.TableName(), p)).
		First(productModel, d.GetId()).Error

	if err != nil {
		logs.Error("db error: %s", err)
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return global.GetNotFoundOrNoPermissionErr
		}
		return global.GetFailedErr
	}
	return nil
}

// Insert 创建Enterprise对象
func (e *Enterprise) Insert(c *dto.EnterpriseInsertReq) error {
	var err error
	var data models.Enterprise
	err = db.DB.Where("name = ? AND dept_id = ?", c.Name, c.DeptId).First(&data).Error
	if err != nil {
		// 没有则创建
		if errors.Is(err, gorm.ErrRecordNotFound) {
			c.Generate(&data)
			err = db.DB.Create(&data).Error
			if err != nil {
				logs.Error("db error: %s", err)
				return global.CreateFailedErr
			}
			c.Id = data.Id
			return nil
		}
		logs.Error("db error: %s", err)
		return global.CreateFailedErr
	}
	c.Id = data.Id
	return nil
}

// Update 修改Enterprise对象
func (e *Enterprise) Update(c *dto.EnterpriseUpdateReq) error {
	var productModel = models.Enterprise{}
	err := db.DB.First(&productModel, c.GetId()).Error
	if err != nil {
		logs.Error("db error: %s", err)
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return global.UpdateNotFoundOrNoPermissionErr
		}
		return global.UpdateFailedErr
	}

	c.Generate(&productModel)

	err = db.DB.Save(&productModel).Error
	if err != nil {
		logs.Error("db error: %s", err)
		return global.UpdateFailedErr
	}
	return nil
}

// Remove 删除Enterprise
func (e *Enterprise) Remove(d *dto.EnterpriseDeleteReq) error {
	var err error
	var data models.Enterprise

	var Count int64
	err = db.DB.Table(models.GetMedicineInfoTableName(d.DeptId)).Where("unit_id = ?", d.GetId()).Count(&Count).Error
	if err != nil {
		logs.Error("Delete error: %s", err)
		return global.DeleteFailedErr
	}
	if Count > 0 {
		return errors.New("该生产企业已被关联，禁止删除")
	}

	result := db.DB.Model(&data).Delete(&data, d.GetId())
	if err = result.Error; err != nil {
		logs.Error("Delete error: %s", err)
		return global.DeleteFailedErr
	}
	if result.RowsAffected == 0 {
		return global.DeleteNotFoundOrNoPermissionErr
	}
	return nil
}