unistack-org/micro-store-redis
4
0
Fork 2
Code Issues Pull Requests Actions Packages Projects Releases 39 Wiki Activity

fixup set connected in hook

Browse Source 
Signed-off-by: Vasiliy Tolstov <v.tolstov@unistack.org>
This commit is contained in:
Василий Толстов 2024-11-28 09:49:29 +03:00
parent 24f9f41de4
commit 27bd074a2f
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
event.go
View File

@ -22,9 +22,14 @@ func newEventHook(connected *atomic.Bool) *eventHook {
func (h *eventHook) DialHook(hook goredis.DialHook) goredis.DialHook { func (h *eventHook) DialHook(hook goredis.DialHook) goredis.DialHook {
return func(ctx context.Context, network, addr string) (net.Conn, error) { return func(ctx context.Context, network, addr string) (net.Conn, error) {
conn, err := hook(ctx, network, addr) conn, err := hook(ctx, network, addr)
if err != nil && !isRedisError(err) { if err != nil {
if !isRedisError(err) {
h.connected.Store(false) h.connected.Store(false)
} }
h.connected.Store(true)
} else {
h.connected.Store(true)
}
return conn, err return conn, err
} }
} }