jdk/src/share/classes/sun/tracing/MultiplexProviderFactory.java

/*
 * Copyright (c) 2008, Oracle and/or its affiliates. All rights reserved.
 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
 *
 * This code is free software; you can redistribute it and/or modify it
 * under the terms of the GNU General Public License version 2 only, as
 * published by the Free Software Foundation.  Oracle designates this
 * particular file as subject to the "Classpath" exception as provided
 * by Oracle in the LICENSE file that accompanied this code.
 *
 * This code is distributed in the hope that it will be useful, but WITHOUT
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
 * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
 * version 2 for more details (a copy is included in the LICENSE file that
 * accompanied this code).
 *
 * You should have received a copy of the GNU General Public License version
 * 2 along with this work; if not, write to the Free Software Foundation,
 * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
 * or visit www.oracle.com if you need additional information or have any
 * questions.
 */

package sun.tracing;

import java.lang.reflect.Method;
import java.lang.reflect.InvocationTargetException;
import java.util.HashMap;
import java.util.HashSet;
import java.util.Set;

import com.sun.tracing.ProviderFactory;
import com.sun.tracing.Provider;
import com.sun.tracing.Probe;

/**
 * Factory class to create tracing Providers.
 *
 * This factory creates a "multiplex provider", which is a provider that
 * encapsulates a list of providers and whose probes trigger a corresponding
 * trigger in each of the encapsulated providers' probes.
 *
 * This is used when there are multiple tracing frameworks activated at once.
 * A user-defined provider gets implementation for each of the activated
 * frameworks and this multiplex framework is what is ultimately passed
 * back to the user.  All probe triggers are multiplexed to each
 * active framework.
 *
 * @since 1.7
 */
public class MultiplexProviderFactory extends ProviderFactory {

    private Set<ProviderFactory> factories;

    public MultiplexProviderFactory(Set<ProviderFactory> factories) {
        this.factories = factories;
    }

    public <T extends Provider> T createProvider(Class<T> cls) {
        HashSet<Provider> providers = new HashSet<Provider>();
        for (ProviderFactory factory : factories) {
            providers.add(factory.createProvider(cls));
        }
        MultiplexProvider provider = new MultiplexProvider(cls, providers);
        provider.init();
        return provider.newProxyInstance();
    }
}

class MultiplexProvider extends ProviderSkeleton {

    private Set<Provider> providers;

    protected ProbeSkeleton createProbe(Method m) {
        return new MultiplexProbe(m, providers);
    }

    MultiplexProvider(Class<? extends Provider> type, Set<Provider> providers) {
        super(type);
        this.providers = providers;
    }

    public void dispose() {
        for (Provider p : providers) {
            p.dispose();
        }
        super.dispose();
    }
}

class MultiplexProbe extends ProbeSkeleton {

    private Set<Probe> probes;

    MultiplexProbe(Method m, Set<Provider> providers) {
        super(m.getParameterTypes());
        probes = new HashSet<Probe>();
        for (Provider p : providers) {
            Probe probe = p.getProbe(m);
            if (probe != null) {
                probes.add(probe);
            }
        }
    }

    public boolean isEnabled() {
        for (Probe p : probes) {
            if (p.isEnabled()) {
                return true;
            }
        }
        return false;
    }

    public void uncheckedTrigger(Object[] args) {
        for (Probe p : probes) {
            try {
                // try the fast path
                ProbeSkeleton ps = (ProbeSkeleton)p;
                ps.uncheckedTrigger(args);
            } catch (ClassCastException e) {
                // Probe.trigger takes an "Object ..." varargs parameter,
                // so we can't call it directly.
                try {
                    Method m = Probe.class.getMethod(
                        "trigger", Class.forName("[java.lang.Object"));
                    m.invoke(p, args);
                } catch (Exception e1) {
                    assert false; // This shouldn't happen
                }
            }
        }
    }
}