this_thread.hpp

浏览该文件的文档.

#ifndef MSTL_CORE_ASYNC_THIS_THREAD_HPP__
#define MSTL_CORE_ASYNC_THIS_THREAD_HPP__

 
#include "MSTL/core/config/c++config.hpp"
#ifdef MSTL_PLATFORM_WINDOWS__
#include <Windows.h>
#ifdef max
#undef max
#endif
#ifdef min
#undef min
#endif
#endif
#ifdef MSTL_PLATFORM_LINUX__
#include <sched.h>
#include <unistd.h>
#endif
#ifdef MSTL_ARCH_RISCV__
#include <riscv_pause.h>
#endif
MSTL_BEGIN_NAMESPACE__

MSTL_BEGIN_THIS_THREAD__


MSTL_ALWAYS_INLINE_INLINE int64_t current_cpu() noexcept {
#ifdef MSTL_PLATFORM_WINDOWS__
    return static_cast<int64_t>(::GetCurrentProcessorNumber());
#else
    return static_cast<int64_t>(::sched_getcpu());
#endif
}

MSTL_ALWAYS_INLINE_INLINE void yield() noexcept {
#ifdef MSTL_PLATFORM_WINDOWS__
    ::SwitchToThread();
#else
    ::sched_yield();
#endif
}

MSTL_ALWAYS_INLINE_INLINE void relax() noexcept {
#ifdef MSTL_PLATFORM_WINDOWS__
    ::_mm_pause();
#else
#if defined(MSTL_ARCH_X86__)
    __builtin_ia32_pause();
#elif defined(MSTL_ARCH_ARM__)
    asm volatile("yield" ::: "memory");
#elif defined(MSTL_ARCH_RISCV__)
    ::riscv_pause();
#elif defined(MSTL_ARCH_LOONGARCH__)
    asm volatile("dbar 0" ::: "memory");
#else
    this_thread::yield();
#endif
#endif
}

MSTL_ALWAYS_INLINE_INLINE void relax(const int count) noexcept {
    if (count < 4) {
        for (int i = 0; i < (1 << count); ++i) {
            this_thread::relax();
        }
    } else if (count < 8) {
        this_thread::relax();
    } else {
        this_thread::yield();
    }
}

MSTL_ALWAYS_INLINE_INLINE void sleep_for_ms(uint32_t milliseconds) noexcept {
#ifdef MSTL_PLATFORM_WINDOWS__
    ::Sleep(milliseconds);
#else
    ::usleep(milliseconds * 1000);
#endif
}

void MSTL_API sleep_for_ms(uint32_t ms, bool busy_wait) noexcept;

void MSTL_API sleep_for_us(uint64_t ms) noexcept;

void MSTL_API sleep_for_ns(uint64_t ns) noexcept;

bool MSTL_API affinity(size_t cpu_mask) noexcept;

bool MSTL_API priority(int priority) noexcept;
 // Thread
 
MSTL_END_THIS_THREAD__
MSTL_END_NAMESPACE__
#endif // MSTL_CORE_ASYNC_THIS_THREAD_HPP__