Merge pull request #342 from KantaTamura/profile-exit-context

fix a bug that can't profiling of context close (io.close)

pfio/v2/local.py

@@ -8,7 +8,7 @@
 from .fs import FS, FileStat, format_repr
 
 
-class LocalProfileWrapper:
+class LocalProfileIOWrapper:
     def __init__(self, fp):
         self.fp = fp
 
@@ -17,7 +17,8 @@ def __enter__(self):
         return self
 
     def __exit__(self, exc_type, exc_value, traceback):
-        self.fp.__exit__(exc_type, exc_value, traceback)
+        with record("pfio.v2.Local:exit-context", trace=True):
+            self.fp.__exit__(exc_type, exc_value, traceback)
 
     def __getattr__(self, name):
         attr = getattr(self.fp, name)
@@ -121,7 +122,7 @@ def open(self, file_path, mode='r',
 
             # Add ppe recorder to io class methods (e.g. read, write)
             if self.trace:
-                return LocalProfileWrapper(fp)
+                return LocalProfileIOWrapper(fp)
             else:
                 return fp
 

tests/v2_tests/test_local.py

@@ -307,6 +307,7 @@ def test_local_rw_profiling():
 
         assert "pfio.v2.Local:open" in keys
         assert "pfio.v2.Local:write" in keys
+        assert "pfio.v2.Local:exit-context" in keys
 
     with Local(trace=True) as fs:
         ppe.profiler.clear_tracer()
@@ -320,6 +321,7 @@ def test_local_rw_profiling():
 
         assert "pfio.v2.Local:open" in keys
         assert "pfio.v2.Local:read" in keys
+        assert "pfio.v2.Local:exit-context" in keys
 
     # no profile case
     with Local(trace=False) as fs: