mirror of
https://github.com/redis/go-redis.git
synced 2025-09-10 07:11:50 +03:00
cb3af0800e
* feat(hitless): Introduce handlers for hitless upgrades This commit includes all the work on hitless upgrades with the addition of: - Pubsub Pool - Examples - Refactor of push - Refactor of pool (using atomics for most things) - Introducing of hooks in pool --------- Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>
48 lines
1.9 KiB
Go
48 lines
1.9 KiB
Go
package hitless
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/redis/go-redis/v9/internal"
|
|
"github.com/redis/go-redis/v9/internal/pool"
|
|
"github.com/redis/go-redis/v9/logging"
|
|
"github.com/redis/go-redis/v9/push"
|
|
)
|
|
|
|
// LoggingHook is an example hook implementation that logs all notifications.
|
|
type LoggingHook struct {
|
|
LogLevel logging.LogLevel
|
|
}
|
|
|
|
// PreHook logs the notification before processing and allows modification.
|
|
func (lh *LoggingHook) PreHook(ctx context.Context, notificationCtx push.NotificationHandlerContext, notificationType string, notification []interface{}) ([]interface{}, bool) {
|
|
if lh.LogLevel.InfoOrAbove() { // Info level
|
|
// Log the notification type and content
|
|
connID := uint64(0)
|
|
if conn, ok := notificationCtx.Conn.(*pool.Conn); ok {
|
|
connID = conn.GetID()
|
|
}
|
|
internal.Logger.Printf(ctx, "hitless: conn[%d] processing %s notification: %v", connID, notificationType, notification)
|
|
}
|
|
return notification, true // Continue processing with unmodified notification
|
|
}
|
|
|
|
// PostHook logs the result after processing.
|
|
func (lh *LoggingHook) PostHook(ctx context.Context, notificationCtx push.NotificationHandlerContext, notificationType string, notification []interface{}, result error) {
|
|
connID := uint64(0)
|
|
if conn, ok := notificationCtx.Conn.(*pool.Conn); ok {
|
|
connID = conn.GetID()
|
|
}
|
|
if result != nil && lh.LogLevel.WarnOrAbove() { // Warning level
|
|
internal.Logger.Printf(ctx, "hitless: conn[%d] %s notification processing failed: %v - %v", connID, notificationType, result, notification)
|
|
} else if lh.LogLevel.DebugOrAbove() { // Debug level
|
|
internal.Logger.Printf(ctx, "hitless: conn[%d] %s notification processed successfully", connID, notificationType)
|
|
}
|
|
}
|
|
|
|
// NewLoggingHook creates a new logging hook with the specified log level.
|
|
// Log levels: LogLevelError=errors, LogLevelWarn=warnings, LogLevelInfo=info, LogLevelDebug=debug
|
|
func NewLoggingHook(logLevel logging.LogLevel) *LoggingHook {
|
|
return &LoggingHook{LogLevel: logLevel}
|
|
}
|