VC++ PDH/性能计数器

例子:

PID=0,缺省为当前进程,但最好是获取当前进程ID传递进去,当然也可以选择其它进程的ID。

PerformanceCounter pc;

pc.Open(0, "//Processor(_Total)//% Processor Time");

源实现:

#include <windows.h>
#include <pdh.h>

        class PerformanceCounter
        {
        public:
            PerformanceCounter() noexcept;
            virtual ~PerformanceCounter() noexcept;

        public:
            virtual double              Next() noexcept;
            virtual void                Open(int pid, LPCSTR counter);
            virtual void                Dispose() noexcept;

        private:
            void                        Release() noexcept;

        private:
            std::atomic<void*>          m_phQuery   = NULL;
            std::atomic<void*>          m_phCounter = NULL;
        };

        PerformanceCounter::PerformanceCounter() noexcept
            : m_phQuery(NULL)
            , m_phCounter(NULL)
        {

        }

        PerformanceCounter::~PerformanceCounter() noexcept
        {
            Release();
        }

        void PerformanceCounter::Open(int pid, LPCSTR counter)
        {
            void* phQuery = NULL;
            if (PdhOpenQueryA(NULL, pid, &phQuery) != ERROR_SUCCESS)
            {
                throw std::exception("The handle to the PerformanceCounter could not be opened.");
            }

            void* phCounter = NULL;
            if (PdhAddCounterA(phQuery, counter, 0, &phCounter) != ERROR_SUCCESS)
            {
                PdhCloseQuery(phCounter);
                throw std::exception("Unable to add a performance counter instance.");
            }
            else
            {
                Release();
            }

            m_phQuery.exchange(phQuery);
            m_phCounter.exchange(phCounter);
        }

        double PerformanceCounter::Next() noexcept
        {
            if (m_phQuery == NULL)
            {
                return 0;
            }
            else
            {
                PdhCollectQueryData(m_phQuery);
            }

            PDH_FMT_COUNTERVALUE counterValue;
            if (PdhGetFormattedCounterValue(m_phCounter, PDH_FMT_DOUBLE, NULL, &counterValue) == ERROR_SUCCESS)
            {
                return counterValue.doubleValue;
            }

            return 0;
        }

        void PerformanceCounter::Dispose() noexcept
        {
            Release();
        }

        void PerformanceCounter::Release() noexcept
        {
            void* phCounter = m_phCounter.exchange(NULL);
            if (NULL != phCounter)
            {
                PdhRemoveCounter(phCounter);
            }

            void* phQuery = m_phQuery.exchange(NULL);
            if (phQuery != NULL)
            {
                PdhCloseQuery(phQuery);
            }
        }

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值