go get github.com/redis/go-redis/v9

docker run -d -p 6379:6379 redis

package main

import (
    "context"
    "fmt"
    "github.com/redis/go-redis/v9"
    "log"
    "time"
)

var ctx = context.Background()
var rdb *redis.Client

func initRedis() {
    rdb = redis.NewClient(&redis.Options{
        Addr:     "localhost:6379",
        Password: "",
        DB:       0,
    })

    if _, err := rdb.Ping(ctx).Result(); err != nil {
        log.Fatal("Redis 連接失敗:", err)
    }
    log.Println("Redis 連接成功")
}

// CheckLoginLimit 檢查用戶是否超過登錄限制
func CheckLoginLimit(username string) bool {
    key := fmt.Sprintf("login:%s", username)
    limit := 5             // 最大嘗試次數(shù)
    blockTime := time.Minute // 超過限制后鎖定時間

    // 自增計數(shù)器
    count, err := rdb.Incr(ctx, key).Result()
    if err != nil {
        log.Println("Redis 錯誤:", err)
        return false
    }

    if count == 1 {
        // 第一次設(shè)置過期時間
        rdb.Expire(ctx, key, blockTime)
    }

    if count > int64(limit) {
        return false // 超過限制
    }

    return true
}

// MockLogin 模擬登錄
func MockLogin(username, password string) {
    if !CheckLoginLimit(username) {
        log.Printf("用戶 %s 登錄失敗: 嘗試次數(shù)過多，請稍后再試\n", username)
        return
    }

    // 假設(shè)正確密碼是 "123456"
    if password == "123456" {
        log.Printf("用戶 %s 登錄成功\n", username)
        // 成功后清除計數(shù)器
        rdb.Del(ctx, fmt.Sprintf("login:%s", username))
    } else {
        log.Printf("用戶 %s 登錄失敗: 密碼錯誤\n", username)
    }
}

func main() {
    initRedis()

    for i := 1; i <= 7; i++ {
        MockLogin("alice", "wrongpass")
    }
}