lab/traefik-cert-exporter
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: write file not truncate

Browse Source
This commit is contained in:
lab
2025-11-13 13:25:46 +08:00
parent 190822a46f
commit 8e35409a88
1 changed files with 26 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
cmd/cert-exporter/main.go
View File

@@ -219,13 +219,19 @@ func DetachFullchainPem(fullChain string) (string, string) {
// a bool value returned means the write action had performed done or not // a bool value returned means the write action had performed done or not
func CompareAndOverWritePemFile(fileDir, filename string, data []byte) (bool, error) { func CompareAndOverWritePemFile(fileDir, filename string, data []byte) (bool, error) {
fp := pemFullPath(fileDir, filename) fp := pemFullPath(fileDir, filename)
fd, err := os.OpenFile(fp, os.O_RDWR|os.O_CREATE, 0644) fdr, err := os.OpenFile(fp, os.O_RDONLY, 0644)
if err != nil {
var originalData []byte
if os.IsNotExist(err) {
fmt.Errorf("file not exist, it will be created when write: %w", err)
} else if err != nil {
return false, fmt.Errorf("failed to open file when attempting to write: %w", err) return false, fmt.Errorf("failed to open file when attempting to write: %w", err)
} } else {
defer fdr.Close()
// read original file content // read original file content
originalData, err := io.ReadAll(fd) originalData, err := io.ReadAll(fdr)
if err != nil { if err != nil {
return false, fmt.Errorf("failed to read original pem file: %w", err) return false, fmt.Errorf("failed to read original pem file: %w", err)
} }
@@ -236,20 +242,20 @@ func CompareAndOverWritePemFile(fileDir, filename string, data []byte) (bool, er
log.Info("pem file content as before, pass", "file", fp) log.Info("pem file content as before, pass", "file", fp)
return false, nil return false, nil
} }
}
fdw, err = os.OpenFile(fp, os.O_WRONLY|os.O_CREATE|os.O_TRUNC, 0644)
defer fdw.Close()
// write data to file // write data to file
if _, err := fd.Write(data); err != nil { if _, err := fdw.Write(data); err != nil {
return false, fmt.Errorf("failed to write file: %w", err) return false, fmt.Errorf("failed to write file: %w", err)
} }
if err := fd.Sync(); err != nil { if err := fdw.Sync(); err != nil {
return false, fmt.Errorf("failed to commits the current contents of the file to stable storage: %w", err) return false, fmt.Errorf("failed to commits the current contents of the file to stable storage: %w", err)
} }
if err := fd.Close(); err != nil {
return false, fmt.Errorf("failed to close file: %s, %w", fp, err)
}
return true, nil return true, nil
} }