docker/buildx
1
0
Fork 0
mirror of https://gitea.com/Lydanne/buildx.git synced 2025-05-18 00:47:48 +08:00
Code Issues Packages Projects Releases 1 Wiki Activity

Use ioutils.AtomicWriteFile from moby

Browse Source 
Signed-off-by: CrazyMax <crazy-max@users.noreply.github.com>
This commit is contained in:
CrazyMax 2021-06-30 08:34:45 +02:00
parent 9215fc56a3
commit ef95f8135b
No known key found for this signature in database
GPG Key ID: 3248E46B6BB8C7F7
1 changed files with 5 additions and 34 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

39
store/store.go
View File

@ -2,13 +2,13 @@ package store
import ( import (
"encoding/json" "encoding/json"
"io"
"io/ioutil" "io/ioutil"
"os" "os"
"path/filepath" "path/filepath"
"regexp" "regexp"
"sort" "sort"
"github.com/docker/docker/pkg/ioutils"
"github.com/gofrs/flock" "github.com/gofrs/flock"
"github.com/opencontainers/go-digest" "github.com/opencontainers/go-digest"
"github.com/pkg/errors" "github.com/pkg/errors"
@ -95,7 +95,7 @@ func (t *Txn) Save(ng *NodeGroup) error {
if err != nil { if err != nil {
return err return err
} }
return atomicWriteFile(filepath.Join(t.s.root, "instances", name), dt, 0600) return ioutils.AtomicWriteFile(filepath.Join(t.s.root, "instances", name), dt, 0600)
} }
func (t *Txn) Remove(name string) error { func (t *Txn) Remove(name string) error {
@ -116,14 +116,14 @@ func (t *Txn) SetCurrent(key, name string, global, def bool) error {
if err != nil { if err != nil {
return err return err
} }
if err := atomicWriteFile(filepath.Join(t.s.root, "current"), dt, 0600); err != nil { if err := ioutils.AtomicWriteFile(filepath.Join(t.s.root, "current"), dt, 0600); err != nil {
return err return err
} }
h := toHash(key) h := toHash(key)
if def { if def {
if err := atomicWriteFile(filepath.Join(t.s.root, "defaults", h), []byte(name), 0600); err != nil { if err := ioutils.AtomicWriteFile(filepath.Join(t.s.root, "defaults", h), []byte(name), 0600); err != nil {
return err return err
} }
} else { } else {
@ -137,7 +137,7 @@ func (t *Txn) reset(key string) error {
if err != nil { if err != nil {
return err return err
} }
if err := atomicWriteFile(filepath.Join(t.s.root, "current"), dt, 0600); err != nil { if err := ioutils.AtomicWriteFile(filepath.Join(t.s.root, "current"), dt, 0600); err != nil {
return err return err
} }
return nil return nil
@ -202,35 +202,6 @@ type current struct {
var namePattern = regexp.MustCompile(`^[a-zA-Z][a-zA-Z0-9\.\-_]*$`) var namePattern = regexp.MustCompile(`^[a-zA-Z][a-zA-Z0-9\.\-_]*$`)
func atomicWriteFile(filename string, data []byte, perm os.FileMode) error {
f, err := ioutil.TempFile(filepath.Dir(filename), ".tmp-"+filepath.Base(filename))
if err != nil {
return err
}
err = os.Chmod(f.Name(), perm)
if err != nil {
f.Close()
return err
}
n, err := f.Write(data)
if err == nil && n < len(data) {
f.Close()
return io.ErrShortWrite
}
if err != nil {
f.Close()
return err
}
if err := f.Sync(); err != nil {
f.Close()
return err
}
if err := f.Close(); err != nil {
return err
}
return os.Rename(f.Name(), filename)
}
func toHash(in string) string { func toHash(in string) string {
return digest.FromBytes([]byte(in)).Hex()[:20] return digest.FromBytes([]byte(in)).Hex()[:20]
} }