init

lib/redis.go

@@ -0,0 +1,69 @@
+package lib
+
+import (
+	"context"
+	"fmt"
+	"github.com/go-redis/redis/v8"
+	"sync/atomic"
+)
+
+var (
+	rdb     *redis.Client
+	ctx     = context.Background()
+	counter uint64
+)
+
+// 初始化 Redis 客户端
+func init() {
+	rdb = redis.NewClient(&redis.Options{
+		Addr:     "127.0.0.1:6379",
+		Password: "WKJXXmPchENCYK7x",
+		DB:       2,
+	})
+}
+
+func GetAllLinks() ([]string, error) {
+	// 使用 LRange 获取整个列表，范围为 0 到 -1（表示获取所有元素）
+	links, err := rdb.LRange(ctx, "links", 0, -1).Result()
+	if err != nil {
+		return nil, err
+	}
+	return links, nil
+}
+
+// 添加链接到 Redis List
+func AddLink(link string) error {
+	return rdb.LPush(ctx, "links", link).Err()
+}
+
+// 删除指定的链接
+func DeleteLink(link string) error {
+	// LREM count 参数为 0 表示删除所有匹配的元素
+	return rdb.LRem(ctx, "links", 0, link).Err()
+}
+
+// 修改指定索引位置的链接
+func UpdateLink(index int64, newLink string) error {
+	return rdb.LSet(ctx, "links", index, newLink).Err()
+}
+
+func GetLink() (string, error) {
+	length, err := rdb.LLen(ctx, "links").Result()
+	if err != nil || length == 0 {
+		return "", fmt.Errorf("no available links")
+	}
+	index := atomic.AddUint64(&counter, 1) % uint64(length)
+	return rdb.LIndex(ctx, "links", int64(index)).Result()
+}
+
+func ClearRedis() string {
+	// 清空当前数据库中的所有数据
+	err := rdb.FlushDB(ctx).Err()
+	if err != nil {
+		mes := fmt.Sprintf("无法清空链接: %v", err)
+		return mes
+	} else {
+
+		return "成功清空链接，请即使新增链接"
+	}
+}