package mq

import (
	"github.com/go-redis/redis"
	"wss-pool/logging/applogger"
)

var Rdb *redis.Client

const (
	PublishKey = "websocket"
)

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

// Publish 发布消息到redis
// channel是发布的目标信道
// payload是要发布的消息内容
func Publish(channel string, payload string) error {
	var err error
	applogger.Debug("[Redis] publish [%s]: %s", channel, payload)
	err = Rdb.Publish(channel, payload).Err()
	if err != nil {
		applogger.Error("[Redis] pulish error: %s", err.Error())
		return err
	}
	return err
}

// Subscribe 订阅redis消息
// channel是订阅的目标信道
func Subscribe(channel string) (string, error) {
	applogger.Info("[Redis] subscribe [%s]", channel)
	sub := Rdb.Subscribe(channel)
	msg, err := sub.ReceiveMessage()
	if err != nil {
		applogger.Error("[Redis] subscribe [%s]", channel)
		return "", err
	}
	applogger.Debug("[Redis] subscribe [%s]: %s", channel, msg.String())
	return msg.Payload, err
}