Go语言使用Redis和Etcd实现高性能分布式锁

package main
import (
    "context"
    "github.com/go-redis/redis/v8"
    "time"
)
var redisClient *redis.Client
func init() {
    redisClient = redis.NewClient(&redis.Options{
        Addr: "localhost:6379",
        Password: "",
        DB: 0,
    })
}
func acquireLock(lockKey string, expiration time.Duration) bool {
    ctx := context.Background()
    result, err := redisClient.SetNX(ctx, lockKey, "locked", expiration).Result()
    if err != nil {
        panic(err)
    }
    return result
}
func releaseLock(lockKey string) {
    ctx := context.Background()
    redisClient.Del(ctx, lockKey)
}
func main() {
    lockKey := "my_lock_key"
    locked := acquireLock(lockKey, 10*time.Second)
    if locked {
        defer releaseLock(lockKey)
        // 执行业务逻辑
    }
}

package main
import (
    "context"
    "go.etcd.io/etcd/client/v3"
    "go.etcd.io/etcd/client/v3/concurrency"
    "log"
    "time"
)
func main() {
    cli, err := clientv3.New(clientv3.Config{
        Endpoints:   []string{"localhost:2379"},
        DialTimeout: 5 * time.Second,
    })
    if err != nil {
        log.Fatal(err)
    }
    defer cli.Close()
    session, err := concurrency.NewSession(cli)
    if err != nil {
        log.Fatal(err)
    }
    defer session.Close()
    mutex := concurrency.NewMutex(session, "/my-lock/")
    if err := mutex.Lock(context.Background()); err != nil {
        log.Fatal(err)
    }
    log.Println("acquired lock")
    // 执行业务逻辑
    if err := mutex.Unlock(context.Background()); err != nil {
        log.Fatal(err)
    }
    log.Println("released lock")
}