| /* |
| * 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.apache.brooklyn.entity.zookeeper; |
| |
| import java.util.concurrent.TimeUnit; |
| |
| import javax.management.ObjectName; |
| |
| import org.apache.brooklyn.entity.java.JavaSoftwareProcessDriver; |
| import org.apache.brooklyn.entity.software.base.SoftwareProcessImpl; |
| import org.apache.brooklyn.feed.jmx.JmxAttributePollConfig; |
| import org.apache.brooklyn.feed.jmx.JmxFeed; |
| import org.apache.brooklyn.feed.jmx.JmxHelper; |
| import org.slf4j.Logger; |
| import org.slf4j.LoggerFactory; |
| |
| import com.google.common.base.Functions; |
| import com.google.common.base.Objects.ToStringHelper; |
| |
| /** |
| * An {@link org.apache.brooklyn.api.entity.Entity} that represents a single Apache ZooKeeper instance. |
| */ |
| public abstract class AbstractZooKeeperImpl extends SoftwareProcessImpl implements ZooKeeperNode { |
| |
| @SuppressWarnings("unused") |
| private static final Logger log = LoggerFactory.getLogger(AbstractZooKeeperImpl.class); |
| private static final ObjectName ZOOKEEPER_MBEAN = JmxHelper.createObjectName("org.apache.ZooKeeperService:name0=StandaloneServer_port-1"); |
| |
| private volatile JmxFeed jmxFeed; |
| |
| public AbstractZooKeeperImpl() { |
| } |
| |
| @Override |
| public Integer getZookeeperPort() { return getAttribute(ZOOKEEPER_PORT); } |
| |
| @Override |
| public String getHostname() { return getAttribute(HOSTNAME); } |
| |
| @Override |
| public void waitForServiceUp(long duration, TimeUnit units) { |
| super.waitForServiceUp(duration, units); |
| |
| if (((JavaSoftwareProcessDriver)getDriver()).isJmxEnabled()) { |
| // Wait for the MBean to exist |
| JmxHelper helper = new JmxHelper(this); |
| try { |
| helper.assertMBeanExistsEventually(ZOOKEEPER_MBEAN, units.toMillis(duration)); |
| } finally { |
| helper.terminate(); |
| } |
| } |
| } |
| |
| @Override |
| protected void connectSensors() { |
| connectServiceUpIsRunning(); |
| |
| if (((JavaSoftwareProcessDriver)getDriver()).isJmxEnabled()) { |
| jmxFeed = JmxFeed.builder() |
| .entity(this) |
| .period(500, TimeUnit.MILLISECONDS) |
| .pollAttribute(new JmxAttributePollConfig<Long>(OUTSTANDING_REQUESTS) |
| .objectName(ZOOKEEPER_MBEAN) |
| .attributeName("OutstandingRequests") |
| .onFailureOrException(Functions.constant(-1l))) |
| .pollAttribute(new JmxAttributePollConfig<Long>(PACKETS_RECEIVED) |
| .objectName(ZOOKEEPER_MBEAN) |
| .attributeName("PacketsReceived") |
| .onFailureOrException(Functions.constant(-1l))) |
| .pollAttribute(new JmxAttributePollConfig<Long>(PACKETS_SENT) |
| .objectName(ZOOKEEPER_MBEAN) |
| .attributeName("PacketsSent") |
| .onFailureOrException(Functions.constant(-1l))) |
| .build(); |
| } |
| } |
| |
| @Override |
| public void disconnectSensors() { |
| if (jmxFeed != null) jmxFeed.stop(); |
| disconnectServiceUpIsRunning(); |
| super.disconnectSensors(); |
| } |
| |
| @Override |
| protected ToStringHelper toStringHelper() { |
| return super.toStringHelper() |
| .add("zookeeperPort", getZookeeperPort()); |
| } |
| |
| } |