summaryrefslogtreecommitdiffstats
path: root/src/go/plugin/go.d/modules/intelgpu/intelgpu.go
diff options
context:
space:
mode:
Diffstat (limited to 'src/go/plugin/go.d/modules/intelgpu/intelgpu.go')
-rw-r--r--src/go/plugin/go.d/modules/intelgpu/intelgpu.go108
1 files changed, 108 insertions, 0 deletions
diff --git a/src/go/plugin/go.d/modules/intelgpu/intelgpu.go b/src/go/plugin/go.d/modules/intelgpu/intelgpu.go
new file mode 100644
index 000000000..8e98c688d
--- /dev/null
+++ b/src/go/plugin/go.d/modules/intelgpu/intelgpu.go
@@ -0,0 +1,108 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+package intelgpu
+
+import (
+ _ "embed"
+ "errors"
+
+ "github.com/netdata/netdata/go/plugins/plugin/go.d/agent/module"
+)
+
+//go:embed "config_schema.json"
+var configSchema string
+
+func init() {
+ module.Register("intelgpu", module.Creator{
+ JobConfigSchema: configSchema,
+ Create: func() module.Module { return New() },
+ Config: func() any { return &Config{} },
+ })
+}
+
+func New() *IntelGPU {
+ return &IntelGPU{
+ ndsudoName: "ndsudo",
+ charts: charts.Copy(),
+ engines: make(map[string]bool),
+ }
+}
+
+type Config struct {
+ UpdateEvery int `yaml:"update_every,omitempty" json:"update_every"`
+ Device string `yaml:"device,omitempty" json:"device"`
+}
+
+type (
+ IntelGPU struct {
+ module.Base
+ Config `yaml:",inline" json:""`
+
+ charts *module.Charts
+
+ exec intelGpuTop
+ ndsudoName string
+
+ engines map[string]bool
+ }
+ intelGpuTop interface {
+ queryGPUSummaryJson() ([]byte, error)
+ stop() error
+ }
+)
+
+func (ig *IntelGPU) Configuration() any {
+ return ig.Config
+}
+
+func (ig *IntelGPU) Init() error {
+ topExec, err := ig.initIntelGPUTopExec()
+ if err != nil {
+ ig.Error(err)
+ return err
+ }
+
+ ig.exec = topExec
+
+ return nil
+}
+
+func (ig *IntelGPU) Check() error {
+ mx, err := ig.collect()
+ if err != nil {
+ ig.Error(err)
+ return err
+ }
+
+ if len(mx) == 0 {
+ return errors.New("no metrics collected")
+ }
+
+ return nil
+}
+
+func (ig *IntelGPU) Charts() *module.Charts {
+ return ig.charts
+}
+
+func (ig *IntelGPU) Collect() map[string]int64 {
+ mx, err := ig.collect()
+ if err != nil {
+ ig.Error(err)
+ }
+
+ if len(mx) == 0 {
+ return nil
+ }
+
+ return mx
+}
+
+func (ig *IntelGPU) Cleanup() {
+ if ig.exec != nil {
+ if err := ig.exec.stop(); err != nil {
+ ig.Error(err)
+ }
+ ig.exec = nil
+ }
+}