DependencyProviders and utilities for nameko services to interface with a relational database using SQLAlchemy.
from nameko_sqlalchemy import Database
from .models import Model, DeclarativeBase
class Service(object):
name = "service"
db = Database(DeclarativeBase)
@entrypoint
def write_to_db(self):
model = Model(...)
with self.db.get_session() as session:
session.add(model)
@entrypoint
def query_db(self):
queryset = self.db.session.query(Model).filter(...)
...The nameko_sqlalchemy.Database DependencyProvider can be used in three ways:
As a context manager that issues a commit or rollback on exit:
@entrypoint
def method(self):
with self.db.get_session() as session:
session.add(model)To explicitly retrieve sessions that can be manually manipulated:
@entrypoint
def method(self):
session1 = self.db.get_session()
session1.add(...)
session2 = self.db.get_session()
session2.add(...)
session1.commit()
session1.close()
session2.commit()
session2.close()To manage a session that is lazily opened on first use and closed when the Nameko entrypoint exits:
@entrypoint
def method(self):
self.db.session.add(...)
self.db.session.commit()The nameko_sqlalchemy.DatabaseSession DependencyProvider maintains the original interface from the early versions of the library. It behaves similarly to the third example above, except that the session is opened before the entrypoint fires, rather than lazily.
class Service:
name = "legacy"
session = DatabaseSession()
@entrypoint
def method(self):
self.session.add(...)
self.session.commit()You may use any database driver compatible with SQLAlchemy provided it is safe to use with eventlet. This will include all pure-python drivers. Known safe drivers are:
This decorator automatically retries the wrapped function when a database connection error occurs.
If the optional session argument is passed it will issue a rollback on it before retrying so the transaction can be processed again.
The session argument can either be the sqlalchemy.orm.session.Session or an operator.attrgetter object if the session is a class attribute.
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from nameko_sqlalchemy import transaction_retry
engine = create_engine('postgresql://username:password@localhost/test')
Session = sessionmaker(bind=engine)
db_session = Session()
@transaction_retry()
def get_example_data():
db_session.query(ExampleModel).all()
example_data = get_example_data()or using with the Database dependency provider
from sqlalchemy.ext.declarative import declarative_base
from nameko_sqlalchemy import Database, transaction_retry
DeclBase = declarative_base(name='examplebase')
class ExampleService:
name = 'exampleservice'
db = Database(DeclBase)
@entrypoint
@transaction_retry
def get_examples(self):
with self.db.get_session() as session:
return session.query(ExampleModel).all()
@entrypoint
def get_examples_with_retry_inside(self):
with self.db.get_session() as session:
@transaction_retry(session=session)
def foo():
return session.query(ExampleModel).all()
return foo()
@entrypoint
@transaction_retry
def create_example_without_using_context_manager(self):
session = self.db.get_session()
session.add(ExampleModel(data='hello'))
session.commit()
@entrypoint
@transaction_retry(session=operator.attrgetter('db.session'))
def create_example_with_worker_scoped_session(self):
self.db.session.add(ExampleModel(data='hello'))
self.db.session.commit()Optionally, the transaction may be retried multiple times with an exponential backoff delay. If given, the backoff factor is applied between attempts after the second try (most errors are resolved immediately by a second try without a delay). The applied delay will then be:
{backoff_factor} * (2 ** ({number of total retries} - 1))
seconds. The delay will never be longer than backoff_max seconds. By default, backoff is disabled (backoff_factor set to 0). Finally, if the connection has still not recovered after total tries, the error is reraised. The following code will thus wait for 0.0s, 0.2s, 0.4s, 0.8s, 1.0s before raising an error:
@transaction_retry(total=5, backoff_factor=0.1, backoff_max=1.0)
def get_example_data():
db_session.query(ExampleModel).all()
example_data = get_example_data()Caution!
Using the decorator may cause unanticipated consequences when the decorated function uses more than one transaction.
It should only be used around single transactions because all transactions inside the decorator will be re-executed if there is a connection error during any of them. Take a look at the following example:
class ExampleService:
db = Database(DeclBase)
@entrypoint
@transaction_retry
def method(self):
with self.db.get_session() as session:
session.add(something)
do_something() # during this a network error occurs
with self.db.get_session() as session:
session.add(something_else) # throws error because the db connection is gone, method will be executed againSince the method is retried all of the statements are executed twice, including the ones that didn't fail. As a result of that something will be added twice.
In order to avoid that one may want to do something like this:
class ExampleService:
db = Database(DeclBase)
@entrypoint
def method(self):
with self.db.get_session() as session:
@transaction_retry(session=session)
def add_two_things():
session.add(something)
do_something()
session.add(something_else)
add_two_things()In this case the failed transaction will be rolled back (because the session is passed to the decorator) and records will not be duplicated.
Pytest fixtures to allow for easy testing are available.
db_sessionfixture (which depends ondb_connectionfixture) will instantiate test database and tear it down at the end of each test.model_basefixture can be overridden to provide customdeclarative_base.db_engine_optionsfixture can be overriden to provide additional keyword arguments tosqlalchemy.create_engine.databasefixture which is similar todb_sessionbut can be passed asDatabasedependency replacement when usingworker_factoryorreplace_dependencies.
import pytest
from sqlalchemy import Column, Integer, String
from sqlalchemy.ext.declarative import declarative_base
class Base(object):
pass
DeclarativeBase = declarative_base(cls=Base)
class User(DeclarativeBase):
__tablename__ = "users"
id = Column(Integer, primary_key=True)
name = Column(String)
@pytest.fixture(scope='session')
def model_base():
return DeclarativeBase
@pytest.fixture(scope='session')
def db_engine_options():
return dict(client_encoding='utf8')
def test_users(db_session):
user = User(id=1, name='Joe')
db_session.add(user)
db_session.commit()
saved_user = db_session.query(User).get(user.id)
assert saved_user.id > 0
assert saved_user.name == 'Joe'When running tests you can pass database test url with --test-db-url parameter or override db_url fixture.
By default SQLite memory database will be used.
pytest test --test-db-url=sqlite:///test_db.sql
pytest test --test-db-url=mysql+mysqlconnector://root:password@localhost:3306/nameko_sqlalchemy_testSome of the tests use toxiproxy to simulate network errors. In order to be able to run those tests you need a toxiproxy server to be in place. You may install it manually or by running the following command (docker is required):
make test-depsThis will setup a mysql and a toxiproxy server with a proxy set up to the database.
Once the containers have been set up the tests can be run by running the following command:
make testThree extra parameters may be passed to pytest:
test-db-url: The database URLtoxiproxy-api-url: The url of the Toxiproxy HTTP APItoxiproxy-db-url: The url of the database through Toxiproxy
If toxiproxy-api-url and toxiproxy-db-url parameters are provided the tests assume that the toxiproxy endpoint is already set up to a database upstream and this proxy can be disabled and enabled via the HTTP API of toxiproxy.
pytest test \
--test-db-url="mysql+pymysql://test_user:password@database_host:3306/nameko_sqlalchemy_test" \
--toxiproxy-api-url="http://toxiproxy_server:8474"
--toxiproxy-db-url="http://toxiproxy_server:3306"if no toxiproxy-api-url and toxiproxy-db-url parameter was provided the tests that require toxiproxy will be skipped.