-
Notifications
You must be signed in to change notification settings - Fork 60
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[COMMON] separate JndiClient from MBeanClient (#1679)
- Loading branch information
Showing
47 changed files
with
349 additions
and
333 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
184 changes: 184 additions & 0 deletions
184
common/src/main/java/org/astraea/common/metrics/JndiClient.java
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,184 @@ | ||
| /* | ||
| * Licensed to the Apache Software Foundation (ASF) under one or more | ||
| * contributor license agreements. See the NOTICE file distributed with | ||
| * this work for additional information regarding copyright ownership. | ||
| * The ASF licenses this file to You under the Apache License, Version 2.0 | ||
| * (the "License"); you may not use this file except in compliance with | ||
| * the License. You may obtain a copy of the License at | ||
| * | ||
| * http://www.apache.org/licenses/LICENSE-2.0 | ||
| * | ||
| * Unless required by applicable law or agreed to in writing, software | ||
| * distributed under the License is distributed on an "AS IS" BASIS, | ||
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | ||
| * See the License for the specific language governing permissions and | ||
| * limitations under the License. | ||
| */ | ||
| package org.astraea.common.metrics; | ||
|
|
||
| import java.io.IOException; | ||
| import java.lang.management.ManagementFactory; | ||
| import java.net.MalformedURLException; | ||
| import java.util.Arrays; | ||
| import java.util.Collection; | ||
| import java.util.HashMap; | ||
| import java.util.List; | ||
| import java.util.function.Consumer; | ||
| import java.util.stream.Collectors; | ||
| import java.util.stream.Stream; | ||
| import javax.management.AttributeNotFoundException; | ||
| import javax.management.InstanceNotFoundException; | ||
| import javax.management.MBeanException; | ||
| import javax.management.MBeanFeatureInfo; | ||
| import javax.management.MBeanServerConnection; | ||
| import javax.management.ObjectInstance; | ||
| import javax.management.ReflectionException; | ||
| import javax.management.RuntimeMBeanException; | ||
| import javax.management.remote.JMXConnectorFactory; | ||
| import javax.management.remote.JMXServiceURL; | ||
| import org.astraea.common.Utils; | ||
|
|
||
| /** A MBeanClient used to retrieve mbean value from remote Jmx server. */ | ||
| public interface JndiClient extends MBeanClient, AutoCloseable { | ||
|
|
||
| /** | ||
| * @param host the address of jmx server | ||
| * @param port the port of jmx server | ||
| * @return a mbean client using JNDI to lookup metrics. | ||
| */ | ||
| static JndiClient of(String host, int port) { | ||
| try { | ||
| return of( | ||
| new JMXServiceURL( | ||
| String.format( | ||
| "service:jmx:rmi://%s:%s/jndi/rmi://%s:%s/jmxrmi", host, port, host, port))); | ||
| } catch (MalformedURLException e) { | ||
| throw new IllegalArgumentException(e); | ||
| } | ||
| } | ||
|
|
||
| static JndiClient of(JMXServiceURL jmxServiceURL) { | ||
| return Utils.packException( | ||
| () -> { | ||
| var jmxConnector = JMXConnectorFactory.connect(jmxServiceURL); | ||
| return new BasicMBeanClient( | ||
| jmxConnector.getMBeanServerConnection(), | ||
| jmxServiceURL.getHost(), | ||
| jmxServiceURL.getPort()) { | ||
| @Override | ||
| public void close() { | ||
| Utils.close(jmxConnector); | ||
| } | ||
| }; | ||
| }); | ||
| } | ||
|
|
||
| static JndiClient local() { | ||
| return new BasicMBeanClient(ManagementFactory.getPlatformMBeanServer(), Utils.hostname(), -1); | ||
| } | ||
|
|
||
| @Override | ||
| default void close() {} | ||
|
|
||
| class BasicMBeanClient implements JndiClient { | ||
|
|
||
| private final MBeanServerConnection connection; | ||
| final String host; | ||
|
|
||
| final int port; | ||
|
|
||
| BasicMBeanClient(MBeanServerConnection connection, String host, int port) { | ||
| this.connection = connection; | ||
| this.host = host; | ||
| this.port = port; | ||
| } | ||
|
|
||
| @Override | ||
| public BeanObject bean(BeanQuery beanQuery) { | ||
| return Utils.packException( | ||
| () -> { | ||
| // ask for MBeanInfo | ||
| var mBeanInfo = connection.getMBeanInfo(beanQuery.objectName()); | ||
|
|
||
| // create a list builder all available attributes name | ||
| var attributeName = | ||
| Arrays.stream(mBeanInfo.getAttributes()) | ||
| .map(MBeanFeatureInfo::getName) | ||
| .collect(Collectors.toList()); | ||
|
|
||
| // query the result | ||
| return queryBean(beanQuery, attributeName); | ||
| }); | ||
| } | ||
|
|
||
| BeanObject queryBean(BeanQuery beanQuery, Collection<String> attributeNameCollection) | ||
| throws ReflectionException, | ||
| InstanceNotFoundException, | ||
| IOException, | ||
| AttributeNotFoundException, | ||
| MBeanException { | ||
| // fetch attribute value from mbean server | ||
| var attributeNameArray = attributeNameCollection.toArray(new String[0]); | ||
| var attributeList = | ||
| connection.getAttributes(beanQuery.objectName(), attributeNameArray).asList(); | ||
|
|
||
| // collect attribute name & value into a map | ||
| var attributes = new HashMap<String, Object>(); | ||
| attributeList.forEach(attribute -> attributes.put(attribute.getName(), attribute.getValue())); | ||
|
|
||
| // according to the javadoc of MBeanServerConnection#getAttributes, the API will | ||
| // ignore any error occurring during the fetch process (for example, attribute not | ||
| // exists). Below code check for such condition and try to figure out what exactly | ||
| // the error is. put it into attributes return result. | ||
| for (var str : attributeNameArray) { | ||
| if (attributes.containsKey(str)) continue; | ||
| try { | ||
| attributes.put(str, connection.getAttribute(beanQuery.objectName(), str)); | ||
| } catch (RuntimeMBeanException e) { | ||
| if (!(e.getCause() instanceof UnsupportedOperationException)) | ||
| throw new IllegalStateException(e); | ||
| // the UnsupportedOperationException is thrown when we query unacceptable | ||
| // attribute. we just skip it as it is normal case to | ||
| // return "acceptable" attribute only | ||
| } | ||
| } | ||
|
|
||
| // collect result, and build a new BeanObject as return result | ||
| return new BeanObject(beanQuery.domainName(), beanQuery.properties(), attributes); | ||
| } | ||
|
|
||
| @Override | ||
| public Collection<BeanObject> beans( | ||
| BeanQuery beanQuery, Consumer<RuntimeException> errorHandle) { | ||
| return Utils.packException( | ||
| () -> | ||
| connection.queryMBeans(beanQuery.objectName(), null).stream() | ||
| // Parallelize the sampling of bean objects. The underlying RMI is thread-safe. | ||
| // https://github.com/skiptests/astraea/issues/1553#issuecomment-1461143723 | ||
| .parallel() | ||
| .map(ObjectInstance::getObjectName) | ||
| .map(BeanQuery::fromObjectName) | ||
| .flatMap( | ||
| query -> { | ||
| try { | ||
| return Stream.of(bean(query)); | ||
| } catch (RuntimeException e) { | ||
| errorHandle.accept(e); | ||
| return Stream.empty(); | ||
| } | ||
| }) | ||
| .collect(Collectors.toUnmodifiableList())); | ||
| } | ||
|
|
||
| /** | ||
| * Returns the list of domains in which any MBean is currently registered. | ||
| * | ||
| * <p>The order of strings within the returned array is not defined. | ||
| * | ||
| * @return a {@link List} of domain name {@link String} | ||
| */ | ||
| List<String> domains() { | ||
| return Utils.packException(() -> Arrays.asList(connection.getDomains())); | ||
| } | ||
| } | ||
| } |
Oops, something went wrong.