fusen/fusenapi
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'develop' of gitee.com:fusenpack/fusenapi into develop

Browse Source
This commit is contained in:
laodaming 2023-11-27 16:32:15 +08:00
commit af60136caf
1 changed files with 38 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
server/order/internal/logic/createorderlogic.go
View File

@ -6,6 +6,8 @@ import (
"fusenapi/service/repositories" "fusenapi/service/repositories"
"fusenapi/utils/auth" "fusenapi/utils/auth"
"fusenapi/utils/basic" "fusenapi/utils/basic"
"strconv"
"sync"
"time" "time"
"context" "context"
@ -34,6 +36,9 @@ func NewCreateOrderLogic(ctx context.Context, svcCtx *svc.ServiceContext) *Creat
// func (l *CreateOrderLogic) BeforeLogic(w http.ResponseWriter, r *http.Request) { // func (l *CreateOrderLogic) BeforeLogic(w http.ResponseWriter, r *http.Request) {
// } // }
var locks map[string]*sync.Mutex
var locksMutex sync.Mutex
func (l *CreateOrderLogic) CreateOrder(req *types.CreateOrderReq, userinfo *auth.UserInfo) (resp *basic.Response) { func (l *CreateOrderLogic) CreateOrder(req *types.CreateOrderReq, userinfo *auth.UserInfo) (resp *basic.Response) {
// 返回值必须调用Set重新返回, resp可以空指针调用 resp.SetStatus(basic.CodeOK, data) // 返回值必须调用Set重新返回, resp可以空指针调用 resp.SetStatus(basic.CodeOK, data)
// userinfo 传入值时, 一定不为null // userinfo 传入值时, 一定不为null
@ -41,6 +46,16 @@ func (l *CreateOrderLogic) CreateOrder(req *types.CreateOrderReq, userinfo *auth
// 如果是，返回未授权的错误码 // 如果是，返回未授权的错误码
return resp.SetStatus(basic.CodeUnAuth) return resp.SetStatus(basic.CodeUnAuth)
} }
var lockKey string
for _, v := range req.CartIds {
var vStr = strconv.Itoa(int(v))
lockKey = lockKey + "|" + vStr
}
// 分布式锁--防止重复下单
LockWith(lockKey) //获取锁
fmt.Println("获取到了锁")
tPlus60Days := time.Now().AddDate(0, 0, 60).UTC() tPlus60Days := time.Now().AddDate(0, 0, 60).UTC()
res, err := l.svcCtx.Repositories.NewOrder.Create(l.ctx, &repositories.CreateReq{ res, err := l.svcCtx.Repositories.NewOrder.Create(l.ctx, &repositories.CreateReq{
ExpectedDeliveryTime: tPlus60Days, ExpectedDeliveryTime: tPlus60Days,
@ -51,6 +66,8 @@ func (l *CreateOrderLogic) CreateOrder(req *types.CreateOrderReq, userinfo *auth
DeliveryMethod: constants.DELIVERYMETHODDSCLOUDSTORE, DeliveryMethod: constants.DELIVERYMETHODDSCLOUDSTORE,
}) })
UnlockWith(lockKey) //释放锁
if err != nil { if err != nil {
return resp.SetStatus(&res.ErrorCode) return resp.SetStatus(&res.ErrorCode)
} }
@ -76,6 +93,27 @@ func (l *CreateOrderLogic) CreateOrder(req *types.CreateOrderReq, userinfo *auth
}) })
} }
func LockWith(resource string) {
locksMutex.Lock()
lock, ok := locks[resource]
if !ok {
lock = &sync.Mutex{}
locks[resource] = lock
}
locksMutex.Unlock()
lock.Lock()
}
func UnlockWith(resource string) {
locksMutex.Lock()
lock, ok := locks[resource]
if ok {
lock.Unlock()
}
locksMutex.Unlock()
}
// 处理逻辑后 w,r 如:重定向, resp 必须重新处理 // 处理逻辑后 w,r 如:重定向, resp 必须重新处理
// func (l *CreateOrderLogic) AfterLogic(w http.ResponseWriter, r *http.Request, resp *basic.Response) { // func (l *CreateOrderLogic) AfterLogic(w http.ResponseWriter, r *http.Request, resp *basic.Response) {
// // httpx.OkJsonCtx(r.Context(), w, resp) // // httpx.OkJsonCtx(r.Context(), w, resp)