优化全局变量

modules/core/dao/user_role_dao.go

@@ -1,20 +1,20 @@
 package dao
 
 import (
-	"cutego/modules"
 	"cutego/modules/core/api/v1/request"
-	"cutego/modules/core/dataobject"
+	"cutego/modules/core/entity"
 	"cutego/pkg/common"
+	"cutego/refs"
 )
 
 type UserRoleDao struct {
 }
 
 // BatchInsert 批量新增用户角色信息
-func (d UserRoleDao) BatchInsert(roles []dataobject.SysUserRole) {
-	session := modules.SqlDB.NewSession()
+func (d UserRoleDao) BatchInsert(roles []entity.SysUserRole) {
+	session := refs.SqlDB.NewSession()
 	session.Begin()
-	_, err := session.Table(dataobject.SysUserRole{}.TableName()).Insert(&roles)
+	_, err := session.Table(entity.SysUserRole{}.TableName()).Insert(&roles)
 	if err != nil {
 		common.ErrorLog(err)
 		session.Rollback()
@@ -25,10 +25,10 @@ func (d UserRoleDao) BatchInsert(roles []dataobject.SysUserRole) {
 
 // Delete 删除用户和角色关系
 func (d UserRoleDao) Delete(id int64) {
-	role := dataobject.SysUserRole{
+	role := entity.SysUserRole{
 		UserId: id,
 	}
-	session := modules.SqlDB.NewSession()
+	session := refs.SqlDB.NewSession()
 	session.Begin()
 	_, err := session.Delete(&role)
 	if err != nil {
@@ -40,8 +40,8 @@ func (d UserRoleDao) Delete(id int64) {
 }
 
 // DeleteAuthUser 取消用户授权
-func (d UserRoleDao) DeleteAuthUser(role dataobject.SysUserRole) int64 {
-	session := modules.SqlDB.NewSession()
+func (d UserRoleDao) DeleteAuthUser(role entity.SysUserRole) int64 {
+	session := refs.SqlDB.NewSession()
 	session.Begin()
 	i, err := session.Delete(&role)
 	if err != nil {
@@ -56,15 +56,15 @@ func (d UserRoleDao) DeleteAuthUser(role dataobject.SysUserRole) int64 {
 // InsertAuthUsers 批量选择用户授权
 func (d UserRoleDao) InsertAuthUsers(body request.UserRoleBody) int64 {
 	ids := body.UserIds
-	roles := make([]dataobject.SysUserRole, 0)
+	roles := make([]entity.SysUserRole, 0)
 	for i := 0; i < len(ids); i++ {
-		role := dataobject.SysUserRole{
+		role := entity.SysUserRole{
 			RoleId: body.RoleId,
 			UserId: ids[i],
 		}
 		roles = append(roles, role)
 	}
-	session := modules.SqlDB.NewSession()
+	session := refs.SqlDB.NewSession()
 	session.Begin()
 	insert, err := session.Insert(&roles)
 	if err != nil {