Маршрутизация базы данных по параметру URL - PullRequest
2 голосов
/ 13 мая 2019

Я работаю над проектом Django с несколькими базами данных.В приложении мне нужно переключить соединение базы данных с Development-Database на Test-DB или Production-DB, основываясь на запросе пользователя.(Архитектура БД установлена ​​и не подлежит изменению!)

Я испытал свою удачу и с этим старым Руководством здесь Не работает.Внутри DB Router у меня нет доступа к threading.locals.

Я также попытался настроить собственный маршрутизатор БД.Над переменной сеанса я попытался установить строку подключения.Чтобы прочитать пользовательский сеанс в dbRouter, мне нужен точный ключ сеанса, или я должен выполнить цикл выброса всех сеансов.

Путь через object.using ('DB_CONNECTION) является неприемлемым решением ... Для многих зависимостей,Я хочу установить глобальное соединение для вошедшего в систему пользователя, не назначая соединение с БД для каждой функции модели….

Пожалуйста, дайте мне некоторое представление о том, как решить эту проблему.

Я должен быть в состояниичтобы вернуть dbConnection в db Router на основе значения сеанса ...

def db_for_read|write|*():
   from django.contrib.sessions.backends.db import SessionStore
   session = SessionStore(session_key='WhyINeedHereAKey_SessionKeyCouldBeUserId')
   return session['app.dbConnection']

Update1: Спасибо @victorT за ваши выводы.Я только что попробовал это с приведенными примерами.Все еще не достигли цели ...

Вот что я пытался.В Mabe вы увидите ошибку конфигурации.

Django Version:     2.1.4
Python Version:     3.6.3
Exception Value:    (1146, "Table 'app.myModel' doesn't exist")

.app / views / myView.py

from ..models import myModel
from ..thread_local import thread_local

class myView:
    @thread_local(DB_FOR_READ_OVERRIDE='MY_DATABASE')
    def get_queryset(self, *args, **kwargs):
        return myModel.objects.get_queryset()

.app / myRouter.py

from .thread_local import get_thread_local

class myRouter:
    def db_for_read(self, model, **hints):
        myDbCon = get_thread_local('DB_FOR_READ_OVERRIDE', 'default')
        print('Returning myDbCon:', myDbCon)
        return myDbCon

.app / thread_local.py

import threading
from functools import wraps


threadlocal = threading.local()


class thread_local(object):
    def __init__(self, **kwargs):
        self.options = kwargs

    def __enter__(self):
        for attr, value in self.options.items():
            print(attr, value)
            setattr(threadlocal, attr, value)

    def __exit__(self, exc_type, exc_value, traceback):
        for attr in self.options.keys():
            setattr(threadlocal, attr, None)

    def __call__(self, test_func):

        @wraps(test_func)
        def inner(*args, **kwargs):
            # the thread_local class is also a context manager
            # which means it will call __enter__ and __exit__
            with self:
                return test_func(*args, **kwargs)

        return inner

def get_thread_local(attr, default=None):
    """ use this method from lower in the stack to get the value """
    return getattr(threadlocal, attr, default)

Это вывод:

Returning myDbCon: default              
DEBUG (0.000) None; args=None
DEBUG (0.000) None; args=None
DEBUG (0.000) None; args=('2019-05-14 06:13:39.477467', '4agimu6ctbwgykvu31tmdvuzr5u94tgk')
DEBUG (0.001) None; args=(1,)
DB_FOR_READ_OVERRIDE MY_DATABASE        # The local_router seems to get the given db Name,
Returning myDbCon: None                 # But disapears in the Router
DEBUG (0.000) None; args=()
Returning myDbCon: None
DEBUG (0.001) None; args=()
Returning myDbCon: None
DEBUG (0.001) None; args=()
Returning myDbCon: None
DEBUG (0.001) None; args=()
Returning myDbCon: None
DEBUG (0.001) None; args=()
Returning myDbCon: None
DEBUG (0.002) None; args=()
ERROR Internal Server Error: /app/env/list/    # It switches back to the default
Traceback (most recent call last):
  File "/.../lib64/python3.6/site-packages/django/db/backends/utils.py", line 85, in _execute
    return self.cursor.execute(sql, params)
  File "/.../lib64/python3.6/site-packages/django/db/backends/mysql/base.py", line 71, in execute
    return self.cursor.execute(query, args)
  File "/.../lib64/python3.6/site-packages/MySQLdb/cursors.py", line 255, in execute
    self.errorhandler(self, exc, value)
  File "/.../lib64/python3.6/site-packages/MySQLdb/connections.py", line 50, in defaulterrorhandler
    raise errorvalue
  File "/.../lib64/python3.6/site-packages/MySQLdb/cursors.py", line 252, in execute
    res = self._query(query)
  File "/.../lib64/python3.6/site-packages/MySQLdb/cursors.py", line 378, in _query
    db.query(q)
  File "/.../lib64/python3.6/site-packages/MySQLdb/connections.py", line 280, in query
    _mysql.connection.query(self, query)
_mysql_exceptions.ProgrammingError: (1146, "Table 'app.myModel' doesn't exist")

The above exception was the direct cause of the following exception:

Update2: Это попытка с использованием сеансов.

Я храню соединение с БД через промежуточное ПО в сеансе.В маршрутизаторе я хочу получить доступ к сеансу, который запрашивает.Я ожидал бы, что Джанго управляет этим и знает запрашивающего.Но я должен указать ключ сеанса как

 s = SessionStore(session_key='???')

, чтобы я не смог добраться до маршрутизатора ...

.middleware.py

from django.contrib.sessions.backends.file import SessionStore

class myMiddleware:

    def  process_view(self, request, view_func, view_args, view_kwargs):
        s = SessionStore()
        s['app.dbConnection'] = view_kwargs['MY_DATABASE']
        s.create()    

.myRouter.py

class myRouter:
    def db_for_read(self, model, **hints):
        from django.contrib.sessions.backends.file import SessionStore
        s = SessionStore(session_key='???')
        return s['app.dbConnection']

Это приводит к тому же, что и threading.local ... значение emtpy: - (

Ответы [ 2 ]

0 голосов
/ 16 мая 2019

По крайней мере я успел протестировать пакет threadlocals , и он работает с Django 2.1 и Python 3.6.3.

.app / middleware.py

from threadlocals.threadlocals import set_request_variable

try:
    from django.utils.deprecation import MiddlewareMixin
except ImportError:
    MiddlewareMixin = object

class MyMiddleware(MiddlewareMixin):
    def  process_view(self, request, view_func, view_args, view_kwargs):
        set_request_variable('dbConnection', view_kwargs['environment'])

...

.app / router.py

from threadlocals.threadlocals import get_request_variable

class MyRouter:
    def db_for_read(self, model, **hints):
        if model._meta.app_label == 'app':
            return get_request_variable("dbConnection")
        return None
...
0 голосов
/ 13 мая 2019

Используйте threading.local для установки переменных при ответе на запрос ( doc ). Настраиваемый маршрутизатор - это путь.

Источники:

Взято из django-dynamic-db-router:

from dynamic_db_router import in_database

with in_database('non-default-db'):
    result = run_complex_query()

Обратите внимание, что проект предназначен для Django 1.11 max и может иметь проблемы с совместимостью. Тем не менее, класс маршрутизатора и декоратор, описанные в обоих источниках, довольно просты.

В вашем случае установите переменную для каждого пользователя. В качестве напоминания вы найдете пользователя с request.user.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...