Skip to content
Snippets Groups Projects
Commit 9e506f39 authored by Tim Steiner's avatar Tim Steiner
Browse files

Fix varnish cloudwatch, add IMDSv2 tokens.

parent 0a0f93a8
Branches
No related tags found
No related merge requests found
#!/usr/bin/python #!/usr/bin/python
import boto3 import boto3
import urllib2 #import urllib2
import json import json
import requests
import sys import sys
#import re #import re
#import tempfile #import tempfile
...@@ -25,7 +26,7 @@ def main(): ...@@ -25,7 +26,7 @@ def main():
previous_data = None previous_data = None
metrics = {} metrics = {}
while True: while True:
current_data = json.loads(subprocess.check_output(['varnishstat', '-j', '-1', '-f', 'MAIN.client_req', '-f', 'MAIN.cache_hit'])) current_data = json.loads(subprocess.check_output(['varnishstat', '-j', '-1', '-f', 'MAIN.client_req', '-f', 'MAIN.cache_hit']))['counters']
if previous_data: if previous_data:
metric_data = [] metric_data = []
for key in current_data: for key in current_data:
...@@ -56,8 +57,10 @@ def main(): ...@@ -56,8 +57,10 @@ def main():
# Returns the region the instance is currently running in (eg: 'us-east') # Returns the region the instance is currently running in (eg: 'us-east')
def get_region(): def get_region():
instance_info = urllib2.urlopen('http://169.254.169.254/2018-09-24/dynamic/instance-identity/document/').read() session_token = requests.put('http://169.254.169.254/latest/api/token', headers={'X-aws-ec2-metadata-token-ttl-seconds': 60}).text
instance_info = json.loads(instance_info) instance_info = requests.get('http://169.254.169.254/2018-09-24/dynamic/instance-identity/document/', headers={'X-aws-ec2-metadata-token': session_token}).json()
#instance_info = urllib2.urlopen('http://169.254.169.254/2018-09-24/dynamic/instance-identity/document/').read()
#instance_info = json.loads(instance_info)
return instance_info['region'] return instance_info['region']
# Returns the instance_id of this VM. # Returns the instance_id of this VM.
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment