feat: monitor button

2026-02-04 20:14:25 +08:00
parent 47596fdb1a
commit 4e8b5e26ae
40 changed files with 352 additions and 176 deletions
@@ -0,0 +1,81 @@
+import { systemConfigAtom } from '@/atoms/system';
+import IconFont from '@/components/icon-font';
+import { GPUSTACK_API_BASE_URL } from '@/config/settings';
+import { useIntl } from '@umijs/max';
+import { Button } from 'antd';
+import { useAtomValue } from 'jotai';
+
+const useGranfanaLink = (options: {
+  type: 'model' | 'worker' | 'instance' | 'cluster';
+  dataList: any[];
+}) => {
+  const intl = useIntl();
+  const systemConfig = useAtomValue(systemConfigAtom);
+
+  const goToModelGrafana = (row: { id: number }) => {
+    window.open(
+      `/${GPUSTACK_API_BASE_URL}/models/${row.id}/dashboard`,
+      '_blank'
+    );
+  };
+
+  const goToWorkerGrafana = (row: { id: number }) => {
+    window.open(
+      `/${GPUSTACK_API_BASE_URL}/workers/${row.id}/dashboard`,
+      '_blank'
+    );
+  };
+
+  const goToInstanceGrafana = (row: { id: number }) => {
+    window.open(
+      `/${GPUSTACK_API_BASE_URL}/model-instances/${row.id}/dashboard`,
+      '_blank'
+    );
+  };
+
+  const goToClusterGrafana = (row: { id: number }) => {
+    window.open(
+      `/${GPUSTACK_API_BASE_URL}/clusters/${row.id}/dashboard`,
+      '_blank'
+    );
+  };
+
+  const goToGrafana = (row: { id: number }) => {
+    if (options.type === 'model') {
+      goToModelGrafana(row);
+    } else if (options.type === 'worker') {
+      goToWorkerGrafana(row);
+    } else if (options.type === 'instance') {
+      goToInstanceGrafana(row);
+    } else if (options.type === 'cluster') {
+      goToClusterGrafana(row);
+    }
+  };
+
+  const handleClick = () => {
+    if (options.dataList?.length > 0) {
+      goToGrafana(options.dataList?.[0]);
+    }
+  };
+
+  const ActionButton = () => {
+    if (systemConfig?.showMonitoring) {
+      return (
+        <Button
+          onClick={handleClick}
+          icon={<IconFont type="icon-metrics" style={{ fontSize: 16 }} />}
+        >
+          {intl.formatMessage({ id: 'resources.metrics.details' })}
+        </Button>
+      );
+    }
+    return null;
+  };
+
+  return {
+    goToGrafana,
+    ActionButton
+  };
+};
+
+export default useGranfanaLink;