From 54c41c1cd4c5820870bba5976bdcde2b2e08248d Mon Sep 17 00:00:00 2001
From: Akizon77 <i@akz.moe>
Date: Thu, 28 Aug 2025 12:45:27 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20=E5=A2=9E=E5=BC=BA=E5=AE=B9=E5=99=A8?=
 =?UTF-8?q?=E6=A3=80=E6=B5=8B?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Dockerfile                        | 2 ++
 monitoring/unit/virtualization.go | 4 +++-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/Dockerfile b/Dockerfile
index 14cf00a..a80d0b8 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -10,6 +10,8 @@ COPY komari-agent-${TARGETOS}-${TARGETARCH} /app/komari-agent
 
 RUN chmod +x /app/komari-agent
 
+RUN touch /.komari-agent-container
+
 ENTRYPOINT ["/app/komari-agent"]
 # 运行时请指定参数
 # Please specify parameters at runtime.
diff --git a/monitoring/unit/virtualization.go b/monitoring/unit/virtualization.go
index a653477..32d6f14 100644
--- a/monitoring/unit/virtualization.go
+++ b/monitoring/unit/virtualization.go
@@ -84,7 +84,9 @@ func detectContainer() string {
 	if s := parseCgroupForContainer(); s != "" {
 		return s
 	}
-
+	if fileExists("/.komari-agent-container") {
+		return "container"
+	}
 	// (Removed mountinfo heuristics which caused host false positives when Docker/Kube tools are installed.)
 	return ""
 }