oshi.util.platform.windows.PerfCounterQuery Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of driver-cql-shaded Show documentation
Show all versions of driver-cql-shaded Show documentation
A Shaded CQL ActivityType driver for http://nosqlbench.io/
/**
* MIT License
*
* Copyright (c) 2010 - 2020 The OSHI Project Contributors: https://github.com/oshi/oshi/graphs/contributors
*
* Permission is hereby granted, free of charge, to any person obtaining a copy
* of this software and associated documentation files (the "Software"), to deal
* in the Software without restriction, including without limitation the rights
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
* copies of the Software, and to permit persons to whom the Software is
* furnished to do so, subject to the following conditions:
*
* The above copyright notice and this permission notice shall be included in all
* copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
package oshi.util.platform.windows;
import java.util.EnumMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.locks.ReentrantLock;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.sun.jna.platform.win32.PdhUtil; //NOSONAR
import com.sun.jna.platform.win32.PdhUtil.PdhException;
import com.sun.jna.platform.win32.Win32Exception;
import com.sun.jna.platform.win32.COM.Wbemcli;
import com.sun.jna.platform.win32.COM.WbemcliUtil.WmiQuery;
import com.sun.jna.platform.win32.COM.WbemcliUtil.WmiResult;
import oshi.annotation.concurrent.GuardedBy;
import oshi.annotation.concurrent.ThreadSafe;
import oshi.util.platform.windows.PerfDataUtil.PerfCounter;
/**
* Enables queries of Performance Counters using wild cards to filter instances
*/
@ThreadSafe
public final class PerfCounterQuery {
private static final Logger LOG = LoggerFactory.getLogger(PerfCounterQuery.class);
// Use a map to cache failed pdh queries
@GuardedBy("failedQueryCacheLock")
private static final Set failedQueryCache = new HashSet<>();
private static final ReentrantLock failedQueryCacheLock = new ReentrantLock();
/*
* Multiple classes use these constants
*/
public static final String TOTAL_INSTANCE = "_Total";
public static final String TOTAL_INSTANCES = "*_Total";
public static final String NOT_TOTAL_INSTANCE = "^" + TOTAL_INSTANCE;
public static final String NOT_TOTAL_INSTANCES = "^" + TOTAL_INSTANCES;
private PerfCounterQuery() {
}
/**
* Query the a Performance Counter using PDH, with WMI backup on failure, for
* values corresponding to the property enum.
*
* @param
* The enum type of {@code propertyEnum}
* @param propertyEnum
* An enum which implements
* {@link oshi.util.platform.windows.PerfCounterQuery.PdhCounterProperty}
* and contains the WMI field (Enum value) and PDH Counter string
* (instance and counter)
* @param perfObject
* The PDH object for this counter; all counters on this object will
* be refreshed at the same time
* @param perfWmiClass
* The WMI PerfData_RawData_* class corresponding to the PDH object
* @return An {@link EnumMap} of the values indexed by {@code propertyEnum} on
* success, or an empty map if both PDH and WMI queries failed.
*/
public static > Map queryValues(Class propertyEnum, String perfObject,
String perfWmiClass) {
// Check without locking for performance
if (!failedQueryCache.contains(perfObject)) {
failedQueryCacheLock.lock();
try {
// Double check lock
if (!failedQueryCache.contains(perfObject)) {
Map valueMap = queryValuesFromPDH(propertyEnum, perfObject);
if (!valueMap.isEmpty()) {
return valueMap;
}
// If we are here, query failed
LOG.warn("Disabling further attempts to query {}.", perfObject);
failedQueryCache.add(perfObject);
}
} finally {
failedQueryCacheLock.unlock();
}
}
return queryValuesFromWMI(propertyEnum, perfWmiClass);
}
/**
* Query the a Performance Counter using PDH for values corresponding to the
* property enum.
*
* @param
* The enum type of {@code propertyEnum}
* @param propertyEnum
* An enum which implements
* {@link oshi.util.platform.windows.PerfCounterQuery.PdhCounterProperty}
* and contains the WMI field (Enum value) and PDH Counter string
* (instance and counter)
* @param perfObject
* The PDH object for this counter; all counters on this object will
* be refreshed at the same time
* @return An {@link EnumMap} of the values indexed by {@code propertyEnum} on
* success, or an empty map if the PDH query failed.
*/
public static > Map queryValuesFromPDH(Class propertyEnum, String perfObject) {
T[] props = propertyEnum.getEnumConstants();
String perfObjectLocalized = localize(perfObject);
EnumMap counterMap = new EnumMap<>(propertyEnum);
EnumMap valueMap = new EnumMap<>(propertyEnum);
try (PerfCounterQueryHandler pdhQueryHandler = new PerfCounterQueryHandler()) {
// Set up the query and counter handles
for (T prop : props) {
PerfCounter counter = PerfDataUtil.createCounter(perfObjectLocalized,
((PdhCounterProperty) prop).getInstance(), ((PdhCounterProperty) prop).getCounter());
counterMap.put(prop, counter);
if (!pdhQueryHandler.addCounterToQuery(counter)) {
return valueMap;
}
}
// And then query. Zero timestamp means update failed
if (0 < pdhQueryHandler.updateQuery()) {
for (T prop : props) {
valueMap.put(prop, pdhQueryHandler.queryCounter(counterMap.get(prop)));
}
}
}
return valueMap;
}
/**
* Query the a Performance Counter using WMI for values corresponding to the
* property enum.
*
* @param
* The enum type of {@code propertyEnum}
* @param propertyEnum
* An enum which implements
* {@link oshi.util.platform.windows.PerfCounterQuery.PdhCounterProperty}
* and contains the WMI field (Enum value) and PDH Counter string
* (instance and counter)
* @param wmiClass
* The WMI PerfData_RawData_* class corresponding to the PDH object
* @return An {@link EnumMap} of the values indexed by {@code propertyEnum} if
* successful, an empty map if the WMI query failed.
*/
public static > Map queryValuesFromWMI(Class propertyEnum, String wmiClass) {
WmiQuery query = new WmiQuery<>(wmiClass, propertyEnum);
WmiResult result = WmiQueryHandler.createInstance().queryWMI(query);
EnumMap valueMap = new EnumMap<>(propertyEnum);
if (result.getResultCount() > 0) {
for (T prop : propertyEnum.getEnumConstants()) {
switch (result.getCIMType(prop)) {
case Wbemcli.CIM_UINT16:
valueMap.put(prop, Long.valueOf(WmiUtil.getUint16(result, prop, 0)));
break;
case Wbemcli.CIM_UINT32:
valueMap.put(prop, WmiUtil.getUint32asLong(result, prop, 0));
break;
case Wbemcli.CIM_UINT64:
valueMap.put(prop, WmiUtil.getUint64(result, prop, 0));
break;
case Wbemcli.CIM_DATETIME:
valueMap.put(prop, WmiUtil.getDateTime(result, prop, 0).toInstant().toEpochMilli());
break;
default:
throw new ClassCastException("Unimplemented CIM Type Mapping.");
}
}
}
return valueMap;
}
/**
* Localize a PerfCounter string. English counter names should normally be in
* {@code HKEY_LOCAL_MACHINE\SOFTWARE\Microsoft\Windows
* NT\CurrentVersion\Perflib\009\Counter}, but language manipulations may delete
* the {@code 009} index. In this case we can assume English must be the
* language and continue. We may still fail to match the name if the assumption
* is wrong but it's better than nothing.
*
* @param perfObject
* A String to localize
* @return The localized string if localization successful, or the original
* string otherwise.
*/
public static String localize(String perfObject) {
String localized = perfObject;
try {
localized = PdhUtil.PdhLookupPerfNameByIndex(null, PdhUtil.PdhLookupPerfIndexByEnglishName(perfObject));
} catch (Win32Exception e) {
LOG.warn(
"Unable to locate English counter names in registry Perflib 009. Assuming English counters. Error {}. {}",
String.format("0x%x", e.getHR().intValue()),
"See https://support.microsoft.com/en-us/help/300956/how-to-manually-rebuild-performance-counter-library-values");
} catch (PdhException e) {
LOG.warn("Unable to localize {} performance counter. Error {}.", perfObject,
String.format("0x%x", e.getErrorCode()));
}
if (localized.isEmpty()) {
return perfObject;
}
LOG.debug("Localized {} to {}", perfObject, localized);
return localized;
}
/**
* Contract for Counter Property Enums
*/
public interface PdhCounterProperty {
/**
* @return Returns the instance.
*/
String getInstance();
/**
* @return Returns the counter.
*/
String getCounter();
}
}
© 2015 - 2025 Weber Informatics LLC | Privacy Policy