Pulse/internal/ceph/collector.go
rcourtman 67351bf58c feat: AI integration, Docker metrics, RAID display, and infrastructure improvements
- Add Claude OAuth authentication support with hybrid API key/OAuth flow
- Implement Docker container historical metrics in backend and charts API
- Add CEPH cluster data collection and new Ceph page
- Enhance RAID status display with detailed tooltips and visual indicators
- Fix host deduplication logic with Docker bridge IP filtering
- Fix NVMe temperature collection in host agent
- Add comprehensive test coverage for new features
- Improve frontend sparklines and metrics history handling
- Fix navigation issues and frontend reload loops
2025-12-09 09:29:27 +00:00

353 lines
10 KiB
Go

// Package ceph provides functionality for collecting Ceph cluster status
// directly from the local system using the ceph CLI.
package ceph
import (
"bytes"
"context"
"encoding/json"
"fmt"
"os/exec"
"strings"
"time"
)
// ClusterStatus represents the complete Ceph cluster status as collected by the agent.
type ClusterStatus struct {
FSID string `json:"fsid"`
Health HealthStatus `json:"health"`
MonMap MonitorMap `json:"monMap,omitempty"`
MgrMap ManagerMap `json:"mgrMap,omitempty"`
OSDMap OSDMap `json:"osdMap"`
PGMap PGMap `json:"pgMap"`
Pools []Pool `json:"pools,omitempty"`
Services []ServiceInfo `json:"services,omitempty"`
CollectedAt time.Time `json:"collectedAt"`
}
// HealthStatus represents Ceph cluster health.
type HealthStatus struct {
Status string `json:"status"` // HEALTH_OK, HEALTH_WARN, HEALTH_ERR
Checks map[string]Check `json:"checks,omitempty"`
Summary []HealthSummary `json:"summary,omitempty"`
}
// Check represents a health check detail.
type Check struct {
Severity string `json:"severity"`
Message string `json:"message,omitempty"`
Detail []string `json:"detail,omitempty"`
}
// HealthSummary represents a health summary message.
type HealthSummary struct {
Severity string `json:"severity"`
Message string `json:"message"`
}
// MonitorMap represents Ceph monitor information.
type MonitorMap struct {
Epoch int `json:"epoch"`
NumMons int `json:"numMons"`
Monitors []Monitor `json:"monitors,omitempty"`
}
// Monitor represents a single Ceph monitor.
type Monitor struct {
Name string `json:"name"`
Rank int `json:"rank"`
Addr string `json:"addr,omitempty"`
Status string `json:"status,omitempty"`
}
// ManagerMap represents Ceph manager information.
type ManagerMap struct {
Available bool `json:"available"`
NumMgrs int `json:"numMgrs"`
ActiveMgr string `json:"activeMgr,omitempty"`
Standbys int `json:"standbys"`
}
// OSDMap represents OSD status summary.
type OSDMap struct {
Epoch int `json:"epoch"`
NumOSDs int `json:"numOsds"`
NumUp int `json:"numUp"`
NumIn int `json:"numIn"`
NumDown int `json:"numDown,omitempty"`
NumOut int `json:"numOut,omitempty"`
}
// PGMap represents placement group statistics.
type PGMap struct {
NumPGs int `json:"numPgs"`
BytesTotal uint64 `json:"bytesTotal"`
BytesUsed uint64 `json:"bytesUsed"`
BytesAvailable uint64 `json:"bytesAvailable"`
DataBytes uint64 `json:"dataBytes,omitempty"`
UsagePercent float64 `json:"usagePercent"`
DegradedRatio float64 `json:"degradedRatio,omitempty"`
MisplacedRatio float64 `json:"misplacedRatio,omitempty"`
ReadBytesPerSec uint64 `json:"readBytesPerSec,omitempty"`
WriteBytesPerSec uint64 `json:"writeBytesPerSec,omitempty"`
ReadOpsPerSec uint64 `json:"readOpsPerSec,omitempty"`
WriteOpsPerSec uint64 `json:"writeOpsPerSec,omitempty"`
}
// Pool represents a Ceph pool.
type Pool struct {
ID int `json:"id"`
Name string `json:"name"`
BytesUsed uint64 `json:"bytesUsed"`
BytesAvailable uint64 `json:"bytesAvailable"`
Objects uint64 `json:"objects"`
PercentUsed float64 `json:"percentUsed"`
}
// ServiceInfo represents a Ceph service summary.
type ServiceInfo struct {
Type string `json:"type"` // mon, mgr, osd, mds, rgw
Running int `json:"running"`
Total int `json:"total"`
Daemons []string `json:"daemons,omitempty"`
}
// IsAvailable checks if the ceph CLI is available on the system.
func IsAvailable(ctx context.Context) bool {
cmd := exec.CommandContext(ctx, "which", "ceph")
return cmd.Run() == nil
}
// Collect gathers Ceph cluster status using the ceph CLI.
// Returns nil if Ceph is not available or not configured on this host.
func Collect(ctx context.Context) (*ClusterStatus, error) {
// Check if ceph CLI is available
if !IsAvailable(ctx) {
return nil, nil
}
// Try to get ceph status - this will fail if not a ceph node
statusJSON, err := runCephCommand(ctx, "status", "--format", "json")
if err != nil {
// Not an error - just means this isn't a Ceph node
return nil, nil
}
status, err := parseStatus(statusJSON)
if err != nil {
return nil, fmt.Errorf("parse ceph status: %w", err)
}
// Get pool usage from ceph df
dfJSON, err := runCephCommand(ctx, "df", "--format", "json")
if err == nil {
pools, usagePercent, err := parseDF(dfJSON)
if err == nil {
status.Pools = pools
if status.PGMap.UsagePercent == 0 && usagePercent > 0 {
status.PGMap.UsagePercent = usagePercent
}
}
}
status.CollectedAt = time.Now().UTC()
return status, nil
}
// runCephCommand executes a ceph command and returns the output.
func runCephCommand(ctx context.Context, args ...string) ([]byte, error) {
// Use a reasonable timeout for each command
cmdCtx, cancel := context.WithTimeout(ctx, 10*time.Second)
defer cancel()
cmd := exec.CommandContext(cmdCtx, "ceph", args...)
var stdout, stderr bytes.Buffer
cmd.Stdout = &stdout
cmd.Stderr = &stderr
if err := cmd.Run(); err != nil {
return nil, fmt.Errorf("ceph %s failed: %w (stderr: %s)",
strings.Join(args, " "), err, stderr.String())
}
return stdout.Bytes(), nil
}
// parseStatus parses the output of `ceph status --format json`.
func parseStatus(data []byte) (*ClusterStatus, error) {
var raw struct {
FSID string `json:"fsid"`
Health struct {
Status string `json:"status"`
Checks map[string]struct {
Severity string `json:"severity"`
Summary struct {
Message string `json:"message"`
} `json:"summary"`
Detail []struct {
Message string `json:"message"`
} `json:"detail"`
} `json:"checks"`
} `json:"health"`
MonMap struct {
Epoch int `json:"epoch"`
Mons []struct {
Name string `json:"name"`
Rank int `json:"rank"`
Addr string `json:"addr"`
} `json:"mons"`
} `json:"monmap"`
MgrMap struct {
Available bool `json:"available"`
NumActive int `json:"num_active_name,omitempty"`
ActiveName string `json:"active_name"`
Standbys []struct {
Name string `json:"name"`
} `json:"standbys"`
} `json:"mgrmap"`
OSDMap struct {
Epoch int `json:"epoch"`
NumOSD int `json:"num_osds"`
NumUp int `json:"num_up_osds"`
NumIn int `json:"num_in_osds"`
} `json:"osdmap"`
PGMap struct {
NumPGs int `json:"num_pgs"`
BytesTotal uint64 `json:"bytes_total"`
BytesUsed uint64 `json:"bytes_used"`
BytesAvail uint64 `json:"bytes_avail"`
DataBytes uint64 `json:"data_bytes"`
DegradedRatio float64 `json:"degraded_ratio"`
MisplacedRatio float64 `json:"misplaced_ratio"`
ReadBytesPerSec uint64 `json:"read_bytes_sec"`
WriteBytesPerSec uint64 `json:"write_bytes_sec"`
ReadOpsPerSec uint64 `json:"read_op_per_sec"`
WriteOpsPerSec uint64 `json:"write_op_per_sec"`
} `json:"pgmap"`
}
if err := json.Unmarshal(data, &raw); err != nil {
return nil, fmt.Errorf("unmarshal: %w", err)
}
status := &ClusterStatus{
FSID: raw.FSID,
Health: HealthStatus{
Status: raw.Health.Status,
Checks: make(map[string]Check),
},
MonMap: MonitorMap{
Epoch: raw.MonMap.Epoch,
NumMons: len(raw.MonMap.Mons),
},
MgrMap: ManagerMap{
Available: raw.MgrMap.Available,
NumMgrs: 1 + len(raw.MgrMap.Standbys),
ActiveMgr: raw.MgrMap.ActiveName,
Standbys: len(raw.MgrMap.Standbys),
},
OSDMap: OSDMap{
Epoch: raw.OSDMap.Epoch,
NumOSDs: raw.OSDMap.NumOSD,
NumUp: raw.OSDMap.NumUp,
NumIn: raw.OSDMap.NumIn,
NumDown: raw.OSDMap.NumOSD - raw.OSDMap.NumUp,
NumOut: raw.OSDMap.NumOSD - raw.OSDMap.NumIn,
},
PGMap: PGMap{
NumPGs: raw.PGMap.NumPGs,
BytesTotal: raw.PGMap.BytesTotal,
BytesUsed: raw.PGMap.BytesUsed,
BytesAvailable: raw.PGMap.BytesAvail,
DataBytes: raw.PGMap.DataBytes,
DegradedRatio: raw.PGMap.DegradedRatio,
MisplacedRatio: raw.PGMap.MisplacedRatio,
ReadBytesPerSec: raw.PGMap.ReadBytesPerSec,
WriteBytesPerSec: raw.PGMap.WriteBytesPerSec,
ReadOpsPerSec: raw.PGMap.ReadOpsPerSec,
WriteOpsPerSec: raw.PGMap.WriteOpsPerSec,
},
}
// Calculate usage percent
if raw.PGMap.BytesTotal > 0 {
status.PGMap.UsagePercent = float64(raw.PGMap.BytesUsed) / float64(raw.PGMap.BytesTotal) * 100
}
// Parse monitors
for _, mon := range raw.MonMap.Mons {
status.MonMap.Monitors = append(status.MonMap.Monitors, Monitor{
Name: mon.Name,
Rank: mon.Rank,
Addr: mon.Addr,
})
}
// Parse health checks
for name, check := range raw.Health.Checks {
details := make([]string, 0, len(check.Detail))
for _, d := range check.Detail {
details = append(details, d.Message)
}
status.Health.Checks[name] = Check{
Severity: check.Severity,
Message: check.Summary.Message,
Detail: details,
}
}
// Build service summary
status.Services = []ServiceInfo{
{Type: "mon", Running: len(raw.MonMap.Mons), Total: len(raw.MonMap.Mons)},
{Type: "mgr", Running: boolToInt(raw.MgrMap.Available), Total: status.MgrMap.NumMgrs},
{Type: "osd", Running: raw.OSDMap.NumUp, Total: raw.OSDMap.NumOSD},
}
return status, nil
}
// parseDF parses the output of `ceph df --format json`.
func parseDF(data []byte) ([]Pool, float64, error) {
var raw struct {
Stats struct {
TotalBytes uint64 `json:"total_bytes"`
TotalUsedBytes uint64 `json:"total_used_bytes"`
PercentUsed float64 `json:"percent_used"`
} `json:"stats"`
Pools []struct {
ID int `json:"id"`
Name string `json:"name"`
Stats struct {
BytesUsed uint64 `json:"bytes_used"`
MaxAvail uint64 `json:"max_avail"`
Objects uint64 `json:"objects"`
PercentUsed float64 `json:"percent_used"`
} `json:"stats"`
} `json:"pools"`
}
if err := json.Unmarshal(data, &raw); err != nil {
return nil, 0, fmt.Errorf("unmarshal df: %w", err)
}
pools := make([]Pool, 0, len(raw.Pools))
for _, p := range raw.Pools {
pools = append(pools, Pool{
ID: p.ID,
Name: p.Name,
BytesUsed: p.Stats.BytesUsed,
BytesAvailable: p.Stats.MaxAvail,
Objects: p.Stats.Objects,
PercentUsed: p.Stats.PercentUsed * 100, // Convert from 0-1 to 0-100
})
}
return pools, raw.Stats.PercentUsed * 100, nil
}
func boolToInt(b bool) int {
if b {
return 1
}
return 0
}