alu/cs239-caching
1
0
Fork 0
mirror of https://github.com/ltcptgeneral/cs239-caching.git synced 2025-09-08 16:27:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed POST requests in system

Browse Source
This commit is contained in:
Derek Wang
2025-03-03 19:28:15 -08:00
parent 8430009f8c
commit 32ac22806c
4 changed files with 40 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
tests/varied_workload.py
View File

@@ -30,12 +30,13 @@ def generate_request(workload, last_updated=None):
"""Generate read or write requests based on workload type"""
if random.random() < workload["read"]:
user_id = select_user(workload, last_updated)
return baseurl + f"/user/{user_id}", "GET"
return baseurl + f"/user/{user_id}", None, "GET"
# Write operation (updates user profile)
user_id = select_user(workload, last_updated)
url = baseurl + f"/update_user/?user_id={user_id}&name=UpdatedUser&followers=500&bio=Updated&posts=UpdatedPost"
return url, "POST"
write_obj = { "user_id":user_id,"name": "UpdatedUser", "followers":"500","bio":"Updated","posts":"UpdatedPost"}
url = baseurl + f"/update_user/"
return url, write_obj, "POST"
def select_user(workload, last_updated):
"""Selects a user based on workload type"""
@@ -55,8 +56,12 @@ def run_workload(name, workload):
last_updated = None
for _ in tqdm(range(10000), desc=f"Running {name}"):
url, method = generate_request(workload, last_updated)
response = requests.request(method, url)
url, data, method = generate_request(workload, last_updated)
if( method == "GET" ):
response = requests.request(method, url)
else:
response = requests.post(url, json = data)
try:
content = json.loads(response.content)