org.tinygroup.service.registry.impl.ServiceRegistryImpl Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of org.tinygroup.service Show documentation
The newest version!
/**
 *  Copyright (c) 1997-2013, www.tinygroup.org ([email protected]).
 *
 *  Licensed under the GPL, Version 3.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.gnu.org/licenses/gpl.html
 *
 *  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.tinygroup.service.registry.impl;

import org.tinygroup.logger.LogLevel;
import org.tinygroup.logger.Logger;
import org.tinygroup.logger.LoggerFactory;
import org.tinygroup.service.Service;
import org.tinygroup.service.registry.ServiceRegistry;
import org.tinygroup.service.registry.ServiceRegistryItem;

import java.util.*;

public class ServiceRegistryImpl implements ServiceRegistry {
	/**
	 * 指定服务ID，默认服务MAP
	 */
	private final Map serviceIdMap = new HashMap();

	private final Map serviceToServiceRegistryItem = new HashMap();
	private static final Logger LOGGER = LoggerFactory
			.getLogger(ServiceRegistryImpl.class);

	private boolean change = true;

	public void registerService(ServiceRegistryItem serviceRegistryItem) {
		change = true;
		if (serviceIdMap.containsKey(serviceRegistryItem.getServiceId())) {
			LOGGER.logMessage(LogLevel.WARN, "服务号:[{0}]已经存在,之前的服务将被覆盖",
					serviceRegistryItem.getServiceId());
			ServiceRegistryItem item = serviceIdMap.get(serviceRegistryItem.getServiceId());
			if(item!=null){
				serviceToServiceRegistryItem.remove(item.getService());
			}
		}
		LOGGER.logMessage(LogLevel.INFO, "添加服务[serviceId:{0}]",
				serviceRegistryItem.getServiceId());
		serviceIdMap.put(serviceRegistryItem.getServiceId(),
				serviceRegistryItem);
		serviceToServiceRegistryItem.put(serviceRegistryItem.getService(),
				serviceRegistryItem);

	}

	public void registerService(List serviceRegistryItems) {
		for (ServiceRegistryItem serviceRegistryItem : serviceRegistryItems) {
			registerService(serviceRegistryItem);
		}
	}

	public void registerService(ServiceRegistryItem[] serviceRegistryItems) {
		for (ServiceRegistryItem serviceRegistryItem : serviceRegistryItems) {
			registerService(serviceRegistryItem);
		}
	}

	public void registerService(Set serviceRegistryItems) {
		for (ServiceRegistryItem serviceRegistryItem : serviceRegistryItems) {
			registerService(serviceRegistryItem);
		}
	}

	public void removeService(String serviceId) {
		removeService(serviceIdMap.get(serviceId));

	}

	private void removeService(ServiceRegistryItem serviceRegistryItem) {
		if (serviceRegistryItem == null) {
			return;
		}
		change = true;
		LOGGER.logMessage(LogLevel.INFO, "删除服务[serviceId:{0}]",
				serviceRegistryItem.getServiceId());
		ServiceRegistryItem removedItem = serviceIdMap
				.remove(serviceRegistryItem.getServiceId());
		serviceToServiceRegistryItem.remove(removedItem.getService());
	}

	public int size() {
		return serviceToServiceRegistryItem.size();
	}

	public void clear() {
		serviceIdMap.clear();
		serviceToServiceRegistryItem.clear();
	}

	public ServiceRegistryItem getServiceRegistryItem(Service service) {
		return serviceToServiceRegistryItem.get(service);
	}

	public ServiceRegistryItem getServiceRegistryItem(String serviceId) {
		return serviceIdMap.get(serviceId);
	}

	public Collection getServiceRegistryItems() {

		return serviceIdMap.values();
	}

	public boolean isChange() {
		return change;
	}

	public void setChange(boolean change) {
		this.change = change;
	}

	


	

}