你可以这样使用:
public static void main(String[] args) {
SparkSession spark = SparkSession
.builder()
.appName("javaGraphx")
.getOrCreate();
List<Tuple2<Object, Data>> vectorList = Arrays.asList(
new Tuple2[]{
new Tuple2(1L, new Data("A", 1)),
new Tuple2(2L, new Data("B", 1)),
new Tuple2(3L, new Data("C", 1))}
);
JavaRDD<Tuple2<Object, Data>> users = spark.sparkContext().parallelize(
JavaConverters.asScalaIteratorConverter(vectorList.iterator()).asScala()
.toSeq(), 1, scala.reflect.ClassTag$.MODULE$.apply(Tuple2.class)
).toJavaRDD();
List<Edge<Integer>> edgeList = Arrays.asList(
new Edge[]{
new Edge(1L, 2L, 1),
new Edge(2L, 3L, 1)});
JavaRDD<Edge<Integer>> followers = spark.sparkContext().parallelize(
JavaConverters.asScalaIteratorConverter(edgeList.iterator()).asScala()
.toSeq(), 1, scala.reflect.ClassTag$.MODULE$.apply(Edge.class)
).toJavaRDD();
// construct graph
Graph<OddRange, Integer> followerGraph =
GraphImpl
.apply(
users.rdd(),
followers.rdd(),
new Data("OVER", 0),
StorageLevel.MEMORY_AND_DISK(),
StorageLevel.MEMORY_AND_DISK(),
scala.reflect.ClassTag$.MODULE$.apply(Data.class),
scala.reflect.ClassTag$.MODULE$.apply(Integer.class)
);
//If you wanna use pregel, you can use it like this.
Graph<OddRange, Integer> dd = Pregel.apply(followerGraph,
new Data("", 0),
5,
EdgeDirection.Out(),
new Vprog(),
new SendMsg(),
new MergemMsg(),
scala.reflect.ClassTag$.MODULE$.apply(OddRange.class),
scala.reflect.ClassTag$.MODULE$.apply(Integer.class),
scala.reflect.ClassTag$.MODULE$.apply(OddRange.class)
);
RDD<OddRange> r = dd
.vertices()
.toJavaRDD()
.map(t->t._2)
.rdd();
r.toJavaRDD().foreach(
o->{
System.out.println(o.getS());
}
);
}
//If you wanna use pregel, you can use it like this.
static class Vprog extends AbstractFunction3< Object, Data, Data, Data> implements Serializable {
@Override
public Data apply(Object l, Data self, Data sumOdd) {
System.out.println(l + "---" + self.getS()+self.getI()+" ---> "+sumOdd.getS()+sumOdd.getI());
self.setS(sumOdd.getS() + self.getS());
self.setI(self.getI() + sumOdd.getI());
System.out.println(l + "---" + self.getS()+self.getI()+" ---> "+sumOdd.getS()+sumOdd.getI());
//Don't just return self here, return a new one;
return new Data(self.getS(), self.getI());
}
}
static class SendMsg extends AbstractFunction1<EdgeTriplet<Data,Integer>, scala.collection.Iterator<Tuple2<Object, Data>>> implements Serializable {
@Override
public scala.collection.Iterator<Tuple2<Object, Data>> apply(EdgeTriplet<Data,Integer> t) {
System.out.println(t.srcId()+" ---> "+t.dstId()+" with: "+t.srcAttr().getS()+t.srcAttr().getI()+" ---> "+t.dstAttr().getS()+t.dstAttr().getI());
if(t.srcAttr().getI() <= 8){
List<Tuple2<Object, Data>> data = new ArrayList();
data.add(new Tuple2<>( t.dstId(), new Data(t.srcAttr().getS(), t.srcAttr().getI())));
return JavaConverters.asScalaIteratorConverter(data.iterator()).asScala();
}else{
return JavaConverters.asScalaIteratorConverter(new ArrayList<Tuple2<Object, Data>>().iterator()).asScala();
}
}
}
static class MergemMsg extends AbstractFunction2< Data, Data, Data> implements Serializable {
@Override
public Data apply(Data a, Data b) {
return new Data( "" + a.getS() + b.getS(), a.getI() + b.getI());
}
}