/*
* 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.karaf.scr.management.internal;
import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;
import javax.management.*;
import javax.management.openmbean.TabularData;
import aQute.bnd.annotation.component.Activate;
import aQute.bnd.annotation.component.Deactivate;
import aQute.bnd.annotation.component.Reference;
import org.apache.felix.scr.Component;
import org.apache.felix.scr.ScrService;
import org.apache.karaf.scr.management.ScrServiceMBean;
import org.apache.karaf.scr.management.codec.JmxComponent;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
@aQute.bnd.annotation.component.Component(
name = ScrServiceMBeanImpl.COMPONENT_NAME,
enabled = true,
immediate = true,
properties = {"hidden.component=true"})
public class ScrServiceMBeanImpl extends StandardMBean implements ScrServiceMBean {
public static final String OBJECT_NAME = "org.apache.karaf:type=scr,name=" + System.getProperty("karaf.name", "root");
public static final String COMPONENT_NAME = "ScrServiceMBean";
public static final String COMPONENT_LABEL = "Apache Karaf SCR Service MBean";
private static final Logger LOGGER = LoggerFactory.getLogger(ScrServiceMBeanImpl.class);
private MBeanServer mBeanServer;
private ScrService scrService;
private ReadWriteLock lock = new ReentrantReadWriteLock();
/**
* Creates new Declarative Services MBean.
*
* @throws NotCompliantMBeanException If the MBean is not a valid MBean.
*/
public ScrServiceMBeanImpl() throws NotCompliantMBeanException {
super(ScrServiceMBean.class);
}
/**
* Service component activation call back. Called when all dependencies are satisfied.
*
* @throws Exception If the activation fails.
*/
@Activate
public void activate() throws Exception {
LOGGER.info("Activating the " + COMPONENT_LABEL);
Map<Object, String> mbeans = new HashMap<Object, String>();
mbeans.put(this, "org.apache.karaf:type=scr,name=${karaf.name}");
try {
lock.writeLock().lock();
if (mBeanServer != null) {
mBeanServer.registerMBean(this, new ObjectName(OBJECT_NAME));
}
} catch (Exception e) {
LOGGER.error("Exception registering the SCR Management MBean: " + e.getLocalizedMessage(), e);
} finally {
lock.writeLock().unlock();
}
}
/**
* Service component deactivation call back. Called after the component is in an active
* state when any dependencies become unsatisfied.
*
* @throws Exception If the deactivation fails.
*/
@Deactivate
public void deactivate() throws Exception {
LOGGER.info("Deactivating the " + COMPONENT_LABEL);
try {
lock.writeLock().lock();
if (mBeanServer != null) {
mBeanServer.unregisterMBean(new ObjectName(OBJECT_NAME));
}
} finally {
lock.writeLock().unlock();
}
}
@Override
public TabularData getComponents() {
try {
return JmxComponent.tableFrom(safe(scrService.getComponents()));
} catch (Exception e) {
e.printStackTrace(System.out);
return null;
}
}
public String[] listComponents() {
Component[] components = safe(scrService.getComponents());
String[] componentNames = new String[components.length];
for (int i = 0; i < componentNames.length; i++) {
componentNames[i] = components[i].getName();
}
return componentNames;
}
public boolean isComponentActive(String componentName) throws MBeanException {
try {
return (componentState(componentName) == Component.STATE_ACTIVE) ? true : false;
} catch (Exception e) {
throw new MBeanException(null, e.toString());
}
}
public int componentState(String componentName) {
int state = -1;
final Component component = findComponent(componentName);
if (component != null)
state = component.getState();
else
LOGGER.warn("No component found for name: " + componentName);
return state;
}
public void activateComponent(String componentName) {
final Component component = findComponent(componentName);
if (component != null)
component.enable();
else
LOGGER.warn("No component found for name: " + componentName);
}
public void deactivateComponent(String componentName) {
final Component component = findComponent(componentName);
if (component != null)
component.disable();
else
LOGGER.warn("No component found for name: " + componentName);
}
private Component findComponent(String componentName) {
Component answer = null;
if (scrService.getComponents(componentName) != null) {
Component[] components = scrService.getComponents(componentName);
for (Component component : safe(components)) {
answer = component;
}
}
return answer;
}
private Component[] safe(Component[] components) {
return components == null ? new Component[0] : components;
}
@Reference
public void setmBeanServer(MBeanServer mBeanServer) {
this.mBeanServer = mBeanServer;
}
public void unsetmBeanServer(MBeanServer mBeanServer) {
this.mBeanServer = null;
}
@Reference
public void setScrService(ScrService scrService) {
this.scrService = scrService;
}
public void unsetScrService(ScrService scrService) {
this.scrService = null;
}
}