zlei6/go-zero
1
0
Fork 0
mirror of https://github.com/zeromicro/go-zero.git synced 2026-05-14 18:30:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

chore: coding style (#4082)

Browse Source
This commit is contained in:
Kevin Wan
2024-04-17 23:37:35 +08:00
committed by GitHub
parent 62c88a84d1
commit e9dc96af17
10 changed files with 24 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
core/limit/periodlimit.go
View File

@@ -30,9 +30,8 @@ var (
ErrUnknownCode = errors.New("unknown status code")
//go:embed periodscript.lua
periodScript string
scriptPeriod = redis.NewScript(periodScript)
periodLuaScript string
periodScript = redis.NewScript(periodLuaScript)
)
type (
@@ -73,7 +72,7 @@ func (h *PeriodLimit) Take(key string) (int, error) {
// TakeCtx requests a permit with context, it returns the permit state.
func (h *PeriodLimit) TakeCtx(ctx context.Context, key string) (int, error) {
resp, err := h.limitStore.ScriptRunCtx(ctx, scriptPeriod, []string{h.keyPrefix + key}, []string{
resp, err := h.limitStore.ScriptRunCtx(ctx, periodScript, []string{h.keyPrefix + key}, []string{
strconv.Itoa(h.quota),
strconv.Itoa(h.calcExpireSeconds()),
})