package gojobs

import (
	"context"
	"fmt"
	"github.com/go-redis/redis/v9"
	"go.dtapp.net/gotime"
	"testing"
)
import "go.dtapp.net/dorm"

// 下发
func TestRedisPublish(t *testing.T) {
	client, err := dorm.NewRedisClient(&dorm.ConfigRedisClient{
		Addr:     "119.29.14.159:6379",
		Password: "980202",
		DB:       5,
	})
	if err != nil {
		t.Error(err)
	}

	publish, err := client.Db.Publish(context.Background(), "ch_user_1", "测试").Result()
	t.Log(publish)
	t.Log(err)
}

// 订阅
func TestRedisPSubscribe(t *testing.T) {
	client, err := dorm.NewRedisClient(&dorm.ConfigRedisClient{
		Addr:     "119.29.14.159:6379",
		Password: "980202",
		DB:       5,
	})
	if err != nil {
		t.Error(err)
	}

	// 订阅channel1这个channel
	sub := client.Db.PSubscribe(context.Background(), "ch_user_*")
	t.Log(sub)
	// sub.Channel() 返回go channel，可以循环读取redis服务器发过来的消息
	for msg := range sub.Channel() {
		// 打印收到的消息
		t.Log(msg.Channel)
		t.Log(msg.Payload)
	}
}

// 添加一个或者多个元素到集合，如果元素已经存在则更新分数
func TestRedisZAdd(t *testing.T) {
	client, err := dorm.NewRedisClient(&dorm.ConfigRedisClient{
		Addr:     "119.29.14.159:6379",
		Password: "980202",
		DB:       5,
	})
	if err != nil {
		t.Error(err)
	}

	// 删除小于当前时间的客户端
	client.Db.ZRemRangeByScore(context.Background(), "cron_jobs", "0", fmt.Sprintf("(%v", gotime.Current().Timestamp())).Result()

	// 添加客户端或者更新客户端
	publish, err := client.Db.ZAdd(context.Background(), "cron_jobs", redis.Z{
		Score:  float64(gotime.Current().AfterMinute(10).Timestamp()),
		Member: "127.0.0.1",
	}).Result()
	t.Log(publish)
	t.Log(err)
}

// 返回集合中某个索引范围的元素，根据分数从小到大排序
func TestRedisZRange(t *testing.T) {
	client, err := dorm.NewRedisClient(&dorm.ConfigRedisClient{
		Addr:     "119.29.14.159:6379",
		Password: "980202",
		DB:       5,
	})
	if err != nil {
		t.Error(err)
	}

	vals, err := client.Db.ZRange(context.Background(), "cron_jobs", 0, -1).Result()
	t.Log(err)

	for _, val := range vals {
		t.Log(val)
	}

}

func TestRedisKeys(t *testing.T) {
	client, err := dorm.NewRedisClient(&dorm.ConfigRedisClient{
		Addr:     "119.29.14.159:6379",
		Password: "980202",
		DB:       5,
	})
	if err != nil {
		t.Error(err)
	}

	client.Db.Set(context.Background(), "cron_jobs_client_127.0.0.1", "127.0.0.1", 0)
	client.Db.Set(context.Background(), "cron_jobs_client_43.135.79.235", "43.135.79.235", 0)

	result, err := client.Keys(context.Background(), "cron_jobs_client_*").Result()
	t.Log(result)
	t.Log(err)
	for _, val := range result {
		t.Log(val)
	}
}