* -------------------------------------------------------------------------
* This file is part of the MindStudio project.
* Copyright (c) 2025 Huawei Technologies Co.,Ltd.
*
* MindStudio is licensed under Mulan PSL v2.
* You can use this software according to the terms and conditions of the Mulan PSL v2.
* You may obtain a copy of Mulan PSL v2 at:
*
* http://license.coscl.org.cn/MulanPSL2
*
* THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND,
* EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT,
* MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
* See the Mulan PSL v2 for more details.
* -------------------------------------------------------------------------
*/
#include "pch.h"
#include "AdvisorProtocolRequest.h"
#include "AdvisorProtocolResponse.h"
#include "AffinityAPIAdvisor.h"
#include "WsSessionManager.h"
#include "TraceTime.h"
#include "QueryAffinityAPIAdvice.h"
namespace Dic::Module::Advisor {
using namespace Dic::Server;
bool QueryAffinityAPIAdvice::HandleRequest(std::unique_ptr<Protocol::Request> requestPtr) {
auto &request = dynamic_cast<AffinityAPIRequest &>(*requestPtr);
std::unique_ptr<AffinityAPIResponse> responsePtr = std::make_unique<AffinityAPIResponse>();
AffinityAPIResponse &response = *responsePtr;
SetBaseResponse(request, response);
SetResponseResult(response, true);
uint64_t minTimestamp = Timeline::TraceTime::Instance().GetStartTime();
std::string error;
request.params.Check(minTimestamp, error);
if (!std::empty(error)) {
ServerLog::Error(error);
SetAdvisorError(ErrorCode::PARAMS_ERROR);
SendResponse(std::move(responsePtr), false);
return false;
}
if (!AffinityAPIAdvisor::Process(request.params, response.body)) {
ServerLog::Error("Failed to Query Affinity API Advice for rank ", request.params.rankId);
SendResponse(std::move(responsePtr), false);
return false;
}
SendResponse(std::move(responsePtr), true);
return true;
}
}
