OMC/be.ems
1
0
Fork 0
Code Issues Releases 3 Activity

fix: multi-tenant phase 2

Browse Source
This commit is contained in:
simonzhangsz
2024-06-26 10:30:53 +08:00
parent 267904be29
commit f514818e5d
16 changed files with 56 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
src/framework/middleware/collectlogs/operate_log.go
View File

@@ -121,6 +121,7 @@ func OperateLog(options Options) gin.HandlerFunc {
OperName: loginUser.User.UserName,
DeptName: loginUser.User.Dept.DeptName,
TenantName: loginUser.User.Tenant.TenantName,
TenantID: loginUser.User.Tenant.TenantID,
}
if loginUser.User.UserType == "sys" {

14
src/modules/network_element/repository/udm_sub.impl.go
View File

@@ -114,17 +114,21 @@ func (r *UDMSubImpl) SelectPage(query map[string]any) map[string]any {
params = append(params, tenantID[0])
}
} else if v, ok := query["userName"]; ok && v != "" {
var tenantID []string
err := dborm.DefaultDB().Table("sys_user").
Where("user_name=?", v).Cols("tenant_id").Distinct().Find(&tenantID)
var tenantID string
_, err := dborm.DefaultDB().Table("sys_user").
Where("user_name=?", v).Cols("tenant_id").Distinct().Get(&tenantID)
if err != nil {
log.Errorf("Find tenant_id from sys_user err => %v", err)
}
log.Tracef("userName=%v, tenantID=%v", v, tenantID)
if len(tenantID) > 0 {
if tenantID != "" {
conditions = append(conditions, "s.tenant_id = ?")
params = append(params, tenantID[0])
params = append(params, tenantID)
}
// if len(tenantID) > 0 {
// conditions = append(conditions, "s.tenant_id = ?")
// params = append(params, tenantID[0])
// }
}
// 构建查询条件语句

4
src/modules/system/controller/sys_tenant.go
View File

@@ -132,7 +132,7 @@ func (s *SysTenantController) Add(c *gin.Context) {
}
// 检查租户名称唯一
if body.ParentID == "1" {
if body.ParentID == "0" {
uniqueTenantName := s.sysTenantService.CheckUniqueTenantName(body.TenantName, body.ParentID, "")
if !uniqueTenantName {
// 租户新增【%s】失败，租户名称已存在
@@ -197,7 +197,7 @@ func (s *SysTenantController) Edit(c *gin.Context) {
}
// 检查租户名称唯一
if body.ParentID == "1" {
if body.ParentID == "0" {
uniqueTenantName := s.sysTenantService.CheckUniqueTenantName(body.TenantName, body.ParentID, body.TenantID)
if !uniqueTenantName {
// 租户修改【%s】失败，租户名称已存在

3
src/modules/system/model/sys_log_operate.go
View File

@@ -34,6 +34,7 @@ type SysLogOperate struct {
OperTime int64 `json:"operTime"`
// 消耗时间（毫秒）
CostTime int64 `json:"costTime"`
// Tenant name refer to tenant_name of sys_tenant
// TenantName and TenantID refer to sys_tenant
TenantName string `json:"tenantName"`
TenantID string `json:"tenantId"`
}

32
src/modules/system/repository/sys_log_operate.impl.go
View File

@@ -4,6 +4,8 @@ import (
"strings"
"time"
dborm "be.ems/lib/core/datasource"
"be.ems/lib/log"
"be.ems/src/framework/datasource"
"be.ems/src/framework/logger"
"be.ems/src/framework/utils/parse"
@@ -14,9 +16,11 @@ import (
// 实例化数据层 SysLogOperateImpl 结构体
var NewSysLogOperateImpl = &SysLogOperateImpl{
selectSql: `select
oper_id, title, business_type, method, request_method, operator_type, oper_name, dept_name,
oper_url, oper_ip, oper_location, oper_param, oper_msg, status, oper_time, cost_time
from sys_log_operate`,
l.oper_id, l.title, l.business_type, l.method, l.request_method, l.operator_type, l.oper_name, l.dept_name,
l.oper_url, l.oper_ip, l.oper_location, l.oper_param, l.oper_msg, l.status, l.oper_time, l.cost_time,
t.tenant_name
from sys_log_operate l
left join sys_tenant t on t.tenant_id = l.tenant_id`,
resultMap: map[string]string{
"oper_id": "OperID",
@@ -35,6 +39,7 @@ var NewSysLogOperateImpl = &SysLogOperateImpl{
"status": "Status",
"oper_time": "OperTime",
"cost_time": "CostTime",
"tenant_name": "TenantName",
},
}
@@ -100,9 +105,18 @@ func (r *SysLogOperateImpl) SelectSysLogOperatePage(query map[string]any) map[st
}
// multi-tenancy, only filter user setting tenant_id
if v, ok := query["operName"]; ok && v != "" {
conditions = append(conditions, "oper_name=?")
params = append(params, v)
if v, ok := query["tenantName"]; ok && v != "" {
var tenantID string
_, err := dborm.DefaultDB().Table("sys_tenant").
Where("tenant_name=?", v).Cols("tenant_id").Distinct().Get(&tenantID)
if err != nil {
log.Errorf("get tenant_id err => %v", err)
}
if tenantID != "" {
conditions = append(conditions, "l.tenant_id = ?")
params = append(params, tenantID)
}
log.Tracef("tenantName=%v, tenantID=%v", v, tenantID)
}
// 构建查询条件语句
@@ -118,7 +132,7 @@ func (r *SysLogOperateImpl) SelectSysLogOperatePage(query map[string]any) map[st
}
// 查询数量 长度为0直接返回
totalSql := "select count(1) as 'total' from sys_log_operate"
totalSql := "select count(1) as 'total' from sys_log_operate l"
totalRows, err := datasource.RawDB("", totalSql+whereSql, params)
if err != nil {
logger.Errorf("total err => %v", err)
@@ -253,8 +267,8 @@ func (r *SysLogOperateImpl) InsertSysLogOperate(SysLogOperate model.SysLogOperat
if SysLogOperate.CostTime > 0 {
params["cost_time"] = SysLogOperate.CostTime
}
if SysLogOperate.TenantName != "" {
params["tenant_name"] = SysLogOperate.TenantName
if SysLogOperate.TenantID != "" {
params["tenant_id"] = SysLogOperate.TenantID
}
// 构建执行语句

6
src/modules/system/repository/sys_tenant.impl.go
View File

@@ -236,6 +236,7 @@ func (r *SysTenantImpl) IsUniqueTenancy(sysTenant model.SysTenant) bool {
// 查询条件拼接
var conditions []string
var params []any
if sysTenant.TenantID != "" {
conditions = append(conditions, "tenant_id != ?")
params = append(params, sysTenant.TenantID)
@@ -245,8 +246,9 @@ func (r *SysTenantImpl) IsUniqueTenancy(sysTenant model.SysTenant) bool {
params = append(params, sysTenant.TenancyType)
}
if sysTenant.TenancyKey != "" {
conditions = append(conditions, "? like tenancy_key")
params = append(params, sysTenant.TenancyKey)
conditions = append(conditions, "(? like tenancy_key or tenancy_key like ?)")
//conditions = append(conditions, "(LOCATE(?, tenancy_key)>0 or LOCATE(tenancy_key, ?)>0)")
params = append(params, sysTenant.TenancyKey, sysTenant.TenancyKey)
}
// 构建查询条件语句