zjxpcyc 6 年之前
父節點
當前提交
aa40d80bdb
共有 1 個文件被更改,包括 8 次插入4 次删除
  1. 8
    4
      service/user.go

+ 8
- 4
service/user.go 查看文件

@@ -70,7 +70,12 @@ func (s *UserServ) GetUserList(username, typeid, caseids string, page int, pageS
70 70
 // GetCurrentEnvVars 获取当前环境相关变量
71 71
 func (s *UserServ) GetCurrentEnvVars() (map[string]interface{}, error) {
72 72
 	// 当前用户信息
73
-	user := s.ctx.Get("user").(model.SysUser)
73
+	userRaw := s.ctx.Get("user")
74
+	if userRaw == nil {
75
+		return nil, utils.LogError("无有效用户信息")
76
+	}
77
+
78
+	user := userRaw.(model.SysUser)
74 79
 	user.Pwd = ""
75 80
 	// 当前用户案场
76 81
 	sessionCase := s.ctx.Get("cases").([]model.SysUserCase)
@@ -89,14 +94,13 @@ func (s *UserServ) GetCurrentEnvVars() (map[string]interface{}, error) {
89 94
 	// 当前用户角色
90 95
 	roles, err := s.dao.GetUserRole(user.UserId)
91 96
 	if err != nil {
92
-		beego.Error(err)
93
-		return nil, err
97
+		return nil, utils.LogError(err)
94 98
 	}
95 99
 	// 当前用户菜单
96 100
 	menus, err := s.dao.GetUserMenu(user.UserId)
97 101
 	if err != nil {
98 102
 		beego.Error(err)
99
-		return nil, err
103
+		return nil, utils.LogError(err)
100 104
 	}
101 105
 
102 106
 	rtn := map[string]interface{}{