🧱 Introduce etcd for high availability (HA)
This commit is contained in:
@ -1,6 +1,10 @@
|
||||
package kv
|
||||
|
||||
import (
|
||||
"context"
|
||||
"fmt"
|
||||
"github.com/rs/zerolog/log"
|
||||
"github.com/samber/lo"
|
||||
clientv3 "go.etcd.io/etcd/client/v3"
|
||||
"time"
|
||||
)
|
||||
@ -12,8 +16,22 @@ func ConnectEtcd(endpoints []string) error {
|
||||
Endpoints: endpoints,
|
||||
DialTimeout: 10 * time.Second,
|
||||
})
|
||||
if err == nil {
|
||||
Kv = conn
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
var status []bool
|
||||
for _, endpoint := range endpoints {
|
||||
ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
|
||||
_, err := conn.Status(ctx, endpoint)
|
||||
if err != nil {
|
||||
log.Warn().Str("endpoint", endpoint).Err(err).Msg("An KV endpoint is not available...")
|
||||
}
|
||||
status = append(status, err == nil)
|
||||
cancel()
|
||||
}
|
||||
if len(lo.Filter(status, func(s bool, _ int) bool { return s })) == 0 {
|
||||
return fmt.Errorf("unable to connect to all KV endpoints")
|
||||
}
|
||||
Kv = conn
|
||||
return err
|
||||
}
|
||||
|
Reference in New Issue
Block a user