MySql 手动获取自增主键

• 最近在项目中遇到有在批量插入数据的时候有些数据需要关联父级节点的情况使用mysql 原来的自增有些无力。
• 废话不多说先上场景案例：
• 公司原本的项目用的是Oracle数据库，最近说是要把数据部署的云端让统一把数据库替换成MySql，所以大刀阔斧就开始了。
• 在原本的业务逻辑中所有的主键数据都是通过oracle来获取的，但转成MySql后有些场景的数据就要稍加变动一下了。
• 比如：我们会通过一些固定的数据去生成业绩（和绩效差不多一个意思），但是生成的业绩是多条并且业绩下还可以挂子业绩，这就涉及到关联父级节点的情况了。原来是先获取自增序列在赋值，现在是使用MySql表中自增主键在批量插入数据的时候无法把子业绩关联到父级上。

解决方法：

• 通过创建存储过程外加函数的方式维护一张sequences表来获取数据（一次解决终身受用）。
• 可以指定每次自增的大小，以及初始数。

 

  

select nextval('TESTDATA') AS batchId

CREATE DEFINER=`admin`@`%` FUNCTION `nextval`(seq_name VARCHAR (50)) RETURNS bigint(20)
BEGIN
  UPDATE
    SEQUENCES
  SET
    last_number = last_number + increment_by
  WHERE sequence_name = seq_name;
  RETURN currval (seq_name);
END;

CREATE DEFINER=`admin`@`%` FUNCTION `currval`(seq_name VARCHAR (50)) RETURNS bigint(20)
    NO SQL
BEGIN
  
  SELECT
    last_number INTO @VALUE
  FROM
    SEQUENCES
  WHERE sequence_name = seq_name;
  RETURN @VALUE;
END;

DROP TABLE IF EXISTS `sequences`;
CREATE TABLE `sequences`  (
  `sequence_owner` varchar(30) CHARACTER SET utf8mb4 COLLATE utf8mb4_bin NOT NULL,
  `sequence_name` varchar(30) CHARACTER SET utf8mb4 COLLATE utf8mb4_bin NOT NULL,
  `min_value` bigint(20) NULL DEFAULT 1,
  `max_value` bigint(20) NULL DEFAULT NULL,
  `increment_by` bigint(20) NOT NULL DEFAULT 1,
  `cycle_flag` varchar(1) CHARACTER SET utf8mb4 COLLATE utf8mb4_bin NULL DEFAULT NULL,
  `order_flag` varchar(1) CHARACTER SET utf8mb4 COLLATE utf8mb4_bin NULL DEFAULT NULL,
  `cache_size` bigint(20) NOT NULL,
  `last_number` bigint(20) NOT NULL,
  PRIMARY KEY (`sequence_name`) USING BTREE,
  UNIQUE INDEX `sel`(`sequence_owner`, `sequence_name`) USING BTREE
) ENGINE = InnoDB CHARACTER SET = utf8mb4 COLLATE = utf8mb4_bin ROW_FORMAT = Compact;

-- ----------------------------
-- Records of sequences
-- ----------------------------
INSERT INTO `sequences` VALUES ('SCM', 'TESTDATA', 1, 9999999999, 1, 'N', 'Y', 20, 0);
INSERT INTO `sequences` VALUES ('SCM', 'SEQ', 1, 9999999999, 1, 'N', 'N', 20, 0);

SET FOREIGN_KEY_CHECKS = 1;

 

如要转载请标明出处。