From 1fe20f7ec098ed43ad0083a608521f09197a6341 Mon Sep 17 00:00:00 2001
From: beichen <43266206+BeichenDream@users.noreply.github.com>
Date: Sat, 2 Dec 2023 00:14:41 +0800
Subject: [PATCH] fix error get cgroup

---
 toa.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/toa.py b/toa.py
index c44d6a9..23bfa2f 100644
--- a/toa.py
+++ b/toa.py
@@ -27,13 +27,16 @@ class UserOption:
         return struct.pack('=B I H', self.toa_kind, self.toa_tcp_host, self.toa_tcp_port)
 
 def get_current_cgroup():
+    if os.path.exists('/sys/fs/cgroup/unified/'):
+        return '/sys/fs/cgroup/unified'
+
     with open('/proc/self/cgroup', 'r') as file:
         lines = file.readlines()
-    
+
     cgroup_info = []
     for line in lines:
         cgroup_info = line.strip().split(':')
-    
+
     return '/sys/fs/cgroup/' + cgroup_info[2][0:cgroup_info[2].index('/')]
 
 def execute_command(cmd):