Test env updates for dmaap messages in batch and for running external agent
[nonrtric.git] / test / cr / cr.py
1
2 #  ============LICENSE_START===============================================
3 #  Copyright (C) 2020 Nordix Foundation. All rights reserved.
4 #  ========================================================================
5 #  Licensed under the Apache License, Version 2.0 (the "License");
6 #  you may not use this file except in compliance with the License.
7 #  You may obtain a copy of the License at
8 #
9 #       http://www.apache.org/licenses/LICENSE-2.0
10 #
11 #  Unless required by applicable law or agreed to in writing, software
12 #  distributed under the License is distributed on an "AS IS" BASIS,
13 #  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 #  See the License for the specific language governing permissions and
15 #  limitations under the License.
16 #  ============LICENSE_END=================================================
17 #
18
19 from flask import Flask, request
20 from time import sleep
21 import time
22 import datetime
23 import json
24 from flask import Flask
25 from flask import Response
26 import traceback
27
28 app = Flask(__name__)
29
30 # list of callback messages
31 msg_callbacks={}
32
33 # Server info
34 HOST_IP = "::"
35 HOST_PORT = 8090
36
37 # Metrics vars
38 cntr_msg_callbacks=0
39 cntr_msg_fetched=0
40
41 # Request and response constants
42 CALLBACK_URL="/callbacks/<string:id>"
43 APP_READ_URL="/get-event/<string:id>"
44
45 MIME_TEXT="text/plain"
46 MIME_JSON="application/json"
47 CAUGHT_EXCEPTION="Caught exception: "
48 SERVER_ERROR="Server error :"
49
50 #I'm alive function
51 @app.route('/',
52     methods=['GET'])
53 def index():
54     return 'OK', 200
55
56 ### Callback interface, for control
57
58 # Fetch the oldest callback message for an id
59 # URI and parameter, (GET): /get-event/<id>
60 # response: message + 200 or 204
61 @app.route(APP_READ_URL,
62     methods=['GET'])
63 def receiveresponse(id):
64     global msg_callbacks
65     global cntr_msg_fetched
66
67     try:
68         if ((id in msg_callbacks.keys()) and (len(msg_callbacks[id]) > 0)):
69             cntr_msg_fetched+=1
70             msg=str(msg_callbacks[id][0])
71             print("Fetching msg for id: "+id+", msg="+msg)
72             del msg_callbacks[id][0]
73             return msg,200
74         print("No messages for id: "+id)
75     except Exception as e:
76         print(CAUGHT_EXCEPTION+str(e))
77
78     return "",204
79
80
81 # Receive a callback message
82 # URI and payload, (PUT or POST): /callbacks/<id> <json array of response messages>
83 # response: OK 200 or 500 for other errors
84 @app.route(CALLBACK_URL,
85     methods=['PUT','POST'])
86 def events_write(id):
87     global msg_callbacks
88     global cntr_msg_callbacks
89
90     try:
91         print("Received callback for id: "+id +", content-type="+request.content_type)
92         try:
93             if (request.content_type == MIME_JSON):
94                 msg = request.json
95                 print("Payload(json): "+str(msg))
96             elif (request.content_type == MIME_TEXT):
97                 msg= request.form
98                 print("Payload(text): "+str(msg))
99             else:
100                 msg="\"\""
101                 print("Payload(content-type="+request.content_type+"). Setting data to empty, quoted, string")
102         except:
103             msg="\"\""
104             print("(Exception) Payload does not contain any json or text data, setting empty string as payload")
105
106         cntr_msg_callbacks += 1
107         if (id in msg_callbacks.keys()):
108             msg_callbacks[id].append(msg)
109         else:
110             msg_callbacks[id]=[]
111             msg_callbacks[id].append(msg)
112     except Exception as e:
113         print(CAUGHT_EXCEPTION+str(e))
114         return 'OK',500
115
116     return 'OK',200
117
118
119 ### Functions for metrics read out ###
120
121 @app.route('/counter/received_callbacks',
122     methods=['GET'])
123 def requests_submitted():
124     return Response(str(cntr_msg_callbacks), status=200, mimetype=MIME_TEXT)
125
126 @app.route('/counter/fetched_callbacks',
127     methods=['GET'])
128 def requests_fetched():
129     return Response(str(cntr_msg_fetched), status=200, mimetype=MIME_TEXT)
130
131 @app.route('/counter/current_messages',
132     methods=['GET'])
133 def current_messages():
134     return Response(str(cntr_msg_callbacks-cntr_msg_fetched), status=200, mimetype=MIME_TEXT)
135
136
137
138 ### Admin ###
139
140 # Reset all messsages and counters
141 @app.route('/reset',
142     methods=['GET', 'POST', 'PUT'])
143 def reset():
144     global msg_callbacks
145     global cntr_msg_fetched
146     global cntr_msg_callbacks
147
148     msg_callbacks={}
149     cntr_msg_fetched=0
150     cntr_msg_callbacks=0
151
152     return Response('OK', status=200, mimetype=MIME_TEXT)
153
154 ### Main function ###
155
156 if __name__ == "__main__":
157     app.run(port=HOST_PORT, host=HOST_IP)