forked from uncloud/uncloud
22 lines
436 B
Python
22 lines
436 B
Python
|
from etcd3_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
|