mCaptcha/dcache
2
0
Fork 0
Code Issues 1 Pull requests 10 Projects Releases Packages Wiki Activity

feat: post to leadere

Browse source
This commit is contained in:
Aravinth Manivannan 2023-05-27 10:29:05 +05:30
parent 9c0a85915c
commit 6ba293d287
Signed by: realaravinth
GPG key ID: AD9F0F08E855ED88
1 changed files with 27 additions and 16 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

43
test.py
View file

@ -1,17 +1,17 @@
#!/bin/env /bin/python
# mCaptcha - A proof of work based DoS protection system
# Copyright © 2023 Aravinth Manivannan <realravinth@batsense.net>
#
#
# 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/>.
@ -44,11 +44,12 @@ def metrics(host: str):
def write(host, data):
resp = requests.post(f"http://{host}/write", json=data)
print(f"RPC Status: {resp.status_code}")
data = resp.json()
if "Err" in data:
leader = data["Err"]["APIError"]["ForwardToLeader"]["leader_node"]["addr"]
write(leader, data)
return data['Ok']['data']
resp = resp.json()
if "Err" in resp:
leader = resp["Err"]["APIError"]["ForwardToLeader"]["leader_node"]["addr"]
print(f"Forwarding write to leader {leader}")
return write(leader, data)
return resp["Ok"]["data"]
def add_vote(host: str, captcha_id: str):
@ -77,14 +78,24 @@ def add_captcha(host: str, captcha_id: str):
pprint(f"Captcha added {captcha_id}: {resp}")
if __name__ == "__main__":
host = "localhost:9001"
# init(host)
# add_host(host, id=2, peer="localhost:9002")
# switch_to_cluster(host, nodes=[1, 2])
# metrics(host)
captcha_id = "test_1"
host = "localhost:9001"
peers = [(2, "localhost:9002"), (3, "localhost:9003"), (4, "localhost:9004")]
captcha_id = "test_1"
def initialize_cluster():
init(host)
for peer_id, peer in peers:
add_host(host=host, id=peer_id, peer=peer)
switch_to_cluster(host, nodes=[1, 2,3,4])
add_captcha(host, captcha_id)
add_vote(host, captcha_id)
for _ in range(0,600):
for _ in range(0, 600):
add_vote(host, captcha_id)
if __name__ == "__main__":
add_vote("localhost:9002", captcha_id)