2025/1/12

import org.apache.spark.graphx._
import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}

val conf = new SparkConf().setAppName("GraphX Example").setMaster("local")
val sc = new SparkContext(conf)

// 创建图
val vertices: RDD[(VertexId, String)] = sc.parallelize(Array((1L, "Alice"), (2L, "Bob"), (3L, "Charlie")))
val edges: RDD[Edge[String]] = sc.parallelize(Array(Edge(1L, 2L, "Friend"), Edge(2L, 3L, "Follow")))
val graph = Graph(vertices, edges)

// 计算PageRank
val ranks = graph.pageRank(0.0001).vertices
ranks.collect().foreach { case (id, rank) => println(s"$id has rank: $rank") }

sc.stop()