HistoryPurge: Clearing 219 old commits

2024-10-24 23:09:21 +08:00
commit d08b7cac59
348 changed files with 376141 additions and 0 deletions
@@ -0,0 +1,65 @@
+import json
+import socket
+import boto3
+import logging
+
+logger = logging.getLogger()
+logger.setLevel(logging.INFO)
+
+def lambda_handler(event, context):
+    hosts=['10.129.72.63', '10.135.72.66', '10.129.72.64', '10.135.72.67']
+    port=636
+    timeout_seconds=3
+    test_results = 0
+    metric_value = 0
+    for host in hosts:
+        sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+        sock.settimeout(timeout_seconds)
+        result = sock.connect_ex((host,int(port)))
+        
+        if result == 0:
+            logger.info("Host {}:{} - Up".format(host, port))
+            test_results += 1
+            metric_value = 1
+        else:
+            logger.error("Host {}:{} - Down".format(host, port))
+            metric_value = 0
+        sock.close()
+        
+        logger.info("Publishing cloudwatch metric...")
+        cloudwatch = boto3.client('cloudwatch')
+        try:
+            response = cloudwatch.put_metric_data(
+                MetricData=[
+                    {
+                        'MetricName': 'Ldap Tcp Test',
+                        'Dimensions': [
+                            {
+                                'Name': 'LdapHost',
+                                'Value': host
+                            }
+                        ],
+                        'Unit': 'None',
+                        'Value': metric_value
+                    },
+                ],
+                Namespace='Custom/Lambda'
+            )
+            logger.info("Successfully published cloudwatch metric")
+            # logger.info(response)
+        except Exception as e:
+            logger.error("Error publishing cloudwatch metric: {}".format(str(e)))
+
+        
+    if test_results == 4:
+        return {
+            'message' : 'Successfully connected to all LDAP servers' 
+        }
+    else:
+        raise Exception('Not all LDAP servers can be connected!')
+
+    return {
+        'statusCode': 200,
+        'body': json.dumps("Finished")
+    }
+