MySQL中通过SQL语句实现雪花ID生成方案实例

一、雪花算法简介分布式系统中生成唯一ID的需求日益普遍，雪花算法因其高效可靠的特性成为主流解决方案。该算法通过64位二进制数字构造ID，将标识符划分为符号位、时间戳、工作机器ID和序列号四个关键部分，确保生成的ID具备全局唯一性和时间有序性。这种结构化设计使得系统能够高效处理数据存储与查询任务，特别适用于大规模分布式环境。二、问题描述(项目场景)在实际开发中遇到需要将A表特定数据迁移至B表的情况。由于B表采用雪花算法生成的主键ID，且原始数据已使用代码生成雪花ID，本次迁移需通过SQL语句实现相同规则的ID生成功能。关键点在于确保新生成的ID符合原系统的数据规范，同时完成跨表数据插入操作。insert into select语句是实现这一需求的核心语法。-- 使用insert into select语法实现跨表数据迁移insert into table_name2(id, project_code)select 雪花算法id, project_code from table_name1 where type = 1;三、sql生成雪花算法(已验证)3.1 存储函数创建SET @last_timestamp = -1; -- 初始化全局变量SET @sequence = 0; -- 初始化全局变量DROP FUNCTION IF EXISTS generate_snowflake_id;DELIMITER //CREATE FUNCTION generate_snowflake_id() RETURNS BIGINT READS SQL DATABEGIN DECLARE timestamp BIGINT; DECLARE machine_id BIGINT DEFAULT 1; -- 假设机器 ID 为 1 DECLARE data_center_id BIGINT DEFAULT 0; -- 假设数据中心 ID 为 0 DECLARE epoch BIGINT DEFAULT 1288834974657; -- 2010-01-01 00:00:00 UTC 的毫秒数 -- 获取当前时间戳（毫秒） SET timestamp = FLOOR(UNIX_TIMESTAMP(NOW(3)) * 1000) - epoch; -- 检查当前时间戳 IF timestamp = @last_timestamp THEN SET @sequence = (@sequence + 1) % 4096; -- 在同一毫秒内递增序列号 ELSE SET @sequence = 0; -- 不同毫秒重置序列号 END IF; SET @last_timestamp = timestamp; -- 存储当前时间戳 -- 生成雪花 ID RETURN (timestamp