70 lines
1.5 KiB
Go
70 lines
1.5 KiB
Go
package cache
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/redis/go-redis/v9"
|
|
|
|
"accounting-app/internal/config"
|
|
)
|
|
|
|
// RedisClient wraps the Redis client with additional functionality
|
|
type RedisClient struct {
|
|
client *redis.Client
|
|
cfg *config.Config
|
|
}
|
|
|
|
// NewRedisClient creates a new Redis client from the configuration
|
|
func NewRedisClient(cfg *config.Config) (*RedisClient, error) {
|
|
client := redis.NewClient(&redis.Options{
|
|
Addr: cfg.RedisAddr,
|
|
Password: cfg.RedisPassword,
|
|
DB: cfg.RedisDB,
|
|
})
|
|
|
|
rc := &RedisClient{
|
|
client: client,
|
|
cfg: cfg,
|
|
}
|
|
|
|
// Test the connection
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
|
|
if err := rc.Ping(ctx); err != nil {
|
|
return nil, fmt.Errorf("failed to connect to Redis: %w", err)
|
|
}
|
|
|
|
return rc, nil
|
|
}
|
|
|
|
// Ping checks if the Redis connection is healthy
|
|
func (rc *RedisClient) Ping(ctx context.Context) error {
|
|
_, err := rc.client.Ping(ctx).Result()
|
|
return err
|
|
}
|
|
|
|
// HealthCheck performs a health check on the Redis connection
|
|
func (rc *RedisClient) HealthCheck() error {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 3*time.Second)
|
|
defer cancel()
|
|
return rc.Ping(ctx)
|
|
}
|
|
|
|
// Close closes the Redis connection
|
|
func (rc *RedisClient) Close() error {
|
|
return rc.client.Close()
|
|
}
|
|
|
|
// Client returns the underlying Redis client for direct access
|
|
func (rc *RedisClient) Client() *redis.Client {
|
|
return rc.client
|
|
}
|
|
|
|
// GetConfig returns the configuration used by this client
|
|
func (rc *RedisClient) GetConfig() *config.Config {
|
|
return rc.cfg
|
|
}
|