SQL查询获取自增主键

前言

在业务中经常会遇到类似以下模型的表：
表1 STUDENT表(学生表)

stu_id	stu_name
1	张三
2	李四

表2 SCHOOL表(学校表)

school_id	school_name
1	aa大学
2	bb大学

表3 RELATION表(学校与学生的关系表)

school_id	stu_id
1	2
2	1

其中stu_id与school_id均为自增主键，此时要一次性新增数据就比较麻烦，因为要获取新增学生和学校的自增id写入关系表。

@@identity

使用

SELECT @@identity;

可以得到上一次插入记录时自动产生的ID，在一条INSERT语句执行完毕后，@@identity返回的即为该INSERT语句的自增id，如果是多条执行完毕后，@@identity返回的为最后一条数据的自增id。

解决办法

借助@@identity可以轻松地解决之前的问题，假如现在有一个新学生王五，在CC大学就读，此时我们就可以利用下列语句来新增数据。

INSERT INTO STUDENT(stu_name) VALUES("王五");
# 使用变量stu1 存储 新增学生的自增id
SELECT @stu1 := @@identity;

INSERT INTO SCHOOL(school_name) VALUES("CC大学");
# 使用变量 school1 存储 新增学校的自增ID
SELECT @school1 := @@identity;
#写入关系表
INSERT INTO RELATION(school_id, stu_id) VALUES(@school1, @stu1);