simono41/image-checker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

add cache

Browse source
This commit is contained in:
Simon Rieger 2025-07-17 10:24:53 +02:00
parent e88bb581c6
commit a92ed9fe8f
1 changed files with 178 additions and 129 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

121
go/main.go
View file

@ -8,6 +8,7 @@ import (
"regexp"
"strings"
"sync"
"time"
"github.com/docker/docker/api/types/container"
"github.com/docker/docker/api/types/image"
@ -18,6 +19,23 @@ import (
"github.com/regclient/regclient/types/ref"
)
type ImageStatus struct {
ContainerName, Image, Tag string
UpdateAvailable float64
LocalDigest, RemoteDigest string
}
type StatusCache struct {
sync.RWMutex
Data []ImageStatus
LastCheck time.Time
}
var (
cache = &StatusCache{Data: []ImageStatus{}, LastCheck: time.Time{}}
interval = 6 * time.Hour
)
func toRegistryImage(imageTag string) (string, error) {
r := regexp.MustCompile(`^(?:(?P<registry>[^/]+)/)?(?P<repo>[^:]+)(?::(?P<tag>.+))?$`)
match := r.FindStringSubmatch(imageTag)
@ -45,26 +63,7 @@ func extractDigest(s string) string {
return s
}
type imageUpdateCollector struct {
metric *prometheus.Desc
}
func newImageUpdateCollector() *imageUpdateCollector {
return &imageUpdateCollector{
metric: prometheus.NewDesc(
"docker_image_update_available",
"Ob Update für das lokale Docker-Image des laufenden Containers für das Tag im Registry verfügbar ist (1=Update, 0=aktuell)",
[]string{"container_name", "image", "tag"},
nil,
),
}
}
func (c *imageUpdateCollector) Describe(ch chan<- *prometheus.Desc) {
ch <- c.metric
}
func (c *imageUpdateCollector) Collect(ch chan<- prometheus.Metric) {
func checkImageUpdates() {
ctx := context.Background()
cli, err := client.NewClientWithOpts(client.FromEnv)
if err != nil {
@ -74,7 +73,6 @@ func (c *imageUpdateCollector) Collect(ch chan<- prometheus.Metric) {
defer cli.Close()
rc := regclient.New()
// Hole nur aktive (laufende) Container
containers, err := cli.ContainerList(ctx, container.ListOptions{All: false})
if err != nil {
log.Printf("Fehler bei ContainerList: %v", err)
@ -99,18 +97,22 @@ func (c *imageUpdateCollector) Collect(ch chan<- prometheus.Metric) {
}
var wg sync.WaitGroup
results := make([]ImageStatus, 0)
resultsLock := sync.Mutex{}
for _, ctr := range containers {
tag := ctr.Image
imageRef, err := toRegistryImage(tag)
if err != nil {
log.Printf("ImageRef-Fehler bei %s: %v", tag, err)
continue
}
localDigest := imageTagToDigest[tag]
containerName := "unknown"
if len(ctr.Names) > 0 {
containerName = ctr.Names[0]
}
wg.Add(1)
go func(ctr container.Summary, tag, localDigest, imageRef string) {
go func(containerName, tag, localDigest, imageRef string) {
defer wg.Done()
refObj, err := ref.New(imageRef)
if err != nil {
@ -124,30 +126,77 @@ func (c *imageUpdateCollector) Collect(ch chan<- prometheus.Metric) {
}
remoteDigest := desc.GetDigest().String()
fmt.Printf("Container: %s\n Image: %s\n Local Digest: %s\n Remote Digest: %s\n", ctr.Names[0], tag, localDigest, remoteDigest)
var update float64 = 0
update := 0.0
if localDigest != remoteDigest {
fmt.Println(" -> Update verfügbar!")
update = 1
update = 1.0
fmt.Printf("Container: %s\n Image: %s\n Local Digest: %s\n Remote Digest: %s\n -> Update verfügbar!\n", containerName, tag, localDigest, remoteDigest)
} else {
fmt.Println(" -> Kein Update verfügbar.")
fmt.Printf("Container: %s\n Image: %s\n Local Digest: %s\n Remote Digest: %s\n -> Kein Update verfügbar.\n", containerName, tag, localDigest, remoteDigest)
}
// Labels container_name, image, tag
labelImg, labelTag := tag, "latest"
if cp := strings.Split(tag, ":"); len(cp) == 2 {
labelImg, labelTag = cp[0], cp[1]
}
ch <- prometheus.MustNewConstMetric(
c.metric, prometheus.GaugeValue,
update, ctr.Names[0], labelImg, labelTag,
)
}(ctr, tag, localDigest, imageRef)
resultsLock.Lock()
results = append(results, ImageStatus{
ContainerName: containerName, Image: labelImg, Tag: labelTag,
UpdateAvailable: update, LocalDigest: localDigest, RemoteDigest: remoteDigest,
})
resultsLock.Unlock()
}(containerName, tag, localDigest, imageRef)
}
wg.Wait()
// Im Cache speichern
cache.Lock()
cache.Data = results
cache.LastCheck = time.Now()
cache.Unlock()
}
type imageUpdateCollector struct {
metric *prometheus.Desc
}
func newImageUpdateCollector() *imageUpdateCollector {
return &imageUpdateCollector{
metric: prometheus.NewDesc(
"docker_image_update_available",
"Ob Update für das lokale Docker-Image des laufenden Containers für das Tag im Registry verfügbar ist (1=Update, 0=aktuell)",
[]string{"container_name", "image", "tag"},
nil,
),
}
}
func (c *imageUpdateCollector) Describe(ch chan<- *prometheus.Desc) {
ch <- c.metric
}
func (c *imageUpdateCollector) Collect(ch chan<- prometheus.Metric) {
cache.RLock()
for _, stat := range cache.Data {
ch <- prometheus.MustNewConstMetric(
c.metric, prometheus.GaugeValue,
stat.UpdateAvailable, stat.ContainerName, stat.Image, stat.Tag,
)
}
cache.RUnlock()
}
func main() {
log.Println("Starte Prometheus Exporter für laufende Container-Images (Port 9788)...")
log.Printf("Starte Docker-Image-Update-Exporter mit 6h-Intervall...")
// Hintergrund: alle 6h Update, zu Beginn auch direkt
go func() {
for {
checkImageUpdates()
time.Sleep(interval)
}
}()
checkImageUpdates() // Initiales Scrape
exporter := newImageUpdateCollector()
prometheus.MustRegister(exporter)
http.Handle("/metrics", promhttp.Handler())