ch.genevajug.crappy.serviceregistor.impl.ServiceRegistrator.java Source code

Introduction

Here is the source code for ch.genevajug.crappy.serviceregistor.impl.ServiceRegistrator.java

Source

/*
 * Copyright 2008 the original author or authors.
 *
 * 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 ch.genevajug.crappy.serviceregistor.impl;

import ch.genevajug.crappy.serviceregistor.IServiceRegistrator;
import ch.genevajug.crappy.serviceregistor.osgi.BundleContext;
import ch.genevajug.crappy.serviceregistor.osgi.ServiceRegistration;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

/**
 * An "OSGi"-ish implementation of the {@link IServiceRegistrator} interface.
 * The test for this class demonstrates static mocking as well as getting and
 * setting internal state.
 * 
 */
public class ServiceRegistrator implements IServiceRegistrator {

    @Autowired
    private BundleContext bundleContext;

    /**
     * Holds all services registrations that has been registered by this service
     * registrator.
     */
    private final Map<Long, ServiceRegistration> serviceRegistrations;

    /**
     * Default constructor, initializes internal state.
     */
    public ServiceRegistrator() {
        serviceRegistrations = new ConcurrentHashMap<Long, ServiceRegistration>();
    }

    /**
     * {@inheritDoc}
     */
    public long registerService(String name, Object serviceImplementation) {
        ServiceRegistration registerService = bundleContext.registerService(name, serviceImplementation, null);
        final long id = IdGenerator.generateNewId();
        serviceRegistrations.put(id, registerService);
        return id;
    }

    /**           Mock for ServiceRegistratio
     * {@inheritDoc}
     */
    public void unregisterService(long id) {
        final ServiceRegistration registration = serviceRegistrations.remove(id);
        if (registration == null) {
            throw new IllegalStateException(
                    "Registration with id " + id + " has already been removed or has never been registered");
        }
        registration.unregister();
    }

}