mirror of
https://github.com/prometheus/node_exporter.git
synced 2024-12-30 07:59:53 -08:00
179 lines
5 KiB
Go
179 lines
5 KiB
Go
|
// Copyright 2021 The Prometheus Authors
|
||
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
||
|
// you may not use this file except in compliance with the License.
|
||
|
// You may obtain a copy of the License at
|
||
|
//
|
||
|
// http://www.apache.org/licenses/LICENSE-2.0
|
||
|
//
|
||
|
// Unless required by applicable law or agreed to in writing, software
|
||
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
||
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||
|
// See the License for the specific language governing permissions and
|
||
|
// limitations under the License.
|
||
|
|
||
|
package collector
|
||
|
|
||
|
import (
|
||
|
"errors"
|
||
|
"io"
|
||
|
"os"
|
||
|
"regexp"
|
||
|
"strconv"
|
||
|
"strings"
|
||
|
"sync"
|
||
|
"time"
|
||
|
|
||
|
"github.com/go-kit/log"
|
||
|
"github.com/go-kit/log/level"
|
||
|
envparse "github.com/hashicorp/go-envparse"
|
||
|
"github.com/prometheus/client_golang/prometheus"
|
||
|
)
|
||
|
|
||
|
const (
|
||
|
etcOSRelease = "/etc/os-release"
|
||
|
usrLibOSRelease = "/usr/lib/os-release"
|
||
|
)
|
||
|
|
||
|
var (
|
||
|
versionRegex = regexp.MustCompile(`^[0-9]+\.?[0-9]*`)
|
||
|
)
|
||
|
|
||
|
type osRelease struct {
|
||
|
Name string
|
||
|
ID string
|
||
|
IDLike string
|
||
|
PrettyName string
|
||
|
Variant string
|
||
|
VariantID string
|
||
|
Version string
|
||
|
VersionID string
|
||
|
VersionCodename string
|
||
|
BuildID string
|
||
|
ImageID string
|
||
|
ImageVersion string
|
||
|
}
|
||
|
|
||
|
type osReleaseCollector struct {
|
||
|
infoDesc *prometheus.Desc
|
||
|
logger log.Logger
|
||
|
os *osRelease
|
||
|
osFilename string // file name of cached release information
|
||
|
osMtime time.Time // mtime of cached release file
|
||
|
osMutex sync.Mutex
|
||
|
osReleaseFilenames []string // all os-release file names to check
|
||
|
version float64
|
||
|
versionDesc *prometheus.Desc
|
||
|
}
|
||
|
|
||
|
func init() {
|
||
|
registerCollector("os", defaultEnabled, NewOSCollector)
|
||
|
}
|
||
|
|
||
|
// NewOSCollector returns a new Collector exposing os-release information.
|
||
|
func NewOSCollector(logger log.Logger) (Collector, error) {
|
||
|
return &osReleaseCollector{
|
||
|
logger: logger,
|
||
|
infoDesc: prometheus.NewDesc(
|
||
|
prometheus.BuildFQName(namespace, "os", "info"),
|
||
|
"A metric with a constant '1' value labeled by build_id, id, id_like, image_id, image_version, "+
|
||
|
"name, pretty_name, variant, variant_id, version, version_codename, version_id.",
|
||
|
[]string{"build_id", "id", "id_like", "image_id", "image_version", "name", "pretty_name",
|
||
|
"variant", "variant_id", "version", "version_codename", "version_id"}, nil,
|
||
|
),
|
||
|
osReleaseFilenames: []string{etcOSRelease, usrLibOSRelease},
|
||
|
versionDesc: prometheus.NewDesc(
|
||
|
prometheus.BuildFQName(namespace, "os", "version"),
|
||
|
"Metric containing the major.minor part of the OS version.",
|
||
|
[]string{"id", "id_like", "name"}, nil,
|
||
|
),
|
||
|
}, nil
|
||
|
}
|
||
|
|
||
|
func parseOSRelease(r io.Reader) (*osRelease, error) {
|
||
|
env, err := envparse.Parse(r)
|
||
|
return &osRelease{
|
||
|
Name: env["NAME"],
|
||
|
ID: env["ID"],
|
||
|
IDLike: env["ID_LIKE"],
|
||
|
PrettyName: env["PRETTY_NAME"],
|
||
|
Variant: env["VARIANT"],
|
||
|
VariantID: env["VARIANT_ID"],
|
||
|
Version: env["VERSION"],
|
||
|
VersionID: env["VERSION_ID"],
|
||
|
VersionCodename: env["VERSION_CODENAME"],
|
||
|
BuildID: env["BUILD_ID"],
|
||
|
ImageID: env["IMAGE_ID"],
|
||
|
ImageVersion: env["IMAGE_VERSION"],
|
||
|
}, err
|
||
|
}
|
||
|
|
||
|
func (c *osReleaseCollector) UpdateStruct(path string) error {
|
||
|
releaseFile, err := os.Open(path)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
defer releaseFile.Close()
|
||
|
|
||
|
stat, err := releaseFile.Stat()
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
t := stat.ModTime()
|
||
|
if path == c.osFilename && t == c.osMtime {
|
||
|
// osReleaseCollector struct is already up-to-date.
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// Acquire a lock to update the osReleaseCollector struct.
|
||
|
c.osMutex.Lock()
|
||
|
defer c.osMutex.Unlock()
|
||
|
|
||
|
level.Debug(c.logger).Log("msg", "file modification time has changed",
|
||
|
"file", path, "old_value", c.osMtime, "new_value", t)
|
||
|
c.osFilename = path
|
||
|
c.osMtime = t
|
||
|
|
||
|
c.os, err = parseOSRelease(releaseFile)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
majorMinor := versionRegex.FindString(c.os.VersionID)
|
||
|
if majorMinor != "" {
|
||
|
c.version, err = strconv.ParseFloat(majorMinor, 64)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
} else {
|
||
|
c.version = 0
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
func (c *osReleaseCollector) Update(ch chan<- prometheus.Metric) error {
|
||
|
for i, path := range c.osReleaseFilenames {
|
||
|
err := c.UpdateStruct(*rootfsPath + path)
|
||
|
if err == nil {
|
||
|
break
|
||
|
}
|
||
|
if errors.Is(err, os.ErrNotExist) {
|
||
|
if i >= (len(c.osReleaseFilenames) - 1) {
|
||
|
level.Debug(c.logger).Log("msg", "no os-release file found", "files", strings.Join(c.osReleaseFilenames, ","))
|
||
|
return ErrNoData
|
||
|
}
|
||
|
continue
|
||
|
}
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
ch <- prometheus.MustNewConstMetric(c.infoDesc, prometheus.GaugeValue, 1.0,
|
||
|
c.os.BuildID, c.os.ID, c.os.IDLike, c.os.ImageID, c.os.ImageVersion, c.os.Name, c.os.PrettyName,
|
||
|
c.os.Variant, c.os.VariantID, c.os.Version, c.os.VersionCodename, c.os.VersionID)
|
||
|
if c.version > 0 {
|
||
|
ch <- prometheus.MustNewConstMetric(c.versionDesc, prometheus.GaugeValue, c.version,
|
||
|
c.os.ID, c.os.IDLike, c.os.Name)
|
||
|
}
|
||
|
return nil
|
||
|
}
|