beenull/crowdsec
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

hub update: reload crowdsec if only data files have changed (#2912)

Browse source
This commit is contained in:
mmetc 2024-03-20 15:46:14 +01:00 committed by GitHub
parent 75a50c0c9d
commit 7779c7ff0c
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 8 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
pkg/cwhub/dataset.go
View file

@ -32,6 +32,9 @@ func downloadFile(url string, destPath string) error {
return fmt.Errorf("bad http code %d for %s", resp.StatusCode, url) return fmt.Errorf("bad http code %d for %s", resp.StatusCode, url)
} }
// Download to a temporary location to avoid corrupting files
// that are currently in use or memory mapped.
tmpFile, err := os.CreateTemp(filepath.Dir(destPath), filepath.Base(destPath)+".*.tmp") tmpFile, err := os.CreateTemp(filepath.Dir(destPath), filepath.Base(destPath)+".*.tmp")
if err != nil { if err != nil {
return err return err
@ -57,6 +60,11 @@ func downloadFile(url string, destPath string) error {
return err return err
} }
// a check on stdout is used while scripting to know if the hub has been upgraded
// and a configuration reload is required
// TODO: use a better way to communicate this
fmt.Printf("updated %s\n", filepath.Base(destPath))
if err = os.Rename(tmpFileName, destPath); err != nil { if err = os.Rename(tmpFileName, destPath); err != nil {
return err return err
} }