112 lines
3.0 KiB
Python
112 lines
3.0 KiB
Python
"""This module contains basic functions to instantiate the BigchainDB API.
|
|
|
|
The application is implemented in Flask and runs using Gunicorn.
|
|
"""
|
|
|
|
import copy
|
|
import multiprocessing
|
|
|
|
from flask import Flask
|
|
from flask_cors import CORS
|
|
import gunicorn.app.base
|
|
|
|
from bigchaindb import utils
|
|
from bigchaindb import Bigchain
|
|
from bigchaindb.web.routes import add_routes
|
|
|
|
|
|
# TODO: Figure out if we do we need all this boilerplate.
|
|
class StandaloneApplication(gunicorn.app.base.BaseApplication):
|
|
"""Run a **wsgi** app wrapping it in a Gunicorn Base Application.
|
|
|
|
Adapted from:
|
|
- http://docs.gunicorn.org/en/latest/custom.html
|
|
"""
|
|
|
|
def __init__(self, app, *, options=None):
|
|
'''Initialize a new standalone application.
|
|
|
|
Args:
|
|
app: A wsgi Python application.
|
|
options (dict): the configuration.
|
|
|
|
'''
|
|
self.options = options or {}
|
|
self.application = app
|
|
super().__init__()
|
|
|
|
def load_config(self):
|
|
config = dict((key, value) for key, value in self.options.items()
|
|
if key in self.cfg.settings and value is not None)
|
|
|
|
for key, value in config.items():
|
|
# not sure if we need the `key.lower` here, will just keep
|
|
# keep it for now.
|
|
self.cfg.set(key.lower(), value)
|
|
|
|
def load(self):
|
|
return self.application
|
|
|
|
|
|
def create_app(*, debug=False, threads=4):
|
|
"""Return an instance of the Flask application.
|
|
|
|
Args:
|
|
debug (bool): a flag to activate the debug mode for the app
|
|
(default: False).
|
|
threads (int): number of threads to use
|
|
Return:
|
|
an instance of the Flask application.
|
|
"""
|
|
|
|
app = Flask(__name__)
|
|
|
|
CORS(app,
|
|
allow_headers=(
|
|
'x-requested-with',
|
|
'content-type',
|
|
'accept',
|
|
'origin',
|
|
'authorization',
|
|
'x-csrftoken',
|
|
'withcredentials',
|
|
'cache-control',
|
|
'cookie',
|
|
'session-id',
|
|
),
|
|
supports_credentials=True)
|
|
|
|
app.debug = debug
|
|
|
|
app.config['bigchain_pool'] = utils.pool(Bigchain, size=threads)
|
|
|
|
add_routes(app)
|
|
|
|
return app
|
|
|
|
|
|
def create_server(settings):
|
|
"""Wrap and return an application ready to be run.
|
|
|
|
Args:
|
|
settings (dict): a dictionary containing the settings, more info
|
|
here http://docs.gunicorn.org/en/latest/settings.html
|
|
|
|
Return:
|
|
an initialized instance of the application.
|
|
"""
|
|
|
|
settings = copy.deepcopy(settings)
|
|
|
|
if not settings.get('workers'):
|
|
settings['workers'] = (multiprocessing.cpu_count() * 2) + 1
|
|
|
|
if not settings.get('threads'):
|
|
settings['threads'] = (multiprocessing.cpu_count() * 2) + 1
|
|
|
|
settings['logger_class'] = 'bigchaindb.log.loggers.HttpServerLogger'
|
|
app = create_app(debug=settings.get('debug', False),
|
|
threads=settings['threads'])
|
|
standalone = StandaloneApplication(app, options=settings)
|
|
return standalone
|