用Java在GraphFrames中获取shortestPaths

<dependency>
    <groupId>graphframes</groupId>
    <artifactId>graphframes</artifactId>
    <version>0.8.1-spark3.0-s_2.12</version>
</dependency>

import org.apache.spark.sql.Dataset;
import org.apache.spark.sql.Row;
import org.apache.spark.sql.SparkSession;
import org.graphframes.GraphFrame;

SparkSession spark = SparkSession.builder()
        .appName("ShortestPathsExample")
        .master("local[*]")  // 根据实际情况设置Master URL
        .getOrCreate();

// 创建顶点数据集
List<Row> verticesList = Arrays.asList(
        RowFactory.create("A", "Alice"),
        RowFactory.create("B", "Bob"),
        RowFactory.create("C", "Charlie"),
        RowFactory.create("D", "David"),
        RowFactory.create("E", "Esther"),
        RowFactory.create("F", "Fanny")
);
StructType verticesSchema = new StructType(new StructField[]{
        new StructField("id", DataTypes.StringType, false, Metadata.empty()),
        new StructField("name", DataTypes.StringType, false, Metadata.empty())
});
Dataset<Row> vertices = spark.createDataFrame(verticesList, verticesSchema);

// 创建边数据集
List<Row> edgesList = Arrays.asList(
        RowFactory.create("A", "B", 1),
        RowFactory.create("B", "C", 2),
        RowFactory.create("B", "D", 3),
        RowFactory.create("C", "D", 1),
        RowFactory.create("D", "E", 2),
        RowFactory.create("E", "F", 3),
        RowFactory.create("F", "A", 4)
);
StructType edgesSchema = new StructType(new StructField[]{
        new StructField("src", DataTypes.StringType, false, Metadata.empty()),
        new StructField("dst", DataTypes.StringType, false, Metadata.empty()),
        new StructField("weight", DataTypes.IntegerType, false, Metadata.empty())
});
Dataset<Row> edges = spark.createDataFrame(edgesList, edgesSchema);

GraphFrame graph = new GraphFrame(vertices, edges);

Dataset<Row> result = graph.shortestPaths().landmarks(Arrays.asList("A", "E")).run();
result.show();