mysql - 加载数据 INFILE - 绝望地卡住了

Question

希望我能很好地解释这一点......我有一个名为 artists 的表，基本上有 2 列，artist_pk(主键)和 artist_name。我正在使用 LOAD DATA INFILE 通过 CSV 文件将 artist_name 列的记录导入表 artists，并让 MySQL 提供 artist_pk 值。这一切都完成了。

现在我想为名为 albums 的表导入数据(通过相同的导入方法)。此处的三个相关列是 album_pk(主要)、album_name 和 artist_pk(来自 artists 表的外键)。同样，我将让 mySQL 分配 album_pk 值。

在我的 CSV 数据中，我有 album_name 和 artist_name。我没有 artist_pk 值。所以我的问题是:我可以导入 artist_name CSV 列作为我的 tables 导入的一部分，而不是按原样使用它，而是指示 mySQL 使用关联的 artist_pk artists 表中的 PRIMARY KEY 值？

例如artists 表中的一条记录:

artist_pk | artist_name  |
+-----------+--------------+
|       1 | Depeche Mode |
+-----------+--------------+

现在是我的 CSV 文件的摘录(我想将其放入我的 albums 表中)。

album_name  artist_name
Violator    Depeche Mode

'Violator' 将填充 albums.album_name。但是要填充 albums.artist_pk，我希望 MySQL 使用 artists.artist_name 中的“Depeche Mode”来获取其关联的 artist_pk 值(在本例中为 1)- 这是将进入表 albums 的值。

提前致谢!

Answer 1

假设 artists 表已经填充，您可以在 LOAD DATA INFILE 中利用 session 变量和 SET 子句在加载您的作品时进行必要的查找数据

LOAD DATA INFILE '/path/to/albums.txt'
INTO TABLE albums
FIELDS TERMINATED BY '\t' OPTIONALLY ENCLOSED BY '"'
LINES TERMINATED BY '\n'
IGNORE 1 LINES
(album_name, @artist_name) -- use a session variable to store a value read from the file
SET artist_pk = 
(
  SELECT artist_pk 
    FROM artists
   WHERE artist_name = @artist_name -- use a session variable to get an artist_pk
   LIMIT 1 -- this is to ensure that only one row will be returned
) 

让我们试试看

mysql> CREATE TABLE artists (`artist_pk` int not null auto_increment primary key, `artist_name` varchar(12));Query OK, 0 rows affected (0.07 sec)mysql> INSERT INTO artists (`artist_name`) VALUES ('Depeche Mode');Query OK, 1 row affected (0.00 sec)mysql> CREATE TABLE albums (`album_pk` int not null auto_increment primary key, album_name varchar(255), artist_pk int, foreign key (artist_pk) references artists(artist_pk));Query OK, 0 rows affected (0.03 sec)mysql> LOAD DATA INFILE '/tmp/albums.txt'    -> INTO TABLE albums    -> FIELDS TERMINATED BY '\t' OPTIONALLY ENCLOSED BY '"'    -> LINES TERMINATED BY '\n'    -> IGNORE 1 LINES    -> (album_name, @artist_name)    -> SET artist_pk =     -> (    ->   SELECT artist_pk    ->     FROM artists    ->    WHERE artist_name = @artist_name    -> );Query OK, 1 row affected (0.01 sec)Records: 1  Deleted: 0  Skipped: 0  Warnings: 0mysql> select * from albums;                                                                                       +----------+------------+-----------+| album_pk | album_name | artist_pk |+----------+------------+-----------+|        1 | Violator   |         1 |+----------+------------+-----------+1 row in set (0.00 sec)