ChatLab/src/utils/sqlExport.ts at dev · ChatLab/ChatLab

66 lines (56 loc) · 1.68 KB

import dayjs from 'dayjs'

export type SQLExportFormat = 'csv' | 'json'

export interface SQLExportData {
  columns: string[]
  rows: any[][]
}

/**
 * 将 SQL 结果格式化为 CSV
 */
export function formatAsCSV(data: SQLExportData): string {
  const header = data.columns.join(',')
  const rows = data.rows.map((row) =>
    row.map((cell) => (cell === null ? '' : `"${String(cell).replace(/"/g, '""')}"`)).join(',')
  )
  return [header, ...rows].join('\n')
}

/**
 * 将 SQL 结果格式化为 JSON（数组对象形式）
 */
export function formatAsJSON(data: SQLExportData): string {
  const jsonData = data.rows.map((row) => {
    const obj: Record<string, unknown> = {}
    data.columns.forEach((col, idx) => {
      obj[col] = row[idx]
    })
    return obj
  })
  return JSON.stringify(jsonData, null, 2)
}

/**
 * 导出 SQL 结果到文件
 */
export async function exportSQLResult(
  data: SQLExportData,
  format: SQLExportFormat
): Promise<{ success: boolean; filePath?: string; error?: string }> {
  if (data.rows.length === 0) {
    return { success: false, error: 'No data to export' }
  }

  const timestamp = dayjs().format('YYYYMMDD_HHmmss')
  const filename = `sql_result_${timestamp}.${format}`

  let content: string
  let mimeType: string

  if (format === 'json') {
    content = formatAsJSON(data)
    mimeType = 'application/json'
  } else {
    content = formatAsCSV(data)
    mimeType = 'text/csv'
  }

  // 转换为 data URL 并保存
  const dataUrl = `data:${mimeType};charset=utf-8,${encodeURIComponent(content)}`
  const { useCacheService } = await import('@/services/cache/service')
  const result = await useCacheService().saveToDownloads(filename, dataUrl)

  return result
}

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Expand file tree

Search code, repositories, users, issues, pull requests...

FilesExpand file tree

sqlExport.ts

Latest commit

History

sqlExport.ts

File metadata and controls

Expand file tree