导入CSV文件数据

本文以一个示例说明如何使用Exchange将存储在HDFS或本地的CSV文件数据导入Nebula Graph。

如果要向Nebula Graph导入本地CSV文件，请参见Nebula Importer。

数据集

本文以basketballplayer数据集为例。

环境配置

本文示例在MacOS下完成，以下是相关的环境配置信息：

  • 硬件规格：

    • CPU：1.7 GHz Quad-Core Intel Core i7
    • 内存：16 GB

  • Spark：2.4.7 单机版

  • Hadoop：2.9.2 伪分布式部署

  • Nebula Graph：2.0.1。使用Docker Compose部署。

前提条件

开始导入数据之前，用户需要确认以下信息：

  • 已经安装部署Nebula Graph并获取如下信息：

    • Graph服务和Meta服务的的IP地址和端口。

    • 拥有Nebula Graph写权限的用户名和密码。

  • 已经编译Exchange。详情请参见编译Exchange。本示例中使用Exchange 2.1.0。

  • 已经安装Spark。

  • 了解Nebula Graph中创建Schema的信息，包括Tag和Edge type的名称、属性等。

  • 如果文件存储在HDFS上，需要确认Hadoop服务运行正常。

  • 如果文件存储在本地且Nebula Graph是集群架构，需要在集群每台机器本地相同目录下放置文件。

操作步骤

步骤 1：在Nebula Graph中创建Schema

分析CSV文件中的数据，按以下步骤在Nebula Graph中创建Schema：

  1. 确认Schema要素。Nebula Graph中的Schema要素如下表所示。

    要素名称属性
    Tagplayername string, age int
    Tagteamname string
    Edge Typefollowdegree int
    Edge Typeservestart_year int, end_year int

  2. 使用Nebula Console创建一个图空间basketballplayer，并创建一个Schema，如下所示。

    1. ## 创建图空间
    2. nebula> CREATE SPACE basketballplayer \
    3.         (partition_num = 10, \
    4.         replica_factor = 1, \
    5.         vid_type = FIXED_STRING(30));
    7. ## 选择图空间basketballplayer
    8. nebula> USE basketballplayer;
    10. ## 创建Tag player
    11. nebula> CREATE TAG player(name string, age int);
    13. ## 创建Tag team
    14. nebula> CREATE TAG team(name string);
    16. ## 创建Edge type follow
    17. nebula> CREATE EDGE follow(degree int);
    19. ## 创建Edge type serve
    20. nebula> CREATE EDGE serve(start_year int, end_year int);

更多信息，请参见快速开始。

步骤 2：处理CSV文件

确认以下信息：

  1. 处理CSV文件以满足Schema的要求。

  2. 获取CSV文件存储路径。

步骤 3：修改配置文件

编译Exchange后，复制target/classes/application.conf文件设置CSV数据源相关的配置。在本示例中，复制的文件名为csv_application.conf。各个配置项的详细说明请参见配置说明。

  1. {
  2.   # Spark相关配置
  3.   spark: {
  4.     app: {
  5.       name: Nebula Exchange 2.1.0
  6.     }
  7.     driver: {
  8.       cores: 1
  9.       maxResultSize: 1G
  10.     }
  11.     executor: {
  12.         memory:1G
  13.     }
  15.     cores {
  16.       max: 16
  17.     }
  18.   }
  20.   # Nebula Graph相关配置
  21.   nebula: {
  22.     address:{
  23.       # 指定Graph服务和所有Meta服务的IP地址和端口。
  24.       # 如果有多台服务器，地址之间用英文逗号（,）分隔。
  25.       # 格式: "ip1:port","ip2:port","ip3:port"
  26.       graph:["127.0.0.1:9669"]
  27.       meta:["127.0.0.1:9559"]
  28.     }
  30.     # 指定拥有Nebula Graph写权限的用户名和密码。
  31.     user: root
  32.     pswd: nebula
  34.     # 指定图空间名称。
  35.     space: basketballplayer
  36.     connection {
  37.       timeout: 3000
  38.       retry: 3
  39.     }
  40.     execution {
  41.       retry: 3
  42.     }
  43.     error: {
  44.       max: 32
  45.       output: /tmp/errors
  46.     }
  47.     rate: {
  48.       limit: 1024
  49.       timeout: 1000
  50.     }
  51.   }
  53.   # 处理点
  54.   tags: [
  55.     # 设置Tag player相关信息。
  56.     {
  57.       # 指定Nebula Graph中定义的Tag名称。
  58.       name: player
  59.       type: {
  60.         # 指定数据源，使用CSV。
  61.         source: csv
  63.         # 指定如何将点数据导入Nebula Graph：Client或SST。
  64.         sink: client
  65.       }
  67.       # 指定CSV文件的路径。
  68.       # 如果文件存储在HDFS上，用双引号括起路径，以hdfs://开头，例如"hdfs://ip:port/xx/xx"。
  69.       # 如果文件存储在本地，用双引号括起路径，以file://开头，例如"file:///tmp/xx.csv"。
  70.       path: "hdfs://192.168.11.13:9000/data/vertex_player.csv"
  72.       # 如果CSV文件没有表头，使用[_c0, _c1, _c2, ..., _cn]表示其表头，并将列指示为属性值的源。
  73.       # 如果CSV文件有表头，则使用实际的列名。
  74.       fields: [_c1, _c2]
  76.       # 指定Nebula Graph中定义的属性名称。
  77.       # fields与nebula.fields的顺序必须一一对应。
  78.       nebula.fields: [age, name]
  80.       # 指定一个列作为VID的源。
  81.       # vertex的值必须与上述fields或者csv.fields中的列名保持一致。
  82.       # 目前，Nebula Graph 2.0.1仅支持字符串或整数类型的VID。
  83.       # 不要使用vertex.policy映射。
  84.       vertex: {
  85.         field:_c0
  86.         # policy:hash
  87.       }
  89.       # 指定的分隔符。默认值为英文逗号（,）。
  90.       separator: ","
  92.       # 如果CSV文件有表头，请将header设置为true。
  93.       # 如果CSV文件没有表头，请将header设置为false。默认值为false。
  94.       header: false
  96.       # 指定单批次写入Nebula Graph的最大点数量。
  97.       batch: 256
  99.       # 指定Spark分片数量。
  100.       partition: 32
  101.     }
  103.     # 设置Tag team相关信息。
  104.     {
  105.       # 指定Nebula Graph中定义的Tag名称。
  106.       name: team
  107.       type: {
  108.         # 指定数据源，使用CSV。
  109.         source: csv
  111.         # 指定如何将点数据导入Nebula Graph：Client或SST。
  112.         sink: client
  113.       }
  115.       # 指定CSV文件的路径。
  116.       # 如果文件存储在HDFS上，用双引号括起路径，以hdfs://开头，例如"hdfs://ip:port/xx/xx"。
  117.       # 如果文件存储在本地，用双引号括起路径，以file://开头，例如"file:///tmp/xx.csv"。
  118.       path: "hdfs://192.168.11.13:9000/data/vertex_team.csv"
  120.       # 如果CSV文件没有表头，使用[_c0, _c1, _c2, ..., _cn]表示其表头，并将列指示为属性值的源。
  121.       # 如果CSV文件有表头，则使用实际的列名。
  122.       fields: [_c1]
  124.       # 指定Nebula Graph中定义的属性名称。
  125.       # fields与nebula.fields的顺序必须一一对应。
  126.       nebula.fields: [name]
  128.       # 指定一个列作为VID的源。
  129.       # vertex的值必须与上述fields或者csv.fields中的列名保持一致。
  130.       # 目前，Nebula Graph 2.0.1仅支持字符串或整数类型的VID。
  131.       # 不要使用vertex.policy映射。
  132.       vertex: {
  133.         field:_c0
  134.         # policy:hash
  135.       }
  137.       # 指定的分隔符。默认值为英文逗号（,）。
  138.       separator: ","
  140.       # 如果CSV文件有表头，请将header设置为true。
  141.       # 如果CSV文件没有表头，请将header设置为false。默认值为false。
  142.       header: false
  144.       # 指定单批次写入Nebula Graph的最大点数量。
  145.       batch: 256
  147.       # 指定Spark分片数量。
  148.       partition: 32
  149.     }
  152.     # 如果需要添加更多点，请参考前面的配置进行添加。
  153.   ]
  154.   # 处理边
  155.   edges: [
  156.     # 设置Edge type follow相关信息。
  157.     {
  158.       # 指定Nebula Graph中定义的Edge type名称。
  159.       name: follow
  160.       type: {
  161.         # 指定数据源，使用CSV。
  162.         source: csv
  164.         # 指定如何将点数据导入Nebula Graph：Client或SST。
  165.         sink: client
  166.       }
  168.       # 指定CSV文件的路径。
  169.       # 如果文件存储在HDFS上，用双引号括起路径，以hdfs://开头，例如"hdfs://ip:port/xx/xx"。
  170.       # 如果文件存储在本地，用双引号括起路径，以file://开头，例如"file:///tmp/xx.csv"。
  171.       path: "hdfs://192.168.11.13:9000/data/edge_follow.csv"
  173.       # 如果CSV文件没有表头，使用[_c0, _c1, _c2, ..., _cn]表示其表头，并将列指示为属性值的源。
  174.       # 如果CSV文件有表头，则使用实际的列名。
  175.       fields: [_c2]
  177.       # 指定Nebula Graph中定义的属性名称。
  178.       # fields与nebula.fields的顺序必须一一对应。
  179.       nebula.fields: [degree]
  181.       # 指定一个列作为起始点和目的点的源。
  182.       # vertex的值必须与上述fields或者csv.fields中的列名保持一致。
  183.       # 目前，Nebula Graph 2.0.1仅支持字符串或整数类型的VID。
  184.       # 不要使用vertex.policy映射。
  185.       source: {
  186.         field: _c0
  187.       }
  188.       target: {
  189.         field: _c1
  190.       }
  192.       # 指定的分隔符。默认值为英文逗号（,）。
  193.       separator: ","
  195.       # 指定一个列作为rank的源(可选)。
  197.       #ranking: rank
  199.       # 如果CSV文件有表头，请将header设置为true。
  200.       # 如果CSV文件没有表头，请将header设置为false。默认值为false。
  201.       header: false
  203.       # 指定单批次写入Nebula Graph的最大边数量。
  204.       batch: 256
  206.       # 指定Spark分片数量。
  207.       partition: 32
  208.     }
  210.     # 设置Edge type serve相关信息。
  211.     {
  212.       # 指定Nebula Graph中定义的Edge type名称。
  213.       name: serve
  214.       type: {
  215.         # 指定数据源，使用CSV。
  216.         source: csv
  218.         # 指定如何将点数据导入Nebula Graph：Client或SST。
  219.         sink: client
  220.       }
  222.       # 指定CSV文件的路径。
  223.       # 如果文件存储在HDFS上，用双引号括起路径，以hdfs://开头，例如"hdfs://ip:port/xx/xx"。
  224.       # 如果文件存储在本地，用双引号括起路径，以file://开头，例如"file:///tmp/xx.csv"。
  225.       path: "hdfs://192.168.11.13:9000/data/edge_serve.csv"
  227.       # 如果CSV文件没有表头，使用[_c0, _c1, _c2, ..., _cn]表示其表头，并将列指示为属性值的源。
  228.       # 如果CSV文件有表头，则使用实际的列名。
  229.       fields: [_c2,_c3]
  231.       # 指定Nebula Graph中定义的属性名称。
  232.       # fields与nebula.fields的顺序必须一一对应。
  233.       nebula.fields: [start_year, end_year]
  235.       # 指定一个列作为起始点和目的点的源。
  236.       # vertex的值必须与上述fields或者csv.fields中的列名保持一致。
  237.       # 目前，Nebula Graph 2.0.1仅支持字符串或整数类型的VID。
  238.       # 不要使用vertex.policy映射。
  239.       source: {
  240.         field: _c0
  241.       }
  242.       target: {
  243.         field: _c1
  244.       }
  246.       # 指定的分隔符。默认值为英文逗号（,）。
  247.       separator: ","
  249.       # 指定一个列作为rank的源(可选)。
  250.       #ranking: _c5
  252.       # 如果CSV文件有表头，请将header设置为true。
  253.       # 如果CSV文件没有表头，请将header设置为false。默认值为false。
  254.       header: false
  256.       # 指定单批次写入Nebula Graph的最大边数量。
  257.       batch: 256
  259.       # 指定Spark分片数量。
  260.       partition: 32
  261.     }
  263.   ]
  264.   # 如果需要添加更多边，请参考前面的配置进行添加。
  265. }

步骤 4：向Nebula Graph导入数据

运行如下命令将CSV文件数据导入到Nebula Graph中。关于参数的说明，请参见导入命令参数。

  1. ${SPARK_HOME}/bin/spark-submit --master "local" --class com.vesoft.nebula.exchange.Exchange <nebula-exchange-2.1.0.jar_path> -c <csv_application.conf_path>

示例：

  1. ${SPARK_HOME}/bin/spark-submit  --master "local" --class com.vesoft.nebula.exchange.Exchange  /root/nebula-spark-utils/nebula-exchange/target/nebula-exchange-2.1.0.jar  -c /root/nebula-spark-utils/nebula-exchange/target/classes/csv_application.conf

用户可以在返回信息中搜索batchSuccess.<tag_name/edge_name>，确认成功的数量。例如batchSuccess.follow: 300。

步骤 5：（可选）验证数据

用户可以在Nebula Graph客户端（例如Nebula Graph Studio）中执行查询语句，确认数据是否已导入。例如：

  1. GO FROM "player100" OVER follow;

用户也可以使用命令SHOW STATS查看统计数据。

步骤 6：（如有）在Nebula Graph中重建索引

导入数据后，用户可以在Nebula Graph中重新创建并重建索引。详情请参见索引介绍。