Django+PostgreSQL测试库自定义函数重建丢失的解决方案咨询

1. 查询pg_proc判断函数是否存在的方法是否可行？ #

这个方法完全可行，但要注意一个关键细节：PostgreSQL允许不同schema下存在同名函数，仅通过proname判断可能出现误判。建议补充schema过滤条件，确保精准定位目标函数（以默认的public schema为例）：

SELECT 1 FROM pg_proc 
WHERE proname = 'test_function'
AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'public');

你可以把判断逻辑封装到SQL的DO块中，实现自动创建：

DO $$
BEGIN
    IF NOT EXISTS (
        SELECT 1 FROM pg_proc 
        WHERE proname = 'test_function'
        AND pronamespace = (SELECT oid FROM pg_namespace WHERE nspname = 'public')
    ) THEN
        CREATE FUNCTION test_function()
        RETURNS void AS $$
        BEGIN
            -- 这里写函数具体逻辑
        END;
        $$ LANGUAGE plpgsql;
    END IF;
END $$;

2. Django REST Framework结合PostgreSQL的更优方案 #

推荐直接利用Django原生迁移系统管理自定义函数，这是最贴合DRF+Django生态的方案，能让函数的创建/更新与数据库结构变更完全同步，无需手动维护脚本：

• 第一步，生成空迁移文件：

  python manage.py makemigrations your_app --empty
  

• 第二步，打开生成的迁移文件，用RunSQL执行函数创建逻辑。PostgreSQL 11及以上版本支持CREATE OR REPLACE FUNCTION，既可以创建新函数，也能在函数已存在时更新定义，比判断存在再创建更灵活：

  from django.db import migrations, models
  
  class Migration(migrations.Migration):
      dependencies = [
          # 填写依赖的前置迁移，比如('your_app', '0001_initial')
      ]
  
      operations = [
          migrations.RunSQL(
              """
              CREATE OR REPLACE FUNCTION test_function()
              RETURNS void AS $$
              BEGIN
                  -- 函数具体逻辑
              END;
              $$ LANGUAGE plpgsql;
              """
          )
      ]
  

• 每次重建测试数据库后，只需执行python manage.py migrate，Django会自动执行所有迁移操作，包括创建/更新自定义函数，全程无需手动干预。

如果你的函数需要和Django模型深度绑定，也可以尝试django-pg-extensions这类第三方库，但原生迁移系统已经能覆盖绝大多数场景，且无需额外依赖。

内容的提问来源于stack exchange，提问作者Petie Pete