导入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));
    6. ## 选择图空间basketballplayer
    7. nebula> USE basketballplayer;
    8. ## 创建Tag player
    9. nebula> CREATE TAG player(name string, age int);
    10. ## 创建Tag team
    11. nebula> CREATE TAG team(name string);
    12. ## 创建Edge type follow
    13. nebula> CREATE EDGE follow(degree int);
    14. ## 创建Edge type serve
    15. 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. }
  14. cores {
  15. max: 16
  16. }
  17. }
  18. # Nebula Graph相关配置
  19. nebula: {
  20. address:{
  21. # 指定Graph服务和所有Meta服务的IP地址和端口。
  22. # 如果有多台服务器,地址之间用英文逗号(,)分隔。
  23. # 格式: "ip1:port","ip2:port","ip3:port"
  24. graph:["127.0.0.1:9669"]
  25. meta:["127.0.0.1:9559"]
  26. }
  27. # 指定拥有Nebula Graph写权限的用户名和密码。
  28. user: root
  29. pswd: nebula
  30. # 指定图空间名称。
  31. space: basketballplayer
  32. connection {
  33. timeout: 3000
  34. retry: 3
  35. }
  36. execution {
  37. retry: 3
  38. }
  39. error: {
  40. max: 32
  41. output: /tmp/errors
  42. }
  43. rate: {
  44. limit: 1024
  45. timeout: 1000
  46. }
  47. }
  48. # 处理点
  49. tags: [
  50. # 设置Tag player相关信息。
  51. {
  52. # 指定Nebula Graph中定义的Tag名称。
  53. name: player
  54. type: {
  55. # 指定数据源,使用CSV。
  56. source: csv
  57. # 指定如何将点数据导入Nebula Graph:Client或SST。
  58. sink: client
  59. }
  60. # 指定CSV文件的路径。
  61. # 如果文件存储在HDFS上,用双引号括起路径,以hdfs://开头,例如"hdfs://ip:port/xx/xx"。
  62. # 如果文件存储在本地,用双引号括起路径,以file://开头,例如"file:///tmp/xx.csv"。
  63. path: "hdfs://192.168.11.13:9000/data/vertex_player.csv"
  64. # 如果CSV文件没有表头,使用[_c0, _c1, _c2, ..., _cn]表示其表头,并将列指示为属性值的源。
  65. # 如果CSV文件有表头,则使用实际的列名。
  66. fields: [_c1, _c2]
  67. # 指定Nebula Graph中定义的属性名称。
  68. # fields与nebula.fields的顺序必须一一对应。
  69. nebula.fields: [age, name]
  70. # 指定一个列作为VID的源。
  71. # vertex的值必须与上述fields或者csv.fields中的列名保持一致。
  72. # 目前,Nebula Graph 2.0.1仅支持字符串或整数类型的VID。
  73. # 不要使用vertex.policy映射。
  74. vertex: {
  75. field:_c0
  76. # policy:hash
  77. }
  78. # 指定的分隔符。默认值为英文逗号(,)。
  79. separator: ","
  80. # 如果CSV文件有表头,请将header设置为true。
  81. # 如果CSV文件没有表头,请将header设置为false。默认值为false。
  82. header: false
  83. # 指定单批次写入Nebula Graph的最大点数量。
  84. batch: 256
  85. # 指定Spark分片数量。
  86. partition: 32
  87. }
  88. # 设置Tag team相关信息。
  89. {
  90. # 指定Nebula Graph中定义的Tag名称。
  91. name: team
  92. type: {
  93. # 指定数据源,使用CSV。
  94. source: csv
  95. # 指定如何将点数据导入Nebula Graph:Client或SST。
  96. sink: client
  97. }
  98. # 指定CSV文件的路径。
  99. # 如果文件存储在HDFS上,用双引号括起路径,以hdfs://开头,例如"hdfs://ip:port/xx/xx"。
  100. # 如果文件存储在本地,用双引号括起路径,以file://开头,例如"file:///tmp/xx.csv"。
  101. path: "hdfs://192.168.11.13:9000/data/vertex_team.csv"
  102. # 如果CSV文件没有表头,使用[_c0, _c1, _c2, ..., _cn]表示其表头,并将列指示为属性值的源。
  103. # 如果CSV文件有表头,则使用实际的列名。
  104. fields: [_c1]
  105. # 指定Nebula Graph中定义的属性名称。
  106. # fields与nebula.fields的顺序必须一一对应。
  107. nebula.fields: [name]
  108. # 指定一个列作为VID的源。
  109. # vertex的值必须与上述fields或者csv.fields中的列名保持一致。
  110. # 目前,Nebula Graph 2.0.1仅支持字符串或整数类型的VID。
  111. # 不要使用vertex.policy映射。
  112. vertex: {
  113. field:_c0
  114. # policy:hash
  115. }
  116. # 指定的分隔符。默认值为英文逗号(,)。
  117. separator: ","
  118. # 如果CSV文件有表头,请将header设置为true。
  119. # 如果CSV文件没有表头,请将header设置为false。默认值为false。
  120. header: false
  121. # 指定单批次写入Nebula Graph的最大点数量。
  122. batch: 256
  123. # 指定Spark分片数量。
  124. partition: 32
  125. }
  126. # 如果需要添加更多点,请参考前面的配置进行添加。
  127. ]
  128. # 处理边
  129. edges: [
  130. # 设置Edge type follow相关信息。
  131. {
  132. # 指定Nebula Graph中定义的Edge type名称。
  133. name: follow
  134. type: {
  135. # 指定数据源,使用CSV。
  136. source: csv
  137. # 指定如何将点数据导入Nebula Graph:Client或SST。
  138. sink: client
  139. }
  140. # 指定CSV文件的路径。
  141. # 如果文件存储在HDFS上,用双引号括起路径,以hdfs://开头,例如"hdfs://ip:port/xx/xx"。
  142. # 如果文件存储在本地,用双引号括起路径,以file://开头,例如"file:///tmp/xx.csv"。
  143. path: "hdfs://192.168.11.13:9000/data/edge_follow.csv"
  144. # 如果CSV文件没有表头,使用[_c0, _c1, _c2, ..., _cn]表示其表头,并将列指示为属性值的源。
  145. # 如果CSV文件有表头,则使用实际的列名。
  146. fields: [_c2]
  147. # 指定Nebula Graph中定义的属性名称。
  148. # fields与nebula.fields的顺序必须一一对应。
  149. nebula.fields: [degree]
  150. # 指定一个列作为起始点和目的点的源。
  151. # vertex的值必须与上述fields或者csv.fields中的列名保持一致。
  152. # 目前,Nebula Graph 2.0.1仅支持字符串或整数类型的VID。
  153. # 不要使用vertex.policy映射。
  154. source: {
  155. field: _c0
  156. }
  157. target: {
  158. field: _c1
  159. }
  160. # 指定的分隔符。默认值为英文逗号(,)。
  161. separator: ","
  162. # 指定一个列作为rank的源(可选)。
  163. #ranking: rank
  164. # 如果CSV文件有表头,请将header设置为true。
  165. # 如果CSV文件没有表头,请将header设置为false。默认值为false。
  166. header: false
  167. # 指定单批次写入Nebula Graph的最大边数量。
  168. batch: 256
  169. # 指定Spark分片数量。
  170. partition: 32
  171. }
  172. # 设置Edge type serve相关信息。
  173. {
  174. # 指定Nebula Graph中定义的Edge type名称。
  175. name: serve
  176. type: {
  177. # 指定数据源,使用CSV。
  178. source: csv
  179. # 指定如何将点数据导入Nebula Graph:Client或SST。
  180. sink: client
  181. }
  182. # 指定CSV文件的路径。
  183. # 如果文件存储在HDFS上,用双引号括起路径,以hdfs://开头,例如"hdfs://ip:port/xx/xx"。
  184. # 如果文件存储在本地,用双引号括起路径,以file://开头,例如"file:///tmp/xx.csv"。
  185. path: "hdfs://192.168.11.13:9000/data/edge_serve.csv"
  186. # 如果CSV文件没有表头,使用[_c0, _c1, _c2, ..., _cn]表示其表头,并将列指示为属性值的源。
  187. # 如果CSV文件有表头,则使用实际的列名。
  188. fields: [_c2,_c3]
  189. # 指定Nebula Graph中定义的属性名称。
  190. # fields与nebula.fields的顺序必须一一对应。
  191. nebula.fields: [start_year, end_year]
  192. # 指定一个列作为起始点和目的点的源。
  193. # vertex的值必须与上述fields或者csv.fields中的列名保持一致。
  194. # 目前,Nebula Graph 2.0.1仅支持字符串或整数类型的VID。
  195. # 不要使用vertex.policy映射。
  196. source: {
  197. field: _c0
  198. }
  199. target: {
  200. field: _c1
  201. }
  202. # 指定的分隔符。默认值为英文逗号(,)。
  203. separator: ","
  204. # 指定一个列作为rank的源(可选)。
  205. #ranking: _c5
  206. # 如果CSV文件有表头,请将header设置为true。
  207. # 如果CSV文件没有表头,请将header设置为false。默认值为false。
  208. header: false
  209. # 指定单批次写入Nebula Graph的最大边数量。
  210. batch: 256
  211. # 指定Spark分片数量。
  212. partition: 32
  213. }
  214. ]
  215. # 如果需要添加更多边,请参考前面的配置进行添加。
  216. }

步骤 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中重新创建并重建索引。详情请参见索引介绍