diff --git a/sjrpc/utils/datastructures.py b/sjrpc/utils/datastructures.py index 2f62e34eee4eade95ea9afa8940c2339ed7a46de..d2fbcd2123a2225034fdeb63a88a4e06621f77a6 100644 --- a/sjrpc/utils/datastructures.py +++ b/sjrpc/utils/datastructures.py @@ -63,3 +63,9 @@ class BytesBuffer(object): ''' return len(self._buf) + + def __enter__(self): + return self._lock.__enter__() + + def __exit__(self): + return self._lock.__exit__()