67 lines
1.4 KiB
Go
67 lines
1.4 KiB
Go
package lib
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/joho/godotenv"
|
|
"github.com/redis/go-redis/v9"
|
|
)
|
|
|
|
var RedisClient *redis.Client
|
|
|
|
type RedisPubSubMessage struct {
|
|
pubsub *redis.PubSub
|
|
}
|
|
|
|
// RedisPubSub is a Redis implementation of the PubSub interface.
|
|
type RedisPubSub struct {
|
|
Client *redis.Client
|
|
}
|
|
|
|
func NewRedisClient() *redis.Client {
|
|
if RedisClient != nil {
|
|
return RedisClient
|
|
}
|
|
|
|
godotenv.Load(".env")
|
|
redis_host := os.Getenv("REDIS_HOST")
|
|
redis_password := os.Getenv("REDIS_PASSWORD")
|
|
|
|
log.Printf("Connecting to Redis at %s", redis_host)
|
|
RedisClient = redis.NewClient(&redis.Options{
|
|
Addr: redis_host,
|
|
Password: redis_password,
|
|
DB: 0,
|
|
})
|
|
|
|
return RedisClient
|
|
}
|
|
|
|
func (m *RedisPubSubMessage) ReceiveMessage(ctx context.Context) (*Message, error) {
|
|
msg, err := m.pubsub.ReceiveMessage(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &Message{Payload: msg.Payload}, nil
|
|
}
|
|
|
|
func (ps *RedisPubSub) SubscribeToChannel(channel string) (PubSubMessage, error) {
|
|
pubsub := ps.Client.Subscribe(context.Background(), channel)
|
|
_, err := pubsub.Receive(context.Background())
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return &RedisPubSubMessage{pubsub: pubsub}, nil
|
|
}
|
|
|
|
func (r *RedisPubSub) PublishToChannel(channel string, message string) error {
|
|
err := r.Client.Publish(context.Background(), channel, message).Err()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|