mirror of
https://github.com/rcourtman/Pulse.git
synced 2026-05-13 15:28:38 +00:00
d07b471e40
- Separated metrics collection into internal/dockeragent/collect.go - Added agent self-update pre-flight check (--self-test) - Implemented signed binary verification with key rotation for updates - Added batch update support to frontend with parallel processing - Cleaned up agent.go and added startup cleanup for backup containers - Updated documentation for Docker features and agent security
69 lines
2.1 KiB
Go
69 lines
2.1 KiB
Go
package dockeragent
|
|
|
|
import (
|
|
"context"
|
|
"crypto/rand"
|
|
"encoding/json"
|
|
"io"
|
|
"os"
|
|
"os/exec"
|
|
"runtime"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/docker/docker/client"
|
|
"github.com/rcourtman/pulse-go-rewrite/internal/hostmetrics"
|
|
)
|
|
|
|
var (
|
|
connectRuntimeFn = connectRuntime
|
|
hostmetricsCollect = hostmetrics.Collect
|
|
newTickerFn = time.NewTicker
|
|
newTimerFn = time.NewTimer
|
|
randomDurationFn = randomDuration
|
|
nowFn = time.Now
|
|
sleepFn = time.Sleep
|
|
jsonMarshalFn = json.Marshal
|
|
normalizeTargetsFn = normalizeTargets
|
|
buildRuntimeCandidatesFn = buildRuntimeCandidates
|
|
tryRuntimeCandidateFn = tryRuntimeCandidate
|
|
randIntFn = rand.Int
|
|
osExecutableFn = os.Executable
|
|
osCreateTempFn = os.CreateTemp
|
|
closeFileFn = func(f *os.File) error { return f.Close() }
|
|
osRenameFn = os.Rename
|
|
osChmodFn = os.Chmod
|
|
osRemoveFn = os.Remove
|
|
osReadFileFn = os.ReadFile
|
|
osWriteFileFn = os.WriteFile
|
|
osStatFn = os.Stat
|
|
syscallExecFn = syscall.Exec
|
|
goArch = runtime.GOARCH
|
|
unameMachine = func() (string, error) {
|
|
out, err := exec.Command("uname", "-m").Output()
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(out), nil
|
|
}
|
|
machineIDPaths = []string{
|
|
"/etc/machine-id",
|
|
"/var/lib/dbus/machine-id",
|
|
}
|
|
unraidVersionPath = "/etc/unraid-version"
|
|
unraidPersistPath = "/boot/config/plugins/pulse-docker-agent/pulse-docker-agent"
|
|
unraidStartupScriptPath = "/boot/config/go.d/pulse-docker-agent.sh"
|
|
agentLogPath = "/var/log/pulse-docker-agent.log"
|
|
openProcUptime = func() (io.ReadCloser, error) {
|
|
return os.Open("/proc/uptime")
|
|
}
|
|
newDockerClientFn = func(opts ...client.Opt) (dockerClient, error) {
|
|
return client.NewClientWithOpts(opts...)
|
|
}
|
|
selfUpdateFunc = func(a *Agent, ctx context.Context) error {
|
|
return a.selfUpdate(ctx)
|
|
}
|
|
execCommandContextFn = func(ctx context.Context, name string, arg ...string) *exec.Cmd {
|
|
return exec.CommandContext(ctx, name, arg...)
|
|
}
|
|
)
|