- private static final Logger LOG = LoggerFactory.getLogger(ShardMBeanFactory.class);
-
- private static Cache<String,ShardStats> shardMBeansCache =
- CacheBuilder.newBuilder().weakValues().build();
-
- public static ShardStats getShardStatsMBean(final String shardName, final String mxBeanType) {
- final String finalMXBeanType = mxBeanType != null ? mxBeanType : "DistDataStore";
- try {
- return shardMBeansCache.get(shardName, new Callable<ShardStats>() {
- @Override
- public ShardStats call() throws Exception {
- ShardStats shardStatsMBeanImpl = new ShardStats(shardName, finalMXBeanType);
- shardStatsMBeanImpl.registerMBean();
- return shardStatsMBeanImpl;
- }
- });
- } catch(ExecutionException e) {
- LOG.error(String.format("Could not create MXBean for shard: %s", shardName), e);
- // Just return an instance that isn't registered.
- return new ShardStats(shardName, finalMXBeanType);
- }
+ public static ShardStats getShardStatsMBean(final String shardName, final String mxBeanType,
+ @Nonnull final Shard shard) {
+ String finalMXBeanType = mxBeanType != null ? mxBeanType : "DistDataStore";
+ ShardStats shardStatsMBeanImpl = new ShardStats(shardName, finalMXBeanType, shard);
+ shardStatsMBeanImpl.registerMBean();
+ return shardStatsMBeanImpl;