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:53:03 +03:00
parent 7fc5a26347
commit e19c67ccc5
1 changed files with 14 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
event.go
View File

@ -37,8 +37,13 @@ func (h *eventHook) DialHook(hook goredis.DialHook) goredis.DialHook {
func (h *eventHook) ProcessHook(hook goredis.ProcessHook) goredis.ProcessHook { func (h *eventHook) ProcessHook(hook goredis.ProcessHook) goredis.ProcessHook {
return func(ctx context.Context, cmd goredis.Cmder) error { return func(ctx context.Context, cmd goredis.Cmder) error {
err := hook(ctx, cmd) err := hook(ctx, cmd)
if err != nil && !isRedisError(err) { if err != nil {
h.connected.Store(false) if !isRedisError(err) {
h.connected.Store(false)
}
h.connected.Store(true)
} else {
h.connected.Store(true)
} }
return err return err
} }
@ -47,8 +52,13 @@ func (h *eventHook) ProcessHook(hook goredis.ProcessHook) goredis.ProcessHook {
func (h *eventHook) ProcessPipelineHook(hook goredis.ProcessPipelineHook) goredis.ProcessPipelineHook { func (h *eventHook) ProcessPipelineHook(hook goredis.ProcessPipelineHook) goredis.ProcessPipelineHook {
return func(ctx context.Context, cmds []goredis.Cmder) error { return func(ctx context.Context, cmds []goredis.Cmder) error {
err := hook(ctx, cmds) err := hook(ctx, cmds)
if err != nil && !isRedisError(err) { if err != nil {
h.connected.Store(false) if !isRedisError(err) {
h.connected.Store(false)
}
h.connected.Store(true)
} else {
h.connected.Store(true)
} }
return err return err
} }