package data_manage_permission import ( "eta_gn/eta_api/global" "eta_gn/eta_api/utils" "fmt" "strconv" "time" ) type ChartInfoPermission struct { ChartInfoPermissionId int64 `json:"chart_info_permission_id" orm:"column(chart_info_permission_id);pk" gorm:"primaryKey" ` ChartInfoId int32 `json:"chart_info_id"` // 图表id Source int32 `json:"source"` // 1:ETA图库;2:商品价格曲线;3:相关性图 SysUserId int32 `json:"sys_user_id"` // 系统用户id ModifyTime time.Time `json:"modify_time"` // 变更时间 CreateTime time.Time `json:"create_time"` // 关系建立时间 } type ChartClassifyPermission struct { ChartClassifyPermissionId int64 `json:"chart_classify_permission_id" orm:"column(chart_classify_permission_id);pk" gorm:"primaryKey" ` ChartClassifyId int32 `json:"chart_classify_id"` // 分类id Source int32 `json:"source"` // 分类来源: 1-图库; 2-商品价格曲线; 3-相关性图表; 6-拟合方程图表; 7-统计特征 SysUserId int32 `json:"sys_user_id"` // 系统用户id ModifyTime time.Time `json:"modify_time"` // 变更时间 CreateTime time.Time `json:"create_time"` // 关系建立时间 } func SetIsPermissionByChartClassifyIdList(classifyIdList []int, chartClassifySource int) (err error) { num := len(classifyIdList) to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() sql := `UPDATE chart_classify SET is_join_permission=?,modify_time=now() WHERE is_join_permission = 1 AND source = ?` err = to.Exec(sql, 0, chartClassifySource).Error if err != nil { return } if num > 0 { sql = `UPDATE chart_classify SET is_join_permission=?,modify_time=now() WHERE source = ? AND chart_classify_id in (` + utils.GetOrmInReplace(num) + `) ` err = to.Exec(sql, 1, chartClassifySource, classifyIdList).Error if err != nil { return } } return } func SetPermissionByChartIdList(chartIdList []string, userIdList []int, source int) (err error) { chartNum := len(chartIdList) if chartNum <= 0 { return } to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() chartInfoPermissionList := make([]*ChartInfoPermission, 0) sql := `SELECT * FROM chart_info_permission WHERE source = ? AND chart_info_id in (` + utils.GetOrmInReplace(chartNum) + `) ` err = to.Raw(sql, source, chartIdList).Find(&chartInfoPermissionList).Error if err != nil { return } chartInfoPermissionMap := make(map[string]*ChartInfoPermission) for _, v := range chartInfoPermissionList { chartInfoPermissionMap[fmt.Sprint(v.ChartInfoId, "_", v.SysUserId)] = v } { isJoinPermission := 1 if len(userIdList) <= 0 { isJoinPermission = 0 } sql = `UPDATE chart_info SET is_join_permission=?,modify_time=now() WHERE source = ? AND chart_info_id in (` + utils.GetOrmInReplace(chartNum) + `) ` err = to.Exec(sql, isJoinPermission, source, chartIdList).Error if err != nil { return } } addList := make([]*ChartInfoPermission, 0) for _, chartInfoIdStr := range chartIdList { chartInfoId, tmpErr := strconv.ParseInt(chartInfoIdStr, 10, 64) if tmpErr != nil { err = tmpErr return } for _, userId := range userIdList { key := fmt.Sprint(chartInfoId, "_", userId) if _, ok := chartInfoPermissionMap[key]; ok { delete(chartInfoPermissionMap, key) } else { addList = append(addList, &ChartInfoPermission{ ChartInfoId: int32(chartInfoId), SysUserId: int32(userId), Source: int32(source), ModifyTime: time.Now(), CreateTime: time.Now(), }) } } } if len(addList) > 0 { err = to.CreateInBatches(addList, utils.MultiAddNum).Error if err != nil { return } } { deletePermissionIdList := make([]int64, 0) for _, v := range chartInfoPermissionMap { deletePermissionIdList = append(deletePermissionIdList, v.ChartInfoPermissionId) } deletePermissionIdNum := len(deletePermissionIdList) if deletePermissionIdNum > 0 { sql = "DELETE FROM chart_info_permission WHERE chart_info_permission_id in (" + utils.GetOrmInReplace(deletePermissionIdNum) + ")" err = to.Exec(sql, deletePermissionIdList).Error if err != nil { return } } } return } func SetPermissionByChartClassifyIdList(classifyIdList []int, userIdList []int, classifyType int) (err error) { userNum := len(userIdList) if userNum <= 0 { return } to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() classifyPermissionList := make([]*ChartClassifyPermission, 0) sql := `SELECT * FROM chart_classify_permission WHERE source = ? AND sys_user_id in (` + utils.GetOrmInReplace(userNum) + `) ` err = to.Raw(sql, classifyType, userIdList).Find(&classifyPermissionList).Error if err != nil { return } classifyPermissionMap := make(map[string]*ChartClassifyPermission) for _, v := range classifyPermissionList { classifyPermissionMap[fmt.Sprint(v.ChartClassifyId, "_", v.SysUserId)] = v } addList := make([]*ChartClassifyPermission, 0) for _, userId := range userIdList { for _, classifyId := range classifyIdList { key := fmt.Sprint(classifyId, "_", userId) if _, ok := classifyPermissionMap[key]; ok { delete(classifyPermissionMap, key) } else { addList = append(addList, &ChartClassifyPermission{ ChartClassifyId: int32(classifyId), Source: int32(classifyType), SysUserId: int32(userId), ModifyTime: time.Now(), CreateTime: time.Now(), }) } } } if len(addList) > 0 { err = to.CreateInBatches(addList, utils.MultiAddNum).Error if err != nil { return } } { deletePermissionIdList := make([]int64, 0) for _, v := range classifyPermissionMap { deletePermissionIdList = append(deletePermissionIdList, v.ChartClassifyPermissionId) } deletePermissionIdNum := len(deletePermissionIdList) if deletePermissionIdNum > 0 { sql = "DELETE FROM chart_classify_permission WHERE chart_classify_permission_id in (" + utils.GetOrmInReplace(deletePermissionIdNum) + ")" err = to.Exec(sql, deletePermissionIdList).Error if err != nil { return } } } return } func GetPermissionChartClassifyIdListByUserId(userId int, classifyType int) (chartClassifyIdList []int, err error) { sql := `SELECT chart_classify_id FROM chart_classify_permission WHERE source = ? AND sys_user_id = ? ` err = global.DmSQL["data"].Raw(sql, classifyType, userId).Find(&chartClassifyIdList).Error return } func GetPermissionUserIdListByChartId(dataId int, source int) (userIdList []int, err error) { sql := `SELECT sys_user_id FROM chart_info_permission WHERE source = ? AND chart_info_id= ? ` err = global.DmSQL["data"].Raw(sql, source, dataId).Find(&userIdList).Error return } func GetPermissionUserIdListByChartClassifyId(classifyId int, source int) (userIdList []int, err error) { sql := `SELECT sys_user_id FROM chart_classify_permission WHERE source = ? AND chart_classify_id= ? ` err = global.DmSQL["data"].Raw(sql, source, classifyId).Find(&userIdList).Error return } func GetPermissionChartIdList(userId, chartInfoId int) (idList []int, err error) { pars := []interface{}{userId} sql := `SELECT chart_info_id FROM chart_info_permission WHERE sys_user_id = ? ` if chartInfoId > 0 { sql += ` AND chart_info_id = ? ` pars = append(pars, chartInfoId) } err = global.DmSQL["data"].Raw(sql, pars...).Find(&idList).Error return } func GetPermissionChartClassifyIdList(userId, classifyId int) (idList []int, err error) { pars := []interface{}{userId} sql := `SELECT chart_classify_id FROM chart_classify_permission WHERE sys_user_id = ? ` if classifyId > 0 { sql += ` AND chart_classify_id = ? ` pars = append(pars, classifyId) } err = global.DmSQL["data"].Raw(sql, pars...).Find(&idList).Error return } func InheritParentClassifyByChartClassifyId(dataSource, chartSource, classifyId, parentClassifyId int, classifyName, uniqueCode string) (err error) { to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() sql := `UPDATE chart_classify SET is_join_permission=?,modify_time=now() WHERE source = ? AND chart_classify_id = ? ` err = to.Exec(sql, 1, chartSource, classifyId).Error if err != nil { return } { var parentRecordItems []*ChartInfoClassifyPermissionNoAuthRecord sql = `SELECT * FROM data_permission_classify_no_auth_record WHERE classify_id = ? AND source = ? AND sub_source = ? ORDER BY data_permission_classify_no_auth_record_id desc LIMIT ?,? ` err = to.Raw(sql, parentClassifyId, dataSource, chartSource).Find(&parentRecordItems).Error addNoAuthRecordItems := make([]*ChartInfoClassifyPermissionNoAuthRecord, 0) for _, v := range parentRecordItems { addNoAuthRecordItems = append(addNoAuthRecordItems, &ChartInfoClassifyPermissionNoAuthRecord{ ChartInfoClassifyPermissionNoAuthRecordId: 0, Source: v.Source, OpUniqueCode: uniqueCode, ClassifyId: fmt.Sprint(classifyId), ClassifyName: classifyName, SysUserId: v.SysUserId, CreateTime: time.Now(), }) } if len(addNoAuthRecordItems) > 0 { err = to.CreateInBatches(addNoAuthRecordItems, len(addNoAuthRecordItems)).Error if err != nil { return } } } { parentClassifyPermissionList := make([]*ChartClassifyPermission, 0) sql = `SELECT * FROM chart_classify_permission WHERE source = ? AND chart_classify_id = ? ` err = to.Raw(sql, chartSource, parentClassifyId).Find(&parentClassifyPermissionList).Error if err != nil { return } addList := make([]*ChartClassifyPermission, 0) for _, v := range parentClassifyPermissionList { addList = append(addList, &ChartClassifyPermission{ ChartClassifyId: int32(classifyId), Source: int32(chartSource), SysUserId: v.SysUserId, ModifyTime: time.Now(), CreateTime: time.Now(), }) } if len(addList) > 0 { err = to.CreateInBatches(addList, utils.MultiAddNum).Error if err != nil { return } } } return } type ChartInfoPermissionNoAuthRecord struct { ChartInfoPermissionNoAuthRecordId int64 `json:"chart_info_permission_no_auth_record_id" orm:"column(chart_info_permission_no_auth_record_id);pk" gorm:"primaryKey" ` // 资产数据操作记录id OpUniqueCode string `json:"op_unique_code"` // 操作的唯一编码,主要是记录统一操作的日志 Source int32 `json:"source"` // 1:ETA图库;2:商品价格曲线;3:相关性图 ChartInfoId int32 `json:"chart_info_id"` // 指标id ChartName string `json:"chart_name"` // 图表名称 SysUserId int32 `json:"sys_user_id"` // 系统用户id CreateTime time.Time `json:"create_time"` // 创建时间 } func AddChartInfoPermissionNoAuthRecordBySourceAndDataIdList(source, chartSource int, dataList []DataItem, noAuthUserIdList, authUserIdList []int, uniqueCode, title, content string, opUserId int) (err error) { num := len(dataList) if num <= 0 { return } dataIdList := make([]int, 0) for _, v := range dataList { dataIdList = append(dataIdList, v.DataId) } userNum := len(noAuthUserIdList) if userNum <= 0 { return } to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() var existList []*ChartInfoPermissionNoAuthRecord sql := `SELECT * FROM chart_info_permission_no_auth_record WHERE source = ? AND chart_info_id in (` + utils.GetOrmInReplace(num) + `)` err = to.Raw(sql, chartSource, dataIdList).Find(&existList).Error if err != nil { return } existMap := make(map[int32]map[string]*ChartInfoPermissionNoAuthRecord) for _, v := range existList { tmpUserExistMap, ok := existMap[v.SysUserId] if !ok { tmpUserExistMap = make(map[string]*ChartInfoPermissionNoAuthRecord) } key := fmt.Sprint(v.Source, "_", v.ChartInfoId) tmpUserExistMap[key] = v existMap[v.SysUserId] = tmpUserExistMap } addMessageList := make([]*DataPermissionMessage, 0) addRecordList := make([]*ChartInfoPermissionNoAuthRecord, 0) for _, userId := range noAuthUserIdList { isAdd := false tmpUserExistMap, userExistOk := existMap[int32(userId)] for _, dataItem := range dataList { if userExistOk { key := fmt.Sprint(chartSource, "_", dataItem.DataId) _, ok := tmpUserExistMap[key] if ok { continue } } isAdd = true addRecordList = append(addRecordList, &ChartInfoPermissionNoAuthRecord{ ChartInfoPermissionNoAuthRecordId: 0, OpUniqueCode: uniqueCode, Source: int32(chartSource), ChartInfoId: int32(dataItem.DataId), ChartName: dataItem.DataName, SysUserId: int32(userId), CreateTime: time.Now(), }) } if isAdd { addMessageList = append(addMessageList, &DataPermissionMessage{ DataPermissionMessageId: 0, SendUserId: int32(opUserId), ReceiveUserId: int32(userId), Content: title, Remark: content, OpType: 3, Source: int32(source), SubSource: int32(chartSource), OpUniqueCode: uniqueCode, IsRead: 0, CreateTime: time.Now(), ModifyTime: time.Now(), }) } } if len(addMessageList) > 0 { err = to.CreateInBatches(addMessageList, utils.MultiAddNum).Error if err != nil { return } } if len(addRecordList) > 0 { err = to.CreateInBatches(addRecordList, utils.MultiAddNum).Error if err != nil { return } } authUserIdNum := len(authUserIdList) if authUserIdNum > 0 { sql = `DELETE FROM chart_info_permission_no_auth_record WHERE source = ? AND sys_user_id in (` + utils.GetOrmInReplace(authUserIdNum) + `) AND chart_info_id in (` + utils.GetOrmInReplace(num) + `)` err = to.Exec(sql, chartSource, authUserIdList, dataIdList).Error } return } func DeleteChartInfoPermissionNoAuthRecordBySourceAndDataIdList(chartSource int, dataIdList []string) (err error) { num := len(dataIdList) if num <= 0 { return } sql := `DELETE FROM chart_info_permission_no_auth_record WHERE source = ? AND chart_info_id in (` + utils.GetOrmInReplace(num) + `)` err = global.DmSQL["data"].Exec(sql, chartSource, dataIdList).Error return } func GetChartInfoDataPermissionNoAuthRecordListByUserId(userId int32, chartSource, startSize, pageSize int) (total int, items []*DataPermissionNoAuthRecord, err error) { sql := `SELECT count(1) AS total FROM chart_info_permission_no_auth_record WHERE sys_user_id = ? AND source = ? ` err = global.DmSQL["data"].Raw(sql, userId, chartSource).Scan(&total).Error if err != nil { return } sql = `SELECT chart_info_permission_no_auth_record_id as data_permission_no_auth_record_id,op_unique_code,source as sub_source,chart_info_id as data_id,chart_name as data_name,sys_user_id,create_time FROM chart_info_permission_no_auth_record WHERE sys_user_id = ? AND source = ? ORDER BY chart_info_permission_no_auth_record_id desc LIMIT ?,? ` err = global.DmSQL["data"].Raw(sql, userId, chartSource, startSize, pageSize).Find(&items).Error return } type ChartInfoClassifyPermissionNoAuthRecord struct { ChartInfoClassifyPermissionNoAuthRecordId int64 `json:"chart_info_classify_permission_no_auth_record_id" orm:"column(chart_info_classify_permission_no_auth_record_id);pk" gorm:"primaryKey" ` // 资产分类数据操作记录id Source int32 `json:"source"` // 子来源 :ETA表格中的各种表格类型,以及图表的来源(这个是后续的扩展方向) OpUniqueCode string `json:"op_unique_code"` // 操作的唯一编码,主要是记录统一操作的日志 ClassifyId string `json:"classify_id"` // 图表资产分类id ClassifyName string `json:"classify_name"` // 图表资产分类名称 SysUserId int32 `json:"sys_user_id"` // 系统用户id CreateTime time.Time `json:"create_time"` // 创建时间 } func AddChartInfoClassifyNoAuthRecordBySourceAndClassifyIdList(source, chartSource int, classifyInfoList []ClassifyDataItem, noAuthUserIdList []int, uniqueCode, title, content string, opUserId int) (err error) { num := len(classifyInfoList) if num <= 0 { return } classifyIdList := make([]int, 0) for _, v := range classifyInfoList { classifyIdList = append(classifyIdList, v.ClassifyId) } userNum := len(noAuthUserIdList) if userNum <= 0 { return } to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() var existList []*ChartInfoClassifyPermissionNoAuthRecord sql := `SELECT * FROM chart_info_classify_permission_no_auth_record WHERE source = ? AND classify_id in (` + utils.GetOrmInReplace(num) + `)` err = to.Raw(sql, chartSource, classifyIdList).Find(&existList).Error if err != nil { return } existMap := make(map[int32]map[string]*ChartInfoClassifyPermissionNoAuthRecord) for _, v := range existList { tmpUserExistMap, ok := existMap[v.SysUserId] if !ok { tmpUserExistMap = make(map[string]*ChartInfoClassifyPermissionNoAuthRecord) } key := fmt.Sprint(v.Source, "_", v.ClassifyId) tmpUserExistMap[key] = v existMap[v.SysUserId] = tmpUserExistMap } addMessageList := make([]*DataPermissionMessage, 0) addRecordList := make([]*ChartInfoClassifyPermissionNoAuthRecord, 0) for _, userId := range noAuthUserIdList { isAdd := false tmpUserExistMap, userExistOk := existMap[int32(userId)] for _, dataItem := range classifyInfoList { if userExistOk { key := fmt.Sprint(chartSource, "_", dataItem.ClassifyId) _, ok := tmpUserExistMap[key] if ok { continue } } isAdd = true addRecordList = append(addRecordList, &ChartInfoClassifyPermissionNoAuthRecord{ ChartInfoClassifyPermissionNoAuthRecordId: 0, Source: int32(chartSource), OpUniqueCode: uniqueCode, ClassifyId: fmt.Sprint(dataItem.ClassifyId), ClassifyName: dataItem.ClassifyName, SysUserId: int32(userId), CreateTime: time.Now(), }) } if isAdd { addMessageList = append(addMessageList, &DataPermissionMessage{ DataPermissionMessageId: 0, SendUserId: int32(opUserId), ReceiveUserId: int32(userId), Content: title, Remark: content, OpType: 4, Source: int32(source), SubSource: int32(chartSource), OpUniqueCode: uniqueCode, IsRead: 0, CreateTime: time.Now(), ModifyTime: time.Now(), }) } } if len(addMessageList) > 0 { err = to.CreateInBatches(addMessageList, utils.MultiAddNum).Error if err != nil { return } } if len(addRecordList) > 0 { err = to.CreateInBatches(addRecordList, utils.MultiAddNum).Error if err != nil { return } } authUserIdNum := len(classifyIdList) if authUserIdNum > 0 { sql = `DELETE FROM chart_info_classify_permission_no_auth_record WHERE source = ? AND classify_id not in (` + utils.GetOrmInReplace(num) + `)` err = to.Exec(sql, chartSource, classifyIdList).Error } return } func AddChartInfoClassifyNoAuthRecordBySourceAndUserIdList(source, chartSource int, noAuthClassifyMap map[int]ClassifyDataItem, configUserIdList []int, uniqueCode, title, content string, opUserId int) (err error) { configUserNum := len(configUserIdList) if configUserNum <= 0 { return } to := global.DmSQL["data"].Begin() defer func() { if err != nil { _ = to.Rollback() } else { _ = to.Commit() } }() var existList []*ChartInfoClassifyPermissionNoAuthRecord sql := `SELECT * FROM chart_info_classify_permission_no_auth_record WHERE source = ? AND sys_user_id in (` + utils.GetOrmInReplace(configUserNum) + `)` err = to.Raw(sql, chartSource, configUserIdList).Find(&existList).Error if err != nil { return } existMap := make(map[int32]map[string]*ChartInfoClassifyPermissionNoAuthRecord) delRecordIdMap := make(map[int64]int64) for _, v := range existList { tmpUserExistMap, ok := existMap[v.SysUserId] if !ok { tmpUserExistMap = make(map[string]*ChartInfoClassifyPermissionNoAuthRecord) } tmpUserExistMap[v.ClassifyId] = v existMap[v.SysUserId] = tmpUserExistMap delRecordIdMap[v.ChartInfoClassifyPermissionNoAuthRecordId] = v.ChartInfoClassifyPermissionNoAuthRecordId } addMessageList := make([]*DataPermissionMessage, 0) addRecordList := make([]*ChartInfoClassifyPermissionNoAuthRecord, 0) for _, userId := range configUserIdList { isAdd := false tmpUserExistMap, userExistOk := existMap[int32(userId)] for _, dataItem := range noAuthClassifyMap { if userExistOk { key := fmt.Sprint(dataItem.ClassifyId) tmpUserRecord, ok := tmpUserExistMap[key] if ok { delete(delRecordIdMap, tmpUserRecord.ChartInfoClassifyPermissionNoAuthRecordId) continue } } isAdd = true addRecordList = append(addRecordList, &ChartInfoClassifyPermissionNoAuthRecord{ ChartInfoClassifyPermissionNoAuthRecordId: 0, Source: int32(chartSource), OpUniqueCode: uniqueCode, ClassifyId: fmt.Sprint(dataItem.ClassifyId), ClassifyName: dataItem.ClassifyName, SysUserId: int32(userId), CreateTime: time.Now(), }) } if isAdd { addMessageList = append(addMessageList, &DataPermissionMessage{ DataPermissionMessageId: 0, SendUserId: int32(opUserId), ReceiveUserId: int32(userId), Content: content, Remark: content, OpType: 4, Source: int32(source), SubSource: int32(chartSource), OpUniqueCode: uniqueCode, IsRead: 0, CreateTime: time.Now(), ModifyTime: time.Now(), }) } } if len(addMessageList) > 0 { err = to.CreateInBatches(addMessageList, utils.MultiAddNum).Error if err != nil { return } } if len(addRecordList) > 0 { err = to.CreateInBatches(addRecordList, utils.MultiAddNum).Error if err != nil { return } } delRecordIdNum := len(delRecordIdMap) if delRecordIdNum > 0 { delRecordIdList := make([]int64, 0) for _, v := range delRecordIdMap { delRecordIdList = append(delRecordIdList, v) } sql = `DELETE FROM chart_info_classify_permission_no_auth_record WHERE chart_info_classify_permission_no_auth_record_id in (` + utils.GetOrmInReplace(delRecordIdNum) + `) ` err = to.Exec(sql, delRecordIdList).Error } return } func DeleteChartInfoClassifyNoAuthRecordBySourceAndClassifyIdList(chartSource int) (err error) { sql := `DELETE FROM chart_info_classify_permission_no_auth_record WHERE source = ?` err = global.DmSQL["data"].Exec(sql, chartSource).Error return } func GetChartInfoDataPermissionClassifyNoAuthRecordListByUserId(userId int32, chartSource, startSize, pageSize int) (total int, items []*DataPermissionClassifyNoAuthRecord, err error) { sql := `SELECT count(1) AS total FROM chart_info_classify_permission_no_auth_record WHERE sys_user_id = ? AND source = ? ` err = global.DmSQL["data"].Raw(sql, userId, chartSource).Scan(&total).Error if err != nil { return } sql = `SELECT chart_info_classify_permission_no_auth_record_id as data_permission_classify_no_auth_record_id,source as sub_source,op_unique_code,classify_id,classify_name,sys_user_id,create_time FROM chart_info_classify_permission_no_auth_record WHERE sys_user_id = ? AND source = ? ORDER BY chart_info_classify_permission_no_auth_record_id desc LIMIT ?,? ` err = global.DmSQL["data"].Raw(sql, userId, chartSource, startSize, pageSize).Find(&items).Error return }