|
@@ -1,9 +1,9 @@
|
|
|
package data_manage_permission
|
|
|
|
|
|
import (
|
|
|
+ "eta_gn/eta_api/global"
|
|
|
"eta_gn/eta_api/utils"
|
|
|
"fmt"
|
|
|
- "github.com/beego/beego/v2/client/orm"
|
|
|
"strconv"
|
|
|
"time"
|
|
|
)
|
|
@@ -37,15 +37,25 @@ type EdbClassifyPermission struct {
|
|
|
// @param classifyIdList []int
|
|
|
func SetIsPermissionEdbChartByEdbClassifyIdList(classifyIdList []int, classifyType int) (err error) {
|
|
|
num := len(classifyIdList)
|
|
|
- o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
- if err != nil {
|
|
|
- return
|
|
|
- }
|
|
|
+ //o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ //defer func() {
|
|
|
+ // if err != nil {
|
|
|
+ // _ = o.Rollback()
|
|
|
+ // } else {
|
|
|
+ // _ = o.Commit()
|
|
|
+ // }
|
|
|
+ //}()
|
|
|
+
|
|
|
+ to := global.DmSQL["data"].Begin()
|
|
|
+
|
|
|
defer func() {
|
|
|
if err != nil {
|
|
|
- _ = o.Rollback()
|
|
|
+ _ = to.Rollback()
|
|
|
} else {
|
|
|
- _ = o.Commit()
|
|
|
+ _ = to.Commit()
|
|
|
}
|
|
|
}()
|
|
|
|
|
@@ -62,16 +72,28 @@ func SetIsPermissionEdbChartByEdbClassifyIdList(classifyIdList []int, classifyTy
|
|
|
//}
|
|
|
|
|
|
// 先将所有已经设置了涉密的分类设置为不涉密
|
|
|
+ //sql := `UPDATE edb_classify SET is_join_permission=?,modify_time=now() WHERE is_join_permission = 1 AND classify_type = ?`
|
|
|
+ //_, err = o.Raw(sql, 0, classifyType).Exec()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
sql := `UPDATE edb_classify SET is_join_permission=?,modify_time=now() WHERE is_join_permission = 1 AND classify_type = ?`
|
|
|
- _, err = o.Raw(sql, 0, classifyType).Exec()
|
|
|
+ err = to.Exec(sql, 0, classifyType).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
|
|
|
if num > 0 {
|
|
|
// 将对应的分类设置为涉密
|
|
|
+ //sql = `UPDATE edb_classify SET is_join_permission=?,modify_time=now() WHERE classify_type = ? AND classify_id in (` + utils.GetOrmInReplace(num) + `) `
|
|
|
+ //_, err = o.Raw(sql, 1, classifyType, classifyIdList).Exec()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
sql = `UPDATE edb_classify SET is_join_permission=?,modify_time=now() WHERE classify_type = ? AND classify_id in (` + utils.GetOrmInReplace(num) + `) `
|
|
|
- _, err = o.Raw(sql, 1, classifyType, classifyIdList).Exec()
|
|
|
+ err = to.Exec(sql, 1, classifyType, classifyIdList).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -93,25 +115,42 @@ func SetPermissionByEdbIdList(edbIdList []string, userIdList []int, edbInfoType
|
|
|
if edbNum <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
- if err != nil {
|
|
|
- return
|
|
|
- }
|
|
|
+ //o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ //defer func() {
|
|
|
+ // if err != nil {
|
|
|
+ // _ = o.Rollback()
|
|
|
+ // } else {
|
|
|
+ // _ = o.Commit()
|
|
|
+ // }
|
|
|
+ //}()
|
|
|
+
|
|
|
+ to := global.DmSQL["data"].Begin()
|
|
|
+
|
|
|
defer func() {
|
|
|
if err != nil {
|
|
|
- _ = o.Rollback()
|
|
|
+ _ = to.Rollback()
|
|
|
} else {
|
|
|
- _ = o.Commit()
|
|
|
+ _ = to.Commit()
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
// 获取已经配置的指标权限用户
|
|
|
edbInfoPermissionList := make([]*EdbInfoPermission, 0)
|
|
|
+ //sql := `SELECT * FROM edb_info_permission WHERE edb_info_type = ? AND edb_info_id in (` + utils.GetOrmInReplace(edbNum) + `) `
|
|
|
+ //_, err = o.Raw(sql, edbInfoType, edbIdList).QueryRows(&edbInfoPermissionList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
sql := `SELECT * FROM edb_info_permission WHERE edb_info_type = ? AND edb_info_id in (` + utils.GetOrmInReplace(edbNum) + `) `
|
|
|
- _, err = o.Raw(sql, edbInfoType, edbIdList).QueryRows(&edbInfoPermissionList)
|
|
|
+ err = to.Raw(sql, edbInfoType, edbIdList).Find(&edbInfoPermissionList).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
+
|
|
|
edbInfoPermissionMap := make(map[string]*EdbInfoPermission)
|
|
|
for _, v := range edbInfoPermissionList {
|
|
|
edbInfoPermissionMap[fmt.Sprint(v.EdbInfoId, "_", v.SysUserId)] = v
|
|
@@ -126,8 +165,14 @@ func SetPermissionByEdbIdList(edbIdList []string, userIdList []int, edbInfoType
|
|
|
// 标记指标为不纳入权限管控
|
|
|
isJoinPermission = 0
|
|
|
}
|
|
|
+ //sql = `UPDATE edb_info SET is_join_permission=?,modify_time=now() WHERE edb_info_type = ? AND edb_info_id in (` + utils.GetOrmInReplace(edbNum) + `) `
|
|
|
+ //_, err = o.Raw(sql, isJoinPermission, edbInfoType, edbIdList).Exec()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
sql = `UPDATE edb_info SET is_join_permission=?,modify_time=now() WHERE edb_info_type = ? AND edb_info_id in (` + utils.GetOrmInReplace(edbNum) + `) `
|
|
|
- _, err = o.Raw(sql, isJoinPermission, edbInfoType, edbIdList).Exec()
|
|
|
+ err = to.Raw(sql, isJoinPermission, edbInfoType, edbIdList).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -164,10 +209,16 @@ func SetPermissionByEdbIdList(edbIdList []string, userIdList []int, edbInfoType
|
|
|
|
|
|
// 添加待配置项
|
|
|
if len(addList) > 0 {
|
|
|
- _, err = o.InsertMulti(500, addList)
|
|
|
+ //_, err = o.InsertMulti(500, addList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
+ err = to.CreateInBatches(addList, utils.MultiAddNum).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
+
|
|
|
}
|
|
|
|
|
|
// 移除废弃的配置项
|
|
@@ -180,8 +231,14 @@ func SetPermissionByEdbIdList(edbIdList []string, userIdList []int, edbInfoType
|
|
|
|
|
|
deletePermissionIdNum := len(deletePermissionIdList)
|
|
|
if deletePermissionIdNum > 0 {
|
|
|
+ //sql = "DELETE FROM edb_info_permission WHERE edb_info_permission_id in (" + utils.GetOrmInReplace(deletePermissionIdNum) + ")"
|
|
|
+ //_, err = o.Raw(sql, deletePermissionIdList).Exec()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
sql = "DELETE FROM edb_info_permission WHERE edb_info_permission_id in (" + utils.GetOrmInReplace(deletePermissionIdNum) + ")"
|
|
|
- _, err = o.Raw(sql, deletePermissionIdList).Exec()
|
|
|
+ err = to.Exec(sql, deletePermissionIdList).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -203,22 +260,32 @@ func SetPermissionByEdbClassifyIdList(classifyIdList []int, userIdList []int, cl
|
|
|
if userNum <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
- if err != nil {
|
|
|
- return
|
|
|
- }
|
|
|
+ //o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ //defer func() {
|
|
|
+ // if err != nil {
|
|
|
+ // _ = o.Rollback()
|
|
|
+ // } else {
|
|
|
+ // _ = o.Commit()
|
|
|
+ // }
|
|
|
+ //}()
|
|
|
+
|
|
|
+ to := global.DmSQL["data"].Begin()
|
|
|
+
|
|
|
defer func() {
|
|
|
if err != nil {
|
|
|
- _ = o.Rollback()
|
|
|
+ _ = to.Rollback()
|
|
|
} else {
|
|
|
- _ = o.Commit()
|
|
|
+ _ = to.Commit()
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
// 获取当前选择用户已经配置的指标分类权限
|
|
|
classifyPermissionList := make([]*EdbClassifyPermission, 0)
|
|
|
sql := `SELECT * FROM edb_classify_permission WHERE edb_classify_type = ? AND sys_user_id in (` + utils.GetOrmInReplace(userNum) + `) `
|
|
|
- _, err = o.Raw(sql, classifyType, userIdList).QueryRows(&classifyPermissionList)
|
|
|
+ err = to.Raw(sql, classifyType, userIdList).Find(&classifyPermissionList).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -253,7 +320,12 @@ func SetPermissionByEdbClassifyIdList(classifyIdList []int, userIdList []int, cl
|
|
|
|
|
|
// 添加待配置项
|
|
|
if len(addList) > 0 {
|
|
|
- _, err = o.InsertMulti(500, addList)
|
|
|
+ //_, err = o.InsertMulti(500, addList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
+ err = to.CreateInBatches(addList, utils.MultiAddNum).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -269,8 +341,14 @@ func SetPermissionByEdbClassifyIdList(classifyIdList []int, userIdList []int, cl
|
|
|
|
|
|
deletePermissionIdNum := len(deletePermissionIdList)
|
|
|
if deletePermissionIdNum > 0 {
|
|
|
+ //sql = "DELETE FROM edb_classify_permission WHERE edb_classify_permission_id in (" + utils.GetOrmInReplace(deletePermissionIdNum) + ")"
|
|
|
+ //_, err = o.Raw(sql, deletePermissionIdList).Exec()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
sql = "DELETE FROM edb_classify_permission WHERE edb_classify_permission_id in (" + utils.GetOrmInReplace(deletePermissionIdNum) + ")"
|
|
|
- _, err = o.Raw(sql, deletePermissionIdList).Exec()
|
|
|
+ err = to.Exec(sql, deletePermissionIdList).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -289,10 +367,12 @@ func SetPermissionByEdbClassifyIdList(classifyIdList []int, userIdList []int, cl
|
|
|
// @return edbClassifyIdList []int
|
|
|
// @return err error
|
|
|
func GetPermissionEdbClassifyIdListByUserId(userId int, classifyType int) (edbClassifyIdList []int, err error) {
|
|
|
- o := orm.NewOrmUsingDB("data")
|
|
|
- sql := `SELECT edb_classify_id FROM edb_classify_permission WHERE edb_classify_type = ? AND sys_user_id = ? `
|
|
|
- _, err = o.Raw(sql, classifyType, userId).QueryRows(&edbClassifyIdList)
|
|
|
+ //o := orm.NewOrmUsingDB("data")
|
|
|
+ //sql := `SELECT edb_classify_id FROM edb_classify_permission WHERE edb_classify_type = ? AND sys_user_id = ? `
|
|
|
+ //_, err = o.Raw(sql, classifyType, userId).QueryRows(&edbClassifyIdList)
|
|
|
|
|
|
+ sql := `SELECT edb_classify_id FROM edb_classify_permission WHERE edb_classify_type = ? AND sys_user_id = ? `
|
|
|
+ err = global.DmSQL["data"].Raw(sql, classifyType, userId).Find(&edbClassifyIdList).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -305,9 +385,12 @@ func GetPermissionEdbClassifyIdListByUserId(userId int, classifyType int) (edbCl
|
|
|
// @return userIdList []int
|
|
|
// @return err error
|
|
|
func GetPermissionUserIdListByEdbId(dataId int, edbInfoType int) (userIdList []int, err error) {
|
|
|
- o := orm.NewOrmUsingDB("data")
|
|
|
+ //o := orm.NewOrmUsingDB("data")
|
|
|
+ //sql := `SELECT sys_user_id FROM edb_info_permission WHERE edb_info_type = ? AND edb_info_id= ? `
|
|
|
+ //_, err = o.Raw(sql, edbInfoType, dataId).QueryRows(&userIdList)
|
|
|
+
|
|
|
sql := `SELECT sys_user_id FROM edb_info_permission WHERE edb_info_type = ? AND edb_info_id= ? `
|
|
|
- _, err = o.Raw(sql, edbInfoType, dataId).QueryRows(&userIdList)
|
|
|
+ err = global.DmSQL["data"].Raw(sql, edbInfoType, dataId).Find(&userIdList).Error
|
|
|
|
|
|
return
|
|
|
}
|
|
@@ -321,9 +404,12 @@ func GetPermissionUserIdListByEdbId(dataId int, edbInfoType int) (userIdList []i
|
|
|
// @return userIdList []int
|
|
|
// @return err error
|
|
|
func GetPermissionUserIdListByEdbClassifyId(classifyId int, edbClassifyType int) (userIdList []int, err error) {
|
|
|
- o := orm.NewOrmUsingDB("data")
|
|
|
+ //o := orm.NewOrmUsingDB("data")
|
|
|
+ //sql := `SELECT sys_user_id FROM edb_classify_permission WHERE edb_classify_type = ? AND edb_classify_id= ? `
|
|
|
+ //_, err = o.Raw(sql, edbClassifyType, classifyId).QueryRows(&userIdList)
|
|
|
+
|
|
|
sql := `SELECT sys_user_id FROM edb_classify_permission WHERE edb_classify_type = ? AND edb_classify_id= ? `
|
|
|
- _, err = o.Raw(sql, edbClassifyType, classifyId).QueryRows(&userIdList)
|
|
|
+ err = global.DmSQL["data"].Raw(sql, edbClassifyType, classifyId).Find(&userIdList).Error
|
|
|
|
|
|
return
|
|
|
}
|
|
@@ -338,13 +424,20 @@ func GetPermissionUserIdListByEdbClassifyId(classifyId int, edbClassifyType int)
|
|
|
// @return err error
|
|
|
func GetPermissionEdbIdList(userId, edbInfoId int) (idList []int, err error) {
|
|
|
pars := []interface{}{userId}
|
|
|
- o := orm.NewOrmUsingDB("data")
|
|
|
+ //o := orm.NewOrmUsingDB("data")
|
|
|
+ //sql := `SELECT edb_info_id FROM edb_info_permission WHERE sys_user_id = ? `
|
|
|
+ //if edbInfoId > 0 {
|
|
|
+ // sql += ` AND edb_info_id = ? `
|
|
|
+ // pars = append(pars, edbInfoId)
|
|
|
+ //}
|
|
|
+ //_, err = o.Raw(sql, pars).QueryRows(&idList)
|
|
|
+
|
|
|
sql := `SELECT edb_info_id FROM edb_info_permission WHERE sys_user_id = ? `
|
|
|
if edbInfoId > 0 {
|
|
|
sql += ` AND edb_info_id = ? `
|
|
|
pars = append(pars, edbInfoId)
|
|
|
}
|
|
|
- _, err = o.Raw(sql, pars).QueryRows(&idList)
|
|
|
+ err = global.DmSQL["data"].Raw(sql, pars).Find(&idList).Error
|
|
|
|
|
|
return
|
|
|
}
|
|
@@ -359,14 +452,20 @@ func GetPermissionEdbIdList(userId, edbInfoId int) (idList []int, err error) {
|
|
|
// @return err error
|
|
|
func GetPermissionEdbClassifyIdList(userId, classifyId int) (idList []int, err error) {
|
|
|
pars := []interface{}{userId}
|
|
|
- o := orm.NewOrmUsingDB("data")
|
|
|
+ //o := orm.NewOrmUsingDB("data")
|
|
|
+ //sql := `SELECT edb_classify_id FROM edb_classify_permission WHERE sys_user_id = ? `
|
|
|
+ //if classifyId > 0 {
|
|
|
+ // sql += ` AND edb_classify_id = ? `
|
|
|
+ // pars = append(pars, classifyId)
|
|
|
+ //}
|
|
|
+ //_, err = o.Raw(sql, pars).QueryRows(&idList)
|
|
|
+
|
|
|
sql := `SELECT edb_classify_id FROM edb_classify_permission WHERE sys_user_id = ? `
|
|
|
if classifyId > 0 {
|
|
|
sql += ` AND edb_classify_id = ? `
|
|
|
pars = append(pars, classifyId)
|
|
|
}
|
|
|
- _, err = o.Raw(sql, pars).QueryRows(&idList)
|
|
|
-
|
|
|
+ err = global.DmSQL["data"].Raw(sql, pars).Find(&idList).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -382,21 +481,37 @@ func GetPermissionEdbClassifyIdList(userId, classifyId int) (idList []int, err e
|
|
|
// @param uniqueCode string
|
|
|
// @return err error
|
|
|
func InheritParentClassifyByEdbClassifyId(source, classifyType, classifyId, parentClassifyId int, classifyName, uniqueCode string) (err error) {
|
|
|
- o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
- if err != nil {
|
|
|
- return
|
|
|
- }
|
|
|
+ //o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ //defer func() {
|
|
|
+ // if err != nil {
|
|
|
+ // _ = o.Rollback()
|
|
|
+ // } else {
|
|
|
+ // _ = o.Commit()
|
|
|
+ // }
|
|
|
+ //}()
|
|
|
+
|
|
|
+ to := global.DmSQL["data"].Begin()
|
|
|
+
|
|
|
defer func() {
|
|
|
if err != nil {
|
|
|
- _ = o.Rollback()
|
|
|
+ _ = to.Rollback()
|
|
|
} else {
|
|
|
- _ = o.Commit()
|
|
|
+ _ = to.Commit()
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
// 将对应的分类设置为涉密
|
|
|
+ //sql := `UPDATE edb_classify SET is_join_permission=?,modify_time=now() WHERE classify_type = ? AND classify_id = ? `
|
|
|
+ //_, err = o.Raw(sql, 1, classifyType, classifyId).Exec()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
sql := `UPDATE edb_classify SET is_join_permission=?,modify_time=now() WHERE classify_type = ? AND classify_id = ? `
|
|
|
- _, err = o.Raw(sql, 1, classifyType, classifyId).Exec()
|
|
|
+ err = to.Exec(sql, 1, classifyType, classifyId).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -405,8 +520,11 @@ func InheritParentClassifyByEdbClassifyId(source, classifyType, classifyId, pare
|
|
|
{
|
|
|
// 获取父级未授权的用户记录
|
|
|
var parentRecordItems []*EdbInfoClassifyPermissionNoAuthRecord
|
|
|
+ //sql = `SELECT * FROM edb_info_classify_permission_no_auth_record WHERE classify_id = ? AND edb_classify_type = ? ORDER BY edb_info_classify_permission_no_auth_record_id desc LIMIT ?,? `
|
|
|
+ //_, err = o.Raw(sql, parentClassifyId, source, classifyType).QueryRows(&parentRecordItems)
|
|
|
+
|
|
|
sql = `SELECT * FROM edb_info_classify_permission_no_auth_record WHERE classify_id = ? AND edb_classify_type = ? ORDER BY edb_info_classify_permission_no_auth_record_id desc LIMIT ?,? `
|
|
|
- _, err = o.Raw(sql, parentClassifyId, source, classifyType).QueryRows(&parentRecordItems)
|
|
|
+ err = to.Raw(sql, parentClassifyId, source, classifyType).Find(&parentRecordItems).Error
|
|
|
|
|
|
addNoAuthRecordItems := make([]*EdbInfoClassifyPermissionNoAuthRecord, 0)
|
|
|
for _, v := range parentRecordItems {
|
|
@@ -423,7 +541,12 @@ func InheritParentClassifyByEdbClassifyId(source, classifyType, classifyId, pare
|
|
|
|
|
|
// 添加待配置项
|
|
|
if len(addNoAuthRecordItems) > 0 {
|
|
|
- _, err = o.InsertMulti(500, addNoAuthRecordItems)
|
|
|
+ //_, err = o.InsertMulti(500, addNoAuthRecordItems)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
+ err = to.CreateInBatches(addNoAuthRecordItems, utils.MultiAddNum).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -435,8 +558,14 @@ func InheritParentClassifyByEdbClassifyId(source, classifyType, classifyId, pare
|
|
|
{
|
|
|
// 获取父级分类已经授权的用户
|
|
|
parentClassifyPermissionList := make([]*EdbClassifyPermission, 0)
|
|
|
+ //sql = `SELECT * FROM edb_classify_permission WHERE edb_classify_type = ? AND edb_classify_id = ? `
|
|
|
+ //_, err = o.Raw(sql, classifyType, parentClassifyId).QueryRows(&parentClassifyPermissionList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
sql = `SELECT * FROM edb_classify_permission WHERE edb_classify_type = ? AND edb_classify_id = ? `
|
|
|
- _, err = o.Raw(sql, classifyType, parentClassifyId).QueryRows(&parentClassifyPermissionList)
|
|
|
+ err = to.Raw(sql, classifyType, parentClassifyId).Find(&parentClassifyPermissionList).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -455,7 +584,12 @@ func InheritParentClassifyByEdbClassifyId(source, classifyType, classifyId, pare
|
|
|
}
|
|
|
// 添加待配置项
|
|
|
if len(addList) > 0 {
|
|
|
- _, err = o.InsertMulti(500, addList)
|
|
|
+ //_, err = o.InsertMulti(500, addList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
+ err = to.CreateInBatches(addList, utils.MultiAddNum).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -504,22 +638,38 @@ func AddEdbInfoPermissionNoAuthRecordBySourceAndDataIdList(source, edbInfoType i
|
|
|
if userNum <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
- if err != nil {
|
|
|
- return
|
|
|
- }
|
|
|
+ //o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ //defer func() {
|
|
|
+ // if err != nil {
|
|
|
+ // _ = o.Rollback()
|
|
|
+ // } else {
|
|
|
+ // _ = o.Commit()
|
|
|
+ // }
|
|
|
+ //}()
|
|
|
+
|
|
|
+ to := global.DmSQL["data"].Begin()
|
|
|
+
|
|
|
defer func() {
|
|
|
if err != nil {
|
|
|
- _ = o.Rollback()
|
|
|
+ _ = to.Rollback()
|
|
|
} else {
|
|
|
- _ = o.Commit()
|
|
|
+ _ = to.Commit()
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
// 根据指标获取已经存在的未授权记录
|
|
|
var existList []*EdbInfoPermissionNoAuthRecord
|
|
|
+ //sql := `SELECT * FROM edb_info_permission_no_auth_record WHERE edb_info_type = ? AND edb_info_id in (` + utils.GetOrmInReplace(num) + `)`
|
|
|
+ //_, err = o.Raw(sql, edbInfoType, dataIdList).QueryRows(&existList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
sql := `SELECT * FROM edb_info_permission_no_auth_record WHERE edb_info_type = ? AND edb_info_id in (` + utils.GetOrmInReplace(num) + `)`
|
|
|
- _, err = o.Raw(sql, edbInfoType, dataIdList).QueryRows(&existList)
|
|
|
+ err = to.Raw(sql, edbInfoType, dataIdList).Find(&existList).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -588,7 +738,12 @@ func AddEdbInfoPermissionNoAuthRecordBySourceAndDataIdList(source, edbInfoType i
|
|
|
|
|
|
// 添加消息
|
|
|
if len(addMessageList) > 0 {
|
|
|
- _, err = o.InsertMulti(500, addMessageList)
|
|
|
+ //_, err = o.InsertMulti(500, addMessageList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
+ err = to.CreateInBatches(addMessageList, utils.MultiAddNum).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -596,7 +751,12 @@ func AddEdbInfoPermissionNoAuthRecordBySourceAndDataIdList(source, edbInfoType i
|
|
|
|
|
|
// 添加记录
|
|
|
if len(addRecordList) > 0 {
|
|
|
- _, err = o.InsertMulti(500, addRecordList)
|
|
|
+ //_, err = o.InsertMulti(500, addRecordList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
+ err = to.CreateInBatches(addRecordList, utils.MultiAddNum).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -605,8 +765,11 @@ func AddEdbInfoPermissionNoAuthRecordBySourceAndDataIdList(source, edbInfoType i
|
|
|
// 已经授权了的用户,需要删除未授权记录
|
|
|
authUserIdNum := len(authUserIdList)
|
|
|
if authUserIdNum > 0 {
|
|
|
+ //sql = `DELETE FROM edb_info_permission_no_auth_record WHERE edb_info_type = ? AND sys_user_id in (` + utils.GetOrmInReplace(authUserIdNum) + `) AND edb_info_id in (` + utils.GetOrmInReplace(num) + `)`
|
|
|
+ //_, err = o.Raw(sql, edbInfoType, authUserIdList, dataIdList).Exec()
|
|
|
+
|
|
|
sql = `DELETE FROM edb_info_permission_no_auth_record WHERE edb_info_type = ? AND sys_user_id in (` + utils.GetOrmInReplace(authUserIdNum) + `) AND edb_info_id in (` + utils.GetOrmInReplace(num) + `)`
|
|
|
- _, err = o.Raw(sql, edbInfoType, authUserIdList, dataIdList).Exec()
|
|
|
+ err = to.Exec(sql, edbInfoType, authUserIdList, dataIdList).Error
|
|
|
}
|
|
|
|
|
|
return
|
|
@@ -625,11 +788,13 @@ func DeleteEdbInfoPermissionNoAuthRecordBySourceAndDataIdList(edbInfoType int, d
|
|
|
if num <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- o := orm.NewOrmUsingDB("data")
|
|
|
+ //o := orm.NewOrmUsingDB("data")
|
|
|
+ //
|
|
|
+ //sql := `DELETE FROM edb_info_permission_no_auth_record WHERE edb_info_type = ? AND edb_info_id in (` + utils.GetOrmInReplace(num) + `)`
|
|
|
+ //_, err = o.Raw(sql, edbInfoType, dataIdList).Exec()
|
|
|
|
|
|
sql := `DELETE FROM edb_info_permission_no_auth_record WHERE edb_info_type = ? AND edb_info_id in (` + utils.GetOrmInReplace(num) + `)`
|
|
|
- _, err = o.Raw(sql, edbInfoType, dataIdList).Exec()
|
|
|
-
|
|
|
+ err = global.DmSQL["data"].Exec(sql, edbInfoType, dataIdList).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -645,18 +810,27 @@ func DeleteEdbInfoPermissionNoAuthRecordBySourceAndDataIdList(edbInfoType int, d
|
|
|
// @return items []*DataPermissionNoAuthRecord
|
|
|
// @return err error
|
|
|
func GetEdbInfoDataPermissionNoAuthRecordListByUserId(userId int32, edbInfoType, startSize, pageSize int) (total int, items []*DataPermissionNoAuthRecord, err error) {
|
|
|
- o := orm.NewOrmUsingDB("data")
|
|
|
+ //o := orm.NewOrmUsingDB("data")
|
|
|
+ //
|
|
|
+ //// 获取总数
|
|
|
+ //sql := `SELECT count(1) AS total FROM edb_info_permission_no_auth_record WHERE sys_user_id = ? AND edb_info_type = ? `
|
|
|
+ //err = o.Raw(sql, userId, edbInfoType).QueryRow(&total)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ //
|
|
|
+ //sql = `SELECT edb_info_permission_no_auth_record_id as data_permission_no_auth_record_id,op_unique_code,edb_info_type as sub_source,edb_info_id as data_id,edb_code as data_code,edb_name as data_name,sys_user_id,create_time FROM edb_info_permission_no_auth_record WHERE sys_user_id = ? AND edb_info_type = ? ORDER BY edb_info_permission_no_auth_record_id desc LIMIT ?,? `
|
|
|
+ //_, err = o.Raw(sql, userId, edbInfoType, startSize, pageSize).QueryRows(&items)
|
|
|
|
|
|
// 获取总数
|
|
|
sql := `SELECT count(1) AS total FROM edb_info_permission_no_auth_record WHERE sys_user_id = ? AND edb_info_type = ? `
|
|
|
- err = o.Raw(sql, userId, edbInfoType).QueryRow(&total)
|
|
|
+ err = global.DmSQL["data"].Raw(sql, userId, edbInfoType).First(&total).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
|
|
|
sql = `SELECT edb_info_permission_no_auth_record_id as data_permission_no_auth_record_id,op_unique_code,edb_info_type as sub_source,edb_info_id as data_id,edb_code as data_code,edb_name as data_name,sys_user_id,create_time FROM edb_info_permission_no_auth_record WHERE sys_user_id = ? AND edb_info_type = ? ORDER BY edb_info_permission_no_auth_record_id desc LIMIT ?,? `
|
|
|
- _, err = o.Raw(sql, userId, edbInfoType, startSize, pageSize).QueryRows(&items)
|
|
|
-
|
|
|
+ err = global.DmSQL["data"].Raw(sql, userId, edbInfoType, startSize, pageSize).Find(&items).Error
|
|
|
return
|
|
|
}
|
|
|
|
|
@@ -697,22 +871,38 @@ func AddEdbInfoClassifyNoAuthRecordBySourceAndClassifyIdList(source, edbClassify
|
|
|
if userNum <= 0 {
|
|
|
return
|
|
|
}
|
|
|
- o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
- if err != nil {
|
|
|
- return
|
|
|
- }
|
|
|
+ //o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ //defer func() {
|
|
|
+ // if err != nil {
|
|
|
+ // _ = o.Rollback()
|
|
|
+ // } else {
|
|
|
+ // _ = o.Commit()
|
|
|
+ // }
|
|
|
+ //}()
|
|
|
+
|
|
|
+ to := global.DmSQL["data"].Begin()
|
|
|
+
|
|
|
defer func() {
|
|
|
if err != nil {
|
|
|
- _ = o.Rollback()
|
|
|
+ _ = to.Rollback()
|
|
|
} else {
|
|
|
- _ = o.Commit()
|
|
|
+ _ = to.Commit()
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
// 根据分类获取已经存在的未授权记录
|
|
|
var existList []*EdbInfoClassifyPermissionNoAuthRecord
|
|
|
+ //sql := `SELECT * FROM edb_info_classify_permission_no_auth_record WHERE edb_classify_type = ? AND classify_id in (` + utils.GetOrmInReplace(num) + `)`
|
|
|
+ //_, err = o.Raw(sql, edbClassifyType, classifyIdList).QueryRows(&existList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
sql := `SELECT * FROM edb_info_classify_permission_no_auth_record WHERE edb_classify_type = ? AND classify_id in (` + utils.GetOrmInReplace(num) + `)`
|
|
|
- _, err = o.Raw(sql, edbClassifyType, classifyIdList).QueryRows(&existList)
|
|
|
+ err = to.Raw(sql, edbClassifyType, classifyIdList).Find(&existList).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -780,15 +970,24 @@ func AddEdbInfoClassifyNoAuthRecordBySourceAndClassifyIdList(source, edbClassify
|
|
|
|
|
|
// 添加消息
|
|
|
if len(addMessageList) > 0 {
|
|
|
- _, err = o.InsertMulti(500, addMessageList)
|
|
|
+ //_, err = o.InsertMulti(500, addMessageList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ err = to.CreateInBatches(addMessageList, utils.MultiAddNum).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
+
|
|
|
}
|
|
|
|
|
|
// 添加记录
|
|
|
if len(addRecordList) > 0 {
|
|
|
- _, err = o.InsertMulti(500, addRecordList)
|
|
|
+ //_, err = o.InsertMulti(500, addRecordList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ err = to.CreateInBatches(addRecordList, utils.MultiAddNum).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -797,8 +996,11 @@ func AddEdbInfoClassifyNoAuthRecordBySourceAndClassifyIdList(source, edbClassify
|
|
|
// 移除已经公开了的分类权限,需要删除未授权记录
|
|
|
authUserIdNum := len(classifyIdList)
|
|
|
if authUserIdNum > 0 {
|
|
|
+ //sql = `DELETE FROM edb_info_classify_permission_no_auth_record WHERE edb_classify_type = ? AND classify_id not in (` + utils.GetOrmInReplace(num) + `)`
|
|
|
+ //_, err = o.Raw(sql, edbClassifyType, classifyIdList).Exec()
|
|
|
+
|
|
|
sql = `DELETE FROM edb_info_classify_permission_no_auth_record WHERE edb_classify_type = ? AND classify_id not in (` + utils.GetOrmInReplace(num) + `)`
|
|
|
- _, err = o.Raw(sql, edbClassifyType, classifyIdList).Exec()
|
|
|
+ err = to.Exec(sql, edbClassifyType, classifyIdList).Error
|
|
|
}
|
|
|
|
|
|
return
|
|
@@ -829,22 +1031,38 @@ func AddEdbInfoClassifyNoAuthRecordBySourceAndUserIdList(source, edbClassifyType
|
|
|
// return
|
|
|
//}
|
|
|
|
|
|
- o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
- if err != nil {
|
|
|
- return
|
|
|
- }
|
|
|
+ //o, err := orm.NewOrmUsingDB("data").Begin()
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ //defer func() {
|
|
|
+ // if err != nil {
|
|
|
+ // _ = o.Rollback()
|
|
|
+ // } else {
|
|
|
+ // _ = o.Commit()
|
|
|
+ // }
|
|
|
+ //}()
|
|
|
+
|
|
|
+ to := global.DmSQL["data"].Begin()
|
|
|
+
|
|
|
defer func() {
|
|
|
if err != nil {
|
|
|
- _ = o.Rollback()
|
|
|
+ _ = to.Rollback()
|
|
|
} else {
|
|
|
- _ = o.Commit()
|
|
|
+ _ = to.Commit()
|
|
|
}
|
|
|
}()
|
|
|
|
|
|
// 根据当前配置用户获取已经存在的未授权记录
|
|
|
var existList []*EdbInfoClassifyPermissionNoAuthRecord
|
|
|
+ //sql := `SELECT * FROM edb_info_classify_permission_no_auth_record WHERE edb_classify_type = ? AND sys_user_id in (` + utils.GetOrmInReplace(configUserNum) + `)`
|
|
|
+ //_, err = o.Raw(sql, edbClassifyType, configUserIdList).QueryRows(&existList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
sql := `SELECT * FROM edb_info_classify_permission_no_auth_record WHERE edb_classify_type = ? AND sys_user_id in (` + utils.GetOrmInReplace(configUserNum) + `)`
|
|
|
- _, err = o.Raw(sql, edbClassifyType, configUserIdList).QueryRows(&existList)
|
|
|
+ err = to.Raw(sql, edbClassifyType, configUserIdList).Find(&existList).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -916,7 +1134,12 @@ func AddEdbInfoClassifyNoAuthRecordBySourceAndUserIdList(source, edbClassifyType
|
|
|
|
|
|
// 添加消息
|
|
|
if len(addMessageList) > 0 {
|
|
|
- _, err = o.InsertMulti(500, addMessageList)
|
|
|
+ //_, err = o.InsertMulti(500, addMessageList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
+ err = to.CreateInBatches(addMessageList, utils.MultiAddNum).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -924,7 +1147,12 @@ func AddEdbInfoClassifyNoAuthRecordBySourceAndUserIdList(source, edbClassifyType
|
|
|
|
|
|
// 添加记录
|
|
|
if len(addRecordList) > 0 {
|
|
|
- _, err = o.InsertMulti(500, addRecordList)
|
|
|
+ //_, err = o.InsertMulti(500, addRecordList)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+
|
|
|
+ err = to.CreateInBatches(addRecordList, utils.MultiAddNum).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
@@ -937,8 +1165,10 @@ func AddEdbInfoClassifyNoAuthRecordBySourceAndUserIdList(source, edbClassifyType
|
|
|
for _, v := range delRecordIdMap {
|
|
|
delRecordIdList = append(delRecordIdList, v)
|
|
|
}
|
|
|
+ //sql = `DELETE FROM edb_info_classify_permission_no_auth_record WHERE edb_info_classify_permission_no_auth_record_id in (` + utils.GetOrmInReplace(delRecordIdNum) + `) `
|
|
|
+ //_, err = o.Raw(sql, delRecordIdList).Exec()
|
|
|
sql = `DELETE FROM edb_info_classify_permission_no_auth_record WHERE edb_info_classify_permission_no_auth_record_id in (` + utils.GetOrmInReplace(delRecordIdNum) + `) `
|
|
|
- _, err = o.Raw(sql, delRecordIdList).Exec()
|
|
|
+ err = to.Exec(sql, delRecordIdList).Error
|
|
|
}
|
|
|
|
|
|
return
|
|
@@ -952,10 +1182,13 @@ func AddEdbInfoClassifyNoAuthRecordBySourceAndUserIdList(source, edbClassifyType
|
|
|
// @param edbClassifyType int
|
|
|
// @return err error
|
|
|
func DeleteEdbInfoClassifyNoAuthRecordBySourceAndClassifyIdList(edbClassifyType int) (err error) {
|
|
|
- o := orm.NewOrmUsingDB("data")
|
|
|
+ //o := orm.NewOrmUsingDB("data")
|
|
|
+ //
|
|
|
+ //sql := `DELETE FROM edb_info_classify_permission_no_auth_record WHERE edb_classify_type = ?`
|
|
|
+ //_, err = o.Raw(sql, edbClassifyType).Exec()
|
|
|
|
|
|
sql := `DELETE FROM edb_info_classify_permission_no_auth_record WHERE edb_classify_type = ?`
|
|
|
- _, err = o.Raw(sql, edbClassifyType).Exec()
|
|
|
+ err = global.DmSQL["data"].Exec(sql, edbClassifyType).Error
|
|
|
|
|
|
return
|
|
|
}
|
|
@@ -973,17 +1206,26 @@ func DeleteEdbInfoClassifyNoAuthRecordBySourceAndClassifyIdList(edbClassifyType
|
|
|
// @return items []*DataPermissionClassifyNoAuthRecord
|
|
|
// @return err error
|
|
|
func GetEdbInfoDataPermissionClassifyNoAuthRecordListByUserId(userId int32, edbClassifyType, startSize, pageSize int) (total int, items []*DataPermissionClassifyNoAuthRecord, err error) {
|
|
|
- o := orm.NewOrmUsingDB("data")
|
|
|
+ //o := orm.NewOrmUsingDB("data")
|
|
|
+ //
|
|
|
+ //// 获取总数
|
|
|
+ //sql := `SELECT count(1) AS total FROM edb_info_classify_permission_no_auth_record WHERE sys_user_id = ? AND edb_classify_type = ? `
|
|
|
+ //err = o.Raw(sql, userId, edbClassifyType).QueryRow(&total)
|
|
|
+ //if err != nil {
|
|
|
+ // return
|
|
|
+ //}
|
|
|
+ //
|
|
|
+ //sql = `SELECT edb_info_classify_permission_no_auth_record_id as data_permission_classify_no_auth_record_id,edb_classify_type as sub_source,op_unique_code,classify_id,classify_name,sys_user_id,create_time FROM edb_info_classify_permission_no_auth_record WHERE sys_user_id = ? AND edb_classify_type = ? ORDER BY edb_info_classify_permission_no_auth_record_id desc LIMIT ?,? `
|
|
|
+ //_, err = o.Raw(sql, userId, edbClassifyType, startSize, pageSize).QueryRows(&items)
|
|
|
|
|
|
// 获取总数
|
|
|
sql := `SELECT count(1) AS total FROM edb_info_classify_permission_no_auth_record WHERE sys_user_id = ? AND edb_classify_type = ? `
|
|
|
- err = o.Raw(sql, userId, edbClassifyType).QueryRow(&total)
|
|
|
+ err = global.DmSQL["data"].Raw(sql, userId, edbClassifyType).Scan(&total).Error
|
|
|
if err != nil {
|
|
|
return
|
|
|
}
|
|
|
|
|
|
sql = `SELECT edb_info_classify_permission_no_auth_record_id as data_permission_classify_no_auth_record_id,edb_classify_type as sub_source,op_unique_code,classify_id,classify_name,sys_user_id,create_time FROM edb_info_classify_permission_no_auth_record WHERE sys_user_id = ? AND edb_classify_type = ? ORDER BY edb_info_classify_permission_no_auth_record_id desc LIMIT ?,? `
|
|
|
- _, err = o.Raw(sql, userId, edbClassifyType, startSize, pageSize).QueryRows(&items)
|
|
|
-
|
|
|
+ err = global.DmSQL["data"].Raw(sql, userId, edbClassifyType, startSize, pageSize).Find(&items).Error
|
|
|
return
|
|
|
}
|