mysql+redis

Author: irr

tornado/web/Web.py

@@ -23,8 +23,9 @@
 
 class WebApplication(tornado.web.Application):
     def __init__(self, **kwargs):
-        kwargs["redis"] = tornadoredis.ConnectionPool(max_connections=10,
-                                                      wait_for_available=True)
+        kwargs["redis"] = tornadoredis.ConnectionPool(max_connections = 10,
+                                                      wait_for_available = True)
+        kwargs["mysql"] = { 'host':'localhost', 'database':'mysql', 'user':'root', 'password':'mysql' }
         handlers = [(r"/", IndexHandler, kwargs)]
         tornado.web.Application.__init__(self, handlers)
 

tornado/web/handlers/Index.py

@@ -14,8 +14,8 @@
 class IndexHandler(RequestHandler):
     def initialize(self, **kwargs):
         self.logger = kwargs.get("logger")
+        self.mysql = kwargs.get("mysql")
         self.redis = tornadoredis.Client(connection_pool=kwargs["redis"])
-        self.mysql = torndb.Connection("localhost", "mysql")
         self.template = Template("op -cmd {{ cmd }} -value {{ value }}")
 
     def _customize(self):
@@ -24,20 +24,25 @@ def _customize(self):
 
     @tornado.web.asynchronous
     @tornado.gen.engine
-    def _redis(self):
-        info = yield tornado.gen.Task(self.redis.info)
-        yield tornado.gen.Task(self.redis.disconnect)
-        return info
-
     def _handle(self, **kwargs):
         self._customize()
-        try:            
+        db = None
+        try:   
+            db = torndb.Connection(self.mysql['host'], self.mysql['database'], 
+                user=self.mysql['user'], password=self.mysql['password'])
+            hosts = [host for host in db.query("SELECT Host FROM user WHERE User = 'root'")]
+            info = yield tornado.gen.Task(self.redis.info)
+            yield tornado.gen.Task(self.redis.disconnect)
             data = { 'cmd': "any", 'value': str(time.time()) }
             response = { 'status': 200,
                          'msg': self.template.generate(**data),
-                         'redis': _redis['redis_version'] }
+                         'redis': info['redis_version'],
+                         'mysql': hosts }
         except Exception as ex:
             response = { 'status': 500, 'msg': str(ex) }
+        finally:
+            if db:
+                db.close()
         if response['status'] != 200:
             self.set_status(response['status'])            
         self.write(json_encode(response))