给定自动递增主键整数的 RangeA(1 到 32767)和 RangeB(32768 到 2147483647)。如果条件为真,则保存在 RangeA 中分配主键的对象,否则将其保存在 RangeB 中。
管理员(我)将是唯一保存到 RangeA 的用户。如果上述方法不可行:如果 Django 始终保存在 RangeB 中并且需要进入 shell 以保存在 RangeA 中,这将不理想但仍然可用。
如何使用 Django 和 Postgres 来做到这一点?
给定自动递增主键整数的 RangeA(1 到 32767)和 RangeB(32768 到 2147483647)。如果条件为真,则保存在 RangeA 中分配主键的对象,否则将其保存在 RangeB 中。
管理员(我)将是唯一保存到 RangeA 的用户。如果上述方法不可行:如果 Django 始终保存在 RangeB 中并且需要进入 shell 以保存在 RangeA 中,这将不理想但仍然可用。
如何使用 Django 和 Postgres 来做到这一点?
很有可能。首先是更改您的模型,使其不使用标准 AutoField 作为主键
class MyModel(models.Model):
id = models.IntegerField(primary_key=True)
然后你需要连接到 postgresql 并创建两个不同的序列。
CREATE SEQUENCE small START 1;
CREATE SEQUENCE big START 32768;
除了在 PSQL 控制台中输入,您还可以考虑编辑 django 迁移(使用 RunSQL 指令)来创建这些。
下一步是覆盖保存方法
def save(self,*args, **kwargs)
if not self.id :
cursor = connection.cursor()
if small condition:
cursor.execute("select nextval('small')")
else:
cursor.execute("select nextval('big')")
self.id = cursor.fetchone()[0]
super(MyModel,self).save(*args,**kwargs)
覆盖保存方法的替代方法是在插入触发器之前创建一个 postgresql。获得 nextval 的往返成本不是很高,但如果这是一个问题,则可以选择创建触发器。在这种情况下,您不需要覆盖 save 方法,但您必须在触发器中实现相同的逻辑。