From 9f8e7533fb0f490de3ddccbf7c25f04a0edafc6c Mon Sep 17 00:00:00 2001
From: YunaiV <zhijiantianya@gmail.com>
Date: Tue, 6 May 2025 22:22:32 +0800
Subject: [PATCH] =?UTF-8?q?fix=EF=BC=9A=E7=A7=9F=E6=88=B7=E5=88=87?=
 =?UTF-8?q?=E6=8D=A2=E9=A6=96=E6=AC=A1=E6=97=A0=E6=B3=95=E5=8A=A0=E8=BD=BD?=
 =?UTF-8?q?=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../layouts/components/tenant-dropdown.vue    | 47 +++++++++----------
 packages/stores/src/modules/access.ts         | 11 ++---
 2 files changed, 25 insertions(+), 33 deletions(-)

diff --git a/apps/web-antd/src/layouts/components/tenant-dropdown.vue b/apps/web-antd/src/layouts/components/tenant-dropdown.vue
index a272d562..5bb888b9 100644
--- a/apps/web-antd/src/layouts/components/tenant-dropdown.vue
+++ b/apps/web-antd/src/layouts/components/tenant-dropdown.vue
@@ -1,6 +1,8 @@
 <script lang="ts" setup>
 import type { SelectValue } from 'ant-design-vue/es/select';
 
+import type { SystemTenantApi } from '#/api/system/tenant';
+
 import { onMounted, ref } from 'vue';
 
 import { useAccess } from '@vben/access';
@@ -9,56 +11,51 @@ import { useAccessStore } from '@vben/stores';
 
 import { message, Select } from 'ant-design-vue';
 
-import { getTenantList } from '#/api/system/tenant';
+import { getSimpleTenantList } from '#/api/system/tenant';
 import { $t } from '#/locales';
-import { resetRoutes } from '#/router';
-import { useAuthStore } from '#/store';
 
-const { closeOtherTabs, refreshTab, closeAllTabs } = useTabs();
+const { closeOtherTabs, refreshTab } = useTabs();
 
 const { hasAccessByCodes } = useAccess();
 const accessStore = useAccessStore();
 
-const authStore = useAuthStore();
-
 const tenantEnable = isTenantEnable();
 
-const visitTenantList = accessStore.visitTenantId;
+const value = ref<number>(accessStore.visitTenantId ?? undefined); // 当前访问的租户 ID
+const tenants = ref<SystemTenantApi.Tenant[]>([]); // 租户列表
 
-const tenant = ref<SelectValue>(accessStore.tenantId ?? 0);
-
-async function handleClick(id: SelectValue) {
-  accessStore.setTenantId(Number(id));
-  await authStore.fetchUserInfo();
-  await resetRoutes();
+async function handleChange(id: SelectValue) {
+  // 设置访问租户 ID
+  accessStore.setVisitTenantId(id as number);
+  // 关闭其他标签页，只保留当前页
   await closeOtherTabs();
-  await closeAllTabs();
+  // 刷新当前页面
   await refreshTab();
-  message.success($t('page.tenant.success'));
+  // 提示切换成功
+  const tenant = tenants.value.find((item) => item.id === id);
+  if (tenant) {
+    message.success(`切换当前租户为: ${tenant.name}`);
+  }
 }
 
 onMounted(async () => {
-  if (tenantEnable) {
-    const resp = await getTenantList();
-    accessStore.setVisitTenantId(
-      resp
-        .map((item) => ({ id: item.id, name: item.name }))
-        .filter((item): item is { id: number; name: string } => !!item.id),
-    );
+  if (!tenantEnable) {
+    return;
   }
+  tenants.value = await getSimpleTenantList();
 });
 </script>
 <template>
   <div v-if="tenantEnable && hasAccessByCodes(['system:tenant:visit'])">
     <Select
-      v-model:value="tenant"
+      v-model:value="value"
       :field-names="{ label: 'name', value: 'id' }"
-      :options="visitTenantList"
+      :options="tenants"
       :placeholder="$t('page.tenant.placeholder')"
       :dropdown-style="{ position: 'fixed', zIndex: 1666 }"
       allow-clear
       class="w-40"
-      @change="handleClick"
+      @change="handleChange"
     />
   </div>
 </template>
diff --git a/packages/stores/src/modules/access.ts b/packages/stores/src/modules/access.ts
index 605e78dd..785fffdf 100644
--- a/packages/stores/src/modules/access.ts
+++ b/packages/stores/src/modules/access.ts
@@ -6,11 +6,6 @@ import { acceptHMRUpdate, defineStore } from 'pinia';
 
 type AccessToken = null | string;
 
-type VisitTenantId = {
-  id: number;
-  name: string;
-};
-
 interface AccessState {
   /**
    * 权限码
@@ -55,7 +50,7 @@ interface AccessState {
   /**
    * 访问租户编号
    */
-  visitTenantId: VisitTenantId[];
+  visitTenantId: null | number;
 }
 
 /**
@@ -110,7 +105,7 @@ export const useAccessStore = defineStore('core-access', {
     setTenantId(tenantId: null | number) {
       this.tenantId = tenantId;
     },
-    setVisitTenantId(visitTenantId: VisitTenantId[]) {
+    setVisitTenantId(visitTenantId: number) {
       this.visitTenantId = visitTenantId;
     },
     unlockScreen() {
@@ -141,7 +136,7 @@ export const useAccessStore = defineStore('core-access', {
     loginExpired: false,
     refreshToken: null,
     tenantId: null,
-    visitTenantId: [],
+    visitTenantId: null,
   }),
 });