refactor: 将分布式锁从 redis 迁移至 utils 包

utils/utils.go

@@ -3,6 +3,7 @@ package utils
 import (
 	"context"
 	"encoding/json"
+	"errors"
 	"fmt"
 	"net"
 	"reflect"
@@ -13,10 +14,12 @@ import (
 	"time"
 
 	"gitea.com/red-future/common/beans"
+	"github.com/gogf/gf/v2/container/gvar"
 	"github.com/gogf/gf/v2/database/gredis"
 	"github.com/gogf/gf/v2/errors/gcode"
 	"github.com/gogf/gf/v2/errors/gerror"
 	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/os/glog"
 	"github.com/gogf/gf/v2/os/gtime"
 	"github.com/gogf/gf/v2/util/gconv"
 	"github.com/tiger1103/gfast-token/gftoken"
@@ -397,6 +400,7 @@ func GetFileAddressPrefix(ctx context.Context) (imageUrl string, err error) {
 	return
 }
 
+// GetBucketName 获取bucket名称
 func GetBucketName(ctx context.Context) (bucketName string, err error) {
 	user, err := GetUserInfo(ctx)
 	if err != nil {
@@ -405,3 +409,37 @@ func GetBucketName(ctx context.Context) (bucketName string, err error) {
 	bucketName = fmt.Sprintf("tenantid-%d", user.TenantId)
 	return
 }
+
+// Lock 分布式锁
+func Lock(ctx context.Context, key string, expireSeconds int64, fn func(ctx context.Context) error) (success bool, err error) {
+	limit := 3
+LOOP:
+	if limit < 0 {
+		return false, errors.New("锁重试次数耗尽")
+	}
+	limit--
+	var val *gvar.Var
+	if val, err = g.Redis().Set(ctx, key, true, gredis.SetOption{
+		TTLOption: gredis.TTLOption{
+			EX: &expireSeconds,
+		},
+		NX: true,
+	}); err != nil {
+		return false, err
+	}
+
+	if val.Bool() {
+		defer func(ctx context.Context, key string) {
+			if _, err = g.Redis().Del(ctx, key); err != nil {
+				glog.Errorf(ctx, "redis client Del error: %v", err)
+			}
+		}(ctx, key)
+		if err = fn(ctx); err != nil {
+			return false, err
+		}
+		return true, nil
+	}
+
+	time.Sleep(time.Second)
+	goto LOOP
+}