package repository

import (
	"be.ems/src/framework/database/db"
	"be.ems/src/framework/logger"
	"be.ems/src/modules/system/model"
)

// NewSysRoleMenu 实例化数据层
var NewSysRoleMenu = &SysRoleMenu{}

// SysRoleMenu 角色与菜单关联表 数据层处理
type SysRoleMenu struct{}

// ExistRoleByMenuId 存在角色使用数量By菜单
func (r SysRoleMenu) ExistRoleByMenuId(menuId int64) int64 {
	if menuId <= 0 {
		return 0
	}
	tx := db.DB("").Model(&model.SysRoleMenu{})
	tx = tx.Where("menu_id = ?", menuId)
	// 查询数据
	var count int64 = 0
	if err := tx.Count(&count).Error; err != nil {
		logger.Errorf("query find err => %v", err.Error())
		return count
	}
	return count
}

// DeleteByRoleIds 批量删除关联By角色
func (r SysRoleMenu) DeleteByRoleIds(roleIds []int64) int64 {
	if len(roleIds) <= 0 {
		return 0
	}
	tx := db.DB("").Where("role_id in ?", roleIds)
	// 执行删除
	if err := tx.Delete(&model.SysRoleMenu{}).Error; err != nil {
		logger.Errorf("delete err => %v", err.Error())
		return 0
	}
	return tx.RowsAffected
}

// DeleteByMenuIds 批量删除关联By菜单
func (r SysRoleMenu) DeleteByMenuIds(menuIds []int64) int64 {
	if len(menuIds) <= 0 {
		return 0
	}
	tx := db.DB("").Where("menu_id in ?", menuIds)
	// 执行删除
	if err := tx.Delete(&model.SysRoleMenu{}).Error; err != nil {
		logger.Errorf("delete err => %v", err.Error())
		return 0
	}
	return tx.RowsAffected
}

// BatchInsert 批量新增信息
func (r SysRoleMenu) BatchInsert(sysRoleMenus []model.SysRoleMenu) int64 {
	if len(sysRoleMenus) <= 0 {
		return 0
	}
	// 执行批量删除
	tx := db.DB("").CreateInBatches(sysRoleMenus, 500)
	if err := tx.Error; err != nil {
		logger.Errorf("delete batch err => %v", err.Error())
		return 0
	}
	return tx.RowsAffected
}