diff options
Diffstat (limited to 'slog/counter.py')
-rw-r--r-- | slog/counter.py | 34 |
1 files changed, 34 insertions, 0 deletions
diff --git a/slog/counter.py b/slog/counter.py new file mode 100644 index 0000000..4760fe8 --- /dev/null +++ b/slog/counter.py @@ -0,0 +1,34 @@ +# Copyright 2015, 2016 Suomen Tilaajavastuu Oy +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as published by +# the Free Software Foundation, either version 3 of the License, or +# (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + + +import _thread + + +class Counter: + + '''A threadsafe incrementing counter.''' + + def __init__(self): + self._lock = _thread.allocate_lock() + self._counter = 0 + + def get(self): + return self._counter + + def increment(self): + with self._lock: + self._counter += 1 + return self._counter |