背景

本文介绍 MySQL 8.0 shell 子模块 Util 的两个导入特性 importTable/import_table(JS和python 版本的命名差异)、importJson/import_json的使用方法。其中 import_table 是通过传统 MySQL 协议来通信,Import_json 是通过 X 插件协议来通信。MySQL 一直以来提供导入文件 SQL 命令 load data infile(单线程)以及对应的可执行文件 mysqlimport(多线程)。比如我导入 100W 行示例数据到表 ytt.tl1,花了 24 秒。这个已经是 MySQL 默认导入来的最快的。

  1. [root@mysql-dev ytt]# time mysqlimport  --login-path=ytt_master --fields-terminated-by=, --use-threads=4 ytt  /var/lib/mysql-files/tl1.csv

  2. ytt.tl1: Records: 1048576  Deleted: 0  Skipped: 0  Warnings: 0


  3. real    0m24.815s

  4. user    0m0.013s

  5. sys     0m0.031s


分析
那我们现在看下 mysqlimport 工具的升级版,mysqlshell 的 util 工具集。使用这两个工具之前,必须得临时开启 local_infile 选项。1. import_table

  1. [root@mysql-dev ytt]# mysqlsh

  2. MySQL Shell 8.0.17


  3. Copyright (c) 2016, 2019, Oracle and/or its affiliates. All rights reserved.

  4. Oracle is a registered trademark of Oracle Corporation and/or its affiliates.

  5. Other names may be trademarks of their respective owners.


  6. Type '\help' or '\?' for help; '\quit' to exit.

建立 3306 端口的新连接

MySQL  JS > \c admin@127.0.0.1:3306Creating a session to 'admin@127.0.0.1:3306'Fetching schema names for autocompletion... Press ^C to stop.Your MySQL connection id is 56Server version: 8.0.17 MySQL Community Server - GPLNo default schema selected; type \use <schema>to set one.

我这里切换为 python 模式

MySQL  127.0.0.1:3306 ssl  JS > \pySwitching to Python mode...MySQL  127.0.0.1:3306 ssl  Py > \use yttDefault schema set to `ytt`.

清空掉示例表 Ytt.tl1

MySQL  127.0.0.1:3306 ssl  ytt  Py > \sql truncate tl1;Fetching table and column names from `ytt` for auto-completion... Press ^C to stop.Query OK, 0 rows affected (0.2354 sec)

示例表表结构

MySQL  127.0.0.1:3306 ssl  ytt  Py > \sql show create table tl1\G*************************** 1. row ***************************       Table: tl1Create Table: CREATE TABLE `tl1` (  `id` bigint(20) unsigned NOT NULL AUTO_INCREMENT,  `r1` int(11) DEFAULT NULL,  `r2` int(11) DEFAULT NULL,  `r3` varchar(30) COLLATE utf8mb4_general_ci DEFAULT NULL,  `r4` datetime DEFAULT NULL,  PRIMARY KEY (`id`),  UNIQUE KEY `id` (`id`)) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci1 row in set (0.0011 sec)

import_table 有两个参数,第一个参数定义导入文件的路径,第二个定义相关选项,比如导入的格式,并发的数量等。定义文件路径(参数1)

MySQL  127.0.0.1:3306 ssl  ytt  Py > y_file1='/var/lib/mysql-files/tl1.csv'

定义选项(参数2)

MySQL  127.0.0.1:3306 ssl  ytt  Py > y_options1={"schema":"ytt","table":"tl1","fieldsTerminatedBy":",","showProgress":True,"threads":4}

执行导入:

MySQL  127.0.0.1:3306 ssl  ytt  Py > util.import_table(y_file1,y_options1);Importing from file '/var/lib/mysql-files/tl1.csv' to table `ytt`.`tl1` in MySQL Server at 127.0.0.1:3306 using 1 thread[Worker000] ytt.tl1: Records: 1048576  Deleted: 0  Skipped: 0  Warnings: 0100% (40.87 MB / 40.87 MB), 2.14 MB/sFile '/var/lib/mysql-files/tl1.csv' (40.87 MB) was imported in 16.7394 sec at 2.44 MB/sTotal rows affected in ytt.tl1: Records: 1048576  Deleted: 0  Skipped: 0  Warnings: 0

只花了不到 17 秒,比传统 mysqlimport 快了不少。

我们上面指定了显式指定了字段分隔符,那有没有已经定义好的组合格式呢? 答案是有的,选项 dialect 可以指定以下格式:csv,tsvjsoncsv-unix

那么上面的导入,我们可以更简单,改下变量 y_options1 的定义

