[VOL-5486] Fix deprecated versions
Change-Id: If0b888d6c2f33b2f415c8b03b08dc994bb3df3f4
Signed-off-by: Abhay Kumar <abhay.kumar@radisys.com>
diff --git a/vendor/go.etcd.io/etcd/client/v3/snapshot/doc.go b/vendor/go.etcd.io/etcd/client/v3/snapshot/doc.go
new file mode 100644
index 0000000..1c761be
--- /dev/null
+++ b/vendor/go.etcd.io/etcd/client/v3/snapshot/doc.go
@@ -0,0 +1,16 @@
+// Copyright 2018 The etcd Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+// Package snapshot implements utilities around etcd snapshot.
+package snapshot
diff --git a/vendor/go.etcd.io/etcd/client/v3/snapshot/v3_snapshot.go b/vendor/go.etcd.io/etcd/client/v3/snapshot/v3_snapshot.go
new file mode 100644
index 0000000..5f263ea
--- /dev/null
+++ b/vendor/go.etcd.io/etcd/client/v3/snapshot/v3_snapshot.go
@@ -0,0 +1,123 @@
+// Copyright 2018 The etcd Authors
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+// http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package snapshot
+
+import (
+ "context"
+ "crypto/sha256"
+ "errors"
+ "fmt"
+ "io"
+ "os"
+ "time"
+
+ "github.com/dustin/go-humanize"
+ "go.uber.org/zap"
+
+ "go.etcd.io/etcd/client/pkg/v3/fileutil"
+ clientv3 "go.etcd.io/etcd/client/v3"
+)
+
+// hasChecksum returns "true" if the file size "n"
+// has appended sha256 hash digest.
+func hasChecksum(n int64) bool {
+ // 512 is chosen because it's a minimum disk sector size
+ // smaller than (and multiplies to) OS page size in most systems
+ return (n % 512) == sha256.Size
+}
+
+// SaveWithVersion fetches snapshot from remote etcd server, saves data
+// to target path and returns server version. If the context "ctx" is canceled or timed out,
+// snapshot save stream will error out (e.g. context.Canceled,
+// context.DeadlineExceeded). Make sure to specify only one endpoint
+// in client configuration. Snapshot API must be requested to a
+// selected node, and saved snapshot is the point-in-time state of
+// the selected node.
+// Etcd <v3.6 will return "" as version.
+func SaveWithVersion(ctx context.Context, lg *zap.Logger, cfg clientv3.Config, dbPath string) (string, error) {
+ cfg.Logger = lg.Named("client")
+ if len(cfg.Endpoints) != 1 {
+ return "", fmt.Errorf("snapshot must be requested to one selected node, not multiple %v", cfg.Endpoints)
+ }
+ cli, err := clientv3.New(cfg)
+ if err != nil {
+ return "", err
+ }
+ defer func() {
+ err = cli.Close()
+ if err != nil {
+ lg.Error("Failed to close client", zap.Error(err))
+ }
+ }()
+
+ partpath := dbPath + ".part"
+ defer func() {
+ err = os.RemoveAll(partpath)
+ if err != nil {
+ lg.Error("Failed to cleanup .part file", zap.Error(err))
+ }
+ }()
+
+ f, err := os.OpenFile(partpath, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, fileutil.PrivateFileMode)
+ if err != nil {
+ return "", fmt.Errorf("could not open %s (%w)", partpath, err)
+ }
+ defer func() {
+ err = f.Close()
+ if err != nil && !errors.Is(err, os.ErrClosed) {
+ lg.Error("Could not close file descriptor", zap.Error(err))
+ }
+ }()
+ lg.Info("created temporary db file", zap.String("path", partpath))
+
+ start := time.Now()
+ resp, err := cli.SnapshotWithVersion(ctx)
+ if err != nil {
+ return "", err
+ }
+ defer func() {
+ err = resp.Snapshot.Close()
+ if err != nil {
+ lg.Error("Could not close snapshot stream", zap.Error(err))
+ }
+ }()
+ lg.Info("fetching snapshot", zap.String("endpoint", cfg.Endpoints[0]))
+ var size int64
+ size, err = io.Copy(f, resp.Snapshot)
+ if err != nil {
+ return resp.Version, fmt.Errorf("could not write snapshot: %w", err)
+ }
+ if !hasChecksum(size) {
+ return resp.Version, fmt.Errorf("sha256 checksum not found [bytes: %d]", size)
+ }
+ if err = fileutil.Fsync(f); err != nil {
+ return resp.Version, fmt.Errorf("could not fsync snapshot: %w", err)
+ }
+ if err = f.Close(); err != nil {
+ return resp.Version, fmt.Errorf("could not close file descriptor: %w", err)
+ }
+ lg.Info("fetched snapshot",
+ zap.String("endpoint", cfg.Endpoints[0]),
+ zap.String("size", humanize.Bytes(uint64(size))),
+ zap.Duration("took", time.Since(start)),
+ zap.String("etcd-version", resp.Version),
+ )
+
+ if err = os.Rename(partpath, dbPath); err != nil {
+ return resp.Version, fmt.Errorf("could not rename %s to %s (%w)", partpath, dbPath, err)
+ }
+ lg.Info("saved", zap.String("path", dbPath))
+ return resp.Version, nil
+}