package org.jboss.gravia.provision.internal;
/*
* #%L
* Gravia :: Provision
* %%
* Copyright (C) 2013 - 2014 JBoss by Red Hat
* %%
* 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.
* #L%
*/
import java.util.concurrent.atomic.AtomicReference;
import org.jboss.gravia.Constants;
import org.jboss.gravia.provision.DefaultProvisioner;
import org.jboss.gravia.provision.Provisioner;
import org.jboss.gravia.provision.spi.RuntimeEnvironment;
import org.jboss.gravia.repository.Repository;
import org.jboss.gravia.resolver.Resolver;
import org.jboss.gravia.runtime.Runtime;
import org.jboss.gravia.runtime.RuntimeLocator;
import org.osgi.framework.BundleActivator;
import org.osgi.framework.BundleContext;
import org.osgi.framework.ServiceEvent;
import org.osgi.framework.ServiceListener;
import org.osgi.framework.ServiceReference;
import org.osgi.framework.ServiceRegistration;
/**
* Activate the {@link Resolver} in the runtime.
*
* @author thomas.diesler@jboss.com
* @since 20-Dec-2012
*/
public final class ProvisionerActivator implements BundleActivator {
private AtomicReference<ServiceReference<Resolver>> resolverRef = new AtomicReference<ServiceReference<Resolver>>();
private AtomicReference<ServiceReference<Repository>> repositoryRef = new AtomicReference<ServiceReference<Repository>>();
private ServiceRegistration<Provisioner> registration;
@Override
public void start(final BundleContext context) throws Exception {
// Add a listener that tracks {@link Resolver} and {@link Repository}
ServiceListener listener = new ServiceListener() {
@Override
@SuppressWarnings("unchecked")
public void serviceChanged(ServiceEvent event) {
if (event.getType() == ServiceEvent.REGISTERED) {
ServiceReference<?> sref = event.getServiceReference();
String[] objectClass = (String[]) sref.getProperty(Constants.OBJECTCLASS);
if (Resolver.class.getName().equals(objectClass[0])) {
resolverRef.compareAndSet(null, (ServiceReference<Resolver>) sref);
registerProvisionerService(context, this);
} else if (Repository.class.getName().equals(objectClass[0])) {
repositoryRef.compareAndSet(null, (ServiceReference<Repository>) sref);
registerProvisionerService(context, this);
}
}
}
};
context.addServiceListener(listener);
// Double check instead we already missed the event
resolverRef.compareAndSet(null, context.getServiceReference(Resolver.class));
repositoryRef.compareAndSet(null, context.getServiceReference(Repository.class));
// Setup and register the {@link Provisioner} service
registerProvisionerService(context, listener);
}
private synchronized void registerProvisionerService(BundleContext context, ServiceListener listener) {
ServiceReference<Resolver> resref = resolverRef.get();
ServiceReference<Repository> repref = repositoryRef.get();
if (registration == null && resref != null & repref != null) {
context.removeServiceListener(listener);
Resolver resolver = context.getService(resref);
Repository repository = context.getService(repref);
Runtime runtime = RuntimeLocator.getRequiredRuntime();
RuntimeEnvironment environment = new RuntimeEnvironment(runtime).initDefaultContent();
BundleResourceInstaller installer = new BundleResourceInstaller(context, environment);
Provisioner provisioner = new DefaultProvisioner(environment, resolver, repository, installer);
registration = context.registerService(Provisioner.class, provisioner, null);
}
}
@Override
public void stop(BundleContext context) throws Exception {
if (registration != null) {
registration.unregister();
}
ServiceReference<Resolver> resref = resolverRef.get();
if (resref != null) {
context.ungetService(resref);
}
ServiceReference<Repository> repref = repositoryRef.get();
if (repref != null) {
context.ungetService(repref);
}
}
}