package main

import (
	"context"
	"fmt"
	"log"

	"github.com/redis/go-redis/v9"
)

func main() {
	// 创建 Redis 客户端
	client := redis.NewClient(&redis.Options{
		Addr: "localhost:6379",
	})
	ctx := context.Background()

	// 删除已有的布隆过滤器（如果存在）
	fmt.Println("Deleting existing Bloom Filter...")
	_, err := client.Del(ctx, "my_filter").Result()
	if err != nil {
		log.Fatalf("Error deleting filter: %v", err)
	}

	// 创建布隆过滤器
	fmt.Println("Creating Bloom Filter...")
	_, err = client.BFReserve(ctx, "my_filter", 0.01, 10000).Result()
	if err != nil {
		log.Fatalf("Error creating filter: %v", err)
	}

	// 添加元素
	fmt.Println("Adding elements to Bloom Filter...")
	elements := []string{"element1", "element2", "element3"}
	for _, el := range elements {
		_, err := client.BFAdd(ctx, "my_filter", el).Result()
		if err != nil {
			log.Fatalf("Error adding element %s: %v", el, err)
		}
	}

	// 检查元素是否存在
	fmt.Println("Checking elements existence...")
	for _, el := range elements {
		exists, err := client.BFExists(ctx, "my_filter", el).Result()
		if err != nil {
			log.Fatalf("Error checking element %s: %v", el, err)
		}
		fmt.Printf("Element %s exists: %v\n", el, exists)
	}

	// 获取布隆过滤器的信息
	fmt.Println("Getting Bloom Filter information...")
	info, err := client.BFInfo(ctx, "my_filter").Result()
	if err != nil {
		log.Fatalf("Error getting filter info: %v", err)
	}
	fmt.Printf("Filter Info: %v\n", info)

	// 获取布隆过滤器的特定信息
	fmt.Println("Getting Bloom Filter specific information...")
	capacity, err := client.BFInfoCapacity(ctx, "my_filter").Result()
	if err != nil {
		log.Fatalf("Error getting filter capacity: %v", err)
	}
	size, err := client.BFInfoSize(ctx, "my_filter").Result()
	if err != nil {
		log.Fatalf("Error getting filter size: %v", err)
	}
	filters, err := client.BFInfoFilters(ctx, "my_filter").Result()
	if err != nil {
		log.Fatalf("Error getting filter filters: %v", err)
	}
	items, err := client.BFInfoItems(ctx, "my_filter").Result()
	if err != nil {
		log.Fatalf("Error getting filter items: %v", err)
	}
	fmt.Printf("Filter Capacity: %d\n", capacity)
	fmt.Printf("Filter Size: %d\n", size)
	fmt.Printf("Filter Filters: %v\n", filters)
	fmt.Printf("Filter Items: %d\n", items)

	// 获取布隆过滤器的扩展信息
	fmt.Println("Getting Bloom Filter expansion information...")
	expansion, err := client.BFInfoExpansion(ctx, "my_filter").Result()
	if err != nil {
		log.Fatalf("Error getting filter expansion: %v", err)
	}
	fmt.Printf("Filter Expansion: %d\n", expansion)

	// 批量插入元素
	fmt.Println("Batch inserting elements...")
	insertOptions := &redis.BFInsertOptions{}
	insertResult, err := client.BFInsert(ctx, "my_filter", insertOptions, "element4", "element5").Result()
	if err != nil {
		log.Fatalf("Error batch inserting elements: %v", err)
	}
	fmt.Printf("Insert Results: %v\n", insertResult)

	// 批量检查元素
	fmt.Println("Batch checking elements...")
	existResults, err := client.BFMExists(ctx, "my_filter", "element1", "element4").Result()
	if err != nil {
		log.Fatalf("Error batch checking elements: %v", err)
	}
	fmt.Printf("Batch Exist Results: %v\n", existResults)

	// 返回布隆过滤器的基数，即：布隆过滤器中添加的项目数量
	fmt.Println("Getting Bloom Filter card...")
	card, err := client.BFCard(ctx, "my_filter").Result()
	if err != nil {
		log.Fatalf("Error getting filter card: %v", err)
	}
	fmt.Printf("Filter Card: %d\n", card)

	// 结束
	fmt.Println("All operations completed.")
}

Deleting existing Bloom Filter...
Creating Bloom Filter...
Adding elements to Bloom Filter...
Checking elements existence...
Element element1 exists: true
Element element2 exists: true
Element element3 exists: true
Getting Bloom Filter information...
Filter Info: {10000 13888 1 3 2}
Getting Bloom Filter specific information...
Filter Capacity: {10000 0 0 0 0}
Filter Size: {0 13888 0 0 0}
Filter Filters: {0 0 1 0 0}
Filter Items: {0 0 0 3 0}
Getting Bloom Filter expansion information...
Filter Expansion: {0 0 0 0 2}
Batch inserting elements...
Insert Results: [true true]
Batch checking elements...
Batch Exist Results: [true true]
Getting Bloom Filter card...
Filter Card: 5
All operations completed.