如何在Tornado中传递环境变量(数据库URL)?

时间:2019-07-11 00:48:08

标签: tornado tornado-motor

我有一个弹性URL作为环境变量传递给post方法,我已经从tornado_sqlalchemy中声明了make_session_factory,并传递给requesthandler,我应该如何自我接收呢,下面是我的代码

class MyHandler(RequestHandler):
    _thread_pool = ThreadPoolExecutor(max_workers=10)

    def initialize(self, session_factory): 
      self.session_factory = session_factory

    @gen.coroutine
    def post(self):
        try:
            data = tornado.escape.json_decode(self.request.body)
            yield self.predict('mod1')
        except Exception:
            self.respond('server_error', 500)

    @concurrent.run_on_executor(executor='_thread_pool')
    def _b_run(self, mod, X):
        results = do some calculation
        return results

    @gen.coroutine
    def predict(self, mod):  
        model = mod(load from database)
        values = (load from database)
        results = yield self._b_run(model, values)
        self.respond(results)

    def respond(self, code=200):
        self.set_status(code)
        self.write(json.JSONEncoder().encode({
            "status": code
        }))
        self.finish()


def mod(model):
    #Elastic Client
    es = Elasticsearch([{'host': 'localhost', 'port': 9200}])
    ……...Load model
    return model


factory = make_session_factory(os.environ.get('DATABASE_URL', ''))
define('port', default=8888, help='Tornado port to listen on')

def make_app():  
    url = [(r"/pred", PredictionHandler)]
    return Application(url,session_factory=factory)

if __name__ == "__main__":
    application = make_app()
    http_server = HTTPServer(application)
    http_server.listen(options.port)
    IOLoop.current().start()

目前,我已经在我的mod函数中将URL编码为localhost 9200

要将其作为环境变量传递,我已经创建了工厂,并在应用程序中将其作为session_factory传递给了工厂,

我必须在我的requesthandler post方法中接收它,以便我可以传递给我的预测方法,该方法随后将调用我的mod函数,我可能知道如何接收它。我尝试使用Initialize,但无法正常工作。

还发现了另一种使用龙卷风电机的选择

import motor.motor_tornado
client = motor.motor_tornado.MotorClient('localhost', 9200)
db = motor.motor_tornado.MotorClient().test_database

application = tornado.web.Application([
    (r'/', MainHandler)
], db=db)

class MainHandler(tornado.web.RequestHandler):
    def get(self):
        db = self.settings['db']

我还必须在发帖方式中接收

1 个答案:

答案 0 :(得分:1)

您可以通过Application中的任何请求处理程序使用传递给self.settings类的任何其他参数。

因此,您可以像这样修改initialize方法:

def initialize(self): 
    self.session_factory = self.settings['session_factory']

# and just use `self.session_factory` in any other method where you need it