【Go - 超实用，3行代码实现个自增器】

场景

自增器的作用是生成一个唯一的递增序列号。这在一些需要生成自增id的场景十分有用，比如自增的订单号，任务号，序列号。

要点

• 全局统一：在整个服务体系下，多个服务或者进程，都统一调用这个自增器，来获取自增ID。
• 严格自增：避免竞争，写冲突造成写覆盖等，导致不严格自增

实现

根据上面要点，需要跨服务进程可以访问，且保障严格自增。综上考虑， 依赖MonogoDB来实现这个自增器，以下是代码实现，

代码

mongodb-conn.go

package main

import (
    "context"
    "os"
    "sync"

    mongo "go.mongodb.org/mongo-driver/mongo"
    "go.mongodb.org/mongo-driver/mongo/options"
)

var (
    singletonMongoClient *mongo.Client
    once                 sync.Once
)

var MONGODB_URI = os.Getenv("MONGODB_URI")
var MONGODB_DATABASE = os.Getenv("MONGODB_DATABASE")

func getSingletonMongoClient() *mongo.Client {

    once.Do(func() {
        // 创建连接到 MongoDB 的客户端
        uri := MONGODB_URI
        client, err := mongo.Connect(context.TODO(), options.Client().
            ApplyURI(uri))

        if err != nil {
            panic(err)
        }
        singletonMongoClient = client
    })
    return singletonMongoClient
}
func GetMongoConn() *mongo.Database {
    return getSingletonMongoClient().Database(MONGODB_DATABASE)
}

main.go

package main

import (
    "context"

    bson "go.mongodb.org/mongo-driver/bson"
    "go.mongodb.org/mongo-driver/mongo/options"
)

type UniqSeq struct {
    Key string `bson:"key"`
    Seq int64  `bson:"seq"`
}

func GetNextSeq(key string) (int64, error) {
    conn := GetMongoConn()
    var collectionName = "uniqseqz"
    collection := conn.Collection(collectionName)
    filter := bson.M{"key": key}
    update := bson.M{"$inc": bson.M{"seq": 1}}
    opts := options.FindOneAndUpdate().SetUpsert(true).SetReturnDocument(options.After)

    var result UniqSeq
    err := collection.FindOneAndUpdate(context.Background(), filter, update, opts).Decode(&result)

    if err != nil {
        return 0, err
    }
    return result.Seq, nil
}

func main() {
    // Test
    key := "T"
    seq, err := GetNextSeq(key)
    if err != nil {
        panic(err)
    }
    println(seq)
}

这里的核心就在下面三行

// 这里是借助了MongoDB $inc 和 upsert特性
// 先按key过滤，找出对应的document，然后upsert，没有就插入一条，有就加一。
// 以此来完成了自增
filter := bson.M{"key": key}
update := bson.M{"$inc": bson.M{"seq": 1}}
opts := options.FindOneAndUpdate().SetUpsert(true).SetReturnDocument(options.After)

关于竞争，

// 这个操作调用的时候，MongoDB Server端会加锁执行，保障操作的原子性，
// 对于调用方可以看以下一行是原子性的，不用担心写冲突或者写覆盖。
var result UniqSeq
    err := collection.FindOneAndUpdate(context.Background(), filter, update, opts).Decode(&result)

运行

最后可以带入运行一下，

MONGODB_URI="${YOUR_MONGODB_URL}" MONGODB_DATABASE="${YOUR_MONGODB_DATABASE}" go run mongodb-conn.go main.go