Merge pull request #9 from mathieupassenaud/add-list-metrics

Add list metrics

Author: swallez

Bash/README.md

@@ -49,3 +49,9 @@ Or send key:value list with `kv_to_put_json` function:
             "tags":{}
         }]
     }' | delete
+
+### List Metrics
+
+listMetrics
+
+returns all metrics for the current application

Bash/lib/iot-lib.sh

@@ -44,9 +44,19 @@ delete() {
        $OPTS
 }
 
+# Get metrics
+#
+# @param none
+#
+listMetrics() {
+     curl -s \
+       -u $READ_TOKEN_ID:$READ_TOKEN_KEY \
+       -XGET "$IOT_API/api/suggest?type=metrics" \
+       $OPTS
+}
+
 # Utils
 
-# Transform a key/value list in JSON 'OpenTSDB put' data.
 # (http://opentsdb.net/docs/build/html/api_http/put.html)
 #
 # @param tag    optional JSON object for the tags (example: {"host": localhost})

Bash/list-metrics

@@ -0,0 +1,10 @@
+#!/bin/bash
+set -euo pipefail
+
+source lib/iot-lib.sh
+
+#
+# @return a JSON object representing an OpenTSDB suggest metrics query
+#
+
+listMetrics

Python/Metric.py

@@ -0,0 +1,28 @@
+# -*- coding: utf8 -*-
+
+import json
+from json import JSONEncoder
+
+#  
+# A metric for an OpenTSDB container
+#
+class Metric:
+
+    metric = ""
+    value = 0.0
+    timestamp = 0
+    tags = {}
+    
+    def __init__(self, metricName, value, timestamp, tags):
+        self.metric = metricName
+        self.value = value
+        self.timestamp = timestamp
+        self.tags = tags
+
+    def getAsJsonString(self):
+        return MetricEncoder().encode(self)
+        #return json.dumps(self)
+
+class MetricEncoder(JSONEncoder):
+    def default(self, o):
+        return o.__dict__

Python/OpenTSDBProfile.py

@@ -0,0 +1,21 @@
+# -*- coding: utf8 -*-
+
+#
+# Your OpenTSDB Account
+#
+# Only use the default constructor with your tokens
+# and server URI
+#
+
+class OpenTSDBProfile:
+
+    token_id = ""
+    token_password = ""
+    url = ""
+    port = 0
+
+    def __init__(self, token_id, token_password, url, port):
+        self.token_id = token_id
+        self.token_password = token_password
+        self.url = url
+        self.port = port

Python/OpenTSDBPusherHttp.py

@@ -0,0 +1,39 @@
+# -*- coding: utf8 -*-
+
+###############################################################################################
+#                   OpenTSDB REST API session in a HTTPS connection library                   #
+# from https://github.com/runabove/iot-push-examples/blob/master/Python/python-client-http.py #
+###############################################################################################
+
+from __future__ import print_function, unicode_literals
+import json
+
+# Requests library that hides much of the complexity of dealing with
+# HTTP requests in Python. Can either be installed with pip:
+#     pip install requests
+# or with your package manager:
+#     apt-get install python-requests
+#     yum install python-requests
+import requests
+from Metric import Metric
+from OpenTSDBProfile import OpenTSDBProfile
+
+class OpenTSDBPusher:
+
+    def pushData(self, openTSDBProfile, metric):
+        try:
+            # Send request and fetch response
+            response = requests.post(openTSDBProfile.url, data=metric.getAsJsonString(),
+                                     auth=(openTSDBProfile.token_id, openTSDBProfile.token_password))
+
+            # Raise error if any
+            response.raise_for_status()
+
+            # Print the http response code on success
+            print('Send successful\nResponse code from server: {}'.
+                  format(response.status_code))
+
+        except requests.exceptions.HTTPError as e:
+            print('HTTP code is {} and reason is {}'.format(e.response.status_code,
+                                                            e.response.reason))
+            print('data : ' + metric.getAsJsonString())

Python/OpenTSDBPusherTelnet.py

@@ -0,0 +1,64 @@
+# -*- coding: utf8 -*-
+
+###############################################################################################
+#                   OpenTSDB REST API session in a HTTPS connection library                   #
+# from https://github.com/runabove/iot-push-examples/blob/master/Python/python-client-http.py #
+###############################################################################################
+
+from __future__ import print_function, unicode_literals
+from socket import socket, AF_INET, SOCK_STREAM, SHUT_RDWR
+from ssl import wrap_socket, PROTOCOL_TLSv1
+import json
+
+# Requests library that hides much of the complexity of dealing with
+# HTTP requests in Python. Can either be installed with pip:
+#     pip install requests
+# or with your package manager:
+#     apt-get install python-requests
+#     yum install python-requests
+import requests
+from Metric import Metric
+from OpenTSDBProfile import OpenTSDBProfile
+
+class OpenTSDBPusher:
+
+    def pushData(self, openTSDBProfile, metric):
+        client_socket = socket(AF_INET, SOCK_STREAM)
+        client_socket.settimeout(1)
+
+        tls_client = wrap_socket(client_socket, ssl_version=PROTOCOL_TLSv1)
+
+        print('Opening connection')
+        # Connect to the echo server
+        tls_client.connect((openTSDBProfile.url, openTSDBProfile.port))
+
+        print('Authenticating')
+        # Send auth command
+        auth_command = ('auth {}:{}\n'.format(openTSDBProfile.token_id, openTSDBProfile.token_password))
+        tls_client.send(auth_command.encode('utf-8'))
+
+        # Read received data
+        data_in = tls_client.recv(1024)
+
+        # Decode and print message
+        response = data_in.decode()
+        print('Read response: ' + response)
+
+        if response == 'ok\n':
+            # Send data
+            print('Sending data')
+            tagsAsString = ""
+            for key, value in metric.tags.iteritems():
+                tagsAsString = tagsAsString + " " + key + "=" + value
+
+            tls_client.send(b'put '+ metric.metric +' '+ metric.timestamp +' '+ metric.value + ' ' + tagsAsString + '\n')
+            print('Data sent')
+        else:
+            print('Auth failed, not sending data.')
+
+        # Send exit command to close connection
+        tls_client.send(b'exit\n')
+
+        # Close the socket
+        tls_client.shutdown(SHUT_RDWR)
+        tls_client.close()