com.microsoft.reef.io.network.naming.NameRegistryClient Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of reef-io Show documentation
There is a newer version: 0.9
/**
 * Copyright (C) 2013 Microsoft Corporation
 *
 * 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 com.microsoft.reef.io.network.naming;

import com.microsoft.reef.io.naming.NamingRegistry;
import com.microsoft.reef.io.network.naming.exception.NamingException;
import com.microsoft.reef.io.network.naming.serialization.NamingMessage;
import com.microsoft.reef.io.network.naming.serialization.NamingRegisterRequest;
import com.microsoft.reef.io.network.naming.serialization.NamingRegisterResponse;
import com.microsoft.reef.io.network.naming.serialization.NamingUnregisterRequest;
import com.microsoft.wake.EventHandler;
import com.microsoft.wake.Identifier;
import com.microsoft.wake.IdentifierFactory;
import com.microsoft.wake.Stage;
import com.microsoft.wake.impl.SyncStage;
import com.microsoft.wake.remote.Codec;
import com.microsoft.wake.remote.NetUtils;
import com.microsoft.wake.remote.impl.TransportEvent;
import com.microsoft.wake.remote.transport.Link;
import com.microsoft.wake.remote.transport.LinkListener;
import com.microsoft.wake.remote.transport.Transport;
import com.microsoft.wake.remote.transport.netty.LoggingLinkListener;
import com.microsoft.wake.remote.transport.netty.NettyMessagingTransport;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.SocketAddress;
import java.util.concurrent.BlockingQueue;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.TimeUnit;
import java.util.logging.Level;
import java.util.logging.Logger;

/**
 * Naming registry client
 */
public class NameRegistryClient implements Stage, NamingRegistry {
  private static final Logger LOG = Logger.getLogger(NameRegistryClient.class.getName());

  private final SocketAddress serverSocketAddr;
  private final Transport transport;
  private final Codec codec;
  private final BlockingQueue replyQueue;
  private final long timeout;

  /**
   * Constructs a naming registry client
   * 
   * @param serverAddr a name server address
   * @param serverPort a name server port
   * @param factory an identifier factory
   */
  public NameRegistryClient(String serverAddr, int serverPort, IdentifierFactory factory) {   
    this(serverAddr, serverPort, 10000, factory);
  }
  
  /**
   * Constructs a naming registry client
   * 
   * @param serverAddr a name server address
   * @param serverPort a name server port
   * @param timeout timeout in ms
   * @param factory an identifier factory
   */
  public NameRegistryClient(String serverAddr, int serverPort, long timeout, IdentifierFactory factory) {   
    serverSocketAddr = new InetSocketAddress(serverAddr, serverPort);
    this.timeout = timeout;
    codec = NamingCodecFactory.createRegistryCodec(factory);
    replyQueue = new LinkedBlockingQueue();
    transport = new NettyMessagingTransport(NetUtils.getLocalAddress(), 0, new SyncStage(new NamingRegistryClientHandler(new NamingRegistryResponseHandler(replyQueue), codec)), null);
  }
  
  NameRegistryClient(String serverAddr, int serverPort, long timeout, IdentifierFactory factory, 
      BlockingQueue replyQueue, Transport transport) {   
    serverSocketAddr = new InetSocketAddress(serverAddr, serverPort);
    this.timeout = timeout;
    codec = NamingCodecFactory.createFullCodec(factory);
    this.replyQueue = replyQueue;
    this.transport = transport;
  }
  
  /**
   * Registers an (identifier, address) mapping
   * 
   * @param id an identifier
   * @param addr an Internet socket address
   */
  @Override
  public void register(Identifier id, InetSocketAddress addr)
      throws Exception {
   
    // needed to keep threads from reading the wrong response
    // TODO: better fix matches replies to threads with a map after REEF-198
    synchronized (this) {
      LOG.log(Level.FINE, id + " " + addr);
      Link link = transport.open(serverSocketAddr, codec, 
          new LoggingLinkListener());
      link.write(new NamingRegisterRequest(new NameAssignmentTuple(id, addr)));

      while(true) {
        try {
          replyQueue.poll(timeout, TimeUnit.MILLISECONDS);
          break;
        } catch (InterruptedException e) {
          e.printStackTrace();
          throw new NamingException(e);
        }
      }
    }
  }

  /**
   * Unregisters an identifier
   * 
   * @param id an identifier
   */
  @Override
  public void unregister(Identifier id) throws IOException {
    Link link = transport.open(serverSocketAddr, codec, 
        new LinkListener() {
          @Override
          public void messageReceived(NamingMessage message) {
          }
        });
    link.write(new NamingUnregisterRequest(id));
  }

  /**
   * Closes resources
   */
  @Override
  public void close() throws Exception {
    // Should not close transport as we did not
    // create it
  }

}

/**
 * Naming registry client transport event handler
 */
class NamingRegistryClientHandler implements EventHandler {
  private static final Logger LOG = Logger.getLogger(NamingRegistryClientHandler.class.getName());

  private final EventHandler handler;
  private final Codec codec;
  
  NamingRegistryClientHandler(EventHandler handler, Codec codec) {
    this.handler = handler;
    this.codec = codec;
  }
  
  @Override
  public void onNext(TransportEvent value) {
    LOG.log(Level.FINE, value.toString());
    handler.onNext((NamingRegisterResponse)codec.decode(value.getData()));
  }
}

/**
 * Naming register response handler
 */
class NamingRegistryResponseHandler implements EventHandler {

  private final BlockingQueue replyQueue;

  NamingRegistryResponseHandler(BlockingQueue replyQueue) {
    this.replyQueue = replyQueue;
  }
  
  @Override
  public void onNext(NamingRegisterResponse value) {
    replyQueue.offer(value);
  }
  
}