图计算11.11优惠活动

import org.apache.giraph.graph.BasicComputation;
import org.apache.giraph.graph.Vertex;
import org.apache.hadoop.io.IntWritable;
import org.apache.hadoop.io.LongWritable;

public class ShortestPathComputation extends BasicComputation<
    LongWritable, IntWritable, NullWritable, IntWritable> {

  @Override
  public void compute(Vertex<LongWritable, IntWritable, NullWritable> vertex,
                      Iterable<IntWritable> messages) {
    int minDist = isSource(vertex) ? 0 : Integer.MAX_VALUE;
    for (IntWritable msg : messages) {
      minDist = Math.min(minDist, msg.get());
    }
    vertex.setValue(new IntWritable(minDist));
    if (minDist < Integer.MAX_VALUE) {
      for (Edge<LongWritable, NullWritable> edge : vertex.getEdges()) {
        sendMessage(edge.getTargetVertexId(), new IntWritable(minDist + 1));
      }
    }
  }

  private boolean isSource(Vertex<LongWritable, IntWritable, NullWritable> vertex) {
    return vertex.getId().get() == getSourceId();
  }
}