/*

  * Licensed to the Apache Software Foundation (ASF) under one

  * or more contributor license agreements.  See the NOTICE file

  * distributed with this work for additional information

  * regarding copyright ownership.  The ASF licenses this file

  * to you 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.apache.giraph.comm.netty;

 import org.apache.giraph.bsp.CentralizedServiceWorker;

 import org.apache.giraph.comm.ServerData;

 import org.apache.giraph.comm.WorkerServer;

 import org.apache.giraph.comm.flow_control.FlowControl;

 import org.apache.giraph.comm.netty.handler.WorkerRequestServerHandler;

 import org.apache.giraph.conf.ImmutableClassesGiraphConfiguration;

 import org.apache.hadoop.io.Writable;

 import org.apache.hadoop.io.WritableComparable;

 import org.apache.hadoop.mapreduce.Mapper;

 import org.apache.log4j.Logger;

 import java.net.InetSocketAddress;

 /**

  * Netty worker server that implement {@link WorkerServer} and contains

  * the actual {@link ServerData}.

  *

  * @param <I> Vertex id

  * @param <V> Vertex data

  * @param <E> Edge data

  */

 @SuppressWarnings("rawtypes")

 public class NettyWorkerServer<I extends WritableComparable,

     V extends Writable, E extends Writable>

     implements WorkerServer<I, V, E> {

   /** Class logger */

   private static final Logger LOG =

     Logger.getLogger(NettyWorkerServer.class);

   /** Hadoop configuration */

   private final ImmutableClassesGiraphConfiguration<I, V, E> conf;

   /** Service worker */

   private final CentralizedServiceWorker<I, V, E> service;

   /** Netty server that does that actual I/O */

   private final NettyServer nettyServer;

   /** Server data storage */

   private final ServerData<I, V, E> serverData;

   /** Mapper context */

   private final Mapper<?, ?, ?, ?>.Context context;

   /**

    * Constructor to start the server.

    *

    * @param conf Configuration

    * @param service Service to get partition mappings

    * @param context Mapper context

    * @param exceptionHandler handle uncaught exceptions

    */

   public NettyWorkerServer(ImmutableClassesGiraphConfiguration<I, V, E> conf,

       CentralizedServiceWorker<I, V, E> service,

       Mapper<?, ?, ?, ?>.Context context,

       Thread.UncaughtExceptionHandler exceptionHandler) {

     this.conf = conf;

     this.service = service;

     this.context = context;

     serverData =

         new ServerData<I, V, E>(service, this, conf, context);

     nettyServer = new NettyServer(conf,

         new WorkerRequestServerHandler.Factory<I, V, E>(serverData),

         service.getWorkerInfo(), context, exceptionHandler);

     nettyServer.start();

   }

   @Override

   public InetSocketAddress getMyAddress() {

     return nettyServer.getMyAddress();

   }

   @Override

   public String getLocalHostOrIp() {

     return nettyServer.getLocalHostOrIp();

   }

   @Override

   public void prepareSuperstep() {

     serverData.prepareSuperstep(); // updates the current message-store

   }

   @Override

   public ServerData<I, V, E> getServerData() {

     return serverData;

   }

   @Override

   public void close() {

     nettyServer.stop();

   }

   @Override

   public void setFlowControl(FlowControl flowControl) {

     nettyServer.setFlowControl(flowControl);

   }

   @Override

   public long getBytesReceivedPerSuperstep() {

     return nettyServer.getInByteCounter().getBytesReceivedPerSuperstep();

   }

   @Override

   public void resetBytesReceivedPerSuperstep() {

     nettyServer.getInByteCounter().resetBytesReceivedPerSuperstep();

   }

   @Override

   public long getBytesReceived() {

     return nettyServer.getInByteCounter().getBytesReceived();

   }

 }