multivimbroker/multivimbroker/middleware.py

   1 # Copyright (c) 2017-2018 VMware, Inc.
   2 #
   3 # Licensed under the Apache License, Version 2.0 (the "License");
   4 # you may not use this file except in compliance with the License.
   5 # You may obtain a copy of the License at:
   6 #
   7 #       http://www.apache.org/licenses/LICENSE-2.0
   8 #
   9 # Unless required by applicable law or agreed to in writing, software
  10 # distributed under the License is distributed on an "AS IS" BASIS,
  11 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  12
  13
  14 import uuid
  15 from onaplogging.mdcContext import MDC
  16 from multivimbroker.pub.config.config import SERVICE_NAME
  17 from multivimbroker.pub.config.config import FORWARDED_FOR_FIELDS
  18
  19
  20 class LogContextMiddleware(object):
  21
  22     #  the last IP behind multiple proxies,  if no exist proxies
  23     #  get local host ip.
  24     def _getLastIp(self, request):
  25
  26         ip = ""
  27         try:
  28             for field in FORWARDED_FOR_FIELDS:
  29                 if field in request.META:
  30                     if ',' in request.META[field]:
  31                         parts = request.META[field].split(',')
  32                         ip = parts[-1].strip().split(":")[0]
  33                     else:
  34                         ip = request.META[field].split(":")[0]
  35
  36             if ip == "":
  37                 ip = request.META.get("HTTP_HOST").split(":")[0]
  38
  39         except Exception:
  40             pass
  41
  42         return ip
  43
  44     def process_request(self, request):
  45
  46         # Fetch TRANSACTIONID Id and pass to plugin server
  47         ReqeustID = request.META.get("HTTP_X_TRANSACTIONID", None)
  48         if ReqeustID is None:
  49             ReqeustID = uuid.uuid3(uuid.NAMESPACE_URL, SERVICE_NAME)
  50             request.META["HTTP_X_TRANSACTIONID"] = ReqeustID
  51         MDC.put("requestID", ReqeustID)
  52         # generate the unique  id
  53         InovocationID = uuid.uuid3(uuid.NAMESPACE_DNS, SERVICE_NAME)
  54         MDC.put("invocationID", InovocationID)
  55         MDC.put("serviceName", SERVICE_NAME)
  56         # access ip
  57         MDC.put("serviceIP", self._getLastIp(request))
  58
  59         return None
  60
  61     def process_response(self, request, response):
  62
  63         MDC.clear()
  64         return response