mirror of
https://github.com/offen/docker-volume-backup.git
synced 2025-12-24 09:41:12 +01:00
Compare commits
5 Commits
v2.37.0-al
...
v2.37.0
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
4c74313222 | ||
|
|
de03d4f704 | ||
|
|
65626dd3d4 | ||
|
|
69eceb3982 | ||
|
|
1d45062100 |
@@ -9,6 +9,7 @@ import (
|
||||
"log/slog"
|
||||
"os"
|
||||
"os/signal"
|
||||
"runtime"
|
||||
"syscall"
|
||||
|
||||
"github.com/robfig/cron/v3"
|
||||
@@ -122,7 +123,7 @@ func runScript(c *Config) (err error) {
|
||||
return runErr
|
||||
}
|
||||
|
||||
func (c *command) runInForeground() error {
|
||||
func (c *command) runInForeground(profileCronExpression string) error {
|
||||
cr := cron.New(
|
||||
cron.WithParser(
|
||||
cron.NewParser(
|
||||
@@ -189,6 +190,28 @@ func (c *command) runInForeground() error {
|
||||
}
|
||||
}
|
||||
|
||||
if profileCronExpression != "" {
|
||||
if _, err := cr.AddFunc(profileCronExpression, func() {
|
||||
memStats := runtime.MemStats{}
|
||||
runtime.ReadMemStats(&memStats)
|
||||
c.logger.Info(
|
||||
"Collecting runtime information",
|
||||
"num_goroutines",
|
||||
runtime.NumGoroutine(),
|
||||
"memory_heap_alloc",
|
||||
formatBytes(memStats.HeapAlloc, false),
|
||||
"memory_heap_inuse",
|
||||
formatBytes(memStats.HeapInuse, false),
|
||||
"memory_heap_sys",
|
||||
formatBytes(memStats.HeapSys, false),
|
||||
"memory_heap_objects",
|
||||
memStats.HeapObjects,
|
||||
)
|
||||
}); err != nil {
|
||||
return fmt.Errorf("runInForeground: error adding profiling job: %w", err)
|
||||
}
|
||||
}
|
||||
|
||||
var quit = make(chan os.Signal, 1)
|
||||
signal.Notify(quit, syscall.SIGTERM, syscall.SIGINT)
|
||||
cr.Start()
|
||||
@@ -214,11 +237,12 @@ func (c *command) runAsCommand() error {
|
||||
|
||||
func main() {
|
||||
foreground := flag.Bool("foreground", false, "run the tool in the foreground")
|
||||
profile := flag.String("profile", "", "collect runtime metrics and log them periodically on the given cron expression")
|
||||
flag.Parse()
|
||||
|
||||
c := newCommand()
|
||||
if *foreground {
|
||||
c.must(c.runInForeground())
|
||||
c.must(c.runInForeground(*profile))
|
||||
} else {
|
||||
c.must(c.runAsCommand())
|
||||
}
|
||||
|
||||
@@ -112,6 +112,12 @@ func newScript(c *Config) (*script, error) {
|
||||
return nil, fmt.Errorf("newScript: failed to create docker client")
|
||||
}
|
||||
s.cli = cli
|
||||
s.registerHook(hookLevelPlumbing, func(err error) error {
|
||||
if err := s.cli.Close(); err != nil {
|
||||
return fmt.Errorf("newScript: failed to close docker client: %w", err)
|
||||
}
|
||||
return nil
|
||||
})
|
||||
}
|
||||
|
||||
logFunc := func(logType storage.LogLevel, context string, msg string, params ...any) {
|
||||
|
||||
Reference in New Issue
Block a user