Browse Source

added vote logging

Bret Fisher 4 years ago
parent
commit
d1920ed44a
1 changed files with 6 additions and 0 deletions
  1. 6 0
      vote/app.py

+ 6 - 0
vote/app.py

@@ -4,6 +4,7 @@ import os
 import socket
 import random
 import json
+import logging
 
 option_a = os.getenv('OPTION_A', "Cats")
 option_b = os.getenv('OPTION_B', "Dogs")
@@ -11,6 +12,10 @@ hostname = socket.gethostname()
 
 app = Flask(__name__)
 
+gunicorn_error_logger = logging.getLogger('gunicorn.error')
+app.logger.handlers.extend(gunicorn_error_logger.handlers)
+app.logger.setLevel(logging.INFO)
+
 def get_redis():
     if not hasattr(g, 'redis'):
         g.redis = Redis(host="redis", db=0, socket_timeout=5)
@@ -27,6 +32,7 @@ def hello():
     if request.method == 'POST':
         redis = get_redis()
         vote = request.form['vote']
+        app.logger.info('Received vote for %s', vote)
         data = json.dumps({'voter_id': voter_id, 'vote': vote})
         redis.rpush('votes', data)