Asp.Net Core Web API客户端：管理员如何导出PostgreSQL为SQL并恢复数据

一、数据库备份实现 #

PostgreSQL官方提供的pg_dump工具是生成.sql备份文件的标准方式，我们可以在API中通过调用系统进程执行该命令：

1. 编写备份接口
   创建一个仅管理员可访问的接口，执行pg_dump并将输出流返回给客户端下载：

[Authorize(Roles = "Admin")]
[HttpGet("backup")]
public async Task<IActionResult> BackupDatabase()
{
    var connectionString = _configuration.GetConnectionString("DefaultConnection");
    var connBuilder = new NpgsqlConnectionStringBuilder(connectionString);
    
    // 构造pg_dump命令参数
    var args = $"--host={connBuilder.Host} --port={connBuilder.Port} --username={connBuilder.Username} --dbname={connBuilder.Database} --password={connBuilder.Password} --format=p";
    
    var processStartInfo = new ProcessStartInfo
    {
        FileName = "pg_dump",
        Arguments = args,
        RedirectStandardOutput = true,
        UseShellExecute = false,
        CreateNoWindow = true
    };
    
    using var process = Process.Start(processStartInfo);
    if (process == null) return BadRequest("无法启动备份进程");
    
    var memoryStream = new MemoryStream();
    await process.StandardOutput.BaseStream.CopyToAsync(memoryStream);
    memoryStream.Position = 0;
    
    await process.WaitForExitAsync();
    if (process.ExitCode != 0) return StatusCode(StatusCodes.Status500InternalServerError, "备份失败");
    
    return File(memoryStream, "application/sql", $"backup_{DateTime.Now:yyyyMMddHHmmss}.sql");
}

2. 关键注意事项

• 确保服务器已安装PostgreSQL客户端工具（含pg_dump），且工具路径已加入系统环境变量
• 避免明文传递密码，可改用pgpass文件或环境变量存储数据库密码
• 接口必须添加管理员角色验证，防止未授权访问

二、数据库恢复实现 #

使用PostgreSQL的psql工具执行.sql文件的恢复操作，API需接收客户端上传的备份文件并执行命令：

1. 编写恢复接口

[Authorize(Roles = "Admin")]
[HttpPost("restore")]
public async Task<IActionResult> RestoreDatabase(IFormFile backupFile)
{
    if (backupFile == null || backupFile.Length == 0)
        return BadRequest("请上传有效的备份文件");
    
    var connectionString = _configuration.GetConnectionString("DefaultConnection");
    var connBuilder = new NpgsqlConnectionStringBuilder(connectionString);
    
    // 构造psql命令参数
    var args = $"--host={connBuilder.Host} --port={connBuilder.Port} --username={connBuilder.Username} --dbname={connBuilder.Database} --password={connBuilder.Password} -f -";
    
    var processStartInfo = new ProcessStartInfo
    {
        FileName = "psql",
        Arguments = args,
        RedirectStandardInput = true,
        UseShellExecute = false,
        CreateNoWindow = true
    };
    
    using var process = Process.Start(processStartInfo);
    if (process == null) return BadRequest("无法启动恢复进程");
    
    await backupFile.CopyToAsync(process.StandardInput.BaseStream);
    process.StandardInput.Close(); // 关闭输入流触发命令执行
    
    await process.WaitForExitAsync();
    if (process.ExitCode != 0) return StatusCode(StatusCodes.Status500InternalServerError, "恢复失败");
    
    return Ok("数据恢复成功");
}

2. 关键注意事项

• 恢复操作会覆盖现有数据，建议客户端添加二次确认弹窗，避免误操作
• 确保数据库用户拥有CREATEDB、OWNER等足够权限执行恢复
• 大文件恢复时需调整API请求超时时间，可在Program.cs中配置：builder.Services.Configure<KestrelServerOptions>(options => options.Limits.RequestHeadersTimeout = TimeSpan.FromMinutes(10));

三、客户端交互实现 #

客户端需提供两个核心操作入口：

• 备份按钮：点击后调用备份接口，触发浏览器下载.sql文件
• 恢复按钮：弹出文件选择框，选择本地备份文件后上传到恢复接口

以Vue客户端为例，核心代码：

<template>
  <div v-if="isAdmin">
    <button @click="handleBackup">备份数据库</button>
    <input type="file" accept=".sql" @change="handleRestore" style="margin-left:10px;">
  </div>
</template>

<script>
export default {
  data() {
    return { isAdmin: false };
  },
  async mounted() {
    // 验证当前用户是否为管理员
    this.isAdmin = await this.checkAdminRole();
  },
  methods: {
    async handleBackup() {
      const response = await fetch('/api/database/backup', {
        headers: { 'Authorization': `Bearer ${localStorage.getItem('token')}` }
      });
      const blob = await response.blob();
      const url = window.URL.createObjectURL(blob);
      const a = document.createElement('a');
      a.href = url;
      a.download = `backup_${new Date().toISOString().slice(0,19).replace(/[:T]/g,'')}.sql`;
      a.click();
      window.URL.revokeObjectURL(url);
    },
    async handleRestore(e) {
      const file = e.target.files[0];
      if (!file) return;
      
      const formData = new FormData();
      formData.append('backupFile', file);
      
      const response = await fetch('/api/database/restore', {
        method: 'POST',
        headers: { 'Authorization': `Bearer ${localStorage.getItem('token')}` },
        body: formData
      });
      
      if (response.ok) {
        alert('恢复成功');
      } else {
        const error = await response.text();
        alert(`恢复失败: ${error}`);
      }
    }
  }
}
</script>

四、安全与优化建议 #

• 权限加固：除角色验证外，可添加IP白名单限制，仅允许指定管理员IP访问备份/恢复接口
• 操作日志：记录每次备份/恢复的时间、操作者、结果，便于后续审计
• 备份存储：可将备份文件同时存储到服务器本地或云存储，避免客户端丢失备份
• 增量备份：若数据量较大，可使用pg_dump的增量备份参数（如--data-only、--table），减少备份文件体积

内容的提问来源于stack exchange，提问作者Semir Hamid