src/emuvim/cli/prometheus.py

   1 # Copyright (c) 2015 SONATA-NFV and Paderborn University
   2 # ALL RIGHTS RESERVED.
   3 #
   4 # Licensed under the Apache License, Version 2.0 (the "License");
   5 # you may not use this file except in compliance with the License.
   6 # You may obtain a copy of the License at
   7 #
   8 #    http://www.apache.org/licenses/LICENSE-2.0
   9 #
  10 # Unless required by applicable law or agreed to in writing, software
  11 # distributed under the License is distributed on an "AS IS" BASIS,
  12 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13 # See the License for the specific language governing permissions and
  14 # limitations under the License.
  15 #
  16 # Neither the name of the SONATA-NFV, Paderborn University
  17 # nor the names of its contributors may be used to endorse or promote
  18 # products derived from this software without specific prior written
  19 # permission.
  20 #
  21 # This work has been performed in the framework of the SONATA project,
  22 # funded by the European Commission under Grant number 671517 through
  23 # the Horizon 2020 and 5G-PPP programmes. The authors would like to
  24 # acknowledge the contributions of their colleagues of the SONATA
  25 # partner consortium (www.sonata-nfv.eu).
  26 import requests
  27
  28
  29 # set this to localhost for now
  30 # this is correct for son-emu started outside of a container or as a container with net=host
  31 # TODO if prometheus sdk DB is started outside of emulator, place these
  32 # globals in an external SDK config file?
  33 prometheus_ip = 'localhost'
  34 # when sdk is started with docker-compose, we could use
  35 # prometheus_ip = 'prometheus'
  36 prometheus_port = '9090'
  37 prometheus_REST_api = 'http://{0}:{1}'.format(prometheus_ip, prometheus_port)
  38
  39
  40 def query_Prometheus(query):
  41     url = prometheus_REST_api + '/' + 'api/v1/query?query=' + query
  42     # logging.info('query:{0}'.format(url))
  43     req = requests.get(url)
  44     ret = req.json()
  45     if ret['status'] == 'success':
  46         # logging.info('return:{0}'.format(ret))
  47         try:
  48             ret = ret['data']['result'][0]['value']
  49         except BaseException:
  50             ret = None
  51     else:
  52         ret = None
  53     return ret