nurse/protocols/redis/ping.go

89 lines
1.8 KiB
Go
Raw Normal View History

2022-04-28 16:35:02 +00:00
package redis
import (
"context"
"fmt"
"github.com/redis/go-redis/v9"
2022-04-28 16:35:02 +00:00
2022-09-22 09:46:36 +00:00
"code.icb4dc0.de/prskr/nurse/check"
"code.icb4dc0.de/prskr/nurse/config"
"code.icb4dc0.de/prskr/nurse/grammar"
"code.icb4dc0.de/prskr/nurse/validation"
2022-04-28 16:35:02 +00:00
)
var _ check.SystemChecker = (*PingCheck)(nil)
2022-04-28 16:35:02 +00:00
type PingCheck struct {
redis.UniversalClient
2022-06-09 20:12:45 +00:00
validators validation.Validator[redis.Cmder]
2022-04-28 16:35:02 +00:00
Message string
}
func (p PingCheck) Execute(ctx check.Context) error {
2022-04-28 16:35:02 +00:00
if p.Message == "" {
return p.Ping(ctx).Err()
}
for {
select {
case <-ctx.Done():
return ctx.Err()
default:
attemptCtx, cancel := ctx.AttemptContext()
err := p.executeAttempt(attemptCtx)
cancel()
if err == nil {
return nil
}
}
}
}
func (p PingCheck) executeAttempt(ctx context.Context) error {
2022-04-28 16:35:02 +00:00
if resp, err := p.Do(ctx, "PING", p.Message).Text(); err != nil {
return err
} else if resp != p.Message {
return fmt.Errorf("expected value %s got %s", p.Message, resp)
}
return nil
}
func (p *PingCheck) UnmarshalCheck(c grammar.Check, lookup config.ServerLookup) error {
2022-04-28 16:35:02 +00:00
const (
serverOnlyArgCount = 1
serverAndMessageArgCount = 2
)
val, _ := GenericCommandValidatorFor("PONG")
2022-06-09 20:12:45 +00:00
validators := validation.Chain[redis.Cmder]{}
validators = append(validators, val)
p.validators = validators
2022-04-28 16:35:02 +00:00
init := c.Initiator
switch len(init.Params) {
case 0:
return grammar.ErrMissingServer
case serverAndMessageArgCount:
if msg, err := init.Params[1].AsString(); err != nil {
return err
} else {
val, _ := GenericCommandValidatorFor(msg)
2022-06-09 20:12:45 +00:00
p.validators = validation.Chain[redis.Cmder]{val}
2022-04-28 16:35:02 +00:00
}
fallthrough
case serverOnlyArgCount:
if cli, err := clientFromParam(init.Params[0], lookup); err != nil {
2022-04-28 16:35:02 +00:00
return err
} else {
p.UniversalClient = cli
}
return nil
default:
return grammar.ErrAmbiguousParamCount
}
}