74 lines
1.6 KiB
Python
74 lines
1.6 KiB
Python
import json
|
|
import flask
|
|
import os
|
|
import redis
|
|
import rq
|
|
import sentry_sdk
|
|
import time
|
|
import urllib3
|
|
|
|
from sentry_sdk.integrations.flask import FlaskIntegration
|
|
from sentry_sdk.integrations.rq import RqIntegration
|
|
|
|
|
|
app = flask.Flask(__name__)
|
|
redis_conn = redis.Redis()
|
|
http = urllib3.PoolManager()
|
|
queue = rq.Queue(connection=redis_conn)
|
|
|
|
|
|
def write_event(event):
|
|
with open("events", "a") as f:
|
|
f.write(json.dumps(event))
|
|
f.write("\n")
|
|
|
|
|
|
sentry_sdk.init(
|
|
integrations=[FlaskIntegration(), RqIntegration()],
|
|
traces_sample_rate=1.0,
|
|
traceparent_v2=True,
|
|
debug=True,
|
|
transport=write_event,
|
|
)
|
|
|
|
|
|
def decode_base64(encoded, redis_key):
|
|
time.sleep(1)
|
|
r = http.request("GET", "http://httpbin.org/base64/{}".format(encoded))
|
|
redis_conn.set(redis_key, r.data)
|
|
|
|
|
|
@app.route("/")
|
|
def index():
|
|
return flask.render_template(
|
|
"index.html",
|
|
sentry_dsn=os.environ["SENTRY_DSN"],
|
|
traceparent=dict(sentry_sdk.Hub.current.iter_trace_propagation_headers()),
|
|
)
|
|
|
|
|
|
@app.route("/compute/<input>")
|
|
def compute(input):
|
|
redis_key = "sentry-python-tracing-example-result:{}".format(input)
|
|
redis_conn.delete(redis_key)
|
|
queue.enqueue(decode_base64, encoded=input, redis_key=redis_key)
|
|
|
|
return redis_key
|
|
|
|
|
|
@app.route("/wait/<redis_key>")
|
|
def wait(redis_key):
|
|
result = redis_conn.get(redis_key)
|
|
if result is None:
|
|
return "NONE"
|
|
else:
|
|
redis_conn.delete(redis_key)
|
|
return "RESULT: {}".format(result)
|
|
|
|
|
|
@app.cli.command("worker")
|
|
def run_worker():
|
|
print("WORKING")
|
|
worker = rq.Worker([queue], connection=queue.connection)
|
|
worker.work()
|