fix:增加AI状态检查功能

Author: fenyf

src/common/network/ai.ts

@@ -0,0 +1,56 @@
+/*
+ * Copyright 2025 OceanBase
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+import request from '@/util/request';
+
+/**
+ * AI功能状态响应接口
+ */
+export interface IAIStatusResponse {
+  /** AI功能是否启用 */
+  enabled: boolean;
+  /** AI功能是否可用 */
+  available: boolean;
+  /** 使用的AI模型 */
+  model: string;
+  /** API基础URL */
+  baseUrl: string;
+  /** API密钥是否已配置 */
+  apiKeyConfigured: boolean;
+}
+
+/**
+ * 查询AI功能状态
+ * @returns AI功能状态信息
+ */
+export async function getAIStatus(): Promise<IAIStatusResponse> {
+  const res = await request.get('/api/v2/ai/status');
+  return res?.data;
+}
+
+/**
+ * 检查AI功能是否可用
+ * @returns 是否可用
+ */
+export async function isAIAvailable(): Promise<boolean> {
+  try {
+    const status = await getAIStatus();
+    return status.enabled && status.available;
+  } catch (error) {
+    console.warn('Failed to check AI status:', error);
+    return false;
+  }
+}

src/component/SensitiveColumnIndicator/index.tsx

@@ -8,6 +8,7 @@ import { EyeInvisibleOutlined, SecurityScanOutlined, SyncOutlined } from '@ant-d
 import { formatMessage } from '@/util/intl';
 import sensitiveColumnScanner, { IScanResult } from '@/service/sensitiveColumnScanner';
 import { ISensitiveColumnInfo } from '@/d.ts/sensitiveColumn';
+import { isAIAvailable } from '@/common/network/ai';
 import styles from './index.less';
 
 interface ISensitiveColumnIndicatorProps {
@@ -72,6 +73,21 @@ const SensitiveColumnIndicator: React.FC<ISensitiveColumnIndicatorProps> = ({
   const [scanError, setScanError] = useState<string | null>(null);
   const [retryCount, setRetryCount] = useState(0);
   const [isRetrying, setIsRetrying] = useState(false);
+  const [aiAvailable, setAiAvailable] = useState<boolean>(true);
+
+  // 检查AI功能状态
+  useEffect(() => {
+    const checkAIStatus = async () => {
+      try {
+        const available = await isAIAvailable();
+        setAiAvailable(available);
+      } catch (error) {
+        console.warn('检查AI状态失败:', error);
+        setAiAvailable(false);
+      }
+    };
+    checkAIStatus();
+  }, []);
 
   // 使用传入的敏感列数据或扫描结果
   const sensitiveColumns = propSensitiveColumns || scanResult?.sensitiveColumns || [];
@@ -199,6 +215,15 @@ const SensitiveColumnIndicator: React.FC<ISensitiveColumnIndicatorProps> = ({
       sensitiveColumnScanner.removeCacheChangeListener(handleCacheChange);
     };
   }, [tableName, databaseName, sessionId, autoScan, propSensitiveColumns]);
+
+  // 如果外部已提供敏感列数据，直接显示，不受AI状态影响
+  if (propSensitiveColumns && propSensitiveColumns.length > 0) {
+    // 外部提供的数据，直接渲染，跳过AI状态检查
+  } else if (!aiAvailable) {
+    // 只有在需要扫描且AI未开启时才隐藏组件
+    return null;
+  }
+
   // 如果正在扫描，显示扫描状态
   if (isScanning) {
     return (

src/page/Project/Sensitive/components/SensitiveColumn/components/SacnRule.tsx

@@ -30,6 +30,7 @@ import { useContext, useEffect, useState } from 'react';
 import SensitiveContext from '../../../SensitiveContext';
 import MultipleDatabaseSelect from '@/component/Task/component/MultipleDatabaseSelect/index';
 import { isConnectTypeBeFileSystemGroup } from '@/util/connection';
+import { isAIAvailable } from '@/common/network/ai';
 
 const ScanRule = ({ formRef, reset, setManageSensitiveRuleDrawerOpen }) => {
   const context = useContext(ProjectContext);
@@ -42,6 +43,35 @@ const ScanRule = ({ formRef, reset, setManageSensitiveRuleDrawerOpen }) => {
   const [dataSourceOptions, setDataSourceOptions] = useState<SelectItemProps[]>([]);
   const [sensitiveOptions, setSensitiveOptions] = useState<SelectItemProps[]>([]);
   const [rawData, setRawData] = useState<IResponseData<IConnection>>();
+  const [aiAvailable, setAiAvailable] = useState<boolean>(true);
+  // 检查AI功能状态
+  useEffect(() => {
+    const checkAIStatus = async () => {
+      try {
+        const available = await isAIAvailable();
+        setAiAvailable(available);
+
+        // 如果AI不可用且当前选择的是AI增强识别，自动切换到传统规则识别
+        if (!available && scanningMode === 'JOINT_RECOGNITION') {
+          await formRef.setFieldsValue({
+            scanningMode: 'RULES_ONLY',
+          });
+        }
+      } catch (error) {
+        console.warn('检查AI状态失败:', error);
+        setAiAvailable(false);
+
+        // AI检查失败时也切换到传统规则识别
+        if (scanningMode === 'JOINT_RECOGNITION') {
+          await formRef.setFieldsValue({
+            scanningMode: 'RULES_ONLY',
+          });
+        }
+      }
+    };
+    checkAIStatus();
+  }, []);
+
   const initDataSources = async () => {
     const rawData = await getConnectionList({
       projectId: sensitiveContext.projectId,
@@ -283,11 +313,12 @@ const ScanRule = ({ formRef, reset, setManageSensitiveRuleDrawerOpen }) => {
                 defaultMessage: '传统规则识别',
               })}
             </Radio>
-            <Radio value="JOINT_RECOGNITION">
+            <Radio value="JOINT_RECOGNITION" disabled={!aiAvailable}>
               {formatMessage({
                 id: 'odc.SensitiveColumn.components.SacnRule.AIEnhanced',
                 defaultMessage: 'AI增强识别',
               })}
+              {!aiAvailable && <span style={{ color: '#999', marginLeft: 8 }}>(AI功能未开启)</span>}
             </Radio>
           </Radio.Group>
         </Form.Item>