// Copyright (c) 2021 Huawei Device Co., Ltd.
// 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 { Args } from '../CommonArgs';
import { TraficEnum } from '../utils/QueryEnum';

export const chartHiperfProcessData10MSProtoSql = (args: Args): string => {
  return `select startNS as startNS,
                 max(event_count)                                                         eventCount,
                 sample_count as sampleCount,
                 event_type_id as eventTypeId,
                 callchain_id as callchainId,
                 (startNS / (${Math.floor((args.endNS - args.startNS) / args.width)})) AS px
          from (SELECT sp.callchain_id,
                       (sp.timestamp_trace - ${args.recordStartNS}) / 10000000 * 10000000 startNS,
                       sum(event_count)                                                   event_count,
                       count(event_count)                                                 sample_count,
                       event_type_id
                from perf_sample sp
                where sp.thread_id in (select thread_id
                                       from perf_thread
                                       where perf_thread.process_id = ${args.pid})
                  and sp.thread_id != 0 ${args.drawType >= 0 ? 'and event_type_id =' + args.drawType : ''}
                group by startNS)
          where startNS + 10000000 >= ${Math.floor(args.startNS)}
            and startNS <= ${Math.floor(args.endNS)}
          group by px;`;
};
export const chartHiperfProcessDataProtoSql = (args: Args): string => {
  return `SELECT (sp.timestamp_trace - ${args.recordStartNS})          startNS,
                 event_count as eventCount,
                 1 as sampleCount,
                 event_type_id as eventTypeId,
                 sp.callchain_id as callchainId,
                 (sp.timestamp_trace - ${args.recordStartNS}) / (${Math.floor(
    (args.endNS - args.startNS) / args.width
  )}) AS px
          from perf_sample sp
          where sp.thread_id in (select thread_id
                                 from perf_thread
                                 where perf_thread.process_id = ${args.pid})
            and sp.thread_id != 0 ${args.drawType >= 0 ? 'and event_type_id =' + args.drawType : ''}
            and startNS >= ${Math.floor(args.startNS)}
            and startNS <= ${Math.floor(args.endNS)}
          group by px;`;
};

export function hiperfProcessDataReceiver(data: unknown, proc: Function): void {
  let sql: string;
  // @ts-ignore
  if (data.params.scale > 30_000_000) {
    // @ts-ignore
    sql = chartHiperfProcessData10MSProtoSql(data.params);
  } else {
    // @ts-ignore
    sql = chartHiperfProcessDataProtoSql(data.params);
  }
  let res = proc(sql);
  // @ts-ignore
  arrayBufferHandler(data, res, data.params.trafic !== TraficEnum.SharedArrayBuffer);
}

function arrayBufferHandler(data: unknown, res: unknown[], transfer: boolean): void {
  // @ts-ignore
  let maxCpuCount = data.params.maxCpuCount;
  // @ts-ignore
  let intervalPerf = data.params.intervalPerf;
  // @ts-ignore
  let usage = data.params.drawType === -2;
  let perfProcess = new PerfProcess(data, transfer, res.length);
  let maxEventCount = Math.max(
    ...res.map((it) => {
      // @ts-ignore
      data.params.trafic === TraficEnum.ProtoBuffer && (it = it.hiperfData);
      // @ts-ignore
      return it.eventCount;
    })
  );
  res.forEach((it, i) => {
    // @ts-ignore
    data.params.trafic === TraficEnum.ProtoBuffer && (it = it.hiperfData);
    // @ts-ignore
    perfProcess.startNS[i] = it.startNS || it.startNs;
    // @ts-ignore
    perfProcess.eventCount[i] = it.eventCount;
    // @ts-ignore
    perfProcess.sampleCount[i] = it.sampleCount;
    // @ts-ignore
    perfProcess.eventTypeId[i] = it.eventTypeId;
    // @ts-ignore
    perfProcess.callChainId[i] = it.callchainId;
    if (usage) {
      if (maxCpuCount === -1) {
        // @ts-ignore
        perfProcess.height[i] = Math.floor((it.sampleCount / (10 / intervalPerf)) * 40);
      } else {
        // @ts-ignore
        perfProcess.height[i] = Math.floor((it.sampleCount / (10 / intervalPerf) / maxCpuCount) * 40);
      }
    } else {
      // @ts-ignore
      perfProcess.height[i] = Math.floor((it.eventCount / maxEventCount) * 40);
    }
  });
  postPerfProcessMessage(data, transfer, perfProcess, res.length);
}
function postPerfProcessMessage(data: unknown, transfer: boolean, perfProcess: PerfProcess, len: number): void {
  (self as unknown as Worker).postMessage(
    {
      // @ts-ignore
      id: data.id,
      // @ts-ignore
      action: data.action,
      results: transfer
        ? {
            startNS: perfProcess.startNS.buffer,
            eventCount: perfProcess.eventCount.buffer,
            sampleCount: perfProcess.sampleCount.buffer,
            eventTypeId: perfProcess.eventTypeId.buffer,
            callChainId: perfProcess.callChainId.buffer,
            height: perfProcess.height.buffer,
          }
        : {},
      len: len,
      transfer: transfer,
    },
    transfer
      ? [
          perfProcess.startNS.buffer,
          perfProcess.eventCount.buffer,
          perfProcess.sampleCount.buffer,
          perfProcess.eventTypeId.buffer,
          perfProcess.callChainId.buffer,
          perfProcess.height.buffer,
        ]
      : []
  );
}
class PerfProcess {
  startNS: Float64Array;
  eventCount: Int32Array;
  sampleCount: Int32Array;
  eventTypeId: Int32Array;
  callChainId: Int32Array;
  height: Int32Array;
  constructor(data: unknown, transfer: boolean, len: number) {
    // @ts-ignore
    this.startNS = new Float64Array(transfer ? len : data.params.sharedArrayBuffers.startNS);
    // @ts-ignore
    this.eventCount = new Int32Array(transfer ? len : data.params.sharedArrayBuffers.eventCount);
    // @ts-ignore
    this.sampleCount = new Int32Array(transfer ? len : data.params.sharedArrayBuffers.sampleCount);
    // @ts-ignore
    this.eventTypeId = new Int32Array(transfer ? len : data.params.sharedArrayBuffers.eventTypeId);
    // @ts-ignore
    this.callChainId = new Int32Array(transfer ? len : data.params.sharedArrayBuffers.callChainId);
    // @ts-ignore
    this.height = new Int32Array(transfer ? len : data.params.sharedArrayBuffers.height);
  }
}