ThreadSafeCounter python

ThreadSafeCounter.py

# -*- coding: utf-8 -*-

import threading


class ThreadSafeCounter:
    def __init__(self, counter=0, max_counter=0):
        self.lock = threading.Lock()
        self.counter = counter
        self.max_counter = max_counter

    def get(self):
        return self.counter

    def set(self, count):
        self.counter = count

    def incr(self, count=1):
        with self.lock:
            self.counter += count

    def get_max(self):
        return self.max_counter

    def set_max(self, count):
        self.max_counter = count