pq: got 86575 parameters but PostgreSQL only supports 65535 parameters

import (
    "github.com/lib/pq"
    "gorm.io/gorm"
)

// BatchCreate 批量創(chuàng)建消息接收者記錄 - 使用 COPY 協(xié)議
func (r *receiverRepo) BatchCreate(ctx context.Context, db *gorm.DB, data []*define.WecomMsgReceiver) (rowsAffected int64, err error) {
    db = r.WithTrace(ctx, db)
    db = db.Table(r.TableName())

    if len(data) == 0 {
        return 0, nil
    }

    // 過濾掉 nil 的數(shù)據(jù)
    validData := make([]*define.WecomMsgReceiver, 0, len(data))
    for _, item := range data {
        if item != nil {
            validData = append(validData, item)
        }
    }
    if len(validData) == 0 {
        return 0, nil
    }

    // 獲取底層 sql.DB
    sqlDB := db.DB()
    
    // 開始事務
    tx, err := sqlDB.BeginTx(ctx, nil)
    if err != nil {
        return 0, fmt.Errorf("開始事務失?。?+v", err)
    }
    defer func() {
        if err != nil {
            tx.Rollback()
        }
    }()

    // 創(chuàng)建 COPY writer
    stmt, err := tx.Prepare(pq.CopyIn(r.TableName(), "send_log_id", "user_id", "status", "created_at", "updated_at"))
    if err != nil {
        return 0, fmt.Errorf("準備 COPY 語句失敗：%+v", err)
    }
    defer stmt.Close()

    // 批量寫入數(shù)據(jù)
    for _, item := range validData {
        _, err = stmt.Exec(item.SendLogID, item.UserID, item.Status, item.CreatedAt, item.UpdatedAt)
        if err != nil {
            return 0, fmt.Errorf("寫入數(shù)據(jù)失?。?+v", err)
        }
    }

    // 執(zhí)行 COPY
    _, err := stmt.Exec()
    if err != nil {
        return 0, fmt.Errorf("執(zhí)行 COPY 失?。?+v", err)
    }

    // 提交事務
    if err = tx.Commit(); err != nil {
        return 0, fmt.Errorf("提交事務失?。?+v", err)
    }
    
    rowsAffected = int64(len(validData))
    return rowsAffected, nil
}

// 設(shè)置測試數(shù)據(jù)庫
func setupTestDB() (*gorm.DB, error) {
    ctx := context.Background()
    postgres, err := infrastructure.DialPostgres(ctx, infrastructure.PostgresConfig{
        Host:     "host",
        Port:     5432,
        Username: "postgres",
        Password: "xxxxx",
        Database: "xxxxx",
    })
    if err != nil {
        return nil, err
    }

    return postgres, nil
}

func setupLogger() factory.LogFactory {
    logger, _ := factory.NewJsonFactory(factory.NewLevel("info"), factory.NewZapOption(factory.AddCallerSkip(0)))
    return logger
}

func TestReceiverRepo_BatchCreate(t *testing.T) {
    db, err := setupTestDB()
    require.NoError(t, err)
    defer db.Close()

    // 創(chuàng)建日志工廠
    logger := setupLogger()

    // 創(chuàng)建 repository 實例
    repo := NewReceiverRepository(db, logger)

    // 準備測試數(shù)據(jù) - 20000 條記錄，使用負的 send_log_id 避免污染數(shù)據(jù)
    testData := make([]*define.WecomMsgReceiver, 0, 20000)
    now := time.Now()
    negativeSendLogID := int64(-100000) // 使用負的 send_log_id

    for i := 0; i < 20000; i++ {
        testData = append(testData, &define.WecomMsgReceiver{
            SendLogID: negativeSendLogID,
            UserID:    "test_user_" + fmt.Sprint(i),
            Status:    1,
            CreatedAt: now,
            UpdatedAt: now,
        })
    }

    ctx := context.Background()

    // 執(zhí)行批量插入
    rowsAffected, err := repo.BatchCreate(ctx, db, testData)

    // 驗證結(jié)果
    assert.NoError(t, err)
    assert.Equal(t, int64(20000), rowsAffected)

    // 驗證數(shù)據(jù)是否正確插入
    var count int64
    query := "SELECT COUNT(*) FROM wecom_msg_receiver WHERE send_log_id < 0 AND send_log_id >= ?"
    err = db.Raw(query, negativeSendLogID).Count(&count).Error
    assert.NoError(t, err)
    assert.Equal(t, int64(20000), count)

    // 清理測試數(shù)據(jù)
    deleteQuery := "DELETE FROM wecom_msg_receiver WHERE send_log_id < 0 AND send_log_id >= ?"
    result := db.Exec(deleteQuery, negativeSendLogID)
    assert.NoError(t, result.Error)
    assert.Equal(t, int64(20000), result.RowsAffected)

    // 驗證清理是否成功
    err = db.Raw(query, negativeSendLogID).Count(&count).Error
    assert.NoError(t, err)
    assert.Equal(t, int64(0), count)
}