feat: initial commit
All checks were successful
Cadoles/bouncer/pipeline/head This commit looks good
All checks were successful
Cadoles/bouncer/pipeline/head This commit looks good
This commit is contained in:
93
internal/store/redis/helper.go
Normal file
93
internal/store/redis/helper.go
Normal file
@ -0,0 +1,93 @@
|
||||
package redis
|
||||
|
||||
import (
|
||||
"context"
|
||||
"encoding/json"
|
||||
"strings"
|
||||
|
||||
"github.com/pkg/errors"
|
||||
"github.com/redis/go-redis/v9"
|
||||
)
|
||||
|
||||
type jsonWrapper[T any] struct {
|
||||
value T
|
||||
}
|
||||
|
||||
func (w *jsonWrapper[T]) MarshalBinary() ([]byte, error) {
|
||||
data, err := json.Marshal(w.value)
|
||||
if err != nil {
|
||||
return nil, errors.WithStack(err)
|
||||
}
|
||||
|
||||
return data, nil
|
||||
}
|
||||
|
||||
func (w *jsonWrapper[T]) UnmarshalBinary(data []byte) error {
|
||||
if err := json.Unmarshal(data, &w.value); err != nil {
|
||||
return errors.WithStack(err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (w *jsonWrapper[T]) UnmarshalText(data []byte) error {
|
||||
if err := json.Unmarshal(data, &w.value); err != nil {
|
||||
return errors.WithStack(err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (w *jsonWrapper[T]) Value() T {
|
||||
return w.value
|
||||
}
|
||||
|
||||
func wrap[T any](v T) *jsonWrapper[T] {
|
||||
return &jsonWrapper[T]{v}
|
||||
}
|
||||
|
||||
func unwrap[T any](v any) (T, error) {
|
||||
str, ok := v.(string)
|
||||
if !ok {
|
||||
return *new(T), errors.Errorf("could not unwrap value of type '%T'", v)
|
||||
}
|
||||
|
||||
u := new(T)
|
||||
|
||||
if err := json.Unmarshal([]byte(str), u); err != nil {
|
||||
return *new(T), errors.WithStack(err)
|
||||
}
|
||||
|
||||
return *u, nil
|
||||
}
|
||||
|
||||
func key(parts ...string) string {
|
||||
return strings.Join(parts, ":")
|
||||
}
|
||||
|
||||
func WithTx(ctx context.Context, client redis.UniversalClient, key string, fn func(ctx context.Context, tx *redis.Tx) error) error {
|
||||
txf := func(tx *redis.Tx) error {
|
||||
if err := fn(ctx, tx); err != nil {
|
||||
return errors.WithStack(err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
err := client.Watch(ctx, txf, key)
|
||||
if err != nil {
|
||||
return errors.WithStack(err)
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func contains[T ~string](values []T, v T) bool {
|
||||
for _, vv := range values {
|
||||
if vv == v {
|
||||
return true
|
||||
}
|
||||
}
|
||||
|
||||
return false
|
||||
}
|
Reference in New Issue
Block a user