feat: 更新 Dockerfile 以支持 Pig 包管理器，添加扩展安装和中文分词功能测试脚本，更新 README.md 以反映新特性

Author: Ricco Xie

Dockerfile

@@ -4,66 +4,34 @@ FROM postgres:16
 # 切换为root用户执行安装操作
 USER root
 
-# 安装依赖包
+# 安装基础依赖包
 RUN apt-get update && apt-get install -y --no-install-recommends \
     apt-transport-https \
-    bzip2 \
     ca-certificates \
     curl \
-    gcc \
-    libc6-dev \
-    make \
-    wget \
-    unzip \
-    cmake \
-    openssl \
-    clang \
-    && rm -rf /var/lib/apt/lists/* \
-    && update-ca-certificates
-
-# 安装 PostgreSQL 开发包（容错安装）
-RUN apt-get update && apt-get install -y --no-install-recommends \
-    libpq-dev \
-    && (apt-get install -y postgresql-server-dev-16 || \
-        apt-get install -y postgresql-server-dev-all || \
-        apt-get install -y postgresql-server-dev || \
-        echo "PostgreSQL server dev packages not available for this architecture") \
+    gnupg \
+    lsb-release \
     && rm -rf /var/lib/apt/lists/*
 
-# 验证 PostgreSQL 开发包是否安装成功
-RUN if [ ! -f "/usr/include/postgresql/16/server/postgres.h" ]; then \
-        echo "PostgreSQL headers not found, trying alternative installation..."; \
-        apt-get update && apt-get install -y --no-install-recommends \
-        postgresql-server-dev-16 || \
-        echo "Failed to install PostgreSQL development headers"; \
-    fi
+# 安装 Pig 包管理器
+RUN curl -fsSL https://repo.pigsty.cc/get | bash
+
+# 配置 Pig 仓库
+RUN pig repo add all
 
-# 安装SCWS库和zhparser扩展（使用官方发布版本）
-RUN wget -q -O - "http://www.xunsearch.com/scws/down/scws-1.2.3.tar.bz2" | tar xjf - && \
-    ZHPARSER_URL="https://github.com/amutu/zhparser/archive/master.tar.gz" && \
-    curl -sSkLf "${ZHPARSER_URL}" | tar xzf - && \
-    cd scws-1.2.3 && \
-    ./configure && \
-    make -j$(nproc) install V=0 && \
-    ldconfig && \
-    cd /zhparser-master && \
-    # 检查 PostgreSQL 头文件是否存在
-    if [ -f "/usr/include/postgresql/16/server/postgres.h" ]; then \
-        echo "PostgreSQL headers found, proceeding with zhparser compilation..."; \
-        make -j$(nproc) install; \
-    else \
-        echo "PostgreSQL headers not found, skipping zhparser compilation..."; \
-        echo "zhparser will not be available in this build"; \
-    fi && \
-    rm -rf /scws-1.2.3 /zhparser-master
+# 使用 Pig 安装 PostgreSQL 扩展
+RUN pig install pg_trgm hstore btree_gist pg_stat_statements unaccent
 
-# 注意：pg_jieba 已移除，因为 zhparser 已经提供了完整的中文分词功能
-# 如果需要结巴分词功能，可以单独安装 pg_jieba 扩展
+# 尝试安装中文分词扩展（如果可用）
+RUN pig install zhparser || echo "zhparser not available for this architecture, skipping..."
 
-# 安装其他常用扩展包
+# 安装其他常用扩展
+RUN pig install postgresql-contrib || echo "postgresql-contrib not available via pig, using apt..."
+
+# 如果 Pig 无法安装某些扩展，回退到 apt 安装
 RUN apt-get update && apt-get install -y --no-install-recommends \
     postgresql-contrib \
-    && rm -rf /var/lib/apt/lists/*
+    && rm -rf /var/lib/apt/lists/* || echo "Some extensions may not be available for this architecture"
 
 # 复制初始化脚本到容器中
 COPY init-scripts/ /docker-entrypoint-initdb.d/

README.md

@@ -24,6 +24,7 @@
 - **全文搜索**: 支持中文全文搜索和模糊匹配
 - **扩展丰富**: 预装常用 PostgreSQL 扩展
 - **容错构建**: 智能处理不同架构的包兼容性问题
+- **Pig 包管理**: 使用 Pig 包管理器统一管理 PostgreSQL 扩展
 
 ### 📦 预装扩展
 
@@ -217,6 +218,30 @@ cd docker-postgres-zh
 docker build -t riccoxie/postgres-zh .
 ```
 
+### Pig 包管理器
+
+本镜像使用 [Pig](https://repo.pigsty.cc/) 作为 PostgreSQL 扩展的包管理器，提供以下优势：
+
+- **统一管理**: 通过 Pig 统一管理所有 PostgreSQL 扩展
+- **多架构支持**: 自动处理不同架构的扩展兼容性
+- **容错安装**: 智能处理扩展安装失败的情况
+
+#### 扩展管理
+
+```bash
+# 查看已安装的扩展
+docker exec -it postgres-zh pig list
+
+# 安装新扩展
+docker exec -it postgres-zh pig install <extension_name>
+
+# 更新扩展
+docker exec -it postgres-zh pig update <extension_name>
+
+# 查看已安装的扩展
+docker exec -it postgres-zh pig list
+```
+
 ### 本地开发
 
 ```bash

init-scripts/01-extensions.sql

@@ -1,20 +1,48 @@
--- 启用核心扩展
+-- 使用 Pig 包管理器管理的扩展初始化脚本
+-- 此脚本会在 PostgreSQL 启动时自动执行
+
+-- 启用核心扩展（通过 Pig 安装）
 CREATE EXTENSION IF NOT EXISTS pg_trgm;
 CREATE EXTENSION IF NOT EXISTS hstore;
 CREATE EXTENSION IF NOT EXISTS btree_gist;
 CREATE EXTENSION IF NOT EXISTS pg_stat_statements;
 CREATE EXTENSION IF NOT EXISTS unaccent;
 
--- 启用中文分词扩展
-CREATE EXTENSION IF NOT EXISTS zhparser;
+-- 启用中文分词扩展（如果通过 Pig 安装成功）
+-- 使用容错方式处理，避免在某些架构上失败
+DO $$
+BEGIN
+    -- 尝试创建 zhparser 扩展
+    BEGIN
+        CREATE EXTENSION IF NOT EXISTS zhparser;
+        
+        -- 为zhparser创建中文全文搜索配置
+        -- 包含常用的中文词性：名词、动词、形容词、习语、叹词、习用语、简称、时间词
+        IF NOT EXISTS (SELECT 1 FROM pg_ts_config WHERE cfgname = 'chinese_zh') THEN
+            CREATE TEXT SEARCH CONFIGURATION chinese_zh (PARSER = zhparser);
+            ALTER TEXT SEARCH CONFIGURATION chinese_zh ADD MAPPING FOR n,v,a,i,e,l,j,t WITH simple;
+        END IF;
+        
+        RAISE NOTICE 'zhparser extension and chinese_zh configuration created successfully';
+    EXCEPTION
+        WHEN OTHERS THEN
+            RAISE NOTICE 'zhparser extension not available for this architecture: %', SQLERRM;
+    END;
+END $$;
 
--- 为zhparser创建中文全文搜索配置
--- 包含常用的中文词性：名词、动词、形容词、习语、叹词、习用语、简称、时间词
+-- 显示已安装的扩展信息
 DO $$
+DECLARE
+    ext_record RECORD;
 BEGIN
-    IF NOT EXISTS (SELECT 1 FROM pg_ts_config WHERE cfgname = 'chinese_zh') THEN
-        CREATE TEXT SEARCH CONFIGURATION chinese_zh (PARSER = zhparser);
-        ALTER TEXT SEARCH CONFIGURATION chinese_zh ADD MAPPING FOR n,v,a,i,e,l,j,t WITH simple;
-    END IF;
+    RAISE NOTICE '=== PostgreSQL Extensions Status ===';
+    FOR ext_record IN 
+        SELECT extname, extversion 
+        FROM pg_extension 
+        ORDER BY extname
+    LOOP
+        RAISE NOTICE 'Extension: % (version: %)', ext_record.extname, ext_record.extversion;
+    END LOOP;
+    RAISE NOTICE '================================';
 END $$;
 

test-pig-build.sh

@@ -0,0 +1,44 @@
+#!/bin/bash
+
+# 测试 Pig 包管理器构建的脚本
+
+set -e
+
+echo "=== 测试 Pig 包管理器构建 ==="
+
+# 构建镜像
+echo "构建 Docker 镜像..."
+docker build -t postgres-zh-pig-test .
+
+# 启动容器
+echo "启动测试容器..."
+docker run -d \
+  --name postgres-zh-pig-test \
+  -e POSTGRES_PASSWORD=test_password \
+  -e POSTGRES_DB=test_db \
+  postgres-zh-pig-test
+
+# 等待 PostgreSQL 启动
+echo "等待 PostgreSQL 启动..."
+sleep 10
+
+# 测试 Pig 包管理器
+echo "测试 Pig 包管理器..."
+docker exec postgres-zh-pig-test pig list
+
+# 测试扩展安装
+echo "测试扩展状态..."
+docker exec postgres-zh-pig-test psql -U postgres -d test_db -c "\dx"
+
+# 测试中文分词功能
+echo "测试中文分词功能..."
+docker exec postgres-zh-pig-test psql -U postgres -d test_db -c "
+SELECT to_tsvector('chinese_zh', '这是一个中文分词的测试');
+"
+
+# 清理
+echo "清理测试容器..."
+docker stop postgres-zh-pig-test
+docker rm postgres-zh-pig-test
+
+echo "=== 测试完成 ==="