1// Copyright (c) 2021 Huawei Device Co., Ltd.
2// Licensed under the Apache License, Version 2.0 (the "License");
3// you may not use this file except in compliance with the License.
4// You may obtain a copy of the License at
5//
6//     http://www.apache.org/licenses/LICENSE-2.0
7//
8// Unless required by applicable law or agreed to in writing, software
9// distributed under the License is distributed on an "AS IS" BASIS,
10// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
11// See the License for the specific language governing permissions and
12// limitations under the License.
13
14import { CHART_OFFSET_LEFT, MAX_COUNT, QueryEnum, TraficEnum } from './utils/QueryEnum';
15import { threadPool } from '../SqlLite';
16import { TraceRow } from '../../component/trace/base/TraceRow';
17import { EBPFChartStruct } from '../ui-worker/ProcedureWorkerEBPF';
18
19export function fileSystemSender(
20  type: number,
21  scale: number,
22  row: TraceRow<EBPFChartStruct>
23): Promise<EBPFChartStruct[]> {
24  let trafic: number = TraficEnum.TransferArrayBuffer;
25  let width = row.clientWidth - CHART_OFFSET_LEFT;
26  if (trafic === TraficEnum.SharedArrayBuffer && !row.sharedArrayBuffers) {
27    row.sharedArrayBuffers = {
28      endNS: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT),
29      startNS: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT),
30      size: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT),
31    };
32  }
33  return new Promise((resolve, reject): void => {
34    threadPool.submitProto(
35      QueryEnum.FileSystemData,
36      {
37        startNS: TraceRow.range?.startNS || 0,
38        endNS: TraceRow.range?.endNS || 0,
39        recordStartNS: window.recordStartNS,
40        recordEndNS: window.recordEndNS,
41        width: width,
42        trafic: trafic,
43        sharedArrayBuffers: row.sharedArrayBuffers,
44        type: type,
45        scale: scale,
46      },
47      (res: unknown, len: number, transfer: boolean): void => {
48        resolve(arrayBufferHandler(transfer ? res : row.sharedArrayBuffers, len));
49      }
50    );
51  });
52}
53
54export function diskIoSender(
55  all: boolean,
56  ipid: number,
57  typeArr: Array<number>,
58  scale: number,
59  row: TraceRow<EBPFChartStruct>
60): Promise<EBPFChartStruct[]> {
61  let trafic: number = TraficEnum.TransferArrayBuffer;
62  let width = row.clientWidth - CHART_OFFSET_LEFT;
63  if (trafic === TraficEnum.SharedArrayBuffer && !row.sharedArrayBuffers) {
64    row.sharedArrayBuffers = {
65      endNS: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT),
66      startNS: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT),
67      dur: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT),
68    };
69  }
70  return new Promise((resolve, reject): void => {
71    threadPool.submitProto(
72      QueryEnum.DiskIoData,
73      {
74        startNS: TraceRow.range?.startNS || 0,
75        endNS: TraceRow.range?.endNS || 0,
76        recordStartNS: window.recordStartNS,
77        recordEndNS: window.recordEndNS,
78        width: width,
79        trafic: trafic,
80        sharedArrayBuffers: row.sharedArrayBuffers,
81        all: all,
82        ipid: ipid,
83        typeArr: typeArr,
84        scale: scale,
85      },
86      (res: unknown, len: number, transfer: boolean) => {
87        resolve(arrayBufferHandler(transfer ? res : row.sharedArrayBuffers, len));
88      }
89    );
90  });
91}
92export function fileSysVMSender(scale: number, row: TraceRow<EBPFChartStruct>): Promise<EBPFChartStruct[]> {
93  let trafic: number = TraficEnum.TransferArrayBuffer;
94  let width = row.clientWidth - CHART_OFFSET_LEFT;
95  if (trafic === TraficEnum.SharedArrayBuffer && !row.sharedArrayBuffers) {
96    row.sharedArrayBuffers = {
97      endNS: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT),
98      startNS: new SharedArrayBuffer(Float64Array.BYTES_PER_ELEMENT * MAX_COUNT),
99      dur: new SharedArrayBuffer(Int32Array.BYTES_PER_ELEMENT * MAX_COUNT),
100    };
101  }
102  return new Promise((resolve, reject) => {
103    threadPool.submitProto(
104      QueryEnum.EBPFVm,
105      {
106        startNS: TraceRow.range?.startNS || 0,
107        endNS: TraceRow.range?.endNS || 0,
108        recordStartNS: window.recordStartNS,
109        recordEndNS: window.recordEndNS,
110        width: width,
111        trafic: trafic,
112        sharedArrayBuffers: row.sharedArrayBuffers,
113        scale: scale,
114      },
115      (res: unknown, len: number, transfer: boolean) => {
116        resolve(arrayBufferHandler(transfer ? res : row.sharedArrayBuffers, len));
117      }
118    );
119  });
120}
121function arrayBufferHandler(buffers: unknown, len: number): EBPFChartStruct[] {
122  let outArr: EBPFChartStruct[] = [];
123  // @ts-ignore
124  let endNS = new Float64Array(buffers.endNS);
125  // @ts-ignore
126  let startNS = new Float64Array(buffers.startNS);
127  // @ts-ignore
128  let size = new Float64Array(buffers.size);
129  // @ts-ignore
130  let dur = new Float64Array(buffers.dur);
131  // @ts-ignore
132  let height = new Int32Array(buffers.height);
133  for (let i = 0; i < len; i++) {
134    outArr.push({
135      size: size[i],
136      dur: dur[i],
137      endNS: endNS[i],
138      startNS: startNS[i],
139      height: height[i],
140    } as unknown as EBPFChartStruct);
141  }
142  return outArr;
143}
144