需求描述
最近在写Django DRM,我这里有一个Model,是记录节点信息的,几个相关字段如下:
字段名 | 字段描述 | 备注 |
---|---|---|
node_id | 节点的ID | 主键,但是根据IP生成 |
node_ip | 节点的IP | 节点的IP地址 |
web_url | 节点服务的URL | 和IP也是关联的 |
现在的需求就是,在创建、更改一条节点信息时,node_id和web_url同步做校验修改,如下:
实现方式
这里采用Django的信号机制来实现这个需求。编写函数到signals_receiver.py:
from django.db.models import signals
from django.dispatch import receiver
from prometheus.models import PrometheusStatus
@receiver(signals.post_save, sender=PrometheusStatus)
def monitor_prometheus_create(instance, **kwargs):
node_id = 'P'
for num in instance.node_ip.split('.'):
node_id += num.zfill(3)
web_url = "{protocol}://{node_ip}:{port}".format(protocol='http', node_ip=instance.node_ip, port='9090')
if instance.node_id == node_id and instance.web_url == web_url:
return
PrometheusStatus.objects.filter(node_ip=instance.node_ip).update(node_id=node_id, web_url=web_url)
说明一下,node_id的规则是P+IP补零后的12位写法,比如192.168.1.1的ID就是P192168001001;
这里的信号网上有些解决方案是使用post_init,我个人认为post_save更合适,这样无论是新建还是后续进行变更,都会更新对应的值;
这里我修改一下IP:
保存以后查看结果:
另外,因为我的signals_receiver.py文件是单独放在app的details目录下的,所以我还在urls.py里面import了一下:
from .details import signals_receiver