new: gpu status (#201)

2025-12-17 07:14:28 +01:00 · 2023-11-27 16:49:26 +08:00
parent ac94b33ad4
commit 05de760c41
11 changed files with 1126 additions and 65 deletions
--- a/lib/data/model/server/nvdia.dart
+++ b/lib/data/model/server/nvdia.dart
@@ -0,0 +1,132 @@
+import 'package:xml/xml.dart';
+
+/// [
+///   {
+///     "name": "GeForce RTX 3090",
+///     "temp": 40,
+///     "power": "30W / 350W",
+///     "memory": {
+///       "total": 24268,
+///       "used": 240,
+///       "unit": "MiB",
+///       "processes": [
+///         {
+///           "pid": 1456,
+///           "name": "/usr/lib/xorg/Xorg",
+///           "memory": 40
+///         },
+///       ]
+///     },
+///   }
+/// ]
+///
+
+class NvdiaSmi {
+  static List<NvdiaSmiItem> fromXml(String raw) {
+    final xmlData = XmlDocument.parse(raw);
+    final gpus = xmlData.findAllElements('gpu');
+    final result = List<NvdiaSmiItem?>.generate(gpus.length, (index) {
+      final gpu = gpus.elementAt(index);
+      final name = gpu.findElements('product_name').firstOrNull?.innerText;
+      final temp = gpu
+          .findElements('temperature')
+          .firstOrNull
+          ?.findElements('gpu_temp')
+          .firstOrNull
+          ?.innerText;
+      final power = gpu.findElements('gpu_power_readings').firstOrNull;
+      final powerDraw =
+          power?.findElements('power_draw').firstOrNull?.innerText;
+      final powerLimit =
+          power?.findElements('current_power_limit').firstOrNull?.innerText;
+      final memory = gpu.findElements('fb_memory_usage').firstOrNull;
+      final memoryUsed = memory?.findElements('used').firstOrNull?.innerText;
+      final memoryTotal = memory?.findElements('total').firstOrNull?.innerText;
+      final processes = gpu
+          .findElements('processes')
+          .firstOrNull
+          ?.findElements('process_info');
+      final memoryProcesses =
+          List<NvdiaSmiMemProcess?>.generate(processes?.length ?? 0, (index) {
+        final process = processes?.elementAt(index);
+        final pid = process?.findElements('pid').firstOrNull?.innerText;
+        final name =
+            process?.findElements('process_name').firstOrNull?.innerText;
+        final memory =
+            process?.findElements('used_memory').firstOrNull?.innerText;
+        if (pid != null && name != null && memory != null) {
+          return NvdiaSmiMemProcess(
+            int.parse(pid),
+            name,
+            int.parse(
+              memory.split(' ').firstOrNull ?? '0',
+            ),
+          );
+        }
+        return null;
+      });
+      memoryProcesses.removeWhere((element) => element == null);
+      if (name != null &&
+          temp != null &&
+          powerDraw != null &&
+          powerLimit != null &&
+          memory != null) {
+        return NvdiaSmiItem(
+          name,
+          int.parse(temp.split(' ').firstOrNull ?? '0'),
+          '$powerDraw / $powerLimit',
+          NvdiaSmiMem(
+            int.parse(memoryTotal?.split(' ').firstOrNull ?? '0'),
+            int.parse(memoryUsed?.split(' ').firstOrNull ?? '0'),
+            'MiB',
+            List.from(memoryProcesses),
+          ),
+        );
+      }
+      return null;
+    });
+    result.removeWhere((element) => element == null);
+    return List.from(result);
+  }
+}
+
+class NvdiaSmiItem {
+  final String name;
+  final int temp;
+  final String power;
+  final NvdiaSmiMem memory;
+
+  const NvdiaSmiItem(this.name, this.temp, this.power, this.memory);
+
+  @override
+  String toString() {
+    return 'NvdiaSmiItem{name: $name, temp: $temp, power: $power, memory: $memory}';
+  }
+}
+
+class NvdiaSmiMem {
+  final int total;
+  final int used;
+  final String unit;
+  final List<NvdiaSmiMemProcess> processes;
+
+  const NvdiaSmiMem(this.total, this.used, this.unit, this.processes);
+
+  @override
+  String toString() {
+    return 'NvdiaSmiMem{total: $total, used: $used, unit: $unit, processes: $processes}';
+  }
+}
+
+class NvdiaSmiMemProcess {
+  final int pid;
+  final String name;
+  final int memory;
+
+  const NvdiaSmiMemProcess(this.pid, this.name, this.memory);
+
+  @override
+  String toString() {
+    return 'NvdiaSmiMemProcess{pid: $pid, name: $name, memory: $memory}';
+  }
+}