Compare commits

..

2 Commits

Author SHA1 Message Date
cbf00a0981 fixup set connected in hook
Some checks failed
codeql / analyze (go) (push) Failing after 48s
build / test (push) Failing after 4m57s
build / lint (push) Successful in 9m38s
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
2024-11-28 09:53:03 +03:00
7fc5a26347 fixup set connected in hook
Some checks failed
codeql / analyze (go) (push) Failing after 46s
build / test (push) Has been cancelled
build / lint (push) Has been cancelled
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
2024-11-28 09:49:29 +03:00

View File

@@ -22,8 +22,13 @@ func newEventHook(connected *atomic.Bool) *eventHook {
func (h *eventHook) DialHook(hook goredis.DialHook) goredis.DialHook {
return func(ctx context.Context, network, addr string) (net.Conn, error) {
conn, err := hook(ctx, network, addr)
if err != nil && !isRedisError(err) {
h.connected.Store(false)
if err != nil {
if !isRedisError(err) {
h.connected.Store(false)
}
h.connected.Store(true)
} else {
h.connected.Store(true)
}
return conn, err
}
@@ -32,8 +37,13 @@ func (h *eventHook) DialHook(hook goredis.DialHook) goredis.DialHook {
func (h *eventHook) ProcessHook(hook goredis.ProcessHook) goredis.ProcessHook {
return func(ctx context.Context, cmd goredis.Cmder) error {
err := hook(ctx, cmd)
if err != nil && !isRedisError(err) {
h.connected.Store(false)
if err != nil {
if !isRedisError(err) {
h.connected.Store(false)
}
h.connected.Store(true)
} else {
h.connected.Store(true)
}
return err
}
@@ -42,8 +52,13 @@ func (h *eventHook) ProcessHook(hook goredis.ProcessHook) goredis.ProcessHook {
func (h *eventHook) ProcessPipelineHook(hook goredis.ProcessPipelineHook) goredis.ProcessPipelineHook {
return func(ctx context.Context, cmds []goredis.Cmder) error {
err := hook(ctx, cmds)
if err != nil && !isRedisError(err) {
h.connected.Store(false)
if err != nil {
if !isRedisError(err) {
h.connected.Store(false)
}
h.connected.Store(true)
} else {
h.connected.Store(true)
}
return err
}