1 Star 0 Fork 2

who7708/etcd

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
.github
Documentation
alarm
auth
client
clientv3
cmd
compactor
contrib
discovery
e2e
error
etcdctl
etcdmain
etcdserver
hack
integration
lease
logos
mvcc
pkg
proxy
raft
rafthttp
scripts
snap
snappb
db.go
message.go
metrics.go
snapshotter.go
snapshotter_test.go
store
tools
version
wal
.dockerignore
.gitignore
.godir
.header
.travis.yml
CONTRIBUTING.md
DCO
Dockerfile
Dockerfile-release
LICENSE
MAINTAINERS
NOTICE
Procfile
README.md
ROADMAP.md
V2Procfile
build
build.bat
build.ps1
cover
etcd.conf.yml.sample
main.go
test
克隆/下载
db.go 1.84 KB
一键复制 编辑 原始数据 按行查看 历史
Gyu-Ho Lee 提交于 9年前 . *: update LICENSE header
// Copyright 2015 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 snap
import (
"fmt"
"io"
"io/ioutil"
"os"
"path"
"github.com/coreos/etcd/pkg/fileutil"
)
// SaveDBFrom saves snapshot of the database from the given reader. It
// guarantees the save operation is atomic.
func (s *Snapshotter) SaveDBFrom(r io.Reader, id uint64) (int64, error) {
f, err := ioutil.TempFile(s.dir, "tmp")
if err != nil {
return 0, err
}
var n int64
n, err = io.Copy(f, r)
if err == nil {
err = fileutil.Fsync(f)
}
f.Close()
if err != nil {
os.Remove(f.Name())
return n, err
}
fn := path.Join(s.dir, fmt.Sprintf("%016x.snap.db", id))
if fileutil.Exist(fn) {
os.Remove(f.Name())
return n, nil
}
err = os.Rename(f.Name(), fn)
if err != nil {
os.Remove(f.Name())
return n, err
}
plog.Infof("saved database snapshot to disk [total bytes: %d]", n)
return n, nil
}
// DBFilePath returns the file path for the snapshot of the database with
// given id. If the snapshot does not exist, it returns error.
func (s *Snapshotter) DBFilePath(id uint64) (string, error) {
fns, err := fileutil.ReadDir(s.dir)
if err != nil {
return "", err
}
wfn := fmt.Sprintf("%016x.snap.db", id)
for _, fn := range fns {
if fn == wfn {
return path.Join(s.dir, fn), nil
}
}
return "", fmt.Errorf("snap: snapshot file doesn't exist")
}
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/who7708/etcd.git
git@gitee.com:who7708/etcd.git
who7708
etcd
etcd
v3.0.10

搜索帮助