From 96cedf88f309ecb503cabfdd1ae4eac6544a93ad Mon Sep 17 00:00:00 2001
From: Jonny Ervine <jonny@ervine.org>
Date: Tue, 15 Dec 2020 02:57:02 +0000
Subject: [PATCH] Add main.py

Initial commit of code
---
 main.py | 46 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)
 create mode 100644 main.py

diff --git a/main.py b/main.py
new file mode 100644
index 0000000..c80e065
--- /dev/null
+++ b/main.py
@@ -0,0 +1,46 @@
+import os, sys, json, base64, re;
+from kubernetes import client, config
+
+def handler(event, context):
+    if 'data' in event:
+        message = json.loads(base64.b64decode(event['data']).decode('utf-8'))
+        print('Data found - for debugging purposes, message contents printed in next line')
+        print(message)
+    else:
+        print('Data not found - this is an unexpected error, and the function has been called without a Pub/Sub message. This should be checked')
+        return f'Data not found'
+    if 'pod_name' in message['incident']['resource']['labels']:
+        namespace = message["incident"]['resource']['labels']['namespace_name']
+        pod = message["incident"]['resource']['labels']['pod_name']
+        cluster = message['incident']['resource']['labels']['cluster_name']        
+        print("Deleting " + pod + " in namespace " + namespace + " in cluster " + cluster + " because it has utilised more than 90% of the configured memory limit")
+    elif "deadlock_detected" in message['incident']['metric']['type']:
+        namespace = namespace = message["incident"]['resource']['labels']['namespace_name']
+        sum = message['incident']['summary']
+        extract = sum.split("pod_name=", 1)[-1]
+        pod = re.findall("[\w-]*", extract)[0]
+        cluster = message['incident']['resource']['labels']['cluster_name']
+        print("Deleting " + pod + " in namespace " + namespace + " in cluster " + cluster + " because it has encountered a deadlock condition")
+    elif "request-timeout" in message['incident']['metric']['type']:
+        namespace = namespace = message["incident"]['resource']['labels']['namespace_name']
+        sum = message['incident']['summary']
+        extract = sum.split("pod_name=", 1)[-1]
+        pod = re.findall("[\w-]*", extract)[0]
+        cluster = message['incident']['resource']['labels']['cluster_name']
+        print("Deleting " + pod + " in namespace " + namespace + " in cluster " + cluster + " because it has encountered a frequent request-timeout condition")
+    else:
+        print('From unknown source')
+
+    if 'KUBERNETES_LOAD_KUBE_CONFIG' in os.environ:
+        if cluster == 'ggv-k8s-prod':
+            config.load_kube_config(context='gke_api-project-1004752207173_us-east4_ggv-k8s-prod', persist_config=False)
+        if cluster == 'ggv-k8s-cluster':
+            config.load_kube_config(context='gke_api-project-1004752207173_asia-southeast1_ggv-k8s-cluster', persist_config=False)
+        if cluster == 'ggv-k8s-production-sg':
+            config.load_kube_config(context='gke_api-project-1004752207173_asia-southeast1_ggv-k8s-production-sg', persist_config=False)
+    else:
+        config.load_incluster_config()
+    v1 = client.CoreV1Api()
+    body = client.V1beta1Eviction(metadata=client.V1ObjectMeta(name=pod, namespace=namespace))
+    api_response = v1.create_namespaced_pod_eviction(name=pod, namespace=namespace, body=body)
+    print('Evicted pod?')
\ No newline at end of file