MySQL  127.0.0.1:3306 ssl  ytt  Py > \sql truncate tl1;Query OK, 0 rows affected (0.0776 sec)MySQL  127.0.0.1:3306 ssl  ytt  Py > y_options1={"schema":"ytt","table":"tl1","dialect":"csv-unix","showProgress":True,"threads":4}MySQL  127.0.0.1:3306 ssl  ytt  Py > util.import_table(y_file1,y_options1);Importing from file '/var/lib/mysql-files/tl1.csv' to table `ytt`.`tl1` in MySQL Server at 127.0.0.1:3306 using 1 thread[Worker000] ytt.tl1: Records: 1048576  Deleted: 0  Skipped: 0  Warnings: 0100% (40.87 MB / 40.87 MB), 2.67 MB/sFile '/var/lib/mysql-files/tl1.csv' (40.87 MB) was imported in 14.1000 sec at 2.90 MB/sTotal rows affected in ytt.tl1: Records: 1048576  Deleted: 0  Skipped: 0  Warnings: 0

导入时间差不多。这里要说明下,dialect 选项的优先级比较低,比如添加了'linesTerminatedBy':'\r\n', 则覆盖他自己的'\n'。选项 diaelect 还有一个可选值为 json,可以直接把 json 结果导入到文档表里。比如我新建一张表 tl1_json

MySQL  127.0.0.1:3306 ssl  ytt  Py > \sql create table tl1_json(     id bigint primary key,     content json);Query OK, 0 rows affected (0.3093 sec)

重新定义文件以及导入选项。

MySQL  127.0.0.1:3306 ssl  ytt  Py > y_file2='/var/lib/mysql-files/tl1.json'MySQL  127.0.0.1:3306 ssl  ytt  Py > rows=['content']MySQL  127.0.0.1:3306 ssl  ytt  Py > y_options2={"schema":"ytt","table":"tl1_json","dialect":"json","showProgress":True,"threads":4,'columns':rows}

导入 JSON 数据

MySQL  127.0.0.1:3306 ssl  ytt  Py > util.import_table(y_file2,y_options2)Importing from file '/var/lib/mysql-files/tl1.json' to table `ytt`.`tl1_json` in MySQL Server at 127.0.0.1:3306 using 2 threads[Worker000] ytt.tl1_json: Records: 464633  Deleted: 0  Skipped: 0  Warnings: 0[Worker001] ytt.tl1_json: Records: 583943  Deleted: 0  Skipped: 0  Warnings: 0100% (90.15 MB / 90.15 MB), 2.71 MB/sFile '/var/lib/mysql-files/tl1.json' (90.15 MB) was imported in 23.3530 sec at 3.86 MB/sTotal rows affected in ytt.tl1_json: Records: 1048576  Deleted: 0  Skipped: 0  Warnings: 0

速度也还可以,不到 24 秒。那导入 json 数据,就必须得提到以 X 插件协议通信的工具 import_json了。
2. imort_json我们切换到 mysqlx 端口

MySQL  127.0.0.1:3306 ssl  ytt  Py > \c admin@127.0.0.1:33060Creating a session to 'admin@127.0.0.1:33060'Fetching schema names for autocompletion... Press ^C to stop.Closing old connection...Your MySQL connection id is 16 (X protocol)Server version: 8.0.17 MySQL Community Server - GPLNo default schema selected; type \use <schema> to set one. MySQL  127.0.0.1:33060+ ssl  Py > \use yttDefault schema `ytt` accessible through db.-- 清空表tl1_json MySQL  127.0.0.1:33060+ ssl  ytt  Py > \sql truncate tl1_json;Query OK, 0 rows affected (0.1098 sec)

import_json 参数和 Import_table 参数类似,这里我改下选项

  1. MySQL  127.0.0.1:33060+ ssl  ytt  Py > y_file3=y_file2


  2. MySQL  127.0.0.1:33060+ ssl  ytt  Py > y_options3={"schema":"ytt","table":"tl1_json",'tableColumn':'content'}


  3. MySQL  127.0.0.1:33060+ ssl  ytt  Py > util.import_json(y_file3,y_options3)

  4. Importing from file "/var/lib/mysql-files/tl1.json" to table `ytt`.`tl1_json` in MySQL Server at 127.0.0.1:33060


  5. .. 517776.. 1032724.. 1048576.. 1048576

  6. Processed 90.15 MB in 1048576 documents in 35.2400 sec (29.76K documents/s)

  7. Total successfully imported documents 1048576 (29.76K documents/s)

我在手册上没有看到多线程的选项,所以单线程跑 35 秒慢了些。查看刚刚导入的数据

MySQL  127.0.0.1:33060+ ssl  ytt  Py > \sql select  id,json_pretty(content) from tl1_json limit 1\G*************************** 1. row ***************************                  id: 1json_pretty(content): {  "id": 1,  "r1": 10,  "r2": 10,  "r3": "mysql",  "r4": "2019-09-16 16:49:50.000000"}1 row in set (0.0007 sec)

import_json 不仅仅可以导入 Json 数据,更重要的是可以在 BSON 和 JSON 之间平滑的转换,有兴趣的同学可以去 TRY 下。