for (const [oldId, newId] of importResult.idMappings) {
  idMappings.set(oldId, newId);
}

// 在 controller.ts 的 runOfflineMigration 方法中,迁移完成后添加:
if (idMappings.size > 0) {
  addLog.info(`[Migration] Updating MongoDB ID mappings for ${idMappings.size} vectors...`);

  const MongoClient = require('mongodb').MongoClient;
  // 批量更新 dataset_datas 集合
  // 更新 indexes[].dataId 为新 ID
}

const data: InsertReq[] = records.map((record) => {
  const newId = this.idCounter++;
  return {
    // ... 使用自增 ID
  };
});

// 方案1:使用 Milvus 自动生成的 ID
const data: InsertReq[] = records.map((record, idx) => {
  return {
    data: [record.vector],
    meta: {
      teamId: record.teamId,
      datasetId: record.datasetId,
      collectionId: record.collectionId,
      oldId: record.id, // 保存旧 ID
      createTime: dayjs(record.createTime).unix()
    }
  };
});

// 获取 Milvus 返回的自动生成 ID,建立映射
const insertResp = await this.client.insert({ ... });
const insertIds = insertResp.IDs?.int_id?.data || [];

// 添加测试文件:
// packages/service/common/vectorDB/migration/__tests__/exporters.test.ts
// packages/service/common/vectorDB/migration/__tests__/importers.test.ts
// packages/service/common/vectorDB/migration/__tests__/controller.test.ts

describe('PgExporter', () => {
  it('should export vectors with pagination', async () => {
    // 测试分页导出
  });

  it('should handle empty result', async () => {
    // 测试空结果
  });
});

const whereConditions: string[] = [];
const params: (string | number)[] = [];
let paramIndex = 1;

if (afterId) {
  whereConditions.push(`id > $${paramIndex++}`);
  params.push(afterId);
}

if (afterId) {
  const numericId = parseInt(afterId, 10);
  if (isNaN(numericId)) {
    throw new Error(`Invalid afterId: ${afterId}. Must be a number.`);
  }
  whereConditions.push(`id > $${paramIndex++}`);
  params.push(numericId);
}

this.client = new MilvusClient({
  address: this.address,
  token: this.token
});
await this.client.connectPromise;

this.client = new MilvusClient({
  address: this.address,
  token: this.token,
  timeout: 30000, // 30秒超时
  connectTimeout: 10000 // 10秒连接超时
});

return {
  passed: countMatch,
  countMatch: {
    source: sourceCount,
    target: targetCount,
    match: countMatch
  },
  sampleMatch: {
    checked: 0,
    matched: 0,
    mismatchedIds: []
  }
};

// 随机抽取 100 条记录进行向量内容对比
const sampleSize = 100;
const sampleRecords = await exporter.exportBatch({ limit: sampleSize });
let matchedCount = 0;
const mismatchedIds: string[] = [];

for (const record of sampleRecords.records) {
  const targetVector = await importer.getVectorById(record.id);
  if (targetVector && vectorsEqual(record.vector, targetVector)) {
    matchedCount++;
  } else {
    mismatchedIds.push(record.id);
  }
}

return {
  passed: countMatch && mismatchedIds.length === 0,
  countMatch: { ... },
  sampleMatch: {
    checked: sampleSize,
    matched: matchedCount,
    mismatchedIds
  }
};

// 关键错误
addLog.error(`[Migration] Failed to insert batch: ${error.message}`);

// 警告
addLog.warn(`[Migration] Retry attempt ${retryCount}/${maxRetries}`);

// 调试信息
addLog.debug(`[Migration] Processing batch ${currentBatch}/${totalBatches}`);

// 一般信息
addLog.info(`[Migration] Progress: ${processedCount}/${totalCount}`);

// 对于 PostgreSQL,使用游标
const cursor = pg.query(new Cursor(`
  SELECT id, vector::text, team_id, ...
  FROM ${DatasetVectorTableName}
  ${whereClause}
  ORDER BY id ASC
`));

const batches = [];
while (true) {
  const batch = await cursor.read(batchSize);
  if (batch.length === 0) break;
  batches.push(batch);
}

// 抽取通用逻辑
abstract class SqlExporter implements VectorExporter {
  abstract getPlaceholder(index: number): string;

  buildQuery(filters: FilterConfig): { query: string; params: any[] } {
    const whereConditions: string[] = [];
    const params: any[] = [];
    let paramIndex = 1;

    if (filters.afterId) {
      whereConditions.push(`id > ${this.getPlaceholder(paramIndex++)}`);
      params.push(filters.afterId);
    }
    // ...
  }
}

class PgExporter extends SqlExporter {
  getPlaceholder(index: number) {
    return `$${index}`;
  }
}

class OceanBaseExporter extends SqlExporter {
  getPlaceholder(index: number) {
    return '?';
  }
}

export type VectorDbType = 'pg' | 'oceanbase' | 'milvus';
export type MigrationStatus = 'idle' | 'preparing' | 'full_sync' | 'incremental_sync' | 'completed' | 'failed' | 'cancelled';

export enum VectorDbType {
  Postgres = 'pg',
  OceanBase = 'oceanbase',
  Milvus = 'milvus'
}

export enum MigrationStatus {
  Idle = 'idle',
  Preparing = 'preparing',
  FullSync = 'full_sync',
  IncrementalSync = 'incremental_sync',
  Completed = 'completed',
  Failed = 'failed',
  Cancelled = 'cancelled'
}