【MongoDB】在Django中MySQL和MongoDB共用 | EMQX 用户数据库

简要

使用mongoengine、singals实现MySQL和MongoDB的共用。

需求

在不影响原有MySQL表的情况下，加入MongoDB来处理EMQX的用户验证。

代码

模型定义

# models.py
class MQTTUser(models.Model):
    username = models.CharField(max_length=255, db_index=True, null=False, blank=False, unique=True)
    password = models.CharField(max_length=255,  null=False, blank=False)
    salt = models.CharField(max_length=100, null=True, blank=True) #  # 自动生成
    is_superuser = models.BooleanField(default=False)  # 固定
    expired_date = models.DateTimeField(default=timezone.now, null=True)  # 过期时间 null则为不限时
    create_by = models.ForeignKey(User, on_delete=models.CASCADE) 

上面的为MYSQL所使用的模型，那现在我们要使用mongoengine定义Document与MYSQL的模型进行同步。

# models.py
import mongoengine

class MQTTUser_MongoModel(mongoengine.Document):
    username = mongoengine.StringField(max_length=255)
    password = mongoengine.StringField(max_length=255)
    salt = mongoengine.StringField(max_length=100)
    is_superuser = mongoengine.BooleanField(default=False)
    created = mongoengine.DateTimeField()

    meta ={
        "collection": "mqtt_user"
    }

类名定义随意，我习惯性后面加上下划线进行分类，meta中的collection为集合名称，如果不定义集合名称，它自动生成的是根据类名大写来的，如：m_q_t_t_user_mongo_model，建议自定义好类名。

连接

原有settings中存在MySQL引擎的连接。

mongo的连接

定义好两个模型后，我们需要先加入mongo的连接定义，在Django的配置文件数据库配置后面加上即可。

# settings.py
...
# 原有的MySQL配置
DATABASES = {
    'default': {
        'ENGINE': 'django.db.backends.mysql',
        'NAME': '',
        'USER': '',
        'PASSWORD': '',
        'HOST': '127.0.0.1',
        'PORT': '3306',
    },
}
# 添加
from mongoengine import connect  
connect(host='mongodb://<username>:<password>@<host>:<port>/<database>?authSource=admin') 

connect的参数如下：

• db=None,
• name=None,
• host=None,
• port=None,
• read_preference=READ_PREFERENCE,
• username=None,
• password=None,
• authentication_source=None,
• authentication_mechanism=None,

这里用到host参数，如mongodb://admin:admin@localhost:27017/emqx?authSource=admin
后面的authSource=admin是必要的，指明mongo的验证集合。

模型之间的连接

我们可以使用Django的信号函数，在MQTTUser模型创建新纪录后（post_save）调用create_mqtt_user_in_emq函数，传入实例，在函数中调用实例创建后信息进行Mongo中的创建。

from django.dispatch import receiver
from django.db.models import signals
@receiver(signals.post_save, sender=MQTTUser)
def create_mqtt_user_in_emq(sender, instance, created, **kwargs):
    if created:
        MQTTUser_MongoModel.objects.create(
            username=instance.username, 
            password=instance.password, 
            salt=instance.salt, 
            is_superuser=instance.is_superuser, 
            created=instance.created, 
            )