forked from uncloud/uncloud
21 lines
436 B
Python
21 lines
436 B
Python
from .etcd_wrapper import Etcd3Wrapper
|
|
|
|
|
|
def increment_etcd_counter(etcd_client: Etcd3Wrapper, key):
|
|
kv = etcd_client.get(key)
|
|
|
|
if kv:
|
|
counter = int(kv.value)
|
|
counter = counter + 1
|
|
else:
|
|
counter = 1
|
|
|
|
etcd_client.put(key, str(counter))
|
|
return counter
|
|
|
|
|
|
def get_etcd_counter(etcd_client: Etcd3Wrapper, key):
|
|
kv = etcd_client.get(key)
|
|
if kv:
|
|
return int(kv.value)
|
|
return None
|