NodeRegistrationInfo.java

Go to the documentation of this file.

/*
 * Copyright (C) 2012 Google Inc.
 * 
 * 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.ros.internal.node.server.master;

import com.google.common.collect.ImmutableSet;
import com.google.common.collect.Sets;

import org.ros.namespace.GraphName;

import java.net.URI;
import java.util.Set;

public class NodeRegistrationInfo {

  private final GraphName nodeName;

  private final URI nodeSlaveUri;

  private final Set<TopicRegistrationInfo> publishers;

  private final Set<TopicRegistrationInfo> subscribers;

  private final Set<ServiceRegistrationInfo> services;

  public NodeRegistrationInfo(GraphName nodeName, URI nodeSlaveUri) {
    this.nodeName = nodeName;
    this.nodeSlaveUri = nodeSlaveUri;
    this.publishers = Sets.newHashSet();
    this.subscribers = Sets.newHashSet();
    this.services = Sets.newHashSet();
  }

  public GraphName getNodeName() {
    return nodeName;
  }

  public URI getNodeSlaveUri() {
    return nodeSlaveUri;
  }

  public boolean hasRegistrations() {
    return !publishers.isEmpty() || !subscribers.isEmpty() || !services.isEmpty();
  }
  
  public Set<TopicRegistrationInfo> getPublishers() {
    return ImmutableSet.copyOf(publishers);
  }

  public void addPublisher(TopicRegistrationInfo publisherTopic) {
    publishers.add(publisherTopic);
  }
  
  public boolean removePublisher(TopicRegistrationInfo publisherTopic) {
    return publishers.remove(publisherTopic);
  }
  
  public Set<TopicRegistrationInfo> getSubscribers() {
    return ImmutableSet.copyOf(subscribers);
  }
  
  public void addSubscriber(TopicRegistrationInfo subscriberTopic) {
    subscribers.add(subscriberTopic);
  }
  
  public boolean removeSubscriber(TopicRegistrationInfo subscriberTopic) {
    return subscribers.remove(subscriberTopic);
  }
  
  public Set<ServiceRegistrationInfo> getServices() {
    return ImmutableSet.copyOf(services);
  }
  
  public void addService(ServiceRegistrationInfo service) {
    services.add(service);
  }
  
  public boolean removeService(ServiceRegistrationInfo service) {
    return services.remove(service);
  }

  @Override
  public int hashCode() {
    final int prime = 31;
    int result = 1;
    result = prime * result + nodeName.hashCode();
    return result;
  }

  @Override
  public boolean equals(Object obj) {
    if (this == obj)
      return true;
    if (obj == null)
      return false;
    if (getClass() != obj.getClass())
      return false;
    NodeRegistrationInfo other = (NodeRegistrationInfo) obj;
    if (!nodeName.equals(other.nodeName))
      return false;
    return true;
  }
}