package redis

import "context"

const SCAN_COUNT = 10000

func (c *cacheRepo) GetScanKeys(pattern string) (result []string, err error) {
	result = make([]string, 0)
	ctx := context.Background()
	var cursor uint64
	var keys []string
	for {
		if keys, cursor, err = c.client.Scan(ctx, cursor, pattern, SCAN_COUNT).Result(); err != nil {
			break
		}
		if len(keys) > 0 {
			result = append(result, keys...)
		}
		if cursor == 0 {
			break
		}
	}

	return
}

func (c *cacheRepo) GetSScanKeys(key string, pattern string) (result []string, err error) {
	result = make([]string, 0)
	ctx := context.Background()
	var cursor uint64
	var keys []string
	for {
		if keys, cursor, err = c.client.SScan(ctx, key, cursor, pattern, SCAN_COUNT).Result(); err != nil {
			break
		}
		if len(keys) > 0 {
			result = append(result, keys...)
		}
		if cursor == 0 {
			break
		}
	}

	return
}

func (c *cacheRepo) GetHScanKeys(key string, pattern string) (result []string, err error) {
	result = make([]string, 0)
	ctx := context.Background()
	var cursor uint64
	var keys []string
	for {
		if keys, cursor, err = c.client.HScan(ctx, key, cursor, pattern, SCAN_COUNT).Result(); err != nil {
			break
		}
		if len(keys) > 0 {
			result = append(result, keys...)
		}
		if cursor == 0 {
			break
		}
	}

	return
}