osdfapp.py

   1 # -------------------------------------------------------------------------
   2 #   Copyright (c) 2015-2017 AT&T Intellectual Property
   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 # -------------------------------------------------------------------------
  17 #
  18
  19 """
  20 OSDF Manager Main Flask Application
  21 """
  22
  23 import json
  24
  25 from threading import Thread  # for scaling up, may need celery with RabbitMQ or redis
  26
  27 from flask import request, g
  28
  29 from osdf.apps.baseapp import app, run_app
  30 from apps.nst.models.api.nstSelectionRequest import NSTSelectionAPI
  31 from apps.pci.models.api.pciOptimizationRequest import PCIOptimizationAPI
  32 from apps.nst.optimizers.nst_select_processor import process_nst_selection
  33 from apps.pci.optimizers.pci_opt_processor import process_pci_optimation
  34 from apps.placement.models.api.placementRequest import PlacementAPI
  35 from apps.placement.optimizers.conductor.remote_opt_processor import process_placement_opt
  36 from apps.route.optimizers.simple_route_opt import RouteOpt
  37 from apps.slice_selection.models.api.nsi_selection_request import NSISelectionAPI
  38 from apps.slice_selection.optimizers.conductor.remote_opt_processor import process_nsi_selection_opt
  39 from osdf.adapters.policy.interface import get_policies
  40 from osdf.adapters.policy.interface import upload_policy_models
  41 from osdf.config.base import osdf_config
  42 from osdf.logging.osdf_logging import MH, audit_log
  43 from osdf.operation.responses import osdf_response_for_request_accept as req_accept
  44 from osdf.utils import api_data_utils
  45 from osdf.webapp.appcontroller import auth_basic
  46
  47
  48 @app.route("/api/oof/v1/healthcheck", methods=["GET"])
  49 def do_osdf_health_check():
  50     """Simple health check"""
  51     audit_log.info("A health check request is processed!")
  52     return "OK"
  53
  54
  55 @app.route("/api/oof/loadmodels/v1", methods=["GET"])
  56 def do_osdf_load_policies():
  57     audit_log.info("Uploading policy models")
  58     """Upload policy models"""
  59     response = upload_policy_models()
  60     audit_log.info(response)
  61     return "OK"
  62
  63
  64 @app.route("/api/oof/v1/placement", methods=["POST"])
  65 @auth_basic.login_required
  66 def do_placement_opt():
  67     return placement_rest_api()
  68
  69
  70 @app.route("/api/oof/placement/v1", methods=["POST"])
  71 @auth_basic.login_required
  72 def do_placement_opt_common_versioning():
  73     return placement_rest_api()
  74
  75
  76 def placement_rest_api():
  77     """Perform placement optimization after validating the request and fetching policies
  78     Make a call to the call-back URL with the output of the placement request.
  79     Note: Call to Conductor for placement optimization may have redirects, so account for them
  80     """
  81     request_json = request.get_json()
  82     req_id = request_json['requestInfo']['requestId']
  83     g.request_id = req_id
  84     audit_log.info(MH.received_request(request.url, request.remote_addr, json.dumps(request_json)))
  85     api_version_info = api_data_utils.retrieve_version_info(request, req_id)
  86     PlacementAPI(request_json).validate()
  87     policies = get_policies(request_json, "placement")
  88     audit_log.info(MH.new_worker_thread(req_id, "[for placement]"))
  89     t = Thread(target=process_placement_opt, args=(request_json, policies, osdf_config))
  90     t.start()
  91     audit_log.info(MH.accepted_valid_request(req_id, request))
  92     return req_accept(request_id=req_id,
  93                       transaction_id=request_json['requestInfo']['transactionId'],
  94                       version_info=api_version_info, request_status="accepted", status_message="")
  95
  96
  97 @app.route("/api/oof/route/v1", methods=["POST"])
  98 def do_route_calc():
  99     """
 100     Perform the basic route calculations and returnn the vpn-bindings
 101     """
 102     request_json = request.get_json()
 103     audit_log.info("Calculate Route request received!")
 104     response = RouteOpt().get_route(request_json, osdf_config)
 105     return response
 106
 107 @app.route("/api/oof/v1/selection/nst", methods=["POST"])
 108 def do_nst_selection():
 109     request_json = request.get_json()
 110     req_id = request_json['requestInfo']['requestId']
 111     NSTSelectionAPI(request_json).validate()
 112     response = process_nst_selection(request_json, osdf_config)
 113     return response
 114
 115 @app.route("/api/oof/v1/pci", methods=["POST"])
 116 @app.route("/api/oof/pci/v1", methods=["POST"])
 117 @auth_basic.login_required
 118 def do_pci_optimization():
 119     request_json = request.get_json()
 120     audit_log.info('request json obtained==>')
 121     audit_log.info(request_json)
 122
 123     req_id = request_json['requestInfo']['requestId']
 124     audit_log.info('requestID obtained==>')
 125     audit_log.info(req_id)
 126
 127     g.request_id = req_id
 128     audit_log.info(MH.received_request(request.url, request.remote_addr, json.dumps(request_json)))
 129     PCIOptimizationAPI(request_json).validate()
 130     # disable policy retrieval
 131     # policies = get_policies(request_json, "pciopt")
 132     audit_log.info(MH.new_worker_thread(req_id, "[for pciopt]"))
 133     t = Thread(target=process_pci_optimation, args=(request_json, osdf_config, None))
 134     t.start()
 135     audit_log.info(MH.accepted_valid_request(req_id, request))
 136     audit_log.info('reached upto return')
 137     return req_accept(request_id=req_id,
 138                       transaction_id=request_json['requestInfo']['transactionId'],
 139                       request_status="accepted", status_message="")
 140
 141
 142 @app.route("/api/oof/selection/nsi/v1", methods=["POST"])
 143 def do_nsi_selection():
 144     request_json = request.get_json()
 145     req_id = request_json['requestInfo']['requestId']
 146     g.request_id = req_id
 147     audit_log.info(MH.received_request(request.url, request.remote_addr, json.dumps(request_json)))
 148     NSISelectionAPI(request_json).validate()
 149     audit_log.info(MH.new_worker_thread(req_id, "[for NSI selection]"))
 150     t = Thread(target=process_nsi_selection_opt, args=(request_json, osdf_config))
 151     t.start()
 152     return req_accept(request_id=req_id,
 153                       transaction_id=request_json['requestInfo']['transactionId'],
 154                       request_status="accepted", status_message="")
 155
 156
 157 if __name__ == "__main__":
 158     run_app()