org.weakref.jmx.guice.GuiceMBeanExporter Maven / Gradle / Ivy

Go to download
/**
 *  Copyright 2009 Martin Traverso
 *
 *  Licensed 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.weakref.jmx.guice;

import com.google.inject.Inject;
import com.google.inject.Injector;
import org.weakref.jmx.MBeanExporter;

import javax.management.ObjectName;
import java.util.Map;
import java.util.Set;

class GuiceMBeanExporter
{
    @Inject
    public GuiceMBeanExporter(Set mappings,
            Set> setMappings,
            Set> mapMappings,
            MBeanExporter exporter,
            Injector injector)
    {
        export(mappings, exporter, injector);

        // cast to Object to get around Java's broken generics
        exportSets((Set>) (Object) setMappings, exporter, injector);
        exportMaps((Set>) (Object) mapMappings, exporter, injector);
    }

    private  void exportMaps(Set> mapMappings, MBeanExporter exporter, Injector injector)
    {
        for (MapMapping mapping : mapMappings) {
            ObjectNameFunction> namingFunction = mapping.getObjectNameFunction();

            Map map = injector.getInstance(mapping.getKey());

            for (Map.Entry entry : map.entrySet()) {
                ObjectName name = namingFunction.name(entry);
                exporter.export(name, entry.getValue());
            }
        }
    }

    private  void exportSets(Set> setMappings, MBeanExporter exporter, Injector injector)
    {
        for (SetMapping mapping : setMappings) {
            ObjectNameFunction objectNameFunction = mapping.getObjectNameFunction();

            Set set = injector.getInstance(mapping.getKey());

            for (T instance : set) {
                ObjectName name = objectNameFunction.name(instance);
                exporter.export(name, instance);
            }
        }
    }

    private void export(Set mappings, MBeanExporter exporter, Injector injector)
    {
        for (Mapping mapping : mappings) {
            exporter.export(mapping.getName(), injector.getInstance(mapping.getKey()));
        }
    }
}