/*
 * 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.
 */
#include "ecmascript/cpu_profiler/profile_processor.h"

#include <csignal>
#include <sys/time.h>
#include "ecmascript/cpu_profiler/cpu_profiler.h"
#include "ecmascript/interpreter/interpreter.h"

namespace panda::ecmascript {
JSThread *ProfileProcessor::thread_ = nullptr;
bool ProfileProcessor::isStart_ = true;
ProfileProcessor::ProfileProcessor(ProfileGenerator *generator, const EcmaVM *vm, int interval)
{
    generator_ = generator;
    interval_ = interval;
    pid_ = pthread_self();
    thread_ = vm->GetAssociatedJSThread();
}
ProfileProcessor::~ProfileProcessor() {}

bool ProfileProcessor::Run(uint32_t threadIndex)
{
    uint64_t startTime = 0;
    uint64_t endTime = 0;
    startTime = GetMicrosecondsTimeStamp();
    generator_->SetThreadStartTime(startTime);
    while (isStart_) {
        static int collectCount = 1;
#if ECMASCRIPT_ENABLE_ACTIVE_CPUPROFILER
        JSThread *thread = GetJSThread();
        ProfileGenerator::staticGcState_ = thread->GetGcState();
        if (!ProfileGenerator::staticGcState_) {
            thread->SetGetStackSignal(true);
            if (sem_wait(&CpuProfiler::sem_) != 0) {
                LOG(ERROR, RUNTIME) << "sem_ wait failed";
            }
        }
#else
        if (pthread_kill(pid_, SIGINT) != 0) {
            LOG(ERROR, RUNTIME) << "pthread_kill signal failed";
            return false;
        }
        if (sem_wait(&CpuProfiler::sem_) != 0) {
            LOG(ERROR, RUNTIME) << "sem_ wait failed";
            return false;
        }
#endif
        endTime = GetMicrosecondsTimeStamp();
        generator_->AddSample(CpuProfiler::staticFrameStack_, endTime);
        if (generator_->GetMethodNodes().size() >= 10 || // 10:Number of nodes currently stored
            generator_->GetSamples().size() == 100) { // 100:Number of Samples currently stored
            generator_->WriteMethodsAndSampleInfo(false);
        }
        int64_t ts = interval_ - (endTime - startTime);
        if (ts > 0) {
            usleep(ts);
        }
        startTime = GetMicrosecondsTimeStamp();
        if (collectCount % 50 == 0) { // 50:The sampling times reached 50 times.
            WriteSampleDataToFile();
        }
        collectCount++;
    }
    if (sem_post(&CpuProfiler::sem_) != 0) {
        LOG(ERROR, RUNTIME) << "sem_ post failed";
        return false;
    }
    return true;
}

uint64_t ProfileProcessor::GetMicrosecondsTimeStamp()
{
    struct timeval time;
    gettimeofday(&time, NULL);
    return (time.tv_sec * 1000000 + time.tv_usec); // 1000000:Second to subtle
}

JSThread *ProfileProcessor::GetJSThread()
{
    return thread_;
}

void ProfileProcessor::SetIsStart(bool isStart)
{
    isStart_ = isStart;
}

void ProfileProcessor::WriteSampleDataToFile()
{
    static int flag = 0;
    if (!generator_->fileHandle_.is_open() || generator_->GetSampleData().size() < 1024 * 1024) { // 1024 * 1024:1M
        return;
    }
    if (flag == 0) {
        generator_->fileHandle_ << generator_->GetSampleData();
        generator_->ClearSampleData();
        generator_->fileHandle_.close();
        generator_->fileHandle_.open(generator_->GetFileName().c_str(), std::ios::app);
        flag++;
        return;
    }
    generator_->fileHandle_ << generator_->GetSampleData();
    generator_->ClearSampleData();
}
} // namespace panda::ecmascript