exporter

type AccountCsvFetcher struct {
	// contains filtered or unexported fields
}

type AccountLimitMetric struct {
	Account string
	// limit to the amount of resources for a particular account in the RUNNING state
	AllocatedMem  float64
	AllocatedCPU  float64
	AllocatedJobs float64
	// limit to the amount of resources that can be either PENDING or RUNNING
	TotalJobs float64
}

type AccountMetric struct {
	// contains filtered or unexported fields
}

type AtomicProcFetcher struct {
	sync.Mutex
	Info map[int64]*TraceInfo
	// contains filtered or unexported fields
}

type AtomicThrottledCache[C SlurmPrimitiveMetric] struct {
	sync.Mutex
	// contains filtered or unexported fields
}

type CliFlags struct {
	SlurmLicEnabled      bool
	SlurmDiagEnabled     bool
	SlurmCliFallback     bool
	TraceEnabled         bool
	SacctEnabled         bool
	SlurmPollLimit       float64
	LogLevel             string
	ListenAddress        string
	MetricsPath          string
	SlurmSqueueOverride  string
	SlurmSinfoOverride   string
	SlurmDiagOverride    string
	SlurmAcctOverride    string
	TraceRate            uint64
	TracePath            string
	SlurmLicenseOverride string
}

type CliJsonLicMetricFetcher struct {
	// contains filtered or unexported fields
}

type CliOpts struct {
	// contains filtered or unexported fields
}

type CliScraper struct {
	// contains filtered or unexported fields
}

type CoercedInt int

type Config struct {
	TraceConf     *TraceConfig
	PollLimit     float64
	LogLevel      slog.Level
	ListenAddress string
	MetricsPath   string
	// contains filtered or unexported fields
}

type CpuSummaryMetric struct {
	Total    float64
	Idle     float64
	Load     float64
	PerState map[string]*PerStateMetric
}

type DiagMetric struct {
	ServerThreadCount     int              `json:"server_thread_count"`
	DBDAgentQueueSize     int              `json:"dbd_agent_queue_size"`
	RpcByUser             []UserRpcInfo    `json:"rpcs_by_user"`
	RpcByMessageType      []MessageRpcInfo `json:"rpcs_by_message_type"`
	BackfillJobCount      int              `json:"bf_backfilled_jobs"`
	BackfillCycleCountSum int              `json:"bf_cycle_sum"`
	BackfillCycleCounter  int              `json:"bf_cycle_counter"`
	BackfillLastDepth     int              `json:"bf_last_depth"`
	BackfillLastDepthTry  int              `json:"bf_last_depth_try"`
}

type DiagnosticsCollector struct {
	// contains filtered or unexported fields
}

type FeatureJobMetric struct {
	// contains filtered or unexported fields
}

type IntFromOptionalStruct int

type JobCliFallbackFetcher struct {
	// contains filtered or unexported fields
}

type JobJsonFetcher struct {
	// contains filtered or unexported fields
}

type JobMetric struct {
	Account      string      `json:"account"`
	JobId        float64     `json:"job_id"`
	EndTime      float64     `json:"end_time"`
	JobState     string      `json:"job_state"`
	Partition    string      `json:"partition"`
	UserName     string      `json:"user_name"`
	Features     string      `json:"features"`
	JobResources JobResource `json:"job_resources"`
	StateReason  string      `json:"state_reason"`
}

type JobResource struct {
	AllocCpus  float64                  `json:"allocated_cpus"`
	AllocNodes map[string]*NodeResource `json:"allocated_nodes"`
}

type JobsCollector struct {
	// contains filtered or unexported fields
}

type LicCollector struct {
	// contains filtered or unexported fields
}

type LicenseMetric struct {
	LicenseName  string `json:"LicenseName"`
	Total        int    `json:"Total"`
	Used         int    `json:"Used"`
	Free         int    `json:"Free"`
	Remote       bool   `json:"Remote"`
	Reserved     int    `json:"Reserved"`
	LastConsumed int    `json:"LastConsumed"`
	LastDeficit  int    `json:"LastDeficit"`
}

type LimitCollector struct {
	// contains filtered or unexported fields
}

type MemSummaryMetric struct {
	AllocMemory float64
	FreeMemory  float64
	RealMemory  float64
}

