数据库相关操作

自定义bat程序，批量执行sql语句

提示：该文章针对postgres数据库

---

---

前言

工作中遇到的需求，小小记录一下。说明：对于一个完整的pg数据库备份后的恢复，需要保留某些表的数据，清空某些表的数据，并且需要将清空数据库的自增序列，初始化，为避免每次恢复数据库后一个表一个表的手动操作，设计一个可以一次操作完成所有操作的程序或者工具，因为在windows环境下，所以想到用批处理程序来实现。

---

一、自定义data.sql文件

示例：

\encoding UTF8;
\cd D:/java/pg/bin
\set ON_ERROR_STOP 1
\set ECHO all
\timing on
\i D:/test3.sql;
\i D:/test4.sql;

说明：
\encoding 设置客户端字符集编码(可选)

\cd 切换目录。切换到本机pgsql安装的bin目录

\set ON_ERROR_STOP 设置变量ON_ERROR_STOP(可选)，当脚本执行出现错误时自动终止。不设则默认继续执行

\set ECHO all 设置变量ECHO(可选)，用于将每一步的执行脚本都显示到标准输出

\timing on 打开计时功能(可选)，计算脚本每一步执行的时间

\i 执行脚本文件。需要sql完整的路径，一个sql文件写一条要执行的sql语句
D:/test4.sql 表示一条sql语句，每条sql语句单独用一个sql文件表示

二、自定义bat文件

示例：

@echo off
D:
cd /java/pg/bin
psql -h localhost -U postgres -d database -p 5432 -a -f "D:\data.sql"
pause

说明：

database: 要操作的数据库名称
D:\data.sql：第一部中自定义的sql文件路径
pause：执行命令后不自动关闭，方便查看sql运行结果

三、运行bat脚本

需要输入postgres的口令
运行结果如下：

---

总结

以上内容即对批量操作pg数据库sql语句的实现。另外，对于windows系统来说，很多需要用到cmd执行命令的操作，都可以使用bat来实现。