登录成功后存储当前数据源信息

ruoyi-ui/src/store/modules/user.js

@@ -1,5 +1,5 @@
-import { login, logout, getInfo } from '@/api/login'
-import { getToken, setToken, removeToken } from '@/utils/auth'
+import {login, logout, getInfo} from '@/api/login'
+import {getToken, setToken, removeToken} from '@/utils/auth'
 
 const user = {
   state: {
@@ -8,7 +8,8 @@ const user = {
     avatar: '',
     roles: [],
     permissions: [],
-    tenant:{}
+    tenant: {},
+    dataSource: {}
   },
 
   mutations: {
@@ -29,12 +30,15 @@ const user = {
     },
     SET_TENANT: (state, tenant) => {
       state.tenant = tenant
+    },
+    SET_DATASOURCE: (state, dataSource) => {
+      state.dataSource = dataSource
     }
   },
 
   actions: {
     // 登录
-    Login({ commit }, userInfo) {
+    Login({commit}, userInfo) {
       const username = userInfo.username.trim()
       const password = userInfo.password
       const code = userInfo.code
@@ -51,7 +55,7 @@ const user = {
     },
 
     // 获取用户信息
-    GetInfo({ commit, state }) {
+    GetInfo({commit, state}) {
       return new Promise((resolve, reject) => {
         getInfo().then(res => {
           const user = res.user
@@ -62,8 +66,9 @@ const user = {
           } else {
             commit('SET_ROLES', ['ROLE_DEFAULT'])
           }
-          if(res.tenant != null){
-            commit('SET_TENANT',res.tenant)
+          if (res.tenant != null) {
+            commit('SET_TENANT', res.tenant)
+            commit('SET_DATASOURCE', res.dataSource)
           }
           commit('SET_NAME', user.userName)
           commit('SET_AVATAR', avatar)
@@ -75,7 +80,7 @@ const user = {
     },
 
     // 退出系统
-    LogOut({ commit, state }) {
+    LogOut({commit, state}) {
       return new Promise((resolve, reject) => {
         logout(state.token).then(() => {
           commit('SET_TOKEN', '')
@@ -91,7 +96,7 @@ const user = {
     },
 
     // 前端 登出
-    FedLogOut({ commit }) {
+    FedLogOut({commit}) {
       return new Promise(resolve => {
         commit('SET_TOKEN', '')
         removeToken()