/*
 * Copyright (c) 2022-2023 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 "ressched_utils.h"

#include <dlfcn.h>
#include "cgroup_sched_log.h"
#include "hisysevent.h"
#include "nlohmann/json.hpp"

#undef LOG_TAG
#define LOG_TAG "ResSchedUtils"

namespace OHOS {
namespace ResourceSchedule {
namespace {
    const std::string RES_SCHED_SERVICE_SO = "libresschedsvc.z.so";
    const std::string RES_SCHED_CG_EXT_SO = "libcgroup_sched_ext.z.so";
}

ResSchedUtils& ResSchedUtils::GetInstance()
{
    static ResSchedUtils instance;
    return instance;
}

void ResSchedUtils::LoadUtils()
{
    auto handle = dlopen(RES_SCHED_SERVICE_SO.c_str(), RTLD_NOW);
    if (!handle) {
        CGS_LOGW("%{public}s load %{public}s failed!", __func__, RES_SCHED_SERVICE_SO.c_str());
        HiSysEventWrite(HiviewDFX::HiSysEvent::Domain::RSS, "INIT_FAULT", HiviewDFX::HiSysEvent::EventType::FAULT,
                        "COMPONENT_NAME", RES_SCHED_SERVICE_SO,
                        "ERR_TYPE", "plugin failure",
                        "ERR_MSG", "ResSchedUtils dlopen " + RES_SCHED_SERVICE_SO + " failed!");
        return;
    }

    reportFunc_ = reinterpret_cast<ReportDataFunc>(dlsym(handle, "ReportDataInProcess"));
    if (!reportFunc_) {
        CGS_LOGW("%{public}s load function:ReportDataInProcess failed!", __func__);
        HiSysEventWrite(HiviewDFX::HiSysEvent::Domain::RSS, "INIT_FAULT", HiviewDFX::HiSysEvent::EventType::FAULT,
                        "COMPONENT_NAME", RES_SCHED_SERVICE_SO,
                        "ERR_TYPE", "plugin failure",
                        "ERR_MSG", "ResSchedUtils don't found dlsym " + RES_SCHED_SERVICE_SO + "!");
        dlclose(handle);
        return;
    }
}

void ResSchedUtils::LoadUtilsExtra()
{
    auto handle = dlopen(RES_SCHED_CG_EXT_SO.c_str(), RTLD_NOW);
    if (!handle) {
        CGS_LOGD("%{public}s load %{public}s failed! errno:%{public}d", __func__, RES_SCHED_CG_EXT_SO.c_str(), errno);
        HiSysEventWrite(HiviewDFX::HiSysEvent::Domain::RSS, "INIT_FAULT", HiviewDFX::HiSysEvent::EventType::FAULT,
                        "COMPONENT_NAME", RES_SCHED_SERVICE_SO,
                        "ERR_TYPE", "plugin failure",
                        "ERR_MSG", "ResSchedUtils dlopen " + RES_SCHED_SERVICE_SO + " failed!");
        return;
    }

    reportArbitrationResultFunc_ =
        reinterpret_cast<ReportArbitrationResultFunc>(dlsym(handle, "ReportArbitrationResult"));
    if (!reportArbitrationResultFunc_) {
        CGS_LOGD("%{public}s load function:ReportArbitrationResult failed! errno:%{public}d", __func__, errno);
        HiSysEventWrite(HiviewDFX::HiSysEvent::Domain::RSS, "INIT_FAULT", HiviewDFX::HiSysEvent::EventType::FAULT,
                        "COMPONENT_NAME", RES_SCHED_SERVICE_SO,
                        "ERR_TYPE", "plugin failure",
                        "ERR_MSG", "ResSchedUtils don't found dlsym " + RES_SCHED_SERVICE_SO + "!");
        dlclose(handle);
        return;
    }

    reportSysEventFunc_ =
        reinterpret_cast<ReportSysEventFunc>(dlsym(handle, "ReportSysEvent"));
    if (!reportSysEventFunc_) {
        CGS_LOGD("%{public}s load function:ReportSysEvent failed! errno:%{public}d", __func__, errno);
        HiSysEventWrite(HiviewDFX::HiSysEvent::Domain::RSS, "INIT_FAULT", HiviewDFX::HiSysEvent::EventType::FAULT,
                        "COMPONENT_NAME", RES_SCHED_SERVICE_SO,
                        "ERR_TYPE", "plugin failure",
                        "ERR_MSG", "ResSchedUtils don't found ReportSysEvent in " + RES_SCHED_SERVICE_SO + "!");
        dlclose(handle);
        return;
    }
}

void ResSchedUtils::ReportDataInProcess(uint32_t resType, int64_t value, const nlohmann::json& payload)
{
    if (!reportFunc_) {
        CGS_LOGD("%{public}s failed, function nullptr.", __func__);
        return;
    }
    reportFunc_(resType, value, payload);
}

void ResSchedUtils::ReportArbitrationResult(Application &app, ProcessRecord &pr, AdjustSource source)
{
    if (!reportArbitrationResultFunc_) {
        CGS_LOGD("%{public}s failed, function nullptr.", __func__);
        return;
    }
    reportArbitrationResultFunc_(app, pr, source);
}

void ResSchedUtils::ReportSysEvent(Application &app, ProcessRecord &pr, uint32_t resType, int32_t state)
{
    if (!reportSysEventFunc_) {
        CGS_LOGD("%{public}s failed, function nullptr.", __func__);
        return;
    }
    reportSysEventFunc_(app, pr, resType, state);
}
} // namespace ResourceSchedule
} // namespace OHOS