[mysql] 如何让string类型ID 实现自增长

V_Jan · · 116 次点击 · · 开始浏览    

mysql里 如何让string类型ID 实现自增长?这里有2种方法:
首先,数据库类型中能实现自增比较常见的是int类型,它可以很方便的设置为AUTO_INCREMENT, 所以思路就是,假如A表需要这样的ID,那就在某个地方用int保存一个对应的序列。


image.png

为方便讲解,假设我们有张Application表,它需要有如下数据结构:

application

id的值预期是APP-1, APP-2, APP-3...APP-n
方法:
1.(条件允许,比较推荐)利用2种数据库 redis+mysql:
因为redis的操作很快(1s的读取次数可以自行搜索下),而且是原子性的,因此可以忽略锁的问题。
Redis存在一张与application表对应的表,比如application_seq. 里面只存当前application表APP-N的最大值,然后取出后调用Redis的Incr(key)实现自增,该函数会返回自增后的值,那这个值就是我们要的,然后在拼上特定前缀,就是mysql需要的ID值。
代码实现(golang+gorm),在数据插入之前,去Redis取当前

func (a *Application) InsertAppWithSE(siteEnvIds []string) (err error) {
    nextId, err := GetNextSeq("application") //去Redis取ID
    if err != nil {
        return
    }
    a.ID = "APP-" + nextId //拼凑成mysql ID
    err = db.Transaction(func(tx *gorm.DB) error {
        if err := tx.Create(a).Error; err != nil {
            return errors.New("fail to create new application, " + err.Error())
        }
        if len(siteEnvIds) > 0 {
            for _, seId := range siteEnvIds {
                r := SiteEnvApplication{
                    Model: Model{
                        CreatedBy: a.CreatedBy,
                    },
                    SiteEnvId: seId,
                    AppId:     a.ID,
                }
                err = r.Insert()
                if err != nil {
                    return errors.New("insert new site_env application failed:" + err.Error())
                }
            }
        }
        return nil
    })
    return
}
//获取mysql中ID的最大值,仅在Redis里没有值的时候用
func GetBiggestIDOfTable(tableName string) (id string, err error) {
    var existIds []string
    err = db.Table(tableName).Select("id").Pluck("id", &existIds).Error
    if err != nil {
        return "", errors.New("error when try to get all exist ids of table " + tableName)
    }
    var ids []int
    for _, id := range existIds {
        tmp := strings.Split(id, "-")
        i, _ := strconv.Atoi(tmp[1])
        ids = append(ids, i)
    }
    sort.Sort(sort.Reverse(sort.IntSlice(ids)))
    return strconv.Itoa(ids[0]), nil
}

func GetNextSeq(tableName string) (nextId string, err error) {
    key := tableName + "_seq"
    lastId, err := redis.Get(key)
    if err != nil {
        return "", err
    }

    if lastId == "" || lastId == "0" {
        biggestID, err := GetBiggestIDOfTable(tableName)
        if err != nil {
            return "", errors.New("error when try to get next seq of table " + tableName)
        }
        err = redis.Set(key, biggestID, 0)
        if err != nil {
            return "", errors.New("fail to save latest id to redis when the seq of " + tableName + " is empty")
        }
    }
    nextId, err = redis.Incr(key)
    return
}

2. (在并发的时候,可能会导致锁表问题)利用1种数据库(mysql),2张表(application, application_seq), 利用trigger实现ID自动拼凑。
2.1创建专门用于存放自增ID的表,只存放一个ID值,并且设置为自增

CREATE TABLE application_seq
(
  id INT NOT NULL AUTO_INCREMENT PRIMARY KEY
);
// 如果app表已经存在一段时间,且ID已经不是1的时候,可以设置对应application_seq的起始值
// ALTER TABLE application_seq AUTO_INCREMENT = 20;

2.2创建一个mysql trigger, 每当要插入application表之前,会调用这个trigger,这个trigger会往专门存放application表自增ID的application_seq里插入一条数据,并且马上调用LAST_INSERT_ID()获取上条执行命令返回的自增ID值(注:这个方法只有在紧接着INSERT语句,且insert语句所操作的表有AUTO_INCREMENT ID才有效),然后拼上特定的前缀,再写入要插入application的数据的ID。

DELIMITER $$
CREATE TRIGGER app_insert
BEFORE INSERT ON application
FOR EACH ROW
BEGIN
  INSERT INTO app_seq VALUES (NULL);
  SET NEW.id = CONCAT('APP-', LAST_INSERT_ID());
END$$
DELIMITER ;

2.3 测试,插入一条数据:

INSERT INTO application (name, repoId) VALUES ("myapplication1","REPO-1")

表里便有
APP-1|myapplication1|REPO-1

但当并发操作的时候,且可能是个长事务的时候,application_seq表可能会被锁住(AUTO-INC lock)。其他要插入application表的操作会因为这个锁而被迫等待前面那个事务完全处理完成才能继续操作。

截止发稿,mysql官网说明有8种锁, 下面是AUTO-INC lock


image.png

参考:
https://dev.mysql.com/doc/refman/8.0/en/innodb-locking.html#:~:text=b60000019d0110%3B%20asc%20%3B%3B-,Gap%20Locks,of%2015%20into%20column%20t.
https://www.cnblogs.com/lisq/p/12634475.html
https://stackoverflow.com/questions/17893988/how-to-make-mysql-table-primary-key-auto-increment-with-some-prefix


有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:V_Jan

查看原文:[mysql] 如何让string类型ID 实现自增长

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:1006366459

116 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传