149 lines
4.2 KiB
TypeScript
149 lines
4.2 KiB
TypeScript
/*
|
||
* Copyright 2025 coze-dev Authors
|
||
*
|
||
* 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 { devtools } from 'zustand/middleware';
|
||
import { create } from 'zustand';
|
||
import { type CSpan } from '@coze-devops/common-modules/query-trace';
|
||
import {
|
||
type GetTracesMetaInfoData,
|
||
type Span,
|
||
} from '@coze-arch/bot-api/ob_query_api';
|
||
|
||
import {
|
||
type BasicInfo,
|
||
type QueryFilterItemId,
|
||
type TargetOverallSpanInfo,
|
||
} from '../typings';
|
||
import { IS_DEV_MODE } from '../consts/env';
|
||
import { FILTERING_OPTION_ALL } from '../consts';
|
||
|
||
export type SpanCategory = GetTracesMetaInfoData['span_category'];
|
||
|
||
interface DebugPanelStore {
|
||
isPanelShow: boolean;
|
||
basicInfo: BasicInfo;
|
||
/**
|
||
* 当前选中的Query LogID
|
||
*/
|
||
entranceMessageLogId?: string;
|
||
/**
|
||
* 日期筛选结果
|
||
*/
|
||
targetDateId?: QueryFilterItemId;
|
||
/**
|
||
* 状态筛选结果
|
||
*/
|
||
targetExecuteStatusId?: QueryFilterItemId;
|
||
/**
|
||
* 当前选中的Trace节点信息
|
||
*/
|
||
targetOverallSpanInfo?: TargetOverallSpanInfo;
|
||
/**
|
||
* 当前计算后的Trace列表
|
||
*/
|
||
enhancedOverallSpans: CSpan[];
|
||
/**
|
||
* 某条Trace下Span节点列表
|
||
*/
|
||
orgDetailSpans?: Span[];
|
||
/**
|
||
* 额外Span类型信息（服务端提供）
|
||
*/
|
||
spanCategory?: SpanCategory;
|
||
/**
|
||
* 当前选中的Span节点信息
|
||
*/
|
||
targetDetailSpan?: CSpan;
|
||
curBatchPage?: number;
|
||
}
|
||
|
||
interface DebugPanelAction {
|
||
setIsPanelShow: (isPanelShow: boolean) => void;
|
||
setBasicInfo: (basicInfo: BasicInfo) => void;
|
||
setEntranceMessageLogId: (entranceMessageLogId: string) => void;
|
||
setTargetOverallSpanInfo: (overallSpanInfo: TargetOverallSpanInfo) => void;
|
||
onSelectDate: (dateId: QueryFilterItemId) => void;
|
||
onSelectExecuteStatus: (executeStatusId: QueryFilterItemId) => void;
|
||
setEnhancedOverallSpans: (enhancedOverallSpans: CSpan[]) => void;
|
||
setOrgDetailSpans: (orgDetailSpans: Span[]) => void;
|
||
setSpanCategory: (spanCategory?: SpanCategory) => void;
|
||
setTargetDetailSpan: (targetDetailSpan?: CSpan) => void;
|
||
setCurBatchPage: (curBatchPage: number) => void;
|
||
resetStore: () => void;
|
||
}
|
||
|
||
const initialStore: DebugPanelStore = {
|
||
isPanelShow: false,
|
||
basicInfo: {
|
||
placement: 'left',
|
||
},
|
||
entranceMessageLogId: undefined,
|
||
targetDateId: FILTERING_OPTION_ALL,
|
||
targetExecuteStatusId: FILTERING_OPTION_ALL,
|
||
enhancedOverallSpans: [],
|
||
targetOverallSpanInfo: undefined,
|
||
orgDetailSpans: undefined,
|
||
targetDetailSpan: undefined,
|
||
};
|
||
|
||
export const useDebugPanelStore = create<DebugPanelStore & DebugPanelAction>()(
|
||
devtools(
|
||
set => ({
|
||
...initialStore,
|
||
setIsPanelShow: (isPanelShow: boolean) => {
|
||
set({ isPanelShow });
|
||
},
|
||
setBasicInfo: basicInfo => {
|
||
set({ basicInfo });
|
||
},
|
||
setTargetOverallSpanInfo: overallSpanInfo => {
|
||
set({ targetOverallSpanInfo: overallSpanInfo });
|
||
},
|
||
onSelectDate: dateId => {
|
||
set({ targetDateId: dateId });
|
||
},
|
||
onSelectExecuteStatus: executeStatusId => {
|
||
set({ targetExecuteStatusId: executeStatusId });
|
||
},
|
||
setEnhancedOverallSpans: enhancedOverallSpans => {
|
||
set({ enhancedOverallSpans });
|
||
},
|
||
setEntranceMessageLogId: entranceMessageLogId => {
|
||
set({ entranceMessageLogId });
|
||
},
|
||
setOrgDetailSpans: orgDetailSpans => {
|
||
set({ orgDetailSpans });
|
||
},
|
||
setSpanCategory: spanCategory => {
|
||
set({ spanCategory });
|
||
},
|
||
setTargetDetailSpan: targetDetailSpan => {
|
||
set({ targetDetailSpan });
|
||
},
|
||
setCurBatchPage: curBatchPage => {
|
||
set({ curBatchPage });
|
||
},
|
||
resetStore: () => {
|
||
set(initialStore);
|
||
},
|
||
}),
|
||
{
|
||
enabled: IS_DEV_MODE,
|
||
name: 'debug.debugPanelStore',
|
||
},
|
||
),
|
||
);
|