type MessageRpcInfo struct {
	MessageType string                `json:"message_type"`
	TypeId      int                   `json:"type_id"`
	Count       int                   `json:"count"`
	AvgTime     IntFromOptionalStruct `json:"average_time"`
	TotalTime   int                   `json:"total_time"`
}

type MockFetchErrored struct{}

type MockScraper struct {
	CallCount int
	// contains filtered or unexported fields
}

type NAbleFloat float64

type NAbleTime struct{ time.Time }

type NodeCliFallbackFetcher struct {
	// contains filtered or unexported fields
}

type NodeJsonFetcher struct {
	// contains filtered or unexported fields
}

type NodeMetric struct {
	AllocMemory float64  `json:"alloc_memory"`
	AllocCpus   float64  `json:"alloc_cpus"`
	Cpus        float64  `json:"cpus"`
	CpuLoad     float64  `json:"cpu_load"`
	FreeMemory  float64  `json:"free_memory"`
	Hostname    string   `json:"hostname"`
	IdleCpus    float64  `json:"idle_cpus"`
	Partitions  []string `json:"partitions"`
	RealMemory  float64  `json:"real_memory"`
	State       string   `json:"state"`
	Weight      float64  `json:"weight"`
}

type NodeResource struct {
	Mem float64 `json:"memory"`
}

type NodesCollector struct {
	// contains filtered or unexported fields
}

type PartitionJobMetric struct {
	// contains filtered or unexported fields
}

type PartitionMetric struct {
	TotalCpus        float64
	RealMemory       float64
	FreeMemory       float64
	StateAllocMemory map[string]float64
	StateAllocCpus   map[string]float64
	StateNodeCount   map[string]float64
	CpuLoad          float64
	IdleCpus         float64
	Weight           float64
}

type PerStateMetric struct {
	Cpus  float64
	Count float64
}

type SdiagResponse struct {
	// Response coercible between slurm 23 and 24 data versions
	Meta struct {
		SlurmVersion SlurmVersion      `json:"Slurm"`
		Plugins      map[string]string `json:"plugins"`
		Plugin       map[string]string `json:"plugin"`
	} `json:"meta"`
	Statistics DiagMetric
	Errors     []string `json:"errors"`
	Warnings   []string `json:"warnings"`
}

type SlurmByteScraper interface {
	FetchRawBytes() ([]byte, error)
	Duration() time.Duration
}

type SlurmMetricFetcher[M SlurmPrimitiveMetric] interface {
	FetchMetrics() ([]M, error)
	ScrapeDuration() time.Duration
	ScrapeError() prometheus.Counter
}

type SlurmPrimitiveMetric interface {
	NodeMetric | JobMetric | DiagMetric | LicenseMetric | AccountLimitMetric
}

type SlurmVersion struct {
	Version struct {
		Major CoercedInt `json:"major"`
		Micro CoercedInt `json:"micro"`
		Minor CoercedInt `json:"minor"`
	} `json:"version"`
	Release string `json:"release"`
}

type StateReasonMetric struct {
	// contains filtered or unexported fields
}

type StringByteScraper struct {
	Callcount int
	// contains filtered or unexported fields
}

type TraceCollector struct {
	ProcessFetcher *AtomicProcFetcher
	// contains filtered or unexported fields
}

type TraceConfig struct {
	// contains filtered or unexported fields
}

type TraceInfo struct {
	JobId      int64   `json:"job_id"`
	Pid        int64   `json:"pid"`
	Cpus       float64 `json:"cpus"`
	WriteBytes float64 `json:"write_bytes"`
	ReadBytes  float64 `json:"read_bytes"`
	Threads    float64 `json:"threads"`
	Mem        float64 `json:"mem"`
	Username   string  `json:"username"`
	Hostname   string  `json:"hostname"`
	// contains filtered or unexported fields
}

type UserJobMetric struct {
	// contains filtered or unexported fields
}

type UserRpcInfo struct {
	User      string                `json:"user"`
	UserId    int                   `json:"user_id"`
	Count     int                   `json:"count"`
	AvgTime   IntFromOptionalStruct `json:"average_time"`
	TotalTime int                   `json:"total_time"`
}