implement multithread

Author: anmolnagpal

README.md

@@ -82,9 +82,10 @@ module "cloudtrail-slack-notification" {
   variables = {
     Website_URL = jsonencode(["https://google.com"]),
     metricname  = "Site Availability"
+    timeout = 5
   }
   slack_variables = {
-    SLACK_WEBHOOK = "https://hooks.slack.com/services/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r1Y"
+    SLACK_WEBHOOK = "https://hooks.slack.com/services/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yrrh08dJo5r1Y"
   }
 }
 ```

README.yaml

@@ -46,9 +46,10 @@ usage : |-
     variables = {
       Website_URL = jsonencode(["https://google.com"]),
       metricname  = "Site Availability"
+      timeout = 5
     }
     slack_variables = {
-      SLACK_WEBHOOK = "https://hooks.slack.com/services/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r1Y"
+      SLACK_WEBHOOK = "https://hooks.slack.com/services/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yrrh08dJo5r1Y"
     }
   }
   ```

_example/example.tf

@@ -14,9 +14,10 @@ module "site-monitor" {
   variables = {
     Website_URL = jsonencode(["https://google.com"]),
     metricname  = "Site Availability"
+    timeout     = 5
   }
   slack_variables = {
-    SLACK_WEBHOOK = "https://hooks.slack.com/services/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r1Y"
+    SLACK_WEBHOOK = "https://hooks.slack.com/services/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yrrh08dJo5r1Y"
   }
 }
 
@@ -40,7 +41,7 @@ module "alarm" {
   alarm_actions       = [module.site-monitor.sns_arn]
 
   actions_enabled           = true
-  insufficient_data_actions = []
+  insufficient_data_actions = [module.site-monitor.sns_arn]
   ok_actions                = [module.site-monitor.sns_arn]
   dimensions = {
     Website = "https://google.com",

monitor/src/index.py

@@ -4,6 +4,7 @@
 import urllib.request as urllib2
 import os
 import json
+import threading
 
 website_url           = json.loads(os.environ['Website_URL'])
 metricname            = os.environ['metricname']
@@ -36,17 +37,16 @@ def check_site(url, metric):
 	STAT = 1
 	print("Checking %s " % url)
 	request = urllib2.Request(url)
-
 	try:
-		response = urllib2.urlopen(request, timeout=timeout)
+		response = urllib2.urlopen(request,timeout=timeout)
 		response.close()
 	except urllib2.URLError as e:
 		if hasattr(e, 'code'):
 			print("1st if")
 			print ("[Error:] Connection to %s failed with code: " %url +str(e.code))
 			STAT = int(e.code)
 			# write_metric(STAT, metric, url)
-		if hasattr(e, 'reason'):
+		elif hasattr(e, 'reason'):
 			print("2nd if")
 			print ("[Error:] Connection to %s failed with code: " % url +str(e.reason))
 			STAT = 501
@@ -57,27 +57,36 @@ def check_site(url, metric):
 			print ("[Error:] Connection to %s failed with code: " % url + str(e.code))
 			STAT = int(e.code)
 			# write_metric(STAT, metric, url)
-		if hasattr(e, 'reason'):
+		elif hasattr(e, 'reason'):
 			print("4th if")
 			print ("[Error:] Connection to %s failed with code: " % url + str(e.reason))
 			STAT = 501
 			# write_metric(STAT, metric, url)
-	if STAT != 501:
+	print(STAT)
+	if STAT != 501 and STAT ==1:
 		STAT = response.getcode()
 
 	return STAT
 
+def run_thread(site):
+  	r = check_site(site,metricname)
+  	if r == 200 or r == 304 or r == 400:
+  		print("Site %s is up" %site)
+  		write_metric(200, metricname, site)
+  	else:
+  		print("[Error:] Site %s down" %site)
+  		write_metric(int(r), metricname, site)
+
 def handler(event, context):
 
 	# Change these to your actual websites.  Remember, the more websites you list
         # the longer the lambda function will run
 	websiteurls = website_url
+	t = [0]*len(website_url)
+	for i in range(len(websiteurls)):
+		t[i] = threading.Thread(target=run_thread, args=(website_url[i],))
+		t[i].start()
 
-	for site in websiteurls:
-		r = check_site(site,metricname)
-		if r == 200 or r == 304 or r == 400:
-			print("Site %s is up" %site)
-			write_metric(200, metricname, site)
-		else:
-			print("[Error:] Site %s down" %site)
-			write_metric(int(r), metricname, site)
+	for i in range(len(t)):
+		t[i].join()
+	print("Done!")

slack/src/index.js

@@ -55,8 +55,10 @@ var handleCloudWatch = function(event, context) {
 
   if (message.NewStateValue === "ALARM") {
       color = "danger";
+      message = "Monitor: endpoint is down (!200)";
   } else if (message.NewStateValue === "OK") {
       color = "good";
+      message = "Monitor: endpoint is up (200)";
   }
 
   var slackMessage = {
@@ -68,6 +70,7 @@ var handleCloudWatch = function(event, context) {
           { "title": "URL", "value": resource, "short": true },
           { "title": "status", "value": newState, "short": true },
           { "title": "Time", "value": new Date(event.Records[0].Sns.Timestamp), "short": true},
+          { "title": "Message", "value": message, "short": true},
           {
             "title": "Link to Alarm",
             "value": "https://console.aws.amazon.com/cloudwatch/home?region=" + region + "#alarm:alarmFilter=ANY;name=" + encodeURIComponent(alarmName),