149 lines
4.9 KiB
Python
149 lines
4.9 KiB
Python
import time
|
|
import random
|
|
import string
|
|
|
|
|
|
def hash_int(x: int):
|
|
x = ((x >> 16) ^ x) * 0x45D9F3B
|
|
x = ((x >> 16) ^ x) * 0x45D9F3B
|
|
x = (x >> 16) ^ x
|
|
return x
|
|
|
|
|
|
def generate_mock_pod(index: int, i: int, j: int):
|
|
names = [
|
|
"agent-cooper",
|
|
"black-lodge",
|
|
"bob",
|
|
"bobby-briggs",
|
|
"laura-palmer",
|
|
"leland-palmer",
|
|
"log-lady",
|
|
"sheriff-truman",
|
|
]
|
|
labels = {"env": ["prod", "dev"], "owner": ["x-wing", "iris"]}
|
|
pod_phases = ["Pending", "Running", "Running", "Failed"]
|
|
|
|
pod_labels = {}
|
|
for li, k in enumerate(labels):
|
|
v = labels[k]
|
|
label_choice = hash_int((index + 1) * (i + 1) * (j + 1) * (li + 1)) % (
|
|
len(v) + 1
|
|
)
|
|
if label_choice != 0:
|
|
pod_labels[k] = v[label_choice - 1]
|
|
|
|
phase = pod_phases[hash_int((index + 1) * (i + 1) * (j + 1)) % len(pod_phases)]
|
|
containers = []
|
|
for k in range(1 + j % 2):
|
|
# generate "more real data"
|
|
requests_cpu = random.randint(10, 50)
|
|
requests_memory = random.randint(64, 256)
|
|
# with max, we defend ourselves against negative cpu/memory ;)
|
|
usage_cpu = max(requests_cpu + random.randint(-30, 30), 1)
|
|
usage_memory = max(requests_memory + random.randint(-64, 128), 1)
|
|
container = {
|
|
"name": "myapp",
|
|
"image": "foo/bar/{}".format(j),
|
|
"resources": {
|
|
"requests": {
|
|
"cpu": f"{requests_cpu}m",
|
|
"memory": f"{requests_memory}Mi",
|
|
},
|
|
"limits": {},
|
|
"usage": {"cpu": f"{usage_cpu}m", "memory": f"{usage_memory}Mi"},
|
|
},
|
|
"ready": True,
|
|
"state": {"running": {}},
|
|
}
|
|
if phase == "Running":
|
|
if j % 13 == 0:
|
|
container.update(
|
|
**{
|
|
"ready": False,
|
|
"state": {"waiting": {"reason": "CrashLoopBackOff"}},
|
|
}
|
|
)
|
|
elif j % 7 == 0:
|
|
container.update(
|
|
**{"ready": False, "state": {"running": {}}, "restartCount": 3}
|
|
)
|
|
elif phase == "Failed":
|
|
del container["state"]
|
|
del container["ready"]
|
|
containers.append(container)
|
|
pod = {
|
|
"name": "{}-{}-{}".format(
|
|
names[hash_int((i + 1) * (j + 1)) % len(names)], i, j
|
|
),
|
|
"namespace": "kube-system" if j < 3 else "default",
|
|
"labels": pod_labels,
|
|
"phase": phase,
|
|
"containers": containers,
|
|
}
|
|
if phase == "Running" and j % 17 == 0:
|
|
pod["deleted"] = 123
|
|
|
|
return pod
|
|
|
|
|
|
def query_mock_cluster(cluster):
|
|
"""Generate deterministic (no randomness!) mock data"""
|
|
index = int(cluster.id.split("-")[-1])
|
|
nodes = {}
|
|
for i in range(10):
|
|
# add/remove the second to last node every 13 seconds
|
|
if i == 8 and int(time.time() / 13) % 2 == 0:
|
|
continue
|
|
labels = {}
|
|
# only the first two clusters have master nodes
|
|
if i < 2 and index < 2:
|
|
if index == 0:
|
|
labels["kubernetes.io/role"] = "master"
|
|
elif index == 1:
|
|
labels["node-role.kubernetes.io/master"] = ""
|
|
else:
|
|
labels["master"] = "true"
|
|
pods = {}
|
|
for j in range(hash_int((index + 1) * (i + 1)) % 32):
|
|
# add/remove some pods every 7 seconds
|
|
if j % 17 == 0 and int(time.time() / 7) % 2 == 0:
|
|
pass
|
|
else:
|
|
pod = generate_mock_pod(index, i, j)
|
|
pods["{}/{}".format(pod["namespace"], pod["name"])] = pod
|
|
|
|
# use data from containers (usage)
|
|
usage_cpu = 0
|
|
usage_memory = 0
|
|
for p in pods.values():
|
|
for c in p["containers"]:
|
|
usage_cpu += int(c["resources"]["usage"]["cpu"].split("m")[0])
|
|
usage_memory += int(c["resources"]["usage"]["memory"].split("Mi")[0])
|
|
|
|
# generate longer name for a node
|
|
suffix = "".join(
|
|
[random.choice(string.ascii_letters) for n in range(random.randint(1, 20))]
|
|
)
|
|
|
|
node = {
|
|
"name": f"node-{i}-{suffix}",
|
|
"labels": labels,
|
|
"status": {
|
|
"capacity": {"cpu": "8", "memory": "64Gi", "pods": "110"},
|
|
"allocatable": {"cpu": "7800m", "memory": "62Gi"},
|
|
},
|
|
"pods": pods,
|
|
# get data from containers (usage)
|
|
"usage": {"cpu": f"{usage_cpu}m", "memory": f"{usage_memory}Mi"},
|
|
}
|
|
nodes[node["name"]] = node
|
|
pod = generate_mock_pod(index, 11, index)
|
|
unassigned_pods = {"{}/{}".format(pod["namespace"], pod["name"]): pod}
|
|
return {
|
|
"id": "mock-cluster-{}".format(index),
|
|
"api_server_url": "https://kube-{}.example.org".format(index),
|
|
"nodes": nodes,
|
|
"unassigned_pods": unassigned_pods,
|
|
}
|