mirror of
https://github.com/tiennm99/keepalive.git
synced 2026-10-13 18:15:42 +00:00
tiennm99
e3b7ea5011
Consolidates the *-keepalive family (redis, valkey, postgresql, mysql, mongodb, couchbase) into a single binary with a runtime DB_TYPE dispatch and an Adapter interface. Original repos archived with a pointer here.
44 lines
820 B
Go
44 lines
820 B
Go
package adapter
|
|
|
|
import (
|
|
"context"
|
|
|
|
"github.com/valkey-io/valkey-go"
|
|
)
|
|
|
|
func init() {
|
|
Registry["valkey"] = func() (Adapter, error) { return &valkeyAdapter{}, nil }
|
|
}
|
|
|
|
type valkeyAdapter struct {
|
|
client valkey.Client
|
|
key string
|
|
}
|
|
|
|
func (a *valkeyAdapter) Connect(_ context.Context) error {
|
|
url, err := envOrFail("VALKEY_URL")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
opt, err := valkey.ParseURL(url)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
client, err := valkey.NewClient(opt)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
a.client = client
|
|
a.key = envOr("COUNTER_KEY", "counter")
|
|
return nil
|
|
}
|
|
|
|
func (a *valkeyAdapter) Increment(ctx context.Context) (int64, error) {
|
|
return a.client.Do(ctx, a.client.B().Incr().Key(a.key).Build()).AsInt64()
|
|
}
|
|
|
|
func (a *valkeyAdapter) Close(_ context.Context) error {
|
|
a.client.Close()
|
|
return nil
|
|
}
|