ark_runtime_core/libpandabase/os/windows/thread.cpp

/*
 * Copyright (c) 2021-2022 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 "os/thread.h"
#include "utils/logger.h"

#include <errhandlingapi.h>
#include <handleapi.h>
#include <processthreadsapi.h>
#include <thread>

namespace panda::os::thread {
ThreadId GetCurrentThreadId()
{
    // The function is provided by mingw
    return ::GetCurrentThreadId();
}

int GetPid()
{
    return _getpid();
}

int SetPriority(DWORD thread_id, int prio)
{
    // The priority can be set within range [-2, 2], and -2 is the lowest priority.
    HANDLE thread = OpenThread(THREAD_SET_INFORMATION, false, thread_id);
    if (thread == NULL) {
        LOG(FATAL, COMMON) << "OpenThread failed, error code " << GetLastError();
    }
    auto ret = SetThreadPriority(thread, prio);
    CloseHandle(thread);
    // The return value is nonzero if the function succeeds, and zero if it fails.
    return ret;
}

int GetPriority(DWORD thread_id)
{
    HANDLE thread = OpenThread(THREAD_QUERY_INFORMATION, false, thread_id);
    if (thread == NULL) {
        LOG(FATAL, COMMON) << "OpenThread failed, error code " << GetLastError();
    }
    auto ret = GetThreadPriority(thread);
    CloseHandle(thread);
    return ret;
}

int SetThreadName(native_handle_type pthread_handle, const char *name)
{
    ASSERT(pthread_handle != 0);
    pthread_t thread = reinterpret_cast<pthread_t>(pthread_handle);
    return pthread_setname_np(thread, name);
}

native_handle_type GetNativeHandle()
{
    return reinterpret_cast<native_handle_type>(pthread_self());
}

void Yield()
{
    std::this_thread::yield();
}

void NativeSleep(unsigned int ms)
{
    std::this_thread::sleep_for(std::chrono::milliseconds(ms));
}

void ThreadDetach(native_handle_type pthread_handle)
{
    pthread_detach(reinterpret_cast<pthread_t>(pthread_handle));
}

void ThreadExit(void *ret)
{
    pthread_exit(ret);
}

void ThreadJoin(native_handle_type pthread_handle, void **ret)
{
    pthread_join(reinterpret_cast<pthread_t>(pthread_handle), ret);
}
}  // namespace panda::os::thread