Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance. Project price only 1 $
You can buy this project and download/modify it how often you want.
/**
* 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.ArrayList;
import java.util.Collections;
import java.util.EnumMap;
import java.util.HashSet;
import java.util.List;
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.PdhEnumObjectItems;
import com.sun.jna.platform.win32.PdhUtil.PdhException;
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.Util;
import oshi.util.platform.windows.PerfDataUtil.PerfCounter;
import oshi.util.tuples.Pair;
/**
* Enables queries of Performance Counters using wild cards to filter instances
*/
@ThreadSafe
public final class PerfCounterWildcardQuery {
private static final Logger LOG = LoggerFactory.getLogger(PerfCounterWildcardQuery.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();
private PerfCounterWildcardQuery() {
}
/**
* 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 pair containing a list of instances and an {@link EnumMap} of the
* corresponding values indexed by {@code propertyEnum} on success, or
* an empty list and empty map if both PDH and WMI queries failed.
*/
public static > Pair, Map>> queryInstancesAndValues(
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)) {
Pair, Map>> instancesAndValuesMap = queryInstancesAndValuesFromPDH(
propertyEnum, perfObject);
if (!instancesAndValuesMap.getA().isEmpty()) {
return instancesAndValuesMap;
}
// If we are here, query failed
LOG.warn("Disabling further attempts to query {}.", perfObject);
failedQueryCache.add(perfObject);
}
} finally {
failedQueryCacheLock.unlock();
}
}
return queryInstancesAndValuesFromWMI(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 pair containing a list of instances and an {@link EnumMap} of the
* corresponding values indexed by {@code propertyEnum} on success, or
* an empty list and empty map if the PDH query failed.
*/
public static > Pair, Map>> queryInstancesAndValuesFromPDH(
Class propertyEnum, String perfObject) {
T[] props = propertyEnum.getEnumConstants();
if (props.length < 2) {
throw new IllegalArgumentException("Enum " + propertyEnum.getName()
+ " must have at least two elements, an instance filter and a counter.");
}
String instanceFilter = ((PdhCounterWildcardProperty) propertyEnum.getEnumConstants()[0]).getCounter()
.toLowerCase();
String perfObjectLocalized = PerfCounterQuery.localize(perfObject);
// Get list of instances
final PdhEnumObjectItems objectItems;
try {
objectItems = PdhUtil.PdhEnumObjectItems(null, null, perfObjectLocalized, 100);
} catch (PdhException e) {
return new Pair<>(Collections.emptyList(), Collections.emptyMap());
}
List instances = objectItems.getInstances();
// Filter out instances not matching filter
instances.removeIf(i -> !Util.wildcardMatch(i.toLowerCase(), instanceFilter));
EnumMap> valuesMap = new EnumMap<>(propertyEnum);
try (PerfCounterQueryHandler pdhQueryHandler = new PerfCounterQueryHandler()) {
// Set up the query and counter handles
EnumMap> counterListMap = new EnumMap<>(propertyEnum);
// Start at 1, first counter defines instance filter
for (int i = 1; i < props.length; i++) {
T prop = props[i];
List counterList = new ArrayList<>(instances.size());
for (String instance : instances) {
PerfCounter counter = PerfDataUtil.createCounter(perfObject, instance,
((PdhCounterWildcardProperty) prop).getCounter());
if (!pdhQueryHandler.addCounterToQuery(counter)) {
return new Pair<>(Collections.emptyList(), Collections.emptyMap());
}
counterList.add(counter);
}
counterListMap.put(prop, counterList);
}
// And then query. Zero timestamp means update failed
if (0 < pdhQueryHandler.updateQuery()) {
// Start at 1, first counter defines instance filter
for (int i = 1; i < props.length; i++) {
T prop = props[i];
List values = new ArrayList<>();
for (PerfCounter counter : counterListMap.get(prop)) {
values.add(pdhQueryHandler.queryCounter(counter));
}
valuesMap.put(prop, values);
}
}
}
return new Pair<>(instances, valuesMap);
}
/**
* 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 pair containing a list of instances and an {@link EnumMap} of the
* corresponding values indexed by {@code propertyEnum} on success, or
* an empty list and empty map if the WMI query failed.
*/
public static > Pair, Map>> queryInstancesAndValuesFromWMI(
Class propertyEnum, String wmiClass) {
List instances = new ArrayList<>();
EnumMap> valuesMap = new EnumMap<>(propertyEnum);
WmiQuery query = new WmiQuery<>(wmiClass, propertyEnum);
WmiResult result = WmiQueryHandler.createInstance().queryWMI(query);
if (result.getResultCount() > 0) {
for (T prop : propertyEnum.getEnumConstants()) {
// First element is instance name
if (prop.ordinal() == 0) {
for (int i = 0; i < result.getResultCount(); i++) {
instances.add(WmiUtil.getString(result, prop, i));
}
} else {
List values = new ArrayList<>();
for (int i = 0; i < result.getResultCount(); i++) {
switch (result.getCIMType(prop)) {
case Wbemcli.CIM_UINT16:
values.add(Long.valueOf(WmiUtil.getUint16(result, prop, i)));
break;
case Wbemcli.CIM_UINT32:
values.add(WmiUtil.getUint32asLong(result, prop, i));
break;
case Wbemcli.CIM_UINT64:
values.add(WmiUtil.getUint64(result, prop, i));
break;
case Wbemcli.CIM_DATETIME:
values.add(WmiUtil.getDateTime(result, prop, i).toInstant().toEpochMilli());
break;
default:
throw new ClassCastException("Unimplemented CIM Type Mapping.");
}
}
valuesMap.put(prop, values);
}
}
}
return new Pair<>(instances, valuesMap);
}
/**
* Contract for Counter Property Enums
*/
public interface PdhCounterWildcardProperty {
/**
* @return Returns the counter. The first element of the enum will return the
* instance filter rather than a counter.
*/
String getCounter();
}
}
